Skip to content

File make_list_tag.h

File List > code_source > templat > include > TempLat > util > rangeiteration > make_list_tag.h

Go to the documentation of this file

#ifndef TEMPLAT_UTIL_MAKELISTTAG_H
#define TEMPLAT_UTIL_MAKELISTTAG_H

/* This file is part of TempLat, available at https://cosmolattice.github.io/templat .
   Copyright 2021-2026 The TempLat authors, see AUTHORS.md.
   Released under the MIT license, see LICENSE.md. */

// File info: Main contributor(s): Adrien Florio, Year: 2019

#include "TempLat/util/assignabletuple.h"

namespace TempLat
{

  template <int Start, typename F, int... I>
  inline auto make_list_tag_impl(F &&f, std::integer_sequence<int, I...> iseq)
  {
    return make_list(f(Tag<I + Start>())...);
  }
  template <int Start, int End, typename F> inline constexpr auto make_list_tag(F &&f)
  {
    if constexpr (End > Start)
      return make_list_tag_impl<Start>(std::forward<F>(f), std::make_integer_sequence<int, End - Start>());
    else
      return std::tuple<>();
  }

  template <int End, typename F> inline constexpr auto make_list_tag(F &&f)
  {
    if constexpr (End > 0)
      return make_list_tag_impl<0>(std::forward<F>(f), std::make_integer_sequence<int, End>());
    else
      return std::tuple<>();
  }

  template <int Start, typename F, int... I>
  inline auto make_vector_tag_impl(F &&f, std::integer_sequence<int, I...> iseq)
  {
    return make_vector(f(Tag<I + Start>())...);
  }
  template <int Start, int End, typename F> inline constexpr auto make_vector_tag(F &&f)
  {
    if constexpr (End > Start)
      return make_vector_tag_impl<Start>(std::forward<F>(f), std::make_integer_sequence<int, End - Start>());
    else
      return std::tuple<>();
  }

  template <int End, typename F> inline constexpr auto make_vector_tag(F &&f)
  {
    if constexpr (End > 0)
      return make_vector_tag_impl<0>(std::forward<F>(f), std::make_integer_sequence<int, End>());
    else
      return std::tuple<>();
  }

#define MakeArray(i, beg, end, expr)                                                                                   \
  make_list_tag<beg, end + 1>([&](auto i) {                                                                            \
    {                                                                                                                  \
      return expr;                                                                                                     \
    };                                                                                                                 \
  })
#define MakeVector(i, beg, end, expr) make_vector_tag<beg, end + 1>([&](auto i) { return expr; })

} // namespace TempLat

#endif