Split HackyAI.BaseBuilder into its own file; Move hackyAI components into OpenRA.Mods.RA.AI namespace and dir

2011-10-24 19:10:07 +13:00
parent bfddbeeb02
commit 3a5cce1f1f
3 changed files with 114 additions and 89 deletions
--- a/OpenRA.Mods.RA/AI/BaseBuilder.cs
+++ b/OpenRA.Mods.RA/AI/BaseBuilder.cs
@@ -0,0 +1,102 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2011 The OpenRA Developers (see AUTHORS)
+ * This file is part of OpenRA, which is free software. It is made
+ * available to you under the terms of the GNU General Public License
+ * as published by the Free Software Foundation. For more information,
+ * see COPYING.
+ */
+#endregion
+
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using OpenRA.FileFormats;
+using OpenRA.Mods.RA.Buildings;
+using OpenRA.Traits;
+using OpenRA.Mods.RA.Activities;
+using XRandom = OpenRA.Thirdparty.Random;
+
+namespace OpenRA.Mods.RA.AI
+{
+	class BaseBuilder
+	{
+		enum BuildState	{ ChooseItem, WaitForProduction, WaitForFeedback }
+
+		BuildState state = BuildState.WaitForFeedback;
+		string category;
+		HackyAI ai;
+		int lastThinkTick;
+		Func<ProductionQueue, ActorInfo> chooseItem;
+
+		public BaseBuilder(HackyAI ai, string category, Func<ProductionQueue, ActorInfo> chooseItem)
+		{
+			this.ai = ai;
+			this.category = category;
+			this.chooseItem = chooseItem;
+		}
+
+		public void Tick()
+		{
+			// Pick a free queue
+			var queue = ai.FindQueues( category ).FirstOrDefault();
+			if (queue == null)
+				return;
+
+			var currentBuilding = queue.CurrentItem();
+			switch (state)
+			{
+				case BuildState.ChooseItem:
+					{
+						var item = chooseItem(queue);
+						if (item == null)
+						{
+							state = BuildState.WaitForFeedback;
+							lastThinkTick = ai.ticks;
+						}
+						else
+						{
+							HackyAI.BotDebug("AI: Starting production of {0}".F(item.Name));
+							state = BuildState.WaitForProduction;
+							ai.world.IssueOrder(Order.StartProduction(queue.self, item.Name, 1));
+						}
+					}
+					break;
+
+				case BuildState.WaitForProduction:
+					if (currentBuilding == null) return;	/* let it happen.. */
+
+					else if (currentBuilding.Paused)
+						ai.world.IssueOrder(Order.PauseProduction(queue.self, currentBuilding.Item, false));
+					else if (currentBuilding.Done)
+					{
+						state = BuildState.WaitForFeedback;
+						lastThinkTick = ai.ticks;
+
+						/* place the building */
+						var location = ai.ChooseBuildLocation(currentBuilding);
+						if (location == null)
+						{
+							HackyAI.BotDebug("AI: Nowhere to place {0}".F(currentBuilding.Item));
+							ai.world.IssueOrder(Order.CancelProduction(queue.self, currentBuilding.Item, 1));
+						}
+						else
+						{
+							ai.world.IssueOrder(new Order("PlaceBuilding", ai.p.PlayerActor, false)
+								{
+									TargetLocation = location.Value,
+									TargetString = currentBuilding.Item
+								});
+						}
+					}
+					break;
+
+				case BuildState.WaitForFeedback:
+					if (ai.ticks - lastThinkTick > HackyAI.feedbackTime)
+						state = BuildState.ChooseItem;
+					break;
+			}
+		}
+	}
+}
+