summaryrefslogtreecommitdiff
path: root/src/util/random.h
blob: 480271c03ec906916d8b15a95be9ee6fb6c86081 (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
/*********************                                                        */
/*! \file random.h
 ** \verbatim
 ** Top contributors (to current version):
 **   Aina Niemetz
 ** This file is part of the CVC4 project.
 ** Copyright (c) 2009-2018 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 "cvc4_private.h"

#ifndef __CVC4__UTIL__RANDOM_H
#define __CVC4__UTIL__RANDOM_H

namespace CVC4 {

class Random
{
 public:
  Random(uint64_t seed) { setSeed(seed); }

  /* Get current RNG (singleton).  */
  static Random& getRandom()
  {
    static thread_local Random s_current(0);
    return s_current;
  }

  /* Set seed of Random.  */
  void setSeed(uint64_t seed)
  {
    d_seed = seed == 0 ? ~seed : seed;
    d_state = d_seed;
  }

  /* Next random uint64_t number. */
  uint64_t rand();
  /* Pick random uint64_t number between from and to (inclusive). */
  uint64_t pick(uint64_t from, uint64_t to);
  /* Pick random double number between from and to (inclusive). */
  double pickDouble(double from, double to);
  /* Pick with given probability (yes / no). */
  bool pickWithProb(double probability);

 private:
  /* The seed of the RNG. */
  uint64_t d_seed;
  /* The current state of the RNG. */
  uint64_t d_state;
};

}  // namespace CVC4
#endif
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback