93 lines
1.8 KiB
C++
93 lines
1.8 KiB
C++
|
#pragma once
|
||
|
|
#include "common.hpp"
|
||
|
|
#include "logger.hpp"
|
||
|
|
#include "script.hpp"
|
||
|
|
|
||
|
|
namespace big
|
||
|
|
{
|
||
|
|
static void script_exception_handler(PEXCEPTION_POINTERS exp)
|
||
|
|
{
|
||
|
|
LOG_ERROR("Script threw an exception!");
|
||
|
|
g_stackwalker.ShowCallstack(GetCurrentThread(), exp->ContextRecord);
|
||
|
|
}
|
||
|
|
|
||
|
|
script::script(func_t func, std::optional<std::size_t> stack_size) :
|
||
|
|
m_func(func),
|
||
|
|
m_script_fiber(nullptr),
|
||
|
|
m_main_fiber(nullptr)
|
||
|
|
{
|
||
|
|
m_script_fiber = CreateFiber(stack_size.has_value() ? stack_size.value() : 0, [](void* param)
|
||
|
|
{
|
||
|
|
auto this_script = static_cast<script*>(param);
|
||
|
|
this_script->fiber_func();
|
||
|
|
}, this);
|
||
|
|
}
|
||
|
|
|
||
|
|
script::~script()
|
||
|
|
{
|
||
|
|
if (m_script_fiber)
|
||
|
|
DeleteFiber(m_script_fiber);
|
||
|
|
}
|
||
|
|
|
||
|
|
void script::tick()
|
||
|
|
{
|
||
|
|
m_main_fiber = GetCurrentFiber();
|
||
|
|
if (!m_wake_time.has_value() || m_wake_time.value() <= std::chrono::high_resolution_clock::now())
|
||
|
|
{
|
||
|
|
SwitchToFiber(m_script_fiber);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void script::yield(std::optional<std::chrono::high_resolution_clock::duration> time)
|
||
|
|
{
|
||
|
|
if (time.has_value())
|
||
|
|
{
|
||
|
|
m_wake_time = std::chrono::high_resolution_clock::now() + time.value();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_wake_time = std::nullopt;
|
||
|
|
}
|
||
|
|
|
||
|
|
SwitchToFiber(m_main_fiber);
|
||
|
|
}
|
||
|
|
|
||
|
|
script *script::get_current()
|
||
|
|
{
|
||
|
|
return static_cast<script*>(GetFiberData());
|
||
|
|
}
|
||
|
|
|
||
|
|
void script::fiber_func()
|
||
|
|
{
|
||
|
|
__try
|
||
|
|
{
|
||
|
|
[this]()
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
m_func();
|
||
|
|
}
|
||
|
|
catch (std::exception const &ex)
|
||
|
|
{
|
||
|
|
auto ex_class = typeid(ex).name() + 6;
|
||
|
|
LOG_INFO("Script threw an C++ expection! {}: {}", ex_class, ex.what());
|
||
|
|
}
|
||
|
|
catch (...)
|
||
|
|
{
|
||
|
|
LOG_INFO("Script threw a C++ exception!");
|
||
|
|
}
|
||
|
|
}();
|
||
|
|
}
|
||
|
|
__except (script_exception_handler(GetExceptionInformation()), EXCEPTION_EXECUTE_HANDLER)
|
||
|
|
{
|
||
|
|
LOG_INFO("Script threw an exception!");
|
||
|
|
}
|
||
|
|
|
||
|
|
LOG_INFO("Script finished!");
|
||
|
|
while (true)
|
||
|
|
{
|
||
|
|
yield();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|