distribution_uniform.hpp

Go to the documentation of this file.

#pragma once
 
#include "distribution.hpp"
#include <random>
 
namespace netdem {
 
class UniformDistribution : public Distribution {
public:
  double bound_min, bound_max;
 
  UniformDistribution() : bound_min(0), bound_max(1) {
    std::random_device rd;
    mt_eng = std::mt19937(rd());
    real_dist = std::uniform_real_distribution<double>(0, 1);
  }
 
  UniformDistribution(double bound_min, double bound_max)
      : bound_min(bound_min), bound_max(bound_max) {
    std::random_device rd;
    mt_eng = std::mt19937(rd());
    real_dist = std::uniform_real_distribution<double>(bound_min, bound_max);
  }
 
  double Get() override { return real_dist(mt_eng); }
 
  VecXT<double> Get(int num) override {
    VecXT<double> num_list(num, 0);
 
    for (int i = 0; i < num; i++) {
      num_list[i] = real_dist(mt_eng);
    }
 
    return num_list;
  }
 
private:
  std::mt19937 mt_eng; 
  std::uniform_real_distribution<double>
      real_dist; 
};
 
} // namespace netdem