Began adding support for output of loaded BMD data to other model formats using AssImp.

Author: Sage-of-Mirrors

SuperBMD/source/BMD/INF1.cs

@@ -111,6 +111,11 @@ private void GetNodesRecursive(Rigging.Bone bone, List<Rigging.Bone> skeleton, S
             }
         }
 
+        public void FillScene(Scene scene, List<Rigging.Bone> flatSkeleton)
+        {
+
+        }
+
         public void Write(EndianBinaryWriter writer, int packetCount, int vertexCount)
         {
             long start = writer.BaseStream.Position;

SuperBMD/source/BMD/MAT3.cs

@@ -659,6 +659,38 @@ private void FillMaterialDataBlocks()
             }
         }
 
+        public void FillScene(Assimp.Scene scene, TEX1 textures, string fileDir)
+        {
+            foreach (Material mat in m_Materials)
+            {
+                Assimp.Material assMat = new Assimp.Material();
+
+                if (mat.TextureIndices[0] != -1)
+                {
+                    int texIndex = mat.TextureIndices[0];
+                    string texPath = textures[texIndex].SaveImageToDisk(fileDir);
+
+                    Assimp.TextureSlot tex = new Assimp.TextureSlot(texPath, Assimp.TextureType.Diffuse, 0,
+                        Assimp.TextureMapping.FromUV, 0, 1.0f, Assimp.TextureOperation.Add,
+                        textures[texIndex].WrapS.ToAssImpWrapMode(), textures[texIndex].WrapT.ToAssImpWrapMode(), 0);
+
+                    assMat.AddMaterialTexture(ref tex);
+                }
+
+                if (mat.MaterialColors[0] != null)
+                {
+                    assMat.ColorDiffuse = mat.MaterialColors[0].Value.ToColor4D();
+                }
+
+                if (mat.AmbientColors[0] != null)
+                {
+                    assMat.ColorAmbient = mat.AmbientColors[0].Value.ToColor4D();
+                }
+
+                scene.Materials.Add(assMat);
+            }
+        }
+
         public void Write(EndianBinaryWriter writer)
         {
             long start = writer.BaseStream.Position;

SuperBMD/source/BMD/SHP1.cs

@@ -164,91 +164,6 @@ public static SHP1 Create(Scene scene, Dictionary<string, int> boneNames, Vertex
             return shp1;
         }
 
-        private void SetMatrixIndices(Vertex vert, EVP1 envelopes, DRW1 partialWeight, List<int> matrixIndices)
-        {
-            vert.SetAttributeIndex(GXVertexAttribute.PositionMatrixIdx, (uint)matrixIndices.Count);
-            matrixIndices.Add(partialWeight.WeightTypeCheck.Count);
-
-            if (vert.VertexWeight.WeightCount > 1)
-            {
-                partialWeight.WeightTypeCheck.Add(true);
-                partialWeight.Indices.Add(envelopes.Weights.Count);
-                envelopes.Weights.Add(vert.VertexWeight);
-            }
-            else
-            {
-                partialWeight.WeightTypeCheck.Add(false);
-                partialWeight.Indices.Add(vert.VertexWeight.BoneIndices[0]);
-            }
-        }
-
-        private void SetVertexIndices(Mesh mesh, Vertex vert, VertexData vertData, ShapeVertexDescriptor descriptor, int vertIndex)
-        {
-            if (descriptor.CheckAttribute(GXVertexAttribute.Position))
-            {
-                Vector3D posVec = mesh.Vertices[vertIndex];
-                uint posIndex = (uint)vertData.Positions.IndexOf(posVec.ToOpenTKVector3());
-                vert.SetAttributeIndex(GXVertexAttribute.Position, posIndex);
-            }
-            if (descriptor.CheckAttribute(GXVertexAttribute.Normal))
-            {
-                Vector3D normVec = mesh.Normals[vertIndex];
-                uint normIndex = (uint)vertData.Normals.IndexOf(normVec.ToOpenTKVector3());
-                vert.SetAttributeIndex(GXVertexAttribute.Normal, normIndex);
-            }
-
-            for (int color = 0; color < 2; color++)
-            {
-                if (descriptor.CheckAttribute(GXVertexAttribute.Color0 + color))
-                {
-                    Color4D assimpColor = mesh.VertexColorChannels[color][vertIndex];
-                    List<Color> colorData = (List<Color>)vertData.GetAttributeData(GXVertexAttribute.Color0 + color);
-                    uint colIndex = (uint)colorData.IndexOf(assimpColor.ToSuperBMDColorRGBA());
-                    vert.SetAttributeIndex(GXVertexAttribute.Color0 + color, colIndex);
-                }
-            }
-
-            for (int tex = 0; tex < 8; tex++)
-            {
-                if (descriptor.CheckAttribute(GXVertexAttribute.Tex0 + tex))
-                {
-                    Vector3D texVec = mesh.TextureCoordinateChannels[tex][vertIndex];
-                    List<OpenTK.Vector2> texData = (List<OpenTK.Vector2>)vertData.GetAttributeData(GXVertexAttribute.Tex0 + tex);
-                    uint texIndex = (uint)texData.IndexOf(texVec.ToOpenTKVector2());
-                    vert.SetAttributeIndex(GXVertexAttribute.Tex0 + tex, texIndex);
-                }
-            }
-        }
-
-        public void DistributeWeights(EVP1 envelopes, DRW1 partialWeights)
-        {
-            foreach (Shape shape in Shapes)
-            {
-                foreach (Packet pack in shape.Packets)
-                {
-                    foreach (Primitive prim in pack.Primitives)
-                    {
-                        foreach (Vertex vert in prim.Vertices)
-                        {
-                            uint drw1Index = vert.GetAttributeIndex(GXVertexAttribute.PositionMatrixIdx);
-
-                            if (partialWeights.WeightTypeCheck[(int)drw1Index])
-                            {
-                                vert.SetWeight(envelopes.Weights[partialWeights.Indices[(int)drw1Index]]);
-                            }
-                            else
-                            {
-                                Rigging.Weight newWeight = new Rigging.Weight();
-                                newWeight.AddWeight(1.0f, partialWeights.Indices[(int)drw1Index]);
-
-                                vert.SetWeight(newWeight);
-                            }
-                        }
-                    }
-                }
-            }
-        }
-
         public void Write(EndianBinaryWriter writer)
         {
             List<Tuple<ShapeVertexDescriptor, int>> descriptorOffsets; // Contains the offsets for each unique vertex descriptor

SuperBMD/source/Materials/BinaryTextureImage.cs

@@ -236,7 +236,7 @@ public void Load(Assimp.TextureSlot texture, string modelDirectory)
             texData.Dispose();
         }
 
-        public void SaveImageToDisk(string outputFile)
+        public string SaveImageToDisk(string outputFile)
         {
             string fileName = Path.Combine(outputFile, $"{ Name }.bmp");
 
@@ -246,6 +246,8 @@ public void SaveImageToDisk(string outputFile)
                 Directory.CreateDirectory(Path.GetDirectoryName(fileName));
                 bmp.Save(fileName);
             }
+
+            return fileName;
         }
 
         public Bitmap CreateBitmap()

