Switch-Toolbox/Switch_Toolbox_Library/FileFormats/DAE/ColladaWriter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Xml;
using System.IO;

namespace Toolbox.Library.Collada
{
    public class ColladaWriter : IDisposable
    {
        //Based around this nice Writer https://raw.githubusercontent.com/Ploaj/SSBHLib/master/CrossMod/IO/DAEWriter.cs
        //With adjustments and expanded usage.

        private XmlTextWriter Writer;
        private DAE.ExportSettings Settings;
        private DAE.Version Version;

        private List<Tuple<string, SemanticType, TriangleList[], int>> GeometrySources = new List<Tuple<string, SemanticType, TriangleList[], int>>();

        private Dictionary<string, int> AttributeIdList = new Dictionary<string, int>();

        private Dictionary<string, Tuple<List<int[]>, List<float[]>>> GeometryControllers = new Dictionary<string, Tuple<List<int[]>, List<float[]>>>();

        //Keep a dictionary of names and assigned id names
        private Dictionary<string, string> MeshIdList = new Dictionary<string, string>();
        private Dictionary<string, List<string>> MaterialIdList = new Dictionary<string, List<string>>();

        private Dictionary<string, string> MeshSkinIdList = new Dictionary<string, string>();

        private List<Joint> Joints = new List<Joint>();

        public string CurrentGeometryID;
        public string CurrentMaterial;

        public bool Optimize = false;

        public ColladaWriter(string fileName, DAE.ExportSettings settings)
        {
            Settings = settings;
            Version = settings.FileVersion;
            Writer = new XmlTextWriter(fileName, Encoding.UTF8)
            {
                Formatting = Formatting.Indented,
                Indentation = 2,
            };

            WriteDAEHeader();
        }

        public void WriteDAEHeader()
        {
            Writer.WriteStartDocument();
            Writer.WriteStartElement("COLLADA");
            Writer.WriteAttributeString("xmlns", "http://www.collada.org/2005/11/COLLADASchema");
            Writer.WriteAttributeString("version", $"{Version.Major}.{Version.Minor}.{Version.Micro}");
        }

        public void WriteAsset()
        {
            Writer.WriteStartElement("asset");
            Writer.WriteEndElement();
        }

        public void WriteFileSettings()
        {
            Writer.WriteStartElement("asset");
            {
                Writer.WriteStartElement("contributor");
                Writer.WriteElementString("authoring_tool", System.Windows.Forms.Application.ProductName);
                Writer.WriteEndElement();

                Writer.WriteStartElement("created");
                Writer.WriteString(DateTime.UtcNow.ToString("s", System.Globalization.CultureInfo.InvariantCulture) + "Z");
                Writer.WriteEndElement();

                Writer.WriteStartElement("modified");
                Writer.WriteString(DateTime.UtcNow.ToString("s", System.Globalization.CultureInfo.InvariantCulture) + "Z");
                Writer.WriteEndElement();

                Writer.WriteStartElement("unit");
                if (Settings.Preset == ProgramPreset.MAX)
                {
                    Writer.WriteAttributeString("meter", "1");
                    Writer.WriteAttributeString("name", "centimeter");
                }
                else
                {
                    Writer.WriteAttributeString("meter", "0.01");
                    Writer.WriteAttributeString("name", "centimeter");
                }
                Writer.WriteEndElement();

                Writer.WriteStartElement("up_axis");
                Writer.WriteString("Y_UP");
                Writer.WriteEndElement();
            }
            Writer.WriteEndElement();
        }

        public void WriteLibraryImages(string[] textureNames = null)
        {
            Writer.WriteStartElement("library_images");
            for (int i = 0; i < textureNames?.Length; i++)
            {
                Writer.WriteStartElement("image");
                Writer.WriteAttributeString("id", textureNames[i]);
                Writer.WriteStartElement("init_from");
                Writer.WriteString($"{Settings.ImageFolder}{textureNames[i]}{Settings.ImageExtension}");
                Writer.WriteEndElement();
                Writer.WriteEndElement();
            }
            Writer.WriteEndElement();
        }

