多维std :: array的可变参数模板

时间:2018-10-05 13:43:47

标签: c++ c++17 generic-programming

我们可以为多维数组添加别名,如下所示:

template<typename T, size_t size1, size_t size2>
using myArray = std::array<std::array<T, size2>, size1>;

但这仅允许我们预定义数量的尺寸。有没有办法将其转换为可变参数模板,以便我们可以编写其中的任何一个:

myArray<int, 2, 2, 2> arr3d;
myArray<int, 2, 2, 2, 2> arr4d;

我尝试了一些事情,但对其中任何一个都不完全满意。

此:

template<typename T, size_t size>
using myArray<T, size> = std::array<T, size>;

template<typename T, size_t size, size_t... more>
using myArray = std::array<myArray<T, more...>, size>;

甚至不编译,因为别名模板显然不允许模板专门化。

这是目前最好的解决方案,但删除了我想保留的所有std :: array构造函数:

template<typename T, size_t size, size_t... more>
struct myArray : public std::array<myArray<T, more...>, size> {};

template<typename T, size_t size>
struct myArray<T, size> : public std::array<T, size>{};

使用这种解决方案,我总是必须在每次访问数组之前都写“ .internal”:

template<typename T, size_t size, size_t... more>
struct myArr {
    std::array<myArr<T, more...>, size> internal;
};

template<typename T, size_t size>
struct myArr<T, size> {
    std::array<T, size> internal;
};

那么,谁能想到第二种解决方案,但我可以将构造函数保留在哪里呢?我不能。

2 个答案:

答案 0 :(得分:5)

我建议您简单地将维度相乘并得到一个std::array。例如:std::array<int, D0 * D1 * D2 * D3>。然后,您可以提供实用程序函数或包装器类,以将多维索引转换为一维。

反正...

这是一个使用显式模板专门化的简单递归解决方案:

template <typename T, std::size_t... Ds>
struct nested_array;

template <typename T, std::size_t D>
struct nested_array<T, D>
{
    using type = std::array<T, D>;
};

template <typename T, std::size_t D, std::size_t... Ds>
struct nested_array<T, D, Ds...>
{
    using type = std::array<typename nested_array<T, Ds...>::type, D>;
};

static_assert(std::is_same_v<
    typename nested_array<int, 1, 2, 3>::type,
    std::array<std::array<std::array<int, 3>, 2>, 1>
>);

live example on wandbox


以下是基于可变模板部分专业化的解决方案:

template <typename T>
struct t { using type = T; };

template <typename T>
using unwrap = typename T::type;

template <typename T, std::size_t... Ds>
constexpr auto nested_array = t<void>{};

template <typename T, std::size_t D>
constexpr auto nested_array<T, D> = t<std::array<T, D>>{};

template <typename T, std::size_t D, std::size_t... Ds>
constexpr auto nested_array<T, D, Ds...> = 
    t<std::array<unwrap<decltype(nested_array<T, Ds...>)>, D>>{};

static_assert(std::is_same_v<
    unwrap<decltype(nested_array<int, 1, 2, 3>)>,
    std::array<std::array<std::array<int, 3>, 2>, 1>
>);

live example on wandbox

答案 1 :(得分:1)

您可以使用额外的层:

template<typename T, size_t size, size_t... more>
struct myArray_impl
{
    using type = std::array<typename myArray_impl<T, more...>::type, size>;
};

template<typename T, size_t size>
struct myArray_impl<T, size>
{
    using type = std::array<T, size>;
};

template<typename T, size_t size, size_t... more>
using myArray = typename myArray_impl<T, size, more...>::type;