Gr.cpp 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338
  1. // Copyright (C) 2009-2016, Panagiotis Christopoulos Charitos and contributors.
  2. // All rights reserved.
  3. // Code licensed under the BSD License.
  4. // http://www.anki3d.org/LICENSE
  5. #include <tests/framework/Framework.h>
  6. #include <anki/Gr.h>
  7. #include <anki/core/NativeWindow.h>
  8. #include <anki/core/Config.h>
  9. #include <anki/util/HighRezTimer.h>
  10. #include <anki/core/StagingGpuMemoryManager.h>
  11. namespace anki
  12. {
  13. const U WIDTH = 1024;
  14. const U HEIGHT = 768;
  15. static const char* VERT_SRC = R"(
  16. out gl_PerVertex
  17. {
  18. vec4 gl_Position;
  19. };
  20. void main()
  21. {
  22. const vec2 POSITIONS[3] = vec2[](vec2(-1.0, 1.0), vec2(0.0, -1.0), vec2(1.0, 1.0));
  23. ANKI_WRITE_POSITION(vec4(POSITIONS[gl_VertexID % 3], 0.0, 1.0));
  24. })";
  25. static const char* VERT_UBO_SRC = R"(
  26. out gl_PerVertex
  27. {
  28. vec4 gl_Position;
  29. };
  30. layout(ANKI_UBO_BINDING(0, 0)) uniform u0_
  31. {
  32. vec4 u_color[3];
  33. };
  34. layout(ANKI_UBO_BINDING(0, 1)) uniform u1_
  35. {
  36. vec4 u_rotation2d;
  37. };
  38. layout(location = 0) out vec3 out_color;
  39. void main()
  40. {
  41. out_color = u_color[gl_VertexID].rgb;
  42. const vec2 POSITIONS[3] = vec2[](vec2(-1.0, 1.0), vec2(0.0, -1.0), vec2(1.0, 1.0));
  43. mat2 rot = mat2(
  44. u_rotation2d.x, u_rotation2d.y, u_rotation2d.z, u_rotation2d.w);
  45. vec2 pos = rot * POSITIONS[gl_VertexID % 3];
  46. gl_Position = vec4(pos, 0.0, 1.0);
  47. })";
  48. static const char* VERT_INP_SRC = R"(
  49. layout(location = 0) in vec3 in_position;
  50. layout(location = 1) in vec3 in_color0;
  51. layout(location = 2) in vec3 in_color1;
  52. out gl_PerVertex
  53. {
  54. vec4 gl_Position;
  55. };
  56. layout(location = 0) out vec3 out_color0;
  57. layout(location = 1) out vec3 out_color1;
  58. void main()
  59. {
  60. gl_Position = vec4(in_position, 1.0);
  61. out_color0 = in_color0;
  62. out_color1 = in_color1;
  63. })";
  64. static const char* VERT_QUAD_SRC = R"(
  65. out gl_PerVertex
  66. {
  67. vec4 gl_Position;
  68. };
  69. layout(location = 0) out vec2 out_uv;
  70. void main()
  71. {
  72. const vec2 POSITIONS[6] =
  73. vec2[](vec2(-1.0, 1.0), vec2(-1.0, -1.0), vec2(1.0, -1.0),
  74. vec2(1.0, -1.0), vec2(1.0, 1.0), vec2(-1.0, 1.0));
  75. ANKI_WRITE_POSITION(vec4(POSITIONS[gl_VertexID], 0.0, 1.0));
  76. out_uv = POSITIONS[gl_VertexID] / 2.0 + 0.5;
  77. })";
  78. static const char* VERT_MRT_SRC = R"(
  79. out gl_PerVertex
  80. {
  81. vec4 gl_Position;
  82. };
  83. layout(location = 0) in vec3 in_pos;
  84. layout(ANKI_UBO_BINDING(0, 0), std140, row_major) uniform u0_
  85. {
  86. mat4 u_mvp;
  87. };
  88. void main()
  89. {
  90. ANKI_WRITE_POSITION(u_mvp * vec4(in_pos, 1.0));
  91. })";
  92. static const char* FRAG_SRC = R"(layout (location = 0) out vec4 out_color;
  93. void main()
  94. {
  95. out_color = vec4(0.5);
  96. })";
  97. static const char* FRAG_UBO_SRC = R"(layout (location = 0) out vec4 out_color;
  98. layout(location = 0) in vec3 in_color;
  99. void main()
  100. {
  101. out_color = vec4(in_color, 1.0);
  102. })";
  103. static const char* FRAG_INP_SRC = R"(layout (location = 0) out vec4 out_color;
  104. layout(location = 0) in vec3 in_color0;
  105. layout(location = 1) in vec3 in_color1;
  106. void main()
  107. {
  108. out_color = vec4(in_color0 + in_color1, 1.0);
  109. })";
  110. static const char* FRAG_TEX_SRC = R"(layout (location = 0) out vec4 out_color;
  111. layout(location = 0) in vec2 in_uv;
  112. layout(ANKI_TEX_BINDING(0, 0)) uniform sampler2D u_tex0;
  113. layout(ANKI_TEX_BINDING(0, 1)) uniform sampler2D u_tex1;
  114. ANKI_USING_FRAG_COORD(768)
  115. void main()
  116. {
  117. if(anki_fragCoord.x < 1024 / 2)
  118. {
  119. if(anki_fragCoord.y < 768 / 2)
  120. {
  121. vec2 uv = in_uv * 2.0;
  122. out_color = textureLod(u_tex0, uv, 0.0);
  123. }
  124. else
  125. {
  126. vec2 uv = in_uv * 2.0 - vec2(0.0, 1.0);
  127. out_color = textureLod(u_tex0, uv, 1.0);
  128. }
  129. }
  130. else
  131. {
  132. if(anki_fragCoord.y < 768 / 2)
  133. {
  134. vec2 uv = in_uv * 2.0 - vec2(1.0, 0.0);
  135. out_color = textureLod(u_tex1, uv, 0.0);
  136. }
  137. else
  138. {
  139. vec2 uv = in_uv * 2.0 - vec2(1.0, 1.0);
  140. out_color = textureLod(u_tex1, uv, 1.0);
  141. }
  142. }
  143. })";
  144. static const char* FRAG_TEX3D_SRC = R"(layout (location = 0) out vec4 out_color;
  145. layout(ANKI_UBO_BINDING(0, 0)) uniform u0_
  146. {
  147. vec4 u_uv;
  148. };
  149. layout(ANKI_TEX_BINDING(0, 0)) uniform sampler3D u_tex;
  150. void main()
  151. {
  152. out_color = textureLod(u_tex, u_uv.xyz, u_uv.w);
  153. })";
  154. static const char* FRAG_MRT_SRC = R"(layout (location = 0) out vec4 out_color0;
  155. layout (location = 1) out vec4 out_color1;
  156. layout(ANKI_UBO_BINDING(0, 1), std140) uniform u1_
  157. {
  158. vec4 u_color0;
  159. vec4 u_color1;
  160. };
  161. void main()
  162. {
  163. out_color0 = u_color0;
  164. out_color1 = u_color1;
  165. })";
  166. static const char* FRAG_MRT2_SRC = R"(layout (location = 0) out vec4 out_color;
  167. layout(location = 0) in vec2 in_uv;
  168. layout(ANKI_TEX_BINDING(0, 0)) uniform sampler2D u_tex0;
  169. layout(ANKI_TEX_BINDING(0, 1)) uniform sampler2D u_tex1;
  170. void main()
  171. {
  172. vec2 uv = in_uv;
  173. #ifdef ANKI_VK
  174. uv.y = 1.0 - uv.y;
  175. #endif
  176. float factor = uv.x;
  177. vec3 col0 = texture(u_tex0, uv).rgb;
  178. vec3 col1 = texture(u_tex1, uv).rgb;
  179. out_color = vec4(col1 + col0, 1.0);
  180. })";
  181. static const char* FRAG_SIMPLE_TEX_SRC = R"(
  182. layout (location = 0) out vec4 out_color;
  183. layout(location = 0) in vec2 in_uv;
  184. layout(ANKI_TEX_BINDING(0, 0)) uniform sampler2D u_tex0;
  185. void main()
  186. {
  187. out_color = textureLod(u_tex0, in_uv, 1.0);
  188. })";
  189. static const char* COMP_WRITE_IMAGE_SRC = R"(
  190. layout(ANKI_IMAGE_BINDING(0, 0), rgba8) writeonly uniform image2D u_img;
  191. layout(local_size_x = 1, local_size_y = 1, local_size_z = 1) in;
  192. layout(ANKI_SS_BINDING(1, 0)) buffer ss1_
  193. {
  194. vec4 u_color;
  195. };
  196. void main()
  197. {
  198. imageStore(u_img, ivec2(gl_WorkGroupID.x, gl_WorkGroupID.y), u_color);
  199. })";
  200. static NativeWindow* win = nullptr;
  201. static GrManager* gr = nullptr;
  202. static StagingGpuMemoryManager* stagingMem = nullptr;
  203. #define COMMON_BEGIN() \
  204. stagingMem = new StagingGpuMemoryManager(); \
  205. createGrManager(win, gr); \
  206. ANKI_TEST_EXPECT_NO_ERR(stagingMem->init(gr, Config())); \
  207. {
  208. #define COMMON_END() \
  209. } \
  210. delete stagingMem; \
  211. delete gr; \
  212. delete win; \
  213. win = nullptr; \
  214. gr = nullptr; \
  215. stagingMem = nullptr;
  216. static void* setUniforms(PtrSize size, CommandBufferPtr& cmdb, U set, U binding)
  217. {
  218. StagingGpuMemoryToken token;
  219. void* ptr = stagingMem->allocateFrame(size, StagingGpuMemoryType::UNIFORM, token);
  220. cmdb->bindUniformBuffer(set, binding, token.m_buffer, token.m_offset, token.m_range);
  221. return ptr;
  222. }
  223. static void* setStorage(PtrSize size, CommandBufferPtr& cmdb, U set, U binding)
  224. {
  225. StagingGpuMemoryToken token;
  226. void* ptr = stagingMem->allocateFrame(size, StagingGpuMemoryType::STORAGE, token);
  227. cmdb->bindStorageBuffer(set, binding, token.m_buffer, token.m_offset, token.m_range);
  228. return ptr;
  229. }
  230. #define SET_UNIFORMS(type_, size_, cmdb_, set_, binding_) static_cast<type_>(setUniforms(size_, cmdb_, set_, binding_))
  231. #define SET_STORAGE(type_, size_, cmdb_, set_, binding_) static_cast<type_>(setStorage(size_, cmdb_, set_, binding_))
  232. #define UPLOAD_TEX_SURFACE(cmdb_, tex_, surf_, ptr_, size_) \
  233. do \
  234. { \
  235. StagingGpuMemoryToken token; \
  236. void* f = stagingMem->allocateFrame(size_, StagingGpuMemoryType::TRANSFER, token); \
  237. memcpy(f, ptr_, size_); \
  238. cmdb_->copyBufferToTextureSurface(token.m_buffer, token.m_offset, token.m_range, tex_, surf_); \
  239. } while(0)
  240. #define UPLOAD_TEX_VOL(cmdb_, tex_, vol_, ptr_, size_) \
  241. do \
  242. { \
  243. StagingGpuMemoryToken token; \
  244. void* f = stagingMem->allocateFrame(size_, StagingGpuMemoryType::TRANSFER, token); \
  245. memcpy(f, ptr_, size_); \
  246. cmdb_->copyBufferToTextureVolume(token.m_buffer, token.m_offset, token.m_range, tex_, vol_); \
  247. } while(0)
  248. const PixelFormat DS_FORMAT = PixelFormat(ComponentFormat::D24S8, TransformFormat::UNORM);
  249. static NativeWindow* createWindow()
  250. {
  251. HeapAllocator<U8> alloc(allocAligned, nullptr);
  252. NativeWindowInitInfo inf;
  253. inf.m_width = WIDTH;
  254. inf.m_height = HEIGHT;
  255. NativeWindow* win = new NativeWindow();
  256. ANKI_TEST_EXPECT_NO_ERR(win->init(inf, alloc));
  257. return win;
  258. }
  259. static void createGrManager(NativeWindow*& win, GrManager*& gr)
  260. {
  261. win = createWindow();
  262. gr = new GrManager();
  263. Config cfg;
  264. cfg.set("debugContext", true);
  265. cfg.set("vsync", false);
  266. GrManagerInitInfo inf;
  267. inf.m_allocCallback = allocAligned;
  268. inf.m_cacheDirectory = "./";
  269. inf.m_config = &cfg;
  270. inf.m_window = win;
  271. ANKI_TEST_EXPECT_NO_ERR(gr->init(inf));
  272. }
  273. static ShaderProgramPtr createProgram(CString vertSrc, CString fragSrc, GrManager& gr)
  274. {
  275. ShaderPtr vert = gr.newInstance<Shader>(ShaderType::VERTEX, vertSrc);
  276. ShaderPtr frag = gr.newInstance<Shader>(ShaderType::FRAGMENT, fragSrc);
  277. return gr.newInstance<ShaderProgram>(vert, frag);
  278. }
  279. static FramebufferPtr createDefaultFb(GrManager& gr)
  280. {
  281. FramebufferInitInfo fbinit;
  282. fbinit.m_colorAttachmentCount = 1;
  283. fbinit.m_colorAttachments[0].m_clearValue.m_colorf = {{1.0, 0.0, 1.0, 1.0}};
  284. return gr.newInstance<Framebuffer>(fbinit);
  285. }
  286. static void createCube(GrManager& gr, BufferPtr& verts, BufferPtr& indices)
  287. {
  288. static const Array<F32, 8 * 3> pos = {
  289. {1, 1, 1, -1, 1, 1, -1, -1, 1, 1, -1, 1, 1, 1, -1, -1, 1, -1, -1, -1, -1, 1, -1, -1}};
  290. static const Array<U16, 6 * 2 * 3> idx = {
  291. {0, 1, 3, 3, 1, 2, 1, 5, 6, 1, 6, 2, 7, 4, 0, 7, 0, 3, 6, 5, 7, 7, 5, 4, 0, 4, 5, 0, 5, 1, 3, 2, 6, 3, 6, 7}};
  292. verts = gr.newInstance<Buffer>(sizeof(pos), BufferUsageBit::VERTEX, BufferMapAccessBit::WRITE);
  293. void* mapped = verts->map(0, sizeof(pos), BufferMapAccessBit::WRITE);
  294. memcpy(mapped, &pos[0], sizeof(pos));
  295. verts->unmap();
  296. indices = gr.newInstance<Buffer>(sizeof(idx), BufferUsageBit::INDEX, BufferMapAccessBit::WRITE);
  297. mapped = indices->map(0, sizeof(idx), BufferMapAccessBit::WRITE);
  298. memcpy(mapped, &idx[0], sizeof(idx));
  299. indices->unmap();
  300. }
  301. ANKI_TEST(Gr, GrManager)
  302. {
  303. COMMON_BEGIN()
  304. COMMON_END()
  305. }
  306. ANKI_TEST(Gr, Shader)
  307. {
  308. COMMON_BEGIN()
  309. ShaderPtr shader = gr->newInstance<Shader>(ShaderType::FRAGMENT, FRAG_MRT_SRC);
  310. COMMON_END()
  311. }
  312. ANKI_TEST(Gr, ShaderProgram)
  313. {
  314. COMMON_BEGIN()
  315. ShaderProgramPtr ppline = createProgram(VERT_SRC, FRAG_SRC, *gr);
  316. COMMON_END()
  317. }
  318. ANKI_TEST(Gr, ClearScreen)
  319. {
  320. COMMON_BEGIN()
  321. FramebufferPtr fb = createDefaultFb(*gr);
  322. U iterations = 100;
  323. while(iterations--)
  324. {
  325. HighRezTimer timer;
  326. timer.start();
  327. gr->beginFrame();
  328. CommandBufferInitInfo cinit;
  329. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK | CommandBufferFlag::SMALL_BATCH;
  330. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cinit);
  331. cmdb->beginRenderPass(fb);
  332. cmdb->endRenderPass();
  333. cmdb->flush();
  334. gr->swapBuffers();
  335. timer.stop();
  336. const F32 TICK = 1.0 / 30.0;
  337. if(timer.getElapsedTime() < TICK)
  338. {
  339. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  340. }
  341. }
  342. COMMON_END()
  343. }
  344. ANKI_TEST(Gr, SimpleDrawcall)
  345. {
  346. COMMON_BEGIN()
  347. ANKI_TEST_LOGI("Expect to see a grey triangle appearing in the 4 corners");
  348. ShaderProgramPtr prog = createProgram(VERT_SRC, FRAG_SRC, *gr);
  349. FramebufferPtr fb = createDefaultFb(*gr);
  350. static const Array2d<U, 4, 4> VIEWPORTS = {{{{0, 0, WIDTH / 2, HEIGHT / 2}},
  351. {{WIDTH / 2, 0, WIDTH, HEIGHT / 2}},
  352. {{WIDTH / 2, HEIGHT / 2, WIDTH, HEIGHT}},
  353. {{0, HEIGHT / 2, WIDTH / 2, HEIGHT}}}};
  354. const U ITERATIONS = 200;
  355. for(U i = 0; i < ITERATIONS; ++i)
  356. {
  357. HighRezTimer timer;
  358. timer.start();
  359. gr->beginFrame();
  360. CommandBufferInitInfo cinit;
  361. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK;
  362. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cinit);
  363. auto vp = VIEWPORTS[(i / 30) % 4];
  364. cmdb->setViewport(vp[0], vp[1], vp[2], vp[3]);
  365. cmdb->bindShaderProgram(prog);
  366. cmdb->beginRenderPass(fb);
  367. cmdb->drawArrays(PrimitiveTopology::TRIANGLES, 3);
  368. cmdb->endRenderPass();
  369. cmdb->flush();
  370. gr->swapBuffers();
  371. timer.stop();
  372. const F32 TICK = 1.0 / 30.0;
  373. if(timer.getElapsedTime() < TICK)
  374. {
  375. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  376. }
  377. }
  378. COMMON_END()
  379. }
  380. ANKI_TEST(Gr, Buffer)
  381. {
  382. COMMON_BEGIN()
  383. BufferPtr a = gr->newInstance<Buffer>(512, BufferUsageBit::UNIFORM_ALL, BufferMapAccessBit::NONE);
  384. BufferPtr b =
  385. gr->newInstance<Buffer>(64, BufferUsageBit::STORAGE_ALL, BufferMapAccessBit::WRITE | BufferMapAccessBit::READ);
  386. void* ptr = b->map(0, 64, BufferMapAccessBit::WRITE);
  387. ANKI_TEST_EXPECT_NEQ(ptr, nullptr);
  388. U8 ptr2[64];
  389. memset(ptr, 0xCC, 64);
  390. memset(ptr2, 0xCC, 64);
  391. b->unmap();
  392. ptr = b->map(0, 64, BufferMapAccessBit::READ);
  393. ANKI_TEST_EXPECT_NEQ(ptr, nullptr);
  394. ANKI_TEST_EXPECT_EQ(memcmp(ptr, ptr2, 64), 0);
  395. b->unmap();
  396. COMMON_END()
  397. }
  398. ANKI_TEST(Gr, DrawWithUniforms)
  399. {
  400. COMMON_BEGIN()
  401. // A non-uploaded buffer
  402. BufferPtr b = gr->newInstance<Buffer>(sizeof(Vec4) * 3, BufferUsageBit::UNIFORM_ALL, BufferMapAccessBit::WRITE);
  403. Vec4* ptr = static_cast<Vec4*>(b->map(0, sizeof(Vec4) * 3, BufferMapAccessBit::WRITE));
  404. ANKI_TEST_EXPECT_NEQ(ptr, nullptr);
  405. ptr[0] = Vec4(1.0, 0.0, 0.0, 0.0);
  406. ptr[1] = Vec4(0.0, 1.0, 0.0, 0.0);
  407. ptr[2] = Vec4(0.0, 0.0, 1.0, 0.0);
  408. b->unmap();
  409. // Progm
  410. ShaderProgramPtr prog = createProgram(VERT_UBO_SRC, FRAG_UBO_SRC, *gr);
  411. // FB
  412. FramebufferPtr fb = createDefaultFb(*gr);
  413. const U ITERATION_COUNT = 100;
  414. U iterations = ITERATION_COUNT;
  415. while(iterations--)
  416. {
  417. HighRezTimer timer;
  418. timer.start();
  419. gr->beginFrame();
  420. CommandBufferInitInfo cinit;
  421. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK;
  422. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cinit);
  423. cmdb->setViewport(0, 0, WIDTH, HEIGHT);
  424. cmdb->bindShaderProgram(prog);
  425. cmdb->beginRenderPass(fb);
  426. cmdb->bindUniformBuffer(0, 0, b, 0, MAX_PTR_SIZE);
  427. // Uploaded buffer
  428. Vec4* rotMat = SET_UNIFORMS(Vec4*, sizeof(Vec4), cmdb, 0, 1);
  429. F32 angle = toRad(360.0f / ITERATION_COUNT * iterations);
  430. (*rotMat)[0] = cos(angle);
  431. (*rotMat)[1] = -sin(angle);
  432. (*rotMat)[2] = sin(angle);
  433. (*rotMat)[3] = cos(angle);
  434. cmdb->drawArrays(PrimitiveTopology::TRIANGLES, 3);
  435. cmdb->endRenderPass();
  436. cmdb->flush();
  437. gr->swapBuffers();
  438. timer.stop();
  439. const F32 TICK = 1.0 / 30.0;
  440. if(timer.getElapsedTime() < TICK)
  441. {
  442. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  443. }
  444. }
  445. COMMON_END()
  446. }
  447. ANKI_TEST(Gr, DrawWithVertex)
  448. {
  449. COMMON_BEGIN()
  450. // The buffers
  451. struct Vert
  452. {
  453. Vec3 m_pos;
  454. Array<U8, 4> m_color;
  455. };
  456. static_assert(sizeof(Vert) == sizeof(Vec4), "See file");
  457. BufferPtr b = gr->newInstance<Buffer>(sizeof(Vert) * 3, BufferUsageBit::VERTEX, BufferMapAccessBit::WRITE);
  458. Vert* ptr = static_cast<Vert*>(b->map(0, sizeof(Vert) * 3, BufferMapAccessBit::WRITE));
  459. ANKI_TEST_EXPECT_NEQ(ptr, nullptr);
  460. ptr[0].m_pos = Vec3(-1.0, 1.0, 0.0);
  461. ptr[1].m_pos = Vec3(0.0, -1.0, 0.0);
  462. ptr[2].m_pos = Vec3(1.0, 1.0, 0.0);
  463. ptr[0].m_color = {{255, 0, 0}};
  464. ptr[1].m_color = {{0, 255, 0}};
  465. ptr[2].m_color = {{0, 0, 255}};
  466. b->unmap();
  467. BufferPtr c = gr->newInstance<Buffer>(sizeof(Vec3) * 3, BufferUsageBit::VERTEX, BufferMapAccessBit::WRITE);
  468. Vec3* otherColor = static_cast<Vec3*>(c->map(0, sizeof(Vec3) * 3, BufferMapAccessBit::WRITE));
  469. otherColor[0] = Vec3(0.0, 1.0, 1.0);
  470. otherColor[1] = Vec3(1.0, 0.0, 1.0);
  471. otherColor[2] = Vec3(1.0, 1.0, 0.0);
  472. c->unmap();
  473. // Prog
  474. ShaderProgramPtr prog = createProgram(VERT_INP_SRC, FRAG_INP_SRC, *gr);
  475. // FB
  476. FramebufferPtr fb = createDefaultFb(*gr);
  477. U iterations = 100;
  478. while(iterations--)
  479. {
  480. HighRezTimer timer;
  481. timer.start();
  482. gr->beginFrame();
  483. CommandBufferInitInfo cinit;
  484. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK;
  485. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cinit);
  486. cmdb->bindVertexBuffer(0, b, 0, sizeof(Vert));
  487. cmdb->bindVertexBuffer(1, c, 0, sizeof(Vec3));
  488. cmdb->setVertexAttribute(0, 0, PixelFormat(ComponentFormat::R32G32B32, TransformFormat::FLOAT), 0);
  489. cmdb->setVertexAttribute(1, 0, PixelFormat(ComponentFormat::R8G8B8, TransformFormat::UNORM), sizeof(Vec3));
  490. cmdb->setVertexAttribute(2, 1, PixelFormat(ComponentFormat::R32G32B32, TransformFormat::FLOAT), 0);
  491. cmdb->setViewport(0, 0, WIDTH, HEIGHT);
  492. cmdb->setPolygonOffset(0.0, 0.0);
  493. cmdb->bindShaderProgram(prog);
  494. cmdb->beginRenderPass(fb);
  495. cmdb->drawArrays(PrimitiveTopology::TRIANGLES, 3);
  496. cmdb->endRenderPass();
  497. cmdb->flush();
  498. gr->swapBuffers();
  499. timer.stop();
  500. const F32 TICK = 1.0 / 30.0;
  501. if(timer.getElapsedTime() < TICK)
  502. {
  503. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  504. }
  505. }
  506. COMMON_END()
  507. }
  508. ANKI_TEST(Gr, Sampler)
  509. {
  510. COMMON_BEGIN()
  511. SamplerInitInfo init;
  512. SamplerPtr b = gr->newInstance<Sampler>(init);
  513. COMMON_END()
  514. }
  515. ANKI_TEST(Gr, Texture)
  516. {
  517. COMMON_BEGIN()
  518. TextureInitInfo init;
  519. init.m_depth = 1;
  520. init.m_format = PixelFormat(ComponentFormat::R8G8B8, TransformFormat::UNORM);
  521. init.m_usage = TextureUsageBit::SAMPLED_FRAGMENT;
  522. init.m_height = 4;
  523. init.m_width = 4;
  524. init.m_mipmapsCount = 2;
  525. init.m_depth = 1;
  526. init.m_layerCount = 1;
  527. init.m_samples = 1;
  528. init.m_sampling.m_minMagFilter = SamplingFilter::LINEAR;
  529. init.m_sampling.m_mipmapFilter = SamplingFilter::LINEAR;
  530. init.m_type = TextureType::_2D;
  531. TexturePtr b = gr->newInstance<Texture>(init);
  532. COMMON_END()
  533. }
  534. ANKI_TEST(Gr, DrawWithTexture)
  535. {
  536. COMMON_BEGIN()
  537. //
  538. // Create texture A
  539. //
  540. TextureInitInfo init;
  541. init.m_depth = 1;
  542. init.m_format = PixelFormat(ComponentFormat::R8G8B8, TransformFormat::UNORM);
  543. init.m_usage = TextureUsageBit::SAMPLED_FRAGMENT | TextureUsageBit::TRANSFER_DESTINATION;
  544. init.m_initialUsage = TextureUsageBit::SAMPLED_FRAGMENT;
  545. init.m_height = 2;
  546. init.m_width = 2;
  547. init.m_mipmapsCount = 2;
  548. init.m_samples = 1;
  549. init.m_depth = 1;
  550. init.m_layerCount = 1;
  551. init.m_sampling.m_repeat = false;
  552. init.m_sampling.m_minMagFilter = SamplingFilter::NEAREST;
  553. init.m_sampling.m_mipmapFilter = SamplingFilter::LINEAR;
  554. init.m_type = TextureType::_2D;
  555. TexturePtr a = gr->newInstance<Texture>(init);
  556. //
  557. // Create texture B
  558. //
  559. init.m_width = 4;
  560. init.m_height = 4;
  561. init.m_mipmapsCount = 3;
  562. init.m_usage =
  563. TextureUsageBit::SAMPLED_FRAGMENT | TextureUsageBit::TRANSFER_DESTINATION | TextureUsageBit::GENERATE_MIPMAPS;
  564. init.m_initialUsage = TextureUsageBit::NONE;
  565. TexturePtr b = gr->newInstance<Texture>(init);
  566. //
  567. // Upload all textures
  568. //
  569. Array<U8, 2 * 2 * 3> mip0 = {{255, 0, 0, 0, 255, 0, 0, 0, 255, 255, 0, 255}};
  570. Array<U8, 3> mip1 = {{128, 128, 128}};
  571. Array<U8, 4 * 4 * 3> bmip0 = {{255,
  572. 0,
  573. 0,
  574. 0,
  575. 255,
  576. 0,
  577. 0,
  578. 0,
  579. 255,
  580. 255,
  581. 255,
  582. 0,
  583. 255,
  584. 0,
  585. 255,
  586. 0,
  587. 255,
  588. 255,
  589. 255,
  590. 255,
  591. 255,
  592. 128,
  593. 0,
  594. 0,
  595. 0,
  596. 128,
  597. 0,
  598. 0,
  599. 0,
  600. 128,
  601. 128,
  602. 128,
  603. 0,
  604. 128,
  605. 0,
  606. 128,
  607. 0,
  608. 128,
  609. 128,
  610. 128,
  611. 128,
  612. 128,
  613. 255,
  614. 128,
  615. 0,
  616. 0,
  617. 128,
  618. 255}};
  619. CommandBufferInitInfo cmdbinit;
  620. cmdbinit.m_flags = CommandBufferFlag::TRANSFER_WORK;
  621. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cmdbinit);
  622. // Set barriers
  623. cmdb->setTextureSurfaceBarrier(
  624. a, TextureUsageBit::SAMPLED_FRAGMENT, TextureUsageBit::TRANSFER_DESTINATION, TextureSurfaceInfo(0, 0, 0, 0));
  625. cmdb->setTextureSurfaceBarrier(
  626. a, TextureUsageBit::SAMPLED_FRAGMENT, TextureUsageBit::TRANSFER_DESTINATION, TextureSurfaceInfo(1, 0, 0, 0));
  627. cmdb->setTextureSurfaceBarrier(
  628. b, TextureUsageBit::NONE, TextureUsageBit::TRANSFER_DESTINATION, TextureSurfaceInfo(0, 0, 0, 0));
  629. UPLOAD_TEX_SURFACE(cmdb, a, TextureSurfaceInfo(0, 0, 0, 0), &mip0[0], sizeof(mip0));
  630. UPLOAD_TEX_SURFACE(cmdb, a, TextureSurfaceInfo(1, 0, 0, 0), &mip1[0], sizeof(mip1));
  631. UPLOAD_TEX_SURFACE(cmdb, b, TextureSurfaceInfo(0, 0, 0, 0), &bmip0[0], sizeof(bmip0));
  632. // Gen mips
  633. cmdb->setTextureSurfaceBarrier(
  634. b, TextureUsageBit::TRANSFER_DESTINATION, TextureUsageBit::GENERATE_MIPMAPS, TextureSurfaceInfo(0, 0, 0, 0));
  635. cmdb->generateMipmaps2d(b, 0, 0);
  636. // Set barriers
  637. cmdb->setTextureSurfaceBarrier(
  638. a, TextureUsageBit::TRANSFER_DESTINATION, TextureUsageBit::SAMPLED_FRAGMENT, TextureSurfaceInfo(0, 0, 0, 0));
  639. cmdb->setTextureSurfaceBarrier(
  640. a, TextureUsageBit::TRANSFER_DESTINATION, TextureUsageBit::SAMPLED_FRAGMENT, TextureSurfaceInfo(1, 0, 0, 0));
  641. for(U i = 0; i < 3; ++i)
  642. {
  643. cmdb->setTextureSurfaceBarrier(
  644. b, TextureUsageBit::GENERATE_MIPMAPS, TextureUsageBit::SAMPLED_FRAGMENT, TextureSurfaceInfo(i, 0, 0, 0));
  645. }
  646. cmdb->flush();
  647. //
  648. // Create prog
  649. //
  650. ShaderProgramPtr prog = createProgram(VERT_QUAD_SRC, FRAG_TEX_SRC, *gr);
  651. //
  652. // Create FB
  653. //
  654. FramebufferPtr fb = createDefaultFb(*gr);
  655. //
  656. // Draw
  657. //
  658. const U ITERATION_COUNT = 200;
  659. U iterations = ITERATION_COUNT;
  660. while(iterations--)
  661. {
  662. HighRezTimer timer;
  663. timer.start();
  664. gr->beginFrame();
  665. CommandBufferInitInfo cinit;
  666. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK;
  667. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cinit);
  668. cmdb->setViewport(0, 0, WIDTH, HEIGHT);
  669. cmdb->bindShaderProgram(prog);
  670. cmdb->beginRenderPass(fb);
  671. for(U i = 0; i < 2; ++i)
  672. {
  673. cmdb->informTextureSurfaceCurrentUsage(
  674. a, TextureSurfaceInfo(i, 0, 0, 0), TextureUsageBit::SAMPLED_FRAGMENT);
  675. cmdb->informTextureSurfaceCurrentUsage(
  676. b, TextureSurfaceInfo(i, 0, 0, 0), TextureUsageBit::SAMPLED_FRAGMENT);
  677. }
  678. cmdb->informTextureSurfaceCurrentUsage(b, TextureSurfaceInfo(2, 0, 0, 0), TextureUsageBit::SAMPLED_FRAGMENT);
  679. cmdb->bindTexture(0, 0, a);
  680. cmdb->bindTexture(0, 1, b);
  681. cmdb->drawArrays(PrimitiveTopology::TRIANGLES, 6);
  682. cmdb->endRenderPass();
  683. cmdb->flush();
  684. gr->swapBuffers();
  685. timer.stop();
  686. const F32 TICK = 1.0 / 30.0;
  687. if(timer.getElapsedTime() < TICK)
  688. {
  689. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  690. }
  691. }
  692. COMMON_END()
  693. }
  694. static void drawOffscreenDrawcalls(GrManager& gr,
  695. ShaderProgramPtr prog,
  696. CommandBufferPtr cmdb,
  697. U viewPortSize,
  698. BufferPtr indexBuff,
  699. BufferPtr vertBuff)
  700. {
  701. static F32 ang = -2.5f;
  702. ang += toRad(2.5f);
  703. Mat4 viewMat(Vec4(0.0, 0.0, 5.0, 1.0), Mat3::getIdentity(), 1.0f);
  704. viewMat.invert();
  705. Mat4 projMat = Mat4::calculatePerspectiveProjectionMatrix(toRad(60.0), toRad(60.0), 0.1f, 100.0f);
  706. Mat4 modelMat(Vec4(-0.5, -0.5, 0.0, 1.0), Mat3(Euler(ang, ang / 2.0f, ang / 3.0f)), 1.0f);
  707. Mat4* mvp = SET_UNIFORMS(Mat4*, sizeof(*mvp), cmdb, 0, 0);
  708. *mvp = projMat * viewMat * modelMat;
  709. Vec4* color = SET_UNIFORMS(Vec4*, sizeof(*color) * 2, cmdb, 0, 1);
  710. *color++ = Vec4(1.0, 0.0, 0.0, 0.0);
  711. *color = Vec4(0.0, 1.0, 0.0, 0.0);
  712. cmdb->bindVertexBuffer(0, vertBuff, 0, sizeof(Vec3));
  713. cmdb->setVertexAttribute(0, 0, PixelFormat(ComponentFormat::R32G32B32, TransformFormat::FLOAT), 0);
  714. cmdb->bindShaderProgram(prog);
  715. cmdb->bindIndexBuffer(indexBuff, 0, IndexType::U16);
  716. cmdb->setViewport(0, 0, viewPortSize, viewPortSize);
  717. cmdb->drawElements(PrimitiveTopology::TRIANGLES, 6 * 2 * 3);
  718. // 2nd draw
  719. modelMat = Mat4(Vec4(0.5, 0.5, 0.0, 1.0), Mat3(Euler(ang * 2.0, ang, ang / 3.0f * 2.0)), 1.0f);
  720. mvp = SET_UNIFORMS(Mat4*, sizeof(*mvp), cmdb, 0, 0);
  721. *mvp = projMat * viewMat * modelMat;
  722. color = SET_UNIFORMS(Vec4*, sizeof(*color) * 2, cmdb, 0, 1);
  723. *color++ = Vec4(0.0, 0.0, 1.0, 0.0);
  724. *color = Vec4(0.0, 1.0, 1.0, 0.0);
  725. cmdb->drawElements(PrimitiveTopology::TRIANGLES, 6 * 2 * 3);
  726. }
  727. static void drawOffscreen(GrManager& gr, Bool useSecondLevel)
  728. {
  729. //
  730. // Create textures
  731. //
  732. const PixelFormat COL_FORMAT = PixelFormat(ComponentFormat::R8G8B8A8, TransformFormat::UNORM);
  733. const U TEX_SIZE = 256;
  734. TextureInitInfo init;
  735. init.m_depth = 1;
  736. init.m_format = COL_FORMAT;
  737. init.m_usage = TextureUsageBit::SAMPLED_FRAGMENT | TextureUsageBit::FRAMEBUFFER_ATTACHMENT_READ_WRITE;
  738. init.m_height = TEX_SIZE;
  739. init.m_width = TEX_SIZE;
  740. init.m_mipmapsCount = 1;
  741. init.m_depth = 1;
  742. init.m_layerCount = 1;
  743. init.m_samples = 1;
  744. init.m_sampling.m_minMagFilter = SamplingFilter::LINEAR;
  745. init.m_sampling.m_mipmapFilter = SamplingFilter::LINEAR;
  746. init.m_type = TextureType::_2D;
  747. TexturePtr col0 = gr.newInstance<Texture>(init);
  748. TexturePtr col1 = gr.newInstance<Texture>(init);
  749. init.m_format = DS_FORMAT;
  750. TexturePtr dp = gr.newInstance<Texture>(init);
  751. //
  752. // Create FB
  753. //
  754. FramebufferInitInfo fbinit;
  755. fbinit.m_colorAttachmentCount = 2;
  756. fbinit.m_colorAttachments[0].m_texture = col0;
  757. fbinit.m_colorAttachments[0].m_clearValue.m_colorf = {{0.1, 0.0, 0.0, 0.0}};
  758. fbinit.m_colorAttachments[1].m_texture = col1;
  759. fbinit.m_colorAttachments[1].m_clearValue.m_colorf = {{0.0, 0.1, 0.0, 0.0}};
  760. fbinit.m_depthStencilAttachment.m_texture = dp;
  761. fbinit.m_depthStencilAttachment.m_aspect = DepthStencilAspectBit::DEPTH;
  762. fbinit.m_depthStencilAttachment.m_clearValue.m_depthStencil.m_depth = 1.0;
  763. FramebufferPtr fb = gr.newInstance<Framebuffer>(fbinit);
  764. //
  765. // Create default FB
  766. //
  767. FramebufferPtr dfb = createDefaultFb(gr);
  768. //
  769. // Create buffs
  770. //
  771. BufferPtr verts, indices;
  772. createCube(gr, verts, indices);
  773. //
  774. // Create progs
  775. //
  776. ShaderProgramPtr prog = createProgram(VERT_MRT_SRC, FRAG_MRT_SRC, gr);
  777. ShaderProgramPtr resolveProg = createProgram(VERT_QUAD_SRC, FRAG_MRT2_SRC, gr);
  778. //
  779. // Draw
  780. //
  781. const U ITERATION_COUNT = 200;
  782. U iterations = ITERATION_COUNT;
  783. while(iterations--)
  784. {
  785. HighRezTimer timer;
  786. timer.start();
  787. gr.beginFrame();
  788. CommandBufferInitInfo cinit;
  789. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK;
  790. CommandBufferPtr cmdb = gr.newInstance<CommandBuffer>(cinit);
  791. cmdb->setPolygonOffset(0.0, 0.0);
  792. cmdb->setTextureSurfaceBarrier(
  793. col0, TextureUsageBit::NONE, TextureUsageBit::FRAMEBUFFER_ATTACHMENT_WRITE, TextureSurfaceInfo(0, 0, 0, 0));
  794. cmdb->setTextureSurfaceBarrier(
  795. col1, TextureUsageBit::NONE, TextureUsageBit::FRAMEBUFFER_ATTACHMENT_WRITE, TextureSurfaceInfo(0, 0, 0, 0));
  796. cmdb->setTextureSurfaceBarrier(dp,
  797. TextureUsageBit::NONE,
  798. TextureUsageBit::FRAMEBUFFER_ATTACHMENT_READ_WRITE,
  799. TextureSurfaceInfo(0, 0, 0, 0));
  800. cmdb->beginRenderPass(fb);
  801. if(!useSecondLevel)
  802. {
  803. drawOffscreenDrawcalls(gr, prog, cmdb, TEX_SIZE, indices, verts);
  804. }
  805. else
  806. {
  807. CommandBufferInitInfo cinit;
  808. cinit.m_flags = CommandBufferFlag::SECOND_LEVEL | CommandBufferFlag::GRAPHICS_WORK;
  809. cinit.m_framebuffer = fb;
  810. CommandBufferPtr cmdb2 = gr.newInstance<CommandBuffer>(cinit);
  811. cmdb2->informTextureSurfaceCurrentUsage(
  812. col0, TextureSurfaceInfo(0, 0, 0, 0), TextureUsageBit::FRAMEBUFFER_ATTACHMENT_WRITE);
  813. cmdb2->informTextureSurfaceCurrentUsage(
  814. col1, TextureSurfaceInfo(0, 0, 0, 0), TextureUsageBit::FRAMEBUFFER_ATTACHMENT_WRITE);
  815. cmdb2->informTextureSurfaceCurrentUsage(
  816. dp, TextureSurfaceInfo(0, 0, 0, 0), TextureUsageBit::FRAMEBUFFER_ATTACHMENT_READ_WRITE);
  817. drawOffscreenDrawcalls(gr, prog, cmdb2, TEX_SIZE, indices, verts);
  818. cmdb2->flush();
  819. cmdb->pushSecondLevelCommandBuffer(cmdb2);
  820. }
  821. cmdb->endRenderPass();
  822. cmdb->setTextureSurfaceBarrier(col0,
  823. TextureUsageBit::FRAMEBUFFER_ATTACHMENT_WRITE,
  824. TextureUsageBit::SAMPLED_FRAGMENT,
  825. TextureSurfaceInfo(0, 0, 0, 0));
  826. cmdb->setTextureSurfaceBarrier(col1,
  827. TextureUsageBit::FRAMEBUFFER_ATTACHMENT_WRITE,
  828. TextureUsageBit::SAMPLED_FRAGMENT,
  829. TextureSurfaceInfo(0, 0, 0, 0));
  830. cmdb->setTextureSurfaceBarrier(dp,
  831. TextureUsageBit::FRAMEBUFFER_ATTACHMENT_READ_WRITE,
  832. TextureUsageBit::SAMPLED_FRAGMENT,
  833. TextureSurfaceInfo(0, 0, 0, 0));
  834. // Draw quad
  835. cmdb->beginRenderPass(dfb);
  836. cmdb->bindShaderProgram(resolveProg);
  837. cmdb->setViewport(0, 0, WIDTH, HEIGHT);
  838. cmdb->bindTexture(0, 0, col0);
  839. cmdb->bindTexture(0, 1, col1);
  840. cmdb->drawArrays(PrimitiveTopology::TRIANGLES, 6);
  841. cmdb->endRenderPass();
  842. cmdb->flush();
  843. // End
  844. gr.swapBuffers();
  845. timer.stop();
  846. const F32 TICK = 1.0 / 30.0;
  847. if(timer.getElapsedTime() < TICK)
  848. {
  849. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  850. }
  851. }
  852. }
  853. ANKI_TEST(Gr, DrawOffscreen)
  854. {
  855. COMMON_BEGIN()
  856. drawOffscreen(*gr, false);
  857. COMMON_END()
  858. }
  859. ANKI_TEST(Gr, DrawWithSecondLevel)
  860. {
  861. COMMON_BEGIN()
  862. drawOffscreen(*gr, true);
  863. COMMON_END()
  864. }
  865. ANKI_TEST(Gr, ImageLoadStore)
  866. {
  867. COMMON_BEGIN()
  868. TextureInitInfo init;
  869. init.m_width = init.m_height = 4;
  870. init.m_mipmapsCount = 2;
  871. init.m_usage = TextureUsageBit::CLEAR | TextureUsageBit::SAMPLED_ALL | TextureUsageBit::IMAGE_COMPUTE_WRITE;
  872. init.m_type = TextureType::_2D;
  873. init.m_format = PixelFormat(ComponentFormat::R8G8B8A8, TransformFormat::UNORM);
  874. init.m_sampling.m_mipmapFilter = SamplingFilter::LINEAR;
  875. TexturePtr tex = gr->newInstance<Texture>(init);
  876. // Prog
  877. ShaderProgramPtr prog = createProgram(VERT_QUAD_SRC, FRAG_SIMPLE_TEX_SRC, *gr);
  878. // Create shader & compute prog
  879. ShaderPtr shader = gr->newInstance<Shader>(ShaderType::COMPUTE, COMP_WRITE_IMAGE_SRC);
  880. ShaderProgramPtr compProg = gr->newInstance<ShaderProgram>(shader);
  881. // FB
  882. FramebufferPtr dfb = createDefaultFb(*gr);
  883. // Write texture data
  884. CommandBufferInitInfo cmdbinit;
  885. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cmdbinit);
  886. cmdb->setTextureSurfaceBarrier(tex, TextureUsageBit::NONE, TextureUsageBit::CLEAR, TextureSurfaceInfo(0, 0, 0, 0));
  887. ClearValue clear;
  888. clear.m_colorf = {{0.0, 1.0, 0.0, 1.0}};
  889. cmdb->clearTextureSurface(tex, TextureSurfaceInfo(0, 0, 0, 0), clear);
  890. cmdb->setTextureSurfaceBarrier(
  891. tex, TextureUsageBit::CLEAR, TextureUsageBit::SAMPLED_FRAGMENT, TextureSurfaceInfo(0, 0, 0, 0));
  892. cmdb->setTextureSurfaceBarrier(tex, TextureUsageBit::NONE, TextureUsageBit::CLEAR, TextureSurfaceInfo(1, 0, 0, 0));
  893. clear.m_colorf = {{0.0, 0.0, 1.0, 1.0}};
  894. cmdb->clearTextureSurface(tex, TextureSurfaceInfo(1, 0, 0, 0), clear);
  895. cmdb->setTextureSurfaceBarrier(
  896. tex, TextureUsageBit::CLEAR, TextureUsageBit::IMAGE_COMPUTE_WRITE, TextureSurfaceInfo(1, 0, 0, 0));
  897. cmdb->flush();
  898. const U ITERATION_COUNT = 100;
  899. U iterations = ITERATION_COUNT;
  900. while(iterations--)
  901. {
  902. HighRezTimer timer;
  903. timer.start();
  904. gr->beginFrame();
  905. CommandBufferInitInfo cinit;
  906. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK | CommandBufferFlag::COMPUTE_WORK;
  907. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cinit);
  908. // Write image
  909. Vec4* col = SET_STORAGE(Vec4*, sizeof(*col), cmdb, 1, 0);
  910. *col = Vec4(iterations / F32(ITERATION_COUNT));
  911. cmdb->setTextureSurfaceBarrier(
  912. tex, TextureUsageBit::NONE, TextureUsageBit::IMAGE_COMPUTE_WRITE, TextureSurfaceInfo(1, 0, 0, 0));
  913. cmdb->bindShaderProgram(compProg);
  914. cmdb->bindImage(0, 0, tex, 1);
  915. cmdb->dispatchCompute(WIDTH / 2, HEIGHT / 2, 1);
  916. cmdb->setTextureSurfaceBarrier(tex,
  917. TextureUsageBit::IMAGE_COMPUTE_WRITE,
  918. TextureUsageBit::SAMPLED_FRAGMENT,
  919. TextureSurfaceInfo(1, 0, 0, 0));
  920. // Present image
  921. cmdb->setViewport(0, 0, WIDTH, HEIGHT);
  922. cmdb->bindShaderProgram(prog);
  923. cmdb->beginRenderPass(dfb);
  924. cmdb->informTextureSurfaceCurrentUsage(tex, TextureSurfaceInfo(0, 0, 0, 0), TextureUsageBit::SAMPLED_FRAGMENT);
  925. cmdb->bindTexture(0, 0, tex);
  926. cmdb->drawArrays(PrimitiveTopology::TRIANGLES, 6);
  927. cmdb->endRenderPass();
  928. cmdb->flush();
  929. // End
  930. gr->swapBuffers();
  931. timer.stop();
  932. const F32 TICK = 1.0 / 30.0;
  933. if(timer.getElapsedTime() < TICK)
  934. {
  935. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  936. }
  937. }
  938. COMMON_END()
  939. }
  940. ANKI_TEST(Gr, 3DTextures)
  941. {
  942. COMMON_BEGIN()
  943. //
  944. // Create texture A
  945. //
  946. TextureInitInfo init;
  947. init.m_depth = 1;
  948. init.m_format = PixelFormat(ComponentFormat::R8G8B8A8, TransformFormat::UNORM);
  949. init.m_usage = TextureUsageBit::SAMPLED_FRAGMENT | TextureUsageBit::TRANSFER_DESTINATION;
  950. init.m_initialUsage = TextureUsageBit::TRANSFER_DESTINATION;
  951. init.m_usageWhenEncountered = TextureUsageBit::SAMPLED_FRAGMENT;
  952. init.m_height = 2;
  953. init.m_width = 2;
  954. init.m_mipmapsCount = 2;
  955. init.m_samples = 1;
  956. init.m_depth = 2;
  957. init.m_layerCount = 1;
  958. init.m_sampling.m_repeat = false;
  959. init.m_sampling.m_minMagFilter = SamplingFilter::NEAREST;
  960. init.m_sampling.m_mipmapFilter = SamplingFilter::NEAREST;
  961. init.m_type = TextureType::_3D;
  962. TexturePtr a = gr->newInstance<Texture>(init);
  963. //
  964. // Upload all textures
  965. //
  966. Array<U8, 2 * 2 * 2 * 4> mip0 = {{255,
  967. 0,
  968. 0,
  969. 0,
  970. 0,
  971. 255,
  972. 0,
  973. 0,
  974. 0,
  975. 0,
  976. 255,
  977. 0,
  978. 255,
  979. 255,
  980. 0,
  981. 0,
  982. 255,
  983. 0,
  984. 255,
  985. 0,
  986. 0,
  987. 255,
  988. 255,
  989. 0,
  990. 255,
  991. 255,
  992. 255,
  993. 0,
  994. 0,
  995. 0,
  996. 0,
  997. 0}};
  998. Array<U8, 4> mip1 = {{128, 128, 128, 0}};
  999. CommandBufferInitInfo cmdbinit;
  1000. cmdbinit.m_flags = CommandBufferFlag::TRANSFER_WORK;
  1001. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cmdbinit);
  1002. cmdb->setTextureVolumeBarrier(
  1003. a, TextureUsageBit::NONE, TextureUsageBit::TRANSFER_DESTINATION, TextureVolumeInfo(0));
  1004. cmdb->setTextureVolumeBarrier(
  1005. a, TextureUsageBit::NONE, TextureUsageBit::TRANSFER_DESTINATION, TextureVolumeInfo(1));
  1006. UPLOAD_TEX_VOL(cmdb, a, TextureVolumeInfo(0), &mip0[0], sizeof(mip0));
  1007. UPLOAD_TEX_VOL(cmdb, a, TextureVolumeInfo(1), &mip1[0], sizeof(mip1));
  1008. cmdb->setTextureVolumeBarrier(
  1009. a, TextureUsageBit::TRANSFER_DESTINATION, TextureUsageBit::SAMPLED_FRAGMENT, TextureVolumeInfo(0));
  1010. cmdb->setTextureVolumeBarrier(
  1011. a, TextureUsageBit::TRANSFER_DESTINATION, TextureUsageBit::SAMPLED_FRAGMENT, TextureVolumeInfo(1));
  1012. cmdb->flush();
  1013. //
  1014. // Rest
  1015. //
  1016. ShaderProgramPtr prog = createProgram(VERT_QUAD_SRC, FRAG_TEX3D_SRC, *gr);
  1017. FramebufferPtr dfb = createDefaultFb(*gr);
  1018. static Array<Vec4, 9> TEX_COORDS_LOD = {{Vec4(0, 0, 0, 0),
  1019. Vec4(1, 0, 0, 0),
  1020. Vec4(0, 1, 0, 0),
  1021. Vec4(1, 1, 0, 0),
  1022. Vec4(0, 0, 1, 0),
  1023. Vec4(1, 0, 1, 0),
  1024. Vec4(0, 1, 1, 0),
  1025. Vec4(1, 1, 1, 0),
  1026. Vec4(0, 0, 0, 1)}};
  1027. const U ITERATION_COUNT = 100;
  1028. U iterations = ITERATION_COUNT;
  1029. while(iterations--)
  1030. {
  1031. HighRezTimer timer;
  1032. timer.start();
  1033. gr->beginFrame();
  1034. CommandBufferInitInfo cinit;
  1035. cinit.m_flags = CommandBufferFlag::GRAPHICS_WORK;
  1036. CommandBufferPtr cmdb = gr->newInstance<CommandBuffer>(cinit);
  1037. cmdb->setViewport(0, 0, WIDTH, HEIGHT);
  1038. cmdb->beginRenderPass(dfb);
  1039. cmdb->bindShaderProgram(prog);
  1040. Vec4* uv = SET_UNIFORMS(Vec4*, sizeof(Vec4), cmdb, 0, 0);
  1041. U idx = (F32(ITERATION_COUNT - iterations - 1) / ITERATION_COUNT) * TEX_COORDS_LOD.getSize();
  1042. *uv = TEX_COORDS_LOD[idx];
  1043. cmdb->bindTexture(0, 0, a);
  1044. cmdb->drawArrays(PrimitiveTopology::TRIANGLES, 6);
  1045. cmdb->endRenderPass();
  1046. cmdb->flush();
  1047. // End
  1048. gr->swapBuffers();
  1049. timer.stop();
  1050. const F32 TICK = 1.0 / 15.0;
  1051. if(timer.getElapsedTime() < TICK)
  1052. {
  1053. HighRezTimer::sleep(TICK - timer.getElapsedTime());
  1054. }
  1055. }
  1056. COMMON_END()
  1057. }
  1058. } // end namespace anki