PKHeX/SAV/SAV_EventFlags.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using System.IO;

namespace PKHeX
{
    public partial class SAV_EventFlags : Form
    {
        public SAV_EventFlags(Form1 frm1)
        {
            InitializeComponent();
            m_parent = frm1;
            savshift = 0x7F000 * m_parent.savindex;

            Setup();

            nud.Text = "0"; // Prompts an update for flag 0.
        }
        Form1 m_parent;
        public int savshift;
        bool setup = true;
        public CheckBox[] chka;
        public bool[] flags = new bool[3072];
        private void B_Cancel_Click(object sender, EventArgs e)
        {
            Close();
        }
        private void B_Save_Click(object sender, EventArgs e)
        {
            // Gather Updated Flags
            for (int i = 0; i < chka.Length; i++)
                flags[getFlagNum(chka[i])] = chka[i].Checked;

            byte[] data = new byte[flags.Length / 8]; 
            
            for (int i = 0; i < flags.Length; i++)
            {
                if (flags[i])
                    data[i/8] |= (byte)(1 << i%8);
            }
            Array.Copy(data, 0, m_parent.savefile, 0x1A0FC + savshift, 0x180);
            this.Close();
        }
        private void Setup()
        {
            // Fill Bit arrays
            
            chka = new CheckBox[] {
                flag_0001,flag_0002,flag_0003,flag_0004,flag_0005,
                flag_2237,flag_2238,flag_2239,
                flag_0115,flag_0963, // Mewtwo
                flag_0114,flag_0790, // Zygarde
                flag_0285,flag_0286,flag_0287,flag_0288,flag_0289, // Statuettes
                flag_0290,flag_0291,flag_0292,flag_0293,flag_0294, // Super Unlocks
                flag_0675, // Chatelaine 50
                flag_2546, // Pokedex
            };
            int offset = 0x1A0FC + savshift;
            byte[] data = new Byte[0x180];
            Array.Copy(m_parent.savefile, offset, data, 0, 0x180);
            BitArray BitRegion = new BitArray(data);
            BitRegion.CopyTo(flags, 0);

            // Populate Flags
            setup = true;
            popFlags();
        }
        private void popFlags()
        {
            if (setup)
            {
                for (int i = 0; i < chka.Length; i++)
                    chka[i].Checked = flags[getFlagNum(chka[i])];

                changeCustomFlag(null, null);
            }
        }

        private int getFlagNum(CheckBox chk)
        {
            try
            {
                string source = chk.Name;
                return Convert.ToInt32(source.Substring(Math.Max(0, source.Length - 4)));
            }
            catch { return 0; }
        }
        private void changeCustomBool(object sender, EventArgs e)
        {
            flags[(int)nud.Value] = CHK_CustomFlag.Checked;
            popFlags();
        }
        private void changeCustomFlag(object sender, EventArgs e)
        {
            int flag = (int)nud.Value;
            if (flag >= 3072)
            {
                CHK_CustomFlag.Checked = false;
                CHK_CustomFlag.Enabled = false;
                nud.BackColor = Color.Red;
            }
            else
            {
                CHK_CustomFlag.Enabled = true;
                nud.BackColor = Color.White;
                CHK_CustomFlag.Checked = flags[flag];
            }
        }
        private void changeCustomFlag(object sender, KeyEventArgs e)
        {
            changeCustomFlag(null, (EventArgs)e);
        }

        private void toggleFlag(object sender, EventArgs e)
        {
            flags[getFlagNum((CheckBox)(sender))] = ((CheckBox)(sender)).Checked;
            changeCustomFlag(sender, e);
        }

        private void changeSAV(object sender, EventArgs e)
        {
            if (TB_NewSAV.Text.Length > 0 && TB_OldSAV.Text.Length > 0)
            {
                diffSaves();
            }
        }
        private void diffSaves()
        {
            BitArray oldBits = new BitArray(olddata);
            BitArray newBits = new BitArray(newdata);

            for (int i = 0; i < oldBits.Length; i++)
            {
                if (oldBits[i] != newBits[i])
                {
                    if (newBits[i])
                        TB_IsSet.AppendText(i.ToString("0000") + ",");
                    else
                        TB_UnSet.AppendText(i.ToString("0000") + ",");
                }
            }
        }
        private byte[] olddata = new Byte[0x180];
        private byte[] newdata = new Byte[0x180];
        private void openSAV(object sender, EventArgs e)
        {
            byte[] eventflags = new byte[0x180];
            string path = "";
            OpenFileDialog ofd = new OpenFileDialog();
            if (ofd.ShowDialog() == DialogResult.OK)
            {
                path = ofd.FileName;
                FileInfo fi = new FileInfo(path);
                if (fi.Length == 0x100000)
                {
                    byte[] data = File.ReadAllBytes(path);
                    Array.Copy(data, 0x1A0FC, eventflags, 0, 0x180);
                }
                else if (fi.Length == 0x65600)
                {
                    byte[] data = File.ReadAllBytes(path);
                    Array.Copy(data, 0x1A0FC - 0x5400, eventflags, 0, 0x180);
                }
                else
                {
                    MessageBox.Show("Invalid SAV Size", "Error"); return;
                }
            }
            else return;
            
            Button bs = (Button)sender;
            if (bs.Name == "B_LoadOld")
            {
                Array.Copy(eventflags, olddata, 0x180);
                TB_OldSAV.Text = path;
            }
            else 
            {
                Array.Copy(eventflags, newdata, 0x180);
                TB_NewSAV.Text = path;
            }
        }
    }
}