set_difference.hpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375
  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 <vector>
  23. #include <range/v3/core.hpp>
  24. #include <range/v3/algorithm/fill.hpp>
  25. #include <range/v3/algorithm/set_algorithm.hpp>
  26. #include <range/v3/algorithm/lexicographical_compare.hpp>
  27. #include "../simple_test.hpp"
  28. #include "../test_utils.hpp"
  29. #include "../test_iterators.hpp"
  30. template<class Iter1, class Iter2, class OutIter>
  31. void
  32. test_iter()
  33. {
  34. int ia[] = {1, 2, 2, 3, 3, 3, 4, 4, 4, 4};
  35. static const int sa = sizeof(ia)/sizeof(ia[0]);
  36. int ib[] = {2, 4, 4, 6};
  37. static const int sb = sizeof(ib)/sizeof(ib[0]);
  38. int ic[20];
  39. int ir[] = {1, 2, 3, 3, 3, 4, 4};
  40. static const int sr = sizeof(ir)/sizeof(ir[0]);
  41. auto set_difference = ::make_testable_2<false, true>(ranges::set_difference);
  42. set_difference(Iter1(ia), Iter1(ia+sa), Iter2(ib), Iter2(ib+sb), OutIter(ic)).
  43. check([&](ranges::set_difference_result<Iter1, OutIter> res)
  44. {
  45. CHECK((base(res.in1) - ia) == sa);
  46. CHECK((base(res.out) - ic) == sr);
  47. CHECK(std::lexicographical_compare(ic, base(res.out), ir, ir+sr) == false);
  48. ranges::fill(ic, 0);
  49. }
  50. );
  51. int irr[] = {6};
  52. static const int srr = sizeof(irr)/sizeof(irr[0]);
  53. set_difference(Iter1(ib), Iter1(ib+sb), Iter2(ia), Iter2(ia+sa), OutIter(ic)).
  54. check([&](ranges::set_difference_result<Iter1, OutIter> res)
  55. {
  56. CHECK((base(res.in1) - ib) == sb);
  57. CHECK((base(res.out) - ic) == srr);
  58. CHECK(std::lexicographical_compare(ic, base(res.out), irr, irr+srr) == false);
  59. ranges::fill(ic, 0);
  60. }
  61. );
  62. }
  63. template<class Iter1, class Iter2, class OutIter>
  64. void
  65. test_comp()
  66. {
  67. int ia[] = {1, 2, 2, 3, 3, 3, 4, 4, 4, 4};
  68. static const int sa = sizeof(ia)/sizeof(ia[0]);
  69. int ib[] = {2, 4, 4, 6};
  70. static const int sb = sizeof(ib)/sizeof(ib[0]);
  71. int ic[20];
  72. int ir[] = {1, 2, 3, 3, 3, 4, 4};
  73. static const int sr = sizeof(ir)/sizeof(ir[0]);
  74. auto set_difference = ::make_testable_2<false, true>(ranges::set_difference);
  75. set_difference(Iter1(ia), Iter1(ia+sa), Iter2(ib), Iter2(ib+sb), OutIter(ic), std::less<int>()).
  76. check([&](ranges::set_difference_result<Iter1, OutIter> res)
  77. {
  78. CHECK((base(res.in1) - ia) == sa);
  79. CHECK((base(res.out) - ic) == sr);
  80. CHECK(std::lexicographical_compare(ic, base(res.out), ir, ir+sr) == false);
  81. ranges::fill(ic, 0);
  82. }
  83. );
  84. int irr[] = {6};
  85. static const int srr = sizeof(irr)/sizeof(irr[0]);
  86. set_difference(Iter1(ib), Iter1(ib+sb), Iter2(ia), Iter2(ia+sa), OutIter(ic), std::less<int>()).
  87. check([&](ranges::set_difference_result<Iter1, OutIter> res)
  88. {
  89. CHECK((base(res.in1) - ib) == sb);
  90. CHECK((base(res.out) - ic) == srr);
  91. CHECK(std::lexicographical_compare(ic, base(res.out), irr, irr+srr) == false);
  92. ranges::fill(ic, 0);
  93. }
  94. );
  95. }
  96. template<class Iter1, class Iter2, class OutIter>
  97. void test()
  98. {
  99. test_iter<Iter1, Iter2, OutIter>();
  100. test_comp<Iter1, Iter2, OutIter>();
  101. }
  102. struct S
  103. {
  104. int i;
  105. };
  106. struct T
  107. {
  108. int j;
  109. };
  110. struct U
  111. {
  112. int k;
  113. U& operator=(S s) { k = s.i; return *this;}
  114. U& operator=(T t) { k = t.j; return *this;}
  115. };
  116. constexpr bool test_constexpr()
  117. {
  118. using namespace ranges;
  119. using IL = std::initializer_list<int>;
  120. int ia[] = {1, 2, 2, 3, 3, 3, 4, 4, 4, 4};
  121. const int sa = sizeof(ia) / sizeof(ia[0]);
  122. int ib[] = {2, 4, 4, 6};
  123. const int sb = sizeof(ib) / sizeof(ib[0]);
  124. int ic[20] = {0};
  125. int ir[] = {1, 2, 3, 3, 3, 4, 4};
  126. const int sr = sizeof(ir) / sizeof(ir[0]);
  127. const auto res = set_difference(ia, IL{2, 4, 4, 6}, ic, less{});
  128. STATIC_CHECK_RETURN((res.in1 - ia) == sa);
  129. STATIC_CHECK_RETURN((res.out - ic) == sr);
  130. STATIC_CHECK_RETURN(lexicographical_compare(ic, res.out, ir, ir + sr, less{}) == 0);
  131. fill(ic, 0);
  132. int irr[] = {6};
  133. const int srr = sizeof(irr) / sizeof(irr[0]);
  134. const auto res2 = set_difference(ib, IL{1, 2, 2, 3, 3, 3, 4, 4, 4, 4}, ic, less{});
  135. STATIC_CHECK_RETURN((res2.in1 - ib) == sb);
  136. STATIC_CHECK_RETURN((res2.out - ic) == srr);
  137. STATIC_CHECK_RETURN(lexicographical_compare(ic, res2.out, irr, irr + srr, less{}) == 0);
  138. return true;
  139. }
  140. int main()
  141. {
  142. #ifdef SET_DIFFERENCE_1
  143. test<InputIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  144. test<InputIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  145. test<InputIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  146. test<InputIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  147. test<InputIterator<const int*>, InputIterator<const int*>, int*>();
  148. test<InputIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  149. test<InputIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  150. test<InputIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  151. test<InputIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  152. test<InputIterator<const int*>, ForwardIterator<const int*>, int*>();
  153. test<InputIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  154. test<InputIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  155. test<InputIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  156. test<InputIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  157. test<InputIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  158. test<InputIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  159. test<InputIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  160. test<InputIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  161. test<InputIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  162. test<InputIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  163. test<InputIterator<const int*>, const int*, OutputIterator<int*> >();
  164. test<InputIterator<const int*>, const int*, ForwardIterator<int*> >();
  165. test<InputIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  166. test<InputIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  167. test<InputIterator<const int*>, const int*, int*>();
  168. #endif
  169. #ifdef SET_DIFFERENCE_2
  170. test<ForwardIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  171. test<ForwardIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  172. test<ForwardIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  173. test<ForwardIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  174. test<ForwardIterator<const int*>, InputIterator<const int*>, int*>();
  175. test<ForwardIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  176. test<ForwardIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  177. test<ForwardIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  178. test<ForwardIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  179. test<ForwardIterator<const int*>, ForwardIterator<const int*>, int*>();
  180. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  181. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  182. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  183. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  184. test<ForwardIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  185. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  186. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  187. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  188. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  189. test<ForwardIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  190. test<ForwardIterator<const int*>, const int*, OutputIterator<int*> >();
  191. test<ForwardIterator<const int*>, const int*, ForwardIterator<int*> >();
  192. test<ForwardIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  193. test<ForwardIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  194. test<ForwardIterator<const int*>, const int*, int*>();
  195. #endif
  196. #ifdef SET_DIFFERENCE_3
  197. test<BidirectionalIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  198. test<BidirectionalIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  199. test<BidirectionalIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  200. test<BidirectionalIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  201. test<BidirectionalIterator<const int*>, InputIterator<const int*>, int*>();
  202. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  203. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  204. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  205. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  206. test<BidirectionalIterator<const int*>, ForwardIterator<const int*>, int*>();
  207. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  208. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  209. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  210. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  211. test<BidirectionalIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  212. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  213. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  214. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  215. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  216. test<BidirectionalIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  217. test<BidirectionalIterator<const int*>, const int*, OutputIterator<int*> >();
  218. test<BidirectionalIterator<const int*>, const int*, ForwardIterator<int*> >();
  219. test<BidirectionalIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  220. test<BidirectionalIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  221. test<BidirectionalIterator<const int*>, const int*, int*>();
  222. #endif
  223. #ifdef SET_DIFFERENCE_4
  224. test<RandomAccessIterator<const int*>, InputIterator<const int*>, OutputIterator<int*> >();
  225. test<RandomAccessIterator<const int*>, InputIterator<const int*>, ForwardIterator<int*> >();
  226. test<RandomAccessIterator<const int*>, InputIterator<const int*>, BidirectionalIterator<int*> >();
  227. test<RandomAccessIterator<const int*>, InputIterator<const int*>, RandomAccessIterator<int*> >();
  228. test<RandomAccessIterator<const int*>, InputIterator<const int*>, int*>();
  229. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, OutputIterator<int*> >();
  230. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, ForwardIterator<int*> >();
  231. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  232. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  233. test<RandomAccessIterator<const int*>, ForwardIterator<const int*>, int*>();
  234. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  235. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  236. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  237. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  238. test<RandomAccessIterator<const int*>, BidirectionalIterator<const int*>, int*>();
  239. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  240. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  241. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  242. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  243. test<RandomAccessIterator<const int*>, RandomAccessIterator<const int*>, int*>();
  244. test<RandomAccessIterator<const int*>, const int*, OutputIterator<int*> >();
  245. test<RandomAccessIterator<const int*>, const int*, ForwardIterator<int*> >();
  246. test<RandomAccessIterator<const int*>, const int*, BidirectionalIterator<int*> >();
  247. test<RandomAccessIterator<const int*>, const int*, RandomAccessIterator<int*> >();
  248. test<RandomAccessIterator<const int*>, const int*, int*>();
  249. #endif
  250. #ifdef SET_DIFFERENCE_5
  251. test<const int*, InputIterator<const int*>, OutputIterator<int*> >();
  252. test<const int*, InputIterator<const int*>, BidirectionalIterator<int*> >(); test<const int*, InputIterator<const int*>, BidirectionalIterator<int*> >();
  253. test<const int*, InputIterator<const int*>, RandomAccessIterator<int*> >();
  254. test<const int*, InputIterator<const int*>, int*>();
  255. test<const int*, ForwardIterator<const int*>, OutputIterator<int*> >();
  256. test<const int*, ForwardIterator<const int*>, ForwardIterator<int*> >();
  257. test<const int*, ForwardIterator<const int*>, BidirectionalIterator<int*> >();
  258. test<const int*, ForwardIterator<const int*>, RandomAccessIterator<int*> >();
  259. test<const int*, ForwardIterator<const int*>, int*>();
  260. test<const int*, BidirectionalIterator<const int*>, OutputIterator<int*> >();
  261. test<const int*, BidirectionalIterator<const int*>, ForwardIterator<int*> >();
  262. test<const int*, BidirectionalIterator<const int*>, BidirectionalIterator<int*> >();
  263. test<const int*, BidirectionalIterator<const int*>, RandomAccessIterator<int*> >();
  264. test<const int*, BidirectionalIterator<const int*>, int*>();
  265. test<const int*, RandomAccessIterator<const int*>, OutputIterator<int*> >();
  266. test<const int*, RandomAccessIterator<const int*>, ForwardIterator<int*> >();
  267. test<const int*, RandomAccessIterator<const int*>, BidirectionalIterator<int*> >();
  268. test<const int*, RandomAccessIterator<const int*>, RandomAccessIterator<int*> >();
  269. test<const int*, RandomAccessIterator<const int*>, int*>();
  270. test<const int*, const int*, OutputIterator<int*> >();
  271. test<const int*, const int*, ForwardIterator<int*> >();
  272. test<const int*, const int*, BidirectionalIterator<int*> >();
  273. test<const int*, const int*, RandomAccessIterator<int*> >();
  274. test<const int*, const int*, int*>();
  275. #endif
  276. #ifdef SET_DIFFERENCE_6
  277. // Test projections
  278. {
  279. S ia[] = {S{1}, S{2}, S{2}, S{3}, S{3}, S{3}, S{4}, S{4}, S{4}, S{4}};
  280. static const int sa = sizeof(ia)/sizeof(ia[0]);
  281. T ib[] = {T{2}, T{4}, T{4}, T{6}};
  282. static const int sb = sizeof(ib)/sizeof(ib[0]);
  283. U ic[20];
  284. int ir[] = {1, 2, 3, 3, 3, 4, 4};
  285. static const int sr = sizeof(ir)/sizeof(ir[0]);
  286. ranges::set_difference_result<S *, U *> res = ranges::set_difference(ia, ib, ic, std::less<int>(), &S::i, &T::j);
  287. CHECK((res.in1 - ia) == sa);
  288. CHECK((res.out - ic) == sr);
  289. CHECK(ranges::lexicographical_compare(ic, res.out, ir, ir+sr, std::less<int>(), &U::k) == false);
  290. ranges::fill(ic, U{0});
  291. int irr[] = {6};
  292. static const int srr = sizeof(irr)/sizeof(irr[0]);
  293. ranges::set_difference_result<T *, U *> res2 = ranges::set_difference(ib, ia, ic, std::less<int>(), &T::j, &S::i);
  294. CHECK((res2.in1 - ib) == sb);
  295. CHECK((res2.out - ic) == srr);
  296. CHECK(ranges::lexicographical_compare(ic, res2.out, ir, irr+srr, std::less<int>(), &U::k) == false);
  297. }
  298. // Test rvalue ranges
  299. {
  300. S ia[] = {S{1}, S{2}, S{2}, S{3}, S{3}, S{3}, S{4}, S{4}, S{4}, S{4}};
  301. T ib[] = {T{2}, T{4}, T{4}, T{6}};
  302. static const int sb = sizeof(ib)/sizeof(ib[0]);
  303. U ic[20];
  304. int ir[] = {1, 2, 3, 3, 3, 4, 4};
  305. static const int sr = sizeof(ir)/sizeof(ir[0]);
  306. auto res = ranges::set_difference(std::move(ia), ranges::views::all(ib), ic, std::less<int>(), &S::i, &T::j);
  307. #ifndef RANGES_WORKAROUND_MSVC_573728
  308. CHECK(::is_dangling(res.in1));
  309. #endif // RANGES_WORKAROUND_MSVC_573728
  310. CHECK((res.out - ic) == sr);
  311. CHECK(ranges::lexicographical_compare(ic, res.out, ir, ir+sr, std::less<int>(), &U::k) == false);
  312. ranges::fill(ic, U{0});
  313. int irr[] = {6};
  314. static const int srr = sizeof(irr)/sizeof(irr[0]);
  315. auto res2 = ranges::set_difference(ranges::views::all(ib), ranges::views::all(ia), ic, std::less<int>(), &T::j, &S::i);
  316. CHECK((res2.in1 - ib) == sb);
  317. CHECK((res2.out - ic) == srr);
  318. CHECK(ranges::lexicographical_compare(ic, res2.out, ir, irr+srr, std::less<int>(), &U::k) == false);
  319. ranges::fill(ic, U{0});
  320. std::vector<S> vec{S{1}, S{2}, S{2}, S{3}, S{3}, S{3}, S{4}, S{4}, S{4}, S{4}};
  321. auto res3 = ranges::set_difference(std::move(vec), ranges::views::all(ib), ic, std::less<int>(), &S::i, &T::j);
  322. CHECK(::is_dangling(res3.in1));
  323. CHECK((res3.out - ic) == sr);
  324. CHECK(ranges::lexicographical_compare(ic, res3.out, ir, ir+sr, std::less<int>(), &U::k) == false);
  325. }
  326. {
  327. STATIC_CHECK(test_constexpr());
  328. }
  329. #endif
  330. return ::test_result();
  331. }