repo_model_export_assimp.h

#pragma once

#include <string>

#include <assimp/scene.h>
#include <assimp/Exporter.hpp>

#include "../../../core/model/bson/repo_node_camera.h"
#include "../../../core/model/bson/repo_node_material.h"
#include "../../../core/model/bson/repo_node_mesh.h"
#include "../../../core/model/collection/repo_scene.h"
#include "repo_model_export_abstract.h"

namespace repo{
    namespace manipulator{
        namespace modelconvertor{//x3d shader
            class AssimpModelExport : public AbstractModelExport
            {
            public:
                AssimpModelExport(
                    const repo::core::model::RepoScene *scene);

                virtual ~AssimpModelExport();

                bool exportToFile(
                    const std::string                  &filePath);

                static std::string getSupportedFormats();

            private:

                aiNode* constructAiSceneRecursively(
                    const repo::core::model::RepoScene            *scene,
                    const repo::core::model::RepoNode             *currNode,
                    std::vector<aiMesh*>                          &meshVec,
                    std::vector<aiMaterial*>                      &matVec,
                    std::vector<aiCamera*>                        &camVec,
                    repo::core::model::RepoNodeSet                &textNodes,
                    const repo::core::model::RepoScene::GraphType &gType
                    = repo::core::model::RepoScene::GraphType::DEFAULT);

                aiNode* constructAiSceneRecursively(
                    const repo::core::model::RepoScene                        *scene,
                    const repo::core::model::RepoNode                         *currNode,
                    std::vector<aiMesh*>                                      &meshVec,
                    std::vector<aiMaterial*>                                  &matVec,
                    std::vector<aiCamera*>                                    &camVec,
                    std::unordered_map<repoUUID, aiMesh*, RepoUUIDHasher>     &meshMap,
                    std::unordered_map<repoUUID, aiMaterial*, RepoUUIDHasher> &materialMap,
                    std::unordered_map<repoUUID, aiCamera*, RepoUUIDHasher>   &camMap,
                    repo::core::model::RepoNodeSet                            &textNodes,
                    const repo::core::model::RepoScene::GraphType             &gType
                    = repo::core::model::RepoScene::GraphType::DEFAULT);

                aiCamera* convertCamera(
                    const repo::core::model::RepoScene  *scene,
                    const repo::core::model::CameraNode *camNode,
                    const std::string                   &name = std::string());

                aiMaterial* convertMaterial(
                    const repo::core::model::RepoScene            *scene,
                    const repo::core::model::MaterialNode         *matNode,
                    repo::core::model::RepoNodeSet                &textNodes,
                    const repo::core::model::RepoScene::GraphType &gType
                    = repo::core::model::RepoScene::GraphType::DEFAULT);

                aiMesh* convertMesh(
                    const repo::core::model::RepoScene                        *scene,
                    const repo::core::model::MeshNode                         *meshNode,
                    std::vector<aiMaterial*>                                  &matVec,
                    std::unordered_map<repoUUID, aiMaterial*, RepoUUIDHasher> &matMap,
                    repo::core::model::RepoNodeSet                            &textNodes,
                    const repo::core::model::RepoScene::GraphType             &gType
                    = repo::core::model::RepoScene::GraphType::DEFAULT);

                aiScene* convertToAssimp(
                    const repo::core::model::RepoScene            *scene,
                    repo::core::model::RepoNodeSet                &textNodes,
                    const repo::core::model::RepoScene::GraphType &gType
                    = repo::core::model::RepoScene::GraphType::DEFAULT);

                std::string getExportFormatID(
                    const std::string &fileExtension);

                bool writeTexturesToFiles(
                    const repo::core::model::RepoNodeSet &nodes,
                    const std::string                    &filePath);

                bool writeSceneToFile(
                    const aiScene     *scene,
                    const std::string &filePath);
            };
        } //namespace modelconvertor
    } //namespace manipulator
} //namespace repo