2017-08-24 21:15:48 +00:00
/*
Open Asset Import Library ( assimp )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
2021-02-28 11:17:54 +00:00
Copyright ( c ) 2006 - 2021 , assimp team
2018-01-28 18:42:05 +00:00
2017-08-24 21:15:48 +00:00
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 .
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
*/
2020-07-05 18:22:31 +00:00
# if !defined(ASSIMP_BUILD_NO_GLTF_IMPORTER) && !defined(ASSIMP_BUILD_NO_GLTF2_IMPORTER)
2017-08-24 21:15:48 +00:00
2020-05-02 13:14:38 +00:00
# include "AssetLib/glTF2/glTF2Importer.h"
2019-11-16 07:08:57 +00:00
# include "PostProcessing/MakeVerboseFormat.h"
2020-05-02 13:14:38 +00:00
# include "AssetLib/glTF2/glTF2Asset.h"
2021-04-16 18:44:40 +00:00
# if !defined(ASSIMP_BUILD_NO_EXPORT)
2020-05-02 13:14:38 +00:00
# include "AssetLib/glTF2/glTF2AssetWriter.h"
2021-04-16 18:44:40 +00:00
# endif
2019-06-10 21:26:00 +00:00
2019-11-16 07:08:57 +00:00
# include <assimp/CreateAnimMesh.h>
2018-01-06 00:18:33 +00:00
# include <assimp/StringComparison.h>
# include <assimp/StringUtils.h>
2017-08-24 21:15:48 +00:00
# include <assimp/ai_assert.h>
# include <assimp/importerdesc.h>
2019-11-16 07:08:57 +00:00
# include <assimp/scene.h>
# include <assimp/DefaultLogger.hpp>
# include <assimp/Importer.hpp>
2019-12-05 11:25:12 +00:00
# include <assimp/commonMetaData.h>
2017-08-24 21:15:48 +00:00
# include <memory>
2018-10-26 22:36:34 +00:00
# include <unordered_map>
2017-08-24 21:15:48 +00:00
2017-09-06 18:50:02 +00:00
# include <rapidjson/document.h>
# include <rapidjson/rapidjson.h>
2017-08-24 21:15:48 +00:00
using namespace Assimp ;
using namespace glTF2 ;
2019-09-11 15:00:39 +00:00
using namespace glTFCommon ;
2017-08-24 21:15:48 +00:00
2018-07-05 13:28:29 +00:00
namespace {
2019-11-16 07:08:57 +00:00
// generate bi-tangents from normals and tangents according to spec
struct Tangent {
2020-10-16 15:09:17 +00:00
aiVector3D xyz ;
ai_real w ;
2019-11-16 07:08:57 +00:00
} ;
2018-07-05 13:28:29 +00:00
} // namespace
2017-08-24 21:15:48 +00:00
//
// glTF2Importer
//
static const aiImporterDesc desc = {
2020-10-16 15:09:17 +00:00
" glTF2 Importer " ,
" " ,
" " ,
" " ,
aiImporterFlags_SupportTextFlavour | aiImporterFlags_SupportBinaryFlavour | aiImporterFlags_LimitedSupport | aiImporterFlags_Experimental ,
0 ,
0 ,
0 ,
0 ,
" gltf glb "
2017-08-24 21:15:48 +00:00
} ;
2019-11-16 07:08:57 +00:00
glTF2Importer : : glTF2Importer ( ) :
2020-10-16 15:09:17 +00:00
BaseImporter ( ) ,
meshOffsets ( ) ,
embeddedTexIdxs ( ) ,
mScene ( nullptr ) {
// empty
2017-08-24 21:15:48 +00:00
}
glTF2Importer : : ~ glTF2Importer ( ) {
2020-10-16 15:09:17 +00:00
// empty
2017-08-24 21:15:48 +00:00
}
2019-11-16 07:08:57 +00:00
const aiImporterDesc * glTF2Importer : : GetInfo ( ) const {
2020-10-16 15:09:17 +00:00
return & desc ;
2017-08-24 21:15:48 +00:00
}
2019-11-16 07:08:57 +00:00
bool glTF2Importer : : CanRead ( const std : : string & pFile , IOSystem * pIOHandler , bool /* checkSig */ ) const {
2020-10-16 15:09:17 +00:00
const std : : string & extension = GetExtension ( pFile ) ;
2017-09-06 18:50:02 +00:00
2020-10-16 15:09:17 +00:00
if ( extension ! = " gltf " & & extension ! = " glb " ) {
2020-03-15 09:17:54 +00:00
return false ;
2020-10-16 15:09:17 +00:00
}
2017-09-06 18:50:02 +00:00
2020-10-16 15:09:17 +00:00
if ( pIOHandler ) {
glTF2 : : Asset asset ( pIOHandler ) ;
asset . Load ( pFile , extension = = " glb " ) ;
std : : string version = asset . asset . version ;
return ! version . empty ( ) & & version [ 0 ] = = ' 2 ' ;
}
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
return false ;
2017-08-24 21:15:48 +00:00
}
2019-11-16 07:08:57 +00:00
static aiTextureMapMode ConvertWrappingMode ( SamplerWrap gltfWrapMode ) {
2020-10-16 15:09:17 +00:00
switch ( gltfWrapMode ) {
case SamplerWrap : : Mirrored_Repeat :
return aiTextureMapMode_Mirror ;
case SamplerWrap : : Clamp_To_Edge :
return aiTextureMapMode_Clamp ;
case SamplerWrap : : UNSET :
case SamplerWrap : : Repeat :
default :
return aiTextureMapMode_Wrap ;
}
2018-05-25 00:25:26 +00:00
}
2017-08-24 21:15:48 +00:00
2019-11-16 07:08:57 +00:00
inline void SetMaterialColorProperty ( Asset & /*r*/ , vec4 & prop , aiMaterial * mat , const char * pKey , unsigned int type , unsigned int idx ) {
2020-10-16 15:09:17 +00:00
aiColor4D col ;
CopyValue ( prop , col ) ;
mat - > AddProperty ( & col , 1 , pKey , type , idx ) ;
2017-08-30 21:25:11 +00:00
}
2019-11-16 07:08:57 +00:00
inline void SetMaterialColorProperty ( Asset & /*r*/ , vec3 & prop , aiMaterial * mat , const char * pKey , unsigned int type , unsigned int idx ) {
2020-10-16 15:09:17 +00:00
aiColor4D col ;
glTFCommon : : CopyValue ( prop , col ) ;
mat - > AddProperty ( & col , 1 , pKey , type , idx ) ;
2017-08-30 21:25:11 +00:00
}
2019-11-16 07:08:57 +00:00
inline void SetMaterialTextureProperty ( std : : vector < int > & embeddedTexIdxs , Asset & /*r*/ , glTF2 : : TextureInfo prop , aiMaterial * mat , aiTextureType texType , unsigned int texSlot = 0 ) {
2020-10-16 15:09:17 +00:00
if ( prop . texture & & prop . texture - > source ) {
aiString uri ( prop . texture - > source - > uri ) ;
int texIdx = embeddedTexIdxs [ prop . texture - > source . GetIndex ( ) ] ;
if ( texIdx ! = - 1 ) { // embedded
// setup texture reference string (copied from ColladaLoader::FindFilenameForEffectTexture)
uri . data [ 0 ] = ' * ' ;
uri . length = 1 + ASSIMP_itoa10 ( uri . data + 1 , MAXLEN - 1 , texIdx ) ;
}
2019-11-16 07:08:57 +00:00
2019-11-16 14:51:26 +00:00
mat - > AddProperty ( & uri , AI_MATKEY_TEXTURE ( texType , texSlot ) ) ;
2021-06-11 14:17:25 +00:00
const int uvIndex = static_cast < int > ( prop . texCoord ) ;
mat - > AddProperty ( & uvIndex , 1 , AI_MATKEY_UVWSRC ( texType , texSlot ) ) ;
2019-11-16 14:51:26 +00:00
2020-10-16 15:09:17 +00:00
if ( prop . textureTransformSupported ) {
aiUVTransform transform ;
transform . mScaling . x = prop . TextureTransformExt_t . scale [ 0 ] ;
transform . mScaling . y = prop . TextureTransformExt_t . scale [ 1 ] ;
transform . mRotation = - prop . TextureTransformExt_t . rotation ; // must be negated
// A change of coordinates is required to map glTF UV transformations into the space used by
// Assimp. In glTF all UV origins are at 0,1 (top left of texture) in Assimp space. In Assimp
// rotation occurs around the image center (0.5,0.5) where as in glTF rotation is around the
// texture origin. All three can be corrected for solely by a change of the translation since
// the transformations available are shape preserving. Note the importer already flips the V
// coordinate of the actual meshes during import.
const ai_real rcos ( cos ( - transform . mRotation ) ) ;
const ai_real rsin ( sin ( - transform . mRotation ) ) ;
transform . mTranslation . x = ( static_cast < ai_real > ( 0.5 ) * transform . mScaling . x ) * ( - rcos + rsin + 1 ) + prop . TextureTransformExt_t . offset [ 0 ] ;
transform . mTranslation . y = ( ( static_cast < ai_real > ( 0.5 ) * transform . mScaling . y ) * ( rsin + rcos - 1 ) ) + 1 - transform . mScaling . y - prop . TextureTransformExt_t . offset [ 1 ] ; ;
mat - > AddProperty ( & transform , 1 , _AI_MATKEY_UVTRANSFORM_BASE , texType , texSlot ) ;
}
if ( prop . texture - > sampler ) {
Ref < Sampler > sampler = prop . texture - > sampler ;
aiString name ( sampler - > name ) ;
aiString id ( sampler - > id ) ;
mat - > AddProperty ( & name , AI_MATKEY_GLTF_MAPPINGNAME ( texType , texSlot ) ) ;
mat - > AddProperty ( & id , AI_MATKEY_GLTF_MAPPINGID ( texType , texSlot ) ) ;
aiTextureMapMode wrapS = ConvertWrappingMode ( sampler - > wrapS ) ;
aiTextureMapMode wrapT = ConvertWrappingMode ( sampler - > wrapT ) ;
mat - > AddProperty ( & wrapS , 1 , AI_MATKEY_MAPPINGMODE_U ( texType , texSlot ) ) ;
mat - > AddProperty ( & wrapT , 1 , AI_MATKEY_MAPPINGMODE_V ( texType , texSlot ) ) ;
if ( sampler - > magFilter ! = SamplerMagFilter : : UNSET ) {
mat - > AddProperty ( & sampler - > magFilter , 1 , AI_MATKEY_GLTF_MAPPINGFILTER_MAG ( texType , texSlot ) ) ;
}
if ( sampler - > minFilter ! = SamplerMinFilter : : UNSET ) {
mat - > AddProperty ( & sampler - > minFilter , 1 , AI_MATKEY_GLTF_MAPPINGFILTER_MIN ( texType , texSlot ) ) ;
}
2021-06-11 13:34:42 +00:00
} else {
// Use glTFv2 default sampler
const aiTextureMapMode default_wrap = aiTextureMapMode_Wrap ;
mat - > AddProperty ( & default_wrap , 1 , AI_MATKEY_MAPPINGMODE_U ( texType , texSlot ) ) ;
mat - > AddProperty ( & default_wrap , 1 , AI_MATKEY_MAPPINGMODE_V ( texType , texSlot ) ) ;
2020-10-16 15:09:17 +00:00
}
}
2017-08-24 21:15:48 +00:00
}
2019-11-16 07:08:57 +00:00
inline void SetMaterialTextureProperty ( std : : vector < int > & embeddedTexIdxs , Asset & r , glTF2 : : NormalTextureInfo & prop , aiMaterial * mat , aiTextureType texType , unsigned int texSlot = 0 ) {
2020-10-16 15:09:17 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , ( glTF2 : : TextureInfo ) prop , mat , texType , texSlot ) ;
2018-01-08 23:19:27 +00:00
2020-10-16 15:09:17 +00:00
if ( prop . texture & & prop . texture - > source ) {
mat - > AddProperty ( & prop . scale , 1 , AI_MATKEY_GLTF_TEXTURE_SCALE ( texType , texSlot ) ) ;
}
2018-01-08 23:19:27 +00:00
}
2019-11-16 07:08:57 +00:00
inline void SetMaterialTextureProperty ( std : : vector < int > & embeddedTexIdxs , Asset & r , glTF2 : : OcclusionTextureInfo & prop , aiMaterial * mat , aiTextureType texType , unsigned int texSlot = 0 ) {
2020-10-16 15:09:17 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , ( glTF2 : : TextureInfo ) prop , mat , texType , texSlot ) ;
2018-01-08 23:19:27 +00:00
2020-10-16 15:09:17 +00:00
if ( prop . texture & & prop . texture - > source ) {
mat - > AddProperty ( & prop . strength , 1 , AI_MATKEY_GLTF_TEXTURE_STRENGTH ( texType , texSlot ) ) ;
}
2018-01-08 23:19:27 +00:00
}
2019-11-16 07:08:57 +00:00
static aiMaterial * ImportMaterial ( std : : vector < int > & embeddedTexIdxs , Asset & r , Material & mat ) {
2020-10-16 15:09:17 +00:00
aiMaterial * aimat = new aiMaterial ( ) ;
2017-08-24 21:15:48 +00:00
2020-12-28 07:48:54 +00:00
try {
if ( ! mat . name . empty ( ) ) {
aiString str ( mat . name ) ;
2017-08-24 21:15:48 +00:00
2020-12-28 07:48:54 +00:00
aimat - > AddProperty ( & str , AI_MATKEY_NAME ) ;
}
2017-08-24 21:15:48 +00:00
2021-06-10 17:13:46 +00:00
// Set Assimp DIFFUSE and BASE COLOR to the pbrMetallicRoughness base color and texture for backwards compatibility
// Technically should not load any pbrMetallicRoughness if extensionsRequired contains KHR_materials_pbrSpecularGlossiness
2020-12-28 07:48:54 +00:00
SetMaterialColorProperty ( r , mat . pbrMetallicRoughness . baseColorFactor , aimat , AI_MATKEY_COLOR_DIFFUSE ) ;
2021-06-10 17:13:46 +00:00
SetMaterialColorProperty ( r , mat . pbrMetallicRoughness . baseColorFactor , aimat , AI_MATKEY_BASE_COLOR ) ;
2017-09-04 02:11:20 +00:00
2020-12-28 07:48:54 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , mat . pbrMetallicRoughness . baseColorTexture , aimat , aiTextureType_DIFFUSE ) ;
2021-06-10 17:13:46 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , mat . pbrMetallicRoughness . baseColorTexture , aimat , aiTextureType_BASE_COLOR ) ;
2017-08-24 21:15:48 +00:00
2020-12-28 07:48:54 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , mat . pbrMetallicRoughness . metallicRoughnessTexture , aimat , AI_MATKEY_GLTF_PBRMETALLICROUGHNESS_METALLICROUGHNESS_TEXTURE ) ;
2017-10-19 16:01:40 +00:00
2021-06-10 17:13:46 +00:00
aimat - > AddProperty ( & mat . pbrMetallicRoughness . metallicFactor , 1 , AI_MATKEY_METALLIC_FACTOR ) ;
aimat - > AddProperty ( & mat . pbrMetallicRoughness . roughnessFactor , 1 , AI_MATKEY_ROUGHNESS_FACTOR ) ;
2017-10-19 16:01:40 +00:00
2020-12-28 07:48:54 +00:00
float roughnessAsShininess = 1 - mat . pbrMetallicRoughness . roughnessFactor ;
roughnessAsShininess * = roughnessAsShininess * 1000 ;
aimat - > AddProperty ( & roughnessAsShininess , 1 , AI_MATKEY_SHININESS ) ;
2017-10-19 16:30:32 +00:00
2020-12-28 07:48:54 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , mat . normalTexture , aimat , aiTextureType_NORMALS ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , mat . occlusionTexture , aimat , aiTextureType_LIGHTMAP ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , mat . emissiveTexture , aimat , aiTextureType_EMISSIVE ) ;
SetMaterialColorProperty ( r , mat . emissiveFactor , aimat , AI_MATKEY_COLOR_EMISSIVE ) ;
2017-08-24 21:15:48 +00:00
2020-12-28 07:48:54 +00:00
aimat - > AddProperty ( & mat . doubleSided , 1 , AI_MATKEY_TWOSIDED ) ;
2021-06-24 17:25:44 +00:00
aimat - > AddProperty ( & mat . pbrMetallicRoughness . baseColorFactor [ 3 ] , 1 , AI_MATKEY_OPACITY ) ;
2017-10-19 16:30:32 +00:00
2020-12-28 07:48:54 +00:00
aiString alphaMode ( mat . alphaMode ) ;
aimat - > AddProperty ( & alphaMode , AI_MATKEY_GLTF_ALPHAMODE ) ;
aimat - > AddProperty ( & mat . alphaCutoff , 1 , AI_MATKEY_GLTF_ALPHACUTOFF ) ;
2017-08-30 21:25:11 +00:00
2020-12-28 07:48:54 +00:00
//pbrSpecularGlossiness
if ( mat . pbrSpecularGlossiness . isPresent ) {
PbrSpecularGlossiness & pbrSG = mat . pbrSpecularGlossiness . value ;
2017-09-18 16:19:55 +00:00
2020-12-28 07:48:54 +00:00
SetMaterialColorProperty ( r , pbrSG . diffuseFactor , aimat , AI_MATKEY_COLOR_DIFFUSE ) ;
SetMaterialColorProperty ( r , pbrSG . specularFactor , aimat , AI_MATKEY_COLOR_SPECULAR ) ;
2017-08-31 05:35:10 +00:00
2020-12-28 07:48:54 +00:00
float glossinessAsShininess = pbrSG . glossinessFactor * 1000.0f ;
aimat - > AddProperty ( & glossinessAsShininess , 1 , AI_MATKEY_SHININESS ) ;
2021-06-10 17:13:46 +00:00
aimat - > AddProperty ( & pbrSG . glossinessFactor , 1 , AI_MATKEY_GLOSSINESS_FACTOR ) ;
2017-09-05 22:04:09 +00:00
2020-12-28 07:48:54 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , pbrSG . diffuseTexture , aimat , aiTextureType_DIFFUSE ) ;
2017-10-19 16:30:32 +00:00
2020-12-28 07:48:54 +00:00
SetMaterialTextureProperty ( embeddedTexIdxs , r , pbrSG . specularGlossinessTexture , aimat , aiTextureType_SPECULAR ) ;
}
2021-06-10 17:13:46 +00:00
// glTFv2 is either PBR or Unlit
aiShadingMode shadingMode = aiShadingMode_PBR_BRDF ;
2020-12-28 07:48:54 +00:00
if ( mat . unlit ) {
2021-06-10 17:13:46 +00:00
shadingMode = aiShadingMode_Unlit ;
2020-12-28 07:48:54 +00:00
}
2020-12-23 09:43:01 +00:00
2021-06-10 17:13:46 +00:00
aimat - > AddProperty ( & shadingMode , 1 , AI_MATKEY_SHADING_MODEL ) ;
2021-06-11 12:44:52 +00:00
// KHR_materials_sheen
2020-12-28 07:48:54 +00:00
if ( mat . materialSheen . isPresent ) {
MaterialSheen & sheen = mat . materialSheen . value ;
2021-06-11 12:44:52 +00:00
// Default value {0,0,0} disables Sheen
if ( sheen . sheenColorFactor ! = defaultSheenFactor ) {
SetMaterialColorProperty ( r , sheen . sheenColorFactor , aimat , AI_MATKEY_SHEEN_COLOR_FACTOR ) ;
aimat - > AddProperty ( & sheen . sheenRoughnessFactor , 1 , AI_MATKEY_SHEEN_ROUGHNESS_FACTOR ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , sheen . sheenColorTexture , aimat , AI_MATKEY_SHEEN_COLOR_TEXTURE ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , sheen . sheenRoughnessTexture , aimat , AI_MATKEY_SHEEN_ROUGHNESS_TEXTURE ) ;
}
2020-12-28 07:48:54 +00:00
}
2021-06-11 12:44:52 +00:00
// KHR_materials_clearcoat
2020-12-28 07:48:54 +00:00
if ( mat . materialClearcoat . isPresent ) {
MaterialClearcoat & clearcoat = mat . materialClearcoat . value ;
2021-06-11 12:44:52 +00:00
// Default value 0.0 disables clearcoat
if ( clearcoat . clearcoatFactor ! = 0.0f ) {
aimat - > AddProperty ( & clearcoat . clearcoatFactor , 1 , AI_MATKEY_CLEARCOAT_FACTOR ) ;
aimat - > AddProperty ( & clearcoat . clearcoatRoughnessFactor , 1 , AI_MATKEY_CLEARCOAT_ROUGHNESS_FACTOR ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , clearcoat . clearcoatTexture , aimat , AI_MATKEY_CLEARCOAT_TEXTURE ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , clearcoat . clearcoatRoughnessTexture , aimat , AI_MATKEY_CLEARCOAT_ROUGHNESS_TEXTURE ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , clearcoat . clearcoatNormalTexture , aimat , AI_MATKEY_CLEARCOAT_NORMAL_TEXTURE ) ;
}
2020-12-28 07:48:54 +00:00
}
2021-06-11 12:44:52 +00:00
// KHR_materials_transmission
2020-12-28 07:48:54 +00:00
if ( mat . materialTransmission . isPresent ) {
MaterialTransmission & transmission = mat . materialTransmission . value ;
2017-10-19 16:24:25 +00:00
2021-06-11 12:44:52 +00:00
aimat - > AddProperty ( & transmission . transmissionFactor , 1 , AI_MATKEY_TRANSMISSION_FACTOR ) ;
SetMaterialTextureProperty ( embeddedTexIdxs , r , transmission . transmissionTexture , aimat , AI_MATKEY_TRANSMISSION_TEXTURE ) ;
2020-12-28 07:48:54 +00:00
}
return aimat ;
} catch ( . . . ) {
delete aimat ;
throw ;
}
2018-01-08 22:57:55 +00:00
}
2017-10-19 16:24:25 +00:00
2019-11-16 07:08:57 +00:00
void glTF2Importer : : ImportMaterials ( glTF2 : : Asset & r ) {
2020-10-16 15:09:17 +00:00
const unsigned int numImportedMaterials = unsigned ( r . materials . Size ( ) ) ;
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_DEBUG ( " Importing " , numImportedMaterials , " materials " ) ;
2020-10-16 15:09:17 +00:00
Material defaultMaterial ;
2018-01-08 22:57:55 +00:00
2020-10-16 15:09:17 +00:00
mScene - > mNumMaterials = numImportedMaterials + 1 ;
mScene - > mMaterials = new aiMaterial * [ mScene - > mNumMaterials ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( mScene - > mMaterials , mScene - > mMaterials + mScene - > mNumMaterials , nullptr ) ;
2020-10-16 15:09:17 +00:00
mScene - > mMaterials [ numImportedMaterials ] = ImportMaterial ( embeddedTexIdxs , r , defaultMaterial ) ;
2018-01-08 22:57:55 +00:00
2020-10-16 15:09:17 +00:00
for ( unsigned int i = 0 ; i < numImportedMaterials ; + + i ) {
mScene - > mMaterials [ i ] = ImportMaterial ( embeddedTexIdxs , r , r . materials [ i ] ) ;
}
2017-08-24 21:15:48 +00:00
}
2020-07-15 11:22:28 +00:00
static inline void SetFaceAndAdvance1 ( aiFace * & face , unsigned int numVertices , unsigned int a ) {
2020-10-16 15:09:17 +00:00
if ( a > = numVertices ) {
return ;
}
face - > mNumIndices = 1 ;
face - > mIndices = new unsigned int [ 1 ] ;
face - > mIndices [ 0 ] = a ;
+ + face ;
2017-08-24 21:15:48 +00:00
}
2020-07-15 11:22:28 +00:00
static inline void SetFaceAndAdvance2 ( aiFace * & face , unsigned int numVertices , unsigned int a , unsigned int b ) {
2020-10-16 15:09:17 +00:00
if ( ( a > = numVertices ) | | ( b > = numVertices ) ) {
return ;
}
face - > mNumIndices = 2 ;
face - > mIndices = new unsigned int [ 2 ] ;
face - > mIndices [ 0 ] = a ;
face - > mIndices [ 1 ] = b ;
+ + face ;
2017-08-24 21:15:48 +00:00
}
2020-07-15 11:22:28 +00:00
static inline void SetFaceAndAdvance3 ( aiFace * & face , unsigned int numVertices , unsigned int a , unsigned int b , unsigned int c ) {
2020-10-16 15:09:17 +00:00
if ( ( a > = numVertices ) | | ( b > = numVertices ) | | ( c > = numVertices ) ) {
return ;
}
face - > mNumIndices = 3 ;
face - > mIndices = new unsigned int [ 3 ] ;
face - > mIndices [ 0 ] = a ;
face - > mIndices [ 1 ] = b ;
face - > mIndices [ 2 ] = c ;
+ + face ;
2017-08-24 21:15:48 +00:00
}
2017-11-02 10:13:52 +00:00
# ifdef ASSIMP_BUILD_DEBUG
2019-11-16 07:08:57 +00:00
static inline bool CheckValidFacesIndices ( aiFace * faces , unsigned nFaces , unsigned nVerts ) {
2020-10-16 15:09:17 +00:00
for ( unsigned i = 0 ; i < nFaces ; + + i ) {
for ( unsigned j = 0 ; j < faces [ i ] . mNumIndices ; + + j ) {
unsigned idx = faces [ i ] . mIndices [ j ] ;
if ( idx > = nVerts ) {
return false ;
}
}
}
return true ;
2017-08-24 21:15:48 +00:00
}
2017-11-02 10:13:52 +00:00
# endif // ASSIMP_BUILD_DEBUG
2017-08-24 21:15:48 +00:00
2021-01-15 02:56:45 +00:00
template < typename T >
aiColor4D * GetVertexColorsForType ( glTF2 : : Ref < glTF2 : : Accessor > input ) {
float max = std : : numeric_limits < T > : : max ( ) ;
aiColor4t < T > * colors ;
input - > ExtractData ( colors ) ;
auto output = new aiColor4D [ input - > count ] ;
for ( size_t i = 0 ; i < input - > count ; i + + ) {
output [ i ] = aiColor4D (
colors [ i ] . r / max , colors [ i ] . g / max ,
colors [ i ] . b / max , colors [ i ] . a / max
) ;
}
delete [ ] colors ;
return output ;
}
2019-11-16 07:08:57 +00:00
void glTF2Importer : : ImportMeshes ( glTF2 : : Asset & r ) {
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_DEBUG ( " Importing " , r . meshes . Size ( ) , " meshes " ) ;
2020-10-16 15:09:17 +00:00
std : : vector < std : : unique_ptr < aiMesh > > meshes ;
2019-11-16 07:08:57 +00:00
2020-10-16 15:09:17 +00:00
unsigned int k = 0 ;
2020-02-14 07:41:55 +00:00
meshOffsets . clear ( ) ;
2019-11-16 07:08:57 +00:00
2020-10-16 15:09:17 +00:00
for ( unsigned int m = 0 ; m < r . meshes . Size ( ) ; + + m ) {
Mesh & mesh = r . meshes [ m ] ;
meshOffsets . push_back ( k ) ;
k + = unsigned ( mesh . primitives . size ( ) ) ;
for ( unsigned int p = 0 ; p < mesh . primitives . size ( ) ; + + p ) {
Mesh : : Primitive & prim = mesh . primitives [ p ] ;
aiMesh * aim = new aiMesh ( ) ;
meshes . push_back ( std : : unique_ptr < aiMesh > ( aim ) ) ;
aim - > mName = mesh . name . empty ( ) ? mesh . id : mesh . name ;
if ( mesh . primitives . size ( ) > 1 ) {
ai_uint32 & len = aim - > mName . length ;
aim - > mName . data [ len ] = ' - ' ;
len + = 1 + ASSIMP_itoa10 ( aim - > mName . data + len + 1 , unsigned ( MAXLEN - len - 1 ) , p ) ;
}
switch ( prim . mode ) {
case PrimitiveMode_POINTS :
aim - > mPrimitiveTypes | = aiPrimitiveType_POINT ;
break ;
case PrimitiveMode_LINES :
case PrimitiveMode_LINE_LOOP :
case PrimitiveMode_LINE_STRIP :
aim - > mPrimitiveTypes | = aiPrimitiveType_LINE ;
break ;
case PrimitiveMode_TRIANGLES :
case PrimitiveMode_TRIANGLE_STRIP :
case PrimitiveMode_TRIANGLE_FAN :
aim - > mPrimitiveTypes | = aiPrimitiveType_TRIANGLE ;
break ;
}
Mesh : : Primitive : : Attributes & attr = prim . attributes ;
if ( attr . position . size ( ) > 0 & & attr . position [ 0 ] ) {
aim - > mNumVertices = static_cast < unsigned int > ( attr . position [ 0 ] - > count ) ;
attr . position [ 0 ] - > ExtractData ( aim - > mVertices ) ;
}
if ( attr . normal . size ( ) > 0 & & attr . normal [ 0 ] ) {
2021-02-24 14:19:29 +00:00
if ( attr . normal [ 0 ] - > count ! = aim - > mNumVertices ) {
2021-05-13 11:05:31 +00:00
DefaultLogger : : get ( ) - > warn ( " Normal count in mesh \" " , mesh . name , " \" does not match the vertex count, normals ignored. " ) ;
2021-02-24 14:19:29 +00:00
} else {
attr . normal [ 0 ] - > ExtractData ( aim - > mNormals ) ;
// only extract tangents if normals are present
if ( attr . tangent . size ( ) > 0 & & attr . tangent [ 0 ] ) {
if ( attr . tangent [ 0 ] - > count ! = aim - > mNumVertices ) {
2021-05-13 11:05:31 +00:00
DefaultLogger : : get ( ) - > warn ( " Tangent count in mesh \" " , mesh . name , " \" does not match the vertex count, tangents ignored. " ) ;
2021-02-24 14:19:29 +00:00
} else {
// generate bitangents from normals and tangents according to spec
Tangent * tangents = nullptr ;
attr . tangent [ 0 ] - > ExtractData ( tangents ) ;
aim - > mTangents = new aiVector3D [ aim - > mNumVertices ] ;
aim - > mBitangents = new aiVector3D [ aim - > mNumVertices ] ;
for ( unsigned int i = 0 ; i < aim - > mNumVertices ; + + i ) {
aim - > mTangents [ i ] = tangents [ i ] . xyz ;
aim - > mBitangents [ i ] = ( aim - > mNormals [ i ] ^ tangents [ i ] . xyz ) * tangents [ i ] . w ;
}
delete [ ] tangents ;
}
2020-10-16 15:09:17 +00:00
}
}
}
for ( size_t c = 0 ; c < attr . color . size ( ) & & c < AI_MAX_NUMBER_OF_COLOR_SETS ; + + c ) {
if ( attr . color [ c ] - > count ! = aim - > mNumVertices ) {
2021-05-13 11:05:31 +00:00
DefaultLogger : : get ( ) - > warn ( " Color stream size in mesh \" " , mesh . name ,
2020-10-16 15:09:17 +00:00
" \" does not match the vertex count " ) ;
continue ;
}
2021-06-12 03:16:53 +00:00
2021-01-15 02:56:45 +00:00
auto componentType = attr . color [ c ] - > componentType ;
if ( componentType = = glTF2 : : ComponentType_FLOAT ) {
attr . color [ c ] - > ExtractData ( aim - > mColors [ c ] ) ;
} else {
if ( componentType = = glTF2 : : ComponentType_UNSIGNED_BYTE ) {
aim - > mColors [ c ] = GetVertexColorsForType < unsigned char > ( attr . color [ c ] ) ;
} else if ( componentType = = glTF2 : : ComponentType_UNSIGNED_SHORT ) {
aim - > mColors [ c ] = GetVertexColorsForType < unsigned short > ( attr . color [ c ] ) ;
}
}
2020-10-16 15:09:17 +00:00
}
for ( size_t tc = 0 ; tc < attr . texcoord . size ( ) & & tc < AI_MAX_NUMBER_OF_TEXTURECOORDS ; + + tc ) {
2020-06-09 05:49:38 +00:00
if ( ! attr . texcoord [ tc ] ) {
2020-06-12 12:52:02 +00:00
DefaultLogger : : get ( ) - > warn ( " Texture coordinate accessor not found or non-contiguous texture coordinate sets. " ) ;
continue ;
2020-06-09 05:49:38 +00:00
}
2020-10-16 15:09:17 +00:00
if ( attr . texcoord [ tc ] - > count ! = aim - > mNumVertices ) {
2021-05-13 11:05:31 +00:00
DefaultLogger : : get ( ) - > warn ( " Texcoord stream size in mesh \" " , mesh . name ,
2020-10-16 15:09:17 +00:00
" \" does not match the vertex count " ) ;
continue ;
}
attr . texcoord [ tc ] - > ExtractData ( aim - > mTextureCoords [ tc ] ) ;
aim - > mNumUVComponents [ tc ] = attr . texcoord [ tc ] - > GetNumComponents ( ) ;
aiVector3D * values = aim - > mTextureCoords [ tc ] ;
for ( unsigned int i = 0 ; i < aim - > mNumVertices ; + + i ) {
values [ i ] . y = 1 - values [ i ] . y ; // Flip Y coords
}
}
std : : vector < Mesh : : Primitive : : Target > & targets = prim . targets ;
if ( targets . size ( ) > 0 ) {
aim - > mNumAnimMeshes = ( unsigned int ) targets . size ( ) ;
aim - > mAnimMeshes = new aiAnimMesh * [ aim - > mNumAnimMeshes ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( aim - > mAnimMeshes , aim - > mAnimMeshes + aim - > mNumAnimMeshes , nullptr ) ;
2020-10-16 15:09:17 +00:00
for ( size_t i = 0 ; i < targets . size ( ) ; i + + ) {
2020-11-03 14:59:30 +00:00
bool needPositions = targets [ i ] . position . size ( ) > 0 ;
2021-03-24 11:04:39 +00:00
bool needNormals = ( targets [ i ] . normal . size ( ) > 0 ) & & aim - > HasNormals ( ) ;
bool needTangents = ( targets [ i ] . tangent . size ( ) > 0 ) & & aim - > HasTangentsAndBitangents ( ) ;
2020-11-03 14:59:30 +00:00
// GLTF morph does not support colors and texCoords
aim - > mAnimMeshes [ i ] = aiCreateAnimMesh ( aim ,
needPositions , needNormals , needTangents , false , false ) ;
2020-10-16 15:09:17 +00:00
aiAnimMesh & aiAnimMesh = * ( aim - > mAnimMeshes [ i ] ) ;
Mesh : : Primitive : : Target & target = targets [ i ] ;
2020-11-03 14:59:30 +00:00
if ( needPositions ) {
2021-03-24 10:55:40 +00:00
if ( target . position [ 0 ] - > count ! = aim - > mNumVertices ) {
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_WARN ( " Positions of target " , i , " in mesh \" " , mesh . name , " \" does not match the vertex count " ) ;
2021-03-24 10:55:40 +00:00
} else {
aiVector3D * positionDiff = nullptr ;
target . position [ 0 ] - > ExtractData ( positionDiff ) ;
for ( unsigned int vertexId = 0 ; vertexId < aim - > mNumVertices ; vertexId + + ) {
aiAnimMesh . mVertices [ vertexId ] + = positionDiff [ vertexId ] ;
}
delete [ ] positionDiff ;
2020-10-16 15:09:17 +00:00
}
}
2020-11-03 14:59:30 +00:00
if ( needNormals ) {
2021-03-24 10:55:40 +00:00
if ( target . normal [ 0 ] - > count ! = aim - > mNumVertices ) {
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_WARN ( " Normals of target " , i , " in mesh \" " , mesh . name , " \" does not match the vertex count " ) ;
2021-03-24 10:55:40 +00:00
} else {
aiVector3D * normalDiff = nullptr ;
target . normal [ 0 ] - > ExtractData ( normalDiff ) ;
for ( unsigned int vertexId = 0 ; vertexId < aim - > mNumVertices ; vertexId + + ) {
aiAnimMesh . mNormals [ vertexId ] + = normalDiff [ vertexId ] ;
}
delete [ ] normalDiff ;
2020-10-16 15:09:17 +00:00
}
}
2020-11-03 14:59:30 +00:00
if ( needTangents ) {
2021-03-24 10:55:40 +00:00
if ( target . tangent [ 0 ] - > count ! = aim - > mNumVertices ) {
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_WARN ( " Tangents of target " , i , " in mesh \" " , mesh . name , " \" does not match the vertex count " ) ;
2021-03-24 10:55:40 +00:00
} else {
Tangent * tangent = nullptr ;
attr . tangent [ 0 ] - > ExtractData ( tangent ) ;
2020-10-16 15:09:17 +00:00
2021-03-24 10:55:40 +00:00
aiVector3D * tangentDiff = nullptr ;
target . tangent [ 0 ] - > ExtractData ( tangentDiff ) ;
2020-10-16 15:09:17 +00:00
2021-03-24 10:55:40 +00:00
for ( unsigned int vertexId = 0 ; vertexId < aim - > mNumVertices ; + + vertexId ) {
tangent [ vertexId ] . xyz + = tangentDiff [ vertexId ] ;
aiAnimMesh . mTangents [ vertexId ] = tangent [ vertexId ] . xyz ;
aiAnimMesh . mBitangents [ vertexId ] = ( aiAnimMesh . mNormals [ vertexId ] ^ tangent [ vertexId ] . xyz ) * tangent [ vertexId ] . w ;
}
delete [ ] tangent ;
delete [ ] tangentDiff ;
2020-10-16 15:09:17 +00:00
}
}
if ( mesh . weights . size ( ) > i ) {
aiAnimMesh . mWeight = mesh . weights [ i ] ;
}
if ( mesh . targetNames . size ( ) > i ) {
aiAnimMesh . mName = mesh . targetNames [ i ] ;
}
}
}
aiFace * faces = nullptr ;
aiFace * facePtr = nullptr ;
size_t nFaces = 0 ;
if ( prim . indices ) {
size_t count = prim . indices - > count ;
Accessor : : Indexer data = prim . indices - > GetIndexer ( ) ;
2020-12-28 07:48:54 +00:00
if ( ! data . IsValid ( ) ) {
throw DeadlyImportError ( " GLTF: Invalid accessor without data in mesh " , getContextForErrorMessages ( mesh . id , mesh . name ) ) ;
}
2020-10-16 15:09:17 +00:00
switch ( prim . mode ) {
case PrimitiveMode_POINTS : {
nFaces = count ;
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < count ; + + i ) {
SetFaceAndAdvance1 ( facePtr , aim - > mNumVertices , data . GetUInt ( i ) ) ;
}
break ;
}
case PrimitiveMode_LINES : {
nFaces = count / 2 ;
if ( nFaces * 2 ! = count ) {
ASSIMP_LOG_WARN ( " The number of vertices was not compatible with the LINES mode. Some vertices were dropped. " ) ;
count = nFaces * 2 ;
}
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < count ; i + = 2 ) {
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , data . GetUInt ( i ) , data . GetUInt ( i + 1 ) ) ;
}
break ;
}
case PrimitiveMode_LINE_LOOP :
case PrimitiveMode_LINE_STRIP : {
nFaces = count - ( ( prim . mode = = PrimitiveMode_LINE_STRIP ) ? 1 : 0 ) ;
facePtr = faces = new aiFace [ nFaces ] ;
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , data . GetUInt ( 0 ) , data . GetUInt ( 1 ) ) ;
for ( unsigned int i = 2 ; i < count ; + + i ) {
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , data . GetUInt ( i - 1 ) , data . GetUInt ( i ) ) ;
}
if ( prim . mode = = PrimitiveMode_LINE_LOOP ) { // close the loop
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , data . GetUInt ( static_cast < int > ( count ) - 1 ) , faces [ 0 ] . mIndices [ 0 ] ) ;
}
break ;
}
case PrimitiveMode_TRIANGLES : {
nFaces = count / 3 ;
if ( nFaces * 3 ! = count ) {
ASSIMP_LOG_WARN ( " The number of vertices was not compatible with the TRIANGLES mode. Some vertices were dropped. " ) ;
count = nFaces * 3 ;
}
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < count ; i + = 3 ) {
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , data . GetUInt ( i ) , data . GetUInt ( i + 1 ) , data . GetUInt ( i + 2 ) ) ;
}
break ;
}
case PrimitiveMode_TRIANGLE_STRIP : {
nFaces = count - 2 ;
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < nFaces ; + + i ) {
//The ordering is to ensure that the triangles are all drawn with the same orientation
if ( ( i + 1 ) % 2 = = 0 ) {
//For even n, vertices n + 1, n, and n + 2 define triangle n
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , data . GetUInt ( i + 1 ) , data . GetUInt ( i ) , data . GetUInt ( i + 2 ) ) ;
} else {
//For odd n, vertices n, n+1, and n+2 define triangle n
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , data . GetUInt ( i ) , data . GetUInt ( i + 1 ) , data . GetUInt ( i + 2 ) ) ;
}
}
break ;
}
case PrimitiveMode_TRIANGLE_FAN :
nFaces = count - 2 ;
facePtr = faces = new aiFace [ nFaces ] ;
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , data . GetUInt ( 0 ) , data . GetUInt ( 1 ) , data . GetUInt ( 2 ) ) ;
for ( unsigned int i = 1 ; i < nFaces ; + + i ) {
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , data . GetUInt ( 0 ) , data . GetUInt ( i + 1 ) , data . GetUInt ( i + 2 ) ) ;
}
break ;
}
} else { // no indices provided so directly generate from counts
// use the already determined count as it includes checks
unsigned int count = aim - > mNumVertices ;
switch ( prim . mode ) {
case PrimitiveMode_POINTS : {
nFaces = count ;
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < count ; + + i ) {
SetFaceAndAdvance1 ( facePtr , aim - > mNumVertices , i ) ;
}
break ;
}
case PrimitiveMode_LINES : {
nFaces = count / 2 ;
if ( nFaces * 2 ! = count ) {
ASSIMP_LOG_WARN ( " The number of vertices was not compatible with the LINES mode. Some vertices were dropped. " ) ;
count = ( unsigned int ) nFaces * 2 ;
}
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < count ; i + = 2 ) {
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , i , i + 1 ) ;
}
break ;
}
case PrimitiveMode_LINE_LOOP :
case PrimitiveMode_LINE_STRIP : {
nFaces = count - ( ( prim . mode = = PrimitiveMode_LINE_STRIP ) ? 1 : 0 ) ;
facePtr = faces = new aiFace [ nFaces ] ;
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , 0 , 1 ) ;
for ( unsigned int i = 2 ; i < count ; + + i ) {
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , i - 1 , i ) ;
}
if ( prim . mode = = PrimitiveMode_LINE_LOOP ) { // close the loop
SetFaceAndAdvance2 ( facePtr , aim - > mNumVertices , count - 1 , 0 ) ;
}
break ;
}
case PrimitiveMode_TRIANGLES : {
nFaces = count / 3 ;
if ( nFaces * 3 ! = count ) {
ASSIMP_LOG_WARN ( " The number of vertices was not compatible with the TRIANGLES mode. Some vertices were dropped. " ) ;
count = ( unsigned int ) nFaces * 3 ;
}
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < count ; i + = 3 ) {
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , i , i + 1 , i + 2 ) ;
}
break ;
}
case PrimitiveMode_TRIANGLE_STRIP : {
nFaces = count - 2 ;
facePtr = faces = new aiFace [ nFaces ] ;
for ( unsigned int i = 0 ; i < nFaces ; + + i ) {
//The ordering is to ensure that the triangles are all drawn with the same orientation
if ( ( i + 1 ) % 2 = = 0 ) {
//For even n, vertices n + 1, n, and n + 2 define triangle n
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , i + 1 , i , i + 2 ) ;
} else {
//For odd n, vertices n, n+1, and n+2 define triangle n
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , i , i + 1 , i + 2 ) ;
}
}
break ;
}
case PrimitiveMode_TRIANGLE_FAN :
nFaces = count - 2 ;
facePtr = faces = new aiFace [ nFaces ] ;
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , 0 , 1 , 2 ) ;
for ( unsigned int i = 1 ; i < nFaces ; + + i ) {
SetFaceAndAdvance3 ( facePtr , aim - > mNumVertices , 0 , i + 1 , i + 2 ) ;
}
break ;
}
}
if ( faces ) {
aim - > mFaces = faces ;
2020-07-15 08:12:52 +00:00
const unsigned int actualNumFaces = static_cast < unsigned int > ( facePtr - faces ) ;
2020-10-16 15:09:17 +00:00
if ( actualNumFaces < nFaces ) {
ASSIMP_LOG_WARN ( " Some faces had out-of-range indices. Those faces were dropped. " ) ;
}
if ( actualNumFaces = = 0 )
{
throw DeadlyImportError ( " Mesh \" " , aim - > mName . C_Str ( ) , " \" has no faces " ) ;
}
aim - > mNumFaces = actualNumFaces ;
ai_assert ( CheckValidFacesIndices ( faces , actualNumFaces , aim - > mNumVertices ) ) ;
}
if ( prim . material ) {
aim - > mMaterialIndex = prim . material . GetIndex ( ) ;
} else {
aim - > mMaterialIndex = mScene - > mNumMaterials - 1 ;
}
}
}
meshOffsets . push_back ( k ) ;
CopyVector ( meshes , mScene - > mMeshes , mScene - > mNumMeshes ) ;
2017-08-24 21:15:48 +00:00
}
2019-11-16 07:08:57 +00:00
void glTF2Importer : : ImportCameras ( glTF2 : : Asset & r ) {
2020-10-16 15:09:17 +00:00
if ( ! r . cameras . Size ( ) ) return ;
2019-11-16 07:08:57 +00:00
2020-05-14 16:43:31 +00:00
const unsigned int numCameras = r . cameras . Size ( ) ;
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_DEBUG ( " Importing " , numCameras , " cameras " ) ;
2020-10-16 15:09:17 +00:00
mScene - > mNumCameras = numCameras ;
mScene - > mCameras = new aiCamera * [ numCameras ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( mScene - > mCameras , mScene - > mCameras + numCameras , nullptr ) ;
2020-10-16 15:09:17 +00:00
for ( size_t i = 0 ; i < numCameras ; + + i ) {
Camera & cam = r . cameras [ i ] ;
aiCamera * aicam = mScene - > mCameras [ i ] = new aiCamera ( ) ;
// cameras point in -Z by default, rest is specified in node transform
aicam - > mLookAt = aiVector3D ( 0.f , 0.f , - 1.f ) ;
if ( cam . type = = Camera : : Perspective ) {
aicam - > mAspect = cam . cameraProperties . perspective . aspectRatio ;
aicam - > mHorizontalFOV = cam . cameraProperties . perspective . yfov * ( ( aicam - > mAspect = = 0.f ) ? 1.f : aicam - > mAspect ) ;
aicam - > mClipPlaneFar = cam . cameraProperties . perspective . zfar ;
aicam - > mClipPlaneNear = cam . cameraProperties . perspective . znear ;
} else {
aicam - > mClipPlaneFar = cam . cameraProperties . ortographic . zfar ;
aicam - > mClipPlaneNear = cam . cameraProperties . ortographic . znear ;
aicam - > mHorizontalFOV = 0.0 ;
aicam - > mOrthographicWidth = cam . cameraProperties . ortographic . xmag ;
aicam - > mAspect = 1.0f ;
if ( 0.f ! = cam . cameraProperties . ortographic . ymag ) {
aicam - > mAspect = cam . cameraProperties . ortographic . xmag / cam . cameraProperties . ortographic . ymag ;
}
}
}
2017-08-24 21:15:48 +00:00
}
2019-11-16 07:08:57 +00:00
void glTF2Importer : : ImportLights ( glTF2 : : Asset & r ) {
2020-10-16 15:09:17 +00:00
if ( ! r . lights . Size ( ) )
return ;
2019-11-16 07:08:57 +00:00
2020-05-14 16:43:31 +00:00
const unsigned int numLights = r . lights . Size ( ) ;
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_DEBUG ( " Importing " , numLights , " lights " ) ;
2020-10-16 15:09:17 +00:00
mScene - > mNumLights = numLights ;
mScene - > mLights = new aiLight * [ numLights ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( mScene - > mLights , mScene - > mLights + numLights , nullptr ) ;
2020-10-16 15:09:17 +00:00
for ( size_t i = 0 ; i < numLights ; + + i ) {
Light & light = r . lights [ i ] ;
aiLight * ail = mScene - > mLights [ i ] = new aiLight ( ) ;
switch ( light . type ) {
case Light : : Directional :
ail - > mType = aiLightSource_DIRECTIONAL ;
break ;
case Light : : Point :
ail - > mType = aiLightSource_POINT ;
break ;
case Light : : Spot :
ail - > mType = aiLightSource_SPOT ;
break ;
}
if ( ail - > mType ! = aiLightSource_POINT ) {
ail - > mDirection = aiVector3D ( 0.0f , 0.0f , - 1.0f ) ;
ail - > mUp = aiVector3D ( 0.0f , 1.0f , 0.0f ) ;
}
vec3 colorWithIntensity = { light . color [ 0 ] * light . intensity , light . color [ 1 ] * light . intensity , light . color [ 2 ] * light . intensity } ;
CopyValue ( colorWithIntensity , ail - > mColorAmbient ) ;
CopyValue ( colorWithIntensity , ail - > mColorDiffuse ) ;
CopyValue ( colorWithIntensity , ail - > mColorSpecular ) ;
if ( ail - > mType = = aiLightSource_DIRECTIONAL ) {
ail - > mAttenuationConstant = 1.0 ;
ail - > mAttenuationLinear = 0.0 ;
ail - > mAttenuationQuadratic = 0.0 ;
} else {
//in PBR attenuation is calculated using inverse square law which can be expressed
//using assimps equation: 1/(att0 + att1 * d + att2 * d*d) with the following parameters
//this is correct equation for the case when range (see
//https://github.com/KhronosGroup/glTF/tree/master/extensions/2.0/Khronos/KHR_lights_punctual)
//is not present. When range is not present it is assumed that it is infinite and so numerator is 1.
//When range is present then numerator might be any value in range [0,1] and then assimps equation
//will not suffice. In this case range is added into metadata in ImportNode function
//and its up to implementation to read it when it wants to
ail - > mAttenuationConstant = 0.0 ;
ail - > mAttenuationLinear = 0.0 ;
ail - > mAttenuationQuadratic = 1.0 ;
}
if ( ail - > mType = = aiLightSource_SPOT ) {
ail - > mAngleInnerCone = light . innerConeAngle ;
ail - > mAngleOuterCone = light . outerConeAngle ;
}
}
2019-08-03 18:51:00 +00:00
}
2019-11-16 07:08:57 +00:00
static void GetNodeTransform ( aiMatrix4x4 & matrix , const glTF2 : : Node & node ) {
2020-10-16 15:09:17 +00:00
if ( node . matrix . isPresent ) {
CopyValue ( node . matrix . value , matrix ) ;
} else {
if ( node . translation . isPresent ) {
aiVector3D trans ;
CopyValue ( node . translation . value , trans ) ;
aiMatrix4x4 t ;
aiMatrix4x4 : : Translation ( trans , t ) ;
matrix = matrix * t ;
}
if ( node . rotation . isPresent ) {
aiQuaternion rot ;
CopyValue ( node . rotation . value , rot ) ;
matrix = matrix * aiMatrix4x4 ( rot . GetMatrix ( ) ) ;
}
if ( node . scale . isPresent ) {
aiVector3D scal ( 1.f ) ;
CopyValue ( node . scale . value , scal ) ;
aiMatrix4x4 s ;
aiMatrix4x4 : : Scaling ( scal , s ) ;
matrix = matrix * s ;
}
}
2018-10-26 22:36:34 +00:00
}
2017-08-24 21:15:48 +00:00
2019-11-16 07:08:57 +00:00
static void BuildVertexWeightMapping ( Mesh : : Primitive & primitive , std : : vector < std : : vector < aiVertexWeight > > & map ) {
2020-10-16 15:09:17 +00:00
Mesh : : Primitive : : Attributes & attr = primitive . attributes ;
if ( attr . weight . empty ( ) | | attr . joint . empty ( ) ) {
return ;
}
if ( attr . weight [ 0 ] - > count ! = attr . joint [ 0 ] - > count ) {
return ;
}
size_t num_vertices = attr . weight [ 0 ] - > count ;
struct Weights {
float values [ 4 ] ;
} ;
Weights * weights = nullptr ;
attr . weight [ 0 ] - > ExtractData ( weights ) ;
struct Indices8 {
uint8_t values [ 4 ] ;
} ;
struct Indices16 {
uint16_t values [ 4 ] ;
} ;
Indices8 * indices8 = nullptr ;
Indices16 * indices16 = nullptr ;
if ( attr . joint [ 0 ] - > GetElementSize ( ) = = 4 ) {
attr . joint [ 0 ] - > ExtractData ( indices8 ) ;
} else {
attr . joint [ 0 ] - > ExtractData ( indices16 ) ;
}
//
if ( nullptr = = indices8 & & nullptr = = indices16 ) {
// Something went completely wrong!
ai_assert ( false ) ;
return ;
}
for ( size_t i = 0 ; i < num_vertices ; + + i ) {
for ( int j = 0 ; j < 4 ; + + j ) {
const unsigned int bone = ( indices8 ! = nullptr ) ? indices8 [ i ] . values [ j ] : indices16 [ i ] . values [ j ] ;
const float weight = weights [ i ] . values [ j ] ;
if ( weight > 0 & & bone < map . size ( ) ) {
map [ bone ] . reserve ( 8 ) ;
map [ bone ] . emplace_back ( static_cast < unsigned int > ( i ) , weight ) ;
}
}
}
delete [ ] weights ;
delete [ ] indices8 ;
delete [ ] indices16 ;
2018-10-26 22:36:34 +00:00
}
2019-11-16 07:08:57 +00:00
static std : : string GetNodeName ( const Node & node ) {
2020-10-16 15:09:17 +00:00
return node . name . empty ( ) ? node . id : node . name ;
2019-04-24 13:58:35 +00:00
}
2020-06-27 02:53:26 +00:00
void ParseExtensions ( aiMetadata * metadata , const CustomExtension & extension ) {
2020-10-16 15:09:17 +00:00
if ( extension . mStringValue . isPresent ) {
metadata - > Add ( extension . name . c_str ( ) , aiString ( extension . mStringValue . value ) ) ;
} else if ( extension . mDoubleValue . isPresent ) {
metadata - > Add ( extension . name . c_str ( ) , extension . mDoubleValue . value ) ;
} else if ( extension . mUint64Value . isPresent ) {
metadata - > Add ( extension . name . c_str ( ) , extension . mUint64Value . value ) ;
} else if ( extension . mInt64Value . isPresent ) {
metadata - > Add ( extension . name . c_str ( ) , static_cast < int32_t > ( extension . mInt64Value . value ) ) ;
} else if ( extension . mBoolValue . isPresent ) {
metadata - > Add ( extension . name . c_str ( ) , extension . mBoolValue . value ) ;
} else if ( extension . mValues . isPresent ) {
aiMetadata val ;
for ( size_t i = 0 ; i < extension . mValues . value . size ( ) ; + + i ) {
ParseExtensions ( & val , extension . mValues . value [ i ] ) ;
}
metadata - > Add ( extension . name . c_str ( ) , val ) ;
}
2020-06-26 04:26:45 +00:00
}
2021-06-27 04:53:40 +00:00
void ParseExtras ( aiMetadata * metadata , const CustomExtension & extension ) {
if ( extension . mValues . isPresent ) {
for ( size_t i = 0 ; i < extension . mValues . value . size ( ) ; + + i ) {
ParseExtensions ( metadata , extension . mValues . value [ i ] ) ;
}
}
}
2019-11-16 07:08:57 +00:00
aiNode * ImportNode ( aiScene * pScene , glTF2 : : Asset & r , std : : vector < unsigned int > & meshOffsets , glTF2 : : Ref < glTF2 : : Node > & ptr ) {
2020-10-16 15:09:17 +00:00
Node & node = * ptr ;
aiNode * ainode = new aiNode ( GetNodeName ( node ) ) ;
2020-12-28 07:48:54 +00:00
try {
if ( ! node . children . empty ( ) ) {
ainode - > mNumChildren = unsigned ( node . children . size ( ) ) ;
ainode - > mChildren = new aiNode * [ ainode - > mNumChildren ] ;
std : : fill ( ainode - > mChildren , ainode - > mChildren + ainode - > mNumChildren , nullptr ) ;
2020-10-16 15:09:17 +00:00
2020-12-28 07:48:54 +00:00
for ( unsigned int i = 0 ; i < ainode - > mNumChildren ; + + i ) {
aiNode * child = ImportNode ( pScene , r , meshOffsets , node . children [ i ] ) ;
child - > mParent = ainode ;
ainode - > mChildren [ i ] = child ;
}
2020-10-16 15:09:17 +00:00
}
2021-06-27 04:53:40 +00:00
if ( node . customExtensions | | node . extras ) {
2020-12-28 07:48:54 +00:00
ainode - > mMetaData = new aiMetadata ;
2021-06-27 04:53:40 +00:00
if ( node . customExtensions ) {
ParseExtensions ( ainode - > mMetaData , node . customExtensions ) ;
}
if ( node . extras ) {
ParseExtras ( ainode - > mMetaData , node . extras ) ;
}
2020-12-28 07:48:54 +00:00
}
2020-10-16 15:09:17 +00:00
2020-12-28 07:48:54 +00:00
GetNodeTransform ( ainode - > mTransformation , node ) ;
2020-10-16 15:09:17 +00:00
2020-12-28 07:48:54 +00:00
if ( ! node . meshes . empty ( ) ) {
// GLTF files contain at most 1 mesh per node.
if ( node . meshes . size ( ) > 1 )
{
throw DeadlyImportError ( " GLTF: Invalid input, found " , node . meshes . size ( ) , " meshes in " , getContextForErrorMessages ( node . id , node . name ) , " , but only 1 mesh per node allowed. " ) ;
}
int mesh_idx = node . meshes [ 0 ] . GetIndex ( ) ;
int count = meshOffsets [ mesh_idx + 1 ] - meshOffsets [ mesh_idx ] ;
ainode - > mNumMeshes = count ;
ainode - > mMeshes = new unsigned int [ count ] ;
if ( node . skin ) {
for ( int primitiveNo = 0 ; primitiveNo < count ; + + primitiveNo ) {
aiMesh * mesh = pScene - > mMeshes [ meshOffsets [ mesh_idx ] + primitiveNo ] ;
unsigned int numBones = static_cast < unsigned int > ( node . skin - > jointNames . size ( ) ) ;
std : : vector < std : : vector < aiVertexWeight > > weighting ( numBones ) ;
BuildVertexWeightMapping ( node . meshes [ 0 ] - > primitives [ primitiveNo ] , weighting ) ;
mesh - > mNumBones = static_cast < unsigned int > ( numBones ) ;
mesh - > mBones = new aiBone * [ mesh - > mNumBones ] ;
std : : fill ( mesh - > mBones , mesh - > mBones + mesh - > mNumBones , nullptr ) ;
// GLTF and Assimp choose to store bone weights differently.
// GLTF has each vertex specify which bones influence the vertex.
// Assimp has each bone specify which vertices it has influence over.
// To convert this data, we first read over the vertex data and pull
// out the bone-to-vertex mapping. Then, when creating the aiBones,
// we copy the bone-to-vertex mapping into the bone. This is unfortunate
// both because it's somewhat slow and because, for many applications,
// we then need to reconvert the data back into the vertex-to-bone
// mapping which makes things doubly-slow.
mat4 * pbindMatrices = nullptr ;
node . skin - > inverseBindMatrices - > ExtractData ( pbindMatrices ) ;
for ( uint32_t i = 0 ; i < numBones ; + + i ) {
const std : : vector < aiVertexWeight > & weights = weighting [ i ] ;
aiBone * bone = new aiBone ( ) ;
Ref < Node > joint = node . skin - > jointNames [ i ] ;
if ( ! joint - > name . empty ( ) ) {
bone - > mName = joint - > name ;
} else {
// Assimp expects each bone to have a unique name.
static const std : : string kDefaultName = " bone_ " ;
char postfix [ 10 ] = { 0 } ;
ASSIMP_itoa10 ( postfix , i ) ;
bone - > mName = ( kDefaultName + postfix ) ;
}
GetNodeTransform ( bone - > mOffsetMatrix , * joint ) ;
CopyValue ( pbindMatrices [ i ] , bone - > mOffsetMatrix ) ;
bone - > mNumWeights = static_cast < uint32_t > ( weights . size ( ) ) ;
if ( bone - > mNumWeights > 0 ) {
bone - > mWeights = new aiVertexWeight [ bone - > mNumWeights ] ;
memcpy ( bone - > mWeights , weights . data ( ) , bone - > mNumWeights * sizeof ( aiVertexWeight ) ) ;
} else {
// Assimp expects all bones to have at least 1 weight.
bone - > mWeights = new aiVertexWeight [ 1 ] ;
bone - > mNumWeights = 1 ;
bone - > mWeights - > mVertexId = 0 ;
bone - > mWeights - > mWeight = 0.f ;
}
mesh - > mBones [ i ] = bone ;
2020-10-16 15:09:17 +00:00
}
2020-12-28 07:48:54 +00:00
if ( pbindMatrices ) {
delete [ ] pbindMatrices ;
}
2020-10-16 15:09:17 +00:00
}
}
2020-12-28 07:48:54 +00:00
int k = 0 ;
for ( unsigned int j = meshOffsets [ mesh_idx ] ; j < meshOffsets [ mesh_idx + 1 ] ; + + j , + + k ) {
ainode - > mMeshes [ k ] = j ;
}
2020-10-16 15:09:17 +00:00
}
2020-12-28 07:48:54 +00:00
if ( node . camera ) {
pScene - > mCameras [ node . camera . GetIndex ( ) ] - > mName = ainode - > mName ;
if ( node . translation . isPresent ) {
aiVector3D trans ;
CopyValue ( node . translation . value , trans ) ;
pScene - > mCameras [ node . camera . GetIndex ( ) ] - > mPosition = trans ;
}
2020-10-16 15:09:17 +00:00
}
2020-12-28 07:48:54 +00:00
if ( node . light ) {
pScene - > mLights [ node . light . GetIndex ( ) ] - > mName = ainode - > mName ;
2020-10-16 15:09:17 +00:00
2020-12-28 07:48:54 +00:00
//range is optional - see https://github.com/KhronosGroup/glTF/tree/master/extensions/2.0/Khronos/KHR_lights_punctual
//it is added to meta data of parent node, because there is no other place to put it
if ( node . light - > range . isPresent ) {
if ( ! ainode - > mMetaData ) {
ainode - > mMetaData = aiMetadata : : Alloc ( 1 ) ;
ainode - > mMetaData - > Set ( 0 , " PBR_LightRange " , node . light - > range . value ) ;
}
else {
ainode - > mMetaData - > Add ( " PBR_LightRange " , node . light - > range . value ) ;
}
2020-10-16 15:09:17 +00:00
}
}
2020-12-28 07:48:54 +00:00
return ainode ;
} catch ( . . . ) {
delete ainode ;
throw ;
}
2017-08-24 21:15:48 +00:00
}
2019-11-16 07:08:57 +00:00
void glTF2Importer : : ImportNodes ( glTF2 : : Asset & r ) {
2020-10-16 15:09:17 +00:00
if ( ! r . scene ) {
throw DeadlyImportError ( " GLTF: No scene " ) ;
}
ASSIMP_LOG_DEBUG ( " Importing nodes " ) ;
std : : vector < Ref < Node > > rootNodes = r . scene - > nodes ;
// The root nodes
unsigned int numRootNodes = unsigned ( rootNodes . size ( ) ) ;
if ( numRootNodes = = 1 ) { // a single root node: use it
mScene - > mRootNode = ImportNode ( mScene , r , meshOffsets , rootNodes [ 0 ] ) ;
} else if ( numRootNodes > 1 ) { // more than one root node: create a fake root
2020-12-28 07:48:54 +00:00
aiNode * root = mScene - > mRootNode = new aiNode ( " ROOT " ) ;
2020-10-16 15:09:17 +00:00
root - > mChildren = new aiNode * [ numRootNodes ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( root - > mChildren , root - > mChildren + numRootNodes , nullptr ) ;
2020-10-16 15:09:17 +00:00
for ( unsigned int i = 0 ; i < numRootNodes ; + + i ) {
aiNode * node = ImportNode ( mScene , r , meshOffsets , rootNodes [ i ] ) ;
node - > mParent = root ;
root - > mChildren [ root - > mNumChildren + + ] = node ;
}
} else {
mScene - > mRootNode = new aiNode ( " ROOT " ) ;
}
2017-08-24 21:15:48 +00:00
}
2018-10-26 22:36:34 +00:00
struct AnimationSamplers {
2020-10-16 15:09:17 +00:00
AnimationSamplers ( ) :
translation ( nullptr ) ,
rotation ( nullptr ) ,
scale ( nullptr ) ,
weight ( nullptr ) {
// empty
}
Animation : : Sampler * translation ;
Animation : : Sampler * rotation ;
Animation : : Sampler * scale ;
Animation : : Sampler * weight ;
2018-10-26 22:36:34 +00:00
} ;
2020-03-08 20:24:01 +00:00
aiNodeAnim * CreateNodeAnim ( glTF2 : : Asset & , Node & node , AnimationSamplers & samplers ) {
2020-10-16 15:09:17 +00:00
aiNodeAnim * anim = new aiNodeAnim ( ) ;
2020-12-28 07:48:54 +00:00
try {
anim - > mNodeName = GetNodeName ( node ) ;
static const float kMillisecondsFromSeconds = 1000.f ;
2021-04-21 15:17:03 +00:00
if ( samplers . translation & & samplers . translation - > input & & samplers . translation - > output ) {
2020-12-28 07:48:54 +00:00
float * times = nullptr ;
samplers . translation - > input - > ExtractData ( times ) ;
aiVector3D * values = nullptr ;
samplers . translation - > output - > ExtractData ( values ) ;
anim - > mNumPositionKeys = static_cast < uint32_t > ( samplers . translation - > input - > count ) ;
anim - > mPositionKeys = new aiVectorKey [ anim - > mNumPositionKeys ] ;
unsigned int ii = ( samplers . translation - > interpolation = = Interpolation_CUBICSPLINE ) ? 1 : 0 ;
for ( unsigned int i = 0 ; i < anim - > mNumPositionKeys ; + + i ) {
anim - > mPositionKeys [ i ] . mTime = times [ i ] * kMillisecondsFromSeconds ;
anim - > mPositionKeys [ i ] . mValue = values [ ii ] ;
ii + = ( samplers . translation - > interpolation = = Interpolation_CUBICSPLINE ) ? 3 : 1 ;
}
delete [ ] times ;
delete [ ] values ;
} else if ( node . translation . isPresent ) {
anim - > mNumPositionKeys = 1 ;
anim - > mPositionKeys = new aiVectorKey [ anim - > mNumPositionKeys ] ;
anim - > mPositionKeys - > mTime = 0.f ;
anim - > mPositionKeys - > mValue . x = node . translation . value [ 0 ] ;
anim - > mPositionKeys - > mValue . y = node . translation . value [ 1 ] ;
anim - > mPositionKeys - > mValue . z = node . translation . value [ 2 ] ;
2020-10-16 15:09:17 +00:00
}
2021-04-21 15:17:03 +00:00
if ( samplers . rotation & & samplers . rotation - > input & & samplers . rotation - > output ) {
2020-12-28 07:48:54 +00:00
float * times = nullptr ;
samplers . rotation - > input - > ExtractData ( times ) ;
aiQuaternion * values = nullptr ;
samplers . rotation - > output - > ExtractData ( values ) ;
anim - > mNumRotationKeys = static_cast < uint32_t > ( samplers . rotation - > input - > count ) ;
anim - > mRotationKeys = new aiQuatKey [ anim - > mNumRotationKeys ] ;
unsigned int ii = ( samplers . rotation - > interpolation = = Interpolation_CUBICSPLINE ) ? 1 : 0 ;
for ( unsigned int i = 0 ; i < anim - > mNumRotationKeys ; + + i ) {
anim - > mRotationKeys [ i ] . mTime = times [ i ] * kMillisecondsFromSeconds ;
anim - > mRotationKeys [ i ] . mValue . x = values [ ii ] . w ;
anim - > mRotationKeys [ i ] . mValue . y = values [ ii ] . x ;
anim - > mRotationKeys [ i ] . mValue . z = values [ ii ] . y ;
anim - > mRotationKeys [ i ] . mValue . w = values [ ii ] . z ;
ii + = ( samplers . rotation - > interpolation = = Interpolation_CUBICSPLINE ) ? 3 : 1 ;
}
delete [ ] times ;
delete [ ] values ;
} else if ( node . rotation . isPresent ) {
anim - > mNumRotationKeys = 1 ;
anim - > mRotationKeys = new aiQuatKey [ anim - > mNumRotationKeys ] ;
anim - > mRotationKeys - > mTime = 0.f ;
anim - > mRotationKeys - > mValue . x = node . rotation . value [ 0 ] ;
anim - > mRotationKeys - > mValue . y = node . rotation . value [ 1 ] ;
anim - > mRotationKeys - > mValue . z = node . rotation . value [ 2 ] ;
anim - > mRotationKeys - > mValue . w = node . rotation . value [ 3 ] ;
2020-10-16 15:09:17 +00:00
}
2021-04-21 15:17:03 +00:00
if ( samplers . scale & & samplers . scale - > input & & samplers . scale - > output ) {
2020-12-28 07:48:54 +00:00
float * times = nullptr ;
samplers . scale - > input - > ExtractData ( times ) ;
aiVector3D * values = nullptr ;
samplers . scale - > output - > ExtractData ( values ) ;
anim - > mNumScalingKeys = static_cast < uint32_t > ( samplers . scale - > input - > count ) ;
anim - > mScalingKeys = new aiVectorKey [ anim - > mNumScalingKeys ] ;
unsigned int ii = ( samplers . scale - > interpolation = = Interpolation_CUBICSPLINE ) ? 1 : 0 ;
for ( unsigned int i = 0 ; i < anim - > mNumScalingKeys ; + + i ) {
anim - > mScalingKeys [ i ] . mTime = times [ i ] * kMillisecondsFromSeconds ;
anim - > mScalingKeys [ i ] . mValue = values [ ii ] ;
ii + = ( samplers . scale - > interpolation = = Interpolation_CUBICSPLINE ) ? 3 : 1 ;
}
delete [ ] times ;
delete [ ] values ;
} else if ( node . scale . isPresent ) {
anim - > mNumScalingKeys = 1 ;
anim - > mScalingKeys = new aiVectorKey [ anim - > mNumScalingKeys ] ;
anim - > mScalingKeys - > mTime = 0.f ;
anim - > mScalingKeys - > mValue . x = node . scale . value [ 0 ] ;
anim - > mScalingKeys - > mValue . y = node . scale . value [ 1 ] ;
anim - > mScalingKeys - > mValue . z = node . scale . value [ 2 ] ;
2020-10-16 15:09:17 +00:00
}
2020-12-28 07:48:54 +00:00
return anim ;
} catch ( . . . ) {
delete anim ;
throw ;
}
2018-10-26 22:36:34 +00:00
}
2020-03-08 20:24:01 +00:00
aiMeshMorphAnim * CreateMeshMorphAnim ( glTF2 : : Asset & , Node & node , AnimationSamplers & samplers ) {
2020-10-16 15:09:17 +00:00
aiMeshMorphAnim * anim = new aiMeshMorphAnim ( ) ;
2020-12-28 07:48:54 +00:00
try {
anim - > mName = GetNodeName ( node ) ;
static const float kMillisecondsFromSeconds = 1000.f ;
2021-05-05 13:09:43 +00:00
if ( samplers . weight & & samplers . weight - > input & & samplers . weight - > output ) {
2020-12-28 07:48:54 +00:00
float * times = nullptr ;
samplers . weight - > input - > ExtractData ( times ) ;
float * values = nullptr ;
samplers . weight - > output - > ExtractData ( values ) ;
anim - > mNumKeys = static_cast < uint32_t > ( samplers . weight - > input - > count ) ;
// for Interpolation_CUBICSPLINE can have more outputs
const unsigned int weightStride = ( unsigned int ) samplers . weight - > output - > count / anim - > mNumKeys ;
const unsigned int numMorphs = ( samplers . weight - > interpolation = = Interpolation_CUBICSPLINE ) ? weightStride - 2 : weightStride ;
anim - > mKeys = new aiMeshMorphKey [ anim - > mNumKeys ] ;
unsigned int ii = ( samplers . weight - > interpolation = = Interpolation_CUBICSPLINE ) ? 1 : 0 ;
for ( unsigned int i = 0u ; i < anim - > mNumKeys ; + + i ) {
unsigned int k = weightStride * i + ii ;
anim - > mKeys [ i ] . mTime = times [ i ] * kMillisecondsFromSeconds ;
anim - > mKeys [ i ] . mNumValuesAndWeights = numMorphs ;
anim - > mKeys [ i ] . mValues = new unsigned int [ numMorphs ] ;
anim - > mKeys [ i ] . mWeights = new double [ numMorphs ] ;
for ( unsigned int j = 0u ; j < numMorphs ; + + j , + + k ) {
anim - > mKeys [ i ] . mValues [ j ] = j ;
anim - > mKeys [ i ] . mWeights [ j ] = ( 0.f > values [ k ] ) ? 0.f : values [ k ] ;
}
2020-10-16 15:09:17 +00:00
}
2020-12-28 07:48:54 +00:00
delete [ ] times ;
delete [ ] values ;
2020-10-16 15:09:17 +00:00
}
2020-12-28 07:48:54 +00:00
return anim ;
} catch ( . . . ) {
delete anim ;
throw ;
2020-10-16 15:09:17 +00:00
}
2019-09-25 08:57:53 +00:00
}
2019-11-16 07:08:57 +00:00
std : : unordered_map < unsigned int , AnimationSamplers > GatherSamplers ( Animation & anim ) {
2020-10-16 15:09:17 +00:00
std : : unordered_map < unsigned int , AnimationSamplers > samplers ;
for ( unsigned int c = 0 ; c < anim . channels . size ( ) ; + + c ) {
Animation : : Channel & channel = anim . channels [ c ] ;
2020-12-28 07:48:54 +00:00
if ( channel . sampler < 0 | | channel . sampler > = static_cast < int > ( anim . samplers . size ( ) ) ) {
2020-10-16 15:09:17 +00:00
continue ;
}
const unsigned int node_index = channel . target . node . GetIndex ( ) ;
AnimationSamplers & sampler = samplers [ node_index ] ;
if ( channel . target . path = = AnimationPath_TRANSLATION ) {
sampler . translation = & anim . samplers [ channel . sampler ] ;
} else if ( channel . target . path = = AnimationPath_ROTATION ) {
sampler . rotation = & anim . samplers [ channel . sampler ] ;
} else if ( channel . target . path = = AnimationPath_SCALE ) {
sampler . scale = & anim . samplers [ channel . sampler ] ;
} else if ( channel . target . path = = AnimationPath_WEIGHTS ) {
sampler . weight = & anim . samplers [ channel . sampler ] ;
}
}
return samplers ;
2018-10-26 22:36:34 +00:00
}
2019-11-16 07:08:57 +00:00
void glTF2Importer : : ImportAnimations ( glTF2 : : Asset & r ) {
2020-10-16 15:09:17 +00:00
if ( ! r . scene ) return ;
2019-11-16 07:08:57 +00:00
2020-05-14 16:43:31 +00:00
const unsigned numAnimations = r . animations . Size ( ) ;
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_DEBUG ( " Importing " , numAnimations , " animations " ) ;
2020-10-16 15:09:17 +00:00
mScene - > mNumAnimations = numAnimations ;
if ( mScene - > mNumAnimations = = 0 ) {
return ;
}
2019-11-16 07:08:57 +00:00
2020-05-14 16:43:31 +00:00
mScene - > mAnimations = new aiAnimation * [ numAnimations ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( mScene - > mAnimations , mScene - > mAnimations + numAnimations , nullptr ) ;
2020-10-16 15:09:17 +00:00
for ( unsigned int i = 0 ; i < numAnimations ; + + i ) {
2020-12-28 07:48:54 +00:00
aiAnimation * ai_anim = mScene - > mAnimations [ i ] = new aiAnimation ( ) ;
2020-10-16 15:09:17 +00:00
Animation & anim = r . animations [ i ] ;
ai_anim - > mName = anim . name ;
ai_anim - > mDuration = 0 ;
ai_anim - > mTicksPerSecond = 0 ;
std : : unordered_map < unsigned int , AnimationSamplers > samplers = GatherSamplers ( anim ) ;
uint32_t numChannels = 0u ;
uint32_t numMorphMeshChannels = 0u ;
for ( auto & iter : samplers ) {
if ( ( nullptr ! = iter . second . rotation ) | | ( nullptr ! = iter . second . scale ) | | ( nullptr ! = iter . second . translation ) ) {
+ + numChannels ;
}
if ( nullptr ! = iter . second . weight ) {
+ + numMorphMeshChannels ;
}
}
ai_anim - > mNumChannels = numChannels ;
if ( ai_anim - > mNumChannels > 0 ) {
ai_anim - > mChannels = new aiNodeAnim * [ ai_anim - > mNumChannels ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( ai_anim - > mChannels , ai_anim - > mChannels + ai_anim - > mNumChannels , nullptr ) ;
2020-10-16 15:09:17 +00:00
int j = 0 ;
for ( auto & iter : samplers ) {
if ( ( nullptr ! = iter . second . rotation ) | | ( nullptr ! = iter . second . scale ) | | ( nullptr ! = iter . second . translation ) ) {
ai_anim - > mChannels [ j ] = CreateNodeAnim ( r , r . nodes [ iter . first ] , iter . second ) ;
+ + j ;
}
}
}
ai_anim - > mNumMorphMeshChannels = numMorphMeshChannels ;
if ( ai_anim - > mNumMorphMeshChannels > 0 ) {
ai_anim - > mMorphMeshChannels = new aiMeshMorphAnim * [ ai_anim - > mNumMorphMeshChannels ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( ai_anim - > mMorphMeshChannels , ai_anim - > mMorphMeshChannels + ai_anim - > mNumMorphMeshChannels , nullptr ) ;
2020-10-16 15:09:17 +00:00
int j = 0 ;
for ( auto & iter : samplers ) {
if ( nullptr ! = iter . second . weight ) {
ai_anim - > mMorphMeshChannels [ j ] = CreateMeshMorphAnim ( r , r . nodes [ iter . first ] , iter . second ) ;
+ + j ;
}
}
}
// Use the latest keyframe for the duration of the animation
double maxDuration = 0 ;
unsigned int maxNumberOfKeys = 0 ;
for ( unsigned int j = 0 ; j < ai_anim - > mNumChannels ; + + j ) {
auto chan = ai_anim - > mChannels [ j ] ;
if ( chan - > mNumPositionKeys ) {
auto lastPosKey = chan - > mPositionKeys [ chan - > mNumPositionKeys - 1 ] ;
if ( lastPosKey . mTime > maxDuration ) {
maxDuration = lastPosKey . mTime ;
}
maxNumberOfKeys = std : : max ( maxNumberOfKeys , chan - > mNumPositionKeys ) ;
}
if ( chan - > mNumRotationKeys ) {
auto lastRotKey = chan - > mRotationKeys [ chan - > mNumRotationKeys - 1 ] ;
if ( lastRotKey . mTime > maxDuration ) {
maxDuration = lastRotKey . mTime ;
}
maxNumberOfKeys = std : : max ( maxNumberOfKeys , chan - > mNumRotationKeys ) ;
}
if ( chan - > mNumScalingKeys ) {
auto lastScaleKey = chan - > mScalingKeys [ chan - > mNumScalingKeys - 1 ] ;
if ( lastScaleKey . mTime > maxDuration ) {
maxDuration = lastScaleKey . mTime ;
}
maxNumberOfKeys = std : : max ( maxNumberOfKeys , chan - > mNumScalingKeys ) ;
}
}
for ( unsigned int j = 0 ; j < ai_anim - > mNumMorphMeshChannels ; + + j ) {
const auto * const chan = ai_anim - > mMorphMeshChannels [ j ] ;
if ( 0u ! = chan - > mNumKeys ) {
const auto & lastKey = chan - > mKeys [ chan - > mNumKeys - 1u ] ;
if ( lastKey . mTime > maxDuration ) {
maxDuration = lastKey . mTime ;
}
maxNumberOfKeys = std : : max ( maxNumberOfKeys , chan - > mNumKeys ) ;
}
}
ai_anim - > mDuration = maxDuration ;
ai_anim - > mTicksPerSecond = 1000.0 ;
}
2018-10-26 22:36:34 +00:00
}
2019-11-16 07:08:57 +00:00
void glTF2Importer : : ImportEmbeddedTextures ( glTF2 : : Asset & r ) {
2020-10-16 15:09:17 +00:00
embeddedTexIdxs . resize ( r . images . Size ( ) , - 1 ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
int numEmbeddedTexs = 0 ;
for ( size_t i = 0 ; i < r . images . Size ( ) ; + + i ) {
if ( r . images [ i ] . HasData ( ) ) {
numEmbeddedTexs + = 1 ;
}
}
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
if ( numEmbeddedTexs = = 0 )
return ;
2017-08-24 21:15:48 +00:00
2021-05-13 09:25:27 +00:00
ASSIMP_LOG_DEBUG ( " Importing " , numEmbeddedTexs , " embedded textures " ) ;
2020-05-14 16:43:31 +00:00
2020-10-16 15:09:17 +00:00
mScene - > mTextures = new aiTexture * [ numEmbeddedTexs ] ;
2020-12-28 07:48:54 +00:00
std : : fill ( mScene - > mTextures , mScene - > mTextures + numEmbeddedTexs , nullptr ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
// Add the embedded textures
for ( size_t i = 0 ; i < r . images . Size ( ) ; + + i ) {
Image & img = r . images [ i ] ;
if ( ! img . HasData ( ) ) {
continue ;
}
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
int idx = mScene - > mNumTextures + + ;
embeddedTexIdxs [ i ] = idx ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
aiTexture * tex = mScene - > mTextures [ idx ] = new aiTexture ( ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
size_t length = img . GetDataLength ( ) ;
void * data = img . StealData ( ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
tex - > mFilename = img . name ;
tex - > mWidth = static_cast < unsigned int > ( length ) ;
tex - > mHeight = 0 ;
tex - > pcData = reinterpret_cast < aiTexel * > ( data ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
if ( ! img . mimeType . empty ( ) ) {
const char * ext = strchr ( img . mimeType . c_str ( ) , ' / ' ) + 1 ;
if ( ext ) {
if ( strcmp ( ext , " jpeg " ) = = 0 ) {
2020-01-07 19:41:37 +00:00
ext = " jpg " ;
}
2021-05-07 23:27:23 +00:00
else if ( strcmp ( ext , " ktx2 " ) = = 0 ) { //basisu: ktx remains
ext = " kx2 " ;
2021-05-06 22:10:06 +00:00
}
else if ( strcmp ( ext , " basis " ) = = 0 ) { //basisu
ext = " bu " ;
}
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
size_t len = strlen ( ext ) ;
if ( len < = 3 ) {
strcpy ( tex - > achFormatHint , ext ) ;
}
}
}
}
2017-08-24 21:15:48 +00:00
}
2019-12-05 11:25:12 +00:00
void glTF2Importer : : ImportCommonMetadata ( glTF2 : : Asset & a ) {
2020-10-16 15:09:17 +00:00
ASSIMP_LOG_DEBUG ( " Importing metadata " ) ;
2019-12-05 11:25:12 +00:00
ai_assert ( mScene - > mMetaData = = nullptr ) ;
2019-12-05 11:37:22 +00:00
const bool hasVersion = ! a . asset . version . empty ( ) ;
const bool hasGenerator = ! a . asset . generator . empty ( ) ;
2019-12-09 14:30:12 +00:00
const bool hasCopyright = ! a . asset . copyright . empty ( ) ;
2021-06-12 03:16:53 +00:00
const bool hasSceneMetadata = a . scene - > customExtensions ;
2021-06-12 01:44:28 +00:00
if ( hasVersion | | hasGenerator | | hasCopyright | | hasSceneMetadata ) {
2019-12-05 11:37:22 +00:00
mScene - > mMetaData = new aiMetadata ;
2020-01-07 19:41:37 +00:00
if ( hasVersion ) {
2019-12-05 11:37:22 +00:00
mScene - > mMetaData - > Add ( AI_METADATA_SOURCE_FORMAT_VERSION , aiString ( a . asset . version ) ) ;
}
2020-01-07 19:41:37 +00:00
if ( hasGenerator ) {
2019-12-05 11:37:22 +00:00
mScene - > mMetaData - > Add ( AI_METADATA_SOURCE_GENERATOR , aiString ( a . asset . generator ) ) ;
}
2020-01-07 19:41:37 +00:00
if ( hasCopyright ) {
2019-12-09 14:30:12 +00:00
mScene - > mMetaData - > Add ( AI_METADATA_SOURCE_COPYRIGHT , aiString ( a . asset . copyright ) ) ;
}
2021-06-12 01:44:28 +00:00
if ( hasSceneMetadata ) {
2021-06-12 03:16:53 +00:00
ParseExtensions ( mScene - > mMetaData , a . scene - > customExtensions ) ;
2021-06-12 01:44:28 +00:00
}
2019-12-05 11:37:22 +00:00
}
2019-12-05 11:25:12 +00:00
}
2019-11-16 07:08:57 +00:00
void glTF2Importer : : InternReadFile ( const std : : string & pFile , aiScene * pScene , IOSystem * pIOHandler ) {
2020-05-14 16:43:31 +00:00
2020-05-15 08:29:20 +00:00
ASSIMP_LOG_DEBUG ( " Reading GLTF2 file " ) ;
2020-05-14 16:43:31 +00:00
2020-10-16 15:09:17 +00:00
// clean all member arrays
meshOffsets . clear ( ) ;
embeddedTexIdxs . clear ( ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
this - > mScene = pScene ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
// read the asset file
glTF2 : : Asset asset ( pIOHandler ) ;
asset . Load ( pFile , GetExtension ( pFile ) = = " glb " ) ;
2020-11-22 05:49:41 +00:00
if ( asset . scene ) {
pScene - > mName = asset . scene - > name ;
}
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
//
// Copy the data out
//
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
ImportEmbeddedTextures ( asset ) ;
ImportMaterials ( asset ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
ImportMeshes ( asset ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
ImportCameras ( asset ) ;
ImportLights ( asset ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
ImportNodes ( asset ) ;
2017-08-24 21:15:48 +00:00
2020-10-16 15:09:17 +00:00
ImportAnimations ( asset ) ;
2018-10-26 22:36:34 +00:00
2019-12-05 11:25:12 +00:00
ImportCommonMetadata ( asset ) ;
2020-10-16 15:09:17 +00:00
if ( pScene - > mNumMeshes = = 0 ) {
pScene - > mFlags | = AI_SCENE_FLAGS_INCOMPLETE ;
}
2017-08-24 21:15:48 +00:00
}
# endif // ASSIMP_BUILD_NO_GLTF_IMPORTER