using System; using System.Collections.Generic; using System.IO; using Toolbox.Library; using Toolbox.Library.IO; using Toolbox.Library.Forms; using System.Windows.Forms; using aampv1 = AampV1Library; using aampv2 = AampV2Library; using FirstPlugin.Forms; namespace FirstPlugin { public class AAMP : IEditor, IFileFormat, IConvertableTextFormat { public FileType FileType { get; set; } = FileType.Parameter; public bool CanSave { get; set; } public string[] Description { get; set; } = new string[] { "AAMP" }; public string[] Extension { get; set; } = new string[] { "*.aamp", "*.bglpbd" }; 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 types = new List(); return types.ToArray(); } } 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(); } } #region Text Converter Interface public TextFileType TextFileType => TextFileType.Yaml; public bool CanConvertBack => false; public string ConvertToString() { if (aampFileV1 != null) return AampYamlConverter.ToYaml(aampFileV1); else return AampYamlConverter.ToYaml(aampFileV2); } public void ConvertFromString(string text) { if (aampFileV1 != null) AampYamlConverter.ToAamp(aampFileV1); } #endregion bool IsSaveDialog = false; public static void GenerateProbeBoundings() { OpenFileDialog ofd1 = new OpenFileDialog(); ofd1.Filter = Utils.GetAllFilters(typeof(AAMP)); if (ofd1.ShowDialog() != DialogResult.OK) return; //Load the AAMP var File = STFileLoader.OpenFileFormat(ofd1.FileName); if (File == null || !(File is AAMP)) throw new Exception("File not a valid AAMP file!"); //Load bfres for generating the bounds OpenFileDialog ofd = new OpenFileDialog(); ofd.Filter = Utils.GetAllFilters(typeof(BFRES)); if (ofd.ShowDialog() == DialogResult.OK) { //Open and check bfres var Bfres = STFileLoader.OpenFileFormat(ofd.FileName); if (Bfres == null || !(Bfres is BFRES)) throw new Exception("File not a valid BFRES file!"); //Check version instance if (((AAMP)File).aampFileV1 != null) { foreach (var val in ((AAMP)File).aampFileV1.RootNode.childParams) { foreach (var param in val.paramObjects) { switch (param.HashString) { case "grid": GenerateGridData((BFRES)Bfres, param.paramEntries); break; } } } foreach (var param in ((AAMP)File).aampFileV1.RootNode.paramObjects) { switch (param.HashString) { case "root_grid": GenerateGridData((BFRES)Bfres, param.paramEntries); break; } } } else { foreach (var val in ((AAMP)File).aampFileV2.RootNode.childParams) { foreach (var param in val.paramObjects) { switch (param.HashString) { case "grid": GenerateGridData((BFRES)Bfres, param.paramEntries); break; } } } foreach (var param in ((AAMP)File).aampFileV2.RootNode.paramObjects) { switch (param.HashString) { case "root_grid": GenerateGridData((BFRES)Bfres, param.paramEntries); break; } } } SaveFileDialog sfd = new SaveFileDialog(); sfd.Filter = Utils.GetAllFilters(File); if (sfd.ShowDialog() == DialogResult.OK) { //Save the aamp back STFileSaver.SaveFileFormat(File, sfd.FileName); } File.Unload(); Bfres.Unload(); } } private static void GenerateGridData(BFRES bfres, aampv2.ParamEntry[] paramEntries) { //Load the grid min nad max and set them var boundings = bfres.GetBoundingBox(); foreach (var entry in paramEntries) { if (entry.HashString == "aabb_min_pos") entry.Value = new Syroot.Maths.Vector3F(boundings.Min.X, boundings.Min.Y, boundings.Min.Z); if (entry.HashString == "aabb_max_pos") entry.Value = new Syroot.Maths.Vector3F(boundings.Max.X, boundings.Max.Y, boundings.Max.Z); } } private static void GenerateGridData(BFRES bfres, aampv1.ParamEntry[] paramEntries) { var boundings = bfres.GetBoundingBox(); foreach (var entry in paramEntries) { if (entry.HashString == "aabb_min_pos") entry.Value = new Syroot.Maths.Vector3F(boundings.Min.X, boundings.Min.Y, boundings.Min.Z); if (entry.HashString == "aabb_max_pos") entry.Value = new Syroot.Maths.Vector3F(boundings.Max.X, boundings.Max.Y, boundings.Max.Z); } } public AampEditorBase OpenForm() { if (aampFileV1 != null) { AampV1Editor editor = new AampV1Editor(this, IsSaveDialog); editor.Text = FileName; editor.Dock = DockStyle.Fill; return editor; } else { AampV2Editor editor = new AampV2Editor(this, IsSaveDialog); editor.Text = FileName; editor.Dock = DockStyle.Fill; return editor; } } public void FillEditor(UserControl control) { } public aampv1.AampFile aampFileV1; public aampv2.AampFile aampFileV2; public void Load(Stream stream) { CanSave = true; IsSaveDialog = IFileInfo != null && IFileInfo.InArchive; uint Version = CheckVersion(stream); if (Version == 1) { aampFileV1 = new aampv1.AampFile(stream); } else if (Version == 2) { aampFileV2 = new aampv2.AampFile(stream); } else { throw new Exception($"Unsupported AAMP version! {Version}"); } } public void Unload() { } public byte[] Save() { var mem = new MemoryStream(); if (aampFileV1 != null) aampFileV1.Save(mem); else aampFileV2.Save(mem); return mem.ToArray(); } } }