NHSE/NHSE.WinForms/Subforms/Map/PatternEditor.cs

using NHSE.Core;
using NHSE.Sprites;
using System;
using System.Drawing;
using System.Drawing.Imaging;
using System.IO;
using System.Runtime.Intrinsics.Arm;
using System.Windows.Forms;

namespace NHSE.WinForms;

public partial class PatternEditor : Form
{
    private readonly DesignPattern[] Patterns;
    private readonly Player Player;

    private int Index;
    private const int scale = 8;

    public PatternEditor(DesignPattern[] patterns, Player player)
    {
        InitializeComponent();
        this.TranslateInterface(GameInfo.CurrentLanguage);
        Patterns = patterns;
        Player = player;
        DialogResult = DialogResult.Cancel;

        foreach (var p in patterns)
            LB_Items.Items.Add(GetPatternSummary(p));

        LB_Items.SelectedIndex = 0;
        SetCM_DLDesign();
    }

    private void SetCM_DLDesign()
    {
        var arr = LB_Items.Items.Count == 0
            ? new ToolStripItem[] { B_DumpDesign, B_LoadDesign }
            : new ToolStripItem[] { B_DumpDesign, B_DumpDesignAll, B_LoadDesign, B_LoadDesignAll };
        CM_DLDesign.Items.AddRange(arr);
    }

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

    private void B_Save_Click(object sender, EventArgs e)
    {
        DialogResult = DialogResult.OK;
        Close();
    }

    private void LB_Items_SelectedIndexChanged(object sender, EventArgs e)
    {
        if (LB_Items.SelectedIndex < 0)
            return;
        LoadPattern(Patterns[Index = LB_Items.SelectedIndex]);
    }

    private static string GetPatternSummary(DesignPattern p) => p.DesignName;
    private static void ShowContextMenuBelow(ToolStripDropDown c, Control n) => c.Show(n.PointToScreen(new Point(0, n.Height)));
    private void B_DumpLoadDesign_Click(object sender, EventArgs e) => ShowContextMenuBelow(CM_DLDesign, B_DumpLoadDesign);

    private void B_DumpDesign_Click(object sender, EventArgs e)
    {
        if (sender == B_DumpDesignAll)
        {
            using var fbd = new FolderBrowserDialog();
            if (fbd.ShowDialog() != DialogResult.OK)
                return;

            var dir = Path.GetDirectoryName(fbd.SelectedPath);
            if (dir == null || !Directory.Exists(dir))
                return;
            bool indexed = WinFormsUtil.Prompt(MessageBoxButtons.YesNo, MessageStrings.MsgAskExportResultsWithFileIndex) == DialogResult.Yes;
            Patterns.Dump(fbd.SelectedPath, indexed);
            return;
        }

        var original = Patterns[Index];
        var name = original.DesignName;
        using var sfd = new SaveFileDialog();
        sfd.Filter = "New Horizons Design Pattern (*.nhd)|*.nhd|All files (*.*)|*.*";
        sfd.FileName = $"{name}.nhd";
        if (sfd.ShowDialog() != DialogResult.OK)
            return;

        var d = original;
        File.WriteAllBytes(sfd.FileName, d.Data);
    }

