blob: 494dbb29aa3d82ecfa632750f6c98665499094c2 [file] [log] [blame]
/*
* 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 "base/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; }; // NOLINT [readability/braces]
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; }; // NOLINT [readability/braces]
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