专门化模板派生类

时间:2016-05-28 09:26:22

标签: c++ templates inheritance

我有以下代码(为了清楚起见,非常简化):

class Base
{
    virtual int DoStuff(int arg) = 0;
};

template <typename T>
class Derived : public Base
{
    int DoStuff(int arg) override
    {
         // do some stuff
         return 0;
    }
};

这很有效。现在我想实现一个DoStuff的特殊(矢量化)实现。我需要根据Derived的类型T来具体实现,如下所示:

class Base
{
    virtual int DoStuff(int arg) = 0;
    virtual int DoStuffVectorized(int arg) = 0;
};

template <typename T>
class Derived : public Base
{
    int DoStuff(int arg) override
    {
         // do some stuff
         return 0;
    }

    int DoStuffVectorized<char>(int arg) override
    {
         // do some stuff for T == char
         return 0;
    }

    int DoStuffVectorized<int>(int arg) override
    {
         // do some stuff for T == int
         return 0;
    }
};

但是我无法完成这项工作。

修改 我收到以下错误消息:错误C2143:语法错误:缺少&#39 ;;&#39;之前&#39;&lt;&#39;在int DoStuffVectorized<char>(int arg) override行。

当我将其更改为: template<char> int DoStuffVectorized(int arg) override我得到:错误C2898:...&#39;:成员函数模板不能是虚拟的

关于如何实现这样的事情的任何建议?我需要它的原因是我有一个std :: vector存储各种类型的数据(通过使用Derived&lt;&gt;)。这样我就可以使用相同的简单代码,无论存储的类型如何,即使使用DoStuff特殊的矢量化实现,我也希望这是真实的。

2 个答案:

答案 0 :(得分:3)

您必须在类之外专门化模板成员函数:

#include <iostream>

class Base
{
    public:
    virtual int DoStuffVectorized(int arg) = 0;
};

template <typename T>
class Derived : public Base
{
    public:
    int DoStuffVectorized(int arg) override;
};

template <>
int Derived<char>::DoStuffVectorized(int arg)
{
     std::cout <<  "T == char\n";
     return 0;
}

template <>
int Derived<int>::DoStuffVectorized(int arg)
{
     std::cout <<  "T == int\n";
     return 0;
}

int main(){
    Derived<char> c;
    Derived<int> i;
    Base* b[] = { &c, &i };
    for(auto* x : b)
        x->DoStuffVectorized(0);
    // undefined reference to `Derived<double>::DoStuffVectorized(int)'
    // Derived<double> d;
}

如果要在编译时捕获非预期的实例化:

#include <type_traits>

// A std::false_type (useful in a static_assert)
template <typename T>
struct static_false : std::false_type
{};

template <typename T>
int Derived<T>::DoStuffVectorized(int arg)
{
    static_assert(static_false<T>::value, "Neither 'char' or 'int'");
    return 0;
}

答案 1 :(得分:2)

DoStuffVectorized<char>语法不正确,DoStuffVectorized不是模板本身。

请参阅template specialization

template <typename T>
class Derived : public Base
{
    int DoStuff(int arg) override
    {
         // do some stuff
         return 0;
    }
    int DoStuffVectorized(int arg) override
    {
         // do some stuff (primary template)
         return 0;
    }
};

template <>
int Derived<int>::DoStuffVectorized(int) {
     // do some stuff for T == char
     return 0;
}

template <>
int Derived<char>::DoStuffVectorized(int) {
     // do some stuff for T == char
     return 0;
}