File: test_transform_iterator.cpp

package info (click to toggle)
boost1.90 1.90.0-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 593,120 kB
  • sloc: cpp: 4,190,908; xml: 196,648; python: 34,618; ansic: 23,145; asm: 5,468; sh: 3,774; makefile: 1,161; perl: 1,020; sql: 728; ruby: 676; yacc: 478; java: 77; lisp: 24; csh: 6
file content (108 lines) | stat: -rw-r--r-- 3,043 bytes parent folder | download | duplicates (18)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
//---------------------------------------------------------------------------//
// 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.
//---------------------------------------------------------------------------//

#define BOOST_TEST_MODULE TestTransformIterator
#include <boost/test/unit_test.hpp>

#include <iterator>

#include <boost/type_traits.hpp>
#include <boost/static_assert.hpp>

#include <boost/compute/types.hpp>
#include <boost/compute/functional.hpp>
#include <boost/compute/algorithm/copy.hpp>
#include <boost/compute/container/vector.hpp>
#include <boost/compute/iterator/buffer_iterator.hpp>
#include <boost/compute/iterator/transform_iterator.hpp>

#include "check_macros.hpp"
#include "context_setup.hpp"

BOOST_AUTO_TEST_CASE(value_type)
{
    using boost::compute::float4_;

    BOOST_STATIC_ASSERT((
        boost::is_same<
            boost::compute::transform_iterator<
                boost::compute::buffer_iterator<float>,
                boost::compute::sqrt<float>
            >::value_type,
            float
        >::value
    ));
    BOOST_STATIC_ASSERT((
        boost::is_same<
            boost::compute::transform_iterator<
                boost::compute::buffer_iterator<float4_>,
                boost::compute::length<float4_>
            >::value_type,
            float
        >::value
    ));
}

BOOST_AUTO_TEST_CASE(base_type)
{
    BOOST_STATIC_ASSERT((
        boost::is_same<
            boost::compute::transform_iterator<
                boost::compute::buffer_iterator<int>, boost::compute::abs<int>
            >::base_type,
            boost::compute::buffer_iterator<int>
        >::value
    ));
}

BOOST_AUTO_TEST_CASE(copy)
{
    int data[] = { 1, -2, 3, -4, 5 };
    boost::compute::vector<int> a(data, data + 5, queue);

    boost::compute::vector<int> b(5, context);
    boost::compute::copy(
        boost::compute::make_transform_iterator(
            a.begin(),
            boost::compute::abs<int>()
        ),
        boost::compute::make_transform_iterator(
            a.end(),
            boost::compute::abs<int>()
        ),
        b.begin(),
        queue
    );
    CHECK_RANGE_EQUAL(int, 5, b, (1, 2, 3, 4, 5));
}

BOOST_AUTO_TEST_CASE(copy_abs_doctest)
{
    int data[] = { -1, -2, -3, -4 };
    boost::compute::vector<int> input(data, data + 4, queue);
    boost::compute::vector<int> output(4, context);

//! [copy_abs]
// use abs() from boost.compute
using boost::compute::abs;

// copy the absolute value for each element in input to output
boost::compute::copy(
    boost::compute::make_transform_iterator(input.begin(), abs<int>()),
    boost::compute::make_transform_iterator(input.end(), abs<int>()),
    output.begin(),
    queue
);
//! [copy_abs]

    CHECK_RANGE_EQUAL(int, 4, output, (1, 2, 3, 4));
}

BOOST_AUTO_TEST_SUITE_END()