/usr/include/boost/compute/algorithm/detail
Edit: /usr/include/boost/compute/algorithm/detail/insertion_sort.hpp (6030B)
//---------------------------------------------------------------------------//
// Copyright (c) 2013 Kyle Lutz
//
// Distributed under the Boost Software License, Version 1.0
// See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt
//
// See http://boostorg.github.com/compute for more information.
//---------------------------------------------------------------------------//
#ifndef BOOST_COMPUTE_ALGORITHM_DETAIL_INSERTION_SORT_HPP
#define BOOST_COMPUTE_ALGORITHM_DETAIL_INSERTION_SORT_HPP
#include
#include
#include
#include
#include
#include
namespace boost {
namespace compute {
namespace detail {
template
inline void serial_insertion_sort(Iterator first,
Iterator last,
Compare compare,
command_queue &queue)
{
typedef typename std::iterator_traits::value_type T;
size_t count = iterator_range_size(first, last);
if(count < 2){
return;
}
meta_kernel k("serial_insertion_sort");
size_t local_data_arg = k.add_arg(memory_object::local_memory, "data");
size_t count_arg = k.add_arg("n");
k <<
// copy data to local memory
"for(uint i = 0; i < n; i++){\n" <<
" data[i] = " << first[k.var("i")] << ";\n"
"}\n"
// sort data in local memory
"for(uint i = 1; i < n; i++){\n" <<
" " << k.decl("value") << " = data[i];\n" <<
" uint pos = i;\n" <<
" while(pos > 0 && " <<
compare(k.var("value"),
k.var("data[pos-1]")) << "){\n" <<
" data[pos] = data[pos-1];\n" <<
" pos--;\n" <<
" }\n" <<
" data[pos] = value;\n" <<
"}\n" <<
// copy sorted data to output
"for(uint i = 0; i < n; i++){\n" <<
" " << first[k.var("i")] << " = data[i];\n"
"}\n";
const context &context = queue.get_context();
::boost::compute::kernel kernel = k.compile(context);
kernel.set_arg(local_data_arg, local_buffer(count));
kernel.set_arg(count_arg, static_cast(count));
queue.enqueue_task(kernel);
}
template
inline void serial_insertion_sort(Iterator first,
Iterator last,
command_queue &queue)
{
typedef typename std::iterator_traits::value_type T;
::boost::compute::less less;
return serial_insertion_sort(first, last, less, queue);
}
template
inline void serial_insertion_sort_by_key(KeyIterator keys_first,
KeyIterator keys_last,
ValueIterator values_first,
Compare compare,
command_queue &queue)
{
typedef typename std::iterator_traits::value_type key_type;
typedef typename std::iterator_traits::value_type value_type;
size_t count = iterator_range_size(keys_first, keys_last);
if(count < 2){
return;
}
meta_kernel k("serial_insertion_sort_by_key");
size_t local_keys_arg = k.add_arg(memory_object::local_memory, "keys");
size_t local_data_arg = k.add_arg(memory_object::local_memory, "data");
size_t count_arg = k.add_arg("n");
k <<
// copy data to local memory
"for(uint i = 0; i < n; i++){\n" <<
" keys[i] = " << keys_first[k.var("i")] << ";\n"
" data[i] = " << values_first[k.var("i")] << ";\n"
"}\n"
// sort data in local memory
"for(uint i = 1; i < n; i++){\n" <<
" " << k.decl("key") << " = keys[i];\n" <<
" " << k.decl("value") << " = data[i];\n" <<
" uint pos = i;\n" <<
" while(pos > 0 && " <<
compare(k.var("key"),
k.var("keys[pos-1]")) << "){\n" <<
" keys[pos] = keys[pos-1];\n" <<
" data[pos] = data[pos-1];\n" <<
" pos--;\n" <<
" }\n" <<
" keys[pos] = key;\n" <<
" data[pos] = value;\n" <<
"}\n" <<
// copy sorted data to output
"for(uint i = 0; i < n; i++){\n" <<
" " << keys_first[k.var("i")] << " = keys[i];\n"
" " << values_first[k.var("i")] << " = data[i];\n"
"}\n";
const context &context = queue.get_context();
::boost::compute::kernel kernel = k.compile(context);
kernel.set_arg(local_keys_arg, static_cast(count * sizeof(key_type)), 0);
kernel.set_arg(local_data_arg, static_cast(count * sizeof(value_type)), 0);
kernel.set_arg(count_arg, static_cast(count));
queue.enqueue_task(kernel);
}
template
inline void serial_insertion_sort_by_key(KeyIterator keys_first,
KeyIterator keys_last,
ValueIterator values_first,
command_queue &queue)
{
typedef typename std::iterator_traits::value_type key_type;
serial_insertion_sort_by_key(
keys_first,
keys_last,
values_first,
boost::compute::less(),
queue
);
}
} // end detail namespace
} // end compute namespace
} // end boost namespace
#endif // BOOST_COMPUTE_ALGORITHM_DETAIL_INSERTION_SORT_HPP