Begin splitting power into its own trait; incomplete and non-working.

OpenRA.Game/OpenRA.Game.csproj

@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="utf-8"?>
+<?xml version="1.0" encoding="utf-8"?>
 <Project DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003" ToolsVersion="3.5">
   <PropertyGroup>
     <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
@@ -227,6 +227,7 @@
     <Compile Include="Graphics\CursorProvider.cs" />
     <Compile Include="Traits\Player\TechTree.cs" />
     <Compile Include="Traits\Player\ClassicProductionQueue.cs" />
+    <Compile Include="Traits\Player\PowerManager.cs" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\OpenRA.FileFormats\OpenRA.FileFormats.csproj">

OpenRA.Game/Traits/Player/ClassicProductionQueue.cs

@@ -15,7 +15,7 @@ using OpenRA.FileFormats;
 
 namespace OpenRA.Traits
 {
-	public class ClassicProductionQueueInfo : ProductionQueueInfo, ITraitPrerequisite<TechTreeInfo>
+	public class ClassicProductionQueueInfo : ProductionQueueInfo, ITraitPrerequisite<TechTreeInfo>, ITraitPrerequisite<PowerManagerInfo>, ITraitPrerequisite<PlayerResourcesInfo>
 	{
 		public override object Create(ActorInitializer init) { return new ClassicProductionQueue(init.self, this); }
 	}

OpenRA.Game/Traits/Player/PlayerResources.cs

@@ -45,17 +45,53 @@ namespace OpenRA.Traits
 		public int OreCapacity;
 		[Sync]
 		public int DisplayOre;
+
+		public float GetSiloFullness() { return (float)Ore / OreCapacity; }
+
+		public void GiveOre(int num)
+		{
+			Ore += num;
+			
+			if (Ore > OreCapacity)
+			{
+				nextSiloAdviceTime = 0;
+				Ore = OreCapacity;
+			}
+		}
 		
-		[Sync]
-		public int PowerProvided;
-		[Sync]
-		public int PowerDrained;
+		public bool TakeOre(int num)
+		{
+			if (Ore < num) return false;
+			Ore -= num;
+			
+			return true;
+		}
 		
+		public void GiveCash(int num)
+		{
+			Cash += num;
+		}
+		
+		public bool TakeCash(int num)
+		{			
+			if (Cash + Ore < num) return false;
+			
+			// Spend ore before cash
+			Ore -= num;
+			if (Ore < 0)
+			{
+				Cash += Ore;
+				Ore = 0;	
+			}
+			
+			return true;
+		}
+
 		const float displayCashFracPerFrame = .07f;
 		const int displayCashDeltaPerFrame = 37;
 		int nextSiloAdviceTime = 0;
 
-		void TickOre(Actor self)
+		public void Tick(Actor self)
 		{
 			OreCapacity = self.World.Queries.OwnedBy[Owner].WithTrait<IStoreOre>()
 				.Sum(a => a.Trait.Capacity);
@@ -102,91 +138,5 @@ namespace OpenRA.Traits
 				Sound.PlayToPlayer(self.Owner, eva.CashTickDown);
 			}
 		}
