expon.h

// Copyright 2009-2021 NTESS. Under the terms
// of Contract DE-NA0003525 with NTESS, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2021, NTESS
// All rights reserved.
//
// This file is part of the SST software package. For license
// information, see the LICENSE file in the top level directory of the
// distribution.

#ifndef _H_SST_CORE_RNG_EXP
#define _H_SST_CORE_RNG_EXP

#include "math.h"

#include "distrib.h"
#include "mersenne.h"

using namespace SST::RNG;

namespace SST {
namespace RNG {

/**
    \class SSTExponentialDistribution expon.h "sst/core/rng/expon.h"

    Creates an exponential distribution for use within SST. This distribution is the same across
    platforms and compilers.
*/
class SSTExponentialDistribution : public SSTRandomDistribution {

    public:
        /**
            Creates an exponential distribution with a specific lambda
            \param mn The lambda of the exponential distribution
        */
    SSTExponentialDistribution(const double mn)  :
    SSTRandomDistribution() {

        lambda = mn;
        baseDistrib = new MersenneRNG();
        deleteDistrib = true;
    }

        /**
            Creates an exponential distribution with a specific lambda and a base random number generator
            \param mn The lambda of the exponential distribution
            \param baseDist The base random number generator to take the distribution from.
        */
    SSTExponentialDistribution(const double mn, SSTRandom* baseDist)  :
    SSTRandomDistribution() {

        lambda = mn;
        baseDistrib = baseDist;
        deleteDistrib = false;
    }

        /**
            Destroys the exponential distribution
        */
    ~SSTExponentialDistribution() {
        if(deleteDistrib) {
            delete baseDistrib;
        }
    }

        /**
            Gets the next (random) double value in the distribution
            \return The next random double from the distribution
        */
    double getNextDouble()  {
        const double next = baseDistrib->nextUniform();
        return log(1 - next) / ( -1 * lambda );
    }

        /**
            Gets the lambda with which the distribution was created
            \return The lambda which the user created the distribution with
        */
    double getLambda()  {
        return lambda;
    }


    protected:
        /**
            Sets the lambda of the exponential distribution.
        */
        double lambda;
        /**
            Sets the base random number generator for the distribution.
        */
        SSTRandom* baseDistrib;

        /**
            Controls whether the base distribution should be deleted when this class is destructed.
        */
        bool deleteDistrib;

};

}
}

#endif