MFEMBlockRestrictable.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 "MFEMBlockRestrictable.h"
 
InputParameters
MFEMBlockRestrictable::validParams()
{
  // Create InputParameters object that will be appended to the parameters for the inheriting object
  InputParameters params = emptyInputParameters();
  params.addParam<std::vector<SubdomainName>>(
      "block",
      {},
      "The list of subdomains (names or ids) that this object will be restricted to. "
      "Leave empty to apply to all subdomains.");
  return params;
}
 
MFEMBlockRestrictable::MFEMBlockRestrictable(const InputParameters & parameters,
                                             const mfem::ParMesh & mfem_mesh)
  : _mfem_mesh(mfem_mesh),
    _subdomain_names(parameters.get<std::vector<SubdomainName>>("block")),
    _subdomain_attributes(subdomainsToAttributes(parameters)),
    _subdomain_markers(_mfem_mesh.attributes.IsEmpty()
                           ? mfem::Array<int>()
                           : mfem::AttributeSets::AttrToMarker(_mfem_mesh.attributes.Max(),
                                                               _subdomain_attributes.IsEmpty()
                                                                   ? _mfem_mesh.attributes
                                                                   : _subdomain_attributes))
{
}
 
mfem::Array<int>
MFEMBlockRestrictable::subdomainsToAttributes(const InputParameters & parameters)
{
  mfem::Array<int> attributes;
  auto & mesh = getMesh();
 
  for (const SubdomainName & subdomain_name : _subdomain_names)
  {
    try
    {
      // Is this a block ID?
      const int attribute_id = std::stoi(subdomain_name);
      attributes.Append(attribute_id);
    }
    catch (...)
    {
      // It was not
      if (!mesh.attribute_sets.AttributeSetExists(subdomain_name))
        parameters.paramError("block", "the mesh has no block named '", subdomain_name, "'.");
      auto & subdomain_ids = mesh.attribute_sets.GetAttributeSet(subdomain_name);
      for (const auto & subdomain_id : subdomain_ids)
        attributes.Append(subdomain_id);
    }
  }
 
  for (const auto attribute : attributes)
    if (mesh.attributes.Find(attribute) == -1)
      parameters.paramError("block", "the mesh has no block with ID ", attribute, ".");
  return attributes;
}
 
std::vector<std::string>
MFEMBlockRestrictable::subdomainsToStrings()
{
  const auto & attrs = _subdomain_attributes;
  std::vector<std::string> strs(attrs.Size());
  std::transform(attrs.begin(), attrs.end(), strs.begin(), [](int n) { return std::to_string(n); });
  return strs;
}
 
#endif