refactor AI system

2026-06-15 09:16:56 +02:00
parent 8451f5a281
commit e8dd73bcb0
67 changed files with 1731 additions and 919 deletions
--- a/src/lib/ecs/system/SalvagerSystem.cpp
+++ b/src/lib/ecs/system/SalvagerSystem.cpp
@@ -0,0 +1,79 @@
+#include "SalvagerSystem.h"
+
+#include <vector>
+
+#include <QVector2D>
+
+#include "Building.h"
+#include "BuildingSystem.h"
+#include "DeliverScrapBehavior.h"
+#include "EntityAdmin.h"
+#include "ModuleOwnerComponent.h"
+#include "PositionComponent.h"
+#include "SalvageCargoComponent.h"
+#include "ScrapSystem.h"
+#include "tracing.h"
+
+SalvagerSystem::SalvagerSystem(EntityAdmin& admin)
+    : m_admin(admin)
+{
+}
+
+void SalvagerSystem::tick(ScrapSystem& scraps, BuildingSystem& buildings)
+{
+    TRACE();
+    const std::vector<ScrapInfo> allScrap = scraps.allScrapInfo();
+
+    // Tick down per-module collection cooldowns.
+    m_admin.forEach<SalvageCargoComponent>(
+        [](entt::entity /*e*/, SalvageCargoComponent& c)
+        {
+            if (c.cooldownTicksRemaining > 0) { --c.cooldownTicksRemaining; }
+        });
+
+    // Collection: each ready, in-range module collects one scrap.
+    m_admin.forEach<SalvageCargoComponent, ModuleOwnerComponent>(
+        [&](entt::entity /*ce*/, SalvageCargoComponent& c, const ModuleOwnerComponent& o)
+        {
+            if (c.current >= c.capacity || c.cooldownTicksRemaining > 0) { return; }
+            if (!m_admin.hasAll<PositionComponent>(o.owner)) { return; }
+
+            const QVector2D ownerPos = m_admin.get<PositionComponent>(o.owner).value;
+            for (const ScrapInfo& si : allScrap)
+            {
+                if ((si.position - ownerPos).length() > c.collectionRange_tiles) { continue; }
+                if (scraps.consume(si.entity))
+                {
+                    ++c.current;
+                    c.cooldownTicksRemaining = c.collectionIntervalTicks;
+                    break;
+                }
+            }
+        });
+
+    // Delivery: a ship at its assigned bay hands over one unit of cargo per tick.
+    m_admin.forEach<DeliverScrapBehavior, PositionComponent>(
+        [&](entt::entity ship, const DeliverScrapBehavior& deliver, const PositionComponent& pos)
+        {
+            if (deliver.deliveryBay == kInvalidBuildingId) { return; }
+            const Building* bay = buildings.findBuilding(deliver.deliveryBay);
+            if (!bay) { return; }
+
+            const QVector2D bayCenter(bay->anchor.x() + bay->footprint.width()  / 2.0f,
+                                      bay->anchor.y() + bay->footprint.height() / 2.0f);
+            if ((pos.value - bayCenter).length() > 1.0f) { return; }
+
+            // Decrement the first non-empty salvage child belonging to this ship.
+            bool delivered = false;
+            m_admin.forEach<SalvageCargoComponent, ModuleOwnerComponent>(
+                [&](entt::entity /*ce*/, SalvageCargoComponent& c, const ModuleOwnerComponent& o)
+                {
+                    if (delivered || o.owner != ship || c.current <= 0) { return; }
+                    if (buildings.deliverScrapToSalvageBay(deliver.deliveryBay))
+                    {
+                        --c.current;
+                        delivered = true;
+                    }
+                });
+        });
+}