123 lines
3.2 KiB
C++
123 lines
3.2 KiB
C++
#include <std_include.hpp>
|
|
#include "loader/component_loader.hpp"
|
|
#include "fastfiles.hpp"
|
|
|
|
#include "command.hpp"
|
|
#include "console.hpp"
|
|
#include "localized_strings.hpp"
|
|
|
|
#include <utils/hook.hpp>
|
|
#include <utils/concurrency.hpp>
|
|
#include <utils/string.hpp>
|
|
|
|
namespace fastfiles
|
|
{
|
|
static utils::concurrency::container<std::string> current_fastfile;
|
|
|
|
namespace
|
|
{
|
|
utils::hook::detour db_try_load_x_file_internal_hook;
|
|
utils::hook::detour db_find_xasset_header;
|
|
|
|
void db_try_load_x_file_internal(const char* zone_name, const int flags)
|
|
{
|
|
console::info("Loading fastfile %s\n", zone_name);
|
|
current_fastfile.access([&](std::string& fastfile)
|
|
{
|
|
fastfile = zone_name;
|
|
});
|
|
return db_try_load_x_file_internal_hook.invoke<void>(zone_name, flags);
|
|
}
|
|
|
|
game::XAssetHeader db_find_xasset_header_stub(game::XAssetType type, const char* name, int allow_create_default)
|
|
{
|
|
const auto start = game::Sys_Milliseconds();
|
|
const auto result = db_find_xasset_header.invoke<game::XAssetHeader>(type, name, allow_create_default);
|
|
const auto diff = game::Sys_Milliseconds() - start;
|
|
|
|
if (diff > 100)
|
|
{
|
|
console::print(
|
|
result.data == nullptr
|
|
? console::con_type_error
|
|
: console::con_type_warning,
|
|
"Waited %i msec for %sasset \"%s\", of type \"%s\"\n",
|
|
diff,
|
|
result.data == nullptr
|
|
? "missing "
|
|
: "",
|
|
name,
|
|
game::g_assetNames[type]
|
|
);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
void add_missing_localized_strings()
|
|
{
|
|
for (auto map = &game::maps[0]; map->unk; ++map)
|
|
{
|
|
const auto str = utils::string::va("LUA_MENU_SP_LOCATION_%s",
|
|
utils::string::to_upper(map->name).data());
|
|
localized_strings::override(str, str);
|
|
}
|
|
}
|
|
}
|
|
|
|
void enum_assets(const game::XAssetType type, const std::function<void(game::XAssetHeader)>& callback, const bool includeOverride)
|
|
{
|
|
game::DB_EnumXAssets_Internal(type, static_cast<void(*)(game::XAssetHeader, void*)>([](game::XAssetHeader header, void* data)
|
|
{
|
|
const auto& cb = *static_cast<const std::function<void(game::XAssetHeader)>*>(data);
|
|
cb(header);
|
|
}), &callback, includeOverride);
|
|
}
|
|
|
|
std::string get_current_fastfile()
|
|
{
|
|
std::string fastfile_copy;
|
|
return current_fastfile.access<std::string>([&](std::string& fastfile)
|
|
{
|
|
return fastfile;
|
|
});
|
|
}
|
|
|
|
class component final : public component_interface
|
|
{
|
|
public:
|
|
void post_unpack() override
|
|
{
|
|
db_try_load_x_file_internal_hook.create(0x1404173B0, db_try_load_x_file_internal);
|
|
db_find_xasset_header.create(game::DB_FindXAssetHeader, db_find_xasset_header_stub);
|
|
|
|
add_missing_localized_strings();
|
|
|
|
command::add("loadzone", [](const command::params& params)
|
|
{
|
|
if (params.size() < 2)
|
|
{
|
|
console::info("usage: loadzone <zone>\n");
|
|
return;
|
|
}
|
|
|
|
game::XZoneInfo info{};
|
|
info.name = params.get(1);
|
|
info.allocFlags = 1;
|
|
info.freeFlags = 0;
|
|
game::DB_LoadXAssets(&info, 1u, game::DBSyncMode::DB_LOAD_SYNC);
|
|
});
|
|
|
|
command::add("g_poolSizes", []()
|
|
{
|
|
for (auto i = 0; i < game::ASSET_TYPE_COUNT; i++)
|
|
{
|
|
console::info("g_poolSize[%i]: %i // %s\n", i, game::g_poolSize[i], game::g_assetNames[i]);
|
|
}
|
|
});
|
|
}
|
|
};
|
|
}
|
|
|
|
REGISTER_COMPONENT(fastfiles::component)
|