MFEMVectorQuadratureFunctionCoefficient.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 "MFEMVectorQuadratureFunctionCoefficient.h"
#include "libmesh/int_range.h"

MFEMVectorQuadratureFunctionCoefficient::MFEMVectorQuadratureFunctionCoefficient(
    mfem::VectorCoefficient & source,
    mfem::QuadratureFunction & qf,
    UpdatePolicy update_policy,
    const std::string & name)
  : mfem::VectorQuadratureFunctionCoefficient(qf),
    MFEMQuadratureFunctionCoefficientBase(update_policy, name),
    _source(source),
    _qf(qf)
{
}

void
MFEMVectorQuadratureFunctionCoefficient::SetTime(mfem::real_t t)
{
  mfem::VectorCoefficient::SetTime(t);
  MarkTimeChanged();
}

void
MFEMVectorQuadratureFunctionCoefficient::Eval(mfem::Vector & V,
                                              mfem::ElementTransformation & T,
                                              const mfem::IntegrationPoint & ip)
{
  if (_dirty)
    Refresh();
  CheckIntegrationRule(_qf, T, ip);
  mfem::VectorQuadratureFunctionCoefficient::Eval(V, T, ip);
}

void
MFEMVectorQuadratureFunctionCoefficient::Project(mfem::QuadratureFunction & qf)
{
  if (_dirty)
    Refresh();
  mfem::VectorQuadratureFunctionCoefficient::Project(qf);
}

void
MFEMVectorQuadratureFunctionCoefficient::Refresh()
{
  // Equivalent to _source.Project(_qf), except performed with a caller-owned element
  // transformation: the mesh-owned shared transformation used by mfem::VectorCoefficient::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::DenseMatrix values;
  mfem::Vector col;
  _qf.HostWrite();
  for (const auto iel : libMesh::make_range(qspace.GetNE()))
  {
    _qf.GetValues(iel, values);
    const mfem::IntegrationRule & ir = qspace.GetIntRule(iel);
    mesh.GetElementTransformation(iel, &T);
    for (const auto iq : libMesh::make_range(ir.Size()))
    {
      const mfem::IntegrationPoint & ip = ir[iq];
      T.SetIntPoint(&ip);
      values.GetColumnReference(iq, col);
      _source.Eval(col, T, ip);
    }
  }
  _dirty = false;
}

#endif