universalmapper.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:
// $Id$
 
#ifndef DUNE_UNIVERSALMAPPER_HH
#define DUNE_UNIVERSALMAPPER_HH
 
#include <iostream>
#include <map>
#include "mapper.hh"
 
namespace Dune
{
  template <typename G, typename IDS>
  class UniversalMapper :
    public Mapper<G,UniversalMapper<G,IDS> >
  {
    typedef typename IDS::IdType IdType;
  public:
 
    using Mapper< G, UniversalMapper >::map;
    using Mapper< G, UniversalMapper >::contains;
 
    UniversalMapper (const G& grid, const IDS& idset)
      : g(grid), ids(idset), index()
    {
      n=0;     // zero data elements
    }
 
    template<class EntityType>
    int map (const EntityType& e) const
    {
      IdType id = ids.id(e);                                 // get id
      typename std::map<IdType,int>::iterator it = index.find(id);    // look up in map
      if (it!=index.end()) return it->second;                // return index if found
      index[id] = n++;                                       // put next index in map
      return n-1;                                            // and return it
    }
 
 
    int map (const typename G::Traits::template Codim<0>::Entity& e, int i, int cc) const
    {
      IdType id = ids.subId(e,i,cc);           // get id
      typename std::map<IdType,int>::iterator it = index.find(id);    // look up in map
      if (it!=index.end()) return it->second;                // return index if found
      index[id] = n++;                                       // put next index in map
      return n-1;                                            // and return it
    }
 
    int size () const
    {
      return n;
    }
 
    template<class EntityType>
    bool contains (const EntityType& e, int& result) const
    {
      IdType id = ids.id(e);                                 // get id
      typename std::map<IdType,int>::iterator it = index.find(id);    // look up in map
      if (it!=index.end())
      {
        result = it->second;
        return true;
      }
      else
        return false;
    }
 
    bool contains (const typename G::Traits::template Codim<0>::Entity& e, int i, int cc, int& result) const
    {
      IdType id = ids.subId(e,i,cc);           // get id
      typename std::map<IdType,int>::iterator it = index.find(id);    // look up in map
      if (it!=index.end())
      {
        result = it->second;
        return true;
      }
      else
        return false;
    }
 
    void update ()
    {     // nothing to do here
    }
 
    // clear the mapper
    void clear ()
    {
      index.clear();
      n = 0;
    }
 
  private:
    mutable int n;     // number of data elements required
    const G& g;
    const IDS& ids;
    mutable std::map<IdType,int> index;
  };
 
 
 
 
  template <typename G>
  class GlobalUniversalMapper : public UniversalMapper<G,typename G::Traits::GlobalIdSet>
  {
  public:
    /* @brief The constructor
       @param grid A reference to a grid.
     */
    GlobalUniversalMapper (const G& grid)
      : UniversalMapper<G,typename G::Traits::GlobalIdSet>(grid,grid.globalIdSet())
    {}
  };
 
  template <typename G>
  class LocalUniversalMapper : public UniversalMapper<G,typename G::Traits::LocalIdSet>
  {
  public:
    /* @brief The constructor
       @param grid A reference to a grid.
     */
    LocalUniversalMapper (const G& grid)
      : UniversalMapper<G,typename G::Traits::LocalIdSet>(grid,grid.localIdSet())
    {}
  };
 
 
}
#endif