godot/thirdparty/etcpak/Semaphore.hpp

47 lines
830 B
C++
Raw Normal View History

#ifndef __DARKRL__SEMAPHORE_HPP__
#define __DARKRL__SEMAPHORE_HPP__
#include <condition_variable>
#include <mutex>
class Semaphore
{
public:
Semaphore( int count ) : m_count( count ) {}
void lock()
{
std::unique_lock<std::mutex> lock( m_mutex );
m_cv.wait( lock, [this](){ return m_count != 0; } );
m_count--;
}
void unlock()
{
std::lock_guard<std::mutex> lock( m_mutex );
m_count++;
m_cv.notify_one();
}
bool try_lock()
{
std::lock_guard<std::mutex> lock( m_mutex );
if( m_count == 0 )
{
return false;
}
else
{
m_count--;
return true;
}
}
private:
std::mutex m_mutex;
std::condition_variable m_cv;
unsigned int m_count;
};
#endif