Merge pull request #10570 from abcdefg30/InsufficientFunds

Add support for playing an InsufficientFunds warning

OpenRA.Mods.Common/Traits/CustomBuildTimeValue.cs

@@ -25,6 +25,8 @@ namespace OpenRA.Mods.Common.Traits
 
 	public static class CustomBuildTimeValueExts
 	{
+		const int FramesPerMin = 25 * 60;
+
 		public static int GetBuildTime(this ActorInfo a)
 		{
 			var csv = a.TraitInfoOrDefault<CustomBuildTimeValueInfo>();
@@ -32,11 +34,7 @@ namespace OpenRA.Mods.Common.Traits
 				return csv.Value;
 
 			var cost = a.HasTraitInfo<ValuedInfo>() ? a.TraitInfo<ValuedInfo>().Cost : 0;
-			var time = cost
-							* (25 * 60) /* frames per min */
-							/ 1000;
-			return
-				time;
+			return cost * FramesPerMin / 1000;
 		}
 	}
 }

OpenRA.Mods.Common/Traits/Player/InsufficientFundsWarning.cs

@@ -0,0 +1,53 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2016 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 OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Traits
+{
+	[Desc("Provides the player with an audible warning when they run out of money while producing.")]
+	public class InsufficientFundsWarningInfo : ITraitInfo, Requires<PlayerResourcesInfo>
+	{
+		[Desc("The speech to play for the warning.")]
+		public readonly string Notification = "InsufficientFunds";
+
+		public object Create(ActorInitializer init) { return new InsufficientFundsWarning(this); }
+	}
+
+	public class InsufficientFundsWarning : INotifyInsufficientFunds
+	{
+		readonly InsufficientFundsWarningInfo info;
+
+		bool played;
+
+		public InsufficientFundsWarning(InsufficientFundsWarningInfo info)
+		{
+			this.info = info;
+		}
+
+		void INotifyInsufficientFunds.InsufficientFunds(Actor self)
+		{
+			Game.RunAfterTick(() =>
+			{
+				if (played)
+					return;
+
+				played = true;
+				var owner = self.Owner;
+				Game.Sound.PlayNotification(self.World.Map.Rules, owner, "Speech", info.Notification, owner.Faction.InternalName);
+			});
+		}
+
+		void INotifyInsufficientFunds.SufficientFunds(Actor self)
+		{
+			played = false;
+		}
+	}
+}

OpenRA.Mods.Common/Traits/Player/ProductionQueue.cs

@@ -415,8 +415,12 @@ namespace OpenRA.Mods.Common.Traits
 		public bool Started { get; private set; }
 		public int Slowdown { get; private set; }
 
+		readonly INotifyInsufficientFunds[] insufficientFunds;
+		readonly Player owner;
 		readonly PowerManager pm;
 
+		bool insufficientFundsPlayed;
+
 		public ProductionItem(ProductionQueue queue, string item, int cost, PowerManager pm, Action onComplete)
 		{
 			Item = item;
@@ -425,6 +429,8 @@ namespace OpenRA.Mods.Common.Traits
 			OnComplete = onComplete;
 			Queue = queue;
 			this.pm = pm;
+			owner = queue.Actor.Owner;
+			insufficientFunds = owner.PlayerActor.TraitsImplementing<INotifyInsufficientFunds>().ToArray();
 		}
 
 		public void Tick(PlayerResources pr)
@@ -435,6 +441,13 @@ namespace OpenRA.Mods.Common.Traits
 				if (time > 0)
 					RemainingTime = TotalTime = time;
 
+				// Don't play a QueuedAudio notification when we can't start building (because we don't have the money to)
+				// Also don't play it when the time to build is actually 0 (i.e. normally dev cheats)
+				// to prevent overlapping with the ReadyAudio notification
+				var initialCost = RemainingCost / RemainingTime;
+				if (time != 0 && initialCost != 0 && pr.Cash + pr.Resources > initialCost)
+					Game.Sound.PlayNotification(owner.World.Map.Rules, owner, "Speech", Queue.Info.QueuedAudio, owner.Faction.InternalName);
+
 				Started = true;
 			}
 
@@ -459,7 +472,22 @@ namespace OpenRA.Mods.Common.Traits
 
 			var costThisFrame = RemainingCost / RemainingTime;
 			if (costThisFrame != 0 && !pr.TakeCash(costThisFrame))
+			{
+				if (!insufficientFundsPlayed)
+				{
+					insufficientFundsPlayed = true;
+					foreach (var funds in insufficientFunds)
+						funds.InsufficientFunds(owner.PlayerActor);
+				}
+
 				return;
+			}
+
+			if (insufficientFundsPlayed)
+				insufficientFundsPlayed = false;
+
+			foreach (var funds in insufficientFunds)
+				funds.SufficientFunds(owner.PlayerActor);
 
 			RemainingCost -= costThisFrame;
 			RemainingTime -= 1;