vk_rasterizer.cpp 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149
  1. // Copyright 2019 yuzu Emulator Project
  2. // Licensed under GPLv2 or any later version
  3. // Refer to the license.txt file included.
  4. #include <algorithm>
  5. #include <array>
  6. #include <memory>
  7. #include <mutex>
  8. #include <vector>
  9. #include <boost/container/static_vector.hpp>
  10. #include <boost/functional/hash.hpp>
  11. #include "common/alignment.h"
  12. #include "common/assert.h"
  13. #include "common/logging/log.h"
  14. #include "common/microprofile.h"
  15. #include "core/core.h"
  16. #include "core/memory.h"
  17. #include "video_core/engines/kepler_compute.h"
  18. #include "video_core/engines/maxwell_3d.h"
  19. #include "video_core/renderer_vulkan/declarations.h"
  20. #include "video_core/renderer_vulkan/fixed_pipeline_state.h"
  21. #include "video_core/renderer_vulkan/maxwell_to_vk.h"
  22. #include "video_core/renderer_vulkan/renderer_vulkan.h"
  23. #include "video_core/renderer_vulkan/vk_buffer_cache.h"
  24. #include "video_core/renderer_vulkan/vk_compute_pass.h"
  25. #include "video_core/renderer_vulkan/vk_compute_pipeline.h"
  26. #include "video_core/renderer_vulkan/vk_descriptor_pool.h"
  27. #include "video_core/renderer_vulkan/vk_device.h"
  28. #include "video_core/renderer_vulkan/vk_graphics_pipeline.h"
  29. #include "video_core/renderer_vulkan/vk_pipeline_cache.h"
  30. #include "video_core/renderer_vulkan/vk_rasterizer.h"
  31. #include "video_core/renderer_vulkan/vk_renderpass_cache.h"
  32. #include "video_core/renderer_vulkan/vk_resource_manager.h"
  33. #include "video_core/renderer_vulkan/vk_sampler_cache.h"
  34. #include "video_core/renderer_vulkan/vk_scheduler.h"
  35. #include "video_core/renderer_vulkan/vk_staging_buffer_pool.h"
  36. #include "video_core/renderer_vulkan/vk_state_tracker.h"
  37. #include "video_core/renderer_vulkan/vk_texture_cache.h"
  38. #include "video_core/renderer_vulkan/vk_update_descriptor.h"
  39. namespace Vulkan {
  40. using Maxwell = Tegra::Engines::Maxwell3D::Regs;
  41. MICROPROFILE_DEFINE(Vulkan_WaitForWorker, "Vulkan", "Wait for worker", MP_RGB(255, 192, 192));
  42. MICROPROFILE_DEFINE(Vulkan_Drawing, "Vulkan", "Record drawing", MP_RGB(192, 128, 128));
  43. MICROPROFILE_DEFINE(Vulkan_Compute, "Vulkan", "Record compute", MP_RGB(192, 128, 128));
  44. MICROPROFILE_DEFINE(Vulkan_Clearing, "Vulkan", "Record clearing", MP_RGB(192, 128, 128));
  45. MICROPROFILE_DEFINE(Vulkan_Geometry, "Vulkan", "Setup geometry", MP_RGB(192, 128, 128));
  46. MICROPROFILE_DEFINE(Vulkan_ConstBuffers, "Vulkan", "Setup constant buffers", MP_RGB(192, 128, 128));
  47. MICROPROFILE_DEFINE(Vulkan_GlobalBuffers, "Vulkan", "Setup global buffers", MP_RGB(192, 128, 128));
  48. MICROPROFILE_DEFINE(Vulkan_RenderTargets, "Vulkan", "Setup render targets", MP_RGB(192, 128, 128));
  49. MICROPROFILE_DEFINE(Vulkan_Textures, "Vulkan", "Setup textures", MP_RGB(192, 128, 128));
  50. MICROPROFILE_DEFINE(Vulkan_Images, "Vulkan", "Setup images", MP_RGB(192, 128, 128));
  51. MICROPROFILE_DEFINE(Vulkan_PipelineCache, "Vulkan", "Pipeline cache", MP_RGB(192, 128, 128));
  52. namespace {
  53. constexpr auto ComputeShaderIndex = static_cast<std::size_t>(Tegra::Engines::ShaderType::Compute);
  54. vk::Viewport GetViewportState(const VKDevice& device, const Maxwell& regs, std::size_t index) {
  55. const auto& viewport = regs.viewport_transform[index];
  56. const float x = viewport.translate_x - viewport.scale_x;
  57. const float y = viewport.translate_y - viewport.scale_y;
  58. const float width = viewport.scale_x * 2.0f;
  59. const float height = viewport.scale_y * 2.0f;
  60. const float reduce_z = regs.depth_mode == Maxwell::DepthMode::MinusOneToOne;
  61. float near = viewport.translate_z - viewport.scale_z * reduce_z;
  62. float far = viewport.translate_z + viewport.scale_z;
  63. if (!device.IsExtDepthRangeUnrestrictedSupported()) {
  64. near = std::clamp(near, 0.0f, 1.0f);
  65. far = std::clamp(far, 0.0f, 1.0f);
  66. }
  67. return vk::Viewport(x, y, width != 0 ? width : 1.0f, height != 0 ? height : 1.0f, near, far);
  68. }
  69. constexpr vk::Rect2D GetScissorState(const Maxwell& regs, std::size_t index) {
  70. const auto& scissor = regs.scissor_test[index];
  71. if (!scissor.enable) {
  72. return {{0, 0}, {INT32_MAX, INT32_MAX}};
  73. }
  74. const u32 width = scissor.max_x - scissor.min_x;
  75. const u32 height = scissor.max_y - scissor.min_y;
  76. return {{static_cast<s32>(scissor.min_x), static_cast<s32>(scissor.min_y)}, {width, height}};
  77. }
  78. std::array<GPUVAddr, Maxwell::MaxShaderProgram> GetShaderAddresses(
  79. const std::array<Shader, Maxwell::MaxShaderProgram>& shaders) {
  80. std::array<GPUVAddr, Maxwell::MaxShaderProgram> addresses;
  81. for (std::size_t i = 0; i < std::size(addresses); ++i) {
  82. addresses[i] = shaders[i] ? shaders[i]->GetGpuAddr() : 0;
  83. }
  84. return addresses;
  85. }
  86. void TransitionImages(const std::vector<ImageView>& views, vk::PipelineStageFlags pipeline_stage,
  87. vk::AccessFlags access) {
  88. for (auto& [view, layout] : views) {
  89. view->Transition(*layout, pipeline_stage, access);
  90. }
  91. }
  92. template <typename Engine, typename Entry>
  93. Tegra::Texture::FullTextureInfo GetTextureInfo(const Engine& engine, const Entry& entry,
  94. std::size_t stage) {
  95. const auto stage_type = static_cast<Tegra::Engines::ShaderType>(stage);
  96. if (entry.IsBindless()) {
  97. const Tegra::Texture::TextureHandle tex_handle =
  98. engine.AccessConstBuffer32(stage_type, entry.GetBuffer(), entry.GetOffset());
  99. return engine.GetTextureInfo(tex_handle);
  100. }
  101. if constexpr (std::is_same_v<Engine, Tegra::Engines::Maxwell3D>) {
  102. return engine.GetStageTexture(stage_type, entry.GetOffset());
  103. } else {
  104. return engine.GetTexture(entry.GetOffset());
  105. }
  106. }
  107. } // Anonymous namespace
  108. class BufferBindings final {
  109. public:
  110. void AddVertexBinding(const vk::Buffer* buffer, vk::DeviceSize offset) {
  111. vertex.buffer_ptrs[vertex.num_buffers] = buffer;
  112. vertex.offsets[vertex.num_buffers] = offset;
  113. ++vertex.num_buffers;
  114. }
  115. void SetIndexBinding(const vk::Buffer* buffer, vk::DeviceSize offset, vk::IndexType type) {
  116. index.buffer = buffer;
  117. index.offset = offset;
  118. index.type = type;
  119. }
  120. void Bind(VKScheduler& scheduler) const {
  121. // Use this large switch case to avoid dispatching more memory in the record lambda than
  122. // what we need. It looks horrible, but it's the best we can do on standard C++.
  123. switch (vertex.num_buffers) {
  124. case 0:
  125. return BindStatic<0>(scheduler);
  126. case 1:
  127. return BindStatic<1>(scheduler);
  128. case 2:
  129. return BindStatic<2>(scheduler);
  130. case 3:
  131. return BindStatic<3>(scheduler);
  132. case 4:
  133. return BindStatic<4>(scheduler);
  134. case 5:
  135. return BindStatic<5>(scheduler);
  136. case 6:
  137. return BindStatic<6>(scheduler);
  138. case 7:
  139. return BindStatic<7>(scheduler);
  140. case 8:
  141. return BindStatic<8>(scheduler);
  142. case 9:
  143. return BindStatic<9>(scheduler);
  144. case 10:
  145. return BindStatic<10>(scheduler);
  146. case 11:
  147. return BindStatic<11>(scheduler);
  148. case 12:
  149. return BindStatic<12>(scheduler);
  150. case 13:
  151. return BindStatic<13>(scheduler);
  152. case 14:
  153. return BindStatic<14>(scheduler);
  154. case 15:
  155. return BindStatic<15>(scheduler);
  156. case 16:
  157. return BindStatic<16>(scheduler);
  158. case 17:
  159. return BindStatic<17>(scheduler);
  160. case 18:
  161. return BindStatic<18>(scheduler);
  162. case 19:
  163. return BindStatic<19>(scheduler);
  164. case 20:
  165. return BindStatic<20>(scheduler);
  166. case 21:
  167. return BindStatic<21>(scheduler);
  168. case 22:
  169. return BindStatic<22>(scheduler);
  170. case 23:
  171. return BindStatic<23>(scheduler);
  172. case 24:
  173. return BindStatic<24>(scheduler);
  174. case 25:
  175. return BindStatic<25>(scheduler);
  176. case 26:
  177. return BindStatic<26>(scheduler);
  178. case 27:
  179. return BindStatic<27>(scheduler);
  180. case 28:
  181. return BindStatic<28>(scheduler);
  182. case 29:
  183. return BindStatic<29>(scheduler);
  184. case 30:
  185. return BindStatic<30>(scheduler);
  186. case 31:
  187. return BindStatic<31>(scheduler);
  188. case 32:
  189. return BindStatic<32>(scheduler);
  190. }
  191. UNREACHABLE();
  192. }
  193. private:
  194. // Some of these fields are intentionally left uninitialized to avoid initializing them twice.
  195. struct {
  196. std::size_t num_buffers = 0;
  197. std::array<const vk::Buffer*, Maxwell::NumVertexArrays> buffer_ptrs;
  198. std::array<vk::DeviceSize, Maxwell::NumVertexArrays> offsets;
  199. } vertex;
  200. struct {
  201. const vk::Buffer* buffer = nullptr;
  202. vk::DeviceSize offset;
  203. vk::IndexType type;
  204. } index;
  205. template <std::size_t N>
  206. void BindStatic(VKScheduler& scheduler) const {
  207. if (index.buffer != nullptr) {
  208. BindStatic<N, true>(scheduler);
  209. } else {
  210. BindStatic<N, false>(scheduler);
  211. }
  212. }
  213. template <std::size_t N, bool is_indexed>
  214. void BindStatic(VKScheduler& scheduler) const {
  215. static_assert(N <= Maxwell::NumVertexArrays);
  216. if constexpr (N == 0) {
  217. return;
  218. }
  219. std::array<vk::Buffer, N> buffers;
  220. std::transform(vertex.buffer_ptrs.begin(), vertex.buffer_ptrs.begin() + N, buffers.begin(),
  221. [](const auto ptr) { return *ptr; });
  222. std::array<vk::DeviceSize, N> offsets;
  223. std::copy(vertex.offsets.begin(), vertex.offsets.begin() + N, offsets.begin());
  224. if constexpr (is_indexed) {
  225. // Indexed draw
  226. scheduler.Record([buffers, offsets, index_buffer = *index.buffer,
  227. index_offset = index.offset,
  228. index_type = index.type](auto cmdbuf, auto& dld) {
  229. cmdbuf.bindIndexBuffer(index_buffer, index_offset, index_type, dld);
  230. cmdbuf.bindVertexBuffers(0, static_cast<u32>(N), buffers.data(), offsets.data(),
  231. dld);
  232. });
  233. } else {
  234. // Array draw
  235. scheduler.Record([buffers, offsets](auto cmdbuf, auto& dld) {
  236. cmdbuf.bindVertexBuffers(0, static_cast<u32>(N), buffers.data(), offsets.data(),
  237. dld);
  238. });
  239. }
  240. }
  241. };
  242. void RasterizerVulkan::DrawParameters::Draw(vk::CommandBuffer cmdbuf,
  243. const vk::DispatchLoaderDynamic& dld) const {
  244. if (is_indexed) {
  245. cmdbuf.drawIndexed(num_vertices, num_instances, 0, base_vertex, base_instance, dld);
  246. } else {
  247. cmdbuf.draw(num_vertices, num_instances, base_vertex, base_instance, dld);
  248. }
  249. }
  250. RasterizerVulkan::RasterizerVulkan(Core::System& system, Core::Frontend::EmuWindow& renderer,
  251. VKScreenInfo& screen_info, const VKDevice& device,
  252. VKResourceManager& resource_manager,
  253. VKMemoryManager& memory_manager, StateTracker& state_tracker,
  254. VKScheduler& scheduler)
  255. : RasterizerAccelerated{system.Memory()}, system{system}, render_window{renderer},
  256. screen_info{screen_info}, device{device}, resource_manager{resource_manager},
  257. memory_manager{memory_manager}, state_tracker{state_tracker}, scheduler{scheduler},
  258. staging_pool(device, memory_manager, scheduler), descriptor_pool(device),
  259. update_descriptor_queue(device, scheduler),
  260. quad_array_pass(device, scheduler, descriptor_pool, staging_pool, update_descriptor_queue),
  261. uint8_pass(device, scheduler, descriptor_pool, staging_pool, update_descriptor_queue),
  262. texture_cache(system, *this, device, resource_manager, memory_manager, scheduler,
  263. staging_pool),
  264. pipeline_cache(system, *this, device, scheduler, descriptor_pool, update_descriptor_queue),
  265. buffer_cache(*this, system, device, memory_manager, scheduler, staging_pool),
  266. sampler_cache(device), query_cache(system, *this, device, scheduler) {
  267. scheduler.SetQueryCache(query_cache);
  268. }
  269. RasterizerVulkan::~RasterizerVulkan() = default;
  270. void RasterizerVulkan::Draw(bool is_indexed, bool is_instanced) {
  271. MICROPROFILE_SCOPE(Vulkan_Drawing);
  272. FlushWork();
  273. query_cache.UpdateCounters();
  274. const auto& gpu = system.GPU().Maxwell3D();
  275. GraphicsPipelineCacheKey key{GetFixedPipelineState(gpu.regs)};
  276. buffer_cache.Map(CalculateGraphicsStreamBufferSize(is_indexed));
  277. BufferBindings buffer_bindings;
  278. const DrawParameters draw_params =
  279. SetupGeometry(key.fixed_state, buffer_bindings, is_indexed, is_instanced);
  280. update_descriptor_queue.Acquire();
  281. sampled_views.clear();
  282. image_views.clear();
  283. const auto shaders = pipeline_cache.GetShaders();
  284. key.shaders = GetShaderAddresses(shaders);
  285. SetupShaderDescriptors(shaders);
  286. buffer_cache.Unmap();
  287. const auto texceptions = UpdateAttachments();
  288. SetupImageTransitions(texceptions, color_attachments, zeta_attachment);
  289. key.renderpass_params = GetRenderPassParams(texceptions);
  290. auto& pipeline = pipeline_cache.GetGraphicsPipeline(key);
  291. scheduler.BindGraphicsPipeline(pipeline.GetHandle());
  292. const auto renderpass = pipeline.GetRenderPass();
  293. const auto [framebuffer, render_area] = ConfigureFramebuffers(renderpass);
  294. scheduler.RequestRenderpass({renderpass, framebuffer, {{0, 0}, render_area}, 0, nullptr});
  295. UpdateDynamicStates();
  296. buffer_bindings.Bind(scheduler);
  297. if (device.IsNvDeviceDiagnosticCheckpoints()) {
  298. scheduler.Record(
  299. [&pipeline](auto cmdbuf, auto& dld) { cmdbuf.setCheckpointNV(&pipeline, dld); });
  300. }
  301. const auto pipeline_layout = pipeline.GetLayout();
  302. const auto descriptor_set = pipeline.CommitDescriptorSet();
  303. scheduler.Record([pipeline_layout, descriptor_set, draw_params](auto cmdbuf, auto& dld) {
  304. if (descriptor_set) {
  305. cmdbuf.bindDescriptorSets(vk::PipelineBindPoint::eGraphics, pipeline_layout,
  306. DESCRIPTOR_SET, 1, &descriptor_set, 0, nullptr, dld);
  307. }
  308. draw_params.Draw(cmdbuf, dld);
  309. });
  310. }
  311. void RasterizerVulkan::Clear() {
  312. MICROPROFILE_SCOPE(Vulkan_Clearing);
  313. query_cache.UpdateCounters();
  314. const auto& gpu = system.GPU().Maxwell3D();
  315. if (!system.GPU().Maxwell3D().ShouldExecute()) {
  316. return;
  317. }
  318. const auto& regs = gpu.regs;
  319. const bool use_color = regs.clear_buffers.R || regs.clear_buffers.G || regs.clear_buffers.B ||
  320. regs.clear_buffers.A;
  321. const bool use_depth = regs.clear_buffers.Z;
  322. const bool use_stencil = regs.clear_buffers.S;
  323. if (!use_color && !use_depth && !use_stencil) {
  324. return;
  325. }
  326. // Clearing images requires to be out of a renderpass
  327. scheduler.RequestOutsideRenderPassOperationContext();
  328. // TODO(Rodrigo): Implement clears rendering a quad or using beginning a renderpass.
  329. if (use_color) {
  330. View color_view;
  331. {
  332. MICROPROFILE_SCOPE(Vulkan_RenderTargets);
  333. color_view = texture_cache.GetColorBufferSurface(regs.clear_buffers.RT.Value(), false);
  334. }
  335. color_view->Transition(vk::ImageLayout::eTransferDstOptimal,
  336. vk::PipelineStageFlagBits::eTransfer,
  337. vk::AccessFlagBits::eTransferWrite);
  338. const std::array clear_color = {regs.clear_color[0], regs.clear_color[1],
  339. regs.clear_color[2], regs.clear_color[3]};
  340. const vk::ClearColorValue clear(clear_color);
  341. scheduler.Record([image = color_view->GetImage(),
  342. subresource = color_view->GetImageSubresourceRange(),
  343. clear](auto cmdbuf, auto& dld) {
  344. cmdbuf.clearColorImage(image, vk::ImageLayout::eTransferDstOptimal, clear, subresource,
  345. dld);
  346. });
  347. }
  348. if (use_depth || use_stencil) {
  349. View zeta_surface;
  350. {
  351. MICROPROFILE_SCOPE(Vulkan_RenderTargets);
  352. zeta_surface = texture_cache.GetDepthBufferSurface(false);
  353. }
  354. zeta_surface->Transition(vk::ImageLayout::eTransferDstOptimal,
  355. vk::PipelineStageFlagBits::eTransfer,
  356. vk::AccessFlagBits::eTransferWrite);
  357. const vk::ClearDepthStencilValue clear(regs.clear_depth,
  358. static_cast<u32>(regs.clear_stencil));
  359. scheduler.Record([image = zeta_surface->GetImage(),
  360. subresource = zeta_surface->GetImageSubresourceRange(),
  361. clear](auto cmdbuf, auto& dld) {
  362. cmdbuf.clearDepthStencilImage(image, vk::ImageLayout::eTransferDstOptimal, clear,
  363. subresource, dld);
  364. });
  365. }
  366. }
  367. void RasterizerVulkan::DispatchCompute(GPUVAddr code_addr) {
  368. MICROPROFILE_SCOPE(Vulkan_Compute);
  369. update_descriptor_queue.Acquire();
  370. sampled_views.clear();
  371. image_views.clear();
  372. query_cache.UpdateCounters();
  373. const auto& launch_desc = system.GPU().KeplerCompute().launch_description;
  374. const ComputePipelineCacheKey key{
  375. code_addr,
  376. launch_desc.shared_alloc,
  377. {launch_desc.block_dim_x, launch_desc.block_dim_y, launch_desc.block_dim_z}};
  378. auto& pipeline = pipeline_cache.GetComputePipeline(key);
  379. // Compute dispatches can't be executed inside a renderpass
  380. scheduler.RequestOutsideRenderPassOperationContext();
  381. buffer_cache.Map(CalculateComputeStreamBufferSize());
  382. const auto& entries = pipeline.GetEntries();
  383. SetupComputeConstBuffers(entries);
  384. SetupComputeGlobalBuffers(entries);
  385. SetupComputeTexelBuffers(entries);
  386. SetupComputeTextures(entries);
  387. SetupComputeImages(entries);
  388. buffer_cache.Unmap();
  389. TransitionImages(sampled_views, vk::PipelineStageFlagBits::eComputeShader,
  390. vk::AccessFlagBits::eShaderRead);
  391. TransitionImages(image_views, vk::PipelineStageFlagBits::eComputeShader,
  392. vk::AccessFlagBits::eShaderRead | vk::AccessFlagBits::eShaderWrite);
  393. if (device.IsNvDeviceDiagnosticCheckpoints()) {
  394. scheduler.Record(
  395. [&pipeline](auto cmdbuf, auto& dld) { cmdbuf.setCheckpointNV(nullptr, dld); });
  396. }
  397. scheduler.Record([grid_x = launch_desc.grid_dim_x, grid_y = launch_desc.grid_dim_y,
  398. grid_z = launch_desc.grid_dim_z, pipeline_handle = pipeline.GetHandle(),
  399. layout = pipeline.GetLayout(),
  400. descriptor_set = pipeline.CommitDescriptorSet()](auto cmdbuf, auto& dld) {
  401. cmdbuf.bindPipeline(vk::PipelineBindPoint::eCompute, pipeline_handle, dld);
  402. cmdbuf.bindDescriptorSets(vk::PipelineBindPoint::eCompute, layout, DESCRIPTOR_SET, 1,
  403. &descriptor_set, 0, nullptr, dld);
  404. cmdbuf.dispatch(grid_x, grid_y, grid_z, dld);
  405. });
  406. }
  407. void RasterizerVulkan::ResetCounter(VideoCore::QueryType type) {
  408. query_cache.ResetCounter(type);
  409. }
  410. void RasterizerVulkan::Query(GPUVAddr gpu_addr, VideoCore::QueryType type,
  411. std::optional<u64> timestamp) {
  412. query_cache.Query(gpu_addr, type, timestamp);
  413. }
  414. void RasterizerVulkan::FlushAll() {}
  415. void RasterizerVulkan::FlushRegion(CacheAddr addr, u64 size) {
  416. texture_cache.FlushRegion(addr, size);
  417. buffer_cache.FlushRegion(addr, size);
  418. query_cache.FlushRegion(addr, size);
  419. }
  420. void RasterizerVulkan::InvalidateRegion(CacheAddr addr, u64 size) {
  421. texture_cache.InvalidateRegion(addr, size);
  422. pipeline_cache.InvalidateRegion(addr, size);
  423. buffer_cache.InvalidateRegion(addr, size);
  424. query_cache.InvalidateRegion(addr, size);
  425. }
  426. void RasterizerVulkan::FlushAndInvalidateRegion(CacheAddr addr, u64 size) {
  427. FlushRegion(addr, size);
  428. InvalidateRegion(addr, size);
  429. }
  430. void RasterizerVulkan::FlushCommands() {
  431. if (draw_counter > 0) {
  432. draw_counter = 0;
  433. scheduler.Flush();
  434. }
  435. }
  436. void RasterizerVulkan::TickFrame() {
  437. draw_counter = 0;
  438. update_descriptor_queue.TickFrame();
  439. buffer_cache.TickFrame();
  440. staging_pool.TickFrame();
  441. }
  442. bool RasterizerVulkan::AccelerateSurfaceCopy(const Tegra::Engines::Fermi2D::Regs::Surface& src,
  443. const Tegra::Engines::Fermi2D::Regs::Surface& dst,
  444. const Tegra::Engines::Fermi2D::Config& copy_config) {
  445. texture_cache.DoFermiCopy(src, dst, copy_config);
  446. return true;
  447. }
  448. bool RasterizerVulkan::AccelerateDisplay(const Tegra::FramebufferConfig& config,
  449. VAddr framebuffer_addr, u32 pixel_stride) {
  450. if (!framebuffer_addr) {
  451. return false;
  452. }
  453. const u8* host_ptr{system.Memory().GetPointer(framebuffer_addr)};
  454. const auto surface{texture_cache.TryFindFramebufferSurface(host_ptr)};
  455. if (!surface) {
  456. return false;
  457. }
  458. // Verify that the cached surface is the same size and format as the requested framebuffer
  459. const auto& params{surface->GetSurfaceParams()};
  460. const auto& pixel_format{
  461. VideoCore::Surface::PixelFormatFromGPUPixelFormat(config.pixel_format)};
  462. ASSERT_MSG(params.width == config.width, "Framebuffer width is different");
  463. ASSERT_MSG(params.height == config.height, "Framebuffer height is different");
  464. screen_info.image = &surface->GetImage();
  465. screen_info.width = params.width;
  466. screen_info.height = params.height;
  467. screen_info.is_srgb = surface->GetSurfaceParams().srgb_conversion;
  468. return true;
  469. }
  470. void RasterizerVulkan::SetupDirtyFlags() {
  471. state_tracker.Initialize();
  472. }
  473. void RasterizerVulkan::FlushWork() {
  474. static constexpr u32 DRAWS_TO_DISPATCH = 4096;
  475. // Only check multiples of 8 draws
  476. static_assert(DRAWS_TO_DISPATCH % 8 == 0);
  477. if ((++draw_counter & 7) != 7) {
  478. return;
  479. }
  480. if (draw_counter < DRAWS_TO_DISPATCH) {
  481. // Send recorded tasks to the worker thread
  482. scheduler.DispatchWork();
  483. return;
  484. }
  485. // Otherwise (every certain number of draws) flush execution.
  486. // This submits commands to the Vulkan driver.
  487. scheduler.Flush();
  488. draw_counter = 0;
  489. }
  490. RasterizerVulkan::Texceptions RasterizerVulkan::UpdateAttachments() {
  491. MICROPROFILE_SCOPE(Vulkan_RenderTargets);
  492. auto& dirty = system.GPU().Maxwell3D().dirty.flags;
  493. const bool update_rendertargets = dirty[VideoCommon::Dirty::RenderTargets];
  494. dirty[VideoCommon::Dirty::RenderTargets] = false;
  495. texture_cache.GuardRenderTargets(true);
  496. Texceptions texceptions;
  497. for (std::size_t rt = 0; rt < Maxwell::NumRenderTargets; ++rt) {
  498. if (update_rendertargets) {
  499. color_attachments[rt] = texture_cache.GetColorBufferSurface(rt, true);
  500. }
  501. if (color_attachments[rt] && WalkAttachmentOverlaps(*color_attachments[rt])) {
  502. texceptions[rt] = true;
  503. }
  504. }
  505. if (update_rendertargets) {
  506. zeta_attachment = texture_cache.GetDepthBufferSurface(true);
  507. }
  508. if (zeta_attachment && WalkAttachmentOverlaps(*zeta_attachment)) {
  509. texceptions[ZETA_TEXCEPTION_INDEX] = true;
  510. }
  511. texture_cache.GuardRenderTargets(false);
  512. return texceptions;
  513. }
  514. bool RasterizerVulkan::WalkAttachmentOverlaps(const CachedSurfaceView& attachment) {
  515. bool overlap = false;
  516. for (auto& [view, layout] : sampled_views) {
  517. if (!attachment.IsSameSurface(*view)) {
  518. continue;
  519. }
  520. overlap = true;
  521. *layout = vk::ImageLayout::eGeneral;
  522. }
  523. return overlap;
  524. }
  525. std::tuple<vk::Framebuffer, vk::Extent2D> RasterizerVulkan::ConfigureFramebuffers(
  526. vk::RenderPass renderpass) {
  527. FramebufferCacheKey key{renderpass, std::numeric_limits<u32>::max(),
  528. std::numeric_limits<u32>::max(), std::numeric_limits<u32>::max()};
  529. const auto try_push = [&](const View& view) {
  530. if (!view) {
  531. return false;
  532. }
  533. key.views.push_back(view->GetHandle());
  534. key.width = std::min(key.width, view->GetWidth());
  535. key.height = std::min(key.height, view->GetHeight());
  536. key.layers = std::min(key.layers, view->GetNumLayers());
  537. return true;
  538. };
  539. for (std::size_t index = 0; index < std::size(color_attachments); ++index) {
  540. if (try_push(color_attachments[index])) {
  541. texture_cache.MarkColorBufferInUse(index);
  542. }
  543. }
  544. if (try_push(zeta_attachment)) {
  545. texture_cache.MarkDepthBufferInUse();
  546. }
  547. const auto [fbentry, is_cache_miss] = framebuffer_cache.try_emplace(key);
  548. auto& framebuffer = fbentry->second;
  549. if (is_cache_miss) {
  550. const vk::FramebufferCreateInfo framebuffer_ci(
  551. {}, key.renderpass, static_cast<u32>(key.views.size()), key.views.data(), key.width,
  552. key.height, key.layers);
  553. const auto dev = device.GetLogical();
  554. const auto& dld = device.GetDispatchLoader();
  555. framebuffer = dev.createFramebufferUnique(framebuffer_ci, nullptr, dld);
  556. }
  557. return {*framebuffer, vk::Extent2D{key.width, key.height}};
  558. }
  559. RasterizerVulkan::DrawParameters RasterizerVulkan::SetupGeometry(FixedPipelineState& fixed_state,
  560. BufferBindings& buffer_bindings,
  561. bool is_indexed,
  562. bool is_instanced) {
  563. MICROPROFILE_SCOPE(Vulkan_Geometry);
  564. const auto& gpu = system.GPU().Maxwell3D();
  565. const auto& regs = gpu.regs;
  566. SetupVertexArrays(fixed_state.vertex_input, buffer_bindings);
  567. const u32 base_instance = regs.vb_base_instance;
  568. const u32 num_instances = is_instanced ? gpu.mme_draw.instance_count : 1;
  569. const u32 base_vertex = is_indexed ? regs.vb_element_base : regs.vertex_buffer.first;
  570. const u32 num_vertices = is_indexed ? regs.index_array.count : regs.vertex_buffer.count;
  571. DrawParameters params{base_instance, num_instances, base_vertex, num_vertices, is_indexed};
  572. SetupIndexBuffer(buffer_bindings, params, is_indexed);
  573. return params;
  574. }
  575. void RasterizerVulkan::SetupShaderDescriptors(
  576. const std::array<Shader, Maxwell::MaxShaderProgram>& shaders) {
  577. texture_cache.GuardSamplers(true);
  578. for (std::size_t stage = 0; stage < Maxwell::MaxShaderStage; ++stage) {
  579. // Skip VertexA stage
  580. const auto& shader = shaders[stage + 1];
  581. if (!shader) {
  582. continue;
  583. }
  584. const auto& entries = shader->GetEntries();
  585. SetupGraphicsConstBuffers(entries, stage);
  586. SetupGraphicsGlobalBuffers(entries, stage);
  587. SetupGraphicsTexelBuffers(entries, stage);
  588. SetupGraphicsTextures(entries, stage);
  589. SetupGraphicsImages(entries, stage);
  590. }
  591. texture_cache.GuardSamplers(false);
  592. }
  593. void RasterizerVulkan::SetupImageTransitions(
  594. Texceptions texceptions, const std::array<View, Maxwell::NumRenderTargets>& color_attachments,
  595. const View& zeta_attachment) {
  596. TransitionImages(sampled_views, vk::PipelineStageFlagBits::eAllGraphics,
  597. vk::AccessFlagBits::eShaderRead);
  598. TransitionImages(image_views, vk::PipelineStageFlagBits::eAllGraphics,
  599. vk::AccessFlagBits::eShaderRead | vk::AccessFlagBits::eShaderWrite);
  600. for (std::size_t rt = 0; rt < std::size(color_attachments); ++rt) {
  601. const auto color_attachment = color_attachments[rt];
  602. if (color_attachment == nullptr) {
  603. continue;
  604. }
  605. const auto image_layout =
  606. texceptions[rt] ? vk::ImageLayout::eGeneral : vk::ImageLayout::eColorAttachmentOptimal;
  607. color_attachment->Transition(
  608. image_layout, vk::PipelineStageFlagBits::eColorAttachmentOutput,
  609. vk::AccessFlagBits::eColorAttachmentRead | vk::AccessFlagBits::eColorAttachmentWrite);
  610. }
  611. if (zeta_attachment != nullptr) {
  612. const auto image_layout = texceptions[ZETA_TEXCEPTION_INDEX]
  613. ? vk::ImageLayout::eGeneral
  614. : vk::ImageLayout::eDepthStencilAttachmentOptimal;
  615. zeta_attachment->Transition(image_layout, vk::PipelineStageFlagBits::eLateFragmentTests,
  616. vk::AccessFlagBits::eDepthStencilAttachmentRead |
  617. vk::AccessFlagBits::eDepthStencilAttachmentWrite);
  618. }
  619. }
  620. void RasterizerVulkan::UpdateDynamicStates() {
  621. auto& gpu = system.GPU().Maxwell3D();
  622. UpdateViewportsState(gpu);
  623. UpdateScissorsState(gpu);
  624. UpdateDepthBias(gpu);
  625. UpdateBlendConstants(gpu);
  626. UpdateDepthBounds(gpu);
  627. UpdateStencilFaces(gpu);
  628. }
  629. void RasterizerVulkan::SetupVertexArrays(FixedPipelineState::VertexInput& vertex_input,
  630. BufferBindings& buffer_bindings) {
  631. const auto& regs = system.GPU().Maxwell3D().regs;
  632. for (u32 index = 0; index < static_cast<u32>(Maxwell::NumVertexAttributes); ++index) {
  633. const auto& attrib = regs.vertex_attrib_format[index];
  634. if (!attrib.IsValid()) {
  635. continue;
  636. }
  637. const auto& buffer = regs.vertex_array[attrib.buffer];
  638. ASSERT(buffer.IsEnabled());
  639. vertex_input.attributes[vertex_input.num_attributes++] =
  640. FixedPipelineState::VertexAttribute(index, attrib.buffer, attrib.type, attrib.size,
  641. attrib.offset);
  642. }
  643. for (u32 index = 0; index < static_cast<u32>(Maxwell::NumVertexArrays); ++index) {
  644. const auto& vertex_array = regs.vertex_array[index];
  645. if (!vertex_array.IsEnabled()) {
  646. continue;
  647. }
  648. const GPUVAddr start{vertex_array.StartAddress()};
  649. const GPUVAddr end{regs.vertex_array_limit[index].LimitAddress()};
  650. ASSERT(end > start);
  651. const std::size_t size{end - start + 1};
  652. const auto [buffer, offset] = buffer_cache.UploadMemory(start, size);
  653. vertex_input.bindings[vertex_input.num_bindings++] = FixedPipelineState::VertexBinding(
  654. index, vertex_array.stride,
  655. regs.instanced_arrays.IsInstancingEnabled(index) ? vertex_array.divisor : 0);
  656. buffer_bindings.AddVertexBinding(buffer, offset);
  657. }
  658. }
  659. void RasterizerVulkan::SetupIndexBuffer(BufferBindings& buffer_bindings, DrawParameters& params,
  660. bool is_indexed) {
  661. const auto& regs = system.GPU().Maxwell3D().regs;
  662. switch (regs.draw.topology) {
  663. case Maxwell::PrimitiveTopology::Quads:
  664. if (params.is_indexed) {
  665. UNIMPLEMENTED();
  666. } else {
  667. const auto [buffer, offset] =
  668. quad_array_pass.Assemble(params.num_vertices, params.base_vertex);
  669. buffer_bindings.SetIndexBinding(&buffer, offset, vk::IndexType::eUint32);
  670. params.base_vertex = 0;
  671. params.num_vertices = params.num_vertices * 6 / 4;
  672. params.is_indexed = true;
  673. }
  674. break;
  675. default: {
  676. if (!is_indexed) {
  677. break;
  678. }
  679. const GPUVAddr gpu_addr = regs.index_array.IndexStart();
  680. auto [buffer, offset] = buffer_cache.UploadMemory(gpu_addr, CalculateIndexBufferSize());
  681. auto format = regs.index_array.format;
  682. const bool is_uint8 = format == Maxwell::IndexFormat::UnsignedByte;
  683. if (is_uint8 && !device.IsExtIndexTypeUint8Supported()) {
  684. std::tie(buffer, offset) = uint8_pass.Assemble(params.num_vertices, *buffer, offset);
  685. format = Maxwell::IndexFormat::UnsignedShort;
  686. }
  687. buffer_bindings.SetIndexBinding(buffer, offset, MaxwellToVK::IndexFormat(device, format));
  688. break;
  689. }
  690. }
  691. }
  692. void RasterizerVulkan::SetupGraphicsConstBuffers(const ShaderEntries& entries, std::size_t stage) {
  693. MICROPROFILE_SCOPE(Vulkan_ConstBuffers);
  694. const auto& gpu = system.GPU().Maxwell3D();
  695. const auto& shader_stage = gpu.state.shader_stages[stage];
  696. for (const auto& entry : entries.const_buffers) {
  697. SetupConstBuffer(entry, shader_stage.const_buffers[entry.GetIndex()]);
  698. }
  699. }
  700. void RasterizerVulkan::SetupGraphicsGlobalBuffers(const ShaderEntries& entries, std::size_t stage) {
  701. MICROPROFILE_SCOPE(Vulkan_GlobalBuffers);
  702. auto& gpu{system.GPU()};
  703. const auto cbufs{gpu.Maxwell3D().state.shader_stages[stage]};
  704. for (const auto& entry : entries.global_buffers) {
  705. const auto addr = cbufs.const_buffers[entry.GetCbufIndex()].address + entry.GetCbufOffset();
  706. SetupGlobalBuffer(entry, addr);
  707. }
  708. }
  709. void RasterizerVulkan::SetupGraphicsTexelBuffers(const ShaderEntries& entries, std::size_t stage) {
  710. MICROPROFILE_SCOPE(Vulkan_Textures);
  711. const auto& gpu = system.GPU().Maxwell3D();
  712. for (const auto& entry : entries.texel_buffers) {
  713. const auto image = GetTextureInfo(gpu, entry, stage).tic;
  714. SetupTexelBuffer(image, entry);
  715. }
  716. }
  717. void RasterizerVulkan::SetupGraphicsTextures(const ShaderEntries& entries, std::size_t stage) {
  718. MICROPROFILE_SCOPE(Vulkan_Textures);
  719. const auto& gpu = system.GPU().Maxwell3D();
  720. for (const auto& entry : entries.samplers) {
  721. const auto texture = GetTextureInfo(gpu, entry, stage);
  722. SetupTexture(texture, entry);
  723. }
  724. }
  725. void RasterizerVulkan::SetupGraphicsImages(const ShaderEntries& entries, std::size_t stage) {
  726. MICROPROFILE_SCOPE(Vulkan_Images);
  727. const auto& gpu = system.GPU().KeplerCompute();
  728. for (const auto& entry : entries.images) {
  729. const auto tic = GetTextureInfo(gpu, entry, stage).tic;
  730. SetupImage(tic, entry);
  731. }
  732. }
  733. void RasterizerVulkan::SetupComputeConstBuffers(const ShaderEntries& entries) {
  734. MICROPROFILE_SCOPE(Vulkan_ConstBuffers);
  735. const auto& launch_desc = system.GPU().KeplerCompute().launch_description;
  736. for (const auto& entry : entries.const_buffers) {
  737. const auto& config = launch_desc.const_buffer_config[entry.GetIndex()];
  738. const std::bitset<8> mask = launch_desc.const_buffer_enable_mask.Value();
  739. Tegra::Engines::ConstBufferInfo buffer;
  740. buffer.address = config.Address();
  741. buffer.size = config.size;
  742. buffer.enabled = mask[entry.GetIndex()];
  743. SetupConstBuffer(entry, buffer);
  744. }
  745. }
  746. void RasterizerVulkan::SetupComputeGlobalBuffers(const ShaderEntries& entries) {
  747. MICROPROFILE_SCOPE(Vulkan_GlobalBuffers);
  748. const auto cbufs{system.GPU().KeplerCompute().launch_description.const_buffer_config};
  749. for (const auto& entry : entries.global_buffers) {
  750. const auto addr{cbufs[entry.GetCbufIndex()].Address() + entry.GetCbufOffset()};
  751. SetupGlobalBuffer(entry, addr);
  752. }
  753. }
  754. void RasterizerVulkan::SetupComputeTexelBuffers(const ShaderEntries& entries) {
  755. MICROPROFILE_SCOPE(Vulkan_Textures);
  756. const auto& gpu = system.GPU().KeplerCompute();
  757. for (const auto& entry : entries.texel_buffers) {
  758. const auto image = GetTextureInfo(gpu, entry, ComputeShaderIndex).tic;
  759. SetupTexelBuffer(image, entry);
  760. }
  761. }
  762. void RasterizerVulkan::SetupComputeTextures(const ShaderEntries& entries) {
  763. MICROPROFILE_SCOPE(Vulkan_Textures);
  764. const auto& gpu = system.GPU().KeplerCompute();
  765. for (const auto& entry : entries.samplers) {
  766. const auto texture = GetTextureInfo(gpu, entry, ComputeShaderIndex);
  767. SetupTexture(texture, entry);
  768. }
  769. }
  770. void RasterizerVulkan::SetupComputeImages(const ShaderEntries& entries) {
  771. MICROPROFILE_SCOPE(Vulkan_Images);
  772. const auto& gpu = system.GPU().KeplerCompute();
  773. for (const auto& entry : entries.images) {
  774. const auto tic = GetTextureInfo(gpu, entry, ComputeShaderIndex).tic;
  775. SetupImage(tic, entry);
  776. }
  777. }
  778. void RasterizerVulkan::SetupConstBuffer(const ConstBufferEntry& entry,
  779. const Tegra::Engines::ConstBufferInfo& buffer) {
  780. // Align the size to avoid bad std140 interactions
  781. const std::size_t size =
  782. Common::AlignUp(CalculateConstBufferSize(entry, buffer), 4 * sizeof(float));
  783. ASSERT(size <= MaxConstbufferSize);
  784. const auto [buffer_handle, offset] =
  785. buffer_cache.UploadMemory(buffer.address, size, device.GetUniformBufferAlignment());
  786. update_descriptor_queue.AddBuffer(buffer_handle, offset, size);
  787. }
  788. void RasterizerVulkan::SetupGlobalBuffer(const GlobalBufferEntry& entry, GPUVAddr address) {
  789. auto& memory_manager{system.GPU().MemoryManager()};
  790. const auto actual_addr = memory_manager.Read<u64>(address);
  791. const auto size = memory_manager.Read<u32>(address + 8);
  792. if (size == 0) {
  793. // Sometimes global memory pointers don't have a proper size. Upload a dummy entry because
  794. // Vulkan doesn't like empty buffers.
  795. constexpr std::size_t dummy_size = 4;
  796. const auto buffer = buffer_cache.GetEmptyBuffer(dummy_size);
  797. update_descriptor_queue.AddBuffer(buffer, 0, dummy_size);
  798. return;
  799. }
  800. const auto [buffer, offset] = buffer_cache.UploadMemory(
  801. actual_addr, size, device.GetStorageBufferAlignment(), entry.IsWritten());
  802. update_descriptor_queue.AddBuffer(buffer, offset, size);
  803. }
  804. void RasterizerVulkan::SetupTexelBuffer(const Tegra::Texture::TICEntry& tic,
  805. const TexelBufferEntry& entry) {
  806. const auto view = texture_cache.GetTextureSurface(tic, entry);
  807. ASSERT(view->IsBufferView());
  808. update_descriptor_queue.AddTexelBuffer(view->GetBufferView());
  809. }
  810. void RasterizerVulkan::SetupTexture(const Tegra::Texture::FullTextureInfo& texture,
  811. const SamplerEntry& entry) {
  812. auto view = texture_cache.GetTextureSurface(texture.tic, entry);
  813. ASSERT(!view->IsBufferView());
  814. const auto image_view = view->GetHandle(texture.tic.x_source, texture.tic.y_source,
  815. texture.tic.z_source, texture.tic.w_source);
  816. const auto sampler = sampler_cache.GetSampler(texture.tsc);
  817. update_descriptor_queue.AddSampledImage(sampler, image_view);
  818. const auto image_layout = update_descriptor_queue.GetLastImageLayout();
  819. *image_layout = vk::ImageLayout::eShaderReadOnlyOptimal;
  820. sampled_views.push_back(ImageView{std::move(view), image_layout});
  821. }
  822. void RasterizerVulkan::SetupImage(const Tegra::Texture::TICEntry& tic, const ImageEntry& entry) {
  823. auto view = texture_cache.GetImageSurface(tic, entry);
  824. if (entry.IsWritten()) {
  825. view->MarkAsModified(texture_cache.Tick());
  826. }
  827. UNIMPLEMENTED_IF(tic.IsBuffer());
  828. const auto image_view = view->GetHandle(tic.x_source, tic.y_source, tic.z_source, tic.w_source);
  829. update_descriptor_queue.AddImage(image_view);
  830. const auto image_layout = update_descriptor_queue.GetLastImageLayout();
  831. *image_layout = vk::ImageLayout::eGeneral;
  832. image_views.push_back(ImageView{std::move(view), image_layout});
  833. }
  834. void RasterizerVulkan::UpdateViewportsState(Tegra::Engines::Maxwell3D& gpu) {
  835. if (!state_tracker.TouchViewports()) {
  836. return;
  837. }
  838. const auto& regs = gpu.regs;
  839. const std::array viewports{
  840. GetViewportState(device, regs, 0), GetViewportState(device, regs, 1),
  841. GetViewportState(device, regs, 2), GetViewportState(device, regs, 3),
  842. GetViewportState(device, regs, 4), GetViewportState(device, regs, 5),
  843. GetViewportState(device, regs, 6), GetViewportState(device, regs, 7),
  844. GetViewportState(device, regs, 8), GetViewportState(device, regs, 9),
  845. GetViewportState(device, regs, 10), GetViewportState(device, regs, 11),
  846. GetViewportState(device, regs, 12), GetViewportState(device, regs, 13),
  847. GetViewportState(device, regs, 14), GetViewportState(device, regs, 15)};
  848. scheduler.Record([viewports](auto cmdbuf, auto& dld) {
  849. cmdbuf.setViewport(0, static_cast<u32>(viewports.size()), viewports.data(), dld);
  850. });
  851. }
  852. void RasterizerVulkan::UpdateScissorsState(Tegra::Engines::Maxwell3D& gpu) {
  853. if (!state_tracker.TouchScissors()) {
  854. return;
  855. }
  856. const auto& regs = gpu.regs;
  857. const std::array scissors = {
  858. GetScissorState(regs, 0), GetScissorState(regs, 1), GetScissorState(regs, 2),
  859. GetScissorState(regs, 3), GetScissorState(regs, 4), GetScissorState(regs, 5),
  860. GetScissorState(regs, 6), GetScissorState(regs, 7), GetScissorState(regs, 8),
  861. GetScissorState(regs, 9), GetScissorState(regs, 10), GetScissorState(regs, 11),
  862. GetScissorState(regs, 12), GetScissorState(regs, 13), GetScissorState(regs, 14),
  863. GetScissorState(regs, 15)};
  864. scheduler.Record([scissors](auto cmdbuf, auto& dld) {
  865. cmdbuf.setScissor(0, static_cast<u32>(scissors.size()), scissors.data(), dld);
  866. });
  867. }
  868. void RasterizerVulkan::UpdateDepthBias(Tegra::Engines::Maxwell3D& gpu) {
  869. if (!state_tracker.TouchDepthBias()) {
  870. return;
  871. }
  872. const auto& regs = gpu.regs;
  873. scheduler.Record([constant = regs.polygon_offset_units, clamp = regs.polygon_offset_clamp,
  874. factor = regs.polygon_offset_factor](auto cmdbuf, auto& dld) {
  875. cmdbuf.setDepthBias(constant, clamp, factor / 2.0f, dld);
  876. });
  877. }
  878. void RasterizerVulkan::UpdateBlendConstants(Tegra::Engines::Maxwell3D& gpu) {
  879. if (!state_tracker.TouchBlendConstants()) {
  880. return;
  881. }
  882. const std::array blend_color = {gpu.regs.blend_color.r, gpu.regs.blend_color.g,
  883. gpu.regs.blend_color.b, gpu.regs.blend_color.a};
  884. scheduler.Record([blend_color](auto cmdbuf, auto& dld) {
  885. cmdbuf.setBlendConstants(blend_color.data(), dld);
  886. });
  887. }
  888. void RasterizerVulkan::UpdateDepthBounds(Tegra::Engines::Maxwell3D& gpu) {
  889. if (!state_tracker.TouchDepthBounds()) {
  890. return;
  891. }
  892. const auto& regs = gpu.regs;
  893. scheduler.Record([min = regs.depth_bounds[0], max = regs.depth_bounds[1]](
  894. auto cmdbuf, auto& dld) { cmdbuf.setDepthBounds(min, max, dld); });
  895. }
  896. void RasterizerVulkan::UpdateStencilFaces(Tegra::Engines::Maxwell3D& gpu) {
  897. const auto& regs = gpu.regs;
  898. if (regs.stencil_two_side_enable) {
  899. // Separate values per face
  900. scheduler.Record(
  901. [front_ref = regs.stencil_front_func_ref, front_write_mask = regs.stencil_front_mask,
  902. front_test_mask = regs.stencil_front_func_mask, back_ref = regs.stencil_back_func_ref,
  903. back_write_mask = regs.stencil_back_mask,
  904. back_test_mask = regs.stencil_back_func_mask](auto cmdbuf, auto& dld) {
  905. // Front face
  906. cmdbuf.setStencilReference(vk::StencilFaceFlagBits::eFront, front_ref, dld);
  907. cmdbuf.setStencilWriteMask(vk::StencilFaceFlagBits::eFront, front_write_mask, dld);
  908. cmdbuf.setStencilCompareMask(vk::StencilFaceFlagBits::eFront, front_test_mask, dld);
  909. // Back face
  910. cmdbuf.setStencilReference(vk::StencilFaceFlagBits::eBack, back_ref, dld);
  911. cmdbuf.setStencilWriteMask(vk::StencilFaceFlagBits::eBack, back_write_mask, dld);
  912. cmdbuf.setStencilCompareMask(vk::StencilFaceFlagBits::eBack, back_test_mask, dld);
  913. });
  914. } else {
  915. // Front face defines both faces
  916. scheduler.Record([ref = regs.stencil_back_func_ref, write_mask = regs.stencil_back_mask,
  917. test_mask = regs.stencil_back_func_mask](auto cmdbuf, auto& dld) {
  918. cmdbuf.setStencilReference(vk::StencilFaceFlagBits::eFrontAndBack, ref, dld);
  919. cmdbuf.setStencilWriteMask(vk::StencilFaceFlagBits::eFrontAndBack, write_mask, dld);
  920. cmdbuf.setStencilCompareMask(vk::StencilFaceFlagBits::eFrontAndBack, test_mask, dld);
  921. });
  922. }
  923. }
  924. std::size_t RasterizerVulkan::CalculateGraphicsStreamBufferSize(bool is_indexed) const {
  925. std::size_t size = CalculateVertexArraysSize();
  926. if (is_indexed) {
  927. size = Common::AlignUp(size, 4) + CalculateIndexBufferSize();
  928. }
  929. size += Maxwell::MaxConstBuffers * (MaxConstbufferSize + device.GetUniformBufferAlignment());
  930. return size;
  931. }
  932. std::size_t RasterizerVulkan::CalculateComputeStreamBufferSize() const {
  933. return Tegra::Engines::KeplerCompute::NumConstBuffers *
  934. (Maxwell::MaxConstBufferSize + device.GetUniformBufferAlignment());
  935. }
  936. std::size_t RasterizerVulkan::CalculateVertexArraysSize() const {
  937. const auto& regs = system.GPU().Maxwell3D().regs;
  938. std::size_t size = 0;
  939. for (u32 index = 0; index < Maxwell::NumVertexArrays; ++index) {
  940. // This implementation assumes that all attributes are used in the shader.
  941. const GPUVAddr start{regs.vertex_array[index].StartAddress()};
  942. const GPUVAddr end{regs.vertex_array_limit[index].LimitAddress()};
  943. DEBUG_ASSERT(end > start);
  944. size += (end - start + 1) * regs.vertex_array[index].enable;
  945. }
  946. return size;
  947. }
  948. std::size_t RasterizerVulkan::CalculateIndexBufferSize() const {
  949. const auto& regs = system.GPU().Maxwell3D().regs;
  950. return static_cast<std::size_t>(regs.index_array.count) *
  951. static_cast<std::size_t>(regs.index_array.FormatSizeInBytes());
  952. }
  953. std::size_t RasterizerVulkan::CalculateConstBufferSize(
  954. const ConstBufferEntry& entry, const Tegra::Engines::ConstBufferInfo& buffer) const {
  955. if (entry.IsIndirect()) {
  956. // Buffer is accessed indirectly, so upload the entire thing
  957. return buffer.size;
  958. } else {
  959. // Buffer is accessed directly, upload just what we use
  960. return entry.GetSize();
  961. }
  962. }
  963. RenderPassParams RasterizerVulkan::GetRenderPassParams(Texceptions texceptions) const {
  964. using namespace VideoCore::Surface;
  965. const auto& regs = system.GPU().Maxwell3D().regs;
  966. RenderPassParams renderpass_params;
  967. for (std::size_t rt = 0; rt < static_cast<std::size_t>(regs.rt_control.count); ++rt) {
  968. const auto& rendertarget = regs.rt[rt];
  969. if (rendertarget.Address() == 0 || rendertarget.format == Tegra::RenderTargetFormat::NONE) {
  970. continue;
  971. }
  972. renderpass_params.color_attachments.push_back(RenderPassParams::ColorAttachment{
  973. static_cast<u32>(rt), PixelFormatFromRenderTargetFormat(rendertarget.format),
  974. texceptions[rt]});
  975. }
  976. renderpass_params.has_zeta = regs.zeta_enable;
  977. if (renderpass_params.has_zeta) {
  978. renderpass_params.zeta_pixel_format = PixelFormatFromDepthFormat(regs.zeta.format);
  979. renderpass_params.zeta_texception = texceptions[ZETA_TEXCEPTION_INDEX];
  980. }
  981. return renderpass_params;
  982. }
  983. } // namespace Vulkan