        public void WriteLibraryMaterials(List<Material> materials)
        {
            Writer.WriteStartElement("library_materials");
            for (int i = 0; i < materials?.Count; i++)
            {
                Writer.WriteStartElement("material");
                Writer.WriteAttributeString("id", materials[i].Name);
                Writer.WriteStartElement("instance_effect");
                Writer.WriteAttributeString("url", "#Effect_" + materials[i].Name);
                Writer.WriteEndElement();
                Writer.WriteEndElement();
            }
            Writer.WriteEndElement();
        }

        public void WriteLibraryEffects(List<Material> materials)
        {
            Writer.WriteStartElement("library_effects");
            for (int i = 0; i < materials?.Count; i++)
            {
                Writer.WriteStartElement("effect");
                Writer.WriteAttributeString("id", $"Effect_{materials[i].Name}");
                Writer.WriteStartElement("profile_COMMON");
                {
                    for (int t = 0; t < materials[i].Textures.Count; t++)
                    {
                        var tex = materials[i].Textures[t];
                        //Here we write 2 things. Surface then sampler info

                        //Surface info
                        Writer.WriteStartElement("newparam");
                        Writer.WriteAttributeString("sid", $"surface_{materials[i].Name}-{tex.Type}");
                        Writer.WriteStartElement("surface");
                        Writer.WriteAttributeString("type", "2D");
                        {
                            Writer.WriteStartElement("init_from");
                            Writer.WriteString(tex.Name);
                            Writer.WriteEndElement();
                            Writer.WriteStartElement("format");
                            //Formats are uncompressed types. 
                            //Todo these should be adjusted for DDS
                            Writer.WriteString("A8R8G8B8");
                            Writer.WriteEndElement();
                        }
                        Writer.WriteEndElement();
                        Writer.WriteEndElement();

                        {
                            //Sampler info
                            Writer.WriteStartElement("newparam");
                            Writer.WriteAttributeString("sid", $"sampler_{materials[i].Name}-{tex.Type}");
                            Writer.WriteStartElement("sampler2D");

                            Writer.WriteStartElement("source");
                            Writer.WriteString($"surface_{materials[i].Name}-{tex.Type}");
                            Writer.WriteEndElement();

                            //Add sampler wrap modes
                            Writer.WriteStartElement("wrap_s");
                            Writer.WriteString(tex.WrapModeS.ToString());
                            Writer.WriteEndElement();

                            Writer.WriteStartElement("wrap_t");
                            Writer.WriteString(tex.WrapModeT.ToString());
                            Writer.WriteEndElement();


                            Writer.WriteEndElement();
                            Writer.WriteEndElement();
                        }

                    }

                    //Now write texture info
                    //Todo support PBR workflow

                    Writer.WriteStartElement("technique");
                    Writer.WriteAttributeString("sid", "common");
                    Writer.WriteStartElement("phong");

                    bool HasSpecular = false;
                    bool HasEmission = false;
                    bool HasDiffuse = false;

                    for (int t = 0; t < materials[i].Textures.Count; t++)
                    {
                        var tex = materials[i].Textures[t];

                        if (tex.Type == PhongTextureType.diffuse)
                            HasDiffuse = true;
                        if (tex.Type == PhongTextureType.emission)
                            HasEmission = true;
                        if (tex.Type == PhongTextureType.specular)
                            HasSpecular = true;

                        Writer.WriteStartElement($"{tex.Type}");
                        Writer.WriteStartElement("texture");
                        Writer.WriteAttributeString("texture", $"sampler_{materials[i].Name}-{tex.Type}");
                        Writer.WriteAttributeString("texcoord", $"{tex.TextureChannel}");

                        Writer.WriteEndElement();
                        Writer.WriteEndElement();
                    }

                    if (!HasDiffuse)
                        WriteMaterialColor("diffuse", new float[] { 1, 1, 1, 1 });
                    if (!HasEmission)
                        WriteMaterialColor("emission", new float[] { 0, 0, 0, 1 });
                    if (!HasSpecular)
                        WriteMaterialColor("specular", new float[] { 0, 0, 0, 1 });

                    Writer.WriteEndElement();
                    Writer.WriteEndElement();
                }
                Writer.WriteEndElement();
                Writer.WriteEndElement();
            }
            Writer.WriteEndElement();
        }

