Gamma.h

Go to the documentation of this file.

#ifndef GAMMA_H_
#define GAMMA_H_
 
#include "MUQ/Modeling/Distributions/Distribution.h"
 
namespace muq {
  namespace Modeling {
 
    class Gamma : public Distribution {
    public:
 
      Gamma(double       alphaIn,
            double       betaIn);
 
      Gamma(Eigen::VectorXd const& alphaIn,
            Eigen::VectorXd const& betaIn);
 
      virtual ~Gamma() = default;
 
      static std::shared_ptr<Gamma> FromMoments(Eigen::VectorXd const& mean,
                                                Eigen::VectorXd const& var);
 
      const Eigen::VectorXd alpha;
      const Eigen::VectorXd beta;
 
    private:
 
      static double ComputeConstant(Eigen::VectorXd const& alphaIn,
                                    Eigen::VectorXd const& betaIn);
 
      const double logConst; // sum( log( beta^alpha / Gamma(alpha) ) )
 
      virtual double LogDensityImpl(ref_vector<Eigen::VectorXd> const& inputs) override;
 
      virtual Eigen::VectorXd GradLogDensity(unsigned int wrt, 
                                             ref_vector<Eigen::VectorXd> const& inputs) override;
 
      virtual Eigen::VectorXd ApplyLogDensityHessian(unsigned int                const  inWrt1,
                                                     unsigned int                const  inWrt2,
                                                     ref_vector<Eigen::VectorXd> const& input,
                                                     Eigen::VectorXd             const& vec) override;
 
      virtual Eigen::VectorXd SampleImpl(ref_vector<Eigen::VectorXd> const& inputs) override;
 
    };
  } // namespace Modeling
} // namespace muq
 
#endif