在C++泛型编程中如何只特化类的某个成员函数

我们知道在C++模板编程中如果我们特化或是偏特化某个模板类， 我们需要重写整个模板类中的所有函数， 但是这些代码通常是非常相似的， 甚至在某些情况下可能只有一两个函数会不一样，其他函数都是一样的。在这种情况下，同时存在多份相同的代码，对我们维护这些代码是非常不利的， 我们最好只需要特化其中不一样的那个函数。

比如下面这个模板类：

 

template<typename T, unsigned B>
struct Base
{
     // other function
     // ....
     void Func(){ cout <<  " primary function " << endl; }
};
void test1()
{
    Base< int,  1> a;
    a.Func();
    Base< int,  16> b;
    b.Func();
}
int main()
{
     test1();
}

 

 

只有当B等于16时， Func这个函数需要特化， 但是其他函数无论什么情况下都是一样的。

下面是我们的一些可能解决方案:

方法1:

 

template<typename T>
struct Base<T,  16>
{
     // other function
     // ....
     void Func(){ cout <<  " specialization function " << endl; }
};

 

点评：通过偏特化实现，需要重写所有的类成员方法。

 

方法2：

 

template<typename T, unsigned B>
struct Base
{
     // other function
     // ....
     void Func()
    {
         if(B ==  16)
        {
            cout <<  " primary function " << endl;
        }
         else
        {
            cout <<  " specialization function " << endl;
        }
    }
};

 

点评: 通过运行时判断， 容易理解，但是相对低效。

 

方法3：

 

template<typename T, unsigned B>
struct Base
{
     // other function
     // ....
     void Func()
    {
#if B!=16
            cout <<  " primary function " << endl;
#else
            cout <<  " specialization function " << endl;
#endif
    }
};

 

点评： 试图通过预编译来实现，但是这个方法是错误的。C++模板编译包括预编译，语法检查，模板实例化等阶段，在预编译阶段模板参数都还没有实例化呢。

 

方法4：

 

template<typename T, unsigned B>
struct Base
{
     // other function
     // ....
    template<unsigned S>
     struct FuncObj
    {
         void  operator()()
        {
            cout<< " primary function "<<endl;
        }
    };
    template<>
     struct FuncObj< 16>
    {
         void  operator()()
        {
            cout<< " specialization function "<<endl;
        }
    };
    FuncObj<B> Func;
};

 

点评: 通过成员类以防函数的形式特化， 增加了类成员变量。

 

方法5：

 

template<typename T, unsigned B>
struct Base
{
     // other function
     // ....
    template<unsigned N>
     void FuncImpl()
    {
        cout<< " primary function "<<endl;
    }
    template<>
     void FuncImpl< 16>()
    {
        cout<< " specialization function "<<endl;
    }
     void Func()
    {
        FuncImpl<B>();
    }
};

 

点评：通过类成员模板函数特化来实现。

 

方法6：

 

template<typename T, unsigned B>
struct Base
{
     // other function
     // ....
    template<unsigned N> 
     class Int2Type
    {
         enum { value = N };
    };
    template<unsigned V>
     void FuncImpl( const Int2Type<V>)
    {
        cout<< " primary function "<<endl;
    }
     void FuncImpl( const Int2Type< 16>)
    {
        cout<< " specialization function "<<endl;
    }
     void Func()
    {
        FuncImpl(Int2Type<B>());
    }
};

 

点评： 通过将int根据值的不同转成不同的类型，然后通过函数重载实现。

 

方法7：

 

namespace
{
    template < bool,typename T,typename>  struct conditional { typedef T type; };
    template <typename T,typename U>  struct conditional< false,T,U> {typedef U type; };
}
template< class T, unsigned B>
struct Base
{
     // other function
     // ....

     void Func ()
    {
        typedef typename ::conditional<B!= 16,primary_t,spec_t>::type type;
        Func_impl(type());
    }
private:
     struct primary_t { };
     struct spec_t    { };
     void Func_impl (primary_t) { std::cout <<  " primary function " << std::endl; }
     void Func_impl (spec_t   ) { std::cout <<  " specialization function " << std::endl; }
};

 

