New extendable mechanism for disabling units

OpenRa.Game/Traits/Building.cs

@@ -1,7 +1,10 @@
 using System;
+using System.Linq;
 using OpenRa.Effects;
 using OpenRa.GameRules;
 using OpenRa.Traits.Activities;
+using OpenRa.Graphics;
+using System.Collections.Generic;
 
 namespace OpenRa.Traits
 {
@@ -17,7 +20,6 @@ namespace OpenRa.Traits
 	public class BuildingInfo : OwnedActorInfo, ITraitInfo
 	{
 		public readonly int Power = 0;
-		public readonly bool RequiresPower = false;
 		public readonly bool BaseNormal = true;
 		public readonly int Adjacent = 2;
 		public readonly bool Bib = false;
@@ -31,25 +33,18 @@ namespace OpenRa.Traits
 		public object Create(Actor self) { return new Building(self); }
 	}
 
-	public class Building : INotifyDamage, IResolveOrder, ITick
+	public class Building : INotifyDamage, IResolveOrder, ITick, IRenderModifier
 	{
 		readonly Actor self;
 		public readonly BuildingInfo Info;
 		[Sync]
 		bool isRepairing = false;
-		[Sync]
-		bool manuallyDisabled = false;
-		public bool ManuallyDisabled { get { return manuallyDisabled; } }
+
 		public bool Disabled
 		{
-			get
-			{
-				return (manuallyDisabled || 
-					(Info.RequiresPower && self.Owner.GetPowerState() != PowerState.Normal));
-			}
+			get	{ return self.traits.WithInterface<IDisable>().Any(t => t.Disabled); }
 		}
-		bool wasDisabled = false;
-		
+
 		public Building(Actor self)
 		{
 			this.self = self;
@@ -60,15 +55,17 @@ namespace OpenRa.Traits
 		
 		public int GetPowerUsage()
 		{
-			if (manuallyDisabled)
-				return 0;
-
+			var modifier = self.traits
+				.WithInterface<IPowerModifier>()
+				.Select(t => t.GetPowerModifier())
+				.Product();
+				
 			var maxHP = self.Info.Traits.Get<BuildingInfo>().HP;
 
 			if (Info.Power > 0)
-				return (self.Health * Info.Power) / maxHP;
+				return (int)(modifier*(self.Health * Info.Power) / maxHP);
 			else
-				return Info.Power;
+				return (int)(modifier * Info.Power);
 		}
 
 		public void Damaged(Actor self, AttackInfo e)
@@ -89,23 +86,12 @@ namespace OpenRa.Traits
 			{
 				isRepairing = !isRepairing;
 			}
-			
-			if (order.OrderString == "PowerDown")
-			{
-				manuallyDisabled = !manuallyDisabled;
-				Sound.Play((manuallyDisabled) ? "bleep12.aud" : "bleep11.aud");
-			}
 		}
 
 		int remainingTicks;
 
 		public void Tick(Actor self)
 		{
-			// If the disabled state has changed since the last frame
-			if (Disabled ^ wasDisabled 
-				&& (wasDisabled = Disabled)) // Yes, I mean assignment
-					self.World.AddFrameEndTask(w => w.Add(new PowerDownIndicator(self)));
-			
 			if (!isRepairing) return;
 
 			if (remainingTicks == 0)
@@ -132,5 +118,15 @@ namespace OpenRa.Traits
 			else
 				--remainingTicks;
 		}
+		
+		public IEnumerable<Renderable> ModifyRender(Actor self, IEnumerable<Renderable> r)
+		{
+			foreach (var a in r)
+			{
+				yield return a;
+				if (Disabled)
+					yield return a.WithPalette(PaletteType.Disabled);
+			}
+		}
 	}
 }

OpenRa.Game/Traits/CanPowerDown.cs

@@ -0,0 +1,41 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using OpenRa.Traits;
+
+namespace OpenRa.Traits
+{
+	public class CanPowerDownInfo : ITraitInfo
+	{
+		public object Create(Actor self) { return new CanPowerDown(self); }
+	}
+
+	public class CanPowerDown : IDisable, IPowerModifier, IResolveOrder
+	{
+		readonly Actor self;
+		[Sync]
+		bool IsDisabled = false;
+		
+		public CanPowerDown(Actor self)
+		{
+			this.self = self;
+		}
+
+		public bool Disabled
+		{
+			get { return IsDisabled; }
+			set { IsDisabled = value; }
+		}
+		
+		public float GetPowerModifier() { return (IsDisabled) ? 0.0f : 1.0f; }	
+		
+		public void ResolveOrder(Actor self, Order order)
+		{
+			if (order.OrderString == "PowerDown")
+			{
+				IsDisabled = !IsDisabled;
+				Sound.Play((IsDisabled) ? "bleep12.aud" : "bleep11.aud");
+			}
+		}
+	}
+}

OpenRa.Game/Traits/TraitsInterfaces.cs

@@ -26,7 +26,9 @@ namespace OpenRa.Traits
 	public interface IAcceptSpy { void OnInfiltrate(Actor self, Actor spy); }
 
 	public interface ICustomTerrain { float GetCost(int2 p, UnitMovementType umt); }
-
+	
+	public interface IDisable {	bool Disabled { get; set; } }
+	
 	interface IProducer
 	{
 		bool Produce( Actor self, ActorInfo producee );
@@ -37,6 +39,7 @@ namespace OpenRa.Traits
 	public interface IRenderModifier { IEnumerable<Renderable> ModifyRender(Actor self, IEnumerable<Renderable> r); }
 	public interface IDamageModifier { float GetDamageModifier(); }
 	public interface ISpeedModifier { float GetSpeedModifier(); }
+	public interface IPowerModifier { float GetPowerModifier(); }
 	public interface IPaletteModifier { void AdjustPalette(Bitmap b); }
 	public interface IPips { IEnumerable<PipType> GetPips(Actor self); }
 	public interface ITags { IEnumerable<TagType> GetTags(); }