1 //===----------------------------------------------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is dual licensed under the MIT and the University of Illinois Open 6 // Source Licenses. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 // <algorithm> 11 12 // template <class InputIterator, class OutputIterator1, 13 // class OutputIterator2, class Predicate> 14 // constexpr pair<OutputIterator1, OutputIterator2> // constexpr after C++17 15 // partition_copy(InputIterator first, InputIterator last, 16 // OutputIterator1 out_true, OutputIterator2 out_false, 17 // Predicate pred); 18 19 #include <algorithm> 20 #include <cassert> 21 22 #include "test_macros.h" 23 #include "test_iterators.h" 24 25 struct is_odd 26 { 27 TEST_CONSTEXPR bool operator()(const int& i) const {return i & 1;} 28 }; 29 30 #if TEST_STD_VER > 17 31 TEST_CONSTEXPR bool test_constexpr() { 32 int ia[] = {1, 3, 5, 2, 4, 6}; 33 int r1[10] = {0}; 34 int r2[10] = {0}; 35 36 auto p = std::partition_copy(std::begin(ia), std::end(ia), 37 std::begin(r1), std::begin(r2), is_odd()); 38 39 return std::all_of(std::begin(r1), p.first, is_odd()) 40 && std::all_of(p.first, std::end(r1), [](int a){return a == 0;}) 41 && std::none_of(std::begin(r2), p.second, is_odd()) 42 && std::all_of(p.second, std::end(r2), [](int a){return a == 0;}) 43 ; 44 } 45 #endif 46 47 int main() 48 { 49 { 50 const int ia[] = {1, 2, 3, 4, 6, 8, 5, 7}; 51 int r1[10] = {0}; 52 int r2[10] = {0}; 53 typedef std::pair<output_iterator<int*>, int*> P; 54 P p = std::partition_copy(input_iterator<const int*>(std::begin(ia)), 55 input_iterator<const int*>(std::end(ia)), 56 output_iterator<int*>(r1), r2, is_odd()); 57 assert(p.first.base() == r1 + 4); 58 assert(r1[0] == 1); 59 assert(r1[1] == 3); 60 assert(r1[2] == 5); 61 assert(r1[3] == 7); 62 assert(p.second == r2 + 4); 63 assert(r2[0] == 2); 64 assert(r2[1] == 4); 65 assert(r2[2] == 6); 66 assert(r2[3] == 8); 67 } 68 69 #if TEST_STD_VER > 17 70 static_assert(test_constexpr()); 71 #endif 72 } 73