NVBIO
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
transform_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 
28 #pragma once
29 
30 #include <nvbio/basic/types.h>
31 #include <nvbio/basic/iterator.h>
32 
33 namespace nvbio {
34 
37 
40 
45 template <typename T, typename Transform>
46 struct transform_iterator
47 {
48  typedef typename Transform::result_type value_type;
49  typedef value_type& reference;
50  typedef value_type const_reference;
51  typedef value_type* pointer;
52  typedef typename std::iterator_traits<T>::difference_type difference_type;
53  //typedef typename std::iterator_traits<T>::distance_type distance_type;
54  typedef typename std::iterator_traits<T>::iterator_category iterator_category;
55 
58  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
59  transform_iterator() {}
60 
63  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
64  transform_iterator(const T base, const Transform f) : m_base( base ), m_f( f ) {}
65 
68  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
69  transform_iterator(const transform_iterator& it) : m_base( it.m_base ), m_f( it.m_f ) {}
70 
73  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
74  const_reference operator[](const uint32 i) const { return m_f( m_base[i] ); }
75 
78  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
79  value_type operator*() const { return m_f( m_base[0] ); }
80 
83  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
84  transform_iterator<T,Transform>& operator++()
85  {
86  ++m_base;
87  return *this;
88  }
89 
92  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
93  transform_iterator<T,Transform> operator++(int i)
94  {
95  transform_iterator<T,Transform> r( m_base, m_f );
96  ++m_base;
97  return r;
98  }
99 
102  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
103  transform_iterator<T,Transform>& operator--()
104  {
105  --m_base;
106  return *this;
107  }
108 
111  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
112  transform_iterator<T,Transform> operator--(int i)
113  {
114  transform_iterator<T,Transform> r( m_base, m_f );
115  --m_base;
116  return r;
117  }
118 
121  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
122  transform_iterator<T,Transform> operator+(const difference_type i) const
123  {
124  return transform_iterator( m_base + i, m_f );
125  }
126 
129  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
130  transform_iterator<T,Transform> operator-(const difference_type i) const
131  {
132  return transform_iterator( m_base - i, m_f );
133  }
134 
137  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
138  transform_iterator<T,Transform>& operator+=(const difference_type i)
139  {
140  m_base += i;
141  return *this;
142  }
143 
146  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
147  transform_iterator<T,Transform>& operator-=(const difference_type i)
148  {
149  m_base -= i;
150  return *this;
151  }
152 
155  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
156  difference_type operator-(const transform_iterator<T,Transform> it) const
157  {
158  return m_base - it.m_base;
159  }
160 
163  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
164  transform_iterator& operator=(const transform_iterator<T,Transform>& it)
165  {
166  m_base = it.m_base;
167  m_f = it.m_f;
168  return *this;
169  }
170 
171  T m_base;
172  Transform m_f;
173 };
174 
177 template <typename T, typename Transform>
178 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
179 transform_iterator<T,Transform> make_transform_iterator(const T it, const Transform f)
180 {
181  return transform_iterator<T,Transform>( it, f );
182 }
183 
184 
187 template <typename T, typename Transform>
188 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
189 bool operator==(const transform_iterator<T,Transform> it1, const transform_iterator<T,Transform> it2)
190 {
191  return (it1.m_base == it2.m_base)/* && (it1.m_f == it2.m_f)*/;
192 }
195 template <typename T, typename Transform>
196 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
197 bool operator!=(const transform_iterator<T,Transform> it1, const transform_iterator<T,Transform> it2)
198 {
199  return (it1.m_base != it2.m_base)/* || (it1.m_f != it2.m_f)*/;
200 }
203 template <typename T, typename Transform>
204 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
205 bool operator<(const transform_iterator<T,Transform> it1, const transform_iterator<T,Transform> it2) { return (it1.m_base < it2.m_base); }
208 template <typename T, typename Transform>
209 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
210 bool operator<=(const transform_iterator<T,Transform> it1, const transform_iterator<T,Transform> it2) { return (it1.m_base <= it2.m_base); }
213 template <typename T, typename Transform>
214 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
215 bool operator>(const transform_iterator<T,Transform> it1, const transform_iterator<T,Transform> it2) { return (it1.m_base > it2.m_base); }
218 template <typename T, typename Transform>
219 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
220 bool operator>=(const transform_iterator<T,Transform> it1, const transform_iterator<T,Transform> it2) { return (it1.m_base >= it2.m_base); }
221 
222 
227 template <typename T1, typename T2, typename Transform>
228 struct binary_transform_iterator
229 {
230  typedef typename Transform::result_type value_type;
231  typedef value_type& reference;
232  typedef value_type const_reference;
233  typedef value_type* pointer;
234  typedef typename std::iterator_traits<T1>::difference_type difference_type;
235  //typedef typename std::iterator_traits<T1>::distance_type distance_type;
236  typedef typename std::iterator_traits<T1>::iterator_category iterator_category;
237 
240  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
241  binary_transform_iterator() {}
242 
245  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
246  binary_transform_iterator(const T1 base1, const T2 base2, const Transform f) : m_base1( base1 ), m_base2( base2 ), m_f( f ) {}
247 
250  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
251  binary_transform_iterator(const binary_transform_iterator& it) : m_base1( it.m_base1 ), m_base2( it.m_base2 ), m_f( it.m_f ) {}
252 
255  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
256  const_reference operator[](const uint32 i) const { return m_f( m_base1[i], m_base2[i] ); }
257 
260  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
261  value_type operator*() const { return m_f( m_base1[0], m_base2[0] ); }
262 
265  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
266  binary_transform_iterator<T1,T2,Transform>& operator++()
267  {
268  ++m_base1;
269  ++m_base2;
270  return *this;
271  }
272 
275  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
276  binary_transform_iterator<T1,T2,Transform> operator++(int i)
277  {
278  binary_transform_iterator<T1,T2,Transform> r( m_base1, m_base2, m_f );
279  ++m_base1;
280  ++m_base2;
281  return r;
282  }
283 
286  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
287  binary_transform_iterator<T1,T2,Transform>& operator--()
288  {
289  --m_base1;
290  --m_base2;
291  return *this;
292  }
293 
296  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
297  binary_transform_iterator<T1,T2,Transform> operator--(int i)
298  {
299  binary_transform_iterator<T1,T2,Transform> r( m_base1, m_base2, m_f );
300  --m_base1;
301  --m_base2;
302  return r;
303  }
304 
307  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
308  binary_transform_iterator<T1,T2,Transform> operator+(const difference_type i) const
309  {
310  return binary_transform_iterator( m_base1 + i, m_base2 + i, m_f );
311  }
312 
315  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
316  binary_transform_iterator<T1,T2,Transform> operator-(const difference_type i) const
317  {
318  return binary_transform_iterator( m_base1 - i, m_base2 - i, m_f );
319  }
320 
323  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
324  binary_transform_iterator<T1,T2,Transform>& operator+=(const difference_type i)
325  {
326  m_base1 += i;
327  m_base2 += i;
328  return *this;
329  }
330 
333  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
334  binary_transform_iterator<T1,T2,Transform>& operator-=(const difference_type i)
335  {
336  m_base1 -= i;
337  m_base2 -= i;
338  return *this;
339  }
340 
343  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
344  difference_type operator-(const binary_transform_iterator<T1,T2,Transform> it) const
345  {
346  return m_base1 - it.m_base1;
347  }
348 
351  NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
352  binary_transform_iterator& operator=(const binary_transform_iterator<T1,T2,Transform>& it)
353  {
354  m_base1 = it.m_base1;
355  m_base2 = it.m_base2;
356  m_f = it.m_f;
357  return *this;
358  }
359 
360  T1 m_base1;
361  T2 m_base2;
362  Transform m_f;
363 };
364 
367 template <typename T1, typename T2, typename Transform>
368 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
369 binary_transform_iterator<T1,T2,Transform> make_binary_transform_iterator(const T1 it1, const T2 it2, const Transform f)
370 {
371  return binary_transform_iterator<T1,T2,Transform>( it1, it2, f );
372 }
373 
374 
377 template <typename T1, typename T2, typename Transform>
378 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
379 bool operator==(const binary_transform_iterator<T1,T2,Transform> it1, const binary_transform_iterator<T1,T2,Transform> it2)
380 {
381  return (it1.m_base1 == it2.m_base1) && (it1.m_base2 == it2.m_base2)/* && (it1.m_f == it2.m_f)*/;
382 }
385 template <typename T1, typename T2, typename Transform>
386 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
387 bool operator!=(const binary_transform_iterator<T1,T2,Transform> it1, const binary_transform_iterator<T1,T2,Transform> it2)
388 {
389  return (it1.m_base1 != it2.m_base1) || (it1.m_base2 != it2.m_base2)/* || (it1.m_f != it2.m_f)*/;
390 }
393 template <typename T1, typename T2, typename Transform>
394 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
395 bool operator<(const binary_transform_iterator<T1,T2,Transform> it1, const binary_transform_iterator<T1,T2,Transform> it2) { return (it1.m_base1 < it2.m_base1); }
398 template <typename T1, typename T2, typename Transform>
399 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
400 bool operator<=(const binary_transform_iterator<T1,T2,Transform> it1, const binary_transform_iterator<T1,T2,Transform> it2) { return (it1.m_base1 <= it2.m_base1); }
403 template <typename T1, typename T2, typename Transform>
404 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
405 bool operator>(const binary_transform_iterator<T1,T2,Transform> it1, const binary_transform_iterator<T1,T2,Transform> it2) { return (it1.m_base1 > it2.m_base1); }
408 template <typename T1, typename T2, typename Transform>
409 NVBIO_FORCEINLINE NVBIO_HOST_DEVICE
410 bool operator>=(const binary_transform_iterator<T1,T2,Transform> it1, const binary_transform_iterator<T1,T2,Transform> it2) { return (it1.m_base1 >= it2.m_base1); }
411 
414 
415 } // namespace nvbio
Generated on Wed Feb 25 2015 08:32:57 for NVBIO by   doxygen 1.8.4