        public void WriteMaterialColor(string type, float[] color)
        {
            Writer.WriteStartElement(type);
            Writer.WriteStartElement("color");
            Writer.WriteAttributeString("sid", type);
            Writer.WriteString(string.Join(" ", color));
            Writer.WriteEndElement();
            Writer.WriteEndElement();
        }

        public void StartLibraryGeometries() {
            Writer.WriteStartElement("library_geometries");
        }

        public void StartGeometry(string name)
        {
            CurrentGeometryID = GetUniqueID($"{name}-mesh");
            MeshSkinIdList.Add(CurrentGeometryID, "");
            MeshIdList.Add(CurrentGeometryID, name);
            Writer.WriteStartElement("geometry");
            Writer.WriteAttributeString("id", CurrentGeometryID);
            Writer.WriteAttributeString("name", name);
            Writer.WriteStartElement("mesh");
        }

        private void OptimizeSource<T>(T[] values, TriangleList[] triangleLists, int stride, out T[] newValues, out TriangleList[] newIndices) where T : struct
        {
            var optimizedIndices = new List<uint>();
            var optimizedValues = new List<T>();

            // Use a special class to store values, so we can have the performance benefits of a dictionary.
            var vertexBank = new Dictionary<ValueContainer<T>, uint>();

            for (int t = 0; t < triangleLists.Length; t++)
            {
                for (int i = 0; i < triangleLists[t].Indices.Count; i++)
                {
                    var vertexValues = new ValueContainer<T>(GetVertexValues(values, triangleLists[t].Indices.ToArray(), stride, i));

                    if (!vertexBank.ContainsKey(vertexValues))
                    {
                        uint index = (uint)vertexBank.Count;
                        optimizedIndices.Add(index);
                        vertexBank.Add(vertexValues, index);
                        optimizedValues.AddRange(vertexValues.Values);
                    }
                    else
                    {
                        optimizedIndices.Add(vertexBank[vertexValues]);

                    }
                }

                triangleLists[t].Indices = optimizedIndices;
            }

            newIndices = triangleLists;
            newValues = optimizedValues.ToArray();
        }

        private static T[] GetVertexValues<T>(T[] values, uint[] indices, int stride, int i)
        {
            var vertexValues = new T[stride];
            for (int j = 0; j < stride; j++)
                vertexValues[j] = values[indices[i] * stride + j];

            return vertexValues;
        }

