Remove a pile of duplication on mcv undeploy

2010-07-30 20:36:04 +12:00
parent 119f85dbf4
commit bde16e4e15
8 changed files with 52 additions and 122 deletions
--- a/OpenRA.Mods.RA/Activities/Transform.cs
+++ b/OpenRA.Mods.RA/Activities/Transform.cs
@@ -0,0 +1,87 @@
+#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;
+using System.Linq;
+using System.Collections.Generic;
+using OpenRA.Traits;
+using OpenRA.Mods.RA.Render;
+
+namespace OpenRA.Mods.RA.Activities
+{
+	class Transform : IActivity
+	{
+		string actor = null;
+		int2 offset;
+		string[] sounds = null;		
+		bool isCanceled;
+		int facing;
+		
+		RenderBuilding rb;
+		public Transform(Actor self, string toActor, int2 offset, int facing, string[] sounds)
+		{
+			this.actor = toActor;
+			this.offset = offset;
+			this.sounds = sounds;
+			this.facing = facing;
+			rb = self.traits.GetOrDefault<RenderBuilding>();
+		}
+		
+		public IActivity NextActivity { get; set; }
+
+		
+		void DoTransform(Actor self)
+		{
+			self.World.AddFrameEndTask(w =>
+			{
+				var selected = w.Selection.Contains(self);
+				
+				self.World.Remove(self);
+				foreach (var s in sounds)
+					Sound.PlayToPlayer(self.Owner, s, self.CenterLocation);
+
+				var a = w.CreateActor(actor, self.Location + offset, self.Owner);
+				var oldHealth = self.traits.GetOrDefault<Health>();
+				var newHealth = a.traits.GetOrDefault<Health>();
+				if (oldHealth != null && newHealth != null)
+					newHealth.HPFraction = oldHealth.HPFraction;
+				
+				var unit = a.traits.GetOrDefault<Unit>();
+				if (unit != null)
+					unit.Facing = facing;
+				
+				if (selected)
+					w.Selection.Add(w, a);
+			});
+		}
+		
+		bool started = false;
+		public IActivity Tick( Actor self )
+		{
+			if (isCanceled) return NextActivity;
+			if (started) return this;
+			
+			if (rb == null)
+				DoTransform(self);
+			else
+			{
+				rb.PlayCustomAnimBackwards(self, "make", () => DoTransform(self));
+				
+				foreach (var s in self.Info.Traits.Get<BuildingInfo>().SellSounds)
+					Sound.PlayToPlayer(self.Owner, s, self.CenterLocation);
+				
+				started = true;	
+			}
+			return this;
+		}
+		
+		public void Cancel(Actor self) { isCanceled = true; NextActivity = null; }
+	}
+}