all_codim_mesh_data_set.h

 
#ifndef INCGdf4311acf6554f11919b7b1edfc5b3dd
#define INCGdf4311acf6554f11919b7b1edfc5b3dd
 
#include <lf/mesh/mesh.h>
 
#include <boost/container/vector.hpp>
#include <concepts>
 
#include "lf/base/lf_assert.h"
#include "mesh_data_set.h"
 
namespace lf::mesh::utils {

template <class T>
class AllCodimMeshDataSet : public MeshDataSet<T> {
 public:
  using size_type = Mesh::size_type;
  using dim_t = base::RefEl::dim_t;
 
  AllCodimMeshDataSet(const AllCodimMeshDataSet&) = delete;
  AllCodimMeshDataSet(AllCodimMeshDataSet&&) noexcept = default;
  AllCodimMeshDataSet& operator=(const AllCodimMeshDataSet&) = delete;
  AllCodimMeshDataSet& operator=(AllCodimMeshDataSet&&) noexcept = default;
  ~AllCodimMeshDataSet() override = default;

  [[nodiscard]] T& operator()(const Entity& e) {
    LF_ASSERT_MSG(DefinedOn(e), "MeshDataSet is not defined on this entity.");
    return data_[e.Codim()][mesh_->Index(e)];
  }
  // NOLINTNEXTLINE(readability-const-return-type)
  [[nodiscard]] const T& operator()(const Entity& e) const override {
    LF_ASSERT_MSG(DefinedOn(e), "MeshDataSet is not defined on this entity.");
    return data_[e.Codim()][mesh_->Index(e)];
  }
  [[nodiscard]] bool DefinedOn(const Entity& e) const override {
    return mesh_->Contains(e);
  }

  explicit AllCodimMeshDataSet(
      const std::shared_ptr<const lf::mesh::Mesh>& mesh)
      : MeshDataSet<T>(),
        dim_mesh_(mesh->DimMesh()),
        mesh_(mesh),
        data_(dim_mesh_ + 1) {
    for (dim_t codim = 0; codim <= dim_mesh_; ++codim) {
      data_[codim].resize(mesh_->NumEntities(codim));
    }
  }

  template <class X,
            class = typename std::enable_if_t<std::is_convertible_v<X, T> &&
                                              std::is_copy_constructible_v<X>>>
  AllCodimMeshDataSet(const std::shared_ptr<const lf::mesh::Mesh>& mesh,
                      X init_value)
      : MeshDataSet<T>(),
        dim_mesh_(mesh->DimMesh()),
        mesh_(mesh),
        data_(dim_mesh_ + 1) {
    for (dim_t codim = 0; codim <= dim_mesh_; ++codim) {
      data_[codim] =
          boost::container::vector<T>(mesh_->NumEntities(codim), init_value);
    }
  }
 
 private:
  dim_t dim_mesh_;
  std::shared_ptr<const lf::mesh::Mesh> mesh_;
 
  // Note(craffael): We use boost container here which doesn't have the ridicule
  // specialization that std::vector<bool> has. In this way we can actually
  // return references in operator() const.
  boost::container::vector<boost::container::vector<T>> data_;
};

template <class T>
std::shared_ptr<AllCodimMeshDataSet<T>> make_AllCodimMeshDataSet(
    const std::shared_ptr<const lf::mesh::Mesh>& mesh) {
  using impl_t = AllCodimMeshDataSet<T>;
  auto t = new AllCodimMeshDataSet<T>(mesh);
  return std::shared_ptr<impl_t>(t);
}

template <std::copy_constructible T>
std::shared_ptr<AllCodimMeshDataSet<T>> make_AllCodimMeshDataSet(
    const std::shared_ptr<const lf::mesh::Mesh>& mesh, T init_value) {
  using impl_t = AllCodimMeshDataSet<T>;
  return std::make_shared<impl_t>(mesh, init_value);
}
 
}  // namespace lf::mesh::utils
 
#endif  // INCGdf4311acf6554f11919b7b1edfc5b3dd