-
Notifications
You must be signed in to change notification settings - Fork 182
/
range_stream.hpp
102 lines (81 loc) · 2.49 KB
/
range_stream.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
* Copyright (c) Facebook, Inc. and its affiliates.
*
* Licensed under the Apache License Version 2.0 with LLVM Exceptions
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* https://llvm.org/LICENSE.txt
*
* 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.
*/
#pragma once
#include <unifex/blocking.hpp>
#include <unifex/just_done.hpp>
#include <unifex/receiver_concepts.hpp>
#include <unifex/stream_concepts.hpp>
#include <type_traits>
#include <utility>
#include <unifex/detail/prologue.hpp>
namespace unifex {
namespace _range {
struct stream;
template <typename Receiver>
struct _op {
struct type;
};
template <typename Receiver>
using operation = typename _op<remove_cvref_t<Receiver>>::type;
template <typename Receiver>
struct _op<Receiver>::type {
stream& stream_;
Receiver receiver_;
void start() noexcept;
};
struct next_sender {
stream& stream_;
template <
template <typename...>
class Variant,
template <typename...>
class Tuple>
using value_types = Variant<Tuple<int>>;
template <template <typename...> class Variant>
using error_types = Variant<>;
static constexpr bool sends_done = true;
static constexpr blocking_kind blocking = blocking_kind::always_inline;
template <typename Receiver>
operation<Receiver> connect(Receiver&& receiver) && {
return operation<Receiver>{stream_, (Receiver &&) receiver};
}
template <typename Receiver>
void connect(Receiver&& receiver) const& = delete;
};
struct stream {
int next_;
int max_;
explicit stream(int max) : next_(0), max_(max) {}
explicit stream(int start, int max) : next_(start), max_(max) {}
friend next_sender tag_invoke(tag_t<next>, stream& s) noexcept {
return next_sender{s};
}
friend auto tag_invoke(tag_t<cleanup>, stream&) noexcept {
return just_done();
}
};
template <typename Receiver>
void _op<Receiver>::type::start() noexcept {
if (stream_.next_ < stream_.max_) {
unifex::set_value(std::move(receiver_), stream_.next_++);
} else {
unifex::set_done(std::move(receiver_));
}
}
} // namespace _range
using range_stream = _range::stream;
} // namespace unifex
#include <unifex/detail/epilogue.hpp>