PolynomialChaosReporter.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#pragma once
 
#include "GeneralReporter.h"
#include "SurrogateModelInterface.h"
 
#include "PolynomialChaos.h"
#include "nlohmann/json.h"
 
class PolynomialChaosReporter : public GeneralReporter, public SurrogateModelInterface
{
public:
  static InputParameters validParams();
  PolynomialChaosReporter(const InputParameters & parameters);
 
  virtual void initialize() override {}
  virtual void execute() override;
  virtual void finalize() override {}
 
private:
  static std::vector<Real> computeLocalSensitivity(const PolynomialChaos & pc,
                                                   const std::vector<Real> & data);
 
  const std::vector<std::vector<Real>> & _loc_point;
  std::vector<Sampler *> _loc_sampler;
 
  std::vector<const PolynomialChaos *> _pc;
 
  std::vector<std::vector<std::vector<Real>> *> _loc_point_sense;
  std::vector<std::vector<std::vector<Real>> *> _loc_sampler_sense;
};
 
void to_json(nlohmann::json & json, const PolynomialChaos * const & pc);
 
template <>
inline void
dataStore(std::ostream &, const PolynomialChaos *&, void *)
{
}
template <>
inline void
dataLoad(std::istream &, const PolynomialChaos *&, void *)
{
}
 
template <typename OutType>
class PCStatisticsContext : public ReporterGeneralContext<std::pair<OutType, std::vector<OutType>>>
{
public:
  PCStatisticsContext(const libMesh::ParallelObject & other,
                      const MooseObject & producer,
                      ReporterState<std::pair<OutType, std::vector<OutType>>> & state,
                      const PolynomialChaos & pc,
                      const MooseEnumItem & stat);
 
  virtual void finalize() override;
  virtual void storeInfo(nlohmann::json & json) const override;
 
private:
  const PolynomialChaos & _pc;
  const MooseEnumItem _stat;
};
 
template <typename OutType>
class PCSobolContext : public ReporterGeneralContext<
                           std::pair<std::vector<OutType>, std::vector<std::vector<OutType>>>>
{
public:
  PCSobolContext(
      const libMesh::ParallelObject & other,
      const MooseObject & producer,
      ReporterState<std::pair<std::vector<OutType>, std::vector<std::vector<OutType>>>> & state,
      const PolynomialChaos & pc);
 
  virtual void finalize() override;
  virtual std::string type() const override
  {
    return "SobolIndices<" + MooseUtils::prettyCppType<OutType>() + ">";
  }
 
protected:
  virtual void store(nlohmann::json & json) const override;
 
private:
  const PolynomialChaos & _pc;
};