1
0
mirror of synced 2024-12-13 08:11:07 +01:00
Switch-Toolbox/Switch_FileFormatsMain/FileFormats/AAMP/AAMP.cs

257 lines
8.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 : IEditor<AampEditorBase>, 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<Type> types = new List<Type>();
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();
}
}
}