diff options
Diffstat (limited to 'boost/compute/algorithm/transform.hpp')
-rw-r--r-- | boost/compute/algorithm/transform.hpp | 76 |
1 files changed, 76 insertions, 0 deletions
diff --git a/boost/compute/algorithm/transform.hpp b/boost/compute/algorithm/transform.hpp new file mode 100644 index 0000000000..022a4988bd --- /dev/null +++ b/boost/compute/algorithm/transform.hpp @@ -0,0 +1,76 @@ +//---------------------------------------------------------------------------// +// Copyright (c) 2013 Kyle Lutz <kyle.r.lutz@gmail.com> +// +// 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_TRANSFORM_HPP +#define BOOST_COMPUTE_ALGORITHM_TRANSFORM_HPP + +#include <boost/compute/system.hpp> +#include <boost/compute/command_queue.hpp> +#include <boost/compute/algorithm/copy.hpp> +#include <boost/compute/iterator/transform_iterator.hpp> +#include <boost/compute/iterator/zip_iterator.hpp> +#include <boost/compute/functional/detail/unpack.hpp> + +namespace boost { +namespace compute { + +/// Transforms the elements in the range [\p first, \p last) using +/// \p transform and stores the results in the range beginning at +/// \p result. +/// +/// For example, to calculate the absolute value for each element in a vector: +/// +/// \snippet test/test_transform.cpp transform_abs +/// +/// \see copy() +template<class InputIterator, class OutputIterator, class UnaryOperator> +inline OutputIterator transform(InputIterator first, + InputIterator last, + OutputIterator result, + UnaryOperator op, + command_queue &queue = system::default_queue()) +{ + return copy( + ::boost::compute::make_transform_iterator(first, op), + ::boost::compute::make_transform_iterator(last, op), + result, + queue + ); +} + +/// \overload +template<class InputIterator1, + class InputIterator2, + class OutputIterator, + class BinaryOperator> +inline OutputIterator transform(InputIterator1 first1, + InputIterator1 last1, + InputIterator2 first2, + OutputIterator result, + BinaryOperator op, + command_queue &queue = system::default_queue()) +{ + typedef typename std::iterator_traits<InputIterator1>::difference_type difference_type; + + difference_type n = std::distance(first1, last1); + + return transform( + make_zip_iterator(boost::make_tuple(first1, first2)), + make_zip_iterator(boost::make_tuple(last1, first2 + n)), + result, + detail::unpack(op), + queue + ); +} + +} // end compute namespace +} // end boost namespace + +#endif // BOOST_COMPUTE_ALGORITHM_TRANSFORM_HPP |