Move Air activities and traits to Activities\Air and Traits\Air, respectively

OpenRA.Mods.RA/Traits/Air/ReturnOnIdle.cs

@@ -0,0 +1,64 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2014 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.Linq;
+using OpenRA.Mods.RA;
+using OpenRA.Mods.RA.Activities;
+using OpenRA.Mods.RA.Buildings;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.RA.Traits
+{
+	[Desc("Return to a player owned RearmBuildings. If none available, head back to base and circle over it.")]
+	class ReturnOnIdleInfo : TraitInfo<ReturnOnIdle> { }
+
+	class ReturnOnIdle : INotifyIdle
+	{
+		public void TickIdle(Actor self)
+		{
+			// We're on the ground, let's stay there.
+			if (self.CenterPosition.Z == 0)
+				return;
+
+			var airfield = ReturnToBase.ChooseAirfield(self, true);
+			if (airfield != null)
+			{
+				self.QueueActivity(new ReturnToBase(self, airfield));
+				self.QueueActivity(new ResupplyAircraft());
+			}
+			else
+			{
+				// nowhere to land, pick something friendly and circle over it.
+
+				// i'd prefer something we own
+				var someBuilding = self.World.ActorsWithTrait<Building>()
+					.Select(a => a.Actor)
+					.FirstOrDefault(a => a.Owner == self.Owner);
+
+				// failing that, something unlikely to shoot at us
+				if (someBuilding == null)
+					someBuilding = self.World.ActorsWithTrait<Building>()
+						.Select(a => a.Actor)
+						.FirstOrDefault(a => self.Owner.Stances[a.Owner] == Stance.Ally);
+
+				if (someBuilding == null)
+				{
+					// ... going down the garden to eat worms ...
+					self.QueueActivity(new FlyOffMap());
+					self.QueueActivity(new RemoveSelf());
+					return;
+				}
+
+				self.QueueActivity(new Fly(self, Target.FromActor(someBuilding)));
+				self.QueueActivity(new FlyCircle());
+			}
+		}
+	}
+}