Ticket #4142: 4142_petra_regicide_support_v1.4.patch

binaries/data/mods/public/simulation/ai/common-api/entity.js

@@ -929 +929 @@
             if (item.progress < percentToStopAt)
                 Engine.PostCommand(PlayerID,{ "type": "stop-production", "entity": this.id(), "id": item.id });
         return this;
+    },
+
+    "canGuard": function() { return this.get("UnitAI/CanGuard") === "true"; },
+
+    "guard": function(target, queued = false) {
+        Engine.PostCommand(PlayerID, { "type": "guard", "entities": [this.id()], "target": target.id(), "queued": queued });
+        return this;
+    },
+
+    "removeGuard": function() {
+        Engine.PostCommand(PlayerID, { "type": "remove-guard", "entities": [this.id()] });
+        return this;
     }
 });
 

binaries/data/mods/public/simulation/ai/petra/gameTypeManager.js

@@ -7 +7 @@
  * TODO: Assign military units to guard the hero in regicide
  */
 
-m.GameTypeManager = function()
+m.GameTypeManager = function(Config)
 {
+    this.Config = Config;
     this.heroGarrisonEmergency = false;
+    this.healersAssignedToHero = 0;
+    this.heroGuards = [];
 };
 
 /**
@@ -80 +83 @@
         }
     }
 
-    if (!this.heroGarrisonEmergency)
-        return;
+    // check if new healers/guards need to be assigned to the hero
+    for (let evt of events.Destroy)
+    {
+        if (!evt.entityObj || evt.entityObj.owner() !== PlayerID ||
+            this.heroGuards.indexOf(evt.entityObj) === -1)
+            continue;
 
+        this.heroGuards.splice(this.heroGuards.indexOf(evt.entityObj), 1);
+        if (evt.entityObj.hasClass("Healer"))
+            --this.healersAssignedToHero;
+    }
+
+    for (let evt of events.TrainingFinished)
+        for (let entId of evt.entities)
+        {
+            let ent = gameState.getEntityById(entId);
+            if (!ent || !ent.isOwn(PlayerID) || ent.getMetadata(PlayerID, "role") !== "regicideHealer")
+                continue;
+
+            let heroEnt = gameState.getOwnEntitiesByClass("Hero", true).toEntityArray()[0];
+            let access = gameState.ai.accessibility.getAccessValue(ent.position());
+
+            if (heroEnt && heroEnt.position() && ent.canGuard() &&
+                access === gameState.ai.accessibility.getAccessValue(heroEnt.position()))
+            {
+                ent.guard(heroEnt);
+                this.heroGuards.push(ent);
+            }
+        }
+
     for (let evt of events.Garrison)
-        if (gameState.getEntityById(evt.entity).hasClass("Hero"))
+    {
+        let ent = gameState.getEntityById(evt.entity)
+        if (!ent || !ent.isOwn(PlayerID) || !ent.hasClass("Hero"))
+            continue;
+
+        if (this.heroGarrisonEmergency)
             this.heroGarrisonEmergency = false;
+
+        if (!gameState.getEntityById(evt.holder).hasClass("Ship"))
+            continue;
+
+        // If the hero is garrisoned on a ship, remove its guards
+        for (let guardEnt of this.heroGuards)
+            if (guardEnt)
+                guardEnt.removeGuard();
+    }
 };
 
 m.GameTypeManager.prototype.buildWonder = function(gameState, queues)
@@ -116 +160 @@
         heroEnt.move(basePos.anchor.position()[0], basePos.anchor.position()[1]);
 };
 
+m.GameTypeManager.prototype.assignHealerToRegicideHero = function(gameState, queues)
+{
+    if (gameState.ai.HQ.saveResources || !gameState.getOwnEntitiesByClass("Temple", true).hasEntities())
+        return;
+
+    let template = gameState.applyCiv("units/{civ}_support_healer_b");
+
+    queues.villager.addPlan(new m.TrainingPlan(gameState, template, { "role": "regicideHealer", "base": 0 }, 1, 1));
+    ++this.healersAssignedToHero;
+};
+
 m.GameTypeManager.prototype.update = function(gameState, events, queues)
 {
     this.checkEvents(gameState, events);
@@ -122 +177 @@
 
     if (gameState.getGameType() === "wonder")
         this.buildWonder(gameState, queues);
-    else if (gameState.getGameType() === "regicide" && gameState.ai.playedTurn % 50 === 0)
+
+    if (gameState.getGameType() !== "regicide")
+        return;
+
+    if (gameState.ai.playedTurn % 50 === 0)
     {
         let heroEnt = gameState.getOwnEntitiesByClass("Hero", true).toEntityArray()[0];
         if (heroEnt && heroEnt.healthLevel() > 0.7)
             heroEnt.setStance("aggressive");
     }
+
+    if (this.healersAssignedToHero < Math.min(Math.round(this.Config.personality.defensive * 10), 4))
+        this.assignHealerToRegicideHero(gameState, queues);
 };
 
 m.GameTypeManager.prototype.Serialize = function()
 {
-    return { "heroGarrisonEmergency": this.heroGarrisonEmergency };
+    return {
+        "heroGarrisonEmergency": this.heroGarrisonEmergency,
+        "healersAssignedToHero": this.healersAssignedToHero,
+        "heroGuards": this.heroGuards
+    };
 };
 
 m.GameTypeManager.prototype.Deserialize = function(data)

binaries/data/mods/public/simulation/ai/petra/headquarters.js

@@ -51 +51 @@
     this.researchManager = new m.ResearchManager(this.Config);
     this.diplomacyManager = new m.DiplomacyManager(this.Config);
     this.garrisonManager = new m.GarrisonManager();
-    this.gameTypeManager = new m.GameTypeManager();
+    this.gameTypeManager = new m.GameTypeManager(this.Config);
 };
 
 /** More initialisation for stuff that needs the gameState */
@@ -2279 +2279 @@
     this.garrisonManager = new m.GarrisonManager();
     this.garrisonManager.Deserialize(data.garrisonManager);
 
-    this.gameTypeManager = new m.GameTypeManager();
+    this.gameTypeManager = new m.GameTypeManager(this.Config);
     this.gameTypeManager.Deserialize(data.gameTypeManager);
 };