2
0

bgfx_utils.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932
  1. /*
  2. * Copyright 2011-2016 Branimir Karadzic. All rights reserved.
  3. * License: https://github.com/bkaradzic/bgfx#license-bsd-2-clause
  4. */
  5. #include <string.h> // strlen
  6. #include "common.h"
  7. #include <tinystl/allocator.h>
  8. #include <tinystl/vector.h>
  9. #include <tinystl/string.h>
  10. namespace stl = tinystl;
  11. #include <bgfx/bgfx.h>
  12. #include <bx/commandline.h>
  13. #include <bx/endian.h>
  14. #include <bx/fpumath.h>
  15. #include <bx/readerwriter.h>
  16. #include <bx/string.h>
  17. #include "entry/entry.h"
  18. #include <ib-compress/indexbufferdecompression.h>
  19. BX_PRAGMA_DIAGNOSTIC_PUSH()
  20. BX_PRAGMA_DIAGNOSTIC_IGNORED_CLANG_GCC("-Wtype-limits")
  21. #define TINYEXR_IMPLEMENTATION
  22. #include <tinyexr/tinyexr.h>
  23. BX_PRAGMA_DIAGNOSTIC_POP()
  24. #define LODEPNG_NO_COMPILE_ENCODER
  25. #define LODEPNG_NO_COMPILE_DISK
  26. #define LODEPNG_NO_COMPILE_ANCILLARY_CHUNKS
  27. #define LODEPNG_NO_COMPILE_ERROR_TEXT
  28. #define LODEPNG_NO_COMPILE_ALLOCATORS
  29. #define LODEPNG_NO_COMPILE_CPP
  30. #include <lodepng/lodepng.h>
  31. #include "bgfx_utils.h"
  32. void* load(bx::FileReaderI* _reader, bx::AllocatorI* _allocator, const char* _filePath, uint32_t* _size)
  33. {
  34. if (bx::open(_reader, _filePath) )
  35. {
  36. uint32_t size = (uint32_t)bx::getSize(_reader);
  37. void* data = BX_ALLOC(_allocator, size);
  38. bx::read(_reader, data, size);
  39. bx::close(_reader);
  40. if (NULL != _size)
  41. {
  42. *_size = size;
  43. }
  44. return data;
  45. }
  46. else
  47. {
  48. DBG("Failed to open: %s.", _filePath);
  49. }
  50. if (NULL != _size)
  51. {
  52. *_size = 0;
  53. }
  54. return NULL;
  55. }
  56. void* load(const char* _filePath, uint32_t* _size)
  57. {
  58. return load(entry::getFileReader(), entry::getAllocator(), _filePath, _size);
  59. }
  60. void unload(void* _ptr)
  61. {
  62. BX_FREE(entry::getAllocator(), _ptr);
  63. }
  64. static const bgfx::Memory* loadMem(bx::FileReaderI* _reader, const char* _filePath)
  65. {
  66. if (bx::open(_reader, _filePath) )
  67. {
  68. uint32_t size = (uint32_t)bx::getSize(_reader);
  69. const bgfx::Memory* mem = bgfx::alloc(size+1);
  70. bx::read(_reader, mem->data, size);
  71. bx::close(_reader);
  72. mem->data[mem->size-1] = '\0';
  73. return mem;
  74. }
  75. DBG("Failed to load %s.", _filePath);
  76. return NULL;
  77. }
  78. static void* loadMem(bx::FileReaderI* _reader, bx::AllocatorI* _allocator, const char* _filePath, uint32_t* _size)
  79. {
  80. if (bx::open(_reader, _filePath) )
  81. {
  82. uint32_t size = (uint32_t)bx::getSize(_reader);
  83. void* data = BX_ALLOC(_allocator, size);
  84. bx::read(_reader, data, size);
  85. bx::close(_reader);
  86. if (NULL != _size)
  87. {
  88. *_size = size;
  89. }
  90. return data;
  91. }
  92. DBG("Failed to load %s.", _filePath);
  93. return NULL;
  94. }
  95. static bgfx::ShaderHandle loadShader(bx::FileReaderI* _reader, const char* _name)
  96. {
  97. char filePath[512];
  98. const char* shaderPath = "???";
  99. switch (bgfx::getRendererType() )
  100. {
  101. case bgfx::RendererType::Noop:
  102. case bgfx::RendererType::Direct3D9: shaderPath = "shaders/dx9/"; break;
  103. case bgfx::RendererType::Direct3D11:
  104. case bgfx::RendererType::Direct3D12: shaderPath = "shaders/dx11/"; break;
  105. case bgfx::RendererType::Gnm: shaderPath = "shaders/pssl/"; break;
  106. case bgfx::RendererType::Metal: shaderPath = "shaders/metal/"; break;
  107. case bgfx::RendererType::OpenGL: shaderPath = "shaders/glsl/"; break;
  108. case bgfx::RendererType::OpenGLES: shaderPath = "shaders/essl/"; break;
  109. case bgfx::RendererType::Vulkan: shaderPath = "shaders/spirv/"; break;
  110. case bgfx::RendererType::Count:
  111. BX_CHECK(false, "You should not be here!");
  112. break;
  113. }
  114. strcpy(filePath, shaderPath);
  115. strcat(filePath, _name);
  116. strcat(filePath, ".bin");
  117. return bgfx::createShader(loadMem(_reader, filePath) );
  118. }
  119. bgfx::ShaderHandle loadShader(const char* _name)
  120. {
  121. return loadShader(entry::getFileReader(), _name);
  122. }
  123. bgfx::ProgramHandle loadProgram(bx::FileReaderI* _reader, const char* _vsName, const char* _fsName)
  124. {
  125. bgfx::ShaderHandle vsh = loadShader(_reader, _vsName);
  126. bgfx::ShaderHandle fsh = BGFX_INVALID_HANDLE;
  127. if (NULL != _fsName)
  128. {
  129. fsh = loadShader(_reader, _fsName);
  130. }
  131. return bgfx::createProgram(vsh, fsh, true /* destroy shaders when program is destroyed */);
  132. }
  133. bgfx::ProgramHandle loadProgram(const char* _vsName, const char* _fsName)
  134. {
  135. return loadProgram(entry::getFileReader(), _vsName, _fsName);
  136. }
  137. typedef unsigned char stbi_uc;
  138. extern "C" stbi_uc* stbi_load_from_memory(stbi_uc const* _buffer, int _len, int* _x, int* _y, int* _comp, int _req_comp);
  139. extern "C" void stbi_image_free(void* _ptr);
  140. extern void lodepng_free(void* _ptr);
  141. static void exrRelease(void* _ptr)
  142. {
  143. BX_FREE(entry::getAllocator(), _ptr);
  144. }
  145. bgfx::TextureHandle loadTexture(bx::FileReaderI* _reader, const char* _filePath, uint32_t _flags, uint8_t _skip, bgfx::TextureInfo* _info)
  146. {
  147. if (NULL != bx::stristr(_filePath, ".dds")
  148. || NULL != bx::stristr(_filePath, ".pvr")
  149. || NULL != bx::stristr(_filePath, ".ktx") )
  150. {
  151. const bgfx::Memory* mem = loadMem(_reader, _filePath);
  152. if (NULL != mem)
  153. {
  154. return bgfx::createTexture(mem, _flags, _skip, _info);
  155. }
  156. bgfx::TextureHandle handle = BGFX_INVALID_HANDLE;
  157. DBG("Failed to load %s.", _filePath);
  158. return handle;
  159. }
  160. bgfx::TextureHandle handle = BGFX_INVALID_HANDLE;
  161. bx::AllocatorI* allocator = entry::getAllocator();
  162. uint32_t size = 0;
  163. void* data = loadMem(_reader, allocator, _filePath, &size);
  164. if (NULL != data)
  165. {
  166. bgfx::TextureFormat::Enum format = bgfx::TextureFormat::RGBA8;
  167. uint32_t bpp = 32;
  168. uint32_t width = 0;
  169. uint32_t height = 0;
  170. typedef void (*ReleaseFn)(void* _ptr);
  171. ReleaseFn release = stbi_image_free;
  172. uint8_t* out = NULL;
  173. static uint8_t pngMagic[] = { 0x89, 0x50, 0x4E, 0x47, 0x0d, 0x0a };
  174. if (0 == memcmp(data, pngMagic, sizeof(pngMagic) ) )
  175. {
  176. release = lodepng_free;
  177. unsigned error;
  178. LodePNGState state;
  179. lodepng_state_init(&state);
  180. state.decoder.color_convert = 0;
  181. error = lodepng_decode(&out, &width, &height, &state, (uint8_t*)data, size);
  182. if (0 == error)
  183. {
  184. switch (state.info_raw.bitdepth)
  185. {
  186. case 8:
  187. switch (state.info_raw.colortype)
  188. {
  189. case LCT_GREY:
  190. format = bgfx::TextureFormat::R8;
  191. bpp = 8;
  192. break;
  193. case LCT_GREY_ALPHA:
  194. format = bgfx::TextureFormat::RG8;
  195. bpp = 16;
  196. break;
  197. case LCT_RGB:
  198. format = bgfx::TextureFormat::RGB8;
  199. bpp = 24;
  200. break;
  201. case LCT_RGBA:
  202. format = bgfx::TextureFormat::RGBA8;
  203. bpp = 32;
  204. break;
  205. case LCT_PALETTE:
  206. format = bgfx::TextureFormat::R8;
  207. bpp = 8;
  208. break;
  209. }
  210. break;
  211. case 16:
  212. switch (state.info_raw.colortype)
  213. {
  214. case LCT_GREY:
  215. for (uint32_t ii = 0, num = width*height; ii < num; ++ii)
  216. {
  217. uint16_t* rgba = (uint16_t*)out + ii*4;
  218. rgba[0] = bx::toHostEndian(rgba[0], false);
  219. }
  220. format = bgfx::TextureFormat::R16;
  221. bpp = 16;
  222. break;
  223. case LCT_GREY_ALPHA:
  224. for (uint32_t ii = 0, num = width*height; ii < num; ++ii)
  225. {
  226. uint16_t* rgba = (uint16_t*)out + ii*4;
  227. rgba[0] = bx::toHostEndian(rgba[0], false);
  228. rgba[1] = bx::toHostEndian(rgba[1], false);
  229. }
  230. format = bgfx::TextureFormat::R16;
  231. bpp = 16;
  232. break;
  233. case LCT_RGBA:
  234. for (uint32_t ii = 0, num = width*height; ii < num; ++ii)
  235. {
  236. uint16_t* rgba = (uint16_t*)out + ii*4;
  237. rgba[0] = bx::toHostEndian(rgba[0], false);
  238. rgba[1] = bx::toHostEndian(rgba[1], false);
  239. rgba[2] = bx::toHostEndian(rgba[2], false);
  240. rgba[3] = bx::toHostEndian(rgba[3], false);
  241. }
  242. format = bgfx::TextureFormat::RGBA16;
  243. bpp = 64;
  244. break;
  245. case LCT_RGB:
  246. case LCT_PALETTE:
  247. break;
  248. }
  249. break;
  250. default:
  251. break;
  252. }
  253. }
  254. lodepng_state_cleanup(&state);
  255. }
  256. else
  257. {
  258. EXRVersion exrVersion;
  259. int result = ParseEXRVersionFromMemory(&exrVersion, (uint8_t*)data, size);
  260. if (TINYEXR_SUCCESS == result)
  261. {
  262. const char* err = NULL;
  263. EXRHeader exrHeader;
  264. result = ParseEXRHeaderFromMemory(&exrHeader, &exrVersion, (uint8_t*)data, size, &err);
  265. if (TINYEXR_SUCCESS == result)
  266. {
  267. EXRImage exrImage;
  268. InitEXRImage(&exrImage);
  269. result = LoadEXRImageFromMemory(&exrImage, &exrHeader, (uint8_t*)data, size, &err);
  270. if (TINYEXR_SUCCESS == result)
  271. {
  272. uint8_t idxR = UINT8_MAX;
  273. uint8_t idxG = UINT8_MAX;
  274. uint8_t idxB = UINT8_MAX;
  275. uint8_t idxA = UINT8_MAX;
  276. for (uint8_t ii = 0, num = uint8_t(exrHeader.num_channels); ii < num; ++ii)
  277. {
  278. const EXRChannelInfo& channel = exrHeader.channels[ii];
  279. if (UINT8_MAX == idxR
  280. && 0 == strcmp(channel.name, "R") )
  281. {
  282. idxR = ii;
  283. }
  284. else if (UINT8_MAX == idxG
  285. && 0 == strcmp(channel.name, "G") )
  286. {
  287. idxG = ii;
  288. }
  289. else if (UINT8_MAX == idxB
  290. && 0 == strcmp(channel.name, "B") )
  291. {
  292. idxB = ii;
  293. }
  294. else if (UINT8_MAX == idxA
  295. && 0 == strcmp(channel.name, "A") )
  296. {
  297. idxA = ii;
  298. }
  299. }
  300. if (UINT8_MAX != idxR)
  301. {
  302. const bool asFloat = exrHeader.pixel_types[idxR] == TINYEXR_PIXELTYPE_FLOAT;
  303. uint32_t srcBpp = 32;
  304. uint32_t dstBpp = asFloat ? 32 : 16;
  305. format = asFloat ? bgfx::TextureFormat::R32F : bgfx::TextureFormat::R16F;
  306. uint32_t stepR = 1;
  307. uint32_t stepG = 0;
  308. uint32_t stepB = 0;
  309. uint32_t stepA = 0;
  310. if (UINT8_MAX != idxG)
  311. {
  312. srcBpp += 32;
  313. dstBpp = asFloat ? 64 : 32;
  314. format = asFloat ? bgfx::TextureFormat::RG32F : bgfx::TextureFormat::RG16F;
  315. stepG = 1;
  316. }
  317. if (UINT8_MAX != idxB)
  318. {
  319. srcBpp += 32;
  320. dstBpp = asFloat ? 128 : 64;
  321. format = asFloat ? bgfx::TextureFormat::RGBA32F : bgfx::TextureFormat::RGBA16F;
  322. stepB = 1;
  323. }
  324. if (UINT8_MAX != idxA)
  325. {
  326. srcBpp += 32;
  327. dstBpp = asFloat ? 128 : 64;
  328. format = asFloat ? bgfx::TextureFormat::RGBA32F : bgfx::TextureFormat::RGBA16F;
  329. stepA = 1;
  330. }
  331. release = exrRelease;
  332. out = (uint8_t*)BX_ALLOC(allocator, exrImage.width * exrImage.height * dstBpp/8);
  333. const float zero = 0.0f;
  334. const float* srcR = UINT8_MAX == idxR ? &zero : (const float*)(exrImage.images)[idxR];
  335. const float* srcG = UINT8_MAX == idxG ? &zero : (const float*)(exrImage.images)[idxG];
  336. const float* srcB = UINT8_MAX == idxB ? &zero : (const float*)(exrImage.images)[idxB];
  337. const float* srcA = UINT8_MAX == idxA ? &zero : (const float*)(exrImage.images)[idxA];
  338. const uint32_t bytesPerPixel = dstBpp/8;
  339. for (uint32_t ii = 0, num = exrImage.width * exrImage.height; ii < num; ++ii)
  340. {
  341. float rgba[4] =
  342. {
  343. *srcR,
  344. *srcG,
  345. *srcB,
  346. *srcA,
  347. };
  348. memcpy(&out[ii * bytesPerPixel], rgba, bytesPerPixel);
  349. srcR += stepR;
  350. srcG += stepG;
  351. srcB += stepB;
  352. srcA += stepA;
  353. }
  354. }
  355. FreeEXRImage(&exrImage);
  356. }
  357. FreeEXRHeader(&exrHeader);
  358. }
  359. }
  360. else
  361. {
  362. int comp = 0;
  363. out = stbi_load_from_memory( (uint8_t*)data, size, (int*)&width, (int*)&height, &comp, 4);
  364. }
  365. }
  366. BX_FREE(allocator, data);
  367. if (NULL != out)
  368. {
  369. handle = bgfx::createTexture2D(
  370. uint16_t(width)
  371. , uint16_t(height)
  372. , false
  373. , 1
  374. , format
  375. , _flags
  376. , bgfx::copy(out, width*height*bpp/8)
  377. );
  378. release(out);
  379. if (NULL != _info)
  380. {
  381. bgfx::calcTextureSize(
  382. *_info
  383. , uint16_t(width)
  384. , uint16_t(height)
  385. , 0
  386. , false
  387. , false
  388. , 1
  389. , format
  390. );
  391. }
  392. }
  393. }
  394. else
  395. {
  396. DBG("Failed to load %s.", _filePath);
  397. }
  398. return handle;
  399. }
  400. bgfx::TextureHandle loadTexture(const char* _name, uint32_t _flags, uint8_t _skip, bgfx::TextureInfo* _info)
  401. {
  402. return loadTexture(entry::getFileReader(), _name, _flags, _skip, _info);
  403. }
  404. void calcTangents(void* _vertices, uint16_t _numVertices, bgfx::VertexDecl _decl, const uint16_t* _indices, uint32_t _numIndices)
  405. {
  406. struct PosTexcoord
  407. {
  408. float m_x;
  409. float m_y;
  410. float m_z;
  411. float m_pad0;
  412. float m_u;
  413. float m_v;
  414. float m_pad1;
  415. float m_pad2;
  416. };
  417. float* tangents = new float[6*_numVertices];
  418. memset(tangents, 0, 6*_numVertices*sizeof(float) );
  419. PosTexcoord v0;
  420. PosTexcoord v1;
  421. PosTexcoord v2;
  422. for (uint32_t ii = 0, num = _numIndices/3; ii < num; ++ii)
  423. {
  424. const uint16_t* indices = &_indices[ii*3];
  425. uint32_t i0 = indices[0];
  426. uint32_t i1 = indices[1];
  427. uint32_t i2 = indices[2];
  428. bgfx::vertexUnpack(&v0.m_x, bgfx::Attrib::Position, _decl, _vertices, i0);
  429. bgfx::vertexUnpack(&v0.m_u, bgfx::Attrib::TexCoord0, _decl, _vertices, i0);
  430. bgfx::vertexUnpack(&v1.m_x, bgfx::Attrib::Position, _decl, _vertices, i1);
  431. bgfx::vertexUnpack(&v1.m_u, bgfx::Attrib::TexCoord0, _decl, _vertices, i1);
  432. bgfx::vertexUnpack(&v2.m_x, bgfx::Attrib::Position, _decl, _vertices, i2);
  433. bgfx::vertexUnpack(&v2.m_u, bgfx::Attrib::TexCoord0, _decl, _vertices, i2);
  434. const float bax = v1.m_x - v0.m_x;
  435. const float bay = v1.m_y - v0.m_y;
  436. const float baz = v1.m_z - v0.m_z;
  437. const float bau = v1.m_u - v0.m_u;
  438. const float bav = v1.m_v - v0.m_v;
  439. const float cax = v2.m_x - v0.m_x;
  440. const float cay = v2.m_y - v0.m_y;
  441. const float caz = v2.m_z - v0.m_z;
  442. const float cau = v2.m_u - v0.m_u;
  443. const float cav = v2.m_v - v0.m_v;
  444. const float det = (bau * cav - bav * cau);
  445. const float invDet = 1.0f / det;
  446. const float tx = (bax * cav - cax * bav) * invDet;
  447. const float ty = (bay * cav - cay * bav) * invDet;
  448. const float tz = (baz * cav - caz * bav) * invDet;
  449. const float bx = (cax * bau - bax * cau) * invDet;
  450. const float by = (cay * bau - bay * cau) * invDet;
  451. const float bz = (caz * bau - baz * cau) * invDet;
  452. for (uint32_t jj = 0; jj < 3; ++jj)
  453. {
  454. float* tanu = &tangents[indices[jj]*6];
  455. float* tanv = &tanu[3];
  456. tanu[0] += tx;
  457. tanu[1] += ty;
  458. tanu[2] += tz;
  459. tanv[0] += bx;
  460. tanv[1] += by;
  461. tanv[2] += bz;
  462. }
  463. }
  464. for (uint32_t ii = 0; ii < _numVertices; ++ii)
  465. {
  466. const float* tanu = &tangents[ii*6];
  467. const float* tanv = &tangents[ii*6 + 3];
  468. float normal[4];
  469. bgfx::vertexUnpack(normal, bgfx::Attrib::Normal, _decl, _vertices, ii);
  470. float ndt = bx::vec3Dot(normal, tanu);
  471. float nxt[3];
  472. bx::vec3Cross(nxt, normal, tanu);
  473. float tmp[3];
  474. tmp[0] = tanu[0] - normal[0] * ndt;
  475. tmp[1] = tanu[1] - normal[1] * ndt;
  476. tmp[2] = tanu[2] - normal[2] * ndt;
  477. float tangent[4];
  478. bx::vec3Norm(tangent, tmp);
  479. tangent[3] = bx::vec3Dot(nxt, tanv) < 0.0f ? -1.0f : 1.0f;
  480. bgfx::vertexPack(tangent, true, bgfx::Attrib::Tangent, _decl, _vertices, ii);
  481. }
  482. delete [] tangents;
  483. }
  484. struct Aabb
  485. {
  486. float m_min[3];
  487. float m_max[3];
  488. };
  489. struct Obb
  490. {
  491. float m_mtx[16];
  492. };
  493. struct Sphere
  494. {
  495. float m_center[3];
  496. float m_radius;
  497. };
  498. struct Primitive
  499. {
  500. uint32_t m_startIndex;
  501. uint32_t m_numIndices;
  502. uint32_t m_startVertex;
  503. uint32_t m_numVertices;
  504. Sphere m_sphere;
  505. Aabb m_aabb;
  506. Obb m_obb;
  507. };
  508. typedef stl::vector<Primitive> PrimitiveArray;
  509. struct Group
  510. {
  511. Group()
  512. {
  513. reset();
  514. }
  515. void reset()
  516. {
  517. m_vbh.idx = bgfx::invalidHandle;
  518. m_ibh.idx = bgfx::invalidHandle;
  519. m_prims.clear();
  520. }
  521. bgfx::VertexBufferHandle m_vbh;
  522. bgfx::IndexBufferHandle m_ibh;
  523. Sphere m_sphere;
  524. Aabb m_aabb;
  525. Obb m_obb;
  526. PrimitiveArray m_prims;
  527. };
  528. namespace bgfx
  529. {
  530. int32_t read(bx::ReaderI* _reader, bgfx::VertexDecl& _decl, bx::Error* _err = NULL);
  531. }
  532. struct Mesh
  533. {
  534. void load(bx::ReaderSeekerI* _reader)
  535. {
  536. #define BGFX_CHUNK_MAGIC_VB BX_MAKEFOURCC('V', 'B', ' ', 0x1)
  537. #define BGFX_CHUNK_MAGIC_IB BX_MAKEFOURCC('I', 'B', ' ', 0x0)
  538. #define BGFX_CHUNK_MAGIC_IBC BX_MAKEFOURCC('I', 'B', 'C', 0x0)
  539. #define BGFX_CHUNK_MAGIC_PRI BX_MAKEFOURCC('P', 'R', 'I', 0x0)
  540. using namespace bx;
  541. using namespace bgfx;
  542. Group group;
  543. bx::AllocatorI* allocator = entry::getAllocator();
  544. uint32_t chunk;
  545. bx::Error err;
  546. while (4 == bx::read(_reader, chunk, &err)
  547. && err.isOk() )
  548. {
  549. switch (chunk)
  550. {
  551. case BGFX_CHUNK_MAGIC_VB:
  552. {
  553. read(_reader, group.m_sphere);
  554. read(_reader, group.m_aabb);
  555. read(_reader, group.m_obb);
  556. read(_reader, m_decl);
  557. uint16_t stride = m_decl.getStride();
  558. uint16_t numVertices;
  559. read(_reader, numVertices);
  560. const bgfx::Memory* mem = bgfx::alloc(numVertices*stride);
  561. read(_reader, mem->data, mem->size);
  562. group.m_vbh = bgfx::createVertexBuffer(mem, m_decl);
  563. }
  564. break;
  565. case BGFX_CHUNK_MAGIC_IB:
  566. {
  567. uint32_t numIndices;
  568. read(_reader, numIndices);
  569. const bgfx::Memory* mem = bgfx::alloc(numIndices*2);
  570. read(_reader, mem->data, mem->size);
  571. group.m_ibh = bgfx::createIndexBuffer(mem);
  572. }
  573. break;
  574. case BGFX_CHUNK_MAGIC_IBC:
  575. {
  576. uint32_t numIndices;
  577. bx::read(_reader, numIndices);
  578. const bgfx::Memory* mem = bgfx::alloc(numIndices*2);
  579. uint32_t compressedSize;
  580. bx::read(_reader, compressedSize);
  581. void* compressedIndices = BX_ALLOC(allocator, compressedSize);
  582. bx::read(_reader, compressedIndices, compressedSize);
  583. ReadBitstream rbs( (const uint8_t*)compressedIndices, compressedSize);
  584. DecompressIndexBuffer( (uint16_t*)mem->data, numIndices / 3, rbs);
  585. BX_FREE(allocator, compressedIndices);
  586. group.m_ibh = bgfx::createIndexBuffer(mem);
  587. }
  588. break;
  589. case BGFX_CHUNK_MAGIC_PRI:
  590. {
  591. uint16_t len;
  592. read(_reader, len);
  593. stl::string material;
  594. material.resize(len);
  595. read(_reader, const_cast<char*>(material.c_str() ), len);
  596. uint16_t num;
  597. read(_reader, num);
  598. for (uint32_t ii = 0; ii < num; ++ii)
  599. {
  600. read(_reader, len);
  601. stl::string name;
  602. name.resize(len);
  603. read(_reader, const_cast<char*>(name.c_str() ), len);
  604. Primitive prim;
  605. read(_reader, prim.m_startIndex);
  606. read(_reader, prim.m_numIndices);
  607. read(_reader, prim.m_startVertex);
  608. read(_reader, prim.m_numVertices);
  609. read(_reader, prim.m_sphere);
  610. read(_reader, prim.m_aabb);
  611. read(_reader, prim.m_obb);
  612. group.m_prims.push_back(prim);
  613. }
  614. m_groups.push_back(group);
  615. group.reset();
  616. }
  617. break;
  618. default:
  619. DBG("%08x at %d", chunk, bx::skip(_reader, 0) );
  620. break;
  621. }
  622. }
  623. }
  624. void unload()
  625. {
  626. for (GroupArray::const_iterator it = m_groups.begin(), itEnd = m_groups.end(); it != itEnd; ++it)
  627. {
  628. const Group& group = *it;
  629. bgfx::destroyVertexBuffer(group.m_vbh);
  630. if (bgfx::isValid(group.m_ibh) )
  631. {
  632. bgfx::destroyIndexBuffer(group.m_ibh);
  633. }
  634. }
  635. m_groups.clear();
  636. }
  637. void submit(uint8_t _id, bgfx::ProgramHandle _program, const float* _mtx, uint64_t _state) const
  638. {
  639. if (BGFX_STATE_MASK == _state)
  640. {
  641. _state = 0
  642. | BGFX_STATE_RGB_WRITE
  643. | BGFX_STATE_ALPHA_WRITE
  644. | BGFX_STATE_DEPTH_WRITE
  645. | BGFX_STATE_DEPTH_TEST_LESS
  646. | BGFX_STATE_CULL_CCW
  647. | BGFX_STATE_MSAA
  648. ;
  649. }
  650. bgfx::setTransform(_mtx);
  651. bgfx::setState(_state);
  652. for (GroupArray::const_iterator it = m_groups.begin(), itEnd = m_groups.end(); it != itEnd; ++it)
  653. {
  654. const Group& group = *it;
  655. bgfx::setIndexBuffer(group.m_ibh);
  656. bgfx::setVertexBuffer(group.m_vbh);
  657. bgfx::submit(_id, _program, 0, it != itEnd-1);
  658. }
  659. }
  660. void submit(const MeshState*const* _state, uint8_t _numPasses, const float* _mtx, uint16_t _numMatrices) const
  661. {
  662. uint32_t cached = bgfx::setTransform(_mtx, _numMatrices);
  663. for (uint32_t pass = 0; pass < _numPasses; ++pass)
  664. {
  665. bgfx::setTransform(cached, _numMatrices);
  666. const MeshState& state = *_state[pass];
  667. bgfx::setState(state.m_state);
  668. for (uint8_t tex = 0; tex < state.m_numTextures; ++tex)
  669. {
  670. const MeshState::Texture& texture = state.m_textures[tex];
  671. bgfx::setTexture(texture.m_stage
  672. , texture.m_sampler
  673. , texture.m_texture
  674. , texture.m_flags
  675. );
  676. }
  677. for (GroupArray::const_iterator it = m_groups.begin(), itEnd = m_groups.end(); it != itEnd; ++it)
  678. {
  679. const Group& group = *it;
  680. bgfx::setIndexBuffer(group.m_ibh);
  681. bgfx::setVertexBuffer(group.m_vbh);
  682. bgfx::submit(state.m_viewId, state.m_program, 0, it != itEnd-1);
  683. }
  684. }
  685. }
  686. bgfx::VertexDecl m_decl;
  687. typedef stl::vector<Group> GroupArray;
  688. GroupArray m_groups;
  689. };
  690. Mesh* meshLoad(bx::ReaderSeekerI* _reader)
  691. {
  692. Mesh* mesh = new Mesh;
  693. mesh->load(_reader);
  694. return mesh;
  695. }
  696. Mesh* meshLoad(const char* _filePath)
  697. {
  698. bx::FileReaderI* reader = entry::getFileReader();
  699. if (bx::open(reader, _filePath) )
  700. {
  701. Mesh* mesh = meshLoad(reader);
  702. bx::close(reader);
  703. return mesh;
  704. }
  705. return NULL;
  706. }
  707. void meshUnload(Mesh* _mesh)
  708. {
  709. _mesh->unload();
  710. delete _mesh;
  711. }
  712. MeshState* meshStateCreate()
  713. {
  714. MeshState* state = (MeshState*)BX_ALLOC(entry::getAllocator(), sizeof(MeshState) );
  715. return state;
  716. }
  717. void meshStateDestroy(MeshState* _meshState)
  718. {
  719. BX_FREE(entry::getAllocator(), _meshState);
  720. }
  721. void meshSubmit(const Mesh* _mesh, uint8_t _id, bgfx::ProgramHandle _program, const float* _mtx, uint64_t _state)
  722. {
  723. _mesh->submit(_id, _program, _mtx, _state);
  724. }
  725. void meshSubmit(const Mesh* _mesh, const MeshState*const* _state, uint8_t _numPasses, const float* _mtx, uint16_t _numMatrices)
  726. {
  727. _mesh->submit(_state, _numPasses, _mtx, _numMatrices);
  728. }
  729. Args::Args(int _argc, char** _argv)
  730. : m_type(bgfx::RendererType::Count)
  731. , m_pciId(BGFX_PCI_ID_NONE)
  732. {
  733. bx::CommandLine cmdLine(_argc, (const char**)_argv);
  734. if (cmdLine.hasArg("gl") )
  735. {
  736. m_type = bgfx::RendererType::OpenGL;
  737. }
  738. else if (cmdLine.hasArg("vk") )
  739. {
  740. m_type = bgfx::RendererType::Vulkan;
  741. }
  742. else if (cmdLine.hasArg("noop") )
  743. {
  744. m_type = bgfx::RendererType::Noop;
  745. }
  746. else if (BX_ENABLED(BX_PLATFORM_WINDOWS) )
  747. {
  748. if (cmdLine.hasArg("d3d9") )
  749. {
  750. m_type = bgfx::RendererType::Direct3D9;
  751. }
  752. else if (cmdLine.hasArg("d3d11") )
  753. {
  754. m_type = bgfx::RendererType::Direct3D11;
  755. }
  756. else if (cmdLine.hasArg("d3d12") )
  757. {
  758. m_type = bgfx::RendererType::Direct3D12;
  759. }
  760. }
  761. else if (BX_ENABLED(BX_PLATFORM_OSX) )
  762. {
  763. if (cmdLine.hasArg("mtl") )
  764. {
  765. m_type = bgfx::RendererType::Metal;
  766. }
  767. }
  768. if (cmdLine.hasArg("amd") )
  769. {
  770. m_pciId = BGFX_PCI_ID_AMD;
  771. }
  772. else if (cmdLine.hasArg("nvidia") )
  773. {
  774. m_pciId = BGFX_PCI_ID_NVIDIA;
  775. }
  776. else if (cmdLine.hasArg("intel") )
  777. {
  778. m_pciId = BGFX_PCI_ID_INTEL;
  779. }
  780. else if (cmdLine.hasArg("sw") )
  781. {
  782. m_pciId = BGFX_PCI_ID_SOFTWARE_RASTERIZER;
  783. }
  784. }