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