summaryrefslogtreecommitdiff
path: root/include/llvm/Support/RandomNumberGenerator.h
blob: 1ec2c7eca944620bdac2200659d17b79f075365d (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
//==- llvm/Support/RandomNumberGenerator.h - RNG for diversity ---*- C++ -*-==//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file defines an abstraction for random number generation (RNG).
// Note that the current implementation is not cryptographically secure
// as it uses the C++11 <random> facilities.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_SUPPORT_RANDOMNUMBERGENERATOR_H_
#define LLVM_SUPPORT_RANDOMNUMBERGENERATOR_H_

#include "llvm/ADT/StringRef.h"
#include "llvm/Support/DataTypes.h" // Needed for uint64_t on Windows.
#include <random>

namespace llvm {

/// A random number generator.
/// Instances of this class should not be shared across threads.
class RandomNumberGenerator {
public:
  /// Seeds and salts the underlying RNG engine. The salt of type StringRef
  /// is passed into the constructor. The seed can be set on the command
  /// line via -rng-seed=<uint64>.
  /// The reason for the salt is to ensure different random streams even if
  /// the same seed is used for multiple invocations of the compiler.
  /// A good salt value should add additional entropy and be constant across
  /// different machines (i.e., no paths) to allow for reproducible builds.
  /// An instance of this class can be retrieved from the current Module.
  /// \see Module::getRNG
  RandomNumberGenerator(StringRef Salt);

  /// Returns a random number in the range [0, Max).
  uint64_t next(uint64_t Max);

private:
  // 64-bit Mersenne Twister by Matsumoto and Nishimura, 2000
  // http://en.cppreference.com/w/cpp/numeric/random/mersenne_twister_engine
  std::mt19937_64 Generator;

  // Noncopyable.
  RandomNumberGenerator(const RandomNumberGenerator &other) = delete;
  RandomNumberGenerator &operator=(const RandomNumberGenerator &other) = delete;
};
}

#endif