NVBIO
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
iterator.h
Go to the documentation of this file.
1 /*
2  * nvbio
3  * Copyright (c) 2011-2014, NVIDIA CORPORATION. All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  * * Redistributions of source code must retain the above copyright
8  * notice, this list of conditions and the following disclaimer.
9  * * Redistributions in binary form must reproduce the above copyright
10  * notice, this list of conditions and the following disclaimer in the
11  * documentation and/or other materials provided with the distribution.
12  * * Neither the name of the NVIDIA CORPORATION nor the
13  * names of its contributors may be used to endorse or promote products
14  * derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19  * DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANY
20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
33 #pragma once
34 
35 #include <iterator>
36 #include <nvbio/basic/types.h>
37 #include <thrust/iterator/iterator_categories.h>
38 
39 #if defined(__CUDACC__)
40 
41 namespace std
42 {
43 
44 // extend the std::iterator_traits with support for CUDA's __restrict__ pointers
45 template<class _Ty>
46 struct iterator_traits<const _Ty * __restrict__>
47 { // get traits from const pointer
48  typedef random_access_iterator_tag iterator_category;
49  typedef _Ty value_type;
50  typedef ptrdiff_t difference_type;
51  //typedef ptrdiff_t distance_type; // retained
52  typedef const _Ty* __restrict__ pointer;
53  typedef const _Ty& reference;
54 };
55 
56 } // namespace std
57 
58 #endif // __CUDACC__
59 
60 namespace nvbio {
61 
64 
65 typedef std::input_iterator_tag input_host_iterator_tag;
66 typedef std::output_iterator_tag output_host_iterator_tag;
67 typedef std::forward_iterator_tag forward_host_iterator_tag;
68 typedef std::bidirectional_iterator_tag bidirectional_host_iterator_tag;
69 typedef std::random_access_iterator_tag random_access_host_iterator_tag;
70 
71 typedef thrust::input_device_iterator_tag input_device_iterator_tag;
72 typedef thrust::output_device_iterator_tag output_device_iterator_tag;
73 typedef thrust::forward_device_iterator_tag forward_device_iterator_tag;
74 typedef thrust::bidirectional_device_iterator_tag bidirectional_device_iterator_tag;
75 typedef thrust::random_access_device_iterator_tag random_access_device_iterator_tag;
76 
77 typedef thrust::input_universal_iterator_tag input_universal_iterator_tag;
78 typedef thrust::output_universal_iterator_tag output_universal_iterator_tag;
79 typedef thrust::forward_universal_iterator_tag forward_universal_iterator_tag;
80 typedef thrust::bidirectional_universal_iterator_tag bidirectional_universal_iterator_tag;
81 typedef thrust::random_access_universal_iterator_tag random_access_universal_iterator_tag;
82 
83 template <typename iterator_category> struct iterator_category_system {};
84 template <> struct iterator_category_system<input_host_iterator_tag> { typedef host_tag type; };
85 template <> struct iterator_category_system<output_host_iterator_tag> { typedef host_tag type; };
86 template <> struct iterator_category_system<forward_host_iterator_tag> { typedef host_tag type; };
87 template <> struct iterator_category_system<bidirectional_host_iterator_tag> { typedef host_tag type; };
88 template <> struct iterator_category_system<random_access_host_iterator_tag> { typedef host_tag type; };
89 template <> struct iterator_category_system<input_device_iterator_tag> { typedef device_tag type; };
90 template <> struct iterator_category_system<output_device_iterator_tag> { typedef device_tag type; };
91 template <> struct iterator_category_system<forward_device_iterator_tag> { typedef device_tag type; };
92 template <> struct iterator_category_system<bidirectional_device_iterator_tag> { typedef device_tag type; };
93 template <> struct iterator_category_system<random_access_device_iterator_tag> { typedef device_tag type; };
94 template <> struct iterator_category_system<input_universal_iterator_tag> { typedef device_tag type; };
95 template <> struct iterator_category_system<output_universal_iterator_tag> { typedef device_tag type; };
96 template <> struct iterator_category_system<forward_universal_iterator_tag> { typedef device_tag type; };
97 template <> struct iterator_category_system<bidirectional_universal_iterator_tag> { typedef device_tag type; };
98 template <> struct iterator_category_system<random_access_universal_iterator_tag> { typedef device_tag type; };
99 
100 template <typename iterator>
101 struct iterator_system
102 {
103  typedef typename std::iterator_traits<iterator>::iterator_category iterator_category;
104  typedef typename iterator_category_system<iterator_category>::type type;
105 };
106 
109 template <typename T>
110 struct iterator_traits : public std::iterator_traits<T>
111 {
112  typedef typename std::iterator_traits<T>::iterator_category iterator_category;
113  typedef typename std::iterator_traits<T>::value_type value_type;
114  typedef typename std::iterator_traits<T>::difference_type difference_type;
115  typedef typename std::iterator_traits<T>::pointer pointer;
116  typedef typename std::iterator_traits<T>::reference reference;
117  typedef T forward_iterator;
118 };
119 
121 
122 } // namespace nvbio
Generated on Wed Feb 25 2015 08:32:55 for NVBIO by   doxygen 1.8.4