pkNX/pkNX.WinForms/Subforms/MapViewer8a.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Windows.Forms;
using pkNX.Containers;
using pkNX.Game;
using pkNX.Structures;
using pkNX.Structures.FlatBuffers;

namespace pkNX.WinForms.Subforms
{
    public partial class MapViewer8a : Form
    {
        private readonly GameManager ROM;
        private readonly GFPack Resident;

        public readonly AreaInstance8a[] Areas;
        private readonly bool Loading = true;

        public MapViewer8a(GameManager rom, GFPack resident)
        {
            ROM = rom;
            Resident = resident;
            InitializeComponent();

            var pd = ROM.GetFile(GameFile.PersonalStats)[0];
            var po = FlatBufferConverter.DeserializeFrom<PersonalTableLA>(pd);
            var test = PersonalConverter.FromArceus(po);
            var pt = new PersonalTable(test, 905);
            Areas = PLAInfo.AreaNames.Select(z => AreaInstance8a.Create(Resident, z)).ToArray();
            var speciesNames = ROM.GetStrings(TextName.SpeciesNames);
            CB_Map.Items.AddRange(Areas.Select(z => z.ParentArea?.AreaName ?? z.AreaName).ToArray());

            var nameList = new List<ComboItem>();
            foreach (var e in pt.Table.OfType<PersonalInfoLA>())
            {
                if (!e.IsPresentInGame)
                    continue;

                var species = e.Species;
                if (!nameList.Any(z => z.Value == species))
                    nameList.Add(new(speciesNames[species], species));
            }

            nameList.Sort((x, y) => x.Text.CompareTo(y.Text));

            CB_Species.DisplayMember = nameof(ComboItem.Text);
            CB_Species.ValueMember = nameof(ComboItem.Value);
            CB_Species.DataSource = new BindingSource(nameList, null);

            CB_Species.SelectedValue = 399;
            Loading = false;
            CB_Map.SelectedIndex = 0;
        }
        public class ComboItem
        {
            public ComboItem(string text, int value)
            {
                Text = text;
                Value = value;
            }

            public string Text { get; }
            public int Value { get; }
        }

        private void CB_Map_SelectedIndexChanged(object sender, EventArgs e)
        {
            UpdateMap(CB_Map.SelectedIndex, (int)CB_Species.SelectedValue);
        }

        private void CB_Species_SelectedIndexChanged(object sender, EventArgs e)
        {
            UpdateMap(CB_Map.SelectedIndex, (int)CB_Species.SelectedValue);
        }

        private void UpdateMap(int map, int species)
        {
            if (Loading)
                return;

            var mapfile = $"map_pla\\map_lmap_pic_l_{map-1:00}.png";
            if (!System.IO.File.Exists(mapfile))
            {
                pictureBox1.BackgroundImage = null;
                return;
            }

            var area = Areas[map];
            var coordinates = GetSpawnerInfo(species, area);

            var img = Image.FromFile(mapfile);
            using var gr = Graphics.FromImage(img);
            var r = new SolidBrush(Color.FromArgb(100, 255, 0, 0));
            var g = new SolidBrush(Color.FromArgb(100, 20, 255, 10));
            var b = new SolidBrush(Color.FromArgb(100, 10, 0, 255));
            var c = new SolidBrush(Color.FromArgb(100, 0, 255, 255));

            var rs = new Pen(Color.FromArgb(255, 255, 0, 0)) { Width = 3 };
            var gs = new Pen(Color.FromArgb(255, 20, 255, 10)) { Width = 3 };
            var bs = new Pen(Color.FromArgb(255, 10, 0, 255)) { Width = 3 };
            var cs = new Pen(Color.FromArgb(255, 10, 255, 255)) { Width = 3 };
            foreach (var o in coordinates)
            {
                var pen = o.Type switch
                {
                    SpawnerType.Spawner => r,
                    SpawnerType.Wormhole => g,
                    SpawnerType.Landmark => b,
                    SpawnerType.Unown => c,
                    _ => throw new ArgumentOutOfRangeException(nameof(o.Type)),
                };
                var penS = o.Type switch
                {
                    SpawnerType.Spawner => rs,
                    SpawnerType.Wormhole => gs,
                    SpawnerType.Landmark => bs,
                    SpawnerType.Unown => cs,
                    _ => throw new ArgumentOutOfRangeException(nameof(o.Type)),
                };
                var p = o.Position;
                var s = o.Scale;
                var x = (p.X * 2) - (s / 2);
                var y = (p.Z * 2) - (s / 2);
                gr.FillEllipse(pen, x, y, s, s);
                gr.DrawEllipse(penS, x, y, s, s);
            }

            pictureBox1.BackgroundImage = img;
        }

        private static List<AreaDef> GetSpawnerInfo(int species, AreaInstance8a area)
        {
            var result = new List<AreaDef>();

            foreach (var s in area.Spawners.Concat(area.SubAreas.SelectMany(z => z.Spawners)))
            {
                var table = s.Field_20_Value.EncounterTableID;
                var slots = Array.Find(area.Encounters.Table, z => z.TableID == table);
                if (slots == null)
                    continue;

                if (slots.Table.All(z => z.Species != species))
                    continue;

                result.Add(new(s.Parameters.Coordinates, SpawnerType.Spawner, slots.Table, s.Scalar * 4));
            }

            foreach (var s in area.Wormholes.Concat(area.SubAreas.SelectMany(z => z.Wormholes)))
            {
                var table = s.Field_20_Value.EncounterTableID;
                var slots = Array.Find(area.Encounters.Table, z => z.TableID == table);
                if (slots == null)
                    continue;

                if (slots.Table.All(z => z.Species != species))
                    continue;

                result.Add(new(s.Parameters.Coordinates, SpawnerType.Wormhole, slots.Table, Math.Max(s.Scalar * 4, 50)));
            }

            foreach (var a in area.LandMarks.Concat(area.SubAreas.SelectMany(z => z.LandMarks)))
            {
                var table = a.LandmarkItemSpawnTableID;
                foreach (var l in area.LandItems.Concat(area.SubAreas.SelectMany(z => z.LandItems)))
                {
                    if (l.LandmarkItemSpawnTableID != table)
                        continue;
                    var st = l.EncounterTableID;
                    var slots = Array.Find(area.Encounters.Table, z => z.TableID == st);
                    if (slots == null)
                        continue;

                    if (slots.Table.All(z => z.Species != species))
                        continue;

                    result.Add(new(a.Parameters.Coordinates, SpawnerType.Landmark, slots.Table, Math.Max(a.Scalar, 1) * 4));
                }
            }

            if (species is not 201)
                return result;

            foreach (var u in area.Unown.Concat(area.SubAreas.SelectMany(z => z.Unown)))
            {
                var slots = Unown;
                result.Add(new(u.Parameters.Coordinates, SpawnerType.Unown, slots, u.Number * 2));
            }

            return result;
        }

        private static readonly EncounterSlot8a[] Unown = { new() { Species = 201 } };
    }

    public class AreaDef
    {
        public readonly PlacementV3f8a Position;
        public readonly SpawnerType Type;
        public readonly EncounterSlot8a[] Slots;
        public readonly float Scale;

        public AreaDef(PlacementV3f8a position, SpawnerType type, EncounterSlot8a[] slots, float scale)
        {
            Position = position;
            Type = type;
            Slots = slots;
            Scale = scale;
        }
    }
}