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.
122 lines
2.9 KiB
122 lines
2.9 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/type_traits/has_nested_type.h>
|
||
|
|
#include <thrust/detail/raw_pointer_cast.h>
|
||
|
|
#include <thrust/detail/type_traits.h>
|
||
|
|
#include <thrust/iterator/detail/tuple_of_iterator_references.h>
|
||
|
|
|
||
|
|
namespace thrust
|
||
|
|
{
|
||
|
|
namespace detail
|
||
|
|
{
|
||
|
|
|
||
|
|
|
||
|
|
__THRUST_DEFINE_HAS_NESTED_TYPE(is_wrapped_reference, wrapped_reference_hint)
|
||
|
|
|
||
|
|
namespace raw_reference_detail
|
||
|
|
{
|
||
|
|
|
||
|
|
template<typename T, typename Enable = void>
|
||
|
|
struct raw_reference
|
||
|
|
: add_reference<T>
|
||
|
|
{};
|
||
|
|
|
||
|
|
|
||
|
|
// XXX consider making raw_reference<T&> an error
|
||
|
|
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
struct raw_reference<
|
||
|
|
T,
|
||
|
|
typename thrust::detail::enable_if<
|
||
|
|
is_wrapped_reference<
|
||
|
|
typename remove_cv<T>::type
|
||
|
|
>::value
|
||
|
|
>::type
|
||
|
|
>
|
||
|
|
{
|
||
|
|
typedef typename add_reference<
|
||
|
|
typename pointer_element<typename T::pointer>::type
|
||
|
|
>::type type;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // end raw_reference_ns
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
struct raw_reference :
|
||
|
|
raw_reference_detail::raw_reference<T>
|
||
|
|
{};
|
||
|
|
|
||
|
|
|
||
|
|
// wrapped reference-like things which aren't strictly wrapped references
|
||
|
|
// (e.g. tuples of wrapped references) are considered unwrappable
|
||
|
|
template<typename T>
|
||
|
|
struct is_unwrappable
|
||
|
|
: is_wrapped_reference<T>
|
||
|
|
{};
|
||
|
|
|
||
|
|
|
||
|
|
template<typename T, typename Result = void>
|
||
|
|
struct enable_if_unwrappable
|
||
|
|
: enable_if<
|
||
|
|
is_unwrappable<T>::value,
|
||
|
|
Result
|
||
|
|
>
|
||
|
|
{};
|
||
|
|
|
||
|
|
|
||
|
|
} // end detail
|
||
|
|
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
inline __host__ __device__ typename detail::raw_reference<T>::type raw_reference_cast(T &ref)
|
||
|
|
{
|
||
|
|
return *thrust::raw_pointer_cast(&ref);
|
||
|
|
} // end raw_reference_cast
|
||
|
|
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
inline __host__ __device__ typename detail::raw_reference<const T>::type raw_reference_cast(const T &ref)
|
||
|
|
{
|
||
|
|
return *thrust::raw_pointer_cast(&ref);
|
||
|
|
} // end raw_reference_cast
|
||
|
|
|
||
|
|
|
||
|
|
template<
|
||
|
|
typename T0, typename T1, typename T2,
|
||
|
|
typename T3, typename T4, typename T5,
|
||
|
|
typename T6, typename T7, typename T8,
|
||
|
|
typename T9
|
||
|
|
>
|
||
|
|
inline __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(detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> t);
|
||
|
|
|
||
|
|
|
||
|
|
} // end thrust
|
||
|
|
|
||
|
|
#include <thrust/detail/raw_reference_cast.inl>
|
||
|
|
|