如何为2个不同的值专门化模板?

时间:2017-05-04 14:51:28

标签: c++ c++11 templates

鉴于下面的代码,我想知道是否可以对一组值进行专门化。在我的示例中,我想为N=3N=4创建一个专门化,以使用已知大小的数组。在这种情况下是否可以避免代码重复?

template <typename T, unsigned int N>
class A
{
public:
    T* data;
};

template <typename T>
class A<T, 3>
{
public:
    T data[3];
};

template <typename T>
class A<T, 4>
{
public:
    T data[4];
};

int main()
{
    A<int, 1> u;
    std::cout << sizeof(u.data) << std::endl; // Size of pointer

    A<int, 3> v;
    std::cout << sizeof(v.data) << std::endl; // Size of data

    A<int, 4> w;
    std::cout << sizeof(w.data) << std::endl; // Size of data

    return 0;
}

1 个答案:

答案 0 :(得分:3)

通过在一般情况下引入默认void模板参数,您可以使用std::enable_if

template <typename T, unsigned int N, typename = void>
class A
{
public:
    T* data;
};

template <typename T, unsigned int N>
class A<T, N, typename std::enable_if<N == 3 || N == 4>::type>
{
public:
    T data[N];
};

live wandbox example

如果N == 3 || N == 4true,则typename std::enable_if<N == 3 || N == 4>::type 格式正确,评估为void。然后选择专业化。

如果N == 3 || N == 4false,则typename std::enable_if<N == 3 || N == 4>::type 格式错误,专业化为“SFINAEd away”。然后选择一般情况。