containers.hpp

// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef GKO_PUBLIC_CORE_SYNTHESIZER_CONTAINERS_HPP_
#define GKO_PUBLIC_CORE_SYNTHESIZER_CONTAINERS_HPP_
 
 
#include <array>
#include <type_traits>
 
 
namespace gko {
namespace syn {
 

template <typename T, T... Values>
struct value_list {};
 

template <typename... Types>
struct type_list {};
 

template <int Start, int End, int Step = 1>
struct range {};
 
 
namespace detail {
 

template <typename List1, typename List2>
struct concatenate_impl;

template <typename T, T... Values1, T... Values2>
struct concatenate_impl<value_list<T, Values1...>, value_list<T, Values2...>> {
    using type = value_list<T, Values1..., Values2...>;
};
 
 
}  // namespace detail
 

template <typename List1, typename List2>
using concatenate = typename detail::concatenate_impl<List1, List2>::type;
 
 
namespace detail {
 

template <typename T, typename = void>
struct as_list_impl;

template <typename T, T... Values>
struct as_list_impl<value_list<T, Values...>> {
    using type = value_list<T, Values...>;
};

template <typename... Types>
struct as_list_impl<type_list<Types...>> {
    using type = type_list<Types...>;
};

template <int Start, int End, int Step>
struct as_list_impl<range<Start, End, Step>, std::enable_if_t<(Start < End)>> {
    using type = concatenate<
        value_list<int, Start>,
        typename as_list_impl<range<Start + Step, End, Step>>::type>;
};

template <int Start, int End, int Step>
struct as_list_impl<range<Start, End, Step>, std::enable_if_t<(Start >= End)>> {
    using type = value_list<int>;
};
 
 
}  // namespace detail
 

template <typename T>
using as_list = typename detail::as_list_impl<T>::type;
 

template <typename T, T... Value>
constexpr std::array<T, sizeof...(Value)> as_array(value_list<T, Value...> vl)
{
    return std::array<T, sizeof...(Value)>{Value...};
}
 
 
}  // namespace syn
}  // namespace gko
 
 
#endif  // GKO_PUBLIC_CORE_SYNTHESIZER_CONTAINERS_HPP_