leafiterator.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_MMESH_GRID_MMESHLEAFITERATOR_HH
#define DUNE_MMESH_GRID_MMESHLEAFITERATOR_HH
 
// Dune includes
#include <dune/grid/common/gridenums.hh>
 
namespace Dune {
 
template <int codim, PartitionIteratorType pitype, class GridImp,
          typename Enable = void>
class MMeshLeafIteratorImp {};
 
template <int codim, PartitionIteratorType pitype, class GridImp>
using MMeshLeafIterator = MMeshLeafIteratorImp<codim, pitype, GridImp>;
 
template <PartitionIteratorType pitype, class GridImp>
class MMeshLeafIteratorImp<0, pitype, GridImp,
                           std::enable_if_t<GridImp::dimension == 2>> {
 public:
  using HostGridLeafIterator =
      typename GridImp::HostGridType::Finite_faces_iterator;
  enum { codimension = 0 };
 
  typedef typename GridImp::template Codim<0>::Entity Entity;
 
  explicit MMeshLeafIteratorImp() : mMesh_(nullptr) {}
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh)
      : mMesh_(mMesh),
        hostLeafIterator_(pitype == Interior_Partition
                              ? mMesh->partitionHelper().leafInteriorBegin()
                              : mMesh->getHostGrid().finite_faces_begin()) {
    while (proceed()) increment();
  }
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh, bool endDummy)
      : mMesh_(mMesh),
        hostLeafIterator_(pitype == Interior_Partition
                              ? mMesh->partitionHelper().leafInteriorEnd()
                              : mMesh->getHostGrid().finite_faces_end()) {}
 
  void increment() {
    do {
      ++hostLeafIterator_;
    } while (proceed());
  }
 
  Entity dereference() const { return Entity{{mMesh_, hostLeafIterator_}}; }
 
  bool equals(const MMeshLeafIteratorImp& i) const {
    return hostLeafIterator_ == i.hostLeafIterator_;
  }
 
 private:
  bool proceed() {
    const auto endIterator = (pitype == Interior_Partition
                                  ? mMesh_->partitionHelper().leafInteriorEnd()
                                  : mMesh_->getHostGrid().finite_faces_end());
    if (hostLeafIterator_ == endIterator) return false;
    return !mMesh_->partitionHelper().contains(
        pitype, hostLeafIterator_->info().partition);
  }
 
  const GridImp* mMesh_;
  HostGridLeafIterator hostLeafIterator_;
};
 
template <PartitionIteratorType pitype, class GridImp>
class MMeshLeafIteratorImp<1, pitype, GridImp,
                           std::enable_if_t<GridImp::dimension == 2>> {
 private:
  using HostGridLeafIterator =
      typename GridImp::HostGridType::Finite_edges_iterator;
 
 public:
  enum { codimension = 1 };
 
  typedef typename GridImp::template Codim<1>::Entity Entity;
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_edges_begin()) {
    while (proceed()) increment();
  }
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh, bool endDummy)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_edges_end()) {}
 
  void increment() {
    do {
      ++hostLeafIterator_;
    } while (proceed());
  }
 
  Entity dereference() const { return Entity{{mMesh_, *hostLeafIterator_}}; }
 
  bool equals(const MMeshLeafIteratorImp& i) const {
    return hostLeafIterator_ == i.hostLeafIterator_;
  }
 
 private:
  bool proceed() {
    if (hostLeafIterator_ == mMesh_->getHostGrid().finite_edges_end())
      return false;
    return !mMesh_->partitionHelper().contains(pitype, dereference());
  }
  const GridImp* mMesh_;
 
  HostGridLeafIterator hostLeafIterator_;
};
 
template <PartitionIteratorType pitype, class GridImp>
class MMeshLeafIteratorImp<2, pitype, GridImp,
                           std::enable_if_t<GridImp::dimension == 2>> {
 private:
  using HostGridLeafIterator =
      typename GridImp::HostGridType::Finite_vertices_iterator;
 
 public:
  enum { codimension = GridImp::dimension };
 
  typedef typename GridImp::template Codim<codimension>::Entity Entity;
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_vertices_begin()) {
    while (proceed()) increment();
  }
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh, bool endDummy)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_vertices_end()) {}
 
  void increment() {
    do {
      ++hostLeafIterator_;
    } while (proceed());
  }
 
  Entity dereference() const { return Entity{{mMesh_, hostLeafIterator_}}; }
 
  bool equals(const MMeshLeafIteratorImp& i) const {
    return hostLeafIterator_ == i.hostLeafIterator_;
  }
 
 private:
  bool proceed() {
    if (hostLeafIterator_ == mMesh_->getHostGrid().finite_vertices_end())
      return false;
    return !mMesh_->partitionHelper().contains(
        pitype, hostLeafIterator_->info().partition);
  }
 
  const GridImp* mMesh_;
  HostGridLeafIterator hostLeafIterator_;
};
 
