mirror of
https://github.com/XRPLF/clio.git
synced 2026-06-03 16:56:45 +00:00
110 lines
3.1 KiB
C++
110 lines
3.1 KiB
C++
//------------------------------------------------------------------------------
|
|
/*
|
|
This file is part of clio: https://github.com/XRPLF/clio
|
|
Copyright (c) 2023, the clio developers.
|
|
|
|
Permission to use, copy, modify, and distribute this software for any
|
|
purpose with or without fee is hereby granted, provided that the above
|
|
copyright notice and this permission notice appear in all copies.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
//==============================================================================
|
|
#pragma once
|
|
|
|
#include "util/Assert.hpp"
|
|
|
|
#include <cstddef>
|
|
#include <random>
|
|
|
|
namespace util {
|
|
|
|
/**
|
|
* @brief Random number generator interface
|
|
*/
|
|
class RandomGeneratorInterface {
|
|
public:
|
|
virtual ~RandomGeneratorInterface() = default;
|
|
|
|
using SeedType = typename std::mt19937_64::result_type;
|
|
|
|
/**
|
|
* @brief Generate a random number between min and max
|
|
*
|
|
* @param min Minimum value
|
|
* @param max Maximum value
|
|
* @return Random number between min and max
|
|
*/
|
|
[[nodiscard]]
|
|
virtual size_t
|
|
uniform(size_t min, size_t max) = 0;
|
|
|
|
/**
|
|
* @brief Set the seed for the random number generator
|
|
*
|
|
* @param seed Seed to set
|
|
*/
|
|
virtual void
|
|
setSeed(SeedType seed) = 0;
|
|
};
|
|
|
|
/**
|
|
* @brief Mersenne Twister random number generator
|
|
*/
|
|
class MTRandomGenerator : public RandomGeneratorInterface {
|
|
public:
|
|
MTRandomGenerator();
|
|
|
|
/**
|
|
* @brief Generate a random number between min and max
|
|
*
|
|
* @param min Minimum value
|
|
* @param max Maximum value
|
|
* @return Random number between min and max
|
|
*/
|
|
[[nodiscard]]
|
|
size_t
|
|
uniform(size_t min, size_t max) override;
|
|
|
|
/**
|
|
* @brief Generate a random number between min and max
|
|
*
|
|
* @tparam T Type of the number to generate
|
|
* @param min Minimum value
|
|
* @param max Maximum value
|
|
* @return Random number between min and max
|
|
*/
|
|
template <typename T>
|
|
[[nodiscard]]
|
|
T
|
|
uniformImpl(T min, T max)
|
|
{
|
|
ASSERT(min <= max, "Min cannot be greater than max. min: {}, max: {}", min, max);
|
|
if constexpr (std::is_floating_point_v<T>) {
|
|
std::uniform_real_distribution<T> distribution(min, max);
|
|
return distribution(generator_);
|
|
}
|
|
std::uniform_int_distribution<T> distribution(min, max);
|
|
return distribution(generator_);
|
|
}
|
|
|
|
/**
|
|
* @brief Set the seed for the random number generator
|
|
*
|
|
* @param seed Seed to set
|
|
*/
|
|
void
|
|
setSeed(SeedType seed) override;
|
|
|
|
private:
|
|
std::mt19937_64 generator_;
|
|
};
|
|
|
|
} // namespace util
|