51 lines
957 B
C++
51 lines
957 B
C++
#ifndef UNIFORM_H
|
|
#define UNIFORM_H
|
|
|
|
#include <cmath>
|
|
#include <random>
|
|
#include "../random/RandomGenerator.h"
|
|
|
|
#include <type_traits>
|
|
|
|
namespace Distribution {
|
|
|
|
/** uniform distribution */
|
|
template <typename T> class Uniform {
|
|
|
|
private:
|
|
|
|
Random::RandomGenerator gen;
|
|
|
|
/** depending on T, Dist is either a uniform_real or uniform_int distribution */
|
|
typedef typename std::conditional< std::is_floating_point<T>::value, std::uniform_real_distribution<T>, std::uniform_int_distribution<T> >::type Dist;
|
|
Dist dist;
|
|
|
|
|
|
public:
|
|
|
|
/** ctor */
|
|
Uniform(const T min, const T max) : gen(RANDOM_SEED), dist(min, max) {
|
|
|
|
}
|
|
|
|
/** ctor with seed */
|
|
Uniform(const T min, const T max, const uint32_t seed) : gen(seed), dist(min, max) {
|
|
|
|
}
|
|
|
|
/** get a uniformaly distributed random number */
|
|
T draw() {
|
|
return dist(gen);
|
|
}
|
|
|
|
/** set the seed to use */
|
|
void setSeed(const long seed) {
|
|
gen.seed(seed);
|
|
}
|
|
|
|
};
|
|
|
|
}
|
|
|
|
#endif // UNIFORM_H
|