1 | /* |
2 | Open Asset Import Library (assimp) |
3 | ---------------------------------------------------------------------- |
4 | |
5 | Copyright (c) 2006-2017, assimp team |
6 | |
7 | All rights reserved. |
8 | |
9 | Redistribution and use of this software in source and binary forms, |
10 | with or without modification, are permitted provided that the |
11 | following conditions are met: |
12 | |
13 | * Redistributions of source code must retain the above |
14 | copyright notice, this list of conditions and the |
15 | following disclaimer. |
16 | |
17 | * Redistributions in binary form must reproduce the above |
18 | copyright notice, this list of conditions and the |
19 | following disclaimer in the documentation and/or other |
20 | materials provided with the distribution. |
21 | |
22 | * Neither the name of the assimp team, nor the names of its |
23 | contributors may be used to endorse or promote products |
24 | derived from this software without specific prior |
25 | written permission of the assimp team. |
26 | |
27 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
28 | "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
29 | LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
30 | A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
31 | OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
32 | SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
33 | LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
34 | DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
35 | THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
36 | (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
37 | OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
38 | |
39 | ---------------------------------------------------------------------- |
40 | */ |
41 | |
42 | /** @file GltfExporter.h |
43 | * Declares the exporter class to write a scene to a gltf/glb file |
44 | */ |
45 | #ifndef AI_GLTF2EXPORTER_H_INC |
46 | #define AI_GLTF2EXPORTER_H_INC |
47 | |
48 | #ifndef ASSIMP_BUILD_NO_GLTF_IMPORTER |
49 | |
50 | #include <assimp/types.h> |
51 | #include <assimp/material.h> |
52 | |
53 | #include <sstream> |
54 | #include <vector> |
55 | #include <map> |
56 | #include <memory> |
57 | |
58 | struct aiScene; |
59 | struct aiNode; |
60 | struct aiMaterial; |
61 | |
62 | namespace glTF2 |
63 | { |
64 | template<class T> |
65 | class Ref; |
66 | |
67 | class Asset; |
68 | struct TexProperty; |
69 | struct TextureInfo; |
70 | struct NormalTextureInfo; |
71 | struct OcclusionTextureInfo; |
72 | struct Node; |
73 | struct Texture; |
74 | |
75 | // Vec/matrix types, as raw float arrays |
76 | typedef float (vec3)[3]; |
77 | typedef float (vec4)[4]; |
78 | } |
79 | |
80 | namespace Assimp |
81 | { |
82 | class IOSystem; |
83 | class IOStream; |
84 | class ExportProperties; |
85 | |
86 | // ------------------------------------------------------------------------------------------------ |
87 | /** Helper class to export a given scene to an glTF file. */ |
88 | // ------------------------------------------------------------------------------------------------ |
89 | class glTF2Exporter |
90 | { |
91 | public: |
92 | /// Constructor for a specific scene to export |
93 | glTF2Exporter(const char* filename, IOSystem* pIOSystem, const aiScene* pScene, |
94 | const ExportProperties* pProperties, bool binary); |
95 | |
96 | private: |
97 | |
98 | const char* mFilename; |
99 | IOSystem* mIOSystem; |
100 | const aiScene* mScene; |
101 | const ExportProperties* mProperties; |
102 | |
103 | std::map<std::string, unsigned int> mTexturesByPath; |
104 | |
105 | std::shared_ptr<glTF2::Asset> mAsset; |
106 | |
107 | std::vector<unsigned char> mBodyData; |
108 | |
109 | void WriteBinaryData(IOStream* outfile, std::size_t sceneLength); |
110 | |
111 | void GetTexSampler(const aiMaterial* mat, glTF2::Ref<glTF2::Texture> texture, aiTextureType tt, unsigned int slot); |
112 | void GetMatTexProp(const aiMaterial* mat, unsigned int& prop, const char* propName, aiTextureType tt, unsigned int idx); |
113 | void GetMatTexProp(const aiMaterial* mat, float& prop, const char* propName, aiTextureType tt, unsigned int idx); |
114 | void GetMatTex(const aiMaterial* mat, glTF2::Ref<glTF2::Texture>& texture, aiTextureType tt, unsigned int slot); |
115 | void GetMatTex(const aiMaterial* mat, glTF2::TextureInfo& prop, aiTextureType tt, unsigned int slot); |
116 | void GetMatTex(const aiMaterial* mat, glTF2::NormalTextureInfo& prop, aiTextureType tt, unsigned int slot); |
117 | void GetMatTex(const aiMaterial* mat, glTF2::OcclusionTextureInfo& prop, aiTextureType tt, unsigned int slot); |
118 | aiReturn GetMatColor(const aiMaterial* mat, glTF2::vec4& prop, const char* propName, int type, int idx); |
119 | aiReturn GetMatColor(const aiMaterial* mat, glTF2::vec3& prop, const char* propName, int type, int idx); |
120 | void ExportMetadata(); |
121 | void ExportMaterials(); |
122 | void ExportMeshes(); |
123 | void MergeMeshes(); |
124 | unsigned int ExportNodeHierarchy(const aiNode* n); |
125 | unsigned int ExportNode(const aiNode* node, glTF2::Ref<glTF2::Node>& parent); |
126 | void ExportScene(); |
127 | void ExportAnimations(); |
128 | }; |
129 | |
130 | } |
131 | |
132 | #endif // ASSIMP_BUILD_NO_GLTF_IMPORTER |
133 | |
134 | #endif // AI_GLTF2EXPORTER_H_INC |
135 | |