diff options
Diffstat (limited to 'src/boost/libs/compute/test/test_functional_bind.cpp')
-rw-r--r-- | src/boost/libs/compute/test/test_functional_bind.cpp | 237 |
1 files changed, 237 insertions, 0 deletions
diff --git a/src/boost/libs/compute/test/test_functional_bind.cpp b/src/boost/libs/compute/test/test_functional_bind.cpp new file mode 100644 index 00000000..327232d8 --- /dev/null +++ b/src/boost/libs/compute/test/test_functional_bind.cpp @@ -0,0 +1,237 @@ +//---------------------------------------------------------------------------// +// Copyright (c) 2013-2014 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. +//---------------------------------------------------------------------------// + +#define BOOST_TEST_MODULE TestFunctionalBind +#include <boost/test/unit_test.hpp> + +#include <iostream> + +#include <boost/compute/function.hpp> +#include <boost/compute/algorithm/copy_n.hpp> +#include <boost/compute/algorithm/count_if.hpp> +#include <boost/compute/algorithm/find_if.hpp> +#include <boost/compute/algorithm/transform.hpp> +#include <boost/compute/container/vector.hpp> +#include <boost/compute/functional/bind.hpp> +#include <boost/compute/functional/common.hpp> +#include <boost/compute/functional/operator.hpp> +#include <boost/compute/types/struct.hpp> + +// simple test struct +struct data_struct +{ + int int_value; + float float_value; +}; + +BOOST_COMPUTE_ADAPT_STRUCT(data_struct, data_struct, (int_value, float_value)) + +#include "quirks.hpp" +#include "check_macros.hpp" +#include "context_setup.hpp" + +namespace compute = boost::compute; + +using compute::placeholders::_1; +using compute::placeholders::_2; + +BOOST_AUTO_TEST_CASE(transform_plus_two) +{ + int data[] = { 1, 2, 3, 4 }; + compute::vector<int> vector(4, context); + compute::copy_n(data, 4, vector.begin(), queue); + + compute::transform( + vector.begin(), vector.end(), vector.begin(), + compute::bind(compute::plus<int>(), _1, 2), + queue + ); + + CHECK_RANGE_EQUAL(int, 4, vector, (3, 4, 5, 6)); +} + +BOOST_AUTO_TEST_CASE(transform_pow_two) +{ + float data[] = { 2, 3, 4, 5 }; + compute::vector<float> vector(4, context); + compute::copy_n(data, 4, vector.begin(), queue); + + compute::transform( + vector.begin(), vector.end(), vector.begin(), + compute::bind(compute::pow<float>(), 2.0f, _1), + queue + ); + + compute::copy(vector.begin(), vector.end(), data, queue); + BOOST_CHECK_CLOSE(data[0], 4.0f, 1e-4); + BOOST_CHECK_CLOSE(data[1], 8.0f, 1e-4); + BOOST_CHECK_CLOSE(data[2], 16.0f, 1e-4); + BOOST_CHECK_CLOSE(data[3], 32.0f, 1e-4); +} + +BOOST_AUTO_TEST_CASE(find_if_equal) +{ + int data[] = { 1, 2, 3, 4 }; + compute::vector<int> vector(4, context); + compute::copy_n(data, 4, vector.begin(), queue); + + BOOST_CHECK( + compute::find_if( + vector.begin(), vector.end(), + compute::bind(compute::equal_to<int>(), _1, 3), + queue + ) == vector.begin() + 2 + ); +} + +BOOST_AUTO_TEST_CASE(compare_less_than) +{ + int data[] = { 1, 2, 3, 4 }; + compute::vector<int> vector(data, data + 4, queue); + + int count = boost::compute::count_if( + vector.begin(), vector.end(), + compute::bind(compute::less<int>(), _1, int(3)), + queue + ); + BOOST_CHECK_EQUAL(count, 2); + + count = boost::compute::count_if( + vector.begin(), vector.end(), + compute::bind(compute::less<int>(), int(3), _1), + queue + ); + BOOST_CHECK_EQUAL(count, 1); +} + +BOOST_AUTO_TEST_CASE(subtract_ranges) +{ + int data1[] = { 1, 2, 3, 4 }; + int data2[] = { 4, 3, 2, 1 }; + + compute::vector<int> vector1(data1, data1 + 4, queue); + compute::vector<int> vector2(data2, data2 + 4, queue); + + compute::vector<int> result(4, context); + + compute::transform( + vector1.begin(), + vector1.end(), + vector2.begin(), + result.begin(), + compute::bind(compute::minus<int>(), _1, _2), + queue + ); + CHECK_RANGE_EQUAL(int, 4, result, (-3, -1, 1, 3)); + + compute::transform( + vector1.begin(), + vector1.end(), + vector2.begin(), + result.begin(), + compute::bind(compute::minus<int>(), _2, _1), + queue + ); + CHECK_RANGE_EQUAL(int, 4, result, (3, 1, -1, -3)); + + compute::transform( + vector1.begin(), + vector1.end(), + vector2.begin(), + result.begin(), + compute::bind(compute::minus<int>(), 5, _1), + queue + ); + CHECK_RANGE_EQUAL(int, 4, result, (4, 3, 2, 1)); + + compute::transform( + vector1.begin(), + vector1.end(), + vector2.begin(), + result.begin(), + compute::bind(compute::minus<int>(), 5, _2), + queue + ); + CHECK_RANGE_EQUAL(int, 4, result, (1, 2, 3, 4)); +} + +BOOST_AUTO_TEST_CASE(clamp_values) +{ + int data[] = { 1, 2, 3, 4 }; + compute::vector<int> vector(data, data + 4, queue); + + compute::transform( + vector.begin(), vector.end(), vector.begin(), + compute::bind(compute::clamp<int>(), _1, 2, 3), + queue + ); + CHECK_RANGE_EQUAL(int, 4, vector, (2, 2, 3, 3)); +} + +BOOST_AUTO_TEST_CASE(bind_custom_function) +{ + int data[] = { 1, 2, 3, 4 }; + compute::vector<int> vector(data, data + 4, queue); + + BOOST_COMPUTE_FUNCTION(int, x_if_odd_else_y, (int x, int y), + { + if(x & 1) + return x; + else + return y; + }); + + compute::transform( + vector.begin(), vector.end(), vector.begin(), + compute::bind(x_if_odd_else_y, _1, 9), + queue + ); + CHECK_RANGE_EQUAL(int, 4, vector, (1, 9, 3, 9)); + + compute::copy( + data, data + 4, vector.begin(), queue + ); + + compute::transform( + vector.begin(), vector.end(), vector.begin(), + compute::bind(x_if_odd_else_y, 2, _1), + queue + ); + CHECK_RANGE_EQUAL(int, 4, vector, (1, 2, 3, 4)); +} + +BOOST_AUTO_TEST_CASE(bind_struct) +{ + if(bug_in_struct_assignment(device)){ + std::cerr << "skipping bind_struct test" << std::endl; + return; + } + + BOOST_COMPUTE_FUNCTION(int, add_struct_value, (int x, data_struct s), + { + return s.int_value + x; + }); + + data_struct data; + data.int_value = 3; + data.float_value = 4.56f; + + int input[] = { 1, 2, 3, 4 }; + compute::vector<int> vec(input, input + 4, queue); + + compute::transform( + vec.begin(), vec.end(), vec.begin(), + compute::bind(add_struct_value, _1, data), + queue + ); + CHECK_RANGE_EQUAL(int, 4, vec, (4, 5, 6, 7)); +} + +BOOST_AUTO_TEST_SUITE_END() |