PKHeX/PKHeX.WinForms/Controls/PKM Editor/TrainerID.cs

using System;
using System.Collections.Generic;
using System.Windows.Forms;
using PKHeX.Core;

namespace PKHeX.WinForms.Controls
{
    public partial class TrainerID : UserControl
    {
        public TrainerID() => InitializeComponent();
        public event EventHandler? UpdatedID;

        private int Format = -1;
        private ITrainerID Trainer = null!;

        public void UpdateTSV()
        {
            var tsv = GetTSV();
            if (tsv < 0)
                return;

            string IDstr = $"TSV: {tsv:d4}{Environment.NewLine}{GetAlternateRepresentation(Trainer, Format)}";
            TSVTooltip.SetToolTip(TB_TID, IDstr);
            TSVTooltip.SetToolTip(TB_SID, IDstr);
            TSVTooltip.SetToolTip(TB_TID7, IDstr);
            TSVTooltip.SetToolTip(TB_SID7, IDstr);
        }

        private static string GetAlternateRepresentation(ITrainerID tr, int format)
        {
            if (format >= 7)
                return $"ID: {tr.TID:D5}/{tr.SID:D5}";
            var repack = (uint)((tr.SID << 16) | tr.TID); // 32bit
            return $"G7ID: ({repack / 1_000_000:D4}){repack % 1_000_000:D6}";
        }

        private int GetTSV()
        {
            if (Format <= 2)
                return -1;
            var xor = Trainer.SID ^ Trainer.TID;
            if (Format <= 5)
                return xor >> 3;
            return xor >> 4;
        }

        public void LoadIDValues(ITrainerID tr)
        {
            Trainer = tr;
            int format = tr.GetTrainerIDFormat();
            SetFormat(format);
            LoadValues();
        }

        public void UpdateSID() => LoadValues();

        public void LoadInfo(ITrainerInfo info)
        {
            Trainer.TID = info.TID;
            Trainer.SID = info.SID;
            LoadValues();
        }

        private void LoadValues()
        {
            if (Format <= 2)
                TB_TID.Text = Trainer.TID.ToString();
            else if (Format <= 6)
                LoadTID(Trainer.TID, Trainer.SID);
            else
                LoadTID7(Trainer.TID, Trainer.SID);
        }

        private void LoadTID(int tid, int sid)
        {
            TB_TID.Text = tid.ToString("D5");
            TB_SID.Text = sid.ToString("D5");
        }

        private void LoadTID7(int tid, int sid)
        {
            var repack = (uint)((sid << 16) | tid);
            sid = (int)(repack / 1_000_000);
            tid = (int)(repack % 1_000_000);

            TB_TID7.Text = tid.ToString("D6");
            TB_SID7.Text = sid.ToString("D4");
        }

        private void SetFormat(int format)
        {
            if (format == Format)
                return;

            var controls = GetControlsForFormat(format);
            FLP.Controls.Clear(); int i = 0;
            foreach (var c in controls)
            {
                FLP.Controls.Add(c);
                FLP.Controls.SetChildIndex(c, i++); // because you don't listen the first time
            }

            Format = format;
        }

        private IEnumerable<Control> GetControlsForFormat(int format) => format switch
        {
            >= 7 => new Control[] {Label_SID, TB_SID7, Label_TID, TB_TID7},
            >= 3 => new Control[] {Label_TID, TB_TID,  Label_SID, TB_SID },
               _ => new Control[] {Label_TID, TB_TID}, // Gen1/2
        };

        private void UpdateTSV(object sender, EventArgs e) => UpdateTSV();

        private void Update_ID(object sender, EventArgs e)
        {
            if (sender is not MaskedTextBox mt)
                return;

            if (!int.TryParse(mt.Text, out var value))
                value = 0;
            if (mt == TB_TID7)
            {
                if (value > 999_999)
                {
                    mt.Text = "999999";
                    return;
                }
                if (!int.TryParse(TB_SID7.Text, out var sid))
                    sid = 0;
                SanityCheckSID7(value, sid);
            }
            else if (mt == TB_SID7)
            {
                if (value > 4294) // max 4 digits of 32bit int
                {
                    mt.Text = "4294";
                    return;
                }
                if (!int.TryParse(TB_TID7.Text, out var tid))
                    tid = 0;
                SanityCheckSID7(tid, value);
            }
            else
            {
                if (value > ushort.MaxValue) // prior to gen7
                    mt.Text = (value = ushort.MaxValue).ToString();

                if (mt == TB_TID)
                    Trainer.TID = value;
                else
                    Trainer.SID = value;
            }

            UpdatedID?.Invoke(sender, e);
        }

        private void SanityCheckSID7(int tid, int sid)
        {
            var repack = ((long)sid * 1_000_000) + tid;
            if (repack > uint.MaxValue)
            {
                TB_SID7.Text = (sid - 1).ToString();
                return;
            }

            Trainer.SID = (ushort)(repack >> 16);
            Trainer.TID = (ushort)repack;
        }
    }
}