xref: /llvm-project/pstl/test/std/algorithms/alg.modifying.operations/swap_ranges.pass.cpp (revision b5e896c0493d44c2d9819d584fe2814af7cfd476)
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