cmAlgorithms.h 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145
  1. /* Distributed under the OSI-approved BSD 3-Clause License. See accompanying
  2. file Copyright.txt or https://cmake.org/licensing for details. */
  3. #pragma once
  4. #include "cmConfigure.h" // IWYU pragma: keep
  5. #include <algorithm>
  6. #include <functional>
  7. #include <iterator>
  8. #include <memory>
  9. #include <unordered_set>
  10. #include <utility>
  11. #include <vector>
  12. #include <cmext/algorithm>
  13. #include "cmRange.h"
  14. template <typename FwdIt>
  15. FwdIt cmRotate(FwdIt first, FwdIt middle, FwdIt last)
  16. {
  17. const typename std::iterator_traits<FwdIt>::difference_type dist =
  18. std::distance(middle, last);
  19. std::rotate(first, middle, last);
  20. std::advance(first, dist);
  21. return first;
  22. }
  23. namespace ContainerAlgorithms {
  24. template <typename FwdIt>
  25. FwdIt RemoveN(FwdIt i1, FwdIt i2, size_t n)
  26. {
  27. FwdIt m = i1;
  28. std::advance(m, n);
  29. return cmRotate(i1, m, i2);
  30. }
  31. template <typename Range>
  32. struct BinarySearcher
  33. {
  34. using argument_type = typename Range::value_type;
  35. BinarySearcher(Range const& r)
  36. : m_range(r)
  37. {
  38. }
  39. bool operator()(argument_type const& item) const
  40. {
  41. return std::binary_search(this->m_range.begin(), this->m_range.end(),
  42. item);
  43. }
  44. private:
  45. Range const& m_range;
  46. };
  47. }
  48. class cmListFileBacktrace;
  49. using cmBacktraceRange =
  50. cmRange<std::vector<cmListFileBacktrace>::const_iterator>;
  51. template <typename Range>
  52. typename Range::const_iterator cmRemoveN(Range& r, size_t n)
  53. {
  54. return ContainerAlgorithms::RemoveN(r.begin(), r.end(), n);
  55. }
  56. template <typename Range, typename InputRange>
  57. typename Range::const_iterator cmRemoveIndices(Range& r, InputRange const& rem)
  58. {
  59. typename InputRange::const_iterator remIt = rem.begin();
  60. typename InputRange::const_iterator remEnd = rem.end();
  61. const auto rangeEnd = r.end();
  62. if (remIt == remEnd) {
  63. return rangeEnd;
  64. }
  65. auto writer = r.begin();
  66. std::advance(writer, *remIt);
  67. auto pivot = writer;
  68. typename InputRange::value_type prevRem = *remIt;
  69. ++remIt;
  70. size_t count = 1;
  71. for (; writer != rangeEnd && remIt != remEnd; ++count, ++remIt) {
  72. std::advance(pivot, *remIt - prevRem);
  73. prevRem = *remIt;
  74. writer = ContainerAlgorithms::RemoveN(writer, pivot, count);
  75. }
  76. return ContainerAlgorithms::RemoveN(writer, rangeEnd, count);
  77. }
  78. template <typename Range, typename MatchRange>
  79. typename Range::const_iterator cmRemoveMatching(Range& r, MatchRange const& m)
  80. {
  81. return std::remove_if(r.begin(), r.end(),
  82. ContainerAlgorithms::BinarySearcher<MatchRange>(m));
  83. }
  84. template <typename ForwardIterator>
  85. ForwardIterator cmRemoveDuplicates(ForwardIterator first, ForwardIterator last)
  86. {
  87. using Value = typename std::iterator_traits<ForwardIterator>::value_type;
  88. struct Hash
  89. {
  90. std::size_t operator()(ForwardIterator it) const
  91. {
  92. return std::hash<Value>{}(*it);
  93. }
  94. };
  95. struct Equal
  96. {
  97. bool operator()(ForwardIterator it1, ForwardIterator it2) const
  98. {
  99. return *it1 == *it2;
  100. }
  101. };
  102. std::unordered_set<ForwardIterator, Hash, Equal> uniq;
  103. ForwardIterator result = first;
  104. while (first != last) {
  105. if (!cm::contains(uniq, first)) {
  106. if (result != first) {
  107. *result = std::move(*first);
  108. }
  109. uniq.insert(result);
  110. ++result;
  111. }
  112. ++first;
  113. }
  114. return result;
  115. }
  116. template <typename Range>
  117. typename Range::const_iterator cmRemoveDuplicates(Range& r)
  118. {
  119. return cmRemoveDuplicates(r.begin(), r.end());
  120. }
  121. template <typename Range, typename T>
  122. typename Range::const_iterator cmFindNot(Range const& r, T const& t)
  123. {
  124. return std::find_if(r.begin(), r.end(), [&t](T const& i) { return i != t; });
  125. }