bernoulli_dist.h

Go to the documentation of this file.

#ifndef BERNOULLI_DIST_H
#define BERNOULLI_DIST_H
 
#include "bitrl/bitrl_consts.h"
#include "bitrl/bitrl_types.h"
 
#include <random>
#include <vector>
 
namespace bitrl
{
namespace utils::maths::stats
{
 
class BernoulliDist
{
public:
    typedef bool result_type;
 
    explicit BernoulliDist(real_t p = 0.5);
 
    result_type sample() const;
 
    result_type sample(uint_t seed) const;
 
    std::vector<result_type> sample_many(uint_t size) const;
 
    std::vector<result_type> sample_many(uint_t size, uint_t seed) const;
 
    real_t mean() const { return p_; }
 
    real_t variance() const { return p_ * (1.0 - p_); }
 
    void reset() { dist_.reset(); }
 
    void reset(real_t p);
 
private:
    real_t p_;
    mutable std::bernoulli_distribution dist_;
};
 
}
} // namespace bitrl
 
#endif