Rim/iw5-mod
Rim/iw5-mod
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Prepare scripting scheduler

Browse Source
This commit is contained in:
momo5502 2019-01-20 18:29:13 +01:00
parent 4c86842b7f
commit 7256cc786c
8 changed files with 194 additions and 154 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/game/scripting/context.cpp
View File

@ -5,7 +5,7 @@ namespace game
{
namespace scripting
{
context::context() : executer_(this), parameters_(this), event_handler_(this)
context::context() : executer_(this), scheduler_(this), parameters_(this), event_handler_(this)
{
context_initializer::initialize(this);
}
@ -15,6 +15,11 @@ namespace game
return &this->executer_;
}
scheduler* context::get_scheduler()
{
return &this->scheduler_;
}
parameters* context::get_parameters()
{
return &this->parameters_;

3
src/game/scripting/context.hpp
View File

@ -1,5 +1,6 @@
#pragma once
#include "executer.hpp"
#include "scheduler.hpp"
#include "parameters.hpp"
#include "event_handler.hpp"
@ -13,12 +14,14 @@ namespace game
context();
executer* get_executer();
scheduler* get_scheduler();
parameters* get_parameters();
event_handler* get_event_handler();
chaiscript::ChaiScript* get_chai();
private:
executer executer_;
scheduler scheduler_;
parameters parameters_;
event_handler event_handler_;
chaiscript::ChaiScript chai_;

5
src/game/scripting/event_handler.cpp
View File

@ -9,6 +9,11 @@ namespace game
{
}
void event_handler::run_frame()
{
}
void event_handler::dispatch(event* event)
{
try

1
src/game/scripting/event_handler.hpp
View File

@ -31,6 +31,7 @@ namespace game
public:
explicit event_handler(context* context);
void run_frame();
void dispatch(event* event);
void add_event_listener(const event_listener& listener);

17
src/game/scripting/scheduler.cpp Normal file
View File

@ -0,0 +1,17 @@
#include "std_include.hpp"
#include "context.hpp"
namespace game
{
namespace scripting
{
scheduler::scheduler(context* context) : context_(context)
{
}
void scheduler::run_frame()
{
}
}
}

20
src/game/scripting/scheduler.hpp Normal file
View File

@ -0,0 +1,20 @@
#pragma once
namespace game
{
namespace scripting
{
class context;
class scheduler final
{
public:
explicit scheduler(context* context);
void run_frame();
private:
context* context_;
};
}
}

264
src/module/scripting.cpp
View File

@ -1,131 +1,151 @@
#include <std_include.hpp>
#include "loader/module_loader.hpp"
#include "utils/hook.hpp"
#include "utils/io.hpp"
#include "game/scripting/context.hpp"
#include "scheduler.hpp"
#include "scripting.hpp"
class scripting final : public module
{
public:
void post_load() override
{
start_hook_.initialize(SELECT_VALUE(0x50C575, 0x50D4F2, 0x48A026), &start_execution_stub, HOOK_CALL) //
->install() //
->quick();
stop_hook_.initialize(SELECT_VALUE(0x528B04, 0x569E46, 0x4F03FA), &stop_execution_stub, HOOK_CALL) //
->install() //
->quick();
utils::hook(SELECT_VALUE(0x6109F3, 0x56B637, 0x4EDFF7), &vm_notify_stub, HOOK_CALL).install()->quick();
utils::hook(SELECT_VALUE(0x6128BE, 0x56D541, 0x4EFAF9), &vm_notify_stub, HOOK_CALL).install()->quick();
if (game::is_sp())
{
utils::hook(0x610970, &vm_notify_stub, HOOK_JUMP).install()->quick();
}
scheduler::on_frame(std::bind(&scripting::run_frame, this));
}
void pre_destroy() override
{
this->scripts_.clear();
}
private:
std::vector<std::unique_ptr<game::scripting::context>> scripts_;
void load_scripts()
{
const auto scripts = utils::io::list_files("open-iw5/scripts/");
for (const auto& script : scripts)
{
if (script.substr(script.find_last_of('.') + 1) == "chai")
{
try
{
auto context = std::make_unique<game::scripting::context>();
context->get_chai()->eval_file(script);
this->scripts_.push_back(std::move(context));
}
catch (chaiscript::exception::eval_error& e)
{
throw std::runtime_error(e.pretty_print());
}
}
}
}
void start_execution()
{
try
{
this->load_scripts();
}
catch (std::exception& e)
{
propagate_error(e);
}
}
void stop_execution()
{
this->scripts_.clear();
}
void run_frame()
{
for (const auto& script : this->scripts_)
{
script->get_scheduler()->run_frame();
}
}
void dispatch(game::scripting::event* event)
{
for (const auto& script : this->scripts_)
{
script->get_event_handler()->dispatch(event);
}
}
static utils::hook start_hook_;
static utils::hook stop_hook_;
static void propagate_error(const std::exception& e)
{
printf("\n******* Script execution error *******\n");
printf("%s\n", e.what());
printf("**************************************\n\n");
scheduler::error("Script execution error\n(see console for actual details)\n", 5);
}
static void start_execution_stub()
{
module_loader::get<scripting>()->start_execution();
reinterpret_cast<void(*)()>(start_hook_.get_original())();
}
static void stop_execution_stub()
{
module_loader::get<scripting>()->stop_execution();
reinterpret_cast<void(*)()>(stop_hook_.get_original())();
}
static void vm_notify_stub(const unsigned int notify_id, const unsigned short type,
game::native::VariableValue* stack)
{
try
{
game::scripting::event e;
e.name = game::native::SL_ConvertToString(type);
e.entity_id = notify_id;
if (e.name == "touch") return; // Skip that for now
//printf("%X: %s\n", e.entity_id, e.name.data());
for (auto value = stack; value->type != game::native::SCRIPT_END; --value)
{
e.arguments.emplace_back(*value);
}
module_loader::get<scripting>()->dispatch(&e);
}
catch (std::exception& e)
{
propagate_error(e);
}
game::native::VM_Notify(notify_id, type, stack);
}
};
utils::hook scripting::start_hook_;
utils::hook scripting::stop_hook_;
void scripting::post_load()
{
start_hook_.initialize(SELECT_VALUE(0x50C575, 0x50D4F2, 0x48A026), &start_execution_stub, HOOK_CALL) //
->install() //
->quick();
stop_hook_.initialize(SELECT_VALUE(0x528B04, 0x569E46, 0x4F03FA), &stop_execution_stub, HOOK_CALL) //
->install() //
->quick();
utils::hook(SELECT_VALUE(0x6109F3, 0x56B637, 0x4EDFF7), &vm_notify_stub, HOOK_CALL).install()->quick();
utils::hook(SELECT_VALUE(0x6128BE, 0x56D541, 0x4EFAF9), &vm_notify_stub, HOOK_CALL).install()->quick();
if (game::is_sp())
{
utils::hook(0x610970, &vm_notify_stub, HOOK_JUMP).install()->quick();
}
}
void scripting::pre_destroy()
{
this->scripts_.clear();
}
void scripting::load_scripts()
{
const auto scripts = utils::io::list_files("open-iw5/scripts/");
for (const auto& script : scripts)
{
if (script.substr(script.find_last_of('.') + 1) == "chai")
{
try
{
auto context = std::make_unique<game::scripting::context>();
context->get_chai()->eval_file(script);
this->scripts_.push_back(std::move(context));
}
catch (chaiscript::exception::eval_error& e)
{
throw std::runtime_error(e.pretty_print());
}
}
}
}
void scripting::start_execution()
{
try
{
this->load_scripts();
}
catch (std::exception& e)
{
propagate_error(e);
}
}
void scripting::stop_execution()
{
this->scripts_.clear();
}
void scripting::dispatch(game::scripting::event* event)
{
for (const auto& script : this->scripts_)
{
script->get_event_handler()->dispatch(event);
}
}
void scripting::propagate_error(const std::exception& e)
{
printf("\n******* Script execution error *******\n");
printf("%s\n", e.what());
printf("**************************************\n\n");
scheduler::error("Script execution error\n(see console for actual details)\n", 5);
}
void scripting::start_execution_stub()
{
module_loader::get<scripting>()->start_execution();
reinterpret_cast<void(*)()>(start_hook_.get_original())();
}
void scripting::stop_execution_stub()
{
module_loader::get<scripting>()->stop_execution();
reinterpret_cast<void(*)()>(stop_hook_.get_original())();
}
void scripting::vm_notify_stub(const unsigned int notify_id, const unsigned short type,
game::native::VariableValue* stack)
{
try
{
game::scripting::event e;
e.name = game::native::SL_ConvertToString(type);
e.entity_id = notify_id;
if (e.name == "touch") return; // Skip that for now
//printf("%X: %s\n", e.entity_id, e.name.data());
for (auto value = stack; value->type != game::native::SCRIPT_END; --value)
{
e.arguments.emplace_back(*value);
}
module_loader::get<scripting>()->dispatch(&e);
}
catch (std::exception& e)
{
propagate_error(e);
}
game::native::VM_Notify(notify_id, type, stack);
}
REGISTER_MODULE(scripting)

31
src/module/scripting.hpp
View File

@ -1,31 +0,0 @@
#pragma once
#include "loader/module_loader.hpp"
#include "utils/hook.hpp"
#include "game/scripting/context.hpp"
class scripting final : public module
{
public:
void post_load() override;
void pre_destroy() override;
static void propagate_error(const std::exception& e);
private:
std::vector<std::unique_ptr<game::scripting::context>> scripts_;
void load_scripts();
void start_execution();
void stop_execution();
void dispatch(game::scripting::event* event);
static utils::hook start_hook_;
static utils::hook stop_hook_;
static void start_execution_stub();
static void stop_execution_stub();
static void vm_notify_stub(const unsigned int notify_id, const unsigned short type,
game::native::VariableValue* stack);
};