#pragma once

#include <initializer_list>

#include <amdis/common/Index.hpp>
#include <amdis/common/Range.hpp>

namespace AMDiS
{
  namespace Ranges
  {
    template <class Functor, class... Args>
    constexpr void forVariadic(Functor&& f, Args&&... args)
    {
      (f(FWD(args)),...);
    }

    template <std::size_t... I, class Tuple, class Functor>
    constexpr void forEach(std::index_sequence<I...>, Tuple&& tuple, Functor&& f)
    {
      using std::get;
      (f(get<I>(tuple)),...);
    }

    template <class Tuple, class Functor>
    constexpr void forEach(Tuple&& tuple, Functor&& f)
    {
      Ranges::forEach(std::make_index_sequence<static_size_v<Tuple>>{}, FWD(tuple), FWD(f));
    }

    template <class T, std::size_t n, class Functor>
    constexpr void forEach(std::array<T,n> const& a, Functor&& f)
    {
      for (auto const& ai : a)
        f(ai);
    }

    template <class T, class Functor>
    constexpr void forEach(std::vector<T> const& v, Functor&& f)
    {
      for (auto const& vi : v)
        f(vi);
    }


    template <std::size_t I0 = 0, std::size_t... I, class Functor>
    constexpr void forIndices(std::index_sequence<I...>, Functor&& f)
    {
      (f(index_t<I0+I>{}),...);
    }

    template <std::size_t I0, std::size_t I1, class Functor>
    constexpr void forIndices(index_t<I0>, index_t<I1>, Functor&& f)
    {
      Ranges::forIndices<I0>(std::make_index_sequence<std::size_t(I1-I0)>{}, FWD(f));
    }

    template <std::size_t N, class Functor>
    constexpr void forIndices(index_t<N>, Functor&& f)
    {
      Ranges::forIndices(std::make_index_sequence<N>{}, FWD(f));
    }

    template <std::size_t I0, std::size_t I1, class Functor>
    constexpr void forIndices(Functor&& f)
    {
      Ranges::forIndices<I0>(std::make_index_sequence<std::size_t(I1-I0)>{}, FWD(f));
    }

    template <std::size_t N, class Functor>
    constexpr void forIndices(Functor&& f)
    {
      Ranges::forIndices<0>(std::make_index_sequence<N>{}, FWD(f));
    }

  } // end namespace Ranges
} // end namespace AMDiS