forked from catid/longhair
-
Notifications
You must be signed in to change notification settings - Fork 0
/
SiameseTools.h
239 lines (193 loc) · 6.82 KB
/
SiameseTools.h
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
/** \file
\brief Siamese FEC Implementation: Tools
\copyright Copyright (c) 2017 Christopher A. Taylor. All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright notice,
this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright notice,
this list of conditions and the following disclaimer in the documentation
and/or other materials provided with the distribution.
* Neither the name of Siamese nor the names of its contributors may be
used to endorse or promote products derived from this software without
specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
*/
#pragma once
/**
Tools:
+ System headers
+ Debug breakpoints/asserts
+ Compiler-specific code wrappers
+ PCGRandom implementation
+ Microsecond timing
+ Windowed minimum/maximum
*/
#include <stdint.h> // uint32_t
#include <string.h> // memcpy
#include <new> // std::nothrow
//------------------------------------------------------------------------------
// Portability macros
// Compiler-specific debug break
#if defined(_DEBUG) || defined(DEBUG)
#define SIAMESE_DEBUG
#ifdef _WIN32
#define SIAMESE_DEBUG_BREAK() __debugbreak()
#else
#define SIAMESE_DEBUG_BREAK() __builtin_trap()
#endif
#define SIAMESE_DEBUG_ASSERT(cond) { if (!(cond)) { SIAMESE_DEBUG_BREAK(); } }
#else
#define SIAMESE_DEBUG_BREAK() do {} while (false);
#define SIAMESE_DEBUG_ASSERT(cond) do {} while (false);
#endif
// Compiler-specific force inline keyword
#ifdef _MSC_VER
#define SIAMESE_FORCE_INLINE inline __forceinline
#else
#define SIAMESE_FORCE_INLINE inline __attribute__((always_inline))
#endif
namespace siamese {
//------------------------------------------------------------------------------
// PCG PRNG
/// From http://www.pcg-random.org/
class PCGRandom
{
public:
void Seed(uint64_t y, uint64_t x = 0)
{
State = 0;
Inc = (y << 1u) | 1u;
Next();
State += x;
Next();
}
uint32_t Next()
{
const uint64_t oldstate = State;
State = oldstate * UINT64_C(6364136223846793005) + Inc;
const uint32_t xorshifted = (uint32_t)(((oldstate >> 18) ^ oldstate) >> 27);
const uint32_t rot = oldstate >> 59;
return (xorshifted >> rot) | (xorshifted << ((uint32_t)(-(int32_t)rot) & 31));
}
uint64_t State = 0, Inc = 0;
};
//------------------------------------------------------------------------------
// Timing
/// Platform independent high-resolution timers
uint64_t GetTimeUsec();
uint64_t GetTimeMsec();
//------------------------------------------------------------------------------
// WindowedMinMax
template<typename T> struct WindowedMinCompare
{
SIAMESE_FORCE_INLINE bool operator()(const T x, const T y) const
{
return x <= y;
}
};
template<typename T> struct WindowedMaxCompare
{
SIAMESE_FORCE_INLINE bool operator()(const T x, const T y) const
{
return x >= y;
}
};
/// Templated class that calculates a running windowed minimum or maximum with
/// a fixed time and resource cost.
template<typename T, class CompareT> class WindowedMinMax
{
public:
typedef uint64_t TimeT;
CompareT Compare;
struct Sample
{
/// Sample value
T Value;
/// Timestamp of data collection
TimeT Timestamp;
/// Default values and initializing constructor
explicit Sample(T value = 0, TimeT timestamp = 0)
: Value(value)
, Timestamp(timestamp)
{
}
/// Check if a timeout expired
inline bool TimeoutExpired(TimeT now, TimeT timeout)
{
return (TimeT)(now - Timestamp) > timeout;
}
};
static const unsigned kSampleCount = 3;
Sample Samples[kSampleCount];
bool IsValid() const
{
return Samples[0].Value != 0; ///< ish
}
T GetBest() const
{
return Samples[0].Value;
}
void Reset(const Sample sample = Sample())
{
Samples[0] = Samples[1] = Samples[2] = sample;
}
void Update(T value, TimeT timestamp, const TimeT windowLengthTime)
{
const Sample sample(value, timestamp);
// On the first sample, new best sample, or if window length has expired:
if (!IsValid() ||
Compare(value, Samples[0].Value) ||
Samples[2].TimeoutExpired(sample.Timestamp, windowLengthTime))
{
Reset(sample);
return;
}
// Insert the new value into the sorted array
if (Compare(value, Samples[1].Value))
Samples[2] = Samples[1] = sample;
else if (Compare(value, Samples[2].Value))
Samples[2] = sample;
// Expire best if it has been the best for a long time
if (Samples[0].TimeoutExpired(sample.Timestamp, windowLengthTime))
{
// Also expire the next best if needed
if (Samples[1].TimeoutExpired(sample.Timestamp, windowLengthTime))
{
Samples[0] = Samples[2];
Samples[1] = sample;
}
else
{
Samples[0] = Samples[1];
Samples[1] = Samples[2];
}
Samples[2] = sample;
return;
}
// Quarter of window has gone by without a better value - Use the second-best
if (Samples[1].Value == Samples[0].Value &&
Samples[1].TimeoutExpired(sample.Timestamp, windowLengthTime / 4))
{
Samples[2] = Samples[1] = sample;
return;
}
// Half the window has gone by without a better value - Use the third-best one
if (Samples[2].Value == Samples[1].Value &&
Samples[2].TimeoutExpired(sample.Timestamp, windowLengthTime / 2))
{
Samples[2] = sample;
}
}
};
} // namespace siamese