iterator.hh

// -*- mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=2 sw=2 sts=2:
#ifndef DUNE_POLYHEDRALGRID_ITERATOR_HH
#define DUNE_POLYHEDRALGRID_ITERATOR_HH
 
#include <dune/geometry/referenceelements.hh>
 
#include <dune/grid/common/entityiterator.hh>
 
#include <opm/grid/polyhedralgrid/entitypointer.hh>
 
namespace Dune
{
 
  // PolyhedralGridIterator
  // --------------
 
  template< int codim, class Grid, PartitionIteratorType pitype >
  class PolyhedralGridIterator
  : public PolyhedralGridEntityPointer< codim, Grid >
  {
    typedef PolyhedralGridIterator< codim, Grid, pitype > This;
    typedef PolyhedralGridEntityPointer< codim, Grid > Base;
 
  protected:
    typedef typename Base::ExtraData ExtraData;
    using Base :: entityImpl;
 
  public:
    typedef typename Grid::Traits::template Codim<codim>::EntitySeed EntitySeed;
    typedef typename Grid::Traits::template Codim<codim>::EntityImpl EntityImpl;
 
    PolyhedralGridIterator ( ExtraData data, const bool beginIterator )
    : Base( data )
    {
      if( beginIterator )
        entityImpl() = EntityImpl( data, EntitySeed( 0 ) );
    }

    void increment ()
    {
      int index = entityImpl().seed().index();
      ++index;
 
      if( index >= entityImpl().data()->size( codim ) )
        entityImpl() = EntityImpl( entityImpl().data() );
      else
        entityImpl() = EntityImpl( entityImpl().data(), EntitySeed( index ) );
    }
  };
 
} // namespace Dune
 
#endif // #ifndef DUNE_POLYHEDRALGRID_ITERATOR_HH