ChordStepDistributions.hpp

Go to the documentation of this file.

#ifndef HOPS_CHORDSTEPDISTRIBUTIONS_HPP
#define HOPS_CHORDSTEPDISTRIBUTIONS_HPP
 
#include "../../RandomNumberGenerator/RandomNumberGenerator.hpp"
#include <random>
#include <string>
#include "TruncatedNormalDistribution.hpp"
 
namespace hops {
    template<typename RealType = double>
    class UniformStepDistribution {
    public:
        RealType draw(RandomNumberGenerator &randomNumberGenerator, RealType lowerLimit, RealType upperLimit) {
            typename std::uniform_real_distribution<RealType>::param_type params(lowerLimit, upperLimit);
            return uniformRealDistribution(randomNumberGenerator, params);
        }
 
        constexpr RealType computeInverseNormalizationConstant(RealType, RealType, RealType) {
            return 1.;
        }
 
    private:
        std::uniform_real_distribution<RealType> uniformRealDistribution;
    };
 
    template<typename RealType = double>
    class GaussianStepDistribution {
    public:
        RealType draw(RandomNumberGenerator &randomNumberGenerator, RealType lowerLimit,
                      RealType upperLimit) {
            return truncatedNormalDistribution(randomNumberGenerator, {stepSize, lowerLimit, upperLimit});
        }
 
        RealType draw(RandomNumberGenerator &randomNumberGenerator,
                          double m_sigma,
                          RealType lowerLimit,
                          RealType upperLimit) {
            return truncatedNormalDistribution(randomNumberGenerator, {m_sigma, lowerLimit, upperLimit});
        }
 
        RealType getStepSize() const {
            return stepSize;
        }
 
        void setStepSize(RealType newStepSize) {
            stepSize = newStepSize;
        }
 
        RealType computeInverseNormalizationConstant(RealType m_sigma, RealType m_lowerBound, RealType m_upperBound) {
            return truncatedNormalDistribution.inverseNormalization({m_sigma, m_lowerBound, m_upperBound});
        }
 
        RealType computeProbabilityDensity(RealType x, RealType m_sigma, RealType m_lowerBound, RealType m_upperBound){
            return truncatedNormalDistribution.probabilityDensity(x, m_sigma, m_lowerBound, m_upperBound);
        }
 
    private:
        RealType stepSize = 1.;
        TruncatedNormalDistribution<RealType> truncatedNormalDistribution;
    };
}
 
#endif //HOPS_CHORDSTEPDISTRIBUTIONS_HPP