1 //===- iterator.h - Utilities for using and defining iterators --*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #ifndef LLVM_ADT_ITERATOR_H
11 #define LLVM_ADT_ITERATOR_H
13 #include "llvm/ADT/iterator_range.h"
17 #include <type_traits>
22 /// CRTP base class which implements the entire standard iterator facade
23 /// in terms of a minimal subset of the interface.
25 /// Use this when it is reasonable to implement most of the iterator
26 /// functionality in terms of a core subset. If you need special behavior or
27 /// there are performance implications for this, you may want to override the
28 /// relevant members instead.
30 /// Note, one abstraction that this does *not* provide is implementing
31 /// subtraction in terms of addition by negating the difference. Negation isn't
32 /// always information preserving, and I can see very reasonable iterator
33 /// designs where this doesn't work well. It doesn't really force much added
34 /// boilerplate anyways.
36 /// Another abstraction that this doesn't provide is implementing increment in
37 /// terms of addition of one. These aren't equivalent for all iterator
38 /// categories, and respecting that adds a lot of complexity for little gain.
40 /// Classes wishing to use `iterator_facade_base` should implement the following
43 /// Forward Iterators:
44 /// (All of the following methods)
45 /// - DerivedT &operator=(const DerivedT &R);
46 /// - bool operator==(const DerivedT &R) const;
47 /// - const T &operator*() const;
49 /// - DerivedT &operator++();
51 /// Bidirectional Iterators:
52 /// (All methods of forward iterators, plus the following)
53 /// - DerivedT &operator--();
55 /// Random-access Iterators:
56 /// (All methods of bidirectional iterators excluding the following)
57 /// - DerivedT &operator++();
58 /// - DerivedT &operator--();
59 /// (and plus the following)
60 /// - bool operator<(const DerivedT &RHS) const;
61 /// - DifferenceTypeT operator-(const DerivedT &R) const;
62 /// - DerivedT &operator+=(DifferenceTypeT N);
63 /// - DerivedT &operator-=(DifferenceTypeT N);
65 template <typename DerivedT, typename IteratorCategoryT, typename T,
66 typename DifferenceTypeT = std::ptrdiff_t, typename PointerT = T *,
67 typename ReferenceT = T &>
68 class iterator_facade_base
69 : public std::iterator<IteratorCategoryT, T, DifferenceTypeT, PointerT,
73 IsRandomAccess = std::is_base_of<std::random_access_iterator_tag,
74 IteratorCategoryT>::value,
75 IsBidirectional = std::is_base_of<std::bidirectional_iterator_tag,
76 IteratorCategoryT>::value,
79 /// A proxy object for computing a reference via indirecting a copy of an
80 /// iterator. This is used in APIs which need to produce a reference via
81 /// indirection but for which the iterator object might be a temporary. The
82 /// proxy preserves the iterator internally and exposes the indirected
83 /// reference via a conversion operator.
84 class ReferenceProxy {
85 friend iterator_facade_base;
89 ReferenceProxy(DerivedT I) : I(std::move(I)) {}
92 operator ReferenceT() const { return *I; }
96 DerivedT operator+(DifferenceTypeT n) const {
97 static_assert(std::is_base_of<iterator_facade_base, DerivedT>::value,
98 "Must pass the derived type to this template!");
101 "The '+' operator is only defined for random access iterators.");
102 DerivedT tmp = *static_cast<const DerivedT *>(this);
106 friend DerivedT operator+(DifferenceTypeT n, const DerivedT &i) {
109 "The '+' operator is only defined for random access iterators.");
112 DerivedT operator-(DifferenceTypeT n) const {
115 "The '-' operator is only defined for random access iterators.");
116 DerivedT tmp = *static_cast<const DerivedT *>(this);
121 DerivedT &operator++() {
122 static_assert(std::is_base_of<iterator_facade_base, DerivedT>::value,
123 "Must pass the derived type to this template!");
124 return static_cast<DerivedT *>(this)->operator+=(1);
126 DerivedT operator++(int) {
127 DerivedT tmp = *static_cast<DerivedT *>(this);
128 ++*static_cast<DerivedT *>(this);
131 DerivedT &operator--() {
134 "The decrement operator is only defined for bidirectional iterators.");
135 return static_cast<DerivedT *>(this)->operator-=(1);
137 DerivedT operator--(int) {
140 "The decrement operator is only defined for bidirectional iterators.");
141 DerivedT tmp = *static_cast<DerivedT *>(this);
142 --*static_cast<DerivedT *>(this);
146 bool operator!=(const DerivedT &RHS) const {
147 return !static_cast<const DerivedT *>(this)->operator==(RHS);
150 bool operator>(const DerivedT &RHS) const {
153 "Relational operators are only defined for random access iterators.");
154 return !static_cast<const DerivedT *>(this)->operator<(RHS) &&
155 !static_cast<const DerivedT *>(this)->operator==(RHS);
157 bool operator<=(const DerivedT &RHS) const {
160 "Relational operators are only defined for random access iterators.");
161 return !static_cast<const DerivedT *>(this)->operator>(RHS);
163 bool operator>=(const DerivedT &RHS) const {
166 "Relational operators are only defined for random access iterators.");
167 return !static_cast<const DerivedT *>(this)->operator<(RHS);
170 PointerT operator->() { return &static_cast<DerivedT *>(this)->operator*(); }
171 PointerT operator->() const {
172 return &static_cast<const DerivedT *>(this)->operator*();
174 ReferenceProxy operator[](DifferenceTypeT n) {
175 static_assert(IsRandomAccess,
176 "Subscripting is only defined for random access iterators.");
177 return ReferenceProxy(static_cast<DerivedT *>(this)->operator+(n));
179 ReferenceProxy operator[](DifferenceTypeT n) const {
180 static_assert(IsRandomAccess,
181 "Subscripting is only defined for random access iterators.");
182 return ReferenceProxy(static_cast<const DerivedT *>(this)->operator+(n));
186 /// CRTP base class for adapting an iterator to a different type.
188 /// This class can be used through CRTP to adapt one iterator into another.
189 /// Typically this is done through providing in the derived class a custom \c
190 /// operator* implementation. Other methods can be overridden as well.
192 typename DerivedT, typename WrappedIteratorT,
193 typename IteratorCategoryT =
194 typename std::iterator_traits<WrappedIteratorT>::iterator_category,
195 typename T = typename std::iterator_traits<WrappedIteratorT>::value_type,
196 typename DifferenceTypeT =
197 typename std::iterator_traits<WrappedIteratorT>::difference_type,
198 typename PointerT = typename std::conditional<
199 std::is_same<T, typename std::iterator_traits<
200 WrappedIteratorT>::value_type>::value,
201 typename std::iterator_traits<WrappedIteratorT>::pointer, T *>::type,
202 typename ReferenceT = typename std::conditional<
203 std::is_same<T, typename std::iterator_traits<
204 WrappedIteratorT>::value_type>::value,
205 typename std::iterator_traits<WrappedIteratorT>::reference, T &>::type,
206 // Don't provide these, they are mostly to act as aliases below.
207 typename WrappedTraitsT = std::iterator_traits<WrappedIteratorT>>
208 class iterator_adaptor_base
209 : public iterator_facade_base<DerivedT, IteratorCategoryT, T,
210 DifferenceTypeT, PointerT, ReferenceT> {
211 using BaseT = typename iterator_adaptor_base::iterator_facade_base;
216 iterator_adaptor_base() = default;
218 explicit iterator_adaptor_base(WrappedIteratorT u) : I(std::move(u)) {
219 static_assert(std::is_base_of<iterator_adaptor_base, DerivedT>::value,
220 "Must pass the derived type to this template!");
223 const WrappedIteratorT &wrapped() const { return I; }
226 using difference_type = DifferenceTypeT;
228 DerivedT &operator+=(difference_type n) {
230 BaseT::IsRandomAccess,
231 "The '+=' operator is only defined for random access iterators.");
233 return *static_cast<DerivedT *>(this);
235 DerivedT &operator-=(difference_type n) {
237 BaseT::IsRandomAccess,
238 "The '-=' operator is only defined for random access iterators.");
240 return *static_cast<DerivedT *>(this);
242 using BaseT::operator-;
243 difference_type operator-(const DerivedT &RHS) const {
245 BaseT::IsRandomAccess,
246 "The '-' operator is only defined for random access iterators.");
250 // We have to explicitly provide ++ and -- rather than letting the facade
251 // forward to += because WrappedIteratorT might not support +=.
252 using BaseT::operator++;
253 DerivedT &operator++() {
255 return *static_cast<DerivedT *>(this);
257 using BaseT::operator--;
258 DerivedT &operator--() {
260 BaseT::IsBidirectional,
261 "The decrement operator is only defined for bidirectional iterators.");
263 return *static_cast<DerivedT *>(this);
266 bool operator==(const DerivedT &RHS) const { return I == RHS.I; }
267 bool operator<(const DerivedT &RHS) const {
269 BaseT::IsRandomAccess,
270 "Relational operators are only defined for random access iterators.");
274 ReferenceT operator*() const { return *I; }
277 /// An iterator type that allows iterating over the pointees via some
280 /// The typical usage of this is to expose a type that iterates over Ts, but
281 /// which is implemented with some iterator over T*s:
284 /// using iterator = pointee_iterator<SmallVectorImpl<T *>::iterator>;
286 template <typename WrappedIteratorT,
287 typename T = typename std::remove_reference<
288 decltype(**std::declval<WrappedIteratorT>())>::type>
289 struct pointee_iterator
290 : iterator_adaptor_base<
291 pointee_iterator<WrappedIteratorT, T>, WrappedIteratorT,
292 typename std::iterator_traits<WrappedIteratorT>::iterator_category,
294 pointee_iterator() = default;
295 template <typename U>
296 pointee_iterator(U &&u)
297 : pointee_iterator::iterator_adaptor_base(std::forward<U &&>(u)) {}
299 T &operator*() const { return **this->I; }
302 template <typename RangeT, typename WrappedIteratorT =
303 decltype(std::begin(std::declval<RangeT>()))>
304 iterator_range<pointee_iterator<WrappedIteratorT>>
305 make_pointee_range(RangeT &&Range) {
306 using PointeeIteratorT = pointee_iterator<WrappedIteratorT>;
307 return make_range(PointeeIteratorT(std::begin(std::forward<RangeT>(Range))),
308 PointeeIteratorT(std::end(std::forward<RangeT>(Range))));
311 template <typename WrappedIteratorT,
312 typename T = decltype(&*std::declval<WrappedIteratorT>())>
313 class pointer_iterator
314 : public iterator_adaptor_base<pointer_iterator<WrappedIteratorT, T>,
315 WrappedIteratorT, T> {
319 pointer_iterator() = default;
321 explicit pointer_iterator(WrappedIteratorT u)
322 : pointer_iterator::iterator_adaptor_base(std::move(u)) {}
324 T &operator*() { return Ptr = &*this->I; }
325 const T &operator*() const { return Ptr = &*this->I; }
328 template <typename RangeT, typename WrappedIteratorT =
329 decltype(std::begin(std::declval<RangeT>()))>
330 iterator_range<pointer_iterator<WrappedIteratorT>>
331 make_pointer_range(RangeT &&Range) {
332 using PointerIteratorT = pointer_iterator<WrappedIteratorT>;
333 return make_range(PointerIteratorT(std::begin(std::forward<RangeT>(Range))),
334 PointerIteratorT(std::end(std::forward<RangeT>(Range))));
337 } // end namespace llvm
339 #endif // LLVM_ADT_ITERATOR_H