Moves BaseAttackNotifier to Mods.Common

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

@@ -0,0 +1,77 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2015 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.Drawing;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Traits
+{
+	[Desc("Plays an audio notification and shows a radar ping when a building is attacked.",
+		"Attach this to the player actor.")]
+	public class BaseAttackNotifierInfo : ITraitInfo
+	{
+		[Desc("Minimum duration (in seconds) between notification events.")]
+		public readonly int NotifyInterval = 30;
+
+		public readonly Color RadarPingColor = Color.Red;
+
+		[Desc("Length of time (in ticks) to display a location ping in the minimap.")]
+		public readonly int RadarPingDuration = 10 * 25;
+
+		[Desc("The audio notification type to play.")]
+		public string Notification = "BaseAttack";
+
+		public object Create(ActorInitializer init) { return new BaseAttackNotifier(init.Self, this); }
+	}
+
+	public class BaseAttackNotifier : INotifyDamage
+	{
+		readonly RadarPings radarPings;
+		readonly BaseAttackNotifierInfo info;
+
+		int lastAttackTime;
+
+		public BaseAttackNotifier(Actor self, BaseAttackNotifierInfo info)
+		{
+			radarPings = self.World.WorldActor.TraitOrDefault<RadarPings>();
+			this.info = info;
+			lastAttackTime = -info.NotifyInterval * 25;
+		}
+
+		public void Damaged(Actor self, AttackInfo e)
+		{
+			// only track last hit against our base
+			if (!self.HasTrait<Building>())
+				return;
+
+			if (e.Attacker == null)
+				return;
+
+			if (e.Attacker.Owner == self.Owner)
+				return;
+
+			if (e.Attacker == self.World.WorldActor)
+				return;
+
+			if (e.Attacker.Owner.IsAlliedWith(self.Owner) && e.Damage <= 0)
+				return;
+
+			if (self.World.WorldTick - lastAttackTime > info.NotifyInterval * 25)
+			{
+				Sound.PlayNotification(self.World.Map.Rules, self.Owner, "Speech", info.Notification, self.Owner.Country.Race);
+
+				if (radarPings != null)
+					radarPings.Add(() => self.Owner == self.World.LocalPlayer, self.CenterPosition, info.RadarPingColor, info.RadarPingDuration);
+			}
+
+			lastAttackTime = self.World.WorldTick;
+		}
+	}
+}