Commit | Line | Data |
---|---|---|
f00ff009 LOK |
1 | #pragma once |
2 | /* | |
3 | * This file is part of the libCEC(R) library. | |
4 | * | |
b492c10e | 5 | * libCEC(R) is Copyright (C) 2011-2012 Pulse-Eight Limited. All rights reserved. |
f00ff009 LOK |
6 | * libCEC(R) is an original work, containing original code. |
7 | * | |
8 | * libCEC(R) is a trademark of Pulse-Eight Limited. | |
9 | * | |
10 | * This program is dual-licensed; you can redistribute it and/or modify | |
11 | * it under the terms of the GNU General Public License as published by | |
12 | * the Free Software Foundation; either version 2 of the License, or | |
13 | * (at your option) any later version. | |
14 | * | |
15 | * This program is distributed in the hope that it will be useful, | |
16 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
17 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
18 | * GNU General Public License for more details. | |
19 | * | |
20 | * You should have received a copy of the GNU General Public License | |
21 | * along with this program; if not, write to the Free Software | |
22 | * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | |
23 | * | |
24 | * | |
25 | * Alternatively, you can license this library under a commercial license, | |
26 | * please contact Pulse-Eight Licensing for more information. | |
27 | * | |
28 | * For more information contact: | |
29 | * Pulse-Eight Licensing <license@pulse-eight.com> | |
30 | * http://www.pulse-eight.com/ | |
31 | * http://www.pulse-eight.net/ | |
32 | */ | |
33 | ||
34 | namespace PLATFORM | |
35 | { | |
36 | #define thread_t HANDLE | |
37 | #define ThreadsWait(thread, retVal) (::WaitForSingleObject(thread, INFINITE) < 0) | |
38 | #define ThreadsCreate(thread, func, arg) ((thread = ::CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)func, arg, 0, NULL)) == NULL ? false : true) | |
39 | ||
40 | typedef CRITICAL_SECTION* mutex_t; | |
41 | #define MutexCreate(mutex) ::InitializeCriticalSection(mutex = new CRITICAL_SECTION) | |
42 | #define MutexDelete(mutex) ::DeleteCriticalSection(mutex); delete mutex | |
43 | #define MutexLock(mutex) ::EnterCriticalSection(mutex) | |
44 | #define MutexTryLock(mutex) (::TryEnterCriticalSection(mutex) != 0) | |
45 | #define MutexUnlock(mutex) ::LeaveCriticalSection(mutex) | |
46 | ||
3a590d6a LOK |
47 | // windows vista+ conditions |
48 | typedef VOID (WINAPI *ConditionArg) (CONDITION_VARIABLE*); | |
49 | typedef BOOL (WINAPI *ConditionMutexArg)(CONDITION_VARIABLE*, CRITICAL_SECTION*, DWORD); | |
50 | static ConditionArg g_InitializeConditionVariable; | |
51 | static ConditionArg g_WakeConditionVariable; | |
52 | static ConditionArg g_WakeAllConditionVariable; | |
53 | static ConditionMutexArg g_SleepConditionVariableCS; | |
54 | ||
55 | // check whether vista+ conditions are available at runtime | |
56 | static bool CheckVistaConditionFunctions(void) | |
6a3e7856 | 57 | { |
3a590d6a LOK |
58 | static int iHasVistaConditionFunctions(-1); |
59 | if (iHasVistaConditionFunctions == -1) | |
60 | { | |
61 | HMODULE handle = GetModuleHandle("Kernel32"); | |
62 | if (handle == NULL) | |
63 | { | |
64 | iHasVistaConditionFunctions = 0; | |
65 | } | |
66 | else | |
67 | { | |
68 | g_InitializeConditionVariable = (ConditionArg) GetProcAddress(handle,"InitializeConditionVariable"); | |
69 | g_WakeConditionVariable = (ConditionArg) GetProcAddress(handle,"WakeConditionVariable"); | |
70 | g_WakeAllConditionVariable = (ConditionArg) GetProcAddress(handle,"WakeAllConditionVariable"); | |
71 | g_SleepConditionVariableCS = (ConditionMutexArg)GetProcAddress(handle,"SleepConditionVariableCS"); | |
72 | ||
73 | // 1 when everything is resolved, 0 otherwise | |
74 | iHasVistaConditionFunctions = g_InitializeConditionVariable && | |
75 | g_WakeConditionVariable && | |
76 | g_WakeAllConditionVariable && | |
77 | g_SleepConditionVariableCS ? 1 : 0; | |
78 | } | |
79 | } | |
80 | return iHasVistaConditionFunctions == 1; | |
6a3e7856 | 81 | } |
3a590d6a LOK |
82 | |
83 | class CConditionImpl | |
84 | { | |
85 | public: | |
86 | CConditionImpl(void) | |
87 | { | |
88 | m_bOnVista = CheckVistaConditionFunctions(); | |
89 | if (m_bOnVista) | |
90 | (*g_InitializeConditionVariable)(m_conditionVista = new CONDITION_VARIABLE); | |
91 | else | |
92 | m_conditionPreVista = ::CreateEvent(NULL, TRUE, FALSE, NULL); | |
93 | } | |
94 | ||
95 | virtual ~CConditionImpl(void) | |
96 | { | |
97 | if (m_bOnVista) | |
98 | delete m_conditionVista; | |
99 | else | |
100 | ::CloseHandle(m_conditionPreVista); | |
101 | } | |
102 | ||
103 | void Signal(void) | |
104 | { | |
105 | if (m_bOnVista) | |
106 | (*g_WakeConditionVariable)(m_conditionVista); | |
107 | else | |
108 | ::SetEvent(m_conditionPreVista); | |
109 | } | |
110 | ||
111 | void Broadcast(void) | |
112 | { | |
113 | if (m_bOnVista) | |
114 | (*g_WakeAllConditionVariable)(m_conditionVista); | |
115 | else | |
116 | ::SetEvent(m_conditionPreVista); | |
117 | } | |
118 | ||
fd09ec02 LOK |
119 | bool Wait(mutex_t &mutex) |
120 | { | |
121 | if (m_bOnVista) | |
122 | { | |
123 | return ((*g_SleepConditionVariableCS)(m_conditionVista, mutex, INFINITE) ? true : false); | |
124 | } | |
125 | else | |
126 | { | |
127 | ::ResetEvent(m_conditionPreVista); | |
128 | MutexUnlock(mutex); | |
129 | DWORD iWaitReturn = ::WaitForSingleObject(m_conditionPreVista, 1000); | |
130 | MutexLock(mutex); | |
131 | return (iWaitReturn == 0); | |
132 | } | |
133 | } | |
134 | ||
3a590d6a LOK |
135 | bool Wait(mutex_t &mutex, uint32_t iTimeoutMs) |
136 | { | |
fd09ec02 LOK |
137 | if (iTimeoutMs == 0) |
138 | return Wait(mutex); | |
139 | ||
3a590d6a LOK |
140 | if (m_bOnVista) |
141 | { | |
fd09ec02 | 142 | return ((*g_SleepConditionVariableCS)(m_conditionVista, mutex, iTimeoutMs) ? true : false); |
3a590d6a LOK |
143 | } |
144 | else | |
145 | { | |
146 | ::ResetEvent(m_conditionPreVista); | |
147 | MutexUnlock(mutex); | |
fd09ec02 | 148 | DWORD iWaitReturn = ::WaitForSingleObject(m_conditionPreVista, iTimeoutMs); |
3a590d6a LOK |
149 | MutexLock(mutex); |
150 | return (iWaitReturn == 0); | |
151 | } | |
152 | } | |
153 | ||
154 | bool m_bOnVista; | |
155 | CONDITION_VARIABLE *m_conditionVista; | |
156 | HANDLE m_conditionPreVista; | |
157 | }; | |
f00ff009 | 158 | } |