template <PartitionIteratorType pitype, class GridImp>
class MMeshLeafIteratorImp<0, pitype, GridImp,
                           std::enable_if_t<GridImp::dimension == 3>> {
 public:
  using HostGridLeafIterator =
      typename GridImp::HostGridType::Finite_cells_iterator;
 
  enum { codimension = 0 };
 
  typedef typename GridImp::template Codim<0>::Entity Entity;
 
  explicit MMeshLeafIteratorImp() : mMesh_(nullptr) {}
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh)
      : mMesh_(mMesh),
        hostLeafIterator_(pitype == Interior_Partition
                              ? mMesh->partitionHelper().leafInteriorBegin()
                              : mMesh->getHostGrid().finite_cells_begin()) {
    while (proceed()) increment();
  }
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh, bool endDummy)
      : mMesh_(mMesh),
        hostLeafIterator_(pitype == Interior_Partition
                              ? mMesh->partitionHelper().leafInteriorEnd()
                              : mMesh->getHostGrid().finite_cells_end()) {}
 
  void increment() {
    do {
      ++hostLeafIterator_;
    } while (proceed());
  }
 
  Entity dereference() const { return Entity{{mMesh_, hostLeafIterator_}}; }
 
  bool equals(const MMeshLeafIteratorImp& i) const {
    return hostLeafIterator_ == i.hostLeafIterator_;
  }
 
 private:
  bool proceed() {
    const auto endIterator = (pitype == Interior_Partition
                                  ? mMesh_->partitionHelper().leafInteriorEnd()
                                  : mMesh_->getHostGrid().finite_cells_end());
    if (hostLeafIterator_ == endIterator) return false;
    return !mMesh_->partitionHelper().contains(
        pitype, hostLeafIterator_->info().partition);
  }
 
  const GridImp* mMesh_;
  HostGridLeafIterator hostLeafIterator_;
};
 
template <PartitionIteratorType pitype, class GridImp>
class MMeshLeafIteratorImp<1, pitype, GridImp,
                           std::enable_if_t<GridImp::dimension == 3>> {
 private:
  using HostGridLeafIterator =
      typename GridImp::HostGridType::Finite_facets_iterator;
 
 public:
  enum { codimension = 1 };
 
  typedef typename GridImp::template Codim<1>::Entity Entity;
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_facets_begin()) {
    while (proceed()) increment();
  }
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh, bool endDummy)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_facets_end()) {}
 
  void increment() {
    do {
      ++hostLeafIterator_;
    } while (proceed());
  }
 
  Entity dereference() const { return Entity{{mMesh_, *hostLeafIterator_}}; }
 
  bool equals(const MMeshLeafIteratorImp& i) const {
    return hostLeafIterator_ == i.hostLeafIterator_;
  }
 
 private:
  bool proceed() {
    if (hostLeafIterator_ == mMesh_->getHostGrid().finite_facets_end())
      return false;
    return !mMesh_->partitionHelper().contains(pitype, dereference());
  }
 
  const GridImp* mMesh_;
  HostGridLeafIterator hostLeafIterator_;
};
 
template <PartitionIteratorType pitype, class GridImp>
class MMeshLeafIteratorImp<2, pitype, GridImp,
                           std::enable_if_t<GridImp::dimension == 3>> {
 private:
  using HostGridLeafIterator =
      typename GridImp::HostGridType::Finite_edges_iterator;
 
 public:
  enum { codimension = 2 };
 
  typedef typename GridImp::template Codim<2>::Entity Entity;
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_edges_begin()) {
    while (proceed()) increment();
  }
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh, bool endDummy)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_edges_end()) {}
 
  void increment() {
    do {
      ++hostLeafIterator_;
    } while (proceed());
  }
 
  Entity dereference() const { return Entity{{mMesh_, *hostLeafIterator_}}; }
 
  bool equals(const MMeshLeafIteratorImp& i) const {
    return hostLeafIterator_ == i.hostLeafIterator_;
  }
 
 private:
  bool proceed() {
    if (hostLeafIterator_ == mMesh_->getHostGrid().finite_edges_end())
      return false;
    return !mMesh_->partitionHelper().contains(pitype, dereference());
  }
 
  const GridImp* mMesh_;
  HostGridLeafIterator hostLeafIterator_;
};
 
template <PartitionIteratorType pitype, class GridImp>
class MMeshLeafIteratorImp<3, pitype, GridImp,
                           std::enable_if_t<GridImp::dimension == 3>> {
 private:
  typedef typename GridImp::HostGridType::Vertex_iterator HostGridLeafIterator;
 
 public:
  enum { codimension = GridImp::dimension };
 
  typedef typename GridImp::template Codim<codimension>::Entity Entity;
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_vertices_begin()) {
    while (proceed()) increment();
  }
 
  explicit MMeshLeafIteratorImp(const GridImp* mMesh, bool endDummy)
      : mMesh_(mMesh),
        hostLeafIterator_(mMesh->getHostGrid().finite_vertices_end()) {}
 
  void increment() {
    do {
      ++hostLeafIterator_;
    } while (proceed());
  }
 
  Entity dereference() const { return Entity{{mMesh_, hostLeafIterator_}}; }
 
  bool equals(const MMeshLeafIteratorImp& i) const {
    return hostLeafIterator_ == i.hostLeafIterator_;
  }
 
 private:
  bool proceed() {
    if (hostLeafIterator_ ==
        HostGridLeafIterator(mMesh_->getHostGrid().finite_vertices_end()))
      return false;
    if (hostLeafIterator_ == mMesh_->getHostGrid().infinite_vertex())
      return true;
    return !mMesh_->partitionHelper().contains(
        pitype, hostLeafIterator_->info().partition);
  }
 
  const GridImp* mMesh_;
  HostGridLeafIterator hostLeafIterator_;
};
 
}  // namespace Dune
 
#endif