C++使用模版实现按成员变量或者成员函数排序

来源：互联网发布：身价最高淘宝模特是谁编辑：程序博客网时间：2024/06/01 07:33

例如有一个这样的类:

class CA{public:    CA():      m_a(0),      m_b(0)      {      }      CA(int a, int b):m_a(a),m_b(b){}    int m_a;    int m_b;    int Func1()    {        return m_a + m_b;    }    double Func2( )    {        return double( m_a - m_b);    }};

有一个该类的list：

std::list<CA> listA;

现在不同场合需要根据成员变量m_a进行排序，有时需要根据m_b排序，有时还需要根据Func1的返回值进行排序。为了实现这个功能，折腾了两个模版类。

template<class T1, class T2>class CCmpTT{public:     CCmpTT(T1 T2::*ptr);     bool operator()(T2 v1, T2 v2);     T1 T2::*m_ptr;};template<class T1, class T2>CCmpTT<T1, T2>::CCmpTT(T1 T2::*ptr){    m_ptr = ptr;}template<class T1, class T2>bool CCmpTT<T1, T2>::operator()(T2 v1, T2 v2){    return v1.*m_ptr < v2.*m_ptr;}#define Get_CmpByMemberValue(ClassType, memberType, member) (CCmpTT<memberType,ClassType>(&ClassType::member))template<class T1, class T2>class CCmpTTFunc{public:    CCmpTTFunc(T1 (T2::*ptr)());    bool operator()(T2 v1, T2 v2);    T1 (T2::*m_ptr)();};template<class T1, class T2>CCmpTTFunc<T1, T2>::CCmpTTFunc(T1 (T2::*ptr)()){    m_ptr = ptr;}template<class T1, class T2>bool CCmpTTFunc<T1, T2>::operator()(T2 v1, T2 v2){    return (v1.*m_ptr)() < (v2.*m_ptr)();}#define Get_CmpByMemberFunc(ClassType, FuncRetType, Func) (CCmpTTFunc<FuncRetType, ClassType>(&ClassType::Func))

这两组模版和宏定义分别实现了按照某个类的成员变量或者函数获得比较函数。使用方法如下：

    listA.sort( Get_CmpByMemberValue(CA, int, m_a));    listA.sort( Get_CmpByMemberValue(CA, int, m_b));    listA.sort( Get_CmpByMemberFunc(CA, int, Func1));    listA.sort( Get_CmpByMemberFunc(CA, double, Func2));

继续改进：经陈启华同学建议，不要使用宏，使用模版函数会更简单：

template<typename T1, typename T2> CCmpTT<T1, T2> Get_CMP( T1 T2::*ptr){    return CCmpTT<T1, T2>( ptr );} template<typename T1, typename T2> CCmpTTFunc<T1, T2> Get_CMPByFunc( T1 (T2::*ptr)() ){    return CCmpTTFunc<T1, T2>( ptr );}

使用如下：

    listA.sort( Get_CMP(&CA::m_a));    listA.sort( Get_CMP(&CA::m_b));    listA.sort( Get_CMPByFunc(&CA::Func1));    listA.sort( Get_CMPByFunc( &CA::Func2 ));

TODO: 增加到但参数的成员函数的支持。