repo_model_export_gltf.h

#pragma once

#include <string>

#include "repo_model_export_web.h"
#include "../../../lib/repo_property_tree.h"
#include "../../../core/model/collection/repo_scene.h"

namespace repo{
    namespace manipulator{
        namespace modelconvertor{
            class GLTFModelExport : public WebModelExport
            {
            public:
                GLTFModelExport(const repo::core::model::RepoScene *scene);

                virtual ~GLTFModelExport();

                repo_web_buffers_t getAllFilesExportedAsBuffer() const;

            private:
                std::unordered_map<std::string, std::vector<uint8_t>> fullDataBuffer;

                void addAccessors(
                    const std::string              &accName,
                    const std::string              &buffViewName,
                    repo::lib::PropertyTree        &tree,
                    const std::vector<uint16_t>    &faces,
                    const uint32_t                 &addrFrom,
                    const uint32_t                 &addrTo,
                    const std::string              &refId = std::string(),
                    const std::vector<uint16_t>    &lod = std::vector<uint16_t>(),
                    const size_t                   &offset = 0);

                void addAccessors(
                    const std::string              &accName,
                    const std::string              &buffViewName,
                    repo::lib::PropertyTree        &tree,
                    const std::vector<float>       &data,
                    const uint32_t                 &addrFrom,
                    const uint32_t                 &addrTo,
                    const std::string              &refId,
                    const size_t                   &offset);

                void addAccessors(
                    const std::string                  &accName,
                    const std::string                  &buffViewName,
                    repo::lib::PropertyTree            &tree,
                    const std::vector<repo_vector2d_t> &buffer,
                    const uint32_t                     &addrFrom,
                    const uint32_t                     &addrTo,
                    const std::string                  &refId = std::string(),
                    const size_t                       &offset = 0);

                void addAccessors(
                    const std::string                &accName,
                    const std::string                &buffViewName,
                    repo::lib::PropertyTree          &tree,
                    const std::vector<repo_vector_t> &buffer,
                    const uint32_t                   &addrFrom,
                    const uint32_t                   &addrTo,
                    const std::string                &refId = std::string(),
                    const size_t                     &offset = 0);

                void addAccessors(
                    const std::string              &accName,
                    const std::string              &buffViewName,
                    repo::lib::PropertyTree        &tree,
                    const size_t                   &count,
                    const size_t                   &offset,
                    const size_t                   &stride,
                    const uint32_t                 &componentType,
                    const std::string              &bufferType,
                    const std::vector<float>       &min,
                    const std::vector<float>       &max,
                    const std::string              &refId = std::string(),
                    const std::vector<uint16_t>    &lod = std::vector<uint16_t>());

                void addBufferView(
                    const std::string                   &name,
                    const std::string                   &fileName,
                    repo::lib::PropertyTree             &tree,
                    const std::vector<uint16_t>         &buffer,
                    const size_t                        &offset,
                    const size_t                        &count,
                    const std::string                   &refId = std::string()
                    );

                void addBufferView(
                    const std::string                   &name,
                    const std::string                   &fileName,
                    repo::lib::PropertyTree             &tree,
                    const std::vector<float>            &buffer,
                    const size_t                        &offset,
                    const size_t                        &count,
                    const std::string                   &refId);

                void addBufferView(
                    const std::string                   &name,
                    const std::string                   &fileName,
                    repo::lib::PropertyTree             &tree,
                    const std::vector<repo_vector_t>    &buffer,
                    const size_t                        &offset,
                    const size_t                        &count,
                    const std::string                   &refId = std::string()
                    );

                void addBufferView(
                    const std::string                   &name,
                    const std::string                   &fileName,
                    repo::lib::PropertyTree             &tree,
                    const std::vector<repo_vector2d_t>  &buffer,
                    const size_t                        &offset,
                    const size_t                        &count,
                    const std::string                   &refId = std::string()
                    );

                void addBufferView(
                    const std::string                   &name,
                    const std::string                   &fileName,
                    repo::lib::PropertyTree             &tree,
                    const size_t                        &byteLength,
                    const size_t                        &offset,
                    const uint32_t                      &bufferTarget,
                    const std::string                   &refId = std::string()
                    );

                size_t addToDataBuffer(
                    const std::string              &bufferName,
                    const uint8_t                  *buffer,
                    const size_t                   &byteLength
                    );

                template <typename T>
                size_t addToDataBuffer(
                    const std::string              &bufferName,
                    const std::vector<T>           &buffer
                    )
                {
                    return addToDataBuffer(bufferName, (uint8_t*)buffer.data(), buffer.size() * sizeof(T));
                }

                bool constructScene(
                    repo::lib::PropertyTree &tree);

                bool generateTreeRepresentation();

                repo::lib::PropertyTree generateSpatialPartitioning();

                std::unordered_map<std::string, std::vector<uint8_t>> getGLTFFilesAsBuffer() const;

                bool reIndexFaces(
                    const std::vector<std::vector<repo_mesh_mapping_t>> &matMap,
                    std::vector<uint16_t>                               &faces);

                void processNodeChildren(
                    const repo::core::model::RepoNode *node,
                    repo::lib::PropertyTree          &tree,
                    const std::unordered_map<repoUUID, uint32_t, RepoUUIDHasher> &subMeshCounts
                    );

                void populateWithCameras(
                    repo::lib::PropertyTree          &tree);

                void populateWithMaterials(
                    repo::lib::PropertyTree          &tree);

                std::unordered_map<repoUUID, uint32_t, RepoUUIDHasher>
                    populateWithMeshes(
                    repo::lib::PropertyTree           &tree);

                void populateWithTextures(
                    repo::lib::PropertyTree          &tree);

                void populateWithNodes(
                    repo::lib::PropertyTree          &tree,
                    const std::unordered_map<repoUUID, uint32_t, RepoUUIDHasher> &subMeshCounts);

                std::vector<std::vector<std::vector<uint16_t>>> reorderFaces(
                    std::vector<uint16_t>                               &faces,
                    const std::vector<repo_vector_t>                    &vertices,
                    const std::vector<std::vector<repo_mesh_mapping_t>> &mapping);

                std::vector<uint16_t> reorderFaces(
                    const std::vector<uint16_t>      &faces,
                    const std::vector<repo_vector_t> &vertices,
                    const repo_mesh_mapping_t        &mapping,
                    std::vector<uint16_t>      &lods) const;

                std::vector<uint16_t> serialiseFaces(
                    const std::vector<repo_face_t> &faces) const;

                void writeBuffers(
                    repo::lib::PropertyTree &tree);

                void writeDefaultSampler(
                    repo::lib::PropertyTree &tree
                    );

                void writeDefaultTechnique(
                    repo::lib::PropertyTree &tree
                    );
            };
        } //namespace modelconvertor
    } //namespace manipulator
} //namespace repo