MFEMSolverBase.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 "MFEMObject.h"
 
namespace Moose::MFEM
{
class SolverBase : public MFEMObject
{
public:
  static InputParameters validParams();
 
  SolverBase(const InputParameters & parameters);
 
  mfem::Solver & GetSolver();
 
  void SetSolver(mfem::Solver * solver) { _solver.reset(solver); }
 
  virtual void ConstructSolver() = 0;
 
  void SetOperator(mfem::Operator & op);
 
  void Mult(const mfem::Vector & rhs, mfem::Vector & x) { GetSolver().Mult(rhs, x); }
 
protected:
  virtual void UpdateEquationSystemContext() {}
 
  virtual void SetOperatorImpl(mfem::Operator & op) { GetSolver().SetOperator(op); }
 
  std::unique_ptr<mfem::Solver> _solver;
};
 
inline mfem::Solver &
SolverBase::GetSolver()
{
  mooseAssert(_solver, "Attempting to retrieve solver before it's been constructed");
  return *_solver;
}
} // namespace Moose::MFEM
 
#endif