MFEMContainers.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 MFEM_ENABLED

#pragma once
#include <map>
#include <set>
#include <string>
#include <memory>
#include <vector>
#include "libmesh/ignore_warnings.h"
#include <mfem.hpp>
#include "libmesh/restore_warnings.h"

namespace Moose::MFEM
{

template <typename T>
class NamedFieldsMap
{
public:
  using MapType = std::map<std::string, std::shared_ptr<T>>;
  using const_iterator = typename MapType::const_iterator;

  NamedFieldsMap() = default;

  ~NamedFieldsMap() { DeregisterAll(); }

  template <class FieldType, class... FieldArgs>
  void Register(const std::string & field_name, FieldArgs &&... args)
  {
    Register(field_name, std::make_shared<FieldType>(std::forward<FieldArgs>(args)...));
  }

  void Register(const std::string & field_name, std::shared_ptr<T> field)
  {
    CheckFieldIsRegistrable(field_name, field.get());

    Deregister(field_name);

    _field_map[field_name] = std::move(field);
  }

  void Deregister(const std::string & field_name) { _field_map.erase(field_name); }

  [[nodiscard]] inline bool Has(const std::string & field_name) const
  {
    return FindField(field_name) != end();
  }

  [[nodiscard]] inline std::shared_ptr<T> GetShared(const std::string & field_name) const
  {
    CheckFieldIsRegistered(field_name);

    auto it = FindField(field_name);

    return EnsureFieldPointerIsNonNull(it);
  }

  [[nodiscard]] inline T & GetRef(const std::string & field_name) const
  {
    return *GetShared(field_name);
  }

  [[nodiscard]] inline T * Get(const std::string & field_name) const
  {
    return GetShared(field_name).get();
  }

  template <typename TDerived>
  [[nodiscard]] inline TDerived * Get(const std::string & field_name) const
  {
    auto ptr = Get(field_name);

    return EnsurePointerCastIsNonNull<TDerived>(ptr);
  }

  [[nodiscard]] std::vector<T *> Get(const std::vector<std::string> & keys) const
  {
    std::vector<T *> values;

    for (const auto & key : keys)
    {
      values.push_back(Get(key));
    }

    values.shrink_to_fit();
    return values;
  }

  // NOLINTNEXTLINE(readability-identifier-naming)
  [[nodiscard]] inline const_iterator begin() const { return _field_map.begin(); }

  // NOLINTNEXTLINE(readability-identifier-naming)
  [[nodiscard]] inline const_iterator end() const { return _field_map.end(); }

  // Returns the number of elements in the map
  int size() { return _field_map.size(); }

protected:
  [[nodiscard]] inline const_iterator FindField(const std::string & field_name) const
  {
    return _field_map.find(field_name);
  }

  void CheckFieldIsRegistrable(const std::string & field_name, T * field) const
  {
    if (!field)
    {
      MFEM_ABORT("Cannot register NULL field with name '" << field_name << "'.");
    }

    CheckForDoubleRegistration(field_name, field);
  }

  void CheckForDoubleRegistration(const std::string & field_name, T * field) const
  {
    if (Has(field_name) && Get(field_name) == field)
    {
      MFEM_ABORT("The field '" << field_name << "' is already registered.");
    }
  }

  void CheckFieldIsRegistered(const std::string & field_name) const
  {
    if (!Has(field_name))
    {
      MFEM_ABORT("The field '" << field_name << "' has not been registered.");
    }
  }

  inline std::shared_ptr<T> EnsureFieldPointerIsNonNull(const_iterator & iterator) const
  {
    auto owned_ptr = iterator->second;

    if (!owned_ptr)
    {
      MFEM_ABORT("The field '" << iterator->first << "' is NULL.");
    }

    return owned_ptr;
  }

  template <typename TDerived>
  inline TDerived * EnsurePointerCastIsNonNull(T * ptr) const
  {
    auto derived_ptr = dynamic_cast<TDerived *>(ptr);

    if (!derived_ptr)
    {
      MFEM_ABORT("The dynamic cast performed on the field pointer failed.");
    }

    return derived_ptr;
  }

  void DeregisterAll() { _field_map.clear(); }

private:
  MapType _field_map{};
};

inline std::string
GetTimeDerivativeName(std::string name)
{
  return std::string("d") + name + std::string("_dt");
}

using FECollections = Moose::MFEM::NamedFieldsMap<mfem::FiniteElementCollection>;
using FESpaces = Moose::MFEM::NamedFieldsMap<mfem::ParFiniteElementSpace>;
using GridFunctions = Moose::MFEM::NamedFieldsMap<mfem::ParGridFunction>;
using SubMeshes = Moose::MFEM::NamedFieldsMap<mfem::ParSubMesh>;

} // namespace Moose::MFEM

#endif