SuperBMD/source/Model.cs

@@ -114,7 +114,7 @@ public Model(Scene scene, string modelDirectory)
             vertexCount = VertexData.Attributes.Positions.Count;
         }
 
-        public void Export(string fileName)
+        public void ExportBMD(string fileName)
         {
             using (FileStream stream = new FileStream(fileName, FileMode.Create, FileAccess.Write))
             {
@@ -141,5 +141,27 @@ public void Export(string fileName)
                 writer.Write((int)writer.BaseStream.Length);
             }
         }
+
+        public void ExportAssImp(string fileName, string modelType)
+        {
+            string outDir = Path.GetDirectoryName(fileName);
+
+            Scene outScene = new Scene();
+            outScene.RootNode = new Node("RootNode");
+            Assimp.Node geomNode = new Node(Path.GetFileNameWithoutExtension(fileName), outScene.RootNode);
+
+            for (int i = 0; i < Shapes.Shapes.Count; i++)
+            {
+                geomNode.MeshIndices.Add(i);
+            }
+
+            outScene.RootNode.Children.Add(geomNode);
+
+            Materials.FillScene(outScene, Textures, outDir);
+
+
+            AssimpContext cont = new AssimpContext();
+            cont.ExportFile(outScene, fileName, modelType, PostProcessSteps.ValidateDataStructure);
+        }
     }
 }

SuperBMD/source/Util/Color.cs

@@ -165,6 +165,11 @@ public string ToHexString()
             return sb.ToString();
         }
 
+        public Assimp.Color4D ToColor4D()
+        {
+            return new Assimp.Color4D(R, G, B, A);
+        }
+
         public static Color Lerp(Color a, Color b, float t)
         {
             //t = MathE.ClampNormalized(t);

SuperBMD/source/Util/WrapModeUtility.cs

@@ -26,5 +26,20 @@ public static BinaryTextureImage.WrapModes ToGXWrapMode(this Assimp.TextureWrapM
                     throw new ArgumentException("mode");
             }
         }
+
+        public static Assimp.TextureWrapMode ToAssImpWrapMode(this BinaryTextureImage.WrapModes mode)
+        {
+            switch (mode)
+            {
+                case BinaryTextureImage.WrapModes.ClampToEdge:
+                    return TextureWrapMode.Clamp;
+                case BinaryTextureImage.WrapModes.MirroredRepeat:
+                    return TextureWrapMode.Mirror;
+                case BinaryTextureImage.WrapModes.Repeat:
+                    return TextureWrapMode.Wrap;
+                default:
+                    throw new ArgumentException("mode");
+            }
+        }
     }
 }

SuperBMD_UnitTest/Program.cs

@@ -12,7 +12,8 @@ class Program
         static void Main(string[] args)
         {
             Model mod = Model.Load(args[0]);
-            mod.Export(args[0] + ".bmd");
+            //mod.ExportBMD(args[0] + ".bmd");
+            mod.ExportAssImp(args[2], args[1]);
         }
     }
 }