assimp/code/PlyParser.cpp

1143 lines
31 KiB
C++

/*
---------------------------------------------------------------------------
Open Asset Import Library (assimp)
---------------------------------------------------------------------------
Copyright (c) 2006-2018, 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 <assimp/fast_atof.h>
#include <assimp/DefaultLogger.hpp>
#include <assimp/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);
// go to the next line
PLY::DOM::SkipSpacesAndLineEnd(buffer);
return true;
}
//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);
// 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);
}
}
else
{
const char* pCur = (const char*)&buffer[0];
// 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))))
{
ASSIMP_LOG_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)))
{
ASSIMP_LOG_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:
{
uint32_t t;
memcpy(&t, pCur, sizeof(uint32_t));
pCur += sizeof(uint32_t);
// Swap endianness
if (p_bBE)ByteSwap::Swap(&t);
out->iUInt = t;
break;
}
case EDT_UShort:
{
uint16_t t;
memcpy(&t, pCur, sizeof(uint16_t));
pCur += sizeof(uint16_t);
// Swap endianness
if (p_bBE)ByteSwap::Swap(&t);
out->iUInt = t;
break;
}
case EDT_UChar:
{
uint8_t t;
memcpy(&t, pCur, sizeof(uint8_t));
pCur += sizeof(uint8_t);
out->iUInt = t;
break;
}
case EDT_Int:
{
int32_t t;
memcpy(&t, pCur, sizeof(int32_t));
pCur += sizeof(int32_t);
// Swap endianness
if (p_bBE)ByteSwap::Swap(&t);
out->iInt = t;
break;
}
case EDT_Short:
{
int16_t t;
memcpy(&t, pCur, sizeof(int16_t));
pCur += sizeof(int16_t);
// Swap endianness
if (p_bBE)ByteSwap::Swap(&t);
out->iInt = t;
break;
}
case EDT_Char:
{
int8_t t;
memcpy(&t, pCur, sizeof(int8_t));
pCur += sizeof(int8_t);
out->iInt = t;
break;
}
case EDT_Float:
{
float t;
memcpy(&t, pCur, sizeof(float));
pCur += sizeof(float);
// Swap endianness
if (p_bBE)ByteSwap::Swap(&t);
out->fFloat = t;
break;
}
case EDT_Double:
{
double t;
memcpy(&t, pCur, sizeof(double));
pCur += sizeof(double);
// Swap endianness
if (p_bBE)ByteSwap::Swap(&t);
out->fDouble = t;
break;
}
default:
ret = false;
}
bufferSize -= lsize;
return ret;
}
#endif // !! ASSIMP_BUILD_NO_PLY_IMPORTER