CoreLib/Event.cpp at main · freezerdev/CoreLib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#include "Base.h"
#include "Event.h"
#ifdef __APPLE__
#include <sys/time.h>
#endif

NS_BEGIN

//#################################################################################################
CEvent::CEvent(const bool bSignaled)
#ifndef _WIN32
	: m_bSignaled(bSignaled)
#endif
{
#ifdef _WIN32
	m_hEvent = CreateEventW(nullptr, true, bSignaled, nullptr);
#endif
}

//#################################################################################################
CEvent::~CEvent(void)
{
	Signal();
#ifdef _WIN32
	CloseHandle(m_hEvent);
#endif
}

//#################################################################################################
void CEvent::Signal(void)
{
#ifdef _WIN32
	SetEvent(m_hEvent);
#else
	std::unique_lock<std::mutex> lock(m_mutex);
	m_bSignaled = true;
	lock.unlock();
	m_cv.notify_all();
#endif
}

//#################################################################################################
void CEvent::Reset(void)
{
#ifdef _WIN32
	ResetEvent(m_hEvent);
#else
	std::lock_guard<std::mutex> lock(m_mutex);
	m_bSignaled = false;
#endif
}

//#################################################################################################
bool CEvent::Wait(const size_t nMilliseconds)
{
	bool bSignaled = true;

#ifdef _WIN32
	if(WaitForSingleObjectEx(m_hEvent, (DWORD)nMilliseconds, false) != WAIT_OBJECT_0)
		bSignaled = false;
#else
	if(nMilliseconds == INFINITE_WAIT)
	{
		std::unique_lock<std::mutex> lock(m_mutex);
		while(!m_bSignaled)
			m_cv.wait(lock);
	}
	else
	{
		struct timespec tsWait = {0};
#ifdef __APPLE__
		timeval tv;
		gettimeofday(&tv, nullptr);
		tsWait.tv_sec = tv.tv_sec;
		tsWait.tv_nsec = tv.tv_usec * 1000;
#elif __linux__
		clock_gettime(CLOCK_REALTIME, &tsWait);
#endif

		uint64_t nNS = ((nMilliseconds % 1000) * 1000000) + tsWait.tv_nsec;
		tsWait.tv_sec += (nMilliseconds / 1000) + (nNS / 1000000000);
		tsWait.tv_nsec = nNS % 1000000000;

		std::lock_guard<std::mutex> lock(m_mutex);
		while(!m_bSignaled)
		{
			if(pthread_cond_timedwait(m_cv.native_handle(), m_mutex.native_handle(), &tsWait) == ETIMEDOUT)
			{
				bSignaled = false;
				break;
			}
		}
	}
#endif

	return bSignaled;
}

#ifdef _WIN32
//#################################################################################################
CEvent::operator HANDLE(void) const noexcept
{
	return m_hEvent;
}
#endif

NS_END