entity.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_GEOGRID_ENTITY_HH
#define DUNE_GEOGRID_ENTITY_HH
 
#include <dune/common/nullptr.hh>
 
#include <dune/geometry/referenceelements.hh>
 
#include <dune/grid/common/grid.hh>
#include <dune/grid/geometrygrid/capabilities.hh>
#include <dune/grid/geometrygrid/cornerstorage.hh>
 
namespace Dune
{
 
  namespace GeoGrid
  {
 
    // Internal Forward Declarations
    // -----------------------------
 
    template< int codim, class Grid, bool fake = !(Capabilities::hasHostEntity< Grid, codim >::v) >
    class EntityBase;
 
    template< int codim, int dim, class Grid >
    class Entity;
 
 
 
    // External Forward Declarations
    // -----------------------------
 
    template< class Grid >
    class HierarchicIterator;
 
    template< class Grid, class HostIntersectionIterator >
    class IntersectionIterator;
 
 
 
    // EntityBase (real)
    // -----------------
 
    template< int codim, class Grid >
    class EntityBase< codim, Grid, false >
    {
      typedef typename remove_const< Grid >::type::Traits Traits;
 
    public:
      static const int codimension = codim;
      static const int dimension = Traits::dimension;
      static const int mydimension = dimension - codimension;
      static const int dimensionworld = Traits::dimensionworld;
 
      static const bool fake = false;
 
      typedef typename Traits::ctype ctype;
 
      typedef typename Traits::template Codim< codimension >::Geometry Geometry;
    private:
      typedef typename Traits::HostGrid HostGrid;
      typedef typename Traits::CoordFunction CoordFunction;
 
    public:
      typedef typename HostGrid::template Codim< codimension >::Entity HostEntity;
      typedef typename HostGrid::template Codim< codimension >::EntityPointer HostEntityPointer;
 
      typedef typename Traits::template Codim< codimension >::EntitySeed EntitySeed;
 
      typedef typename HostGrid::template Codim< 0 >::Entity HostElement;
      typedef typename Traits::template Codim< codim >::GeometryImpl GeometryImpl;
 
    private:
      typedef typename HostGrid::template Codim< codimension >::Geometry HostGeometry;
 
      typedef GeoGrid::CoordVector< mydimension, Grid, fake > CoordVector;
 
    public:
      EntityBase ()
        : hostEntity_()
        , grid_( nullptr )
        , geo_()
      {}
 
      EntityBase ( const Grid &grid, const EntitySeed &seed )
        : hostEntity_( grid.hostGrid().entity( grid.getRealImplementation(seed).hostEntitySeed() ) )
        , grid_( &grid )
      {}
 
      EntityBase ( const Grid &grid, const HostElement &hostElement, int i )
        : hostEntity_( hostElement.template subEntity<codim>(i) )
        , grid_( &grid )
      {}
 
 
      EntityBase ( const GeometryImpl &geo, const HostEntity &hostEntity )
        : hostEntity_( hostEntity )
        , grid_( &geo.grid() )
        , geo_( geo )
      {}
 
      EntityBase ( const GeometryImpl &geo, HostEntity&& hostEntity )
        : hostEntity_( std::move( hostEntity ) )
        , grid_( &geo.grid() )
        , geo_( geo )
      {}
 
      EntityBase ( const Grid &grid, const HostEntity& hostEntity )
        : hostEntity_( hostEntity )
        , grid_( &grid )
      {}
 
      EntityBase ( const Grid &grid, HostEntity&& hostEntity )
        : hostEntity_( std::move( hostEntity ) )
        , grid_( &grid )
      {}
 
 
      EntityBase ( const EntityBase &other )
        : hostEntity_( other.hostEntity_ )
        , grid_( other.grid_ )
        , geo_( other.geo_ )
      {}
 
      EntityBase ( EntityBase&& other )
        : hostEntity_( std::move( other.hostEntity_ ) )
        , grid_( other.grid_ )
        , geo_( std::move( other.geo_ ) )
      {}
 
