summaryrefslogtreecommitdiff
path: root/src/util/random.cpp
blob: 89b8c05136407114543a4b41e5b52598d7c90888 (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
69
/*********************                                                        */
/*! \file random.cpp
 ** \verbatim
 ** Top contributors (to current version):
 **   Aina Niemetz
 ** This file is part of the CVC4 project.
 ** Copyright (c) 2009-2020 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.\endverbatim
 **
 ** \brief A Random Number Generator.
 **
 ** 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 CVC4 {

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;
}

}
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback