forked from anse1/sqlsmith
-
Notifications
You must be signed in to change notification settings - Fork 8
/
random.hh
41 lines (31 loc) · 887 Bytes
/
random.hh
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
/// @file
/// @brief randomness
#ifndef RANDOM_HH
#define RANDOM_HH
#include <random>
#include <utility>
#include <stdexcept>
#include <iterator>
namespace smith {
extern std::mt19937_64 rng;
}
template<typename T> T& random_pick(std::vector<T>& container) {
if (!container.size())
throw std::runtime_error("No candidates available");
std::uniform_int_distribution<int> pick(0, container.size()-1);
return container[pick(smith::rng)];
}
template<typename I>
I random_pick(I beg, I end) {
if (beg == end)
throw std::runtime_error("No candidates available");
std::uniform_int_distribution<> pick(0, std::distance(beg, end) - 1);
std::advance(beg, pick(smith::rng));
return beg;
}
template<typename I>
I random_pick(std::pair<I,I> iters) {
return random_pick(iters.first, iters.second);
}
int d6(), d9(), d12(), d20(), d42(), d100();
#endif