Exporter.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629
  1. /*
  2. ---------------------------------------------------------------------------
  3. Open Asset Import Library (assimp)
  4. ---------------------------------------------------------------------------
  5. Copyright (c) 2006-2019, assimp team
  6. All rights reserved.
  7. Redistribution and use of this software in source and binary forms,
  8. with or without modification, are permitted provided that the following
  9. conditions are met:
  10. * Redistributions of source code must retain the above
  11. copyright notice, this list of conditions and the
  12. following disclaimer.
  13. * Redistributions in binary form must reproduce the above
  14. copyright notice, this list of conditions and the
  15. following disclaimer in the documentation and/or other
  16. materials provided with the distribution.
  17. * Neither the name of the assimp team, nor the names of its
  18. contributors may be used to endorse or promote products
  19. derived from this software without specific prior
  20. written permission of the assimp team.
  21. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22. "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23. LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24. A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25. OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26. SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27. LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28. DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29. THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30. (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31. OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32. ---------------------------------------------------------------------------
  33. */
  34. /** @file Exporter.cpp
  35. Assimp export interface. While it's public interface bears many similarities
  36. to the import interface (in fact, it is largely symmetric), the internal
  37. implementations differs a lot. Exporters are considered stateless and are
  38. simple callbacks which we maintain in a global list along with their
  39. description strings.
  40. Here we implement only the C++ interface (Assimp::Exporter).
  41. */
  42. #ifndef ASSIMP_BUILD_NO_EXPORT
  43. #include <assimp/BlobIOSystem.h>
  44. #include <assimp/SceneCombiner.h>
  45. #include <assimp/DefaultIOSystem.h>
  46. #include <assimp/Exporter.hpp>
  47. #include <assimp/mesh.h>
  48. #include <assimp/postprocess.h>
  49. #include <assimp/scene.h>
  50. #include <assimp/Exceptional.h>
  51. #include "Common/DefaultProgressHandler.h"
  52. #include "Common/BaseProcess.h"
  53. #include "Common/ScenePrivate.h"
  54. #include "PostProcessing/CalcTangentsProcess.h"
  55. #include "PostProcessing/MakeVerboseFormat.h"
  56. #include "PostProcessing/JoinVerticesProcess.h"
  57. #include "PostProcessing/ConvertToLHProcess.h"
  58. #include "PostProcessing/PretransformVertices.h"
  59. #include <memory>
  60. namespace Assimp {
  61. // PostStepRegistry.cpp
  62. void GetPostProcessingStepInstanceList(std::vector< BaseProcess* >& out);
  63. // ------------------------------------------------------------------------------------------------
  64. // Exporter worker function prototypes. Should not be necessary to #ifndef them, it's just a prototype
  65. // do not use const, because some exporter need to convert the scene temporary
  66. void ExportSceneCollada(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  67. void ExportSceneXFile(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  68. void ExportSceneStep(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  69. void ExportSceneObj(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  70. void ExportSceneObjNoMtl(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  71. void ExportSceneSTL(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  72. void ExportSceneSTLBinary(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  73. void ExportScenePly(const char*,IOSystem*, const aiScene*, const ExportProperties*);
  74. void ExportScenePlyBinary(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  75. void ExportScene3DS(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  76. void ExportSceneGLTF(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  77. void ExportSceneGLB(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  78. void ExportSceneGLTF2(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  79. void ExportSceneGLB2(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  80. void ExportSceneAssbin(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  81. void ExportSceneAssxml(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  82. void ExportSceneX3D(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  83. void ExportSceneFBX(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  84. void ExportSceneFBXA(const char*, IOSystem*, const aiScene*, const ExportProperties*);
  85. void ExportScene3MF( const char*, IOSystem*, const aiScene*, const ExportProperties* );
  86. void ExportAssimp2Json(const char* , IOSystem*, const aiScene* , const Assimp::ExportProperties*);
  87. // ------------------------------------------------------------------------------------------------
  88. // global array of all export formats which Assimp supports in its current build
  89. Exporter::ExportFormatEntry gExporters[] =
  90. {
  91. #ifndef ASSIMP_BUILD_NO_COLLADA_EXPORTER
  92. Exporter::ExportFormatEntry( "collada", "COLLADA - Digital Asset Exchange Schema", "dae", &ExportSceneCollada ),
  93. #endif
  94. #ifndef ASSIMP_BUILD_NO_X_EXPORTER
  95. Exporter::ExportFormatEntry( "x", "X Files", "x", &ExportSceneXFile,
  96. aiProcess_MakeLeftHanded | aiProcess_FlipWindingOrder | aiProcess_FlipUVs ),
  97. #endif
  98. #ifndef ASSIMP_BUILD_NO_STEP_EXPORTER
  99. Exporter::ExportFormatEntry( "stp", "Step Files", "stp", &ExportSceneStep, 0 ),
  100. #endif
  101. #ifndef ASSIMP_BUILD_NO_OBJ_EXPORTER
  102. Exporter::ExportFormatEntry( "obj", "Wavefront OBJ format", "obj", &ExportSceneObj,
  103. aiProcess_GenSmoothNormals /*| aiProcess_PreTransformVertices */ ),
  104. Exporter::ExportFormatEntry( "objnomtl", "Wavefront OBJ format without material file", "obj", &ExportSceneObjNoMtl,
  105. aiProcess_GenSmoothNormals /*| aiProcess_PreTransformVertices */ ),
  106. #endif
  107. #ifndef ASSIMP_BUILD_NO_STL_EXPORTER
  108. Exporter::ExportFormatEntry( "stl", "Stereolithography", "stl" , &ExportSceneSTL,
  109. aiProcess_Triangulate | aiProcess_GenNormals | aiProcess_PreTransformVertices
  110. ),
  111. Exporter::ExportFormatEntry( "stlb", "Stereolithography (binary)", "stl" , &ExportSceneSTLBinary,
  112. aiProcess_Triangulate | aiProcess_GenNormals | aiProcess_PreTransformVertices
  113. ),
  114. #endif
  115. #ifndef ASSIMP_BUILD_NO_PLY_EXPORTER
  116. Exporter::ExportFormatEntry( "ply", "Stanford Polygon Library", "ply" , &ExportScenePly,
  117. aiProcess_PreTransformVertices
  118. ),
  119. Exporter::ExportFormatEntry( "plyb", "Stanford Polygon Library (binary)", "ply", &ExportScenePlyBinary,
  120. aiProcess_PreTransformVertices
  121. ),
  122. #endif
  123. #ifndef ASSIMP_BUILD_NO_3DS_EXPORTER
  124. Exporter::ExportFormatEntry( "3ds", "Autodesk 3DS (legacy)", "3ds" , &ExportScene3DS,
  125. aiProcess_Triangulate | aiProcess_SortByPType | aiProcess_JoinIdenticalVertices ),
  126. #endif
  127. #ifndef ASSIMP_BUILD_NO_GLTF_EXPORTER
  128. Exporter::ExportFormatEntry( "gltf2", "GL Transmission Format v. 2", "gltf", &ExportSceneGLTF2,
  129. aiProcess_JoinIdenticalVertices | aiProcess_Triangulate | aiProcess_SortByPType ),
  130. Exporter::ExportFormatEntry( "glb2", "GL Transmission Format v. 2 (binary)", "glb", &ExportSceneGLB2,
  131. aiProcess_JoinIdenticalVertices | aiProcess_Triangulate | aiProcess_SortByPType ),
  132. Exporter::ExportFormatEntry( "gltf", "GL Transmission Format", "gltf", &ExportSceneGLTF,
  133. aiProcess_JoinIdenticalVertices | aiProcess_Triangulate | aiProcess_SortByPType ),
  134. Exporter::ExportFormatEntry( "glb", "GL Transmission Format (binary)", "glb", &ExportSceneGLB,
  135. aiProcess_JoinIdenticalVertices | aiProcess_Triangulate | aiProcess_SortByPType ),
  136. #endif
  137. #ifndef ASSIMP_BUILD_NO_ASSBIN_EXPORTER
  138. Exporter::ExportFormatEntry( "assbin", "Assimp Binary File", "assbin" , &ExportSceneAssbin, 0 ),
  139. #endif
  140. #ifndef ASSIMP_BUILD_NO_ASSXML_EXPORTER
  141. Exporter::ExportFormatEntry( "assxml", "Assimp XML Document", "assxml" , &ExportSceneAssxml, 0 ),
  142. #endif
  143. #ifndef ASSIMP_BUILD_NO_X3D_EXPORTER
  144. Exporter::ExportFormatEntry( "x3d", "Extensible 3D", "x3d" , &ExportSceneX3D, 0 ),
  145. #endif
  146. #ifndef ASSIMP_BUILD_NO_FBX_EXPORTER
  147. Exporter::ExportFormatEntry( "fbx", "Autodesk FBX (binary)", "fbx", &ExportSceneFBX, 0 ),
  148. Exporter::ExportFormatEntry( "fbxa", "Autodesk FBX (ascii)", "fbx", &ExportSceneFBXA, 0 ),
  149. #endif
  150. #ifndef ASSIMP_BUILD_NO_3MF_EXPORTER
  151. Exporter::ExportFormatEntry( "3mf", "The 3MF-File-Format", "3mf", &ExportScene3MF, 0 ),
  152. #endif
  153. #ifndef ASSIMP_BUILD_NO_ASSJSON_EXPORTER
  154. Exporter::ExportFormatEntry( "assjson", "Assimp JSON Document", "json", &ExportAssimp2Json, 0)
  155. #endif
  156. };
  157. #define ASSIMP_NUM_EXPORTERS (sizeof(gExporters)/sizeof(gExporters[0]))
  158. class ExporterPimpl {
  159. public:
  160. ExporterPimpl()
  161. : blob()
  162. , mIOSystem(new Assimp::DefaultIOSystem())
  163. , mIsDefaultIOHandler(true)
  164. , mProgressHandler( nullptr )
  165. , mIsDefaultProgressHandler( true )
  166. , mPostProcessingSteps()
  167. , mError()
  168. , mExporters() {
  169. GetPostProcessingStepInstanceList(mPostProcessingSteps);
  170. // grab all built-in exporters
  171. if ( 0 != ( ASSIMP_NUM_EXPORTERS ) ) {
  172. mExporters.resize( ASSIMP_NUM_EXPORTERS );
  173. std::copy( gExporters, gExporters + ASSIMP_NUM_EXPORTERS, mExporters.begin() );
  174. }
  175. }
  176. ~ExporterPimpl() {
  177. delete blob;
  178. // Delete all post-processing plug-ins
  179. for( unsigned int a = 0; a < mPostProcessingSteps.size(); a++) {
  180. delete mPostProcessingSteps[a];
  181. }
  182. delete mProgressHandler;
  183. }
  184. public:
  185. aiExportDataBlob* blob;
  186. std::shared_ptr< Assimp::IOSystem > mIOSystem;
  187. bool mIsDefaultIOHandler;
  188. /** The progress handler */
  189. ProgressHandler *mProgressHandler;
  190. bool mIsDefaultProgressHandler;
  191. /** Post processing steps we can apply at the imported data. */
  192. std::vector< BaseProcess* > mPostProcessingSteps;
  193. /** Last fatal export error */
  194. std::string mError;
  195. /** Exporters, this includes those registered using #Assimp::Exporter::RegisterExporter */
  196. std::vector<Exporter::ExportFormatEntry> mExporters;
  197. };
  198. } // end of namespace Assimp
  199. using namespace Assimp;
  200. // ------------------------------------------------------------------------------------------------
  201. Exporter :: Exporter()
  202. : pimpl(new ExporterPimpl()) {
  203. pimpl->mProgressHandler = new DefaultProgressHandler();
  204. }
  205. // ------------------------------------------------------------------------------------------------
  206. Exporter::~Exporter() {
  207. FreeBlob();
  208. delete pimpl;
  209. }
  210. // ------------------------------------------------------------------------------------------------
  211. void Exporter::SetIOHandler( IOSystem* pIOHandler) {
  212. pimpl->mIsDefaultIOHandler = !pIOHandler;
  213. pimpl->mIOSystem.reset(pIOHandler);
  214. }
  215. // ------------------------------------------------------------------------------------------------
  216. IOSystem* Exporter::GetIOHandler() const {
  217. return pimpl->mIOSystem.get();
  218. }
  219. // ------------------------------------------------------------------------------------------------
  220. bool Exporter::IsDefaultIOHandler() const {
  221. return pimpl->mIsDefaultIOHandler;
  222. }
  223. // ------------------------------------------------------------------------------------------------
  224. void Exporter::SetProgressHandler(ProgressHandler* pHandler) {
  225. ai_assert(nullptr != pimpl);
  226. if ( nullptr == pHandler) {
  227. // Release pointer in the possession of the caller
  228. pimpl->mProgressHandler = new DefaultProgressHandler();
  229. pimpl->mIsDefaultProgressHandler = true;
  230. return;
  231. }
  232. if (pimpl->mProgressHandler == pHandler) {
  233. return;
  234. }
  235. delete pimpl->mProgressHandler;
  236. pimpl->mProgressHandler = pHandler;
  237. pimpl->mIsDefaultProgressHandler = false;
  238. }
  239. // ------------------------------------------------------------------------------------------------
  240. const aiExportDataBlob* Exporter::ExportToBlob( const aiScene* pScene, const char* pFormatId,
  241. unsigned int pPreprocessing, const ExportProperties* pProperties) {
  242. if (pimpl->blob) {
  243. delete pimpl->blob;
  244. pimpl->blob = nullptr;
  245. }
  246. std::shared_ptr<IOSystem> old = pimpl->mIOSystem;
  247. BlobIOSystem* blobio = new BlobIOSystem();
  248. pimpl->mIOSystem = std::shared_ptr<IOSystem>( blobio );
  249. if (AI_SUCCESS != Export(pScene,pFormatId,blobio->GetMagicFileName(), pPreprocessing, pProperties)) {
  250. pimpl->mIOSystem = old;
  251. return nullptr;
  252. }
  253. pimpl->blob = blobio->GetBlobChain();
  254. pimpl->mIOSystem = old;
  255. return pimpl->blob;
  256. }
  257. // ------------------------------------------------------------------------------------------------
  258. aiReturn Exporter::Export( const aiScene* pScene, const char* pFormatId, const char* pPath,
  259. unsigned int pPreprocessing, const ExportProperties* pProperties) {
  260. ASSIMP_BEGIN_EXCEPTION_REGION();
  261. // when they create scenes from scratch, users will likely create them not in verbose
  262. // format. They will likely not be aware that there is a flag in the scene to indicate
  263. // this, however. To avoid surprises and bug reports, we check for duplicates in
  264. // meshes upfront.
  265. const bool is_verbose_format = !(pScene->mFlags & AI_SCENE_FLAGS_NON_VERBOSE_FORMAT) || MakeVerboseFormatProcess::IsVerboseFormat(pScene);
  266. pimpl->mProgressHandler->UpdateFileWrite(0, 4);
  267. pimpl->mError = "";
  268. for (size_t i = 0; i < pimpl->mExporters.size(); ++i) {
  269. const Exporter::ExportFormatEntry& exp = pimpl->mExporters[i];
  270. if (!strcmp(exp.mDescription.id,pFormatId)) {
  271. try {
  272. // Always create a full copy of the scene. We might optimize this one day,
  273. // but for now it is the most pragmatic way.
  274. aiScene* scenecopy_tmp = nullptr;
  275. SceneCombiner::CopyScene(&scenecopy_tmp,pScene);
  276. pimpl->mProgressHandler->UpdateFileWrite(1, 4);
  277. std::unique_ptr<aiScene> scenecopy(scenecopy_tmp);
  278. const ScenePrivateData* const priv = ScenePriv(pScene);
  279. // steps that are not idempotent, i.e. we might need to run them again, usually to get back to the
  280. // original state before the step was applied first. When checking which steps we don't need
  281. // to run, those are excluded.
  282. const unsigned int nonIdempotentSteps = aiProcess_FlipWindingOrder | aiProcess_FlipUVs | aiProcess_MakeLeftHanded;
  283. // Erase all pp steps that were already applied to this scene
  284. const unsigned int pp = (exp.mEnforcePP | pPreprocessing) & ~(priv && !priv->mIsCopy
  285. ? (priv->mPPStepsApplied & ~nonIdempotentSteps)
  286. : 0u);
  287. // If no extra post-processing was specified, and we obtained this scene from an
  288. // Assimp importer, apply the reverse steps automatically.
  289. // TODO: either drop this, or document it. Otherwise it is just a bad surprise.
  290. //if (!pPreprocessing && priv) {
  291. // pp |= (nonIdempotentSteps & priv->mPPStepsApplied);
  292. //}
  293. // If the input scene is not in verbose format, but there is at least post-processing step that relies on it,
  294. // we need to run the MakeVerboseFormat step first.
  295. bool must_join_again = false;
  296. if (!is_verbose_format) {
  297. bool verbosify = false;
  298. for( unsigned int a = 0; a < pimpl->mPostProcessingSteps.size(); a++) {
  299. BaseProcess* const p = pimpl->mPostProcessingSteps[a];
  300. if (p->IsActive(pp) && p->RequireVerboseFormat()) {
  301. verbosify = true;
  302. break;
  303. }
  304. }
  305. if (verbosify || (exp.mEnforcePP & aiProcess_JoinIdenticalVertices)) {
  306. ASSIMP_LOG_DEBUG("export: Scene data not in verbose format, applying MakeVerboseFormat step first");
  307. MakeVerboseFormatProcess proc;
  308. proc.Execute(scenecopy.get());
  309. if(!(exp.mEnforcePP & aiProcess_JoinIdenticalVertices)) {
  310. must_join_again = true;
  311. }
  312. }
  313. }
  314. pimpl->mProgressHandler->UpdateFileWrite(2, 4);
  315. if (pp) {
  316. // the three 'conversion' steps need to be executed first because all other steps rely on the standard data layout
  317. {
  318. FlipWindingOrderProcess step;
  319. if (step.IsActive(pp)) {
  320. step.Execute(scenecopy.get());
  321. }
  322. }
  323. {
  324. FlipUVsProcess step;
  325. if (step.IsActive(pp)) {
  326. step.Execute(scenecopy.get());
  327. }
  328. }
  329. {
  330. MakeLeftHandedProcess step;
  331. if (step.IsActive(pp)) {
  332. step.Execute(scenecopy.get());
  333. }
  334. }
  335. bool exportPointCloud(false);
  336. if (nullptr != pProperties) {
  337. exportPointCloud = pProperties->GetPropertyBool(AI_CONFIG_EXPORT_POINT_CLOUDS);
  338. }
  339. // dispatch other processes
  340. for( unsigned int a = 0; a < pimpl->mPostProcessingSteps.size(); a++) {
  341. BaseProcess* const p = pimpl->mPostProcessingSteps[a];
  342. if (p->IsActive(pp)
  343. && !dynamic_cast<FlipUVsProcess*>(p)
  344. && !dynamic_cast<FlipWindingOrderProcess*>(p)
  345. && !dynamic_cast<MakeLeftHandedProcess*>(p)) {
  346. if (dynamic_cast<PretransformVertices*>(p) && exportPointCloud) {
  347. continue;
  348. }
  349. p->Execute(scenecopy.get());
  350. }
  351. }
  352. ScenePrivateData* const privOut = ScenePriv(scenecopy.get());
  353. ai_assert(nullptr != privOut);
  354. privOut->mPPStepsApplied |= pp;
  355. }
  356. pimpl->mProgressHandler->UpdateFileWrite(3, 4);
  357. if(must_join_again) {
  358. JoinVerticesProcess proc;
  359. proc.Execute(scenecopy.get());
  360. }
  361. ExportProperties emptyProperties; // Never pass NULL ExportProperties so Exporters don't have to worry.
  362. ExportProperties* pProp = pProperties ? (ExportProperties*)pProperties : &emptyProperties;
  363. pProp->SetPropertyBool("bJoinIdenticalVertices", must_join_again);
  364. exp.mExportFunction(pPath,pimpl->mIOSystem.get(),scenecopy.get(), pProp);
  365. exp.mExportFunction(pPath,pimpl->mIOSystem.get(),scenecopy.get(), pProp);
  366. pimpl->mProgressHandler->UpdateFileWrite(4, 4);
  367. } catch (DeadlyExportError& err) {
  368. pimpl->mError = err.what();
  369. return AI_FAILURE;
  370. }
  371. return AI_SUCCESS;
  372. }
  373. }
  374. pimpl->mError = std::string("Found no exporter to handle this file format: ") + pFormatId;
  375. ASSIMP_END_EXCEPTION_REGION(aiReturn);
  376. return AI_FAILURE;
  377. }
  378. // ------------------------------------------------------------------------------------------------
  379. const char* Exporter::GetErrorString() const {
  380. return pimpl->mError.c_str();
  381. }
  382. // ------------------------------------------------------------------------------------------------
  383. void Exporter::FreeBlob() {
  384. delete pimpl->blob;
  385. pimpl->blob = nullptr;
  386. pimpl->mError = "";
  387. }
  388. // ------------------------------------------------------------------------------------------------
  389. const aiExportDataBlob* Exporter::GetBlob() const {
  390. return pimpl->blob;
  391. }
  392. // ------------------------------------------------------------------------------------------------
  393. const aiExportDataBlob* Exporter::GetOrphanedBlob() const {
  394. const aiExportDataBlob* tmp = pimpl->blob;
  395. pimpl->blob = nullptr;
  396. return tmp;
  397. }
  398. // ------------------------------------------------------------------------------------------------
  399. size_t Exporter::GetExportFormatCount() const {
  400. return pimpl->mExporters.size();
  401. }
  402. // ------------------------------------------------------------------------------------------------
  403. const aiExportFormatDesc* Exporter::GetExportFormatDescription( size_t index ) const {
  404. if (index >= GetExportFormatCount()) {
  405. return nullptr;
  406. }
  407. // Return from static storage if the requested index is built-in.
  408. if (index < sizeof(gExporters) / sizeof(gExporters[0])) {
  409. return &gExporters[index].mDescription;
  410. }
  411. return &pimpl->mExporters[index].mDescription;
  412. }
  413. // ------------------------------------------------------------------------------------------------
  414. aiReturn Exporter::RegisterExporter(const ExportFormatEntry& desc) {
  415. for(const ExportFormatEntry& e : pimpl->mExporters) {
  416. if (!strcmp(e.mDescription.id,desc.mDescription.id)) {
  417. return aiReturn_FAILURE;
  418. }
  419. }
  420. pimpl->mExporters.push_back(desc);
  421. return aiReturn_SUCCESS;
  422. }
  423. // ------------------------------------------------------------------------------------------------
  424. void Exporter::UnregisterExporter(const char* id) {
  425. for(std::vector<ExportFormatEntry>::iterator it = pimpl->mExporters.begin();
  426. it != pimpl->mExporters.end(); ++it) {
  427. if (!strcmp((*it).mDescription.id,id)) {
  428. pimpl->mExporters.erase(it);
  429. break;
  430. }
  431. }
  432. }
  433. // ------------------------------------------------------------------------------------------------
  434. ExportProperties::ExportProperties() {
  435. // empty
  436. }
  437. // ------------------------------------------------------------------------------------------------
  438. ExportProperties::ExportProperties(const ExportProperties &other)
  439. : mIntProperties(other.mIntProperties)
  440. , mFloatProperties(other.mFloatProperties)
  441. , mStringProperties(other.mStringProperties)
  442. , mMatrixProperties(other.mMatrixProperties) {
  443. // empty
  444. }
  445. // ------------------------------------------------------------------------------------------------
  446. // Set a configuration property
  447. bool ExportProperties::SetPropertyInteger(const char* szName, int iValue) {
  448. return SetGenericProperty<int>(mIntProperties, szName,iValue);
  449. }
  450. // ------------------------------------------------------------------------------------------------
  451. // Set a configuration property
  452. bool ExportProperties::SetPropertyFloat(const char* szName, ai_real iValue) {
  453. return SetGenericProperty<ai_real>(mFloatProperties, szName,iValue);
  454. }
  455. // ------------------------------------------------------------------------------------------------
  456. // Set a configuration property
  457. bool ExportProperties::SetPropertyString(const char* szName, const std::string& value) {
  458. return SetGenericProperty<std::string>(mStringProperties, szName,value);
  459. }
  460. // ------------------------------------------------------------------------------------------------
  461. // Set a configuration property
  462. bool ExportProperties::SetPropertyMatrix(const char* szName, const aiMatrix4x4& value) {
  463. return SetGenericProperty<aiMatrix4x4>(mMatrixProperties, szName,value);
  464. }
  465. // ------------------------------------------------------------------------------------------------
  466. // Get a configuration property
  467. int ExportProperties::GetPropertyInteger(const char* szName, int iErrorReturn /*= 0xffffffff*/) const {
  468. return GetGenericProperty<int>(mIntProperties,szName,iErrorReturn);
  469. }
  470. // ------------------------------------------------------------------------------------------------
  471. // Get a configuration property
  472. ai_real ExportProperties::GetPropertyFloat(const char* szName, ai_real iErrorReturn /*= 10e10*/) const {
  473. return GetGenericProperty<ai_real>(mFloatProperties,szName,iErrorReturn);
  474. }
  475. // ------------------------------------------------------------------------------------------------
  476. // Get a configuration property
  477. const std::string ExportProperties::GetPropertyString(const char* szName,
  478. const std::string& iErrorReturn /*= ""*/) const {
  479. return GetGenericProperty<std::string>(mStringProperties,szName,iErrorReturn);
  480. }
  481. // ------------------------------------------------------------------------------------------------
  482. // Has a configuration property
  483. const aiMatrix4x4 ExportProperties::GetPropertyMatrix(const char* szName,
  484. const aiMatrix4x4& iErrorReturn /*= aiMatrix4x4()*/) const {
  485. return GetGenericProperty<aiMatrix4x4>(mMatrixProperties,szName,iErrorReturn);
  486. }
  487. // ------------------------------------------------------------------------------------------------
  488. // Has a configuration property
  489. bool ExportProperties::HasPropertyInteger(const char* szName) const {
  490. return HasGenericProperty<int>(mIntProperties, szName);
  491. }
  492. // ------------------------------------------------------------------------------------------------
  493. // Has a configuration property
  494. bool ExportProperties::HasPropertyBool(const char* szName) const {
  495. return HasGenericProperty<int>(mIntProperties, szName);
  496. }
  497. // ------------------------------------------------------------------------------------------------
  498. // Has a configuration property
  499. bool ExportProperties::HasPropertyFloat(const char* szName) const {
  500. return HasGenericProperty<ai_real>(mFloatProperties, szName);
  501. }
  502. // ------------------------------------------------------------------------------------------------
  503. // Has a configuration property
  504. bool ExportProperties::HasPropertyString(const char* szName) const {
  505. return HasGenericProperty<std::string>(mStringProperties, szName);
  506. }
  507. // ------------------------------------------------------------------------------------------------
  508. // Has a configuration property
  509. bool ExportProperties::HasPropertyMatrix(const char* szName) const {
  510. return HasGenericProperty<aiMatrix4x4>(mMatrixProperties, szName);
  511. }
  512. #endif // !ASSIMP_BUILD_NO_EXPORT