geometryc.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904
  1. /*
  2. * Copyright 2011-2012 Branimir Karadzic. All rights reserved.
  3. * License: http://www.opensource.org/licenses/BSD-2-Clause
  4. */
  5. #include <bgfx.h>
  6. #include "../../src/vertexdecl.h"
  7. #include <stdio.h>
  8. #include <stdlib.h>
  9. #include <string.h>
  10. #include <algorithm>
  11. #include <vector>
  12. namespace std { namespace tr1 {} using namespace tr1; } // namespace std
  13. #include <unordered_map>
  14. #include <forsythtriangleorderoptimizer.h>
  15. #if 0
  16. # define BX_TRACE(_format, ...) \
  17. do { \
  18. printf(BX_FILE_LINE_LITERAL "BGFX " _format "\n", ##__VA_ARGS__); \
  19. } while(0)
  20. # define BX_WARN(_condition, _format, ...) \
  21. do { \
  22. if (!(_condition) ) \
  23. { \
  24. BX_TRACE(BX_FILE_LINE_LITERAL "WARN " _format, ##__VA_ARGS__); \
  25. } \
  26. } while(0)
  27. # define BX_CHECK(_condition, _format, ...) \
  28. do { \
  29. if (!(_condition) ) \
  30. { \
  31. BX_TRACE(BX_FILE_LINE_LITERAL "CHECK " _format, ##__VA_ARGS__); \
  32. bx::debugBreak(); \
  33. } \
  34. } while(0)
  35. #endif // 0
  36. #define EXPECT(_condition) \
  37. do { \
  38. if (!(_condition) ) \
  39. { \
  40. printf("Error parsing at:\n" BX_FILE_LINE_LITERAL "\nExpected: " #_condition "\n"); \
  41. exit(EXIT_FAILURE); \
  42. } \
  43. } while(0)
  44. #include <bx/bx.h>
  45. #include <bx/debug.h>
  46. #include <bx/countof.h>
  47. #include <bx/commandline.h>
  48. #include <bx/timer.h>
  49. #include <bx/readerwriter.h>
  50. #include <bx/hash.h>
  51. #include <bx/uint32_t.h>
  52. #include "tokenizecmd.h"
  53. #include "bounds.h"
  54. #include "math.h"
  55. struct Vector3
  56. {
  57. float x;
  58. float y;
  59. float z;
  60. };
  61. typedef std::vector<Vector3> Vector3Array;
  62. struct Index3
  63. {
  64. int32_t m_position;
  65. int32_t m_texcoord;
  66. int32_t m_normal;
  67. int32_t m_vertexIndex;
  68. };
  69. typedef std::unordered_map<uint64_t, Index3> Index3Map;
  70. struct Triangle
  71. {
  72. uint64_t m_index[3];
  73. };
  74. typedef std::vector<Triangle> TriangleArray;
  75. struct Group
  76. {
  77. uint32_t m_startTriangle;
  78. uint32_t m_numTriangles;
  79. std::string m_name;
  80. std::string m_material;
  81. };
  82. typedef std::vector<Group> GroupArray;
  83. struct Primitive
  84. {
  85. uint32_t m_startVertex;
  86. uint32_t m_startIndex;
  87. uint32_t m_numVertices;
  88. uint32_t m_numIndices;
  89. std::string m_name;
  90. };
  91. typedef std::vector<Primitive> PrimitiveArray;
  92. static uint32_t s_obbSteps = 17;
  93. #define BGFX_CHUNK_MAGIC_GEO BX_MAKEFOURCC('G', 'E', 'O', 0x0)
  94. #define BGFX_CHUNK_MAGIC_VB BX_MAKEFOURCC('V', 'B', ' ', 0x0)
  95. #define BGFX_CHUNK_MAGIC_IB BX_MAKEFOURCC('I', 'B', ' ', 0x0)
  96. #define BGFX_CHUNK_MAGIC_PRI BX_MAKEFOURCC('P', 'R', 'I', 0x0)
  97. long int fsize(FILE* _file)
  98. {
  99. long int pos = ftell(_file);
  100. fseek(_file, 0L, SEEK_END);
  101. long int size = ftell(_file);
  102. fseek(_file, pos, SEEK_SET);
  103. return size;
  104. }
  105. void triangleReorder(uint16_t* _indices, uint32_t _numIndices, uint32_t _numVertices, uint16_t _cacheSize)
  106. {
  107. uint16_t* newIndexList = new uint16_t[_numIndices];
  108. Forsyth::OptimizeFaces(_indices, _numIndices, _numVertices, newIndexList, _cacheSize);
  109. memcpy(_indices, newIndexList, _numIndices*2);
  110. delete [] newIndexList;
  111. }
  112. void calcTangents(void* _vertices, uint16_t _numVertices, bgfx::VertexDecl _decl, const uint16_t* _indices, uint32_t _numIndices)
  113. {
  114. struct PosTexcoord
  115. {
  116. float m_x;
  117. float m_y;
  118. float m_z;
  119. float m_pad0;
  120. float m_u;
  121. float m_v;
  122. float m_pad1;
  123. float m_pad2;
  124. };
  125. float* tangents = new float[6*_numVertices];
  126. memset(tangents, 0, 6*_numVertices*sizeof(float) );
  127. PosTexcoord v0;
  128. PosTexcoord v1;
  129. PosTexcoord v2;
  130. for (uint32_t ii = 0, num = _numIndices/3; ii < num; ++ii)
  131. {
  132. const uint16_t* indices = &_indices[ii*3];
  133. uint32_t i0 = indices[0];
  134. uint32_t i1 = indices[1];
  135. uint32_t i2 = indices[2];
  136. bgfx::vertexUnpack(&v0.m_x, bgfx::Attrib::Position, _decl, _vertices, i0);
  137. bgfx::vertexUnpack(&v0.m_u, bgfx::Attrib::TexCoord0, _decl, _vertices, i0);
  138. bgfx::vertexUnpack(&v1.m_x, bgfx::Attrib::Position, _decl, _vertices, i1);
  139. bgfx::vertexUnpack(&v1.m_u, bgfx::Attrib::TexCoord0, _decl, _vertices, i1);
  140. bgfx::vertexUnpack(&v2.m_x, bgfx::Attrib::Position, _decl, _vertices, i2);
  141. bgfx::vertexUnpack(&v2.m_u, bgfx::Attrib::TexCoord0, _decl, _vertices, i2);
  142. const float bax = v1.m_x - v0.m_x;
  143. const float bay = v1.m_y - v0.m_y;
  144. const float baz = v1.m_z - v0.m_z;
  145. const float bau = v1.m_u - v0.m_u;
  146. const float bav = v1.m_v - v0.m_v;
  147. const float cax = v2.m_x - v0.m_x;
  148. const float cay = v2.m_y - v0.m_y;
  149. const float caz = v2.m_z - v0.m_z;
  150. const float cau = v2.m_u - v0.m_u;
  151. const float cav = v2.m_v - v0.m_v;
  152. const float det = (bau * cav - bav * cau);
  153. const float invDet = 1.0f / det;
  154. const float tx = (bax * cav - cax * bav) * invDet;
  155. const float ty = (bay * cav - cay * bav) * invDet;
  156. const float tz = (baz * cav - caz * bav) * invDet;
  157. const float bx = (cax * bau - bax * cau) * invDet;
  158. const float by = (cay * bau - bay * cau) * invDet;
  159. const float bz = (caz * bau - baz * cau) * invDet;
  160. for (uint32_t jj = 0; jj < 3; ++jj)
  161. {
  162. float* tanu = &tangents[indices[jj]*6];
  163. float* tanv = &tanu[3];
  164. tanu[0] += tx;
  165. tanu[1] += ty;
  166. tanu[2] += tz;
  167. tanv[0] += bx;
  168. tanv[1] += by;
  169. tanv[2] += bz;
  170. }
  171. }
  172. for (uint32_t ii = 0; ii < _numVertices; ++ii)
  173. {
  174. const float* tanu = &tangents[ii*6];
  175. const float* tanv = &tangents[ii*6 + 3];
  176. float normal[4];
  177. bgfx::vertexUnpack(normal, bgfx::Attrib::Normal, _decl, _vertices, ii);
  178. float ndt = vec3Dot(normal, tanu);
  179. float nxt[3];
  180. vec3Cross(nxt, normal, tanu);
  181. float tmp[3];
  182. tmp[0] = tanu[0] - normal[0] * ndt;
  183. tmp[1] = tanu[1] - normal[1] * ndt;
  184. tmp[2] = tanu[2] - normal[2] * ndt;
  185. float tangent[4];
  186. vec3Norm(tangent, tmp);
  187. tangent[3] = vec3Dot(nxt, tanv) < 0.0f ? -1.0f : 1.0f;
  188. bgfx::vertexPack(tangent, true, bgfx::Attrib::Tangent, _decl, _vertices, ii);
  189. }
  190. }
  191. void writeBounds(bx::WriterI* _writer, const void* _vertices, uint32_t _numVertices, uint32_t _stride)
  192. {
  193. Sphere maxSphere;
  194. calcMaxBoundingSphere(maxSphere, _vertices, _numVertices, _stride);
  195. Sphere minSphere;
  196. calcMinBoundingSphere(minSphere, _vertices, _numVertices, _stride);
  197. if (minSphere.m_radius > maxSphere.m_radius)
  198. {
  199. bx::write(_writer, maxSphere);
  200. }
  201. else
  202. {
  203. bx::write(_writer, minSphere);
  204. }
  205. Aabb aabb;
  206. calcAabb(aabb, _vertices, _numVertices, _stride);
  207. bx::write(_writer, aabb);
  208. Obb obb;
  209. calcObb(obb, _vertices, _numVertices, _stride, s_obbSteps);
  210. bx::write(_writer, obb);
  211. }
  212. void write(bx::WriterI* _writer, const uint8_t* _vertices, uint32_t _numVertices, const bgfx::VertexDecl& _decl, const uint16_t* _indices, uint32_t _numIndices, const std::string& _material, const PrimitiveArray& _primitives)
  213. {
  214. uint32_t stride = _decl.getStride();
  215. bx::write(_writer, BGFX_CHUNK_MAGIC_VB);
  216. writeBounds(_writer, _vertices, _numVertices, stride);
  217. bx::write(_writer, _decl);
  218. bx::write(_writer, uint16_t(_numVertices) );
  219. bx::write(_writer, _vertices, _numVertices*stride);
  220. bx::write(_writer, BGFX_CHUNK_MAGIC_IB);
  221. bx::write(_writer, _numIndices);
  222. bx::write(_writer, _indices, _numIndices*2);
  223. bx::write(_writer, BGFX_CHUNK_MAGIC_PRI);
  224. uint16_t nameLen = uint16_t(_material.size() );
  225. bx::write(_writer, nameLen);
  226. bx::write(_writer, _material.c_str(), nameLen);
  227. bx::write(_writer, uint16_t(_primitives.size() ) );
  228. for (PrimitiveArray::const_iterator primIt = _primitives.begin(); primIt != _primitives.end(); ++primIt)
  229. {
  230. const Primitive& prim = *primIt;
  231. nameLen = uint16_t(prim.m_name.size() );
  232. bx::write(_writer, nameLen);
  233. bx::write(_writer, prim.m_name.c_str(), nameLen);
  234. bx::write(_writer, prim.m_startIndex);
  235. bx::write(_writer, prim.m_numIndices);
  236. bx::write(_writer, prim.m_startVertex);
  237. bx::write(_writer, prim.m_numVertices);
  238. writeBounds(_writer, &_vertices[prim.m_startVertex*stride], prim.m_numVertices, stride);
  239. }
  240. }
  241. void help(const char* _error = NULL)
  242. {
  243. if (NULL != _error)
  244. {
  245. fprintf(stderr, "Error:\n%s\n\n", _error);
  246. }
  247. fprintf(stderr
  248. , "geometryc, bgfx geometry compiler tool\n"
  249. "Copyright 2011-2012 Branimir Karadzic. All rights reserved.\n"
  250. "License: http://www.opensource.org/licenses/BSD-2-Clause\n\n"
  251. );
  252. fprintf(stderr
  253. , "Usage: geometryc -f <in> -o <out>\n"
  254. "\n"
  255. "Supported input file types:\n"
  256. " *.obj Wavefront\n"
  257. "\n"
  258. "Options:\n"
  259. " -f <file path> Input file path.\n"
  260. " -o <file path> Output file path.\n"
  261. " -s, --scale <num> Scale factor.\n"
  262. " --ccw Counter-clockwise winding order.\n"
  263. " --flipv Flip texture coordinate V.\n"
  264. " --obb <num> Number of steps for calculating oriented bounding box.\n"
  265. " Default value is 17. Less steps less precise OBB is.\n"
  266. " More steps slower calculation.\n"
  267. " --packnormal <num> Normal packing.\n"
  268. " 0 - unpacked 12 bytes (default).\n"
  269. " 1 - packed 4 bytes.\n"
  270. " --packuv <num> Texture coordinate packing.\n"
  271. " 0 - unpacked 8 bytes (default).\n"
  272. " 1 - packed 4 bytes.\n"
  273. " --tangent Calculate tangent vectors (packing mode is the same as normal).\n"
  274. "\n"
  275. "For additional information, see https://github.com/bkaradzic/bgfx\n"
  276. );
  277. }
  278. inline uint32_t rgbaToAbgr(uint8_t _r, uint8_t _g, uint8_t _b, uint8_t _a)
  279. {
  280. return (uint32_t(_r)<<0)
  281. | (uint32_t(_g)<<8)
  282. | (uint32_t(_b)<<16)
  283. | (uint32_t(_a)<<24)
  284. ;
  285. }
  286. int main(int _argc, const char* _argv[])
  287. {
  288. bx::CommandLine cmdLine(_argc, _argv);
  289. const char* filePath = cmdLine.findOption('f');
  290. if (NULL == filePath)
  291. {
  292. help("Input file name must be specified.");
  293. return EXIT_FAILURE;
  294. }
  295. const char* outFilePath = cmdLine.findOption('o');
  296. if (NULL == outFilePath)
  297. {
  298. help("Output file name must be specified.");
  299. return EXIT_FAILURE;
  300. }
  301. float scale = 1.0f;
  302. const char* scaleArg = cmdLine.findOption('s', "scale");
  303. if (NULL != scaleArg)
  304. {
  305. scale = (float)atof(scaleArg);
  306. }
  307. cmdLine.hasArg(s_obbSteps, '\0', "obb");
  308. s_obbSteps = bx::uint32_min(bx::uint32_max(s_obbSteps, 1), 90);
  309. uint32_t packNormal = 0;
  310. cmdLine.hasArg(packNormal, '\0', "packnormal");
  311. uint32_t packUv = 0;
  312. cmdLine.hasArg(packNormal, '\0', "packuv");
  313. bool ccw = cmdLine.hasArg("ccw");
  314. bool flipV = cmdLine.hasArg("flipv");
  315. bool hasTangent = cmdLine.hasArg("tangent");
  316. FILE* file = fopen(filePath, "r");
  317. if (NULL == file)
  318. {
  319. printf("Unable to open input file '%s'.", filePath);
  320. exit(EXIT_FAILURE);
  321. }
  322. int64_t parseElapsed = -bx::getHPCounter();
  323. int64_t triReorderElapsed = 0;
  324. uint32_t size = (uint32_t)fsize(file);
  325. char* data = new char[size+1];
  326. size = (uint32_t)fread(data, 1, size, file);
  327. data[size] = '\0';
  328. fclose(file);
  329. // https://en.wikipedia.org/wiki/Wavefront_.obj_file
  330. Vector3Array positions;
  331. Vector3Array normals;
  332. Vector3Array texcoords;
  333. Index3Map indexMap;
  334. TriangleArray triangles;
  335. GroupArray groups;
  336. uint32_t num = 0;
  337. Group group;
  338. group.m_startTriangle = 0;
  339. group.m_numTriangles = 0;
  340. char commandLine[2048];
  341. uint32_t len = sizeof(commandLine);
  342. int argc;
  343. char* argv[64];
  344. const char* next = data;
  345. do
  346. {
  347. next = tokenizeCommandLine(next, commandLine, len, argc, argv, countof(argv), '\n');
  348. if (0 < argc)
  349. {
  350. if (0 == strcmp(argv[0], "#") )
  351. {
  352. if (2 < argc
  353. && 0 == strcmp(argv[2], "polygons") )
  354. {
  355. }
  356. }
  357. else if (0 == strcmp(argv[0], "f") )
  358. {
  359. Triangle triangle;
  360. for (uint32_t edge = 0, numEdges = argc-1; edge < numEdges; ++edge)
  361. {
  362. Index3 index;
  363. index.m_texcoord = -1;
  364. index.m_normal = -1;
  365. index.m_vertexIndex = -1;
  366. char* vertex = argv[edge+1];
  367. char* texcoord = strchr(vertex, '/');
  368. if (NULL != texcoord)
  369. {
  370. *texcoord++ = '\0';
  371. char* normal = strchr(texcoord, '/');
  372. if (NULL != normal)
  373. {
  374. *normal++ = '\0';
  375. index.m_normal = atoi(normal)-1;
  376. }
  377. index.m_texcoord = atoi(texcoord)-1;
  378. }
  379. index.m_position = atoi(vertex)-1;
  380. uint64_t hash0 = index.m_position;
  381. uint64_t hash1 = uint64_t(index.m_texcoord)<<20;
  382. uint64_t hash2 = uint64_t(index.m_normal)<<40;
  383. uint64_t hash = hash0^hash1^hash2;
  384. std::pair<Index3Map::iterator, bool> result = indexMap.insert(std::make_pair(hash, index) );
  385. if (!result.second)
  386. {
  387. Index3& oldIndex = result.first->second;
  388. BX_UNUSED(oldIndex);
  389. BX_CHECK(oldIndex.m_position == index.m_position
  390. && oldIndex.m_texcoord == index.m_texcoord
  391. && oldIndex.m_normal == index.m_normal
  392. , "Hash collision!"
  393. );
  394. }
  395. switch (edge)
  396. {
  397. case 0:
  398. case 1:
  399. case 2:
  400. triangle.m_index[edge] = hash;
  401. if (2 == edge)
  402. {
  403. if (ccw)
  404. {
  405. std::swap(triangle.m_index[1], triangle.m_index[2]);
  406. }
  407. triangles.push_back(triangle);
  408. }
  409. break;
  410. default:
  411. if (ccw)
  412. {
  413. triangle.m_index[2] = triangle.m_index[1];
  414. triangle.m_index[1] = hash;
  415. }
  416. else
  417. {
  418. triangle.m_index[1] = triangle.m_index[2];
  419. triangle.m_index[2] = hash;
  420. }
  421. triangles.push_back(triangle);
  422. break;
  423. }
  424. }
  425. }
  426. else if (0 == strcmp(argv[0], "g") )
  427. {
  428. EXPECT(1 < argc);
  429. group.m_name = argv[1];
  430. }
  431. else if (*argv[0] == 'v')
  432. {
  433. group.m_numTriangles = (uint32_t)(triangles.size() ) - group.m_startTriangle;
  434. if (0 < group.m_numTriangles)
  435. {
  436. groups.push_back(group);
  437. group.m_startTriangle = (uint32_t)(triangles.size() );
  438. group.m_numTriangles = 0;
  439. }
  440. if (0 == strcmp(argv[0], "vn") )
  441. {
  442. Vector3 normal;
  443. normal.x = (float)atof(argv[1]);
  444. normal.y = (float)atof(argv[2]);
  445. normal.z = (float)atof(argv[3]);
  446. normals.push_back(normal);
  447. }
  448. else if (0 == strcmp(argv[0], "vp") )
  449. {
  450. static bool once = true;
  451. if (once)
  452. {
  453. once = false;
  454. printf("warning: 'parameter space vertices' are unsupported.\n");
  455. }
  456. }
  457. else if (0 == strcmp(argv[0], "vt") )
  458. {
  459. Vector3 texcoord;
  460. texcoord.x = (float)atof(argv[1]);
  461. texcoord.y = 0.0f;
  462. texcoord.z = 0.0f;
  463. switch (argc)
  464. {
  465. case 4:
  466. texcoord.z = (float)atof(argv[3]);
  467. // fallthrough
  468. case 3:
  469. texcoord.y = (float)atof(argv[2]);
  470. break;
  471. default:
  472. break;
  473. }
  474. texcoords.push_back(texcoord);
  475. }
  476. else
  477. {
  478. float px = (float)atof(argv[1]);
  479. float py = (float)atof(argv[2]);
  480. float pz = (float)atof(argv[3]);
  481. float pw = 1.0f;
  482. if (argc > 4)
  483. {
  484. pw = (float)atof(argv[4]);
  485. }
  486. float invW = scale/pw;
  487. px *= invW;
  488. py *= invW;
  489. pz *= invW;
  490. Vector3 pos;
  491. pos.x = px;
  492. pos.y = py;
  493. pos.z = pz;
  494. positions.push_back(pos);
  495. }
  496. }
  497. else if (0 == strcmp(argv[0], "usemtl") )
  498. {
  499. std::string material(argv[1]);
  500. if (material != group.m_material)
  501. {
  502. group.m_numTriangles = (uint32_t)(triangles.size() ) - group.m_startTriangle;
  503. if (0 < group.m_numTriangles)
  504. {
  505. groups.push_back(group);
  506. group.m_startTriangle = (uint32_t)(triangles.size() );
  507. group.m_numTriangles = 0;
  508. }
  509. }
  510. group.m_material = material;
  511. }
  512. // unsupported tags
  513. // else if (0 == strcmp(argv[0], "mtllib") )
  514. // {
  515. // }
  516. // else if (0 == strcmp(argv[0], "o") )
  517. // {
  518. // }
  519. // else if (0 == strcmp(argv[0], "s") )
  520. // {
  521. // }
  522. }
  523. ++num;
  524. }
  525. while ('\0' != *next);
  526. group.m_numTriangles = (uint32_t)(triangles.size() ) - group.m_startTriangle;
  527. if (0 < group.m_numTriangles)
  528. {
  529. groups.push_back(group);
  530. group.m_startTriangle = (uint32_t)(triangles.size() );
  531. group.m_numTriangles = 0;
  532. }
  533. delete [] data;
  534. int64_t now = bx::getHPCounter();
  535. parseElapsed += now;
  536. int64_t convertElapsed = -now;
  537. struct GroupSortByMaterial
  538. {
  539. bool operator()(const Group& _lhs, const Group& _rhs)
  540. {
  541. return _lhs.m_material < _rhs.m_material;
  542. }
  543. };
  544. std::sort(groups.begin(), groups.end(), GroupSortByMaterial() );
  545. bool hasColor = false;
  546. bool hasNormal;
  547. bool hasTexcoord;
  548. {
  549. Index3Map::const_iterator it = indexMap.begin();
  550. hasNormal = -1 != it->second.m_normal;
  551. hasTexcoord = -1 != it->second.m_texcoord;
  552. if (!hasTexcoord
  553. && texcoords.size() == positions.size() )
  554. {
  555. hasTexcoord = true;
  556. for (Index3Map::iterator it = indexMap.begin(), itEnd = indexMap.end(); it != itEnd; ++it)
  557. {
  558. it->second.m_texcoord = it->second.m_position;
  559. }
  560. }
  561. if (!hasNormal
  562. && normals.size() == positions.size() )
  563. {
  564. hasNormal = true;
  565. for (Index3Map::iterator it = indexMap.begin(), itEnd = indexMap.end(); it != itEnd; ++it)
  566. {
  567. it->second.m_normal = it->second.m_position;
  568. }
  569. }
  570. }
  571. bgfx::VertexDecl decl;
  572. decl.begin();
  573. decl.add(bgfx::Attrib::Position, 3, bgfx::AttribType::Float);
  574. if (hasColor)
  575. {
  576. decl.add(bgfx::Attrib::Color0, 4, bgfx::AttribType::Uint8, true);
  577. }
  578. if (hasTexcoord)
  579. {
  580. switch (packUv)
  581. {
  582. default:
  583. case 0:
  584. decl.add(bgfx::Attrib::TexCoord0, 2, bgfx::AttribType::Float);
  585. break;
  586. case 1:
  587. decl.add(bgfx::Attrib::TexCoord0, 2, bgfx::AttribType::Half);
  588. break;
  589. }
  590. }
  591. if (hasNormal)
  592. {
  593. hasTangent &= hasTexcoord;
  594. switch (packNormal)
  595. {
  596. default:
  597. case 0:
  598. decl.add(bgfx::Attrib::Normal, 3, bgfx::AttribType::Float);
  599. if (hasTangent)
  600. {
  601. decl.add(bgfx::Attrib::Tangent, 3, bgfx::AttribType::Float);
  602. }
  603. break;
  604. case 1:
  605. decl.add(bgfx::Attrib::Normal, 4, bgfx::AttribType::Uint8, true, true);
  606. if (hasTangent)
  607. {
  608. decl.add(bgfx::Attrib::Tangent, 4, bgfx::AttribType::Uint8, true, true);
  609. }
  610. break;
  611. }
  612. }
  613. decl.end();
  614. uint32_t stride = decl.getStride();
  615. uint8_t* vertexData = new uint8_t[triangles.size() * 3 * stride];
  616. uint16_t* indexData = new uint16_t[triangles.size() * 3];
  617. int32_t numVertices = 0;
  618. int32_t numIndices = 0;
  619. int32_t numPrimitives = 0;
  620. uint8_t* vertices = vertexData;
  621. uint16_t* indices = indexData;
  622. std::string material = groups.begin()->m_material;
  623. PrimitiveArray primitives;
  624. bx::CrtFileWriter writer;
  625. if (0 != writer.open(outFilePath) )
  626. {
  627. printf("Unable to open output file '%s'.", outFilePath);
  628. exit(EXIT_FAILURE);
  629. }
  630. Primitive prim;
  631. prim.m_startVertex = 0;
  632. prim.m_startIndex = 0;
  633. uint32_t positionOffset = decl.getOffset(bgfx::Attrib::Position);
  634. uint32_t color0Offset = decl.getOffset(bgfx::Attrib::Color0);
  635. uint32_t ii = 0;
  636. for (GroupArray::const_iterator groupIt = groups.begin(); groupIt != groups.end(); ++groupIt, ++ii)
  637. {
  638. for (uint32_t tri = groupIt->m_startTriangle, end = tri + groupIt->m_numTriangles; tri < end; ++tri)
  639. {
  640. if (material != groupIt->m_material
  641. || 65533 < numVertices)
  642. {
  643. prim.m_numVertices = numVertices - prim.m_startVertex;
  644. prim.m_numIndices = numIndices - prim.m_startIndex;
  645. if (0 < prim.m_numVertices)
  646. {
  647. primitives.push_back(prim);
  648. }
  649. triReorderElapsed -= bx::getHPCounter();
  650. for (PrimitiveArray::const_iterator primIt = primitives.begin(); primIt != primitives.end(); ++primIt)
  651. {
  652. const Primitive& prim = *primIt;
  653. triangleReorder(indexData + prim.m_startIndex, prim.m_numIndices, numVertices, 32);
  654. }
  655. triReorderElapsed += bx::getHPCounter();
  656. if (hasTangent)
  657. {
  658. calcTangents(vertexData, numVertices, decl, indexData, numIndices);
  659. }
  660. write(&writer, vertexData, numVertices, decl, indexData, numIndices, material, primitives);
  661. primitives.clear();
  662. for (Index3Map::iterator indexIt = indexMap.begin(); indexIt != indexMap.end(); ++indexIt)
  663. {
  664. indexIt->second.m_vertexIndex = -1;
  665. }
  666. vertices = vertexData;
  667. indices = indexData;
  668. numVertices = 0;
  669. numIndices = 0;
  670. prim.m_startVertex = 0;
  671. prim.m_startIndex = 0;
  672. ++numPrimitives;
  673. material = groupIt->m_material;
  674. }
  675. Triangle& triangle = triangles[tri];
  676. for (uint32_t edge = 0; edge < 3; ++edge)
  677. {
  678. uint64_t hash = triangle.m_index[edge];
  679. Index3& index = indexMap[hash];
  680. if (index.m_vertexIndex == -1)
  681. {
  682. index.m_vertexIndex = numVertices++;
  683. float* position = (float*)(vertices + positionOffset);
  684. memcpy(position, &positions[index.m_position], 3*sizeof(float) );
  685. if (hasColor)
  686. {
  687. uint32_t* color0 = (uint32_t*)(vertices + color0Offset);
  688. *color0 = rgbaToAbgr(numVertices%255, numIndices%255, 0, 0xff);
  689. }
  690. if (hasTexcoord)
  691. {
  692. float uv[2];
  693. memcpy(uv, &texcoords[index.m_texcoord], 2*sizeof(float) );
  694. if (flipV)
  695. {
  696. uv[1] = -uv[1];
  697. }
  698. bgfx::vertexPack(uv, true, bgfx::Attrib::TexCoord0, decl, vertices);
  699. }
  700. if (hasNormal)
  701. {
  702. float normal[4];
  703. vec3Norm(normal, (float*)&normals[index.m_normal]);
  704. bgfx::vertexPack(normal, true, bgfx::Attrib::Normal, decl, vertices);
  705. }
  706. vertices += stride;
  707. }
  708. *indices++ = (uint16_t)index.m_vertexIndex;
  709. ++numIndices;
  710. }
  711. }
  712. if (0 < numVertices)
  713. {
  714. prim.m_numVertices = numVertices - prim.m_startVertex;
  715. prim.m_numIndices = numIndices - prim.m_startIndex;
  716. prim.m_name = groupIt->m_name;
  717. primitives.push_back(prim);
  718. prim.m_startVertex = numVertices;
  719. prim.m_startIndex = numIndices;
  720. }
  721. BX_TRACE("%3d: s %5d, n %5d, %s\n"
  722. , ii
  723. , groupIt->m_startTriangle
  724. , groupIt->m_numTriangles
  725. , groupIt->m_material.c_str()
  726. );
  727. }
  728. if (0 < primitives.size() )
  729. {
  730. triReorderElapsed -= bx::getHPCounter();
  731. for (PrimitiveArray::const_iterator primIt = primitives.begin(); primIt != primitives.end(); ++primIt)
  732. {
  733. const Primitive& prim = *primIt;
  734. triangleReorder(indexData + prim.m_startIndex, prim.m_numIndices, numVertices, 32);
  735. }
  736. triReorderElapsed += bx::getHPCounter();
  737. if (hasTangent)
  738. {
  739. calcTangents(vertexData, numVertices, decl, indexData, numIndices);
  740. }
  741. write(&writer, vertexData, numVertices, decl, indexData, numIndices, material, primitives);
  742. }
  743. printf("size: %d\n", uint32_t(writer.seek() ) );
  744. writer.close();
  745. delete [] indexData;
  746. delete [] vertexData;
  747. now = bx::getHPCounter();
  748. convertElapsed += now;
  749. printf("parse %f [s]\ntri reorder %f [s]\nconvert %f [s]\n# %d, g %d, p %d, v %d, i %d\n"
  750. , double(parseElapsed)/bx::getHPFrequency()
  751. , double(triReorderElapsed)/bx::getHPFrequency()
  752. , double(convertElapsed)/bx::getHPFrequency()
  753. , num
  754. , uint32_t(groups.size() )
  755. , numPrimitives
  756. , numVertices
  757. , numIndices
  758. );
  759. return EXIT_SUCCESS;
  760. }