| 1 | |
| 2 | |
| 3 | |
| 4 | |
| 5 | |
| 6 | |
| 7 | |
| 8 | |
| 9 | |
| 10 | |
| 11 | |
| 12 | |
| 13 | |
| 14 | |
| 15 | |
| 16 | |
| 17 | |
| 18 | |
| 19 | |
| 20 | |
| 21 | |
| 22 | |
| 23 | |
| 24 | #include <util/random.hh> |
| 25 | #include <cmath> |
| 26 | #include <chrono> |
| 27 | |
| 28 | namespace util::rand { |
| 29 | u32 randState[4]; |
| 30 | |
| 31 | static inline u32 leftRot(const u32 x, i32 k) { |
| 32 | return (x << k) | (x >> (32 - k)); |
| 33 | } |
| 34 | |
| 35 | u32 xoshiro128() { |
| 36 | const u32 res = leftRot(randState[1] * 5, 7) * 9; |
| 37 | const u32 t = randState[1] << 9; |
| 38 | |
| 39 | randState[2] ^= randState[0]; |
| 40 | randState[3] ^= randState[1]; |
| 41 | randState[1] ^= randState[2]; |
| 42 | randState[0] ^= randState[3]; |
| 43 | |
| 44 | randState[2] ^= t; |
| 45 | randState[3] = leftRot(randState[3], 11); |
| 46 | |
| 47 | return res; |
| 48 | } |
| 49 | |
| 50 | i32 getValue(i32 min, i32 max) { |
| 51 | u32 range = static_cast<u32>(max - min + 1); |
| 52 | return min + (xoshiro128() % range); |
| 53 | } |
| 54 | |
| 55 | static u32 splitmix32(u32 &x) { |
| 56 | u32 z = (x += 0x9e3779b9); |
| 57 | z = (z ^ (z >> 16)) * 0x85ebca6b; |
| 58 | z = (z ^ (z >> 13)) * 0xc2b2ae35; |
| 59 | return z ^ (z >> 16); |
| 60 | } |
| 61 | |
| 62 | |
| 63 | |
| 64 | void Seed(u32 seed) { |
| 65 | for (int i = 0; i < 4; i++) randState[i] = splitmix32(seed); |
| 66 | } |
| 67 | |
| 68 | i32 GetRandom(i32 min, i32 max) { |
| 69 | if (min > max) { |
| 70 | i32 tmp = max; |
| 71 | max = min; |
| 72 | min = tmp; |
| 73 | } |
| 74 | |
| 75 | return getValue(min, max); |
| 76 | } |
| 77 | } |