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
|
/*
* 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/system/detail/generic/reverse.h>
#include <thrust/advance.h>
#include <thrust/distance.h>
#include <thrust/detail/copy.h>
#include <thrust/swap.h>
#include <thrust/iterator/iterator_traits.h>
#include <thrust/iterator/reverse_iterator.h>
THRUST_NAMESPACE_BEGIN
namespace system
{
namespace detail
{
namespace generic
{
template<typename ExecutionPolicy, typename BidirectionalIterator>
__host__ __device__
void reverse(thrust::execution_policy<ExecutionPolicy> &exec,
BidirectionalIterator first,
BidirectionalIterator last)
{
typedef typename thrust::iterator_difference<BidirectionalIterator>::type difference_type;
// find the midpoint of [first,last)
difference_type N = thrust::distance(first, last);
BidirectionalIterator mid(first);
thrust::advance(mid, N / 2);
// swap elements of [first,mid) with [last - 1, mid)
thrust::swap_ranges(exec, first, mid, thrust::make_reverse_iterator(last));
} // end reverse()
template<typename ExecutionPolicy,
typename BidirectionalIterator,
typename OutputIterator>
__host__ __device__
OutputIterator reverse_copy(thrust::execution_policy<ExecutionPolicy> &exec,
BidirectionalIterator first,
BidirectionalIterator last,
OutputIterator result)
{
return thrust::copy(exec,
thrust::make_reverse_iterator(last),
thrust::make_reverse_iterator(first),
result);
} // end reverse_copy()
} // end namespace generic
} // end namespace detail
} // end namespace system
THRUST_NAMESPACE_END
|