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 "MFEMProblem.h"
#include <functional>

namespace Moose::MFEM
{

class ProblemOperatorBase
{
public:
  ProblemOperatorBase(MFEMProblem & problem);
  virtual ~ProblemOperatorBase() = default;

  virtual void SetGridFunctions();
  virtual void SetTrialVariablesFromTrueVectors();
  virtual void Init(mfem::BlockVector & X);
  virtual void Solve() = 0;

  mfem::Array<int> _block_true_offsets_test;
  mfem::Array<int> _block_true_offsets_trial;

  mfem::BlockVector _true_x, _true_rhs;

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