Rename HeliReturn

Rename HeliReturn to HeliReturnToBase to keep somewhat with the convention established for Plane- against Heli- oriented traits.
2015-09-05 18:01:35 +03:00
parent 15a5bde7e7
commit 0ee38de4bb
5 changed files with 7 additions and 7 deletions
--- a/OpenRA.Mods.Common/Activities/Air/HeliReturnToBase.cs
+++ b/OpenRA.Mods.Common/Activities/Air/HeliReturnToBase.cs
@@ -0,0 +1,74 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2015 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.Activities;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Activities
+{
+	public class HeliReturnToBase : Activity
+	{
+		readonly Aircraft heli;
+
+		public HeliReturnToBase(Actor self)
+		{
+			heli = self.Trait<Aircraft>();
+		}
+
+		public Actor ChooseHelipad(Actor self)
+		{
+			var rearmBuildings = heli.Info.RearmBuildings;
+			return self.World.Actors.Where(a => a.Owner == self.Owner).FirstOrDefault(
+				a => rearmBuildings.Contains(a.Info.Name) && !Reservable.IsReserved(a));
+		}
+
+		public override Activity Tick(Actor self)
+		{
+			if (IsCanceled)
+				return NextActivity;
+
+			var dest = ChooseHelipad(self);
+			var initialFacing = heli.Info.InitialFacing;
+
+			if (dest == null)
+			{
+				var rearmBuildings = heli.Info.RearmBuildings;
+				var nearestHpad = self.World.ActorsWithTrait<Reservable>()
+									.Where(a => a.Actor.Owner == self.Owner && rearmBuildings.Contains(a.Actor.Info.Name))
+									.Select(a => a.Actor)
+									.ClosestTo(self);
+
+				if (nearestHpad == null)
+					return Util.SequenceActivities(new Turn(self, initialFacing), new HeliLand(self, true), NextActivity);
+				else
+					return Util.SequenceActivities(new HeliFly(self, Target.FromActor(nearestHpad)));
+			}
+
+			var res = dest.TraitOrDefault<Reservable>();
+
+			if (res != null)
+			{
+				heli.UnReserve();
+				heli.Reservation = res.Reserve(dest, self, heli);
+			}
+
+			var exit = dest.Info.TraitInfos<ExitInfo>().FirstOrDefault();
+			var offset = (exit != null) ? exit.SpawnOffset : WVec.Zero;
+
+			return Util.SequenceActivities(
+				new HeliFly(self, Target.FromPos(dest.CenterPosition + offset)),
+				new Turn(self, initialFacing),
+				new HeliLand(self, false),
+				new ResupplyAircraft(self));
+		}
+	}
+}