点评： 和方法6类似，通过函数重载实现

 

方法8：

 namespace

 {
    template < bool,typename T =  void>  struct enable_if { typedef T type; };
    template <typename T>  struct enable_if< true,T> {};
}
template< class T, unsigned B>
struct Base
{
     // other function
     // ....

    template <unsigned N>
    typename ::enable_if< 16!=N>::type
        FuncImpl () { std::cout <<  " primary function " << std::endl; }
    template <unsigned N>
    typename ::enable_if< 16==N>::type
        FuncImpl () { std::cout <<  " specialization function " << std::endl; }
     void Func() {
        FuncImpl<B>();
    }
};

点评：通过enable_if, 利用SFINAE实现。

 

我们可以看到根据编译时模板参数int值的不同，我们重写模板类的某个成员函数的方法是多种多样的。针对上面这种情况，个人其实最推荐方法2，我们没必要把简单的问题复杂化。

 

下面我们考虑另外一个需求， 当模板类的某个参数是某种类型时， 我们要求特化其中的一个成员函数:

 

template<typename T1, typename T2>
struct Base
{
     // other function
     // ....
     void Func(){ cout <<  " primary function " << endl; }
};
void test2()
{
    Base< int,  int> a;
    a.Func();
    Base< int,  string> b;
    b.Func();
}
int main()
{
    test2();
}

 

要求上面的模板类如果T2 是string类型， 我们要求对Func特殊重写，其他的成员函数无论什么情况实现都是一样的。

有了上面的那个例子的实现经验， 对这个问题我们解决就方便多了。

 

方法1：

 

template<typename T1, typename T2>
struct Base
{
     // other function
     // ....
     void Func()
    {
         if(typeid(std:: string) == typeid(T2))
        {
            cout<< " specialization function "<<endl;
        }
         else
        {
            cout <<  " primary function " << endl; 
        }
    }
};

 

点评：通过运行时类型识别(RTTI)实现，需要打开相关编译选项，并且低效。

 

方法2：

 

template<typename T1, typename T2>
struct Base
{
     // other function
     // ....
    template<typename T>
     void FuncImpl()
    {
        cout <<  " primary function " << endl; 
    }
    template<>
     void FuncImpl< string>()
    {
        cout <<  " specialization function " << endl; 
    }
     void Func()
    {
        FuncImpl<T2>();
    }
};

 

点评：通过成员函数特化实现

 

方法3：

 

template<typename T1, typename T2>
struct Base
{
     // other function
     // ....
    template<typename T> 
     class Type2Type
    {
        typedef T type;
    };
    template<typename T>
     void FunImpl( const Type2Type<T>)
    {
        cout <<  " primary function " << endl; 
    }
    template<typename T>
     void FunImpl( const Type2Type< string>)
    {
        cout <<  " specialization function " << endl; 
    }
     void Func()
    {
        FunImpl<T2>(Type2Type<T2>());
    }
};

 

点评： 通过函数重载实现

 

方法4：

 

template<typename T>
struct IsString
{
     enum { value =  false };
};
template<>
struct IsString< string>
{
     enum { value =  true };
};
template<typename T1, typename T2>
struct Base
{
     // other function
     // ....
     void Func()
    { 
         if(IsString<T2>::value)
        {
            cout <<  " specialization function " << endl; 
        }
         else
        {
            cout <<  " primary function " << endl; 
        }
    }
};

 

点评： 通过编译时类型判断实现。

 

方法5：

 

template<typename T3,  typename T4>
struct must_be_same_type
{
     enum { ret =  0 };
};
template<>
struct must_be_same_type< string,  string>
{
     enum { ret =  1 };
};
template < typename T1,typename T2 >
class Base{
public:
     // other function
     // ....
     void Func(){
         if(must_be_same_type<T2,  string>::ret)
        {
            cout <<  " specialization function " << endl; 
        }
         else
        {
            cout <<  " primary function " << endl; 
        }
    }
};

 

