gui-cs.Terminal.Gui/Terminal.Gui/Views/ColorPicker.cs

using System;
using System.Reflection;
using NStack;
using static Terminal.Gui.SpinnerStyle;

namespace Terminal.Gui {

	/// <summary>
	/// Event arguments for the <see cref="Color"/> events.
	/// </summary>
	public class ColorEventArgs : EventArgs {

		/// <summary>
		/// Initializes a new instance of <see cref="ColorEventArgs"/>
		/// </summary>
		public ColorEventArgs ()
		{
		}

		/// <summary>
		/// The new Thickness.
		/// </summary>
		public Color Color { get; set; }

		/// <summary>
		/// The previous Thickness.
		/// </summary>
		public Color PreviousColor { get; set; }
	}

	/// <summary>
	/// The <see cref="ColorPicker"/> <see cref="View"/> Color picker.
	/// </summary>
	public class ColorPicker : View {
		private int _selectColorIndex = (int)Color.Black;

		/// <summary>
		/// Columns of color boxes
		/// </summary>
		private int _cols = 8;

		/// <summary>
		/// Rows of color boxes
		/// </summary>
		private int _rows = 2;

		/// <summary>
		/// Width of a color box
		/// </summary>
		public int BoxWidth {
			get => _boxWidth;
			set {
				if (_boxWidth != value) {
					_boxWidth = value;
					SetNeedsLayout ();
				}
			}
		}
		private int _boxWidth = 4;

		/// <summary>
		/// Height of a color box
		/// </summary>
		public int BoxHeight {
			get => _boxHeight;
			set {
				if (_boxHeight != value) {
					_boxHeight = value;
					SetNeedsLayout ();
				}
			}
		}
		private int _boxHeight = 2;

		// Cursor runes.
		private static readonly Rune [] _cursorRunes = new Rune []
		{
			0x250C, 0x2500, 0x2500, 0x2510,
			0x2514, 0x2500, 0x2500, 0x2518
		};

		/// <summary>
		/// Cursor for the selected color.
		/// </summary>
		public Point Cursor {
			get {
				return new Point (_selectColorIndex % _cols, _selectColorIndex / _cols);
			}

			set {
				var colorIndex = value.Y * _cols + value.X;
				SelectedColor = (Color)colorIndex;
			}
		}

		/// <summary>
		/// Fired when a color is picked.
		/// </summary>
		public event EventHandler<ColorEventArgs> ColorChanged;

		/// <summary>
		/// Selected color.
		/// </summary>
		public Color SelectedColor {
			get {
				return (Color)_selectColorIndex;
			}

			set {
				Color prev = (Color)_selectColorIndex;
				_selectColorIndex = (int)value;
				ColorChanged?.Invoke (this, new ColorEventArgs () {
					PreviousColor = prev,
					Color = value,
				});
				SetNeedsDisplay ();
			}
		}

		/// <summary>
		/// Initializes a new instance of <see cref="ColorPicker"/>.
		/// </summary>
		public ColorPicker ()
		{
			SetInitialProperties ();
		}

		private void SetInitialProperties ()
		{
			CanFocus = true;
			AddCommands ();
			AddKeyBindings ();
			LayoutStarted += (o, a) => {
				Bounds = new Rect (Bounds.Location, new Size (_cols * BoxWidth, _rows * BoxHeight));
			};
		}

		/// <summary>
		/// Add the commands.
		/// </summary>
		private void AddCommands ()
		{
			AddCommand (Command.Left, () => MoveLeft ());
			AddCommand (Command.Right, () => MoveRight ());
			AddCommand (Command.LineUp, () => MoveUp ());
			AddCommand (Command.LineDown, () => MoveDown ());
		}

		/// <summary>
		/// Add the KeyBindinds.
		/// </summary>
		private void AddKeyBindings ()
		{
			AddKeyBinding (Key.CursorLeft, Command.Left);
			AddKeyBinding (Key.CursorRight, Command.Right);
			AddKeyBinding (Key.CursorUp, Command.LineUp);
			AddKeyBinding (Key.CursorDown, Command.LineDown);
		}

