RandUtils.h

Go to the documentation of this file.

#ifndef RAND_UTILS_H
#define RAND_UTILS_H
 
#include <cstdint>
#include <random>
#include <map>
#include <memory>
#include <shared_mutex>
#include <unordered_set>
#include <stdexcept>
 
#include "XoshiroCpp.hpp"
#include "CNum/Multithreading/ThreadPool.h"

namespace CNum::Utils::Rand {  
  template<typename T>
  void generate_n_unique_rand_in_range(size_t low_bound,
                       size_t high_bound,
                       T *out,
                       size_t n,
                       uint64_t logical_id = 0);
  
  class RandomGenerator {
  private:
    static uint64_t _epoch;
    static uint64_t _seed;
    static ::std::map<uint64_t, ::XoshiroCpp::Xoshiro256PlusPlus> _registry;
    static ::std::map<uint64_t, uint64_t> _stream_local_epochs;
    static ::std::mutex _seed_mtx;
    
    RandomGenerator() = default;
 
  public:
    static XoshiroCpp::Xoshiro256PlusPlus &instance(uint64_t logical_id = 0);

    static void set_global_seed(uint64_t new_seed);

    static void reset_state();
  };
 
#include "CNum/Utils/RandUtils.tpp"
};
 
#endif