using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Windows.Forms;
using PKHeX.Core;
using PKHeX.Drawing;
using PKHeX.Drawing.Misc;
using PKHeX.Drawing.PokeSprite;

namespace PKHeX.WinForms
{
    public partial class KChart : Form
    {
        private readonly SaveFile SAV;
        private readonly string[] species = GameInfo.Strings.specieslist;
        private readonly string[] abilities = GameInfo.Strings.abilitylist;
        private readonly int[] baseForm;
        private readonly int[] formVal;

        public KChart(SaveFile sav)
        {
            InitializeComponent();
            Icon = Properties.Resources.Icon;
            WinFormsUtil.TranslateInterface(this, Main.CurrentLanguage);
            SAV = sav;

            Array.Resize(ref species, SAV.Personal.TableLength);

            var forms = SAV.Personal.GetFormList(species, SAV.MaxSpeciesID);
            species = SAV.Personal.GetPersonalEntryList(forms, species, SAV.MaxSpeciesID, out baseForm, out formVal);

            DGV.Rows.Clear();
            for (int i = 1; i < species.Length; i++)
                PopEntry(i);

            DGV.DoubleBuffered(true);

            DGV.Sort(DGV.Columns[0], ListSortDirection.Ascending);
        }

        private void PopEntry(int index)
        {
            var p = SAV.Personal[index];
            if (p.HP == 0)
                return;

            int s = index > SAV.MaxSpeciesID ? baseForm[index] : index;
            var f = index <= SAV.MaxSpeciesID ? 0 : formVal[index];

            var row = new DataGridViewRow();
            row.CreateCells(DGV);

            int r = 0;
            row.Cells[r++].Value = s.ToString("000") + (f > 0 ? $"-{f:00}" : "");
            row.Cells[r++].Value = SpriteUtil.GetSprite(s, f, 0, 0, 0, false, false, SAV.Generation);
            row.Cells[r++].Value = species[index];
            row.Cells[r++].Value = GetIsNative(p, s);
            row.Cells[r].Style.BackColor = ColorUtil.ColorBaseStatTotal(p.BST);
            row.Cells[r++].Value = p.BST.ToString("000");
            row.Cells[r++].Value = p.CatchRate.ToString("000");
            row.Cells[r++].Value = TypeSpriteUtil.GetTypeSprite(p.Type1, SAV.Generation);
            row.Cells[r++].Value = p.Type1 == p.Type2 ? SpriteUtil.Spriter.Transparent : TypeSpriteUtil.GetTypeSprite(p.Type2, SAV.Generation);
            row.Cells[r].Style.BackColor = ColorUtil.ColorBaseStat(p.HP);
            row.Cells[r++].Value = p.HP.ToString("000");
            row.Cells[r].Style.BackColor = ColorUtil.ColorBaseStat(p.ATK);
            row.Cells[r++].Value = p.ATK.ToString("000");
            row.Cells[r].Style.BackColor = ColorUtil.ColorBaseStat(p.DEF);
            row.Cells[r++].Value = p.DEF.ToString("000");
            row.Cells[r].Style.BackColor = ColorUtil.ColorBaseStat(p.SPA);
            row.Cells[r++].Value = p.SPA.ToString("000");
            row.Cells[r].Style.BackColor = ColorUtil.ColorBaseStat(p.SPD);
            row.Cells[r++].Value = p.SPD.ToString("000");
            row.Cells[r].Style.BackColor = ColorUtil.ColorBaseStat(p.SPE);
            row.Cells[r++].Value = p.SPE.ToString("000");
            var abils = p.Abilities;
            row.Cells[r++].Value = GetAbility(abils, 0);
            row.Cells[r++].Value = GetAbility(abils, 1);
            row.Cells[r].Value = GetAbility(abils, 2);
            row.Height = SpriteUtil.Spriter.Height + 1;
            DGV.Rows.Add(row);
        }

        private string GetAbility(IReadOnlyList<int> abilityIDs, int index)
        {
            if ((uint)index >= abilityIDs.Count)
                return abilities[0];
            return abilities[abilityIDs[index]];
        }

        private static bool GetIsNative(PersonalInfo personalInfo, int s) => personalInfo switch
        {
            PersonalInfoSM => s > 721 || Legal.PastGenAlolanNatives.Contains(s),
            PersonalInfoSWSH ss => ss.IsInDex,
            PersonalInfoBDSP bs => bs.IsInDex,
            _ => true,
        };
    }
}
