1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
/*
* Copyright 2008-2013 NVIDIA Corporation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once
#include <thrust/detail/config.h>
#include <thrust/iterator/iterator_categories.h>
#include <thrust/iterator/detail/iterator_category_to_traversal.h>
#include <thrust/detail/type_traits.h>
#include <thrust/type_traits/void_t.h>
THRUST_NAMESPACE_BEGIN
template<typename Iterator>
struct iterator_value
{
typedef typename thrust::iterator_traits<Iterator>::value_type type;
}; // end iterator_value
template <typename Iterator>
using iterator_value_t = typename iterator_value<Iterator>::type;
template<typename Iterator>
struct iterator_pointer
{
typedef typename thrust::iterator_traits<Iterator>::pointer type;
}; // end iterator_pointer
template <typename Iterator>
using iterator_pointer_t = typename iterator_pointer<Iterator>::type;
template<typename Iterator>
struct iterator_reference
{
typedef typename iterator_traits<Iterator>::reference type;
}; // end iterator_reference
template <typename Iterator>
using iterator_reference_t = typename iterator_reference<Iterator>::type;
template<typename Iterator>
struct iterator_difference
{
typedef typename thrust::iterator_traits<Iterator>::difference_type type;
}; // end iterator_difference
template <typename Iterator>
using iterator_difference_t = typename iterator_difference<Iterator>::type;
namespace detail
{
template <typename Iterator, typename = void>
struct iterator_system_impl {};
template <typename Iterator>
struct iterator_system_impl<
Iterator
, typename voider<
typename iterator_traits<Iterator>::iterator_category
>::type
>
: detail::iterator_category_to_system<
typename iterator_traits<Iterator>::iterator_category
>
{};
} // namespace detail
template <typename Iterator>
struct iterator_system : detail::iterator_system_impl<Iterator> {};
// specialize iterator_system for void *, which has no category
template<>
struct iterator_system<void *>
{
typedef thrust::iterator_system<int*>::type type;
}; // end iterator_system<void*>
template<>
struct iterator_system<const void *>
{
typedef thrust::iterator_system<const int*>::type type;
}; // end iterator_system<void*>
template <typename Iterator>
using iterator_system_t = typename iterator_system<Iterator>::type;
template <typename Iterator>
struct iterator_traversal
: detail::iterator_category_to_traversal<
typename thrust::iterator_traits<Iterator>::iterator_category
>
{
}; // end iterator_traversal
namespace detail
{
template <typename T>
struct is_iterator_traversal
: thrust::detail::is_convertible<T, incrementable_traversal_tag>
{
}; // end is_iterator_traversal
template<typename T>
struct is_iterator_system
: detail::or_<
detail::is_convertible<T, any_system_tag>,
detail::or_<
detail::is_convertible<T, host_system_tag>,
detail::is_convertible<T, device_system_tag>
>
>
{
}; // end is_iterator_system
} // end namespace detail
THRUST_NAMESPACE_END
|
