PCEFactory.h

Go to the documentation of this file.

#ifndef PCEFACTORY_H
#define PCEFACTORY_H
 
#include "MUQ/Approximation/PolynomialChaos/PolynomialChaosExpansion.h"
#include "MUQ/Approximation/Quadrature/Quadrature.h"
#include "MUQ/Approximation/Quadrature/FullTensorQuadrature.h"
 
#include "MUQ/Utilities/MultiIndices/MultiIndex.h"
 
#include "MUQ/Modeling/ModPiece.h"
 
#include <boost/property_tree/ptree.hpp>
 
namespace muq {
namespace Approximation {
 
  class PCEFactory {
  public:
 
    PCEFactory(std::vector<std::shared_ptr<Quadrature>>         const& quadTypesIn,
               std::vector<std::shared_ptr<IndexedScalarBasis>> const& polyTypesIn);
 
    PCEFactory(std::vector<std::shared_ptr<Quadrature>>         const& quadTypesIn,
               std::shared_ptr<muq::Utilities::MultiIndex>      const& quadOrders,
               std::vector<std::shared_ptr<IndexedScalarBasis>> const& polyTypesIn);
 
    PCEFactory(std::vector<std::shared_ptr<Quadrature>>         const& quadTypesIn,
               std::shared_ptr<muq::Utilities::MultiIndex>      const& quadOrders,
               std::vector<std::shared_ptr<IndexedScalarBasis>> const& polyTypesIn,
               std::shared_ptr<muq::Utilities::MultiIndexSet>   const& polyMultisIn);
 
 
    std::shared_ptr<PolynomialChaosExpansion> Compute(std::shared_ptr<muq::Modeling::ModPiece> const& model);
 
 
    std::shared_ptr<PolynomialChaosExpansion> Compute(std::vector<Eigen::VectorXd> const& quadEvals);
 
    std::shared_ptr<PolynomialChaosExpansion> Compute(std::vector<std::reference_wrapper<const Eigen::VectorXd>> const& quadEvals);
 
    std::shared_ptr<PolynomialChaosExpansion> Compute(std::vector<Eigen::VectorXd>                const& quadEvals,
                                                      std::shared_ptr<muq::Utilities::MultiIndex> const& quadOrders);
 
    std::shared_ptr<PolynomialChaosExpansion> Compute(std::vector<std::reference_wrapper<const Eigen::VectorXd>> const& quadEvals,
                                                      std::shared_ptr<muq::Utilities::MultiIndex> const& quadOrders);
    std::vector<Eigen::VectorXd> const& QuadPts() const{return quadPts;};
    std::vector<Eigen::VectorXd> const& QuadPts(std::shared_ptr<muq::Utilities::MultiIndex> const& quadOrders);
 
 
  protected:
 
 
    void Setup(std::shared_ptr<muq::Utilities::MultiIndex> const& quadOrders);
 
    std::shared_ptr<muq::Utilities::MultiIndex> quadOrdersCache;
 
 
    std::vector<std::shared_ptr<Quadrature>> quadTypes;
    std::vector<std::shared_ptr<IndexedScalarBasis>> polyTypes;
    FullTensorQuadrature tensQuad;
 
    std::shared_ptr<muq::Utilities::MultiIndexSet>   polyMultis;
 
    std::vector<Eigen::VectorXd> quadPts;
    Eigen::VectorXd quadWts;
 
  };
 
}
}
 
 
 
 
#endif // PCEFACTORY_H