stencil.cpp 38 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418
  1. /*
  2. * Copyright 2013-2014 Dario Manesku. All rights reserved.
  3. * License: https://github.com/bkaradzic/bgfx/blob/master/LICENSE
  4. */
  5. #include <string>
  6. #include <vector>
  7. #include "common.h"
  8. #include "bgfx_utils.h"
  9. #include <bx/file.h>
  10. #include "camera.h"
  11. #include "imgui/imgui.h"
  12. namespace bgfx
  13. {
  14. int32_t read(bx::ReaderI* _reader, bgfx::VertexLayout& _layout, bx::Error* _err);
  15. }
  16. namespace
  17. {
  18. #define RENDER_VIEWID_RANGE1_PASS_0 1
  19. #define RENDER_VIEWID_RANGE1_PASS_1 2
  20. #define RENDER_VIEWID_RANGE1_PASS_2 3
  21. #define RENDER_VIEWID_RANGE1_PASS_3 4
  22. #define RENDER_VIEWID_RANGE1_PASS_4 5
  23. #define RENDER_VIEWID_RANGE1_PASS_5 6
  24. #define RENDER_VIEWID_RANGE5_PASS_6 7
  25. #define RENDER_VIEWID_RANGE1_PASS_7 13
  26. #define MAX_NUM_LIGHTS 5
  27. struct PosNormalTexcoordVertex
  28. {
  29. float m_x;
  30. float m_y;
  31. float m_z;
  32. uint32_t m_normal;
  33. float m_u;
  34. float m_v;
  35. static void init()
  36. {
  37. ms_layout
  38. .begin()
  39. .add(bgfx::Attrib::Position, 3, bgfx::AttribType::Float)
  40. .add(bgfx::Attrib::Normal, 4, bgfx::AttribType::Uint8, true, true)
  41. .add(bgfx::Attrib::TexCoord0, 2, bgfx::AttribType::Float)
  42. .end();
  43. }
  44. static bgfx::VertexLayout ms_layout;
  45. };
  46. bgfx::VertexLayout PosNormalTexcoordVertex::ms_layout;
  47. static const float s_texcoord = 5.0f;
  48. static PosNormalTexcoordVertex s_hplaneVertices[] =
  49. {
  50. { -1.0f, 0.0f, 1.0f, encodeNormalRgba8(0.0f, 1.0f, 0.0f), s_texcoord, s_texcoord },
  51. { 1.0f, 0.0f, 1.0f, encodeNormalRgba8(0.0f, 1.0f, 0.0f), s_texcoord, 0.0f },
  52. { -1.0f, 0.0f, -1.0f, encodeNormalRgba8(0.0f, 1.0f, 0.0f), 0.0f, s_texcoord },
  53. { 1.0f, 0.0f, -1.0f, encodeNormalRgba8(0.0f, 1.0f, 0.0f), 0.0f, 0.0f },
  54. };
  55. static PosNormalTexcoordVertex s_vplaneVertices[] =
  56. {
  57. { -1.0f, 1.0f, 0.0f, encodeNormalRgba8(0.0f, 0.0f, -1.0f), 1.0f, 1.0f },
  58. { 1.0f, 1.0f, 0.0f, encodeNormalRgba8(0.0f, 0.0f, -1.0f), 1.0f, 0.0f },
  59. { -1.0f, -1.0f, 0.0f, encodeNormalRgba8(0.0f, 0.0f, -1.0f), 0.0f, 1.0f },
  60. { 1.0f, -1.0f, 0.0f, encodeNormalRgba8(0.0f, 0.0f, -1.0f), 0.0f, 0.0f },
  61. };
  62. static const PosNormalTexcoordVertex s_cubeVertices[] =
  63. {
  64. { -1.0f, 1.0f, 1.0f, encodeNormalRgba8( 0.0f, 1.0f, 0.0f), 1.0f, 1.0f },
  65. { 1.0f, 1.0f, 1.0f, encodeNormalRgba8( 0.0f, 1.0f, 0.0f), 0.0f, 1.0f },
  66. { -1.0f, 1.0f, -1.0f, encodeNormalRgba8( 0.0f, 1.0f, 0.0f), 1.0f, 0.0f },
  67. { 1.0f, 1.0f, -1.0f, encodeNormalRgba8( 0.0f, 1.0f, 0.0f), 0.0f, 0.0f },
  68. { -1.0f, -1.0f, 1.0f, encodeNormalRgba8( 0.0f, -1.0f, 0.0f), 1.0f, 1.0f },
  69. { 1.0f, -1.0f, 1.0f, encodeNormalRgba8( 0.0f, -1.0f, 0.0f), 0.0f, 1.0f },
  70. { -1.0f, -1.0f, -1.0f, encodeNormalRgba8( 0.0f, -1.0f, 0.0f), 1.0f, 0.0f },
  71. { 1.0f, -1.0f, -1.0f, encodeNormalRgba8( 0.0f, -1.0f, 0.0f), 0.0f, 0.0f },
  72. { 1.0f, -1.0f, 1.0f, encodeNormalRgba8( 0.0f, 0.0f, 1.0f), 0.0f, 0.0f },
  73. { 1.0f, 1.0f, 1.0f, encodeNormalRgba8( 0.0f, 0.0f, 1.0f), 0.0f, 1.0f },
  74. { -1.0f, -1.0f, 1.0f, encodeNormalRgba8( 0.0f, 0.0f, 1.0f), 1.0f, 0.0f },
  75. { -1.0f, 1.0f, 1.0f, encodeNormalRgba8( 0.0f, 0.0f, 1.0f), 1.0f, 1.0f },
  76. { 1.0f, -1.0f, -1.0f, encodeNormalRgba8( 0.0f, 0.0f, -1.0f), 0.0f, 0.0f },
  77. { 1.0f, 1.0f, -1.0f, encodeNormalRgba8( 0.0f, 0.0f, -1.0f), 0.0f, 1.0f },
  78. { -1.0f, -1.0f, -1.0f, encodeNormalRgba8( 0.0f, 0.0f, -1.0f), 1.0f, 0.0f },
  79. { -1.0f, 1.0f, -1.0f, encodeNormalRgba8( 0.0f, 0.0f, -1.0f), 1.0f, 1.0f },
  80. { 1.0f, 1.0f, -1.0f, encodeNormalRgba8( 1.0f, 0.0f, 0.0f), 1.0f, 1.0f },
  81. { 1.0f, 1.0f, 1.0f, encodeNormalRgba8( 1.0f, 0.0f, 0.0f), 0.0f, 1.0f },
  82. { 1.0f, -1.0f, -1.0f, encodeNormalRgba8( 1.0f, 0.0f, 0.0f), 1.0f, 0.0f },
  83. { 1.0f, -1.0f, 1.0f, encodeNormalRgba8( 1.0f, 0.0f, 0.0f), 0.0f, 0.0f },
  84. { -1.0f, 1.0f, -1.0f, encodeNormalRgba8(-1.0f, 0.0f, 0.0f), 1.0f, 1.0f },
  85. { -1.0f, 1.0f, 1.0f, encodeNormalRgba8(-1.0f, 0.0f, 0.0f), 0.0f, 1.0f },
  86. { -1.0f, -1.0f, -1.0f, encodeNormalRgba8(-1.0f, 0.0f, 0.0f), 1.0f, 0.0f },
  87. { -1.0f, -1.0f, 1.0f, encodeNormalRgba8(-1.0f, 0.0f, 0.0f), 0.0f, 0.0f },
  88. };
  89. static const uint16_t s_cubeIndices[] =
  90. {
  91. 0, 1, 2,
  92. 1, 3, 2,
  93. 4, 6, 5,
  94. 5, 6, 7,
  95. 8, 9, 10,
  96. 9, 11, 10,
  97. 12, 14, 13,
  98. 13, 14, 15,
  99. 16, 17, 18,
  100. 17, 19, 18,
  101. 20, 22, 21,
  102. 21, 22, 23,
  103. };
  104. static const uint16_t s_planeIndices[] =
  105. {
  106. 0, 1, 2,
  107. 1, 3, 2,
  108. };
  109. static uint32_t s_viewMask = 0;
  110. static uint32_t s_clearMask = 0;
  111. static bgfx::UniformHandle s_texColor;
  112. void setViewClearMask(uint32_t _viewMask, uint8_t _flags, uint32_t _rgba, float _depth, uint8_t _stencil)
  113. {
  114. for (uint32_t view = 0, viewMask = _viewMask; 0 != viewMask; viewMask >>= 1, view += 1 )
  115. {
  116. const uint32_t ntz = bx::uint32_cnttz(viewMask);
  117. viewMask >>= ntz;
  118. view += ntz;
  119. bgfx::setViewClear( (uint8_t)view, _flags, _rgba, _depth, _stencil);
  120. }
  121. }
  122. void setViewTransformMask(uint32_t _viewMask, const void* _view, const void* _proj)
  123. {
  124. for (uint32_t view = 0, viewMask = _viewMask; 0 != viewMask; viewMask >>= 1, view += 1 )
  125. {
  126. const uint32_t ntz = bx::uint32_cnttz(viewMask);
  127. viewMask >>= ntz;
  128. view += ntz;
  129. bgfx::setViewTransform( (uint8_t)view, _view, _proj);
  130. }
  131. }
  132. void setViewRectMask(uint32_t _viewMask, uint16_t _x, uint16_t _y, uint16_t _width, uint16_t _height)
  133. {
  134. for (uint32_t view = 0, viewMask = _viewMask; 0 != viewMask; viewMask >>= 1, view += 1 )
  135. {
  136. const uint32_t ntz = bx::uint32_cnttz(viewMask);
  137. viewMask >>= ntz;
  138. view += ntz;
  139. bgfx::setViewRect( (uint8_t)view, _x, _y, _width, _height);
  140. }
  141. }
  142. void mtxReflected(float* _result, const bx::Vec3& _pos, const bx::Vec3& _normal)
  143. {
  144. const float nx = _normal.x;
  145. const float ny = _normal.y;
  146. const float nz = _normal.z;
  147. _result[ 0] = 1.0f - 2.0f * nx * nx;
  148. _result[ 1] = - 2.0f * nx * ny;
  149. _result[ 2] = - 2.0f * nx * nz;
  150. _result[ 3] = 0.0f;
  151. _result[ 4] = - 2.0f * nx * ny;
  152. _result[ 5] = 1.0f - 2.0f * ny * ny;
  153. _result[ 6] = - 2.0f * ny * nz;
  154. _result[ 7] = 0.0f;
  155. _result[ 8] = - 2.0f * nx * nz;
  156. _result[ 9] = - 2.0f * ny * nz;
  157. _result[10] = 1.0f - 2.0f * nz * nz;
  158. _result[11] = 0.0f;
  159. const float dot = bx::dot(_pos, _normal);
  160. _result[12] = 2.0f * dot * nx;
  161. _result[13] = 2.0f * dot * ny;
  162. _result[14] = 2.0f * dot * nz;
  163. _result[15] = 1.0f;
  164. }
  165. void mtxShadow(float* _result, const float* _ground, const float* _light)
  166. {
  167. const float dot =
  168. _ground[0] * _light[0]
  169. + _ground[1] * _light[1]
  170. + _ground[2] * _light[2]
  171. + _ground[3] * _light[3]
  172. ;
  173. _result[ 0] = dot - _light[0] * _ground[0];
  174. _result[ 1] = 0.0f - _light[1] * _ground[0];
  175. _result[ 2] = 0.0f - _light[2] * _ground[0];
  176. _result[ 3] = 0.0f - _light[3] * _ground[0];
  177. _result[ 4] = 0.0f - _light[0] * _ground[1];
  178. _result[ 5] = dot - _light[1] * _ground[1];
  179. _result[ 6] = 0.0f - _light[2] * _ground[1];
  180. _result[ 7] = 0.0f - _light[3] * _ground[1];
  181. _result[ 8] = 0.0f - _light[0] * _ground[2];
  182. _result[ 9] = 0.0f - _light[1] * _ground[2];
  183. _result[10] = dot - _light[2] * _ground[2];
  184. _result[11] = 0.0f - _light[3] * _ground[2];
  185. _result[12] = 0.0f - _light[0] * _ground[3];
  186. _result[13] = 0.0f - _light[1] * _ground[3];
  187. _result[14] = 0.0f - _light[2] * _ground[3];
  188. _result[15] = dot - _light[3] * _ground[3];
  189. }
  190. void mtxBillboard(float* _result, const float* _view, const float* _pos, const float* _scale)
  191. {
  192. _result[ 0] = _view[0] * _scale[0];
  193. _result[ 1] = _view[4] * _scale[0];
  194. _result[ 2] = _view[8] * _scale[0];
  195. _result[ 3] = 0.0f;
  196. _result[ 4] = _view[1] * _scale[1];
  197. _result[ 5] = _view[5] * _scale[1];
  198. _result[ 6] = _view[9] * _scale[1];
  199. _result[ 7] = 0.0f;
  200. _result[ 8] = _view[2] * _scale[2];
  201. _result[ 9] = _view[6] * _scale[2];
  202. _result[10] = _view[10] * _scale[2];
  203. _result[11] = 0.0f;
  204. _result[12] = _pos[0];
  205. _result[13] = _pos[1];
  206. _result[14] = _pos[2];
  207. _result[15] = 1.0f;
  208. }
  209. struct Uniforms
  210. {
  211. void init()
  212. {
  213. m_params.m_ambientPass = 1.0f;
  214. m_params.m_lightingPass = 1.0f;
  215. m_params.m_lightCount = 4.0f;
  216. m_params.m_lightIndex = 4.0f;
  217. m_ambient[0] = 0.02f;
  218. m_ambient[1] = 0.02f;
  219. m_ambient[2] = 0.02f;
  220. m_ambient[3] = 0.0f; //unused
  221. m_diffuse[0] = 0.2f;
  222. m_diffuse[1] = 0.2f;
  223. m_diffuse[2] = 0.2f;
  224. m_diffuse[3] = 0.0f; //unused
  225. m_specular_shininess[0] = 1.0f;
  226. m_specular_shininess[1] = 1.0f;
  227. m_specular_shininess[2] = 1.0f;
  228. m_specular_shininess[3] = 10.0f; //shininess
  229. m_color[0] = 1.0f;
  230. m_color[1] = 1.0f;
  231. m_color[2] = 1.0f;
  232. m_color[3] = 1.0f;
  233. m_time = 0.0f;
  234. for (uint8_t ii = 0; ii < MAX_NUM_LIGHTS; ++ii)
  235. {
  236. m_lightPosRadius[ii][0] = 0.0f;
  237. m_lightPosRadius[ii][1] = 0.0f;
  238. m_lightPosRadius[ii][2] = 0.0f;
  239. m_lightPosRadius[ii][3] = 1.0f;
  240. m_lightRgbInnerR[ii][0] = 1.0f;
  241. m_lightRgbInnerR[ii][1] = 1.0f;
  242. m_lightRgbInnerR[ii][2] = 1.0f;
  243. m_lightRgbInnerR[ii][3] = 1.0f;
  244. }
  245. u_params = bgfx::createUniform("u_params", bgfx::UniformType::Vec4);
  246. u_ambient = bgfx::createUniform("u_ambient", bgfx::UniformType::Vec4);
  247. u_diffuse = bgfx::createUniform("u_diffuse", bgfx::UniformType::Vec4);
  248. u_specular_shininess = bgfx::createUniform("u_specular_shininess", bgfx::UniformType::Vec4);
  249. u_color = bgfx::createUniform("u_color", bgfx::UniformType::Vec4);
  250. u_lightPosRadius = bgfx::createUniform("u_lightPosRadius", bgfx::UniformType::Vec4, MAX_NUM_LIGHTS);
  251. u_lightRgbInnerR = bgfx::createUniform("u_lightRgbInnerR", bgfx::UniformType::Vec4, MAX_NUM_LIGHTS);
  252. }
  253. //call this once at initialization
  254. void submitConstUniforms()
  255. {
  256. bgfx::setUniform(u_ambient, &m_ambient);
  257. bgfx::setUniform(u_diffuse, &m_diffuse);
  258. bgfx::setUniform(u_specular_shininess, &m_specular_shininess);
  259. }
  260. //call this before each draw call
  261. void submitPerDrawUniforms()
  262. {
  263. bgfx::setUniform(u_params, &m_params);
  264. bgfx::setUniform(u_color, &m_color);
  265. bgfx::setUniform(u_lightPosRadius, &m_lightPosRadius, MAX_NUM_LIGHTS);
  266. bgfx::setUniform(u_lightRgbInnerR, &m_lightRgbInnerR, MAX_NUM_LIGHTS);
  267. }
  268. void destroy()
  269. {
  270. bgfx::destroy(u_params);
  271. bgfx::destroy(u_ambient);
  272. bgfx::destroy(u_diffuse);
  273. bgfx::destroy(u_specular_shininess);
  274. bgfx::destroy(u_color);
  275. bgfx::destroy(u_lightPosRadius);
  276. bgfx::destroy(u_lightRgbInnerR);
  277. }
  278. struct Params
  279. {
  280. float m_ambientPass;
  281. float m_lightingPass;
  282. float m_lightCount;
  283. float m_lightIndex;
  284. };
  285. struct SvParams
  286. {
  287. float m_useStencilTex;
  288. float m_dfail;
  289. float m_unused0;
  290. float m_unused1;
  291. };
  292. Params m_params;
  293. SvParams m_svparams;
  294. float m_ambient[4];
  295. float m_diffuse[4];
  296. float m_specular_shininess[4];
  297. float m_color[4];
  298. float m_time;
  299. float m_lightPosRadius[MAX_NUM_LIGHTS][4];
  300. float m_lightRgbInnerR[MAX_NUM_LIGHTS][4];
  301. /**
  302. * u_params.x - u_ambientPass
  303. * u_params.y - u_lightingPass
  304. * u_params.z - u_lightCount
  305. * u_params.w - u_lightIndex
  306. */
  307. bgfx::UniformHandle u_params;
  308. bgfx::UniformHandle u_ambient;
  309. bgfx::UniformHandle u_diffuse;
  310. bgfx::UniformHandle u_specular_shininess;
  311. bgfx::UniformHandle u_color;
  312. bgfx::UniformHandle u_lightPosRadius;
  313. bgfx::UniformHandle u_lightRgbInnerR;
  314. };
  315. static Uniforms s_uniforms;
  316. //-------------------------------------------------
  317. // Render state
  318. //-------------------------------------------------
  319. struct RenderState
  320. {
  321. enum Enum
  322. {
  323. StencilReflection_CraftStencil = 0,
  324. StencilReflection_DrawReflected,
  325. StencilReflection_BlendPlane,
  326. StencilReflection_DrawScene,
  327. ProjectionShadows_DrawAmbient,
  328. ProjectionShadows_CraftStencil,
  329. ProjectionShadows_DrawDiffuse,
  330. Custom_BlendLightTexture,
  331. Custom_DrawPlaneBottom,
  332. Count
  333. };
  334. uint64_t m_state;
  335. uint32_t m_blendFactorRgba;
  336. uint32_t m_fstencil;
  337. uint32_t m_bstencil;
  338. };
  339. static RenderState s_renderStates[RenderState::Count] =
  340. {
  341. { // StencilReflection_CraftStencil
  342. BGFX_STATE_WRITE_RGB
  343. | BGFX_STATE_WRITE_Z
  344. | BGFX_STATE_DEPTH_TEST_LESS
  345. | BGFX_STATE_MSAA
  346. , UINT32_MAX
  347. , BGFX_STENCIL_TEST_ALWAYS // pass always
  348. | BGFX_STENCIL_FUNC_REF(1) // value = 1
  349. | BGFX_STENCIL_FUNC_RMASK(0xff)
  350. | BGFX_STENCIL_OP_FAIL_S_REPLACE
  351. | BGFX_STENCIL_OP_FAIL_Z_REPLACE
  352. | BGFX_STENCIL_OP_PASS_Z_REPLACE // store the value
  353. , BGFX_STENCIL_NONE
  354. },
  355. { // StencilReflection_DrawReflected
  356. BGFX_STATE_WRITE_RGB
  357. | BGFX_STATE_WRITE_A
  358. | BGFX_STATE_BLEND_FUNC(BGFX_STATE_BLEND_SRC_ALPHA, BGFX_STATE_BLEND_INV_SRC_ALPHA)
  359. | BGFX_STATE_WRITE_Z
  360. | BGFX_STATE_DEPTH_TEST_LESS
  361. | BGFX_STATE_CULL_CW //reflection matrix has inverted normals. using CCW instead of CW.
  362. | BGFX_STATE_MSAA
  363. , UINT32_MAX
  364. , BGFX_STENCIL_TEST_EQUAL
  365. | BGFX_STENCIL_FUNC_REF(1)
  366. | BGFX_STENCIL_FUNC_RMASK(1)
  367. | BGFX_STENCIL_OP_FAIL_S_KEEP
  368. | BGFX_STENCIL_OP_FAIL_Z_KEEP
  369. | BGFX_STENCIL_OP_PASS_Z_KEEP
  370. , BGFX_STENCIL_NONE
  371. },
  372. { // StencilReflection_BlendPlane
  373. BGFX_STATE_WRITE_RGB
  374. | BGFX_STATE_WRITE_Z
  375. | BGFX_STATE_BLEND_FUNC(BGFX_STATE_BLEND_ONE, BGFX_STATE_BLEND_SRC_COLOR)
  376. | BGFX_STATE_DEPTH_TEST_LESS
  377. | BGFX_STATE_CULL_CCW
  378. | BGFX_STATE_MSAA
  379. , UINT32_MAX
  380. , BGFX_STENCIL_NONE
  381. , BGFX_STENCIL_NONE
  382. },
  383. { // StencilReflection_DrawScene
  384. BGFX_STATE_WRITE_RGB
  385. | BGFX_STATE_WRITE_Z
  386. | BGFX_STATE_DEPTH_TEST_LESS
  387. | BGFX_STATE_CULL_CCW
  388. | BGFX_STATE_MSAA
  389. , UINT32_MAX
  390. , BGFX_STENCIL_NONE
  391. , BGFX_STENCIL_NONE
  392. },
  393. { // ProjectionShadows_DrawAmbient
  394. BGFX_STATE_WRITE_RGB
  395. | BGFX_STATE_WRITE_Z // write depth !
  396. | BGFX_STATE_DEPTH_TEST_LESS
  397. | BGFX_STATE_CULL_CCW
  398. | BGFX_STATE_MSAA
  399. , UINT32_MAX
  400. , BGFX_STENCIL_NONE
  401. , BGFX_STENCIL_NONE
  402. },
  403. { // ProjectionShadows_CraftStencil
  404. BGFX_STATE_DEPTH_TEST_LESS
  405. | BGFX_STATE_MSAA
  406. , UINT32_MAX
  407. , BGFX_STENCIL_TEST_ALWAYS // pass always
  408. | BGFX_STENCIL_FUNC_REF(1) // value = 1
  409. | BGFX_STENCIL_FUNC_RMASK(0xff)
  410. | BGFX_STENCIL_OP_FAIL_S_KEEP
  411. | BGFX_STENCIL_OP_FAIL_Z_KEEP
  412. | BGFX_STENCIL_OP_PASS_Z_REPLACE // store the value
  413. , BGFX_STENCIL_NONE
  414. },
  415. { // ProjectionShadows_DrawDiffuse
  416. BGFX_STATE_WRITE_RGB
  417. | BGFX_STATE_BLEND_FUNC(BGFX_STATE_BLEND_ONE, BGFX_STATE_BLEND_ONE)
  418. | BGFX_STATE_DEPTH_TEST_EQUAL
  419. | BGFX_STATE_CULL_CCW
  420. | BGFX_STATE_MSAA
  421. , UINT32_MAX
  422. , BGFX_STENCIL_TEST_NOTEQUAL
  423. | BGFX_STENCIL_FUNC_REF(1)
  424. | BGFX_STENCIL_FUNC_RMASK(1)
  425. | BGFX_STENCIL_OP_FAIL_S_KEEP
  426. | BGFX_STENCIL_OP_FAIL_Z_KEEP
  427. | BGFX_STENCIL_OP_PASS_Z_KEEP
  428. , BGFX_STENCIL_NONE
  429. },
  430. { // Custom_BlendLightTexture
  431. BGFX_STATE_WRITE_RGB
  432. | BGFX_STATE_WRITE_A
  433. | BGFX_STATE_WRITE_Z
  434. | BGFX_STATE_DEPTH_TEST_LESS
  435. | BGFX_STATE_BLEND_FUNC(BGFX_STATE_BLEND_SRC_COLOR, BGFX_STATE_BLEND_INV_SRC_COLOR)
  436. | BGFX_STATE_CULL_CCW
  437. | BGFX_STATE_MSAA
  438. , UINT32_MAX
  439. , BGFX_STENCIL_NONE
  440. , BGFX_STENCIL_NONE
  441. },
  442. { // Custom_DrawPlaneBottom
  443. BGFX_STATE_WRITE_RGB
  444. | BGFX_STATE_CULL_CW
  445. | BGFX_STATE_MSAA
  446. , UINT32_MAX
  447. , BGFX_STENCIL_NONE
  448. , BGFX_STENCIL_NONE
  449. },
  450. };
  451. struct ViewState
  452. {
  453. ViewState(uint32_t _width = 0, uint32_t _height = 0)
  454. : m_width(_width)
  455. , m_height(_height)
  456. {
  457. }
  458. uint32_t m_width;
  459. uint32_t m_height;
  460. float m_view[16];
  461. float m_proj[16];
  462. };
  463. struct ClearValues
  464. {
  465. ClearValues(uint32_t _clearRgba = 0x30303000
  466. , float _clearDepth = 1.0f
  467. , uint8_t _clearStencil = 0
  468. )
  469. : m_clearRgba(_clearRgba)
  470. , m_clearDepth(_clearDepth)
  471. , m_clearStencil(_clearStencil)
  472. { }
  473. uint32_t m_clearRgba;
  474. float m_clearDepth;
  475. uint8_t m_clearStencil;
  476. };
  477. void clearView(bgfx::ViewId _id, uint8_t _flags, const ClearValues& _clearValues)
  478. {
  479. bgfx::setViewClear(_id
  480. , _flags
  481. , _clearValues.m_clearRgba
  482. , _clearValues.m_clearDepth
  483. , _clearValues.m_clearStencil
  484. );
  485. // Keep track of cleared views
  486. s_clearMask |= 1 << _id;
  487. }
  488. void clearViewMask(uint32_t _viewMask, uint8_t _flags, const ClearValues& _clearValues)
  489. {
  490. setViewClearMask(_viewMask
  491. , _flags
  492. , _clearValues.m_clearRgba
  493. , _clearValues.m_clearDepth
  494. , _clearValues.m_clearStencil
  495. );
  496. // Keep track of cleared views
  497. s_clearMask |= _viewMask;
  498. }
  499. struct Aabb
  500. {
  501. float m_min[3];
  502. float m_max[3];
  503. };
  504. struct Obb
  505. {
  506. float m_mtx[16];
  507. };
  508. struct Sphere
  509. {
  510. float m_center[3];
  511. float m_radius;
  512. };
  513. struct Primitive
  514. {
  515. uint32_t m_startIndex;
  516. uint32_t m_numIndices;
  517. uint32_t m_startVertex;
  518. uint32_t m_numVertices;
  519. Sphere m_sphere;
  520. Aabb m_aabb;
  521. Obb m_obb;
  522. };
  523. typedef std::vector<Primitive> PrimitiveArray;
  524. struct Group
  525. {
  526. Group()
  527. {
  528. reset();
  529. }
  530. void reset()
  531. {
  532. m_vbh.idx = bgfx::kInvalidHandle;
  533. m_ibh.idx = bgfx::kInvalidHandle;
  534. m_sphere={};
  535. m_aabb={};
  536. m_obb={};
  537. m_prims.clear();
  538. }
  539. bgfx::VertexBufferHandle m_vbh;
  540. bgfx::IndexBufferHandle m_ibh;
  541. Sphere m_sphere;
  542. Aabb m_aabb;
  543. Obb m_obb;
  544. PrimitiveArray m_prims;
  545. };
  546. struct Mesh
  547. {
  548. void load(const void* _vertices, uint32_t _numVertices, const bgfx::VertexLayout _layout, const uint16_t* _indices, uint32_t _numIndices)
  549. {
  550. Group group;
  551. const bgfx::Memory* mem;
  552. uint32_t size;
  553. size = _numVertices*_layout.getStride();
  554. mem = bgfx::makeRef(_vertices, size);
  555. group.m_vbh = bgfx::createVertexBuffer(mem, _layout);
  556. size = _numIndices*2;
  557. mem = bgfx::makeRef(_indices, size);
  558. group.m_ibh = bgfx::createIndexBuffer(mem);
  559. m_groups.push_back(group);
  560. }
  561. void load(const char* _filePath)
  562. {
  563. #define BGFX_CHUNK_MAGIC_VB BX_MAKEFOURCC('V', 'B', ' ', 0x1)
  564. #define BGFX_CHUNK_MAGIC_IB BX_MAKEFOURCC('I', 'B', ' ', 0x0)
  565. #define BGFX_CHUNK_MAGIC_PRI BX_MAKEFOURCC('P', 'R', 'I', 0x0)
  566. bx::FileReaderI* reader = entry::getFileReader();
  567. bx::open(reader, _filePath);
  568. Group group;
  569. bx::Error err;
  570. uint32_t chunk;
  571. while (4 == bx::read(reader, chunk, &err) )
  572. {
  573. switch (chunk)
  574. {
  575. case BGFX_CHUNK_MAGIC_VB:
  576. {
  577. bx::read(reader, group.m_sphere, &err);
  578. bx::read(reader, group.m_aabb, &err);
  579. bx::read(reader, group.m_obb, &err);
  580. bgfx::read(reader, m_layout, &err);
  581. uint16_t stride = m_layout.getStride();
  582. uint16_t numVertices;
  583. bx::read(reader, numVertices, &err);
  584. const bgfx::Memory* mem = bgfx::alloc(numVertices*stride);
  585. bx::read(reader, mem->data, mem->size, &err);
  586. group.m_vbh = bgfx::createVertexBuffer(mem, m_layout);
  587. }
  588. break;
  589. case BGFX_CHUNK_MAGIC_IB:
  590. {
  591. uint32_t numIndices;
  592. bx::read(reader, numIndices, &err);
  593. const bgfx::Memory* mem = bgfx::alloc(numIndices*2);
  594. bx::read(reader, mem->data, mem->size, &err);
  595. group.m_ibh = bgfx::createIndexBuffer(mem);
  596. }
  597. break;
  598. case BGFX_CHUNK_MAGIC_PRI:
  599. {
  600. uint16_t len;
  601. bx::read(reader, len, &err);
  602. std::string material;
  603. material.resize(len);
  604. bx::read(reader, const_cast<char*>(material.c_str() ), len, &err);
  605. uint16_t num;
  606. bx::read(reader, num, &err);
  607. for (uint32_t ii = 0; ii < num; ++ii)
  608. {
  609. bx::read(reader, len, &err);
  610. std::string name;
  611. name.resize(len);
  612. bx::read(reader, const_cast<char*>(name.c_str() ), len, &err);
  613. Primitive prim;
  614. bx::read(reader, prim.m_startIndex, &err);
  615. bx::read(reader, prim.m_numIndices, &err);
  616. bx::read(reader, prim.m_startVertex, &err);
  617. bx::read(reader, prim.m_numVertices, &err);
  618. bx::read(reader, prim.m_sphere, &err);
  619. bx::read(reader, prim.m_aabb, &err);
  620. bx::read(reader, prim.m_obb, &err);
  621. group.m_prims.push_back(prim);
  622. }
  623. m_groups.push_back(group);
  624. group.reset();
  625. }
  626. break;
  627. default:
  628. DBG("%08x at %d", chunk, bx::seek(reader) );
  629. abort();
  630. break;
  631. }
  632. }
  633. bx::close(reader);
  634. }
  635. void unload()
  636. {
  637. for (GroupArray::const_iterator it = m_groups.begin(), itEnd = m_groups.end(); it != itEnd; ++it)
  638. {
  639. const Group& group = *it;
  640. bgfx::destroy(group.m_vbh);
  641. if (bgfx::isValid(group.m_ibh) )
  642. {
  643. bgfx::destroy(group.m_ibh);
  644. }
  645. }
  646. m_groups.clear();
  647. }
  648. void submit(bgfx::ViewId _id, float* _mtx, bgfx::ProgramHandle _program, const RenderState& _renderState)
  649. {
  650. bgfx::TextureHandle texture = BGFX_INVALID_HANDLE;
  651. submit(_id, _mtx, _program, _renderState, texture);
  652. }
  653. void submit(bgfx::ViewId _id, float* _mtx, bgfx::ProgramHandle _program, const RenderState& _renderState, bgfx::TextureHandle _texture)
  654. {
  655. for (GroupArray::const_iterator it = m_groups.begin(), itEnd = m_groups.end(); it != itEnd; ++it)
  656. {
  657. const Group& group = *it;
  658. // Set uniforms
  659. s_uniforms.submitPerDrawUniforms();
  660. // Set model matrix for rendering.
  661. bgfx::setTransform(_mtx);
  662. bgfx::setIndexBuffer(group.m_ibh);
  663. bgfx::setVertexBuffer(0, group.m_vbh);
  664. // Set texture
  665. bgfx::setTexture(0, s_texColor, _texture);
  666. // Apply render state
  667. bgfx::setStencil(_renderState.m_fstencil, _renderState.m_bstencil);
  668. bgfx::setState(_renderState.m_state, _renderState.m_blendFactorRgba);
  669. // Submit
  670. bgfx::submit(_id, _program);
  671. // Keep track of submitted view ids
  672. s_viewMask |= 1 << _id;
  673. }
  674. }
  675. bgfx::VertexLayout m_layout;
  676. typedef std::vector<Group> GroupArray;
  677. GroupArray m_groups;
  678. };
  679. class ExampleStencil : public entry::AppI
  680. {
  681. public:
  682. ExampleStencil(const char* _name, const char* _description, const char* _url)
  683. : entry::AppI(_name, _description, _url)
  684. {
  685. }
  686. virtual void init(int32_t _argc, const char* const* _argv, uint32_t _width, uint32_t _height) override
  687. {
  688. Args args(_argc, _argv);
  689. m_viewState = ViewState(_width, _height);
  690. m_clearValues = ClearValues(0x30303000, 1.0f, 0);
  691. m_debug = BGFX_DEBUG_NONE;
  692. m_reset = BGFX_RESET_VSYNC;
  693. bgfx::Init init;
  694. init.type = args.m_type;
  695. init.vendorId = args.m_pciId;
  696. init.platformData.nwh = entry::getNativeWindowHandle(entry::kDefaultWindowHandle);
  697. init.platformData.ndt = entry::getNativeDisplayHandle();
  698. init.platformData.type = entry::getNativeWindowHandleType(entry::kDefaultWindowHandle);
  699. init.resolution.width = m_viewState.m_width;
  700. init.resolution.height = m_viewState.m_height;
  701. init.resolution.reset = m_reset;
  702. bgfx::init(init);
  703. // Enable debug text.
  704. bgfx::setDebug(m_debug);
  705. // Imgui.
  706. imguiCreate();
  707. PosNormalTexcoordVertex::init();
  708. s_uniforms.init();
  709. s_texColor = bgfx::createUniform("s_texColor", bgfx::UniformType::Sampler);
  710. m_programTextureLighting = loadProgram("vs_stencil_texture_lighting", "fs_stencil_texture_lighting");
  711. m_programColorLighting = loadProgram("vs_stencil_color_lighting", "fs_stencil_color_lighting" );
  712. m_programColorTexture = loadProgram("vs_stencil_color_texture", "fs_stencil_color_texture" );
  713. m_programColorBlack = loadProgram("vs_stencil_color", "fs_stencil_color_black" );
  714. m_programTexture = loadProgram("vs_stencil_texture", "fs_stencil_texture" );
  715. m_bunnyMesh.load("meshes/bunny.bin");
  716. m_columnMesh.load("meshes/column.bin");
  717. m_cubeMesh.load(s_cubeVertices, BX_COUNTOF(s_cubeVertices), PosNormalTexcoordVertex::ms_layout, s_cubeIndices, BX_COUNTOF(s_cubeIndices) );
  718. m_hplaneMesh.load(s_hplaneVertices, BX_COUNTOF(s_hplaneVertices), PosNormalTexcoordVertex::ms_layout, s_planeIndices, BX_COUNTOF(s_planeIndices) );
  719. m_vplaneMesh.load(s_vplaneVertices, BX_COUNTOF(s_vplaneVertices), PosNormalTexcoordVertex::ms_layout, s_planeIndices, BX_COUNTOF(s_planeIndices) );
  720. m_figureTex = loadTexture("textures/figure-rgba.dds");
  721. m_flareTex = loadTexture("textures/flare.dds");
  722. m_fieldstoneTex = loadTexture("textures/fieldstone-rgba.dds");
  723. // Setup lights.
  724. const float rgbInnerR[][4] =
  725. {
  726. { 1.0f, 0.7f, 0.2f, 0.0f }, //yellow
  727. { 0.7f, 0.2f, 1.0f, 0.0f }, //purple
  728. { 0.2f, 1.0f, 0.7f, 0.0f }, //cyan
  729. { 1.0f, 0.4f, 0.2f, 0.0f }, //orange
  730. { 0.7f, 0.7f, 0.7f, 0.0f }, //white
  731. };
  732. for (uint8_t ii = 0, jj = 0; ii < MAX_NUM_LIGHTS; ++ii, ++jj)
  733. {
  734. const uint8_t index = jj%BX_COUNTOF(rgbInnerR);
  735. m_lightRgbInnerR[ii][0] = rgbInnerR[index][0];
  736. m_lightRgbInnerR[ii][1] = rgbInnerR[index][1];
  737. m_lightRgbInnerR[ii][2] = rgbInnerR[index][2];
  738. m_lightRgbInnerR[ii][3] = rgbInnerR[index][3];
  739. }
  740. bx::memCopy(s_uniforms.m_lightRgbInnerR, m_lightRgbInnerR, MAX_NUM_LIGHTS * 4*sizeof(float) );
  741. // Set view and projection matrices.
  742. const float aspect = float(m_viewState.m_width)/float(m_viewState.m_height);
  743. const bgfx::Caps* caps = bgfx::getCaps();
  744. bx::mtxProj(m_viewState.m_proj, 60.0f, aspect, 0.1f, 100.0f, caps->homogeneousDepth);
  745. cameraCreate();
  746. cameraSetPosition({ 0.0f, 18.0f, -40.0f });
  747. cameraSetVerticalAngle(-0.35f);
  748. cameraGetViewMtx(m_viewState.m_view);
  749. m_timeOffset = bx::getHPCounter();
  750. m_scene = StencilReflectionScene;
  751. m_numLights = 4;
  752. m_reflectionValue = 0.8f;
  753. m_updateLights = true;
  754. m_updateScene = true;
  755. }
  756. virtual int shutdown() override
  757. {
  758. // Cleanup.
  759. m_bunnyMesh.unload();
  760. m_columnMesh.unload();
  761. m_cubeMesh.unload();
  762. m_hplaneMesh.unload();
  763. m_vplaneMesh.unload();
  764. bgfx::destroy(m_figureTex);
  765. bgfx::destroy(m_fieldstoneTex);
  766. bgfx::destroy(m_flareTex);
  767. bgfx::destroy(m_programTextureLighting);
  768. bgfx::destroy(m_programColorLighting);
  769. bgfx::destroy(m_programColorTexture);
  770. bgfx::destroy(m_programColorBlack);
  771. bgfx::destroy(m_programTexture);
  772. bgfx::destroy(s_texColor);
  773. s_uniforms.destroy();
  774. cameraDestroy();
  775. imguiDestroy();
  776. // Shutdown bgfx.
  777. bgfx::shutdown();
  778. return 0;
  779. }
  780. virtual bool update() override
  781. {
  782. if (!entry::processEvents(m_viewState.m_width, m_viewState.m_height, m_debug, m_reset, &m_mouseState) )
  783. {
  784. imguiBeginFrame(m_mouseState.m_mx
  785. , m_mouseState.m_my
  786. , (m_mouseState.m_buttons[entry::MouseButton::Left ] ? IMGUI_MBUT_LEFT : 0)
  787. | (m_mouseState.m_buttons[entry::MouseButton::Right ] ? IMGUI_MBUT_RIGHT : 0)
  788. | (m_mouseState.m_buttons[entry::MouseButton::Middle] ? IMGUI_MBUT_MIDDLE : 0)
  789. , m_mouseState.m_mz
  790. , uint16_t(m_viewState.m_width)
  791. , uint16_t(m_viewState.m_height)
  792. );
  793. showExampleDialog(this);
  794. ImGui::SetNextWindowPos(
  795. ImVec2(m_viewState.m_width - m_viewState.m_width / 5.0f - 10.0f, 10.0f)
  796. , ImGuiCond_FirstUseEver
  797. );
  798. ImGui::SetNextWindowSize(
  799. ImVec2(m_viewState.m_width / 5.0f, m_viewState.m_height / 2.0f)
  800. , ImGuiCond_FirstUseEver
  801. );
  802. ImGui::Begin("Settings"
  803. , NULL
  804. , 0
  805. );
  806. {
  807. bool check = StencilReflectionScene == m_scene;
  808. if (ImGui::Checkbox("Stencil Reflection Scene", &check) )
  809. {
  810. m_scene = StencilReflectionScene;
  811. m_numLights = 4;
  812. }
  813. }
  814. {
  815. bool check = ProjectionShadowsScene == m_scene;
  816. if (ImGui::Checkbox("Projection Shadows Scene", &check) )
  817. {
  818. m_scene = ProjectionShadowsScene;
  819. m_numLights = 1;
  820. }
  821. }
  822. ImGui::SliderInt("Lights", &m_numLights, 1, MAX_NUM_LIGHTS);
  823. if (m_scene == StencilReflectionScene)
  824. {
  825. ImGui::SliderFloat("Reflection value", &m_reflectionValue, 0.0f, 1.0f);
  826. }
  827. ImGui::Checkbox("Update lights", &m_updateLights);
  828. ImGui::Checkbox("Update scene", &m_updateScene);
  829. ImGui::End();
  830. imguiEndFrame();
  831. s_uniforms.submitConstUniforms();
  832. // Update settings.
  833. uint8_t numLights = (uint8_t)m_numLights;
  834. s_uniforms.m_params.m_ambientPass = 1.0f;
  835. s_uniforms.m_params.m_lightingPass = 1.0f;
  836. s_uniforms.m_params.m_lightCount = float(m_numLights);
  837. s_uniforms.m_params.m_lightIndex = 0.0f;
  838. s_uniforms.m_color[3] = m_reflectionValue;
  839. // Time.
  840. int64_t now = bx::getHPCounter();
  841. static int64_t last = now;
  842. const int64_t frameTime = now - last;
  843. last = now;
  844. const double freq = double(bx::getHPFrequency() );
  845. const float time = (float)( (now - m_timeOffset)/double(bx::getHPFrequency() ) );
  846. const float deltaTime = float(frameTime/freq);
  847. s_uniforms.m_time = time;
  848. // Update camera.
  849. cameraUpdate(deltaTime, m_mouseState, ImGui::MouseOverArea() );
  850. cameraGetViewMtx(m_viewState.m_view);
  851. static float lightTimeAccumulator = 0.0f;
  852. if (m_updateLights)
  853. {
  854. lightTimeAccumulator += deltaTime;
  855. }
  856. static float sceneTimeAccumulator = 0.0f;
  857. if (m_updateScene)
  858. {
  859. sceneTimeAccumulator += deltaTime;
  860. }
  861. float lightPosRadius[MAX_NUM_LIGHTS][4];
  862. const float radius = (m_scene == StencilReflectionScene) ? 15.0f : 25.0f;
  863. for (uint8_t ii = 0; ii < numLights; ++ii)
  864. {
  865. lightPosRadius[ii][0] = bx::sin( (lightTimeAccumulator*1.1f + ii*0.03f + ii*bx::kPiHalf*1.07f ) )*20.0f;
  866. lightPosRadius[ii][1] = 8.0f + (1.0f - bx::cos( (lightTimeAccumulator*1.5f + ii*0.29f + bx::kPiHalf*1.49f ) ) )*4.0f;
  867. lightPosRadius[ii][2] = bx::cos( (lightTimeAccumulator*1.3f + ii*0.13f + ii*bx::kPiHalf*1.79f ) )*20.0f;
  868. lightPosRadius[ii][3] = radius;
  869. }
  870. bx::memCopy(s_uniforms.m_lightPosRadius, lightPosRadius, numLights * 4*sizeof(float) );
  871. // Floor position.
  872. float floorMtx[16];
  873. bx::mtxSRT(floorMtx
  874. , 20.0f //scaleX
  875. , 20.0f //scaleY
  876. , 20.0f //scaleZ
  877. , 0.0f //rotX
  878. , 0.0f //rotY
  879. , 0.0f //rotZ
  880. , 0.0f //translateX
  881. , 0.0f //translateY
  882. , 0.0f //translateZ
  883. );
  884. // Bunny position.
  885. float bunnyMtx[16];
  886. bx::mtxSRT(bunnyMtx
  887. , 5.0f
  888. , 5.0f
  889. , 5.0f
  890. , 0.0f
  891. , 1.56f - sceneTimeAccumulator
  892. , 0.0f
  893. , 0.0f
  894. , 2.0f
  895. , 0.0f
  896. );
  897. // Columns position.
  898. const float dist = 14.0f;
  899. const float columnPositions[4][3] =
  900. {
  901. { dist, 0.0f, dist },
  902. { -dist, 0.0f, dist },
  903. { dist, 0.0f, -dist },
  904. { -dist, 0.0f, -dist },
  905. };
  906. float columnMtx[4][16];
  907. for (uint8_t ii = 0; ii < 4; ++ii)
  908. {
  909. bx::mtxSRT(columnMtx[ii]
  910. , 1.0f
  911. , 1.0f
  912. , 1.0f
  913. , 0.0f
  914. , 0.0f
  915. , 0.0f
  916. , columnPositions[ii][0]
  917. , columnPositions[ii][1]
  918. , columnPositions[ii][2]
  919. );
  920. }
  921. const uint8_t numCubes = 9;
  922. float cubeMtx[numCubes][16];
  923. for (uint16_t ii = 0; ii < numCubes; ++ii)
  924. {
  925. bx::mtxSRT(cubeMtx[ii]
  926. , 1.0f
  927. , 1.0f
  928. , 1.0f
  929. , 0.0f
  930. , 0.0f
  931. , 0.0f
  932. , bx::sin(ii * 2.0f + 13.0f - sceneTimeAccumulator) * 13.0f
  933. , 4.0f
  934. , bx::cos(ii * 2.0f + 13.0f - sceneTimeAccumulator) * 13.0f
  935. );
  936. }
  937. // Make sure at the beginning everything gets cleared.
  938. clearView(0, BGFX_CLEAR_COLOR | BGFX_CLEAR_DEPTH | BGFX_CLEAR_STENCIL, m_clearValues);
  939. bgfx::touch(0);
  940. s_viewMask |= 1;
  941. // Bunny and columns color.
  942. s_uniforms.m_color[0] = 0.70f;
  943. s_uniforms.m_color[1] = 0.65f;
  944. s_uniforms.m_color[2] = 0.60f;
  945. switch (m_scene)
  946. {
  947. case StencilReflectionScene:
  948. {
  949. // First pass - Draw plane.
  950. // Setup params for this scene.
  951. s_uniforms.m_params.m_ambientPass = 1.0f;
  952. s_uniforms.m_params.m_lightingPass = 1.0f;
  953. // Floor.
  954. m_hplaneMesh.submit(RENDER_VIEWID_RANGE1_PASS_0
  955. , floorMtx
  956. , m_programColorBlack
  957. , s_renderStates[RenderState::StencilReflection_CraftStencil]
  958. );
  959. // Second pass - Draw reflected objects.
  960. // Clear depth from previous pass.
  961. clearView(RENDER_VIEWID_RANGE1_PASS_1, BGFX_CLEAR_DEPTH, m_clearValues);
  962. // Compute reflected matrix.
  963. float reflectMtx[16];
  964. mtxReflected(reflectMtx, { 0.0f, 0.01f, 0.0f }, { 0.0f, 1.0f, 0.0f });
  965. // Reflect lights.
  966. for (uint8_t ii = 0; ii < numLights; ++ii)
  967. {
  968. bx::Vec3 reflected = bx::mul(bx::load<bx::Vec3>(lightPosRadius[ii]), reflectMtx);
  969. bx::store(&s_uniforms.m_lightPosRadius[ii], reflected);
  970. s_uniforms.m_lightPosRadius[ii][3] = lightPosRadius[ii][3];
  971. }
  972. // Reflect and submit bunny.
  973. float mtxReflectedBunny[16];
  974. bx::mtxMul(mtxReflectedBunny, bunnyMtx, reflectMtx);
  975. m_bunnyMesh.submit(RENDER_VIEWID_RANGE1_PASS_1
  976. , mtxReflectedBunny
  977. , m_programColorLighting
  978. , s_renderStates[RenderState::StencilReflection_DrawReflected]
  979. );
  980. // Reflect and submit columns.
  981. float mtxReflectedColumn[16];
  982. for (uint8_t ii = 0; ii < 4; ++ii)
  983. {
  984. bx::mtxMul(mtxReflectedColumn, columnMtx[ii], reflectMtx);
  985. m_columnMesh.submit(RENDER_VIEWID_RANGE1_PASS_1
  986. , mtxReflectedColumn
  987. , m_programColorLighting
  988. , s_renderStates[RenderState::StencilReflection_DrawReflected]
  989. );
  990. }
  991. // Set lights back.
  992. bx::memCopy(s_uniforms.m_lightPosRadius, lightPosRadius, numLights * 4*sizeof(float) );
  993. // Third pass - Blend plane.
  994. // Floor.
  995. m_hplaneMesh.submit(RENDER_VIEWID_RANGE1_PASS_2
  996. , floorMtx
  997. , m_programTextureLighting
  998. , s_renderStates[RenderState::StencilReflection_BlendPlane]
  999. , m_fieldstoneTex
  1000. );
  1001. // Fourth pass - Draw everything else but the plane.
  1002. // Bunny.
  1003. m_bunnyMesh.submit(RENDER_VIEWID_RANGE1_PASS_3
  1004. , bunnyMtx
  1005. , m_programColorLighting
  1006. , s_renderStates[RenderState::StencilReflection_DrawScene]
  1007. );
  1008. // Columns.
  1009. for (uint8_t ii = 0; ii < 4; ++ii)
  1010. {
  1011. m_columnMesh.submit(RENDER_VIEWID_RANGE1_PASS_3
  1012. , columnMtx[ii]
  1013. , m_programColorLighting
  1014. , s_renderStates[RenderState::StencilReflection_DrawScene]
  1015. );
  1016. }
  1017. }
  1018. break;
  1019. case ProjectionShadowsScene:
  1020. {
  1021. // First pass - Draw entire scene. (ambient only).
  1022. s_uniforms.m_params.m_ambientPass = 1.0f;
  1023. s_uniforms.m_params.m_lightingPass = 0.0f;
  1024. // Bunny.
  1025. m_bunnyMesh.submit(RENDER_VIEWID_RANGE1_PASS_0
  1026. , bunnyMtx
  1027. , m_programColorLighting
  1028. , s_renderStates[RenderState::ProjectionShadows_DrawAmbient]
  1029. );
  1030. // Floor.
  1031. m_hplaneMesh.submit(RENDER_VIEWID_RANGE1_PASS_0
  1032. , floorMtx
  1033. , m_programTextureLighting
  1034. , s_renderStates[RenderState::ProjectionShadows_DrawAmbient]
  1035. , m_fieldstoneTex
  1036. );
  1037. // Cubes.
  1038. for (uint8_t ii = 0; ii < numCubes; ++ii)
  1039. {
  1040. m_cubeMesh.submit(RENDER_VIEWID_RANGE1_PASS_0
  1041. , cubeMtx[ii]
  1042. , m_programTextureLighting
  1043. , s_renderStates[RenderState::ProjectionShadows_DrawAmbient]
  1044. , m_figureTex
  1045. );
  1046. }
  1047. // Ground plane.
  1048. float ground[4] = { 0.0f, 1.0f, 0.0f, -bx::dot(bx::Vec3{ 0.0f, 0.0f, 0.0f }, bx::Vec3{ 0.0f, 1.0f, 0.0f }) - 0.01f };
  1049. for (uint8_t ii = 0, viewId = RENDER_VIEWID_RANGE5_PASS_6; ii < numLights; ++ii, ++viewId)
  1050. {
  1051. // Clear stencil for this light source.
  1052. clearView(viewId, BGFX_CLEAR_STENCIL, m_clearValues);
  1053. // Draw shadow projection of scene objects.
  1054. // Get homogeneous light pos.
  1055. float* lightPos = lightPosRadius[ii];
  1056. float pos[4];
  1057. bx::memCopy(pos, lightPos, sizeof(float) * 3);
  1058. pos[3] = 1.0f;
  1059. // Calculate shadow mtx for current light.
  1060. float shadowMtx[16];
  1061. mtxShadow(shadowMtx, ground, pos);
  1062. // Submit bunny's shadow.
  1063. float mtxShadowedBunny[16];
  1064. bx::mtxMul(mtxShadowedBunny, bunnyMtx, shadowMtx);
  1065. m_bunnyMesh.submit(viewId
  1066. , mtxShadowedBunny
  1067. , m_programColorBlack
  1068. , s_renderStates[RenderState::ProjectionShadows_CraftStencil]
  1069. );
  1070. // Submit cube shadows.
  1071. float mtxShadowedCube[16];
  1072. for (uint8_t jj = 0; jj < numCubes; ++jj)
  1073. {
  1074. bx::mtxMul(mtxShadowedCube, cubeMtx[jj], shadowMtx);
  1075. m_cubeMesh.submit(viewId
  1076. , mtxShadowedCube
  1077. , m_programColorBlack
  1078. , s_renderStates[RenderState::ProjectionShadows_CraftStencil]
  1079. );
  1080. }
  1081. // Draw entire scene. (lighting pass only. blending is on)
  1082. s_uniforms.m_params.m_ambientPass = 0.0f;
  1083. s_uniforms.m_params.m_lightingPass = 1.0f;
  1084. s_uniforms.m_params.m_lightCount = 1.0f;
  1085. s_uniforms.m_params.m_lightIndex = float(ii);
  1086. // Bunny.
  1087. m_bunnyMesh.submit(viewId
  1088. , bunnyMtx
  1089. , m_programColorLighting
  1090. , s_renderStates[RenderState::ProjectionShadows_DrawDiffuse]
  1091. );
  1092. // Floor.
  1093. m_hplaneMesh.submit(viewId
  1094. , floorMtx
  1095. , m_programTextureLighting
  1096. , s_renderStates[RenderState::ProjectionShadows_DrawDiffuse]
  1097. , m_fieldstoneTex
  1098. );
  1099. // Cubes.
  1100. for (uint8_t jj = 0; jj < numCubes; ++jj)
  1101. {
  1102. m_cubeMesh.submit(viewId
  1103. , cubeMtx[jj]
  1104. , m_programTextureLighting
  1105. , s_renderStates[RenderState::ProjectionShadows_DrawDiffuse]
  1106. , m_figureTex
  1107. );
  1108. }
  1109. }
  1110. // Reset these to default..
  1111. s_uniforms.m_params.m_ambientPass = 1.0f;
  1112. s_uniforms.m_params.m_lightingPass = 1.0f;
  1113. }
  1114. break;
  1115. };
  1116. //lights
  1117. const float lightScale[3] = { 1.5f, 1.5f, 1.5f };
  1118. float lightMtx[16];
  1119. for (uint8_t ii = 0; ii < numLights; ++ii)
  1120. {
  1121. s_uniforms.m_color[0] = m_lightRgbInnerR[ii][0];
  1122. s_uniforms.m_color[1] = m_lightRgbInnerR[ii][1];
  1123. s_uniforms.m_color[2] = m_lightRgbInnerR[ii][2];
  1124. mtxBillboard(lightMtx, m_viewState.m_view, lightPosRadius[ii], lightScale);
  1125. m_vplaneMesh.submit(RENDER_VIEWID_RANGE1_PASS_7
  1126. , lightMtx
  1127. , m_programColorTexture
  1128. , s_renderStates[RenderState::Custom_BlendLightTexture]
  1129. , m_flareTex
  1130. );
  1131. }
  1132. // Draw floor bottom.
  1133. float floorBottomMtx[16];
  1134. bx::mtxSRT(floorBottomMtx
  1135. , 20.0f //scaleX
  1136. , 20.0f //scaleY
  1137. , 20.0f //scaleZ
  1138. , 0.0f //rotX
  1139. , 0.0f //rotY
  1140. , 0.0f //rotZ
  1141. , 0.0f //translateX
  1142. , -0.1f //translateY
  1143. , 0.0f //translateZ
  1144. );
  1145. m_hplaneMesh.submit(RENDER_VIEWID_RANGE1_PASS_7
  1146. , floorBottomMtx
  1147. , m_programTexture
  1148. , s_renderStates[RenderState::Custom_DrawPlaneBottom]
  1149. , m_figureTex
  1150. );
  1151. // Setup view rect and transform for all used views.
  1152. setViewRectMask(s_viewMask, 0, 0, uint16_t(m_viewState.m_width), uint16_t(m_viewState.m_height) );
  1153. setViewTransformMask(s_viewMask, m_viewState.m_view, m_viewState.m_proj);
  1154. s_viewMask = 0;
  1155. // Advance to next frame. Rendering thread will be kicked to
  1156. // process submitted rendering primitives.
  1157. bgfx::frame();
  1158. //reset clear values on used views
  1159. clearViewMask(s_clearMask, BGFX_CLEAR_NONE, m_clearValues);
  1160. s_clearMask = 0;
  1161. return true;
  1162. }
  1163. return false;
  1164. }
  1165. ViewState m_viewState;
  1166. entry::MouseState m_mouseState;
  1167. ClearValues m_clearValues;
  1168. uint32_t m_debug;
  1169. uint32_t m_reset;
  1170. bgfx::ProgramHandle m_programTextureLighting;
  1171. bgfx::ProgramHandle m_programColorLighting;
  1172. bgfx::ProgramHandle m_programColorTexture;
  1173. bgfx::ProgramHandle m_programColorBlack;
  1174. bgfx::ProgramHandle m_programTexture;
  1175. Mesh m_bunnyMesh;
  1176. Mesh m_columnMesh;
  1177. Mesh m_cubeMesh;
  1178. Mesh m_hplaneMesh;
  1179. Mesh m_vplaneMesh;
  1180. bgfx::TextureHandle m_figureTex;
  1181. bgfx::TextureHandle m_flareTex;
  1182. bgfx::TextureHandle m_fieldstoneTex;
  1183. float m_lightRgbInnerR[MAX_NUM_LIGHTS][4];
  1184. int64_t m_timeOffset;
  1185. enum Scene
  1186. {
  1187. StencilReflectionScene = 0,
  1188. ProjectionShadowsScene,
  1189. };
  1190. Scene m_scene;
  1191. int32_t m_numLights;
  1192. float m_reflectionValue;
  1193. bool m_updateLights;
  1194. bool m_updateScene;
  1195. };
  1196. } // namespace
  1197. ENTRY_IMPLEMENT_MAIN(
  1198. ExampleStencil
  1199. , "13-stencil"
  1200. , "Stencil reflections and shadows."
  1201. , "https://bkaradzic.github.io/bgfx/examples.html#stencil"
  1202. );