Add sonic all stars transformed archive as it's basically an older version of SP2

This commit is contained in:
KillzXGaming 2019-07-20 20:56:22 -04:00
parent ece27fd3f8
commit ee5f01e44d
8 changed files with 581 additions and 5 deletions

Binary file not shown.

View file

@ -23,9 +23,7 @@ namespace FirstPlugin
{
using (var reader = new Toolbox.Library.IO.FileReader(stream, true))
{
bool IsMatch = reader.ReadUInt32() == 0xD6D1820C;
reader.Position = 0;
return IsMatch || Utils.HasExtension(FileName, ".sp2");
return Utils.HasExtension(FileName, ".sp2");
}
}
@ -91,7 +89,7 @@ namespace FirstPlugin
break;
case ChunkMetaInfo:
break;
case ChunkAnimInfo:
/* case ChunkAnimInfo:
if (chunk.ChunkSize > 0xB0)
{
// reader.Seek(chunk.Position + 0xB0, System.IO.SeekOrigin.Begin);
@ -117,7 +115,7 @@ namespace FirstPlugin
MaterialFile matFile = new MaterialFile();
matFile.Read(reader);
chunk.ChunkData = matFile;
break;
break;*/
}
reader.Seek(chunk.Position + chunk.ChunkSize, System.IO.SeekOrigin.Begin);

View file

