You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
277 lines
7.8 KiB
277 lines
7.8 KiB
/* |
|
* Copyright 2008-2012 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/detail/raw_reference_cast.h> |
|
#include <thrust/detail/type_traits.h> |
|
#include <thrust/detail/tuple_transform.h> |
|
|
|
namespace thrust |
|
{ |
|
namespace detail |
|
{ |
|
|
|
// specialize is_unwrappable |
|
// a tuple is_unwrappable if any of its elements is_unwrappable |
|
template< |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
struct is_unwrappable< |
|
thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
> |
|
: or_< |
|
is_unwrappable<T0>, |
|
is_unwrappable<T1>, |
|
is_unwrappable<T2>, |
|
is_unwrappable<T3>, |
|
is_unwrappable<T4>, |
|
is_unwrappable<T5>, |
|
is_unwrappable<T6>, |
|
is_unwrappable<T7>, |
|
is_unwrappable<T8>, |
|
is_unwrappable<T9> |
|
> |
|
{}; |
|
|
|
|
|
// specialize is_unwrappable |
|
// a tuple_of_iterator_references is_unwrappable if any of its elements is_unwrappable |
|
template< |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
struct is_unwrappable< |
|
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
> |
|
: or_< |
|
is_unwrappable<T0>, |
|
is_unwrappable<T1>, |
|
is_unwrappable<T2>, |
|
is_unwrappable<T3>, |
|
is_unwrappable<T4>, |
|
is_unwrappable<T5>, |
|
is_unwrappable<T6>, |
|
is_unwrappable<T7>, |
|
is_unwrappable<T8>, |
|
is_unwrappable<T9> |
|
> |
|
{}; |
|
|
|
|
|
namespace raw_reference_detail |
|
{ |
|
|
|
// unlike raw_reference, |
|
// raw_reference_tuple_helper needs to return a value |
|
// when it encounters one, rather than a reference |
|
// upon encountering tuple, recurse |
|
// |
|
// we want the following behavior: |
|
// 1. T -> T |
|
// 2. T& -> T& |
|
// 3. null_type -> null_type |
|
// 4. reference<T> -> T& |
|
// 5. tuple_of_iterator_references<T> -> tuple_of_iterator_references<raw_reference_tuple_helper<T>::type> |
|
|
|
|
|
// wrapped references are unwrapped using raw_reference, otherwise, return T |
|
template<typename T> |
|
struct raw_reference_tuple_helper |
|
: eval_if< |
|
is_unwrappable< |
|
typename remove_cv<T>::type |
|
>::value, |
|
raw_reference<T>, |
|
identity_<T> |
|
> |
|
{}; |
|
|
|
|
|
// recurse on tuples |
|
template < |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
struct raw_reference_tuple_helper< |
|
thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
> |
|
{ |
|
typedef thrust::tuple< |
|
typename raw_reference_tuple_helper<T0>::type, |
|
typename raw_reference_tuple_helper<T1>::type, |
|
typename raw_reference_tuple_helper<T2>::type, |
|
typename raw_reference_tuple_helper<T3>::type, |
|
typename raw_reference_tuple_helper<T4>::type, |
|
typename raw_reference_tuple_helper<T5>::type, |
|
typename raw_reference_tuple_helper<T6>::type, |
|
typename raw_reference_tuple_helper<T7>::type, |
|
typename raw_reference_tuple_helper<T8>::type, |
|
typename raw_reference_tuple_helper<T9>::type |
|
> type; |
|
}; |
|
|
|
|
|
template < |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
struct raw_reference_tuple_helper< |
|
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
> |
|
{ |
|
typedef thrust::detail::tuple_of_iterator_references< |
|
typename raw_reference_tuple_helper<T0>::type, |
|
typename raw_reference_tuple_helper<T1>::type, |
|
typename raw_reference_tuple_helper<T2>::type, |
|
typename raw_reference_tuple_helper<T3>::type, |
|
typename raw_reference_tuple_helper<T4>::type, |
|
typename raw_reference_tuple_helper<T5>::type, |
|
typename raw_reference_tuple_helper<T6>::type, |
|
typename raw_reference_tuple_helper<T7>::type, |
|
typename raw_reference_tuple_helper<T8>::type, |
|
typename raw_reference_tuple_helper<T9>::type |
|
> type; |
|
}; |
|
|
|
|
|
} // end raw_reference_detail |
|
|
|
|
|
// if a tuple "tuple_type" is_unwrappable, |
|
// then the raw_reference of tuple_type is a tuple of its members' raw_references |
|
// else the raw_reference of tuple_type is tuple_type & |
|
template < |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
struct raw_reference< |
|
thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
> |
|
{ |
|
private: |
|
typedef thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> tuple_type; |
|
|
|
public: |
|
typedef typename eval_if< |
|
is_unwrappable<tuple_type>::value, |
|
raw_reference_detail::raw_reference_tuple_helper<tuple_type>, |
|
add_reference<tuple_type> |
|
>::type type; |
|
}; |
|
|
|
|
|
template < |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
struct raw_reference< |
|
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
> |
|
{ |
|
private: |
|
typedef detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> tuple_type; |
|
|
|
public: |
|
typedef typename raw_reference_detail::raw_reference_tuple_helper<tuple_type>::type type; |
|
|
|
// XXX figure out why is_unwrappable seems to be broken for tuple_of_iterator_references |
|
//typedef typename eval_if< |
|
// is_unwrappable<tuple_type>::value, |
|
// raw_reference_detail::raw_reference_tuple_helper<tuple_type>, |
|
// add_reference<tuple_type> |
|
//>::type type; |
|
}; |
|
|
|
|
|
struct raw_reference_caster |
|
{ |
|
template<typename T> |
|
__host__ __device__ |
|
typename detail::raw_reference<T>::type operator()(T &ref) |
|
{ |
|
return thrust::raw_reference_cast(ref); |
|
} |
|
|
|
template<typename T> |
|
__host__ __device__ |
|
typename detail::raw_reference<const T>::type operator()(const T &ref) |
|
{ |
|
return thrust::raw_reference_cast(ref); |
|
} |
|
|
|
template< |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
__host__ __device__ |
|
typename detail::raw_reference< |
|
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
>::type |
|
operator()(thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> t, |
|
typename enable_if< |
|
is_unwrappable<thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> >::value |
|
>::type * = 0) |
|
{ |
|
return thrust::raw_reference_cast(t); |
|
} |
|
}; // end raw_reference_caster |
|
|
|
|
|
} // end detail |
|
|
|
|
|
template< |
|
typename T0, typename T1, typename T2, |
|
typename T3, typename T4, typename T5, |
|
typename T6, typename T7, typename T8, |
|
typename T9 |
|
> |
|
__host__ __device__ |
|
typename detail::enable_if_unwrappable< |
|
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>, |
|
typename detail::raw_reference< |
|
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> |
|
>::type |
|
>::type |
|
raw_reference_cast(thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> t) |
|
{ |
|
thrust::detail::raw_reference_caster f; |
|
|
|
// note that we pass raw_reference_tuple_helper, not raw_reference as the unary metafunction |
|
// the subtle difference is important |
|
return thrust::detail::tuple_host_device_transform<detail::raw_reference_detail::raw_reference_tuple_helper>(t, f); |
|
} // end raw_reference_cast |
|
|
|
|
|
} // end thrust |
|
|
|
|