12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397 |
- /*
- ---------------------------------------------------------------------------
- Open Asset Import Library (assimp)
- ---------------------------------------------------------------------------
- Copyright (c) 2006-2021, assimp team
- 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.
- ---------------------------------------------------------------------------
- */
- /** @file Implementation of the XFile parser helper class */
- #ifndef ASSIMP_BUILD_NO_X_IMPORTER
- #include "XFileParser.h"
- #include "XFileHelper.h"
- #include <assimp/ByteSwapper.h>
- #include <assimp/Exceptional.h>
- #include <assimp/StringUtils.h>
- #include <assimp/TinyFormatter.h>
- #include <assimp/fast_atof.h>
- #include <assimp/DefaultLogger.hpp>
- using namespace Assimp;
- using namespace Assimp::XFile;
- using namespace Assimp::Formatter;
- #ifndef ASSIMP_BUILD_NO_COMPRESSED_X
- #ifdef ASSIMP_BUILD_NO_OWN_ZLIB
- #include <zlib.h>
- #else
- #include "../contrib/zlib/zlib.h"
- #endif
- // Magic identifier for MSZIP compressed data
- #define MSZIP_MAGIC 0x4B43
- #define MSZIP_BLOCK 32786
- // ------------------------------------------------------------------------------------------------
- // Dummy memory wrappers for use with zlib
- static void *dummy_alloc(void * /*opaque*/, unsigned int items, unsigned int size) {
- return ::operator new(items *size);
- }
- static void dummy_free(void * /*opaque*/, void *address) {
- return ::operator delete(address);
- }
- #endif // !! ASSIMP_BUILD_NO_COMPRESSED_X
- // ------------------------------------------------------------------------------------------------
- // Throws an exception with a line number and the given text.
- template<typename... T>
- AI_WONT_RETURN void XFileParser::ThrowException(T&&... args) {
- if (mIsBinaryFormat) {
- throw DeadlyImportError(args...);
- } else {
- throw DeadlyImportError("Line ", mLineNumber, ": ", args...);
- }
- }
- // ------------------------------------------------------------------------------------------------
- // Constructor. Creates a data structure out of the XFile given in the memory block.
- XFileParser::XFileParser(const std::vector<char> &pBuffer) :
- mMajorVersion(0), mMinorVersion(0), mIsBinaryFormat(false), mBinaryNumCount(0), mP(nullptr), mEnd(nullptr), mLineNumber(0), mScene(nullptr) {
- // vector to store uncompressed file for INFLATE'd X files
- std::vector<char> uncompressed;
- // set up memory pointers
- mP = &pBuffer.front();
- mEnd = mP + pBuffer.size() - 1;
- // check header
- if (0 != strncmp(mP, "xof ", 4)) {
- throw DeadlyImportError("Header mismatch, file is not an XFile.");
- }
- // read version. It comes in a four byte format such as "0302"
- mMajorVersion = (unsigned int)(mP[4] - 48) * 10 + (unsigned int)(mP[5] - 48);
- mMinorVersion = (unsigned int)(mP[6] - 48) * 10 + (unsigned int)(mP[7] - 48);
- bool compressed = false;
- // txt - pure ASCII text format
- if (strncmp(mP + 8, "txt ", 4) == 0)
- mIsBinaryFormat = false;
- // bin - Binary format
- else if (strncmp(mP + 8, "bin ", 4) == 0)
- mIsBinaryFormat = true;
- // tzip - Inflate compressed text format
- else if (strncmp(mP + 8, "tzip", 4) == 0) {
- mIsBinaryFormat = false;
- compressed = true;
- }
- // bzip - Inflate compressed binary format
- else if (strncmp(mP + 8, "bzip", 4) == 0) {
- mIsBinaryFormat = true;
- compressed = true;
- } else
- ThrowException("Unsupported xfile format '", mP[8], mP[9], mP[10], mP[11], "'");
- // float size
- mBinaryFloatSize = (unsigned int)(mP[12] - 48) * 1000 + (unsigned int)(mP[13] - 48) * 100 + (unsigned int)(mP[14] - 48) * 10 + (unsigned int)(mP[15] - 48);
- if (mBinaryFloatSize != 32 && mBinaryFloatSize != 64)
- ThrowException("Unknown float size ", mBinaryFloatSize, " specified in xfile header.");
- // The x format specifies size in bits, but we work in bytes
- mBinaryFloatSize /= 8;
- mP += 16;
- // If this is a compressed X file, apply the inflate algorithm to it
- if (compressed) {
- #ifdef ASSIMP_BUILD_NO_COMPRESSED_X
- throw DeadlyImportError("Assimp was built without compressed X support");
- #else
- /* ///////////////////////////////////////////////////////////////////////
- * COMPRESSED X FILE FORMAT
- * ///////////////////////////////////////////////////////////////////////
- * [xhead]
- * 2 major
- * 2 minor
- * 4 type // bzip,tzip
- * [mszip_master_head]
- * 4 unkn // checksum?
- * 2 unkn // flags? (seems to be constant)
- * [mszip_head]
- * 2 ofs // offset to next section
- * 2 magic // 'CK'
- * ... ofs bytes of data
- * ... next mszip_head
- *
- * http://www.kdedevelopers.org/node/3181 has been very helpful.
- * ///////////////////////////////////////////////////////////////////////
- */
- // build a zlib stream
- z_stream stream;
- stream.opaque = nullptr;
- stream.zalloc = &dummy_alloc;
- stream.zfree = &dummy_free;
- stream.data_type = (mIsBinaryFormat ? Z_BINARY : Z_ASCII);
- // initialize the inflation algorithm
- ::inflateInit2(&stream, -MAX_WBITS);
- // skip unknown data (checksum, flags?)
- mP += 6;
- // First find out how much storage we'll need. Count sections.
- const char *P1 = mP;
- unsigned int est_out = 0;
- while (P1 + 3 < mEnd) {
- // read next offset
- uint16_t ofs = *((uint16_t *)P1);
- AI_SWAP2(ofs);
- P1 += 2;
- if (ofs >= MSZIP_BLOCK)
- throw DeadlyImportError("X: Invalid offset to next MSZIP compressed block");
- // check magic word
- uint16_t magic = *((uint16_t *)P1);
- AI_SWAP2(magic);
- P1 += 2;
- if (magic != MSZIP_MAGIC)
- throw DeadlyImportError("X: Unsupported compressed format, expected MSZIP header");
- // and advance to the next offset
- P1 += ofs;
- est_out += MSZIP_BLOCK; // one decompressed block is 32786 in size
- }
- // Allocate storage and terminating zero and do the actual uncompressing
- uncompressed.resize(est_out + 1);
- char *out = &uncompressed.front();
- while (mP + 3 < mEnd) {
- uint16_t ofs = *((uint16_t *)mP);
- AI_SWAP2(ofs);
- mP += 4;
- if (mP + ofs > mEnd + 2) {
- throw DeadlyImportError("X: Unexpected EOF in compressed chunk");
- }
- // push data to the stream
- stream.next_in = (Bytef *)mP;
- stream.avail_in = ofs;
- stream.next_out = (Bytef *)out;
- stream.avail_out = MSZIP_BLOCK;
- // and decompress the data ....
- int ret = ::inflate(&stream, Z_SYNC_FLUSH);
- if (ret != Z_OK && ret != Z_STREAM_END)
- throw DeadlyImportError("X: Failed to decompress MSZIP-compressed data");
- ::inflateReset(&stream);
- ::inflateSetDictionary(&stream, (const Bytef *)out, MSZIP_BLOCK - stream.avail_out);
- // and advance to the next offset
- out += MSZIP_BLOCK - stream.avail_out;
- mP += ofs;
- }
- // terminate zlib
- ::inflateEnd(&stream);
- // ok, update pointers to point to the uncompressed file data
- mP = &uncompressed[0];
- mEnd = out;
- // FIXME: we don't need the compressed data anymore, could release
- // it already for better memory usage. Consider breaking const-co.
- ASSIMP_LOG_INFO("Successfully decompressed MSZIP-compressed file");
- #endif // !! ASSIMP_BUILD_NO_COMPRESSED_X
- } else {
- // start reading here
- ReadUntilEndOfLine();
- }
- mScene = new Scene;
- ParseFile();
- // filter the imported hierarchy for some degenerated cases
- if (mScene->mRootNode) {
- FilterHierarchy(mScene->mRootNode);
- }
- }
- // ------------------------------------------------------------------------------------------------
- // Destructor. Destroys all imported data along with it
- XFileParser::~XFileParser() {
- // kill everything we created
- delete mScene;
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseFile() {
- bool running = true;
- while (running) {
- // read name of next object
- std::string objectName = GetNextToken();
- if (objectName.length() == 0)
- break;
- // parse specific object
- if (objectName == "template")
- ParseDataObjectTemplate();
- else if (objectName == "Frame")
- ParseDataObjectFrame(nullptr);
- else if (objectName == "Mesh") {
- // some meshes have no frames at all
- Mesh *mesh = new Mesh;
- ParseDataObjectMesh(mesh);
- mScene->mGlobalMeshes.push_back(mesh);
- } else if (objectName == "AnimTicksPerSecond")
- ParseDataObjectAnimTicksPerSecond();
- else if (objectName == "AnimationSet")
- ParseDataObjectAnimationSet();
- else if (objectName == "Material") {
- // Material outside of a mesh or node
- Material material;
- ParseDataObjectMaterial(&material);
- mScene->mGlobalMaterials.push_back(material);
- } else if (objectName == "}") {
- // whatever?
- ASSIMP_LOG_WARN("} found in dataObject");
- } else {
- // unknown format
- ASSIMP_LOG_WARN("Unknown data object in animation of .x file");
- ParseUnknownDataObject();
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectTemplate() {
- // parse a template data object. Currently not stored.
- std::string name;
- readHeadOfDataObject(&name);
- // read GUID
- std::string guid = GetNextToken();
- // read and ignore data members
- bool running = true;
- while (running) {
- std::string s = GetNextToken();
- if (s == "}")
- break;
- if (s.length() == 0)
- ThrowException("Unexpected end of file reached while parsing template definition");
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectFrame(Node *pParent) {
- // A coordinate frame, or "frame of reference." The Frame template
- // is open and can contain any object. The Direct3D extensions (D3DX)
- // mesh-loading functions recognize Mesh, FrameTransformMatrix, and
- // Frame template instances as child objects when loading a Frame
- // instance.
- std::string name;
- readHeadOfDataObject(&name);
- // create a named node and place it at its parent, if given
- Node *node = new Node(pParent);
- node->mName = name;
- if (pParent) {
- pParent->mChildren.push_back(node);
- } else {
- // there might be multiple root nodes
- if (mScene->mRootNode != nullptr) {
- // place a dummy root if not there
- if (mScene->mRootNode->mName != "$dummy_root") {
- Node *exroot = mScene->mRootNode;
- mScene->mRootNode = new Node(nullptr);
- mScene->mRootNode->mName = "$dummy_root";
- mScene->mRootNode->mChildren.push_back(exroot);
- exroot->mParent = mScene->mRootNode;
- }
- // put the new node as its child instead
- mScene->mRootNode->mChildren.push_back(node);
- node->mParent = mScene->mRootNode;
- } else {
- // it's the first node imported. place it as root
- mScene->mRootNode = node;
- }
- }
- // Now inside a frame.
- // read tokens until closing brace is reached.
- bool running = true;
- while (running) {
- std::string objectName = GetNextToken();
- if (objectName.size() == 0)
- ThrowException("Unexpected end of file reached while parsing frame");
- if (objectName == "}")
- break; // frame finished
- else if (objectName == "Frame")
- ParseDataObjectFrame(node); // child frame
- else if (objectName == "FrameTransformMatrix")
- ParseDataObjectTransformationMatrix(node->mTrafoMatrix);
- else if (objectName == "Mesh") {
- Mesh *mesh = new Mesh(name);
- node->mMeshes.push_back(mesh);
- ParseDataObjectMesh(mesh);
- } else {
- ASSIMP_LOG_WARN("Unknown data object in frame in x file");
- ParseUnknownDataObject();
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectTransformationMatrix(aiMatrix4x4 &pMatrix) {
- // read header, we're not interested if it has a name
- readHeadOfDataObject();
- // read its components
- pMatrix.a1 = ReadFloat();
- pMatrix.b1 = ReadFloat();
- pMatrix.c1 = ReadFloat();
- pMatrix.d1 = ReadFloat();
- pMatrix.a2 = ReadFloat();
- pMatrix.b2 = ReadFloat();
- pMatrix.c2 = ReadFloat();
- pMatrix.d2 = ReadFloat();
- pMatrix.a3 = ReadFloat();
- pMatrix.b3 = ReadFloat();
- pMatrix.c3 = ReadFloat();
- pMatrix.d3 = ReadFloat();
- pMatrix.a4 = ReadFloat();
- pMatrix.b4 = ReadFloat();
- pMatrix.c4 = ReadFloat();
- pMatrix.d4 = ReadFloat();
- // trailing symbols
- CheckForSemicolon();
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectMesh(Mesh *pMesh) {
- std::string name;
- readHeadOfDataObject(&name);
- // read vertex count
- unsigned int numVertices = ReadInt();
- pMesh->mPositions.resize(numVertices);
- // read vertices
- for (unsigned int a = 0; a < numVertices; a++)
- pMesh->mPositions[a] = ReadVector3();
- // read position faces
- unsigned int numPosFaces = ReadInt();
- pMesh->mPosFaces.resize(numPosFaces);
- for (unsigned int a = 0; a < numPosFaces; ++a) {
- // read indices
- unsigned int numIndices = ReadInt();
- Face &face = pMesh->mPosFaces[a];
- for (unsigned int b = 0; b < numIndices; ++b) {
- const int idx(ReadInt());
- if (static_cast<unsigned int>(idx) <= numVertices) {
- face.mIndices.push_back(idx);
- }
- }
- TestForSeparator();
- }
- // here, other data objects may follow
- bool running = true;
- while (running) {
- std::string objectName = GetNextToken();
- if (objectName.empty())
- ThrowException("Unexpected end of file while parsing mesh structure");
- else if (objectName == "}")
- break; // mesh finished
- else if (objectName == "MeshNormals")
- ParseDataObjectMeshNormals(pMesh);
- else if (objectName == "MeshTextureCoords")
- ParseDataObjectMeshTextureCoords(pMesh);
- else if (objectName == "MeshVertexColors")
- ParseDataObjectMeshVertexColors(pMesh);
- else if (objectName == "MeshMaterialList")
- ParseDataObjectMeshMaterialList(pMesh);
- else if (objectName == "VertexDuplicationIndices")
- ParseUnknownDataObject(); // we'll ignore vertex duplication indices
- else if (objectName == "XSkinMeshHeader")
- ParseDataObjectSkinMeshHeader(pMesh);
- else if (objectName == "SkinWeights")
- ParseDataObjectSkinWeights(pMesh);
- else {
- ASSIMP_LOG_WARN("Unknown data object in mesh in x file");
- ParseUnknownDataObject();
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectSkinWeights(Mesh *pMesh) {
- if (nullptr == pMesh) {
- return;
- }
- readHeadOfDataObject();
- std::string transformNodeName;
- GetNextTokenAsString(transformNodeName);
- pMesh->mBones.push_back(Bone());
- Bone &bone = pMesh->mBones.back();
- bone.mName = transformNodeName;
- // read vertex weights
- unsigned int numWeights = ReadInt();
- bone.mWeights.reserve(numWeights);
- for (unsigned int a = 0; a < numWeights; a++) {
- BoneWeight weight;
- weight.mVertex = ReadInt();
- bone.mWeights.push_back(weight);
- }
- // read vertex weights
- for (unsigned int a = 0; a < numWeights; a++)
- bone.mWeights[a].mWeight = ReadFloat();
- // read matrix offset
- bone.mOffsetMatrix.a1 = ReadFloat();
- bone.mOffsetMatrix.b1 = ReadFloat();
- bone.mOffsetMatrix.c1 = ReadFloat();
- bone.mOffsetMatrix.d1 = ReadFloat();
- bone.mOffsetMatrix.a2 = ReadFloat();
- bone.mOffsetMatrix.b2 = ReadFloat();
- bone.mOffsetMatrix.c2 = ReadFloat();
- bone.mOffsetMatrix.d2 = ReadFloat();
- bone.mOffsetMatrix.a3 = ReadFloat();
- bone.mOffsetMatrix.b3 = ReadFloat();
- bone.mOffsetMatrix.c3 = ReadFloat();
- bone.mOffsetMatrix.d3 = ReadFloat();
- bone.mOffsetMatrix.a4 = ReadFloat();
- bone.mOffsetMatrix.b4 = ReadFloat();
- bone.mOffsetMatrix.c4 = ReadFloat();
- bone.mOffsetMatrix.d4 = ReadFloat();
- CheckForSemicolon();
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectSkinMeshHeader(Mesh * /*pMesh*/) {
- readHeadOfDataObject();
- /*unsigned int maxSkinWeightsPerVertex =*/ReadInt();
- /*unsigned int maxSkinWeightsPerFace =*/ReadInt();
- /*unsigned int numBonesInMesh = */ ReadInt();
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectMeshNormals(Mesh *pMesh) {
- readHeadOfDataObject();
- // read count
- unsigned int numNormals = ReadInt();
- pMesh->mNormals.resize(numNormals);
- // read normal vectors
- for (unsigned int a = 0; a < numNormals; ++a) {
- pMesh->mNormals[a] = ReadVector3();
- }
- // read normal indices
- unsigned int numFaces = ReadInt();
- if (numFaces != pMesh->mPosFaces.size()) {
- ThrowException("Normal face count does not match vertex face count.");
- }
- // do not crah when no face definitions are there
- if (numFaces > 0) {
- // normal face creation
- pMesh->mNormFaces.resize(numFaces);
- for (unsigned int a = 0; a < numFaces; ++a) {
- unsigned int numIndices = ReadInt();
- pMesh->mNormFaces[a] = Face();
- Face &face = pMesh->mNormFaces[a];
- for (unsigned int b = 0; b < numIndices; ++b) {
- face.mIndices.push_back(ReadInt());
- }
- TestForSeparator();
- }
- }
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectMeshTextureCoords(Mesh *pMesh) {
- readHeadOfDataObject();
- if (pMesh->mNumTextures + 1 > AI_MAX_NUMBER_OF_TEXTURECOORDS)
- ThrowException("Too many sets of texture coordinates");
- std::vector<aiVector2D> &coords = pMesh->mTexCoords[pMesh->mNumTextures++];
- unsigned int numCoords = ReadInt();
- if (numCoords != pMesh->mPositions.size())
- ThrowException("Texture coord count does not match vertex count");
- coords.resize(numCoords);
- for (unsigned int a = 0; a < numCoords; a++)
- coords[a] = ReadVector2();
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectMeshVertexColors(Mesh *pMesh) {
- readHeadOfDataObject();
- if (pMesh->mNumColorSets + 1 > AI_MAX_NUMBER_OF_COLOR_SETS)
- ThrowException("Too many colorsets");
- std::vector<aiColor4D> &colors = pMesh->mColors[pMesh->mNumColorSets++];
- unsigned int numColors = ReadInt();
- if (numColors != pMesh->mPositions.size())
- ThrowException("Vertex color count does not match vertex count");
- colors.resize(numColors, aiColor4D(0, 0, 0, 1));
- for (unsigned int a = 0; a < numColors; a++) {
- unsigned int index = ReadInt();
- if (index >= pMesh->mPositions.size())
- ThrowException("Vertex color index out of bounds");
- colors[index] = ReadRGBA();
- // HACK: (thom) Maxon Cinema XPort plugin puts a third separator here, kwxPort puts a comma.
- // Ignore gracefully.
- if (!mIsBinaryFormat) {
- FindNextNoneWhiteSpace();
- if (*mP == ';' || *mP == ',')
- mP++;
- }
- }
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectMeshMaterialList(Mesh *pMesh) {
- readHeadOfDataObject();
- // read material count
- /*unsigned int numMaterials =*/ReadInt();
- // read non triangulated face material index count
- unsigned int numMatIndices = ReadInt();
- // some models have a material index count of 1... to be able to read them we
- // replicate this single material index on every face
- if (numMatIndices != pMesh->mPosFaces.size() && numMatIndices != 1)
- ThrowException("Per-Face material index count does not match face count.");
- // read per-face material indices
- for (unsigned int a = 0; a < numMatIndices; a++)
- pMesh->mFaceMaterials.push_back(ReadInt());
- // in version 03.02, the face indices end with two semicolons.
- // commented out version check, as version 03.03 exported from blender also has 2 semicolons
- if (!mIsBinaryFormat) // && MajorVersion == 3 && MinorVersion <= 2)
- {
- if (mP < mEnd && *mP == ';')
- ++mP;
- }
- // if there was only a single material index, replicate it on all faces
- while (pMesh->mFaceMaterials.size() < pMesh->mPosFaces.size())
- pMesh->mFaceMaterials.push_back(pMesh->mFaceMaterials.front());
- // read following data objects
- bool running = true;
- while (running) {
- std::string objectName = GetNextToken();
- if (objectName.size() == 0)
- ThrowException("Unexpected end of file while parsing mesh material list.");
- else if (objectName == "}")
- break; // material list finished
- else if (objectName == "{") {
- // template materials
- std::string matName = GetNextToken();
- Material material;
- material.mIsReference = true;
- material.mName = matName;
- pMesh->mMaterials.push_back(material);
- CheckForClosingBrace(); // skip }
- } else if (objectName == "Material") {
- pMesh->mMaterials.push_back(Material());
- ParseDataObjectMaterial(&pMesh->mMaterials.back());
- } else if (objectName == ";") {
- // ignore
- } else {
- ASSIMP_LOG_WARN("Unknown data object in material list in x file");
- ParseUnknownDataObject();
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectMaterial(Material *pMaterial) {
- std::string matName;
- readHeadOfDataObject(&matName);
- if (matName.empty())
- matName = std::string("material") + ai_to_string(mLineNumber);
- pMaterial->mName = matName;
- pMaterial->mIsReference = false;
- // read material values
- pMaterial->mDiffuse = ReadRGBA();
- pMaterial->mSpecularExponent = ReadFloat();
- pMaterial->mSpecular = ReadRGB();
- pMaterial->mEmissive = ReadRGB();
- // read other data objects
- bool running = true;
- while (running) {
- std::string objectName = GetNextToken();
- if (objectName.size() == 0)
- ThrowException("Unexpected end of file while parsing mesh material");
- else if (objectName == "}")
- break; // material finished
- else if (objectName == "TextureFilename" || objectName == "TextureFileName") {
- // some exporters write "TextureFileName" instead.
- std::string texname;
- ParseDataObjectTextureFilename(texname);
- pMaterial->mTextures.push_back(TexEntry(texname));
- } else if (objectName == "NormalmapFilename" || objectName == "NormalmapFileName") {
- // one exporter writes out the normal map in a separate filename tag
- std::string texname;
- ParseDataObjectTextureFilename(texname);
- pMaterial->mTextures.push_back(TexEntry(texname, true));
- } else {
- ASSIMP_LOG_WARN("Unknown data object in material in x file");
- ParseUnknownDataObject();
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectAnimTicksPerSecond() {
- readHeadOfDataObject();
- mScene->mAnimTicksPerSecond = ReadInt();
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectAnimationSet() {
- std::string animName;
- readHeadOfDataObject(&animName);
- Animation *anim = new Animation;
- mScene->mAnims.push_back(anim);
- anim->mName = animName;
- bool running = true;
- while (running) {
- std::string objectName = GetNextToken();
- if (objectName.length() == 0)
- ThrowException("Unexpected end of file while parsing animation set.");
- else if (objectName == "}")
- break; // animation set finished
- else if (objectName == "Animation")
- ParseDataObjectAnimation(anim);
- else {
- ASSIMP_LOG_WARN("Unknown data object in animation set in x file");
- ParseUnknownDataObject();
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectAnimation(Animation *pAnim) {
- readHeadOfDataObject();
- AnimBone *banim = new AnimBone;
- pAnim->mAnims.push_back(banim);
- bool running = true;
- while (running) {
- std::string objectName = GetNextToken();
- if (objectName.length() == 0)
- ThrowException("Unexpected end of file while parsing animation.");
- else if (objectName == "}")
- break; // animation finished
- else if (objectName == "AnimationKey")
- ParseDataObjectAnimationKey(banim);
- else if (objectName == "AnimationOptions")
- ParseUnknownDataObject(); // not interested
- else if (objectName == "{") {
- // read frame name
- banim->mBoneName = GetNextToken();
- CheckForClosingBrace();
- } else {
- ASSIMP_LOG_WARN("Unknown data object in animation in x file");
- ParseUnknownDataObject();
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectAnimationKey(AnimBone *pAnimBone) {
- readHeadOfDataObject();
- // read key type
- unsigned int keyType = ReadInt();
- // read number of keys
- unsigned int numKeys = ReadInt();
- for (unsigned int a = 0; a < numKeys; a++) {
- // read time
- unsigned int time = ReadInt();
- // read keys
- switch (keyType) {
- case 0: // rotation quaternion
- {
- // read count
- if (ReadInt() != 4)
- ThrowException("Invalid number of arguments for quaternion key in animation");
- aiQuatKey key;
- key.mTime = double(time);
- key.mValue.w = ReadFloat();
- key.mValue.x = ReadFloat();
- key.mValue.y = ReadFloat();
- key.mValue.z = ReadFloat();
- pAnimBone->mRotKeys.push_back(key);
- CheckForSemicolon();
- break;
- }
- case 1: // scale vector
- case 2: // position vector
- {
- // read count
- if (ReadInt() != 3)
- ThrowException("Invalid number of arguments for vector key in animation");
- aiVectorKey key;
- key.mTime = double(time);
- key.mValue = ReadVector3();
- if (keyType == 2)
- pAnimBone->mPosKeys.push_back(key);
- else
- pAnimBone->mScaleKeys.push_back(key);
- break;
- }
- case 3: // combined transformation matrix
- case 4: // denoted both as 3 or as 4
- {
- // read count
- if (ReadInt() != 16)
- ThrowException("Invalid number of arguments for matrix key in animation");
- // read matrix
- MatrixKey key;
- key.mTime = double(time);
- key.mMatrix.a1 = ReadFloat();
- key.mMatrix.b1 = ReadFloat();
- key.mMatrix.c1 = ReadFloat();
- key.mMatrix.d1 = ReadFloat();
- key.mMatrix.a2 = ReadFloat();
- key.mMatrix.b2 = ReadFloat();
- key.mMatrix.c2 = ReadFloat();
- key.mMatrix.d2 = ReadFloat();
- key.mMatrix.a3 = ReadFloat();
- key.mMatrix.b3 = ReadFloat();
- key.mMatrix.c3 = ReadFloat();
- key.mMatrix.d3 = ReadFloat();
- key.mMatrix.a4 = ReadFloat();
- key.mMatrix.b4 = ReadFloat();
- key.mMatrix.c4 = ReadFloat();
- key.mMatrix.d4 = ReadFloat();
- pAnimBone->mTrafoKeys.push_back(key);
- CheckForSemicolon();
- break;
- }
- default:
- ThrowException("Unknown key type ", keyType, " in animation.");
- break;
- } // end switch
- // key separator
- CheckForSeparator();
- }
- CheckForClosingBrace();
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseDataObjectTextureFilename(std::string &pName) {
- readHeadOfDataObject();
- GetNextTokenAsString(pName);
- CheckForClosingBrace();
- // FIX: some files (e.g. AnimationTest.x) have "" as texture file name
- if (!pName.length()) {
- ASSIMP_LOG_WARN("Length of texture file name is zero. Skipping this texture.");
- }
- // some exporters write double backslash paths out. We simply replace them if we find them
- while (pName.find("\\\\") != std::string::npos)
- pName.replace(pName.find("\\\\"), 2, "\\");
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ParseUnknownDataObject() {
- // find opening delimiter
- bool running = true;
- while (running) {
- std::string t = GetNextToken();
- if (t.length() == 0)
- ThrowException("Unexpected end of file while parsing unknown segment.");
- if (t == "{")
- break;
- }
- unsigned int counter = 1;
- // parse until closing delimiter
- while (counter > 0) {
- std::string t = GetNextToken();
- if (t.length() == 0)
- ThrowException("Unexpected end of file while parsing unknown segment.");
- if (t == "{")
- ++counter;
- else if (t == "}")
- --counter;
- }
- }
- // ------------------------------------------------------------------------------------------------
- //! checks for closing curly brace
- void XFileParser::CheckForClosingBrace() {
- if (GetNextToken() != "}")
- ThrowException("Closing brace expected.");
- }
- // ------------------------------------------------------------------------------------------------
- //! checks for one following semicolon
- void XFileParser::CheckForSemicolon() {
- if (mIsBinaryFormat)
- return;
- if (GetNextToken() != ";")
- ThrowException("Semicolon expected.");
- }
- // ------------------------------------------------------------------------------------------------
- //! checks for a separator char, either a ',' or a ';'
- void XFileParser::CheckForSeparator() {
- if (mIsBinaryFormat)
- return;
- std::string token = GetNextToken();
- if (token != "," && token != ";")
- ThrowException("Separator character (';' or ',') expected.");
- }
- // ------------------------------------------------------------------------------------------------
- // tests and possibly consumes a separator char, but does nothing if there was no separator
- void XFileParser::TestForSeparator() {
- if (mIsBinaryFormat)
- return;
- FindNextNoneWhiteSpace();
- if (mP >= mEnd)
- return;
- // test and skip
- if (*mP == ';' || *mP == ',')
- mP++;
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::readHeadOfDataObject(std::string *poName) {
- std::string nameOrBrace = GetNextToken();
- if (nameOrBrace != "{") {
- if (poName)
- *poName = nameOrBrace;
- if (GetNextToken() != "{") {
- delete mScene;
- ThrowException("Opening brace expected.");
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- std::string XFileParser::GetNextToken() {
- std::string s;
- // process binary-formatted file
- if (mIsBinaryFormat) {
- // in binary mode it will only return NAME and STRING token
- // and (correctly) skip over other tokens.
- if (mEnd - mP < 2) {
- return s;
- }
- unsigned int tok = ReadBinWord();
- unsigned int len;
- // standalone tokens
- switch (tok) {
- case 1: {
- // name token
- if (mEnd - mP < 4) {
- return s;
- }
- len = ReadBinDWord();
- const int bounds = int(mEnd - mP);
- const int iLen = int(len);
- if (iLen < 0) {
- return s;
- }
- if (bounds < iLen) {
- return s;
- }
- s = std::string(mP, len);
- mP += len;
- }
- return s;
- case 2:
- // string token
- if (mEnd - mP < 4) return s;
- len = ReadBinDWord();
- if (mEnd - mP < int(len)) return s;
- s = std::string(mP, len);
- mP += (len + 2);
- return s;
- case 3:
- // integer token
- mP += 4;
- return "<integer>";
- case 5:
- // GUID token
- mP += 16;
- return "<guid>";
- case 6:
- if (mEnd - mP < 4) return s;
- len = ReadBinDWord();
- mP += (len * 4);
- return "<int_list>";
- case 7:
- if (mEnd - mP < 4) return s;
- len = ReadBinDWord();
- mP += (len * mBinaryFloatSize);
- return "<flt_list>";
- case 0x0a:
- return "{";
- case 0x0b:
- return "}";
- case 0x0c:
- return "(";
- case 0x0d:
- return ")";
- case 0x0e:
- return "[";
- case 0x0f:
- return "]";
- case 0x10:
- return "<";
- case 0x11:
- return ">";
- case 0x12:
- return ".";
- case 0x13:
- return ",";
- case 0x14:
- return ";";
- case 0x1f:
- return "template";
- case 0x28:
- return "WORD";
- case 0x29:
- return "DWORD";
- case 0x2a:
- return "FLOAT";
- case 0x2b:
- return "DOUBLE";
- case 0x2c:
- return "CHAR";
- case 0x2d:
- return "UCHAR";
- case 0x2e:
- return "SWORD";
- case 0x2f:
- return "SDWORD";
- case 0x30:
- return "void";
- case 0x31:
- return "string";
- case 0x32:
- return "unicode";
- case 0x33:
- return "cstring";
- case 0x34:
- return "array";
- }
- }
- // process text-formatted file
- else {
- FindNextNoneWhiteSpace();
- if (mP >= mEnd)
- return s;
- while ((mP < mEnd) && !isspace((unsigned char)*mP)) {
- // either keep token delimiters when already holding a token, or return if first valid char
- if (*mP == ';' || *mP == '}' || *mP == '{' || *mP == ',') {
- if (!s.size())
- s.append(mP++, 1);
- break; // stop for delimiter
- }
- s.append(mP++, 1);
- }
- }
- return s;
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::FindNextNoneWhiteSpace() {
- if (mIsBinaryFormat)
- return;
- bool running = true;
- while (running) {
- while (mP < mEnd && isspace((unsigned char)*mP)) {
- if (*mP == '\n')
- mLineNumber++;
- ++mP;
- }
- if (mP >= mEnd)
- return;
- // check if this is a comment
- if ((mP[0] == '/' && mP[1] == '/') || mP[0] == '#')
- ReadUntilEndOfLine();
- else
- break;
- }
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::GetNextTokenAsString(std::string &poString) {
- if (mIsBinaryFormat) {
- poString = GetNextToken();
- return;
- }
- FindNextNoneWhiteSpace();
- if (mP >= mEnd) {
- delete mScene;
- ThrowException("Unexpected end of file while parsing string");
- }
- if (*mP != '"') {
- delete mScene;
- ThrowException("Expected quotation mark.");
- }
- ++mP;
- while (mP < mEnd && *mP != '"')
- poString.append(mP++, 1);
- if (mP >= mEnd - 1) {
- delete mScene;
- ThrowException("Unexpected end of file while parsing string");
- }
- if (mP[1] != ';' || mP[0] != '"') {
- delete mScene;
- ThrowException("Expected quotation mark and semicolon at the end of a string.");
- }
- mP += 2;
- }
- // ------------------------------------------------------------------------------------------------
- void XFileParser::ReadUntilEndOfLine() {
- if (mIsBinaryFormat)
- return;
- while (mP < mEnd) {
- if (*mP == '\n' || *mP == '\r') {
- ++mP;
- mLineNumber++;
- return;
- }
- ++mP;
- }
- }
- // ------------------------------------------------------------------------------------------------
- unsigned short XFileParser::ReadBinWord() {
- ai_assert(mEnd - mP >= 2);
- const unsigned char *q = (const unsigned char *)mP;
- unsigned short tmp = q[0] | (q[1] << 8);
- mP += 2;
- return tmp;
- }
- // ------------------------------------------------------------------------------------------------
- unsigned int XFileParser::ReadBinDWord() {
- ai_assert(mEnd - mP >= 4);
- const unsigned char *q = (const unsigned char *)mP;
- unsigned int tmp = q[0] | (q[1] << 8) | (q[2] << 16) | (q[3] << 24);
- mP += 4;
- return tmp;
- }
- // ------------------------------------------------------------------------------------------------
- unsigned int XFileParser::ReadInt() {
- if (mIsBinaryFormat) {
- if (mBinaryNumCount == 0 && mEnd - mP >= 2) {
- unsigned short tmp = ReadBinWord(); // 0x06 or 0x03
- if (tmp == 0x06 && mEnd - mP >= 4) // array of ints follows
- mBinaryNumCount = ReadBinDWord();
- else // single int follows
- mBinaryNumCount = 1;
- }
- --mBinaryNumCount;
- const size_t len(mEnd - mP);
- if (len >= 4) {
- return ReadBinDWord();
- } else {
- mP = mEnd;
- return 0;
- }
- } else {
- FindNextNoneWhiteSpace();
- // TODO: consider using strtol10 instead???
- // check preceding minus sign
- bool isNegative = false;
- if (*mP == '-') {
- isNegative = true;
- mP++;
- }
- // at least one digit expected
- if (!isdigit((unsigned char)*mP))
- ThrowException("Number expected.");
- // read digits
- unsigned int number = 0;
- while (mP < mEnd) {
- if (!isdigit((unsigned char)*mP))
- break;
- number = number * 10 + (*mP - 48);
- mP++;
- }
- CheckForSeparator();
- return isNegative ? ((unsigned int)-int(number)) : number;
- }
- }
- // ------------------------------------------------------------------------------------------------
- ai_real XFileParser::ReadFloat() {
- if (mIsBinaryFormat) {
- if (mBinaryNumCount == 0 && mEnd - mP >= 2) {
- unsigned short tmp = ReadBinWord(); // 0x07 or 0x42
- if (tmp == 0x07 && mEnd - mP >= 4) // array of floats following
- mBinaryNumCount = ReadBinDWord();
- else // single float following
- mBinaryNumCount = 1;
- }
- --mBinaryNumCount;
- if (mBinaryFloatSize == 8) {
- if (mEnd - mP >= 8) {
- double res;
- ::memcpy(&res, mP, 8);
- mP += 8;
- const ai_real result(static_cast<ai_real>(res));
- return result;
- } else {
- mP = mEnd;
- return 0;
- }
- } else {
- if (mEnd - mP >= 4) {
- ai_real result;
- ::memcpy(&result, mP, 4);
- mP += 4;
- return result;
- } else {
- mP = mEnd;
- return 0;
- }
- }
- }
- // text version
- FindNextNoneWhiteSpace();
- // check for various special strings to allow reading files from faulty exporters
- // I mean you, Blender!
- // Reading is safe because of the terminating zero
- if (strncmp(mP, "-1.#IND00", 9) == 0 || strncmp(mP, "1.#IND00", 8) == 0) {
- mP += 9;
- CheckForSeparator();
- return 0.0;
- } else if (strncmp(mP, "1.#QNAN0", 8) == 0) {
- mP += 8;
- CheckForSeparator();
- return 0.0;
- }
- ai_real result = 0.0;
- mP = fast_atoreal_move<ai_real>(mP, result);
- CheckForSeparator();
- return result;
- }
- // ------------------------------------------------------------------------------------------------
- aiVector2D XFileParser::ReadVector2() {
- aiVector2D vector;
- vector.x = ReadFloat();
- vector.y = ReadFloat();
- TestForSeparator();
- return vector;
- }
- // ------------------------------------------------------------------------------------------------
- aiVector3D XFileParser::ReadVector3() {
- aiVector3D vector;
- vector.x = ReadFloat();
- vector.y = ReadFloat();
- vector.z = ReadFloat();
- TestForSeparator();
- return vector;
- }
- // ------------------------------------------------------------------------------------------------
- aiColor4D XFileParser::ReadRGBA() {
- aiColor4D color;
- color.r = ReadFloat();
- color.g = ReadFloat();
- color.b = ReadFloat();
- color.a = ReadFloat();
- TestForSeparator();
- return color;
- }
- // ------------------------------------------------------------------------------------------------
- aiColor3D XFileParser::ReadRGB() {
- aiColor3D color;
- color.r = ReadFloat();
- color.g = ReadFloat();
- color.b = ReadFloat();
- TestForSeparator();
- return color;
- }
- // ------------------------------------------------------------------------------------------------
- // Filters the imported hierarchy for some degenerated cases that some exporters produce.
- void XFileParser::FilterHierarchy(XFile::Node *pNode) {
- // if the node has just a single unnamed child containing a mesh, remove
- // the anonymous node between. The 3DSMax kwXport plugin seems to produce this
- // mess in some cases
- if (pNode->mChildren.size() == 1 && pNode->mMeshes.empty()) {
- XFile::Node *child = pNode->mChildren.front();
- if (child->mName.length() == 0 && child->mMeshes.size() > 0) {
- // transfer its meshes to us
- for (unsigned int a = 0; a < child->mMeshes.size(); a++)
- pNode->mMeshes.push_back(child->mMeshes[a]);
- child->mMeshes.clear();
- // transfer the transform as well
- pNode->mTrafoMatrix = pNode->mTrafoMatrix * child->mTrafoMatrix;
- // then kill it
- delete child;
- pNode->mChildren.clear();
- }
- }
- // recurse
- for (unsigned int a = 0; a < pNode->mChildren.size(); a++)
- FilterHierarchy(pNode->mChildren[a]);
- }
- #endif // !! ASSIMP_BUILD_NO_X_IMPORTER
|