ComplexEquationSystemProblemOperator.C

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
 
#include "ComplexEquationSystemProblemOperator.h"
#include "MFEMProblem.h"
 
namespace Moose::MFEM
{
ComplexEquationSystemProblemOperator::ComplexEquationSystemProblemOperator(
    MFEMProblem & problem, const std::string & weak_form_name)
  : EquationSystemProblemOperator(problem, weak_form_name),
    _equation_system{std::dynamic_pointer_cast<Moose::MFEM::ComplexEquationSystem>(
        problem.getEquationSystem(weak_form_name))}
{
  if (!_equation_system)
    mooseError("The weak form supplying this operator does not provide a ComplexEquationSystem, "
               "which is required by ComplexEquationSystemProblemOperator.");
}
 
void
ComplexEquationSystemProblemOperator::SetGridFunctions()
{
  _trial_var_names = GetEquationSystem()->GetTrialVarNames();
  _test_var_names = GetEquationSystem()->GetTestVarNames();
 
  _cmplx_trial_variables = _problem_data.cmplx_gridfunctions.Get(_trial_var_names);
  _cmplx_test_variables = _problem_data.cmplx_gridfunctions.Get(_test_var_names);
 
  // Set operator size and block structure for trial spaces
  _block_true_offsets_trial.SetSize(_cmplx_trial_variables.size() + 1);
  _block_true_offsets_trial[0] = 0;
  for (const auto ind : index_range(_cmplx_trial_variables))
  {
    _block_true_offsets_trial[ind + 1] =
        2 * _cmplx_trial_variables.at(ind)->ParFESpace()->TrueVSize();
  }
  _block_true_offsets_trial.PartialSum();
 
  // Set operator size and block structure for test spaces
  _block_true_offsets_test.SetSize(_cmplx_test_variables.size() + 1);
  _block_true_offsets_test[0] = 0;
  for (const auto ind : index_range(_cmplx_test_variables))
  {
    _block_true_offsets_test[ind + 1] =
        2 * _cmplx_test_variables.at(ind)->ParFESpace()->TrueVSize();
  }
  _block_true_offsets_test.PartialSum();
 
  _true_x.Update(_block_true_offsets_trial);
  _true_rhs.Update(_block_true_offsets_test);
 
  width = _block_true_offsets_trial[_cmplx_trial_variables.size()];
  height = _block_true_offsets_test[_cmplx_test_variables.size()];
}
 
void
ComplexEquationSystemProblemOperator::Solve()
{
  auto * const es = GetEquationSystem();
 
  {
    TIME_SECTION("ComplexEquationSystemProblemOperator::FormSystem", 2, "Assembling MFEM System");
    FormEquationSystemOperator();
  }
 
  {
    TIME_SECTION("ComplexEquationSystemProblemOperator::SolveSystem", 2, "Solving MFEM System");
    SolveWithOperator(*es, _true_rhs, _true_x);
  }
 
  es->SetTrialVariablesFromTrueVectors(_true_x);
}
 
} // namespace Moose::MFEM
 
#endif