        public void WriteGeometrySource(string name, SemanticType semantic, float[] values, TriangleList[] indices, int set = -1)
        {
            int stride = GetStride(semantic);

            if (Optimize)
                OptimizeSource(values, indices, stride, out values, out indices);

            string sourceid = GetUniqueID(name + "-" + semantic.ToString().ToLower());
            Writer.WriteStartElement("source");
            Writer.WriteAttributeString("id", sourceid);

            Writer.WriteStartElement("float_array");
            string FloatArrayID = GetUniqueID(name + "-" + semantic.ToString().ToLower() + "-array");
            Writer.WriteAttributeString("id", FloatArrayID);
            Writer.WriteAttributeString("count", values.Length.ToString());
            Writer.WriteString(string.Join(" ", values));
            Writer.WriteEndElement();

            Writer.WriteStartElement("technique_common");
            {
                Writer.WriteStartElement("accessor");
                Writer.WriteAttributeString("source", $"#{FloatArrayID}");
                Writer.WriteAttributeString("count", (values.Length / stride).ToString());
                Writer.WriteAttributeString("stride", stride.ToString());
                if (semantic == SemanticType.NORMAL || semantic == SemanticType.POSITION)
                {
                    WriteParam("X", "float");
                    WriteParam("Y", "float");
                    WriteParam("Z", "float");
                }
                if (semantic == SemanticType.TEXCOORD)
                {
                    WriteParam("S", "float");
                    WriteParam("T", "float");
                }
                if (semantic == SemanticType.COLOR)
                {
                    WriteParam("R", "float");
                    WriteParam("G", "float");
                    WriteParam("B", "float");
                    WriteParam("A", "float");
                }
                Writer.WriteEndElement();
            }
            Writer.WriteEndElement();
            Writer.WriteEndElement();

            GeometrySources.Add(new Tuple<string, SemanticType, TriangleList[], int>(sourceid, semantic, indices, set));
        }

        private int GetStride(SemanticType semantic) {
            if (semantic == SemanticType.COLOR)
                return 4;
            else if (semantic == SemanticType.TEXCOORD)
                return 2;
            else if (semantic == SemanticType.JOINT)
                return 1;
            else if (semantic == SemanticType.INV_BIND_MATRIX)
                return 16;
            else if (semantic == SemanticType.WEIGHT)
                return 1;
            else
                return 3;
        }

        private void WriteParam(string Name, string Type)
        {
            Writer.WriteStartElement("param");
            Writer.WriteAttributeString("name", Name);
            Writer.WriteAttributeString("type", Type);
            Writer.WriteEndElement();
        }

        public void EndGeometryMesh()
        {
            Tuple<string, SemanticType, TriangleList[], int> Position = null;
            foreach (var v in GeometrySources)
            {
                if (v.Item2 == SemanticType.POSITION)
                {
                    Position = v;
                    break;
                }
            }

            // can only write vertex information if there is at least position data
            if (Position != null)
            {
                // vertices
                string verticesid = GetUniqueID(Position.Item1.Replace("position", "vertex"));
                Writer.WriteStartElement("vertices");
                Writer.WriteAttributeString("id", verticesid);
                WriteInput(Position.Item2.ToString(), Position.Item1);
                Writer.WriteEndElement();

                List<string> triangleMaterials = new List<string>();

                // triangles
                int triIndex = 0;
                foreach (TriangleList triangleList in Position.Item3)
                {
                    Writer.WriteStartElement("triangles");
                    if (triangleList.Material != "")
                    {
                        Writer.WriteAttributeString("material", $"{triangleList.Material}");
                        triangleMaterials.Add(triangleList.Material);
                    }
                    else if (CurrentMaterial != "")
                    {
                        Writer.WriteAttributeString("material", $"{CurrentMaterial}");
                        triangleMaterials.Add(CurrentMaterial);
                        CurrentMaterial = "";
                    }

                    Writer.WriteAttributeString("count", (triangleList.Indices.Count / 3).ToString());
                    WriteInput("VERTEX", $"{verticesid}", 0);
                    int offset = 1;
                    foreach (var v in GeometrySources)
                        if (v.Item2 != SemanticType.POSITION)
                        {
                            WriteInput(v.Item2.ToString(), $"{v.Item1}", offset++, v.Item4);
                        }
                    // write p
                    StringBuilder p = new StringBuilder();
                    for (int i = 0; i < triangleList.Indices.Count; i++)
                    {
                        p.Append(triangleList.Indices[i] + " ");
                        foreach (var v in GeometrySources)
                            if (v.Item2 != SemanticType.POSITION)
                                p.Append(v.Item3[triIndex].Indices[i] + " ");
                    }
                    Writer.WriteStartElement("p");
                    Writer.WriteString(p.ToString());
                    Writer.WriteEndElement();

                    Writer.WriteEndElement();

                    triIndex++;
                }

                MaterialIdList.Add(CurrentGeometryID, triangleMaterials);
            }

            GeometrySources.Clear();
            Writer.WriteEndElement();
            Writer.WriteEndElement();
        }