@ -0,0 +1,512 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Toolbox.Library;
using Toolbox.Library.IO;
using Toolbox.Library.Forms;
using System.Drawing;
using System.Windows.Forms;
namespace FirstPlugin
{
public class SWU : TreeNodeFile, IArchiveFile, IFileFormat
{
public FileType FileType { get; set; } = FileType.Archive;
public bool CanSave { get; set; }
public string[] Description { get; set; } = new string[] { "Sonic and All Stars Racing Transformed Archive" };
public string[] Extension { get; set; } = new string[] { "*.swu" };
public string FileName { get; set; }
public string FilePath { get; set; }
public IFileInfo IFileInfo { get; set; }
public bool Identify(System.IO.Stream stream)
{
using (var reader = new Toolbox.Library.IO.FileReader(stream, true))
{
return Utils.HasExtension(FileName, ".swu");
}
}
public Type[] Types
{
get
{
List<Type> types = new List<Type>();
return types.ToArray();
}
}
public List<FileInfo> files = new List<FileInfo>();
public IEnumerable<ArchiveFileInfo> Files => files;
public void ClearFiles() { files.Clear(); }
public bool CanAddFiles { get; set; }
public bool CanRenameFiles { get; set; }
public bool CanReplaceFiles { get; set; }
public bool CanDeleteFiles { get; set; }
private const uint ChunkTextureFile = 0xD6D1820C;
private const uint ChunkMetaInfo = 0xB111B40E;
private const uint ChunkAnimInfo = 0x22008309;
private const uint ChunkAnimData = 0x29318F0A;
private const uint ChunkSkeletonData = 0x115AB800;
private const uint ChunkModelData = 0xCA121903;
private const uint ChunkShaderData = 0x777A9B0E;
private const uint ChunkMaterialData = 0x79C90901;
private List<ChunkHeader> Chunks = new List<ChunkHeader>();
public void Load(System.IO.Stream stream)
{
CanSave = false;
using (var reader = new FileReader(stream))
{
reader.SetByteOrder(true);
bool IsWiiU = true;
Text = FileName;
while (!reader.EndOfStream)
{
ChunkHeader chunk = new ChunkHeader();
chunk.Position = reader.Position;
chunk.Identifier = reader.ReadUInt32();
uint unk = reader.ReadUInt32();
chunk.ChunkSize = reader.ReadUInt32();
chunk.ChunkId = reader.ReadUInt32();
chunk.NextFilePtr = reader.ReadUInt32();
chunk.FileSize = reader.ReadUInt32();
uint unk2 = reader.ReadUInt32();
uint unk3 = reader.ReadUInt32();
Chunks.Add(chunk);
var Identifer = chunk.Identifier.Reverse();
switch (Identifer)
{
case ChunkTextureFile:
SWUTexture texture = new SWUTexture();
reader.SeekBegin(chunk.Position + 72);
texture.ImageKey = "texture";
texture.SelectedImageKey = "texture";
texture.ReadChunk(reader, IsWiiU);
chunk.ChunkData = texture;
if (chunk.ChunkSize > 244)
{
reader.Seek(chunk.Position + 244, System.IO.SeekOrigin.Begin);
chunk.FileName = reader.ReadString(Syroot.BinaryData.BinaryStringFormat.ZeroTerminated);
texture.Text = chunk.FileName;
}
Nodes.Add(texture);
break;
case ChunkMetaInfo:
break;
case ChunkAnimInfo:
if (chunk.ChunkSize > 0xB0)
{
// reader.Seek(chunk.Position + 0xB0, System.IO.SeekOrigin.Begin);
// chunk.FileName = reader.ReadString(Syroot.BinaryData.BinaryStringFormat.ZeroTerminated);
}
break;
/* case ChunkAnimData:
AnimationFile animFile = new AnimationFile();
animFile.Read(reader);
chunk.ChunkData = animFile;
break;
case ChunkSkeletonData:
SkeletonFile skelFile = new SkeletonFile();
skelFile.Read(reader);
chunk.ChunkData = skelFile;
break;
case ChunkModelData:
ModelFile modelFile = new ModelFile();
modelFile.Read(reader);
chunk.ChunkData = modelFile;
break;
case ChunkMaterialData:
MaterialFile matFile = new MaterialFile();
matFile.Read(reader);
chunk.ChunkData = matFile;
break;*/
}
reader.Seek(chunk.Position + chunk.ChunkSize, System.IO.SeekOrigin.Begin);
}
ReadGPUFile(FilePath);
}
TreeHelper.CreateFileDirectory(this);
}
public class SWUTexture : STGenericTexture, IChunkData
{
public byte[] ImageData;
public FileType FileType { get; set; } = FileType.Image;
public override bool CanEdit { get; set; } = false;
public override TEX_FORMAT[] SupportedFormats
{
get
{
return new TEX_FORMAT[] {
TEX_FORMAT.R8G8B8A8_UNORM,
};
}
}
public override void OnClick(TreeView treeview)
{
ImageEditorBase editor = (ImageEditorBase)LibraryGUI.GetActiveContent(typeof(ImageEditorBase));
if (editor == null)
{
editor = new ImageEditorBase();
editor.Dock = DockStyle.Fill;
LibraryGUI.LoadEditor(editor);
}
if (GX2Surface != null)
{
var tex = Bfres.Structs.FTEX.FromGx2Surface(GX2Surface, Text);
editor.LoadProperties(tex);
}
editor.Text = Text;
editor.LoadImage(this);
}
GX2.GX2Surface GX2Surface;
public void ReadChunk(FileReader reader, bool IsWiiU)
{
if (IsWiiU)
{
reader.SetByteOrder(true);
Console.WriteLine("TEX pos " + reader.Position);
GX2Surface = new GX2.GX2Surface();
GX2Surface.dim = reader.ReadUInt32();
GX2Surface.width = reader.ReadUInt32();
GX2Surface.height = reader.ReadUInt32();
GX2Surface.depth = reader.ReadUInt32();
GX2Surface.numMips = reader.ReadUInt32();
GX2Surface.format = reader.ReadUInt32();
GX2Surface.aa = reader.ReadUInt32();
GX2Surface.use = reader.ReadUInt32();
GX2Surface.imageSize = reader.ReadUInt32();
GX2Surface.imagePtr = reader.ReadUInt32();
GX2Surface.mipSize = reader.ReadUInt32();
GX2Surface.mipPtr = reader.ReadUInt32();
GX2Surface.tileMode = reader.ReadUInt32();
GX2Surface.swizzle = reader.ReadUInt32();
GX2Surface.alignment = reader.ReadUInt32();
GX2Surface.pitch = reader.ReadUInt32();
GX2Surface.mipOffset = reader.ReadUInt32s(13);
GX2Surface.firstMip = reader.ReadUInt32();
GX2Surface.imageCount = reader.ReadUInt32();
GX2Surface.firstSlice = reader.ReadUInt32();
GX2Surface.numSlices = reader.ReadUInt32();
GX2Surface.compSel = reader.ReadBytes(4);
GX2Surface.texRegs = reader.ReadUInt32s(4);
RedChannel = GX2ChanneToGeneric((Syroot.NintenTools.Bfres.GX2.GX2CompSel)GX2Surface.compSel[0]);
GreenChannel = GX2ChanneToGeneric((Syroot.NintenTools.Bfres.GX2.GX2CompSel)GX2Surface.compSel[1]);
BlueChannel = GX2ChanneToGeneric((Syroot.NintenTools.Bfres.GX2.GX2CompSel)GX2Surface.compSel[2]);
AlphaChannel = GX2ChanneToGeneric((Syroot.NintenTools.Bfres.GX2.GX2CompSel)GX2Surface.compSel[3]);
if (GX2Surface.numMips > 13)
return;
Width = GX2Surface.width;
Height = GX2Surface.height;
MipCount = GX2Surface.numMips;
ArrayCount = GX2Surface.numArray;
Format = Bfres.Structs.FTEX.ConvertFromGx2Format((Syroot.NintenTools.Bfres.GX2.GX2SurfaceFormat)GX2Surface.format);
}
}
private STChannelType GX2ChanneToGeneric(Syroot.NintenTools.Bfres.GX2.GX2CompSel comp)
{
if (comp == Syroot.NintenTools.Bfres.GX2.GX2CompSel.ChannelR) return STChannelType.Red;
else if (comp == Syroot.NintenTools.Bfres.GX2. GX2CompSel.ChannelG) return STChannelType.Green;
else if (comp == Syroot.NintenTools.Bfres.GX2.GX2CompSel.ChannelB) return STChannelType.Blue;
else if (comp == Syroot.NintenTools.Bfres.GX2.GX2CompSel.ChannelA) return STChannelType.Alpha;
else if (comp == Syroot.NintenTools.Bfres.GX2.GX2CompSel.Always0) return STChannelType.Zero;
else return STChannelType.One;
}
public override void SetImageData(Bitmap bitmap, int ArrayLevel)
{
throw new NotImplementedException("Cannot set image data! Operation not implemented!");
}
public override byte[] GetImageData(int ArrayLevel = 0, int MipLevel = 0)
{
if (GX2Surface != null)
{
GX2Surface.data = ImageData;
GX2Surface.mipData = ImageData;
return GX2.Decode(GX2Surface, ArrayLevel, MipLevel);
}
else
{
return ImageData;
}
}
}
private void ReadGPUFile(string FileName)
{
string path = FileName.Replace("cpu", "gpu");
if (!System.IO.File.Exists(path))
return;
int offset = 0;
//Read the data based on CPU chunk info
using (var reader = new FileReader(path))
{
for (int i = 0; i < Chunks.Count; i++)
{
if (Chunks[i].FileSize != 0 || Chunks[i].FileName != string.Empty || Chunks[i].ChunkData != null)
{
long pos = reader.Position;
var identifer = Chunks[i].Identifier.Reverse();
var fileInfo = new FileInfo();
//Get CPU chunk data
if (Chunks[i].ChunkData != null)
{
if (Chunks[i].ChunkData is SWUTexture)
{
SWUTexture texFile = (SWUTexture)Chunks[i].ChunkData;
if (Chunks[i].FileSize != 0)
texFile.ImageData = reader.ReadBytes((int)Chunks[i].FileSize);
continue;
}
if ( Chunks[i].ChunkData is AnimationFile)
{
AnimationFile animFile = (AnimationFile)Chunks[i].ChunkData;
fileInfo.FileName = animFile.FileName;
fileInfo.FileData = animFile.Data;
}
if (Chunks[i].ChunkData is SkeletonFile)
{
SkeletonFile animFile = (SkeletonFile)Chunks[i].ChunkData;
fileInfo.FileName = animFile.FileName;
fileInfo.FileData = animFile.Data;
}
if (Chunks[i].ChunkData is MaterialFile)
{
MaterialFile animFile = (MaterialFile)Chunks[i].ChunkData;
fileInfo.FileName = animFile.FileName;
fileInfo.FileData = animFile.Data;
}
if (Chunks[i].ChunkData is MaterialFile)
{
MaterialFile animFile = (MaterialFile)Chunks[i].ChunkData;
fileInfo.FileName = animFile.FileName;
fileInfo.FileData = animFile.Data;
}
if (Chunks[i].ChunkData is ModelFile)
{
ModelFile modelFile = (ModelFile)Chunks[i].ChunkData;
fileInfo.FileName = modelFile.FileName;
byte[] BufferData = new byte[0];
if (Chunks[i].FileSize != 0)
BufferData = reader.ReadBytes((int)Chunks[i].FileSize);
fileInfo.FileData = Utils.CombineByteArray(modelFile.Data, modelFile.Data2, BufferData);
//Don't advance the stream unless the chunk has a pointer
if (Chunks[i].NextFilePtr != 0)
reader.Seek(pos + Chunks[i].NextFilePtr, System.IO.SeekOrigin.Begin);
}
}
else //Else get the data from GPU
{
if (Chunks[i].FileName != string.Empty)
fileInfo.FileName = $"{Chunks[i].FileName}";
else
fileInfo.FileName = $"{i} {Chunks[i].ChunkId} {identifer.ToString("X")}";
if (Chunks[i].FileSize != 0)
fileInfo.FileData = reader.ReadBytes((int)Chunks[i].FileSize);
else
fileInfo.FileData = new byte[0];
}
files.Add(fileInfo);
//Don't advance the stream unless the chunk has a pointer
if (Chunks[i].NextFilePtr != 0)
reader.Seek(pos + Chunks[i].NextFilePtr, System.IO.SeekOrigin.Begin);
}
}
}
}
public void Unload()
{
}
public interface IChunkData { }
public class ChunkHeader
{
public IChunkData ChunkData;
public uint Identifier;
public long Position;
public uint ChunkSize;
public uint ChunkId;
public uint NextFilePtr;
public uint FileSize;
public string FileName = "";
}
//Info in CPU file about the model
//Note the GPU file chunk linked from this contains the buffers
public class ModelFile : IChunkData
{
public string FileName = "";
public string FileName2 = ""; //Yeah there's another file for some reason
public byte[] Data;
public byte[] Data2;
public void Read(FileReader reader)
{
long pos = reader.Position;
uint unk3 = reader.ReadUInt32();
uint unk4 = reader.ReadUInt32(); //Set to 1
uint SectionSize = reader.ReadUInt32(); //At the end, the file name
uint Padding = reader.ReadUInt32();
uint NextSectionOffset = reader.ReadUInt32();
reader.Seek(pos, System.IO.SeekOrigin.Begin);
//Model FILE
Data = reader.ReadBytes((int)SectionSize);
FileName = reader.ReadString(Syroot.BinaryData.BinaryStringFormat.ZeroTerminated);
//Todo
Data2 = new byte[0];
// reader.Seek(NextSectionOffset, System.IO.SeekOrigin.Begin);
// Data2 = reader.ReadBytes();
}
}
public class MaterialFile : IChunkData
{
public string FileName = "";
public byte[] Data;
public void Read(FileReader reader)
{
long pos = reader.Position;
uint unk3 = reader.ReadUInt32();
uint unk4 = reader.ReadUInt32(); //Set to 1
uint SectionSize = reader.ReadUInt32(); //At the end, the file name
uint Padding = reader.ReadUInt32();
reader.Seek(pos, System.IO.SeekOrigin.Begin);
//Material FILE
Data = reader.ReadBytes((int)SectionSize);
FileName = reader.ReadString(Syroot.BinaryData.BinaryStringFormat.ZeroTerminated);
}
}
public class SkeletonFile : IChunkData
{
public string FileName = "";
public byte[] Data;
public void Read(FileReader reader)
{
long pos = reader.Position;
uint unk3 = reader.ReadUInt32();
uint unk4 = reader.ReadUInt32(); //Set to 1
uint SectionSize = reader.ReadUInt32(); //At the end, the file name
uint Padding = reader.ReadUInt32();
reader.Seek(pos, System.IO.SeekOrigin.Begin);
//SKEL FILE
Data = reader.ReadBytes((int)SectionSize);
FileName = reader.ReadString(Syroot.BinaryData.BinaryStringFormat.ZeroTerminated);
}
}
public class AnimationFile : IChunkData
{
public string FileName = "";
public byte[] Data;
public void Read(FileReader reader)
{
long pos = reader.Position;
uint Hash = reader.ReadUInt32(); //Maybe a hash? Idk
uint unk4 = reader.ReadUInt32(); //Set to 1
uint SectionSize = reader.ReadUInt32(); //At the end, the file name
uint Padding = reader.ReadUInt32();
reader.Seek(pos, System.IO.SeekOrigin.Begin);
//ANIM FILE
Data = reader.ReadBytes((int)SectionSize);
FileName = reader.ReadString(Syroot.BinaryData.BinaryStringFormat.ZeroTerminated);
}
}
public class TextureFile : IChunkData
{
}
public class TextureInfo : IChunkData
{
}
public byte[] Save()
{
var mem = new System.IO.MemoryStream();
return mem.ToArray();
}
public bool AddFile(ArchiveFileInfo archiveFileInfo)
{
return false;
}
public bool DeleteFile(ArchiveFileInfo archiveFileInfo)
{
return false;
}
public class FileInfo : ArchiveFileInfo
{
}
}
}

