ProblemOperatorBase.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
 
#ifdef MOOSE_MFEM_ENABLED
 
#pragma once
 
#include "PerfGraphInterface.h"
#include "libmesh/ignore_warnings.h"
#include "mfem.hpp"
#include "libmesh/restore_warnings.h"
#include <functional>
 
class MFEMProblem;
struct MFEMProblemData;
 
namespace Moose::MFEM
{
 
class ProblemOperatorBase : public PerfGraphInterface
{
public:
  ProblemOperatorBase(MFEMProblem & problem);
  virtual ~ProblemOperatorBase() = default;
 
  virtual void SetGridFunctions();
  virtual void SetTrialVariablesFromTrueVectors();
  virtual void Init();
  virtual void Solve() = 0;
 
  mfem::Array<int> _block_true_offsets_test;
  mfem::Array<int> _block_true_offsets_trial;
 
  mfem::BlockVector _true_x, _true_rhs;
 
  mfem::BlockVector _true_solution;
 
protected:
  void SolveWithOperator(mfem::Operator & system_operator,
                         mfem::Operator & linear_operator,
                         const mfem::Vector & rhs,
                         mfem::Vector & x);
 
  void
  SolveWithOperator(mfem::Operator & system_operator, const mfem::Vector & rhs, mfem::Vector & x)
  {
    return SolveWithOperator(system_operator, system_operator.GetGradient(x), rhs, x);
  }
 
  MFEMProblem & _problem;
  MFEMProblemData & _problem_data;
 
  std::vector<std::string> _trial_var_names;
  std::vector<std::string> _test_var_names;
  std::vector<mfem::ParGridFunction *> _trial_variables;
  std::vector<mfem::ParGridFunction *> _test_variables;
  mfem::Vector * _trial_true_vector = nullptr;
};
}
 
#endif