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
|
/*
* Copyright 2012 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "SkCondVar.h"
#if defined(SK_BUILD_FOR_WIN32)
static void (WINAPI *initialize_condition_variable)(PCONDITION_VARIABLE);
static BOOL (WINAPI *sleep_condition_variable)(PCONDITION_VARIABLE, PCRITICAL_SECTION, DWORD);
static void (WINAPI *wake_condition_variable)(PCONDITION_VARIABLE);
static void (WINAPI *wake_all_condition_variable)(PCONDITION_VARIABLE);
template <typename T>
static void set_fn_ptr(T* ptr, FARPROC fn) { *ptr = reinterpret_cast<T>(fn); }
#endif
bool SkCondVar::Supported() {
#ifdef SK_BUILD_FOR_WIN32
// If we're >= Vista we'll find these functions. Otherwise (XP) SkCondVar is not supported.
HMODULE kernel32 = GetModuleHandleA("kernel32.dll");
set_fn_ptr(&initialize_condition_variable,
GetProcAddress(kernel32, "InitializeConditionVariable"));
set_fn_ptr(&sleep_condition_variable,
GetProcAddress(kernel32, "SleepConditionVariableCS"));
set_fn_ptr(&wake_condition_variable,
GetProcAddress(kernel32, "WakeConditionVariable"));
set_fn_ptr(&wake_all_condition_variable,
GetProcAddress(kernel32, "WakeAllConditionVariable"));
return initialize_condition_variable
&& sleep_condition_variable
&& wake_condition_variable
&& wake_all_condition_variable;
#else
return true;
#endif
}
SkCondVar::SkCondVar() {
#ifdef SK_BUILD_FOR_WIN32
InitializeCriticalSection(&fCriticalSection);
SkASSERT(initialize_condition_variable);
initialize_condition_variable(&fCondition);
#else
pthread_mutex_init(&fMutex, NULL /* default mutex attr */);
pthread_cond_init(&fCond, NULL /* default cond attr */);
#endif
}
SkCondVar::~SkCondVar() {
#ifdef SK_BUILD_FOR_WIN32
DeleteCriticalSection(&fCriticalSection);
// No need to clean up fCondition.
#else
pthread_mutex_destroy(&fMutex);
pthread_cond_destroy(&fCond);
#endif
}
void SkCondVar::lock() {
#ifdef SK_BUILD_FOR_WIN32
EnterCriticalSection(&fCriticalSection);
#else
pthread_mutex_lock(&fMutex);
#endif
}
void SkCondVar::unlock() {
#ifdef SK_BUILD_FOR_WIN32
LeaveCriticalSection(&fCriticalSection);
#else
pthread_mutex_unlock(&fMutex);
#endif
}
void SkCondVar::wait() {
#ifdef SK_BUILD_FOR_WIN32
SkASSERT(sleep_condition_variable);
sleep_condition_variable(&fCondition, &fCriticalSection, INFINITE);
#else
pthread_cond_wait(&fCond, &fMutex);
#endif
}
void SkCondVar::signal() {
#ifdef SK_BUILD_FOR_WIN32
SkASSERT(wake_condition_variable);
wake_condition_variable(&fCondition);
#else
pthread_cond_signal(&fCond);
#endif
}
void SkCondVar::broadcast() {
#ifdef SK_BUILD_FOR_WIN32
SkASSERT(wake_all_condition_variable);
wake_all_condition_variable(&fCondition);
#else
pthread_cond_broadcast(&fCond);
#endif
}
|