Move everything Repair-related to Mods.Common

OpenRA.Mods.Common/Effects/RepairIndicator.cs

@@ -0,0 +1,65 @@
+#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.Collections.Generic;
+using System.Linq;
+using OpenRA.Effects;
+using OpenRA.Graphics;
+using OpenRA.Mods.Common.Traits;
+
+namespace OpenRA.Mods.Common.Effects
+{
+	class RepairIndicator : IEffect
+	{
+		readonly Actor building;
+		readonly string palettePrefix;
+		readonly Animation anim;
+		readonly RepairableBuilding rb;
+		int shownPlayer = 0;
+
+		public RepairIndicator(Actor building, string palettePrefix)
+		{
+			this.building = building;
+			this.palettePrefix = palettePrefix;
+
+			rb = building.TraitOrDefault<RepairableBuilding>();
+			anim = new Animation(building.World, "allyrepair");
+			anim.Paused = () => !rb.RepairActive || rb.IsTraitDisabled;
+
+			CycleRepairer();
+		}
+
+		public void Tick(World world)
+		{
+			if (!building.IsInWorld || building.IsDead ||
+				rb == null || !rb.Repairers.Any())
+				world.AddFrameEndTask(w => w.Remove(this));
+
+			anim.Tick();
+		}
+
+		public IEnumerable<IRenderable> Render(WorldRenderer wr)
+		{
+			if (building.Destroyed || wr.World.FogObscures(building) || rb.Repairers.Count == 0)
+				return SpriteRenderable.None;
+
+			var palette = wr.Palette(palettePrefix + rb.Repairers[shownPlayer % rb.Repairers.Count].InternalName);
+			return anim.Render(building.CenterPosition, palette);
+		}
+
+		void CycleRepairer()
+		{
+			anim.PlayThen("repair", CycleRepairer);
+
+			if (++shownPlayer == rb.Repairers.Count)
+				shownPlayer = 0;
+		}
+	}
+}