Question

我对部分模板特化有点困惑...我有一些代码依赖于算术数据类型T，并且在一个小的整数DIM上。我希望能够为不同的DIM值指定不同的类方法。使用部分模板专业化的不可能性使我探索了enable_if。它正是我所需要的...除了我希望它返回一个数字而不是一个类型。我怎样才能做到这一点？以下代码应说明我想要的内容。

#include <stdio.h>
#include <iostream>
#include <type_traits>

template <typename T, int DIM>
class foo{
    public:
        T function();

};


template <typename T, int DIM>
T foo<T, std::enable_if<DIM == 1>::value>::function(){
    // do something
    return 1.0;
}

template <typename T, int DIM>
T foo<T, std::enable_if<DIM == 2>::value>::function(){  
    // do something else
    return 2342.0;
}

int main(){
    foo<int, 1> object;
    int ak = object.function();
    std::cout << ak << "\n";

    return 0;   
}

Answer 1

您可以使用enable_if完全执行您想要的操作，请记住，当条件为false时，替换必须失败，因此您必须调用type以确保在专门针对各种条件时替换失败

#include <stdio.h>
#include <iostream>
#include <type_traits>

template <typename T, int DIM>
class foo
{
public:
    template <int D = DIM>
    typename std::enable_if<D == 1, T>::type
    function()
    {
        // do something
        return 1.0;
    }

    template <int D = DIM>
    typename std::enable_if<D == 2, T>::type
    function()
    {
        // do something else
        return 2342.0;
    }

};

int main(){
    foo<int, 1> object;
    int ak = object.function();
    std::cout << ak << "\n";

    return 0;
}

对于简单的场景，如上面的场景（检查特定值而不是值范围），您也可以使用局部特化。但是，如果您希望对1-50的所有值，51-200的另一个值进行专门化，然后进行通用的实现，enable_if效果很好。

您还可以在模板签名中使用enable_if。只是一个简单的例子。

#include <stdio.h>
#include <iostream>
#include <type_traits>

template <typename T, int DIM>
class foo
{
public:
    template <int D = DIM, typename std::enable_if<D == 1, void>::type* = nullptr>
    T function()
    {
        // do something
        return 1.0;
    }

    template <int D = DIM, typename std::enable_if<D == 2, void>::type* = nullptr>
    T function()
    {
        // do something else
        return 2342.0;
    }

};

int main(){
    foo<int, 1> object;
    int ak = object.function();
    std::cout << ak << "\n";

    return 0;
}

Answer 2

你可以部分专门化整个班级：

template <typename T, int DIM>
class foo;

template <typename T>
class foo<T, 1>
{
public:
    T function() {
         // do something
         return 1.0;
     }
};

template <typename T>
class foo<T, 2>
{
public:
    T function() {
         // do something
         return 2342.0;
     }
};

如果两个专业化之间有很多共同的代码，你仍然可以使用继承（继承自公共部分或只是专门部分）。

一种简单的替代方法是使用标签调度：

template <typename T, int dim>
class foo
{
public:
    T function();
};

 template <typename T>
 T function_helper(foo<T, 1>&) {
     // do something
     return 1.0;
 }

 template <typename T>
 T function_helper(foo<T, 2>&) {
     // do something
     return 2342.0;
 }

template <typename T, int dim>
T foo::function() {
    return function_helper(*this);
}

Answer 3

使用std :: integral_constant类型来包装您的值，如：

std::integral_constant<int, 2>

c ++ enable_if用于非类型模板参数

3 个答案: