PKHeX/PKHeX.WinForms/Subforms/Save Editors/Gen6/SAV_Pokepuff.cs

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

namespace PKHeX.WinForms
{
    public partial class SAV_Pokepuff : Form
    {
        private readonly ISaveBlock6Main SAV;

        public SAV_Pokepuff(SaveFile sav)
        {
            InitializeComponent();
            WinFormsUtil.TranslateInterface(this, Main.CurrentLanguage);
            SAV = (ISaveBlock6Main)sav;

            var puffs = SAV.Puff.GetPuffs();
            Setup(puffs.Length);
            LoadPuffs(puffs);

            toolTip1.SetToolTip(B_Sort, "Hold CTRL to reverse sort.");
            toolTip2.SetToolTip(B_All, "Hold CTRL to best instead of varied.");
        }

        private readonly string[] pfa = GameInfo.Strings.puffs;

        private void Setup(int rowCount)
        {
            dgv.Rows.Clear();
            dgv.Columns.Clear();

            DataGridViewColumn dgvIndex = new DataGridViewTextBoxColumn();
            {
                dgvIndex.HeaderText = "Slot";
                dgvIndex.DisplayIndex = 0;
                dgvIndex.Width = 45;
                dgvIndex.ReadOnly = true;
                dgvIndex.DefaultCellStyle.Alignment = DataGridViewContentAlignment.MiddleCenter;
            }
            DataGridViewComboBoxColumn dgvPuff = new()
            {
                DisplayStyle = DataGridViewComboBoxDisplayStyle.Nothing,
            };
            {
                foreach (string t in pfa)
                    dgvPuff.Items.Add(t);

                dgvPuff.DisplayIndex = 1;
                dgvPuff.Width = 135;
                dgvPuff.FlatStyle = FlatStyle.Flat;
            }
            dgv.Columns.Add(dgvIndex);
            dgv.Columns.Add(dgvPuff);
            dgv.Rows.Add(rowCount);
        }

        private void LoadPuffs(byte[] Puffs)
        {
            for (int i = 0; i < Puffs.Length; i++)
            {
                dgv.Rows[i].Cells[0].Value = (i + 1).ToString();
                int puffval = Puffs[i];
                if (puffval >= pfa.Length)
                {
                    WinFormsUtil.Error($"Invalid Puff Index: {i}", $"Expected < ${pfa.Length}");
                    puffval = 0;
                }
                dgv.Rows[i].Cells[1].Value = pfa[puffval];
            }
        }

        private void DropClick(object sender, DataGridViewCellEventArgs e)
        {
            if (e.ColumnIndex != 1)
                return;
            ((ComboBox)((DataGridView) sender).EditingControl).DroppedDown = true;
        }

        private void B_Cancel_Click(object sender, EventArgs e)
        {
            Close();
        }

        private void B_All_Click(object sender, EventArgs e)
        {
            SAV.Puff.MaxCheat(ModifierKeys == Keys.Control);
            LoadPuffs(SAV.Puff.GetPuffs());
        }

        private void B_None_Click(object sender, EventArgs e)
        {
            SAV.Puff.Reset();
            LoadPuffs(SAV.Puff.GetPuffs());
        }

        private void B_Sort_Click(object sender, EventArgs e)
        {
            SAV.Puff.Sort(ModifierKeys == Keys.Control);
            LoadPuffs(SAV.Puff.GetPuffs());
        }

        private byte[] GetPuffs()
        {
            var puffs = new List<byte>();
            for (int i = 0; i < dgv.Rows.Count; i++)
            {
                var puff = dgv.Rows[i].Cells[1].Value?.ToString();
                int index = (byte)Array.IndexOf(pfa, puff);
                puffs.Add((byte)index);
            }
            return puffs.ToArray();
        }

        private void B_Save_Click(object sender, EventArgs e)
        {
            var puffs = GetPuffs();
            SAV.Puff.SetPuffs(puffs);
            SAV.Puff.PuffCount = puffs.Length;
            Close();
        }
    }
}