118 lines
3.0 KiB
C++
118 lines
3.0 KiB
C++
#include "CTweakManager.h"
|
|
#include "Core/GameProject/CGameProject.h"
|
|
#include "Core/GameProject/CResourceIterator.h"
|
|
#include "Core/Tweaks/CTweakCooker.h"
|
|
#include "Core/Tweaks/CTweakLoader.h"
|
|
|
|
CTweakManager::CTweakManager(CGameProject* pInProject)
|
|
: mpProject(pInProject)
|
|
{
|
|
}
|
|
|
|
CTweakManager::~CTweakManager()
|
|
{
|
|
ClearTweaks();
|
|
}
|
|
|
|
void CTweakManager::LoadTweaks()
|
|
{
|
|
ASSERT(mTweakObjects.empty());
|
|
|
|
// MP1 - Load all tweak assets into memory
|
|
if (mpProject->Game() <= EGame::Prime)
|
|
{
|
|
for (TResourceIterator<EResourceType::Tweaks> It(mpProject->ResourceStore()); It; ++It)
|
|
{
|
|
if (CTweakData* pTweaks = (CTweakData*)It->Load())
|
|
{
|
|
pTweaks->Lock();
|
|
mTweakObjects.push_back(pTweaks);
|
|
}
|
|
}
|
|
}
|
|
|
|
// MP2+ - Load tweaks from Standard.ntwk
|
|
else
|
|
{
|
|
if (!mpProject->IsWiiBuild())
|
|
{
|
|
mStandardFilePath = mpProject->DiscFilesystemRoot(false) / "Standard.ntwk";
|
|
}
|
|
else
|
|
{
|
|
// For Wii builds, there is another game-dependent subfolder.
|
|
EGame Game = mpProject->Game();
|
|
TString GameName = (Game == EGame::Prime ? "MP1" : Game == EGame::Echoes ? "MP2" : "MP3");
|
|
mStandardFilePath = mpProject->DiscFilesystemRoot(false) / GameName / "Standard.ntwk";
|
|
|
|
// MP3 might actually be FrontEnd
|
|
if (Game == EGame::Corruption && !FileUtil::Exists(mStandardFilePath))
|
|
{
|
|
mStandardFilePath = mpProject->DiscFilesystemRoot(false) / "fe/Standard.ntwk";
|
|
}
|
|
}
|
|
|
|
if (FileUtil::Exists(mStandardFilePath))
|
|
{
|
|
CFileInStream StandardNTWK(mStandardFilePath, EEndian::BigEndian);
|
|
CTweakLoader::LoadNTWK(StandardNTWK, mpProject->Game(), mTweakObjects);
|
|
}
|
|
}
|
|
}
|
|
|
|
bool CTweakManager::SaveTweaks()
|
|
{
|
|
// If we don't have any tweaks loaded, nothing to do
|
|
if (mTweakObjects.empty())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// MP1 - Save all tweak assets
|
|
if (mpProject->Game() <= EGame::Prime)
|
|
{
|
|
bool SavedAll = true, SavedAny = false;
|
|
|
|
for (CTweakData* pTweakData : mTweakObjects)
|
|
{
|
|
if (!pTweakData->Entry()->Save(true))
|
|
{
|
|
SavedAll = false;
|
|
}
|
|
else
|
|
{
|
|
SavedAny = true;
|
|
}
|
|
}
|
|
|
|
if (SavedAny)
|
|
{
|
|
mpProject->ResourceStore()->ConditionalSaveStore();
|
|
}
|
|
|
|
return SavedAll;
|
|
}
|
|
// MP2+ - Save tweaks to Standard.ntwk
|
|
else
|
|
{
|
|
CFileOutStream StandardNTWK(mStandardFilePath, EEndian::BigEndian);
|
|
return CTweakCooker::CookNTWK(mTweakObjects, StandardNTWK);
|
|
}
|
|
}
|
|
|
|
void CTweakManager::ClearTweaks()
|
|
{
|
|
for (CTweakData* pTweakData : mTweakObjects)
|
|
{
|
|
if (pTweakData->Entry() != nullptr)
|
|
{
|
|
pTweakData->Release();
|
|
}
|
|
else
|
|
{
|
|
delete pTweakData;
|
|
}
|
|
}
|
|
mTweakObjects.clear();
|
|
}
|