| /* |
| * Copyright (C) 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <algorithm> |
| #include <vector> |
| |
| #include "gtest/gtest.h" |
| #include "transform_array_ref.h" |
| |
| namespace art { |
| |
| namespace { // anonymous namespace |
| |
| struct ValueHolder { |
| // Deliberately not explicit. |
| ValueHolder(int v) : value(v) { } // NOLINT |
| int value; |
| }; |
| |
| ATTRIBUTE_UNUSED bool operator==(const ValueHolder& lhs, const ValueHolder& rhs) { |
| return lhs.value == rhs.value; |
| } |
| |
| } // anonymous namespace |
| |
| TEST(TransformArrayRef, ConstRefAdd1) { |
| auto add1 = [](const ValueHolder& h) { return h.value + 1; }; |
| std::vector<ValueHolder> input({ 7, 6, 4, 0 }); |
| std::vector<int> output; |
| |
| auto taref = MakeTransformArrayRef(input, add1); |
| using TarefIter = decltype(taref)::iterator; |
| using ConstTarefIter = decltype(taref)::const_iterator; |
| static_assert(std::is_same<int, decltype(taref)::value_type>::value, "value_type"); |
| static_assert(std::is_same<TarefIter, decltype(taref)::pointer>::value, "pointer"); |
| static_assert(std::is_same<int, decltype(taref)::reference>::value, "reference"); |
| static_assert(std::is_same<ConstTarefIter, decltype(taref)::const_pointer>::value, |
| "const_pointer"); |
| static_assert(std::is_same<int, decltype(taref)::const_reference>::value, "const_reference"); |
| |
| std::copy(taref.begin(), taref.end(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 8, 7, 5, 1 }), output); |
| output.clear(); |
| |
| std::copy(taref.cbegin(), taref.cend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 8, 7, 5, 1 }), output); |
| output.clear(); |
| |
| std::copy(taref.rbegin(), taref.rend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 5, 7, 8 }), output); |
| output.clear(); |
| |
| std::copy(taref.crbegin(), taref.crend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 5, 7, 8 }), output); |
| output.clear(); |
| |
| ASSERT_EQ(input.size(), taref.size()); |
| ASSERT_EQ(input.empty(), taref.empty()); |
| ASSERT_EQ(input.front().value + 1, taref.front()); |
| ASSERT_EQ(input.back().value + 1, taref.back()); |
| |
| for (size_t i = 0; i != input.size(); ++i) { |
| ASSERT_EQ(input[i].value + 1, taref[i]); |
| } |
| } |
| |
| TEST(TransformArrayRef, NonConstRefSub1) { |
| auto sub1 = [](ValueHolder& h) { return h.value - 1; }; |
| std::vector<ValueHolder> input({ 4, 4, 5, 7, 10 }); |
| std::vector<int> output; |
| |
| auto taref = MakeTransformArrayRef(input, sub1); |
| using TarefIter = decltype(taref)::iterator; |
| static_assert(std::is_same<void, decltype(taref)::const_iterator>::value, "const_iterator"); |
| static_assert(std::is_same<int, decltype(taref)::value_type>::value, "value_type"); |
| static_assert(std::is_same<TarefIter, decltype(taref)::pointer>::value, "pointer"); |
| static_assert(std::is_same<int, decltype(taref)::reference>::value, "reference"); |
| static_assert(std::is_same<void, decltype(taref)::const_pointer>::value, "const_pointer"); |
| static_assert(std::is_same<void, decltype(taref)::const_reference>::value, "const_reference"); |
| |
| std::copy(taref.begin(), taref.end(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 3, 3, 4, 6, 9 }), output); |
| output.clear(); |
| |
| std::copy(taref.rbegin(), taref.rend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 9, 6, 4, 3, 3 }), output); |
| output.clear(); |
| |
| ASSERT_EQ(input.size(), taref.size()); |
| ASSERT_EQ(input.empty(), taref.empty()); |
| ASSERT_EQ(input.front().value - 1, taref.front()); |
| ASSERT_EQ(input.back().value - 1, taref.back()); |
| |
| for (size_t i = 0; i != input.size(); ++i) { |
| ASSERT_EQ(input[i].value - 1, taref[i]); |
| } |
| } |
| |
| TEST(TransformArrayRef, ConstAndNonConstRef) { |
| struct Ref { |
| int& operator()(ValueHolder& h) const { return h.value; } |
| const int& operator()(const ValueHolder& h) const { return h.value; } |
| }; |
| Ref ref; |
| std::vector<ValueHolder> input({ 1, 0, 1, 0, 3, 1 }); |
| std::vector<int> output; |
| |
| auto taref = MakeTransformArrayRef(input, ref); |
| static_assert(std::is_same<int, decltype(taref)::value_type>::value, "value_type"); |
| static_assert(std::is_same<int*, decltype(taref)::pointer>::value, "pointer"); |
| static_assert(std::is_same<int&, decltype(taref)::reference>::value, "reference"); |
| static_assert(std::is_same<const int*, decltype(taref)::const_pointer>::value, "const_pointer"); |
| static_assert(std::is_same<const int&, decltype(taref)::const_reference>::value, |
| "const_reference"); |
| |
| std::copy(taref.begin(), taref.end(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 0, 1, 0, 3, 1 }), output); |
| output.clear(); |
| |
| std::copy(taref.cbegin(), taref.cend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 0, 1, 0, 3, 1 }), output); |
| output.clear(); |
| |
| std::copy(taref.rbegin(), taref.rend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 3, 0, 1, 0, 1 }), output); |
| output.clear(); |
| |
| std::copy(taref.crbegin(), taref.crend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 3, 0, 1, 0, 1 }), output); |
| output.clear(); |
| |
| ASSERT_EQ(input.size(), taref.size()); |
| ASSERT_EQ(input.empty(), taref.empty()); |
| ASSERT_EQ(input.front().value, taref.front()); |
| ASSERT_EQ(input.back().value, taref.back()); |
| |
| for (size_t i = 0; i != input.size(); ++i) { |
| ASSERT_EQ(input[i].value, taref[i]); |
| } |
| |
| // Test writing through the transform iterator. |
| std::vector<int> transform_input({ 24, 37, 11, 71 }); |
| std::vector<ValueHolder> transformed(transform_input.size(), 0); |
| taref = MakeTransformArrayRef(transformed, ref); |
| for (size_t i = 0; i != transform_input.size(); ++i) { |
| taref[i] = transform_input[i]; |
| } |
| ASSERT_EQ(std::vector<ValueHolder>({ 24, 37, 11, 71 }), transformed); |
| |
| const std::vector<ValueHolder>& cinput = input; |
| |
| auto ctaref = MakeTransformArrayRef(cinput, ref); |
| static_assert(std::is_same<int, decltype(ctaref)::value_type>::value, "value_type"); |
| static_assert(std::is_same<const int*, decltype(ctaref)::pointer>::value, "pointer"); |
| static_assert(std::is_same<const int&, decltype(ctaref)::reference>::value, "reference"); |
| static_assert(std::is_same<const int*, decltype(ctaref)::const_pointer>::value, "const_pointer"); |
| static_assert(std::is_same<const int&, decltype(ctaref)::const_reference>::value, |
| "const_reference"); |
| |
| std::copy(ctaref.begin(), ctaref.end(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 0, 1, 0, 3, 1 }), output); |
| output.clear(); |
| |
| std::copy(ctaref.cbegin(), ctaref.cend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 0, 1, 0, 3, 1 }), output); |
| output.clear(); |
| |
| std::copy(ctaref.rbegin(), ctaref.rend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 3, 0, 1, 0, 1 }), output); |
| output.clear(); |
| |
| std::copy(ctaref.crbegin(), ctaref.crend(), std::back_inserter(output)); |
| ASSERT_EQ(std::vector<int>({ 1, 3, 0, 1, 0, 1 }), output); |
| output.clear(); |
| |
| ASSERT_EQ(cinput.size(), ctaref.size()); |
| ASSERT_EQ(cinput.empty(), ctaref.empty()); |
| ASSERT_EQ(cinput.front().value, ctaref.front()); |
| ASSERT_EQ(cinput.back().value, ctaref.back()); |
| |
| for (size_t i = 0; i != cinput.size(); ++i) { |
| ASSERT_EQ(cinput[i].value, ctaref[i]); |
| } |
| |
| // Test conversion adding const. |
| decltype(ctaref) ctaref2 = taref; |
| ASSERT_EQ(taref.size(), ctaref2.size()); |
| for (size_t i = 0; i != taref.size(); ++i) { |
| ASSERT_EQ(taref[i], ctaref2[i]); |
| } |
| } |
| |
| } // namespace art |