NodeIndices.hpp

#pragma once
 
#include <dune/common/rangeutilities.hh>
 
#include <amdis/common/Concepts.hpp>
#include <amdis/typetree/MultiIndex.hpp>
#include <amdis/utility/MappedRangeView.hpp>
 
namespace AMDiS
{
  template <class LocalView, class Node>
  auto nodeIndices(LocalView const& localView, Node const& node)
  {
    static_assert(Concepts::LocalView<LocalView>, "");
    static_assert(Concepts::BasisTree<Node, typename LocalView::GridView>, "");
 
    return mappedRangeView(Dune::range(node.size()), [&](std::size_t j) -> typename LocalView::MultiIndex {
      return localView.index(node.localIndex(j));
    });
  }
 
  template <class LocalView>
  auto nodeIndices(LocalView const& localView)
  {
    static_assert(Concepts::LocalView<LocalView>, "");
 
    return mappedRangeView(Dune::range(localView.size()), [&](std::size_t i) -> typename LocalView::MultiIndex {
      return localView.index(i);
    });
  }
 
 
  template <class LocalView, class Node>
  std::size_t nodeIndexCount(LocalView const& /*localView*/, Node const& node)
  {
    return node.size();
  }
 
  template <class LocalView>
  std::size_t nodeIndexCount(LocalView const& localView)
  {
    return localView.size();
  }
 
} // end namespace AMDiS