在 C++11 中根据模板参数选择数组大小？

Question

考虑这段代码：

enum class EnumType
{
  Type1,
  Type2
};

constexpr std::size_t the_length;

template <EnumType T>
int function()
{
  std::array<uint8_t, the_length> x;
  //some code here that uses x 
}

我希望数组x 的长度根据T 的类型具有不同的值。例如，如果 T 可以取 2 个值之一（Type1、Type2），我希望 the_length 如果 T==Type1 具有值 10，如果 T==Type2 具有值 20。这可以在 C++11 中完成吗？谢谢

Answer 1

旧的三元运算符有什么问题？

template <EnumType T>
int function()
{
  std::array<SomeType, T == EnumType::Type1 ? 10u : 20u> x;
}

如果T 是typename，而不是某种类型的值，您只需更改测试

template <typename T>
int function()
{
  std::array<T, std::is_same<SomeType1, T>::value ? 10u : 20u> x;
}

Answer 2

正如@templatetypedef 所说，但 C++11 可以做的远不止这些：

#include <array>
#include <cstddef>

enum class EnumType { T1, T2 };

template<EnumType T>
struct my_array_traits;

template<>
struct my_array_traits<EnumType::T1> {
    using type = float;
    constexpr static std::size_t value = 5;
};

template<>
struct my_array_traits<EnumType::T2> {
    using type = double;
    constexpr static std::size_t value = 10;
};

template<EnumType T>
struct my_other_array_traits;

template<>
struct my_other_array_traits<EnumType::T1> {
    using type = short;
    constexpr static std::size_t value = 20;
};

template<>
struct my_other_array_traits<EnumType::T2> {
    using type = long;
    constexpr static std::size_t value = 40;
};


template <EnumType T, template<EnumType> class array_traits>
int function()
{
    std::array<typename array_traits<T>::type,
               array_traits<T>::value> x;
    //some code here that uses x 
    return 0;
}

int main() {
    function<EnumType::T1, my_array_traits>();
    function<EnumType::T2, my_array_traits>();
    function<EnumType::T1, my_other_array_traits>();
    function<EnumType::T2, my_other_array_traits>();
    return 0;
}

Answer 3

当然，这是可能的。您只需要另一个级别的间接性。这是执行此操作的一种方法：

/* array_for<T>::type gives you your array type. */
template <EnumType T> struct array_for;

/* Specialize to give different values based on what T is. */
template <> struct array_for<Option1> {
    using type = std::array<WhateverTypeYouWantToUse1, 10>;
};
template <> struct array_for<Option2> {
    using type = std::array<WhateverTypeYouWantToUse2, 20>;
};

template <EnumType T>
int function()
{
  typename array_for<T>::type myArray;
  /* do whatever coding dance with myArray seems most fitting. */
}

这里，array_for 帮助器 struct 模板将 EnumType 作为输入，然后根据找到的内容输出不同类型的数组。