Implement ExitsDebugOverlay and ExitsDebugOverlayManager

OpenRA.Mods.Common/Traits/World/ExitsDebugOverlayManager.cs

@@ -0,0 +1,65 @@
+#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, either version 3 of
+ * the License, or (at your option) any later version. For more
+ * information, see COPYING.
+ */
+#endregion
+
+using OpenRA.Graphics;
+using OpenRA.Mods.Common.Commands;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Traits
+{
+	public class ExitsDebugOverlayManagerInfo : ITraitInfo
+	{
+		[Desc("The font used to draw cell vectors. Should match the value as-is in the Fonts section of the mod manifest (do not convert to lowercase).")]
+		public readonly string Font = "TinyBold";
+
+		object ITraitInfo.Create(ActorInitializer init) { return new ExitsDebugOverlayManager(init.Self, this); }
+	}
+
+	public class ExitsDebugOverlayManager : IWorldLoaded, IChatCommand
+	{
+		const string CommandName = "exits-overlay";
+		const string CommandHelp = "Displays exits for factories.";
+
+		public readonly SpriteFont Font;
+		public readonly ExitsDebugOverlayManagerInfo Info;
+
+		public bool Enabled;
+
+		readonly Actor self;
+
+		public ExitsDebugOverlayManager(Actor self, ExitsDebugOverlayManagerInfo info)
+		{
+			this.self = self;
+			Info = info;
+
+			if (!Game.Renderer.Fonts.TryGetValue(info.Font, out Font))
+				throw new YamlException("Could not find font '{0}'".F(info.Font));
+		}
+
+		void IWorldLoaded.WorldLoaded(World w, WorldRenderer wr)
+		{
+			var console = self.TraitOrDefault<ChatCommands>();
+			var help = self.TraitOrDefault<HelpCommand>();
+
+			if (console == null || help == null)
+				return;
+
+			console.RegisterCommand(CommandName, this);
+			help.RegisterHelp(CommandName, CommandHelp);
+		}
+
+		void IChatCommand.InvokeCommand(string command, string arg)
+		{
+			if (command == CommandName)
+				Enabled ^= true;
+		}
+	}
+}