mirror of
https://github.com/Laupetin/OpenAssetTools.git
synced 2026-05-16 23:11:42 +00:00
fix(unlinker): respect minimal zone definition output
This commit is contained in:
@@ -67,7 +67,7 @@ namespace
|
||||
}
|
||||
|
||||
const auto* zoneDefWriter = IZoneDefWriter::GetZoneDefWriterForGame(zone.m_game_id);
|
||||
zoneDefWriter->WriteZoneDef(zoneDefinitionFile, zone, m_args.m_use_gdt);
|
||||
zoneDefWriter->WriteZoneDef(zoneDefinitionFile, zone, m_args.m_use_gdt, m_args.m_minimal_zone_def);
|
||||
|
||||
zoneDefinitionFile.close();
|
||||
|
||||
|
||||
@@ -6,7 +6,7 @@ using namespace IW3;
|
||||
|
||||
void ZoneDefWriter::WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const {}
|
||||
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,6 +17,9 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
{
|
||||
case ASSET_TYPE_LOCALIZE_ENTRY:
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace IW3
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
};
|
||||
} // namespace IW3
|
||||
|
||||
@@ -6,7 +6,7 @@ using namespace IW4;
|
||||
|
||||
void ZoneDefWriter::WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const {}
|
||||
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,6 +17,9 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
{
|
||||
case ASSET_TYPE_LOCALIZE_ENTRY:
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace IW4
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
};
|
||||
} // namespace IW4
|
||||
|
||||
@@ -6,7 +6,7 @@ using namespace IW5;
|
||||
|
||||
void ZoneDefWriter::WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const {}
|
||||
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,6 +17,9 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
{
|
||||
case ASSET_TYPE_LOCALIZE_ENTRY:
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace IW5
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
};
|
||||
} // namespace IW5
|
||||
|
||||
@@ -6,7 +6,7 @@ using namespace T5;
|
||||
|
||||
void ZoneDefWriter::WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const {}
|
||||
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,6 +17,9 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
{
|
||||
case ASSET_TYPE_LOCALIZE_ENTRY:
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace T5
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
};
|
||||
} // namespace T5
|
||||
|
||||
@@ -62,7 +62,7 @@ void ZoneDefWriter::WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone
|
||||
}
|
||||
}
|
||||
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -73,6 +73,9 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
{
|
||||
case ASSET_TYPE_LOCALIZE_ENTRY:
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace T6
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
};
|
||||
} // namespace T6
|
||||
|
||||
@@ -5,6 +5,7 @@
|
||||
#include "Game/IW5/Zone/Definition/ZoneDefWriterIW5.h"
|
||||
#include "Game/T5/Zone/Definition/ZoneDefWriterT5.h"
|
||||
#include "Game/T6/Zone/Definition/ZoneDefWriterT6.h"
|
||||
#include "Pool/XAssetInfo.h"
|
||||
|
||||
#include <cassert>
|
||||
|
||||
@@ -25,7 +26,31 @@ const IZoneDefWriter* IZoneDefWriter::GetZoneDefWriterForGame(GameId game)
|
||||
return result;
|
||||
}
|
||||
|
||||
void AbstractZoneDefWriter::WriteZoneDef(std::ostream& stream, const Zone& zone, const bool useGdt) const
|
||||
AbstractZoneDefWriter::DependencyAssetLookup AbstractZoneDefWriter::CreateDependencyAssetLookup(const Zone& zone, const bool minimalZone)
|
||||
{
|
||||
DependencyAssetLookup dependencyAssets;
|
||||
|
||||
if (!minimalZone)
|
||||
return dependencyAssets;
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
for (const auto* dependency : asset->m_dependencies)
|
||||
{
|
||||
if (dependency != nullptr && dependency->m_zone == &zone)
|
||||
dependencyAssets.emplace(dependency);
|
||||
}
|
||||
}
|
||||
|
||||
return dependencyAssets;
|
||||
}
|
||||
|
||||
bool AbstractZoneDefWriter::ShouldWriteAsset(const XAssetInfoGeneric& asset, const DependencyAssetLookup& dependencyAssets)
|
||||
{
|
||||
return dependencyAssets.empty() || !dependencyAssets.contains(&asset);
|
||||
}
|
||||
|
||||
void AbstractZoneDefWriter::WriteZoneDef(std::ostream& stream, const Zone& zone, const bool useGdt, const bool minimalZone) const
|
||||
{
|
||||
ZoneDefinitionOutputStream out(stream);
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
@@ -42,5 +67,7 @@ void AbstractZoneDefWriter::WriteZoneDef(std::ostream& stream, const Zone& zone,
|
||||
}
|
||||
|
||||
WriteMetaData(out, zone);
|
||||
WriteContent(out, zone);
|
||||
|
||||
const auto dependencyAssets = CreateDependencyAssetLookup(zone, minimalZone);
|
||||
WriteContent(out, zone, dependencyAssets);
|
||||
}
|
||||
|
||||
@@ -3,6 +3,10 @@
|
||||
#include "Zone/Definition/ZoneDefinitionStream.h"
|
||||
#include "Zone/Zone.h"
|
||||
|
||||
#include <unordered_set>
|
||||
|
||||
class XAssetInfoGeneric;
|
||||
|
||||
class IZoneDefWriter
|
||||
{
|
||||
public:
|
||||
@@ -13,7 +17,7 @@ public:
|
||||
IZoneDefWriter& operator=(const IZoneDefWriter& other) = default;
|
||||
IZoneDefWriter& operator=(IZoneDefWriter&& other) noexcept = default;
|
||||
|
||||
virtual void WriteZoneDef(std::ostream& stream, const Zone& zone, bool useGdt) const = 0;
|
||||
virtual void WriteZoneDef(std::ostream& stream, const Zone& zone, bool useGdt, bool minimalZone) const = 0;
|
||||
|
||||
static const IZoneDefWriter* GetZoneDefWriterForGame(GameId game);
|
||||
};
|
||||
@@ -24,9 +28,14 @@ protected:
|
||||
static constexpr auto META_DATA_KEY_GAME = "game";
|
||||
static constexpr auto META_DATA_KEY_GDT = "gdt";
|
||||
|
||||
using DependencyAssetLookup = std::unordered_set<const XAssetInfoGeneric*>;
|
||||
|
||||
static DependencyAssetLookup CreateDependencyAssetLookup(const Zone& zone, bool minimalZone);
|
||||
static bool ShouldWriteAsset(const XAssetInfoGeneric& asset, const DependencyAssetLookup& dependencyAssets);
|
||||
|
||||
virtual void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const = 0;
|
||||
virtual void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone) const = 0;
|
||||
virtual void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const = 0;
|
||||
|
||||
public:
|
||||
void WriteZoneDef(std::ostream& stream, const Zone& zone, bool useGdt) const override;
|
||||
void WriteZoneDef(std::ostream& stream, const Zone& zone, bool useGdt, bool minimalZone) const override;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user