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_INTERFACE_MMESHLEAFITERATOR_HH
#define DUNE_MMESH_INTERFACE_MMESHLEAFITERATOR_HH
 
 
// Dune includes
#include <dune/grid/common/gridenums.hh>
 
namespace Dune
{
 
  template<int codim, PartitionIteratorType pitype, class GridImp, typename Enable = void>
  class MMeshInterfaceGridLeafIteratorImp {};
 
  template<int codim, PartitionIteratorType pitype, class GridImp>
  using MMeshInterfaceGridLeafIterator = MMeshInterfaceGridLeafIteratorImp<codim, pitype, GridImp>;
 
  template<PartitionIteratorType pitype, class GridImp>
  class MMeshInterfaceGridLeafIteratorImp<0, pitype, GridImp, std::enable_if_t<GridImp::dimensionworld == 2>>
  {
  private:
   using HostGridLeafIterator = typename GridImp::HostGridType::Finite_faces_iterator;
   using HostGridFacet = typename GridImp::MMeshType::FacetHandle;
 
  public:
    enum {codimension = 0};
 
    typedef typename GridImp::template Codim<0>::Entity Entity;
 
    explicit MMeshInterfaceGridLeafIteratorImp() :
      mMesh_(nullptr)
    {}
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh) :
      mMesh_(mMesh),
      hostLeafIterator_(pitype == Interior_Partition
        ? mMesh->partitionHelper().leafInteriorBegin()
        : mMesh->getHostGrid().finite_faces_begin()),
      face_(0)
    {
      if( proceed() )
        increment();
    }
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh, bool endDummy) :
      mMesh_(mMesh),
      hostLeafIterator_(pitype == Interior_Partition
        ? mMesh->partitionHelper().leafInteriorEnd()
        : mMesh->getHostGrid().finite_faces_end()),
      face_(0)
    {}
 
    void increment() {
      do {
        face_++;
        if (face_ == 3)
        {
          ++hostLeafIterator_;
          face_ = 0;
        }
      }
      while( proceed() );
    }
 
    Entity dereference() const {
      return Entity {{ mMesh_, HostGridFacet( hostLeafIterator_, face_ ) }};
    }
 
    bool equals(const MMeshInterfaceGridLeafIteratorImp& i) const {
      return hostLeafIterator_ == i.hostLeafIterator_ && face_ == i.face_;
    }
 
  private:
    bool proceed()
    {
      const auto endIterator = (pitype == Interior_Partition ? mMesh_->partitionHelper().leafInteriorEnd() : mMesh_->getHostGrid().finite_faces_end());
      if (hostLeafIterator_ == endIterator)
        return false;
 
      HostGridFacet facet ( hostLeafIterator_, face_ );
      if (!mMesh_->isInterface( facet ))
        return true;
 
      const auto mirrored = hostLeafIterator_->neighbor( face_ );
      if ( hostLeafIterator_->info().insertionIndex > mirrored->info().insertionIndex )
        return true;
 
      return !mMesh_->partitionHelper().contains(pitype, dereference());
    }
 
    const GridImp* mMesh_;
 
    HostGridLeafIterator hostLeafIterator_;
    int face_;
  };
 
 
  template<PartitionIteratorType pitype, class GridImp>
  class MMeshInterfaceGridLeafIteratorImp<1, pitype, GridImp, std::enable_if_t<GridImp::dimensionworld == 2>>
  {
  private:
    typedef typename GridImp::HostGridType::Vertex_iterator HostGridLeafIterator;
 
  public:
    enum {codimension = GridImp::dimension};
 
    typedef typename GridImp::template Codim<codimension>::Entity Entity;
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh) :
      mMesh_(mMesh),
      hostLeafIterator_(mMesh->getHostGrid().finite_vertices_begin()),
      hostLeafIteratorEnd_(mMesh->getHostGrid().finite_vertices_end())
    {
      while( proceed() )
        ++hostLeafIterator_;
    }
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh, bool endDummy) :
      mMesh_(mMesh),
      hostLeafIterator_(mMesh->getHostGrid().finite_vertices_end()),
      hostLeafIteratorEnd_(mMesh->getHostGrid().finite_vertices_end())
    {}
 
    void increment() {
      ++hostLeafIterator_;
 
      while( proceed() )
        ++hostLeafIterator_;
    }
 
    Entity dereference() const {
      return Entity {{ mMesh_, hostLeafIterator_ }};
    }
 
    bool equals(const MMeshInterfaceGridLeafIteratorImp& i) const {
      return hostLeafIterator_ == i.hostLeafIterator_;
    }
 
  private:
    bool proceed()
    {
      if (hostLeafIterator_ == hostLeafIteratorEnd_)
        return false;
      if (!hostLeafIterator_->info().isInterface)
        return true;
      return !mMesh_->partitionHelper().contains(pitype, dereference());
    }
 
    const GridImp* mMesh_;
 
    HostGridLeafIterator hostLeafIterator_;
    HostGridLeafIterator hostLeafIteratorEnd_;
  };
 
 
  template<PartitionIteratorType pitype, class GridImp>
  class MMeshInterfaceGridLeafIteratorImp<0, pitype, GridImp, std::enable_if_t<GridImp::dimensionworld == 3>>
  {
  private:
   using HostGridLeafIterator = typename GridImp::HostGridType::Finite_cells_iterator;
   using HostGridFacet = typename GridImp::MMeshType::FacetHandle;
 
  public:
    enum {codimension = 0};
 
    typedef typename GridImp::template Codim<0>::Entity Entity;
 
    explicit MMeshInterfaceGridLeafIteratorImp() :
      mMesh_(nullptr)
    {}
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh) :
      mMesh_(mMesh),
      hostLeafIterator_(pitype == Interior_Partition
        ? mMesh->partitionHelper().leafInteriorBegin()
        : mMesh->getHostGrid().finite_cells_begin()),
      face_(0)
    {
      if( proceed() )
        increment();
    }
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh, bool endDummy) :
      mMesh_(mMesh),
      hostLeafIterator_(pitype == Interior_Partition
        ? mMesh->partitionHelper().leafInteriorEnd()
        : mMesh->getHostGrid().finite_cells_end()),
      face_(0)
    {}
 
    void increment() {
      do {
        face_++;
        if (face_ == 4)
        {
          ++hostLeafIterator_;
          face_ = 0;
        }
      }
      while( proceed() );
    }
 
    Entity dereference() const {
      return Entity {{ mMesh_, HostGridFacet( hostLeafIterator_, face_ ) }};
    }
 
    bool equals(const MMeshInterfaceGridLeafIteratorImp& i) const {
      return hostLeafIterator_ == i.hostLeafIterator_ && face_ == i.face_;
    }
 
  private:
    bool proceed()
    {
      const auto endIterator = (pitype == Interior_Partition ? mMesh_->partitionHelper().leafInteriorEnd() : mMesh_->getHostGrid().finite_cells_end());
      if (hostLeafIterator_ == endIterator)
        return false;
 
      HostGridFacet facet ( hostLeafIterator_, face_ );
      if (!mMesh_->isInterface( facet ))
        return true;
 
      const auto mirrored = hostLeafIterator_->neighbor( face_ );
      if ( hostLeafIterator_->info().insertionIndex > mirrored->info().insertionIndex )
        return true;
 
      return !mMesh_->partitionHelper().contains(pitype, dereference());
    }
 
    const GridImp* mMesh_;
 
    HostGridLeafIterator hostLeafIterator_;
    int face_;
  };
 
  template<PartitionIteratorType pitype, class GridImp>
  class MMeshInterfaceGridLeafIteratorImp<1, pitype, GridImp, std::enable_if_t<GridImp::dimensionworld == 3>>
  {
  private:
   using HostGridLeafIterator = typename GridImp::HostGridType::Finite_edges_iterator;
 
  public:
    enum {codimension = 1};
 
    typedef typename GridImp::template Codim<1>::Entity Entity;
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh) :
      mMesh_(mMesh),
      hostLeafIterator_(mMesh->getHostGrid().finite_edges_begin()),
      hostLeafIteratorEnd_(mMesh->getHostGrid().finite_edges_end())
    {
      while( proceed() )
        ++hostLeafIterator_;
    }
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh, bool endDummy) :
      mMesh_(mMesh),
      hostLeafIterator_(mMesh->getHostGrid().finite_edges_end()),
      hostLeafIteratorEnd_(mMesh->getHostGrid().finite_edges_end())
    {}
 
    void increment() {
      ++hostLeafIterator_;
 
      while( proceed() )
        ++hostLeafIterator_;
    }
 
    Entity dereference() const {
      return Entity {{ mMesh_, *hostLeafIterator_ }};
    }
 
    bool equals(const MMeshInterfaceGridLeafIteratorImp& i) const {
      return hostLeafIterator_ == i.hostLeafIterator_;
    }
 
  private:
    bool proceed()
    {
      if (hostLeafIterator_ == hostLeafIteratorEnd_)
        return false;
      if (!mMesh_->isInterface( *hostLeafIterator_ ))
        return true;
      return !mMesh_->partitionHelper().contains(pitype, dereference());
    }
 
    const GridImp* mMesh_;
 
    HostGridLeafIterator hostLeafIterator_;
    HostGridLeafIterator hostLeafIteratorEnd_;
  };
 
  template<PartitionIteratorType pitype, class GridImp>
  class MMeshInterfaceGridLeafIteratorImp<2, pitype, GridImp, std::enable_if_t<GridImp::dimensionworld == 3>>
  {
  private:
    typedef typename GridImp::HostGridType::Vertex_iterator HostGridLeafIterator;
 
  public:
    enum {codimension = GridImp::dimension};
 
    typedef typename GridImp::template Codim<codimension>::Entity Entity;
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh) :
      mMesh_(mMesh),
      hostLeafIterator_(mMesh->getHostGrid().finite_vertices_begin()),
      hostLeafIteratorEnd_(mMesh->getHostGrid().finite_vertices_end())
    {
      while( proceed() )
        ++hostLeafIterator_;
    }
 
    explicit MMeshInterfaceGridLeafIteratorImp(const GridImp* mMesh, bool endDummy) :
      mMesh_(mMesh),
      hostLeafIterator_(mMesh->getHostGrid().finite_vertices_end()),
      hostLeafIteratorEnd_(mMesh->getHostGrid().finite_vertices_end())
    {}
 
    void increment() {
      ++hostLeafIterator_;
 
      while( proceed() )
        ++hostLeafIterator_;
    }
 
    Entity dereference() const {
      return Entity {{ mMesh_, hostLeafIterator_ }};
    }
 
    bool equals(const MMeshInterfaceGridLeafIteratorImp& i) const {
      return hostLeafIterator_ == i.hostLeafIterator_;
    }
 
  private:
    bool proceed()
    {
      if (hostLeafIterator_ == hostLeafIteratorEnd_)
        return false;
      if (!hostLeafIterator_->info().isInterface)
        return true;
      return !mMesh_->partitionHelper().contains(pitype, dereference());
    }
 
    const GridImp* mMesh_;
 
    HostGridLeafIterator hostLeafIterator_;
    HostGridLeafIterator hostLeafIteratorEnd_;
  };
 
}  // namespace Dune
 
#endif