    private void B_LoadDesign_Click(object sender, EventArgs e)
    {
        if (sender == B_LoadDesignAll)
        {
            using var fbd = new FolderBrowserDialog();
            if (fbd.ShowDialog() != DialogResult.OK)
                return;

            var dir = Path.GetDirectoryName(fbd.SelectedPath);
            if (dir == null || !Directory.Exists(dir))
                return;
            var result = WinFormsUtil.Prompt(MessageBoxButtons.YesNoCancel, MessageStrings.MsgAskUpdateValues);
            Patterns.Load(fbd.SelectedPath, result == DialogResult.Yes);
            LoadPattern(Patterns[Index]);
            RepopulateList(Index);
            return;
        }

        var original = Patterns[Index];
        var name = original.DesignName;
        using var ofd = new OpenFileDialog();
        ofd.Filter = "New Horizons Design Pattern (*.nhd)|*.nhd|All files (*.*)|*.*";
        ofd.FileName = $"{name}.nhd";
        if (ofd.ShowDialog() != DialogResult.OK)
            return;

        var file = ofd.FileName;
        var expectLength = original.Data.Length;
        var fi = new FileInfo(file);
        if (fi.Length != expectLength)
        {
            var msg = string.Format(MessageStrings.MsgDataSizeMismatchImport, fi.Length, expectLength);
            WinFormsUtil.Error(MessageStrings.MsgCanceling, msg);
            return;
        }

        var data = File.ReadAllBytes(ofd.FileName);
        var d = new DesignPattern(data);
        var player0 = original;
        if (!d.IsOriginatedFrom(player0))
        {
            var notHost = string.Format(MessageStrings.MsgDataDidNotOriginateFromHost_0, player0.PlayerName);
            var result = WinFormsUtil.Prompt(MessageBoxButtons.YesNoCancel, notHost, MessageStrings.MsgAskUpdateValues);
            if (result == DialogResult.Cancel)
                return;
            if (result == DialogResult.Yes)
                d.ChangeOrigins(player0, d.Data);
        }

        Patterns[Index] = d;
        LoadPattern(d);
        RepopulateList(Index);
    }

    private void RepopulateList(int index)
    {
        LB_Items.Items.Clear();
        foreach (var p in Patterns)
            LB_Items.Items.Add(GetPatternSummary(p));
        LB_Items.SelectedIndex = index;
    }

    public bool ContainsTransparentPixels(Bitmap image)
    {
        for (int y = 0; y < image.Height; ++y)
        {
            for (int x = 0; x < image.Width; ++x)
            {
                if (image.GetPixel(x, y).A < 255) // check alpha
                {
                    return true;
                }
            }
        }
        return false;
    }

    private void LoadPattern(DesignPattern designPattern)
    {
        if (designPattern.UsageCompatibility is not (0xEE00 or 0xEE02)) // known valid values (00=non-transparent, 02=transparent)
        {
            using var image = designPattern.GetImage();
            if (ContainsTransparentPixels(image))
            {
                designPattern.UsageCompatibility = 0xEE02; // set to transparent
            }
            else
            {
                designPattern.UsageCompatibility = 0xEE00; // reset to default value (non-transparent)
            }
        }

        if (CB_Pattern_OverwriteDesigner.Checked)
        {
            designPattern.PlayerID = Player.Personal.PlayerID;
            designPattern.PlayerName = Player.Personal.PlayerName;
            designPattern.TownID = Player.Personal.TownID;
            designPattern.TownName = Player.Personal.TownName;
        }

        PB_Pattern.Image = ImageUtil.ResizeImage(designPattern.GetImage(), DesignPattern.Width * scale, DesignPattern.Height * scale);
        PB_Palette.Image = ImageUtil.ResizeImage(designPattern.GetPalette(), 150, 10);
        L_PatternName.Text = designPattern.DesignName + Environment.NewLine +
                             designPattern.TownName + Environment.NewLine +
                             designPattern.PlayerName;
    }

    private void PB_Pattern_MouseEnter(object sender, EventArgs e) => PB_Pattern.BackColor = Color.GreenYellow;
    private void PB_Pattern_MouseLeave(object sender, EventArgs e) => PB_Pattern.BackColor = Color.Transparent;

    private void Menu_SavePNG_Click(object sender, EventArgs e)
    {
        if (!WinFormsUtil.TryGetUnderlying<PictureBox>(sender, out var pb) || pb.Image is null)
        {
            WinFormsUtil.Alert(MessageStrings.MsgNoPictureLoaded);
            return;
        }

        var name = Patterns[Index].DesignName;
        using var sfd = new SaveFileDialog();
        sfd.Filter = "png file (*.png)|*.png|All files (*.*)|*.*";
        sfd.FileName = $"{name}.png";
        if (sfd.ShowDialog() != DialogResult.OK)
            return;

        var bmp = pb.Image;
        bmp.Save(sfd.FileName, ImageFormat.Png);
    }
}