ProblemInstat.hpp

#pragma once
 
#include <string>
 
#include <amdis/ProblemInstatBase.hpp>
#include <amdis/ProblemStat.hpp>
#include <amdis/common/TypeTraits.hpp>
 
#include <dune/grid/yaspgrid.hh>
 
namespace AMDiS
{
  // forward declarations
  class AdaptInfo;
 
  template <class Traits>
  class ProblemInstat
      : public ProblemInstatBase
  {
    using Self        = ProblemInstat;
    using ProblemType = ProblemStat<Traits>;
 
    using SolutionVector = typename ProblemType::SolutionVector;
 
  public:
    ProblemInstat(std::string const& name, ProblemType& prob)
      : ProblemInstatBase(name)
      , problemStat_(&prob)
    {}
 
    ProblemInstat(std::string const& name, ProblemType& prob, ProblemStatBase& initialProb)
      : ProblemInstatBase(name, initialProb)
      , problemStat_(&prob)
    {}
 
    void initialize(Flag initFlag = INIT_NOTHING);
 
    void initTimestep(AdaptInfo& adaptInfo) override;
 
    void closeTimestep(AdaptInfo& adaptInfo) override;
 
    ProblemType&       problemStat()       { return *problemStat_; }
    ProblemType const& problemStat() const { return *problemStat_; }
 
    std::shared_ptr<SolutionVector const> oldSolutionVector() const
    {
      test_exit_dbg(bool(oldSolution_),
        "OldSolution need to be created. Call initialize with INIT_UH_OLD.");
      return oldSolution_;
    }
 
    std::shared_ptr<SolutionVector> oldSolutionVector()
    {
      test_exit_dbg(bool(oldSolution_),
        "OldSolution need to be created. Call initialize with INIT_UH_OLD.");
      return oldSolution_;
    }
 
 
    template <class Range = void, class... Indices>
    auto oldSolution(Indices... ii) const
    {
      test_exit_dbg(bool(oldSolution_),
        "OldSolution need to be created. Call initialize with INIT_UH_OLD.");
      return valueOf<Range>(*oldSolution_, ii...);
    }
 
    void transferInitialSolution(AdaptInfo& adaptInfo) override;
 
  protected:
    void createUhOld();
 
  protected:
    ProblemType* problemStat_;
 
    std::shared_ptr<SolutionVector> oldSolution_;
  };
 
 
  // Deduction guides
  template <class Traits>
  ProblemInstat(std::string const& name, ProblemStat<Traits>& prob)
    -> ProblemInstat<Traits>;
 
  template <class Traits>
  ProblemInstat(std::string const& name, ProblemStat<Traits>& prob, ProblemStatBase& initialProb)
    -> ProblemInstat<Traits>;
 
 
  // mark template as explicitly instantiated in cpp file
  extern template class ProblemInstat<LagrangeBasis<Dune::YaspGrid<2>,1>>;
  extern template class ProblemInstat<LagrangeBasis<Dune::YaspGrid<2>,1,1>>;
 
} // end namespace AMDiS
 
#include "ProblemInstat.inc.hpp"