public void SaveFromModel(List <STGenericObject> Meshes, List <STGenericMaterial> Materials, string FileName, List <STGenericTexture> Textures, STSkeleton skeleton = null, List <int> NodeArray = null)
{
ExtractedTextures.Clear();
Scene scene = new Scene();
scene.RootNode = new Node("RootNode");
progressBar = new STProgressBar();
progressBar.Task = "Exporting Skeleton...";
progressBar.Value = 0;
progressBar.StartPosition = FormStartPosition.CenterScreen;
progressBar.Show();
progressBar.Refresh();
SaveSkeleton(skeleton, scene.RootNode);
SaveMaterials(scene, Materials, FileName, Textures);
progressBar.Task = "Exporting Meshes...";
progressBar.Value = 50;
SaveMeshes(scene, Meshes, skeleton, FileName, NodeArray);
progressBar.Task = "Saving File...";
progressBar.Value = 80;
SaveScene(FileName, scene, Meshes);
progressBar.Value = 100;
progressBar.Close();
progressBar.Dispose();
}
private void SaveMaterials(Scene scene, List <STGenericMaterial> Materials, string FileName, List <STGenericTexture> Textures)
{
string TextureExtension = ".png";
string TexturePath = System.IO.Path.GetDirectoryName(FileName);
for (int i = 0; i < Textures.Count; i++)
{
string path = System.IO.Path.Combine(TexturePath, Textures[i].Text + TextureExtension);
if (!ExtractedTextures.Contains(path))
{
ExtractedTextures.Add(path);
progressBar.Task = $"Exporting Texture {Textures[i].Text}";
progressBar.Value = ((i * 100) / Textures.Count);
progressBar.Refresh();
var bitmap = Textures[i].GetBitmap();
bitmap.Save(path);
bitmap.Dispose();
GC.Collect();
}
}
if (Materials.Count == 0)
{
Material material = new Material();
material.Name = "New Material";
scene.Materials.Add(material);
return;
}
foreach (var mat in Materials)
{
var genericMat = (STGenericMaterial)mat;
Material material = new Material();
material.Name = genericMat.Text;
foreach (var tex in genericMat.TextureMaps)
{
int index = Textures.FindIndex(r => r.Text.Equals(tex.Name));
string path = System.IO.Path.Combine(TexturePath, tex.Name + TextureExtension);
if (!File.Exists(path))
{
continue;
}
TextureSlot slot2 = new TextureSlot(path, ConvertToAssimpTextureType(tex.Type), 0, TextureMapping.FromUV,
0, 1.0f, Assimp.TextureOperation.Add, ConvertToAssimpWrapType(tex.WrapModeS), ConvertToAssimpWrapType(tex.WrapModeT), 0);
material.AddMaterialTexture(ref slot2);
}
scene.Materials.Add(material);
}
}
private void RepackAction(object sender, EventArgs args)
{
FolderSelectDialog dlg = new FolderSelectDialog();
if (dlg.ShowDialog() == DialogResult.OK)
{
string FolderPath = dlg.SelectedPath;
STProgressBar progressBar = new STProgressBar();
progressBar.Task = "Reading Directory...";
progressBar.Value = 0;
progressBar.StartPosition = FormStartPosition.CenterScreen;
progressBar.Show();
progressBar.Refresh();
var ProccessedFiles = TreeHelper.ReadFiles(FolderPath);
progressBar.Task = "Repacking Files...";
progressBar.Refresh();
ArchiveFile.ClearFiles();
for (int i = 0; i < ProccessedFiles.Count; i++)
{
progressBar.Value = (i * 100) / ProccessedFiles.Count;
progressBar.Task = $"Packing {ProccessedFiles[i].Item1}";
progressBar.Refresh();
ArchiveFile.AddFile(new ArchiveFileInfo()
{
FileName = ProccessedFiles[i].Item1,
FileData = File.ReadAllBytes(ProccessedFiles[i].Item2),
});
}
progressBar.Close();
progressBar.Dispose();
ProccessedFiles.Clear();
GC.Collect();
FillTreeNodes();
}
}
//jakeroo123's animation adding code.
public static void ExportAnimation(string FileName, Toolbox.Library.Animations.Animation anim = null, STSkeleton skeleton = null, List <int> NodeArray = null)
{
//Always use the same settings
ExportSettings settings = new ExportSettings();
settings.SuppressConfirmDialog = true;
settings.UseMatrixTransform = false;
List <string> failedTextureExport = new List <string>();
STProgressBar progressBar = new STProgressBar();
progressBar.Task = "Exporting Animation \"" + anim.Text + "\"...";
progressBar.Value = 0;
progressBar.StartPosition = System.Windows.Forms.FormStartPosition.CenterScreen;
progressBar.Show();
progressBar.Refresh();
/* I don't see any way to use this
* if (settings.UseOldExporter)
* {
* AssimpSaver saver = new AssimpSaver();
* STGenericModel model = new STGenericModel();
* model.Objects = Meshes;
* model.Materials = Materials;
* saver.SaveFromModel(model, FileName, Textures, skeleton, NodeArray);
* return;
* }
*/
string TexturePath = System.IO.Path.GetDirectoryName(FileName);
using (ColladaWriter writer = new ColladaWriter(FileName, settings))
{
writer.WriteAsset();
//This is where the magic happens - Iterate through bones, and add animations.
if (anim != null && skeleton != null)
{
writer.StartLibraryAnimations();
float[] frames = new float[anim.FrameCount + 1];
for (int i = 0; i <= anim.FrameCount; i++)
{
frames[i] = i / 30f; //DAE appears to save frame times - This is saving the animation as 30 FPS, regardless of anything else.
}
string[] interp = new string[anim.FrameCount + 1];
for (int i = 0; i <= anim.FrameCount; i++)
{
interp[i] = "LINEAR"; //Just use linear.
}
foreach (STBone bone in skeleton.bones)
{
skeleton.reset();
string aid = anim.Text + "_" + bone.Text; //The prefix for animation
string bid = "Armature_" + bone.Text;
Vector3 defaultRotation = bone.EulerRotation;
//Three arrays, for
float[] rotateX = new float[anim.FrameCount + 1];
float[] rotateY = new float[anim.FrameCount + 1];
float[] rotateZ = new float[anim.FrameCount + 1];
Vector3[] scale = new Vector3[anim.FrameCount + 1];
Vector3[] translate = new Vector3[anim.FrameCount + 1];
anim.SetFrame(0);
for (int i = 0; i <= anim.FrameCount; i++)
{
anim.NextFrame(skeleton, false, true, bone.Text);
//Rotation
Vector3 eul = STMath.ToEulerAngles(bone.rot);
rotateX[i] = eul.X * Rad2Deg; //- defaultRotation.X;
rotateY[i] = eul.Y * Rad2Deg; //- defaultRotation.Y;
rotateZ[i] = eul.Z * Rad2Deg; //- defaultRotation.Z;
//Scaling and translation
scale[i] = bone.GetScale();
translate[i] = bone.GetPosition();
}
writer.WriteAnimationVector(aid + "_translate", frames, interp, translate, bid + "/location");
writer.WriteAnimationAngle(aid + "_rotateX", frames, interp, rotateX, bid + "/rotationX.ANGLE");
writer.WriteAnimationAngle(aid + "_rotateY", frames, interp, rotateY, bid + "/rotationY.ANGLE");
writer.WriteAnimationAngle(aid + "_rotateZ", frames, interp, rotateZ, bid + "/rotationZ.ANGLE");
writer.WriteAnimationVector(aid + "_scale", frames, interp, scale, bid + "/scale");
}
writer.EndLibraryAnimations();
}
skeleton.reset();
//Don't bother exporting any textures - It's not really necessary, at this point.
writer.WriteLibraryImages();
if (skeleton != null)
{
//Don't bother searching for rigging, because the mesh will simply be empty.
foreach (var bone in skeleton.bones)
{
//Set the inverse matrix
var inverse = skeleton.GetBoneTransform(bone).Inverted();
var transform = bone.GetTransform();
float[] Transform = new float[] {
transform.M11, transform.M21, transform.M31, transform.M41,
transform.M12, transform.M22, transform.M32, transform.M42,
transform.M13, transform.M23, transform.M33, transform.M43,
transform.M14, transform.M24, transform.M34, transform.M44
};
float[] InvTransform = new float[] {
inverse.M11, inverse.M21, inverse.M31, inverse.M41,
inverse.M12, inverse.M22, inverse.M32, inverse.M42,
inverse.M13, inverse.M23, inverse.M33, inverse.M43,
inverse.M14, inverse.M24, inverse.M34, inverse.M44
};
writer.AddJoint(bone.Text, bone.parentIndex == -1 ? "" :
skeleton.bones[bone.parentIndex].Text, Transform, InvTransform,
new float[3] {
bone.Position.X, bone.Position.Y, bone.Position.Z
},
new float[3] {
bone.EulerRotation.X *Rad2Deg, bone.EulerRotation.Y *Rad2Deg, bone.EulerRotation.Z *Rad2Deg
},
new float[3] {
bone.Scale.X, bone.Scale.Y, bone.Scale.Z
});
}
}
writer.StartLibraryGeometries();
//Try to not write meshes? Let's hope this works.
writer.EndGeometrySection();
}
progressBar?.Close();
}
public static void Export(string FileName, ExportSettings settings,
List <STGenericObject> Meshes, List <STGenericMaterial> Materials,
List <STGenericTexture> Textures, STSkeleton skeleton = null, List <int> NodeArray = null)
{
if (Materials == null)
{
Materials = new List <STGenericMaterial>();
}
if (skeleton != null && skeleton.BoneIndices != null)
{
NodeArray = skeleton.BoneIndices.ToList();
}
List <string> failedTextureExport = new List <string>();
STProgressBar progressBar = new STProgressBar();
progressBar.Task = "Exporting Model...";
progressBar.Value = 0;
progressBar.StartPosition = System.Windows.Forms.FormStartPosition.CenterScreen;
progressBar.Show();
progressBar.Refresh();
if (settings.UseOldExporter)
{
AssimpSaver saver = new AssimpSaver();
STGenericModel model = new STGenericModel();
model.Objects = Meshes;
model.Materials = Materials;
saver.SaveFromModel(model, FileName, Textures, skeleton, NodeArray);
return;
}
string TexturePath = System.IO.Path.GetDirectoryName(FileName);
Dictionary <string, STGenericMaterial> MaterialRemapper = new Dictionary <string, STGenericMaterial>();
using (ColladaWriter writer = new ColladaWriter(FileName, settings))
{
writer.WriteAsset();
if (Materials.Count > 0)
{
List <string> textureNames = new List <string>();
for (int i = 0; i < Textures?.Count; i++)
{
if (!textureNames.Contains(Textures[i].Text))
{
textureNames.Add(Textures[i].Text);
}
if (settings.ExportTextures)
{
progressBar.Task = $"Exporting Texture {Textures[i].Text}";
progressBar.Value = ((i * 100) / Textures.Count);
progressBar.Refresh();
try
{
var bitmap = Textures[i].GetBitmap();
if (bitmap != null)
{
if (settings.UseTextureChannelComponents)
{
bitmap = Textures[i].GetComponentBitmap(bitmap);
}
string textureName = Textures[i].Text;
if (textureName.RemoveIllegaleFileNameCharacters() != textureName)
{
string properName = textureName.RemoveIllegaleFileNameCharacters();
for (int m = 0; m < Materials?.Count; m++)
{
foreach (var tex in Materials[m].TextureMaps)
{
if (tex.Name == textureName)
{
tex.Name = properName;
}
}
}
textureName = properName;
}
bitmap.Save($"{TexturePath}/{textureName}.png");
bitmap.Dispose();
GC.Collect();
}
}
catch (Exception ex) {
failedTextureExport.Add(Textures[i].Text);
}
}
}
List <Material> materials = new List <Material>();
foreach (var mat in Materials)
{
Material material = new Material();
material.Name = mat.Text;
if (!MaterialRemapper.ContainsKey(mat.Text))
{
MaterialRemapper.Add(mat.Text, mat);
}
else
{
string name = Utils.RenameDuplicateString(MaterialRemapper.Keys.ToList(), mat.Text);
MaterialRemapper.Add(name, mat);
material.Name = name;
}
materials.Add(material);
foreach (var tex in mat.TextureMaps)
{
TextureMap texMap = new TextureMap();
texMap.Name = tex.Name;
if (tex.Type == STGenericMatTexture.TextureType.Diffuse)
{
texMap.Type = PhongTextureType.diffuse;
}
else if (tex.Type == STGenericMatTexture.TextureType.Normal)
{
texMap.Type = PhongTextureType.bump;
}
else if (tex.Type == STGenericMatTexture.TextureType.Specular)
{
texMap.Type = PhongTextureType.specular;
}
else if (tex.Type == STGenericMatTexture.TextureType.Emission)
{
texMap.Type = PhongTextureType.emission;
}
else
{
continue; //Skip adding unknown types
}
if (tex.WrapModeS == STTextureWrapMode.Repeat)
{
texMap.WrapModeS = SamplerWrapMode.WRAP;
}
else if (tex.WrapModeS == STTextureWrapMode.Mirror)
{
texMap.WrapModeS = SamplerWrapMode.MIRROR;
}
else if (tex.WrapModeS == STTextureWrapMode.Clamp)
{
texMap.WrapModeS = SamplerWrapMode.CLAMP;
}
if (tex.WrapModeT == STTextureWrapMode.Repeat)
{
texMap.WrapModeT = SamplerWrapMode.WRAP;
}
else if (tex.WrapModeT == STTextureWrapMode.Mirror)
{
texMap.WrapModeT = SamplerWrapMode.MIRROR;
}
else if (tex.WrapModeT == STTextureWrapMode.Clamp)
{
texMap.WrapModeT = SamplerWrapMode.CLAMP;
}
//If no textures are saved, still keep images references
//So the user can still dump textures after
if (Textures?.Count == 0 && !textureNames.Contains(texMap.Name))
{
textureNames.Add($"{texMap.Name}");
}
material.Textures.Add(texMap);
}
}
writer.WriteLibraryImages(textureNames.ToArray());
writer.WriteLibraryMaterials(materials);
writer.WriteLibraryEffects(materials);
}
else
{
writer.WriteLibraryImages();
}
if (skeleton != null)
{
//Search for bones with rigging first
List <string> riggedBones = new List <string>();
if (settings.OnlyExportRiggedBones)
{
for (int i = 0; i < Meshes.Count; i++)
{
for (int v = 0; v < Meshes[i].vertices.Count; v++)
{
var vertex = Meshes[i].vertices[v];
for (int j = 0; j < vertex.boneIds.Count; j++)
{
int id = -1;
if (NodeArray != null && NodeArray.Count > vertex.boneIds[j])
{
id = NodeArray[vertex.boneIds[j]];
}
else
{
id = vertex.boneIds[j];
}
if (id < skeleton.bones.Count && id != -1)
{
riggedBones.Add(skeleton.bones[id].Text);
}
}
}
}
}
foreach (var bone in skeleton.bones)
{
if (settings.OnlyExportRiggedBones && !riggedBones.Contains(bone.Text))
{
Console.WriteLine("Skipping " + bone.Text);
continue;
}
//Set the inverse matrix
var inverse = skeleton.GetBoneTransform(bone).Inverted();
var transform = bone.GetTransform();
float[] Transform = new float[] {
transform.M11, transform.M21, transform.M31, transform.M41,
transform.M12, transform.M22, transform.M32, transform.M42,
transform.M13, transform.M23, transform.M33, transform.M43,
transform.M14, transform.M24, transform.M34, transform.M44
};
float[] InvTransform = new float[] {
inverse.M11, inverse.M21, inverse.M31, inverse.M41,
inverse.M12, inverse.M22, inverse.M32, inverse.M42,
inverse.M13, inverse.M23, inverse.M33, inverse.M43,
inverse.M14, inverse.M24, inverse.M34, inverse.M44
};
writer.AddJoint(bone.Text, bone.parentIndex == -1 ? "" :
skeleton.bones[bone.parentIndex].Text, Transform, InvTransform,
new float[3] {
bone.Position.X, bone.Position.Y, bone.Position.Z
},
new float[3] {
bone.EulerRotation.X *Rad2Deg, bone.EulerRotation.Y *Rad2Deg, bone.EulerRotation.Z *Rad2Deg
},
new float[3] {
bone.Scale.X, bone.Scale.Y, bone.Scale.Z
});
}
}
int meshIndex = 0;
writer.StartLibraryGeometries();
foreach (var mesh in Meshes)
{
progressBar.Task = $"Exporting Mesh {mesh.Text}";
progressBar.Value = ((meshIndex++ *100) / Meshes.Count);
progressBar.Refresh();
int[] IndexTable = null;
if (NodeArray != null)
{
IndexTable = NodeArray.ToArray();
}
writer.StartGeometry(mesh.Text);
if (mesh.MaterialIndex != -1 && Materials.Count > mesh.MaterialIndex)
{
writer.CurrentMaterial = Materials[mesh.MaterialIndex].Text;
Console.WriteLine($"MaterialIndex {mesh.MaterialIndex } {Materials[mesh.MaterialIndex].Text}");
}
if (settings.TransformColorUVs)
{
List <Vertex> transformedVertices = new List <Vertex>();
foreach (var poly in mesh.PolygonGroups)
{
var mat = poly.Material;
if (mat == null)
{
continue;
}
var faces = poly.GetDisplayFace();
for (int v = 0; v < poly.displayFaceSize; v += 3)
{
if (faces.Count < v + 2)
{
break;
}
var diffuse = mat.TextureMaps.FirstOrDefault(x => x.Type == STGenericMatTexture.TextureType.Diffuse);
STTextureTransform transform = new STTextureTransform();
if (diffuse != null)
{
transform = diffuse.Transform;
}
var vertexA = mesh.vertices[faces[v]];
var vertexB = mesh.vertices[faces[v + 1]];
var vertexC = mesh.vertices[faces[v + 2]];
if (!transformedVertices.Contains(vertexA))
{
vertexA.uv0 = (vertexA.uv0 * transform.Scale) + transform.Translate;
transformedVertices.Add(vertexA);
}
if (!transformedVertices.Contains(vertexB))
{
vertexB.uv0 = (vertexB.uv0 * transform.Scale) + transform.Translate;
transformedVertices.Add(vertexB);
}
if (!transformedVertices.Contains(vertexC))
{
vertexC.uv0 = (vertexC.uv0 * transform.Scale) + transform.Translate;
transformedVertices.Add(vertexC);
}
}
}
}
// collect sources
List <float> Position = new List <float>();
List <float> Normal = new List <float>();
List <float> UV0 = new List <float>();
List <float> UV1 = new List <float>();
List <float> UV2 = new List <float>();
List <float> UV3 = new List <float>();
List <float> Color = new List <float>();
List <float> Color2 = new List <float>();
List <int[]> BoneIndices = new List <int[]>();
List <float[]> BoneWeights = new List <float[]>();
bool HasNormals = false;
bool HasColors = false;
bool HasColors2 = false;
bool HasUV0 = false;
bool HasUV1 = false;
bool HasUV2 = false;
bool HasBoneIds = false;
foreach (var vertex in mesh.vertices)
{
if (vertex.nrm != Vector3.Zero)
{
HasNormals = true;
}
if (vertex.col != Vector4.One && settings.UseVertexColors)
{
HasColors = true;
}
if (vertex.col2 != Vector4.One && settings.UseVertexColors)
{
HasColors2 = true;
}
if (vertex.uv0 != Vector2.Zero)
{
HasUV0 = true;
}
if (vertex.uv1 != Vector2.Zero)
{
HasUV1 = true;
}
if (vertex.uv2 != Vector2.Zero)
{
HasUV2 = true;
}
if (vertex.boneIds.Count > 0)
{
HasBoneIds = true;
}
Position.Add(vertex.pos.X); Position.Add(vertex.pos.Y); Position.Add(vertex.pos.Z);
Normal.Add(vertex.nrm.X); Normal.Add(vertex.nrm.Y); Normal.Add(vertex.nrm.Z);
if (settings.FlipTexCoordsVertical)
{
UV0.Add(vertex.uv0.X); UV0.Add(1 - vertex.uv0.Y);
UV1.Add(vertex.uv1.X); UV1.Add(1 - vertex.uv1.Y);
UV2.Add(vertex.uv2.X); UV2.Add(1 - vertex.uv2.Y);
}
else
{
UV0.Add(vertex.uv0.X); UV0.Add(vertex.uv0.Y);
UV1.Add(vertex.uv1.X); UV1.Add(vertex.uv1.Y);
UV2.Add(vertex.uv2.X); UV2.Add(vertex.uv2.Y);
}
Color.AddRange(new float[] { vertex.col.X, vertex.col.Y, vertex.col.Z, vertex.col.W });
Color2.AddRange(new float[] { vertex.col2.X, vertex.col2.Y, vertex.col2.Z, vertex.col2.W });
List <int> bIndices = new List <int>();
List <float> bWeights = new List <float>();
for (int b = 0; b < vertex.boneIds.Count; b++)
{
if (b > mesh.VertexSkinCount - 1)
{
continue;
}
//Skip 0 weights
if (vertex.boneWeights.Count > b)
{
if (vertex.boneWeights[b] == 0)
{
continue;
}
}
int index = -1;
if (IndexTable != null)
{
index = (int)IndexTable[vertex.boneIds[b]];
}
else
{
index = (int)vertex.boneIds[b];
}
//Only map for valid weights/indices
bool hasValidIndex = index != -1 && index < skeleton?.bones.Count;
bool hasValidWeight = vertex.boneWeights.Count > b;
if (hasValidIndex)
{
bIndices.Add(index);
}
if (hasValidWeight && hasValidIndex)
{
bWeights.Add(vertex.boneWeights[b]);
}
}
//Rigid bodies with no direct bone indices
if (bIndices.Count == 0 && mesh.BoneIndex != -1)
{
HasBoneIds = true;
bIndices.Add(mesh.BoneIndex);
bWeights.Add(1);
}
//Bone indices with no weights directly mapped
if (bWeights.Count == 0 && bIndices.Count > 0)
{
bWeights.Add(1.0f);
}
BoneIndices.Add(bIndices.ToArray());
BoneWeights.Add(bWeights.ToArray());
}
List <TriangleList> triangleLists = new List <TriangleList>();
if (mesh.lodMeshes.Count > 0)
{
TriangleList triangleList = new TriangleList();
triangleLists.Add(triangleList);
var lodMesh = mesh.lodMeshes[mesh.DisplayLODIndex];
List <int> faces = new List <int>();
if (lodMesh.PrimativeType == STPrimitiveType.TrangleStrips)
{
faces = STGenericObject.ConvertTriangleStripsToTriangles(lodMesh.faces);
}
else
{
faces = lodMesh.faces;
}
for (int i = 0; i < faces.Count; i++)
{
triangleList.Indices.Add((uint)faces[i]);
}
}
if (mesh.PolygonGroups.Count > 0)
{
foreach (var group in mesh.PolygonGroups)
{
TriangleList triangleList = new TriangleList();
triangleLists.Add(triangleList);
STGenericMaterial material = new STGenericMaterial();
if (group.MaterialIndex != -1 && Materials.Count > group.MaterialIndex)
{
material = Materials[group.MaterialIndex];
}
if (group.Material != null)
{
material = group.Material;
}
if (MaterialRemapper.Values.Any(x => x == material))
{
var key = MaterialRemapper.FirstOrDefault(x => x.Value == material).Key;
triangleList.Material = key;
}
else if (material.Text != string.Empty)
{
triangleList.Material = material.Text;
}
List <int> faces = new List <int>();
if (group.PrimativeType == STPrimitiveType.TrangleStrips)
{
faces = STGenericObject.ConvertTriangleStripsToTriangles(group.faces);
}
else
{
faces = group.faces;
}
for (int i = 0; i < faces.Count; i++)
{
triangleList.Indices.Add((uint)faces[i]);
}
}
}
// write sources
writer.WriteGeometrySource(mesh.Text, SemanticType.POSITION, Position.ToArray(), triangleLists.ToArray());
if (HasNormals)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.NORMAL, Normal.ToArray(), triangleLists.ToArray());
}
if (HasColors)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.COLOR, Color.ToArray(), triangleLists.ToArray(), 0);
}
if (HasColors2)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.COLOR, Color2.ToArray(), triangleLists.ToArray(), 1);
}
if (HasUV0)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.TEXCOORD, UV0.ToArray(), triangleLists.ToArray(), 0);
}
if (HasUV1)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.TEXCOORD, UV1.ToArray(), triangleLists.ToArray(), 1);
}
if (HasUV2)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.TEXCOORD, UV2.ToArray(), triangleLists.ToArray(), 2);
}
if (HasBoneIds)
{
writer.AttachGeometryController(BoneIndices, BoneWeights);
}
writer.EndGeometryMesh();
}
writer.EndGeometrySection();
}
progressBar?.Close();
if (!settings.SuppressConfirmDialog)
{
System.Windows.Forms.MessageBox.Show($"Exported {FileName} Successfuly!");
}
}
public static string[] ExtractAllFiles(string ParentPath, TreeNodeCollection Nodes, string overridePath = "")
{
List <string> filesExtracted = new List <string>();
if (overridePath == string.Empty)
{
FolderSelectDialog folderDialog = new FolderSelectDialog();
if (folderDialog.ShowDialog() != DialogResult.OK)
{
return(new string[0]);
}
overridePath = folderDialog.SelectedPath;
}
STProgressBar progressBar = new STProgressBar();
progressBar.Task = "Extracing Files...";
progressBar.Refresh();
progressBar.Value = 0;
progressBar.StartPosition = FormStartPosition.CenterScreen;
progressBar.Show();
var Collection = TreeViewExtensions.Collect(Nodes);
int Curfile = 0;
foreach (TreeNode file in Collection)
{
if (file is ArchiveFileWrapper)
{
string FilePath = ((ArchiveFileWrapper)file).ArchiveFileInfo.FileName;
string FolderPath = Path.GetDirectoryName(FilePath.RemoveIllegaleFolderNameCharacters());
string FolderPathDir = Path.Combine(overridePath, FolderPath);
if (!Directory.Exists(FolderPathDir))
{
Directory.CreateDirectory(FolderPathDir);
}
string FileName = file.Text.RemoveIllegaleFileNameCharacters();
FilePath = Path.Combine(FolderPath, FileName);
if (ParentPath != string.Empty)
{
FilePath = FilePath.Replace(ParentPath, string.Empty);
}
var path = Path.Combine(overridePath, FilePath);
progressBar.Task = $"Extracting File {file}";
progressBar.Value = (Curfile++ *100) / Collection.Count();
progressBar.Refresh();
CreateDirectoryIfExists($"{path}");
if (file is ArchiveFileWrapper)
{
filesExtracted.Add($"{path}");
File.WriteAllBytes($"{path}",
((ArchiveFileWrapper)file).ArchiveFileInfo.FileData);
}
}
}
progressBar.Value = 100;
progressBar.Refresh();
progressBar.Close();
return(filesExtracted.ToArray());
}
public static void Export(string FileName, ExportSettings settings,
List <STGenericObject> Meshes, List <STGenericMaterial> Materials,
List <STGenericTexture> Textures, STSkeleton skeleton = null, List <int> NodeArray = null)
{
if (Materials == null)
{
Materials = new List <STGenericMaterial>();
}
List <string> failedTextureExport = new List <string>();
STProgressBar progressBar = new STProgressBar();
progressBar.Task = "Exporting Model...";
progressBar.Value = 0;
progressBar.StartPosition = System.Windows.Forms.FormStartPosition.CenterScreen;
progressBar.Show();
progressBar.Refresh();
if (settings.UseOldExporter)
{
AssimpSaver saver = new AssimpSaver();
STGenericModel model = new STGenericModel();
model.Objects = Meshes;
model.Materials = Materials;
saver.SaveFromModel(model, FileName, Textures, skeleton, NodeArray);
return;
}
string TexturePath = System.IO.Path.GetDirectoryName(FileName);
using (ColladaWriter writer = new ColladaWriter(FileName, settings))
{
writer.WriteAsset();
if (Materials.Count > 0)
{
List <string> textureNames = new List <string>();
for (int i = 0; i < Textures?.Count; i++)
{
textureNames.Add(Textures[i].Text);
if (settings.ExportTextures)
{
progressBar.Task = $"Exporting Texture {Textures[i].Text}";
progressBar.Value = ((i * 100) / Textures.Count);
progressBar.Refresh();
try
{
var bitmap = Textures[i].GetBitmap();
if (bitmap != null)
{
string textureName = Textures[i].Text;
if (textureName.RemoveIllegaleFileNameCharacters() != textureName)
{
string properName = textureName.RemoveIllegaleFileNameCharacters();
for (int m = 0; m < Materials?.Count; m++)
{
foreach (var tex in Materials[m].TextureMaps)
{
if (tex.Name == textureName)
{
tex.Name = properName;
}
}
}
textureName = properName;
}
bitmap.Save($"{TexturePath}/{textureName}.png");
bitmap.Dispose();
GC.Collect();
}
}
catch (Exception ex) {
failedTextureExport.Add(Textures[i].Text);
}
}
}
List <Material> materials = new List <Material>();
foreach (var mat in Materials)
{
Material material = new Material();
material.Name = mat.Text;
materials.Add(material);
foreach (var tex in mat.TextureMaps)
{
TextureMap texMap = new TextureMap();
texMap.Name = tex.Name;
if (tex.Type == STGenericMatTexture.TextureType.Diffuse)
{
texMap.Type = PhongTextureType.diffuse;
}
else if (tex.Type == STGenericMatTexture.TextureType.Normal)
{
texMap.Type = PhongTextureType.bump;
}
else if (tex.Type == STGenericMatTexture.TextureType.Specular)
{
texMap.Type = PhongTextureType.specular;
}
else if (tex.Type == STGenericMatTexture.TextureType.Emission)
{
texMap.Type = PhongTextureType.emission;
}
else
{
continue; //Skip adding unknown types
}
if (tex.WrapModeS == STTextureWrapMode.Repeat)
{
texMap.WrapModeS = SamplerWrapMode.WRAP;
}
else if (tex.WrapModeS == STTextureWrapMode.Mirror)
{
texMap.WrapModeS = SamplerWrapMode.MIRROR;
}
else if (tex.WrapModeS == STTextureWrapMode.Clamp)
{
texMap.WrapModeS = SamplerWrapMode.CLAMP;
}
if (tex.WrapModeT == STTextureWrapMode.Repeat)
{
texMap.WrapModeT = SamplerWrapMode.WRAP;
}
else if (tex.WrapModeT == STTextureWrapMode.Mirror)
{
texMap.WrapModeT = SamplerWrapMode.MIRROR;
}
else if (tex.WrapModeT == STTextureWrapMode.Clamp)
{
texMap.WrapModeT = SamplerWrapMode.CLAMP;
}
//If no textures are saved, still keep images references
//So the user can still dump textures after
if (Textures?.Count == 0)
{
textureNames.Add($"{texMap.Name}");
}
material.Textures.Add(texMap);
}
}
writer.WriteLibraryImages(textureNames.ToArray());
writer.WriteLibraryMaterials(materials);
writer.WriteLibraryEffects(materials);
}
else
{
writer.WriteLibraryImages();
}
if (skeleton != null)
{
foreach (var bone in skeleton.bones)
{
//Set the inverse matrix
var inverse = skeleton.GetBoneTransform(bone).Inverted();
var transform = bone.GetTransform();
float[] Transform = new float[] {
transform.M11, transform.M21, transform.M31, transform.M41,
transform.M12, transform.M22, transform.M32, transform.M42,
transform.M13, transform.M23, transform.M33, transform.M43,
transform.M14, transform.M24, transform.M34, transform.M44
};
float[] InvTransform = new float[] {
inverse.M11, inverse.M21, inverse.M31, inverse.M41,
inverse.M12, inverse.M22, inverse.M32, inverse.M42,
inverse.M13, inverse.M23, inverse.M33, inverse.M43,
inverse.M14, inverse.M24, inverse.M34, inverse.M44
};
writer.AddJoint(bone.Text, bone.parentIndex == -1 ? "" :
skeleton.bones[bone.parentIndex].Text, Transform, InvTransform);
}
}
int meshIndex = 0;
writer.StartLibraryGeometries();
foreach (var mesh in Meshes)
{
progressBar.Task = $"Exporting Mesh {mesh.Text}";
progressBar.Value = ((meshIndex++ *100) / Meshes.Count);
progressBar.Refresh();
int[] IndexTable = null;
if (NodeArray != null)
{
IndexTable = NodeArray.ToArray();
}
writer.StartGeometry(mesh.Text);
if (mesh.MaterialIndex != -1 && Materials.Count > mesh.MaterialIndex)
{
writer.CurrentMaterial = Materials[mesh.MaterialIndex].Text;
}
// collect sources
List <float> Position = new List <float>();
List <float> Normal = new List <float>();
List <float> UV0 = new List <float>();
List <float> UV1 = new List <float>();
List <float> UV2 = new List <float>();
List <float> UV3 = new List <float>();
List <float> Color = new List <float>();
List <int[]> BoneIndices = new List <int[]>();
List <float[]> BoneWeights = new List <float[]>();
bool HasNormals = false;
bool HasColors = false;
bool HasUV0 = false;
bool HasUV1 = false;
bool HasUV2 = false;
bool HasBoneIds = false;
foreach (var vertex in mesh.vertices)
{
//Remove zero weights
if (settings.OptmizeZeroWeights)
{
float MaxWeight = 1;
for (int i = 0; i < 4; i++)
{
if (vertex.boneWeights.Count <= i)
{
continue;
}
if (vertex.boneIds.Count < i + 1)
{
vertex.boneWeights[i] = 0;
MaxWeight = 0;
}
else
{
float weight = vertex.boneWeights[i];
if (vertex.boneWeights.Count == i + 1)
{
weight = MaxWeight;
}
if (weight >= MaxWeight)
{
weight = MaxWeight;
MaxWeight = 0;
}
else
{
MaxWeight -= weight;
}
vertex.boneWeights[i] = weight;
}
}
}
if (vertex.nrm != Vector3.Zero)
{
HasNormals = true;
}
if (vertex.col != Vector4.One && settings.UseVertexColors)
{
HasColors = true;
}
if (vertex.uv0 != Vector2.Zero)
{
HasUV0 = true;
}
if (vertex.uv1 != Vector2.Zero)
{
HasUV1 = true;
}
if (vertex.uv2 != Vector2.Zero)
{
HasUV2 = true;
}
if (vertex.boneIds.Count > 0)
{
HasBoneIds = true;
}
Position.Add(vertex.pos.X); Position.Add(vertex.pos.Y); Position.Add(vertex.pos.Z);
Normal.Add(vertex.nrm.X); Normal.Add(vertex.nrm.Y); Normal.Add(vertex.nrm.Z);
if (settings.FlipTexCoordsVertical)
{
UV0.Add(vertex.uv0.X); UV0.Add(1 - vertex.uv0.Y);
UV1.Add(vertex.uv1.X); UV1.Add(1 - vertex.uv1.Y);
UV2.Add(vertex.uv2.X); UV2.Add(1 - vertex.uv2.Y);
}
else
{
UV0.Add(vertex.uv0.X); UV0.Add(vertex.uv0.Y);
UV1.Add(vertex.uv1.X); UV1.Add(vertex.uv1.Y);
UV2.Add(vertex.uv2.X); UV2.Add(vertex.uv2.Y);
}
Color.AddRange(new float[] { vertex.col.X, vertex.col.Y, vertex.col.Z, vertex.col.W });
List <int> bIndices = new List <int>();
List <float> bWeights = new List <float>();
for (int b = 0; b < vertex.boneIds.Count; b++)
{
if (b > mesh.VertexSkinCount - 1)
{
continue;
}
if (vertex.boneWeights.Count > b)
{
if (vertex.boneWeights[b] == 0)
{
continue;
}
}
int index = -1;
if (IndexTable != null)
{
index = (int)IndexTable[vertex.boneIds[b]];
}
else
{
index = (int)vertex.boneIds[b];
}
if (index != -1 && index < skeleton?.bones.Count)
{
bIndices.Add(index);
}
//Some models may only use indices (single bind, rigid skin)
if (vertex.boneWeights.Count > b)
{
bWeights.Add(vertex.boneWeights[b]);
}
else
{
bWeights.Add(1);
}
}
if (bIndices.Count == 0 && mesh.BoneIndex != -1)
{
HasBoneIds = true;
bIndices.Add(mesh.BoneIndex);
bWeights.Add(1);
}
BoneIndices.Add(bIndices.ToArray());
BoneWeights.Add(bWeights.ToArray());
}
List <TriangleList> triangleLists = new List <TriangleList>();
if (mesh.lodMeshes.Count > 0)
{
TriangleList triangleList = new TriangleList();
triangleLists.Add(triangleList);
var lodMesh = mesh.lodMeshes[mesh.DisplayLODIndex];
List <int> faces = new List <int>();
if (lodMesh.PrimativeType == STPrimitiveType.TrangleStrips)
{
faces = STGenericObject.ConvertTriangleStripsToTriangles(lodMesh.faces);
}
else
{
faces = lodMesh.faces;
}
for (int i = 0; i < faces.Count; i++)
{
triangleList.Indices.Add((uint)faces[i]);
}
}
if (mesh.PolygonGroups.Count > 0)
{
foreach (var group in mesh.PolygonGroups)
{
TriangleList triangleList = new TriangleList();
triangleLists.Add(triangleList);
if (group.MaterialIndex != -1 && Materials.Count > group.MaterialIndex)
{
triangleList.Material = Materials[group.MaterialIndex].Text;
}
List <int> faces = new List <int>();
if (group.PrimativeType == STPrimitiveType.TrangleStrips)
{
faces = STGenericObject.ConvertTriangleStripsToTriangles(group.faces);
}
else
{
faces = group.faces;
}
for (int i = 0; i < faces.Count; i++)
{
triangleList.Indices.Add((uint)faces[i]);
}
}
}
// write sources
writer.WriteGeometrySource(mesh.Text, SemanticType.POSITION, Position.ToArray(), triangleLists.ToArray());
if (HasNormals)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.NORMAL, Normal.ToArray(), triangleLists.ToArray());
}
if (HasColors)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.COLOR, Color.ToArray(), triangleLists.ToArray());
}
if (HasUV0)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.TEXCOORD, UV0.ToArray(), triangleLists.ToArray(), 0);
}
if (HasUV1)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.TEXCOORD, UV1.ToArray(), triangleLists.ToArray(), 1);
}
if (HasUV2)
{
writer.WriteGeometrySource(mesh.Text, SemanticType.TEXCOORD, UV2.ToArray(), triangleLists.ToArray(), 2);
}
if (HasBoneIds)
{
writer.AttachGeometryController(BoneIndices, BoneWeights);
}
writer.EndGeometryMesh();
}
writer.EndGeometrySection();
}
progressBar?.Close();
if (!settings.SuppressConfirmDialog)
{
System.Windows.Forms.MessageBox.Show($"Exported {FileName} Successfuly!");
}
}
public static string[] ExtractAllFiles(string ParentPath, TreeNodeCollection Nodes, string overridePath = "")
{
List <string> filesExtracted = new List <string>();
if (overridePath == string.Empty)
{
FolderSelectDialog folderDialog = new FolderSelectDialog();
if (folderDialog.ShowDialog() != DialogResult.OK)
{
return(new string[0]);
}
overridePath = folderDialog.SelectedPath;
}
STProgressBar progressBar = new STProgressBar();
progressBar.Task = "Extracing Files...";
progressBar.Refresh();
progressBar.Value = 0;
progressBar.StartPosition = FormStartPosition.CenterScreen;
progressBar.Show();
Thread Thread = new Thread((ThreadStart)(() =>
{
var Collection = TreeViewExtensions.Collect(Nodes).ToList();
Console.WriteLine($"Collection {Collection.Count}");
int Curfile = 0;
foreach (TreeNode node in Collection)
{
if (progressBar.IsDisposed || progressBar.Disposing)
{
break;
}
ArchiveFileInfo file = null;
if (node.Tag != null && node.Tag is ArchiveFileInfo)
{
file = (ArchiveFileInfo)node.Tag;
}
else if (node is ArchiveFileWrapper)
{
file = ((ArchiveFileWrapper)node).ArchiveFileInfo;
}
if (file != null)
{
string FilePath = file.FileName;
string FolderPath = Path.GetDirectoryName(FilePath.RemoveIllegaleFolderNameCharacters());
string FolderPathDir = Path.Combine(overridePath, FolderPath);
if (!Directory.Exists(FolderPathDir))
{
Directory.CreateDirectory(FolderPathDir);
}
string FileName = Path.GetFileName(file.FileName).RemoveIllegaleFileNameCharacters();
FilePath = Path.Combine(FolderPath, FileName);
if (ParentPath != string.Empty)
{
FilePath = FilePath.Replace(ParentPath, string.Empty);
}
var path = $"{overridePath}/{FilePath}";
if (progressBar.InvokeRequired)
{
progressBar.Invoke((MethodInvoker) delegate {
// Running on the UI thread
progressBar.Task = $"Extracting File {FileName}";
progressBar.Value = (Curfile++ *100) / Collection.Count;
progressBar.Refresh();
});
}
CreateDirectoryIfExists($"{path}");
filesExtracted.Add($"{path}");
if (file.FileFormat != null && file.FileFormat.CanSave)
{
file.SaveFileFormat();
}
if (file.FileDataStream != null)
{
file.FileDataStream.ExportToFile(path);
}
else
{
File.WriteAllBytes($"{path}", file.FileData);
}
}
}
if (progressBar.InvokeRequired)
{
progressBar.Invoke((MethodInvoker) delegate {
progressBar.Value = 100;
progressBar.Refresh();
progressBar.Close();
});
}
else
{
progressBar.Value = 100;
progressBar.Refresh();
progressBar.Close();
}
}));
Thread.Start();
return(filesExtracted.ToArray());
}