cmAlgorithms.h 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149
  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 T>
  52. class BT;
  53. using cmBTStringRange = cmRange<std::vector<BT<std::string>>::const_iterator>;
  54. template <typename Range>
  55. typename Range::const_iterator cmRemoveN(Range& r, size_t n)
  56. {
  57. return ContainerAlgorithms::RemoveN(r.begin(), r.end(), n);
  58. }
  59. template <typename Range, typename InputRange>
  60. typename Range::const_iterator cmRemoveIndices(Range& r, InputRange const& rem)
  61. {
  62. typename InputRange::const_iterator remIt = rem.begin();
  63. typename InputRange::const_iterator remEnd = rem.end();
  64. const auto rangeEnd = r.end();
  65. if (remIt == remEnd) {
  66. return rangeEnd;
  67. }
  68. auto writer = r.begin();
  69. std::advance(writer, *remIt);
  70. auto pivot = writer;
  71. typename InputRange::value_type prevRem = *remIt;
  72. ++remIt;
  73. size_t count = 1;
  74. for (; writer != rangeEnd && remIt != remEnd; ++count, ++remIt) {
  75. std::advance(pivot, *remIt - prevRem);
  76. prevRem = *remIt;
  77. writer = ContainerAlgorithms::RemoveN(writer, pivot, count);
  78. }
  79. return ContainerAlgorithms::RemoveN(writer, rangeEnd, count);
  80. }
  81. template <typename Range, typename MatchRange>
  82. typename Range::const_iterator cmRemoveMatching(Range& r, MatchRange const& m)
  83. {
  84. return std::remove_if(r.begin(), r.end(),
  85. ContainerAlgorithms::BinarySearcher<MatchRange>(m));
  86. }
  87. template <typename ForwardIterator>
  88. ForwardIterator cmRemoveDuplicates(ForwardIterator first, ForwardIterator last)
  89. {
  90. using Value = typename std::iterator_traits<ForwardIterator>::value_type;
  91. struct Hash
  92. {
  93. std::size_t operator()(ForwardIterator it) const
  94. {
  95. return std::hash<Value>{}(*it);
  96. }
  97. };
  98. struct Equal
  99. {
  100. bool operator()(ForwardIterator it1, ForwardIterator it2) const
  101. {
  102. return *it1 == *it2;
  103. }
  104. };
  105. std::unordered_set<ForwardIterator, Hash, Equal> uniq;
  106. ForwardIterator result = first;
  107. while (first != last) {
  108. if (!cm::contains(uniq, first)) {
  109. if (result != first) {
  110. *result = std::move(*first);
  111. }
  112. uniq.insert(result);
  113. ++result;
  114. }
  115. ++first;
  116. }
  117. return result;
  118. }
  119. template <typename Range>
  120. typename Range::const_iterator cmRemoveDuplicates(Range& r)
  121. {
  122. return cmRemoveDuplicates(r.begin(), r.end());
  123. }
  124. template <typename Range, typename T>
  125. typename Range::const_iterator cmFindNot(Range const& r, T const& t)
  126. {
  127. return std::find_if(r.begin(), r.end(), [&t](T const& i) { return i != t; });
  128. }