2017-01-18 17:44:36 -05:00
|
|
|
#include "STDInclude.hpp"
|
|
|
|
|
|
|
|
namespace Components
|
|
|
|
{
|
2017-01-19 15:21:59 -05:00
|
|
|
std::recursive_mutex Localization::LocalizeMutex;
|
2017-01-18 17:44:36 -05:00
|
|
|
Dvar::Var Localization::UseLocalization;
|
|
|
|
Utils::Memory::Allocator Localization::MemAllocator;
|
|
|
|
std::unordered_map<std::string, Game::LocalizeEntry*> Localization::LocalizeMap;
|
|
|
|
std::unordered_map<std::string, Game::LocalizeEntry*> Localization::TempLocalizeMap;
|
|
|
|
|
|
|
|
void Localization::Set(std::string key, std::string value)
|
|
|
|
{
|
2017-01-19 15:21:59 -05:00
|
|
|
std::lock_guard<std::recursive_mutex> _(Localization::LocalizeMutex);
|
2017-01-18 17:44:36 -05:00
|
|
|
|
|
|
|
if (Localization::LocalizeMap.find(key) != Localization::LocalizeMap.end())
|
|
|
|
{
|
|
|
|
Game::LocalizeEntry* entry = Localization::LocalizeMap[key];
|
|
|
|
|
|
|
|
char* newStaticValue = Localization::MemAllocator.duplicateString(value);
|
|
|
|
if (!newStaticValue) return;
|
|
|
|
if (entry->value) Localization::MemAllocator.free(entry->value);
|
|
|
|
entry->value = newStaticValue;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
Game::LocalizeEntry* entry = Localization::MemAllocator.allocate<Game::LocalizeEntry>();
|
|
|
|
if (!entry) return;
|
|
|
|
|
|
|
|
entry->name = Localization::MemAllocator.duplicateString(key);
|
|
|
|
if (!entry->name)
|
|
|
|
{
|
|
|
|
Localization::MemAllocator.free(entry);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
entry->value = Localization::MemAllocator.duplicateString(value);
|
|
|
|
if (!entry->value)
|
|
|
|
{
|
|
|
|
Localization::MemAllocator.free(entry->name);
|
|
|
|
Localization::MemAllocator.free(entry);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
Localization::LocalizeMap[key] = entry;
|
|
|
|
}
|
|
|
|
|
|
|
|
const char* Localization::Get(const char* key)
|
|
|
|
{
|
|
|
|
if (!Localization::UseLocalization.get<bool>()) return key;
|
|
|
|
|
|
|
|
Game::LocalizeEntry* entry = nullptr;
|
2017-01-19 15:21:59 -05:00
|
|
|
std::lock_guard<std::recursive_mutex> _(Localization::LocalizeMutex);
|
2017-01-18 17:44:36 -05:00
|
|
|
|
|
|
|
if (Localization::TempLocalizeMap.find(key) != Localization::TempLocalizeMap.end())
|
|
|
|
{
|
|
|
|
entry = Localization::TempLocalizeMap[key];
|
|
|
|
}
|
|
|
|
else if (Localization::LocalizeMap.find(key) != Localization::LocalizeMap.end())
|
|
|
|
{
|
|
|
|
entry = Localization::LocalizeMap[key];
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!entry || !entry->value)
|
|
|
|
{
|
|
|
|
entry = Game::DB_FindXAssetHeader(Game::XAssetType::ASSET_TYPE_LOCALIZE_ENTRY, key).localize;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (entry && entry->value)
|
|
|
|
{
|
|
|
|
return entry->value;
|
|
|
|
}
|
|
|
|
|
|
|
|
return key;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Localization::SetTemp(std::string key, std::string value)
|
|
|
|
{
|
2017-01-19 15:21:59 -05:00
|
|
|
std::lock_guard<std::recursive_mutex> _(Localization::LocalizeMutex);
|
2017-01-18 17:44:36 -05:00
|
|
|
|
|
|
|
if (Localization::TempLocalizeMap.find(key) != Localization::TempLocalizeMap.end())
|
|
|
|
{
|
|
|
|
Game::LocalizeEntry* entry = Localization::TempLocalizeMap[key];
|
|
|
|
if(entry->value) Localization::MemAllocator.free(entry->value);
|
|
|
|
entry->value = Localization::MemAllocator.duplicateString(value);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
Game::LocalizeEntry* entry = Localization::MemAllocator.allocate<Game::LocalizeEntry>();
|
|
|
|
if (!entry) return;
|
|
|
|
|
|
|
|
entry->name = Localization::MemAllocator.duplicateString(key);
|
|
|
|
if (!entry->name)
|
|
|
|
{
|
|
|
|
Localization::MemAllocator.free(entry);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
entry->value = Localization::MemAllocator.duplicateString(value);
|
|
|
|
if (!entry->value)
|
|
|
|
{
|
|
|
|
Localization::MemAllocator.free(entry->name);
|
|
|
|
Localization::MemAllocator.free(entry);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
Localization::TempLocalizeMap[key] = entry;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Localization::ClearTemp()
|
|
|
|
{
|
2017-01-19 15:21:59 -05:00
|
|
|
std::lock_guard<std::recursive_mutex> _(Localization::LocalizeMutex);
|
2017-01-18 17:44:36 -05:00
|
|
|
|
|
|
|
for (auto i = Localization::TempLocalizeMap.begin(); i != Localization::TempLocalizeMap.end(); ++i)
|
|
|
|
{
|
|
|
|
if (i->second)
|
|
|
|
{
|
|
|
|
if (i->second->name) Localization::MemAllocator.free(i->second->name);
|
|
|
|
if (i->second->value) Localization::MemAllocator.free(i->second->value);
|
|
|
|
Localization::MemAllocator.free(i->second);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
Localization::TempLocalizeMap.clear();
|
|
|
|
}
|
|
|
|
|
|
|
|
void __stdcall Localization::SetStringStub(const char* key, const char* value, bool /*isEnglish*/)
|
|
|
|
{
|
|
|
|
Localization::Set(key, value);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Localization::LoadLanguageStrings()
|
|
|
|
{
|
|
|
|
//if (ZoneBuilder::IsEnabled())
|
|
|
|
{
|
|
|
|
if (FileSystem::File(Utils::String::VA("localizedstrings/iw4x_%s.str", Game::Win_GetLanguage())).exists())
|
|
|
|
{
|
|
|
|
Game::SE_Load(Utils::String::VA("localizedstrings/iw4x_%s.str", Game::Win_GetLanguage()), 0);
|
|
|
|
}
|
|
|
|
else if (FileSystem::File("localizedstrings/iw4x_english.str").exists())
|
|
|
|
{
|
|
|
|
Game::SE_Load("localizedstrings/iw4x_english.str", 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
__declspec(naked) void Localization::SELoadLanguageStub()
|
|
|
|
{
|
|
|
|
__asm
|
|
|
|
{
|
|
|
|
pushad
|
|
|
|
call Localization::LoadLanguageStrings
|
|
|
|
popad
|
|
|
|
|
|
|
|
push 629E20h
|
|
|
|
retn
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
Localization::Localization()
|
|
|
|
{
|
|
|
|
AssetHandler::OnFind(Game::XAssetType::ASSET_TYPE_LOCALIZE_ENTRY, [] (Game::XAssetType, std::string filename)
|
|
|
|
{
|
|
|
|
Game::XAssetHeader header = { 0 };
|
2017-01-19 15:21:59 -05:00
|
|
|
std::lock_guard<std::recursive_mutex> _(Localization::LocalizeMutex);
|
2017-01-18 17:44:36 -05:00
|
|
|
|
|
|
|
if (Localization::TempLocalizeMap.find(filename) != Localization::TempLocalizeMap.end())
|
|
|
|
{
|
|
|
|
header.localize = Localization::TempLocalizeMap[filename];
|
|
|
|
}
|
|
|
|
else if (Localization::LocalizeMap.find(filename) != Localization::LocalizeMap.end())
|
|
|
|
{
|
|
|
|
header.localize = Localization::LocalizeMap[filename];
|
|
|
|
}
|
|
|
|
|
|
|
|
return header;
|
|
|
|
});
|
|
|
|
|
|
|
|
// Resolving hook
|
|
|
|
Utils::Hook(0x629B90, Localization::Get, HOOK_JUMP).install()->quick();
|
|
|
|
|
|
|
|
// Set loading entry point
|
|
|
|
Utils::Hook(0x41D859, Localization::SELoadLanguageStub, HOOK_CALL).install()->quick();
|
|
|
|
|
|
|
|
// Overwrite SetString
|
|
|
|
Utils::Hook(0x4CE5EE, Localization::SetStringStub, HOOK_CALL).install()->quick();
|
|
|
|
|
|
|
|
Localization::UseLocalization = Dvar::Register<bool>("ui_localize", true, Game::dvar_flag::DVAR_FLAG_NONE, "Use localization strings");
|
|
|
|
}
|
|
|
|
|
|
|
|
Localization::~Localization()
|
|
|
|
{
|
|
|
|
Localization::ClearTemp();
|
|
|
|
|
|
|
|
Localization::LocalizeMap.clear();
|
|
|
|
Localization::MemAllocator.clear();
|
|
|
|
}
|
|
|
|
}
|