使用make_zip_iterator、make_tuple同时操作多个stl的迭代器

东方耀

// 1、使用make_zip_iterator、make_tuple同时操作多个stl的迭代器
// 2、泛型里面又是泛型：这种场景
// 3、泛型定义的时候应先满足 大类型，大类型里面的小类型用typedef即可

1. #include <thrust/device_vector.h>
   
2. #include <iostream>
   
3. #include <thrust/iterator/zip_iterator.h>
   
4. #include <thrust/tuple.h>
   
5. #include <string>
   
6. 
   
7. // 1、使用make_zip_iterator、make_tuple同时操作多个stl的迭代器
   
8. // 2、泛型里面又是泛型：这种场景
   
9. // 3、泛型定义的时候应先满足 大类型，大类型里面的小类型用typedef即可
   
10. 
    
11. 
    
12. // template <typename T>
    
13. // void print_vector2(thrust::device_vector<T> &v){
    
14. //     // 这样写 为什么不对？
    
15. //     thrust::device_vector<T>::iterator it;
    
16. //     for(it, it!=v.end();it++){
    
17. //         std::cout << *it << std::endl;
    
18. //     }
    
19. // }
    
20. 
    
21. 
    
22. // template <typename T>
    
23. // void print_vector(thrust::device_vector<T> &v){
    
24. //     thrust::device_vector<T>::iterator it;
    
25. //     for(it, it!=v.end();it++){
    
26. //         std::cout << *it << std::endl;
    
27. //     }
    
28. // }
    
29. 
    
30. 
    
31. template <typename Vector>
    
32. void print(const std::string& s, const Vector& v)
    
33. {
    
34.   //v这个类型为：thrust::device_vector<int> 或 thrust::device_vector<float>
    
35.   //本身就是泛型啊  泛型里面又是泛型：这种场景
    
36.   typedef typename Vector::value_type T;  //这个操作比较关键？  T才相当于int 或 float
    
37.   // 泛型定义的时候应先满足 大类型，大类型里面的小类型用typedef即可
    
38. 
    
39.   std::cout << s;
    
40.   thrust::copy(v.begin(), v.end(), std::ostream_iterator<T>(std::cout, " "));
    
41.   std::cout << std::endl;
    
42. }
    
43. 
    
44. 
    
45. int main(){
    
46.     thrust::device_vector<int> int_in(3), int_out(3);
    
47.     int_in[0] = 0;
    
48.     int_in[1] = 1;
    
49.     int_in[2] = 2;
    
50.     thrust::device_vector<float> float_in(3), float_out(3);
    
51.     float_in[0] =  0.0f;
    
52.     float_in[1] = 10.0f;
    
53.     float_in[2] = 20.0f;
    
54.     // copy (first, last, new_first)
    
55.     // 使用make_zip_iterator、make_tuple同时操作多个stl的迭代器
    
56.     thrust::copy(thrust::make_zip_iterator(thrust::make_tuple(int_in.begin(), float_in.begin())),
    
57.                  thrust::make_zip_iterator(thrust::make_tuple(int_in.end(),   float_in.end())),
    
58.                  thrust::make_zip_iterator(thrust::make_tuple(int_out.begin(),float_out.begin())));
    
59.     // int_out is now [0, 1, 2]
    
60.     //print_vector<int>(int_out);
    
61.     print<thrust::device_vector<int>>("看int_out_jjj=", int_out);
    
62.     print("看int_out(调模板函数，简写，不合理啊)=", int_out);   //为啥不用<>啊
    
63.     // float_out is now [0.0f, 10.0f, 20.0f]
    
64.     print<thrust::device_vector<float>>("看float_out=", float_out);
    
65.     //print_vector<float>(float_out);
    
66.    
    
67. 
    
68.     return 0;
    
69. }
    
70. 
    
71. 
    
72. 
    
73. 
    

复制代码