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"
#include "libmesh/ignore_warnings.h"
#include "mfem/fem/multigrid.hpp"
#include "libmesh/restore_warnings.h"

class MFEMGeometricMultigridSolver : public Moose::MFEM::LinearSolverBase
{
public:
  static InputParameters validParams();

  MFEMGeometricMultigridSolver(const InputParameters & parameters);

  void ConstructSolver() override;

  void SetOperator(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