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.
278 lines
7.8 KiB
278 lines
7.8 KiB
11 years ago
|
/*
|
||
|
|
* 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
|
||
|
|
|