blob: 2073fd76f5a46c923fa30a2913e44af668949821 (
plain) (
blame)
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
|
#include "main.h"
#include "./flagTracker.h"
FlagTracker::FlagTracker()
: flags(0), cache(0), updateRef(0), eventHandler(NULL), eventParam(0)
{
InitializeCriticalSection(&lock);
}
FlagTracker::~FlagTracker()
{
DeleteCriticalSection(&lock);
}
void FlagTracker::Set(UINT nFlags, UINT nMask)
{
EnterCriticalSection(&lock);
nFlags &= nMask;
UINT clearMask = ~(nFlags ^ nMask);
flags &= clearMask;
cache &= clearMask;
if (0 != nFlags)
{
Mark(nFlags);
}
LeaveCriticalSection(&lock);
}
UINT FlagTracker::Get()
{
return flags;
}
void FlagTracker::Mark(UINT nFlags)
{
EnterCriticalSection(&lock);
flags |= nFlags;
if (0 == updateRef)
{
if (NULL != eventHandler && 0 != nFlags)
{
eventHandler(nFlags, this, eventParam);
}
}
else
{
cache |= nFlags;
}
LeaveCriticalSection(&lock);
}
ULONG FlagTracker::BeginUpdate()
{
EnterCriticalSection(&lock);
ULONG r = InterlockedIncrement((LONG*)&updateRef);
LeaveCriticalSection(&lock);
return r;
}
ULONG FlagTracker::EndUpdate()
{
EnterCriticalSection(&lock);
ULONG r;
if (0 == updateRef)
{
r = 0;
}
else
{
r = InterlockedDecrement((LONG*)&updateRef);
if (0 == r && 0 != cache)
{
Mark(cache);
cache = 0;
}
}
LeaveCriticalSection(&lock);
return r;
}
void FlagTracker::SetEventHandler(EventHandler handler, ULONG_PTR user)
{
EnterCriticalSection(&lock);
eventHandler = handler;
eventParam = user;
LeaveCriticalSection(&lock);
}
|
