Random.hpp

#pragma once
 
#include "util/Assert.hpp"
 
#include <cstddef>
#include <random>
 
namespace util {

class RandomGeneratorInterface {
public:
    virtual ~RandomGeneratorInterface() = default;
 
    using SeedType = typename std::mt19937_64::result_type;

    [[nodiscard]]
    virtual size_t
    uniform(size_t min, size_t max) = 0;

    virtual void
    setSeed(SeedType seed) = 0;
};

class MTRandomGenerator : public RandomGeneratorInterface {
public:
    MTRandomGenerator();

    [[nodiscard]]
    size_t
    uniform(size_t min, size_t max) override;

    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_);
    }

    void
    setSeed(SeedType seed) override;
 
private:
    std::mt19937_64 generator_;
};
 
}  // namespace util