View file

@ -211,6 +211,7 @@
<Compile Include="FileFormats\Archives\NXARC.cs" />
<Compile Include="FileFormats\Archives\RARC.cs" />
<Compile Include="FileFormats\Archives\SARC_OLD.cs" />
<Compile Include="FileFormats\Archives\SWU.cs" />
<Compile Include="FileFormats\Archives\SP2.cs" />
<Compile Include="FileFormats\Archives\TMPK.cs" />
<Compile Include="FileFormats\Archives\U8.cs" />

View file

@ -330,6 +330,7 @@ namespace FirstPlugin
Formats.Add(typeof(TEX3DS));
Formats.Add(typeof(NXARC));
Formats.Add(typeof(SP2));
Formats.Add(typeof(SWU));
Formats.Add(typeof(NUSHDB));
Formats.Add(typeof(MKGPDX_PAC));
Formats.Add(typeof(LZARC));

View file

@ -31,6 +31,12 @@ namespace Toolbox.Library.Forms
var FileRoot = new ArchiveRootNodeWrapper(FileFormat.FileName, (IArchiveFile)FileFormat);
FileRoot.FillTreeNodes();
AddNode(FileRoot);
if (FileFormat is TreeNode) //It can still be both, so add all it's nodes
{
foreach (TreeNode n in ((TreeNode)FileFormat).Nodes)
FileRoot.Nodes.Add(n);
}
}
public void AddNodeCollection(TreeNodeCollection nodes, bool ClearNodes)

