2011-03-04 23:11:26 -07:00
|
|
|
|
|
|
|
#ifndef MUTEX_H_
|
|
|
|
#define MUTEX_H_
|
|
|
|
|
|
|
|
#define GCC_VER(x,y,z) ((x) * 10000 + (y) * 100 + (z))
|
|
|
|
#define GCC_VERSION GCC_VER(__GNUC__, __GNUC_MINOR__, __GNUC_PATCHLEVEL__)
|
|
|
|
|
|
|
|
#if GCC_VERSION >= GCC_VER(4,4,0) && __GXX_EXPERIMENTAL_CXX0X__
|
|
|
|
// GCC 4.4 provides <mutex>
|
|
|
|
#include <mutex>
|
|
|
|
#else
|
|
|
|
|
|
|
|
// partial <mutex> implementation for win32/pthread
|
|
|
|
|
|
|
|
#include <algorithm>
|
|
|
|
|
|
|
|
#if defined(_WIN32)
|
|
|
|
// WIN32
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
#include <Windows.h>
|
|
|
|
|
|
|
|
#else
|
|
|
|
// POSIX
|
|
|
|
#include <pthread.h>
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#if (_MSC_VER >= 1600) || (GCC_VERSION >= GCC_VER(4,3,0) && __GXX_EXPERIMENTAL_CXX0X__)
|
|
|
|
#define USE_RVALUE_REFERENCES
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#if defined(_WIN32) && defined(_M_X64)
|
|
|
|
#define USE_SRWLOCKS
|
|
|
|
#endif
|
|
|
|
|
|
|
|
namespace std
|
|
|
|
{
|
|
|
|
|
|
|
|
class recursive_mutex
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
typedef CRITICAL_SECTION native_type;
|
|
|
|
#else
|
|
|
|
typedef pthread_mutex_t native_type;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
public:
|
|
|
|
typedef native_type* native_handle_type;
|
|
|
|
|
|
|
|
recursive_mutex(const recursive_mutex&) /*= delete*/;
|
|
|
|
recursive_mutex& operator=(const recursive_mutex&) /*= delete*/;
|
|
|
|
|
|
|
|
recursive_mutex()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
InitializeCriticalSection(&m_handle);
|
|
|
|
#else
|
|
|
|
pthread_mutexattr_t attr;
|
|
|
|
pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
|
|
|
|
pthread_mutex_init(&m_handle, &attr);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
~recursive_mutex()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
DeleteCriticalSection(&m_handle);
|
|
|
|
#else
|
|
|
|
pthread_mutex_destroy(&m_handle);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
void lock()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
EnterCriticalSection(&m_handle);
|
|
|
|
#else
|
|
|
|
pthread_mutex_lock(&m_handle);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
void unlock()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
LeaveCriticalSection(&m_handle);
|
|
|
|
#else
|
|
|
|
pthread_mutex_unlock(&m_handle);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
bool try_lock()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
return (0 != TryEnterCriticalSection(&m_handle));
|
|
|
|
#else
|
|
|
|
return !pthread_mutex_trylock(&m_handle);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
native_handle_type native_handle()
|
|
|
|
{
|
|
|
|
return &m_handle;
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
native_type m_handle;
|
|
|
|
};
|
|
|
|
|
2011-03-05 17:52:04 -07:00
|
|
|
#if !defined(_WIN32) || defined(USE_SRWLOCKS)
|
2011-03-04 23:11:26 -07:00
|
|
|
|
|
|
|
class mutex
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
typedef SRWLOCK native_type;
|
|
|
|
#else
|
|
|
|
typedef pthread_mutex_t native_type;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
public:
|
|
|
|
typedef native_type* native_handle_type;
|
|
|
|
|
|
|
|
mutex(const mutex&) /*= delete*/;
|
|
|
|
mutex& operator=(const mutex&) /*= delete*/;
|
|
|
|
|
|
|
|
mutex()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
InitializeSRWLock(&m_handle);
|
|
|
|
#else
|
|
|
|
pthread_mutex_init(&m_handle, NULL);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
~mutex()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
#else
|
|
|
|
pthread_mutex_destroy(&m_handle);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
void lock()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
AcquireSRWLockExclusive(&m_handle);
|
|
|
|
#else
|
|
|
|
pthread_mutex_lock(&m_handle);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
void unlock()
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
ReleaseSRWLockExclusive(&m_handle);
|
|
|
|
#else
|
|
|
|
pthread_mutex_unlock(&m_handle);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2011-03-06 17:16:38 -07:00
|
|
|
// TryAcquireSRWLockExclusive requires Windows 7!!
|
|
|
|
// bool try_lock()
|
|
|
|
// {
|
|
|
|
//#ifdef _WIN32
|
|
|
|
// return (0 != TryAcquireSRWLockExclusive(&m_handle));
|
|
|
|
//#else
|
|
|
|
// return !pthread_mutex_trylock(&m_handle);
|
|
|
|
//#endif
|
|
|
|
// }
|
2011-03-04 23:11:26 -07:00
|
|
|
|
|
|
|
native_handle_type native_handle()
|
|
|
|
{
|
|
|
|
return &m_handle;
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
native_type m_handle;
|
|
|
|
};
|
|
|
|
|
|
|
|
#else
|
|
|
|
typedef recursive_mutex mutex; // just use CriticalSections
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
enum defer_lock_t { defer_lock };
|
|
|
|
enum try_to_lock_t { try_to_lock };
|
|
|
|
enum adopt_lock_t { adopt_lock };
|
|
|
|
|
|
|
|
template <class Mutex>
|
|
|
|
class lock_guard
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
typedef Mutex mutex_type;
|
|
|
|
|
|
|
|
explicit lock_guard(mutex_type& m)
|
|
|
|
: pm(m)
|
|
|
|
{
|
|
|
|
m.lock();
|
|
|
|
}
|
|
|
|
|
|
|
|
lock_guard(mutex_type& m, adopt_lock_t)
|
|
|
|
: pm(m)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
~lock_guard()
|
|
|
|
{
|
|
|
|
pm.unlock();
|
|
|
|
}
|
|
|
|
|
|
|
|
lock_guard(lock_guard const&) /*= delete*/;
|
|
|
|
lock_guard& operator=(lock_guard const&) /*= delete*/;
|
|
|
|
|
|
|
|
private:
|
|
|
|
mutex_type& pm;
|
|
|
|
};
|
|
|
|
|
|
|
|
template <class Mutex>
|
|
|
|
class unique_lock
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
typedef Mutex mutex_type;
|
|
|
|
|
|
|
|
unique_lock()
|
|
|
|
: pm(NULL), owns(false)
|
|
|
|
{}
|
|
|
|
|
|
|
|
/*explicit*/ unique_lock(mutex_type& m)
|
|
|
|
: pm(&m), owns(true)
|
|
|
|
{
|
|
|
|
m.lock();
|
|
|
|
}
|
|
|
|
|
|
|
|
unique_lock(mutex_type& m, defer_lock_t)
|
|
|
|
: pm(&m), owns(false)
|
|
|
|
{}
|
|
|
|
|
|
|
|
unique_lock(mutex_type& m, try_to_lock_t)
|
|
|
|
: pm(&m), owns(m.try_lock())
|
|
|
|
{}
|
|
|
|
|
|
|
|
unique_lock(mutex_type& m, adopt_lock_t)
|
|
|
|
: pm(&m), owns(true)
|
|
|
|
{}
|
|
|
|
|
|
|
|
//template <class Clock, class Duration>
|
|
|
|
//unique_lock(mutex_type& m, const chrono::time_point<Clock, Duration>& abs_time);
|
|
|
|
|
|
|
|
//template <class Rep, class Period>
|
|
|
|
//unique_lock(mutex_type& m, const chrono::duration<Rep, Period>& rel_time);
|
|
|
|
|
|
|
|
~unique_lock()
|
|
|
|
{
|
|
|
|
if (owns_lock())
|
|
|
|
mutex()->unlock();
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef USE_RVALUE_REFERENCES
|
|
|
|
unique_lock& operator=(const unique_lock&) /*= delete*/;
|
|
|
|
|
|
|
|
unique_lock& operator=(unique_lock&& other)
|
|
|
|
{
|
|
|
|
#else
|
|
|
|
unique_lock& operator=(const unique_lock& u)
|
|
|
|
{
|
|
|
|
// ugly const_cast to get around lack of rvalue references
|
|
|
|
unique_lock& other = const_cast<unique_lock&>(u);
|
|
|
|
#endif
|
|
|
|
swap(other);
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef USE_RVALUE_REFERENCES
|
|
|
|
unique_lock(const unique_lock&) /*= delete*/;
|
|
|
|
|
|
|
|
unique_lock(unique_lock&& other)
|
|
|
|
: pm(NULL), owns(false)
|
|
|
|
{
|
|
|
|
#else
|
|
|
|
unique_lock(const unique_lock& u)
|
|
|
|
: pm(NULL), owns(false)
|
|
|
|
{
|
|
|
|
// ugly const_cast to get around lack of rvalue references
|
|
|
|
unique_lock& other = const_cast<unique_lock&>(u);
|
|
|
|
#endif
|
|
|
|
swap(other);
|
|
|
|
}
|
|
|
|
|
|
|
|
void lock()
|
|
|
|
{
|
|
|
|
mutex()->lock();
|
|
|
|
owns = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool try_lock()
|
|
|
|
{
|
|
|
|
owns = mutex()->try_lock();
|
|
|
|
return owns;
|
|
|
|
}
|
|
|
|
|
|
|
|
//template <class Rep, class Period>
|
|
|
|
//bool try_lock_for(const chrono::duration<Rep, Period>& rel_time);
|
|
|
|
//template <class Clock, class Duration>
|
|
|
|
//bool try_lock_until(const chrono::time_point<Clock, Duration>& abs_time);
|
|
|
|
|
|
|
|
void unlock()
|
|
|
|
{
|
|
|
|
mutex()->unlock();
|
|
|
|
owns = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
void swap(unique_lock& u)
|
|
|
|
{
|
|
|
|
std::swap(pm, u.pm);
|
|
|
|
std::swap(owns, u.owns);
|
|
|
|
}
|
|
|
|
|
|
|
|
mutex_type* release()
|
|
|
|
{
|
|
|
|
return mutex();
|
|
|
|
pm = NULL;
|
|
|
|
owns = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool owns_lock() const
|
|
|
|
{
|
|
|
|
return owns;
|
|
|
|
}
|
|
|
|
|
|
|
|
//explicit operator bool () const
|
|
|
|
//{
|
|
|
|
// return owns_lock();
|
|
|
|
//}
|
|
|
|
|
|
|
|
mutex_type* mutex() const
|
|
|
|
{
|
|
|
|
return pm;
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
mutex_type* pm;
|
|
|
|
bool owns;
|
|
|
|
};
|
|
|
|
|
|
|
|
template <class Mutex>
|
|
|
|
void swap(unique_lock<Mutex>& x, unique_lock<Mutex>& y)
|
|
|
|
{
|
|
|
|
x.swap(y);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
#endif
|