HierarchicNodeToRangeMap.hpp

#pragma once

#include <utility>
#include <type_traits>

#include <dune/common/concept.hh>

#include <dune/functions/common/indexaccess.hh>

#include <amdis/common/Concepts.hpp>

namespace AMDiS
{
  struct HierarchicNodeToRangeMap
  {
    // Specialization for ranges with operator[] access
    template <class Node, class TreePath, class Range,
      REQUIRES(Concepts::HasIndexAccess<Range, Dune::index_constant<0>>)>
    decltype(auto) operator()(const Node& node, const TreePath& treePath, Range&& y) const
    {
      return Dune::Functions::resolveStaticMultiIndex(y, transformTreePath(treePath));
    }

    // Specialization for non-container ranges
    template <class Node, class TreePath, class Range,
      REQUIRES(not Concepts::HasIndexAccess<Range, Dune::index_constant<0>>)>
    decltype(auto) operator()(const Node& node, const TreePath& treePath, Range&& y) const
    {
      return std::forward<Range>(y);
    }

  private:
    // NOTE: due to a bug in dune-functions <= 2.6, a hybrid-treepath can not be passed to a HierarchicNodeToRangeMap,
    // i.e. the HybridTreePath missed a size() function
    template <class TreePath>
    static auto transformTreePath(TreePath const& treePath)
    {
      return Ranges::apply([](auto... i) { return Dune::makeTupleVector(i...); }, treePath._data);
    }
  };

} // end namespace AMDiS