ScaleIntegrator.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 "libmesh/ignore_warnings.h"
#include "mfem/miniapps/common/pfem_extras.hpp"
#include "libmesh/restore_warnings.h"
 
#include "MooseError.h"
 
namespace Moose::MFEM
{
 
class ScaleIntegrator : public mfem::BilinearFormIntegrator
{
private:
  mfem::BilinearFormIntegrator * _integrator{nullptr};
  mfem::real_t _scale;
  bool _own_integrator;
 
public:
  ScaleIntegrator(mfem::BilinearFormIntegrator * integ)
    : _integrator{integ}, _scale{1}, _own_integrator{true}
  {
  }
  ScaleIntegrator(mfem::BilinearFormIntegrator * integ, mfem::real_t scale)
    : _integrator{integ}, _scale{scale}, _own_integrator{true}
  {
  }
  ScaleIntegrator(mfem::BilinearFormIntegrator * integ, mfem::real_t scale, bool own)
    : _integrator{integ}, _scale{scale}, _own_integrator{own}
  {
  }
 
  void SetIntegrator(mfem::BilinearFormIntegrator * integ)
  {
    if (_integrator && _own_integrator)
    {
      delete _integrator;
    }
 
    _integrator = integ;
  }
 
  void SetScale(mfem::real_t scale) { _scale = scale; }
 
  void SetOwn(bool own) { _own_integrator = own; }
 
  void CheckIntegrator()
  {
    if (!_integrator)
      mooseError("Integrator not set");
  }
 
  virtual void SetIntRule(const mfem::IntegrationRule * ir) override;
 
  virtual void AssembleElementMatrix(const mfem::FiniteElement & el,
                                     mfem::ElementTransformation & Trans,
                                     mfem::DenseMatrix & elmat) override;
  virtual void AssembleElementMatrix2(const mfem::FiniteElement & trial_fe,
                                      const mfem::FiniteElement & test_fe,
                                      mfem::ElementTransformation & Trans,
                                      mfem::DenseMatrix & elmat) override;
 
  using mfem::BilinearFormIntegrator::AssembleFaceMatrix;
  virtual void AssembleFaceMatrix(const mfem::FiniteElement & el1,
                                  const mfem::FiniteElement & el2,
                                  mfem::FaceElementTransformations & Trans,
                                  mfem::DenseMatrix & elmat) override;
 
  using mfem::BilinearFormIntegrator::AssemblePA;
  virtual void AssemblePA(const mfem::FiniteElementSpace & fes) override;
 
  virtual void AssembleDiagonalPA(mfem::Vector & diag) override;
 
  virtual void AssemblePAInteriorFaces(const mfem::FiniteElementSpace & fes) override;
 
  virtual void AssemblePABoundaryFaces(const mfem::FiniteElementSpace & fes) override;
 
  virtual void AddMultTransposePA(const mfem::Vector & x, mfem::Vector & y) const override;
 
  virtual void AddMultPA(const mfem::Vector & x, mfem::Vector & y) const override;
 
  virtual void
  AssembleEA(const mfem::FiniteElementSpace & fes, mfem::Vector & emat, const bool add) override;
 
  virtual void AssembleEABoundary(const mfem::FiniteElementSpace & fes,
                                  mfem::Vector & emat,
                                  const bool add) override;
 
  virtual void AssembleMF(const mfem::FiniteElementSpace & fes) override;
 
  virtual void AddMultMF(const mfem::Vector & x, mfem::Vector & y) const override;
 
  virtual void AssembleDiagonalMF(mfem::Vector & diag) override;
 
  virtual ~ScaleIntegrator();
};
} // namespace Moose::MFEM
 
#endif