MFEMDataCollection.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 "MFEMDataCollection.h"

InputParameters
MFEMDataCollection::validParams()
{
  InputParameters params = FileOutput::validParams();
  params.addClassDescription("Output for controlling MFEMDataCollection inherited data.");
  params.addParam<std::string>("submesh",
                               "Submesh to output variables on. Leave blank to use base mesh.");
  return params;
}

MFEMDataCollection::MFEMDataCollection(const InputParameters & parameters)
  : FileOutput(parameters),
    _problem_data(static_cast<MFEMProblem *>(_problem_ptr)->getProblemData()),
    _pmesh(parameters.isParamValid("submesh")
               ? _problem_data.submeshes.GetRef(getParam<std::string>("submesh"))
               : const_cast<mfem::ParMesh &>(*_problem_data.pmesh.get()))
{
}

void
MFEMDataCollection::registerFields()
{
  mfem::DataCollection & dc(getDataCollection());
  for (auto const & [gf_name, gf_ptr] : _problem_data.gridfunctions)
  {
    if (dc.GetMesh() == gf_ptr->FESpace()->GetMesh())
      dc.RegisterField(gf_name, gf_ptr.get());
    else
      mooseInfo("The variable ",
                gf_name,
                " is not defined on the same mesh as the output DataCollection.");
  }
}

void
MFEMDataCollection::output()
{
  mfem::DataCollection & dc(getDataCollection());
  // Write fields to disk
  dc.SetCycle(getFileNumber());
  dc.SetTime(time());
  dc.Save();
  _file_num++;
}

#endif