1 //===- llvm/ADT/STLExtras.h - Useful STL related functions ------*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file contains some templates that are useful if you are working with the
11 // STL at all.
12 //
13 // No library is required when using these functions.
14 //
15 //===----------------------------------------------------------------------===//
16
17 #ifndef LLVM_ADT_STLEXTRAS_H
18 #define LLVM_ADT_STLEXTRAS_H
19
20 #include <algorithm> // for std::all_of
21 #include <cassert>
22 #include <cstddef> // for std::size_t
23 #include <cstdlib> // for qsort
24 #include <functional>
25 #include <iterator>
26 #include <memory>
27 #include <tuple>
28 #include <utility> // for std::pair
29
30 #include "llvm/ADT/Optional.h"
31 #include "llvm/ADT/iterator.h"
32 #include "llvm/ADT/iterator_range.h"
33 #include "llvm/Support/Compiler.h"
34
35 namespace llvm {
36
37 // Only used by compiler if both template types are the same. Useful when
38 // using SFINAE to test for the existence of member functions.
39 template <typename T, T> struct SameType;
40
41 namespace detail {
42
43 template <typename RangeT>
44 using IterOfRange = decltype(std::begin(std::declval<RangeT &>()));
45
46 } // End detail namespace
47
48 //===----------------------------------------------------------------------===//
49 // Extra additions to <functional>
50 //===----------------------------------------------------------------------===//
51
52 template<class Ty>
53 struct identity : public std::unary_function<Ty, Ty> {
operatoridentity54 Ty &operator()(Ty &self) const {
55 return self;
56 }
operatoridentity57 const Ty &operator()(const Ty &self) const {
58 return self;
59 }
60 };
61
62 template<class Ty>
63 struct less_ptr : public std::binary_function<Ty, Ty, bool> {
operatorless_ptr64 bool operator()(const Ty* left, const Ty* right) const {
65 return *left < *right;
66 }
67 };
68
69 template<class Ty>
70 struct greater_ptr : public std::binary_function<Ty, Ty, bool> {
operatorgreater_ptr71 bool operator()(const Ty* left, const Ty* right) const {
72 return *right < *left;
73 }
74 };
75
76 /// An efficient, type-erasing, non-owning reference to a callable. This is
77 /// intended for use as the type of a function parameter that is not used
78 /// after the function in question returns.
79 ///
80 /// This class does not own the callable, so it is not in general safe to store
81 /// a function_ref.
82 template<typename Fn> class function_ref;
83
84 template<typename Ret, typename ...Params>
85 class function_ref<Ret(Params...)> {
86 Ret (*callback)(intptr_t callable, Params ...params);
87 intptr_t callable;
88
89 template<typename Callable>
callback_fn(intptr_t callable,Params...params)90 static Ret callback_fn(intptr_t callable, Params ...params) {
91 return (*reinterpret_cast<Callable*>(callable))(
92 std::forward<Params>(params)...);
93 }
94
95 public:
96 template <typename Callable>
97 function_ref(Callable &&callable,
98 typename std::enable_if<
99 !std::is_same<typename std::remove_reference<Callable>::type,
100 function_ref>::value>::type * = nullptr)
callback(callback_fn<typename std::remove_reference<Callable>::type>)101 : callback(callback_fn<typename std::remove_reference<Callable>::type>),
102 callable(reinterpret_cast<intptr_t>(&callable)) {}
operator()103 Ret operator()(Params ...params) const {
104 return callback(callable, std::forward<Params>(params)...);
105 }
106 };
107
108 // deleter - Very very very simple method that is used to invoke operator
109 // delete on something. It is used like this:
110 //
111 // for_each(V.begin(), B.end(), deleter<Interval>);
112 //
113 template <class T>
deleter(T * Ptr)114 inline void deleter(T *Ptr) {
115 delete Ptr;
116 }
117
118
119
120 //===----------------------------------------------------------------------===//
121 // Extra additions to <iterator>
122 //===----------------------------------------------------------------------===//
123
124 // mapped_iterator - This is a simple iterator adapter that causes a function to
125 // be dereferenced whenever operator* is invoked on the iterator.
126 //
127 template <class RootIt, class UnaryFunc>
128 class mapped_iterator {
129 RootIt current;
130 UnaryFunc Fn;
131 public:
132 typedef typename std::iterator_traits<RootIt>::iterator_category
133 iterator_category;
134 typedef typename std::iterator_traits<RootIt>::difference_type
135 difference_type;
136 typedef typename std::result_of<
137 UnaryFunc(decltype(*std::declval<RootIt>()))>
138 ::type value_type;
139
140 typedef void pointer;
141 //typedef typename UnaryFunc::result_type *pointer;
142 typedef void reference; // Can't modify value returned by fn
143
144 typedef RootIt iterator_type;
145
getCurrent()146 inline const RootIt &getCurrent() const { return current; }
getFunc()147 inline const UnaryFunc &getFunc() const { return Fn; }
148
mapped_iterator(const RootIt & I,UnaryFunc F)149 inline explicit mapped_iterator(const RootIt &I, UnaryFunc F)
150 : current(I), Fn(F) {}
151
152 inline value_type operator*() const { // All this work to do this
153 return Fn(*current); // little change
154 }
155
156 mapped_iterator &operator++() {
157 ++current;
158 return *this;
159 }
160 mapped_iterator &operator--() {
161 --current;
162 return *this;
163 }
164 mapped_iterator operator++(int) {
165 mapped_iterator __tmp = *this;
166 ++current;
167 return __tmp;
168 }
169 mapped_iterator operator--(int) {
170 mapped_iterator __tmp = *this;
171 --current;
172 return __tmp;
173 }
174 mapped_iterator operator+(difference_type n) const {
175 return mapped_iterator(current + n, Fn);
176 }
177 mapped_iterator &operator+=(difference_type n) {
178 current += n;
179 return *this;
180 }
181 mapped_iterator operator-(difference_type n) const {
182 return mapped_iterator(current - n, Fn);
183 }
184 mapped_iterator &operator-=(difference_type n) {
185 current -= n;
186 return *this;
187 }
188 reference operator[](difference_type n) const { return *(*this + n); }
189
190 bool operator!=(const mapped_iterator &X) const { return !operator==(X); }
191 bool operator==(const mapped_iterator &X) const {
192 return current == X.current;
193 }
194 bool operator<(const mapped_iterator &X) const { return current < X.current; }
195
196 difference_type operator-(const mapped_iterator &X) const {
197 return current - X.current;
198 }
199 };
200
201 template <class Iterator, class Func>
202 inline mapped_iterator<Iterator, Func>
203 operator+(typename mapped_iterator<Iterator, Func>::difference_type N,
204 const mapped_iterator<Iterator, Func> &X) {
205 return mapped_iterator<Iterator, Func>(X.getCurrent() - N, X.getFunc());
206 }
207
208
209 // map_iterator - Provide a convenient way to create mapped_iterators, just like
210 // make_pair is useful for creating pairs...
211 //
212 template <class ItTy, class FuncTy>
map_iterator(const ItTy & I,FuncTy F)213 inline mapped_iterator<ItTy, FuncTy> map_iterator(const ItTy &I, FuncTy F) {
214 return mapped_iterator<ItTy, FuncTy>(I, F);
215 }
216
217 /// Helper to determine if type T has a member called rbegin().
218 template <typename Ty> class has_rbegin_impl {
219 typedef char yes[1];
220 typedef char no[2];
221
222 template <typename Inner>
223 static yes& test(Inner *I, decltype(I->rbegin()) * = nullptr);
224
225 template <typename>
226 static no& test(...);
227
228 public:
229 static const bool value = sizeof(test<Ty>(nullptr)) == sizeof(yes);
230 };
231
232 /// Metafunction to determine if T& or T has a member called rbegin().
233 template <typename Ty>
234 struct has_rbegin : has_rbegin_impl<typename std::remove_reference<Ty>::type> {
235 };
236
237 // Returns an iterator_range over the given container which iterates in reverse.
238 // Note that the container must have rbegin()/rend() methods for this to work.
239 template <typename ContainerTy>
240 auto reverse(ContainerTy &&C,
241 typename std::enable_if<has_rbegin<ContainerTy>::value>::type * =
242 nullptr) -> decltype(make_range(C.rbegin(), C.rend())) {
243 return make_range(C.rbegin(), C.rend());
244 }
245
246 // Returns a std::reverse_iterator wrapped around the given iterator.
247 template <typename IteratorTy>
make_reverse_iterator(IteratorTy It)248 std::reverse_iterator<IteratorTy> make_reverse_iterator(IteratorTy It) {
249 return std::reverse_iterator<IteratorTy>(It);
250 }
251
252 // Returns an iterator_range over the given container which iterates in reverse.
253 // Note that the container must have begin()/end() methods which return
254 // bidirectional iterators for this to work.
255 template <typename ContainerTy>
256 auto reverse(
257 ContainerTy &&C,
258 typename std::enable_if<!has_rbegin<ContainerTy>::value>::type * = nullptr)
259 -> decltype(make_range(llvm::make_reverse_iterator(std::end(C)),
260 llvm::make_reverse_iterator(std::begin(C)))) {
261 return make_range(llvm::make_reverse_iterator(std::end(C)),
262 llvm::make_reverse_iterator(std::begin(C)));
263 }
264
265 /// An iterator adaptor that filters the elements of given inner iterators.
266 ///
267 /// The predicate parameter should be a callable object that accepts the wrapped
268 /// iterator's reference type and returns a bool. When incrementing or
269 /// decrementing the iterator, it will call the predicate on each element and
270 /// skip any where it returns false.
271 ///
272 /// \code
273 /// int A[] = { 1, 2, 3, 4 };
274 /// auto R = make_filter_range(A, [](int N) { return N % 2 == 1; });
275 /// // R contains { 1, 3 }.
276 /// \endcode
277 template <typename WrappedIteratorT, typename PredicateT>
278 class filter_iterator
279 : public iterator_adaptor_base<
280 filter_iterator<WrappedIteratorT, PredicateT>, WrappedIteratorT,
281 typename std::common_type<
282 std::forward_iterator_tag,
283 typename std::iterator_traits<
284 WrappedIteratorT>::iterator_category>::type> {
285 using BaseT = iterator_adaptor_base<
286 filter_iterator<WrappedIteratorT, PredicateT>, WrappedIteratorT,
287 typename std::common_type<
288 std::forward_iterator_tag,
289 typename std::iterator_traits<WrappedIteratorT>::iterator_category>::
290 type>;
291
292 struct PayloadType {
293 WrappedIteratorT End;
294 PredicateT Pred;
295 };
296
297 Optional<PayloadType> Payload;
298
findNextValid()299 void findNextValid() {
300 assert(Payload && "Payload should be engaged when findNextValid is called");
301 while (this->I != Payload->End && !Payload->Pred(*this->I))
302 BaseT::operator++();
303 }
304
305 // Construct the begin iterator. The begin iterator requires to know where end
306 // is, so that it can properly stop when it hits end.
filter_iterator(WrappedIteratorT Begin,WrappedIteratorT End,PredicateT Pred)307 filter_iterator(WrappedIteratorT Begin, WrappedIteratorT End, PredicateT Pred)
308 : BaseT(std::move(Begin)),
309 Payload(PayloadType{std::move(End), std::move(Pred)}) {
310 findNextValid();
311 }
312
313 // Construct the end iterator. It's not incrementable, so Payload doesn't
314 // have to be engaged.
filter_iterator(WrappedIteratorT End)315 filter_iterator(WrappedIteratorT End) : BaseT(End) {}
316
317 public:
318 using BaseT::operator++;
319
320 filter_iterator &operator++() {
321 BaseT::operator++();
322 findNextValid();
323 return *this;
324 }
325
326 template <typename RT, typename PT>
327 friend iterator_range<filter_iterator<detail::IterOfRange<RT>, PT>>
328 make_filter_range(RT &&, PT);
329 };
330
331 /// Convenience function that takes a range of elements and a predicate,
332 /// and return a new filter_iterator range.
333 ///
334 /// FIXME: Currently if RangeT && is a rvalue reference to a temporary, the
335 /// lifetime of that temporary is not kept by the returned range object, and the
336 /// temporary is going to be dropped on the floor after the make_iterator_range
337 /// full expression that contains this function call.
338 template <typename RangeT, typename PredicateT>
339 iterator_range<filter_iterator<detail::IterOfRange<RangeT>, PredicateT>>
make_filter_range(RangeT && Range,PredicateT Pred)340 make_filter_range(RangeT &&Range, PredicateT Pred) {
341 using FilterIteratorT =
342 filter_iterator<detail::IterOfRange<RangeT>, PredicateT>;
343 return make_range(FilterIteratorT(std::begin(std::forward<RangeT>(Range)),
344 std::end(std::forward<RangeT>(Range)),
345 std::move(Pred)),
346 FilterIteratorT(std::end(std::forward<RangeT>(Range))));
347 }
348
349 // forward declarations required by zip_shortest/zip_first
350 template <typename R, typename UnaryPredicate>
351 bool all_of(R &&range, UnaryPredicate P);
352
353 template <size_t... I> struct index_sequence;
354
355 template <class... Ts> struct index_sequence_for;
356
357 namespace detail {
358 template <typename... Iters> class zip_first {
359 public:
360 typedef std::input_iterator_tag iterator_category;
361 typedef std::tuple<decltype(*std::declval<Iters>())...> value_type;
362 std::tuple<Iters...> iterators;
363
364 private:
deres(index_sequence<Ns...>)365 template <size_t... Ns> value_type deres(index_sequence<Ns...>) {
366 return value_type(*std::get<Ns>(iterators)...);
367 }
368
decltype(iterators)369 template <size_t... Ns> decltype(iterators) tup_inc(index_sequence<Ns...>) {
370 return std::tuple<Iters...>(std::next(std::get<Ns>(iterators))...);
371 }
372
373 public:
374 value_type operator*() { return deres(index_sequence_for<Iters...>{}); }
375
376 void operator++() { iterators = tup_inc(index_sequence_for<Iters...>{}); }
377
378 bool operator!=(const zip_first<Iters...> &other) const {
379 return std::get<0>(iterators) != std::get<0>(other.iterators);
380 }
zip_first(Iters &&...ts)381 zip_first(Iters &&... ts) : iterators(std::forward<Iters>(ts)...) {}
382 };
383
384 template <typename... Iters> class zip_shortest : public zip_first<Iters...> {
385 template <size_t... Ns>
test(const zip_first<Iters...> & other,index_sequence<Ns...>)386 bool test(const zip_first<Iters...> &other, index_sequence<Ns...>) const {
387 return all_of(std::initializer_list<bool>{std::get<Ns>(this->iterators) !=
388 std::get<Ns>(other.iterators)...},
389 identity<bool>{});
390 }
391
392 public:
393 bool operator!=(const zip_first<Iters...> &other) const {
394 return test(other, index_sequence_for<Iters...>{});
395 }
zip_shortest(Iters &&...ts)396 zip_shortest(Iters &&... ts)
397 : zip_first<Iters...>(std::forward<Iters>(ts)...) {}
398 };
399
400 template <template <typename...> class ItType, typename... Args> class zippy {
401 public:
402 typedef ItType<decltype(std::begin(std::declval<Args>()))...> iterator;
403
404 private:
405 std::tuple<Args...> ts;
406
begin_impl(index_sequence<Ns...>)407 template <size_t... Ns> iterator begin_impl(index_sequence<Ns...>) {
408 return iterator(std::begin(std::get<Ns>(ts))...);
409 }
end_impl(index_sequence<Ns...>)410 template <size_t... Ns> iterator end_impl(index_sequence<Ns...>) {
411 return iterator(std::end(std::get<Ns>(ts))...);
412 }
413
414 public:
begin()415 iterator begin() { return begin_impl(index_sequence_for<Args...>{}); }
end()416 iterator end() { return end_impl(index_sequence_for<Args...>{}); }
zippy(Args &&...ts_)417 zippy(Args &&... ts_) : ts(std::forward<Args>(ts_)...) {}
418 };
419 } // End detail namespace
420
421 /// zip iterator for two or more iteratable types.
422 template <typename T, typename U, typename... Args>
zip(T && t,U && u,Args &&...args)423 detail::zippy<detail::zip_shortest, T, U, Args...> zip(T &&t, U &&u,
424 Args &&... args) {
425 return detail::zippy<detail::zip_shortest, T, U, Args...>(
426 std::forward<T>(t), std::forward<U>(u), std::forward<Args>(args)...);
427 }
428
429 /// zip iterator that, for the sake of efficiency, assumes the first iteratee to
430 /// be the shortest.
431 template <typename T, typename U, typename... Args>
zip_first(T && t,U && u,Args &&...args)432 detail::zippy<detail::zip_first, T, U, Args...> zip_first(T &&t, U &&u,
433 Args &&... args) {
434 return detail::zippy<detail::zip_first, T, U, Args...>(
435 std::forward<T>(t), std::forward<U>(u), std::forward<Args>(args)...);
436 }
437
438 //===----------------------------------------------------------------------===//
439 // Extra additions to <utility>
440 //===----------------------------------------------------------------------===//
441
442 /// \brief Function object to check whether the first component of a std::pair
443 /// compares less than the first component of another std::pair.
444 struct less_first {
operatorless_first445 template <typename T> bool operator()(const T &lhs, const T &rhs) const {
446 return lhs.first < rhs.first;
447 }
448 };
449
450 /// \brief Function object to check whether the second component of a std::pair
451 /// compares less than the second component of another std::pair.
452 struct less_second {
operatorless_second453 template <typename T> bool operator()(const T &lhs, const T &rhs) const {
454 return lhs.second < rhs.second;
455 }
456 };
457
458 // A subset of N3658. More stuff can be added as-needed.
459
460 /// \brief Represents a compile-time sequence of integers.
461 template <class T, T... I> struct integer_sequence {
462 typedef T value_type;
463
sizeinteger_sequence464 static constexpr size_t size() { return sizeof...(I); }
465 };
466
467 /// \brief Alias for the common case of a sequence of size_ts.
468 template <size_t... I>
469 struct index_sequence : integer_sequence<std::size_t, I...> {};
470
471 template <std::size_t N, std::size_t... I>
472 struct build_index_impl : build_index_impl<N - 1, N - 1, I...> {};
473 template <std::size_t... I>
474 struct build_index_impl<0, I...> : index_sequence<I...> {};
475
476 /// \brief Creates a compile-time integer sequence for a parameter pack.
477 template <class... Ts>
478 struct index_sequence_for : build_index_impl<sizeof...(Ts)> {};
479
480 /// Utility type to build an inheritance chain that makes it easy to rank
481 /// overload candidates.
482 template <int N> struct rank : rank<N - 1> {};
483 template <> struct rank<0> {};
484
485 /// \brief traits class for checking whether type T is one of any of the given
486 /// types in the variadic list.
487 template <typename T, typename... Ts> struct is_one_of {
488 static const bool value = false;
489 };
490
491 template <typename T, typename U, typename... Ts>
492 struct is_one_of<T, U, Ts...> {
493 static const bool value =
494 std::is_same<T, U>::value || is_one_of<T, Ts...>::value;
495 };
496
497 //===----------------------------------------------------------------------===//
498 // Extra additions for arrays
499 //===----------------------------------------------------------------------===//
500
501 /// Find the length of an array.
502 template <class T, std::size_t N>
503 constexpr inline size_t array_lengthof(T (&)[N]) {
504 return N;
505 }
506
507 /// Adapt std::less<T> for array_pod_sort.
508 template<typename T>
509 inline int array_pod_sort_comparator(const void *P1, const void *P2) {
510 if (std::less<T>()(*reinterpret_cast<const T*>(P1),
511 *reinterpret_cast<const T*>(P2)))
512 return -1;
513 if (std::less<T>()(*reinterpret_cast<const T*>(P2),
514 *reinterpret_cast<const T*>(P1)))
515 return 1;
516 return 0;
517 }
518
519 /// get_array_pod_sort_comparator - This is an internal helper function used to
520 /// get type deduction of T right.
521 template<typename T>
522 inline int (*get_array_pod_sort_comparator(const T &))
523 (const void*, const void*) {
524 return array_pod_sort_comparator<T>;
525 }
526
527
528 /// array_pod_sort - This sorts an array with the specified start and end
529 /// extent. This is just like std::sort, except that it calls qsort instead of
530 /// using an inlined template. qsort is slightly slower than std::sort, but
531 /// most sorts are not performance critical in LLVM and std::sort has to be
532 /// template instantiated for each type, leading to significant measured code
533 /// bloat. This function should generally be used instead of std::sort where
534 /// possible.
535 ///
536 /// This function assumes that you have simple POD-like types that can be
537 /// compared with std::less and can be moved with memcpy. If this isn't true,
538 /// you should use std::sort.
539 ///
540 /// NOTE: If qsort_r were portable, we could allow a custom comparator and
541 /// default to std::less.
542 template<class IteratorTy>
543 inline void array_pod_sort(IteratorTy Start, IteratorTy End) {
544 // Don't inefficiently call qsort with one element or trigger undefined
545 // behavior with an empty sequence.
546 auto NElts = End - Start;
547 if (NElts <= 1) return;
548 qsort(&*Start, NElts, sizeof(*Start), get_array_pod_sort_comparator(*Start));
549 }
550
551 template <class IteratorTy>
552 inline void array_pod_sort(
553 IteratorTy Start, IteratorTy End,
554 int (*Compare)(
555 const typename std::iterator_traits<IteratorTy>::value_type *,
556 const typename std::iterator_traits<IteratorTy>::value_type *)) {
557 // Don't inefficiently call qsort with one element or trigger undefined
558 // behavior with an empty sequence.
559 auto NElts = End - Start;
560 if (NElts <= 1) return;
561 qsort(&*Start, NElts, sizeof(*Start),
562 reinterpret_cast<int (*)(const void *, const void *)>(Compare));
563 }
564
565 //===----------------------------------------------------------------------===//
566 // Extra additions to <algorithm>
567 //===----------------------------------------------------------------------===//
568
569 /// For a container of pointers, deletes the pointers and then clears the
570 /// container.
571 template<typename Container>
572 void DeleteContainerPointers(Container &C) {
573 for (auto V : C)
574 delete V;
575 C.clear();
576 }
577
578 /// In a container of pairs (usually a map) whose second element is a pointer,
579 /// deletes the second elements and then clears the container.
580 template<typename Container>
581 void DeleteContainerSeconds(Container &C) {
582 for (auto &V : C)
583 delete V.second;
584 C.clear();
585 }
586
587 /// Provide wrappers to std::all_of which take ranges instead of having to pass
588 /// begin/end explicitly.
589 template <typename R, typename UnaryPredicate>
590 bool all_of(R &&Range, UnaryPredicate P) {
591 return std::all_of(std::begin(Range), std::end(Range), P);
592 }
593
594 /// Provide wrappers to std::any_of which take ranges instead of having to pass
595 /// begin/end explicitly.
596 template <typename R, typename UnaryPredicate>
597 bool any_of(R &&Range, UnaryPredicate P) {
598 return std::any_of(std::begin(Range), std::end(Range), P);
599 }
600
601 /// Provide wrappers to std::none_of which take ranges instead of having to pass
602 /// begin/end explicitly.
603 template <typename R, typename UnaryPredicate>
604 bool none_of(R &&Range, UnaryPredicate P) {
605 return std::none_of(std::begin(Range), std::end(Range), P);
606 }
607
608 /// Provide wrappers to std::find which take ranges instead of having to pass
609 /// begin/end explicitly.
610 template <typename R, typename T>
611 auto find(R &&Range, const T &Val) -> decltype(std::begin(Range)) {
612 return std::find(std::begin(Range), std::end(Range), Val);
613 }
614
615 /// Provide wrappers to std::find_if which take ranges instead of having to pass
616 /// begin/end explicitly.
617 template <typename R, typename UnaryPredicate>
618 auto find_if(R &&Range, UnaryPredicate P) -> decltype(std::begin(Range)) {
619 return std::find_if(std::begin(Range), std::end(Range), P);
620 }
621
622 template <typename R, typename UnaryPredicate>
623 auto find_if_not(R &&Range, UnaryPredicate P) -> decltype(std::begin(Range)) {
624 return std::find_if_not(std::begin(Range), std::end(Range), P);
625 }
626
627 /// Provide wrappers to std::remove_if which take ranges instead of having to
628 /// pass begin/end explicitly.
629 template <typename R, typename UnaryPredicate>
630 auto remove_if(R &&Range, UnaryPredicate P) -> decltype(std::begin(Range)) {
631 return std::remove_if(std::begin(Range), std::end(Range), P);
632 }
633
634 /// Wrapper function around std::find to detect if an element exists
635 /// in a container.
636 template <typename R, typename E>
637 bool is_contained(R &&Range, const E &Element) {
638 return std::find(std::begin(Range), std::end(Range), Element) !=
639 std::end(Range);
640 }
641
642 /// Wrapper function around std::count to count the number of times an element
643 /// \p Element occurs in the given range \p Range.
644 template <typename R, typename E>
645 auto count(R &&Range, const E &Element) -> typename std::iterator_traits<
646 decltype(std::begin(Range))>::difference_type {
647 return std::count(std::begin(Range), std::end(Range), Element);
648 }
649
650 /// Wrapper function around std::count_if to count the number of times an
651 /// element satisfying a given predicate occurs in a range.
652 template <typename R, typename UnaryPredicate>
653 auto count_if(R &&Range, UnaryPredicate P) -> typename std::iterator_traits<
654 decltype(std::begin(Range))>::difference_type {
655 return std::count_if(std::begin(Range), std::end(Range), P);
656 }
657
658 /// Wrapper function around std::transform to apply a function to a range and
659 /// store the result elsewhere.
660 template <typename R, typename OutputIt, typename UnaryPredicate>
661 OutputIt transform(R &&Range, OutputIt d_first, UnaryPredicate P) {
662 return std::transform(std::begin(Range), std::end(Range), d_first, P);
663 }
664
665 //===----------------------------------------------------------------------===//
666 // Extra additions to <memory>
667 //===----------------------------------------------------------------------===//
668
669 // Implement make_unique according to N3656.
670
671 /// \brief Constructs a `new T()` with the given args and returns a
672 /// `unique_ptr<T>` which owns the object.
673 ///
674 /// Example:
675 ///
676 /// auto p = make_unique<int>();
677 /// auto p = make_unique<std::tuple<int, int>>(0, 1);
678 template <class T, class... Args>
679 typename std::enable_if<!std::is_array<T>::value, std::unique_ptr<T>>::type
680 make_unique(Args &&... args) {
681 return std::unique_ptr<T>(new T(std::forward<Args>(args)...));
682 }
683
684 /// \brief Constructs a `new T[n]` with the given args and returns a
685 /// `unique_ptr<T[]>` which owns the object.
686 ///
687 /// \param n size of the new array.
688 ///
689 /// Example:
690 ///
691 /// auto p = make_unique<int[]>(2); // value-initializes the array with 0's.
692 template <class T>
693 typename std::enable_if<std::is_array<T>::value && std::extent<T>::value == 0,
694 std::unique_ptr<T>>::type
695 make_unique(size_t n) {
696 return std::unique_ptr<T>(new typename std::remove_extent<T>::type[n]());
697 }
698
699 /// This function isn't used and is only here to provide better compile errors.
700 template <class T, class... Args>
701 typename std::enable_if<std::extent<T>::value != 0>::type
702 make_unique(Args &&...) = delete;
703
704 struct FreeDeleter {
705 void operator()(void* v) {
706 ::free(v);
707 }
708 };
709
710 template<typename First, typename Second>
711 struct pair_hash {
712 size_t operator()(const std::pair<First, Second> &P) const {
713 return std::hash<First>()(P.first) * 31 + std::hash<Second>()(P.second);
714 }
715 };
716
717 /// A functor like C++14's std::less<void> in its absence.
718 struct less {
719 template <typename A, typename B> bool operator()(A &&a, B &&b) const {
720 return std::forward<A>(a) < std::forward<B>(b);
721 }
722 };
723
724 /// A functor like C++14's std::equal<void> in its absence.
725 struct equal {
726 template <typename A, typename B> bool operator()(A &&a, B &&b) const {
727 return std::forward<A>(a) == std::forward<B>(b);
728 }
729 };
730
731 /// Binary functor that adapts to any other binary functor after dereferencing
732 /// operands.
733 template <typename T> struct deref {
734 T func;
735 // Could be further improved to cope with non-derivable functors and
736 // non-binary functors (should be a variadic template member function
737 // operator()).
738 template <typename A, typename B>
739 auto operator()(A &lhs, B &rhs) const -> decltype(func(*lhs, *rhs)) {
740 assert(lhs);
741 assert(rhs);
742 return func(*lhs, *rhs);
743 }
744 };
745
746 namespace detail {
747 template <typename R> class enumerator_impl {
748 public:
749 template <typename X> struct result_pair {
750 result_pair(std::size_t Index, X Value) : Index(Index), Value(Value) {}
751
752 const std::size_t Index;
753 X Value;
754 };
755
756 class iterator {
757 typedef
758 typename std::iterator_traits<IterOfRange<R>>::reference iter_reference;
759 typedef result_pair<iter_reference> result_type;
760
761 public:
762 iterator(IterOfRange<R> &&Iter, std::size_t Index)
763 : Iter(Iter), Index(Index) {}
764
765 result_type operator*() const { return result_type(Index, *Iter); }
766
767 iterator &operator++() {
768 ++Iter;
769 ++Index;
770 return *this;
771 }
772
773 bool operator!=(const iterator &RHS) const { return Iter != RHS.Iter; }
774
775 private:
776 IterOfRange<R> Iter;
777 std::size_t Index;
778 };
779
780 public:
781 explicit enumerator_impl(R &&Range) : Range(std::forward<R>(Range)) {}
782
783 iterator begin() { return iterator(std::begin(Range), 0); }
784 iterator end() { return iterator(std::end(Range), std::size_t(-1)); }
785
786 private:
787 R Range;
788 };
789 }
790
791 /// Given an input range, returns a new range whose values are are pair (A,B)
792 /// such that A is the 0-based index of the item in the sequence, and B is
793 /// the value from the original sequence. Example:
794 ///
795 /// std::vector<char> Items = {'A', 'B', 'C', 'D'};
796 /// for (auto X : enumerate(Items)) {
797 /// printf("Item %d - %c\n", X.Index, X.Value);
798 /// }
799 ///
800 /// Output:
801 /// Item 0 - A
802 /// Item 1 - B
803 /// Item 2 - C
804 /// Item 3 - D
805 ///
806 template <typename R> detail::enumerator_impl<R> enumerate(R &&Range) {
807 return detail::enumerator_impl<R>(std::forward<R>(Range));
808 }
809
810 namespace detail {
811 template <typename F, typename Tuple, std::size_t... I>
812 auto apply_tuple_impl(F &&f, Tuple &&t, index_sequence<I...>)
813 -> decltype(std::forward<F>(f)(std::get<I>(std::forward<Tuple>(t))...)) {
814 return std::forward<F>(f)(std::get<I>(std::forward<Tuple>(t))...);
815 }
816 }
817
818 /// Given an input tuple (a1, a2, ..., an), pass the arguments of the
819 /// tuple variadically to f as if by calling f(a1, a2, ..., an) and
820 /// return the result.
821 template <typename F, typename Tuple>
822 auto apply_tuple(F &&f, Tuple &&t) -> decltype(detail::apply_tuple_impl(
823 std::forward<F>(f), std::forward<Tuple>(t),
824 build_index_impl<
825 std::tuple_size<typename std::decay<Tuple>::type>::value>{})) {
826 using Indices = build_index_impl<
827 std::tuple_size<typename std::decay<Tuple>::type>::value>;
828
829 return detail::apply_tuple_impl(std::forward<F>(f), std::forward<Tuple>(t),
830 Indices{});
831 }
832 } // End llvm namespace
833
834 #endif
835