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