Switch-Toolbox/Switch_FileFormatsMain/FileFormats/BFRES/BFRES.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.IO;
using System.Data;
using System.Threading;
using System.Windows.Forms;
using Switch_Toolbox.Library;
using Switch_Toolbox.Library.Forms;
using Switch_Toolbox.Library.IO;
using Bfres.Structs;
using ResU = Syroot.NintenTools.Bfres;
using Syroot.NintenTools.NSW.Bfres;
using Switch_Toolbox.Library.Animations;
using Switch_Toolbox.Library.NodeWrappers;
using GL_EditorFramework.Interfaces;
using FirstPlugin.Forms;
using FirstPlugin.NodeWrappers;

namespace FirstPlugin
{
    public class BFRES : BFRESWrapper, IFileFormat
    {
        public bool CanSave { get; set; }
        public string[] Description { get; set; } = new string[] { "BFRES" };
        public string[] Extension { get; set; } = new string[] {
            "*.bfres", "*.sbfres", "*.sbmapopen", "*.sbstftex", "*.sbitemico", "*.sbmaptex", "*.sbreviewtex" };
        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, "FRES");
            }
        }

        public Type[] Types
        {
            get
            {
                List<Type> types = new List<Type>();
                types.Add(typeof(MenuExt));
                return types.ToArray();
            }
        }

        private Thread Thread;

        public override string ExportFilter => Utils.GetAllFilters(new BFRES());

        class MenuExt : IFileMenuExtension
        {
            public STToolStripItem[] NewFileMenuExtensions => newFileExt;
            public STToolStripItem[] NewFromFileMenuExtensions => null;
            public STToolStripItem[] ToolsMenuExtensions => toolExt;
            public STToolStripItem[] TitleBarExtensions => null;
            public STToolStripItem[] CompressionMenuExtensions => null;
            public STToolStripItem[] ExperimentalMenuExtensions => null;

            STToolStripItem[] toolExt = new STToolStripItem[1];
            STToolStripItem[] newFileExt = new STToolStripItem[2];

            public MenuExt()
            {
                toolExt[0] = new STToolStripItem("Open Bfres Debugger");
                toolExt[0].Click += DebugInfo;

                newFileExt[0] = new STToolStripItem("BFRES (Switch)");
                newFileExt[0].Click += NewSwitchBfres;
                newFileExt[1] = new STToolStripItem("BFRES (Wii U)");
                newFileExt[1].Click += NewWiiUBfres;
            }
            private void NewWiiUBfres(object sender, EventArgs args)
            {
                BFRES bfres = new BFRES();
                bfres.IFileInfo = new IFileInfo();
                bfres.FileName = "Untitled.bfres";

                bfres.Load(new MemoryStream(BfresWiiU.CreateNewBFRES("Untitled.bfres")));

                ObjectEditor editor = new ObjectEditor();
                editor.Text = "Untitled-" + 0;
                editor.treeViewCustom1.Nodes.Add(bfres);
                LibraryGUI.Instance.CreateMdiWindow(editor);
            }
            private void NewSwitchBfres(object sender, EventArgs args)
            {
                BFRES bfres = new BFRES();
                bfres.IFileInfo = new IFileInfo();
                bfres.FileName = "Untitled.bfres";

                bfres.Load(new MemoryStream(CreateNewBFRESSwitch("Untitled.bfres")));

                ObjectEditor editor = new ObjectEditor();
                editor.Text = "Untitled-" + 0;
                editor.treeViewCustom1.Nodes.Add(bfres);
                LibraryGUI.Instance.CreateMdiWindow(editor);
            }
            private void DebugInfo(object sender, EventArgs args)
            {
                OpenFileDialog ofd = new OpenFileDialog();
                ofd.Filter = Utils.GetAllFilters(new BFRES());

                if (ofd.ShowDialog() != DialogResult.OK)
                    return;

                var debugInfo = new DebugInfoBox();
                debugInfo.Show();
                debugInfo.PrintDebugInfo(ofd.FileName);
            }
        }

        private static byte[] CreateNewBFRESSwitch(string Name)
        {
            MemoryStream mem = new MemoryStream();

            ResFile resFile = new ResFile();
            resFile.Name = Name;

            resFile.Save(mem);
            var data = mem.ToArray();

            mem.Close();
            mem.Dispose();
            return data;
        }



        public void OnPropertyChanged()
        {
            if (resFile != null)
            {
                Text = resFile.Name;
            }
            else
            {
                Text = resFileU.Name;
            }
            STPropertyGrid editor = (STPropertyGrid)LibraryGUI.Instance.GetActiveContent(typeof(STPropertyGrid));

            if (editor != null)
                editor.Refresh();
        }

        //Used for adding new skeleton drawables
        public void AddSkeletonDrawable(STSkeleton skeleton)
        {
            BfresEditor bfresEditor = (BfresEditor)LibraryGUI.Instance.GetActiveContent(typeof(BfresEditor));
            if (bfresEditor != null)
            {
                bfresEditor.AddDrawable(skeleton);
            }

            drawables.Add(skeleton);
        }

        public void RemoveSkeletonDrawable(STSkeleton skeleton)
        {
            BfresEditor bfresEditor = (BfresEditor)LibraryGUI.Instance.GetActiveContent(typeof(BfresEditor));
            if (bfresEditor != null)
            {
                bfresEditor.RemoveDrawable(skeleton);
            }

            drawables.Remove(skeleton);
        }

        List<AbstractGlDrawable> drawables = new List<AbstractGlDrawable>();
        public void LoadEditors(object SelectedSection)
        {
            BfresEditor bfresEditor = (BfresEditor)LibraryGUI.Instance.GetActiveContent(typeof(BfresEditor));

            bool HasModels = BFRESRender.models.Count > 0;

            if (bfresEditor == null)
            {
                bfresEditor = new BfresEditor(HasModels);
                bfresEditor.Dock = DockStyle.Fill;
                LibraryGUI.Instance.LoadEditor(bfresEditor);
            }

            var toolstrips = new List<ToolStripMenuItem>();
            var menu = new ToolStripMenuItem("Animation Loader", null, AnimLoader);

            toolstrips.Add(menu);

            bool IsLoaded = drawables.Count != 0;

            bfresEditor.IsLoaded = IsLoaded;

            if (drawables.Count <= 0)
            {
                //Add drawables
                drawables.Add(BFRESRender);
                foreach (var mdl in BFRESRender.models)
                    drawables.Add(mdl.Skeleton);
            }
            bfresEditor.LoadViewport(drawables, toolstrips);


            if (!IsLoaded)
            {
                bfresEditor.UpdateVertexData = BFRESRender.UpdateVertexData;
                bfresEditor.UpdateTextureMaps = BFRESRender.UpdateTextureMaps;
                bfresEditor.OnLoadedTab();
            }

            if (SelectedSection is BFRES)
            {
                STPropertyGrid editor = (STPropertyGrid)bfresEditor.GetActiveEditor(typeof(STPropertyGrid));
                if (editor == null)
                {
                    editor = new STPropertyGrid();
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.Text = Text;

                if (resFile != null)
                    editor.LoadProperty(resFile, OnPropertyChanged);
                else
                    editor.LoadProperty(resFileU, OnPropertyChanged);
            }
            else if (SelectedSection is BFRESGroupNode)
            {
                STPropertyGrid editor = (STPropertyGrid)bfresEditor.GetActiveEditor(typeof(STPropertyGrid));
                if (editor == null)
                {
                    editor = new STPropertyGrid();
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.Text = Text;
                editor.LoadProperty(null, null);
            }
            else if(SelectedSection is FSKL.fsklNode)
            {
                FSKLEditor editor = (FSKLEditor)bfresEditor.GetActiveEditor(typeof(FSKLEditor));
                if (editor == null)
                {
                    editor = new FSKLEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadSkeleton(((FSKL.fsklNode)SelectedSection).fskl);
            }
            else if (SelectedSection is BfresBone)
            {
                BfresBoneEditor editor = (BfresBoneEditor)bfresEditor.GetActiveEditor(typeof(BfresBoneEditor));
                if (editor == null)
                {
                    editor = new BfresBoneEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadBone((BfresBone)SelectedSection);
            }
            else if (SelectedSection is FSHP)
            {
                BfresShapeEditor editor = (BfresShapeEditor)bfresEditor.GetActiveEditor(typeof(BfresShapeEditor));
                if (editor == null)
                {
                    editor = new BfresShapeEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadShape((FSHP)SelectedSection);
            }
            else if (SelectedSection is FMAT)
            {
                FMATEditor editor = (FMATEditor)bfresEditor.GetActiveEditor(typeof(FMATEditor));
                if (editor == null)
                {
                    editor = new FMATEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadMaterial((FMAT)SelectedSection);
            }
            else if (SelectedSection is FSKA.BoneAnimNode)
            {
                BoneAnimEditor editor = (BoneAnimEditor)bfresEditor.GetActiveEditor(typeof(BoneAnimEditor));
                if (editor == null)
                {
                    editor = new BoneAnimEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadBoneAnim((FSKA.BoneAnimNode)SelectedSection);
            }
            else if (SelectedSection is FSCN.BfresCameraAnim)
            {
                SceneAnimEditor editor = (SceneAnimEditor)bfresEditor.GetActiveEditor(typeof(SceneAnimEditor));
                if (editor == null)
                {
                    editor = new SceneAnimEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadCameraAnim((FSCN.BfresCameraAnim)SelectedSection);
            }
            else if (SelectedSection is FSCN.BfresLightAnim)
            {
                SceneAnimEditor editor = (SceneAnimEditor)bfresEditor.GetActiveEditor(typeof(SceneAnimEditor));
                if (editor == null)
                {
                    editor = new SceneAnimEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadLightAnim((FSCN.BfresLightAnim)SelectedSection);
            }
            else if (SelectedSection is FSCN.BfresFogAnim)
            {
                SceneAnimEditor editor = (SceneAnimEditor)bfresEditor.GetActiveEditor(typeof(SceneAnimEditor));
                if (editor == null)
                {
                    editor = new SceneAnimEditor();
                    editor.Text = Text;
                    editor.Dock = DockStyle.Fill;
                    bfresEditor.LoadEditor(editor);
                }
                editor.LoadFogAnim((FSCN.BfresFogAnim)SelectedSection);
            }
            else if (SelectedSection is FMDL)
                OpenSubFileEditor<FMDL>(SelectedSection, bfresEditor);
            else if (SelectedSection is FSKA)
                OpenSubFileEditor<FSKA>(SelectedSection, bfresEditor);
            else if (SelectedSection is FSHU)
                OpenSubFileEditor<FSHU>(SelectedSection, bfresEditor);
            else if (SelectedSection is FSCN)
                OpenSubFileEditor<FSCN>(SelectedSection, bfresEditor);
            else if (SelectedSection is FSHA)
                OpenSubFileEditor<FSHA>(SelectedSection, bfresEditor);
            else if (SelectedSection is FTXP)
                OpenSubFileEditor<FTXP>(SelectedSection, bfresEditor);
            else if (SelectedSection is FMAA)
                OpenSubFileEditor<FMAA>(SelectedSection, bfresEditor);
            else if (SelectedSection is FVIS)
                OpenSubFileEditor<FVIS>(SelectedSection, bfresEditor);


           /*   else if (SelectedSection is FMAA && ((FMAA)SelectedSection).AnimType == MaterialAnimation.AnimationType.TexturePattern)
           {
               BfresTexturePatternEditor editor = (BfresTexturePatternEditor)bfresEditor.GetActiveEditor(typeof(BfresTexturePatternEditor));
               if (editor == null)
               {
                   editor = new BfresTexturePatternEditor();
                   bfresEditor.LoadEditor(editor);
               }
               editor.Text = Text;
               editor.Dock = DockStyle.Fill;
               editor.LoadAnim((FMAA)SelectedSection);
           }*/
        }

        private SubFileEditor OpenSubFileEditor<T>(object node, BfresEditor bfresEditor) where T : STGenericWrapper
        {
            SubFileEditor editor = (SubFileEditor)bfresEditor.GetActiveEditor(typeof(SubFileEditor));
            if (editor == null)
            {
                editor = new SubFileEditor();
                editor.Dock = DockStyle.Fill;
                bfresEditor.LoadEditor(editor);
            }
            editor.LoadSubFile<T>(node);

            return editor;
        }

        public void AnimLoader(object sender, EventArgs args)
        {
            AnimationLoader loader = new AnimationLoader();
            loader.TopLevel = true;
            loader.Show();
        }

        public BFRESRender BFRESRender;
        public void Load(System.IO.Stream stream)
        {
            CanSave = true;

            ImageKey = "bfres";
            SelectedImageKey = "bfres";

            using (FileReader reader = new FileReader(stream, true))
            {
                reader.ByteOrder = Syroot.BinaryData.ByteOrder.BigEndian;
                reader.Position = 4;

                if (reader.ReadInt32() != 0x20202020)
                    IsWiiU = true;

                reader.Position = 0;
            }


            LoadMenus(IsWiiU);

       
            BFRESRender = new BFRESRender();
            BFRESRender.ModelTransform = MarioCostumeEditor.SetTransform(FileName);
            BFRESRender.ResFileNode = this;

            if (IsWiiU)
            {
                BFRESRender.LoadFile(new Syroot.NintenTools.Bfres.ResFile(stream));
            }
            else
            {
                BFRESRender.LoadFile(new Syroot.NintenTools.NSW.Bfres.ResFile(stream));
            }

        }
        public void Unload()
        {
            BFRESRender.Destroy();

            foreach (var node in TreeViewExtensions.Collect(BFRESRender.ResFileNode.Nodes))
            {
                if (node is BFRESGroupNode)
                {
                    if (((BFRESGroupNode)node).Type == BRESGroupType.Textures)
                    {
                        if (PluginRuntime.ftexContainers.Contains(((BFRESGroupNode)node)))
                            PluginRuntime.ftexContainers.Remove(((BFRESGroupNode)node));
                    }
                }

                if (node is BNTX)
                    ((BNTX)node).Unload();
            }
            Nodes.Clear();
        }

        public byte[] Save()
        {
            MemoryStream mem = new MemoryStream();

            if (IsWiiU)
                SaveWiiU(mem);
            else
                SaveSwitch(mem);

            return mem.ToArray();
        }

        public ResFile resFile = null;
        public ResU.ResFile resFileU = null;

        public override void OnClick(TreeView treeView)
        {
            LoadEditors(this);
        }

        public void Load(ResU.ResFile res)
        {
            resFileU = res;

            Text = resFileU.Name;

            var modelFolder = new BFRESGroupNode(BRESGroupType.Models);
            var texturesFolder = new BFRESGroupNode(BRESGroupType.Textures);
            var animFolder = new BFRESAnimFolder();
            var externalFilesFolder = new BFRESGroupNode(BRESGroupType.Embedded);

            Nodes.Add(modelFolder);
            Nodes.Add(texturesFolder);
            Nodes.Add(animFolder);
            Nodes.Add(externalFilesFolder);

            animFolder.LoadMenus(IsWiiU);
            PluginRuntime.ftexContainers.Add(texturesFolder);

            if (resFileU.Models.Count > 0)
            {
                foreach (ResU.Model mdl in resFileU.Models.Values)
                {
                    var fmdl = new FMDL();
                    BfresWiiU.ReadModel(fmdl, mdl);
                    modelFolder.AddNode(fmdl);
                }
            }
            if (resFileU.Textures.Count > 0)
            {
                foreach (ResU.Texture tex in resFileU.Textures.Values)
                    texturesFolder.AddNode(new FTEX(tex));
            }
            if (resFileU.SkeletalAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.SkeletalAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.SkeletalAnim anim in resFileU.SkeletalAnims.Values)
                    group.AddNode(new FSKA(anim));
            }
            if (resFileU.ShaderParamAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.ShaderParamAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.ShaderParamAnim anim in resFileU.ShaderParamAnims.Values)
                    group.AddNode(new FSHU(anim, MaterialAnimation.AnimationType.ShaderParam));
            }
            if (resFileU.ColorAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.ColorAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.ShaderParamAnim anim in resFileU.ColorAnims.Values)
                    group.AddNode(new FSHU(anim, MaterialAnimation.AnimationType.Color));
            }
            if (resFileU.TexSrtAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.TexSrtAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.ShaderParamAnim anim in resFileU.TexSrtAnims.Values)
                    group.AddNode(new FSHU(anim, MaterialAnimation.AnimationType.TexturePattern));
            }
            if (resFileU.TexPatternAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.TexPatAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.TexPatternAnim anim in resFileU.TexPatternAnims.Values)
                    group.AddNode(new FTXP(anim));
            }
            if (resFileU.ShapeAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.ShapeAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.ShapeAnim anim in resFileU.ShapeAnims.Values)
                    group.AddNode(new FSHA(anim));
            }
            if (resFileU.BoneVisibilityAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.BoneVisAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.VisibilityAnim anim in resFileU.BoneVisibilityAnims.Values)
                    group.AddNode(new FVIS(anim));
            }
            if (resFileU.MatVisibilityAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.MatVisAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.VisibilityAnim anim in resFileU.BoneVisibilityAnims.Values)
                    group.AddNode(new FVIS(anim));
            }
            if (resFileU.SceneAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.SceneAnim);
                animFolder.Nodes.Add(group);

                foreach (ResU.SceneAnim anim in resFileU.SceneAnims.Values)
                    group.AddNode(new FSCN(anim));
            }
            if (resFileU.ExternalFiles.Count > 0)
            {
                foreach (var anim in resFileU.ExternalFiles)
                {
                    externalFilesFolder.AddNode(new ExternalFileData(anim.Key, anim.Value.Data));
                }
            }
        }
        public void Load(ResFile res)
        {
            resFile = res;

            Text = resFile.Name;

            var modelFolder = new BFRESGroupNode(BRESGroupType.Models);
            var texturesFolder = new BNTX() { ImageKey = "folder", SelectedImageKey = "folder", Text = "Textures" };
            var animFolder = new BFRESAnimFolder();
            var externalFilesFolder = new BFRESGroupNode(BRESGroupType.Embedded);

            //Texture folder acts like a bntx for saving back
            //This will only save if the user adds textures to it or the file has a bntx already
            texturesFolder.IFileInfo = new IFileInfo();
            texturesFolder.FileName = "Textures";
            texturesFolder.Load(new MemoryStream(BNTX.CreateNewBNTX("Textures")));

            animFolder.LoadMenus(IsWiiU);

            Nodes.Add(modelFolder);
            Nodes.Add(texturesFolder);
            Nodes.Add(animFolder);
            Nodes.Add(externalFilesFolder);

            if (resFile.Models.Count > 0)
            {
                foreach (Model mdl in resFile.Models)
                {
                    var fmdl = new FMDL();
                    BfresSwitch.ReadModel(fmdl, mdl);
                    modelFolder.AddNode(fmdl);
                }
            }
            if (resFile.SkeletalAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.SkeletalAnim);
                animFolder.Nodes.Add(group);

                foreach (SkeletalAnim anim in resFile.SkeletalAnims)
                    group.AddNode(new FSKA(anim));
            }
            if (resFile.MaterialAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.ShaderParamAnim);
                var group2 = new BFRESGroupNode(BRESGroupType.TexSrtAnim);
                var group3 = new BFRESGroupNode(BRESGroupType.TexPatAnim);
                var group4 = new BFRESGroupNode(BRESGroupType.ColorAnim);
                var group5 = new BFRESGroupNode(BRESGroupType.MatVisAnim);
                var group6 = new BFRESGroupNode(BRESGroupType.MaterialAnim);

                bool HasShaderParamsAnim = false;
                bool HasTextureSrtAnim = false;
                bool HasTexturePatternAnim = false;
                bool HasColorAnim = false;
                bool HasMatVisAnim = false;
                bool HasMaterialAnim = false;

                foreach (MaterialAnim anim in resFile.MaterialAnims)
                {
                    if (FMAA.IsShaderParamAnimation(anim.Name))
                    {
                        group.AddNode(new FMAA(anim, MaterialAnimation.AnimationType.ShaderParam));
                        HasShaderParamsAnim = true;
                    }
                    else if (FMAA.IsSRTAnimation(anim.Name))
                    {
                        group2.AddNode(new FMAA(anim, MaterialAnimation.AnimationType.TextureSrt));
                        HasTextureSrtAnim = true;
                    }
                    else if(FMAA.IsTexturePattern(anim.Name))
                    {
                        group3.AddNode(new FMAA(anim, MaterialAnimation.AnimationType.TexturePattern));
                        HasTexturePatternAnim = true;
                    }
                    else if (FMAA.IsColorAnimation(anim.Name))
                    {
                        group4.AddNode(new FMAA(anim, MaterialAnimation.AnimationType.Color));
                        HasColorAnim = true;
                    }
                    else if (FMAA.IsVisibiltyAnimation(anim.Name))
                    {
                        group5.AddNode(new FMAA(anim, MaterialAnimation.AnimationType.Visibilty));
                        HasMatVisAnim = true;
                    }
                    else
                    {
                        group6.AddNode(new FMAA(anim, MaterialAnimation.AnimationType.ShaderParam));
                        HasMaterialAnim = true;
                    }
                }

                if (HasShaderParamsAnim)
                    animFolder.Nodes.Add(group);
                if (HasTextureSrtAnim)
                    animFolder.Nodes.Add(group2);
                if (HasTexturePatternAnim)
                    animFolder.Nodes.Add(group3);
                if (HasColorAnim)
                    animFolder.Nodes.Add(group4);
                if (HasMatVisAnim)
                    animFolder.Nodes.Add(group5);
                if (HasMaterialAnim)
                    animFolder.Nodes.Add(group6);
            }
            if (resFile.BoneVisibilityAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.BoneVisAnim);
                animFolder.Nodes.Add(group);

                foreach (VisibilityAnim anim in resFile.BoneVisibilityAnims)
                    group.AddNode(new FVIS(anim));
            }
            if (resFile.SceneAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.SceneAnim);
                animFolder.Nodes.Add(group);

                foreach (SceneAnim anim in resFile.SceneAnims)
                    group.AddNode(new FSCN(anim));
            }
            if (resFile.ShapeAnims.Count > 0)
            {
                var group = new BFRESGroupNode(BRESGroupType.ShapeAnim);
                animFolder.Nodes.Add(group);

                foreach (ShapeAnim anim in resFile.ShapeAnims)
                    group.AddNode(new FSHA(anim));
            }
            if (resFile.ExternalFiles.Count > 0)
            {
                int index = 0;

                //This also can be set to true if we don't want to rebuild the bntx for debug purposes
                bool IsTexturesReplaced = false;

                foreach (var anim in resFile.ExternalFiles)
                {
                    // group.AddNode(new ExternalFileData(Name, anim.Data) { FileFormat = file });

                    string Name = resFile.ExternalFileDict.GetKey(index++);

                    var file = STFileLoader.OpenFileFormat(Name, anim.Data, false, true);

                    //Only do once. There's usually one bntx embedded but incase there are multiple
                    if (file is BNTX && !IsTexturesReplaced)
                    {
                        ((TreeNode)file).Text = texturesFolder.Text;
                        ((TreeNode)file).ImageKey = texturesFolder.ImageKey;
                        ((TreeNode)file).SelectedImageKey = texturesFolder.SelectedImageKey;

                        //Remove temporary bntx file with file one
                        PluginRuntime.bntxContainers.Remove(texturesFolder);

                        ReplaceNode(texturesFolder.Parent, texturesFolder, (TreeNode)file);

                        IsTexturesReplaced = true;
                    }
                    else
                    {
                        externalFilesFolder.AddNode(new ExternalFileData(Name, anim.Data));
                    }
                }
            }
        }

        public static void ReplaceNode(TreeNode node, TreeNode replaceNode, TreeNode NewNode)
        {
            if (NewNode == null)
                return;

            int index = node.Nodes.IndexOf(replaceNode);
            node.Nodes.RemoveAt(index);
            node.Nodes.Insert(index, NewNode);
        }

        public override void Export(string FileName)
        {
            bool IsTex1 = FileName.Contains("Tex1");
            bool HasTextures = false;

            foreach (TreeNode group in Nodes)
            {
                if (group is BFRESGroupNode)
                {
                    if (((BFRESGroupNode)group).Type == BRESGroupType.Textures && group.Nodes.Count > 0)
                        HasTextures = true;
                }
            }
  

            if (IsTex1 && HasTextures)
            {
                STFileSaver.SaveFileFormat(this, FileName);

                byte[] Tex2 = GenerateTex2();

                SaveFileDialog sfd = new SaveFileDialog();
                sfd.FileName = FileName + FileName.Replace("Tex1", "Tex2");
                sfd.DefaultExt = ".sbfres";

                List<IFileFormat> formats = new List<IFileFormat>();
                formats.Add(this);
                sfd.Filter = Utils.GetAllFilters(formats);

                if (sfd.ShowDialog() == DialogResult.OK)
                    STFileSaver.SaveFileFormat(Tex2, true, 0, CompressionType.Yaz0, sfd.FileName);
            }
            else
                STFileSaver.SaveFileFormat(this, FileName);
        }

        private byte[] GenerateTex2()
        {
            var mem = new MemoryStream();

            var resFileU = BFRESRender.ResFileNode.resFileU;

            //Create a tex2 file
            ResU.ResFile resFileTex2 = new ResU.ResFile();
            resFileTex2.Alignment = resFileU.Alignment;
            resFileTex2.Name = resFileU.Name.Replace("Tex1", "Tex2");
            resFileTex2.VersionMajor = resFileU.VersionMajor;
            resFileTex2.VersionMajor2 = resFileU.VersionMajor2;
            resFileTex2.VersionMinor = resFileU.VersionMinor;
            resFileTex2.VersionMinor2 = resFileU.VersionMinor2;
            resFileTex2.Textures = resFileU.Textures;

            foreach (var group in Nodes)
            {
                if (group is BFRESGroupNode)
                {
                    if (((BFRESGroupNode)group).Type != BRESGroupType.Textures)
                        continue;

                    foreach (FTEX tex in ((BFRESGroupNode)group).Nodes)
                    {
                        if (resFileTex2.Textures.ContainsKey(tex.Text))
                        {
                            resFileTex2.Textures[tex.Text].Data = null;
                            resFileTex2.Textures[tex.Text].MipData = tex.texture.MipData;
                            resFileTex2.Textures[tex.Text].MipOffsets = tex.texture.MipOffsets;
                            resFileTex2.Textures[tex.Text].MipCount = tex.texture.MipCount;
                            resFileTex2.Textures[tex.Text].Swizzle = 65536;
                        }
                    }
                }
            }

            resFileTex2.Save(mem);

            return mem.ToArray();
        }

        private void SaveTex2(string fileName)
        {
            var resFileU = BFRESRender.ResFileNode.resFileU;

            bool Compressed = fileName.EndsWith("sbfres");

            byte[] data;
            if (Compressed)
                data = EveryFileExplorer.YAZ0.Decompress(fileName);
            else
                data = File.ReadAllBytes(fileName);

            ResU.ResFile resFileTex2 = new ResU.ResFile(new MemoryStream(data));

        
            foreach (BFRESGroupNode group in Nodes)
            {
                if (group.Type != BRESGroupType.Textures)
                    return;

                foreach (FTEX tex in group.Nodes)
                {
                    if (resFileTex2.Textures.ContainsKey(tex.Text))
                    {
                        resFileTex2.Textures[tex.Text].MipData = tex.texture.MipData;
                        resFileTex2.Textures[tex.Text].MipOffsets = tex.texture.MipOffsets;
                        resFileTex2.Textures[tex.Text].MipCount = tex.texture.MipCount;
                    }
                }
            }
            MemoryStream mem2 = new MemoryStream();
            resFileTex2.Save(mem2);

            SaveFileDialog sfd = new SaveFileDialog();
            sfd.FileName = FileName + "NewTex2.sbfres";

            List<IFileFormat> formats = new List<IFileFormat>();
            formats.Add(this);
            sfd.Filter = Utils.GetAllFilters(formats);

            if (sfd.ShowDialog() == DialogResult.OK)
                STFileSaver.SaveFileFormat(mem2.ToArray(), Compressed, 0, CompressionType.Yaz0, sfd.FileName);
        }
        private void Rename(object sender, EventArgs args)
        {
            RenameDialog dialog = new RenameDialog();
            dialog.SetString(Text);

            if (dialog.ShowDialog() == DialogResult.OK)
            {
                Text = dialog.textBox1.Text;
            }
        }
        private void Remove(object sender, EventArgs args)
        {
            Unload();
        }
        private void SaveSwitch(MemoryStream mem)
        {
            var resFile = BFRESRender.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();
            resFile.ExternalFiles.Clear();
            resFile.ExternalFileDict.Clear();

            foreach (TreeNode node in Nodes)
            {
                if (node is BFRESGroupNode)
                    SaveBfresSwitchGroup((BFRESGroupNode)node, resFile);
                
                if (node is BFRESAnimFolder)
                {
                    foreach (var animGroup in node.Nodes)
                        SaveBfresSwitchGroup((BFRESGroupNode)animGroup, resFile);
                }
                if (node is BNTX)
                {
                    if (((BNTX)node).Textures.Count > 0)
                    {
                        resFile.ExternalFiles.Add(new ExternalFile() { Data = ((BNTX)node).Save() });
                        resFile.ExternalFileDict.Add("textures.bntx");
                    }
                }
            }

            ErrorCheck();
            resFile.Save(mem);
        }

        private static void SaveBfresSwitchGroup(BFRESGroupNode group, ResFile resFile)
        {
            switch (group.Type)
            {
                case BRESGroupType.Models:
                    foreach (FMDL model in group.Nodes)
                    {
                        model.Model.Name = model.Text;
                        resFile.Models.Add(BfresSwitch.SetModel(model));
                    }
                    break;
                case BRESGroupType.SkeletalAnim:
                    foreach (FSKA ska in group.Nodes)
                    {
                        ska.SkeletalAnim.BoneAnims.Clear();

                        foreach (FSKA.BoneAnimNode bone in ska.Bones)
                            ska.SkeletalAnim.BoneAnims.Add(bone.SaveData());

                        ska.SkeletalAnim.Name = ska.Text;
                        resFile.SkeletalAnims.Add(ska.SkeletalAnim);
                    }
                    break;
                case BRESGroupType.TexPatAnim:
                case BRESGroupType.ShaderParamAnim:
                case BRESGroupType.ColorAnim:
                case BRESGroupType.TexSrtAnim:
                case BRESGroupType.MatVisAnim:
                    foreach (FMAA fmaa in group.Nodes)
                    {
                        fmaa.MaterialAnim.MaterialAnimDataList.Clear();
                        foreach (FMAA.MaterialAnimEntry mat in fmaa.Materials)
                            fmaa.MaterialAnim.MaterialAnimDataList.Add(mat.SaveData());

                        fmaa.SetName();
                        fmaa.SaveAnimData();
                        resFile.MaterialAnims.Add(fmaa.MaterialAnim);
                    }
                    break;
                case BRESGroupType.BoneVisAnim:
                    foreach (FVIS fbnv in group.Nodes)
                    {
                        fbnv.SaveData();
                        fbnv.VisibilityAnim.Name = fbnv.Text;
                        resFile.BoneVisibilityAnims.Add(fbnv.VisibilityAnim);
                    }
                    break;
                case BRESGroupType.ShapeAnim:
                    foreach (FSHA fsha in group.Nodes)
                    {
                        fsha.ShapeAnim.VertexShapeAnims.Clear();
                        foreach (FSHA.ShapeAnimEntry shp in fsha.Nodes)
                            fsha.ShapeAnim.VertexShapeAnims.Add(shp.SaveData());

                        fsha.ShapeAnim.Name = fsha.Text;
                        resFile.ShapeAnims.Add(fsha.ShapeAnim);
                    }
                    break;
                case BRESGroupType.SceneAnim:
                    foreach (FSCN fscn in group.Nodes)
                    {
                        fscn.SceneAnim.Name = fscn.Text;
                        resFile.SceneAnims.Add(fscn.SceneAnim);
                    }
                    break;
                case BRESGroupType.Embedded:
                    foreach (var ext in group.Nodes)
                    {
                        if (ext is ExternalFileData)
                        {
                            resFile.ExternalFiles.Add(new ExternalFile() { Data = ((ExternalFileData)ext).Data });
                            resFile.ExternalFileDict.Add(((ExternalFileData)ext).Text);
                        }
                        else if (ext is TreeNodeFile)
                        {
                            resFile.ExternalFiles.Add(new ExternalFile() { Data = ((IFileFormat)ext).Save() });
                            resFile.ExternalFileDict.Add(((TreeNodeFile)ext).Text);
                        }

                    }
                    break;
            }
        }

        private void SaveMaterialAnims(TreeNodeCollection nodes)
        {

        }
            

        private void SaveWiiU(MemoryStream mem)
        {
            var resFileU = BFRESRender.ResFileNode.resFileU;

            resFileU.Models.Clear();
            resFileU.Textures.Clear();
            resFileU.SkeletalAnims.Clear();
            resFileU.ShaderParamAnims.Clear();
            resFileU.ColorAnims.Clear();
            resFileU.TexSrtAnims.Clear();
            resFileU.TexPatternAnims.Clear();
            resFileU.BoneVisibilityAnims.Clear();
            resFileU.MatVisibilityAnims.Clear();
            resFileU.ShapeAnims.Clear();
            resFileU.SceneAnims.Clear();
            resFileU.ExternalFiles.Clear();

            bool IsTex1 = FilePath.Contains("Tex1");

            foreach (TreeNode node in Nodes)
            {
                if (node is BFRESGroupNode)
                    SaveBfresWiiUGroup((BFRESGroupNode)node, resFileU);

                if (node is BFRESAnimFolder)
                {
                    foreach (var animGroup in node.Nodes)
                        SaveBfresWiiUGroup((BFRESGroupNode)animGroup, resFileU);
                }
            }

            //     ErrorCheck();
            resFileU.Save(mem);
        }

        private static void SaveBfresWiiUGroup(BFRESGroupNode group, ResU.ResFile resFileU)
        {
            switch (group.Type)
            {
                case BRESGroupType.Models:
                    foreach (FMDL model in group.Nodes)
                    {
                        model.ModelU.Name = model.Text;
                        resFileU.Models.Add(model.Text, BfresWiiU.SetModel(model));
                    }
                    break;
                case BRESGroupType.Textures:
                    foreach (FTEX tex in group.Nodes)
                    {
                        tex.texture.Name = tex.Text;
                        resFileU.Textures.Add(tex.Text, tex.texture);
                    }
                    break;
                case BRESGroupType.SkeletalAnim:
                    foreach (FSKA ska in group.Nodes)
                    {
                        ska.SkeletalAnimU.BoneAnims.Clear();
                        foreach (FSKA.BoneAnimNode bone in ska.Bones)
                            ska.SkeletalAnimU.BoneAnims.Add(bone.SaveDataU());

                        ska.SkeletalAnimU.Name = ska.Text;
                        resFileU.SkeletalAnims.Add(ska.Text, ska.SkeletalAnimU);
                    }
                    break;
                case BRESGroupType.ShaderParamAnim:
                    foreach (FSHU anim in group.Nodes)
                    {
                        anim.ShaderParamAnim.ShaderParamMatAnims.Clear();
                        foreach (FSHU.MaterialAnimEntry bone in anim.Materials)
                            anim.ShaderParamAnim.ShaderParamMatAnims.Add(bone.SaveData());


                        anim.ShaderParamAnim.Name = anim.Text;
                        resFileU.ShaderParamAnims.Add(anim.Text, anim.ShaderParamAnim);
                    }
                    break;
                case BRESGroupType.ColorAnim:
                    foreach (FSHU anim in group.Nodes)
                    {
                        anim.ShaderParamAnim.ShaderParamMatAnims.Clear();
                        foreach (FSHU.MaterialAnimEntry bone in anim.Materials)
                            anim.ShaderParamAnim.ShaderParamMatAnims.Add(bone.SaveData());

                        anim.ShaderParamAnim.Name = anim.Text;
                        resFileU.ColorAnims.Add(anim.Text, anim.ShaderParamAnim);
                    }
                    break;
                case BRESGroupType.TexSrtAnim:
                    foreach (FSHU anim in group.Nodes)
                    {
                        anim.ShaderParamAnim.ShaderParamMatAnims.Clear();
                        foreach (FSHU.MaterialAnimEntry bone in anim.Materials)
                            anim.ShaderParamAnim.ShaderParamMatAnims.Add(bone.SaveData());

                        anim.ShaderParamAnim.Name = anim.Text;
                        resFileU.TexSrtAnims.Add(anim.Text, anim.ShaderParamAnim);
                    }
                    break;
                case BRESGroupType.TexPatAnim:
                    foreach (FTXP anim in group.Nodes)
                    {
                        anim.TexPatternAnim.Name = anim.Text;
                        resFileU.TexPatternAnims.Add(anim.Text, anim.TexPatternAnim);
                    }
                    break;
                case BRESGroupType.MatVisAnim:
                    foreach (FVIS anim in group.Nodes)
                    {
                        anim.SaveData();
                        anim.VisibilityAnimU.Name = anim.Text;
                        resFileU.MatVisibilityAnims.Add(anim.Text, anim.VisibilityAnimU);
                    }
                    break;
                case BRESGroupType.BoneVisAnim:
                    foreach (FVIS anim in group.Nodes)
                    {
                        anim.SaveData();
                        anim.VisibilityAnimU.Name = anim.Text;
                        resFileU.BoneVisibilityAnims.Add(anim.Text, anim.VisibilityAnimU);
                    }
                    break;
                case BRESGroupType.ShapeAnim:
                    foreach (FSHA anim in group.Nodes)
                    {
                        anim.ShapeAnimU.VertexShapeAnims.Clear();
                        foreach (FSHA.ShapeAnimEntry shp in anim.Nodes)
                            anim.ShapeAnimU.VertexShapeAnims.Add(shp.SaveDataU());

                        anim.ShapeAnimU.Name = anim.Text;
                        resFileU.ShapeAnims.Add(anim.Text, anim.ShapeAnimU);
                    }
                    break;
                case BRESGroupType.SceneAnim:
                    foreach (FSCN anim in group.Nodes)
                    {
                        anim.SceneAnimU.Name = anim.Text;
                        resFileU.SceneAnims.Add(anim.Text, anim.SceneAnimU);
                    }
                    break;
                case BRESGroupType.Embedded:
                    foreach (TreeNode ext in group.Nodes)
                    {
                        if (ext is ExternalFileData)
                        {
                            resFileU.ExternalFiles.Add(ext.Text, new ResU.ExternalFile() { Data = ((ExternalFileData)ext).Data });
                        }
                    }
                    break;
            }
        }

        public static void SetShaderAssignAttributes(FMAT.ShaderAssign shd, FSHP shape)
        {
            foreach (var att in shape.vertexAttributes)
            {
                if (!shd.attributes.ContainsValue(att.Name))
                {
                    try
                    {
                        shd.attributes.Add(att.Name, att.Name);
                    }
                    catch (Exception ex)
                    {
                        Console.WriteLine("Attribute link failed! \n " + ex);
                    }
                }
            }
            foreach (var tex in shape.GetMaterial().TextureMaps)
            {
                if (!shd.samplers.ContainsValue(tex.SamplerName))
                {
                    try
                    {
                        shd.samplers.Add(tex.SamplerName, tex.SamplerName);
                    }
                    catch (Exception ex)
                    {
                        Console.WriteLine("Sampler link failed! \n " + ex);
                    }
                }
            }
        }


        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;
                }
            }
        }

        static bool ImportMissingTextures = false;
        public static void CheckMissingTextures(FSHP shape)
        {
            // FSHP > Objects > FMDL > Models > BFRES
            BFRES root = (BFRES)shape.Parent.Parent.Parent.Parent;

            foreach (var node in root.Nodes)
            {
                if (node is BNTX)
                {
                    BNTX bntx = (BNTX)node;
                    if (bntx.Textures.Count == 0)
                        return;

                    foreach (MatTexture tex in shape.GetMaterial().TextureMaps)
                    {
                        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);
                        }
                    }
                }
            }

            foreach (var node in root.Nodes)
            {
                if (node is BFRESGroupNode && ((BFRESGroupNode)node).Type == BRESGroupType.Textures)
                {
                    if (((BFRESGroupNode)node).ResourceNodes.Count <= 0)
                        return;

                    foreach (MatTexture tex in shape.GetMaterial().TextureMaps)
                    {
                        if (!((BFRESGroupNode)node).ResourceNodes.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)
                                ((BFRESGroupNode)node).ImportPlaceholderTexture(tex.Name);
                        }
                    }
                }
            }
        }

        public void ErrorCheck()
        {
            if (BFRESRender != null)
            {
                List<Errors> Errors = new List<Errors>();
                foreach (FMDL model in BFRESRender.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))
                                    STConsole.WriteLine($"Error! Attribute {vtx.Attributes[att].Name} is unlinked!");
                            }
                            for (int att = 0; att < mat.Samplers.Count; att++)
                            {
                                if (!shdr.SamplerAssigns.Contains(mat.SamplerDict.GetKey(att))) //mat.SamplerDict[att]
                                    STConsole.WriteLine($"Error! Sampler {mat.Samplers[att].Name} 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++)
                            {
                                 ResU.ResString str = new ResU.ResString();
                                str.String = vtx.Attributes[att].Name;
                                if (!shdr.AttribAssigns.ContainsValue(str))
                                    STConsole.WriteLine($"Error! Attribute {vtx.Attributes[att].Name} is unlinked!");
                            }

                            for (int att = 0; att < mat.Samplers.Count; att++)
                            {
                                ResU.ResString str2 = new ResU.ResString();
                                str2.String = mat.Samplers[att].Name;
                                if (!shdr.SamplerAssigns.ContainsValue(str2)) //mat.SamplerDict[att]
                                    STConsole.WriteLine($"Error! Sampler {mat.Samplers[att].Name} 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";
        }
    }
}