codim_mesh_data_set.h

#ifndef INCGbae24f2390174bff85f65c2c2e558a9e
#define INCGbae24f2390174bff85f65c2c2e558a9e
 
#include <boost/container/vector.hpp>
#include <utility>
 
#include "mesh_data_set.h"
 
namespace lf::mesh::utils {

template <class T>
class CodimMeshDataSet : public MeshDataSet<T> {
 public:
  using size_type = Mesh::size_type;
  using dim_t = base::RefEl::dim_t;

  CodimMeshDataSet(std::shared_ptr<const Mesh> mesh, dim_t codim)
      : MeshDataSet<T>(),
        mesh_(std::move(mesh)),
        data_(mesh_->NumEntities(codim)),
        codim_(codim) {}

  CodimMeshDataSet(std::shared_ptr<const Mesh> mesh, dim_t codim, T init)
    requires std::copy_constructible<T>
      : MeshDataSet<T>(),
        mesh_(std::move(mesh)),
        data_(mesh_->NumEntities(codim), init),
        codim_(codim) {}

  [[nodiscard]] T& operator()(const Entity& e) {
    LF_ASSERT_MSG(DefinedOn(e), "MeshDataSet is not defined on this entity.");
    return data_[mesh_->Index(e)];
  }
  // NOLINTNEXTLINE(readability-const-return-type)
  [[nodiscard]] const T& operator()(const Entity& e) const override {
    LF_ASSERT_MSG(DefinedOn(e), "MeshDataSet not defined on this entity.");
    return data_[mesh_->Index(e)];
  }
  [[nodiscard]] bool DefinedOn(const Entity& e) const override {
    return e.Codim() == codim_ && mesh_->Contains(e);
  }
 
 private:
  std::shared_ptr<const Mesh> mesh_;
  boost::container::vector<T> data_;
  dim_t codim_;
};

template <class T>
std::shared_ptr<CodimMeshDataSet<T>> make_CodimMeshDataSet(
    const std::shared_ptr<const lf::mesh::Mesh>& mesh, base::dim_t codim) {
  using impl_t = CodimMeshDataSet<T>;
  return std::shared_ptr<impl_t>(new impl_t(mesh, codim));
}

template <std::copy_constructible T>
std::shared_ptr<CodimMeshDataSet<T>> make_CodimMeshDataSet(
    const std::shared_ptr<const Mesh>& mesh, base::dim_t codim, T init) {
  using impl_t = CodimMeshDataSet<T>;
  return std::shared_ptr<impl_t>(new impl_t(mesh, codim, init));
}
 
}  // namespace lf::mesh::utils
 
#endif  // INCGbae24f2390174bff85f65c2c2e558a9e