dofstorage.hh

#ifndef DUNE_FEM_DOFSTORAGE_HH
#define DUNE_FEM_DOFSTORAGE_HH
 
#include <cassert>
 
namespace Dune
{
 
  namespace Fem
  {
 
    enum DofStoragePolicy { PointBased, VariableBased };
 
    template <DofStoragePolicy p>
    class DofConversionUtility;
 
    template <>
    class DofConversionUtility<PointBased> {
    public:
      DofConversionUtility(int numComponents) :
        numComponents_(numComponents)
      {}
 
      static DofStoragePolicy policy() { return PointBased; }
 
      void newSize(int size) {} // just do nothing
 
      int component(int combinedIndex) const {
        return combinedIndex%numComponents_;
      }
      int containedDof(int combinedIndex) const {
        return combinedIndex/numComponents_;
      }
 
      int combinedDof(int containedIndex, int component) const {
        return containedIndex*numComponents_ + component;
      }
 
    private:
      const int numComponents_;
    };
 
    template <>
    class DofConversionUtility<VariableBased> {
    public:
      DofConversionUtility(int size) :
        size_(size)
      {}
 
      static DofStoragePolicy policy() { return VariableBased; }
 
      void newSize(int size) { size_ = size; }
 
      int component(int combinedIndex) const {
        return combinedIndex/size_;
      }
 
      int containedDof(int combinedIndex) const {
        return combinedIndex%size_;
      }
 
      int combinedDof(int containedIndex, int component) const {
        return containedIndex + component*size_;
      }
 
    private:
      int size_;
    };
 
    template <unsigned int dimRange>
    class PointBasedDofConversionUtility {
    public:
      PointBasedDofConversionUtility(int numComponents)
      {
        // make sure that we use the correct number of components
        assert( numComponents == int(dimRange) );
      }
 
      static DofStoragePolicy policy() { return PointBased; }
 
      void newSize(const int size) {} // just do nothing
 
      int component(const int combinedIndex) const
      {
        return combinedIndex % dimRange;
      }
      int containedDof(const int combinedIndex) const
      {
        return combinedIndex / dimRange;
      }
 
      int combinedDof(const int containedIndex,
                      const int component) const
      {
        return containedIndex * dimRange + component;
      }
    };
 
  } // namespace Fem
 
} // namespace Dune
 
#endif // #ifndef DUNE_FEM_DOFSTORAGE_HH