diff options
Diffstat (limited to 'src/main.cpp')
| -rw-r--r-- | src/main.cpp | 114 |
1 files changed, 114 insertions, 0 deletions
diff --git a/src/main.cpp b/src/main.cpp new file mode 100644 index 0000000..a679828 --- /dev/null +++ b/src/main.cpp @@ -0,0 +1,114 @@ +#include <assert.h> +#include <stdio.h> + +#include <assimp/Importer.hpp> +#include <assimp/postprocess.h> +#include <assimp/scene.h> +#include <assimp/DefaultLogger.hpp> +#include <assimp/LogStream.hpp> + +static void usage(); +static void file_write_error_check(FILE *file, const void *data, size_t size); +static void file_write_uint_error_check(FILE *file, unsigned int value); +static void process_node(const aiScene *scene, const aiNode *node, FILE *output_file); + +int main(int argc, char **argv) { + if(argc != 3) + usage(); + + const char *source_filepath = argv[1]; + const char *destination_filepath = argv[2]; + + Assimp::Logger::LogSeverity severity = Assimp::Logger::VERBOSE; + Assimp::DefaultLogger::create("", severity, aiDefaultLogStream_STDERR); + + Assimp::Importer importer; + const aiScene *scene = importer.ReadFile(source_filepath, aiProcessPreset_TargetRealtime_Quality | aiProcess_Triangulate | aiProcess_FlipUVs); + if(!scene || scene->mFlags & AI_SCENE_FLAGS_INCOMPLETE || !scene->mRootNode) { + fprintf(stderr, "Failed to load file: %s, reason: %s\n", source_filepath, importer.GetErrorString()); + exit(2); + } + + FILE *destination_file = fopen(destination_filepath, "wb"); + if(!destination_file) { + perror(destination_filepath); + exit(3); + } + file_write_uint_error_check(destination_file, 0x036144AF); // magic number + file_write_uint_error_check(destination_file, 0x00000001); // version + + process_node(scene, scene->mRootNode, destination_file); + fclose(destination_file); + Assimp::DefaultLogger::kill(); + return 0; +} + +void usage() { + fprintf(stderr, "usage: amalgine-model-converter <source> <destination>\n"); + exit(1); +} + +void file_write_error_check(FILE *file, const void *data, size_t size) { + if(fwrite(data, 1, size, file) != size) { + fprintf(stderr, "Error: failed to write %ld bytes to output file, error: %s\n", size, strerror(errno)); + exit(4); + } +} + +void file_write_uint_error_check(FILE *file, unsigned int value) { + const size_t size = sizeof(value); + if(fwrite(&value, 1, size, file) != size) { + fprintf(stderr, "Error: failed to write %ld bytes to output file, error: %s\n", size, strerror(errno)); + exit(4); + } +} + +static void process_mesh(const aiScene *scene, const aiMesh *mesh, FILE *output_file) { + file_write_uint_error_check(output_file, mesh->mNumVertices); + file_write_error_check(output_file, &mesh->mVertices, mesh->mNumVertices * sizeof(aiVector3D)); + file_write_error_check(output_file, &mesh->mNormals, mesh->mNumVertices * sizeof(aiVector3D)); + + file_write_uint_error_check(output_file, mesh->mNumFaces); + for(unsigned int i = 0; i < mesh->mNumFaces; ++i) { + // Should always be 3 since the file was loaded with aiProcess_Triangulate + file_write_error_check(output_file, mesh->mFaces[i].mIndices, 3 * sizeof(unsigned int)); + } + + file_write_uint_error_check(output_file, mesh->mTextureCoords[0] ? 1 : 0); + if(mesh->mTextureCoords[0]) + file_write_error_check(output_file, &mesh->mTextureCoords[0], mesh->mNumVertices * sizeof(aiVector3D)); + + file_write_uint_error_check(output_file, mesh->mMaterialIndex < scene->mNumMaterials); + if(mesh->mMaterialIndex < scene->mNumMaterials) { + const unsigned int num_texture_types = 3; + const aiTextureType texture_types[num_texture_types] = { + aiTextureType_DIFFUSE, + aiTextureType_SPECULAR, + aiTextureType_AMBIENT + }; + + const aiMaterial *material = scene->mMaterials[mesh->mMaterialIndex]; + for(unsigned int i = 0; i < num_texture_types; ++i) { + unsigned int num_textures_for_type = material->GetTextureCount(texture_types[i]); + file_write_uint_error_check(output_file, num_textures_for_type); + for(unsigned int j = 0; j < num_textures_for_type; ++j) { + aiString str; + if(material->GetTexture(texture_types[i], j, &str) == aiReturn_SUCCESS) + file_write_error_check(output_file, str.data, str.length + 1); // +1 to include null-terminate character + else + file_write_error_check(output_file, "\0", 1); + } + } + } +} + +void process_node(const aiScene *scene, const aiNode *node, FILE *output_file) { + for(unsigned int i = 0; i < node->mNumMeshes; ++i) { + const aiMesh *mesh = scene->mMeshes[node->mMeshes[i]]; + process_mesh(scene, mesh, output_file); + } + + for(unsigned int i = 0; i < node->mNumChildren; ++i) { + process_node(scene, node->mChildren[i], output_file); + } +}
\ No newline at end of file |