replace_if.cpp 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  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. // HACKHACKHACK silence false positive:
  12. // error: ‘<anonymous>.ranges::istream_view<int>::cursor::rng_’ may be used uninitialized in this function
  13. // triggered on line 39.
  14. #if defined(__GNUC__) && !defined(__clang__) && __GNUC__ >= 7
  15. #include <range/v3/detail/config.hpp>
  16. RANGES_DIAGNOSTIC_IGNORE("-Wmaybe-uninitialized")
  17. #endif
  18. #include <string>
  19. #include <sstream>
  20. #include <vector>
  21. #include <range/v3/core.hpp>
  22. #include <range/v3/view/istream.hpp>
  23. #include <range/v3/functional/reference_wrapper.hpp>
  24. #include <range/v3/utility/copy.hpp>
  25. #include <range/v3/view/iota.hpp>
  26. #include <range/v3/view/take.hpp>
  27. #include <range/v3/view/replace_if.hpp>
  28. #include <range/v3/view/common.hpp>
  29. #include "../simple_test.hpp"
  30. #include "../test_utils.hpp"
  31. int main()
  32. {
  33. using namespace ranges;
  34. std::string str{"1 2 3 4 5 6 7 8 9 1 2 3 4 5 6 7 8 9 1 2 3 4 5 6 7 8 9 "};
  35. std::stringstream sin{str};
  36. {
  37. auto rng = istream<int>(sin) | views::replace_if([](int i){return i == 1; }, 42);
  38. has_type<int const &>(*begin(rng));
  39. CPP_assert(view_<decltype(rng)>);
  40. CPP_assert(!sized_range<decltype(rng)>);
  41. CPP_assert(!common_range<decltype(rng)>);
  42. CPP_assert(input_iterator<decltype(begin(rng))>);
  43. CPP_assert(!forward_iterator<decltype(begin(rng))>);
  44. auto tmp = rng | views::common;
  45. has_type<int const &>(*begin(tmp));
  46. CPP_assert(view_<decltype(tmp)>);
  47. CPP_assert(common_range<decltype(tmp)>);
  48. CPP_assert(!sized_range<decltype(tmp)>);
  49. CPP_assert(input_iterator<decltype(begin(tmp))>);
  50. CPP_assert(!forward_iterator<decltype(begin(tmp))>);
  51. std::vector<int> actual{begin(tmp), end(tmp)};
  52. ::check_equal(actual, {42, 2, 3, 4, 5, 6, 7, 8, 9, 42, 2, 3, 4, 5, 6, 7, 8, 9, 42, 2, 3, 4, 5, 6, 7, 8, 9});
  53. }
  54. {
  55. std::vector<int> vi{1,2,3,4,5,6,7,8,9};
  56. auto rng2 = vi | views::replace_if([](int i){return i == 5;}, 42);
  57. CPP_assert(same_as<range_value_t<decltype(rng2)>, int>);
  58. has_type<int const &>(*begin(rng2));
  59. has_type<int const &>(iter_move(begin(rng2)));
  60. CPP_assert(view_<decltype(rng2)>);
  61. CPP_assert(sized_range<decltype(rng2)>);
  62. CPP_assert(common_range<decltype(rng2)>);
  63. CPP_assert(random_access_iterator<decltype(begin(rng2))>);
  64. ::check_equal(rng2, {1,2,3,4,42,6,7,8,9});
  65. int forty_two = 42;
  66. auto rng3 = vi | views::replace_if([](int i){return i == 5;}, ref(forty_two));
  67. CPP_assert(same_as<range_value_t<decltype(rng3)>, int>);
  68. has_type<int &>(*begin(rng3));
  69. has_type<int const &>(iter_move(begin(rng3)));
  70. CPP_assert(view_<decltype(rng3)>);
  71. CPP_assert(sized_range<decltype(rng3)>);
  72. CPP_assert(common_range<decltype(rng3)>);
  73. CPP_assert(random_access_iterator<decltype(begin(rng3))>);
  74. ::check_equal(rng3, {1,2,3,4,42,6,7,8,9});
  75. }
  76. {
  77. auto rng4 = views::ints | views::replace_if([](int i){return i == 5;},42) | views::take(10);
  78. CPP_assert(same_as<range_value_t<decltype(rng4)>, int>);
  79. has_type<int>(*begin(rng4));
  80. has_type<int>(iter_move(begin(rng4)));
  81. CPP_assert(view_<decltype(rng4)>);
  82. CPP_assert(sized_range<decltype(rng4)>);
  83. CPP_assert(!common_range<decltype(rng4)>);
  84. CPP_assert(random_access_iterator<decltype(begin(rng4))>);
  85. ::check_equal(rng4, {0,1,2,3,4,42,6,7,8,9});
  86. }
  87. {
  88. // Check with a mutable predicate
  89. int rgi[] = {0,1,2,3,4,5,6,7,8,9};
  90. bool flag = false;
  91. auto mutable_only = views::replace_if(rgi, [flag](int) mutable { return flag = !flag;}, 42);
  92. ::check_equal(mutable_only, {42,1,42,3,42,5,42,7,42,9});
  93. CPP_assert(view_<decltype(mutable_only)>);
  94. CPP_assert(!view_<decltype(mutable_only) const>);
  95. }
  96. {
  97. int const some_ints[] = {1,2,3,4,5,6,7,8,9,1,2,3,4,5,6,7,8,9,1,2,3,4,5,6,7,8,9};
  98. auto rng = debug_input_view<int const>{some_ints} |
  99. views::replace_if([](int i){ return i == 1; }, 42);
  100. ::check_equal(rng, {42,2,3,4,5,6,7,8,9,42,2,3,4,5,6,7,8,9,42,2,3,4,5,6,7,8,9});
  101. }
  102. return test_result();
  103. }