|
@@ -2,7 +2,8 @@
|
|
|
Open Asset Import Library (assimp)
|
|
|
----------------------------------------------------------------------
|
|
|
|
|
|
-Copyright (c) 2006-2017, assimp team
|
|
|
+Copyright (c) 2006-2018, assimp team
|
|
|
+
|
|
|
|
|
|
All rights reserved.
|
|
|
|
|
@@ -47,8 +48,8 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
#include <assimp/IOSystem.hpp>
|
|
|
#include <assimp/DefaultLogger.hpp>
|
|
|
#include <assimp/importerdesc.h>
|
|
|
-#include "StringComparison.h"
|
|
|
-#include "StringUtils.h"
|
|
|
+#include <assimp/StringComparison.h>
|
|
|
+#include <assimp/StringUtils.h>
|
|
|
|
|
|
#include <string>
|
|
|
#include <vector>
|
|
@@ -57,17 +58,27 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
#include <memory>
|
|
|
|
|
|
#include "D3MFOpcPackage.h"
|
|
|
-#include <contrib/unzip/unzip.h>
|
|
|
-#include "irrXMLWrapper.h"
|
|
|
+#include <unzip.h>
|
|
|
+#include <assimp/irrXMLWrapper.h>
|
|
|
#include "3MFXmlTags.h"
|
|
|
+#include <assimp/fast_atof.h>
|
|
|
+
|
|
|
+#include <iomanip>
|
|
|
|
|
|
namespace Assimp {
|
|
|
namespace D3MF {
|
|
|
|
|
|
class XmlSerializer {
|
|
|
public:
|
|
|
+ using MatArray = std::vector<aiMaterial*>;
|
|
|
+ using MatId2MatArray = std::map<unsigned int, std::vector<unsigned int>>;
|
|
|
+
|
|
|
XmlSerializer(XmlReader* xmlReader)
|
|
|
- : xmlReader(xmlReader) {
|
|
|
+ : mMeshes()
|
|
|
+ , mMatArray()
|
|
|
+ , mActiveMatGroup( 99999999 )
|
|
|
+ , mMatId2MatArray()
|
|
|
+ , xmlReader(xmlReader){
|
|
|
// empty
|
|
|
}
|
|
|
|
|
@@ -76,14 +87,24 @@ public:
|
|
|
}
|
|
|
|
|
|
void ImportXml(aiScene* scene) {
|
|
|
+ if ( nullptr == scene ) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
scene->mRootNode = new aiNode();
|
|
|
std::vector<aiNode*> children;
|
|
|
|
|
|
+ std::string nodeName;
|
|
|
while(ReadToEndElement(D3MF::XmlTag::model)) {
|
|
|
- if(xmlReader->getNodeName() == D3MF::XmlTag::object) {
|
|
|
+ nodeName = xmlReader->getNodeName();
|
|
|
+ if( nodeName == D3MF::XmlTag::object) {
|
|
|
children.push_back(ReadObject(scene));
|
|
|
- } else if(xmlReader->getNodeName() == D3MF::XmlTag::build) {
|
|
|
-
|
|
|
+ } else if( nodeName == D3MF::XmlTag::build) {
|
|
|
+ //
|
|
|
+ } else if ( nodeName == D3MF::XmlTag::basematerials ) {
|
|
|
+ ReadBaseMaterials();
|
|
|
+ } else if ( nodeName == D3MF::XmlTag::meta ) {
|
|
|
+ ReadMetadata();
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -91,31 +112,47 @@ public:
|
|
|
scene->mRootNode->mName.Set( "3MF" );
|
|
|
}
|
|
|
|
|
|
- scene->mNumMeshes = static_cast<unsigned int>(meshes.size());
|
|
|
+ // import the metadata
|
|
|
+ if ( !mMetaData.empty() ) {
|
|
|
+ const size_t numMeta( mMetaData.size() );
|
|
|
+ scene->mMetaData = aiMetadata::Alloc( numMeta );
|
|
|
+ for ( size_t i = 0; i < numMeta; ++i ) {
|
|
|
+ aiString val( mMetaData[ i ].value );
|
|
|
+ scene->mMetaData->Set( i, mMetaData[ i ].name, val );
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // import the meshes
|
|
|
+ scene->mNumMeshes = static_cast<unsigned int>( mMeshes.size());
|
|
|
scene->mMeshes = new aiMesh*[scene->mNumMeshes]();
|
|
|
+ std::copy( mMeshes.begin(), mMeshes.end(), scene->mMeshes);
|
|
|
|
|
|
- std::copy(meshes.begin(), meshes.end(), scene->mMeshes);
|
|
|
+ // import the materials
|
|
|
+ scene->mNumMaterials = static_cast<unsigned int>( mMatArray.size() );
|
|
|
+ if ( 0 != scene->mNumMaterials ) {
|
|
|
+ scene->mMaterials = new aiMaterial*[ scene->mNumMaterials ];
|
|
|
+ std::copy( mMatArray.begin(), mMatArray.end(), scene->mMaterials );
|
|
|
+ }
|
|
|
|
|
|
+ // create the scenegraph
|
|
|
scene->mRootNode->mNumChildren = static_cast<unsigned int>(children.size());
|
|
|
scene->mRootNode->mChildren = new aiNode*[scene->mRootNode->mNumChildren]();
|
|
|
-
|
|
|
std::copy(children.begin(), children.end(), scene->mRootNode->mChildren);
|
|
|
}
|
|
|
|
|
|
private:
|
|
|
- aiNode* ReadObject(aiScene* scene)
|
|
|
- {
|
|
|
+ aiNode* ReadObject(aiScene* scene) {
|
|
|
std::unique_ptr<aiNode> node(new aiNode());
|
|
|
|
|
|
std::vector<unsigned long> meshIds;
|
|
|
|
|
|
const char *attrib( nullptr );
|
|
|
std::string name, type;
|
|
|
- attrib = xmlReader->getAttributeValue( D3MF::XmlTag::name.c_str() );
|
|
|
+ attrib = xmlReader->getAttributeValue( D3MF::XmlTag::id.c_str() );
|
|
|
if ( nullptr != attrib ) {
|
|
|
name = attrib;
|
|
|
}
|
|
|
- attrib = xmlReader->getAttributeValue( D3MF::XmlTag::name.c_str() );
|
|
|
+ attrib = xmlReader->getAttributeValue( D3MF::XmlTag::type.c_str() );
|
|
|
if ( nullptr != attrib ) {
|
|
|
type = attrib;
|
|
|
}
|
|
@@ -123,19 +160,16 @@ private:
|
|
|
node->mParent = scene->mRootNode;
|
|
|
node->mName.Set(name);
|
|
|
|
|
|
- size_t meshIdx = meshes.size();
|
|
|
+ size_t meshIdx = mMeshes.size();
|
|
|
|
|
|
- while(ReadToEndElement(D3MF::XmlTag::object))
|
|
|
- {
|
|
|
- if(xmlReader->getNodeName() == D3MF::XmlTag::mesh)
|
|
|
- {
|
|
|
+ while(ReadToEndElement(D3MF::XmlTag::object)) {
|
|
|
+ if(xmlReader->getNodeName() == D3MF::XmlTag::mesh) {
|
|
|
auto mesh = ReadMesh();
|
|
|
|
|
|
mesh->mName.Set(name);
|
|
|
- meshes.push_back(mesh);
|
|
|
+ mMeshes.push_back(mesh);
|
|
|
meshIds.push_back(static_cast<unsigned long>(meshIdx));
|
|
|
- meshIdx++;
|
|
|
-
|
|
|
+ ++meshIdx;
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -146,19 +180,14 @@ private:
|
|
|
std::copy(meshIds.begin(), meshIds.end(), node->mMeshes);
|
|
|
|
|
|
return node.release();
|
|
|
-
|
|
|
}
|
|
|
|
|
|
- aiMesh* ReadMesh() {
|
|
|
+ aiMesh *ReadMesh() {
|
|
|
aiMesh* mesh = new aiMesh();
|
|
|
- while(ReadToEndElement(D3MF::XmlTag::mesh))
|
|
|
- {
|
|
|
- if(xmlReader->getNodeName() == D3MF::XmlTag::vertices)
|
|
|
- {
|
|
|
+ while(ReadToEndElement(D3MF::XmlTag::mesh)) {
|
|
|
+ if(xmlReader->getNodeName() == D3MF::XmlTag::vertices) {
|
|
|
ImportVertices(mesh);
|
|
|
- }
|
|
|
- else if(xmlReader->getNodeName() == D3MF::XmlTag::triangles)
|
|
|
- {
|
|
|
+ } else if(xmlReader->getNodeName() == D3MF::XmlTag::triangles) {
|
|
|
ImportTriangles(mesh);
|
|
|
}
|
|
|
}
|
|
@@ -166,14 +195,25 @@ private:
|
|
|
return mesh;
|
|
|
}
|
|
|
|
|
|
- void ImportVertices(aiMesh* mesh)
|
|
|
- {
|
|
|
- std::vector<aiVector3D> vertices;
|
|
|
+ void ReadMetadata() {
|
|
|
+ const std::string name = xmlReader->getAttributeValue( D3MF::XmlTag::meta_name.c_str() );
|
|
|
+ xmlReader->read();
|
|
|
+ const std::string value = xmlReader->getNodeData();
|
|
|
+
|
|
|
+ if ( name.empty() ) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
|
|
|
- while(ReadToEndElement(D3MF::XmlTag::vertices))
|
|
|
- {
|
|
|
- if(xmlReader->getNodeName() == D3MF::XmlTag::vertex)
|
|
|
- {
|
|
|
+ MetaEntry entry;
|
|
|
+ entry.name = name;
|
|
|
+ entry.value = value;
|
|
|
+ mMetaData.push_back( entry );
|
|
|
+ }
|
|
|
+
|
|
|
+ void ImportVertices(aiMesh* mesh) {
|
|
|
+ std::vector<aiVector3D> vertices;
|
|
|
+ while(ReadToEndElement(D3MF::XmlTag::vertices)) {
|
|
|
+ if(xmlReader->getNodeName() == D3MF::XmlTag::vertex) {
|
|
|
vertices.push_back(ReadVertex());
|
|
|
}
|
|
|
}
|
|
@@ -181,11 +221,9 @@ private:
|
|
|
mesh->mVertices = new aiVector3D[mesh->mNumVertices];
|
|
|
|
|
|
std::copy(vertices.begin(), vertices.end(), mesh->mVertices);
|
|
|
-
|
|
|
}
|
|
|
|
|
|
- aiVector3D ReadVertex()
|
|
|
- {
|
|
|
+ aiVector3D ReadVertex() {
|
|
|
aiVector3D vertex;
|
|
|
|
|
|
vertex.x = ai_strtof(xmlReader->getAttributeValue(D3MF::XmlTag::x.c_str()), nullptr);
|
|
@@ -195,16 +233,18 @@ private:
|
|
|
return vertex;
|
|
|
}
|
|
|
|
|
|
- void ImportTriangles(aiMesh* mesh)
|
|
|
- {
|
|
|
+ void ImportTriangles(aiMesh* mesh) {
|
|
|
std::vector<aiFace> faces;
|
|
|
|
|
|
-
|
|
|
- while(ReadToEndElement(D3MF::XmlTag::triangles))
|
|
|
- {
|
|
|
- if(xmlReader->getNodeName() == D3MF::XmlTag::triangle)
|
|
|
- {
|
|
|
+ while(ReadToEndElement(D3MF::XmlTag::triangles)) {
|
|
|
+ const std::string nodeName( xmlReader->getNodeName() );
|
|
|
+ if(xmlReader->getNodeName() == D3MF::XmlTag::triangle) {
|
|
|
faces.push_back(ReadTriangle());
|
|
|
+ const char *pidToken( xmlReader->getAttributeValue( D3MF::XmlTag::p1.c_str() ) );
|
|
|
+ if ( nullptr != pidToken ) {
|
|
|
+ int matIdx( std::atoi( pidToken ) );
|
|
|
+ mesh->mMaterialIndex = matIdx;
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -215,8 +255,7 @@ private:
|
|
|
std::copy(faces.begin(), faces.end(), mesh->mFaces);
|
|
|
}
|
|
|
|
|
|
- aiFace ReadTriangle()
|
|
|
- {
|
|
|
+ aiFace ReadTriangle() {
|
|
|
aiFace face;
|
|
|
|
|
|
face.mNumIndices = 3;
|
|
@@ -228,45 +267,150 @@ private:
|
|
|
return face;
|
|
|
}
|
|
|
|
|
|
+ void ReadBaseMaterials() {
|
|
|
+ std::vector<unsigned int> MatIdArray;
|
|
|
+ const char *baseMaterialId( xmlReader->getAttributeValue( D3MF::XmlTag::basematerials_id.c_str() ) );
|
|
|
+ if ( nullptr != baseMaterialId ) {
|
|
|
+ unsigned int id = std::atoi( baseMaterialId );
|
|
|
+ const size_t newMatIdx( mMatArray.size() );
|
|
|
+ if ( id != mActiveMatGroup ) {
|
|
|
+ mActiveMatGroup = id;
|
|
|
+ MatId2MatArray::const_iterator it( mMatId2MatArray.find( id ) );
|
|
|
+ if ( mMatId2MatArray.end() == it ) {
|
|
|
+ MatIdArray.clear();
|
|
|
+ mMatId2MatArray[ id ] = MatIdArray;
|
|
|
+ } else {
|
|
|
+ MatIdArray = it->second;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ MatIdArray.push_back( static_cast<unsigned int>( newMatIdx ) );
|
|
|
+ mMatId2MatArray[ mActiveMatGroup ] = MatIdArray;
|
|
|
+ }
|
|
|
+
|
|
|
+ while ( ReadToEndElement( D3MF::XmlTag::basematerials ) ) {
|
|
|
+ mMatArray.push_back( readMaterialDef() );
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ bool parseColor( const char *color, aiColor4D &diffuse ) {
|
|
|
+ if ( nullptr == color ) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
+ const size_t len( strlen( color ) );
|
|
|
+ if ( 9 != len ) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
+ const char *buf( color );
|
|
|
+ if ( '#' != *buf ) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ ++buf;
|
|
|
+ char comp[ 3 ] = { 0,0,'\0' };
|
|
|
+
|
|
|
+ comp[ 0 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ comp[ 1 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ diffuse.r = static_cast<ai_real>( strtol( comp, NULL, 16 ) );
|
|
|
+
|
|
|
+
|
|
|
+ comp[ 0 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ comp[ 1 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ diffuse.g = static_cast< ai_real >( strtol( comp, NULL, 16 ) );
|
|
|
+
|
|
|
+ comp[ 0 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ comp[ 1 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ diffuse.b = static_cast< ai_real >( strtol( comp, NULL, 16 ) );
|
|
|
+
|
|
|
+ comp[ 0 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ comp[ 1 ] = *buf;
|
|
|
+ ++buf;
|
|
|
+ diffuse.a = static_cast< ai_real >( strtol( comp, NULL, 16 ) );
|
|
|
+
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+
|
|
|
+ void assignDiffuseColor( aiMaterial *mat ) {
|
|
|
+ const char *color = xmlReader->getAttributeValue( D3MF::XmlTag::basematerials_displaycolor.c_str() );
|
|
|
+ aiColor4D diffuse;
|
|
|
+ if ( parseColor( color, diffuse ) ) {
|
|
|
+ mat->AddProperty<aiColor4D>( &diffuse, 1, AI_MATKEY_COLOR_DIFFUSE );
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+ aiMaterial *readMaterialDef() {
|
|
|
+ aiMaterial *mat( nullptr );
|
|
|
+ const char *name( nullptr );
|
|
|
+ const std::string nodeName( xmlReader->getNodeName() );
|
|
|
+ if ( nodeName == D3MF::XmlTag::basematerials_base ) {
|
|
|
+ name = xmlReader->getAttributeValue( D3MF::XmlTag::basematerials_name.c_str() );
|
|
|
+ std::string stdMatName;
|
|
|
+ aiString matName;
|
|
|
+ std::string strId( to_string( mActiveMatGroup ) );
|
|
|
+ stdMatName += "id";
|
|
|
+ stdMatName += strId;
|
|
|
+ stdMatName += "_";
|
|
|
+ if ( nullptr != name ) {
|
|
|
+ stdMatName += std::string( name );
|
|
|
+ } else {
|
|
|
+ stdMatName += "basemat";
|
|
|
+ }
|
|
|
+ matName.Set( stdMatName );
|
|
|
+
|
|
|
+ mat = new aiMaterial;
|
|
|
+ mat->AddProperty( &matName, AI_MATKEY_NAME );
|
|
|
+
|
|
|
+ assignDiffuseColor( mat );
|
|
|
+ }
|
|
|
+
|
|
|
+ return mat;
|
|
|
+ }
|
|
|
+
|
|
|
private:
|
|
|
- bool ReadToStartElement(const std::string& startTag)
|
|
|
- {
|
|
|
- while(xmlReader->read())
|
|
|
- {
|
|
|
- if (xmlReader->getNodeType() == irr::io::EXN_ELEMENT && xmlReader->getNodeName() == startTag)
|
|
|
- {
|
|
|
+ bool ReadToStartElement(const std::string& startTag) {
|
|
|
+ while(xmlReader->read()) {
|
|
|
+ const std::string &nodeName( xmlReader->getNodeName() );
|
|
|
+ if (xmlReader->getNodeType() == irr::io::EXN_ELEMENT && nodeName == startTag) {
|
|
|
return true;
|
|
|
- }
|
|
|
- else if (xmlReader->getNodeType() == irr::io::EXN_ELEMENT_END &&
|
|
|
- xmlReader->getNodeName() == startTag)
|
|
|
- {
|
|
|
+ } else if (xmlReader->getNodeType() == irr::io::EXN_ELEMENT_END && nodeName == startTag) {
|
|
|
return false;
|
|
|
}
|
|
|
}
|
|
|
- //DefaultLogger::get()->error("unexpected EOF, expected closing <" + closeTag + "> tag");
|
|
|
+
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
- bool ReadToEndElement(const std::string& closeTag)
|
|
|
- {
|
|
|
- while(xmlReader->read())
|
|
|
- {
|
|
|
+ bool ReadToEndElement(const std::string& closeTag) {
|
|
|
+ while(xmlReader->read()) {
|
|
|
+ const std::string &nodeName( xmlReader->getNodeName() );
|
|
|
if (xmlReader->getNodeType() == irr::io::EXN_ELEMENT) {
|
|
|
return true;
|
|
|
- }
|
|
|
- else if (xmlReader->getNodeType() == irr::io::EXN_ELEMENT_END
|
|
|
- && xmlReader->getNodeName() == closeTag)
|
|
|
- {
|
|
|
+ } else if (xmlReader->getNodeType() == irr::io::EXN_ELEMENT_END && nodeName == closeTag) {
|
|
|
return false;
|
|
|
}
|
|
|
}
|
|
|
DefaultLogger::get()->error("unexpected EOF, expected closing <" + closeTag + "> tag");
|
|
|
+
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
-
|
|
|
private:
|
|
|
- std::vector<aiMesh*> meshes;
|
|
|
+ struct MetaEntry {
|
|
|
+ std::string name;
|
|
|
+ std::string value;
|
|
|
+ };
|
|
|
+ std::vector<MetaEntry> mMetaData;
|
|
|
+ std::vector<aiMesh*> mMeshes;
|
|
|
+ MatArray mMatArray;
|
|
|
+ unsigned int mActiveMatGroup;
|
|
|
+ MatId2MatArray mMatId2MatArray;
|
|
|
XmlReader* xmlReader;
|
|
|
};
|
|
|
|
|
@@ -287,7 +431,6 @@ static const aiImporterDesc desc = {
|
|
|
Extension.c_str()
|
|
|
};
|
|
|
|
|
|
-
|
|
|
D3MFImporter::D3MFImporter()
|
|
|
: BaseImporter() {
|
|
|
// empty
|
|
@@ -297,14 +440,19 @@ D3MFImporter::~D3MFImporter() {
|
|
|
// empty
|
|
|
}
|
|
|
|
|
|
-bool D3MFImporter::CanRead(const std::string &pFile, IOSystem *pIOHandler, bool checkSig) const {
|
|
|
- const std::string extension = GetExtension(pFile);
|
|
|
+bool D3MFImporter::CanRead(const std::string &filename, IOSystem *pIOHandler, bool checkSig) const {
|
|
|
+ const std::string extension( GetExtension( filename ) );
|
|
|
if(extension == Extension ) {
|
|
|
return true;
|
|
|
} else if ( !extension.length() || checkSig ) {
|
|
|
- if (nullptr == pIOHandler ) {
|
|
|
- return true;
|
|
|
+ if ( nullptr == pIOHandler ) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ if ( !D3MF::D3MFOpcPackage::isZipArchive( pIOHandler, filename ) ) {
|
|
|
+ return false;
|
|
|
}
|
|
|
+ D3MF::D3MFOpcPackage opcPackage( pIOHandler, filename );
|
|
|
+ return opcPackage.validate();
|
|
|
}
|
|
|
|
|
|
return false;
|
|
@@ -318,8 +466,8 @@ const aiImporterDesc *D3MFImporter::GetInfo() const {
|
|
|
return &desc;
|
|
|
}
|
|
|
|
|
|
-void D3MFImporter::InternReadFile(const std::string &pFile, aiScene *pScene, IOSystem *pIOHandler) {
|
|
|
- D3MF::D3MFOpcPackage opcPackage(pIOHandler, pFile);
|
|
|
+void D3MFImporter::InternReadFile( const std::string &filename, aiScene *pScene, IOSystem *pIOHandler ) {
|
|
|
+ D3MF::D3MFOpcPackage opcPackage(pIOHandler, filename);
|
|
|
|
|
|
std::unique_ptr<CIrrXML_IOStreamReader> xmlStream(new CIrrXML_IOStreamReader(opcPackage.RootStream()));
|
|
|
std::unique_ptr<D3MF::XmlReader> xmlReader(irr::io::createIrrXMLReader(xmlStream.get()));
|