| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150 |
- // Range v3 library
- //
- // Copyright Eric Niebler 2014-present
- //
- // Use, modification and distribution is subject to the
- // Boost Software License, Version 1.0. (See accompanying
- // file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt)
- //
- // Project home: https://github.com/ericniebler/range-v3
- //
- // Copyright 2005 - 2007 Adobe Systems Incorporated
- // Distributed under the MIT License(see accompanying file LICENSE_1_0_0.txt
- // or a copy at http://stlab.adobe.com/licenses.html)
- //===----------------------------------------------------------------------===//
- //
- // The LLVM Compiler Infrastructure
- //
- // This file is dual licensed under the MIT and the University of Illinois Open
- // Source Licenses. See LICENSE.TXT for details.
- //
- //===----------------------------------------------------------------------===//
- #include <utility>
- #include <vector>
- #include <range/v3/core.hpp>
- #include <range/v3/algorithm/replace_if.hpp>
- #include "../simple_test.hpp"
- #include "../test_utils.hpp"
- #include "../test_iterators.hpp"
- template<typename Iter, typename Sent = Iter>
- void test_iter()
- {
- int ia[] = {0, 1, 2, 3, 4};
- const unsigned sa = sizeof(ia)/sizeof(ia[0]);
- Iter i = ranges::replace_if(Iter(ia), Sent(ia+sa), [](int j){return j==2;}, 5);
- CHECK(ia[0] == 0);
- CHECK(ia[1] == 1);
- CHECK(ia[2] == 5);
- CHECK(ia[3] == 3);
- CHECK(ia[4] == 4);
- CHECK(base(i) == ia + sa);
- }
- template<typename Iter, typename Sent = Iter>
- void test_rng()
- {
- int ia[] = {0, 1, 2, 3, 4};
- const unsigned sa = sizeof(ia)/sizeof(ia[0]);
- auto rng = ranges::make_subrange(Iter(ia), Sent(ia+sa));
- Iter i = ranges::replace_if(rng, [](int j){return j==2;}, 5);
- CHECK(ia[0] == 0);
- CHECK(ia[1] == 1);
- CHECK(ia[2] == 5);
- CHECK(ia[3] == 3);
- CHECK(ia[4] == 4);
- CHECK(base(i) == ia + sa);
- }
- constexpr bool equals_two(int i)
- {
- return i == 2;
- }
- constexpr bool test_constexpr()
- {
- using namespace ranges;
- int ia[] = {0, 1, 2, 3, 4};
- constexpr auto sa = ranges::size(ia);
- auto r = ranges::replace_if(ia, equals_two, 42);
- STATIC_CHECK_RETURN(r == ia + sa);
- STATIC_CHECK_RETURN(ia[0] == 0);
- STATIC_CHECK_RETURN(ia[1] == 1);
- STATIC_CHECK_RETURN(ia[2] == 42);
- STATIC_CHECK_RETURN(ia[3] == 3);
- STATIC_CHECK_RETURN(ia[4] == 4);
- return true;
- }
- int main()
- {
- test_iter<ForwardIterator<int*> >();
- test_iter<BidirectionalIterator<int*> >();
- test_iter<RandomAccessIterator<int*> >();
- test_iter<int*>();
- test_iter<ForwardIterator<int*>, Sentinel<int*> >();
- test_iter<BidirectionalIterator<int*>, Sentinel<int*> >();
- test_iter<RandomAccessIterator<int*>, Sentinel<int*> >();
- test_rng<ForwardIterator<int*> >();
- test_rng<BidirectionalIterator<int*> >();
- test_rng<RandomAccessIterator<int*> >();
- test_rng<int*>();
- test_rng<ForwardIterator<int*>, Sentinel<int*> >();
- test_rng<BidirectionalIterator<int*>, Sentinel<int*> >();
- test_rng<RandomAccessIterator<int*>, Sentinel<int*> >();
- // test projection
- {
- using P = std::pair<int,std::string>;
- P ia[] = {{0,"0"}, {1,"1"}, {2,"2"}, {3,"3"}, {4,"4"}};
- P *i = ranges::replace_if(ia, [](int j){return j==2;}, std::make_pair(42,"42"),
- &std::pair<int,std::string>::first);
- CHECK(ia[0] == P{0,"0"});
- CHECK(ia[1] == P{1,"1"});
- CHECK(ia[2] == P{42,"42"});
- CHECK(ia[3] == P{3,"3"});
- CHECK(ia[4] == P{4,"4"});
- CHECK(i == ranges::end(ia));
- }
- // test rvalue ranges
- {
- using P = std::pair<int,std::string>;
- P ia[] = {{0,"0"}, {1,"1"}, {2,"2"}, {3,"3"}, {4,"4"}};
- auto i = ranges::replace_if(std::move(ia), [](int j){return j==2;}, std::make_pair(42,"42"),
- &std::pair<int,std::string>::first);
- #ifndef RANGES_WORKAROUND_MSVC_573728
- CHECK(::is_dangling(i));
- #endif // RANGES_WORKAROUND_MSVC_573728
- CHECK(ia[0] == P{0,"0"});
- CHECK(ia[1] == P{1,"1"});
- CHECK(ia[2] == P{42,"42"});
- CHECK(ia[3] == P{3,"3"});
- CHECK(ia[4] == P{4,"4"});
- }
- {
- using P = std::pair<int,std::string>;
- std::vector<P> ia{{0,"0"}, {1,"1"}, {2,"2"}, {3,"3"}, {4,"4"}};
- auto i = ranges::replace_if(std::move(ia), [](int j){return j==2;}, std::make_pair(42,"42"),
- &std::pair<int,std::string>::first);
- CHECK(::is_dangling(i));
- CHECK(ia[0] == P{0,"0"});
- CHECK(ia[1] == P{1,"1"});
- CHECK(ia[2] == P{42,"42"});
- CHECK(ia[3] == P{3,"3"});
- CHECK(ia[4] == P{4,"4"});
- }
- {
- STATIC_CHECK(test_constexpr());
- }
- return ::test_result();
- }
|