1 | #ifndef UTIL_H__ |
1 | #ifndef UTIL_H__ |
2 | #define UTIL_H__ |
2 | #define UTIL_H__ |
|
|
3 | |
|
|
4 | //#define PREFER_MALLOC |
3 | |
5 | |
4 | #if __GNUC__ >= 3 |
6 | #if __GNUC__ >= 3 |
5 | # define is_constant(c) __builtin_constant_p (c) |
7 | # define is_constant(c) __builtin_constant_p (c) |
6 | #else |
8 | #else |
7 | # define is_constant(c) 0 |
9 | # define is_constant(c) 0 |
… | |
… | |
16 | |
18 | |
17 | #include <shstr.h> |
19 | #include <shstr.h> |
18 | #include <traits.h> |
20 | #include <traits.h> |
19 | |
21 | |
20 | // use a gcc extension for auto declarations until ISO C++ sanctifies them |
22 | // use a gcc extension for auto declarations until ISO C++ sanctifies them |
21 | #define AUTODECL(var,expr) typeof(expr) var = (expr) |
23 | #define auto(var,expr) typeof(expr) var = (expr) |
22 | |
24 | |
23 | // very ugly macro that basicaly declares and initialises a variable |
25 | // very ugly macro that basicaly declares and initialises a variable |
24 | // that is in scope for the next statement only |
26 | // that is in scope for the next statement only |
25 | // works only for stuff that can be assigned 0 and converts to false |
27 | // works only for stuff that can be assigned 0 and converts to false |
26 | // (note: works great for pointers) |
28 | // (note: works great for pointers) |
… | |
… | |
35 | #define IN_RANGE_EXC(val,beg,end) \ |
37 | #define IN_RANGE_EXC(val,beg,end) \ |
36 | ((unsigned int)(val) - (unsigned int)(beg) < (unsigned int)(end) - (unsigned int)(beg)) |
38 | ((unsigned int)(val) - (unsigned int)(beg) < (unsigned int)(end) - (unsigned int)(beg)) |
37 | |
39 | |
38 | void fork_abort (const char *msg); |
40 | void fork_abort (const char *msg); |
39 | |
41 | |
|
|
42 | // rationale for using (U) not (T) is to reduce signed/unsigned issues, |
|
|
43 | // as a is often a constant while b is the variable. it is still a bug, though. |
40 | template<typename T, typename U> static inline T min (T a, U b) { return a < (T)b ? a : (T)b; } |
44 | template<typename T, typename U> static inline T min (T a, U b) { return (U)a < b ? (U)a : b; } |
41 | template<typename T, typename U> static inline T max (T a, U b) { return a > (T)b ? a : (T)b; } |
45 | template<typename T, typename U> static inline T max (T a, U b) { return (U)a > b ? (U)a : b; } |
42 | template<typename T, typename U, typename V> static inline T clamp (T v, U a, V b) { return v < (T)a ? a : v >(T)b ? b : v; } |
46 | template<typename T, typename U, typename V> static inline T clamp (T v, U a, V b) { return v < (T)a ? (T)a : v >(T)b ? (T)b : v; } |
43 | |
47 | |
44 | template<typename T, typename U> static inline void swap (T& a, U& b) { T t=a; a=(T)b; b=(U)t; } |
48 | template<typename T, typename U> static inline void swap (T& a, U& b) { T t=a; a=(T)b; b=(U)t; } |
|
|
49 | |
|
|
50 | // lots of stuff taken from FXT |
|
|
51 | |
|
|
52 | /* Rotate right. This is used in various places for checksumming */ |
|
|
53 | //TODO: that sucks, use a better checksum algo |
|
|
54 | static inline uint32_t |
|
|
55 | rotate_right (uint32_t c, uint32_t count = 1) |
|
|
56 | { |
|
|
57 | return (c << (32 - count)) | (c >> count); |
|
|
58 | } |
|
|
59 | |
|
|
60 | static inline uint32_t |
|
|
61 | rotate_left (uint32_t c, uint32_t count = 1) |
|
|
62 | { |
|
|
63 | return (c >> (32 - count)) | (c << count); |
|
|
64 | } |
|
|
65 | |
|
|
66 | // Return abs(a-b) |
|
|
67 | // Both a and b must not have the most significant bit set |
|
|
68 | static inline uint32_t |
|
|
69 | upos_abs_diff (uint32_t a, uint32_t b) |
|
|
70 | { |
|
|
71 | long d1 = b - a; |
|
|
72 | long d2 = (d1 & (d1 >> 31)) << 1; |
|
|
73 | |
|
|
74 | return d1 - d2; // == (b - d) - (a + d); |
|
|
75 | } |
|
|
76 | |
|
|
77 | // Both a and b must not have the most significant bit set |
|
|
78 | static inline uint32_t |
|
|
79 | upos_min (uint32_t a, uint32_t b) |
|
|
80 | { |
|
|
81 | int32_t d = b - a; |
|
|
82 | d &= d >> 31; |
|
|
83 | return a + d; |
|
|
84 | } |
|
|
85 | |
|
|
86 | // Both a and b must not have the most significant bit set |
|
|
87 | static inline uint32_t |
|
|
88 | upos_max (uint32_t a, uint32_t b) |
|
|
89 | { |
|
|
90 | int32_t d = b - a; |
|
|
91 | d &= d >> 31; |
|
|
92 | return b - d; |
|
|
93 | } |
45 | |
94 | |
46 | // this is much faster than crossfires original algorithm |
95 | // this is much faster than crossfires original algorithm |
47 | // on modern cpus |
96 | // on modern cpus |
48 | inline int |
97 | inline int |
49 | isqrt (int n) |
98 | isqrt (int n) |
… | |
… | |
131 | |
180 | |
132 | // for symmetry |
181 | // for symmetry |
133 | template<typename T> |
182 | template<typename T> |
134 | inline void sfree (T *ptr, int n = 1) throw () |
183 | inline void sfree (T *ptr, int n = 1) throw () |
135 | { |
184 | { |
|
|
185 | #ifdef PREFER_MALLOC |
|
|
186 | free (ptr); |
|
|
187 | #else |
136 | g_slice_free1 (n * sizeof (T), (void *)ptr); |
188 | g_slice_free1 (n * sizeof (T), (void *)ptr); |
|
|
189 | #endif |
137 | } |
190 | } |
138 | |
191 | |
139 | // a STL-compatible allocator that uses g_slice |
192 | // a STL-compatible allocator that uses g_slice |
140 | // boy, this is verbose |
193 | // boy, this is verbose |
141 | template<typename Tp> |
194 | template<typename Tp> |
… | |
… | |
194 | // P. L'Ecuyer, “Maximally Equidistributed Combined Tausworthe Generators”, Mathematics of Computation, 65, 213 (1996), 203–213. |
247 | // P. L'Ecuyer, “Maximally Equidistributed Combined Tausworthe Generators”, Mathematics of Computation, 65, 213 (1996), 203–213. |
195 | // http://www.iro.umontreal.ca/~lecuyer/myftp/papers/tausme.ps |
248 | // http://www.iro.umontreal.ca/~lecuyer/myftp/papers/tausme.ps |
196 | // http://www.iro.umontreal.ca/~lecuyer/myftp/papers/tausme2.ps |
249 | // http://www.iro.umontreal.ca/~lecuyer/myftp/papers/tausme2.ps |
197 | struct tausworthe_random_generator |
250 | struct tausworthe_random_generator |
198 | { |
251 | { |
|
|
252 | // generator |
199 | uint32_t state [4]; |
253 | uint32_t state [4]; |
200 | |
254 | |
201 | tausworthe_random_generator (uint32_t seed); |
255 | void operator =(const tausworthe_random_generator &src) |
|
|
256 | { |
|
|
257 | state [0] = src.state [0]; |
|
|
258 | state [1] = src.state [1]; |
|
|
259 | state [2] = src.state [2]; |
|
|
260 | state [3] = src.state [3]; |
|
|
261 | } |
|
|
262 | |
|
|
263 | void seed (uint32_t seed); |
202 | uint32_t next (); |
264 | uint32_t next (); |
203 | |
265 | |
|
|
266 | // uniform distribution |
204 | uint32_t operator ()(uint32_t r_max) |
267 | uint32_t operator ()(uint32_t num) |
205 | { |
268 | { |
206 | return next () % r_max; |
269 | return is_constant (num) |
|
|
270 | ? (next () * (uint64_t)num) >> 32U |
|
|
271 | : get_range (num); |
207 | } |
272 | } |
208 | |
273 | |
209 | // return a number within (min .. max) |
274 | // return a number within (min .. max) |
210 | int operator () (int r_min, int r_max) |
275 | int operator () (int r_min, int r_max) |
211 | { |
276 | { |
212 | return r_min + (*this) (max (r_max - r_min + 1, 1)); |
277 | return is_constant (r_min) && is_constant (r_max) && r_min <= r_max |
|
|
278 | ? r_min + operator ()(r_max - r_min + 1) |
|
|
279 | : get_range (r_min, r_max); |
213 | } |
280 | } |
214 | |
281 | |
215 | double operator ()() |
282 | double operator ()() |
216 | { |
283 | { |
217 | return next () / (double)0xFFFFFFFFU; |
284 | return this->next () / (double)0xFFFFFFFFU; |
218 | } |
285 | } |
|
|
286 | |
|
|
287 | protected: |
|
|
288 | uint32_t get_range (uint32_t r_max); |
|
|
289 | int get_range (int r_min, int r_max); |
219 | }; |
290 | }; |
220 | |
291 | |
221 | typedef tausworthe_random_generator rand_gen; |
292 | typedef tausworthe_random_generator rand_gen; |
222 | |
293 | |
223 | extern rand_gen rndm; |
294 | extern rand_gen rndm; |
… | |
… | |
328 | } |
399 | } |
329 | |
400 | |
330 | void erase (T *obj) |
401 | void erase (T *obj) |
331 | { |
402 | { |
332 | assert (obj->*index); |
403 | assert (obj->*index); |
333 | int pos = obj->*index; |
404 | unsigned int pos = obj->*index; |
334 | obj->*index = 0; |
405 | obj->*index = 0; |
335 | |
406 | |
336 | if (pos < this->size ()) |
407 | if (pos < this->size ()) |
337 | { |
408 | { |
338 | (*this)[pos - 1] = (*this)[this->size () - 1]; |
409 | (*this)[pos - 1] = (*this)[this->size () - 1]; |
… | |
… | |
363 | // return current time as timestampe |
434 | // return current time as timestampe |
364 | tstamp now (); |
435 | tstamp now (); |
365 | |
436 | |
366 | int similar_direction (int a, int b); |
437 | int similar_direction (int a, int b); |
367 | |
438 | |
|
|
439 | // like printf, but returns a std::string |
|
|
440 | const std::string format (const char *format, ...); |
|
|
441 | |
368 | #endif |
442 | #endif |
369 | |
443 | |