set_intersection.hpp 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286
  1. // Range v3 library
  2. //
  3. // Copyright Eric Niebler 2014-present
  4. //
  5. // Use, modification and distribution is subject to the
  6. // Boost Software License, Version 1.0. (See accompanying
  7. // file LICENSE_1_0.txt or copy at
  8. // http://www.boost.org/LICENSE_1_0.txt)
  9. //
  10. // Project home: https://github.com/ericniebler/range-v3
  11. //
  12. //===----------------------------------------------------------------------===//
  13. //
  14. // The LLVM Compiler Infrastructure
  15. //
  16. // This file is dual licensed under the MIT and the University of Illinois Open
  17. // Source Licenses. See LICENSE.TXT for details.
  18. //
  19. //===----------------------------------------------------------------------===//
  20. #include <algorithm>
  21. #include <functional>
  22. #include <range/v3/core.hpp>
  23. #include <range/v3/algorithm/fill.hpp>
  24. #include <range/v3/algorithm/set_algorithm.hpp>
  25. #include <range/v3/algorithm/lexicographical_compare.hpp>
  26. #include "../simple_test.hpp"
  27. #include "../test_utils.hpp"
  28. #include "../test_iterators.hpp"
  29. template<class Iter1, class Iter2, class OutIter>
  30. void
  31. test()
  32. {
  33. int ia[] = {1, 2, 2, 3, 3, 3, 4, 4, 4, 4};
  34. static const int sa = sizeof(ia)/sizeof(ia[0]);
  35. int ib[] = {2, 4, 4, 6};
  36. static const int sb = sizeof(ib)/sizeof(ib[0]);
  37. int ic[20];
  38. int ir[] = {2, 4, 4};
  39. static const int sr = sizeof(ir)/sizeof(ir[0]);
  40. auto set_intersection = ::make_testable_2<true, true>(ranges::set_intersection);
  41. set_intersection(Iter1(ia), Iter1(ia+sa),
  42. Iter2(ib), Iter2(ib+sb), OutIter(ic)).check([&](OutIter ce)
  43. {
  44. CHECK((base(ce) - ic) == sr);
  45. CHECK(std::lexicographical_compare(ic, base(ce), ir, ir+sr) == false);
  46. ranges::fill(ic, 0);
  47. });
  48. set_intersection(Iter1(ib), Iter1(ib+sb),
  49. Iter2(ia), Iter2(ia+sa), OutIter(ic)).check([&](OutIter ce)
  50. {
  51. CHECK((base(ce) - ic) == sr);
  52. CHECK(std::lexicographical_compare(ic, base(ce), ir, ir+sr) == false);
  53. ranges::fill(ic, 0);
  54. });
  55. set_intersection(Iter1(ia), Iter1(ia+sa),
  56. Iter2(ib), Iter2(ib+sb), OutIter(ic), std::less<int>()).check([&](OutIter ce)
  57. {
  58. CHECK((base(ce) - ic) == sr);
  59. CHECK(std::lexicographical_compare(ic, base(ce), ir, ir+sr) == false);
  60. ranges::fill(ic, 0);
  61. });
  62. set_intersection(Iter1(ib), Iter1(ib+sb),
  63. Iter2(ia), Iter2(ia+sa), OutIter(ic), std::less<int>()).check([&](OutIter ce)
  64. {
  65. CHECK((base(ce) - ic) == sr);
  66. CHECK(std::lexicographical_compare(ic, base(ce), ir, ir+sr) == false);
  67. ranges::fill(ic, 0);
  68. });
  69. }
  70. struct S
  71. {
  72. int i;
  73. };
  74. struct T
  75. {
  76. int j;
  77. };
  78. struct U
  79. {
  80. int k;
  81. U& operator=(S s) { k = s.i; return *this;}
  82. U& operator=(T t) { k = t.j; return *this;}
  83. };
  84. constexpr bool test_constexpr()
  85. {
  86. using namespace ranges;
  87. int ic[20] = {0};
  88. int ia[] = {1, 2, 2, 3, 3, 3, 4, 4, 4, 4};
  89. int ib[4] = {2, 4, 4, 6};
  90. int ir[3] = {2, 4, 4};
  91. constexpr auto sr = size(ir);
  92. int * res = set_intersection(ia, ib, ic, less{});
  93. STATIC_CHECK_RETURN((res - ic) == sr);
  94. STATIC_CHECK_RETURN(lexicographical_compare(ic, res, ir, ir + sr, less{}) == 0);
  95. return true;
  96. }
  97. int main()
  98. {
  99. #ifdef SET_INTERSECTION_1
  100. test<InputIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  101. test<InputIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  102. test<InputIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  103. test<InputIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  104. test<InputIterator<const int*>, InputIterator<const int*>, int*>();
  105. test<InputIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  106. test<InputIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  107. test<InputIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  108. test<InputIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  109. test<InputIterator<const int*>, ForwardIterator<const int*>, int*>();
  110. test<InputIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  111. test<InputIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  112. test<InputIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  113. test<InputIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  114. test<InputIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  115. test<InputIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  116. test<InputIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  117. test<InputIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  118. test<InputIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  119. test<InputIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  120. test<InputIterator<const int*>, const int*, OutputIterator<int*> >();
  121. test<InputIterator<const int*>, const int*, ForwardIterator<int*> >();
  122. test<InputIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  123. test<InputIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  124. test<InputIterator<const int*>, const int*, int*>();
  125. #endif
  126. #ifdef SET_INTERSECTION_2
  127. test<ForwardIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  128. test<ForwardIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  129. test<ForwardIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  130. test<ForwardIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  131. test<ForwardIterator<const int*>, InputIterator<const int*>, int*>();
  132. test<ForwardIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  133. test<ForwardIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  134. test<ForwardIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  135. test<ForwardIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  136. test<ForwardIterator<const int*>, ForwardIterator<const int*>, int*>();
  137. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  138. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  139. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  140. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  141. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  142. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  143. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  144. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  145. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  146. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  147. test<ForwardIterator<const int*>, const int*, OutputIterator<int*> >();
  148. test<ForwardIterator<const int*>, const int*, ForwardIterator<int*> >();
  149. test<ForwardIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  150. test<ForwardIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  151. test<ForwardIterator<const int*>, const int*, int*>();
  152. #endif
  153. #ifdef SET_INTERSECTION_3
  154. test<BidirectionalIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  155. test<BidirectionalIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  156. test<BidirectionalIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  157. test<BidirectionalIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  158. test<BidirectionalIterator<const int*>, InputIterator<const int*>, int*>();
  159. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  160. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  161. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  162. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  163. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, int*>();
  164. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  165. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  166. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  167. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  168. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  169. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  170. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  171. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  172. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  173. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  174. test<BidirectionalIterator<const int*>, const int*, OutputIterator<int*> >();
  175. test<BidirectionalIterator<const int*>, const int*, ForwardIterator<int*> >();
  176. test<BidirectionalIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  177. test<BidirectionalIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  178. test<BidirectionalIterator<const int*>, const int*, int*>();
  179. #endif
  180. #ifdef SET_INTERSECTION_4
  181. test<RandomAccessIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  182. test<RandomAccessIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  183. test<RandomAccessIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  184. test<RandomAccessIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  185. test<RandomAccessIterator<const int*>, InputIterator<const int*>, int*>();
  186. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  187. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  188. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  189. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  190. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, int*>();
  191. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  192. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  193. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  194. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  195. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  196. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  197. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  198. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  199. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  200. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  201. test<RandomAccessIterator<const int*>, const int*, OutputIterator<int*> >();
  202. test<RandomAccessIterator<const int*>, const int*, ForwardIterator<int*> >();
  203. test<RandomAccessIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  204. test<RandomAccessIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  205. test<RandomAccessIterator<const int*>, const int*, int*>();
  206. #endif
  207. #ifdef SET_INTERSECTION_5
  208. test<const int*, InputIterator<const int*>, OutputIterator<int*> >();
  209. test<const int*, InputIterator<const int*>, BidirectionalIterator<int*> >();
  210. test<const int*, InputIterator<const int*>, BidirectionalIterator<int*> >();
  211. test<const int*, InputIterator<const int*>, RandomAccessIterator<int*> >();
  212. test<const int*, InputIterator<const int*>, int*>();
  213. test<const int*, ForwardIterator<const int*>, OutputIterator<int*> >();
  214. test<const int*, ForwardIterator<const int*>, ForwardIterator<int*> >();
  215. test<const int*, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  216. test<const int*, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  217. test<const int*, ForwardIterator<const int*>, int*>();
  218. test<const int*, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  219. test<const int*, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  220. test<const int*, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  221. test<const int*, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  222. test<const int*, BidirectionalIterator<const int*>, int*>();
  223. test<const int*, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  224. test<const int*, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  225. test<const int*, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  226. test<const int*, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  227. test<const int*, RandomAccessIterator<const int*>, int*>();
  228. test<const int*, const int*, OutputIterator<int*> >();
  229. test<const int*, const int*, ForwardIterator<int*> >();
  230. test<const int*, const int*, BidirectionalIterator<int*> >();
  231. test<const int*, const int*, RandomAccessIterator<int*> >();
  232. test<const int*, const int*, int*>();
  233. #endif
  234. #ifdef SET_INTERSECTION_6
  235. // Test projections
  236. {
  237. S ia[] = {S{1}, S{2}, S{2}, S{3}, S{3}, S{3}, S{4}, S{4}, S{4}, S{4}};
  238. T ib[] = {T{2}, T{4}, T{4}, T{6}};
  239. U ic[20];
  240. int ir[] = {2, 4, 4};
  241. static const int sr = sizeof(ir)/sizeof(ir[0]);
  242. U * res = ranges::set_intersection(ranges::views::all(ia), ranges::views::all(ib), ic, std::less<int>(), &S::i, &T::j);
  243. CHECK((res - ic) == sr);
  244. CHECK(ranges::lexicographical_compare(ic, res, ir, ir+sr, std::less<int>(), &U::k) == false);
  245. }
  246. {
  247. STATIC_CHECK(test_constexpr());
  248. }
  249. #endif
  250. return ::test_result();
  251. }