identity.hh

Go to the documentation of this file.

#ifndef DUNE_PDELAB_COMMON_PARTITION_IDENTITY_HH
#define DUNE_PDELAB_COMMON_PARTITION_IDENTITY_HH
 
#include <dune/pdelab/common/partition/halo/region.hh>
 
#include <dune/grid/concepts/gridview.hh>
#include <dune/grid/common/gridenums.hh>
#include <dune/grid/common/partitionset.hh>
 
#include <dune/common/iteratorrange.hh>
 
#include <type_traits>
#include <array>
#include <memory>
 
namespace Dune::PDELab::EntitySetPartition {
 
template<Dune::Concept::GridView ES>
class Identity {
public:
  using EntitySet = ES;
  using Element = typename EntitySet::template Codim<0>::Entity;
  using PatchSet = IteratorRange<typename EntitySet::template Codim<0>::Iterator>;
  using LabelSet = std::array<PatchSet,1>;
  using PartitionSet = std::array<LabelSet,1>;
 
  explicit Identity(const EntitySet& entity_set)
    : _entity_set{entity_set}
  {
    update(entity_set);
  }
 
  [[nodiscard]] constexpr static auto haloRegion(const Dune::Concept::Entity auto& entity) {
    return interior_halo_region;
  }
 
  [[nodiscard]] EntitySet entitySet() const noexcept { return _entity_set; }
 
  [[nodiscard]] auto begin() const { return _partition_set.begin(); }
 
  [[nodiscard]] auto end() const { return _partition_set.end(); }
 
  void update(EntitySet entity_set) {
    _entity_set = entity_set;
    _partition_set = PartitionSet{LabelSet{PatchSet{entity_set.template begin<0>(), entity_set.template end<0>()}}};
  }
 
private:
  EntitySet _entity_set;
  PartitionSet _partition_set;
};
 
} // namespace Dune::PDELab::EntitySetPartition
 
#endif // DUNE_PDELAB_COMMON_PARTITION_IDENTITY_HH