mirror of
https://github.com/boostorg/compute.git
synced 2026-01-27 06:42:19 +00:00
3e840fa306
This adds a new algorithm named transform_if() which applies a given unary function to an input value only if it passes a separate predicate function.
56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
//---------------------------------------------------------------------------//
|
|
// 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://kylelutz.github.com/compute for more information.
|
|
//---------------------------------------------------------------------------//
|
|
|
|
#define BOOST_TEST_MODULE TestTransformIf
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
#include <boost/compute/lambda.hpp>
|
|
#include <boost/compute/functional.hpp>
|
|
#include <boost/compute/algorithm/transform_if.hpp>
|
|
#include <boost/compute/container/vector.hpp>
|
|
|
|
#include "context_setup.hpp"
|
|
|
|
namespace compute = boost::compute;
|
|
|
|
BOOST_AUTO_TEST_CASE(abs_if_odd)
|
|
{
|
|
using compute::lambda::_1;
|
|
|
|
// input data
|
|
int data[] = { -2, -3, -4, -5, -6, -7, -8, -9 };
|
|
compute::vector<int> vector(data, data + 8, context);
|
|
|
|
// calculate absolute value only for odd values
|
|
compute::transform_if(
|
|
vector.begin(),
|
|
vector.end(),
|
|
vector.begin(),
|
|
compute::abs<int>(),
|
|
_1 % 2 != 0,
|
|
queue
|
|
);
|
|
|
|
// copy data back to host
|
|
compute::copy(vector.begin(), vector.end(), data, queue);
|
|
queue.finish();
|
|
|
|
BOOST_CHECK_EQUAL(data[0], -2);
|
|
BOOST_CHECK_EQUAL(data[1], +3);
|
|
BOOST_CHECK_EQUAL(data[2], -4);
|
|
BOOST_CHECK_EQUAL(data[3], +5);
|
|
BOOST_CHECK_EQUAL(data[4], -6);
|
|
BOOST_CHECK_EQUAL(data[5], +7);
|
|
BOOST_CHECK_EQUAL(data[6], -8);
|
|
BOOST_CHECK_EQUAL(data[7], +9);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|