PlyParser.cpp 31 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142
  1. /*
  2. ---------------------------------------------------------------------------
  3. Open Asset Import Library (assimp)
  4. ---------------------------------------------------------------------------
  5. Copyright (c) 2006-2018, assimp team
  6. All rights reserved.
  7. Redistribution and use of this software in source and binary forms,
  8. with or without modification, are permitted provided that the following
  9. conditions are met:
  10. * Redistributions of source code must retain the above
  11. copyright notice, this list of conditions and the
  12. following disclaimer.
  13. * Redistributions in binary form must reproduce the above
  14. copyright notice, this list of conditions and the
  15. following disclaimer in the documentation and/or other
  16. materials provided with the distribution.
  17. * Neither the name of the assimp team, nor the names of its
  18. contributors may be used to endorse or promote products
  19. derived from this software without specific prior
  20. written permission of the assimp team.
  21. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22. "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23. LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24. A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25. OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26. SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27. LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28. DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29. THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30. (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31. OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32. ---------------------------------------------------------------------------
  33. */
  34. /** @file Implementation of the PLY parser class */
  35. #ifndef ASSIMP_BUILD_NO_PLY_IMPORTER
  36. #include <assimp/fast_atof.h>
  37. #include <assimp/DefaultLogger.hpp>
  38. #include <assimp/ByteSwapper.h>
  39. #include "PlyLoader.h"
  40. using namespace Assimp;
  41. // ------------------------------------------------------------------------------------------------
  42. PLY::EDataType PLY::Property::ParseDataType(std::vector<char> &buffer) {
  43. ai_assert(!buffer.empty());
  44. PLY::EDataType eOut = PLY::EDT_INVALID;
  45. if (PLY::DOM::TokenMatch(buffer, "char", 4) ||
  46. PLY::DOM::TokenMatch(buffer, "int8", 4))
  47. {
  48. eOut = PLY::EDT_Char;
  49. }
  50. else if (PLY::DOM::TokenMatch(buffer, "uchar", 5) ||
  51. PLY::DOM::TokenMatch(buffer, "uint8", 5))
  52. {
  53. eOut = PLY::EDT_UChar;
  54. }
  55. else if (PLY::DOM::TokenMatch(buffer, "short", 5) ||
  56. PLY::DOM::TokenMatch(buffer, "int16", 5))
  57. {
  58. eOut = PLY::EDT_Short;
  59. }
  60. else if (PLY::DOM::TokenMatch(buffer, "ushort", 6) ||
  61. PLY::DOM::TokenMatch(buffer, "uint16", 6))
  62. {
  63. eOut = PLY::EDT_UShort;
  64. }
  65. else if (PLY::DOM::TokenMatch(buffer, "int32", 5) || PLY::DOM::TokenMatch(buffer, "int", 3))
  66. {
  67. eOut = PLY::EDT_Int;
  68. }
  69. else if (PLY::DOM::TokenMatch(buffer, "uint32", 6) || PLY::DOM::TokenMatch(buffer, "uint", 4))
  70. {
  71. eOut = PLY::EDT_UInt;
  72. }
  73. else if (PLY::DOM::TokenMatch(buffer, "float", 5) || PLY::DOM::TokenMatch(buffer, "float32", 7))
  74. {
  75. eOut = PLY::EDT_Float;
  76. }
  77. else if (PLY::DOM::TokenMatch(buffer, "double64", 8) || PLY::DOM::TokenMatch(buffer, "double", 6) ||
  78. PLY::DOM::TokenMatch(buffer, "float64", 7))
  79. {
  80. eOut = PLY::EDT_Double;
  81. }
  82. if (PLY::EDT_INVALID == eOut)
  83. {
  84. ASSIMP_LOG_INFO("Found unknown data type in PLY file. This is OK");
  85. }
  86. return eOut;
  87. }
  88. // ------------------------------------------------------------------------------------------------
  89. PLY::ESemantic PLY::Property::ParseSemantic(std::vector<char> &buffer) {
  90. ai_assert(!buffer.empty());
  91. PLY::ESemantic eOut = PLY::EST_INVALID;
  92. if (PLY::DOM::TokenMatch(buffer, "red", 3)) {
  93. eOut = PLY::EST_Red;
  94. }
  95. else if (PLY::DOM::TokenMatch(buffer, "green", 5)) {
  96. eOut = PLY::EST_Green;
  97. }
  98. else if (PLY::DOM::TokenMatch(buffer, "blue", 4)) {
  99. eOut = PLY::EST_Blue;
  100. }
  101. else if (PLY::DOM::TokenMatch(buffer, "alpha", 5)) {
  102. eOut = PLY::EST_Alpha;
  103. }
  104. else if (PLY::DOM::TokenMatch(buffer, "vertex_index", 12) || PLY::DOM::TokenMatch(buffer, "vertex_indices", 14)) {
  105. eOut = PLY::EST_VertexIndex;
  106. }
  107. else if (PLY::DOM::TokenMatch(buffer, "texcoord", 8)) // Manage uv coords on faces
  108. {
  109. eOut = PLY::EST_TextureCoordinates;
  110. }
  111. else if (PLY::DOM::TokenMatch(buffer, "material_index", 14))
  112. {
  113. eOut = PLY::EST_MaterialIndex;
  114. }
  115. else if (PLY::DOM::TokenMatch(buffer, "ambient_red", 11))
  116. {
  117. eOut = PLY::EST_AmbientRed;
  118. }
  119. else if (PLY::DOM::TokenMatch(buffer, "ambient_green", 13))
  120. {
  121. eOut = PLY::EST_AmbientGreen;
  122. }
  123. else if (PLY::DOM::TokenMatch(buffer, "ambient_blue", 12))
  124. {
  125. eOut = PLY::EST_AmbientBlue;
  126. }
  127. else if (PLY::DOM::TokenMatch(buffer, "ambient_alpha", 13))
  128. {
  129. eOut = PLY::EST_AmbientAlpha;
  130. }
  131. else if (PLY::DOM::TokenMatch(buffer, "diffuse_red", 11))
  132. {
  133. eOut = PLY::EST_DiffuseRed;
  134. }
  135. else if (PLY::DOM::TokenMatch(buffer, "diffuse_green", 13))
  136. {
  137. eOut = PLY::EST_DiffuseGreen;
  138. }
  139. else if (PLY::DOM::TokenMatch(buffer, "diffuse_blue", 12))
  140. {
  141. eOut = PLY::EST_DiffuseBlue;
  142. }
  143. else if (PLY::DOM::TokenMatch(buffer, "diffuse_alpha", 13))
  144. {
  145. eOut = PLY::EST_DiffuseAlpha;
  146. }
  147. else if (PLY::DOM::TokenMatch(buffer, "specular_red", 12))
  148. {
  149. eOut = PLY::EST_SpecularRed;
  150. }
  151. else if (PLY::DOM::TokenMatch(buffer, "specular_green", 14))
  152. {
  153. eOut = PLY::EST_SpecularGreen;
  154. }
  155. else if (PLY::DOM::TokenMatch(buffer, "specular_blue", 13))
  156. {
  157. eOut = PLY::EST_SpecularBlue;
  158. }
  159. else if (PLY::DOM::TokenMatch(buffer, "specular_alpha", 14))
  160. {
  161. eOut = PLY::EST_SpecularAlpha;
  162. }
  163. else if (PLY::DOM::TokenMatch(buffer, "opacity", 7))
  164. {
  165. eOut = PLY::EST_Opacity;
  166. }
  167. else if (PLY::DOM::TokenMatch(buffer, "specular_power", 14))
  168. {
  169. eOut = PLY::EST_PhongPower;
  170. }
  171. else if (PLY::DOM::TokenMatch(buffer, "r", 1))
  172. {
  173. eOut = PLY::EST_Red;
  174. }
  175. else if (PLY::DOM::TokenMatch(buffer, "g", 1))
  176. {
  177. eOut = PLY::EST_Green;
  178. }
  179. else if (PLY::DOM::TokenMatch(buffer, "b", 1))
  180. {
  181. eOut = PLY::EST_Blue;
  182. }
  183. // NOTE: Blender3D exports texture coordinates as s,t tuples
  184. 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))
  185. {
  186. eOut = PLY::EST_UTextureCoord;
  187. }
  188. 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))
  189. {
  190. eOut = PLY::EST_VTextureCoord;
  191. }
  192. else if (PLY::DOM::TokenMatch(buffer, "x", 1))
  193. {
  194. eOut = PLY::EST_XCoord;
  195. }
  196. else if (PLY::DOM::TokenMatch(buffer, "y", 1)) {
  197. eOut = PLY::EST_YCoord;
  198. }
  199. else if (PLY::DOM::TokenMatch(buffer, "z", 1)) {
  200. eOut = PLY::EST_ZCoord;
  201. }
  202. else if (PLY::DOM::TokenMatch(buffer, "nx", 2)) {
  203. eOut = PLY::EST_XNormal;
  204. }
  205. else if (PLY::DOM::TokenMatch(buffer, "ny", 2)) {
  206. eOut = PLY::EST_YNormal;
  207. }
  208. else if (PLY::DOM::TokenMatch(buffer, "nz", 2)) {
  209. eOut = PLY::EST_ZNormal;
  210. }
  211. else {
  212. ASSIMP_LOG_INFO("Found unknown property semantic in file. This is ok");
  213. PLY::DOM::SkipLine(buffer);
  214. }
  215. return eOut;
  216. }
  217. // ------------------------------------------------------------------------------------------------
  218. bool PLY::Property::ParseProperty(std::vector<char> &buffer, PLY::Property* pOut)
  219. {
  220. ai_assert(!buffer.empty());
  221. // Forms supported:
  222. // "property float x"
  223. // "property list uchar int vertex_index"
  224. // skip leading spaces
  225. if (!PLY::DOM::SkipSpaces(buffer)) {
  226. return false;
  227. }
  228. // skip the "property" string at the beginning
  229. if (!PLY::DOM::TokenMatch(buffer, "property", 8))
  230. {
  231. // seems not to be a valid property entry
  232. return false;
  233. }
  234. // get next word
  235. if (!PLY::DOM::SkipSpaces(buffer)) {
  236. return false;
  237. }
  238. if (PLY::DOM::TokenMatch(buffer, "list", 4))
  239. {
  240. pOut->bIsList = true;
  241. // seems to be a list.
  242. if (EDT_INVALID == (pOut->eFirstType = PLY::Property::ParseDataType(buffer)))
  243. {
  244. // unable to parse list size data type
  245. PLY::DOM::SkipLine(buffer);
  246. return false;
  247. }
  248. if (!PLY::DOM::SkipSpaces(buffer))return false;
  249. if (EDT_INVALID == (pOut->eType = PLY::Property::ParseDataType(buffer)))
  250. {
  251. // unable to parse list data type
  252. PLY::DOM::SkipLine(buffer);
  253. return false;
  254. }
  255. }
  256. else
  257. {
  258. if (EDT_INVALID == (pOut->eType = PLY::Property::ParseDataType(buffer)))
  259. {
  260. // unable to parse data type. Skip the property
  261. PLY::DOM::SkipLine(buffer);
  262. return false;
  263. }
  264. }
  265. if (!PLY::DOM::SkipSpaces(buffer))
  266. return false;
  267. pOut->Semantic = PLY::Property::ParseSemantic(buffer);
  268. if (PLY::EST_INVALID == pOut->Semantic)
  269. {
  270. ASSIMP_LOG_INFO("Found unknown semantic in PLY file. This is OK");
  271. std::string(&buffer[0], &buffer[0] + strlen(&buffer[0]));
  272. }
  273. PLY::DOM::SkipSpacesAndLineEnd(buffer);
  274. return true;
  275. }
  276. // ------------------------------------------------------------------------------------------------
  277. PLY::EElementSemantic PLY::Element::ParseSemantic(std::vector<char> &buffer)
  278. {
  279. ai_assert(!buffer.empty());
  280. PLY::EElementSemantic eOut = PLY::EEST_INVALID;
  281. if (PLY::DOM::TokenMatch(buffer, "vertex", 6))
  282. {
  283. eOut = PLY::EEST_Vertex;
  284. }
  285. else if (PLY::DOM::TokenMatch(buffer, "face", 4))
  286. {
  287. eOut = PLY::EEST_Face;
  288. }
  289. else if (PLY::DOM::TokenMatch(buffer, "tristrips", 9))
  290. {
  291. eOut = PLY::EEST_TriStrip;
  292. }
  293. #if 0
  294. // TODO: maybe implement this?
  295. else if (PLY::DOM::TokenMatch(buffer,"range_grid",10))
  296. {
  297. eOut = PLY::EEST_Face;
  298. }
  299. #endif
  300. else if (PLY::DOM::TokenMatch(buffer, "edge", 4))
  301. {
  302. eOut = PLY::EEST_Edge;
  303. }
  304. else if (PLY::DOM::TokenMatch(buffer, "material", 8))
  305. {
  306. eOut = PLY::EEST_Material;
  307. }
  308. else if (PLY::DOM::TokenMatch(buffer, "TextureFile", 11))
  309. {
  310. eOut = PLY::EEST_TextureFile;
  311. }
  312. return eOut;
  313. }
  314. // ------------------------------------------------------------------------------------------------
  315. bool PLY::Element::ParseElement(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, PLY::Element* pOut)
  316. {
  317. ai_assert(NULL != pOut);
  318. // Example format: "element vertex 8"
  319. // skip leading spaces
  320. if (!PLY::DOM::SkipSpaces(buffer))
  321. {
  322. return false;
  323. }
  324. // skip the "element" string at the beginning
  325. if (!PLY::DOM::TokenMatch(buffer, "element", 7) && !PLY::DOM::TokenMatch(buffer, "comment", 7))
  326. {
  327. // seems not to be a valid property entry
  328. return false;
  329. }
  330. // get next word
  331. if (!PLY::DOM::SkipSpaces(buffer))
  332. return false;
  333. // parse the semantic of the element
  334. pOut->eSemantic = PLY::Element::ParseSemantic(buffer);
  335. if (PLY::EEST_INVALID == pOut->eSemantic)
  336. {
  337. // if the exact semantic can't be determined, just store
  338. // the original string identifier
  339. pOut->szName = std::string(&buffer[0], &buffer[0] + strlen(&buffer[0]));
  340. }
  341. if (!PLY::DOM::SkipSpaces(buffer))
  342. return false;
  343. if (PLY::EEST_TextureFile == pOut->eSemantic)
  344. {
  345. char* endPos = &buffer[0] + (strlen(&buffer[0]) - 1);
  346. pOut->szName = std::string(&buffer[0], endPos);
  347. // go to the next line
  348. PLY::DOM::SkipSpacesAndLineEnd(buffer);
  349. return true;
  350. }
  351. //parse the number of occurrences of this element
  352. const char* pCur = (char*)&buffer[0];
  353. pOut->NumOccur = strtoul10(pCur, &pCur);
  354. // go to the next line
  355. PLY::DOM::SkipSpacesAndLineEnd(buffer);
  356. // now parse all properties of the element
  357. while (true)
  358. {
  359. streamBuffer.getNextLine(buffer);
  360. pCur = (char*)&buffer[0];
  361. // skip all comments
  362. PLY::DOM::SkipComments(buffer);
  363. PLY::Property prop;
  364. if (!PLY::Property::ParseProperty(buffer, &prop))
  365. break;
  366. pOut->alProperties.push_back(prop);
  367. }
  368. return true;
  369. }
  370. // ------------------------------------------------------------------------------------------------
  371. bool PLY::DOM::SkipSpaces(std::vector<char> &buffer)
  372. {
  373. const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
  374. bool ret = false;
  375. if (pCur)
  376. {
  377. const char* szCur = pCur;
  378. ret = Assimp::SkipSpaces(pCur, &pCur);
  379. uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
  380. buffer.erase(buffer.begin(), buffer.begin() + iDiff);
  381. return ret;
  382. }
  383. return ret;
  384. }
  385. bool PLY::DOM::SkipLine(std::vector<char> &buffer)
  386. {
  387. const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
  388. bool ret = false;
  389. if (pCur)
  390. {
  391. const char* szCur = pCur;
  392. ret = Assimp::SkipLine(pCur, &pCur);
  393. uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
  394. buffer.erase(buffer.begin(), buffer.begin() + iDiff);
  395. return ret;
  396. }
  397. return ret;
  398. }
  399. bool PLY::DOM::TokenMatch(std::vector<char> &buffer, const char* token, unsigned int len)
  400. {
  401. const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
  402. bool ret = false;
  403. if (pCur)
  404. {
  405. const char* szCur = pCur;
  406. ret = Assimp::TokenMatch(pCur, token, len);
  407. uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
  408. buffer.erase(buffer.begin(), buffer.begin() + iDiff);
  409. return ret;
  410. }
  411. return ret;
  412. }
  413. bool PLY::DOM::SkipSpacesAndLineEnd(std::vector<char> &buffer)
  414. {
  415. const char* pCur = buffer.empty() ? NULL : (char*)&buffer[0];
  416. bool ret = false;
  417. if (pCur)
  418. {
  419. const char* szCur = pCur;
  420. ret = Assimp::SkipSpacesAndLineEnd(pCur, &pCur);
  421. uintptr_t iDiff = (uintptr_t)pCur - (uintptr_t)szCur;
  422. buffer.erase(buffer.begin(), buffer.begin() + iDiff);
  423. return ret;
  424. }
  425. return ret;
  426. }
  427. bool PLY::DOM::SkipComments(std::vector<char> &buffer)
  428. {
  429. ai_assert(!buffer.empty());
  430. std::vector<char> nbuffer = buffer;
  431. // skip spaces
  432. if (!SkipSpaces(nbuffer)) {
  433. return false;
  434. }
  435. if (TokenMatch(nbuffer, "comment", 7))
  436. {
  437. if (!SkipSpaces(nbuffer))
  438. SkipLine(nbuffer);
  439. if (!TokenMatch(nbuffer, "TextureFile", 11))
  440. {
  441. SkipLine(nbuffer);
  442. buffer = nbuffer;
  443. return true;
  444. }
  445. return true;
  446. }
  447. return false;
  448. }
  449. // ------------------------------------------------------------------------------------------------
  450. bool PLY::DOM::ParseHeader(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, bool isBinary) {
  451. ASSIMP_LOG_DEBUG("PLY::DOM::ParseHeader() begin");
  452. // parse all elements
  453. while (!buffer.empty())
  454. {
  455. // skip all comments
  456. PLY::DOM::SkipComments(buffer);
  457. PLY::Element out;
  458. if (PLY::Element::ParseElement(streamBuffer, buffer, &out))
  459. {
  460. // add the element to the list of elements
  461. alElements.push_back(out);
  462. }
  463. else if (TokenMatch(buffer, "end_header", 10))
  464. {
  465. // we have reached the end of the header
  466. break;
  467. }
  468. else
  469. {
  470. // ignore unknown header elements
  471. streamBuffer.getNextLine(buffer);
  472. }
  473. }
  474. if (!isBinary) // it would occur an error, if binary data start with values as space or line end.
  475. SkipSpacesAndLineEnd(buffer);
  476. ASSIMP_LOG_DEBUG("PLY::DOM::ParseHeader() succeeded");
  477. return true;
  478. }
  479. // ------------------------------------------------------------------------------------------------
  480. bool PLY::DOM::ParseElementInstanceLists(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, PLYImporter* loader)
  481. {
  482. ASSIMP_LOG_DEBUG("PLY::DOM::ParseElementInstanceLists() begin");
  483. alElementData.resize(alElements.size());
  484. std::vector<PLY::Element>::const_iterator i = alElements.begin();
  485. std::vector<PLY::ElementInstanceList>::iterator a = alElementData.begin();
  486. // parse all element instances
  487. //construct vertices and faces
  488. for (; i != alElements.end(); ++i, ++a)
  489. {
  490. if ((*i).eSemantic == EEST_Vertex || (*i).eSemantic == EEST_Face || (*i).eSemantic == EEST_TriStrip)
  491. {
  492. PLY::ElementInstanceList::ParseInstanceList(streamBuffer, buffer, &(*i), NULL, loader);
  493. }
  494. else
  495. {
  496. (*a).alInstances.resize((*i).NumOccur);
  497. PLY::ElementInstanceList::ParseInstanceList(streamBuffer, buffer, &(*i), &(*a), NULL);
  498. }
  499. }
  500. ASSIMP_LOG_DEBUG("PLY::DOM::ParseElementInstanceLists() succeeded");
  501. return true;
  502. }
  503. // ------------------------------------------------------------------------------------------------
  504. bool PLY::DOM::ParseElementInstanceListsBinary(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer,
  505. const char* &pCur,
  506. unsigned int &bufferSize,
  507. PLYImporter* loader,
  508. bool p_bBE)
  509. {
  510. ASSIMP_LOG_DEBUG("PLY::DOM::ParseElementInstanceListsBinary() begin");
  511. alElementData.resize(alElements.size());
  512. std::vector<PLY::Element>::const_iterator i = alElements.begin();
  513. std::vector<PLY::ElementInstanceList>::iterator a = alElementData.begin();
  514. // parse all element instances
  515. for (; i != alElements.end(); ++i, ++a)
  516. {
  517. if ((*i).eSemantic == EEST_Vertex || (*i).eSemantic == EEST_Face || (*i).eSemantic == EEST_TriStrip)
  518. {
  519. PLY::ElementInstanceList::ParseInstanceListBinary(streamBuffer, buffer, pCur, bufferSize, &(*i), NULL, loader, p_bBE);
  520. }
  521. else
  522. {
  523. (*a).alInstances.resize((*i).NumOccur);
  524. PLY::ElementInstanceList::ParseInstanceListBinary(streamBuffer, buffer, pCur, bufferSize, &(*i), &(*a), NULL, p_bBE);
  525. }
  526. }
  527. ASSIMP_LOG_DEBUG("PLY::DOM::ParseElementInstanceListsBinary() succeeded");
  528. return true;
  529. }
  530. // ------------------------------------------------------------------------------------------------
  531. bool PLY::DOM::ParseInstanceBinary(IOStreamBuffer<char> &streamBuffer, DOM* p_pcOut, PLYImporter* loader, bool p_bBE)
  532. {
  533. ai_assert(NULL != p_pcOut);
  534. ai_assert(NULL != loader);
  535. std::vector<char> buffer;
  536. streamBuffer.getNextLine(buffer);
  537. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstanceBinary() begin");
  538. if (!p_pcOut->ParseHeader(streamBuffer, buffer, true))
  539. {
  540. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstanceBinary() failure");
  541. return false;
  542. }
  543. streamBuffer.getNextBlock(buffer);
  544. unsigned int bufferSize = static_cast<unsigned int>(buffer.size());
  545. const char* pCur = (char*)&buffer[0];
  546. if (!p_pcOut->ParseElementInstanceListsBinary(streamBuffer, buffer, pCur, bufferSize, loader, p_bBE))
  547. {
  548. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstanceBinary() failure");
  549. return false;
  550. }
  551. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstanceBinary() succeeded");
  552. return true;
  553. }
  554. // ------------------------------------------------------------------------------------------------
  555. bool PLY::DOM::ParseInstance(IOStreamBuffer<char> &streamBuffer, DOM* p_pcOut, PLYImporter* loader)
  556. {
  557. ai_assert(NULL != p_pcOut);
  558. ai_assert(NULL != loader);
  559. std::vector<char> buffer;
  560. streamBuffer.getNextLine(buffer);
  561. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstance() begin");
  562. if (!p_pcOut->ParseHeader(streamBuffer, buffer, false))
  563. {
  564. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstance() failure");
  565. return false;
  566. }
  567. //get next line after header
  568. streamBuffer.getNextLine(buffer);
  569. if (!p_pcOut->ParseElementInstanceLists(streamBuffer, buffer, loader))
  570. {
  571. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstance() failure");
  572. return false;
  573. }
  574. ASSIMP_LOG_DEBUG("PLY::DOM::ParseInstance() succeeded");
  575. return true;
  576. }
  577. // ------------------------------------------------------------------------------------------------
  578. bool PLY::ElementInstanceList::ParseInstanceList(
  579. IOStreamBuffer<char> &streamBuffer,
  580. std::vector<char> &buffer,
  581. const PLY::Element* pcElement,
  582. PLY::ElementInstanceList* p_pcOut,
  583. PLYImporter* loader)
  584. {
  585. ai_assert(NULL != pcElement);
  586. // parse all elements
  587. if (EEST_INVALID == pcElement->eSemantic || pcElement->alProperties.empty())
  588. {
  589. // if the element has an unknown semantic we can skip all lines
  590. // However, there could be comments
  591. for (unsigned int i = 0; i < pcElement->NumOccur; ++i)
  592. {
  593. PLY::DOM::SkipComments(buffer);
  594. PLY::DOM::SkipLine(buffer);
  595. streamBuffer.getNextLine(buffer);
  596. }
  597. }
  598. else
  599. {
  600. const char* pCur = (const char*)&buffer[0];
  601. // be sure to have enough storage
  602. for (unsigned int i = 0; i < pcElement->NumOccur; ++i)
  603. {
  604. if (p_pcOut)
  605. PLY::ElementInstance::ParseInstance(pCur, pcElement, &p_pcOut->alInstances[i]);
  606. else
  607. {
  608. ElementInstance elt;
  609. PLY::ElementInstance::ParseInstance(pCur, pcElement, &elt);
  610. // Create vertex or face
  611. if (pcElement->eSemantic == EEST_Vertex)
  612. {
  613. //call loader instance from here
  614. loader->LoadVertex(pcElement, &elt, i);
  615. }
  616. else if (pcElement->eSemantic == EEST_Face)
  617. {
  618. //call loader instance from here
  619. loader->LoadFace(pcElement, &elt, i);
  620. }
  621. else if (pcElement->eSemantic == EEST_TriStrip)
  622. {
  623. //call loader instance from here
  624. loader->LoadFace(pcElement, &elt, i);
  625. }
  626. }
  627. streamBuffer.getNextLine(buffer);
  628. pCur = (buffer.empty()) ? NULL : (const char*)&buffer[0];
  629. }
  630. }
  631. return true;
  632. }
  633. // ------------------------------------------------------------------------------------------------
  634. bool PLY::ElementInstanceList::ParseInstanceListBinary(
  635. IOStreamBuffer<char> &streamBuffer,
  636. std::vector<char> &buffer,
  637. const char* &pCur,
  638. unsigned int &bufferSize,
  639. const PLY::Element* pcElement,
  640. PLY::ElementInstanceList* p_pcOut,
  641. PLYImporter* loader,
  642. bool p_bBE /* = false */)
  643. {
  644. ai_assert(NULL != pcElement);
  645. // we can add special handling code for unknown element semantics since
  646. // we can't skip it as a whole block (we don't know its exact size
  647. // due to the fact that lists could be contained in the property list
  648. // of the unknown element)
  649. for (unsigned int i = 0; i < pcElement->NumOccur; ++i)
  650. {
  651. if (p_pcOut)
  652. PLY::ElementInstance::ParseInstanceBinary(streamBuffer, buffer, pCur, bufferSize, pcElement, &p_pcOut->alInstances[i], p_bBE);
  653. else
  654. {
  655. ElementInstance elt;
  656. PLY::ElementInstance::ParseInstanceBinary(streamBuffer, buffer, pCur, bufferSize, pcElement, &elt, p_bBE);
  657. // Create vertex or face
  658. if (pcElement->eSemantic == EEST_Vertex)
  659. {
  660. //call loader instance from here
  661. loader->LoadVertex(pcElement, &elt, i);
  662. }
  663. else if (pcElement->eSemantic == EEST_Face)
  664. {
  665. //call loader instance from here
  666. loader->LoadFace(pcElement, &elt, i);
  667. }
  668. else if (pcElement->eSemantic == EEST_TriStrip)
  669. {
  670. //call loader instance from here
  671. loader->LoadFace(pcElement, &elt, i);
  672. }
  673. }
  674. }
  675. return true;
  676. }
  677. // ------------------------------------------------------------------------------------------------
  678. bool PLY::ElementInstance::ParseInstance(const char* &pCur,
  679. const PLY::Element* pcElement,
  680. PLY::ElementInstance* p_pcOut)
  681. {
  682. ai_assert(NULL != pcElement);
  683. ai_assert(NULL != p_pcOut);
  684. // allocate enough storage
  685. p_pcOut->alProperties.resize(pcElement->alProperties.size());
  686. std::vector<PLY::PropertyInstance>::iterator i = p_pcOut->alProperties.begin();
  687. std::vector<PLY::Property>::const_iterator a = pcElement->alProperties.begin();
  688. for (; i != p_pcOut->alProperties.end(); ++i, ++a)
  689. {
  690. if (!(PLY::PropertyInstance::ParseInstance(pCur, &(*a), &(*i))))
  691. {
  692. ASSIMP_LOG_WARN("Unable to parse property instance. "
  693. "Skipping this element instance");
  694. PLY::PropertyInstance::ValueUnion v = PLY::PropertyInstance::DefaultValue((*a).eType);
  695. (*i).avList.push_back(v);
  696. }
  697. }
  698. return true;
  699. }
  700. // ------------------------------------------------------------------------------------------------
  701. bool PLY::ElementInstance::ParseInstanceBinary(
  702. IOStreamBuffer<char> &streamBuffer,
  703. std::vector<char> &buffer,
  704. const char* &pCur,
  705. unsigned int &bufferSize,
  706. const PLY::Element* pcElement,
  707. PLY::ElementInstance* p_pcOut,
  708. bool p_bBE /* = false */)
  709. {
  710. ai_assert(NULL != pcElement);
  711. ai_assert(NULL != p_pcOut);
  712. // allocate enough storage
  713. p_pcOut->alProperties.resize(pcElement->alProperties.size());
  714. std::vector<PLY::PropertyInstance>::iterator i = p_pcOut->alProperties.begin();
  715. std::vector<PLY::Property>::const_iterator a = pcElement->alProperties.begin();
  716. for (; i != p_pcOut->alProperties.end(); ++i, ++a)
  717. {
  718. if (!(PLY::PropertyInstance::ParseInstanceBinary(streamBuffer, buffer, pCur, bufferSize, &(*a), &(*i), p_bBE)))
  719. {
  720. ASSIMP_LOG_WARN("Unable to parse binary property instance. "
  721. "Skipping this element instance");
  722. (*i).avList.push_back(PLY::PropertyInstance::DefaultValue((*a).eType));
  723. }
  724. }
  725. return true;
  726. }
  727. // ------------------------------------------------------------------------------------------------
  728. bool PLY::PropertyInstance::ParseInstance(const char* &pCur,
  729. const PLY::Property* prop, PLY::PropertyInstance* p_pcOut)
  730. {
  731. ai_assert(NULL != prop);
  732. ai_assert(NULL != p_pcOut);
  733. // skip spaces at the beginning
  734. if (!SkipSpaces(&pCur))
  735. {
  736. return false;
  737. }
  738. if (prop->bIsList)
  739. {
  740. // parse the number of elements in the list
  741. PLY::PropertyInstance::ValueUnion v;
  742. PLY::PropertyInstance::ParseValue(pCur, prop->eFirstType, &v);
  743. // convert to unsigned int
  744. unsigned int iNum = PLY::PropertyInstance::ConvertTo<unsigned int>(v, prop->eFirstType);
  745. // parse all list elements
  746. p_pcOut->avList.resize(iNum);
  747. for (unsigned int i = 0; i < iNum; ++i)
  748. {
  749. if (!SkipSpaces(&pCur))
  750. return false;
  751. PLY::PropertyInstance::ParseValue(pCur, prop->eType, &p_pcOut->avList[i]);
  752. }
  753. }
  754. else
  755. {
  756. // parse the property
  757. PLY::PropertyInstance::ValueUnion v;
  758. PLY::PropertyInstance::ParseValue(pCur, prop->eType, &v);
  759. p_pcOut->avList.push_back(v);
  760. }
  761. SkipSpacesAndLineEnd(&pCur);
  762. return true;
  763. }
  764. // ------------------------------------------------------------------------------------------------
  765. bool PLY::PropertyInstance::ParseInstanceBinary(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer,
  766. const char* &pCur,
  767. unsigned int &bufferSize,
  768. const PLY::Property* prop,
  769. PLY::PropertyInstance* p_pcOut,
  770. bool p_bBE)
  771. {
  772. ai_assert(NULL != prop);
  773. ai_assert(NULL != p_pcOut);
  774. // parse all elements
  775. if (prop->bIsList)
  776. {
  777. // parse the number of elements in the list
  778. PLY::PropertyInstance::ValueUnion v;
  779. PLY::PropertyInstance::ParseValueBinary(streamBuffer, buffer, pCur, bufferSize, prop->eFirstType, &v, p_bBE);
  780. // convert to unsigned int
  781. unsigned int iNum = PLY::PropertyInstance::ConvertTo<unsigned int>(v, prop->eFirstType);
  782. // parse all list elements
  783. p_pcOut->avList.resize(iNum);
  784. for (unsigned int i = 0; i < iNum; ++i)
  785. {
  786. PLY::PropertyInstance::ParseValueBinary(streamBuffer, buffer, pCur, bufferSize, prop->eType, &p_pcOut->avList[i], p_bBE);
  787. }
  788. }
  789. else
  790. {
  791. // parse the property
  792. PLY::PropertyInstance::ValueUnion v;
  793. PLY::PropertyInstance::ParseValueBinary(streamBuffer, buffer, pCur, bufferSize, prop->eType, &v, p_bBE);
  794. p_pcOut->avList.push_back(v);
  795. }
  796. return true;
  797. }
  798. // ------------------------------------------------------------------------------------------------
  799. PLY::PropertyInstance::ValueUnion PLY::PropertyInstance::DefaultValue(PLY::EDataType eType)
  800. {
  801. PLY::PropertyInstance::ValueUnion out;
  802. switch (eType)
  803. {
  804. case EDT_Float:
  805. out.fFloat = 0.f;
  806. return out;
  807. case EDT_Double:
  808. out.fDouble = 0.;
  809. return out;
  810. default:;
  811. };
  812. out.iUInt = 0;
  813. return out;
  814. }
  815. // ------------------------------------------------------------------------------------------------
  816. bool PLY::PropertyInstance::ParseValue(const char* &pCur,
  817. PLY::EDataType eType,
  818. PLY::PropertyInstance::ValueUnion* out)
  819. {
  820. ai_assert(NULL != pCur);
  821. ai_assert(NULL != out);
  822. //calc element size
  823. bool ret = true;
  824. switch (eType)
  825. {
  826. case EDT_UInt:
  827. case EDT_UShort:
  828. case EDT_UChar:
  829. out->iUInt = (uint32_t)strtoul10(pCur, &pCur);
  830. break;
  831. case EDT_Int:
  832. case EDT_Short:
  833. case EDT_Char:
  834. out->iInt = (int32_t)strtol10(pCur, &pCur);
  835. break;
  836. case EDT_Float:
  837. // technically this should cast to float, but people tend to use float descriptors for double data
  838. // this is the best way to not risk losing precision on import and it doesn't hurt to do this
  839. ai_real f;
  840. pCur = fast_atoreal_move<ai_real>(pCur, f);
  841. out->fFloat = (ai_real)f;
  842. break;
  843. case EDT_Double:
  844. double d;
  845. pCur = fast_atoreal_move<double>(pCur, d);
  846. out->fDouble = (double)d;
  847. break;
  848. case EDT_INVALID:
  849. default:
  850. ret = false;
  851. break;
  852. }
  853. return ret;
  854. }
  855. // ------------------------------------------------------------------------------------------------
  856. bool PLY::PropertyInstance::ParseValueBinary(IOStreamBuffer<char> &streamBuffer,
  857. std::vector<char> &buffer,
  858. const char* &pCur,
  859. unsigned int &bufferSize,
  860. PLY::EDataType eType,
  861. PLY::PropertyInstance::ValueUnion* out,
  862. bool p_bBE)
  863. {
  864. ai_assert(NULL != out);
  865. //calc element size
  866. unsigned int lsize = 0;
  867. switch (eType)
  868. {
  869. case EDT_Char:
  870. case EDT_UChar:
  871. lsize = 1;
  872. break;
  873. case EDT_UShort:
  874. case EDT_Short:
  875. lsize = 2;
  876. break;
  877. case EDT_UInt:
  878. case EDT_Int:
  879. case EDT_Float:
  880. lsize = 4;
  881. break;
  882. case EDT_Double:
  883. lsize = 8;
  884. break;
  885. case EDT_INVALID:
  886. default:
  887. break;
  888. }
  889. //read the next file block if needed
  890. if (bufferSize < lsize)
  891. {
  892. std::vector<char> nbuffer;
  893. if (streamBuffer.getNextBlock(nbuffer))
  894. {
  895. //concat buffer contents
  896. buffer = std::vector<char>(buffer.end() - bufferSize, buffer.end());
  897. buffer.insert(buffer.end(), nbuffer.begin(), nbuffer.end());
  898. nbuffer.clear();
  899. bufferSize = static_cast<unsigned int>(buffer.size());
  900. pCur = (char*)&buffer[0];
  901. }
  902. else
  903. {
  904. throw DeadlyImportError("Invalid .ply file: File corrupted");
  905. }
  906. }
  907. bool ret = true;
  908. switch (eType)
  909. {
  910. case EDT_UInt:
  911. {
  912. uint32_t t;
  913. memcpy(&t, pCur, sizeof(uint32_t));
  914. pCur += sizeof(uint32_t);
  915. // Swap endianness
  916. if (p_bBE)ByteSwap::Swap(&t);
  917. out->iUInt = t;
  918. break;
  919. }
  920. case EDT_UShort:
  921. {
  922. uint16_t t;
  923. memcpy(&t, pCur, sizeof(uint16_t));
  924. pCur += sizeof(uint16_t);
  925. // Swap endianness
  926. if (p_bBE)ByteSwap::Swap(&t);
  927. out->iUInt = t;
  928. break;
  929. }
  930. case EDT_UChar:
  931. {
  932. uint8_t t;
  933. memcpy(&t, pCur, sizeof(uint8_t));
  934. pCur += sizeof(uint8_t);
  935. out->iUInt = t;
  936. break;
  937. }
  938. case EDT_Int:
  939. {
  940. int32_t t;
  941. memcpy(&t, pCur, sizeof(int32_t));
  942. pCur += sizeof(int32_t);
  943. // Swap endianness
  944. if (p_bBE)ByteSwap::Swap(&t);
  945. out->iInt = t;
  946. break;
  947. }
  948. case EDT_Short:
  949. {
  950. int16_t t;
  951. memcpy(&t, pCur, sizeof(int16_t));
  952. pCur += sizeof(int16_t);
  953. // Swap endianness
  954. if (p_bBE)ByteSwap::Swap(&t);
  955. out->iInt = t;
  956. break;
  957. }
  958. case EDT_Char:
  959. {
  960. int8_t t;
  961. memcpy(&t, pCur, sizeof(int8_t));
  962. pCur += sizeof(int8_t);
  963. out->iInt = t;
  964. break;
  965. }
  966. case EDT_Float:
  967. {
  968. float t;
  969. memcpy(&t, pCur, sizeof(float));
  970. pCur += sizeof(float);
  971. // Swap endianness
  972. if (p_bBE)ByteSwap::Swap(&t);
  973. out->fFloat = t;
  974. break;
  975. }
  976. case EDT_Double:
  977. {
  978. double t;
  979. memcpy(&t, pCur, sizeof(double));
  980. pCur += sizeof(double);
  981. // Swap endianness
  982. if (p_bBE)ByteSwap::Swap(&t);
  983. out->fDouble = t;
  984. break;
  985. }
  986. default:
  987. ret = false;
  988. }
  989. bufferSize -= lsize;
  990. return ret;
  991. }
  992. #endif // !! ASSIMP_BUILD_NO_PLY_IMPORTER