1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117 |
- /*
- ---------------------------------------------------------------------------
- Open Asset Import Library (assimp)
- ---------------------------------------------------------------------------
- Copyright (c) 2006-2017, 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 PLY parser class */
- #ifndef ASSIMP_BUILD_NO_PLY_IMPORTER
- #include "fast_atof.h"
- #include <assimp/DefaultLogger.hpp>
- #include "ByteSwapper.h"
- #include "PlyLoader.h"
- using namespace Assimp;
- // ------------------------------------------------------------------------------------------------
- PLY::EDataType PLY::Property::ParseDataType(std::vector<char> &buffer) {
- ai_assert(!buffer.empty());
- PLY::EDataType eOut = PLY::EDT_INVALID;
- if (PLY::DOM::TokenMatch(buffer, "char", 4) ||
- PLY::DOM::TokenMatch(buffer, "int8", 4))
- {
- eOut = PLY::EDT_Char;
- }
- else if (PLY::DOM::TokenMatch(buffer, "uchar", 5) ||
- PLY::DOM::TokenMatch(buffer, "uint8", 5))
- {
- eOut = PLY::EDT_UChar;
- }
- else if (PLY::DOM::TokenMatch(buffer, "short", 5) ||
- PLY::DOM::TokenMatch(buffer, "int16", 5))
- {
- eOut = PLY::EDT_Short;
- }
- else if (PLY::DOM::TokenMatch(buffer, "ushort", 6) ||
- PLY::DOM::TokenMatch(buffer, "uint16", 6))
- {
- eOut = PLY::EDT_UShort;
- }
- else if (PLY::DOM::TokenMatch(buffer, "int32", 5) || PLY::DOM::TokenMatch(buffer, "int", 3))
- {
- eOut = PLY::EDT_Int;
- }
- else if (PLY::DOM::TokenMatch(buffer, "uint32", 6) || PLY::DOM::TokenMatch(buffer, "uint", 4))
- {
- eOut = PLY::EDT_UInt;
- }
- else if (PLY::DOM::TokenMatch(buffer, "float", 5) || PLY::DOM::TokenMatch(buffer, "float32", 7))
- {
- eOut = PLY::EDT_Float;
- }
- else if (PLY::DOM::TokenMatch(buffer, "double64", 8) || PLY::DOM::TokenMatch(buffer, "double", 6) ||
- PLY::DOM::TokenMatch(buffer, "float64", 7))
- {
- eOut = PLY::EDT_Double;
- }
- if (PLY::EDT_INVALID == eOut)
- {
- DefaultLogger::get()->info("Found unknown data type in PLY file. This is OK");
- }
- return eOut;
- }
- // ------------------------------------------------------------------------------------------------
- PLY::ESemantic PLY::Property::ParseSemantic(std::vector<char> &buffer) {
- ai_assert(!buffer.empty());
- PLY::ESemantic eOut = PLY::EST_INVALID;
- if (PLY::DOM::TokenMatch(buffer, "red", 3)) {
- eOut = PLY::EST_Red;
- }
- else if (PLY::DOM::TokenMatch(buffer, "green", 5)) {
- eOut = PLY::EST_Green;
- }
- else if (PLY::DOM::TokenMatch(buffer, "blue", 4)) {
- eOut = PLY::EST_Blue;
- }
- else if (PLY::DOM::TokenMatch(buffer, "alpha", 5)) {
- eOut = PLY::EST_Alpha;
- }
- else if (PLY::DOM::TokenMatch(buffer, "vertex_index", 12) || PLY::DOM::TokenMatch(buffer, "vertex_indices", 14)) {
- eOut = PLY::EST_VertexIndex;
- }
- else if (PLY::DOM::TokenMatch(buffer, "texcoord", 8)) // Manage uv coords on faces
- {
- eOut = PLY::EST_TextureCoordinates;
- }
- else if (PLY::DOM::TokenMatch(buffer, "material_index", 14))
- {
- eOut = PLY::EST_MaterialIndex;
- }
- else if (PLY::DOM::TokenMatch(buffer, "ambient_red", 11))
- {
- eOut = PLY::EST_AmbientRed;
- }
- else if (PLY::DOM::TokenMatch(buffer, "ambient_green", 13))
- {
- eOut = PLY::EST_AmbientGreen;
- }
- else if (PLY::DOM::TokenMatch(buffer, "ambient_blue", 12))
- {
- eOut = PLY::EST_AmbientBlue;
- }
- else if (PLY::DOM::TokenMatch(buffer, "ambient_alpha", 13))
- {
- eOut = PLY::EST_AmbientAlpha;
- }
- else if (PLY::DOM::TokenMatch(buffer, "diffuse_red", 11))
- {
- eOut = PLY::EST_DiffuseRed;
- }
- else if (PLY::DOM::TokenMatch(buffer, "diffuse_green", 13))
- {
- eOut = PLY::EST_DiffuseGreen;
- }
- else if (PLY::DOM::TokenMatch(buffer, "diffuse_blue", 12))
- {
- eOut = PLY::EST_DiffuseBlue;
- }
- else if (PLY::DOM::TokenMatch(buffer, "diffuse_alpha", 13))
- {
- eOut = PLY::EST_DiffuseAlpha;
- }
- else if (PLY::DOM::TokenMatch(buffer, "specular_red", 12))
- {
- eOut = PLY::EST_SpecularRed;
- }
- else if (PLY::DOM::TokenMatch(buffer, "specular_green", 14))
- {
- eOut = PLY::EST_SpecularGreen;
- }
- else if (PLY::DOM::TokenMatch(buffer, "specular_blue", 13))
- {
- eOut = PLY::EST_SpecularBlue;
- }
- else if (PLY::DOM::TokenMatch(buffer, "specular_alpha", 14))
- {
- eOut = PLY::EST_SpecularAlpha;
- }
- else if (PLY::DOM::TokenMatch(buffer, "opacity", 7))
- {
- eOut = PLY::EST_Opacity;
- }
- else if (PLY::DOM::TokenMatch(buffer, "specular_power", 14))
- {
- eOut = PLY::EST_PhongPower;
- }
- else if (PLY::DOM::TokenMatch(buffer, "r", 1))
- {
- eOut = PLY::EST_Red;
- }
- else if (PLY::DOM::TokenMatch(buffer, "g", 1))
- {
- eOut = PLY::EST_Green;
- }
- else if (PLY::DOM::TokenMatch(buffer, "b", 1))
- {
- eOut = PLY::EST_Blue;
- }
- // NOTE: Blender3D exports texture coordinates as s,t tuples
- else if (PLY::DOM::TokenMatch(buffer, "u", 1) || PLY::DOM::TokenMatch(buffer, "s", 1) || PLY::DOM::TokenMatch(buffer, "tx", 2) || PLY::DOM::TokenMatch(buffer, "texture_u", 9))
- {
- eOut = PLY::EST_UTextureCoord;
- }
- else if (PLY::DOM::TokenMatch(buffer, "v", 1) || PLY::DOM::TokenMatch(buffer, "t", 1) || PLY::DOM::TokenMatch(buffer, "ty", 2) || PLY::DOM::TokenMatch(buffer, "texture_v", 9))
- {
- eOut = PLY::EST_VTextureCoord;
- }
- else if (PLY::DOM::TokenMatch(buffer, "x", 1))
- {
- eOut = PLY::EST_XCoord;
- }
- else if (PLY::DOM::TokenMatch(buffer, "y", 1)) {
- eOut = PLY::EST_YCoord;
- }
- else if (PLY::DOM::TokenMatch(buffer, "z", 1)) {
- eOut = PLY::EST_ZCoord;
- }
- else if (PLY::DOM::TokenMatch(buffer, "nx", 2)) {
- eOut = PLY::EST_XNormal;
- }
- else if (PLY::DOM::TokenMatch(buffer, "ny", 2)) {
- eOut = PLY::EST_YNormal;
- }
- else if (PLY::DOM::TokenMatch(buffer, "nz", 2)) {
- eOut = PLY::EST_ZNormal;
- }
- else {
- DefaultLogger::get()->info("Found unknown property semantic in file. This is ok");
- PLY::DOM::SkipLine(buffer);
- }
- return eOut;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::Property::ParseProperty(std::vector<char> &buffer, PLY::Property* pOut)
- {
- ai_assert(!buffer.empty());
- // Forms supported:
- // "property float x"
- // "property list uchar int vertex_index"
- // skip leading spaces
- if (!PLY::DOM::SkipSpaces(buffer)) {
- return false;
- }
- // skip the "property" string at the beginning
- if (!PLY::DOM::TokenMatch(buffer, "property", 8))
- {
- // seems not to be a valid property entry
- return false;
- }
- // get next word
- if (!PLY::DOM::SkipSpaces(buffer)) {
- return false;
- }
- if (PLY::DOM::TokenMatch(buffer, "list", 4))
- {
- pOut->bIsList = true;
- // seems to be a list.
- if (EDT_INVALID == (pOut->eFirstType = PLY::Property::ParseDataType(buffer)))
- {
- // unable to parse list size data type
- PLY::DOM::SkipLine(buffer);
- return false;
- }
- if (!PLY::DOM::SkipSpaces(buffer))return false;
- if (EDT_INVALID == (pOut->eType = PLY::Property::ParseDataType(buffer)))
- {
- // unable to parse list data type
- PLY::DOM::SkipLine(buffer);
- return false;
- }
- }
- else
- {
- if (EDT_INVALID == (pOut->eType = PLY::Property::ParseDataType(buffer)))
- {
- // unable to parse data type. Skip the property
- PLY::DOM::SkipLine(buffer);
- return false;
- }
- }
- if (!PLY::DOM::SkipSpaces(buffer))
- return false;
- pOut->Semantic = PLY::Property::ParseSemantic(buffer);
- if (PLY::EST_INVALID == pOut->Semantic)
- {
- DefaultLogger::get()->info("Found unknown semantic in PLY file. This is OK");
- std::string(&buffer[0], &buffer[0] + strlen(&buffer[0]));
- }
- PLY::DOM::SkipSpacesAndLineEnd(buffer);
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- PLY::EElementSemantic PLY::Element::ParseSemantic(std::vector<char> &buffer)
- {
- ai_assert(!buffer.empty());
- PLY::EElementSemantic eOut = PLY::EEST_INVALID;
- if (PLY::DOM::TokenMatch(buffer, "vertex", 6))
- {
- eOut = PLY::EEST_Vertex;
- }
- else if (PLY::DOM::TokenMatch(buffer, "face", 4))
- {
- eOut = PLY::EEST_Face;
- }
- else if (PLY::DOM::TokenMatch(buffer, "tristrips", 9))
- {
- eOut = PLY::EEST_TriStrip;
- }
- #if 0
- // TODO: maybe implement this?
- else if (PLY::DOM::TokenMatch(buffer,"range_grid",10))
- {
- eOut = PLY::EEST_Face;
- }
- #endif
- else if (PLY::DOM::TokenMatch(buffer, "edge", 4))
- {
- eOut = PLY::EEST_Edge;
- }
- else if (PLY::DOM::TokenMatch(buffer, "material", 8))
- {
- eOut = PLY::EEST_Material;
- }
- else if (PLY::DOM::TokenMatch(buffer, "TextureFile", 11))
- {
- eOut = PLY::EEST_TextureFile;
- }
- return eOut;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::Element::ParseElement(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, PLY::Element* pOut)
- {
- ai_assert(NULL != pOut);
- // Example format: "element vertex 8"
- // skip leading spaces
- if (!PLY::DOM::SkipSpaces(buffer))
- {
- return false;
- }
- // skip the "element" string at the beginning
- if (!PLY::DOM::TokenMatch(buffer, "element", 7) && !PLY::DOM::TokenMatch(buffer, "comment", 7))
- {
- // seems not to be a valid property entry
- return false;
- }
- // get next word
- if (!PLY::DOM::SkipSpaces(buffer))
- return false;
- // parse the semantic of the element
- pOut->eSemantic = PLY::Element::ParseSemantic(buffer);
- if (PLY::EEST_INVALID == pOut->eSemantic)
- {
- // if the exact semantic can't be determined, just store
- // the original string identifier
- pOut->szName = std::string(&buffer[0], &buffer[0] + strlen(&buffer[0]));
- }
- if (!PLY::DOM::SkipSpaces(buffer))
- return false;
- if (PLY::EEST_TextureFile == pOut->eSemantic)
- {
- char* endPos = &buffer[0] + (strlen(&buffer[0]) - 1);
- pOut->szName = std::string(&buffer[0], endPos);
- }
- //parse the number of occurrences of this element
- const char* pCur = (char*)&buffer[0];
- pOut->NumOccur = strtoul10(pCur, &pCur);
- // go to the next line
- PLY::DOM::SkipSpacesAndLineEnd(buffer);
- // now parse all properties of the element
- while (true)
- {
- streamBuffer.getNextLine(buffer);
- pCur = (char*)&buffer[0];
- // skip all comments
- PLY::DOM::SkipComments(buffer);
- PLY::Property prop;
- if (!PLY::Property::ParseProperty(buffer, &prop))
- break;
- pOut->alProperties.push_back(prop);
- }
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::DOM::SkipSpaces(std::vector<char> &buffer)
- {
- const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
- bool ret = false;
- if (pCur)
- {
- const char* szCur = pCur;
- ret = Assimp::SkipSpaces(pCur, &pCur);
- uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
- buffer.erase(buffer.begin(), buffer.begin() + iDiff);
- return ret;
- }
- return ret;
- }
- bool PLY::DOM::SkipLine(std::vector<char> &buffer)
- {
- const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
- bool ret = false;
- if (pCur)
- {
- const char* szCur = pCur;
- ret = Assimp::SkipLine(pCur, &pCur);
- uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
- buffer.erase(buffer.begin(), buffer.begin() + iDiff);
- return ret;
- }
- return ret;
- }
- bool PLY::DOM::TokenMatch(std::vector<char> &buffer, const char* token, unsigned int len)
- {
- const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
- bool ret = false;
- if (pCur)
- {
- const char* szCur = pCur;
- ret = Assimp::TokenMatch(pCur, token, len);
- uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
- buffer.erase(buffer.begin(), buffer.begin() + iDiff);
- return ret;
- }
- return ret;
- }
- bool PLY::DOM::SkipSpacesAndLineEnd(std::vector<char> &buffer)
- {
- const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
- bool ret = false;
- if (pCur)
- {
- const char* szCur = pCur;
- ret = Assimp::SkipSpacesAndLineEnd(pCur, &pCur);
- uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
- buffer.erase(buffer.begin(), buffer.begin() + iDiff);
- return ret;
- }
- return ret;
- }
- bool PLY::DOM::SkipComments(std::vector<char> &buffer)
- {
- ai_assert(!buffer.empty());
- std::vector<char> nbuffer = buffer;
- // skip spaces
- if (!SkipSpaces(nbuffer)) {
- return false;
- }
- if (TokenMatch(nbuffer, "comment", 7))
- {
- if (!SkipSpaces(nbuffer))
- SkipLine(nbuffer);
- if (!TokenMatch(nbuffer, "TextureFile", 11))
- {
- SkipLine(nbuffer);
- buffer = nbuffer;
- return true;
- }
- return true;
- }
- return false;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::DOM::ParseHeader(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, bool isBinary) {
- DefaultLogger::get()->debug("PLY::DOM::ParseHeader() begin");
- // parse all elements
- while (!buffer.empty())
- {
- // skip all comments
- PLY::DOM::SkipComments(buffer);
- PLY::Element out;
- if (PLY::Element::ParseElement(streamBuffer, buffer, &out))
- {
- // add the element to the list of elements
- alElements.push_back(out);
- }
- else if (TokenMatch(buffer, "end_header", 10))
- {
- // we have reached the end of the header
- break;
- }
- else
- {
- // ignore unknown header elements
- streamBuffer.getNextLine(buffer);
- }
- }
- if (!isBinary) // it would occur an error, if binary data start with values as space or line end.
- SkipSpacesAndLineEnd(buffer);
- DefaultLogger::get()->debug("PLY::DOM::ParseHeader() succeeded");
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::DOM::ParseElementInstanceLists(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, PLYImporter* loader)
- {
- DefaultLogger::get()->debug("PLY::DOM::ParseElementInstanceLists() begin");
- alElementData.resize(alElements.size());
- std::vector<PLY::Element>::const_iterator i = alElements.begin();
- std::vector<PLY::ElementInstanceList>::iterator a = alElementData.begin();
- // parse all element instances
- //construct vertices and faces
- for (; i != alElements.end(); ++i, ++a)
- {
- if ((*i).eSemantic == EEST_Vertex || (*i).eSemantic == EEST_Face || (*i).eSemantic == EEST_TriStrip)
- {
- PLY::ElementInstanceList::ParseInstanceList(streamBuffer, buffer, &(*i), NULL, loader);
- }
- else
- {
- (*a).alInstances.resize((*i).NumOccur);
- PLY::ElementInstanceList::ParseInstanceList(streamBuffer, buffer, &(*i), &(*a), NULL);
- }
- }
- DefaultLogger::get()->debug("PLY::DOM::ParseElementInstanceLists() succeeded");
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::DOM::ParseElementInstanceListsBinary(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer,
- const char* &pCur,
- unsigned int &bufferSize,
- PLYImporter* loader,
- bool p_bBE)
- {
- DefaultLogger::get()->debug("PLY::DOM::ParseElementInstanceListsBinary() begin");
- alElementData.resize(alElements.size());
- std::vector<PLY::Element>::const_iterator i = alElements.begin();
- std::vector<PLY::ElementInstanceList>::iterator a = alElementData.begin();
- // parse all element instances
- for (; i != alElements.end(); ++i, ++a)
- {
- if ((*i).eSemantic == EEST_Vertex || (*i).eSemantic == EEST_Face || (*i).eSemantic == EEST_TriStrip)
- {
- PLY::ElementInstanceList::ParseInstanceListBinary(streamBuffer, buffer, pCur, bufferSize, &(*i), NULL, loader, p_bBE);
- }
- else
- {
- (*a).alInstances.resize((*i).NumOccur);
- PLY::ElementInstanceList::ParseInstanceListBinary(streamBuffer, buffer, pCur, bufferSize, &(*i), &(*a), NULL, p_bBE);
- }
- }
- DefaultLogger::get()->debug("PLY::DOM::ParseElementInstanceListsBinary() succeeded");
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::DOM::ParseInstanceBinary(IOStreamBuffer<char> &streamBuffer, DOM* p_pcOut, PLYImporter* loader, bool p_bBE)
- {
- ai_assert(NULL != p_pcOut);
- ai_assert(NULL != loader);
- std::vector<char> buffer;
- streamBuffer.getNextLine(buffer);
- DefaultLogger::get()->debug("PLY::DOM::ParseInstanceBinary() begin");
- if (!p_pcOut->ParseHeader(streamBuffer, buffer, true))
- {
- DefaultLogger::get()->debug("PLY::DOM::ParseInstanceBinary() failure");
- return false;
- }
- streamBuffer.getNextBlock(buffer);
- unsigned int bufferSize = static_cast<unsigned int>(buffer.size());
- const char* pCur = (char*)&buffer[0];
- if (!p_pcOut->ParseElementInstanceListsBinary(streamBuffer, buffer, pCur, bufferSize, loader, p_bBE))
- {
- DefaultLogger::get()->debug("PLY::DOM::ParseInstanceBinary() failure");
- return false;
- }
- DefaultLogger::get()->debug("PLY::DOM::ParseInstanceBinary() succeeded");
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::DOM::ParseInstance(IOStreamBuffer<char> &streamBuffer, DOM* p_pcOut, PLYImporter* loader)
- {
- ai_assert(NULL != p_pcOut);
- ai_assert(NULL != loader);
- std::vector<char> buffer;
- streamBuffer.getNextLine(buffer);
- DefaultLogger::get()->debug("PLY::DOM::ParseInstance() begin");
- if (!p_pcOut->ParseHeader(streamBuffer, buffer, false))
- {
- DefaultLogger::get()->debug("PLY::DOM::ParseInstance() failure");
- return false;
- }
- //get next line after header
- streamBuffer.getNextLine(buffer);
- if (!p_pcOut->ParseElementInstanceLists(streamBuffer, buffer, loader))
- {
- DefaultLogger::get()->debug("PLY::DOM::ParseInstance() failure");
- return false;
- }
- DefaultLogger::get()->debug("PLY::DOM::ParseInstance() succeeded");
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::ElementInstanceList::ParseInstanceList(
- IOStreamBuffer<char> &streamBuffer,
- std::vector<char> &buffer,
- const PLY::Element* pcElement,
- PLY::ElementInstanceList* p_pcOut,
- PLYImporter* loader)
- {
- ai_assert(NULL != pcElement);
- const char* pCur = (const char*)&buffer[0];
- // parse all elements
- if (EEST_INVALID == pcElement->eSemantic || pcElement->alProperties.empty())
- {
- // if the element has an unknown semantic we can skip all lines
- // However, there could be comments
- for (unsigned int i = 0; i < pcElement->NumOccur; ++i)
- {
- PLY::DOM::SkipComments(buffer);
- PLY::DOM::SkipLine(buffer);
- streamBuffer.getNextLine(buffer);
- pCur = (buffer.empty()) ? NULL : (const char*)&buffer[0];
- }
- }
- else
- {
- // be sure to have enough storage
- for (unsigned int i = 0; i < pcElement->NumOccur; ++i)
- {
- if (p_pcOut)
- PLY::ElementInstance::ParseInstance(pCur, pcElement, &p_pcOut->alInstances[i]);
- else
- {
- ElementInstance elt;
- PLY::ElementInstance::ParseInstance(pCur, pcElement, &elt);
- // Create vertex or face
- if (pcElement->eSemantic == EEST_Vertex)
- {
- //call loader instance from here
- loader->LoadVertex(pcElement, &elt, i);
- }
- else if (pcElement->eSemantic == EEST_Face)
- {
- //call loader instance from here
- loader->LoadFace(pcElement, &elt, i);
- }
- else if (pcElement->eSemantic == EEST_TriStrip)
- {
- //call loader instance from here
- loader->LoadFace(pcElement, &elt, i);
- }
- }
- streamBuffer.getNextLine(buffer);
- pCur = (buffer.empty()) ? NULL : (const char*)&buffer[0];
- }
- }
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::ElementInstanceList::ParseInstanceListBinary(
- IOStreamBuffer<char> &streamBuffer,
- std::vector<char> &buffer,
- const char* &pCur,
- unsigned int &bufferSize,
- const PLY::Element* pcElement,
- PLY::ElementInstanceList* p_pcOut,
- PLYImporter* loader,
- bool p_bBE /* = false */)
- {
- ai_assert(NULL != pcElement);
- // we can add special handling code for unknown element semantics since
- // we can't skip it as a whole block (we don't know its exact size
- // due to the fact that lists could be contained in the property list
- // of the unknown element)
- for (unsigned int i = 0; i < pcElement->NumOccur; ++i)
- {
- if (p_pcOut)
- PLY::ElementInstance::ParseInstanceBinary(streamBuffer, buffer, pCur, bufferSize, pcElement, &p_pcOut->alInstances[i], p_bBE);
- else
- {
- ElementInstance elt;
- PLY::ElementInstance::ParseInstanceBinary(streamBuffer, buffer, pCur, bufferSize, pcElement, &elt, p_bBE);
- // Create vertex or face
- if (pcElement->eSemantic == EEST_Vertex)
- {
- //call loader instance from here
- loader->LoadVertex(pcElement, &elt, i);
- }
- else if (pcElement->eSemantic == EEST_Face)
- {
- //call loader instance from here
- loader->LoadFace(pcElement, &elt, i);
- }
- else if (pcElement->eSemantic == EEST_TriStrip)
- {
- //call loader instance from here
- loader->LoadFace(pcElement, &elt, i);
- }
- }
- }
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::ElementInstance::ParseInstance(const char* &pCur,
- const PLY::Element* pcElement,
- PLY::ElementInstance* p_pcOut)
- {
- ai_assert(NULL != pcElement);
- ai_assert(NULL != p_pcOut);
- // allocate enough storage
- p_pcOut->alProperties.resize(pcElement->alProperties.size());
- std::vector<PLY::PropertyInstance>::iterator i = p_pcOut->alProperties.begin();
- std::vector<PLY::Property>::const_iterator a = pcElement->alProperties.begin();
- for (; i != p_pcOut->alProperties.end(); ++i, ++a)
- {
- if (!(PLY::PropertyInstance::ParseInstance(pCur, &(*a), &(*i))))
- {
- DefaultLogger::get()->warn("Unable to parse property instance. "
- "Skipping this element instance");
- PLY::PropertyInstance::ValueUnion v = PLY::PropertyInstance::DefaultValue((*a).eType);
- (*i).avList.push_back(v);
- }
- }
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::ElementInstance::ParseInstanceBinary(
- IOStreamBuffer<char> &streamBuffer,
- std::vector<char> &buffer,
- const char* &pCur,
- unsigned int &bufferSize,
- const PLY::Element* pcElement,
- PLY::ElementInstance* p_pcOut,
- bool p_bBE /* = false */)
- {
- ai_assert(NULL != pcElement);
- ai_assert(NULL != p_pcOut);
- // allocate enough storage
- p_pcOut->alProperties.resize(pcElement->alProperties.size());
- std::vector<PLY::PropertyInstance>::iterator i = p_pcOut->alProperties.begin();
- std::vector<PLY::Property>::const_iterator a = pcElement->alProperties.begin();
- for (; i != p_pcOut->alProperties.end(); ++i, ++a)
- {
- if (!(PLY::PropertyInstance::ParseInstanceBinary(streamBuffer, buffer, pCur, bufferSize, &(*a), &(*i), p_bBE)))
- {
- DefaultLogger::get()->warn("Unable to parse binary property instance. "
- "Skipping this element instance");
- (*i).avList.push_back(PLY::PropertyInstance::DefaultValue((*a).eType));
- }
- }
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::PropertyInstance::ParseInstance(const char* &pCur,
- const PLY::Property* prop, PLY::PropertyInstance* p_pcOut)
- {
- ai_assert(NULL != prop);
- ai_assert(NULL != p_pcOut);
- // skip spaces at the beginning
- if (!SkipSpaces(&pCur))
- {
- return false;
- }
- if (prop->bIsList)
- {
- // parse the number of elements in the list
- PLY::PropertyInstance::ValueUnion v;
- PLY::PropertyInstance::ParseValue(pCur, prop->eFirstType, &v);
- // convert to unsigned int
- unsigned int iNum = PLY::PropertyInstance::ConvertTo<unsigned int>(v, prop->eFirstType);
- // parse all list elements
- p_pcOut->avList.resize(iNum);
- for (unsigned int i = 0; i < iNum; ++i)
- {
- if (!SkipSpaces(&pCur))
- return false;
- PLY::PropertyInstance::ParseValue(pCur, prop->eType, &p_pcOut->avList[i]);
- }
- }
- else
- {
- // parse the property
- PLY::PropertyInstance::ValueUnion v;
- PLY::PropertyInstance::ParseValue(pCur, prop->eType, &v);
- p_pcOut->avList.push_back(v);
- }
- SkipSpacesAndLineEnd(&pCur);
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::PropertyInstance::ParseInstanceBinary(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer,
- const char* &pCur,
- unsigned int &bufferSize,
- const PLY::Property* prop,
- PLY::PropertyInstance* p_pcOut,
- bool p_bBE)
- {
- ai_assert(NULL != prop);
- ai_assert(NULL != p_pcOut);
- // parse all elements
- if (prop->bIsList)
- {
- // parse the number of elements in the list
- PLY::PropertyInstance::ValueUnion v;
- PLY::PropertyInstance::ParseValueBinary(streamBuffer, buffer, pCur, bufferSize, prop->eFirstType, &v, p_bBE);
- // convert to unsigned int
- unsigned int iNum = PLY::PropertyInstance::ConvertTo<unsigned int>(v, prop->eFirstType);
- // parse all list elements
- p_pcOut->avList.resize(iNum);
- for (unsigned int i = 0; i < iNum; ++i)
- {
- PLY::PropertyInstance::ParseValueBinary(streamBuffer, buffer, pCur, bufferSize, prop->eType, &p_pcOut->avList[i], p_bBE);
- }
- }
- else
- {
- // parse the property
- PLY::PropertyInstance::ValueUnion v;
- PLY::PropertyInstance::ParseValueBinary(streamBuffer, buffer, pCur, bufferSize, prop->eType, &v, p_bBE);
- p_pcOut->avList.push_back(v);
- }
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- PLY::PropertyInstance::ValueUnion PLY::PropertyInstance::DefaultValue(PLY::EDataType eType)
- {
- PLY::PropertyInstance::ValueUnion out;
- switch (eType)
- {
- case EDT_Float:
- out.fFloat = 0.f;
- return out;
- case EDT_Double:
- out.fDouble = 0.;
- return out;
- default:;
- };
- out.iUInt = 0;
- return out;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::PropertyInstance::ParseValue(const char* &pCur,
- PLY::EDataType eType,
- PLY::PropertyInstance::ValueUnion* out)
- {
- ai_assert(NULL != pCur);
- ai_assert(NULL != out);
-
- //calc element size
- bool ret = true;
- switch (eType)
- {
- case EDT_UInt:
- case EDT_UShort:
- case EDT_UChar:
- out->iUInt = (uint32_t)strtoul10(pCur, &pCur);
- break;
- case EDT_Int:
- case EDT_Short:
- case EDT_Char:
- out->iInt = (int32_t)strtol10(pCur, &pCur);
- break;
- case EDT_Float:
- // technically this should cast to float, but people tend to use float descriptors for double data
- // this is the best way to not risk losing precision on import and it doesn't hurt to do this
- ai_real f;
- pCur = fast_atoreal_move<ai_real>(pCur, f);
- out->fFloat = (ai_real)f;
- break;
- case EDT_Double:
- double d;
- pCur = fast_atoreal_move<double>(pCur, d);
- out->fDouble = (double)d;
- break;
- case EDT_INVALID:
- default:
- ret = false;
- break;
- }
- return ret;
- }
- // ------------------------------------------------------------------------------------------------
- bool PLY::PropertyInstance::ParseValueBinary(IOStreamBuffer<char> &streamBuffer,
- std::vector<char> &buffer,
- const char* &pCur,
- unsigned int &bufferSize,
- PLY::EDataType eType,
- PLY::PropertyInstance::ValueUnion* out,
- bool p_bBE)
- {
- ai_assert(NULL != out);
- //calc element size
- unsigned int lsize = 0;
- switch (eType)
- {
- case EDT_Char:
- case EDT_UChar:
- lsize = 1;
- break;
- case EDT_UShort:
- case EDT_Short:
- lsize = 2;
- break;
- case EDT_UInt:
- case EDT_Int:
- case EDT_Float:
- lsize = 4;
- break;
- case EDT_Double:
- lsize = 8;
- break;
- case EDT_INVALID:
- default:
- break;
- }
- //read the next file block if needed
- if (bufferSize < lsize)
- {
- std::vector<char> nbuffer;
- if (streamBuffer.getNextBlock(nbuffer))
- {
- //concat buffer contents
- buffer = std::vector<char>(buffer.end() - bufferSize, buffer.end());
- buffer.insert(buffer.end(), nbuffer.begin(), nbuffer.end());
- nbuffer.clear();
- bufferSize = static_cast<unsigned int>(buffer.size());
- pCur = (char*)&buffer[0];
- }
- else
- {
- throw DeadlyImportError("Invalid .ply file: File corrupted");
- }
- }
- bool ret = true;
- switch (eType)
- {
- case EDT_UInt:
- out->iUInt = (uint32_t)*((uint32_t*)pCur);
- pCur += 4;
- // Swap endianness
- if (p_bBE)ByteSwap::Swap((int32_t*)&out->iUInt);
- break;
- case EDT_UShort:
- {
- uint16_t i = *((uint16_t*)pCur);
- // Swap endianness
- if (p_bBE)ByteSwap::Swap(&i);
- out->iUInt = (uint32_t)i;
- pCur += 2;
- break;
- }
- case EDT_UChar:
- {
- out->iUInt = (uint32_t)(*((uint8_t*)pCur));
- pCur++;
- break;
- }
- case EDT_Int:
- out->iInt = *((int32_t*)pCur);
- pCur += 4;
- // Swap endianness
- if (p_bBE)ByteSwap::Swap(&out->iInt);
- break;
- case EDT_Short:
- {
- int16_t i = *((int16_t*)pCur);
- // Swap endianness
- if (p_bBE)ByteSwap::Swap(&i);
- out->iInt = (int32_t)i;
- pCur += 2;
- break;
- }
- case EDT_Char:
- out->iInt = (int32_t)*((int8_t*)pCur);
- pCur++;
- break;
- case EDT_Float:
- {
- out->fFloat = *((float*)pCur);
- // Swap endianness
- if (p_bBE)ByteSwap::Swap((int32_t*)&out->fFloat);
- pCur += 4;
- break;
- }
- case EDT_Double:
- {
- out->fDouble = *((double*)pCur);
- // Swap endianness
- if (p_bBE)ByteSwap::Swap((int64_t*)&out->fDouble);
- pCur += 8;
- break;
- }
- default:
- ret = false;
- }
- bufferSize -= lsize;
- return ret;
- }
- #endif // !! ASSIMP_BUILD_NO_PLY_IMPORTER
|