Fix #197: deleting of constructing building decreased construction limit

dymanoid · Feb 10, 2019 · db0cf73 · db0cf73
1 parent 39483d8
commit db0cf73
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/RealTime/CustomAI/RealTimeBuildingAI.cs b/src/RealTime/CustomAI/RealTimeBuildingAI.cs
@@ -158,7 +158,7 @@ public bool CanBuildOrUpgrade(ItemClass.Service buildingZone, ushort buildingId
             }
 
             HashSet<ushort> buildings = buildingsInConstruction[index];
-            buildings.RemoveWhere(IsBuildingCompleted);
+            buildings.RemoveWhere(IsBuildingCompletedOrMissing);
 
             int allowedCount = GetAllowedConstructingUpradingCount(buildingManager.GeBuildingsCount());
             bool result = buildings.Count < allowedCount;
@@ -417,9 +417,9 @@ private static int GetAllowedConstructingUpradingCount(int currentBuildingCount)
             return MaximumBuildingsInConstruction;
         }
 
-        private bool IsBuildingCompleted(ushort buildingId)
+        private bool IsBuildingCompletedOrMissing(ushort buildingId)
         {
-            return buildingManager.BuildingHasFlags(buildingId, Building.Flags.Completed);
+            return buildingManager.BuildingHasFlags(buildingId, Building.Flags.Completed | Building.Flags.Deleted, includeZero: true);
         }
 
         private void UpdateLightState()