13b62047bSLouis Dionne // -*- C++ -*-
23b62047bSLouis Dionne //===-- swap_ranges.pass.cpp ----------------------------------------------===//
33b62047bSLouis Dionne //
43b62047bSLouis Dionne // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
53b62047bSLouis Dionne // See https://llvm.org/LICENSE.txt for license information.
63b62047bSLouis Dionne // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
73b62047bSLouis Dionne //
83b62047bSLouis Dionne //===----------------------------------------------------------------------===//
93b62047bSLouis Dionne
10*b5e896c0SLouis Dionne // UNSUPPORTED: c++03, c++11, c++14
116ab35c9dSLouis Dionne
123b62047bSLouis Dionne #include "support/pstl_test_config.h"
133b62047bSLouis Dionne
143b62047bSLouis Dionne #include <iterator>
153b62047bSLouis Dionne #include <execution>
163b62047bSLouis Dionne #include <algorithm>
173b62047bSLouis Dionne
183b62047bSLouis Dionne #include "support/utils.h"
193b62047bSLouis Dionne
203b62047bSLouis Dionne using namespace TestUtils;
213b62047bSLouis Dionne
223b62047bSLouis Dionne template <typename T>
233b62047bSLouis Dionne struct wrapper
243b62047bSLouis Dionne {
253b62047bSLouis Dionne T t;
263b62047bSLouis Dionne std::size_t number_of_swaps = 0;
wrapperwrapper273b62047bSLouis Dionne wrapper() {}
wrapperwrapper283b62047bSLouis Dionne explicit wrapper(T t_) : t(t_) {}
293b62047bSLouis Dionne template <typename U>
303b62047bSLouis Dionne void
operator =wrapper313b62047bSLouis Dionne operator=(const U& b)
323b62047bSLouis Dionne {
333b62047bSLouis Dionne t = b;
343b62047bSLouis Dionne }
353b62047bSLouis Dionne bool
operator ==wrapper363b62047bSLouis Dionne operator==(const wrapper<T>& a) const
373b62047bSLouis Dionne {
383b62047bSLouis Dionne return t == a.t;
393b62047bSLouis Dionne }
403b62047bSLouis Dionne };
413b62047bSLouis Dionne
423b62047bSLouis Dionne template <typename T>
433b62047bSLouis Dionne void
swap(wrapper<T> & a,wrapper<T> & b)443b62047bSLouis Dionne swap(wrapper<T>& a, wrapper<T>& b)
453b62047bSLouis Dionne {
463b62047bSLouis Dionne std::swap(a.t, b.t);
473b62047bSLouis Dionne a.number_of_swaps++;
483b62047bSLouis Dionne b.number_of_swaps++;
493b62047bSLouis Dionne }
503b62047bSLouis Dionne
513b62047bSLouis Dionne template <typename T>
523b62047bSLouis Dionne struct check_swap
533b62047bSLouis Dionne {
543b62047bSLouis Dionne bool
operator ()check_swap553491119fSLouis Dionne operator()(T&)
563b62047bSLouis Dionne {
573b62047bSLouis Dionne return true;
583b62047bSLouis Dionne }
593b62047bSLouis Dionne };
603b62047bSLouis Dionne
613b62047bSLouis Dionne template <typename T>
623b62047bSLouis Dionne struct check_swap<wrapper<T>>
633b62047bSLouis Dionne {
643b62047bSLouis Dionne bool
operator ()check_swap653b62047bSLouis Dionne operator()(wrapper<T>& a)
663b62047bSLouis Dionne {
673b62047bSLouis Dionne bool temp = (a.number_of_swaps == 1);
683b62047bSLouis Dionne a.number_of_swaps = 0;
693b62047bSLouis Dionne return temp;
703b62047bSLouis Dionne }
713b62047bSLouis Dionne };
723b62047bSLouis Dionne
733b62047bSLouis Dionne struct test_one_policy
743b62047bSLouis Dionne {
753b62047bSLouis Dionne template <typename ExecutionPolicy, typename Iterator1, typename Iterator2>
763b62047bSLouis Dionne void
operator ()test_one_policy773b62047bSLouis Dionne operator()(ExecutionPolicy&& exec, Iterator1 data_b, Iterator1 data_e, Iterator2 actual_b, Iterator2 actual_e)
783b62047bSLouis Dionne {
793b62047bSLouis Dionne using namespace std;
803b62047bSLouis Dionne using T_ref = typename iterator_traits<Iterator1>::reference;
813b62047bSLouis Dionne using T = typename iterator_traits<Iterator1>::value_type;
823b62047bSLouis Dionne
833b62047bSLouis Dionne iota(data_b, data_e, 0);
843b62047bSLouis Dionne iota(actual_b, actual_e, std::distance(data_b, data_e));
853b62047bSLouis Dionne
863b62047bSLouis Dionne Iterator2 actual_return = swap_ranges(exec, data_b, data_e, actual_b);
873b62047bSLouis Dionne bool check_return = (actual_return == actual_e);
883b62047bSLouis Dionne EXPECT_TRUE(check_return, "wrong result of swap_ranges");
893b62047bSLouis Dionne if (check_return)
903b62047bSLouis Dionne {
913b62047bSLouis Dionne std::size_t i = 0;
923b62047bSLouis Dionne bool check = all_of(actual_b, actual_e, [&i](T_ref a) { return a == T(i++); }) &&
933b62047bSLouis Dionne all_of(data_b, data_e, [&i](T_ref a) { return a == T(i++); });
943b62047bSLouis Dionne
953b62047bSLouis Dionne EXPECT_TRUE(check, "wrong effect of swap_ranges");
963b62047bSLouis Dionne
973b62047bSLouis Dionne if (check)
983b62047bSLouis Dionne {
993b62047bSLouis Dionne bool swap_check =
1003b62047bSLouis Dionne all_of(data_b, data_e, check_swap<T>()) && all_of(actual_b, actual_e, check_swap<T>());
1013b62047bSLouis Dionne EXPECT_TRUE(swap_check, "wrong effect of swap_ranges swap check");
1023b62047bSLouis Dionne }
1033b62047bSLouis Dionne }
1043b62047bSLouis Dionne }
1053b62047bSLouis Dionne };
1063b62047bSLouis Dionne
1073b62047bSLouis Dionne template <typename T>
1083b62047bSLouis Dionne void
test()1093b62047bSLouis Dionne test()
1103b62047bSLouis Dionne {
1113b62047bSLouis Dionne const std::size_t max_len = 100000;
1123b62047bSLouis Dionne
1133b62047bSLouis Dionne Sequence<T> data(max_len);
1143b62047bSLouis Dionne Sequence<T> actual(max_len);
1153b62047bSLouis Dionne
1163b62047bSLouis Dionne for (std::size_t len = 0; len < max_len; len = len <= 16 ? len + 1 : std::size_t(3.1415 * len))
1173b62047bSLouis Dionne {
1183b62047bSLouis Dionne invoke_on_all_policies(test_one_policy(), data.begin(), data.begin() + len, actual.begin(),
1193b62047bSLouis Dionne actual.begin() + len);
1203b62047bSLouis Dionne }
1213b62047bSLouis Dionne }
1223b62047bSLouis Dionne
123249c1c74SLouis Dionne int
main()1243b62047bSLouis Dionne main()
1253b62047bSLouis Dionne {
1263b62047bSLouis Dionne test<wrapper<uint16_t>>();
1273b62047bSLouis Dionne test<wrapper<float64_t>>();
1283b62047bSLouis Dionne test<int32_t>();
1293b62047bSLouis Dionne test<float32_t>();
1303b62047bSLouis Dionne
1313b62047bSLouis Dionne std::cout << done() << std::endl;
1323b62047bSLouis Dionne return 0;
1333b62047bSLouis Dionne }
134