SamplingProblem.h

Go to the documentation of this file.

#ifndef SAMPLINGPROBLEM_H_
#define SAMPLINGPROBLEM_H_
 
// include Density and not ModPiece so that if a SamplingProblem is constructed with a Density the compiler knows it is a child of ModPiece
#include "MUQ/Modeling/Distributions/Density.h"
#include "MUQ/SamplingAlgorithms/AbstractSamplingProblem.h"
 
namespace muq {
  namespace SamplingAlgorithms {
 
    class SamplingProblem : public AbstractSamplingProblem{
    public:
 
      SamplingProblem(std::shared_ptr<muq::Modeling::ModPiece> const& targetIn);
 
      SamplingProblem(std::shared_ptr<muq::Modeling::ModPiece> const& targetIn,
                      std::shared_ptr<muq::Modeling::ModPiece> const& qoiIn);
 
      virtual ~SamplingProblem() = default;
 
      virtual std::shared_ptr<AbstractSamplingProblem> Clone() const override{return std::make_shared<SamplingProblem>(target, qoi);};
 
      virtual double LogDensity(std::shared_ptr<SamplingState> const& state) override;
 
      virtual Eigen::VectorXd GradLogDensity(std::shared_ptr<SamplingState> const& state,
                                             unsigned                       const  blockWrt) override;
 
      std::shared_ptr<muq::Modeling::ModPiece> GetDistribution(){return target;};
 
      virtual std::shared_ptr<SamplingState> QOI() override;
 
    protected:
 
      std::shared_ptr<muq::Modeling::ModPiece> target;
 
      std::shared_ptr<muq::Modeling::ModPiece> qoi;
 
      std::shared_ptr<SamplingState> lastState;
 
    private:
 
      static unsigned GetNumBlocks(std::shared_ptr<muq::Modeling::ModPiece> const& target);
      static std::vector<int> GetBlockSizes(std::shared_ptr<muq::Modeling::ModPiece> const& target);
    };
  } // namespace SamplingAlgorithms
} // namespace muq
 
#endif