libdune-grid-dev 2.2.1-2 / usr / include / dune / grid / common / scsgmapper.hh

// $Id: scsgmapper.hh 7472 2011-02-28 15:09:13Z mnolte $

#ifndef DUNE_SCSGMAPPER_HH
#define DUNE_SCSGMAPPER_HH

#include<iostream>
#include "mapper.hh"

#include <dune/grid/common/grid.hh>

/**
 * @file 
 * @brief  Mapper classes are used to attach data to a grid
 * @author Peter Bastian
 */

namespace Dune
{
  /** 
   * @addtogroup Mapper
   *
   * @{ 
   */
  
  /** @brief Implementation class for a single codim and single geometry type mapper.
   *
   * In this implementation of a mapper the entity set used as domain for the map consists
   * of the entities of a given codimension c for all entities in the given index set. The index
   * set may only contain entities of a single geometry type, otherwise an exception is thrown. This
   * version is usually not used directly but is used to implement versions for leafwise and levelwise
   * entity sets.
   *
   * \tparam GV A Dune grid view type
   *
   * \tparam c A valid codimension
   */
  template <typename GV, int c>
  class SingleCodimSingleGeomTypeMapper :
        public Mapper<typename GV::Grid,SingleCodimSingleGeomTypeMapper<GV,c> >
  {
  public:

    //! import the base class implementation of map and contains (including the deprecated version)
    //! \todo remove in after next release
    using Mapper< typename GV::Grid, SingleCodimSingleGeomTypeMapper >::map;
    using Mapper< typename GV::Grid, SingleCodimSingleGeomTypeMapper >::contains;

	/** @brief Construct mapper from grid and one of its index sets.

	   \param gridView A Dune GridView object.
	 */
    SingleCodimSingleGeomTypeMapper (const GV& gridView);
    
	/** @brief Map entity to array index.

		\param e Reference to codim cc entity, where cc is the template parameter of the function.
		\return An index in the range 0 ... Max number of entities in set - 1.
	 */
	template<class EntityType>
	int map (const EntityType& e) const;
  
	/** @brief Map subentity of codim 0 entity to array index.
	   
	   \param e Reference to codim 0 entity.
	   \param i Number of the subentity of e, where cc is the template parameter of the function.
           \param codim Codimension of the subentity of e
	   \return An index in the range 0 ... Max number of entities in set - 1.
	 */
    int map (const typename GV::template Codim<0>::Entity& e, 
        int i, unsigned int codim) const; 

	/** @brief Return total number of entities in the entity set managed by the mapper.

	   This number can be used to allocate a vector of data elements associated with the 
       entities of the set. In the parallel case this number is per process (i.e. it
       may be different in different processes).

	   \return Size of the entity set.
	*/
	int size () const;

	/** @brief Returns true if the entity is contained in the index set

	   \param e Reference to entity
	   \param result integer reference where corresponding index is  stored if true
	   \return true if entity is in entity set of the mapper
	 */
	template<class EntityType>
	bool contains (const EntityType& e, int& result) const;

	/** @brief Returns true if the entity is contained in the index set

	   \param e Reference to codim 0 entity
	   \param i subentity number
	   \param cc subentity codim
	   \param result integer reference where corresponding index is  stored if true
	   \return true if entity is in entity set of the mapper
	 */
    bool contains (const typename GV::template Codim<0>::Entity& e, int i, int cc, int& result) const;

 	/** @brief Recalculates map after mesh adaptation
	 */
	void update ()
	{ // nothing to do here
	}
 
  private:
    const typename GV::IndexSet& is;
  };

  /** @} */

  template <typename GV, int c>
  SingleCodimSingleGeomTypeMapper<GV,c>::SingleCodimSingleGeomTypeMapper (const GV& gridView) 
    : is(gridView.indexSet())
  {
	// check that grid has only a single geometry type
	if (is.geomTypes(c).size() != 1)
		DUNE_THROW(GridError, "mapper treats only a single codim and a single geometry type");
  }

  template <typename GV, int c>
  template<class EntityType>
  inline int SingleCodimSingleGeomTypeMapper<GV,c>::map (const EntityType& e) const
  {
	enum { cc = EntityType::codimension };
	dune_static_assert(cc == c, "Entity of wrong codim passed to SingleCodimSingleGeomTypeMapper");
	return is.index(e);
  }

  template <typename GV, int c>
  inline int SingleCodimSingleGeomTypeMapper<GV,c>::map (const typename GV::template Codim<0>::Entity& e, int i, unsigned int codim) const
  {
      if (codim != c)
          DUNE_THROW(GridError, "Id of wrong codim requested from SingleCodimSingleGeomTypeMapper");
      return is.subIndex(e,i,codim);
  }

  template <typename GV, int c>
  inline int SingleCodimSingleGeomTypeMapper<GV,c>::size () const
  {
	return is.size(c);
  }

  template <typename GV, int c>
  template<class EntityType>
  inline bool SingleCodimSingleGeomTypeMapper<GV,c>::contains (const EntityType& e, int& result) const
  {
	result = map(e);
	return true;
  }

  template <typename GV, int c>
  inline bool SingleCodimSingleGeomTypeMapper<GV,c>::contains (const typename GV::template Codim<0>::Entity& e, int i, int cc, int& result) const
  {
    result = this->map(e,i,cc);
	return true;
  }

  /** 
   * @addtogroup Mapper
   *
   * @{ 
   */
  /** @brief Single codim and single geometry type mapper for leaf entities.
   
  
  This mapper uses all leaf entities of a certain codimension as its entity set. It is
  assumed (and checked) that the given grid contains only entities of a single geometry type.

  Template parameters are:
  
  \par G
  A Dune grid type.
  \par c
  A valid codimension.
  */
  template <typename G, int c>
  class LeafSingleCodimSingleGeomTypeMapper : public SingleCodimSingleGeomTypeMapper<typename G::LeafGridView,c> {
  public:
	/* @brief The constructor
	   @param grid A reference to a grid.
	 */
	LeafSingleCodimSingleGeomTypeMapper (const G& grid) 
	  : SingleCodimSingleGeomTypeMapper<typename G::LeafGridView,c>(grid.leafView())
	{}
  };

  /** @brief Single codim and single geometry type mapper for entities of one level.
   
  
  This mapper uses all entities of a certain codimension on a given level as its entity set. It is
  assumed (and checked) that the given grid contains only entities of a single geometry type.

  Template parameters are:
  
  \par G
  A Dune grid type.
  \par c
  A valid codimension.
  */
  template <typename G, int c>
  class LevelSingleCodimSingleGeomTypeMapper : public SingleCodimSingleGeomTypeMapper<typename G::LevelGridView,c> {
  public:
	/* @brief The constructor
	   @param grid A reference to a grid.
	   @param level A valid level of the grid.
	 */
	LevelSingleCodimSingleGeomTypeMapper (const G& grid, int level) 
	  : SingleCodimSingleGeomTypeMapper<typename G::LevelGridView,c>(grid.levelView(level))
	{}
  };

  /** @} */
}
#endif