        public void WriteInput(string semantic, string sourceid, int offset = -1, int set = -1)
        {
            Writer.WriteStartElement("input");
            Writer.WriteAttributeString("semantic", semantic);
            Writer.WriteAttributeString("source", $"#{sourceid}");
            if (offset != -1)
                Writer.WriteAttributeString("offset", offset.ToString());
            if (set != -1)
                Writer.WriteAttributeString("set", set.ToString());
            Writer.WriteEndElement();
        }

        /// <summary>
        /// A function for automatically creating and handling the controllers
        /// The Joints must be added before this function is used
        /// </summary>
        /// <param name="BoneIndices">A list of arrays of bone indices for each vertex</param>
        /// <param name="Weights">A list that contains an array of weights per vertex</param>
        public void AttachGeometryController(List<int[]> BoneIndices, List<float[]> Weights)
        {
            GeometryControllers.Add(CurrentGeometryID, new Tuple<List<int[]>, List<float[]>>(BoneIndices, Weights));
        }

        /// <summary>
        /// Ends Writing the Geometry Section
        /// </summary>
        public void EndGeometrySection()
        {
            Writer.WriteEndElement();

            CreateControllerSection();

            CreateVisualNodeSection();
        }

        private void RecursivlyWriteJoints(Joint joint)
        {
            Writer.WriteStartElement("node");
            Writer.WriteAttributeString("id", $"Armature_{joint.Name}");
            Writer.WriteAttributeString("name", joint.Name);
            Writer.WriteAttributeString("sid", joint.Name);
            Writer.WriteAttributeString("type", "JOINT");

            Writer.WriteStartElement("matrix");
            Writer.WriteAttributeString("sid", "transform");
            Writer.WriteString(string.Join(" ", joint.Transform));
            Writer.WriteEndElement();

            foreach (var child in GetChildren(joint))
            {
                RecursivlyWriteJoints(child);
            }

            Writer.WriteEndElement();
        }

        private Joint[] GetChildren(Joint j)
        {
            int parentindex = Joints.IndexOf(j);
            List<Joint> Children = new List<Joint>();
            foreach (var child in Joints)
            {
                if (child.ParentIndex == parentindex)
                    Children.Add(child);
            }
            return Children.ToArray();
        }

        /// <summary>
        /// Starts the library controller section
        /// </summary>
        public void BeginLibraryControllers()
        {
            Writer.WriteStartElement("library_controllers");
        }

        /// <summary>
        /// automatically generates the controller nodes
        /// </summary>
        private void CreateControllerSection()
        {
            BeginLibraryControllers();

            foreach (var v in GeometryControllers.Keys)
            {
                WriteLibraryController(v);
            }

            EndLibraryControllers();
        }


        /// <summary>
        /// Adds a new joint to the default skeletal tree
        /// </summary>
        public void AddJoint(string name, string parentName, float[] Transform, float[] InvWorldTransform)
        {
            Joint j = new Joint();
            j.Name = name;
            j.Transform = Transform;
            j.BindPose = InvWorldTransform;
            foreach (var joint in Joints)
                if (joint.Name.Equals(parentName))
                    j.ParentIndex = Joints.IndexOf(joint);
            Joints.Add(j);
        }

