| //===- llvm/ADT/STLExtras.h - Useful STL related functions ------*- C++ -*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file contains some templates that are useful if you are working with the |
| // STL at all. |
| // |
| // No library is required when using these functions. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef LLVM_ADT_STLEXTRAS_H |
| #define LLVM_ADT_STLEXTRAS_H |
| |
| #include <algorithm> // for std::all_of |
| #include <cassert> |
| #include <cstddef> // for std::size_t |
| #include <cstdlib> // for qsort |
| #include <functional> |
| #include <iterator> |
| #include <memory> |
| #include <tuple> |
| #include <utility> // for std::pair |
| |
| #include "llvm/ADT/Optional.h" |
| #include "llvm/ADT/iterator.h" |
| #include "llvm/ADT/iterator_range.h" |
| #include "llvm/Support/Compiler.h" |
| |
| namespace llvm { |
| |
| // Only used by compiler if both template types are the same. Useful when |
| // using SFINAE to test for the existence of member functions. |
| template <typename T, T> struct SameType; |
| |
| namespace detail { |
| |
| template <typename RangeT> |
| using IterOfRange = decltype(std::begin(std::declval<RangeT &>())); |
| |
| } // End detail namespace |
| |
| //===----------------------------------------------------------------------===// |
| // Extra additions to <functional> |
| //===----------------------------------------------------------------------===// |
| |
| template<class Ty> |
| struct identity : public std::unary_function<Ty, Ty> { |
| Ty &operator()(Ty &self) const { |
| return self; |
| } |
| const Ty &operator()(const Ty &self) const { |
| return self; |
| } |
| }; |
| |
| template<class Ty> |
| struct less_ptr : public std::binary_function<Ty, Ty, bool> { |
| bool operator()(const Ty* left, const Ty* right) const { |
| return *left < *right; |
| } |
| }; |
| |
| template<class Ty> |
| struct greater_ptr : public std::binary_function<Ty, Ty, bool> { |
| bool operator()(const Ty* left, const Ty* right) const { |
| return *right < *left; |
| } |
| }; |
| |
| /// An efficient, type-erasing, non-owning reference to a callable. This is |
| /// intended for use as the type of a function parameter that is not used |
| /// after the function in question returns. |
| /// |
| /// This class does not own the callable, so it is not in general safe to store |
| /// a function_ref. |
| template<typename Fn> class function_ref; |
| |
| template<typename Ret, typename ...Params> |
| class function_ref<Ret(Params...)> { |
| Ret (*callback)(intptr_t callable, Params ...params); |
| intptr_t callable; |
| |
| template<typename Callable> |
| static Ret callback_fn(intptr_t callable, Params ...params) { |
| return (*reinterpret_cast<Callable*>(callable))( |
| std::forward<Params>(params)...); |
| } |
| |
| public: |
| template <typename Callable> |
| function_ref(Callable &&callable, |
| typename std::enable_if< |
| !std::is_same<typename std::remove_reference<Callable>::type, |
| function_ref>::value>::type * = nullptr) |
| : callback(callback_fn<typename std::remove_reference<Callable>::type>), |
| callable(reinterpret_cast<intptr_t>(&callable)) {} |
| Ret operator()(Params ...params) const { |
| return callback(callable, std::forward<Params>(params)...); |
| } |
| }; |
| |
| // deleter - Very very very simple method that is used to invoke operator |
| // delete on something. It is used like this: |
| // |
| // for_each(V.begin(), B.end(), deleter<Interval>); |
| // |
| template <class T> |
| inline void deleter(T *Ptr) { |
| delete Ptr; |
| } |
| |
| |
| |
| //===----------------------------------------------------------------------===// |
| // Extra additions to <iterator> |
| //===----------------------------------------------------------------------===// |
| |
| // mapped_iterator - This is a simple iterator adapter that causes a function to |
| // be dereferenced whenever operator* is invoked on the iterator. |
| // |
| template <class RootIt, class UnaryFunc> |
| class mapped_iterator { |
| RootIt current; |
| UnaryFunc Fn; |
| public: |
| typedef typename std::iterator_traits<RootIt>::iterator_category |
| iterator_category; |
| typedef typename std::iterator_traits<RootIt>::difference_type |
| difference_type; |
| typedef typename std::result_of< |
| UnaryFunc(decltype(*std::declval<RootIt>()))> |
| ::type value_type; |
| |
| typedef void pointer; |
| //typedef typename UnaryFunc::result_type *pointer; |
| typedef void reference; // Can't modify value returned by fn |
| |
| typedef RootIt iterator_type; |
| |
| inline const RootIt &getCurrent() const { return current; } |
| inline const UnaryFunc &getFunc() const { return Fn; } |
| |
| inline explicit mapped_iterator(const RootIt &I, UnaryFunc F) |
| : current(I), Fn(F) {} |
| |
| inline value_type operator*() const { // All this work to do this |
| return Fn(*current); // little change |
| } |
| |
| mapped_iterator &operator++() { |
| ++current; |
| return *this; |
| } |
| mapped_iterator &operator--() { |
| --current; |
| return *this; |
| } |
| mapped_iterator operator++(int) { |
| mapped_iterator __tmp = *this; |
| ++current; |
| return __tmp; |
| } |
| mapped_iterator operator--(int) { |
| mapped_iterator __tmp = *this; |
| --current; |
| return __tmp; |
| } |
| mapped_iterator operator+(difference_type n) const { |
| return mapped_iterator(current + n, Fn); |
| } |
| mapped_iterator &operator+=(difference_type n) { |
| current += n; |
| return *this; |
| } |
| mapped_iterator operator-(difference_type n) const { |
| return mapped_iterator(current - n, Fn); |
| } |
| mapped_iterator &operator-=(difference_type n) { |
| current -= n; |
| return *this; |
| } |
| reference operator[](difference_type n) const { return *(*this + n); } |
| |
| bool operator!=(const mapped_iterator &X) const { return !operator==(X); } |
| bool operator==(const mapped_iterator &X) const { |
| return current == X.current; |
| } |
| bool operator<(const mapped_iterator &X) const { return current < X.current; } |
| |
| difference_type operator-(const mapped_iterator &X) const { |
| return current - X.current; |
| } |
| }; |
| |
| template <class Iterator, class Func> |
| inline mapped_iterator<Iterator, Func> |
| operator+(typename mapped_iterator<Iterator, Func>::difference_type N, |
| const mapped_iterator<Iterator, Func> &X) { |
| return mapped_iterator<Iterator, Func>(X.getCurrent() - N, X.getFunc()); |
| } |
| |
| |
| // map_iterator - Provide a convenient way to create mapped_iterators, just like |
| // make_pair is useful for creating pairs... |
| // |
| template <class ItTy, class FuncTy> |
| inline mapped_iterator<ItTy, FuncTy> map_iterator(const ItTy &I, FuncTy F) { |
| return mapped_iterator<ItTy, FuncTy>(I, F); |
| } |
| |
| /// Helper to determine if type T has a member called rbegin(). |
| template <typename Ty> class has_rbegin_impl { |
| typedef char yes[1]; |
| typedef char no[2]; |
| |
| template <typename Inner> |
| static yes& test(Inner *I, decltype(I->rbegin()) * = nullptr); |
| |
| template <typename> |
| static no& test(...); |
| |
| public: |
| static const bool value = sizeof(test<Ty>(nullptr)) == sizeof(yes); |
| }; |
| |
| /// Metafunction to determine if T& or T has a member called rbegin(). |
| template <typename Ty> |
| struct has_rbegin : has_rbegin_impl<typename std::remove_reference<Ty>::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 <typename ContainerTy> |
| auto reverse(ContainerTy &&C, |
| typename std::enable_if<has_rbegin<ContainerTy>::value>::type * = |
| nullptr) -> decltype(make_range(C.rbegin(), C.rend())) { |
| return make_range(C.rbegin(), C.rend()); |
| } |
| |
| // Returns a std::reverse_iterator wrapped around the given iterator. |
| template <typename IteratorTy> |
| std::reverse_iterator<IteratorTy> make_reverse_iterator(IteratorTy It) { |
| return std::reverse_iterator<IteratorTy>(It); |
| } |
| |
| // Returns an iterator_range over the given container which iterates in reverse. |
| // Note that the container must have begin()/end() methods which return |
| // bidirectional iterators for this to work. |
| template <typename ContainerTy> |
| auto reverse( |
| ContainerTy &&C, |
| typename std::enable_if<!has_rbegin<ContainerTy>::value>::type * = nullptr) |
| -> decltype(make_range(llvm::make_reverse_iterator(std::end(C)), |
| llvm::make_reverse_iterator(std::begin(C)))) { |
| return make_range(llvm::make_reverse_iterator(std::end(C)), |
| llvm::make_reverse_iterator(std::begin(C))); |
| } |
| |
| /// An iterator adaptor that filters the elements of given inner iterators. |
| /// |
| /// The predicate parameter should be a callable object that accepts the wrapped |
| /// iterator's reference type and returns a bool. When incrementing or |
| /// decrementing the iterator, it will call the predicate on each element and |
| /// skip any where it returns false. |
| /// |
| /// \code |
| /// int A[] = { 1, 2, 3, 4 }; |
| /// auto R = make_filter_range(A, [](int N) { return N % 2 == 1; }); |
| /// // R contains { 1, 3 }. |
| /// \endcode |
| template <typename WrappedIteratorT, typename PredicateT> |
| class filter_iterator |
| : public iterator_adaptor_base< |
| filter_iterator<WrappedIteratorT, PredicateT>, WrappedIteratorT, |
| typename std::common_type< |
| std::forward_iterator_tag, |
| typename std::iterator_traits< |
| WrappedIteratorT>::iterator_category>::type> { |
| using BaseT = iterator_adaptor_base< |
| filter_iterator<WrappedIteratorT, PredicateT>, WrappedIteratorT, |
| typename std::common_type< |
| std::forward_iterator_tag, |
| typename std::iterator_traits<WrappedIteratorT>::iterator_category>:: |
| type>; |
| |
| struct PayloadType { |
| WrappedIteratorT End; |
| PredicateT Pred; |
| }; |
| |
| Optional<PayloadType> Payload; |
| |
| void findNextValid() { |
| assert(Payload && "Payload should be engaged when findNextValid is called"); |
| while (this->I != Payload->End && !Payload->Pred(*this->I)) |
| BaseT::operator++(); |
| } |
| |
| // Construct the begin iterator. The begin iterator requires to know where end |
| // is, so that it can properly stop when it hits end. |
| filter_iterator(WrappedIteratorT Begin, WrappedIteratorT End, PredicateT Pred) |
| : BaseT(std::move(Begin)), |
| Payload(PayloadType{std::move(End), std::move(Pred)}) { |
| findNextValid(); |
| } |
| |
| // Construct the end iterator. It's not incrementable, so Payload doesn't |
| // have to be engaged. |
| filter_iterator(WrappedIteratorT End) : BaseT(End) {} |
| |
| public: |
| using BaseT::operator++; |
| |
| filter_iterator &operator++() { |
| BaseT::operator++(); |
| findNextValid(); |
| return *this; |
| } |
| |
| template <typename RT, typename PT> |
| friend iterator_range<filter_iterator<detail::IterOfRange<RT>, PT>> |
| make_filter_range(RT &&, PT); |
| }; |
| |
| /// Convenience function that takes a range of elements and a predicate, |
| /// and return a new filter_iterator range. |
| /// |
| /// FIXME: Currently if RangeT && is a rvalue reference to a temporary, the |
| /// lifetime of that temporary is not kept by the returned range object, and the |
| /// temporary is going to be dropped on the floor after the make_iterator_range |
| /// full expression that contains this function call. |
| template <typename RangeT, typename PredicateT> |
| iterator_range<filter_iterator<detail::IterOfRange<RangeT>, PredicateT>> |
| make_filter_range(RangeT &&Range, PredicateT Pred) { |
| using FilterIteratorT = |
| filter_iterator<detail::IterOfRange<RangeT>, PredicateT>; |
| return make_range(FilterIteratorT(std::begin(std::forward<RangeT>(Range)), |
| std::end(std::forward<RangeT>(Range)), |
| std::move(Pred)), |
| FilterIteratorT(std::end(std::forward<RangeT>(Range)))); |
| } |
| |
| // forward declarations required by zip_shortest/zip_first |
| template <typename R, typename UnaryPredicate> |
| bool all_of(R &&range, UnaryPredicate P); |
| |
| template <size_t... I> struct index_sequence; |
| |
| template <class... Ts> struct index_sequence_for; |
| |
| namespace detail { |
| template <typename... Iters> class zip_first { |
| public: |
| typedef std::input_iterator_tag iterator_category; |
| typedef std::tuple<decltype(*std::declval<Iters>())...> value_type; |
| std::tuple<Iters...> iterators; |
| |
| private: |
| template <size_t... Ns> value_type deres(index_sequence<Ns...>) { |
| return value_type(*std::get<Ns>(iterators)...); |
| } |
| |
| template <size_t... Ns> decltype(iterators) tup_inc(index_sequence<Ns...>) { |
| return std::tuple<Iters...>(std::next(std::get<Ns>(iterators))...); |
| } |
| |
| public: |
| value_type operator*() { return deres(index_sequence_for<Iters...>{}); } |
| |
| void operator++() { iterators = tup_inc(index_sequence_for<Iters...>{}); } |
| |
| bool operator!=(const zip_first<Iters...> &other) const { |
| return std::get<0>(iterators) != std::get<0>(other.iterators); |
| } |
| zip_first(Iters &&... ts) : iterators(std::forward<Iters>(ts)...) {} |
| }; |
| |
| template <typename... Iters> class zip_shortest : public zip_first<Iters...> { |
| template <size_t... Ns> |
| bool test(const zip_first<Iters...> &other, index_sequence<Ns...>) const { |
| return all_of(std::initializer_list<bool>{std::get<Ns>(this->iterators) != |
| std::get<Ns>(other.iterators)...}, |
| identity<bool>{}); |
| } |
| |
| public: |
| bool operator!=(const zip_first<Iters...> &other) const { |
| return test(other, index_sequence_for<Iters...>{}); |
| } |
| zip_shortest(Iters &&... ts) |
| : zip_first<Iters...>(std::forward<Iters>(ts)...) {} |
| }; |
| |
| template <template <typename...> class ItType, typename... Args> class zippy { |
| public: |
| typedef ItType<decltype(std::begin(std::declval<Args>()))...> iterator; |
| |
| private: |
| std::tuple<Args...> ts; |
| |
| template <size_t... Ns> iterator begin_impl(index_sequence<Ns...>) { |
| return iterator(std::begin(std::get<Ns>(ts))...); |
| } |
| template <size_t... Ns> iterator end_impl(index_sequence<Ns...>) { |
| return iterator(std::end(std::get<Ns>(ts))...); |
| } |
| |
| public: |
| iterator begin() { return begin_impl(index_sequence_for<Args...>{}); } |
| iterator end() { return end_impl(index_sequence_for<Args...>{}); } |
| zippy(Args &&... ts_) : ts(std::forward<Args>(ts_)...) {} |
| }; |
| } // End detail namespace |
| |
| /// zip iterator for two or more iteratable types. |
| template <typename T, typename U, typename... Args> |
| detail::zippy<detail::zip_shortest, T, U, Args...> zip(T &&t, U &&u, |
| Args &&... args) { |
| return detail::zippy<detail::zip_shortest, T, U, Args...>( |
| std::forward<T>(t), std::forward<U>(u), std::forward<Args>(args)...); |
| } |
| |
| /// zip iterator that, for the sake of efficiency, assumes the first iteratee to |
| /// be the shortest. |
| template <typename T, typename U, typename... Args> |
| detail::zippy<detail::zip_first, T, U, Args...> zip_first(T &&t, U &&u, |
| Args &&... args) { |
| return detail::zippy<detail::zip_first, T, U, Args...>( |
| std::forward<T>(t), std::forward<U>(u), std::forward<Args>(args)...); |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // Extra additions to <utility> |
| //===----------------------------------------------------------------------===// |
| |
| /// \brief Function object to check whether the first component of a std::pair |
| /// compares less than the first component of another std::pair. |
| struct less_first { |
| template <typename T> bool operator()(const T &lhs, const T &rhs) const { |
| return lhs.first < rhs.first; |
| } |
| }; |
| |
| /// \brief Function object to check whether the second component of a std::pair |
| /// compares less than the second component of another std::pair. |
| struct less_second { |
| template <typename T> bool operator()(const T &lhs, const T &rhs) const { |
| return lhs.second < rhs.second; |
| } |
| }; |
| |
| // A subset of N3658. More stuff can be added as-needed. |
| |
| /// \brief Represents a compile-time sequence of integers. |
| template <class T, T... I> struct integer_sequence { |
| typedef T value_type; |
| |
| static constexpr size_t size() { return sizeof...(I); } |
| }; |
| |
| /// \brief Alias for the common case of a sequence of size_ts. |
| template <size_t... I> |
| struct index_sequence : integer_sequence<std::size_t, I...> {}; |
| |
| template <std::size_t N, std::size_t... I> |
| struct build_index_impl : build_index_impl<N - 1, N - 1, I...> {}; |
| template <std::size_t... I> |
| struct build_index_impl<0, I...> : index_sequence<I...> {}; |
| |
| /// \brief Creates a compile-time integer sequence for a parameter pack. |
| template <class... Ts> |
| struct index_sequence_for : build_index_impl<sizeof...(Ts)> {}; |
| |
| /// Utility type to build an inheritance chain that makes it easy to rank |
| /// overload candidates. |
| template <int N> struct rank : rank<N - 1> {}; |
| template <> struct rank<0> {}; |
| |
| /// \brief traits class for checking whether type T is one of any of the given |
| /// types in the variadic list. |
| template <typename T, typename... Ts> struct is_one_of { |
| static const bool value = false; |
| }; |
| |
| template <typename T, typename U, typename... Ts> |
| struct is_one_of<T, U, Ts...> { |
| static const bool value = |
| std::is_same<T, U>::value || is_one_of<T, Ts...>::value; |
| }; |
| |
| //===----------------------------------------------------------------------===// |
| // Extra additions for arrays |
| //===----------------------------------------------------------------------===// |
| |
| /// Find the length of an array. |
| template <class T, std::size_t N> |
| constexpr inline size_t array_lengthof(T (&)[N]) { |
| return N; |
| } |
| |
| /// Adapt std::less<T> for array_pod_sort. |
| template<typename T> |
| inline int array_pod_sort_comparator(const void *P1, const void *P2) { |
| if (std::less<T>()(*reinterpret_cast<const T*>(P1), |
| *reinterpret_cast<const T*>(P2))) |
| return -1; |
| if (std::less<T>()(*reinterpret_cast<const T*>(P2), |
| *reinterpret_cast<const T*>(P1))) |
| return 1; |
| return 0; |
| } |
| |
| /// get_array_pod_sort_comparator - This is an internal helper function used to |
| /// get type deduction of T right. |
| template<typename T> |
| inline int (*get_array_pod_sort_comparator(const T &)) |
| (const void*, const void*) { |
| return array_pod_sort_comparator<T>; |
| } |
| |
| |
| /// array_pod_sort - This sorts an array with the specified start and end |
| /// extent. This is just like std::sort, except that it calls qsort instead of |
| /// using an inlined template. qsort is slightly slower than std::sort, but |
| /// most sorts are not performance critical in LLVM and std::sort has to be |
| /// template instantiated for each type, leading to significant measured code |
| /// bloat. This function should generally be used instead of std::sort where |
| /// possible. |
| /// |
| /// This function assumes that you have simple POD-like types that can be |
| /// compared with std::less and can be moved with memcpy. If this isn't true, |
| /// you should use std::sort. |
| /// |
| /// NOTE: If qsort_r were portable, we could allow a custom comparator and |
| /// default to std::less. |
| template<class IteratorTy> |
| inline void array_pod_sort(IteratorTy Start, IteratorTy End) { |
| // Don't inefficiently call qsort with one element or trigger undefined |
| // behavior with an empty sequence. |
| auto NElts = End - Start; |
| if (NElts <= 1) return; |
| qsort(&*Start, NElts, sizeof(*Start), get_array_pod_sort_comparator(*Start)); |
| } |
| |
| template <class IteratorTy> |
| inline void array_pod_sort( |
| IteratorTy Start, IteratorTy End, |
| int (*Compare)( |
| const typename std::iterator_traits<IteratorTy>::value_type *, |
| const typename std::iterator_traits<IteratorTy>::value_type *)) { |
| // Don't inefficiently call qsort with one element or trigger undefined |
| // behavior with an empty sequence. |
| auto NElts = End - Start; |
| if (NElts <= 1) return; |
| qsort(&*Start, NElts, sizeof(*Start), |
| reinterpret_cast<int (*)(const void *, const void *)>(Compare)); |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // Extra additions to <algorithm> |
| //===----------------------------------------------------------------------===// |
| |
| /// For a container of pointers, deletes the pointers and then clears the |
| /// container. |
| template<typename Container> |
| void DeleteContainerPointers(Container &C) { |
| for (auto V : C) |
| delete V; |
| C.clear(); |
| } |
| |
| /// In a container of pairs (usually a map) whose second element is a pointer, |
| /// deletes the second elements and then clears the container. |
| template<typename Container> |
| void DeleteContainerSeconds(Container &C) { |
| for (auto &V : C) |
| delete V.second; |
| C.clear(); |
| } |
| |
| /// Provide wrappers to std::all_of which take ranges instead of having to pass |
| /// begin/end explicitly. |
| template <typename R, typename UnaryPredicate> |
| bool all_of(R &&Range, UnaryPredicate P) { |
| return std::all_of(std::begin(Range), std::end(Range), P); |
| } |
| |
| /// Provide wrappers to std::any_of which take ranges instead of having to pass |
| /// begin/end explicitly. |
| template <typename R, typename UnaryPredicate> |
| bool any_of(R &&Range, UnaryPredicate P) { |
| return std::any_of(std::begin(Range), std::end(Range), P); |
| } |
| |
| /// Provide wrappers to std::none_of which take ranges instead of having to pass |
| /// begin/end explicitly. |
| template <typename R, typename UnaryPredicate> |
| bool none_of(R &&Range, UnaryPredicate P) { |
| return std::none_of(std::begin(Range), std::end(Range), P); |
| } |
| |
| /// Provide wrappers to std::find which take ranges instead of having to pass |
| /// begin/end explicitly. |
| template <typename R, typename T> |
| auto find(R &&Range, const T &Val) -> decltype(std::begin(Range)) { |
| return std::find(std::begin(Range), std::end(Range), Val); |
| } |
| |
| /// Provide wrappers to std::find_if which take ranges instead of having to pass |
| /// begin/end explicitly. |
| template <typename R, typename UnaryPredicate> |
| auto find_if(R &&Range, UnaryPredicate P) -> decltype(std::begin(Range)) { |
| return std::find_if(std::begin(Range), std::end(Range), P); |
| } |
| |
| template <typename R, typename UnaryPredicate> |
| auto find_if_not(R &&Range, UnaryPredicate P) -> decltype(std::begin(Range)) { |
| return std::find_if_not(std::begin(Range), std::end(Range), P); |
| } |
| |
| /// Provide wrappers to std::remove_if which take ranges instead of having to |
| /// pass begin/end explicitly. |
| template <typename R, typename UnaryPredicate> |
| auto remove_if(R &&Range, UnaryPredicate P) -> decltype(std::begin(Range)) { |
| return std::remove_if(std::begin(Range), std::end(Range), P); |
| } |
| |
| /// Wrapper function around std::find to detect if an element exists |
| /// in a container. |
| template <typename R, typename E> |
| bool is_contained(R &&Range, const E &Element) { |
| return std::find(std::begin(Range), std::end(Range), Element) != |
| std::end(Range); |
| } |
| |
| /// Wrapper function around std::count to count the number of times an element |
| /// \p Element occurs in the given range \p Range. |
| template <typename R, typename E> |
| auto count(R &&Range, const E &Element) -> typename std::iterator_traits< |
| decltype(std::begin(Range))>::difference_type { |
| return std::count(std::begin(Range), std::end(Range), Element); |
| } |
| |
| /// Wrapper function around std::count_if to count the number of times an |
| /// element satisfying a given predicate occurs in a range. |
| template <typename R, typename UnaryPredicate> |
| auto count_if(R &&Range, UnaryPredicate P) -> typename std::iterator_traits< |
| decltype(std::begin(Range))>::difference_type { |
| return std::count_if(std::begin(Range), std::end(Range), P); |
| } |
| |
| /// Wrapper function around std::transform to apply a function to a range and |
| /// store the result elsewhere. |
| template <typename R, typename OutputIt, typename UnaryPredicate> |
| OutputIt transform(R &&Range, OutputIt d_first, UnaryPredicate P) { |
| return std::transform(std::begin(Range), std::end(Range), d_first, P); |
| } |
| |
| //===----------------------------------------------------------------------===// |
| // Extra additions to <memory> |
| //===----------------------------------------------------------------------===// |
| |
| // Implement make_unique according to N3656. |
| |
| /// \brief Constructs a `new T()` with the given args and returns a |
| /// `unique_ptr<T>` which owns the object. |
| /// |
| /// Example: |
| /// |
| /// auto p = make_unique<int>(); |
| /// auto p = make_unique<std::tuple<int, int>>(0, 1); |
| template <class T, class... Args> |
| typename std::enable_if<!std::is_array<T>::value, std::unique_ptr<T>>::type |
| make_unique(Args &&... args) { |
| return std::unique_ptr<T>(new T(std::forward<Args>(args)...)); |
| } |
| |
| /// \brief Constructs a `new T[n]` with the given args and returns a |
| /// `unique_ptr<T[]>` which owns the object. |
| /// |
| /// \param n size of the new array. |
| /// |
| /// Example: |
| /// |
| /// auto p = make_unique<int[]>(2); // value-initializes the array with 0's. |
| template <class T> |
| typename std::enable_if<std::is_array<T>::value && std::extent<T>::value == 0, |
| std::unique_ptr<T>>::type |
| make_unique(size_t n) { |
| return std::unique_ptr<T>(new typename std::remove_extent<T>::type[n]()); |
| } |
| |
| /// This function isn't used and is only here to provide better compile errors. |
| template <class T, class... Args> |
| typename std::enable_if<std::extent<T>::value != 0>::type |
| make_unique(Args &&...) = delete; |
| |
| struct FreeDeleter { |
| void operator()(void* v) { |
| ::free(v); |
| } |
| }; |
| |
| template<typename First, typename Second> |
| struct pair_hash { |
| size_t operator()(const std::pair<First, Second> &P) const { |
| return std::hash<First>()(P.first) * 31 + std::hash<Second>()(P.second); |
| } |
| }; |
| |
| /// A functor like C++14's std::less<void> in its absence. |
| struct less { |
| template <typename A, typename B> bool operator()(A &&a, B &&b) const { |
| return std::forward<A>(a) < std::forward<B>(b); |
| } |
| }; |
| |
| /// A functor like C++14's std::equal<void> in its absence. |
| struct equal { |
| template <typename A, typename B> bool operator()(A &&a, B &&b) const { |
| return std::forward<A>(a) == std::forward<B>(b); |
| } |
| }; |
| |
| /// Binary functor that adapts to any other binary functor after dereferencing |
| /// operands. |
| template <typename T> struct deref { |
| T func; |
| // Could be further improved to cope with non-derivable functors and |
| // non-binary functors (should be a variadic template member function |
| // operator()). |
| template <typename A, typename B> |
| auto operator()(A &lhs, B &rhs) const -> decltype(func(*lhs, *rhs)) { |
| assert(lhs); |
| assert(rhs); |
| return func(*lhs, *rhs); |
| } |
| }; |
| |
| namespace detail { |
| template <typename R> class enumerator_impl { |
| public: |
| template <typename X> struct result_pair { |
| result_pair(std::size_t Index, X Value) : Index(Index), Value(Value) {} |
| |
| const std::size_t Index; |
| X Value; |
| }; |
| |
| class iterator { |
| typedef |
| typename std::iterator_traits<IterOfRange<R>>::reference iter_reference; |
| typedef result_pair<iter_reference> result_type; |
| |
| public: |
| iterator(IterOfRange<R> &&Iter, std::size_t Index) |
| : Iter(Iter), Index(Index) {} |
| |
| result_type operator*() const { return result_type(Index, *Iter); } |
| |
| iterator &operator++() { |
| ++Iter; |
| ++Index; |
| return *this; |
| } |
| |
| bool operator!=(const iterator &RHS) const { return Iter != RHS.Iter; } |
| |
| private: |
| IterOfRange<R> Iter; |
| std::size_t Index; |
| }; |
| |
| public: |
| explicit enumerator_impl(R &&Range) : Range(std::forward<R>(Range)) {} |
| |
| iterator begin() { return iterator(std::begin(Range), 0); } |
| iterator end() { return iterator(std::end(Range), std::size_t(-1)); } |
| |
| private: |
| R Range; |
| }; |
| } |
| |
| /// Given an input range, returns a new range whose values are are pair (A,B) |
| /// such that A is the 0-based index of the item in the sequence, and B is |
| /// the value from the original sequence. Example: |
| /// |
| /// std::vector<char> Items = {'A', 'B', 'C', 'D'}; |
| /// for (auto X : enumerate(Items)) { |
| /// printf("Item %d - %c\n", X.Index, X.Value); |
| /// } |
| /// |
| /// Output: |
| /// Item 0 - A |
| /// Item 1 - B |
| /// Item 2 - C |
| /// Item 3 - D |
| /// |
| template <typename R> detail::enumerator_impl<R> enumerate(R &&Range) { |
| return detail::enumerator_impl<R>(std::forward<R>(Range)); |
| } |
| |
| namespace detail { |
| template <typename F, typename Tuple, std::size_t... I> |
| auto apply_tuple_impl(F &&f, Tuple &&t, index_sequence<I...>) |
| -> decltype(std::forward<F>(f)(std::get<I>(std::forward<Tuple>(t))...)) { |
| return std::forward<F>(f)(std::get<I>(std::forward<Tuple>(t))...); |
| } |
| } |
| |
| /// Given an input tuple (a1, a2, ..., an), pass the arguments of the |
| /// tuple variadically to f as if by calling f(a1, a2, ..., an) and |
| /// return the result. |
| template <typename F, typename Tuple> |
| auto apply_tuple(F &&f, Tuple &&t) -> decltype(detail::apply_tuple_impl( |
| std::forward<F>(f), std::forward<Tuple>(t), |
| build_index_impl< |
| std::tuple_size<typename std::decay<Tuple>::type>::value>{})) { |
| using Indices = build_index_impl< |
| std::tuple_size<typename std::decay<Tuple>::type>::value>; |
| |
| return detail::apply_tuple_impl(std::forward<F>(f), std::forward<Tuple>(t), |
| Indices{}); |
| } |
| } // End llvm namespace |
| |
| #endif |