INSANEMODE/OpenAssetTools
1
0
Fork 0
mirror of https://github.com/Laupetin/OpenAssetTools.git synced 2025-06-11 07:18:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add load argument to unlinker to load zones before trying to unlink specified zones

Browse Source 
this allows to make sure certain ipaks are loaded before dumping
This commit is contained in:
Jan
2021-03-27 16:45:30 +01:00
parent 7d4029b21f
commit 4f995751ce
10 changed files with 158 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/ZoneLoading/Loading/ZoneLoader.cpp
View File

@ -1,26 +1,27 @@
#include "ZoneLoader.h"
#include "Exception/LoadingException.h"
#include "LoadingFileStream.h"
#include <algorithm>
ZoneLoader::ZoneLoader(Zone* zone)
#include "Exception/LoadingException.h"
#include "LoadingFileStream.h"
ZoneLoader::ZoneLoader(std::unique_ptr<Zone> zone)
: m_processor_chain_dirty(false),
m_zone(std::move(zone))
{
m_zone = zone;
m_processor_chain_dirty = false;
}
ILoadingStream* ZoneLoader::BuildLoadingChain(ILoadingStream* rootStream)
{
auto* currentStream = rootStream;
for(const auto& processor : m_processors)
for (const auto& processor : m_processors)
{
processor->SetBaseStream(currentStream);
currentStream = processor.get();
}
m_processor_chain_dirty = false;
return currentStream;
}
@ -50,9 +51,9 @@ void ZoneLoader::AddStreamProcessor(std::unique_ptr<StreamProcessor> streamProce
void ZoneLoader::RemoveStreamProcessor(StreamProcessor* streamProcessor)
{
for(auto i = m_processors.begin(); i < m_processors.end(); ++i)
for (auto i = m_processors.begin(); i < m_processors.end(); ++i)
{
if(i->get() == streamProcessor)
if (i->get() == streamProcessor)
{
m_processors.erase(i);
m_processor_chain_dirty = true;
@ -61,18 +62,18 @@ void ZoneLoader::RemoveStreamProcessor(StreamProcessor* streamProcessor)
}
}
Zone* ZoneLoader::LoadZone(std::istream& stream)
std::unique_ptr<Zone> ZoneLoader::LoadZone(std::istream& stream)
{
LoadingFileStream fileStream(stream);
auto* endStream = BuildLoadingChain(&fileStream);
try
{
for(const auto& step : m_steps)
for (const auto& step : m_steps)
{
step->PerformStep(this, endStream);
if(m_processor_chain_dirty)
if (m_processor_chain_dirty)
{
endStream = BuildLoadingChain(&fileStream);
}
@ -83,12 +84,10 @@ Zone* ZoneLoader::LoadZone(std::istream& stream)
const auto detailedMessage = e.DetailedMessage();
printf("Loading fastfile failed: %s\n", detailedMessage.c_str());
delete m_zone;
return nullptr;
}
m_zone->Register();
return m_zone;
return std::move(m_zone);
}

7
src/ZoneLoading/Loading/ZoneLoader.h
View File

@ -7,6 +7,7 @@
#include <vector>
#include <istream>
#include <memory>
class ILoadingStep;
@ -17,14 +18,14 @@ class ZoneLoader
bool m_processor_chain_dirty;
Zone* m_zone;
std::unique_ptr<Zone> m_zone;
ILoadingStream* BuildLoadingChain(ILoadingStream* rootStream);
public:
std::vector<XBlock*> m_blocks;
explicit ZoneLoader(Zone* zone);
explicit ZoneLoader(std::unique_ptr<Zone> zone);
void AddXBlock(std::unique_ptr<XBlock> block);
void AddLoadingStep(std::unique_ptr<ILoadingStep> step);
@ -32,5 +33,5 @@ public:
void RemoveStreamProcessor(StreamProcessor* streamProcessor);
Zone* LoadZone(std::istream& stream);
std::unique_ptr<Zone> LoadZone(std::istream& stream);
};