      const EntityBase &operator= ( const EntityBase &other )
      {
        hostEntity_ = other.hostEntity_;
        grid_ = other.grid_;
        geo_ = other.geo_;
        return *this;
      }
 
      const EntityBase &operator= ( EntityBase&& other )
      {
        hostEntity_ = std::move( other.hostEntity_ );
        grid_ = std::move( other.grid_ );
        geo_ = std::move( other.geo_ );
        return *this;
      }
 
      bool equals ( const EntityBase &other) const
      {
        return hostEntity_ == other.hostEntity_;
      }
 
    public:
      GeometryType type () const
      {
        return hostEntity().type();
      }
 
      int level () const
      {
        return hostEntity().level();
      }
 
      PartitionType partitionType () const
      {
        return hostEntity().partitionType();
      }
 
      Geometry geometry () const
      {
        if( !geo_ )
        {
          CoordVector coords( hostEntity(), grid().coordFunction() );
          geo_ = GeometryImpl( grid(), type(), coords );
        }
        return Geometry( geo_ );
      }
 
      EntitySeed seed () const { return typename EntitySeed::Implementation( hostEntity().seed() ); }
      const Grid &grid () const { assert( grid_ ); return *grid_; }
 
      const HostEntity &hostEntity () const
      {
        return hostEntity_;
      }
 
      void initialize ( const HostEntity &hostEntity ) { hostEntity_ = hostEntity; }
 
      template< class HostIndexSet >
      typename HostIndexSet::IndexType
      index ( const HostIndexSet &indexSet ) const
      {
        return indexSet.template index< codimension >( hostEntity() );
      }
 
      template< class HostIndexSet >
      typename HostIndexSet::IndexType
      subIndex ( const HostIndexSet &indexSet, int i, unsigned int cd ) const
      {
        return indexSet.subIndex( hostEntity(), i, cd );
      }
 
      template< class HostIndexSet >
      bool isContained ( const HostIndexSet &indexSet ) const
      {
        return indexSet.contains( hostEntity() );
      }
 
      template< class HostIdSet >
      typename HostIdSet::IdType id ( const HostIdSet &idSet ) const
      {
        return idSet.template id< codimension >( hostEntity() );
      }
    private:
      HostEntity hostEntity_;
      const Grid *grid_;
      mutable GeometryImpl geo_;
    };
 
 
 
    // EntityBase (fake)
    // -----------------
 
    template< int codim, class Grid >
    class EntityBase< codim, Grid, true >
    {
      typedef typename remove_const< Grid >::type::Traits Traits;
 
    public:
      static const int codimension = codim;
      static const int dimension = Traits::dimension;
      static const int mydimension = dimension - codimension;
      static const int dimensionworld = Traits::dimensionworld;
 
      static const bool fake = true;
      typedef typename Traits::ctype ctype;
 
      typedef typename Traits::template Codim< codimension >::Geometry Geometry;
    private:
      typedef typename Traits::HostGrid HostGrid;
      typedef typename Traits::CoordFunction CoordFunction;
 
    public:
      typedef typename HostGrid::template Codim< codimension >::Entity HostEntity;
      typedef typename HostGrid::template Codim< codimension >::EntityPointer HostEntityPointer;
 
      typedef typename Traits::template Codim< codimension >::EntitySeed EntitySeed;
 
      typedef typename HostGrid::template Codim< 0 >::Entity HostElement;
      typedef typename Traits::template Codim< codimension >::GeometryImpl GeometryImpl;
 
    private:
      typedef typename HostGrid::template Codim< 0 >::Geometry HostGeometry;
      typedef typename HostGrid::template Codim< dimension >::EntityPointer HostVertexPointer;
 
      typedef GeoGrid::CoordVector< mydimension, Grid, fake > CoordVector;
 
