MFEMGeometricMultigridSolver.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 "MFEMLinearSolverBase.h"
#include "MFEMFESpaceHierarchy.h"
 
class MFEMGeometricMultigridSolver : public Moose::MFEM::LinearSolverBase
{
public:
  static InputParameters validParams();
 
  MFEMGeometricMultigridSolver(const InputParameters & parameters);
 
  void ConstructSolver() override;
 
protected:
  void SetOperatorImpl(mfem::Operator & op) override;
 
private:
  mfem::AssemblyLevel ParseAssemblyLevel(const std::string & s) const;
 
  void BuildMultigrid(const mfem::Operator & op);
 
  class MGProxy : public mfem::Solver
  {
  public:
    MGProxy(MFEMGeometricMultigridSolver & owner);
 
    void SetMG(mfem::GeometricMultigrid & mg);
 
    void SetOperator(const mfem::Operator & op) override;
 
    void Mult(const mfem::Vector & x, mfem::Vector & y) const override;
 
  private:
    MFEMGeometricMultigridSolver & _owner;
 
    mfem::GeometricMultigrid * _mg = nullptr;
  };
 
  const std::string _var_name;
 
  std::shared_ptr<mfem::ParFiniteElementSpaceHierarchy> _hierarchy;
 
  const std::vector<MFEMSolverName> _smoother_names;
 
  const MFEMSolverName _coarse_solver_name;
 
  std::vector<mfem::AssemblyLevel> _assembly_levels;
 
  MGProxy * _mg_proxy = nullptr;
 
  std::vector<std::shared_ptr<mfem::ParBilinearForm>> _level_blfs;
 
  std::vector<std::unique_ptr<mfem::OperatorHandle>> _level_ops;
 
  std::unique_ptr<mfem::GeometricMultigrid> _mg;
};
 
#endif