Rim/iw4x-client
Rim/iw4x-client
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

[AssetHandler] Use thread_local declaration instead of mutex

Browse Source
This commit is contained in:
momo5502 2016-12-21 17:08:15 +01:00
parent caa9787507
commit fac7f0ef4c
2 changed files with 7 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/Components/Modules/AssetHandler.cpp
View File

@ -2,8 +2,7 @@
namespace Components
{
std::recursive_mutex AssetHandler::BypassMutex;
std::vector<std::thread::id> AssetHandler::BypassThreads;
thread_local bool AssetHandler::BypassState;
std::map<Game::XAssetType, AssetHandler::IAsset*> AssetHandler::AssetInterfaces;
std::map<Game::XAssetType, wink::slot<AssetHandler::Callback>> AssetHandler::TypeCallbacks;
wink::signal<wink::slot<AssetHandler::RestrictCallback>> AssetHandler::RestrictSignal;
@ -56,7 +55,7 @@ namespace Components
if (filename)
{
// Allow call DB_FindXAssetHeader within the hook
AssetHandler::SetThreadBypass();
AssetHandler::BypassState = true;
if (AssetHandler::TypeCallbacks.find(type) != AssetHandler::TypeCallbacks.end())
{
@ -64,7 +63,7 @@ namespace Components
}
// Disallow calling DB_FindXAssetHeader ;)
AssetHandler::ClearThreadBypass();
AssetHandler::BypassState = false;
}
return header;
@ -72,33 +71,7 @@ namespace Components
int AssetHandler::HasThreadBypass()
{
std::lock_guard<std::recursive_mutex> _(AssetHandler::BypassMutex);
return (std::find(AssetHandler::BypassThreads.begin(), AssetHandler::BypassThreads.end(), std::this_thread::get_id()) != AssetHandler::BypassThreads.end()) & 1;
}
void AssetHandler::SetThreadBypass()
{
std::lock_guard<std::recursive_mutex> _(AssetHandler::BypassMutex);
if (!AssetHandler::HasThreadBypass())
{
AssetHandler::BypassThreads.push_back(std::this_thread::get_id());
}
}
void AssetHandler::ClearThreadBypass()
{
std::lock_guard<std::recursive_mutex> _(AssetHandler::BypassMutex);
while (AssetHandler::HasThreadBypass())
{
auto i = std::find(AssetHandler::BypassThreads.begin(), AssetHandler::BypassThreads.end(), std::this_thread::get_id());
if (i != AssetHandler::BypassThreads.end())
{
AssetHandler::BypassThreads.erase(i);
}
}
return AssetHandler::BypassState & 1;
}
__declspec(naked) void AssetHandler::FindAssetStub()
@ -339,13 +312,11 @@ namespace Components
Game::XAssetHeader AssetHandler::FindOriginalAsset(Game::XAssetType type, const char* filename)
{
std::lock_guard<std::recursive_mutex> _(AssetHandler::BypassMutex);
int originalState = AssetHandler::HasThreadBypass();
AssetHandler::SetThreadBypass();
AssetHandler::BypassState = true;
Game::XAssetHeader header = Game::DB_FindXAssetHeader(type, filename);
if (!originalState) AssetHandler::ClearThreadBypass();
if (!originalState) AssetHandler::BypassState = false;
return header;
}

5
src/Components/Modules/AssetHandler.hpp
View File

@ -40,8 +40,7 @@ namespace Components
static void StoreTemporaryAsset(Game::XAssetType type, Game::XAssetHeader asset);
private:
static std::recursive_mutex BypassMutex;
static std::vector<std::thread::id> BypassThreads;
static thread_local bool BypassState;
static std::map<std::string, Game::XAssetHeader> TemporaryAssets[Game::XAssetType::ASSET_TYPE_COUNT];
@ -68,8 +67,6 @@ namespace Components
static void ModifyAsset(Game::XAssetType type, Game::XAssetHeader asset, std::string name);
static int HasThreadBypass();
static void SetThreadBypass();
static void ClearThreadBypass();
};
}