summaryrefslogtreecommitdiff
path: root/src/util/random.h
blob: 738f5c510eafa05f13cdad74d4f4c4f18240d5fe (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
70
71
72
73
/*********************                                                        */
/*! \file random.h
 ** \verbatim
 ** Top contributors (to current version):
 **   Aina Niemetz, Andres Noetzli, Mathias Preiner
 ** This file is part of the CVC4 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.\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 cvc5 {

class Random
{
 public:
  using result_type = uint64_t;

  /** Constructor. */
  Random(uint64_t seed);

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

  /** Get the minimum number that can be picked. */
  static constexpr uint64_t min() { return 0u; }

  /** Get the maximum number that can be picked. */
  static constexpr uint64_t max() { return UINT64_MAX; }

  /** Set seed of Random.  */
  void setSeed(uint64_t seed);

  /** Operator overload to pick random uin64_t number (see rand()). */
  uint64_t operator()();

  /** 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 cvc5
#endif
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback