Switch-Toolbox/Switch_FileFormatsMain/FileFormats/AAMP/AAMP.cs

using System;
using System.Collections.Generic;
using System.IO;
using Switch_Toolbox.Library;
using Switch_Toolbox.Library.IO;
using Switch_Toolbox.Library.Forms;
using System.Windows.Forms;
using aampv1 = AampV1Library;
using aampv2 = AampV2Library;
using FirstPlugin.Forms;

namespace FirstPlugin
{
    public class AAMP : TreeNodeFile, IFileFormat
    {
        public bool CanSave { get; set; }
        public string[] Description { get; set; } = new string[] { "AAMP" };
        public string[] Extension { get; set; } = new string[] { "*.aamp" };
        public string FileName { get; set; }
        public string FilePath { get; set; }
        public IFileInfo IFileInfo { get; set; }

        public bool Identify(Stream stream)
        {
            using (var reader = new FileReader(stream, true))
            {
                return reader.CheckSignature(4, "AAMP");
            }
        }

        public Type[] Types
        {
            get
            {
                List<Type> types = new List<Type>();
                return types.ToArray();
            }
        }

        public override void OnClick(TreeView treeview)
        {
      
        }

        private uint CheckVersion(Stream stream)
        {
            using (FileReader reader = new FileReader(stream, true))
            {
                reader.ByteOrder = Syroot.BinaryData.ByteOrder.LittleEndian;
                reader.ReadSignature(4, "AAMP");
                reader.Position = 4;

                return reader.ReadUInt32();
            }
        }

        aampv1.AampFile aampFileV1;
        aampv2.AampFile aampFileV2;

        AampV1Editor aampEditorV1;
        AampV2Editor aampEditorV2;

        public void Load(Stream stream)
        {
            CanSave = true;
            Text = FileName;

            uint Version = CheckVersion(stream);

            if (Version == 1)
            {
                aampFileV1 = new aampv1.AampFile(stream);
                Text = $"{FileName} Type [{aampFileV1.EffectType}]";
                aampEditorV1 = new AampV1Editor();
                aampEditorV1.LoadFile(aampFileV1, this);
            }
            else if (Version == 2)
            {
                aampFileV2 = new aampv2.AampFile(stream);
                Text = $"{FileName} Type [{aampFileV2.EffectType}]";
                aampEditorV2 = new AampV2Editor();
                aampEditorV2.LoadFile(aampFileV2, this);
            }
            else
            {
                throw new Exception($"Unsupported AAMP version! {Version}");
            }
        }
            
        public override void OnAfterAdded()
        {
            if (aampEditorV1 != null)
            {
                aampEditorV1.LoadImages(TreeView, this);
            }
            if (aampEditorV2 != null)
            {
                aampEditorV2.LoadImages(TreeView, this);
            }
        }

        public void Unload()
        {

        }
        public byte[] Save()
        {
            var mem = new MemoryStream();

            if (aampFileV1 != null)
                aampFileV1.Save(mem);
            else
                aampFileV2.Save(mem);

            return mem.ToArray();
        }
    }
}