diff options
Diffstat (limited to 'core/math/randomizer.hh')
| -rw-r--r-- | core/math/randomizer.hh | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/core/math/randomizer.hh b/core/math/randomizer.hh new file mode 100644 index 0000000..206af42 --- /dev/null +++ b/core/math/randomizer.hh @@ -0,0 +1,60 @@ +#ifndef CORE_MATH_RANDOMIZER_HH +#define CORE_MATH_RANDOMIZER_HH 1 +#pragma once + +namespace math +{ +template<typename T> +class Randomizer final { +public: + explicit Randomizer(void); + explicit Randomizer(std::uint64_t seed); + virtual ~Randomizer(void) = default; + void add(const T& value); + const T& get(void); + void clear(void); + +private: + std::vector<T> m_vector; + std::mt19937_64 m_twister; + std::uniform_int_distribution<std::size_t> m_dist; +}; +} // namespace math + +template<typename T> +inline math::Randomizer<T>::Randomizer(void) +{ + m_vector.clear(); + m_twister.seed(std::random_device()()); + m_dist = std::uniform_int_distribution<std::size_t>(0, 0); +} + +template<typename T> +inline math::Randomizer<T>::Randomizer(std::uint64_t seed) +{ + m_vector.clear(); + m_twister.seed(seed); + m_dist = std::uniform_int_distribution<std::size_t>(0, 0); +} + +template<typename T> +inline void math::Randomizer<T>::add(const T& value) +{ + m_vector.push_back(value); + m_dist = std::uniform_int_distribution<std::size_t>(0, m_vector.size() - 1); +} + +template<typename T> +inline const T& math::Randomizer<T>::get(void) +{ + return m_vector.at(m_dist(m_twister)); +} + +template<typename T> +inline void math::Randomizer<T>::clear(void) +{ + m_vector.clear(); + m_dist = std::uniform_int_distribution<std::size_t>(0, 0); +} + +#endif // CORE_MATH_RANDOMIZER_HH |
