blob: a42d74a3ce1a47703548047e88f383892ff94163 (
plain)
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
|
/******************************************************************************
* Top contributors (to current version):
* Aina Niemetz
*
* This file is part of the cvc5 project.
*
* Copyright (c) 2009-2021 by the authors listed in the file AUTHORS
* in the top-level source directory and their institutional affiliations.
* All rights reserved. See the file COPYING in the top-level source
* directory for licensing information.
* ****************************************************************************
*
* A random number generator, implements the xorshift* generator
* (see S. Vigna, An experimental exploration of Marsaglia's xorshift
* generators, scrambled. ACM Trans. Math. Softw. 42(4): 30:1-30:23, 2016).
*/
#include "util/random.h"
#include <cfloat>
#include "base/check.h"
namespace cvc5 {
Random::Random(uint64_t seed) { setSeed(seed); }
void Random::setSeed(uint64_t seed)
{
d_seed = seed == 0 ? ~seed : seed;
d_state = d_seed;
}
uint64_t Random::operator()() { return rand(); }
uint64_t Random::rand()
{
/* xorshift* generator (see S. Vigna, An experimental exploration of
* Marsaglia's xorshift generators, scrambled. ACM Trans. Math. Softw.
* 42(4): 30:1-30:23, 2016). */
d_state ^= d_state >> 12;
d_state ^= d_state << 25;
d_state ^= d_state >> 27;
return d_state * uint64_t{2685821657736338717};
}
uint64_t Random::pick(uint64_t from, uint64_t to)
{
Assert(from <= to);
Assert(to < UINT64_MAX);
return (Random::rand() % (to - from + 1)) + from;
}
double Random::pickDouble(double from, double to)
{
Assert(from <= to);
Assert(to <= DBL_MAX);
return Random::rand() * (to - from) + from;
}
bool Random::pickWithProb(double probability)
{
Assert(probability <= 1);
uint64_t p = (uint64_t) (probability * 1000);
uint64_t r = pick(0, 999);
return r < p;
}
} // namespace cvc5
|