    public:
      EntityBase () : geo_(), hostElement_(), subEntity_( -1 ) {}
 
 
      EntityBase ( const Grid &grid, const EntitySeed &seed )
        : hostElement_( grid.hostGrid().entity( grid.getRealImplementation(seed).hostElementSeed() ) )
        , grid_( &grid )
        , subEntity_( grid.getRealImplementation(seed).subEntity() )
      {}
 
      EntityBase ( const EntityBase &other )
        : geo_( other.geo_ ),
          hostElement_( other.hostElement_ ),
          subEntity_( other.subEntity_ )
      {}
 
      EntityBase ( EntityBase &&other )
        : geo_( std::move( other.geo_ ) ),
          hostElement_( std::move( other.hostElement_ ) ),
          subEntity_( std::move( other.subEntity_ ) )
      {}
 
      const EntityBase &operator= ( const EntityBase &other )
      {
        geo_ = other.geo_;
        hostElement_ = other.hostElement_;
        subEntity_ = other.subEntity_;
        return *this;
      }
 
      const EntityBase &operator= ( EntityBase&& other )
      {
        geo_ = std::move( other.geo_ );
        hostElement_ = std::move( other.hostElement_ );
        subEntity_ = std::move( other.subEntity_ );
        return *this;
      }
 
      bool equals ( const EntityBase &other) const
      {
        const bool thisEnd = (subEntity() < 0);
        const bool otherEnd = (other.subEntity() < 0);
        if( thisEnd || otherEnd )
          return thisEnd && otherEnd;
 
        const int lvl = level();
        if( lvl != other.level() )
          return false;
 
        const typename Traits::HostGrid::Traits::LevelIndexSet &indexSet
          = grid().hostGrid().levelIndexSet( lvl );
 
        const HostElement &thisElement = hostElement();
        assert( indexSet.contains( thisElement ) );
        const HostElement &otherElement = other.hostElement();
        assert( indexSet.contains( otherElement ) );
 
        const int thisIndex = indexSet.subIndex( thisElement, subEntity(), codimension );
        const int otherIndex = indexSet.subIndex( otherElement, other.subEntity(), codimension );
        return (thisIndex == otherIndex);
      }
 
      GeometryType type () const
      {
        const ReferenceElement< ctype, dimension > &refElement
          = ReferenceElements< ctype, dimension >::general( hostElement().type() );
        return refElement.type( subEntity_, codimension );
      }
 
      int level () const
      {
        return hostElement().level();
      }
 
      PartitionType partitionType () const
      {
        if( !(Capabilities::isParallel< HostGrid >::v) )
          return InteriorEntity;
 
        const ReferenceElement< ctype, dimension > &refElement
          = ReferenceElements< ctype, dimension >::general( hostElement().type() );
 
        PartitionType type = vertexPartitionType( refElement, 0 );
        if( (type != BorderEntity) && (type != FrontEntity) )
          return type;
 
        const int numVertices = refElement.size( subEntity_, codimension, dimension );
        for( int i = 1; i < numVertices; ++i )
        {
          PartitionType vtxType = vertexPartitionType( refElement, i );
          if( (vtxType != BorderEntity) && (vtxType != FrontEntity) )
            return vtxType;
          if( type != vtxType )
            return OverlapEntity;
        }
        assert( (type == BorderEntity) || (type == FrontEntity) );
        return type;
      }
 
      Geometry geometry () const
      {
        if( !geo_ )
        {
          CoordVector coords( hostElement(), subEntity_, grid().coordFunction() );
          geo_ = GeometryImpl( grid(), type(), coords );
        }
        return Geometry( geo_ );
      }
 
      EntitySeed seed () const { return typename EntitySeed::Implementation( hostElement().seed(), subEntity_ ); }
      const Grid &grid () const { assert( grid_ ); return *grid_; }
 
      const HostEntity &hostEntity () const
      {
        DUNE_THROW( NotImplemented, "HostGrid has no entities of codimension " << codimension << "." );
      }
 
      const HostElement &hostElement () const
      {
        assert( *this );
        return hostElement_;
      }
 
      int subEntity () const { return subEntity_; }
 