-		
-		int nextPowerAdviceTime = 0;
-		void TickPower()
-		{
-			var oldBalance = PowerProvided - PowerDrained;
-
-			PowerProvided = 0;
-			PowerDrained = 0;
-
-			var myBuildings = Owner.World.Queries.OwnedBy[Owner].WithTrait<Building>();
-
-			foreach (var a in myBuildings)
-			{
-				var q = a.Trait.GetPowerUsage();
-				if (q > 0)
-					PowerProvided += q;
-				else
-					PowerDrained -= q;
-			}
-
-			if (PowerProvided - PowerDrained < 0)
-				if (PowerProvided - PowerDrained != oldBalance)
-					nextPowerAdviceTime = 0;
-			
-			if (--nextPowerAdviceTime <= 0)
-			{
-				if (PowerProvided - PowerDrained < 0)
-					Owner.GiveAdvice(Rules.Info["world"].Traits.Get<EvaAlertsInfo>().LowPower);
-				
-				nextPowerAdviceTime = AdviceInterval;
-			}
-		}
-
-		public PowerState GetPowerState()
-		{
-			if (PowerProvided >= PowerDrained) return PowerState.Normal;
-			if (PowerProvided > PowerDrained / 2) return PowerState.Low;
-			return PowerState.Critical;
-		}
-
-		public float GetSiloFullness() { return (float)Ore / OreCapacity; }
-
-		public void GiveOre(int num)
-		{
-			Ore += num;
-			
-			if (Ore > OreCapacity)
-			{
-				nextSiloAdviceTime = 0;
-				Ore = OreCapacity;
-			}
-		}
-		
-		public bool TakeOre(int num)
-		{
-			if (Ore < num) return false;
-			Ore -= num;
-			
-			return true;
-		}
-		
-		public void GiveCash(int num)
-		{
-			Cash += num;
-		}
-		
-		public bool TakeCash(int num)
-		{			
-			if (Cash + Ore < num) return false;
-			
-			// Spend ore before cash
-			Ore -= num;
-			if (Ore < 0)
-			{
-				Cash += Ore;
-				Ore = 0;	
-			}
-			
-			return true;
-		}
-
-		public void Tick(Actor self)
-		{
-			TickPower();
-			TickOre(self);
-		}
 	}
 }

OpenRA.Game/Traits/Player/PowerManager.cs

@@ -0,0 +1,108 @@
+#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;
+
+namespace OpenRA.Traits
+{
+	public class PowerManagerInfo : ITraitInfo
+	{
+		public readonly int AdviceInterval = 250;
+		public object Create(ActorInitializer init) { return new PowerManager(init, this); }
+	}
+
+	public class PowerManager : ITick
+	{
+		PowerManagerInfo Info;
+		Player Player;
+		
+		Dictionary<Actor, int> PowerDrain = new Dictionary<Actor, int>(); 
+		[Sync] int totalProvided;
+		public int PowerProvided { get { return totalProvided; } }
+	
+		[Sync] int totalDrained;
+		public int PowerDrained { get { return totalDrained; } }
+		
+		public PowerManager(ActorInitializer init, PowerManagerInfo info)
+		{
+			Info = info;
+			Player = init.self.Owner;
+			
+			init.world.ActorAdded += ActorAdded;
+			init.world.ActorRemoved += ActorRemoved;
+		}
+		
+		public void ActorAdded(Actor a)
+		{
+			if (a.Owner != Player || !a.HasTrait<Building>())
+				return;
+			
+			PowerDrain.Add(a, a.Trait<Building>().GetPowerUsage());
+			UpdateTotals();
+		}
+		
+		public void UpdateActor(Actor a, int newPower)
+		{
+			if (a.Owner != Player || !a.HasTrait<Building>())
+				return;
+			
+			PowerDrain[a] = newPower;
+			UpdateTotals();
+		}
+		
+		public void ActorRemoved(Actor a)
+		{
+			if (a.Owner != Player || !a.HasTrait<Building>())
+				return;
+			
+			PowerDrain.Remove(a);
+			UpdateTotals();
+		}
+		
+		void UpdateTotals()
+		{
+			foreach (var p in PowerDrain.Values)
+			{
+				if (p > 0)
+					totalProvided += p;
+				else
+					totalDrained -= p;
+			}
+		}
+		
+		int nextPowerAdviceTime = 0;
+		bool wasLowPower = false;
+		public void Tick(Actor self)
+		{
+			var lowPower = totalProvided < totalDrained;
+			if (lowPower && !wasLowPower)
+				nextPowerAdviceTime = 0;
+			wasLowPower = lowPower;
+			
+			
+			if (--nextPowerAdviceTime <= 0)
+			{
+				if (lowPower)
+					Player.GiveAdvice(Rules.Info["world"].Traits.Get<EvaAlertsInfo>().LowPower);
+				
+				nextPowerAdviceTime = Info.AdviceInterval;
+			}
+		}
+
+		public PowerState GetPowerState()
+		{
+			if (PowerProvided >= PowerDrained) return PowerState.Normal;
+			if (PowerProvided > PowerDrained / 2) return PowerState.Low;
+			return PowerState.Critical;
+		}
+	}
+}

OpenRA.Game/Traits/Player/ProductionQueue.cs

