Changeset 1101Tweet

Timestamp:

Nov 30, 2011, 7:31:34 PM (11 years ago)

Author:

gary

Message:

core: implement Queue on Win32 and on the PS3.

Location:

trunk/src

Files:

• platform/ps3/threadbase.h (modified) (1 diff)
• thread/threadbase.h (modified) (4 diffs)

trunk/src/platform/ps3/threadbase.h

@@ -53 +53 @@
 };
 
-class ConditionBase
+class QueueBase
 {
 public:
-    ConditionBase()
+    QueueBase()
     {
+        m_start = m_count = 0;
+        m_poppers = m_pushers = 0;
+
+        sys_lwmutex_attribute_t mattr;
+        sys_lwmutex_attribute_initialize(mattr);
+        sys_lwmutex_create(&m_mutex, &mattr);
+
+        sys_lwcond_attribute_t cattr;
+        sys_lwcond_attribute_initialize(cattr);
+        sys_lwcond_create(&m_empty_cond, &m_mutex, &cattr);
+        sys_lwcond_create(&m_full_cond, &m_mutex, &cattr);
     }
 
-    ~ConditionBase()
+    ~QueueBase()
     {
+        while (sys_lwcond_destroy(&m_empty_cond) == EBUSY)
+            ;
+        while (sys_lwcond_destroy(&m_full_cond) == EBUSY)
+            ;
+        while (sys_lwmutex_destroy(&m_mutex) == EBUSY)
+            ;
     }
 
-    void Acquire()
+    void Push(int value)
     {
+        /* FIXME: this is a copy of the pthread implementation, but we
+         * should really use libsync2 instead. */
+        sys_lwmutex_lock(&m_mutex, 0);
+        m_pushers++;
+        while (m_count == CAPACITY)
+            sys_lwcond_wait(&m_full_cond, 0);
+        m_pushers--;
+        m_values[(m_start + m_count) % CAPACITY] = value;
+        m_count++;
+        if (m_poppers)
+            sys_lwcond_signal(&m_empty_cond);
+        sys_lwmutex_unlock(&m_mutex);
     }
 
-    void Release()
+    int Pop()
     {
-    }
-
-    void Wait()
-    {
-    }
-
-    void Notify()
-    {
+        sys_lwmutex_lock(&m_mutex, 0);
+        m_poppers++;
+        while (m_count == 0)
+            sys_lwcond_wait(&m_empty_cond, 0);
+        m_poppers--;
+        int ret = m_values[m_start];
+        m_start = (m_start + 1) % CAPACITY;
+        m_count--;
+        if (m_pushers)
+            sys_lwcond_signal(&m_full_cond);
+        sys_lwmutex_unlock(&m_mutex);
+        return ret;
     }
 
 private:
+    static size_t const CAPACITY = 100;
+    int m_values[CAPACITY];
+    size_t m_start, m_count;
+    size_t m_poppers, m_pushers;
+    sys_lwmutex_t m_mutex;
+    sys_lwcond_t m_empty_cond, m_full_cond;
 };
 

trunk/src/thread/threadbase.h

@@ -81 +81 @@
     QueueBase()
     {
-#if defined __linux__ || defined __native_client__
-        memset(m_values, 0, sizeof(m_values));
         m_start = m_count = 0;
+#if defined __linux__ || defined __native_client__
         m_poppers = m_pushers = 0;
         pthread_mutex_init(&m_mutex, NULL);
         pthread_cond_init(&m_empty_cond, NULL);
         pthread_cond_init(&m_full_cond, NULL);
+#elif defined _WIN32
+        m_empty_sem = CreateSemaphore(NULL, CAPACITY, CAPACITY, NULL);
+        m_full_sem = CreateSemaphore(NULL, 0, CAPACITY, NULL);
+        InitializeCriticalSection(&m_mutex);
 #endif
     }
@@ -97 +100 @@
         pthread_cond_destroy(&m_full_cond);
         pthread_mutex_destroy(&m_mutex);
+#elif defined _WIN32
+        CloseHandle(m_empty_sem);
+        CloseHandle(m_full_sem);
+        DeleteCriticalSection(&m_mutex);
 #endif
     }
@@ -106 +113 @@
         /* If queue is full, wait on the "full" cond var. */
         m_pushers++;
-        while (m_count == 100)
+        while (m_count == CAPACITY)
             pthread_cond_wait(&m_full_cond, &m_mutex);
         m_pushers--;
+#elif defined _WIN32
+        WaitForSingleObject(m_empty_sem, INFINITE);
+        EnterCriticalSection(&m_mutex);
+#endif
+
         /* Push value */
-        m_values[(m_start + m_count) % 100] = value;
+        m_values[(m_start + m_count) % CAPACITY] = value;
         m_count++;
+
+#if defined __linux__ || defined __native_client__
         /* If there were poppers waiting, signal the "empty" cond var. */
         if (m_poppers)
             pthread_cond_signal(&m_empty_cond);
         pthread_mutex_unlock(&m_mutex);
+#elif defined _WIN32
+        LeaveCriticalSection(&m_mutex);
+        ReleaseSemaphore(m_full_sem, 1, NULL);
 #endif
     }
@@ -130 +147 @@
             pthread_cond_wait(&m_empty_cond, &m_mutex);
         m_poppers--;
+#elif defined _WIN32
+        WaitForSingleObject(m_full_sem, INFINITE);
+        EnterCriticalSection(&m_mutex);
+#endif
+
         /* Pop value */
         int ret = m_values[m_start];
-        m_start = (m_start + 1) % 100;
+        m_start = (m_start + 1) % CAPACITY;
         m_count--;
+
+#if defined __linux__ || defined __native_client__
         /* If there were pushers waiting, signal the "full" cond var. */
         if (m_pushers)
             pthread_cond_signal(&m_full_cond);
         pthread_mutex_unlock(&m_mutex);
+#else
+        LeaveCriticalSection(&m_mutex);
+        ReleaseSemaphore(m_empty_sem, 1, NULL);
+#endif
+
         return ret;
-#endif
     }
 
 private:
-#if defined __linux__ || defined __native_client__
-    int m_values[100];
-    size_t m_poppers, m_pushers, m_start, m_count;
+    static size_t const CAPACITY = 100;
+    int m_values[CAPACITY];
+    size_t m_start, m_count;
+#if defined __linux__ || defined __native_client__
+    size_t m_poppers, m_pushers;
     pthread_mutex_t m_mutex;
     pthread_cond_t m_empty_cond, m_full_cond;
+#elif defined _WIN32
+    HANDLE m_empty_sem, m_full_sem;
+    CRITICAL_SECTION m_mutex;
 #endif
 };