| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197 |
- /// \file
- // Range v3 library
- //
- // Copyright Mitsutaka Takeda 2018-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
- //
- #ifndef RANGES_V3_VIEW_EXCLUSIVE_SCAN_HPP
- #define RANGES_V3_VIEW_EXCLUSIVE_SCAN_HPP
- #include <concepts/concepts.hpp>
- #include <range/v3/range_fwd.hpp>
- #include <range/v3/functional/arithmetic.hpp>
- #include <range/v3/functional/bind_back.hpp>
- #include <range/v3/functional/invoke.hpp>
- #include <range/v3/view/adaptor.hpp>
- #include <range/v3/view/view.hpp>
- #include <range/v3/detail/prologue.hpp>
- namespace ranges
- {
- // clang-format off
- /// \concept exclusive_scan_constraints_
- /// \brief The \c exclusive_scan_constraints_ concept
- template(typename Rng, typename T, typename Fun)(
- concept (exclusive_scan_constraints_)(Rng, T, Fun),
- invocable<Fun &, T, range_reference_t<Rng>> AND
- assignable_from<T &, invoke_result_t<Fun &, T, range_reference_t<Rng>>>
- );
- /// \concept exclusive_scan_constraints
- /// \brief The \c exclusive_scan_constraints concept
- template<typename Rng, typename T, typename Fun>
- CPP_concept exclusive_scan_constraints =
- viewable_range<Rng> && input_range<Rng> &&
- copy_constructible<T> &&
- CPP_concept_ref(ranges::exclusive_scan_constraints_, Rng, T, Fun);
- // clang-format on
- /// \addtogroup group-views
- /// @{
- template<typename Rng, typename T, typename Fun>
- struct exclusive_scan_view : view_adaptor<exclusive_scan_view<Rng, T, Fun>, Rng>
- {
- private:
- friend range_access;
- CPP_assert(exclusive_scan_constraints<Rng, T, Fun>);
- semiregular_box_t<T> init_;
- semiregular_box_t<Fun> fun_;
- using single_pass = meta::bool_<single_pass_iterator_<iterator_t<Rng>>>;
- using use_sentinel_t = meta::bool_<!common_range<Rng> || single_pass{}>;
- template<bool IsConst>
- struct adaptor : adaptor_base
- {
- private:
- friend struct adaptor<!IsConst>;
- using exclusive_scan_view_t = meta::const_if_c<IsConst, exclusive_scan_view>;
- using CRng = meta::const_if_c<IsConst, Rng>;
- semiregular_box_t<T> sum_;
- exclusive_scan_view_t * rng_;
- // clang-format off
- auto CPP_auto_fun(move_or_copy_init)(std::false_type)
- (
- return (rng_->init_)
- )
- // If the base range is single-pass, we can move the init value.
- auto CPP_auto_fun(move_or_copy_init)(std::true_type)
- (
- return std::move(rng_->init_)
- )
- // clang-format on
- public : using single_pass = exclusive_scan_view::single_pass;
- adaptor() = default;
- adaptor(exclusive_scan_view_t * rng)
- : rng_(rng)
- {}
- template(bool Other)(
- requires IsConst AND CPP_NOT(Other)) //
- adaptor(adaptor<Other> that)
- : rng_(that.rng_)
- {}
- iterator_t<CRng> begin(exclusive_scan_view_t &)
- {
- sum_ = move_or_copy_init(single_pass{});
- return ranges::begin(rng_->base());
- }
- T read(iterator_t<CRng> const &) const
- {
- return sum_;
- }
- void next(iterator_t<CRng> & it)
- {
- RANGES_EXPECT(it != ranges::end(rng_->base()));
- sum_ = invoke(rng_->fun_, static_cast<T &&>(std::move(sum_)), *it);
- ++it;
- }
- void prev() = delete;
- };
- adaptor<false> begin_adaptor()
- {
- return {this};
- }
- meta::if_<use_sentinel_t, adaptor_base, adaptor<false>> end_adaptor()
- {
- return {this};
- }
- CPP_member
- auto begin_adaptor() const //
- -> CPP_ret(adaptor<true>)(
- requires exclusive_scan_constraints<Rng const, T, Fun const>)
- {
- return {this};
- }
- CPP_member
- auto end_adaptor() const
- -> CPP_ret(meta::if_<use_sentinel_t, adaptor_base, adaptor<true>>)(
- requires exclusive_scan_constraints<Rng const, T, Fun const>)
- {
- return {this};
- }
- public:
- exclusive_scan_view() = default;
- constexpr exclusive_scan_view(Rng rng, T init, Fun fun)
- : exclusive_scan_view::view_adaptor{std::move(rng)}
- , init_(std::move(init))
- , fun_(std::move(fun))
- {}
- CPP_auto_member
- auto CPP_fun(size)()(const
- requires sized_range<Rng const>)
- {
- return ranges::size(this->base());
- }
- CPP_auto_member
- auto CPP_fun(size)()(
- requires sized_range<Rng>)
- {
- return ranges::size(this->base());
- }
- };
- #if RANGES_CXX_DEDUCTION_GUIDES >= RANGES_CXX_DEDUCTION_GUIDES_17
- template(typename Rng, typename T, typename Fun)(
- requires copy_constructible<T> AND copy_constructible<Fun>)
- exclusive_scan_view(Rng &&, T, Fun) //
- -> exclusive_scan_view<views::all_t<Rng>, T, Fun>;
- #endif
- namespace views
- {
- struct exclusive_scan_base_fn
- {
- template(typename Rng, typename T, typename Fun = plus)(
- requires exclusive_scan_constraints<Rng, T, Fun>)
- constexpr exclusive_scan_view<all_t<Rng>, T, Fun> //
- operator()(Rng && rng, T init, Fun fun = Fun{}) const
- {
- return {all(static_cast<Rng &&>(rng)), std::move(init), std::move(fun)};
- }
- };
- struct exclusive_scan_fn : exclusive_scan_base_fn
- {
- using exclusive_scan_base_fn::operator();
- template<typename T, typename Fun = plus>
- constexpr auto operator()(T init, Fun fun = {}) const
- {
- return make_view_closure(
- bind_back(exclusive_scan_base_fn{}, std::move(init), std::move(fun)));
- }
- };
- /// \relates exclusive_scan_fn
- /// \ingroup group-views
- RANGES_INLINE_VARIABLE(exclusive_scan_fn, exclusive_scan)
- } // namespace views
- /// @}
- } // namespace ranges
- #include <range/v3/detail/epilogue.hpp>
- #endif // RANGES_V3_VIEW_EXCLUSIVE_SCAN_HPP
|