src/jthread/jmutex.h

   1 /*
   2
   3     This file is a part of the JThread package, which contains some object-
   4     oriented thread wrappers for different thread implementations.
   5
   6     Copyright (c) 2000-2006  Jori Liesenborgs (jori.liesenborgs@gmail.com)
   7
   8     Permission is hereby granted, free of charge, to any person obtaining a
   9     copy of this software and associated documentation files (the "Software"),
  10     to deal in the Software without restriction, including without limitation
  11     the rights to use, copy, modify, merge, publish, distribute, sublicense,
  12     and/or sell copies of the Software, and to permit persons to whom the
  13     Software is furnished to do so, subject to the following conditions:
  14
  15     The above copyright notice and this permission notice shall be included in
  16     all copies or substantial portions of the Software.
  17
  18     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19     IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20     FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  21     THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22     LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  23     FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  24     DEALINGS IN THE SOFTWARE.
  25
  26 */
  27
  28 #ifndef JMUTEX_H
  29
  30 #define JMUTEX_H
  31
  32 #if (defined(WIN32) || defined(_WIN32_WCE))
  33         #ifndef _WIN32_WINNT
  34                 #define _WIN32_WINNT 0x0501
  35         #endif
  36         #ifndef _WIN32_WCE
  37                 #include <process.h>
  38         #endif // _WIN32_WCE
  39         #include <winsock2.h>
  40         #include <windows.h>
  41         // CriticalSection is way faster than the alternative
  42         #define JMUTEX_CRITICALSECTION
  43 #else // using pthread
  44         #include <pthread.h>
  45 #endif // WIN32
  46
  47 #define ERR_JMUTEX_ALREADYINIT                                          -1
  48 #define ERR_JMUTEX_NOTINIT                                              -2
  49 #define ERR_JMUTEX_CANTCREATEMUTEX                                      -3
  50
  51 class JMutex
  52 {
  53 public:
  54         JMutex();
  55         ~JMutex();
  56         int Init();
  57         int Lock();
  58         int Unlock();
  59         bool IsInitialized()                                            { return initialized; }
  60
  61 private:
  62 #if (defined(WIN32) || defined(_WIN32_WCE))
  63 #ifdef JMUTEX_CRITICALSECTION
  64         CRITICAL_SECTION mutex;
  65 #else // Use standard mutex
  66         HANDLE mutex;
  67 #endif // JMUTEX_CRITICALSECTION
  68 #else // pthread mutex
  69         pthread_mutex_t mutex;
  70
  71         bool IsLocked() {
  72                 if (pthread_mutex_trylock(&mutex)) {
  73                         pthread_mutex_unlock(&mutex);
  74                         return true;
  75                 }
  76                 return false;
  77         }
  78 #endif // WIN32
  79         bool initialized;
  80 };
  81
  82 #ifdef _WIN32
  83
  84 class Event {
  85         HANDLE hEvent;
  86
  87 public:
  88         Event() {
  89                 hEvent = CreateEvent(NULL, 0, 0, NULL);
  90         }
  91
  92         ~Event() {
  93                 CloseHandle(hEvent);
  94         }
  95
  96         void wait() {
  97                 WaitForSingleObject(hEvent, INFINITE);
  98         }
  99
 100         void signal() {
 101                 SetEvent(hEvent);
 102         }
 103 };
 104
 105 #else
 106
 107 #include <semaphore.h>
 108
 109 class Event {
 110         sem_t sem;
 111
 112 public:
 113         Event() {
 114                 sem_init(&sem, 0, 0);
 115         }
 116
 117         ~Event() {
 118                 sem_destroy(&sem);
 119         }
 120
 121         void wait() {
 122                 sem_wait(&sem);
 123         }
 124
 125         void signal() {
 126                 sem_post(&sem);
 127         }
 128 };
 129
 130 #endif
 131
 132 #endif // JMUTEX_H