| 1 | // Boost.Range library |
| 2 | // |
| 3 | // Copyright Neil Groves 2009. Use, modification and |
| 4 | // distribution is subject to the Boost Software License, Version |
| 5 | // 1.0. (See accompanying file LICENSE_1_0.txt or copy at |
| 6 | // http://www.boost.org/LICENSE_1_0.txt) |
| 7 | // |
| 8 | // |
| 9 | // For more information, see http://www.boost.org/libs/range/ |
| 10 | // |
| 11 | #include <boost/range/adaptor/transformed.hpp> |
| 12 | |
| 13 | #include <boost/test/test_tools.hpp> |
| 14 | #include <boost/test/unit_test.hpp> |
| 15 | |
| 16 | #include <boost/assign.hpp> |
| 17 | #include <boost/bind/bind.hpp> |
| 18 | #include <boost/range/algorithm_ext.hpp> |
| 19 | |
| 20 | #include <algorithm> |
| 21 | #include <list> |
| 22 | #include <set> |
| 23 | #include <vector> |
| 24 | |
| 25 | namespace boost |
| 26 | { |
| 27 | namespace |
| 28 | { |
| 29 | struct double_x |
| 30 | { |
| 31 | typedef int result_type; |
| 32 | int operator()(int x) const { return x * 2; } |
| 33 | }; |
| 34 | |
| 35 | struct halve_x |
| 36 | { |
| 37 | typedef int result_type; |
| 38 | int operator()(int x) const { return x / 2; } |
| 39 | }; |
| 40 | |
| 41 | struct lambda_init |
| 42 | { |
| 43 | }; |
| 44 | |
| 45 | struct lambda |
| 46 | { |
| 47 | typedef int result_type; |
| 48 | |
| 49 | lambda(const lambda_init& init) {} |
| 50 | lambda(const lambda& rhs) {} |
| 51 | |
| 52 | int operator()(int x) const { return x + 1; } |
| 53 | |
| 54 | private: |
| 55 | lambda() {} |
| 56 | lambda& operator=(const lambda& rhs) { return *this; } |
| 57 | }; |
| 58 | |
| 59 | template< class Container, class TransformFn > |
| 60 | void transformed_test_impl_core( Container& c, TransformFn fn ) |
| 61 | { |
| 62 | using namespace boost::adaptors; |
| 63 | |
| 64 | std::vector< int > test_result1; |
| 65 | boost::push_back(test_result1, c | transformed(fn)); |
| 66 | |
| 67 | std::vector< int > test_result2; |
| 68 | boost::push_back(test_result2, adaptors::transform(c, fn)); |
| 69 | |
| 70 | std::vector< int > reference; |
| 71 | std::transform(c.begin(), c.end(), std::back_inserter(x&: reference), fn); |
| 72 | |
| 73 | BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(), |
| 74 | test_result1.begin(), test_result1.end() ); |
| 75 | |
| 76 | BOOST_CHECK_EQUAL_COLLECTIONS( reference.begin(), reference.end(), |
| 77 | test_result2.begin(), test_result2.end() ); |
| 78 | } |
| 79 | |
| 80 | template< class Rng > |
| 81 | void check_copy_assign(Rng r) |
| 82 | { |
| 83 | Rng r2 = r; |
| 84 | r2 = r; |
| 85 | } |
| 86 | |
| 87 | template< class Container, class TransformFn > |
| 88 | void transformed_range_copy_assign(Container& c, TransformFn fn) |
| 89 | { |
| 90 | using namespace boost::adaptors; |
| 91 | check_copy_assign(c | transformed(fn)); |
| 92 | check_copy_assign(adaptors::transform(c, fn)); |
| 93 | } |
| 94 | |
| 95 | template< class Container, class TransformFn, class TransformFnInit > |
| 96 | void transformed_test_fn_impl() |
| 97 | { |
| 98 | using namespace boost::assign; |
| 99 | |
| 100 | Container c; |
| 101 | TransformFnInit init; |
| 102 | TransformFn fn( init ); |
| 103 | |
| 104 | // Test empty |
| 105 | transformed_test_impl_core(c, fn); |
| 106 | |
| 107 | // Test one element |
| 108 | c += 1; |
| 109 | transformed_test_impl_core(c, fn); |
| 110 | |
| 111 | // Test many elements |
| 112 | c += 1,1,1,2,2,2,2,2,3,4,5,6,7,8,9; |
| 113 | transformed_test_impl_core(c, fn); |
| 114 | |
| 115 | // test the range and iterator are copy assignable |
| 116 | transformed_range_copy_assign(c, fn); |
| 117 | } |
| 118 | |
| 119 | template< class Container > |
| 120 | void transformed_test_impl() |
| 121 | { |
| 122 | transformed_test_fn_impl< Container, double_x, double_x >(); |
| 123 | transformed_test_fn_impl< Container, halve_x, halve_x >(); |
| 124 | transformed_test_fn_impl< Container, lambda, lambda_init >(); |
| 125 | } |
| 126 | |
| 127 | void transformed_test() |
| 128 | { |
| 129 | transformed_test_impl< std::vector< int > >(); |
| 130 | transformed_test_impl< std::list< int > >(); |
| 131 | transformed_test_impl< std::set< int > >(); |
| 132 | transformed_test_impl< std::multiset< int > >(); |
| 133 | } |
| 134 | |
| 135 | struct foo_bind |
| 136 | { |
| 137 | int foo() const { return 7; } |
| 138 | }; |
| 139 | |
| 140 | void transformed_bind() |
| 141 | { |
| 142 | using namespace boost::adaptors; |
| 143 | using namespace boost::placeholders; |
| 144 | |
| 145 | std::vector<foo_bind> input(5); |
| 146 | std::vector<int> output; |
| 147 | boost::range::push_back( |
| 148 | on&: output, |
| 149 | from: input | transformed(boost::bind(f: &foo_bind::foo, a1: _1))); |
| 150 | |
| 151 | BOOST_CHECK_EQUAL(output.size(), input.size()); |
| 152 | |
| 153 | std::vector<int> reference_output(5, 7); |
| 154 | BOOST_CHECK_EQUAL_COLLECTIONS( |
| 155 | output.begin(), output.end(), |
| 156 | reference_output.begin(), reference_output.end()); |
| 157 | } |
| 158 | } |
| 159 | } |
| 160 | |
| 161 | boost::unit_test::test_suite* |
| 162 | init_unit_test_suite(int argc, char* argv[]) |
| 163 | { |
| 164 | boost::unit_test::test_suite* test |
| 165 | = BOOST_TEST_SUITE( "RangeTestSuite.adaptor.transformed" ); |
| 166 | |
| 167 | test->add(BOOST_TEST_CASE(&boost::transformed_test)); |
| 168 | test->add(BOOST_TEST_CASE(&boost::transformed_bind)); |
| 169 | |
| 170 | return test; |
| 171 | } |
| 172 | |