diff --git a/include/llvm/ADT/STLExtras.h b/include/llvm/ADT/STLExtras.h index 8e4d186d15a..eb515fd4583 100644 --- a/include/llvm/ADT/STLExtras.h +++ b/include/llvm/ADT/STLExtras.h @@ -208,9 +208,8 @@ inline mapped_iterator map_iterator(const ItTy &I, FuncTy F) { return mapped_iterator(I, F); } -/// \brief Metafunction to determine if type T has a member called rbegin(). -template -class has_rbegin { +/// Helper to determine if type T has a member called rbegin(). +template class has_rbegin_impl { typedef char yes[1]; typedef char no[2]; @@ -224,6 +223,11 @@ public: static const bool value = sizeof(test(nullptr)) == sizeof(yes); }; +/// Metafunction to determine if T& or T has a member called rbegin(). +template +struct has_rbegin : has_rbegin_impl::type> { +}; + // Returns an iterator_range over the given container which iterates in reverse. // Note that the container must have rbegin()/rend() methods for this to work. template diff --git a/unittests/ADT/RangeAdapterTest.cpp b/unittests/ADT/RangeAdapterTest.cpp index 2b12fe47173..4c7bef53235 100644 --- a/unittests/ADT/RangeAdapterTest.cpp +++ b/unittests/ADT/RangeAdapterTest.cpp @@ -38,18 +38,18 @@ public: // begin() and end() don't have implementations as this ensures that we will // get a linker error if reverse() chooses begin()/end() over rbegin(), rend(). class BidirectionalVector { - std::vector Vec; + mutable std::vector Vec; public: BidirectionalVector(std::initializer_list list) : Vec(list) {} typedef std::vector::iterator iterator; - iterator begin(); - iterator end(); + iterator begin() const; + iterator end() const; typedef std::vector::reverse_iterator reverse_iterator; - reverse_iterator rbegin() { return Vec.rbegin(); } - reverse_iterator rend() { return Vec.rend(); } + reverse_iterator rbegin() const { return Vec.rbegin(); } + reverse_iterator rend() const { return Vec.rend(); } }; /// This is the same as BidirectionalVector but with the addition of const @@ -75,6 +75,50 @@ public: const_reverse_iterator rend() const { return Vec.rend(); } }; +/// Check that types with custom iterators work. +class CustomIteratorVector { + mutable std::vector V; + +public: + CustomIteratorVector(std::initializer_list list) : V(list) {} + + typedef std::vector::iterator iterator; + class reverse_iterator { + std::vector::iterator I; + + public: + reverse_iterator() = default; + reverse_iterator(const reverse_iterator &) = default; + reverse_iterator &operator=(const reverse_iterator &) = default; + + explicit reverse_iterator(std::vector::iterator I) : I(I) {} + + reverse_iterator &operator++() { + --I; + return *this; + } + reverse_iterator &operator--() { + ++I; + return *this; + } + int &operator*() const { return *std::prev(I); } + int *operator->() const { return &*std::prev(I); } + friend bool operator==(const reverse_iterator &L, + const reverse_iterator &R) { + return L.I == R.I; + } + friend bool operator!=(const reverse_iterator &L, + const reverse_iterator &R) { + return !(L == R); + } + }; + + iterator begin() const { return V.begin(); } + iterator end() const { return V.end(); } + reverse_iterator rbegin() const { return reverse_iterator(V.end()); } + reverse_iterator rend() const { return reverse_iterator(V.begin()); } +}; + template void TestRev(const R &r) { int counter = 3; for (int i : r) @@ -98,9 +142,10 @@ TYPED_TEST(RangeAdapterLValueTest, TrivialOperation) { template struct RangeAdapterRValueTest : testing::Test {}; -typedef ::testing::Types, std::list, ReverseOnlyVector, - BidirectionalVector, - BidirectionalVectorConsts> RangeAdapterRValueTestTypes; +typedef ::testing::Types, std::list, CustomIteratorVector, + ReverseOnlyVector, BidirectionalVector, + BidirectionalVectorConsts> + RangeAdapterRValueTestTypes; TYPED_TEST_CASE(RangeAdapterRValueTest, RangeAdapterRValueTestTypes); TYPED_TEST(RangeAdapterRValueTest, TrivialOperation) { @@ -108,7 +153,20 @@ TYPED_TEST(RangeAdapterRValueTest, TrivialOperation) { } TYPED_TEST(RangeAdapterRValueTest, HasRbegin) { - EXPECT_TRUE(has_rbegin::value); + static_assert(has_rbegin::value, "rbegin() should be defined"); +} + +TYPED_TEST(RangeAdapterRValueTest, RangeType) { + static_assert( + std::is_same< + decltype(reverse(*static_cast(nullptr)).begin()), + decltype(static_cast(nullptr)->rbegin())>::value, + "reverse().begin() should have the same type as rbegin()"); + static_assert( + std::is_same< + decltype(reverse(*static_cast(nullptr)).begin()), + decltype(static_cast(nullptr)->rbegin())>::value, + "reverse().begin() should have the same type as rbegin() [const]"); } } // anonymous namespace