rename ChoosePaletteOnSelect

OpenRA.Mods.RA/World/ChooseBuildTabOnSelect.cs

@@ -0,0 +1,47 @@
+#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.Linq;
+using OpenRA.Traits;
+using OpenRA.Widgets;
+
+namespace OpenRA.Mods.RA.Widgets
+{
+	class ChooseBuildTabOnSelectInfo : ITraitInfo
+	{
+		public object Create( ActorInitializer init ) { return new ChooseBuildTabOnSelect( init ); }
+	}
+
+	class ChooseBuildTabOnSelect : INotifySelection
+	{
+		readonly World world;
+
+		public ChooseBuildTabOnSelect( ActorInitializer init )
+		{
+			world = init.world;
+		}
+
+		public void SelectionChanged()
+		{
+			var firstItem = world.Selection.Actors.FirstOrDefault(
+				a => a.World.LocalPlayer == a.Owner && a.HasTrait<Production>());
+
+			if (firstItem == null)
+				return;
+
+			var produces = firstItem.Info.Traits.Get<ProductionInfo>().Produces.FirstOrDefault();
+			if (produces == null)
+				return;
+
+			Widget.RootWidget.GetWidget<BuildPaletteWidget>("INGAME_BUILD_PALETTE")
+				.SetCurrentTab(produces);
+		}
+	}
+}