stl源码学习之traits技术

简介

第一篇文件我们先来试试水，在SGI STL中 ,你随处可见对模板的操作，在阅读源码的时候阻碍是非常的多的（要是你特别厉害当我没说），我们今天第一篇先来尝试着学习下模板的类型推导，做个开胃小菜。

#类型推导

阅读以下代码，不知道看官是否能自己推导出T是int类型，这个应该不难，不管怎么样我们一起走一圈。从main开始。我们看到 func(&i)，这里的函数接受了一个int的参数，然后看到void func(I iter) 函数定义的地方，这块可以看到传给func_temp 的分别是int 和 int类型（对int* 解引用了）， 然后就顺理成章的推导出来T 是int了，是不是很简单。

template <class I , class T>
void func_temp(I iter, T t) {
    T tmp ;  //这里可以推到出参数T 是int
}

template <class I>
void func(I iter){
    func_temp(iter, *iter);
}


int main() {
    int i ;
    func(&i) ;
    return 0;
}

这个就是 template的参数推导机制，但是他没办法对返回值进行推导，我们需要其他的机制来解决这个问题，我们可以通过声明内嵌类型来解决这个问题，观察如下代码，我们来简单的推到下， 先看MyIter , 其中的value_type 就是T类型了，好了再返回去砍main，func接受了一个MyIter类型，看到typename I::value_type func(I iter)定义的地方，返回值就变成了MyIter::value_type , 这里就可以判断出返回值就是int了。

template<class T>
struct MyIter {
    typedef  T value_type;

    T* ptr;
    MyIter(T* p = 0 ) : ptr(p) {}
    T& operator*() const {return *ptr;}
    //....
};

template <class I>
typename I::value_type func(I iter) {
    return  *iter;
}

int main () {


    MyIter<int> ite(new  int(8));
    std::cout << func(ite);
    return  0;
}

但这个代码在stl迭代器类型中会有大量的运用，但是这个有个陷阱，卖个关子后面再说。有了这部分知识之后我们来可以来看下traits了，traits的中文翻译是特性,有人把他比作榨汁机，可以去提取不同类的共性，以便能够统一处理，traits依靠显示模板特化，把代码中因类型不同而发生的片段提出，用统一的接口来包装，这个接⼝可以包含⼀个C++类所能包含的任何东西，如内嵌类型，成员函数、成员变量。作为客⼾的模板代码，最终可以通过traits模板类所公开的接⼝来间接访问。

我们先来观察下这部分代码

#define N 10
class CIntArray{
public:
    CIntArray() {
        for(int i = 0; i < 10; i++){
            arr[i] = 100 + i;
        }
    }

    int getSum(int times) {
        int total = 0;
        for (int i = 0; i < 10; i++){
            total += arr[i];
        }
        return  total * times;
    }

private:
    int arr[N];
};

class CFloatArray{
public:
    CFloatArray() {
        for(int i = 0; i < 10; i++){
            arr[i] = 100.11 + i;
        }
    }

    float getSum(float times) {
        float total = 0;
        for (int i = 0; i < 10; i++){
            total += arr[i];
        }
        return  total * times;
    }

private:
    float arr[N];
};

int main() {
    CIntArray cint;
    CFloatArray cfloat;
    cout<<cint.getSum(2)<<endl;
    cout<<cfloat.getSum(2.0)<<endl;

    return 0;
}

接下来我们为这部分代码添加一个统一的接口

#define N 10
class CIntArray{
public:
    CIntArray() {
        for(int i = 0; i < 10; i++){
            arr[i] = 100 + i;
        }
    }

    int getSum(int times) {
        int total = 0;
        for (int i = 0; i < 10; i++){
            total += arr[i];
        }
        return  total * times;
    }

private:
    int arr[N];
};


class CFloatArray{
public:
    CFloatArray() {
        for(int i = 0; i < 10; i++){
            arr[i] = 100.11 + i;
        }
    }

    float getSum(float times) {
        float total = 0;
        for (int i = 0; i < 10; i++){
            total += arr[i];
        }
        return  total * times;
    }

private:
    float arr[N];
};

template <class T>
class CApply
{
public:
    int getSum(T& t, int times){
        return t.getSum(times);
    }
};


int main() {
    CIntArray cint;
    CFloatArray cfloat;

    CApply<CIntArray> cl;
    CApply<CFloatArray> c2;

    cout<< cl.getSum(cint , 2) <<endl;
    cout<<c2.getSum(cfloat , 2.0)<<endl;

    return 0;
}

这部分代码已经提供了一个统一的接口，但是你会发现现在返回值只能是int，当要float的时候，就会损失很多精度，这是我们不希望看到的。我们设想下，能不能根据传入的类型自动去匹配呢。这是就可以使用返回值的推导了。
代码如下

#include <iostream>

using  namespace  std;


#define N 10
class CIntArray{
public:
    CIntArray() {
        for(int i = 0; i < 10; i++){
            arr[i] = 100 + i;
        }
    }

    int getSum(int times) {
        int total = 0;
        for (int i = 0; i < 10; i++){
            total += arr[i];
        }
        return  total * times;
    }

private:
    int arr[N];
};


class CFloatArray{
public:
    CFloatArray() {
        for(int i = 0; i < 10; i++){
            arr[i] = 100.11 + i;
        }
    }

    float getSum(float times) {
        float total = 0;
        for (int i = 0; i < 10; i++){
            total += arr[i];
        }
        return  total * times;
    }

private:
    float arr[N];
};

template<typename T>
struct type_traits{};

//特化模板
template<>
struct  type_traits<CIntArray>{
    typedef int result_type;
    typedef int arg_type;
};

template<>
struct  type_traits<CFloatArray>{
    typedef float result_type;
    typedef float arg_type;
};


template <class T>
class CApply
{
public:
    typename type_traits<T>::result_type getSum(T& t,typename type_traits<T>::arg_type times){
        return t.getSum(times);
    }
};





int main() {
    CIntArray cint;
    CFloatArray cfloat;

    CApply<CIntArray> cl;
    CApply<CFloatArray> c2;

    cout<< cl.getSum(cint , 2) <<endl;
    cout<<c2.getSum(cfloat , 2.0)<<endl;

    return 0;
}

理解以上代码的关键就是需要理解模板的特化，其次需要明白traits的目的是去解决的问题（根据对象的不同，能够提取到不同的数据类型）；

总结：
以上是traits的主要思想，SGI STL源码中，你四处可以看到他的身影。你需要做到理解traits的关键点与解决的问题，做到心中有数手中不慌，这样才能看得下去STL源码。