moved traits from engine into mod

2010-09-20 20:56:11 +12:00
parent 65515d54af
commit aff6889995
15 changed files with 43 additions and 44 deletions
--- a/OpenRA.Mods.RA/Production.cs
+++ b/OpenRA.Mods.RA/Production.cs
@@ -0,0 +1,116 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2010 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 LICENSE.
+ */
+#endregion
+
+using System.Drawing;
+using OpenRA.FileFormats;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.RA
+{
+	public class ProductionInfo : ITraitInfo
+	{
+		public readonly string[] Produces = { };
+		
+		public virtual object Create(ActorInitializer init) { return new Production(this); }
+	}
+
+	public class ExitInfo : TraitInfo<Exit>
+	{
+		public readonly float2 SpawnOffset = float2.Zero; // in px relative to CenterLocation
+		public readonly int2 ExitCell = int2.Zero; // in cells relative to TopLeft
+		public readonly int Facing = -1;
+	}
+	public class Exit {}
+	
+	public class Production
+	{		
+		public ProductionInfo Info;
+		public Production(ProductionInfo info)
+		{
+			Info = info;
+		}
+		
+		public void DoProduction(Actor self, ActorInfo producee, ExitInfo exitinfo)
+		{
+			var newUnit = self.World.CreateActor(false, producee.Name, new TypeDictionary
+			{
+			    new OwnerInit( self.Owner ),
+			});
+
+			var exit = self.Location + exitinfo.ExitCell;
+			var spawn = self.CenterLocation + exitinfo.SpawnOffset;
+
+			var move = newUnit.Trait<IMove>();
+			var facing = newUnit.TraitOrDefault<IFacing>();
+			
+			// Set the physical position of the unit as the exit cell
+			move.SetPosition(newUnit,exit);
+			var to = Util.CenterOfCell(exit);
+			newUnit.CenterLocation = spawn;
+			if (facing != null)
+				facing.Facing = exitinfo.Facing < 0 ? Util.GetFacing(to - spawn, facing.Facing) : exitinfo.Facing;
+			self.World.Add(newUnit);
+
+			// Animate the spawn -> exit transition
+			var speed = move.MovementSpeedForCell(self, exit);
+			var length = speed > 0 ? (int)( ( to - spawn ).Length*3 / speed ) : 0;
+			newUnit.QueueActivity(new Traits.Activities.Drag(spawn, to, length));
+
+//			Log.Write("debug", "length={0} facing={1} exit={2} spawn={3}", length, facing.Facing, exit, spawn);
+			
+			// For the target line
+			var target = exit;
+			var rp = self.TraitOrDefault<RallyPoint>();
+			if (rp != null)
+			{
+				target = rp.rallyPoint;
+				// Todo: Move implies unit has Mobile
+				newUnit.QueueActivity(new Traits.Activities.Move(target, 1));
+			}
+			
+			if (newUnit.Owner == self.World.LocalPlayer)
+			{
+				self.World.AddFrameEndTask(w =>
+				{
+					var line = newUnit.TraitOrDefault<DrawLineToTarget>();
+					if (line != null)
+						line.SetTargetSilently(newUnit, Target.FromCell(target), Color.Green);
+				});
+			}
+
+			foreach (var t in self.TraitsImplementing<INotifyProduction>())
+				t.UnitProduced(self, newUnit, exit);
+
+			Log.Write("debug", "{0} #{1} produced by {2} #{3}", newUnit.Info.Name, newUnit.ActorID, self.Info.Name, self.ActorID);
+		}
+		
+		public virtual bool Produce( Actor self, ActorInfo producee )
+		{			
+			// Todo: remove assumption on Mobile; 
+			// required for 3-arg CanEnterCell
+			//var mobile = newUnit.Trait<Mobile>();
+			var mobileInfo = producee.Traits.Get<MobileInfo>();
+			var bim = self.World.WorldActor.Trait<BuildingInfluence>();
+			var uim = self.World.WorldActor.Trait<UnitInfluence>();
+			
+			// Pick a spawn/exit point pair
+			// Todo: Reorder in a synced random way
+			foreach (var s in self.Info.Traits.WithInterface<ExitInfo>())
+				if( Mobile.CanEnterCell( mobileInfo, self.World, uim, bim, self.Location + s.ExitCell,self,true ) )
+				{
+					DoProduction(self, producee, s);
+					return true;
+				}
+			return false;
+		}
+
+
+	}
+}