		///<inheritdoc/>
		public override void Redraw (Rect bounds)
		{
			base.Redraw (bounds);

			Driver.SetAttribute (HasFocus ? ColorScheme.Focus : GetNormalColor ());
			var colorIndex = 0;

			for (var y = 0; y < (Bounds.Height / BoxHeight); y++) {
				for (var x = 0; x < (Bounds.Width / BoxWidth); x++) {
					var foregroundColorIndex = y == 0 ? colorIndex + _cols : colorIndex - _cols;
					Driver.SetAttribute (Driver.MakeAttribute ((Color)foregroundColorIndex, (Color)colorIndex));
					var selected = x == Cursor.X && y == Cursor.Y;
					DrawColorBox (x, y, selected);
					colorIndex++;
				}
			}
		}

		/// <summary>
		/// Draw a box for one color.
		/// </summary>
		/// <param name="x">X location.</param>
		/// <param name="y">Y location</param>
		/// <param name="selected"></param>
		private void DrawColorBox (int x, int y, bool selected)
		{
			var index = 0;

			for (var zoomedY = 0; zoomedY < BoxHeight; zoomedY++) {
				for (var zoomedX = 0; zoomedX < BoxWidth; zoomedX++) {
					Move (x * BoxWidth + zoomedX, y * BoxHeight + zoomedY);
					Driver.AddRune (' ');
					index++;
				}
			}

			if (selected) {
				DrawFocusRect (new Rect (x * BoxWidth, y * BoxHeight, BoxWidth, BoxHeight));
			}
		}

		private void DrawFocusRect (Rect rect)
		{
			var lc = new LineCanvas ();
			if (rect.Width == 1) {
				lc.AddLine (rect.Location, rect.Height, Orientation.Vertical, LineStyle.Dotted);
			} else if (rect.Height == 1) {
				lc.AddLine (rect.Location, rect.Width, Orientation.Horizontal, LineStyle.Dotted);
			} else {
				lc.AddLine (rect.Location, rect.Width, Orientation.Horizontal, LineStyle.Dotted);
				lc.AddLine (new Point (rect.Location.X, rect.Location.Y + rect.Height - 1), rect.Width, Orientation.Horizontal, LineStyle.Dotted);

				lc.AddLine (rect.Location, rect.Height, Orientation.Vertical, LineStyle.Dotted);
				lc.AddLine (new Point (rect.Location.X + rect.Width - 1, rect.Location.Y), rect.Height, Orientation.Vertical, LineStyle.Dotted);
			}
			foreach (var p in lc.GetMap ()) {
				AddRune (p.Key.X, p.Key.Y, p.Value);
			}
		}

		/// <summary>
		/// Moves the selected item index to the previous column.
		/// </summary>
		/// <returns></returns>
		public virtual bool MoveLeft ()
		{
			if (Cursor.X > 0) SelectedColor--;
			return true;
		}

		/// <summary>
		/// Moves the selected item index to the next column.
		/// </summary>
		/// <returns></returns>
		public virtual bool MoveRight ()
		{
			if (Cursor.X < _cols - 1) SelectedColor++;
			return true;
		}

		/// <summary>
		/// Moves the selected item index to the previous row.
		/// </summary>
		/// <returns></returns>
		public virtual bool MoveUp ()
		{
			if (Cursor.Y > 0) SelectedColor -= _cols;
			return true;
		}

		/// <summary>
		/// Moves the selected item index to the next row.
		/// </summary>
		/// <returns></returns>
		public virtual bool MoveDown ()
		{
			if (Cursor.Y < _rows - 1) SelectedColor += _cols;
			return true;
		}

		///<inheritdoc/>
		public override bool ProcessKey (KeyEvent kb)
		{
			var result = InvokeKeybindings (kb);
			if (result != null)
				return (bool)result;

			return false;
		}

		///<inheritdoc/>
		public override bool MouseEvent (MouseEvent me)
		{
			if (!me.Flags.HasFlag (MouseFlags.Button1Clicked) || !CanFocus) {
				return false;
			}

			SetFocus ();
			Cursor = new Point ((me.X - GetFramesThickness ().Left) / _boxWidth, (me.Y - GetFramesThickness ().Top) / _boxHeight);

			return true;
		}
	}
}