MultiIndex.hpp

#pragma once

#include <dune/common/reservedvector.hh>
#include <dune/functions/functionspacebases/flatmultiindex.hh>

namespace AMDiS
{
  namespace Traits
  {
    template <class T>
    struct IsFlatIndex
        : std::is_integral<T> {};

    template <class I>
    struct IsFlatIndex<Dune::Functions::FlatMultiIndex<I> >
        : std::true_type {};

    template <class I>
    struct IsFlatIndex<Dune::ReservedVector<I, 1> >
        : std::true_type {};
  }

  inline std::size_t flatMultiIndex(std::size_t idx)
  {
    return idx;
  }

  inline std::size_t flatMultiIndex(Dune::Functions::FlatMultiIndex<std::size_t> const& idx)
  {
    return idx[0];
  }

  inline std::size_t flatMultiIndex(Dune::ReservedVector<std::size_t,1> const& idx)
  {
    return idx[0];
  }

} // end namespace AMDiS