Initial commit: OpenRA game engine

Fork from OpenRA/OpenRA with one-click launch script (start-ra.cmd)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

OpenRA.Mods.Common/Widgets/Logic/ColorPickerLogic.cs

@@ -0,0 +1,191 @@
+#region Copyright & License Information
+/*
+ * Copyright (c) The OpenRA Developers and Contributors
+ * 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 System;
+using System.Collections.Generic;
+using System.Collections.Immutable;
+using System.Linq;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Primitives;
+using OpenRA.Widgets;
+
+namespace OpenRA.Mods.Common.Widgets.Logic
+{
+	public class ColorPickerLogic : ChromeLogic
+	{
+		static bool paletteTabOpenedLast;
+		int paletteTabHighlighted = 0;
+
+		[ObjectCreator.UseCtor]
+		public ColorPickerLogic(Widget widget, ModData modData, World world, Color initialColor, Action<Color> onChange, Action<Widget> extraLogic,
+			Dictionary<string, MiniYaml> logicArgs)
+		{
+			var mixer = widget.Get<ColorMixerWidget>("MIXER");
+			var hueSlider = widget.Get<HueSliderWidget>("HUE_SLIDER");
+
+			// Set the initial state
+			// All users need to use the same TraitInfo instance, chosen as the default mod rules
+			var colorManager = modData.DefaultRules.Actors[SystemActors.World].TraitInfo<IColorPickerManagerInfo>();
+
+			var (sMin, sMax) = colorManager.SaturationRange;
+			var (vMin, vMax) = colorManager.ValueRange;
+			mixer.SetColorLimits(sMin, sMax, vMin, vMax);
+			mixer.OnChange += () => onChange(mixer.Color);
+			mixer.Set(initialColor);
+
+			hueSlider.OnChange += h =>
+			{
+				mixer.SetColorLimits(sMin, sMax, vMin, vMax, h);
+				var (_, _, s, v) = mixer.Color.ToAhsv();
+				mixer.Set(Color.FromAhsv(h, s, v));
+				onChange(mixer.Color);
+			};
+
+			hueSlider.UpdateValue(initialColor.ToAhsv().H);
+
+			var randomButton = widget.GetOrNull<ButtonWidget>("RANDOM_BUTTON");
+			if (randomButton != null)
+			{
+				var terrainColors = modData.DefaultTerrainInfo
+					.SelectMany(t => t.Value.RestrictedPlayerColors)
+					.Distinct()
+					.ToList();
+				var playerColors = Array.Empty<Color>();
+				randomButton.OnClick = () =>
+				{
+					var randomColor = colorManager.RandomValidColor(world.LocalRandom, terrainColors, playerColors);
+					mixer.Set(randomColor);
+					hueSlider.UpdateValue(randomColor.ToAhsv().H);
+				};
+			}
+
+			// HACK: the value returned from the color mixer will generally not
+			// be equal to the given initialColor due to its internal RGB -> HSL -> RGB
+			// conversion. This conversion can sometimes convert a valid initial value
+			// into an invalid (too close to terrain / another player) color.
+			// We use the original colour here instead of the mixer color to make sure
+			// that we keep the player's previous colour value if they don't change anything
+			onChange(initialColor);
+
+			// Setup tab controls
+			var mixerTab = widget.Get("MIXER_TAB");
+			var paletteTab = widget.Get("PALETTE_TAB");
+			var paletteTabPanel = widget.Get("PALETTE_TAB_PANEL");
+			var mixerTabButton = widget.Get<ButtonWidget>("MIXER_TAB_BUTTON");
+			var paletteTabButton = widget.Get<ButtonWidget>("PALETTE_TAB_BUTTON");
+			var presetArea = paletteTabPanel.Get<ContainerWidget>("PRESET_AREA");
+			var customArea = paletteTabPanel.Get<ContainerWidget>("CUSTOM_AREA");
+			var presetColorTemplate = paletteTabPanel.Get<ColorBlockWidget>("COLORPRESET");
+			var customColorTemplate = paletteTabPanel.Get<ColorBlockWidget>("COLORCUSTOM");
+
+			mixerTab.IsVisible = () => !paletteTabOpenedLast;
+			mixerTabButton.OnClick = () => paletteTabOpenedLast = false;
+			mixerTabButton.IsHighlighted = mixerTab.IsVisible;
+
+			paletteTab.IsVisible = () => paletteTabOpenedLast;
+			paletteTabButton.OnClick = () => paletteTabOpenedLast = true;
+			paletteTabButton.IsHighlighted = () => paletteTab.IsVisible() || paletteTabHighlighted > 0;
+
+			var paletteCols = 8;
+			var palettePresetRows = 2;
+			var paletteCustomRows = 1;
+
+			if (logicArgs.TryGetValue("PaletteColumns", out var yaml) && !int.TryParse(yaml.Value, out paletteCols))
+				throw new YamlException($"Invalid value for PaletteColumns: {yaml.Value}");
+			if (logicArgs.TryGetValue("PalettePresetRows", out yaml) && !int.TryParse(yaml.Value, out palettePresetRows))
+				throw new YamlException($"Invalid value for PalettePresetRows: {yaml.Value}");
+			if (logicArgs.TryGetValue("PaletteCustomRows", out yaml) && !int.TryParse(yaml.Value, out paletteCustomRows))
+				throw new YamlException($"Invalid value for PaletteCustomRows: {yaml.Value}");
+
+			var presetColors = colorManager.PresetColors;
+			for (var j = 0; j < palettePresetRows; j++)
+			{
+				for (var i = 0; i < paletteCols; i++)
+				{
+					var colorIndex = j * paletteCols + i;
+					if (colorIndex >= presetColors.Length)
+						break;
+
+					var color = presetColors[colorIndex];
+
+					var newSwatch = presetColorTemplate.Clone();
+					newSwatch.GetColor = () => color;
+					newSwatch.IsVisible = () => true;
+					newSwatch.Bounds.X = i * newSwatch.Bounds.Width;
+					newSwatch.Bounds.Y = j * newSwatch.Bounds.Height;
+					newSwatch.OnMouseUp = m =>
+					{
+						mixer.Set(color);
+						hueSlider.UpdateValue(color.ToAhsv().H);
+					};
+
+					presetArea.AddChild(newSwatch);
+				}
+			}
+
+			for (var j = 0; j < paletteCustomRows; j++)
+			{
+				for (var i = 0; i < paletteCols; i++)
+				{
+					var colorIndex = j * paletteCols + i;
+
+					var newSwatch = customColorTemplate.Clone();
+					var getColor = new CachedTransform<Color, Color>(c => colorManager.MakeValid(c, world.LocalRandom, [], []));
+
+					newSwatch.GetColor = () => getColor.Update(Game.Settings.Player.CustomColors[colorIndex]);
+					newSwatch.IsVisible = () => Game.Settings.Player.CustomColors.Length > colorIndex;
+					newSwatch.Bounds.X = i * newSwatch.Bounds.Width;
+					newSwatch.Bounds.Y = j * newSwatch.Bounds.Height;
+					newSwatch.OnMouseUp = m =>
+					{
+						var color = newSwatch.GetColor();
+						mixer.Set(color);
+						hueSlider.UpdateValue(color.ToAhsv().H);
+					};
+
+					customArea.AddChild(newSwatch);
+				}
+			}
+
+			// Store color button
+			var storeButton = widget.Get<ButtonWidget>("STORE_BUTTON");
+			if (storeButton != null)
+			{
+				storeButton.OnClick = () =>
+				{
+					// Update the custom color list:
+					//  - Remove any duplicates of the new color
+					//  - Add the new color to the end
+					//  - Save the last N colors
+					Game.Settings.Player.CustomColors = Game.Settings.Player.CustomColors
+						.Where(c => c != mixer.Color)
+						.Append(mixer.Color)
+						.Reverse().Take(paletteCustomRows * paletteCols).Reverse()
+						.ToImmutableArray();
+					Game.Settings.Save();
+
+					// Flash the palette tab to show players that something has happened
+					if (!paletteTabOpenedLast)
+						paletteTabHighlighted = 4;
+				};
+			}
+
+			// Attach logic to preview actor.
+			extraLogic(widget);
+		}
+
+		public override void Tick()
+		{
+			if (paletteTabHighlighted > 0)
+				paletteTabHighlighted--;
+		}
+	}
+}