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