Mesen-S/UI/Forms/Config/Controllers/BaseInputConfigControl.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using Mesen.GUI.Config;
using Mesen.GUI.Controls;

namespace Mesen.GUI.Forms.Config
{
	public class BaseInputConfigControl : BaseControl
	{
		private ToolTip _toolTip = new System.Windows.Forms.ToolTip();

		public event EventHandler Change;
		protected HashSet<Button> _buttons = new HashSet<Button>();

		public enum MappedKeyType
		{
			None,
			Keyboard,
			Controller
		}

		public BaseInputConfigControl()
		{
			_toolTip.AutomaticDelay = 0;
			_toolTip.AutoPopDelay = 32700;
			_toolTip.InitialDelay = 1000;
			_toolTip.ReshowDelay = 10;
		}

		public virtual void Initialize(KeyMapping mappings) { }
		public virtual void UpdateKeyMappings(ref KeyMapping mappings) { }

		protected void InitButton(Button btn, UInt32 scanCode)
		{
			if(!_buttons.Contains(btn)) {
				_buttons.Add(btn);
				btn.Click += btnMapping_Click;
				btn.MouseUp += btnMapping_MouseUp;
				btn.AutoEllipsis = true;
			}
			btn.Text = InputApi.GetKeyName(scanCode);
			_toolTip.SetToolTip(btn, btn.Text);
			btn.Tag = scanCode;
		}

		protected void OnChange()
		{
			this.Change?.Invoke(this, EventArgs.Empty);
		}

		public MappedKeyType GetKeyType()
		{
			MappedKeyType keyType = MappedKeyType.None;
			foreach(Button btn in _buttons) {
				if((UInt32)btn.Tag > 0xFFFF) {
					return MappedKeyType.Controller;
				} else if((UInt32)btn.Tag > 0) {
					keyType = MappedKeyType.Keyboard;
				}
			}
			return keyType;
		}

		public void ClearKeys()
		{
			foreach(Button btn in _buttons) {
				InitButton(btn, 0);
			}
			this.OnChange();
		}

		protected void btnMapping_Click(object sender, EventArgs e)
		{
			using(frmGetKey frm = new frmGetKey(true)) {
				((Form)frm).ShowDialog(this);
				((Button)sender).Text = frm.ShortcutKey.ToString();
				((Button)sender).Tag = frm.ShortcutKey.Key1;
				_toolTip.SetToolTip((Button)sender, ((Button)sender).Text);
			}
			this.OnChange();
		}

		protected void btnMapping_MouseUp(object sender, MouseEventArgs e)
		{
			if(e.Button == MouseButtons.Right) {
				//Clear shortcut on right-click
				((Button)sender).Text = "";
				((Button)sender).Tag = 0U;
				_toolTip.SetToolTip((Button)sender, ((Button)sender).Text);
			}
			this.OnChange();
		}
	}
}