点评： 和方法4类似， 是不过实现方式不一样。

 

最后，探讨下我自己遇到的问题， 我们在写一个事件委托(delegate)类，大概如下：

 

template<typename return_type, typename first_type, typename second_type>
class CEvent 
{
public:
     // other function
     // ....
    return_type  operator()(first_type p1, second_type p2)
    {
        return_type ret = return_type();
         // ...
         // ret = invoker(p1, p2);
         return ret;
    }
};
void test3()
{
    CEvent< int,  int,  int> e1;
    e1( 1,  2);
    CEvent< void,  int,  int> e2;
    e2( 1,  2);
}
int main()
{
    test3();
}

 

我们可以看到，当return_type是void时， 因为没有返回值，上面的代码会编译失败，因此我们只能偏特化这种情况：

 

template<typename first_type, typename second_type>
class CEvent< void, first_type, second_type>
{
public:
     // other function
     // ....
     void  operator()(first_type p1, second_type p2)
    {
         // ...
         // invoker(p1, p2);
         return;
    }
};

 

但是，我们会发现只有这个operator()函数是需要根据return_type特殊对待的，其他函数永远都是一样的。

我们现在的问题就是如何只特化这个函数。

 

首先我们会想到如下的实现方法：

 

template<typename T>
struct IsVoid
{
     enum { value =  false };
};
template<>
struct IsVoid< void>
{
     enum { value =  true };
};
template<typename return_type, typename first_type, typename second_type>
class CEvent 
{
public:
    other function
    ....
    return_type  operator()(first_type p1, second_type p2)
    {
         if(IsVoid<return_type>::value)
        {
            cout <<  " return type is void " << endl;
             // ...
             // invoker(p1, p2);
        }
         else
        {
            cout <<  " return type is not void " << endl;
            return_type ret = return_type();
             // ...
             // ret = invoker(p1, p2);
             return ret;
        }
    }
};

 

但是我们很快会发现这种情况下if语句被编译进去了， 所以return_type是void的情况下还是会编译失败。

我们要解决的问题就是如何把这个if语句变成函数重载，于是我们想到如下实现：

 

template<typename T>
struct IsVoid
{
     enum { value =  false };
};
template<>
struct IsVoid< void>
{
     enum { value =  true };
};
template< int v>
class Int2Type
{
     enum {value = v };
};
template<typename return_type, typename first_type, typename second_type>
class CEvent 
{
public:
     // other function
     // ....
    return_type InvokerImpl(first_type p1, second_type p2, Int2Type< true>)
    {
        cout <<  " return type is void " << endl;
         // ...
         // invoker(p1, p2);
    }
    return_type InvokerImpl(first_type p1, second_type p2, Int2Type< false>)
    {
        cout <<  " return type is not void " << endl;
        return_type ret = return_type();
         // ...
         // ret = invoker(p1, p2);
         return ret;
    }
    return_type  operator()(first_type p1, second_type p2)
    {
         return InvokerImpl(p1, p2, Int2Type<IsVoid<return_type>::value>());
    }
};

 

上面的实现首先通过编译时类型识别，然后再把识别后相应的bool值转成不同类型， 最后再利用不同类型函数重载实现。

 

最后总结下，我们可以看到，从编译时到运行时，从面向对象到普通泛型编程再到模板元编程，C++复杂得让人无语， 也强大得让人无语， 而且C++语言本身是在不断发展的(C++11)， 同一问题在C++中往往有多种解决方案，这些解决方案有的简单，有的复杂，有的高效， 也有的低效， 而我们的目标就是利用C++这把利器寻找简单而高效的解决方案。

 

注：本人初学C++ templates编程，如有错误，欢迎指正。

      参考资料：http://bbs.csdn.net/topics/390116038

                    http://bbs.csdn.net/topics/270041821