bgfx_utils.cpp 22 KB

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