mirror of
https://github.com/Laupetin/OpenAssetTools.git
synced 2026-05-16 23:11:42 +00:00
chore: make DependencyAssetLookup a separate class
This commit is contained in:
@@ -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 DependencyAssetLookup& dependencyAssets) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,7 +17,7 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
if (!filter.ShouldWriteAsset(*asset))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace IW3
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) 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 DependencyAssetLookup& dependencyAssets) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,7 +17,7 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
if (!filter.ShouldWriteAsset(*asset))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace IW4
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) 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 DependencyAssetLookup& dependencyAssets) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,7 +17,7 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
if (!filter.ShouldWriteAsset(*asset))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace IW5
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) 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 DependencyAssetLookup& dependencyAssets) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -17,7 +17,7 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
if (!filter.ShouldWriteAsset(*asset))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace T5
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const override;
|
||||
};
|
||||
} // namespace T5
|
||||
|
||||
@@ -62,7 +62,7 @@ void ZoneDefWriter::WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone
|
||||
}
|
||||
}
|
||||
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const
|
||||
void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const
|
||||
{
|
||||
const auto* game = IGame::GetGameById(zone.m_game_id);
|
||||
|
||||
@@ -73,7 +73,7 @@ void ZoneDefWriter::WriteContent(ZoneDefinitionOutputStream& stream, const Zone&
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
if (!ShouldWriteAsset(*asset, dependencyAssets))
|
||||
if (!filter.ShouldWriteAsset(*asset))
|
||||
continue;
|
||||
|
||||
switch (asset->m_type)
|
||||
|
||||
@@ -8,6 +8,6 @@ namespace T6
|
||||
{
|
||||
protected:
|
||||
void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const DependencyAssetLookup& dependencyAssets) const override;
|
||||
void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const override;
|
||||
};
|
||||
} // namespace T6
|
||||
|
||||
@@ -9,6 +9,38 @@
|
||||
|
||||
#include <cassert>
|
||||
|
||||
namespace
|
||||
{
|
||||
constexpr auto META_DATA_KEY_GAME = "game";
|
||||
constexpr auto META_DATA_KEY_GDT = "gdt";
|
||||
} // namespace
|
||||
|
||||
ZoneDefFilter ZoneDefFilter::AllEntries()
|
||||
{
|
||||
return ZoneDefFilter();
|
||||
}
|
||||
|
||||
ZoneDefFilter ZoneDefFilter::WithOnlyRootAssets(const Zone& zone)
|
||||
{
|
||||
ZoneDefFilter result;
|
||||
|
||||
for (const auto& asset : zone.m_pools)
|
||||
{
|
||||
for (const auto* dependency : asset->m_dependencies)
|
||||
{
|
||||
if (dependency != nullptr && dependency->m_zone == &zone)
|
||||
result.m_non_root_asset_lookup.emplace(dependency);
|
||||
}
|
||||
}
|
||||
|
||||
return std::move(result);
|
||||
}
|
||||
|
||||
bool ZoneDefFilter::ShouldWriteAsset(const XAssetInfoGeneric& asset) const
|
||||
{
|
||||
return m_non_root_asset_lookup.empty() || !m_non_root_asset_lookup.contains(&asset);
|
||||
}
|
||||
|
||||
const IZoneDefWriter* IZoneDefWriter::GetZoneDefWriterForGame(GameId game)
|
||||
{
|
||||
static const IZoneDefWriter* zoneDefWriters[static_cast<unsigned>(GameId::COUNT)]{
|
||||
@@ -26,30 +58,6 @@ const IZoneDefWriter* IZoneDefWriter::GetZoneDefWriterForGame(GameId game)
|
||||
return result;
|
||||
}
|
||||
|
||||
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);
|
||||
@@ -68,6 +76,6 @@ void AbstractZoneDefWriter::WriteZoneDef(std::ostream& stream, const Zone& zone,
|
||||
|
||||
WriteMetaData(out, zone);
|
||||
|
||||
const auto dependencyAssets = CreateDependencyAssetLookup(zone, minimalZone);
|
||||
WriteContent(out, zone, dependencyAssets);
|
||||
const auto filter = minimalZone ? ZoneDefFilter::WithOnlyRootAssets(zone) : ZoneDefFilter::AllEntries();
|
||||
WriteContent(out, zone, filter);
|
||||
}
|
||||
|
||||
@@ -1,11 +1,24 @@
|
||||
#pragma once
|
||||
|
||||
#include "Pool/XAssetInfo.h"
|
||||
#include "Zone/Definition/ZoneDefinitionStream.h"
|
||||
#include "Zone/Zone.h"
|
||||
|
||||
#include <unordered_set>
|
||||
|
||||
class XAssetInfoGeneric;
|
||||
class ZoneDefFilter
|
||||
{
|
||||
public:
|
||||
static ZoneDefFilter AllEntries();
|
||||
static ZoneDefFilter WithOnlyRootAssets(const Zone& zone);
|
||||
|
||||
[[nodiscard]] bool ShouldWriteAsset(const XAssetInfoGeneric& asset) const;
|
||||
|
||||
private:
|
||||
ZoneDefFilter() = default;
|
||||
|
||||
std::unordered_set<const XAssetInfoGeneric*> m_non_root_asset_lookup;
|
||||
};
|
||||
|
||||
class IZoneDefWriter
|
||||
{
|
||||
@@ -24,18 +37,10 @@ public:
|
||||
|
||||
class AbstractZoneDefWriter : public IZoneDefWriter
|
||||
{
|
||||
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 DependencyAssetLookup& dependencyAssets) const = 0;
|
||||
|
||||
public:
|
||||
void WriteZoneDef(std::ostream& stream, const Zone& zone, bool useGdt, bool minimalZone) const override;
|
||||
|
||||
protected:
|
||||
virtual void WriteMetaData(ZoneDefinitionOutputStream& stream, const Zone& zone) const = 0;
|
||||
virtual void WriteContent(ZoneDefinitionOutputStream& stream, const Zone& zone, const ZoneDefFilter& filter) const = 0;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user