SceneCombiner.h

/*
Open Asset Import Library (assimp)
----------------------------------------------------------------------

Copyright (c) 2006-2017, assimp team

All rights reserved.

Redistribution and use of this software in source and binary forms,
with or without modification, are permitted provided that the
following conditions are met:

* Redistributions of source code must retain the above
  copyright notice, this list of conditions and the
  following disclaimer.

* Redistributions in binary form must reproduce the above
  copyright notice, this list of conditions and the
  following disclaimer in the documentation and/or other
  materials provided with the distribution.

* Neither the name of the assimp team, nor the names of its
  contributors may be used to endorse or promote products
  derived from this software without specific prior
  written permission of the assimp team.

THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

----------------------------------------------------------------------
*/

#ifndef AI_SCENE_COMBINER_H_INC
#define AI_SCENE_COMBINER_H_INC

#include <assimp/ai_assert.h>
#include <assimp/types.h>
#include <assimp/Defines.h>
#include <stddef.h>
#include <set>
#include <list>
#include <stdint.h>

#include <vector>

struct aiScene;
struct aiNode;
struct aiMaterial;
struct aiTexture;
struct aiCamera;
struct aiLight;
struct aiMetadata;
struct aiBone;
struct aiMesh;
struct aiAnimation;
struct aiNodeAnim;

namespace Assimp    {

// ---------------------------------------------------------------------------
struct AttachmentInfo
{
    AttachmentInfo()
        :   scene           (NULL)
        ,   attachToNode    (NULL)
    {}

    AttachmentInfo(aiScene* _scene, aiNode* _attachToNode)
        :   scene           (_scene)
        ,   attachToNode    (_attachToNode)
    {}

    aiScene* scene;
    aiNode*  attachToNode;
};

// ---------------------------------------------------------------------------
struct NodeAttachmentInfo
{
    NodeAttachmentInfo()
        :   node            (NULL)
        ,   attachToNode    (NULL)
        ,   resolved        (false)
        ,   src_idx         (SIZE_MAX)
    {}

    NodeAttachmentInfo(aiNode* _scene, aiNode* _attachToNode,size_t idx)
        :   node            (_scene)
        ,   attachToNode    (_attachToNode)
        ,   resolved        (false)
        ,   src_idx         (idx)
    {}

    aiNode*  node;
    aiNode*  attachToNode;
    bool     resolved;
    size_t   src_idx;
};

// ---------------------------------------------------------------------------
#define AI_INT_MERGE_SCENE_GEN_UNIQUE_NAMES      0x1

#define AI_INT_MERGE_SCENE_GEN_UNIQUE_MATNAMES   0x2

#define AI_INT_MERGE_SCENE_DUPLICATES_DEEP_CPY   0x4

#define AI_INT_MERGE_SCENE_RESOLVE_CROSS_ATTACHMENTS 0x8

#define AI_INT_MERGE_SCENE_GEN_UNIQUE_NAMES_IF_NECESSARY 0x10

typedef std::pair<aiBone*,unsigned int> BoneSrcIndex;

// ---------------------------------------------------------------------------
struct BoneWithHash : public std::pair<uint32_t,aiString*>  {
    std::vector<BoneSrcIndex> pSrcBones;
};

// ---------------------------------------------------------------------------
struct SceneHelper
{
    SceneHelper ()
        : scene     (NULL)
        , idlen     (0)
    {
        id[0] = 0;
    }

    explicit SceneHelper (aiScene* _scene)
        : scene     (_scene)
        , idlen     (0)
    {
        id[0] = 0;
    }

    AI_FORCE_INLINE aiScene* operator-> () const
    {
        return scene;
    }

    // scene we're working on
    aiScene* scene;

    // prefix to be added to all identifiers in the scene ...
    char id [32];

    // and its strlen()
    unsigned int idlen;

    // hash table to quickly check whether a name is contained in the scene
    std::set<unsigned int> hashes;
};

// ---------------------------------------------------------------------------
class ASSIMP_API SceneCombiner {
    // class cannot be instanced
    SceneCombiner() {
        // empty
    }

    ~SceneCombiner() {
        // empty
    }

public:
    // -------------------------------------------------------------------
    static void MergeScenes(aiScene** dest,std::vector<aiScene*>& src,
        unsigned int flags = 0);

    // -------------------------------------------------------------------
    static void MergeScenes(aiScene** dest, aiScene* master,
        std::vector<AttachmentInfo>& src,
        unsigned int flags = 0);

    // -------------------------------------------------------------------
    static void MergeMeshes(aiMesh** dest,unsigned int flags,
        std::vector<aiMesh*>::const_iterator begin,
        std::vector<aiMesh*>::const_iterator end);

    // -------------------------------------------------------------------
    static void MergeBones(aiMesh* out,std::vector<aiMesh*>::const_iterator it,
        std::vector<aiMesh*>::const_iterator end);

    // -------------------------------------------------------------------
    static void MergeMaterials(aiMaterial** dest,
        std::vector<aiMaterial*>::const_iterator begin,
        std::vector<aiMaterial*>::const_iterator end);

    // -------------------------------------------------------------------
    static void BuildUniqueBoneList(std::list<BoneWithHash>& asBones,
        std::vector<aiMesh*>::const_iterator it,
        std::vector<aiMesh*>::const_iterator end);

    // -------------------------------------------------------------------
    static void AddNodePrefixes(aiNode* node, const char* prefix,
        unsigned int len);

    // -------------------------------------------------------------------
    static void OffsetNodeMeshIndices (aiNode* node, unsigned int offset);

    // -------------------------------------------------------------------
    static void AttachToGraph ( aiScene* master,
        std::vector<NodeAttachmentInfo>& srcList);

    static void AttachToGraph (aiNode* attach,
        std::vector<NodeAttachmentInfo>& srcList);


    // -------------------------------------------------------------------
    static void CopyScene(aiScene** dest,const aiScene* source,bool allocate = true);


    // -------------------------------------------------------------------
    static void CopySceneFlat(aiScene** dest,const aiScene* source);


    // -------------------------------------------------------------------
    static void Copy     (aiMesh** dest, const aiMesh* src);

    // similar to Copy():
    static void Copy  (aiMaterial** dest, const aiMaterial* src);
    static void Copy  (aiTexture** dest, const aiTexture* src);
    static void Copy  (aiAnimation** dest, const aiAnimation* src);
    static void Copy  (aiCamera** dest, const aiCamera* src);
    static void Copy  (aiBone** dest, const aiBone* src);
    static void Copy  (aiLight** dest, const aiLight* src);
    static void Copy  (aiNodeAnim** dest, const aiNodeAnim* src);
    static void Copy  (aiMetadata** dest, const aiMetadata* src);

    // recursive, of course
    static void Copy     (aiNode** dest, const aiNode* src);


private:

    // -------------------------------------------------------------------
    // Same as AddNodePrefixes, but with an additional check
    static void AddNodePrefixesChecked(aiNode* node, const char* prefix,
        unsigned int len,
        std::vector<SceneHelper>& input,
        unsigned int cur);

    // -------------------------------------------------------------------
    // Add node identifiers to a hashing set
    static void AddNodeHashes(aiNode* node, std::set<unsigned int>& hashes);


    // -------------------------------------------------------------------
    // Search for duplicate names
    static bool FindNameMatch(const aiString& name,
        std::vector<SceneHelper>& input, unsigned int cur);
};

}

#endif // !! AI_SCENE_COMBINER_H_INC