chore: make DependencyAssetLookup a separate class

src/ZoneCommon/Game/IW3/Zone/Definition/ZoneDefWriterIW3.cpp

@@ -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)

src/ZoneCommon/Game/IW3/Zone/Definition/ZoneDefWriterIW3.h

@@ -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

src/ZoneCommon/Game/IW4/Zone/Definition/ZoneDefWriterIW4.cpp

@@ -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)

src/ZoneCommon/Game/IW4/Zone/Definition/ZoneDefWriterIW4.h

@@ -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

src/ZoneCommon/Game/IW5/Zone/Definition/ZoneDefWriterIW5.cpp

@@ -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)

src/ZoneCommon/Game/IW5/Zone/Definition/ZoneDefWriterIW5.h

@@ -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

src/ZoneCommon/Game/T5/Zone/Definition/ZoneDefWriterT5.cpp

@@ -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)

src/ZoneCommon/Game/T5/Zone/Definition/ZoneDefWriterT5.h

@@ -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

src/ZoneCommon/Game/T6/Zone/Definition/ZoneDefWriterT6.cpp

@@ -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)

src/ZoneCommon/Game/T6/Zone/Definition/ZoneDefWriterT6.h

@@ -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

src/ZoneCommon/Zone/Definition/ZoneDefWriter.cpp

@@ -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);
+    const auto filter = minimalZone ? ZoneDefFilter::WithOnlyRootAssets(zone) : ZoneDefFilter::AllEntries();
-    WriteContent(out, zone, dependencyAssets);
+    WriteContent(out, zone, filter);
 }

src/ZoneCommon/Zone/Definition/ZoneDefWriter.h

@@ -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;
 };