X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=common%2Fos_thread.hpp;h=5e9577b56a29924b816abfc10fb16baf70a9bb14;hb=b35ae0df401ff22e3b2dc9b6bfacf1e23baebde0;hp=9d6a9891524564803d844bfe923de13dfd8442c0;hpb=23354536051650b7a8a713e824946d48a7734534;p=apitrace

diff --git a/common/os_thread.hpp b/common/os_thread.hpp
index 9d6a989..5e9577b 100644
--- a/common/os_thread.hpp
+++ b/common/os_thread.hpp
@@ -24,9 +24,9 @@
  **************************************************************************/
 
 /*
- * Simple OS abstraction.
+ * OS native thread abstraction.
  *
- * Mimics C++11 / boost threads.
+ * Mimics C++11 threads.
  */
 
 #ifndef _OS_THREAD_HPP_
@@ -39,6 +39,27 @@
 #include <pthread.h>
 #endif
 
+
+#define USE_WIN32_CONDITION_VARIABLES 0
+
+
+/**
+ * Compiler TLS.
+ *
+ * See also:
+ * - http://gcc.gnu.org/onlinedocs/gcc-4.6.3/gcc/Thread_002dLocal.html
+ * - http://msdn.microsoft.com/en-us/library/9w1sdazb.aspx
+ */
+#if defined(_MSC_VER)
+#  define thread_specific __declspec(thread)
+#elif defined(__GNUC__)
+#  define thread_specific __thread
+#else
+#  define thread_specific
+#  error "Unsupported compiler"
+#endif
+
+
 namespace os {
 
 
@@ -185,16 +206,31 @@ namespace os {
      */
     class condition_variable
     {
-    public:
+    private:
 #ifdef _WIN32
-        /* FIXME */
+#  if USE_WIN32_CONDITION_VARIABLES
+        // XXX: Only supported on Vista an higher. Not yet supported by WINE.
+        typedef CONDITION_VARIABLE native_handle_type;
+        native_handle_type _native_handle;
+#else
+        // http://www.cs.wustl.edu/~schmidt/win32-cv-1.html
+        LONG cWaiters;
+        HANDLE hEvent;
+#endif
 #else
         typedef pthread_cond_t native_handle_type;
+        native_handle_type _native_handle;
 #endif
 
+    public:
         condition_variable() {
 #ifdef _WIN32
-            /* FIXME */
+#  if USE_WIN32_CONDITION_VARIABLES
+            InitializeConditionVariable(&_native_handle);
+#  else
+            cWaiters = 0;
+            hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
+#  endif
 #else
             pthread_cond_init(&_native_handle, NULL);
 #endif
@@ -202,7 +238,11 @@ namespace os {
 
         ~condition_variable() {
 #ifdef _WIN32
-            /* FIXME */
+#  if USE_WIN32_CONDITION_VARIABLES
+            /* No-op */
+#  else
+            CloseHandle(hEvent);
+#  endif
 #else
             pthread_cond_destroy(&_native_handle);
 #endif
@@ -211,7 +251,13 @@ namespace os {
         inline void
         signal(void) {
 #ifdef _WIN32
-            /* FIXME */
+#  if USE_WIN32_CONDITION_VARIABLES
+            WakeConditionVariable(&_native_handle);
+#  else
+            if (cWaiters) {
+                SetEvent(hEvent);
+            }
+#  endif
 #else
             pthread_cond_signal(&_native_handle);
 #endif
@@ -219,110 +265,66 @@ namespace os {
 
         inline void
         wait(unique_lock<mutex> & lock) {
+            mutex::native_handle_type & mutex_native_handle = lock.mutex()->native_handle();
 #ifdef _WIN32
-            /* FIXME */
+#  if USE_WIN32_CONDITION_VARIABLES
+            SleepConditionVariableCS(&_native_handle, &mutex_native_handle, INFINITE);
+#  else
+            InterlockedIncrement(&cWaiters);
+            LeaveCriticalSection(&mutex_native_handle);
+            WaitForSingleObject(hEvent, INFINITE);
+            EnterCriticalSection(&mutex_native_handle);
+            InterlockedDecrement(&cWaiters);
+#  endif
 #else
-            pthread_cond_wait(&_native_handle, &lock.mutex()->native_handle());
+            pthread_cond_wait(&_native_handle, &mutex_native_handle);
 #endif
         }
-
-    protected:
-        native_handle_type _native_handle;
     };
 
 
     /**
-     * Same interface as boost::thread_specific_ptr.
+     * Same interface as std::thread
      */
-    template <typename T>
-    class thread_specific_ptr
-    {
-    private:
-#ifdef _WIN32
-        DWORD dwTlsIndex;
-#else
-        pthread_key_t key;
-
-        static void destructor(void *ptr) {
-            delete static_cast<T *>(ptr);
-        }
-#endif
-
+    class thread {
     public:
-        thread_specific_ptr(void) {
-#ifdef _WIN32
-            dwTlsIndex = TlsAlloc();
-#else
-            pthread_key_create(&key, &destructor);
-#endif
-        }
-
-        ~thread_specific_ptr() {
 #ifdef _WIN32
-            TlsFree(dwTlsIndex);
+        typedef HANDLE native_handle_type;
 #else
-            pthread_key_delete(key);
-#endif
-        }
-
-        T* get(void) const {
-            void *ptr;
-#ifdef _WIN32
-            ptr = TlsGetValue(dwTlsIndex);
-#else
-            ptr = pthread_getspecific(key);
+        typedef pthread_t native_handle_type;
 #endif
-            return static_cast<T*>(ptr);
-        }
 
-        T* operator -> (void) const
+        inline
+        thread() :
+            _native_handle(0)
         {
-            return get();
         }
 
-        T& operator * (void) const
+        inline
+        thread(thread &other) :
+            _native_handle(other._native_handle)
         {
-            return *get();
         }
 
-        void reset(T* new_value=0) {
-            T * old_value = get();
+        template< class Function, class Arg >
+        explicit thread( Function& f, Arg arg ) {
 #ifdef _WIN32
-            TlsSetValue(dwTlsIndex, new_value);
+            DWORD id = 0;
+            _native_handle = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)f, (LPVOID)arg, 0, &id);
 #else
-            pthread_setspecific(key, new_value);
+            pthread_create(&_native_handle, NULL, ( void *(*) (void *))f, arg);
 #endif
-            if (old_value) {
-                delete old_value;
-            }
         }
-    };
-
-
-    /**
-     * Same interface as std::thread
-     */
-    class thread {
-    public:
-#ifdef _WIN32
-        /* FIXME */
-#else
-        typedef pthread_t native_handle_type;
-#endif
 
-        template< class Function, class Arg >
-        explicit thread( Function& f, Arg & arg ) {
-#ifdef _WIN32
-            /* FIXME */
-#else
-            pthread_create(&_native_handle, NULL, f, arg);
-#endif
+        inline bool
+        joinable(void) const {
+            return _native_handle != 0;
         }
 
         inline void
         join() {
 #ifdef _WIN32
-            /* FIXME */
+            WaitForSingleObject(_native_handle, INFINITE);
 #else
             pthread_join(_native_handle, NULL);
 #endif
@@ -330,6 +332,16 @@ namespace os {
 
     private:
         native_handle_type _native_handle;
+
+#if 0
+#ifdef _WIN32
+        template< class Function, class Arg >
+        static DWORD WINAPI
+        ThreadProc(LPVOID lpParameter) {
+
+        );
+#endif
+#endif
     };
 
 } /* namespace os */