Fix lobby color picker

OpenRA.Mods.Cnc/CncColorPickerPaletteModifier.cs

@@ -0,0 +1,48 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2011 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.Collections.Generic;
+using System.Linq;
+using OpenRA.FileFormats;
+using OpenRA.Graphics;
+using OpenRA.Mods.Cnc.Widgets;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Cnc
+{
+	class CncColorPickerPaletteModifierInfo : ITraitInfo
+	{
+		public string PlayerPalette = "player";
+		public object Create( ActorInitializer init ) { return new CncColorPickerPaletteModifier( this ); }
+	}
+	
+	class CncColorPickerPaletteModifier : IPalette, IPaletteModifier
+	{	
+		CncColorPickerPaletteModifierInfo Info;
+		PaletteFormat format;
+		
+		public CncColorPickerPaletteModifier(CncColorPickerPaletteModifierInfo info) { Info = info; }
+		
+		public void InitPalette( WorldRenderer wr )
+		{
+			var info = Rules.Info["player"].Traits.WithInterface<PlayerColorPaletteInfo>()
+				.Where(p => p.BaseName == Info.PlayerPalette)
+				.First();
+			format = info.PaletteFormat;
+			wr.AddPalette("colorpicker", wr.GetPalette(info.BasePalette));
+		}
+
+		public void AdjustPalette(Dictionary<string, Palette> palettes)
+		{
+			palettes["colorpicker"] = new Palette(palettes["colorpicker"],
+			  new PlayerColorRemap(CncLobbyLogic.CurrentColorPreview, format));
+		}
+	}
+}