使用zip_iterator迭代器、make_tuple、thrust::get<0>

东方耀

使用zip_iterator迭代器、make_tuple、thrust::get<0>


// 1、stl的容器都默认会有迭代器的 通用的
// 2、不同容器的迭代器可以组合成一个元组 thrust::tuple
// 3、迭代器元组可以被封装成zip_iterator  thrust::zip_iterator
// 4、zip_iterator的初始化       thrust::make_tuple
// 5、最终我们使用的就是一个zip_iterator的迭代器对象，像使用stl中的普通迭代器一样
// 6、通过zip_iterator取值 thrust::get<1>(zip_iter[2])

1. #include <thrust/device_vector.h>
   
2. #include <iostream>
   
3. #include <thrust/iterator/zip_iterator.h>
   
4. #include <thrust/tuple.h>
   
5. 
   
6. // 1、stl的容器都默认会有迭代器的 通用的
   
7. // 2、不同容器的迭代器可以组合成一个元组 thrust::tuple
   
8. // 3、迭代器元组可以被封装成zip_iterator  thrust::zip_iterator
   
9. // 4、zip_iterator的初始化       thrust::make_tuple
   
10. // 5、最终我们使用的就是一个zip_iterator的迭代器对象，像使用stl中的普通迭代器一样
    
11. // 6、通过zip_iterator取值 thrust::get<1>(zip_iter[2])
    
12. 
    
13. 
    
14. int main(){
    
15.     //使用zip_iterator迭代器、make_tuple、thrust::get<0>
    
16.     thrust::device_vector<int> int_v(3);
    
17.     int_v[0] = 0; int_v[1] = 1; int_v[2] = 2;
    
18.     thrust::device_vector<float> float_v(3);
    
19.     float_v[0] = 0.0f; float_v[1] = 1.0f; float_v[2] = 2.0f;
    
20.     thrust::device_vector<char> char_v(3);
    
21.     char_v[0] = 'a'; char_v[1] = 'b'; char_v[2] = 'c';
    
22. 
    
23. 
    
24.     // typedef these iterators for shorthand  stl的迭代器
    
25.     typedef thrust::device_vector<int>::iterator   IntIterator;
    
26.     typedef thrust::device_vector<float>::iterator FloatIterator;
    
27.     typedef thrust::device_vector<char>::iterator  CharIterator;
    
28.     // typedef a tuple of these iterators  定义一个 迭代器元组
    
29.     typedef thrust::tuple<IntIterator, FloatIterator, CharIterator> IteratorTuple;
    
30.     // typedef the zip_iterator of this tuple  迭代器元组外又封装了一层 zip迭代器
    
31.     typedef thrust::zip_iterator<IteratorTuple> ZipIterator;
    
32.     // finally, create the zip_iterator
    
33.     ZipIterator zip_iter(thrust::make_tuple(int_v.begin(), float_v.begin(), char_v.begin()));
    
34. 
    
35. 
    
36.     std::cout << "开始使用zip迭代器对象了!" << std::endl;
    
37.     // zip_iter[2] = (2, 2.0f, 'c')
    
38.     std::cout << thrust::get<0>(zip_iter[2]) << std::endl;  
    
39.     std::cout << thrust::get<1>(zip_iter[2]) << std::endl;  
    
40.     std::cout << thrust::get<2>(zip_iter[2]) << std::endl;  
    
41. 
    
42. 
    
43.     //std::cout << thrust::get<2>(zip_iter[3]) << std::endl;   越界错误
    
44. 
    
45. 
    
46.     // *zip_iter;   // returns (0, 0.0f, 'a')
    
47.     // zip_iter[0]; // returns (0, 0.0f, 'a')
    
48.     // zip_iter[1]; // returns (1, 1.0f, 'b')
    
49.     // zip_iter[2]; // returns (2, 2.0f, 'c')
    
50.     // thrust::get<0>(zip_iter[2]); // returns 2
    
51.     // thrust::get<1>(zip_iter[0]); // returns 0.0f
    
52.     // thrust::get<2>(zip_iter[1]); // returns 'b'
    
53.     // zip_iter[3] is an out-of-bounds error
    
54.    
    
55. 
    
56.     return 0;
    
57. }
    
58. 
    
59. 
    
60. 
    
61. 
    

复制代码