123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- #pragma once
- /*
- * Copyright 2018-present Facebook, Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include <folly/experimental/pushmi/flow_receiver.h>
- #include <folly/experimental/pushmi/o/extension_operators.h>
- #include <folly/experimental/pushmi/o/submit.h>
- #include <folly/experimental/pushmi/receiver.h>
- namespace pushmi {
- namespace detail {
- template <class F, class Tag, bool IsFlow = false>
- struct transform_on;
- template <class F>
- struct transform_on<F, is_single<>> {
- F f_;
- transform_on() = default;
- constexpr explicit transform_on(F f) : f_(std::move(f)) {}
- struct value_fn {
- F f_;
- value_fn() = default;
- constexpr explicit value_fn(F f) : f_(std::move(f)) {}
- template <class Out, class V0, class... VN>
- auto operator()(Out& out, V0&& v0, VN&&... vn) {
- using Result = ::pushmi::invoke_result_t<F, V0, VN...>;
- static_assert(
- ::pushmi::SemiMovable<Result>,
- "none of the functions supplied to transform can convert this value");
- static_assert(
- ::pushmi::ReceiveValue<Out, Result>,
- "Result of value transform cannot be delivered to Out");
- ::pushmi::set_value(out, f_((V0 &&) v0, (VN &&) vn...));
- }
- };
- template <class Out>
- auto operator()(Out out) const {
- return ::pushmi::make_receiver(std::move(out), value_fn{f_});
- }
- };
- template <class F>
- struct transform_on<F, is_single<>, true> {
- F f_;
- transform_on() = default;
- constexpr explicit transform_on(F f) : f_(std::move(f)) {}
- template <class Out>
- auto operator()(Out out) const {
- return make_flow_single(std::move(out), on_value(*this));
- }
- template <class Out, class V0, class... VN>
- auto operator()(Out& out, V0&& v0, VN&&... vn) {
- using Result = ::pushmi::invoke_result_t<F, V0, VN...>;
- static_assert(
- ::pushmi::SemiMovable<Result>,
- "none of the functions supplied to transform can convert this value");
- static_assert(
- ::pushmi::Flow<Out> && ::pushmi::ReceiveValue<Out, Result>,
- "Result of value transform cannot be delivered to Out");
- ::pushmi::set_value(out, f_((V0 &&) v0, (VN &&) vn...));
- }
- };
- template <class F>
- struct transform_on<F, is_many<>> {
- F f_;
- transform_on() = default;
- constexpr explicit transform_on(F f) : f_(std::move(f)) {}
- template <class Out>
- auto operator()(Out out) const {
- return ::pushmi::make_receiver(std::move(out), on_value(*this));
- }
- template <class Out, class V0, class... VN>
- auto operator()(Out& out, V0&& v0, VN&&... vn) {
- using Result = ::pushmi::invoke_result_t<F, V0, VN...>;
- static_assert(
- ::pushmi::SemiMovable<Result>,
- "none of the functions supplied to transform can convert this value");
- static_assert(
- ::pushmi::ReceiveValue<Out, Result>,
- "Result of value transform cannot be delivered to Out");
- ::pushmi::set_value(out, f_((V0 &&) v0, (VN &&) vn...));
- }
- };
- template <class F>
- struct transform_on<F, is_many<>, true> {
- F f_;
- transform_on() = default;
- constexpr explicit transform_on(F f) : f_(std::move(f)) {}
- template <class Out>
- auto operator()(Out out) const {
- return make_flow_receiver(std::move(out), on_value(*this));
- }
- template <class Out, class V0, class... VN>
- auto operator()(Out& out, V0&& v0, VN&&... vn) {
- using Result = ::pushmi::invoke_result_t<F, V0, VN...>;
- static_assert(
- ::pushmi::SemiMovable<Result>,
- "none of the functions supplied to transform can convert this value");
- static_assert(
- ::pushmi::Flow<Out> && ::pushmi::ReceiveValue<Out, Result>,
- "Result of value transform cannot be delivered to Out");
- ::pushmi::set_value(out, f_((V0 &&) v0, (VN &&) vn...));
- }
- };
- struct transform_fn {
- private:
- template <class F>
- struct impl {
- F f_;
- PUSHMI_TEMPLATE(class In)
- (requires Sender<In>)
- auto operator()(In in) const {
- using Cardinality = property_set_index_t<properties_t<In>, is_single<>>;
- return ::pushmi::detail::sender_from(
- std::move(in),
- ::pushmi::detail::submit_transform_out<In>(
- // copy 'f_' to allow multiple calls to connect to multiple 'in'
- transform_on<
- F,
- Cardinality,
- property_query_v<properties_t<In>, is_flow<>>>{f_}));
- }
- };
- public:
- template <class... FN>
- auto operator()(FN... fn) const {
- auto f = ::pushmi::overload(std::move(fn)...);
- using F = decltype(f);
- return impl<F>{std::move(f)};
- }
- };
- } // namespace detail
- namespace operators {
- PUSHMI_INLINE_VAR constexpr detail::transform_fn transform{};
- } // namespace operators
- } // namespace pushmi
|