Initial Commit

.svn/pristine/14/143dca54dac502a46079a26cfce20a211ab93d9e.svn-base

@@ -0,0 +1,89 @@
+//---------------------------------------------------------------------------//
+// 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_REDUCE_HPP
+#define BOOST_COMPUTE_ALGORITHM_TRANSFORM_REDUCE_HPP
+
+#include <boost/compute/system.hpp>
+#include <boost/compute/algorithm/reduce.hpp>
+#include <boost/compute/iterator/transform_iterator.hpp>
+#include <boost/compute/iterator/zip_iterator.hpp>
+#include <boost/compute/functional/detail/unpack.hpp>
+#include <boost/compute/detail/iterator_range_size.hpp>
+
+namespace boost {
+namespace compute {
+
+/// Transforms each value in the range [\p first, \p last) with the unary
+/// \p transform_function and then reduces each transformed value with
+/// \p reduce_function.
+///
+/// For example, to calculate the sum of the absolute values of a vector
+/// of integers:
+///
+/// \snippet test/test_transform_reduce.cpp sum_abs_int
+///
+/// \see reduce(), inner_product()
+template<class InputIterator,
+         class OutputIterator,
+         class UnaryTransformFunction,
+         class BinaryReduceFunction>
+inline void transform_reduce(InputIterator first,
+                             InputIterator last,
+                             OutputIterator result,
+                             UnaryTransformFunction transform_function,
+                             BinaryReduceFunction reduce_function,
+                             command_queue &queue = system::default_queue())
+{
+    ::boost::compute::reduce(
+        ::boost::compute::make_transform_iterator(first, transform_function),
+        ::boost::compute::make_transform_iterator(last, transform_function),
+        result,
+        reduce_function,
+        queue
+    );
+}
+
+/// \overload
+template<class InputIterator1,
+         class InputIterator2,
+         class OutputIterator,
+         class BinaryTransformFunction,
+         class BinaryReduceFunction>
+inline void transform_reduce(InputIterator1 first1,
+                             InputIterator1 last1,
+                             InputIterator2 first2,
+                             OutputIterator result,
+                             BinaryTransformFunction transform_function,
+                             BinaryReduceFunction reduce_function,
+                             command_queue &queue = system::default_queue())
+{
+    typedef typename std::iterator_traits<InputIterator1>::difference_type difference_type;
+
+    difference_type n = std::distance(first1, last1);
+
+    ::boost::compute::transform_reduce(
+        ::boost::compute::make_zip_iterator(
+            boost::make_tuple(first1, first2)
+        ),
+        ::boost::compute::make_zip_iterator(
+            boost::make_tuple(last1, first2 + n)
+        ),
+        result,
+        detail::unpack(transform_function),
+        reduce_function,
+        queue
+    );
+}
+
+} // end compute namespace
+} // end boost namespace
+
+#endif // BOOST_COMPUTE_ALGORITHM_TRANSFORM_REDUCE_HPP