bgfx_utils.cpp 22 KB

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