2014-05-18 08:57:44 +00:00
/*
Open Asset Import Library ( assimp )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
2018-01-28 18:42:05 +00:00
Copyright ( c ) 2006 - 2018 , assimp team
2017-05-09 17:57:36 +00:00
2014-05-18 08:57:44 +00:00
All rights reserved .
2015-05-19 03:52:10 +00:00
Redistribution and use of this software in source and binary forms ,
with or without modification , are permitted provided that the
2014-05-18 08:57:44 +00:00
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 .
2015-05-19 03:52:10 +00:00
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
" AS IS " AND ANY EXPRESS OR IMPLIED WARRANTIES , INCLUDING , BUT NOT
2014-05-18 08:57:44 +00:00
LIMITED TO , THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
2015-05-19 03:52:10 +00:00
A PARTICULAR PURPOSE ARE DISCLAIMED . IN NO EVENT SHALL THE COPYRIGHT
2014-05-18 08:57:44 +00:00
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT , INDIRECT , INCIDENTAL ,
2015-05-19 03:52:10 +00:00
SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES ( INCLUDING , BUT NOT
2014-05-18 08:57:44 +00:00
LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ; LOSS OF USE ,
2015-05-19 03:52:10 +00:00
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
2014-05-18 08:57:44 +00:00
OF THIS SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
*/
2014-05-20 01:52:53 +00:00
# include "OgreXmlSerializer.h"
2014-05-21 01:00:11 +00:00
# include "OgreBinarySerializer.h"
# include "OgreParsingUtils.h"
2014-05-20 01:52:53 +00:00
2018-01-06 00:18:33 +00:00
# include <assimp/TinyFormatter.h>
2016-06-06 20:04:29 +00:00
# include <assimp/DefaultLogger.hpp>
2016-04-05 21:23:53 +00:00
# include <memory>
2014-05-20 01:52:53 +00:00
# ifndef ASSIMP_BUILD_NO_OGRE_IMPORTER
// Define as 1 to get verbose logging.
# define OGRE_XML_SERIALIZER_DEBUG 0
namespace Assimp
{
namespace Ogre
{
2015-02-13 12:45:36 +00:00
AI_WONT_RETURN void ThrowAttibuteError ( const XmlReader * reader , const std : : string & name , const std : : string & error = " " ) AI_WONT_RETURN_SUFFIX ;
AI_WONT_RETURN void ThrowAttibuteError ( const XmlReader * reader , const std : : string & name , const std : : string & error )
2014-05-20 01:52:53 +00:00
{
2015-05-19 03:57:13 +00:00
if ( ! error . empty ( ) )
{
throw DeadlyImportError ( error + " in node ' " + std : : string ( reader - > getNodeName ( ) ) + " ' and attribute ' " + name + " ' " ) ;
}
else
{
throw DeadlyImportError ( " Attribute ' " + name + " ' does not exist in node ' " + std : : string ( reader - > getNodeName ( ) ) + " ' " ) ;
}
2014-05-20 01:52:53 +00:00
}
2015-05-19 03:52:10 +00:00
template < >
2014-05-20 01:52:53 +00:00
int32_t OgreXmlSerializer : : ReadAttribute < int32_t > ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
if ( HasAttribute ( name . c_str ( ) ) )
{
return static_cast < int32_t > ( m_reader - > getAttributeValueAsInt ( name . c_str ( ) ) ) ;
}
else
{
ThrowAttibuteError ( m_reader , name ) ;
return 0 ;
}
2014-05-20 01:52:53 +00:00
}
2015-05-19 03:52:10 +00:00
template < >
2014-05-20 01:52:53 +00:00
uint32_t OgreXmlSerializer : : ReadAttribute < uint32_t > ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
if ( HasAttribute ( name . c_str ( ) ) )
{
/** @note This is hackish. But we are never expecting unsigned values that go outside the
int32_t range . Just monitor for negative numbers and kill the import . */
int32_t temp = ReadAttribute < int32_t > ( name ) ;
if ( temp > = 0 )
{
return static_cast < uint32_t > ( temp ) ;
}
else
{
ThrowAttibuteError ( m_reader , name , " Found a negative number value where expecting a uint32_t value " ) ;
}
}
else
{
ThrowAttibuteError ( m_reader , name ) ;
}
return 0 ;
2014-05-20 01:52:53 +00:00
}
2015-05-19 03:52:10 +00:00
template < >
2014-05-20 01:52:53 +00:00
uint16_t OgreXmlSerializer : : ReadAttribute < uint16_t > ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
if ( HasAttribute ( name . c_str ( ) ) )
{
return static_cast < uint16_t > ( ReadAttribute < uint32_t > ( name ) ) ;
}
else
{
ThrowAttibuteError ( m_reader , name ) ;
}
return 0 ;
2014-05-20 01:52:53 +00:00
}
2015-05-19 03:52:10 +00:00
template < >
2014-05-20 01:52:53 +00:00
float OgreXmlSerializer : : ReadAttribute < float > ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
if ( HasAttribute ( name . c_str ( ) ) )
{
return m_reader - > getAttributeValueAsFloat ( name . c_str ( ) ) ;
}
else
{
ThrowAttibuteError ( m_reader , name ) ;
return 0 ;
}
2014-05-20 01:52:53 +00:00
}
2015-05-19 03:52:10 +00:00
template < >
2014-05-20 01:52:53 +00:00
std : : string OgreXmlSerializer : : ReadAttribute < std : : string > ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
const char * value = m_reader - > getAttributeValue ( name . c_str ( ) ) ;
if ( value )
{
return std : : string ( value ) ;
}
else
{
ThrowAttibuteError ( m_reader , name ) ;
return " " ;
}
2014-05-20 01:52:53 +00:00
}
2015-05-19 03:52:10 +00:00
template < >
2014-05-20 01:52:53 +00:00
bool OgreXmlSerializer : : ReadAttribute < bool > ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
std : : string value = Ogre : : ToLower ( ReadAttribute < std : : string > ( name ) ) ;
if ( ASSIMP_stricmp ( value , " true " ) = = 0 )
{
return true ;
}
else if ( ASSIMP_stricmp ( value , " false " ) = = 0 )
{
return false ;
}
else
{
ThrowAttibuteError ( m_reader , name , " Boolean value is expected to be 'true' or 'false', encountered ' " + value + " ' " ) ;
return false ;
}
2014-05-20 01:52:53 +00:00
}
bool OgreXmlSerializer : : HasAttribute ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
return ( m_reader - > getAttributeValue ( name . c_str ( ) ) ! = 0 ) ;
2014-05-20 01:52:53 +00:00
}
std : : string & OgreXmlSerializer : : NextNode ( )
{
2015-05-19 03:57:13 +00:00
do
{
if ( ! m_reader - > read ( ) )
{
m_currentNodeName = " " ;
return m_currentNodeName ;
}
}
while ( m_reader - > getNodeType ( ) ! = irr : : io : : EXN_ELEMENT ) ;
CurrentNodeName ( true ) ;
2014-05-20 01:52:53 +00:00
# if (OGRE_XML_SERIALIZER_DEBUG == 1)
2015-05-19 03:57:13 +00:00
DefaultLogger : : get ( ) - > debug ( " < " + m_currentNodeName + " > " ) ;
2014-05-20 01:52:53 +00:00
# endif
2015-05-19 03:57:13 +00:00
return m_currentNodeName ;
2014-05-20 01:52:53 +00:00
}
bool OgreXmlSerializer : : CurrentNodeNameEquals ( const std : : string & name ) const
{
2015-05-19 03:57:13 +00:00
return ( ASSIMP_stricmp ( m_currentNodeName , name ) = = 0 ) ;
2014-05-20 01:52:53 +00:00
}
std : : string OgreXmlSerializer : : CurrentNodeName ( bool forceRead )
{
2015-05-19 03:57:13 +00:00
if ( forceRead )
m_currentNodeName = std : : string ( m_reader - > getNodeName ( ) ) ;
return m_currentNodeName ;
2014-05-20 01:52:53 +00:00
}
std : : string & OgreXmlSerializer : : SkipCurrentNode ( )
{
# if (OGRE_XML_SERIALIZER_DEBUG == 1)
2015-05-19 03:57:13 +00:00
DefaultLogger : : get ( ) - > debug ( " Skipping node < " + m_currentNodeName + " > " ) ;
2014-05-20 01:52:53 +00:00
# endif
2015-05-19 03:57:13 +00:00
for ( ; ; )
{
if ( ! m_reader - > read ( ) )
{
m_currentNodeName = " " ;
return m_currentNodeName ;
}
if ( m_reader - > getNodeType ( ) ! = irr : : io : : EXN_ELEMENT_END )
continue ;
else if ( std : : string ( m_reader - > getNodeName ( ) ) = = m_currentNodeName )
break ;
}
return NextNode ( ) ;
2014-05-20 01:52:53 +00:00
}
2014-05-20 21:09:30 +00:00
// Mesh XML constants
2014-05-20 01:52:53 +00:00
// <mesh>
const std : : string nnMesh = " mesh " ;
const std : : string nnSharedGeometry = " sharedgeometry " ;
const std : : string nnSubMeshes = " submeshes " ;
const std : : string nnSubMesh = " submesh " ;
const std : : string nnSubMeshNames = " submeshnames " ;
const std : : string nnSkeletonLink = " skeletonlink " ;
const std : : string nnLOD = " levelofdetail " ;
const std : : string nnExtremes = " extremes " ;
const std : : string nnPoses = " poses " ;
const std : : string nnAnimations = " animations " ;
// <submesh>
const std : : string nnFaces = " faces " ;
const std : : string nnFace = " face " ;
const std : : string nnGeometry = " geometry " ;
const std : : string nnTextures = " textures " ;
// <mesh/submesh>
const std : : string nnBoneAssignments = " boneassignments " ;
// <sharedgeometry/geometry>
const std : : string nnVertexBuffer = " vertexbuffer " ;
// <vertexbuffer>
const std : : string nnVertex = " vertex " ;
const std : : string nnPosition = " position " ;
const std : : string nnNormal = " normal " ;
const std : : string nnTangent = " tangent " ;
const std : : string nnBinormal = " binormal " ;
const std : : string nnTexCoord = " texcoord " ;
const std : : string nnColorDiffuse = " colour_diffuse " ;
const std : : string nnColorSpecular = " colour_specular " ;
// <boneassignments>
const std : : string nnVertexBoneAssignment = " vertexboneassignment " ;
2014-05-20 21:09:30 +00:00
// Skeleton XML constants
2014-05-20 01:52:53 +00:00
// <skeleton>
2014-05-20 21:09:30 +00:00
const std : : string nnSkeleton = " skeleton " ;
const std : : string nnBones = " bones " ;
const std : : string nnBoneHierarchy = " bonehierarchy " ;
const std : : string nnAnimationLinks = " animationlinks " ;
2014-05-20 01:52:53 +00:00
// <bones>
2014-05-20 21:09:30 +00:00
const std : : string nnBone = " bone " ;
const std : : string nnRotation = " rotation " ;
const std : : string nnAxis = " axis " ;
const std : : string nnScale = " scale " ;
2014-05-20 01:52:53 +00:00
// <bonehierarchy>
2014-05-20 21:09:30 +00:00
const std : : string nnBoneParent = " boneparent " ;
2014-05-20 01:52:53 +00:00
// <animations>
2014-05-20 21:09:30 +00:00
const std : : string nnAnimation = " animation " ;
const std : : string nnTracks = " tracks " ;
2014-05-20 01:52:53 +00:00
// <tracks>
2014-05-20 21:09:30 +00:00
const std : : string nnTrack = " track " ;
const std : : string nnKeyFrames = " keyframes " ;
const std : : string nnKeyFrame = " keyframe " ;
const std : : string nnTranslate = " translate " ;
const std : : string nnRotate = " rotate " ;
// Common XML constants
2014-05-20 01:52:53 +00:00
const std : : string anX = " x " ;
const std : : string anY = " y " ;
const std : : string anZ = " z " ;
2014-05-20 21:09:30 +00:00
// Mesh
2014-05-20 01:52:53 +00:00
2014-05-20 21:09:30 +00:00
MeshXml * OgreXmlSerializer : : ImportMesh ( XmlReader * reader )
2014-05-20 01:52:53 +00:00
{
2015-05-19 03:57:13 +00:00
OgreXmlSerializer serializer ( reader ) ;
2014-05-20 01:52:53 +00:00
2015-05-19 03:57:13 +00:00
MeshXml * mesh = new MeshXml ( ) ;
serializer . ReadMesh ( mesh ) ;
return mesh ;
2014-05-20 01:52:53 +00:00
}
void OgreXmlSerializer : : ReadMesh ( MeshXml * mesh )
{
2015-05-19 03:57:13 +00:00
if ( NextNode ( ) ! = nnMesh ) {
throw DeadlyImportError ( " Root node is < " + m_currentNodeName + " > expecting <mesh> " ) ;
}
DefaultLogger : : get ( ) - > debug ( " Reading Mesh " ) ;
NextNode ( ) ;
// Root level nodes
while ( m_currentNodeName = = nnSharedGeometry | |
m_currentNodeName = = nnSubMeshes | |
m_currentNodeName = = nnSkeletonLink | |
m_currentNodeName = = nnBoneAssignments | |
m_currentNodeName = = nnLOD | |
m_currentNodeName = = nnSubMeshNames | |
m_currentNodeName = = nnExtremes | |
m_currentNodeName = = nnPoses | |
m_currentNodeName = = nnAnimations )
{
if ( m_currentNodeName = = nnSharedGeometry )
{
mesh - > sharedVertexData = new VertexDataXml ( ) ;
ReadGeometry ( mesh - > sharedVertexData ) ;
}
else if ( m_currentNodeName = = nnSubMeshes )
{
NextNode ( ) ;
while ( m_currentNodeName = = nnSubMesh ) {
ReadSubMesh ( mesh ) ;
}
}
else if ( m_currentNodeName = = nnBoneAssignments )
{
ReadBoneAssignments ( mesh - > sharedVertexData ) ;
}
else if ( m_currentNodeName = = nnSkeletonLink )
{
mesh - > skeletonRef = ReadAttribute < std : : string > ( " name " ) ;
DefaultLogger : : get ( ) - > debug ( " Read skeleton link " + mesh - > skeletonRef ) ;
NextNode ( ) ;
}
// Assimp incompatible/ignored nodes
else
SkipCurrentNode ( ) ;
}
2014-05-20 01:52:53 +00:00
}
void OgreXmlSerializer : : ReadGeometry ( VertexDataXml * dest )
{
2015-05-19 03:57:13 +00:00
dest - > count = ReadAttribute < uint32_t > ( " vertexcount " ) ;
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " - Reading geometry of " < < dest - > count < < " vertices " ) ;
2014-05-20 01:52:53 +00:00
2015-05-19 03:57:13 +00:00
NextNode ( ) ;
while ( m_currentNodeName = = nnVertexBuffer ) {
ReadGeometryVertexBuffer ( dest ) ;
}
2014-05-20 01:52:53 +00:00
}
void OgreXmlSerializer : : ReadGeometryVertexBuffer ( VertexDataXml * dest )
{
2015-05-19 03:57:13 +00:00
bool positions = ( HasAttribute ( " positions " ) & & ReadAttribute < bool > ( " positions " ) ) ;
bool normals = ( HasAttribute ( " normals " ) & & ReadAttribute < bool > ( " normals " ) ) ;
bool tangents = ( HasAttribute ( " tangents " ) & & ReadAttribute < bool > ( " tangents " ) ) ;
uint32_t uvs = ( HasAttribute ( " texture_coords " ) ? ReadAttribute < uint32_t > ( " texture_coords " ) : 0 ) ;
// Not having positions is a error only if a previous vertex buffer did not have them.
if ( ! positions & & ! dest - > HasPositions ( ) ) {
throw DeadlyImportError ( " Vertex buffer does not contain positions! " ) ;
}
if ( positions )
{
DefaultLogger : : get ( ) - > debug ( " - Contains positions " ) ;
dest - > positions . reserve ( dest - > count ) ;
}
if ( normals )
{
DefaultLogger : : get ( ) - > debug ( " - Contains normals " ) ;
dest - > normals . reserve ( dest - > count ) ;
}
if ( tangents )
{
DefaultLogger : : get ( ) - > debug ( " - Contains tangents " ) ;
dest - > tangents . reserve ( dest - > count ) ;
}
if ( uvs > 0 )
{
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " - Contains " < < uvs < < " texture coords " ) ;
dest - > uvs . resize ( uvs ) ;
for ( size_t i = 0 , len = dest - > uvs . size ( ) ; i < len ; + + i ) {
dest - > uvs [ i ] . reserve ( dest - > count ) ;
}
}
bool warnBinormal = true ;
bool warnColorDiffuse = true ;
bool warnColorSpecular = true ;
NextNode ( ) ;
while ( m_currentNodeName = = nnVertex | |
m_currentNodeName = = nnPosition | |
m_currentNodeName = = nnNormal | |
m_currentNodeName = = nnTangent | |
m_currentNodeName = = nnBinormal | |
m_currentNodeName = = nnTexCoord | |
m_currentNodeName = = nnColorDiffuse | |
m_currentNodeName = = nnColorSpecular )
{
if ( m_currentNodeName = = nnVertex ) {
NextNode ( ) ;
}
/// @todo Implement nnBinormal, nnColorDiffuse and nnColorSpecular
if ( positions & & m_currentNodeName = = nnPosition )
{
aiVector3D pos ;
pos . x = ReadAttribute < float > ( anX ) ;
pos . y = ReadAttribute < float > ( anY ) ;
pos . z = ReadAttribute < float > ( anZ ) ;
dest - > positions . push_back ( pos ) ;
}
else if ( normals & & m_currentNodeName = = nnNormal )
{
aiVector3D normal ;
normal . x = ReadAttribute < float > ( anX ) ;
normal . y = ReadAttribute < float > ( anY ) ;
normal . z = ReadAttribute < float > ( anZ ) ;
dest - > normals . push_back ( normal ) ;
}
else if ( tangents & & m_currentNodeName = = nnTangent )
{
aiVector3D tangent ;
tangent . x = ReadAttribute < float > ( anX ) ;
tangent . y = ReadAttribute < float > ( anY ) ;
tangent . z = ReadAttribute < float > ( anZ ) ;
dest - > tangents . push_back ( tangent ) ;
}
else if ( uvs > 0 & & m_currentNodeName = = nnTexCoord )
{
2016-05-21 22:44:37 +00:00
for ( auto & uvs : dest - > uvs )
2015-05-19 03:57:13 +00:00
{
if ( m_currentNodeName ! = nnTexCoord ) {
throw DeadlyImportError ( " Vertex buffer declared more UVs than can be found in a vertex " ) ;
}
aiVector3D uv ;
uv . x = ReadAttribute < float > ( " u " ) ;
uv . y = ( ReadAttribute < float > ( " v " ) * - 1 ) + 1 ; // Flip UV from Ogre to Assimp form
2016-05-21 22:44:37 +00:00
uvs . push_back ( uv ) ;
2015-05-19 03:57:13 +00:00
NextNode ( ) ;
}
// Continue main loop as above already read next node
continue ;
}
else
{
/// @todo Remove this stuff once implemented. We only want to log warnings once per element.
bool warn = true ;
if ( m_currentNodeName = = nnBinormal )
{
if ( warnBinormal )
{
warnBinormal = false ;
}
else
{
warn = false ;
}
}
else if ( m_currentNodeName = = nnColorDiffuse )
{
if ( warnColorDiffuse )
{
warnColorDiffuse = false ;
}
else
{
warn = false ;
}
}
else if ( m_currentNodeName = = nnColorSpecular )
{
if ( warnColorSpecular )
{
warnColorSpecular = false ;
}
else
{
warn = false ;
}
}
if ( warn ) {
DefaultLogger : : get ( ) - > warn ( " Vertex buffer attribute read not implemented for element: " + m_currentNodeName ) ;
}
}
// Advance
NextNode ( ) ;
}
// Sanity checks
if ( dest - > positions . size ( ) ! = dest - > count ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " Read only " < < dest - > positions . size ( ) < < " positions when should have read " < < dest - > count ) ;
}
if ( normals & & dest - > normals . size ( ) ! = dest - > count ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " Read only " < < dest - > normals . size ( ) < < " normals when should have read " < < dest - > count ) ;
}
if ( tangents & & dest - > tangents . size ( ) ! = dest - > count ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " Read only " < < dest - > tangents . size ( ) < < " tangents when should have read " < < dest - > count ) ;
}
for ( unsigned int i = 0 ; i < dest - > uvs . size ( ) ; + + i )
{
if ( dest - > uvs [ i ] . size ( ) ! = dest - > count ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " Read only " < < dest - > uvs [ i ] . size ( )
< < " uvs for uv index " < < i < < " when should have read " < < dest - > count ) ;
}
}
2014-05-20 01:52:53 +00:00
}
void OgreXmlSerializer : : ReadSubMesh ( MeshXml * mesh )
{
2015-05-19 03:57:13 +00:00
static const std : : string anMaterial = " material " ;
static const std : : string anUseSharedVertices = " usesharedvertices " ;
static const std : : string anCount = " count " ;
static const std : : string anV1 = " v1 " ;
static const std : : string anV2 = " v2 " ;
static const std : : string anV3 = " v3 " ;
static const std : : string anV4 = " v4 " ;
SubMeshXml * submesh = new SubMeshXml ( ) ;
if ( HasAttribute ( anMaterial ) ) {
submesh - > materialRef = ReadAttribute < std : : string > ( anMaterial ) ;
}
if ( HasAttribute ( anUseSharedVertices ) ) {
submesh - > usesSharedVertexData = ReadAttribute < bool > ( anUseSharedVertices ) ;
}
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " Reading SubMesh " < < mesh - > subMeshes . size ( ) ) ;
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " - Material: ' " < < submesh - > materialRef < < " ' " ) ;
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " - Uses shared geometry: " < < ( submesh - > usesSharedVertexData ? " true " : " false " ) ) ;
// TODO: maybe we have always just 1 faces and 1 geometry and always in this order. this loop will only work correct, when the order
// of faces and geometry changed, and not if we have more than one of one
/// @todo Fix above comment with better read logic below
bool quadWarned = false ;
NextNode ( ) ;
while ( m_currentNodeName = = nnFaces | |
m_currentNodeName = = nnGeometry | |
m_currentNodeName = = nnTextures | |
m_currentNodeName = = nnBoneAssignments )
{
if ( m_currentNodeName = = nnFaces )
{
submesh - > indexData - > faceCount = ReadAttribute < uint32_t > ( anCount ) ;
submesh - > indexData - > faces . reserve ( submesh - > indexData - > faceCount ) ;
NextNode ( ) ;
while ( m_currentNodeName = = nnFace )
{
aiFace face ;
face . mNumIndices = 3 ;
face . mIndices = new unsigned int [ 3 ] ;
face . mIndices [ 0 ] = ReadAttribute < uint32_t > ( anV1 ) ;
face . mIndices [ 1 ] = ReadAttribute < uint32_t > ( anV2 ) ;
face . mIndices [ 2 ] = ReadAttribute < uint32_t > ( anV3 ) ;
/// @todo Support quads if Ogre even supports them in XML (I'm not sure but I doubt it)
if ( ! quadWarned & & HasAttribute ( anV4 ) ) {
DefaultLogger : : get ( ) - > warn ( " Submesh <face> has quads with <v4>, only triangles are supported at the moment! " ) ;
quadWarned = true ;
}
submesh - > indexData - > faces . push_back ( face ) ;
// Advance
NextNode ( ) ;
}
if ( submesh - > indexData - > faces . size ( ) = = submesh - > indexData - > faceCount )
{
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " - Faces " < < submesh - > indexData - > faceCount ) ;
}
else
{
throw DeadlyImportError ( Formatter : : format ( ) < < " Read only " < < submesh - > indexData - > faces . size ( ) < < " faces when should have read " < < submesh - > indexData - > faceCount ) ;
}
}
else if ( m_currentNodeName = = nnGeometry )
{
if ( submesh - > usesSharedVertexData ) {
throw DeadlyImportError ( " Found <geometry> in <submesh> when use shared geometry is true. Invalid mesh file. " ) ;
}
submesh - > vertexData = new VertexDataXml ( ) ;
ReadGeometry ( submesh - > vertexData ) ;
}
else if ( m_currentNodeName = = nnBoneAssignments )
{
ReadBoneAssignments ( submesh - > vertexData ) ;
}
// Assimp incompatible/ignored nodes
else
SkipCurrentNode ( ) ;
}
2016-11-20 01:49:33 +00:00
submesh - > index = static_cast < unsigned int > ( mesh - > subMeshes . size ( ) ) ;
2015-05-19 03:57:13 +00:00
mesh - > subMeshes . push_back ( submesh ) ;
2014-05-20 01:52:53 +00:00
}
void OgreXmlSerializer : : ReadBoneAssignments ( VertexDataXml * dest )
{
2015-05-19 03:57:13 +00:00
if ( ! dest ) {
throw DeadlyImportError ( " Cannot read bone assignments, vertex data is null. " ) ;
}
static const std : : string anVertexIndex = " vertexindex " ;
static const std : : string anBoneIndex = " boneindex " ;
static const std : : string anWeight = " weight " ;
std : : set < uint32_t > influencedVertices ;
NextNode ( ) ;
while ( m_currentNodeName = = nnVertexBoneAssignment )
{
VertexBoneAssignment ba ;
ba . vertexIndex = ReadAttribute < uint32_t > ( anVertexIndex ) ;
ba . boneIndex = ReadAttribute < uint16_t > ( anBoneIndex ) ;
ba . weight = ReadAttribute < float > ( anWeight ) ;
dest - > boneAssignments . push_back ( ba ) ;
influencedVertices . insert ( ba . vertexIndex ) ;
NextNode ( ) ;
}
/** Normalize bone weights.
2016-12-08 02:31:51 +00:00
Some exporters won ' t care if the sum of all bone weights
2015-05-19 03:57:13 +00:00
for a single vertex equals 1 or not , so validate here . */
const float epsilon = 0.05f ;
2016-05-21 22:44:37 +00:00
for ( const uint32_t vertexIndex : influencedVertices )
2015-05-19 03:57:13 +00:00
{
float sum = 0.0f ;
for ( VertexBoneAssignmentList : : const_iterator baIter = dest - > boneAssignments . begin ( ) , baEnd = dest - > boneAssignments . end ( ) ; baIter ! = baEnd ; + + baIter )
{
if ( baIter - > vertexIndex = = vertexIndex )
sum + = baIter - > weight ;
}
if ( ( sum < ( 1.0f - epsilon ) ) | | ( sum > ( 1.0f + epsilon ) ) )
{
2016-05-21 22:44:37 +00:00
for ( auto & boneAssign : dest - > boneAssignments )
2015-05-19 03:57:13 +00:00
{
2016-05-21 22:44:37 +00:00
if ( boneAssign . vertexIndex = = vertexIndex )
boneAssign . weight / = sum ;
2015-05-19 03:57:13 +00:00
}
}
}
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " - " < < dest - > boneAssignments . size ( ) < < " bone assignments " ) ;
2014-05-20 01:52:53 +00:00
}
2014-05-20 21:09:30 +00:00
// Skeleton
2014-05-21 01:06:22 +00:00
bool OgreXmlSerializer : : ImportSkeleton ( Assimp : : IOSystem * pIOHandler , MeshXml * mesh )
2014-05-20 21:09:30 +00:00
{
2015-05-19 03:57:13 +00:00
if ( ! mesh | | mesh - > skeletonRef . empty ( ) )
return false ;
// Highly unusual to see in read world cases but support
// XML mesh referencing a binary skeleton file.
if ( EndsWith ( mesh - > skeletonRef , " .skeleton " , false ) )
{
if ( OgreBinarySerializer : : ImportSkeleton ( pIOHandler , mesh ) )
return true ;
/** Last fallback if .skeleton failed to be read. Try reading from
. skeleton . xml even if the XML file referenced a binary skeleton .
@ note This logic was in the previous version and I don ' t want to break
old code that might depends on it . */
mesh - > skeletonRef = mesh - > skeletonRef + " .xml " ;
}
XmlReaderPtr reader = OpenReader ( pIOHandler , mesh - > skeletonRef ) ;
if ( ! reader . get ( ) )
return false ;
Skeleton * skeleton = new Skeleton ( ) ;
OgreXmlSerializer serializer ( reader . get ( ) ) ;
serializer . ReadSkeleton ( skeleton ) ;
mesh - > skeleton = skeleton ;
return true ;
2014-05-21 01:00:11 +00:00
}
2014-05-21 01:06:22 +00:00
bool OgreXmlSerializer : : ImportSkeleton ( Assimp : : IOSystem * pIOHandler , Mesh * mesh )
2014-05-21 01:00:11 +00:00
{
2015-05-19 03:57:13 +00:00
if ( ! mesh | | mesh - > skeletonRef . empty ( ) )
return false ;
XmlReaderPtr reader = OpenReader ( pIOHandler , mesh - > skeletonRef ) ;
if ( ! reader . get ( ) )
return false ;
Skeleton * skeleton = new Skeleton ( ) ;
OgreXmlSerializer serializer ( reader . get ( ) ) ;
serializer . ReadSkeleton ( skeleton ) ;
mesh - > skeleton = skeleton ;
return true ;
2014-05-21 01:00:11 +00:00
}
XmlReaderPtr OgreXmlSerializer : : OpenReader ( Assimp : : IOSystem * pIOHandler , const std : : string & filename )
{
2015-05-19 03:57:13 +00:00
if ( ! EndsWith ( filename , " .skeleton.xml " , false ) )
{
DefaultLogger : : get ( ) - > error ( " Imported Mesh is referencing to unsupported ' " + filename + " ' skeleton file. " ) ;
return XmlReaderPtr ( ) ;
}
if ( ! pIOHandler - > Exists ( filename ) )
{
DefaultLogger : : get ( ) - > error ( " Failed to find skeleton file ' " + filename + " ' that is referenced by imported Mesh. " ) ;
return XmlReaderPtr ( ) ;
}
2016-04-05 21:23:53 +00:00
std : : unique_ptr < IOStream > file ( pIOHandler - > Open ( filename ) ) ;
2015-05-19 03:57:13 +00:00
if ( ! file . get ( ) ) {
throw DeadlyImportError ( " Failed to open skeleton file " + filename ) ;
}
2016-04-05 21:23:53 +00:00
std : : unique_ptr < CIrrXML_IOStreamReader > stream ( new CIrrXML_IOStreamReader ( file . get ( ) ) ) ;
2015-05-19 03:57:13 +00:00
XmlReaderPtr reader = XmlReaderPtr ( irr : : io : : createIrrXMLReader ( stream . get ( ) ) ) ;
if ( ! reader . get ( ) ) {
throw DeadlyImportError ( " Failed to create XML reader for skeleton file " + filename ) ;
}
return reader ;
2014-05-20 21:09:30 +00:00
}
void OgreXmlSerializer : : ReadSkeleton ( Skeleton * skeleton )
{
2015-05-19 03:57:13 +00:00
if ( NextNode ( ) ! = nnSkeleton ) {
throw DeadlyImportError ( " Root node is < " + m_currentNodeName + " > expecting <skeleton> " ) ;
}
DefaultLogger : : get ( ) - > debug ( " Reading Skeleton " ) ;
// Optional blend mode from root node
if ( HasAttribute ( " blendmode " ) ) {
skeleton - > blendMode = ( ToLower ( ReadAttribute < std : : string > ( " blendmode " ) ) = = " cumulative "
? Skeleton : : ANIMBLEND_CUMULATIVE : Skeleton : : ANIMBLEND_AVERAGE ) ;
}
NextNode ( ) ;
// Root level nodes
while ( m_currentNodeName = = nnBones | |
m_currentNodeName = = nnBoneHierarchy | |
m_currentNodeName = = nnAnimations | |
m_currentNodeName = = nnAnimationLinks )
{
if ( m_currentNodeName = = nnBones )
ReadBones ( skeleton ) ;
else if ( m_currentNodeName = = nnBoneHierarchy )
ReadBoneHierarchy ( skeleton ) ;
else if ( m_currentNodeName = = nnAnimations )
ReadAnimations ( skeleton ) ;
else
SkipCurrentNode ( ) ;
}
2014-05-20 21:09:30 +00:00
}
void OgreXmlSerializer : : ReadAnimations ( Skeleton * skeleton )
{
2015-05-19 03:57:13 +00:00
if ( skeleton - > bones . empty ( ) ) {
throw DeadlyImportError ( " Cannot read <animations> for a Skeleton without bones " ) ;
}
2014-05-20 21:09:30 +00:00
2015-05-19 03:57:13 +00:00
DefaultLogger : : get ( ) - > debug ( " - Animations " ) ;
2015-05-19 03:52:10 +00:00
2015-05-19 03:57:13 +00:00
NextNode ( ) ;
while ( m_currentNodeName = = nnAnimation )
{
Animation * anim = new Animation ( skeleton ) ;
anim - > name = ReadAttribute < std : : string > ( " name " ) ;
anim - > length = ReadAttribute < float > ( " length " ) ;
2015-05-19 03:52:10 +00:00
2015-05-19 03:57:13 +00:00
if ( NextNode ( ) ! = nnTracks ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " No <tracks> found in <animation> " < < anim - > name ) ;
}
2014-05-20 21:09:30 +00:00
2015-05-19 03:57:13 +00:00
ReadAnimationTracks ( anim ) ;
skeleton - > animations . push_back ( anim ) ;
2014-05-20 21:09:30 +00:00
2015-05-19 03:57:13 +00:00
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " " < < anim - > name < < " ( " < < anim - > length < < " sec, " < < anim - > tracks . size ( ) < < " tracks) " ) ;
}
2014-05-20 21:09:30 +00:00
}
void OgreXmlSerializer : : ReadAnimationTracks ( Animation * dest )
{
2015-05-19 03:57:13 +00:00
NextNode ( ) ;
while ( m_currentNodeName = = nnTrack )
{
VertexAnimationTrack track ;
track . type = VertexAnimationTrack : : VAT_TRANSFORM ;
track . boneName = ReadAttribute < std : : string > ( " bone " ) ;
2014-05-20 21:09:30 +00:00
2015-05-19 03:57:13 +00:00
if ( NextNode ( ) ! = nnKeyFrames ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " No <keyframes> found in <track> " < < dest - > name ) ;
}
2014-05-20 21:09:30 +00:00
2015-05-19 03:57:13 +00:00
ReadAnimationKeyFrames ( dest , & track ) ;
2014-05-20 21:09:30 +00:00
2015-05-19 03:57:13 +00:00
dest - > tracks . push_back ( track ) ;
}
2014-05-20 21:09:30 +00:00
}
void OgreXmlSerializer : : ReadAnimationKeyFrames ( Animation * anim , VertexAnimationTrack * dest )
{
2015-05-19 03:57:13 +00:00
const aiVector3D zeroVec ( 0.f , 0.f , 0.f ) ;
NextNode ( ) ;
while ( m_currentNodeName = = nnKeyFrame )
{
TransformKeyFrame keyframe ;
keyframe . timePos = ReadAttribute < float > ( " time " ) ;
NextNode ( ) ;
while ( m_currentNodeName = = nnTranslate | | m_currentNodeName = = nnRotate | | m_currentNodeName = = nnScale )
{
if ( m_currentNodeName = = nnTranslate )
{
keyframe . position . x = ReadAttribute < float > ( anX ) ;
keyframe . position . y = ReadAttribute < float > ( anY ) ;
keyframe . position . z = ReadAttribute < float > ( anZ ) ;
}
else if ( m_currentNodeName = = nnRotate )
{
float angle = ReadAttribute < float > ( " angle " ) ;
if ( NextNode ( ) ! = nnAxis ) {
throw DeadlyImportError ( " No axis specified for keyframe rotation in animation " + anim - > name ) ;
}
aiVector3D axis ;
axis . x = ReadAttribute < float > ( anX ) ;
axis . y = ReadAttribute < float > ( anY ) ;
axis . z = ReadAttribute < float > ( anZ ) ;
if ( axis . Equal ( zeroVec ) )
{
axis . x = 1.0f ;
if ( angle ! = 0 ) {
DefaultLogger : : get ( ) - > warn ( " Found invalid a key frame with a zero rotation axis in animation: " + anim - > name ) ;
}
}
keyframe . rotation = aiQuaternion ( axis , angle ) ;
}
else if ( m_currentNodeName = = nnScale )
{
keyframe . scale . x = ReadAttribute < float > ( anX ) ;
keyframe . scale . y = ReadAttribute < float > ( anY ) ;
keyframe . scale . z = ReadAttribute < float > ( anZ ) ;
}
NextNode ( ) ;
}
dest - > transformKeyFrames . push_back ( keyframe ) ;
}
2014-05-20 21:09:30 +00:00
}
void OgreXmlSerializer : : ReadBoneHierarchy ( Skeleton * skeleton )
{
2015-05-19 03:57:13 +00:00
if ( skeleton - > bones . empty ( ) ) {
throw DeadlyImportError ( " Cannot read <bonehierarchy> for a Skeleton without bones " ) ;
}
while ( NextNode ( ) = = nnBoneParent )
{
const std : : string name = ReadAttribute < std : : string > ( " bone " ) ;
const std : : string parentName = ReadAttribute < std : : string > ( " parent " ) ;
Bone * bone = skeleton - > BoneByName ( name ) ;
Bone * parent = skeleton - > BoneByName ( parentName ) ;
if ( bone & & parent )
parent - > AddChild ( bone ) ;
else
throw DeadlyImportError ( " Failed to find bones for parenting: Child " + name + " for parent " + parentName ) ;
}
// Calculate bone matrices for root bones. Recursively calculates their children.
for ( size_t i = 0 , len = skeleton - > bones . size ( ) ; i < len ; + + i )
{
Bone * bone = skeleton - > bones [ i ] ;
if ( ! bone - > IsParented ( ) )
bone - > CalculateWorldMatrixAndDefaultPose ( skeleton ) ;
}
2014-05-20 21:09:30 +00:00
}
bool BoneCompare ( Bone * a , Bone * b )
{
2015-05-19 03:57:13 +00:00
return ( a - > id < b - > id ) ;
2014-05-20 21:09:30 +00:00
}
void OgreXmlSerializer : : ReadBones ( Skeleton * skeleton )
{
2015-05-19 03:57:13 +00:00
DefaultLogger : : get ( ) - > debug ( " - Bones " ) ;
NextNode ( ) ;
while ( m_currentNodeName = = nnBone )
{
Bone * bone = new Bone ( ) ;
bone - > id = ReadAttribute < uint16_t > ( " id " ) ;
bone - > name = ReadAttribute < std : : string > ( " name " ) ;
NextNode ( ) ;
while ( m_currentNodeName = = nnPosition | |
m_currentNodeName = = nnRotation | |
m_currentNodeName = = nnScale )
{
if ( m_currentNodeName = = nnPosition )
{
bone - > position . x = ReadAttribute < float > ( anX ) ;
bone - > position . y = ReadAttribute < float > ( anY ) ;
bone - > position . z = ReadAttribute < float > ( anZ ) ;
}
else if ( m_currentNodeName = = nnRotation )
{
float angle = ReadAttribute < float > ( " angle " ) ;
if ( NextNode ( ) ! = nnAxis ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " No axis specified for bone rotation in bone " < < bone - > id ) ;
}
aiVector3D axis ;
axis . x = ReadAttribute < float > ( anX ) ;
axis . y = ReadAttribute < float > ( anY ) ;
axis . z = ReadAttribute < float > ( anZ ) ;
bone - > rotation = aiQuaternion ( axis , angle ) ;
}
else if ( m_currentNodeName = = nnScale )
{
/// @todo Implement taking scale into account in matrix/pose calculations!
if ( HasAttribute ( " factor " ) )
{
float factor = ReadAttribute < float > ( " factor " ) ;
bone - > scale . Set ( factor , factor , factor ) ;
}
else
{
if ( HasAttribute ( anX ) )
bone - > scale . x = ReadAttribute < float > ( anX ) ;
if ( HasAttribute ( anY ) )
bone - > scale . y = ReadAttribute < float > ( anY ) ;
if ( HasAttribute ( anZ ) )
bone - > scale . z = ReadAttribute < float > ( anZ ) ;
}
}
NextNode ( ) ;
}
skeleton - > bones . push_back ( bone ) ;
}
// Order bones by Id
std : : sort ( skeleton - > bones . begin ( ) , skeleton - > bones . end ( ) , BoneCompare ) ;
// Validate that bone indexes are not skipped.
/** @note Left this from original authors code, but not sure if this is strictly necessary
as per the Ogre skeleton spec . It might be more that other ( later ) code in this imported does not break . */
for ( size_t i = 0 , len = skeleton - > bones . size ( ) ; i < len ; + + i )
{
Bone * b = skeleton - > bones [ i ] ;
DefaultLogger : : get ( ) - > debug ( Formatter : : format ( ) < < " " < < b - > id < < " " < < b - > name ) ;
if ( b - > id ! = static_cast < uint16_t > ( i ) ) {
throw DeadlyImportError ( Formatter : : format ( ) < < " Bone ids are not in sequence starting from 0. Missing index " < < i ) ;
}
}
2014-05-20 21:09:30 +00:00
}
2014-05-20 01:52:53 +00:00
} // Ogre
} // Assimp
# endif // ASSIMP_BUILD_NO_OGRE_IMPORTER