RecursiveApply.hpp

#pragma once

#include <array>
#include <tuple>
#include <vector>

#include <dune/common/tuplevector.hh>
#include <amdis/common/Apply.hpp>

namespace AMDiS
{
  namespace Recursive
  {
    template <class T>
    struct Apply
    {
      template <class F>
      static auto impl(F&& f, T const& t)
      {
        return f(t);
      }
    };


    template <class F, class T>
    auto apply(F&& f, T const& t)
    {
      return Apply<T>::impl(f,t);
    }


    // specializations for container types

    template <class T, std::size_t n>
    struct Apply<std::array<T,n>>
    {
      template <class F>
      static auto impl(F&& f, std::array<T,n> const& a)
      {
        return Ranges::applyIndices<n>([&](auto... ii) {
          return std::array{Recursive::apply(f,a[ii])...}; });
      }
    };

    template <class... TT>
    struct Apply<std::tuple<TT...>>
    {
      template <class F>
      static auto impl(F&& f, std::tuple<TT...> const& t)
      {
        return Ranges::apply([&](auto const&... ti) {
          return std::tuple{Recursive::apply(f,ti)...}; }, t);
      }
    };

    template <class T1, class T2>
    struct Apply<std::pair<T1,T2>>
    {
      template <class F>
      static auto impl(F&& f, std::pair<T1,T2> const& t)
      {
        return std::pair{Recursive::apply(f,t.first),
                         Recursive::apply(f,t.second)};
      }
    };

    template <class... TT>
    struct Apply<Dune::TupleVector<TT...>>
    {
      template <class F>
      static auto impl(F&& f, Dune::TupleVector<TT...> const& t)
      {
        return Ranges::apply([&](auto const&... ti) {
          return Dune::makeTupleVector(Recursive::apply(f,ti)...); }, t);
      }
    };

    template <class T>
    struct Apply<std::vector<T>>
    {
      template <class F>
      static auto impl(F&& f, std::vector<T> const& v)
      {
        using U = TYPEOF(Recursive::apply(f,std::declval<T>()));
        std::vector<U> out;
        out.reserve(v.size());
        for (std::size_t i = 0; i < v.size(); ++i)
          out.emplace_back(Recursive::apply(f,v[i]));
        return out;
      }
    };

  } // end namespace Recursive
} // end namespace AMDiS