f51dd17f94
Add in nutexb file format. While not finished, it can preview and export them. Batch exporting for nuteb (tools menu). Rework GTX code. This is WIP and not finished. Add XTX code. Unifnished atm. Add saving for wii u and include fmat exporting. Proper error handling for assimp and texture swizzling.
345 lines
13 KiB
C#
345 lines
13 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.IO;
|
|
using System.Threading.Tasks;
|
|
using Switch_Toolbox;
|
|
using System.Windows.Forms;
|
|
using Switch_Toolbox.Library;
|
|
using Switch_Toolbox.Library.Forms;
|
|
using Switch_Toolbox.Library.IO;
|
|
using Bfres.Structs;
|
|
|
|
namespace FirstPlugin
|
|
{
|
|
public class BFRES : IFileFormat
|
|
{
|
|
public bool CanSave { get; set; } = false;
|
|
public bool FileIsEdited { get; set; } = false;
|
|
public bool FileIsCompressed { get; set; } = false;
|
|
public string[] Description { get; set; } = new string[] { "BFRES", "*BFRES", "*BFRES" };
|
|
public string[] Extension { get; set; } = new string[] { "*.bfres", "*.sbfres" };
|
|
public string Magic { get; set; } = "FRES";
|
|
public CompressionType CompressionType { get; set; } = CompressionType.None;
|
|
public byte[] Data { get; set; }
|
|
public string FileName { get; set; }
|
|
public IFileInfo IFileInfo { get; set; }
|
|
|
|
public Type[] Types
|
|
{
|
|
get
|
|
{
|
|
List<Type> types = new List<Type>();
|
|
return types.ToArray();
|
|
}
|
|
}
|
|
|
|
private TreeNodeFile eitorRoot;
|
|
public TreeNodeFile EditorRoot
|
|
{
|
|
get
|
|
{
|
|
return eitorRoot;
|
|
}
|
|
set
|
|
{
|
|
this.eitorRoot = value;
|
|
}
|
|
}
|
|
public void SaveFile()
|
|
{
|
|
List<IFileFormat> formats = new List<IFileFormat>();
|
|
formats.Add(this);
|
|
|
|
SaveFileDialog sfd = new SaveFileDialog();
|
|
sfd.Filter = Utils.GetAllFilters(formats);
|
|
sfd.FileName = FileName;
|
|
|
|
if (sfd.ShowDialog() == DialogResult.OK)
|
|
{
|
|
Cursor.Current = Cursors.WaitCursor;
|
|
SaveCompressFile(Save(), sfd.FileName, Alignment);
|
|
}
|
|
}
|
|
private void SaveCompressFile(byte[] data, string FileName, int Alignment = 0, bool EnableDialog = true)
|
|
{
|
|
if (EnableDialog && CompressionType != CompressionType.None)
|
|
{
|
|
DialogResult save = MessageBox.Show($"Compress file as {CompressionType}?", "File Save", MessageBoxButtons.YesNo);
|
|
|
|
if (save == DialogResult.Yes)
|
|
{
|
|
switch (CompressionType)
|
|
{
|
|
case CompressionType.Yaz0:
|
|
data = EveryFileExplorer.YAZ0.Compress(data, Runtime.Yaz0CompressionLevel, (uint)Alignment);
|
|
break;
|
|
case CompressionType.Lz4f:
|
|
data = STLibraryCompression.Type_LZ4F.Compress(data);
|
|
break;
|
|
case CompressionType.Lz4:
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
File.WriteAllBytes(FileName, data);
|
|
MessageBox.Show($"File has been saved to {FileName}");
|
|
Cursor.Current = Cursors.Default;
|
|
}
|
|
|
|
public bool IsActive { get; set; } = false;
|
|
public bool UseEditMenu { get; set; } = false;
|
|
public int Alignment { get; set; } = 0;
|
|
public string FilePath { get; set; }
|
|
public static bool IsWiiU = false;
|
|
|
|
BFRESRender bfres;
|
|
public void Load()
|
|
{
|
|
IsActive = true;
|
|
CanSave = true;
|
|
|
|
using (FileReader reader = new FileReader(new MemoryStream(Data)))
|
|
{
|
|
reader.Seek(4);
|
|
if (reader.ReadInt32() != 0x20202020)
|
|
{
|
|
IsWiiU = true;
|
|
}
|
|
reader.Close();
|
|
}
|
|
|
|
bfres = new BFRESRender();
|
|
bfres.ResFileNode = new ResourceFile(this);
|
|
bfres.ResFileNode.BFRESRender = bfres;
|
|
|
|
EditorRoot = bfres.ResFileNode;
|
|
|
|
if (IsWiiU)
|
|
{
|
|
bfres.LoadFile(new Syroot.NintenTools.Bfres.ResFile(new System.IO.MemoryStream(Data)));
|
|
}
|
|
else
|
|
{
|
|
bfres.LoadFile(new Syroot.NintenTools.NSW.Bfres.ResFile(new System.IO.MemoryStream(Data)));
|
|
}
|
|
|
|
Runtime.abstractGlDrawables.Add(bfres);
|
|
}
|
|
public void Unload()
|
|
{
|
|
bfres.Destroy();
|
|
bfres.ResFileNode.Nodes.Clear();
|
|
}
|
|
|
|
public byte[] Save()
|
|
{
|
|
MemoryStream mem = new MemoryStream();
|
|
|
|
if (IsWiiU)
|
|
SaveWiiU(mem);
|
|
else
|
|
SaveSwitch(mem);
|
|
|
|
return mem.ToArray();
|
|
}
|
|
private void SaveSwitch(MemoryStream mem)
|
|
{
|
|
var resFile = bfres.ResFileNode.resFile;
|
|
|
|
resFile.Models.Clear();
|
|
resFile.SkeletalAnims.Clear();
|
|
resFile.MaterialAnims.Clear();
|
|
resFile.SceneAnims.Clear();
|
|
resFile.ShapeAnims.Clear();
|
|
resFile.BoneVisibilityAnims.Clear();
|
|
resFile.ModelDict.Clear();
|
|
resFile.SkeletalAnimDict.Clear();
|
|
resFile.MaterialAnimDict.Clear();
|
|
resFile.SceneAnimDict.Clear();
|
|
resFile.ShapeAnimDict.Clear();
|
|
resFile.BoneVisibilityAnimDict.Clear();
|
|
|
|
|
|
int CurMdl = 0;
|
|
if (EditorRoot.Nodes.ContainsKey("FMDL"))
|
|
{
|
|
foreach (FMDL model in EditorRoot.Nodes["FMDL"].Nodes)
|
|
resFile.Models.Add(BfresSwitch.SetModel(model));
|
|
}
|
|
if (EditorRoot.Nodes.ContainsKey("FSKA"))
|
|
{
|
|
foreach (BfresSkeletonAnim ska in EditorRoot.Nodes["FSKA"].Nodes)
|
|
resFile.SkeletalAnims.Add(ska.SkeletalAnim);
|
|
}
|
|
if (EditorRoot.Nodes.ContainsKey("FMAA"))
|
|
{
|
|
foreach (FMAA fmaa in EditorRoot.Nodes["FMAA"].Nodes)
|
|
resFile.MaterialAnims.Add(fmaa.MaterialAnim);
|
|
}
|
|
if (EditorRoot.Nodes.ContainsKey("FBNV"))
|
|
{
|
|
foreach (FBNV fbnv in EditorRoot.Nodes["FBNV"].Nodes)
|
|
resFile.BoneVisibilityAnims.Add(fbnv.VisibilityAnim);
|
|
}
|
|
if (EditorRoot.Nodes.ContainsKey("FSHPA"))
|
|
{
|
|
foreach (FSHA fsha in EditorRoot.Nodes["FSHPA"].Nodes)
|
|
resFile.ShapeAnims.Add(fsha.ShapeAnim);
|
|
}
|
|
if (EditorRoot.Nodes.ContainsKey("FSCN"))
|
|
{
|
|
foreach (FSCN fscn in EditorRoot.Nodes["FSCN"].Nodes)
|
|
resFile.SceneAnims.Add(fscn.SceneAnim);
|
|
}
|
|
|
|
ErrorCheck();
|
|
|
|
BfresSwitch.WriteExternalFiles(resFile, EditorRoot);
|
|
resFile.Save(mem);
|
|
}
|
|
private void SaveWiiU(MemoryStream mem)
|
|
{
|
|
var resFileU = bfres.ResFileNode.resFileU;
|
|
resFileU.Models.Clear();
|
|
// resFileU.SkeletalAnims.Clear();
|
|
// resFileU.SceneAnims.Clear();
|
|
// resFileU.ShapeAnims.Clear();
|
|
// resFileU.BoneVisibilityAnims.Clear();
|
|
// resFileU.Textures.Clear();
|
|
|
|
|
|
int CurMdl = 0;
|
|
if (EditorRoot.Nodes.ContainsKey("FMDL"))
|
|
{
|
|
foreach (FMDL model in EditorRoot.Nodes["FMDL"].Nodes)
|
|
resFileU.Models.Add(model.Text, BfresWiiU.SetModel(model));
|
|
}
|
|
ErrorCheck();
|
|
resFileU.Save(mem);
|
|
}
|
|
|
|
public static void SetShaderAssignAttributes(FMAT.ShaderAssign shd, FSHP shape)
|
|
{
|
|
foreach (var att in shape.vertexAttributes)
|
|
{
|
|
if (!shd.attributes.ContainsValue(att.Name) && !shd.attributes.ContainsKey(att.Name))
|
|
shd.attributes.Add(att.Name, att.Name);
|
|
}
|
|
foreach (var tex in shape.GetMaterial().textures)
|
|
{
|
|
if (!shd.samplers.ContainsValue(tex.SamplerName))
|
|
shd.samplers.Add(tex.SamplerName, tex.SamplerName);
|
|
}
|
|
}
|
|
|
|
|
|
private void SetDuplicateShapeName(FSHP shape)
|
|
{
|
|
DialogResult dialogResult = MessageBox.Show($"A shape {shape.Text} already exists with that name", "", MessageBoxButtons.OK);
|
|
|
|
if (dialogResult == DialogResult.OK)
|
|
{
|
|
RenameDialog renameDialog = new RenameDialog();
|
|
renameDialog.Text = "Rename Texture";
|
|
if (renameDialog.ShowDialog() == DialogResult.OK)
|
|
{
|
|
shape.Text = renameDialog.textBox1.Text;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static void CheckMissingTextures(FSHP shape)
|
|
{
|
|
bool ImportMissingTextures = false;
|
|
foreach (BinaryTextureContainer bntx in PluginRuntime.bntxContainers)
|
|
{
|
|
foreach (MatTexture tex in shape.GetMaterial().textures)
|
|
{
|
|
if (!bntx.Textures.ContainsKey(tex.Name))
|
|
{
|
|
if (!ImportMissingTextures)
|
|
{
|
|
DialogResult result = MessageBox.Show("Missing textures found! Would you like to use placeholders?", "", MessageBoxButtons.YesNo, MessageBoxIcon.Question);
|
|
|
|
if (result == DialogResult.Yes)
|
|
{
|
|
ImportMissingTextures = true;
|
|
}
|
|
else
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (ImportMissingTextures)
|
|
bntx.ImportPlaceholderTexture(tex.Name);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void ErrorCheck()
|
|
{
|
|
if (bfres != null)
|
|
{
|
|
List<Errors> Errors = new List<Errors>();
|
|
foreach (FMDL model in bfres.models)
|
|
{
|
|
foreach (FSHP shp in model.shapes)
|
|
{
|
|
if (!IsWiiU)
|
|
{
|
|
Syroot.NintenTools.NSW.Bfres.VertexBuffer vtx = shp.VertexBuffer;
|
|
Syroot.NintenTools.NSW.Bfres.Material mat = shp.GetMaterial().Material;
|
|
Syroot.NintenTools.NSW.Bfres.ShaderAssign shdr = mat.ShaderAssign;
|
|
|
|
for (int att = 0; att < vtx.Attributes.Count; att++)
|
|
{
|
|
if (!shdr.AttribAssigns.Contains(vtx.Attributes[att].Name))
|
|
MessageBox.Show($"Error! Attribute {vtx.Attributes[att].Name} is unlinked!");
|
|
}
|
|
for (int att = 0; att < mat.TextureRefs.Count; att++)
|
|
{
|
|
if (!shdr.SamplerAssigns.Contains(mat.SamplerDict.GetKey(att))) //mat.SamplerDict[att]
|
|
MessageBox.Show($"Error! Sampler {mat.SamplerDict.GetKey(att)} is unlinked!");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Syroot.NintenTools.Bfres.VertexBuffer vtx = shp.VertexBufferU;
|
|
Syroot.NintenTools.Bfres.Material mat = shp.GetMaterial().MaterialU;
|
|
Syroot.NintenTools.Bfres.ShaderAssign shdr = mat.ShaderAssign;
|
|
|
|
for (int att = 0; att < vtx.Attributes.Count; att++)
|
|
{
|
|
if (!shdr.AttribAssigns.ContainsKey(vtx.Attributes[att].Name))
|
|
MessageBox.Show($"Error! Attribute {vtx.Attributes[att].Name} is unlinked!");
|
|
}
|
|
for (int att = 0; att < mat.TextureRefs.Count; att++)
|
|
{
|
|
string samp = "";
|
|
mat.Samplers.TryGetKey(mat.Samplers[att], out samp);
|
|
if (!shdr.SamplerAssigns.ContainsKey(samp)) //mat.SamplerDict[att]
|
|
MessageBox.Show($"Error! Sampler {samp} is unlinked!");
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|
|
// ErrorList errorList = new ErrorList();
|
|
// errorList.LoadList(Errors);
|
|
// errorList.Show();
|
|
}
|
|
}
|
|
public class Errors
|
|
{
|
|
public string Section = "None";
|
|
public string Section2 = "None";
|
|
public string Message = "";
|
|
public string Type = "Unkown";
|
|
}
|
|
}
|
|
}
|