Merge pull request #7664 from penev92/bleed_harvesterInsurance

Implement Carryall edge spawn, harvester delivery by carryall and harvester insurance for D2k

OpenRA.Mods.D2k/Traits/Buildings/FreeActorWithDelivery.cs

@@ -0,0 +1,110 @@
+#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.IO;
+using OpenRA.Activities;
+using OpenRA.Mods.Common.Activities;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Mods.D2k.Activities;
+using OpenRA.Primitives;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.D2k.Traits
+{
+	[Desc("Player receives a unit for free once the building is placed.",
+		"If you want more than one unit to be delivered, copy this section and assign IDs like FreeActorWithDelivery@2, ...")]
+	public class FreeActorWithDeliveryInfo : FreeActorInfo
+	{
+		[ActorReference]
+		[Desc("Name of the delivering actor. This actor must have the `Carryall` trait")]
+		public readonly string DeliveringActor = null;
+
+		[Desc("Cell coordinates for spawning the delivering actor. If left blank, the closest edge cell will be chosen.")]
+		public readonly CPos SpawnLocation = CPos.Zero;
+
+		[Desc("Offset relative to the top-left cell of the building.")]
+		public readonly CVec DeliveryOffset = CVec.Zero;
+
+		public override object Create(ActorInitializer init) { return new FreeActorWithDelivery(init, this); }
+	}
+
+	public class FreeActorWithDelivery
+	{
+		public readonly FreeActorWithDeliveryInfo Info;
+
+		readonly Actor self;
+
+		public FreeActorWithDelivery(ActorInitializer init, FreeActorWithDeliveryInfo info)
+		{
+			if (string.IsNullOrEmpty(info.Actor))
+				throw new InvalidDataException("Actor type was not specified!");
+			if (string.IsNullOrEmpty(info.DeliveringActor))
+				throw new InvalidDataException("Delivering actor type was not specified!");
+
+			self = init.Self;
+			Info = info;
+
+			DoDelivery(self.Location + info.DeliveryOffset, info.Actor, info.DeliveringActor, info.InitialActivity);
+		}
+
+		public void DoDelivery(CPos location, string actorName, string carrierActorName, string clientInitialActivity)
+		{
+			Actor cargo;
+			Actor carrier;
+
+			CreateActors(actorName, carrierActorName, out cargo, out carrier);
+
+			if (clientInitialActivity != null)
+				cargo.QueueActivity(Game.CreateObject<Activity>(clientInitialActivity));
+
+			cargo.Trait<Carryable>().Destination = location;
+
+			carrier.Trait<Carryall>().AttachCarryable(cargo);
+
+			carrier.QueueActivity(new DeliverUnit(carrier));
+			carrier.QueueActivity(new HeliFly(carrier, Target.FromCell(self.World, self.World.Map.ChooseRandomEdgeCell(self.World.SharedRandom))));
+			carrier.QueueActivity(new RemoveSelf());
+
+			self.World.AddFrameEndTask(w => self.World.Add(carrier));
+		}
+
+		void CreateActors(string actorName, string deliveringActorName, out Actor cargo, out Actor carrier)
+		{
+			// Get a carryall spawn location
+			var location = Info.SpawnLocation;
+			if (location == CPos.Zero)
+				location = self.World.Map.ChooseClosestEdgeCell(self.Location);
+
+			var spawn = self.World.Map.CenterOfCell(location);
+
+			var initialFacing = self.World.Map.FacingBetween(location, self.Location, 0);
+
+			// If aircraft, spawn at cruise altitude
+			var aircraftInfo = self.World.Map.Rules.Actors[deliveringActorName.ToLower()].Traits.GetOrDefault<AircraftInfo>();
+			if (aircraftInfo != null)
+				spawn += new WVec(0, 0, aircraftInfo.CruiseAltitude.Range);
+
+			// Create delivery actor
+			carrier = self.World.CreateActor(false, deliveringActorName, new TypeDictionary
+			{
+				new LocationInit(location),
+				new CenterPositionInit(spawn),
+				new OwnerInit(self.Owner),
+				new FacingInit(initialFacing)
+			});
+
+			// Create delivered actor
+			cargo = self.World.CreateActor(false, actorName, new TypeDictionary
+			{
+				new OwnerInit(self.Owner),
+			});
+		}
+	}
+}

OpenRA.Mods.D2k/Traits/Buildings/ProductionFromMapEdge.cs

@@ -0,0 +1,82 @@
+#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.Drawing;
+using OpenRA.Activities;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Primitives;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.D2k.Traits
+{
+	[Desc("Produce a unit on the closest map edge cell and move into the world.")]
+	class ProductionFromMapEdgeInfo : ProductionInfo
+	{
+		public override object Create(ActorInitializer init) { return new ProductionFromMapEdge(init, this); }
+	}
+
+	class ProductionFromMapEdge : Production
+	{
+		public ProductionFromMapEdge(ActorInitializer init, ProductionInfo info)
+			: base(init, info) { }
+
+		public override bool Produce(Actor self, ActorInfo producee, string raceVariant)
+		{
+			var location = self.World.Map.ChooseClosestEdgeCell(self.Location);
+			var pos = self.World.Map.CenterOfCell(location);
+
+			// If aircraft, spawn at cruise altitude
+			var aircraftInfo = producee.Traits.GetOrDefault<AircraftInfo>();
+			if (aircraftInfo != null)
+				pos += new WVec(0, 0, aircraftInfo.CruiseAltitude.Range);
+
+			var initialFacing = self.World.Map.FacingBetween(location, self.Location, 0);
+
+			self.World.AddFrameEndTask(w =>
+				{
+					var td = new TypeDictionary
+					{
+						new OwnerInit(self.Owner),
+						new LocationInit(location),
+						new CenterPositionInit(pos),
+						new FacingInit(initialFacing)
+					};
+
+					if (raceVariant != null)
+						td.Add(new RaceInit(raceVariant));
+
+					var newUnit = self.World.CreateActor(producee.Name, td);
+
+					var move = newUnit.TraitOrDefault<IMove>();
+					if (move != null)
+						newUnit.QueueActivity(move.MoveIntoWorld(newUnit, self.Location));
+
+					newUnit.SetTargetLine(Target.FromCell(self.World, self.Location), Color.Green, false);
+
+					if (!self.IsDead)
+						foreach (var t in self.TraitsImplementing<INotifyProduction>())
+							t.UnitProduced(self, newUnit, self.Location);
+
+					var notifyOthers = self.World.ActorsWithTrait<INotifyOtherProduction>();
+					foreach (var notify in notifyOthers)
+						notify.Trait.UnitProducedByOther(notify.Actor, self, newUnit);
+
+					var bi = newUnit.Info.Traits.GetOrDefault<BuildableInfo>();
+					if (bi != null && bi.InitialActivity != null)
+						newUnit.QueueActivity(Game.CreateObject<Activity>(bi.InitialActivity));
+
+					foreach (var t in newUnit.TraitsImplementing<INotifyBuildComplete>())
+						t.BuildingComplete(newUnit);
+				});
+
+			return true;
+		}
+	}
+}