        public void WriteLibraryController(string Name)
        {
            if (!GeometryControllers.ContainsKey(Name)) return;

            var BoneWeight = GeometryControllers[Name];
            string SkinID = $"Armature_{Name}";
            MeshSkinIdList[Name] = SkinID;

            Writer.WriteStartElement("controller");
            Writer.WriteAttributeString("id", SkinID);

            Writer.WriteStartElement("skin");
            Writer.WriteAttributeString("source", $"#{Name}");
            {
                Writer.WriteStartElement("bind_shape_matrix");
                Writer.WriteString("1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1");
                Writer.WriteEndElement();

                object[] BoneNames = new string[Joints.Count];
                object[] InvBinds = new object[Joints.Count * 16];
                for (int i = 0; i < BoneNames.Length; i++)
                {
                    BoneNames[i] = Joints[i].Name;
                    for (int j = 0; j < 16; j++)
                    {
                        InvBinds[i * 16 + j] = Joints[i].BindPose[j];
                    }
                }
                var Weights = new List<object>();
                var WeightIndices = new List<int>();
                foreach (var v in BoneWeight.Item2)
                {
                    foreach (var w in v)
                    {
                        int index = Weights.IndexOf(w);
                        if (index == -1)
                        {
                            WeightIndices.Add(Weights.Count);
                            Weights.Add(w);
                        }
                        else
                        {
                            WeightIndices.Add(index);
                        }
                    }
                }
                string Jointid = WriteSkinSource(Name, SemanticType.JOINT, BoneNames);
                string Bindid = WriteSkinSource(Name, SemanticType.INV_BIND_MATRIX, InvBinds);
                string Weightid = WriteSkinSource(Name, SemanticType.WEIGHT, Weights.ToArray());


                Writer.WriteStartElement("joints");
                WriteInput(SemanticType.JOINT.ToString(), Jointid);
                WriteInput(SemanticType.INV_BIND_MATRIX.ToString(), Bindid);
                Writer.WriteEndElement();

                Writer.WriteStartElement("vertex_weights");
                Writer.WriteAttributeString("count", BoneWeight.Item1.Count.ToString());
                WriteInput(SemanticType.JOINT.ToString(), Jointid, 0);
                WriteInput(SemanticType.WEIGHT.ToString(), Weightid, 1);

                // now writing out the counts and such...
                //vcount
                {
                    StringBuilder values = new StringBuilder();
                    foreach (var v in BoneWeight.Item1)
                    {
                        values.Append($"{v.Length} ");
                    }
                    Writer.WriteStartElement("vcount");
                    Writer.WriteString(values.ToString());
                    Writer.WriteEndElement();
                }

                //v
                {
                    StringBuilder values = new StringBuilder();
                    int weightindexcount = 0;
                    for (int i = 0; i < BoneWeight.Item1.Count; i++)
                    {
                        for (int j = 0; j < BoneWeight.Item1[i].Length; j++)
                        {
                            values.Append($"{BoneWeight.Item1[i][j]} {WeightIndices[weightindexcount++]} ");
                        }
                    }
                    Writer.WriteStartElement("v");
                    Writer.WriteString(values.ToString());
                    Writer.WriteEndElement();
                }

                Writer.WriteEndElement();
            }

            Writer.WriteEndElement();
            Writer.WriteEndElement();
        }

        /// <summary>
        /// Writes a skin source accessor for the current skin controller
        /// </summary>
        /// <param name="Name"></param>
        /// <param name="Semantic"></param>
        /// <param name="Values"></param>
        /// <returns>the id of the newly created skin source</returns>
        public string WriteSkinSource(string Name, SemanticType Semantic, object[] Values)
        {
            int Stride = GetStride(Semantic);

            string sourceid = GetUniqueID(Name + "-" + Semantic.ToString().ToLower());
            Writer.WriteStartElement("source");
            Writer.WriteAttributeString("id", sourceid);

            Writer.WriteStartElement(Semantic == SemanticType.JOINT ? "Name_array" : "float_array");
            string FloatArrayID = GetUniqueID(Name + "-" + Semantic.ToString().ToLower() + "-array");
            Writer.WriteAttributeString("id", FloatArrayID);
            Writer.WriteAttributeString("count", Values.Length.ToString());
            Writer.WriteString(string.Join(" ", Values));
            Writer.WriteEndElement();

            Writer.WriteStartElement("technique_common");
            {
                Writer.WriteStartElement("accessor");
                Writer.WriteAttributeString("source", $"#{FloatArrayID}");
                Writer.WriteAttributeString("count", (Values.Length / Stride).ToString());
                Writer.WriteAttributeString("stride", Stride.ToString());
                if (Semantic == SemanticType.JOINT)
                {
                    WriteParam("JOINT", "name");
                }
                if (Semantic == SemanticType.INV_BIND_MATRIX)
                {
                    WriteParam("TRANSFORM", "float4x4");
                }
                if (Semantic == SemanticType.WEIGHT)
                {
                    WriteParam("WEIGHT", "float");
                }
                Writer.WriteEndElement();
            }
            Writer.WriteEndElement();

            Writer.WriteEndElement();

            return sourceid;
        }

