From c3fcfa776f5dd9cbb467027600821e7f4e41a4ed Mon Sep 17 00:00:00 2001
From: Chris Forbes <chrisf@ijw.co.nz>
Date: Tue, 23 Mar 2010 08:09:21 +1300
Subject: [PATCH] helper for 'hide all children, then show THIS menu'

---
 OpenRA.Game/Chrome/DefaultWidgetDelegates.cs | 27 +++++++-------------
 OpenRA.Game/Chrome/Widget.cs                 | 12 +++++++++
 2 files changed, 21 insertions(+), 18 deletions(-)

diff --git a/OpenRA.Game/Chrome/DefaultWidgetDelegates.cs b/OpenRA.Game/Chrome/DefaultWidgetDelegates.cs
index 0dce9f5984..5044021620 100644
--- a/OpenRA.Game/Chrome/DefaultWidgetDelegates.cs
+++ b/OpenRA.Game/Chrome/DefaultWidgetDelegates.cs
@@ -112,15 +112,13 @@ namespace OpenRA.Widgets.Delegates
 		{
 			if (w.Id == "MAINMENU_BUTTON_SETTINGS")
 			{
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = false;
-				Game.chrome.rootWidget.GetWidget("SETTINGS_BG").Visible = true;
+				Game.chrome.rootWidget.ShowMenu("SETTINGS_BG");
 				return true;
 			}
 
 			if (w.Id == "SETTINGS_BUTTON_OK")
 			{
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = true;
-				Game.chrome.rootWidget.GetWidget("SETTINGS_BG").Visible = false;
+				Game.chrome.rootWidget.ShowMenu("MAINMENU_BG");
 				return true;
 			}
 
@@ -155,21 +153,19 @@ namespace OpenRA.Widgets.Delegates
 		{
 			if (w.Id == "MAINMENU_BUTTON_CREATE")
 			{
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = false;
-				Game.chrome.rootWidget.GetWidget("CREATESERVER_BG").Visible = true;
+				Game.chrome.rootWidget.ShowMenu("CREATESERVER_BG");
 				return true;
 			}
 			
 			if (w.Id == "CREATESERVER_BUTTON_CANCEL")
 			{
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = true;
-				Game.chrome.rootWidget.GetWidget("CREATESERVER_BG").Visible = false;
+				Game.chrome.rootWidget.ShowMenu("MAINMENU_BG");
 				return true;
 			}
 			
 			if (w.Id == "CREATESERVER_BUTTON_START")
 			{
-				Game.chrome.rootWidget.GetWidget("CREATESERVER_BG").Visible = false;
+				Game.chrome.rootWidget.ShowMenu(null);
 				Log.Write("Creating server");
 				
 				Server.Server.ServerMain(AdvertiseServerOnline, Game.Settings.MasterServer, 
@@ -195,9 +191,7 @@ namespace OpenRA.Widgets.Delegates
 			// Main Menu root
 			if (w.Id == "MAINMENU_BUTTON_JOIN")
 			{
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = false;
-				Widget bg = Game.chrome.rootWidget.GetWidget("JOINSERVER_BG");
-				bg.Visible = true;
+				var bg = Game.chrome.rootWidget.ShowMenu("JOINSERVER_BG");
 				
 				int height = 50;
 				int width = 300;
@@ -242,8 +236,7 @@ namespace OpenRA.Widgets.Delegates
 				
 			if (w.Id == "JOINSERVER_BUTTON_CANCEL")
 			{
-				Game.chrome.rootWidget.GetWidget("JOINSERVER_BG").Visible = false;
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = true;
+				Game.chrome.rootWidget.ShowMenu("MAINMENU_BG");
 				return true;
 			}
 			
@@ -258,15 +251,13 @@ namespace OpenRA.Widgets.Delegates
 			// Main Menu root
 			if (w.Id == "CONNECTION_BUTTON_ABORT")
 			{
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = true;
-				Game.chrome.rootWidget.GetWidget("CONNECTING_BG").Visible = false;
+				Game.chrome.rootWidget.ShowMenu("MAINMENU_BG");
 				return true;
 			}
 			
 			if (w.Id == "CONNECTION_BUTTON_CANCEL")
 			{
-				Game.chrome.rootWidget.GetWidget("MAINMENU_BG").Visible = true;
-				Game.chrome.rootWidget.GetWidget("CONNECTION_FAILED_BG").Visible = false;
+				Game.chrome.rootWidget.ShowMenu("MAINMENU_BG");
 				return true;
 			}
 			
diff --git a/OpenRA.Game/Chrome/Widget.cs b/OpenRA.Game/Chrome/Widget.cs
index 104553a296..9dc2aa55d2 100644
--- a/OpenRA.Game/Chrome/Widget.cs
+++ b/OpenRA.Game/Chrome/Widget.cs
@@ -143,7 +143,19 @@ namespace OpenRA.Widgets
 			
 			return null;
 		}
+
+		public Widget GetCurrentMenu() { return Children.FirstOrDefault(c => c.Visible); }
 		
+		public Widget ShowMenu(string menu)
+		{
+			GetCurrentMenu().Visible = false;
+
+			var widget = GetWidget(menu);
+			if (widget != null)
+				widget.Visible = true;
+
+			return widget;
+		}
 	}
 
 	class ContainerWidget : Widget { }