AimRT/_deps/libunifex-src/examples/reduce_with_trampoline.cpp
2025-01-12 20:42:42 +08:00

45 lines
1.4 KiB
C++

/*
* 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.
*/
#include <unifex/range_stream.hpp>
#include <unifex/reduce_stream.hpp>
#include <unifex/sync_wait.hpp>
#include <unifex/trampoline_scheduler.hpp>
#include <unifex/then.hpp>
#include <unifex/transform_stream.hpp>
#include <unifex/typed_via_stream.hpp>
#include <cstdio>
using namespace unifex;
// This test uses the trampoline_scheduler to avoid stack-overflow due to very
// deep recursion from a reduce over a synchronous stream.
int main() {
sync_wait(then(
reduce_stream(
typed_via_stream(
trampoline_scheduler{},
transform_stream(
range_stream{0, 100'000},
[](int value) { return value * value; })),
0,
[](int state, int value) { return state + 10 * value; }),
[&](int result) { std::printf("result: %i\n", result); }));
return 0;
}