        /// <summary>
        /// Ends the Library Controller Section
        /// </summary>
        public void EndLibraryControllers()
        {
            Writer.WriteEndElement();
        }



        public void BeginVisualNodeSection()
        {
            Writer.WriteStartElement("library_visual_scenes");
            Writer.WriteStartElement("visual_scene");
            Writer.WriteAttributeString("id", "Scene");
            Writer.WriteAttributeString("name", "Scene");
        }

        public void CreateVisualNodeSection()
        {
            BeginVisualNodeSection();
            Writer.WriteStartElement("node");
            Writer.WriteAttributeString("id", "Armature");
            Writer.WriteAttributeString("name", "Armature");
            Writer.WriteAttributeString("type", "NODE");

            Writer.WriteStartElement("matrix");
            Writer.WriteAttributeString("sid", "transform");
            Writer.WriteString("1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1");
            Writer.WriteEndElement();

            foreach (var joint in Joints)
                if (joint.ParentIndex == -1)
                    RecursivlyWriteJoints(joint);

            Writer.WriteEndElement();

            // write geometry nodes

            foreach (var m in MeshSkinIdList)
            {
                Writer.WriteStartElement("node");
                Writer.WriteAttributeString("id", MeshIdList[m.Key]);
                Writer.WriteAttributeString("name", MeshIdList[m.Key]);
                Writer.WriteAttributeString("type", "NODE");

                if (m.Value.Equals(""))
                {
                    Writer.WriteStartElement("instance_geometry");
                    Writer.WriteAttributeString("url", $"#{m.Key}");
                    Writer.WriteAttributeString("name", MeshIdList[m.Key]);
                    Writer.WriteEndElement();
                }
                else
                {
                    Writer.WriteStartElement("instance_controller");
                    Writer.WriteAttributeString("url", $"#{m.Value}");
                    Writer.WriteStartElement("skeleton");
                    Writer.WriteString("#Armature_" + Joints[0].Name);
                    Writer.WriteEndElement();
                    if (MaterialIdList.ContainsKey(m.Key))
                    {
                        Writer.WriteStartElement("bind_material");
                        Writer.WriteStartElement("technique_common");
                        foreach (var mat in MaterialIdList[m.Key])
                        {
                            Writer.WriteStartElement("instance_material");
                            Writer.WriteAttributeString("symbol", mat);
                            Writer.WriteAttributeString("target", "#" + mat);
                            Writer.WriteEndElement();
                        }
                        Writer.WriteEndElement();
                        Writer.WriteEndElement();
                    }
                    Writer.WriteEndElement();
                }


                Writer.WriteEndElement();
            }

            EndVisualNodeSection();
        }

        public void WriteNode(string Name)
        {
            Writer.WriteStartElement("node");
            Writer.WriteAttributeString("id", Name);
            Writer.WriteAttributeString("name", Name);
            Writer.WriteAttributeString("type", "NODE");
            Writer.WriteEndElement();
        }

