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 <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;
};