      void initialize ( const HostElement &hostElement ) { hostElement_ = hostElement; }
 
      template< class HostIndexSet >
      typename HostIndexSet::IndexType index ( const HostIndexSet &indexSet ) const
      {
        return indexSet.subIndex( hostElement(), subEntity_, codimension );
      }
 
      template< class HostIndexSet >
      typename HostIndexSet::IndexType
      subIndex ( const HostIndexSet &indexSet, int i, unsigned int cd ) const
      {
        const ReferenceElement< ctype, dimension > &refElement
          = ReferenceElements< ctype, dimension >::general( hostElement().type() );
        const int j = refElement.subEntity( subEntity_, codimension, i, codimension+cd );
        return indexSet.subIndex( hostElement(), j, codimension+cd );
      }
 
      template< class HostIndexSet >
      bool isContained ( const HostIndexSet &indexSet ) const
      {
        return indexSet.contains( hostElement() );
      }
 
      template< class HostIdSet >
      typename HostIdSet::IdType id ( const HostIdSet &idSet ) const
      {
        return idSet.subId( hostElement(), subEntity_, codimension );
      }
    private:
      PartitionType
      vertexPartitionType ( const ReferenceElement< ctype, dimension > &refElement, int i ) const
      {
        const int j = refElement.subEntity( subEntity_, codimension, 0, dimension );
        return hostElement().template subEntity< dimension >( j )->partitionType();
      }
 
    private:
      HostElement hostElement_;
      unsigned int subEntity_;
      const Grid *grid_;
      mutable GeometryImpl geo_;
    };
 
 
 
    // Entity
    // ------
 
    template< int codim, int dim, class Grid >
    class Entity
      : public EntityBase< codim, Grid >
    {
      typedef EntityBase< codim, Grid > Base;
 
    public:
      typedef typename Base::HostEntity HostEntity;
      typedef typename Base::HostElement HostElement;
      typedef typename Base::GeometryImpl GeometryImpl;
      typedef typename Base::EntitySeed EntitySeed;
 
      Entity () : Base() {}
 
      Entity ( const Grid &grid, const EntitySeed &seed ) : Base( grid, seed ) {}
 
      Entity ( const Grid &grid, const HostEntity &hostEntity ) : Base( grid, hostEntity ) {}
      Entity ( const Grid &grid, HostEntity&& hostEntity ) : Base( grid, std::move( hostEntity ) ) {}
 
      Entity ( const Grid &grid, const HostElement &hostEntity, int i ) : Base( grid, hostEntity, i ) {}
 
    };
 
 
 
    // Entity for codimension 0
    // ------------------------
 
