bgfx_utils.cpp 22 KB

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