MFEMScalarQuadratureFunctionCoefficient.C

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
 
#include "MFEMScalarQuadratureFunctionCoefficient.h"
#include "libmesh/int_range.h"
 
MFEMScalarQuadratureFunctionCoefficient::MFEMScalarQuadratureFunctionCoefficient(
    mfem::QuadratureFunction & qf, UpdatePolicy update_policy, const std::string & name)
  : mfem::QuadratureFunctionCoefficient(qf),
    MFEMQuadratureFunctionCoefficientBase(update_policy, name),
    _source(nullptr),
    _qf(qf)
{
}
 
void
MFEMScalarQuadratureFunctionCoefficient::SetTime(mfem::real_t t)
{
  mfem::Coefficient::SetTime(t);
  MarkTimeChanged();
}
 
mfem::real_t
MFEMScalarQuadratureFunctionCoefficient::Eval(mfem::ElementTransformation & T,
                                              const mfem::IntegrationPoint & ip)
{
  if (_dirty)
    Refresh();
  CheckIntegrationRule(_qf, T, ip);
  return mfem::QuadratureFunctionCoefficient::Eval(T, ip);
}
 
void
MFEMScalarQuadratureFunctionCoefficient::Project(mfem::QuadratureFunction & qf)
{
  if (_dirty)
    Refresh();
  mfem::QuadratureFunctionCoefficient::Project(qf);
}
 
void
MFEMScalarQuadratureFunctionCoefficient::Refresh()
{
  // Equivalent to _source.Project(_qf), except performed with a caller-owned element
  // transformation: the mesh-owned shared transformation used by mfem::Coefficient::Project
  // may belong to an in-flight assembly loop that is evaluating this coefficient, and
  // projecting through it would corrupt that loop's state.
  const mfem::QuadratureSpaceBase & qspace = *_qf.GetSpace();
  const mfem::Mesh & mesh = *qspace.GetMesh();
  mfem::IsoparametricTransformation T;
  mfem::Vector values;
  _qf.HostWrite();
  for (const auto iel : make_range(qspace.GetNE()))
  {
    _qf.GetValues(iel, values);
    const mfem::IntegrationRule & ir = qspace.GetIntRule(iel);
    mesh.GetElementTransformation(iel, &T);
    for (const auto iq : make_range(ir.Size()))
    {
      const mfem::IntegrationPoint & ip = ir[iq];
      T.SetIntPoint(&ip);
      values[iq] = _source->Eval(T, ip);
    }
  }
  _dirty = false;
}
 
#endif