@@ -26,6 +26,8 @@ namespace OpenRA.Traits
 	{
 		public readonly Actor self;
 		public ProductionQueueInfo Info;
+		readonly PowerManager PlayerPower;
+		readonly PlayerResources PlayerResources;
 		
 		// TODO: sync these
 		// A list of things we are currently building
@@ -38,7 +40,9 @@ namespace OpenRA.Traits
 		{
 			this.self = self;
 			this.Info = info;
-
+			PlayerResources = playerActor.Trait<PlayerResources>();
+			PlayerPower = playerActor.Trait<PowerManager>();
+			
 			var ttc = playerActor.Trait<TechTree>();
 			
 			foreach (var a in AllBuildables(Info.Type))
@@ -119,7 +123,7 @@ namespace OpenRA.Traits
 				FinishProduction();
 			}
 			if( Queue.Count > 0 )
-				Queue[ 0 ].Tick( self.Owner );
+				Queue[ 0 ].Tick( PlayerResources, PlayerPower );
 		}
 
 		public void ResolveOrder( Actor self, Order order )
@@ -249,7 +253,6 @@ namespace OpenRA.Traits
 
 		public bool Paused = false, Done = false;
 		public Action OnComplete;
-
 		int slowdown = 0;
 
 		public ProductionItem(ProductionQueue queue, string item, int time, int cost, Action onComplete)
@@ -264,7 +267,7 @@ namespace OpenRA.Traits
 			Log.Write("debug", "new ProductionItem: {0} time={1} cost={2}", item, time, cost);
 		}
 
-		public void Tick(Player player)
+		public void Tick(PlayerResources pr, PowerManager pm)
 		{
 			if (Done)
 			{
@@ -274,7 +277,7 @@ namespace OpenRA.Traits
 
 			if (Paused) return;
 
-			if (player.PlayerActor.Trait<PlayerResources>().GetPowerState() != PowerState.Normal)
+			if (pm.GetPowerState() != PowerState.Normal)
 			{
 				if (--slowdown <= 0)
 					slowdown = Queue.Info.LowPowerSlowdown; 
@@ -283,7 +286,7 @@ namespace OpenRA.Traits
 			}
 
 			var costThisFrame = RemainingCost / RemainingTime;
-			if (costThisFrame != 0 && !player.PlayerActor.Trait<PlayerResources>().TakeCash(costThisFrame)) return;
+			if (costThisFrame != 0 && !pr.TakeCash(costThisFrame)) return;
 			RemainingCost -= costThisFrame;
 			RemainingTime -= 1;
 			if (RemainingTime > 0) return;

OpenRA.Game/Traits/SupportPower.cs

@@ -12,7 +12,7 @@ using System.Linq;
 
 namespace OpenRA.Traits
 {
-	public abstract class SupportPowerInfo : ITraitInfo, ITraitPrerequisite<TechTreeInfo>
+	public abstract class SupportPowerInfo : ITraitInfo, ITraitPrerequisite<TechTreeInfo>, ITraitPrerequisite<PowerManagerInfo>
 	{
 		public readonly bool RequiresPower = true;
 		public readonly bool OneShot = false;
@@ -52,12 +52,14 @@ namespace OpenRA.Traits
 		bool notifiedCharging;
 		bool notifiedReady;
 
+		readonly PowerManager PlayerPower;
 		public SupportPower(Actor self, SupportPowerInfo info)
 		{
 			Info = info;
 			RemainingTime = TotalTime;
 			Self = self;
 			Owner = self.Owner;
+			PlayerPower = self.Trait<PowerManager>();
 
 			effectivePrereq = Info.Prerequisites
 				.Select(a => a.ToLowerInvariant()).ToArray();
@@ -98,7 +100,7 @@ namespace OpenRA.Traits
 		bool IsPowered()
 		{
 			if (effectivePrereq.Count() == 0) 
-				return Owner.PlayerActor.Trait<PlayerResources>().GetPowerState() == PowerState.Normal;
+				return PlayerPower.GetPowerState() == PowerState.Normal;
 			
 			// Takes 0.3ms on pchote's machine -- calling it every tick for every active special power is retarded
 			var buildings = TechTree.GatherBuildings(Owner);