sol2/examples/source/coroutine.cpp
ThePhD 57d9a05f88
🎨 Refactor tutorial examples
- 🛠 Make sure the tutorials compile across platforms!
- ✍ Redo quite a bit of the documentation
2021-03-06 10:14:48 -05:00

69 lines
1.7 KiB
C++

#define SOL_ALL_SAFETIES_ON 1
#include <sol/sol.hpp>
#include <string>
#include <iostream>
int main() {
std::cout << "=== coroutine ===" << std::endl;
sol::state lua;
std::vector<sol::coroutine> tasks;
lua.open_libraries(sol::lib::base, sol::lib::coroutine);
sol::thread runner_thread = sol::thread::create(lua);
lua.set_function("start_task",
[&runner_thread, &tasks](
sol::function f, sol::variadic_args va) {
// You must ALWAYS get the current state
sol::state_view runner_thread_state
= runner_thread.state();
// Put the task in our task list to keep it alive
// and track it
std::size_t task_index = tasks.size();
tasks.emplace_back(runner_thread_state, f);
sol::coroutine& f_on_runner_thread
= tasks[task_index];
// call coroutine with arguments that came
// from main thread / other thread
// pusher for `variadic_args` and other sol types
// will transfer the arguments from the calling
// thread to the runner thread automatically for
// you using `lua_xmove` internally
int wait = f_on_runner_thread(va);
std::cout << "First return: " << wait
<< std::endl;
// When you call it again, you don't need new
// arguments (they remain the same from the first
// call)
f_on_runner_thread();
std::cout << "Second run complete: " << wait
<< std::endl;
});
lua.script(
R"(
function main(x, y, z)
-- do something
coroutine.yield(20)
-- do something else
-- do ...
print(x, y, z)
end
function main2(x, y)
coroutine.yield(10)
print(x, y)
end
start_task(main, 10, 12, 8)
start_task(main2, 1, 2)
)");
std::cout << std::endl;
return 0;
}