1
0
mirror of synced 2024-12-02 02:57:23 +01:00
Switch-Toolbox/Switch_FileFormatsMain/FileFormats/AAMP/AAMP.cs
2019-04-09 19:43:24 -04:00

119 lines
3.2 KiB
C#

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();
}
}
public aampv1.AampFile aampFileV1;
public 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();
}
}
}