Mirrors/Switch-Toolbox
2
0
Fork 0
mirror of https://github.com/KillzXGaming/Switch-Toolbox synced 2024-11-25 05:50:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

Update to TRPAK Implementation (#546)

Browse source 
* Move TRPAK up the list of Fileformats
prevents false detections of some of the Archives as other File Formats

* add best guess for Paths
if all known paths are the same add that one path to all other Files

* Revert "Move TRPAK up the list of Fileformats"

This reverts commit a1c0c2d630.

* Move TRPAK up the list of Fileformats
this time dont reformat the whole Class

* Added Quick access
Added Quick access Folders in the same style as GFPAK for all Texture Files and all Model Files

* cleanup and set some ImageKeys

* Prefer BNTX Filenames that mathc the Hash
instead of relying on only getting the filenames from the file itself
This commit is contained in:
Teras 2022-11-30 00:32:27 +01:00 committed by GitHub
parent 6839d90a9b
commit 314a880213
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 219 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

224
File_Format_Library/FileFormats/Archives/TRPAK/TRPAK.cs
View file

@ -18,11 +18,29 @@ namespace FirstPlugin
public bool CanRenameFiles => false;
public bool CanReplaceFiles => false;
public bool CanSave { get; set; } = false;
public Dictionary<string, string> CategoryLookup
{
get
{
return new Dictionary<string, string>()
{
{ ".bntx", "Textures" },
{ ".trmbf", "Models" },
{ ".trmdl", "Models" },
{ ".trmsh", "Models" },
{ ".trmtr", "Models" },
{ ".trskl", "Models" },
};
}
}
public string[] Description { get; set; } = new string[] { "tr Package" };
public string[] Extension { get; set; } = new string[] { "*.trpak" };
public string FileName { get; set; }
public string FilePath { get; set; }
public IEnumerable<ArchiveFileInfo> Files { get; set; }
public List<ArchiveFileInfo> files { get; set; }
public IEnumerable<ArchiveFileInfo> Files => files;
public FileType FileType { get; set; } = FileType.Archive;
public IFileInfo IFileInfo { get; set; }
@ -58,12 +76,13 @@ namespace FirstPlugin
public void Load(Stream stream)
{
GFPAKHashCache.EnsureHashCache();
Files = new List<ArchiveFileInfo>();
files = new List<ArchiveFileInfo>();
FlatBuffers.TRPAK.TRPAK trpak = FlatBuffers.TRPAK.TRPAK.GetRootAsTRPAK(new FlatBuffers.ByteBuffer(stream.ToBytes()));
if (trpak.FilesLength != trpak.HashesLength)
{
throw new Exception("not the same amount of Hashes and File Entries in Trpak Container");
}
List<string> paths = new List<string>();
for (int i = 0; i < trpak.FilesLength; i++)
{
FlatBuffers.TRPAK.File? file = trpak.Files(i);
@ -84,10 +103,72 @@ namespace FirstPlugin
}
AFI.FileData = FileData;
AFI.Name = GetName(hash, FileData);
if (AFI.Name.Contains("/"))
{
string tmppath = System.IO.Path.GetDirectoryName(AFI.Name);
if (!paths.Contains(tmppath)) paths.Add(tmppath);
}
AFI.FileName = AFI.Name;
((List<ArchiveFileInfo>)Files).Add(AFI);
files.Add(AFI);
}
}
if (paths.Count == 1)
{
string path = paths[0].Replace("\\", "/");
foreach (var f in files)
{
if (!f.Name.Contains("/"))
{
f.Name = Path.Combine(path, f.Name).Replace("\\", "/");
f.FileName = f.Name;
}
}
}
TreeNode node = new QuickAccessFolder(this, "Quick access");
Nodes.Add(node);
Dictionary<string, TreeNode> folders = new Dictionary<string, TreeNode>();
foreach (var file in files)
{
string ext = Utils.GetExtension(file.FileName);
string folderName = "Other";
if (CategoryLookup.ContainsKey(ext))
folderName = CategoryLookup[ext];
if (!folders.ContainsKey(folderName))
{
TreeNode folder = new GFPAK.QuickAccessFileFolder(folderName);
if (folderName == "Textures")
folder = new TextureFolder(this, "Textures");
if (folderName == "Models")
folder = new GFPAK.QuickAccessFileFolder("Models");
node.Nodes.Add(folder);
folders.Add(folderName, folder);
}
string name = Path.GetFileName(file.FileName);
string imageKey = "fileBlank";
switch (ext)
{
case ".bntx": imageKey = "bntx"; break;
case ".trmdl": imageKey = "model"; break;
case ".trskl": imageKey = "bone"; break;
case ".trmbf": imageKey = "mesh"; break;
case ".trmsh": imageKey = "mesh"; break;
case ".trmtr": imageKey = "material"; break;
}
TreeNode fodlerNode = folders[folderName];
fodlerNode.Nodes.Add(new QuickAccessFile(name)
{
Tag = file,
ImageKey = imageKey,
SelectedImageKey = imageKey,
});
}
GFPAKHashCache.WriteCache();
}
@ -98,7 +179,7 @@ namespace FirstPlugin
public void Unload()
{
foreach (var file in Files)
foreach (var file in files)
{
if (file.FileFormat != null)
file.FileFormat.Unload();
@ -106,7 +187,7 @@ namespace FirstPlugin
file.FileData = null;
}
((List<ArchiveFileInfo>)Files).Clear();
files.Clear();
GC.SuppressFinalize(this);
}
@ -142,7 +223,7 @@ namespace FirstPlugin
{
string fileHashName = GFPAKHashCache.GetHashName(fileHash) ?? "";
string ext = FindMatch(Data, fileHashName);
if (ext == ".bntx" || ext == ".bfres" || ext == ".bnsh" || ext == ".bfsha")
if ((ext == ".bntx" && fileHashName == "") || ext == ".bfres" || ext == ".bnsh" || ext == ".bfsha")
{
string fileName = GetBinaryHeaderName(Data);
//Check for matches for shaders
@ -168,5 +249,136 @@ namespace FirstPlugin
return $"{fileHash.ToString("X")}{ext}";
}
}
public class TextureFolder : TreeNodeCustom, IContextMenuNode
{
private IArchiveFile ArchiveFile;
private bool HasExpanded = false;
private List<STGenericTexture> Textures = new List<STGenericTexture>();
public TextureFolder(IArchiveFile archive, string text)
{
ArchiveFile = archive;
Text = text;
}
public void AddTexture(string fileName)
{
BNTX bntx = BNTX.CreateBNTXFromTexture(fileName);
var mem = new MemoryStream();
bntx.Save(mem);
string filePath = fileName;
ArchiveFile.AddFile(new ArchiveFileInfo()
{
FileData = mem.ToArray(),
FileFormat = bntx,
FileName = filePath,
});
}
public virtual ToolStripItem[] GetContextMenuItems()
{
List<ToolStripItem> Items = new List<ToolStripItem>();
Items.Add(new ToolStripMenuItem("Export All", null, ExportAllAction, Keys.Control | Keys.E));
return Items.ToArray();
}
public void LoadTextures()
{
if (HasExpanded) return;
List<TreeNode> subNodes = new List<TreeNode>();
foreach (TreeNode node in Nodes)
{
var file = (ArchiveFileInfo)node.Tag;
if (file.FileFormat == null)
file.FileFormat = file.OpenFile();
BNTX bntx = file.FileFormat as BNTX;
foreach (var tex in bntx.Textures.Values)
{
tex.OnTextureDeleted += OnTextureDeleted;
//Set tree key for deletion
tex.Name = tex.Text;
tex.Tag = file;
var texNode = new TreeNode(tex.Text);
texNode.Tag = tex;
texNode.ImageKey = tex.ImageKey;
texNode.SelectedImageKey = tex.SelectedImageKey;
subNodes.Add(texNode);
Textures.Add(tex);
}
}
Nodes.Clear();
Nodes.AddRange(subNodes.ToArray());
HasExpanded = true;
}
public override void OnExpand()
{
LoadTextures();
}
private void ExportAllAction(object sender, EventArgs args)
{
LoadTextures();
List<string> Formats = new List<string>();
Formats.Add("Microsoft DDS (.dds)");
Formats.Add("Portable Graphics Network (.png)");
Formats.Add("Joint Photographic Experts Group (.jpg)");
Formats.Add("Bitmap Image (.bmp)");
Formats.Add("Tagged Image File Format (.tiff)");
FolderSelectDialog sfd = new FolderSelectDialog();
if (sfd.ShowDialog() == DialogResult.OK)
{
string folderPath = sfd.SelectedPath;
BatchFormatExport form = new BatchFormatExport(Formats);
if (form.ShowDialog() == DialogResult.OK)
{
foreach (STGenericTexture tex in Textures)
{
if (form.Index == 0)
tex.SaveDDS(folderPath + '\\' + tex.Text + ".dds");
else if (form.Index == 1)
tex.SaveBitMap(folderPath + '\\' + tex.Text + ".png");
else if (form.Index == 2)
tex.SaveBitMap(folderPath + '\\' + tex.Text + ".jpg");
else if (form.Index == 3)
tex.SaveBitMap(folderPath + '\\' + tex.Text + ".bmp");
else if (form.Index == 4)
tex.SaveBitMap(folderPath + '\\' + tex.Text + ".tiff");
}
}
}
}
private void OnTextureDeleted(object sender, EventArgs e)
{
var tex = (TextureData)sender;
foreach (var file in ArchiveFile.Files)
{
if (file.FileFormat != null && file.FileFormat is BNTX)
{
var bntx = (BNTX)file.FileFormat;
if (bntx.Textures.ContainsKey(tex.Text))
{
bntx.RemoveTexture(tex);
bntx.Unload();
ArchiveFile.DeleteFile(file);
Nodes.RemoveByKey(tex.Text);
}
}
}
}
}
}
}

2
File_Format_Library/Main.cs
View file

@ -350,6 +350,7 @@ namespace FirstPlugin
Formats.Add(typeof(BTI));
Formats.Add(typeof(TXE));
Formats.Add(typeof(SARC));
Formats.Add(typeof(TRPAK));
Formats.Add(typeof(BNTX));
Formats.Add(typeof(BEA));
Formats.Add(typeof(BYAML));
@ -460,7 +461,6 @@ namespace FirstPlugin
Formats.Add(typeof(MTXT));
Formats.Add(typeof(NKN));
Formats.Add(typeof(MetroidDreadLibrary.BSMAT));
Formats.Add(typeof(TRPAK));
//Formats.Add(typeof(XLINK_FILE));