    template< int dim, class Grid >
    class Entity< 0, dim, Grid >
      : public EntityBase< 0, Grid >
    {
      typedef EntityBase< 0, Grid > Base;
 
      typedef typename remove_const< Grid >::type::Traits Traits;
 
      typedef typename Traits::HostGrid HostGrid;
 
    public:
      static const int codimension = Base::codimension;
      static const int dimension = Base::dimension;
      static const int mydimension = Base::mydimension;
      static const int dimensionworld = Base::dimensionworld;
 
      static const bool fake = Base::fake;
      typedef typename Traits::template Codim< codimension >::LocalGeometry LocalGeometry;
      typedef typename Traits::template Codim< codimension >::EntityPointer EntityPointer;
 
      typedef Dune::Entity< 0, dim, Grid, Dune::GeoGrid::Entity > EntityFacade;
 
      typedef typename Traits::HierarchicIterator HierarchicIterator;
      typedef typename Traits::LeafIntersectionIterator LeafIntersectionIterator;
      typedef typename Traits::LevelIntersectionIterator LevelIntersectionIterator;
 
      typedef typename Base::HostEntity HostEntity;
      typedef typename Base::HostElement HostElement;
      typedef typename Base::GeometryImpl GeometryImpl;
      typedef typename Base::EntitySeed EntitySeed;
 
      using Base::grid;
      using Base::hostEntity;
 
      Entity () : Base() {}
 
      Entity ( const Grid &grid, const HostEntity &hostEntity ) : Base( grid, hostEntity ) {}
      Entity ( const Grid &grid, HostEntity&& hostEntity ) : Base( grid, std::move( hostEntity ) ) {}
      Entity ( const GeometryImpl &geo, const HostEntity& hostEntity ) : Base( geo, hostEntity ) {}
      Entity ( const GeometryImpl &geo, HostEntity &&hostEntity ) : Base( geo, std::move( hostEntity ) ) {}
 
      Entity ( const Grid &grid, const EntitySeed &seed ) : Base( grid, seed ) {}
 
      Entity ( const Grid &grid, const HostEntity &hostEntity, int i ) : Base( grid, hostEntity )
      {
        assert( i == 0 );
      }
 
      template< int codim >
      int count () const
      {
        return hostEntity().template count< codim >();
      }
 
      unsigned int subEntities (unsigned int codim) const
      {
        return hostEntity().subEntities(codim);
      }
 
      template< int codim >
      typename Grid::template Codim< codim >::Entity
      subEntity ( int i ) const
      {
        typedef typename Traits::template Codim< codim >::EntityImpl EntityImpl;
        return EntityImpl( grid(), hostEntity(), i );
      }
 
      LevelIntersectionIterator ilevelbegin () const
      {
        typedef GeoGrid::IntersectionIterator< Grid, typename HostGrid::LevelIntersectionIterator > LevelIntersectionIteratorImpl;
        return LevelIntersectionIteratorImpl( *this, hostEntity().ilevelbegin() );
      }
 
      LevelIntersectionIterator ilevelend () const
      {
        typedef GeoGrid::IntersectionIterator< Grid, typename HostGrid::LevelIntersectionIterator > LevelIntersectionIteratorImpl;
        return LevelIntersectionIteratorImpl( *this, hostEntity().ilevelend() );
      }
 
      LeafIntersectionIterator ileafbegin () const
      {
        typedef GeoGrid::IntersectionIterator< Grid, typename HostGrid::LeafIntersectionIterator > LeafIntersectionIteratorImpl;
        return LeafIntersectionIteratorImpl( *this, hostEntity().ileafbegin() );
      }
 
      LeafIntersectionIterator ileafend () const
      {
        typedef GeoGrid::IntersectionIterator< Grid, typename HostGrid::LeafIntersectionIterator > LeafIntersectionIteratorImpl;
        return LeafIntersectionIteratorImpl( *this, hostEntity().ileafend() );
      }
 
      bool hasBoundaryIntersections () const
      {
        return hostEntity().hasBoundaryIntersections();
      }
 
      bool isLeaf () const
      {
        return hostEntity().isLeaf();
      }
 
      EntityFacade father () const
      {
        return Entity( grid(), hostEntity().father() );
      }
 
      bool hasFather () const
      {
        return hostEntity().hasFather();
      }
 
      LocalGeometry geometryInFather () const
      {
        return hostEntity().geometryInFather();
      }
 
      HierarchicIterator hbegin ( int maxLevel ) const
      {
        typedef GeoGrid::HierarchicIterator< Grid > HierarchicIteratorImpl;
        return HierarchicIteratorImpl( grid(), hostEntity().hbegin( maxLevel ) );
      }
 
      HierarchicIterator hend ( int maxLevel ) const
      {
        typedef GeoGrid::HierarchicIterator< Grid > HierarchicIteratorImpl;
        return HierarchicIteratorImpl( grid(), hostEntity().hend( maxLevel ) );
      }
 
      bool isRegular () const
      {
        return hostEntity().isRegular();
      }
 
      bool isNew () const
      {
        return hostEntity().isNew();
      }
 
      bool mightVanish () const
      {
        return hostEntity().mightVanish();
      }
    };
 
  } // namespace GeoGrid
 
} // namespace Dune
 
#endif // #ifndef DUNE_GEOGRID_ENTITY_HH