| 
									
										
										
										
											2013-03-29 07:53:45 -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__)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #ifndef __has_include
 | 
					
						
							|  |  |  | #define __has_include(s) 0
 | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-08-29 22:29:33 -05:00
										 |  |  | #if GCC_VERSION >= GCC_VER(4,4,0) && __GXX_EXPERIMENTAL_CXX0X__
 | 
					
						
							| 
									
										
										
										
											2013-03-29 07:53:45 -07:00
										 |  |  | // GCC 4.4 provides <mutex>
 | 
					
						
							|  |  |  | #include <mutex>
 | 
					
						
							| 
									
										
										
										
											2013-04-18 22:52:05 -05:00
										 |  |  | #elif __has_include(<mutex>) && !ANDROID
 | 
					
						
							| 
									
										
										
										
											2013-03-29 07:53:45 -07:00
										 |  |  | // Clang + libc++
 | 
					
						
							|  |  |  | #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_init(&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; | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #if !defined(_WIN32) || defined(USE_SRWLOCKS)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 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
 | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	bool try_lock() | 
					
						
							|  |  |  | 	{ | 
					
						
							|  |  |  | #ifdef _WIN32
 | 
					
						
							|  |  |  | 		// XXX TryAcquireSRWLockExclusive requires Windows 7!
 | 
					
						
							|  |  |  | 		// return (0 != TryAcquireSRWLockExclusive(&m_handle));
 | 
					
						
							|  |  |  | 		return false; | 
					
						
							|  |  |  | #else
 | 
					
						
							|  |  |  | 		return !pthread_mutex_trylock(&m_handle); | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	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() | 
					
						
							|  |  |  | 	{ | 
					
						
							|  |  |  | 		auto const ret = mutex(); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		pm = NULL; | 
					
						
							|  |  |  | 		owns = false; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		return ret; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	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
 |