View file

@ -12,6 +12,64 @@ namespace Toolbox.Library
{
public class TreeHelper
{
public static void CreateFileDirectory(TreeNode root)
{
List<TreeNode> treeNodes = new List<TreeNode>();
foreach (TreeNode n in root.Nodes)
treeNodes.Add(n);
var rootText = root.Text;
var rootTextLength = rootText.Length;
root.Nodes.Clear();
int I = 0;
foreach (TreeNode node in treeNodes)
{
string nodeString = node.Text;
var roots = nodeString.Split(new char[] { '/' },
StringSplitOptions.RemoveEmptyEntries);
// The initial parent is the root node
var parentNode = root;
var sb = new System.Text.StringBuilder(rootText, nodeString.Length + rootTextLength);
for (int rootIndex = 0; rootIndex < roots.Length; rootIndex++)
{
// Build the node name
var parentName = roots[rootIndex];
sb.Append("/");
sb.Append(parentName);
var nodeName = sb.ToString();
// Search for the node
var index = parentNode.Nodes.IndexOfKey(nodeName);
if (index == -1)
{
// Node was not found, add it
var folder = new TreeNode(parentName);
if (rootIndex == roots.Length - 1)
{
node.Text = parentName;
parentNode.Nodes.Add(node);
parentNode = node;
}
else
{
folder.Name = nodeName;
parentNode.Nodes.Add(folder);
parentNode = folder;
}
}
else
{
// Node was found, set that as parent and continue
parentNode = parentNode.Nodes[index];
}
}
}
}
public static void ExtractAllFiles(string ParentPath, TreeNodeCollection Nodes)
{
FolderSelectDialog folderDialog = new FolderSelectDialog();