MFEMHypreAME.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 "MFEMEigensolverBase.h"

class MFEMHypreAME : public Moose::MFEM::EigensolverBase
{
public:
  static InputParameters validParams();

  MFEMHypreAME(const InputParameters & parameters);

  virtual void SetMassMatrix(mfem::Operator & mass) override
  {
    _eigensolver->SetMassMatrix(libMesh::cast_ref<mfem::HypreParMatrix &>(mass));
  }

  virtual void Solve() override { _eigensolver->Solve(); }

  virtual void GetEigenvalues(mfem::Array<mfem::real_t> & eigenvalues) const override
  {
    _eigensolver->GetEigenvalues(eigenvalues);
  }

  virtual const mfem::HypreParVector & GetEigenvector(int index) const override
  {
    return _eigensolver->GetEigenvector(index);
  }

  virtual void ConstructSolver() override;

protected:
  virtual void SetOperatorImpl(mfem::Operator & op) override
  {
    if (_preconditioner)
      _preconditioner->SetOperator(op);
    _eigensolver->SetOperator(libMesh::cast_ref<mfem::HypreParMatrix &>(op));
  }

  std::unique_ptr<mfem::HypreAME> _eigensolver;
};

#endif