        public void EndVisualNodeSection()
        {
            Writer.WriteEndElement();
            Writer.WriteEndElement();

            Writer.WriteStartElement("scene");
            Writer.WriteStartElement("instance_visual_scene");
            Writer.WriteAttributeString("url", "#Scene");
            Writer.WriteEndElement();
            Writer.WriteEndElement();
        }

        public string GetUniqueID(string id)
        {
            if (AttributeIdList.ContainsKey(id))
            {
                AttributeIdList[id]++;
                return $"{id}_{AttributeIdList[id]}";//
            }
            else
            {
                AttributeIdList.Add(id, 0);
                return id;
            }
        }

        public void Dispose()
        {
            Writer?.WriteEndElement();
            Writer?.WriteEndDocument();
            Writer?.Close();
        }


        private class ValueContainer<T> : IEquatable<ValueContainer<T>> where T : struct
        {
            public T[] Values { get; }
            private readonly int hashCode;

            public ValueContainer(T[] values)
            {
                Values = values;
                hashCode = ((System.Collections.IStructuralEquatable)Values).GetHashCode(EqualityComparer<T>.Default);
            }

            public override bool Equals(object obj)
            {
                return Equals(obj as ValueContainer<T>);
            }

            public bool Equals(ValueContainer<T> other)
            {
                // Compare precalculated hash first for performance reasons.
                // The entire sequence needs to be compared to resolve collisions.
                return other != null && hashCode == other.hashCode && Enumerable.SequenceEqual(Values, other.Values);
            }

            public override int GetHashCode()
            {
                return hashCode;
            }
        }
    }

    public class Material
    {
        public string Name { get; set; }

        public List<TextureMap> Textures = new List<TextureMap>();
    }

    public class Geometry
    {
        public string Name { get; set; }

        public Material Material { get; set; }

        public List<float> Position = new List<float>();
        public List<float> Normal = new List<float>();
        public List<float> UV0 = new List<float>();
        public List<float> UV1 = new List<float>();
        public List<float> UV2 = new List<float>();
        public List<float> UV3 = new List<float>();
        public List<float> Color = new List<float>();
        public List<int[]> BoneIndices = new List<int[]>();
        public List<float[]> BoneWeights = new List<float[]>();

        public bool HasNormals = false;
        public bool HasUV0 = false;
        public bool HasUV1 = false;
        public bool HasUV2 = false;
        public bool HasUV3 = false;
        public bool HasColors = false;
        public bool HasBoneIndices = false;
        public bool HasWeights = false;
    }

    public enum PhongTextureType
    {
        diffuse,
        specular,
        emission,
        bump,
    }

    public enum TEXCOORD
    {
        CHANNEL0,
        CHANNEL1,
        CHANNEL2,
        CHANNEL3,
    }

    public enum SemanticType
    {
        None,
        POSITION,
        VERTEX,
        NORMAL,
        TEXCOORD,
        COLOR,
        WEIGHT,
        JOINT,
        INV_BIND_MATRIX,
        TEXTANGENT,
        TEXBINORMAL
    }

    public enum ProgramPreset
    {
        NONE,
        MAX,
        MAYA,
        BLENDER,
    }

    public enum SamplerWrapMode
    {
        NONE,
        WRAP,
        MIRROR,
        CLAMP,
        BORDER,
    }

    public class TextureMap
    {
        public TEXCOORD TextureChannel { get; set; }
        public string Name { get; set; }
        public PhongTextureType Type { get; set; }

        public SamplerWrapMode WrapModeS = SamplerWrapMode.WRAP;
        public SamplerWrapMode WrapModeT = SamplerWrapMode.WRAP;

        public TextureMap()
        {
            TextureChannel = TEXCOORD.CHANNEL0;
            Type = PhongTextureType.diffuse;
            Name = "";
        }
    }

    public class TriangleList
    {
        public string Material = "";
        public List<uint> Indices = new List<uint>();
    }

    public class Joint
    {
        public string Name;
        public int ParentIndex = -1;
        public float[] Transform;
        public float[] BindPose;
    }
}