From 19ec05c70ec061ad46c62ba0c64eef4193214a74 Mon Sep 17 00:00:00 2001
From: Regalis11 <poe.regalis@gmail.com>
Date: Mon, 18 Jan 2016 10:56:25 +0200
Subject: [PATCH] FixLeaks-command (still needs a sprite for the UI)

---
 Subsurface/Content/Orders.xml                 |  6 +-
 .../AI/Objectives/AIObjectiveFixLeaks.cs      | 68 +++++++++++++++++++
 .../AI/Objectives/AIObjectiveManager.cs       | 11 ++-
 3 files changed, 77 insertions(+), 8 deletions(-)
 create mode 100644 Subsurface/Source/Characters/AI/Objectives/AIObjectiveFixLeaks.cs

diff --git a/Subsurface/Content/Orders.xml b/Subsurface/Content/Orders.xml
index c28b1b516..4ae1b0331 100644
--- a/Subsurface/Content/Orders.xml
+++ b/Subsurface/Content/Orders.xml
@@ -19,5 +19,9 @@
 
   <Order name="Operate Railgun" doingtext="Operating Railgun" targetitemtype="Turret" options="Fire at will, Hold fire" color="1.0,0.0,0.0,1.0">
     <Sprite texture="Content/UI/orderSymbols.png" sourcerect="0,64,64,64"/>
-  </Order> 
+  </Order>
+
+  <Order name="Fix Leaks" doingtext="Fixing Leaks" color="0.2,0.2,1.0,1.0">
+    <Sprite texture="Content/UI/orderSymbols.png" sourcerect="0,64,64,64"/>    
+  </Order>
 </Orders>
\ No newline at end of file
diff --git a/Subsurface/Source/Characters/AI/Objectives/AIObjectiveFixLeaks.cs b/Subsurface/Source/Characters/AI/Objectives/AIObjectiveFixLeaks.cs
new file mode 100644
index 000000000..83140a770
--- /dev/null
+++ b/Subsurface/Source/Characters/AI/Objectives/AIObjectiveFixLeaks.cs
@@ -0,0 +1,68 @@
+using Microsoft.Xna.Framework;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+
+namespace Barotrauma
+{
+    class AIObjectiveFixLeaks : AIObjective
+    {
+        const float UpdateGapListInterval = 5.0f;
+
+        private float updateGapListTimer;
+
+        private List<AIObjectiveFixLeak> objectiveList;
+
+        public AIObjectiveFixLeaks(Character character)
+            : base (character, "")
+        {
+            objectiveList = new List<AIObjectiveFixLeak>();
+        }
+
+        public override bool IsCompleted()
+        {
+            return false;
+        }
+
+        protected override void Act(float deltaTime)
+        {
+            updateGapListTimer -= deltaTime;
+
+            if (updateGapListTimer<=0.0f)
+            {
+                UpdateGapList();
+
+                updateGapListTimer = UpdateGapListInterval;
+            }
+
+            if (objectiveList.Any()) objectiveList[0].TryComplete(deltaTime);
+        }
+
+        private void UpdateGapList()
+        {
+            objectiveList.Clear();
+            
+            foreach (Gap gap in Gap.GapList)
+            {
+                if (gap.IsRoomToRoom || gap.ConnectedDoor != null || gap.Open < 0.1f) continue;
+
+                float dist = Vector2.DistanceSquared(character.WorldPosition, gap.WorldPosition);
+
+                int index = 0;
+                while (index<objectiveList.Count && 
+                    Vector2.DistanceSquared(objectiveList[index].Leak.WorldPosition, character.WorldPosition)<dist)
+                {
+                    index++;
+                }
+
+                objectiveList.Insert(index, new AIObjectiveFixLeak(gap, character));
+            }
+        }
+
+        public override bool IsDuplicate(AIObjective otherObjective)
+        {
+            return otherObjective is AIObjectiveFixLeaks;
+        }
+    }
+}
diff --git a/Subsurface/Source/Characters/AI/Objectives/AIObjectiveManager.cs b/Subsurface/Source/Characters/AI/Objectives/AIObjectiveManager.cs
index c648355c8..4d0fe766b 100644
--- a/Subsurface/Source/Characters/AI/Objectives/AIObjectiveManager.cs
+++ b/Subsurface/Source/Characters/AI/Objectives/AIObjectiveManager.cs
@@ -54,13 +54,6 @@ namespace Barotrauma
 
             //sort objectives according to priority
             objectives.Sort((x, y) => y.GetPriority(character).CompareTo(x.GetPriority(character)));
-
-            foreach (Gap gap in Gap.GapList)
-            {
-                if (gap.IsRoomToRoom || gap.ConnectedDoor != null || gap.Open<0.1f) continue;
-
-                AddObjective(new AIObjectiveFixLeak(gap, character));
-            }
         }
 
         public void DoCurrentObjective(float deltaTime)
@@ -87,6 +80,10 @@ namespace Barotrauma
                 case "wait":
                     currentObjective = new AIObjectiveGoTo(character.SimPosition, character, true);
                     break;
+                case "fixleaks":
+                case "fix leaks":
+                    currentObjective = new AIObjectiveFixLeaks(character);
+                    break;
                 default:
                     if (order.TargetItem == null) return;