texture_cache.h 73 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856
  1. // Copyright 2019 yuzu Emulator Project
  2. // Licensed under GPLv2 or any later version
  3. // Refer to the license.txt file included.
  4. #pragma once
  5. #include <unordered_set>
  6. #include "common/alignment.h"
  7. #include "common/settings.h"
  8. #include "video_core/dirty_flags.h"
  9. #include "video_core/engines/kepler_compute.h"
  10. #include "video_core/texture_cache/image_view_base.h"
  11. #include "video_core/texture_cache/samples_helper.h"
  12. #include "video_core/texture_cache/texture_cache_base.h"
  13. #include "video_core/texture_cache/util.h"
  14. namespace VideoCommon {
  15. using Tegra::Texture::SwizzleSource;
  16. using Tegra::Texture::TextureType;
  17. using Tegra::Texture::TICEntry;
  18. using Tegra::Texture::TSCEntry;
  19. using VideoCore::Surface::GetFormatType;
  20. using VideoCore::Surface::IsCopyCompatible;
  21. using VideoCore::Surface::PixelFormat;
  22. using VideoCore::Surface::PixelFormatFromDepthFormat;
  23. using VideoCore::Surface::PixelFormatFromRenderTargetFormat;
  24. using VideoCore::Surface::SurfaceType;
  25. using namespace Common::Literals;
  26. template <class P>
  27. TextureCache<P>::TextureCache(Runtime& runtime_, VideoCore::RasterizerInterface& rasterizer_,
  28. Tegra::Engines::Maxwell3D& maxwell3d_,
  29. Tegra::Engines::KeplerCompute& kepler_compute_,
  30. Tegra::MemoryManager& gpu_memory_)
  31. : runtime{runtime_}, rasterizer{rasterizer_}, maxwell3d{maxwell3d_},
  32. kepler_compute{kepler_compute_}, gpu_memory{gpu_memory_} {
  33. // Configure null sampler
  34. TSCEntry sampler_descriptor{};
  35. sampler_descriptor.min_filter.Assign(Tegra::Texture::TextureFilter::Linear);
  36. sampler_descriptor.mag_filter.Assign(Tegra::Texture::TextureFilter::Linear);
  37. sampler_descriptor.mipmap_filter.Assign(Tegra::Texture::TextureMipmapFilter::Linear);
  38. sampler_descriptor.cubemap_anisotropy.Assign(1);
  39. // Make sure the first index is reserved for the null resources
  40. // This way the null resource becomes a compile time constant
  41. void(slot_images.insert(NullImageParams{}));
  42. void(slot_image_views.insert(runtime, NullImageViewParams{}));
  43. void(slot_samplers.insert(runtime, sampler_descriptor));
  44. if constexpr (HAS_DEVICE_MEMORY_INFO) {
  45. const auto device_memory = runtime.GetDeviceLocalMemory();
  46. const u64 possible_expected_memory = (device_memory * 3) / 10;
  47. const u64 possible_critical_memory = (device_memory * 6) / 10;
  48. expected_memory = std::max(possible_expected_memory, DEFAULT_EXPECTED_MEMORY);
  49. critical_memory = std::max(possible_critical_memory, DEFAULT_CRITICAL_MEMORY);
  50. minimum_memory = 0;
  51. } else {
  52. // On OpenGL we can be more conservatives as the driver takes care.
  53. expected_memory = DEFAULT_EXPECTED_MEMORY + 512_MiB;
  54. critical_memory = DEFAULT_CRITICAL_MEMORY + 1_GiB;
  55. minimum_memory = expected_memory;
  56. }
  57. }
  58. template <class P>
  59. void TextureCache<P>::RunGarbageCollector() {
  60. const bool high_priority_mode = total_used_memory >= expected_memory;
  61. const bool aggressive_mode = total_used_memory >= critical_memory;
  62. const u64 ticks_to_destroy = aggressive_mode ? 10ULL : high_priority_mode ? 25ULL : 100ULL;
  63. size_t num_iterations = aggressive_mode ? 10000 : (high_priority_mode ? 100 : 5);
  64. const auto clean_up = [this, &num_iterations, high_priority_mode](ImageId image_id) {
  65. if (num_iterations == 0) {
  66. return true;
  67. }
  68. --num_iterations;
  69. auto& image = slot_images[image_id];
  70. const bool must_download = image.IsSafeDownload();
  71. if (!high_priority_mode && must_download) {
  72. return false;
  73. }
  74. if (must_download) {
  75. auto map = runtime.DownloadStagingBuffer(image.unswizzled_size_bytes);
  76. const auto copies = FullDownloadCopies(image.info);
  77. image.DownloadMemory(map, copies);
  78. runtime.Finish();
  79. SwizzleImage(gpu_memory, image.gpu_addr, image.info, copies, map.mapped_span);
  80. }
  81. if (True(image.flags & ImageFlagBits::Tracked)) {
  82. UntrackImage(image, image_id);
  83. }
  84. UnregisterImage(image_id);
  85. DeleteImage(image_id);
  86. return false;
  87. };
  88. lru_cache.ForEachItemBelow(frame_tick - ticks_to_destroy, clean_up);
  89. }
  90. template <class P>
  91. void TextureCache<P>::TickFrame() {
  92. if (total_used_memory > minimum_memory) {
  93. RunGarbageCollector();
  94. }
  95. sentenced_images.Tick();
  96. sentenced_framebuffers.Tick();
  97. sentenced_image_view.Tick();
  98. runtime.TickFrame();
  99. ++frame_tick;
  100. }
  101. template <class P>
  102. const typename P::ImageView& TextureCache<P>::GetImageView(ImageViewId id) const noexcept {
  103. return slot_image_views[id];
  104. }
  105. template <class P>
  106. typename P::ImageView& TextureCache<P>::GetImageView(ImageViewId id) noexcept {
  107. return slot_image_views[id];
  108. }
  109. template <class P>
  110. void TextureCache<P>::MarkModification(ImageId id) noexcept {
  111. MarkModification(slot_images[id]);
  112. }
  113. template <class P>
  114. template <bool has_blacklists>
  115. void TextureCache<P>::FillGraphicsImageViews(std::span<ImageViewInOut> views) {
  116. FillImageViews<has_blacklists>(graphics_image_table, graphics_image_view_ids, views);
  117. }
  118. template <class P>
  119. void TextureCache<P>::FillComputeImageViews(std::span<ImageViewInOut> views) {
  120. FillImageViews<true>(compute_image_table, compute_image_view_ids, views);
  121. }
  122. template <class P>
  123. typename P::Sampler* TextureCache<P>::GetGraphicsSampler(u32 index) {
  124. if (index > graphics_sampler_table.Limit()) {
  125. LOG_DEBUG(HW_GPU, "Invalid sampler index={}", index);
  126. return &slot_samplers[NULL_SAMPLER_ID];
  127. }
  128. const auto [descriptor, is_new] = graphics_sampler_table.Read(index);
  129. SamplerId& id = graphics_sampler_ids[index];
  130. if (is_new) {
  131. id = FindSampler(descriptor);
  132. }
  133. return &slot_samplers[id];
  134. }
  135. template <class P>
  136. typename P::Sampler* TextureCache<P>::GetComputeSampler(u32 index) {
  137. if (index > compute_sampler_table.Limit()) {
  138. LOG_DEBUG(HW_GPU, "Invalid sampler index={}", index);
  139. return &slot_samplers[NULL_SAMPLER_ID];
  140. }
  141. const auto [descriptor, is_new] = compute_sampler_table.Read(index);
  142. SamplerId& id = compute_sampler_ids[index];
  143. if (is_new) {
  144. id = FindSampler(descriptor);
  145. }
  146. return &slot_samplers[id];
  147. }
  148. template <class P>
  149. void TextureCache<P>::SynchronizeGraphicsDescriptors() {
  150. using SamplerIndex = Tegra::Engines::Maxwell3D::Regs::SamplerIndex;
  151. const bool linked_tsc = maxwell3d.regs.sampler_index == SamplerIndex::ViaHeaderIndex;
  152. const u32 tic_limit = maxwell3d.regs.tic.limit;
  153. const u32 tsc_limit = linked_tsc ? tic_limit : maxwell3d.regs.tsc.limit;
  154. if (graphics_sampler_table.Synchornize(maxwell3d.regs.tsc.Address(), tsc_limit)) {
  155. graphics_sampler_ids.resize(tsc_limit + 1, CORRUPT_ID);
  156. }
  157. if (graphics_image_table.Synchornize(maxwell3d.regs.tic.Address(), tic_limit)) {
  158. graphics_image_view_ids.resize(tic_limit + 1, CORRUPT_ID);
  159. }
  160. }
  161. template <class P>
  162. void TextureCache<P>::SynchronizeComputeDescriptors() {
  163. const bool linked_tsc = kepler_compute.launch_description.linked_tsc;
  164. const u32 tic_limit = kepler_compute.regs.tic.limit;
  165. const u32 tsc_limit = linked_tsc ? tic_limit : kepler_compute.regs.tsc.limit;
  166. const GPUVAddr tsc_gpu_addr = kepler_compute.regs.tsc.Address();
  167. if (compute_sampler_table.Synchornize(tsc_gpu_addr, tsc_limit)) {
  168. compute_sampler_ids.resize(tsc_limit + 1, CORRUPT_ID);
  169. }
  170. if (compute_image_table.Synchornize(kepler_compute.regs.tic.Address(), tic_limit)) {
  171. compute_image_view_ids.resize(tic_limit + 1, CORRUPT_ID);
  172. }
  173. }
  174. template <class P>
  175. void TextureCache<P>::UpdateRenderTargets(bool is_clear) {
  176. using namespace VideoCommon::Dirty;
  177. auto& flags = maxwell3d.dirty.flags;
  178. if (!flags[Dirty::RenderTargets]) {
  179. for (size_t index = 0; index < NUM_RT; ++index) {
  180. ImageViewId& color_buffer_id = render_targets.color_buffer_ids[index];
  181. PrepareImageView(color_buffer_id, true, is_clear && IsFullClear(color_buffer_id));
  182. }
  183. const ImageViewId depth_buffer_id = render_targets.depth_buffer_id;
  184. PrepareImageView(depth_buffer_id, true, is_clear && IsFullClear(depth_buffer_id));
  185. return;
  186. }
  187. u32 scale_rating = 0;
  188. bool rescaled = false;
  189. std::array<ImageId, NUM_RT> tmp_color_images{};
  190. ImageId tmp_depth_image{};
  191. do {
  192. flags[Dirty::RenderTargets] = false;
  193. has_deleted_images = false;
  194. // Render target control is used on all render targets, so force look ups when this one is
  195. // up
  196. const bool force = flags[Dirty::RenderTargetControl];
  197. flags[Dirty::RenderTargetControl] = false;
  198. scale_rating = 0;
  199. bool any_rescaled = false;
  200. bool can_rescale = true;
  201. bool any_blacklisted = false;
  202. const auto check_rescale = [&](ImageViewId view_id, ImageId& id_save) {
  203. if (view_id != NULL_IMAGE_VIEW_ID && view_id != ImageViewId{}) {
  204. const auto& view = slot_image_views[view_id];
  205. const auto image_id = view.image_id;
  206. id_save = image_id;
  207. auto& image = slot_images[image_id];
  208. can_rescale &= ImageCanRescale(image);
  209. any_blacklisted |= True(image.flags & ImageFlagBits::Blacklisted);
  210. any_rescaled |= True(image.flags & ImageFlagBits::Rescaled);
  211. scale_rating = std::max<u32>(scale_rating, image.scale_tick <= frame_tick
  212. ? image.scale_rating + 1U
  213. : image.scale_rating);
  214. } else {
  215. id_save = CORRUPT_ID;
  216. }
  217. };
  218. for (size_t index = 0; index < NUM_RT; ++index) {
  219. ImageViewId& color_buffer_id = render_targets.color_buffer_ids[index];
  220. if (flags[Dirty::ColorBuffer0 + index] || force) {
  221. flags[Dirty::ColorBuffer0 + index] = false;
  222. BindRenderTarget(&color_buffer_id, FindColorBuffer(index, is_clear));
  223. }
  224. check_rescale(color_buffer_id, tmp_color_images[index]);
  225. }
  226. if (flags[Dirty::ZetaBuffer] || force) {
  227. flags[Dirty::ZetaBuffer] = false;
  228. BindRenderTarget(&render_targets.depth_buffer_id, FindDepthBuffer(is_clear));
  229. }
  230. check_rescale(render_targets.depth_buffer_id, tmp_depth_image);
  231. if (can_rescale) {
  232. rescaled = any_rescaled || scale_rating >= 2;
  233. const auto scale_up = [this](ImageId image_id) {
  234. if (image_id != CORRUPT_ID) {
  235. Image& image = slot_images[image_id];
  236. ScaleUp(image);
  237. }
  238. };
  239. if (rescaled) {
  240. for (size_t index = 0; index < NUM_RT; ++index) {
  241. scale_up(tmp_color_images[index]);
  242. }
  243. scale_up(tmp_depth_image);
  244. scale_rating = 2;
  245. }
  246. } else {
  247. rescaled = false;
  248. const auto scale_down = [this, any_blacklisted](ImageId image_id) {
  249. if (image_id != CORRUPT_ID) {
  250. Image& image = slot_images[image_id];
  251. ScaleDown(image);
  252. if (any_blacklisted) {
  253. image.flags |= ImageFlagBits::Blacklisted;
  254. }
  255. }
  256. };
  257. for (size_t index = 0; index < NUM_RT; ++index) {
  258. scale_down(tmp_color_images[index]);
  259. }
  260. scale_down(tmp_depth_image);
  261. scale_rating = 0;
  262. }
  263. } while (has_deleted_images);
  264. // Rescale End
  265. const auto set_rating = [this, scale_rating](ImageId image_id) {
  266. if (image_id != CORRUPT_ID) {
  267. Image& image = slot_images[image_id];
  268. image.scale_rating = scale_rating;
  269. image.scale_tick = frame_tick + 1;
  270. }
  271. };
  272. for (size_t index = 0; index < NUM_RT; ++index) {
  273. set_rating(tmp_color_images[index]);
  274. }
  275. set_rating(tmp_depth_image);
  276. if (is_rescaling != rescaled) {
  277. flags[Dirty::RescaleViewports] = true;
  278. flags[Dirty::RescaleScissors] = true;
  279. is_rescaling = rescaled;
  280. }
  281. for (size_t index = 0; index < NUM_RT; ++index) {
  282. ImageViewId& color_buffer_id = render_targets.color_buffer_ids[index];
  283. PrepareImageView(color_buffer_id, true, is_clear && IsFullClear(color_buffer_id));
  284. }
  285. const ImageViewId depth_buffer_id = render_targets.depth_buffer_id;
  286. PrepareImageView(depth_buffer_id, true, is_clear && IsFullClear(depth_buffer_id));
  287. for (size_t index = 0; index < NUM_RT; ++index) {
  288. render_targets.draw_buffers[index] = static_cast<u8>(maxwell3d.regs.rt_control.Map(index));
  289. }
  290. u32 up_scale = 1;
  291. u32 down_shift = 0;
  292. if (is_rescaling) {
  293. up_scale = Settings::values.resolution_info.up_scale;
  294. down_shift = Settings::values.resolution_info.down_shift;
  295. }
  296. render_targets.size = Extent2D{
  297. (maxwell3d.regs.render_area.width * up_scale) >> down_shift,
  298. (maxwell3d.regs.render_area.height * up_scale) >> down_shift,
  299. };
  300. flags[Dirty::DepthBiasGlobal] = true;
  301. }
  302. template <class P>
  303. typename P::Framebuffer* TextureCache<P>::GetFramebuffer() {
  304. return &slot_framebuffers[GetFramebufferId(render_targets)];
  305. }
  306. template <class P>
  307. template <bool has_blacklists>
  308. void TextureCache<P>::FillImageViews(DescriptorTable<TICEntry>& table,
  309. std::span<ImageViewId> cached_image_view_ids,
  310. std::span<ImageViewInOut> views) {
  311. bool has_blacklisted;
  312. do {
  313. has_deleted_images = false;
  314. if constexpr (has_blacklists) {
  315. has_blacklisted = false;
  316. }
  317. for (ImageViewInOut& view : views) {
  318. view.id = VisitImageView(table, cached_image_view_ids, view.index);
  319. if constexpr (has_blacklists) {
  320. if (view.blacklist && view.id != NULL_IMAGE_VIEW_ID) {
  321. const ImageViewBase& image_view{slot_image_views[view.id]};
  322. has_blacklisted |= BlackListImage(image_view.image_id);
  323. }
  324. }
  325. }
  326. } while (has_deleted_images || (has_blacklists && has_blacklisted));
  327. }
  328. template <class P>
  329. ImageViewId TextureCache<P>::VisitImageView(DescriptorTable<TICEntry>& table,
  330. std::span<ImageViewId> cached_image_view_ids,
  331. u32 index) {
  332. if (index > table.Limit()) {
  333. LOG_DEBUG(HW_GPU, "Invalid image view index={}", index);
  334. return NULL_IMAGE_VIEW_ID;
  335. }
  336. const auto [descriptor, is_new] = table.Read(index);
  337. ImageViewId& image_view_id = cached_image_view_ids[index];
  338. if (is_new) {
  339. image_view_id = FindImageView(descriptor);
  340. }
  341. if (image_view_id != NULL_IMAGE_VIEW_ID) {
  342. PrepareImageView(image_view_id, false, false);
  343. }
  344. return image_view_id;
  345. }
  346. template <class P>
  347. FramebufferId TextureCache<P>::GetFramebufferId(const RenderTargets& key) {
  348. const auto [pair, is_new] = framebuffers.try_emplace(key);
  349. FramebufferId& framebuffer_id = pair->second;
  350. if (!is_new) {
  351. return framebuffer_id;
  352. }
  353. std::array<ImageView*, NUM_RT> color_buffers;
  354. std::ranges::transform(key.color_buffer_ids, color_buffers.begin(),
  355. [this](ImageViewId id) { return id ? &slot_image_views[id] : nullptr; });
  356. ImageView* const depth_buffer =
  357. key.depth_buffer_id ? &slot_image_views[key.depth_buffer_id] : nullptr;
  358. framebuffer_id = slot_framebuffers.insert(runtime, color_buffers, depth_buffer, key);
  359. return framebuffer_id;
  360. }
  361. template <class P>
  362. void TextureCache<P>::WriteMemory(VAddr cpu_addr, size_t size) {
  363. ForEachImageInRegion(cpu_addr, size, [this](ImageId image_id, Image& image) {
  364. if (True(image.flags & ImageFlagBits::CpuModified)) {
  365. return;
  366. }
  367. image.flags |= ImageFlagBits::CpuModified;
  368. if (True(image.flags & ImageFlagBits::Tracked)) {
  369. UntrackImage(image, image_id);
  370. }
  371. });
  372. }
  373. template <class P>
  374. void TextureCache<P>::DownloadMemory(VAddr cpu_addr, size_t size) {
  375. std::vector<ImageId> images;
  376. ForEachImageInRegion(cpu_addr, size, [this, &images](ImageId image_id, ImageBase& image) {
  377. if (!image.IsSafeDownload()) {
  378. return;
  379. }
  380. image.flags &= ~ImageFlagBits::GpuModified;
  381. images.push_back(image_id);
  382. });
  383. if (images.empty()) {
  384. return;
  385. }
  386. std::ranges::sort(images, [this](ImageId lhs, ImageId rhs) {
  387. return slot_images[lhs].modification_tick < slot_images[rhs].modification_tick;
  388. });
  389. for (const ImageId image_id : images) {
  390. Image& image = slot_images[image_id];
  391. auto map = runtime.DownloadStagingBuffer(image.unswizzled_size_bytes);
  392. const auto copies = FullDownloadCopies(image.info);
  393. image.DownloadMemory(map, copies);
  394. runtime.Finish();
  395. SwizzleImage(gpu_memory, image.gpu_addr, image.info, copies, map.mapped_span);
  396. }
  397. }
  398. template <class P>
  399. void TextureCache<P>::UnmapMemory(VAddr cpu_addr, size_t size) {
  400. std::vector<ImageId> deleted_images;
  401. ForEachImageInRegion(cpu_addr, size, [&](ImageId id, Image&) { deleted_images.push_back(id); });
  402. for (const ImageId id : deleted_images) {
  403. Image& image = slot_images[id];
  404. if (True(image.flags & ImageFlagBits::Tracked)) {
  405. UntrackImage(image, id);
  406. }
  407. UnregisterImage(id);
  408. DeleteImage(id);
  409. }
  410. }
  411. template <class P>
  412. void TextureCache<P>::UnmapGPUMemory(GPUVAddr gpu_addr, size_t size) {
  413. std::vector<ImageId> deleted_images;
  414. ForEachImageInRegionGPU(gpu_addr, size,
  415. [&](ImageId id, Image&) { deleted_images.push_back(id); });
  416. for (const ImageId id : deleted_images) {
  417. Image& image = slot_images[id];
  418. if (True(image.flags & ImageFlagBits::Remapped)) {
  419. continue;
  420. }
  421. image.flags |= ImageFlagBits::Remapped;
  422. if (True(image.flags & ImageFlagBits::Tracked)) {
  423. UntrackImage(image, id);
  424. }
  425. }
  426. }
  427. template <class P>
  428. void TextureCache<P>::BlitImage(const Tegra::Engines::Fermi2D::Surface& dst,
  429. const Tegra::Engines::Fermi2D::Surface& src,
  430. const Tegra::Engines::Fermi2D::Config& copy) {
  431. const BlitImages images = GetBlitImages(dst, src);
  432. const ImageId dst_id = images.dst_id;
  433. const ImageId src_id = images.src_id;
  434. PrepareImage(src_id, false, false);
  435. PrepareImage(dst_id, true, false);
  436. Image& dst_image = slot_images[dst_id];
  437. Image& src_image = slot_images[src_id];
  438. bool is_src_rescaled = True(src_image.flags & ImageFlagBits::Rescaled);
  439. bool is_dst_rescaled = True(dst_image.flags & ImageFlagBits::Rescaled);
  440. const bool is_resolve = src_image.info.num_samples != 1 && dst_image.info.num_samples == 1;
  441. if (is_src_rescaled != is_dst_rescaled) {
  442. if (ImageCanRescale(src_image)) {
  443. ScaleUp(src_image);
  444. is_src_rescaled = True(src_image.flags & ImageFlagBits::Rescaled);
  445. if (is_resolve) {
  446. dst_image.info.rescaleable = true;
  447. for (const auto& alias : dst_image.aliased_images) {
  448. Image& other_image = slot_images[alias.id];
  449. other_image.info.rescaleable = true;
  450. }
  451. }
  452. }
  453. if (ImageCanRescale(dst_image)) {
  454. ScaleUp(dst_image);
  455. is_dst_rescaled = True(dst_image.flags & ImageFlagBits::Rescaled);
  456. }
  457. }
  458. if (is_resolve && (is_src_rescaled != is_dst_rescaled)) {
  459. // A resolve requires both images to be the same dimensions. Resize down if needed.
  460. ScaleDown(src_image);
  461. ScaleDown(dst_image);
  462. is_src_rescaled = True(src_image.flags & ImageFlagBits::Rescaled);
  463. is_dst_rescaled = True(dst_image.flags & ImageFlagBits::Rescaled);
  464. }
  465. const auto& resolution = Settings::values.resolution_info;
  466. const auto scale_up = [&](u32 value) -> u32 {
  467. if (value == 0) {
  468. return 0U;
  469. }
  470. return std::max<u32>((value * resolution.up_scale) >> resolution.down_shift, 1U);
  471. };
  472. const auto scale_region = [&](Region2D& region) {
  473. region.start.x = scale_up(region.start.x);
  474. region.start.y = scale_up(region.start.y);
  475. region.end.x = scale_up(region.end.x);
  476. region.end.y = scale_up(region.end.y);
  477. };
  478. // TODO: Deduplicate
  479. const std::optional src_base = src_image.TryFindBase(src.Address());
  480. const SubresourceRange src_range{.base = src_base.value(), .extent = {1, 1}};
  481. const ImageViewInfo src_view_info(ImageViewType::e2D, images.src_format, src_range);
  482. const auto [src_framebuffer_id, src_view_id] = RenderTargetFromImage(src_id, src_view_info);
  483. const auto [src_samples_x, src_samples_y] = SamplesLog2(src_image.info.num_samples);
  484. Region2D src_region{
  485. Offset2D{.x = copy.src_x0 >> src_samples_x, .y = copy.src_y0 >> src_samples_y},
  486. Offset2D{.x = copy.src_x1 >> src_samples_x, .y = copy.src_y1 >> src_samples_y},
  487. };
  488. if (is_src_rescaled) {
  489. scale_region(src_region);
  490. }
  491. const std::optional dst_base = dst_image.TryFindBase(dst.Address());
  492. const SubresourceRange dst_range{.base = dst_base.value(), .extent = {1, 1}};
  493. const ImageViewInfo dst_view_info(ImageViewType::e2D, images.dst_format, dst_range);
  494. const auto [dst_framebuffer_id, dst_view_id] = RenderTargetFromImage(dst_id, dst_view_info);
  495. const auto [dst_samples_x, dst_samples_y] = SamplesLog2(dst_image.info.num_samples);
  496. Region2D dst_region{
  497. Offset2D{.x = copy.dst_x0 >> dst_samples_x, .y = copy.dst_y0 >> dst_samples_y},
  498. Offset2D{.x = copy.dst_x1 >> dst_samples_x, .y = copy.dst_y1 >> dst_samples_y},
  499. };
  500. if (is_dst_rescaled) {
  501. scale_region(dst_region);
  502. }
  503. // Always call this after src_framebuffer_id was queried, as the address might be invalidated.
  504. Framebuffer* const dst_framebuffer = &slot_framebuffers[dst_framebuffer_id];
  505. if constexpr (FRAMEBUFFER_BLITS) {
  506. // OpenGL blits from framebuffers, not images
  507. Framebuffer* const src_framebuffer = &slot_framebuffers[src_framebuffer_id];
  508. runtime.BlitFramebuffer(dst_framebuffer, src_framebuffer, dst_region, src_region,
  509. copy.filter, copy.operation);
  510. } else {
  511. // Vulkan can blit images, but it lacks format reinterpretations
  512. // Provide a framebuffer in case it's necessary
  513. ImageView& dst_view = slot_image_views[dst_view_id];
  514. ImageView& src_view = slot_image_views[src_view_id];
  515. runtime.BlitImage(dst_framebuffer, dst_view, src_view, dst_region, src_region, copy.filter,
  516. copy.operation);
  517. }
  518. }
  519. template <class P>
  520. typename P::ImageView* TextureCache<P>::TryFindFramebufferImageView(VAddr cpu_addr) {
  521. // TODO: Properly implement this
  522. const auto it = page_table.find(cpu_addr >> PAGE_BITS);
  523. if (it == page_table.end()) {
  524. return nullptr;
  525. }
  526. const auto& image_map_ids = it->second;
  527. for (const ImageMapId map_id : image_map_ids) {
  528. const ImageMapView& map = slot_map_views[map_id];
  529. const ImageBase& image = slot_images[map.image_id];
  530. if (image.cpu_addr != cpu_addr) {
  531. continue;
  532. }
  533. if (image.image_view_ids.empty()) {
  534. continue;
  535. }
  536. return &slot_image_views[image.image_view_ids.at(0)];
  537. }
  538. return nullptr;
  539. }
  540. template <class P>
  541. bool TextureCache<P>::HasUncommittedFlushes() const noexcept {
  542. return !uncommitted_downloads.empty();
  543. }
  544. template <class P>
  545. bool TextureCache<P>::ShouldWaitAsyncFlushes() const noexcept {
  546. return !committed_downloads.empty() && !committed_downloads.front().empty();
  547. }
  548. template <class P>
  549. void TextureCache<P>::CommitAsyncFlushes() {
  550. // This is intentionally passing the value by copy
  551. committed_downloads.push(uncommitted_downloads);
  552. uncommitted_downloads.clear();
  553. }
  554. template <class P>
  555. void TextureCache<P>::PopAsyncFlushes() {
  556. if (committed_downloads.empty()) {
  557. return;
  558. }
  559. const std::span<const ImageId> download_ids = committed_downloads.front();
  560. if (download_ids.empty()) {
  561. committed_downloads.pop();
  562. return;
  563. }
  564. size_t total_size_bytes = 0;
  565. for (const ImageId image_id : download_ids) {
  566. total_size_bytes += slot_images[image_id].unswizzled_size_bytes;
  567. }
  568. auto download_map = runtime.DownloadStagingBuffer(total_size_bytes);
  569. const size_t original_offset = download_map.offset;
  570. for (const ImageId image_id : download_ids) {
  571. Image& image = slot_images[image_id];
  572. const auto copies = FullDownloadCopies(image.info);
  573. image.DownloadMemory(download_map, copies);
  574. download_map.offset += image.unswizzled_size_bytes;
  575. }
  576. // Wait for downloads to finish
  577. runtime.Finish();
  578. download_map.offset = original_offset;
  579. std::span<u8> download_span = download_map.mapped_span;
  580. for (const ImageId image_id : download_ids) {
  581. const ImageBase& image = slot_images[image_id];
  582. const auto copies = FullDownloadCopies(image.info);
  583. SwizzleImage(gpu_memory, image.gpu_addr, image.info, copies, download_span);
  584. download_map.offset += image.unswizzled_size_bytes;
  585. download_span = download_span.subspan(image.unswizzled_size_bytes);
  586. }
  587. committed_downloads.pop();
  588. }
  589. template <class P>
  590. bool TextureCache<P>::IsRescaling() const noexcept {
  591. return is_rescaling;
  592. }
  593. template <class P>
  594. bool TextureCache<P>::IsRescaling(const ImageViewBase& image_view) const noexcept {
  595. if (image_view.type == ImageViewType::Buffer) {
  596. return false;
  597. }
  598. const ImageBase& image = slot_images[image_view.image_id];
  599. return True(image.flags & ImageFlagBits::Rescaled);
  600. }
  601. template <class P>
  602. bool TextureCache<P>::IsRegionGpuModified(VAddr addr, size_t size) {
  603. bool is_modified = false;
  604. ForEachImageInRegion(addr, size, [&is_modified](ImageId, ImageBase& image) {
  605. if (False(image.flags & ImageFlagBits::GpuModified)) {
  606. return false;
  607. }
  608. is_modified = true;
  609. return true;
  610. });
  611. return is_modified;
  612. }
  613. template <class P>
  614. void TextureCache<P>::RefreshContents(Image& image, ImageId image_id) {
  615. if (False(image.flags & ImageFlagBits::CpuModified)) {
  616. // Only upload modified images
  617. return;
  618. }
  619. image.flags &= ~ImageFlagBits::CpuModified;
  620. TrackImage(image, image_id);
  621. if (image.info.num_samples > 1) {
  622. LOG_WARNING(HW_GPU, "MSAA image uploads are not implemented");
  623. return;
  624. }
  625. auto staging = runtime.UploadStagingBuffer(MapSizeBytes(image));
  626. UploadImageContents(image, staging);
  627. runtime.InsertUploadMemoryBarrier();
  628. }
  629. template <class P>
  630. template <typename StagingBuffer>
  631. void TextureCache<P>::UploadImageContents(Image& image, StagingBuffer& staging) {
  632. const std::span<u8> mapped_span = staging.mapped_span;
  633. const GPUVAddr gpu_addr = image.gpu_addr;
  634. if (True(image.flags & ImageFlagBits::AcceleratedUpload)) {
  635. gpu_memory.ReadBlockUnsafe(gpu_addr, mapped_span.data(), mapped_span.size_bytes());
  636. const auto uploads = FullUploadSwizzles(image.info);
  637. runtime.AccelerateImageUpload(image, staging, uploads);
  638. } else if (True(image.flags & ImageFlagBits::Converted)) {
  639. std::vector<u8> unswizzled_data(image.unswizzled_size_bytes);
  640. auto copies = UnswizzleImage(gpu_memory, gpu_addr, image.info, unswizzled_data);
  641. ConvertImage(unswizzled_data, image.info, mapped_span, copies);
  642. image.UploadMemory(staging, copies);
  643. } else {
  644. const auto copies = UnswizzleImage(gpu_memory, gpu_addr, image.info, mapped_span);
  645. image.UploadMemory(staging, copies);
  646. }
  647. }
  648. template <class P>
  649. ImageViewId TextureCache<P>::FindImageView(const TICEntry& config) {
  650. if (!IsValidEntry(gpu_memory, config)) {
  651. return NULL_IMAGE_VIEW_ID;
  652. }
  653. const auto [pair, is_new] = image_views.try_emplace(config);
  654. ImageViewId& image_view_id = pair->second;
  655. if (is_new) {
  656. image_view_id = CreateImageView(config);
  657. }
  658. return image_view_id;
  659. }
  660. template <class P>
  661. ImageViewId TextureCache<P>::CreateImageView(const TICEntry& config) {
  662. const ImageInfo info(config);
  663. if (info.type == ImageType::Buffer) {
  664. const ImageViewInfo view_info(config, 0);
  665. return slot_image_views.insert(runtime, info, view_info, config.Address());
  666. }
  667. const u32 layer_offset = config.BaseLayer() * info.layer_stride;
  668. const GPUVAddr image_gpu_addr = config.Address() - layer_offset;
  669. const ImageId image_id = FindOrInsertImage(info, image_gpu_addr);
  670. if (!image_id) {
  671. return NULL_IMAGE_VIEW_ID;
  672. }
  673. ImageBase& image = slot_images[image_id];
  674. const SubresourceBase base = image.TryFindBase(config.Address()).value();
  675. ASSERT(base.level == 0);
  676. const ImageViewInfo view_info(config, base.layer);
  677. const ImageViewId image_view_id = FindOrEmplaceImageView(image_id, view_info);
  678. ImageViewBase& image_view = slot_image_views[image_view_id];
  679. image_view.flags |= ImageViewFlagBits::Strong;
  680. image.flags |= ImageFlagBits::Strong;
  681. return image_view_id;
  682. }
  683. template <class P>
  684. ImageId TextureCache<P>::FindOrInsertImage(const ImageInfo& info, GPUVAddr gpu_addr,
  685. RelaxedOptions options) {
  686. if (const ImageId image_id = FindImage(info, gpu_addr, options); image_id) {
  687. return image_id;
  688. }
  689. return InsertImage(info, gpu_addr, options);
  690. }
  691. template <class P>
  692. ImageId TextureCache<P>::FindImage(const ImageInfo& info, GPUVAddr gpu_addr,
  693. RelaxedOptions options) {
  694. std::optional<VAddr> cpu_addr = gpu_memory.GpuToCpuAddress(gpu_addr);
  695. if (!cpu_addr) {
  696. cpu_addr = gpu_memory.GpuToCpuAddress(gpu_addr, CalculateGuestSizeInBytes(info));
  697. if (!cpu_addr) {
  698. return ImageId{};
  699. }
  700. }
  701. const bool broken_views = runtime.HasBrokenTextureViewFormats();
  702. const bool native_bgr = runtime.HasNativeBgr();
  703. ImageId image_id;
  704. const auto lambda = [&](ImageId existing_image_id, ImageBase& existing_image) {
  705. if (True(existing_image.flags & ImageFlagBits::Remapped)) {
  706. return false;
  707. }
  708. if (info.type == ImageType::Linear || existing_image.info.type == ImageType::Linear) {
  709. const bool strict_size = False(options & RelaxedOptions::Size) &&
  710. True(existing_image.flags & ImageFlagBits::Strong);
  711. const ImageInfo& existing = existing_image.info;
  712. if (existing_image.gpu_addr == gpu_addr && existing.type == info.type &&
  713. existing.pitch == info.pitch &&
  714. IsPitchLinearSameSize(existing, info, strict_size) &&
  715. IsViewCompatible(existing.format, info.format, broken_views, native_bgr)) {
  716. image_id = existing_image_id;
  717. return true;
  718. }
  719. } else if (IsSubresource(info, existing_image, gpu_addr, options, broken_views,
  720. native_bgr)) {
  721. image_id = existing_image_id;
  722. return true;
  723. }
  724. return false;
  725. };
  726. ForEachImageInRegion(*cpu_addr, CalculateGuestSizeInBytes(info), lambda);
  727. return image_id;
  728. }
  729. template <class P>
  730. bool TextureCache<P>::BlackListImage(ImageId image_id) {
  731. auto& image = slot_images[image_id];
  732. if (True(image.flags & ImageFlagBits::Blacklisted)) {
  733. return false;
  734. }
  735. image.flags |= ImageFlagBits::Blacklisted;
  736. ScaleDown(image);
  737. return true;
  738. }
  739. template <class P>
  740. bool TextureCache<P>::ImageCanRescale(ImageBase& image) {
  741. if (True(image.flags & ImageFlagBits::Blacklisted)) {
  742. return false;
  743. }
  744. if (True(image.flags & (ImageFlagBits::Rescaled | ImageFlagBits::RescaleChecked))) {
  745. return true;
  746. }
  747. if (!image.info.rescaleable) {
  748. image.flags &= ~ImageFlagBits::RescaleChecked;
  749. return false;
  750. }
  751. image.flags |= ImageFlagBits::RescaleChecked;
  752. for (const auto& alias : image.aliased_images) {
  753. Image& other_image = slot_images[alias.id];
  754. if (!ImageCanRescale(other_image)) {
  755. image.flags &= ~ImageFlagBits::RescaleChecked;
  756. return false;
  757. }
  758. }
  759. image.flags &= ~ImageFlagBits::RescaleChecked;
  760. return true;
  761. }
  762. template <class P>
  763. void TextureCache<P>::InvalidateScale(Image& image) {
  764. const std::span<const ImageViewId> image_view_ids = image.image_view_ids;
  765. auto& dirty = maxwell3d.dirty.flags;
  766. dirty[Dirty::RenderTargets] = true;
  767. dirty[Dirty::ZetaBuffer] = true;
  768. for (size_t rt = 0; rt < NUM_RT; ++rt) {
  769. dirty[Dirty::ColorBuffer0 + rt] = true;
  770. }
  771. for (const ImageViewId image_view_id : image_view_ids) {
  772. std::ranges::replace(render_targets.color_buffer_ids, image_view_id, ImageViewId{});
  773. if (render_targets.depth_buffer_id == image_view_id) {
  774. render_targets.depth_buffer_id = ImageViewId{};
  775. }
  776. }
  777. RemoveImageViewReferences(image_view_ids);
  778. RemoveFramebuffers(image_view_ids);
  779. for (const ImageViewId image_view_id : image_view_ids) {
  780. sentenced_image_view.Push(std::move(slot_image_views[image_view_id]));
  781. slot_image_views.erase(image_view_id);
  782. }
  783. image.image_view_ids.clear();
  784. image.image_view_infos.clear();
  785. if constexpr (ENABLE_VALIDATION) {
  786. std::ranges::fill(graphics_image_view_ids, CORRUPT_ID);
  787. std::ranges::fill(compute_image_view_ids, CORRUPT_ID);
  788. }
  789. graphics_image_table.Invalidate();
  790. compute_image_table.Invalidate();
  791. has_deleted_images = true;
  792. }
  793. template <class P>
  794. u64 TextureCache<P>::GetScaledImageSizeBytes(Image& image) {
  795. const f32 add_to_size = Settings::values.resolution_info.up_factor - 1.0f;
  796. const bool sign = std::signbit(add_to_size);
  797. const u32 image_size_bytes = std::max(image.guest_size_bytes, image.unswizzled_size_bytes);
  798. const u64 tentative_size = image_size_bytes * static_cast<u32>(std::abs(add_to_size));
  799. const u64 fitted_size = Common::AlignUp(tentative_size, 1024);
  800. return sign ? -fitted_size : fitted_size;
  801. }
  802. template <class P>
  803. bool TextureCache<P>::ScaleUp(Image& image) {
  804. const bool rescaled = image.ScaleUp();
  805. if (!rescaled) {
  806. return false;
  807. }
  808. total_used_memory += GetScaledImageSizeBytes(image);
  809. InvalidateScale(image);
  810. return true;
  811. }
  812. template <class P>
  813. bool TextureCache<P>::ScaleDown(Image& image) {
  814. const bool rescaled = image.ScaleDown();
  815. if (!rescaled) {
  816. return false;
  817. }
  818. total_used_memory += GetScaledImageSizeBytes(image);
  819. InvalidateScale(image);
  820. return true;
  821. }
  822. template <class P>
  823. ImageId TextureCache<P>::InsertImage(const ImageInfo& info, GPUVAddr gpu_addr,
  824. RelaxedOptions options) {
  825. std::optional<VAddr> cpu_addr = gpu_memory.GpuToCpuAddress(gpu_addr);
  826. if (!cpu_addr) {
  827. const auto size = CalculateGuestSizeInBytes(info);
  828. cpu_addr = gpu_memory.GpuToCpuAddress(gpu_addr, size);
  829. if (!cpu_addr) {
  830. const VAddr fake_addr = ~(1ULL << 40ULL) + virtual_invalid_space;
  831. virtual_invalid_space += Common::AlignUp(size, 32);
  832. cpu_addr = std::optional<VAddr>(fake_addr);
  833. }
  834. }
  835. ASSERT_MSG(cpu_addr, "Tried to insert an image to an invalid gpu_addr=0x{:x}", gpu_addr);
  836. const ImageId image_id = JoinImages(info, gpu_addr, *cpu_addr);
  837. const Image& image = slot_images[image_id];
  838. // Using "image.gpu_addr" instead of "gpu_addr" is important because it might be different
  839. const auto [it, is_new] = image_allocs_table.try_emplace(image.gpu_addr);
  840. if (is_new) {
  841. it->second = slot_image_allocs.insert();
  842. }
  843. slot_image_allocs[it->second].images.push_back(image_id);
  844. return image_id;
  845. }
  846. template <class P>
  847. ImageId TextureCache<P>::JoinImages(const ImageInfo& info, GPUVAddr gpu_addr, VAddr cpu_addr) {
  848. ImageInfo new_info = info;
  849. const size_t size_bytes = CalculateGuestSizeInBytes(new_info);
  850. const bool broken_views = runtime.HasBrokenTextureViewFormats();
  851. const bool native_bgr = runtime.HasNativeBgr();
  852. std::vector<ImageId> overlap_ids;
  853. std::unordered_set<ImageId> overlaps_found;
  854. std::vector<ImageId> left_aliased_ids;
  855. std::vector<ImageId> right_aliased_ids;
  856. std::unordered_set<ImageId> ignore_textures;
  857. std::vector<ImageId> bad_overlap_ids;
  858. std::vector<ImageId> all_siblings;
  859. const bool this_is_linear = info.type == ImageType::Linear;
  860. const auto region_check = [&](ImageId overlap_id, ImageBase& overlap) {
  861. if (True(overlap.flags & ImageFlagBits::Remapped)) {
  862. ignore_textures.insert(overlap_id);
  863. return;
  864. }
  865. const bool overlap_is_linear = overlap.info.type == ImageType::Linear;
  866. if (this_is_linear != overlap_is_linear) {
  867. return;
  868. }
  869. if (this_is_linear && overlap_is_linear) {
  870. if (info.pitch == overlap.info.pitch && gpu_addr == overlap.gpu_addr) {
  871. // Alias linear images with the same pitch
  872. left_aliased_ids.push_back(overlap_id);
  873. }
  874. return;
  875. }
  876. overlaps_found.insert(overlap_id);
  877. static constexpr bool strict_size = true;
  878. const std::optional<OverlapResult> solution = ResolveOverlap(
  879. new_info, gpu_addr, cpu_addr, overlap, strict_size, broken_views, native_bgr);
  880. if (solution) {
  881. gpu_addr = solution->gpu_addr;
  882. cpu_addr = solution->cpu_addr;
  883. new_info.resources = solution->resources;
  884. overlap_ids.push_back(overlap_id);
  885. all_siblings.push_back(overlap_id);
  886. return;
  887. }
  888. static constexpr auto options = RelaxedOptions::Size | RelaxedOptions::Format;
  889. const ImageBase new_image_base(new_info, gpu_addr, cpu_addr);
  890. if (IsSubresource(new_info, overlap, gpu_addr, options, broken_views, native_bgr)) {
  891. left_aliased_ids.push_back(overlap_id);
  892. overlap.flags |= ImageFlagBits::Alias;
  893. all_siblings.push_back(overlap_id);
  894. } else if (IsSubresource(overlap.info, new_image_base, overlap.gpu_addr, options,
  895. broken_views, native_bgr)) {
  896. right_aliased_ids.push_back(overlap_id);
  897. overlap.flags |= ImageFlagBits::Alias;
  898. all_siblings.push_back(overlap_id);
  899. } else {
  900. bad_overlap_ids.push_back(overlap_id);
  901. overlap.flags |= ImageFlagBits::BadOverlap;
  902. }
  903. };
  904. ForEachImageInRegion(cpu_addr, size_bytes, region_check);
  905. const auto region_check_gpu = [&](ImageId overlap_id, ImageBase& overlap) {
  906. if (!overlaps_found.contains(overlap_id)) {
  907. if (True(overlap.flags & ImageFlagBits::Remapped)) {
  908. ignore_textures.insert(overlap_id);
  909. }
  910. if (overlap.gpu_addr == gpu_addr && overlap.guest_size_bytes == size_bytes) {
  911. ignore_textures.insert(overlap_id);
  912. }
  913. }
  914. };
  915. ForEachSparseImageInRegion(gpu_addr, size_bytes, region_check_gpu);
  916. bool can_rescale = info.rescaleable;
  917. bool any_rescaled = false;
  918. bool any_blacklisted = false;
  919. for (const ImageId sibling_id : all_siblings) {
  920. if (!can_rescale) {
  921. break;
  922. }
  923. Image& sibling = slot_images[sibling_id];
  924. can_rescale &= ImageCanRescale(sibling);
  925. any_rescaled |= True(sibling.flags & ImageFlagBits::Rescaled);
  926. any_blacklisted |= True(sibling.flags & ImageFlagBits::Blacklisted);
  927. }
  928. can_rescale &= any_rescaled;
  929. if (can_rescale) {
  930. for (const ImageId sibling_id : all_siblings) {
  931. Image& sibling = slot_images[sibling_id];
  932. ScaleUp(sibling);
  933. }
  934. } else {
  935. for (const ImageId sibling_id : all_siblings) {
  936. Image& sibling = slot_images[sibling_id];
  937. ScaleDown(sibling);
  938. if (any_blacklisted) {
  939. sibling.flags |= ImageFlagBits::Blacklisted;
  940. }
  941. }
  942. }
  943. const ImageId new_image_id = slot_images.insert(runtime, new_info, gpu_addr, cpu_addr);
  944. Image& new_image = slot_images[new_image_id];
  945. if (!gpu_memory.IsContinousRange(new_image.gpu_addr, new_image.guest_size_bytes)) {
  946. new_image.flags |= ImageFlagBits::Sparse;
  947. }
  948. for (const ImageId overlap_id : ignore_textures) {
  949. Image& overlap = slot_images[overlap_id];
  950. if (True(overlap.flags & ImageFlagBits::GpuModified)) {
  951. UNIMPLEMENTED();
  952. }
  953. if (True(overlap.flags & ImageFlagBits::Tracked)) {
  954. UntrackImage(overlap, overlap_id);
  955. }
  956. UnregisterImage(overlap_id);
  957. DeleteImage(overlap_id);
  958. }
  959. // TODO: Only upload what we need
  960. RefreshContents(new_image, new_image_id);
  961. if (can_rescale) {
  962. ScaleUp(new_image);
  963. } else {
  964. ScaleDown(new_image);
  965. }
  966. for (const ImageId overlap_id : overlap_ids) {
  967. Image& overlap = slot_images[overlap_id];
  968. if (overlap.info.num_samples != new_image.info.num_samples) {
  969. LOG_WARNING(HW_GPU, "Copying between images with different samples is not implemented");
  970. } else {
  971. const SubresourceBase base = new_image.TryFindBase(overlap.gpu_addr).value();
  972. auto copies = MakeShrinkImageCopies(new_info, overlap.info, base);
  973. runtime.CopyImage(new_image, overlap, std::move(copies));
  974. }
  975. if (True(overlap.flags & ImageFlagBits::Tracked)) {
  976. UntrackImage(overlap, overlap_id);
  977. }
  978. UnregisterImage(overlap_id);
  979. DeleteImage(overlap_id);
  980. }
  981. ImageBase& new_image_base = new_image;
  982. for (const ImageId aliased_id : right_aliased_ids) {
  983. ImageBase& aliased = slot_images[aliased_id];
  984. AddImageAlias(new_image_base, aliased, new_image_id, aliased_id);
  985. new_image.flags |= ImageFlagBits::Alias;
  986. }
  987. for (const ImageId aliased_id : left_aliased_ids) {
  988. ImageBase& aliased = slot_images[aliased_id];
  989. AddImageAlias(aliased, new_image_base, aliased_id, new_image_id);
  990. new_image.flags |= ImageFlagBits::Alias;
  991. }
  992. for (const ImageId aliased_id : bad_overlap_ids) {
  993. ImageBase& aliased = slot_images[aliased_id];
  994. aliased.overlapping_images.push_back(new_image_id);
  995. new_image.overlapping_images.push_back(aliased_id);
  996. new_image.flags |= ImageFlagBits::BadOverlap;
  997. }
  998. RegisterImage(new_image_id);
  999. return new_image_id;
  1000. }
  1001. template <class P>
  1002. typename TextureCache<P>::BlitImages TextureCache<P>::GetBlitImages(
  1003. const Tegra::Engines::Fermi2D::Surface& dst, const Tegra::Engines::Fermi2D::Surface& src) {
  1004. static constexpr auto FIND_OPTIONS = RelaxedOptions::Format | RelaxedOptions::Samples;
  1005. const GPUVAddr dst_addr = dst.Address();
  1006. const GPUVAddr src_addr = src.Address();
  1007. ImageInfo dst_info(dst);
  1008. ImageInfo src_info(src);
  1009. ImageId dst_id;
  1010. ImageId src_id;
  1011. do {
  1012. has_deleted_images = false;
  1013. dst_id = FindImage(dst_info, dst_addr, FIND_OPTIONS);
  1014. src_id = FindImage(src_info, src_addr, FIND_OPTIONS);
  1015. const ImageBase* const dst_image = dst_id ? &slot_images[dst_id] : nullptr;
  1016. const ImageBase* const src_image = src_id ? &slot_images[src_id] : nullptr;
  1017. DeduceBlitImages(dst_info, src_info, dst_image, src_image);
  1018. if (GetFormatType(dst_info.format) != GetFormatType(src_info.format)) {
  1019. continue;
  1020. }
  1021. if (!dst_id) {
  1022. dst_id = InsertImage(dst_info, dst_addr, RelaxedOptions{});
  1023. }
  1024. if (!src_id) {
  1025. src_id = InsertImage(src_info, src_addr, RelaxedOptions{});
  1026. }
  1027. } while (has_deleted_images);
  1028. return BlitImages{
  1029. .dst_id = dst_id,
  1030. .src_id = src_id,
  1031. .dst_format = dst_info.format,
  1032. .src_format = src_info.format,
  1033. };
  1034. }
  1035. template <class P>
  1036. SamplerId TextureCache<P>::FindSampler(const TSCEntry& config) {
  1037. if (std::ranges::all_of(config.raw, [](u64 value) { return value == 0; })) {
  1038. return NULL_SAMPLER_ID;
  1039. }
  1040. const auto [pair, is_new] = samplers.try_emplace(config);
  1041. if (is_new) {
  1042. pair->second = slot_samplers.insert(runtime, config);
  1043. }
  1044. return pair->second;
  1045. }
  1046. template <class P>
  1047. ImageViewId TextureCache<P>::FindColorBuffer(size_t index, bool is_clear) {
  1048. const auto& regs = maxwell3d.regs;
  1049. if (index >= regs.rt_control.count) {
  1050. return ImageViewId{};
  1051. }
  1052. const auto& rt = regs.rt[index];
  1053. const GPUVAddr gpu_addr = rt.Address();
  1054. if (gpu_addr == 0) {
  1055. return ImageViewId{};
  1056. }
  1057. if (rt.format == Tegra::RenderTargetFormat::NONE) {
  1058. return ImageViewId{};
  1059. }
  1060. const ImageInfo info(regs, index);
  1061. return FindRenderTargetView(info, gpu_addr, is_clear);
  1062. }
  1063. template <class P>
  1064. ImageViewId TextureCache<P>::FindDepthBuffer(bool is_clear) {
  1065. const auto& regs = maxwell3d.regs;
  1066. if (!regs.zeta_enable) {
  1067. return ImageViewId{};
  1068. }
  1069. const GPUVAddr gpu_addr = regs.zeta.Address();
  1070. if (gpu_addr == 0) {
  1071. return ImageViewId{};
  1072. }
  1073. const ImageInfo info(regs);
  1074. return FindRenderTargetView(info, gpu_addr, is_clear);
  1075. }
  1076. template <class P>
  1077. ImageViewId TextureCache<P>::FindRenderTargetView(const ImageInfo& info, GPUVAddr gpu_addr,
  1078. bool is_clear) {
  1079. const auto options = is_clear ? RelaxedOptions::Samples : RelaxedOptions{};
  1080. const ImageId image_id = FindOrInsertImage(info, gpu_addr, options);
  1081. if (!image_id) {
  1082. return NULL_IMAGE_VIEW_ID;
  1083. }
  1084. Image& image = slot_images[image_id];
  1085. const ImageViewType view_type = RenderTargetImageViewType(info);
  1086. SubresourceBase base;
  1087. if (image.info.type == ImageType::Linear) {
  1088. base = SubresourceBase{.level = 0, .layer = 0};
  1089. } else {
  1090. base = image.TryFindBase(gpu_addr).value();
  1091. }
  1092. const s32 layers = image.info.type == ImageType::e3D ? info.size.depth : info.resources.layers;
  1093. const SubresourceRange range{
  1094. .base = base,
  1095. .extent = {.levels = 1, .layers = layers},
  1096. };
  1097. return FindOrEmplaceImageView(image_id, ImageViewInfo(view_type, info.format, range));
  1098. }
  1099. template <class P>
  1100. template <typename Func>
  1101. void TextureCache<P>::ForEachImageInRegion(VAddr cpu_addr, size_t size, Func&& func) {
  1102. using FuncReturn = typename std::invoke_result<Func, ImageId, Image&>::type;
  1103. static constexpr bool BOOL_BREAK = std::is_same_v<FuncReturn, bool>;
  1104. boost::container::small_vector<ImageId, 32> images;
  1105. boost::container::small_vector<ImageMapId, 32> maps;
  1106. ForEachCPUPage(cpu_addr, size, [this, &images, &maps, cpu_addr, size, func](u64 page) {
  1107. const auto it = page_table.find(page);
  1108. if (it == page_table.end()) {
  1109. if constexpr (BOOL_BREAK) {
  1110. return false;
  1111. } else {
  1112. return;
  1113. }
  1114. }
  1115. for (const ImageMapId map_id : it->second) {
  1116. ImageMapView& map = slot_map_views[map_id];
  1117. if (map.picked) {
  1118. continue;
  1119. }
  1120. if (!map.Overlaps(cpu_addr, size)) {
  1121. continue;
  1122. }
  1123. map.picked = true;
  1124. maps.push_back(map_id);
  1125. Image& image = slot_images[map.image_id];
  1126. if (True(image.flags & ImageFlagBits::Picked)) {
  1127. continue;
  1128. }
  1129. image.flags |= ImageFlagBits::Picked;
  1130. images.push_back(map.image_id);
  1131. if constexpr (BOOL_BREAK) {
  1132. if (func(map.image_id, image)) {
  1133. return true;
  1134. }
  1135. } else {
  1136. func(map.image_id, image);
  1137. }
  1138. }
  1139. if constexpr (BOOL_BREAK) {
  1140. return false;
  1141. }
  1142. });
  1143. for (const ImageId image_id : images) {
  1144. slot_images[image_id].flags &= ~ImageFlagBits::Picked;
  1145. }
  1146. for (const ImageMapId map_id : maps) {
  1147. slot_map_views[map_id].picked = false;
  1148. }
  1149. }
  1150. template <class P>
  1151. template <typename Func>
  1152. void TextureCache<P>::ForEachImageInRegionGPU(GPUVAddr gpu_addr, size_t size, Func&& func) {
  1153. using FuncReturn = typename std::invoke_result<Func, ImageId, Image&>::type;
  1154. static constexpr bool BOOL_BREAK = std::is_same_v<FuncReturn, bool>;
  1155. boost::container::small_vector<ImageId, 8> images;
  1156. ForEachGPUPage(gpu_addr, size, [this, &images, gpu_addr, size, func](u64 page) {
  1157. const auto it = gpu_page_table.find(page);
  1158. if (it == gpu_page_table.end()) {
  1159. if constexpr (BOOL_BREAK) {
  1160. return false;
  1161. } else {
  1162. return;
  1163. }
  1164. }
  1165. for (const ImageId image_id : it->second) {
  1166. Image& image = slot_images[image_id];
  1167. if (True(image.flags & ImageFlagBits::Picked)) {
  1168. continue;
  1169. }
  1170. if (!image.OverlapsGPU(gpu_addr, size)) {
  1171. continue;
  1172. }
  1173. image.flags |= ImageFlagBits::Picked;
  1174. images.push_back(image_id);
  1175. if constexpr (BOOL_BREAK) {
  1176. if (func(image_id, image)) {
  1177. return true;
  1178. }
  1179. } else {
  1180. func(image_id, image);
  1181. }
  1182. }
  1183. if constexpr (BOOL_BREAK) {
  1184. return false;
  1185. }
  1186. });
  1187. for (const ImageId image_id : images) {
  1188. slot_images[image_id].flags &= ~ImageFlagBits::Picked;
  1189. }
  1190. }
  1191. template <class P>
  1192. template <typename Func>
  1193. void TextureCache<P>::ForEachSparseImageInRegion(GPUVAddr gpu_addr, size_t size, Func&& func) {
  1194. using FuncReturn = typename std::invoke_result<Func, ImageId, Image&>::type;
  1195. static constexpr bool BOOL_BREAK = std::is_same_v<FuncReturn, bool>;
  1196. boost::container::small_vector<ImageId, 8> images;
  1197. ForEachGPUPage(gpu_addr, size, [this, &images, gpu_addr, size, func](u64 page) {
  1198. const auto it = sparse_page_table.find(page);
  1199. if (it == sparse_page_table.end()) {
  1200. if constexpr (BOOL_BREAK) {
  1201. return false;
  1202. } else {
  1203. return;
  1204. }
  1205. }
  1206. for (const ImageId image_id : it->second) {
  1207. Image& image = slot_images[image_id];
  1208. if (True(image.flags & ImageFlagBits::Picked)) {
  1209. continue;
  1210. }
  1211. if (!image.OverlapsGPU(gpu_addr, size)) {
  1212. continue;
  1213. }
  1214. image.flags |= ImageFlagBits::Picked;
  1215. images.push_back(image_id);
  1216. if constexpr (BOOL_BREAK) {
  1217. if (func(image_id, image)) {
  1218. return true;
  1219. }
  1220. } else {
  1221. func(image_id, image);
  1222. }
  1223. }
  1224. if constexpr (BOOL_BREAK) {
  1225. return false;
  1226. }
  1227. });
  1228. for (const ImageId image_id : images) {
  1229. slot_images[image_id].flags &= ~ImageFlagBits::Picked;
  1230. }
  1231. }
  1232. template <class P>
  1233. template <typename Func>
  1234. void TextureCache<P>::ForEachSparseSegment(ImageBase& image, Func&& func) {
  1235. using FuncReturn = typename std::invoke_result<Func, GPUVAddr, VAddr, size_t>::type;
  1236. static constexpr bool RETURNS_BOOL = std::is_same_v<FuncReturn, bool>;
  1237. const auto segments = gpu_memory.GetSubmappedRange(image.gpu_addr, image.guest_size_bytes);
  1238. for (auto& segment : segments) {
  1239. const auto gpu_addr = segment.first;
  1240. const auto size = segment.second;
  1241. std::optional<VAddr> cpu_addr = gpu_memory.GpuToCpuAddress(gpu_addr);
  1242. ASSERT(cpu_addr);
  1243. if constexpr (RETURNS_BOOL) {
  1244. if (func(gpu_addr, *cpu_addr, size)) {
  1245. break;
  1246. }
  1247. } else {
  1248. func(gpu_addr, *cpu_addr, size);
  1249. }
  1250. }
  1251. }
  1252. template <class P>
  1253. ImageViewId TextureCache<P>::FindOrEmplaceImageView(ImageId image_id, const ImageViewInfo& info) {
  1254. Image& image = slot_images[image_id];
  1255. if (const ImageViewId image_view_id = image.FindView(info); image_view_id) {
  1256. return image_view_id;
  1257. }
  1258. const ImageViewId image_view_id = slot_image_views.insert(runtime, info, image_id, image);
  1259. image.InsertView(info, image_view_id);
  1260. return image_view_id;
  1261. }
  1262. template <class P>
  1263. void TextureCache<P>::RegisterImage(ImageId image_id) {
  1264. ImageBase& image = slot_images[image_id];
  1265. ASSERT_MSG(False(image.flags & ImageFlagBits::Registered),
  1266. "Trying to register an already registered image");
  1267. image.flags |= ImageFlagBits::Registered;
  1268. u64 tentative_size = std::max(image.guest_size_bytes, image.unswizzled_size_bytes);
  1269. if ((IsPixelFormatASTC(image.info.format) &&
  1270. True(image.flags & ImageFlagBits::AcceleratedUpload)) ||
  1271. True(image.flags & ImageFlagBits::Converted)) {
  1272. tentative_size = EstimatedDecompressedSize(tentative_size, image.info.format);
  1273. }
  1274. total_used_memory += Common::AlignUp(tentative_size, 1024);
  1275. image.lru_index = lru_cache.Insert(image_id, frame_tick);
  1276. ForEachGPUPage(image.gpu_addr, image.guest_size_bytes,
  1277. [this, image_id](u64 page) { gpu_page_table[page].push_back(image_id); });
  1278. if (False(image.flags & ImageFlagBits::Sparse)) {
  1279. auto map_id =
  1280. slot_map_views.insert(image.gpu_addr, image.cpu_addr, image.guest_size_bytes, image_id);
  1281. ForEachCPUPage(image.cpu_addr, image.guest_size_bytes,
  1282. [this, map_id](u64 page) { page_table[page].push_back(map_id); });
  1283. image.map_view_id = map_id;
  1284. return;
  1285. }
  1286. std::vector<ImageViewId> sparse_maps{};
  1287. ForEachSparseSegment(
  1288. image, [this, image_id, &sparse_maps](GPUVAddr gpu_addr, VAddr cpu_addr, size_t size) {
  1289. auto map_id = slot_map_views.insert(gpu_addr, cpu_addr, size, image_id);
  1290. ForEachCPUPage(cpu_addr, size,
  1291. [this, map_id](u64 page) { page_table[page].push_back(map_id); });
  1292. sparse_maps.push_back(map_id);
  1293. });
  1294. sparse_views.emplace(image_id, std::move(sparse_maps));
  1295. ForEachGPUPage(image.gpu_addr, image.guest_size_bytes,
  1296. [this, image_id](u64 page) { sparse_page_table[page].push_back(image_id); });
  1297. }
  1298. template <class P>
  1299. void TextureCache<P>::UnregisterImage(ImageId image_id) {
  1300. Image& image = slot_images[image_id];
  1301. ASSERT_MSG(True(image.flags & ImageFlagBits::Registered),
  1302. "Trying to unregister an already registered image");
  1303. image.flags &= ~ImageFlagBits::Registered;
  1304. image.flags &= ~ImageFlagBits::BadOverlap;
  1305. u64 tentative_size = std::max(image.guest_size_bytes, image.unswizzled_size_bytes);
  1306. if ((IsPixelFormatASTC(image.info.format) &&
  1307. True(image.flags & ImageFlagBits::AcceleratedUpload)) ||
  1308. True(image.flags & ImageFlagBits::Converted)) {
  1309. tentative_size = EstimatedDecompressedSize(tentative_size, image.info.format);
  1310. }
  1311. total_used_memory -= Common::AlignUp(tentative_size, 1024);
  1312. lru_cache.Free(image.lru_index);
  1313. const auto& clear_page_table =
  1314. [this, image_id](
  1315. u64 page,
  1316. std::unordered_map<u64, std::vector<ImageId>, IdentityHash<u64>>& selected_page_table) {
  1317. const auto page_it = selected_page_table.find(page);
  1318. if (page_it == selected_page_table.end()) {
  1319. UNREACHABLE_MSG("Unregistering unregistered page=0x{:x}", page << PAGE_BITS);
  1320. return;
  1321. }
  1322. std::vector<ImageId>& image_ids = page_it->second;
  1323. const auto vector_it = std::ranges::find(image_ids, image_id);
  1324. if (vector_it == image_ids.end()) {
  1325. UNREACHABLE_MSG("Unregistering unregistered image in page=0x{:x}",
  1326. page << PAGE_BITS);
  1327. return;
  1328. }
  1329. image_ids.erase(vector_it);
  1330. };
  1331. ForEachGPUPage(image.gpu_addr, image.guest_size_bytes,
  1332. [this, &clear_page_table](u64 page) { clear_page_table(page, gpu_page_table); });
  1333. if (False(image.flags & ImageFlagBits::Sparse)) {
  1334. const auto map_id = image.map_view_id;
  1335. ForEachCPUPage(image.cpu_addr, image.guest_size_bytes, [this, map_id](u64 page) {
  1336. const auto page_it = page_table.find(page);
  1337. if (page_it == page_table.end()) {
  1338. UNREACHABLE_MSG("Unregistering unregistered page=0x{:x}", page << PAGE_BITS);
  1339. return;
  1340. }
  1341. std::vector<ImageMapId>& image_map_ids = page_it->second;
  1342. const auto vector_it = std::ranges::find(image_map_ids, map_id);
  1343. if (vector_it == image_map_ids.end()) {
  1344. UNREACHABLE_MSG("Unregistering unregistered image in page=0x{:x}",
  1345. page << PAGE_BITS);
  1346. return;
  1347. }
  1348. image_map_ids.erase(vector_it);
  1349. });
  1350. slot_map_views.erase(map_id);
  1351. return;
  1352. }
  1353. ForEachGPUPage(image.gpu_addr, image.guest_size_bytes, [this, &clear_page_table](u64 page) {
  1354. clear_page_table(page, sparse_page_table);
  1355. });
  1356. auto it = sparse_views.find(image_id);
  1357. ASSERT(it != sparse_views.end());
  1358. auto& sparse_maps = it->second;
  1359. for (auto& map_view_id : sparse_maps) {
  1360. const auto& map_range = slot_map_views[map_view_id];
  1361. const VAddr cpu_addr = map_range.cpu_addr;
  1362. const std::size_t size = map_range.size;
  1363. ForEachCPUPage(cpu_addr, size, [this, image_id](u64 page) {
  1364. const auto page_it = page_table.find(page);
  1365. if (page_it == page_table.end()) {
  1366. UNREACHABLE_MSG("Unregistering unregistered page=0x{:x}", page << PAGE_BITS);
  1367. return;
  1368. }
  1369. std::vector<ImageMapId>& image_map_ids = page_it->second;
  1370. auto vector_it = image_map_ids.begin();
  1371. while (vector_it != image_map_ids.end()) {
  1372. ImageMapView& map = slot_map_views[*vector_it];
  1373. if (map.image_id != image_id) {
  1374. vector_it++;
  1375. continue;
  1376. }
  1377. if (!map.picked) {
  1378. map.picked = true;
  1379. }
  1380. vector_it = image_map_ids.erase(vector_it);
  1381. }
  1382. });
  1383. slot_map_views.erase(map_view_id);
  1384. }
  1385. sparse_views.erase(it);
  1386. }
  1387. template <class P>
  1388. void TextureCache<P>::TrackImage(ImageBase& image, ImageId image_id) {
  1389. ASSERT(False(image.flags & ImageFlagBits::Tracked));
  1390. image.flags |= ImageFlagBits::Tracked;
  1391. if (False(image.flags & ImageFlagBits::Sparse)) {
  1392. rasterizer.UpdatePagesCachedCount(image.cpu_addr, image.guest_size_bytes, 1);
  1393. return;
  1394. }
  1395. if (True(image.flags & ImageFlagBits::Registered)) {
  1396. auto it = sparse_views.find(image_id);
  1397. ASSERT(it != sparse_views.end());
  1398. auto& sparse_maps = it->second;
  1399. for (auto& map_view_id : sparse_maps) {
  1400. const auto& map = slot_map_views[map_view_id];
  1401. const VAddr cpu_addr = map.cpu_addr;
  1402. const std::size_t size = map.size;
  1403. rasterizer.UpdatePagesCachedCount(cpu_addr, size, 1);
  1404. }
  1405. return;
  1406. }
  1407. ForEachSparseSegment(image,
  1408. [this]([[maybe_unused]] GPUVAddr gpu_addr, VAddr cpu_addr, size_t size) {
  1409. rasterizer.UpdatePagesCachedCount(cpu_addr, size, 1);
  1410. });
  1411. }
  1412. template <class P>
  1413. void TextureCache<P>::UntrackImage(ImageBase& image, ImageId image_id) {
  1414. ASSERT(True(image.flags & ImageFlagBits::Tracked));
  1415. image.flags &= ~ImageFlagBits::Tracked;
  1416. if (False(image.flags & ImageFlagBits::Sparse)) {
  1417. rasterizer.UpdatePagesCachedCount(image.cpu_addr, image.guest_size_bytes, -1);
  1418. return;
  1419. }
  1420. ASSERT(True(image.flags & ImageFlagBits::Registered));
  1421. auto it = sparse_views.find(image_id);
  1422. ASSERT(it != sparse_views.end());
  1423. auto& sparse_maps = it->second;
  1424. for (auto& map_view_id : sparse_maps) {
  1425. const auto& map = slot_map_views[map_view_id];
  1426. const VAddr cpu_addr = map.cpu_addr;
  1427. const std::size_t size = map.size;
  1428. rasterizer.UpdatePagesCachedCount(cpu_addr, size, -1);
  1429. }
  1430. }
  1431. template <class P>
  1432. void TextureCache<P>::DeleteImage(ImageId image_id) {
  1433. ImageBase& image = slot_images[image_id];
  1434. const GPUVAddr gpu_addr = image.gpu_addr;
  1435. const auto alloc_it = image_allocs_table.find(gpu_addr);
  1436. if (alloc_it == image_allocs_table.end()) {
  1437. UNREACHABLE_MSG("Trying to delete an image alloc that does not exist in address 0x{:x}",
  1438. gpu_addr);
  1439. return;
  1440. }
  1441. const ImageAllocId alloc_id = alloc_it->second;
  1442. std::vector<ImageId>& alloc_images = slot_image_allocs[alloc_id].images;
  1443. const auto alloc_image_it = std::ranges::find(alloc_images, image_id);
  1444. if (alloc_image_it == alloc_images.end()) {
  1445. UNREACHABLE_MSG("Trying to delete an image that does not exist");
  1446. return;
  1447. }
  1448. ASSERT_MSG(False(image.flags & ImageFlagBits::Tracked), "Image was not untracked");
  1449. ASSERT_MSG(False(image.flags & ImageFlagBits::Registered), "Image was not unregistered");
  1450. // Mark render targets as dirty
  1451. auto& dirty = maxwell3d.dirty.flags;
  1452. dirty[Dirty::RenderTargets] = true;
  1453. dirty[Dirty::ZetaBuffer] = true;
  1454. for (size_t rt = 0; rt < NUM_RT; ++rt) {
  1455. dirty[Dirty::ColorBuffer0 + rt] = true;
  1456. }
  1457. const std::span<const ImageViewId> image_view_ids = image.image_view_ids;
  1458. for (const ImageViewId image_view_id : image_view_ids) {
  1459. std::ranges::replace(render_targets.color_buffer_ids, image_view_id, ImageViewId{});
  1460. if (render_targets.depth_buffer_id == image_view_id) {
  1461. render_targets.depth_buffer_id = ImageViewId{};
  1462. }
  1463. }
  1464. RemoveImageViewReferences(image_view_ids);
  1465. RemoveFramebuffers(image_view_ids);
  1466. for (const AliasedImage& alias : image.aliased_images) {
  1467. ImageBase& other_image = slot_images[alias.id];
  1468. [[maybe_unused]] const size_t num_removed_aliases =
  1469. std::erase_if(other_image.aliased_images, [image_id](const AliasedImage& other_alias) {
  1470. return other_alias.id == image_id;
  1471. });
  1472. other_image.CheckAliasState();
  1473. ASSERT_MSG(num_removed_aliases == 1, "Invalid number of removed aliases: {}",
  1474. num_removed_aliases);
  1475. }
  1476. for (const ImageId overlap_id : image.overlapping_images) {
  1477. ImageBase& other_image = slot_images[overlap_id];
  1478. [[maybe_unused]] const size_t num_removed_overlaps = std::erase_if(
  1479. other_image.overlapping_images,
  1480. [image_id](const ImageId other_overlap_id) { return other_overlap_id == image_id; });
  1481. other_image.CheckBadOverlapState();
  1482. ASSERT_MSG(num_removed_overlaps == 1, "Invalid number of removed overlapps: {}",
  1483. num_removed_overlaps);
  1484. }
  1485. for (const ImageViewId image_view_id : image_view_ids) {
  1486. sentenced_image_view.Push(std::move(slot_image_views[image_view_id]));
  1487. slot_image_views.erase(image_view_id);
  1488. }
  1489. sentenced_images.Push(std::move(slot_images[image_id]));
  1490. slot_images.erase(image_id);
  1491. alloc_images.erase(alloc_image_it);
  1492. if (alloc_images.empty()) {
  1493. image_allocs_table.erase(alloc_it);
  1494. }
  1495. if constexpr (ENABLE_VALIDATION) {
  1496. std::ranges::fill(graphics_image_view_ids, CORRUPT_ID);
  1497. std::ranges::fill(compute_image_view_ids, CORRUPT_ID);
  1498. }
  1499. graphics_image_table.Invalidate();
  1500. compute_image_table.Invalidate();
  1501. has_deleted_images = true;
  1502. }
  1503. template <class P>
  1504. void TextureCache<P>::RemoveImageViewReferences(std::span<const ImageViewId> removed_views) {
  1505. auto it = image_views.begin();
  1506. while (it != image_views.end()) {
  1507. const auto found = std::ranges::find(removed_views, it->second);
  1508. if (found != removed_views.end()) {
  1509. it = image_views.erase(it);
  1510. } else {
  1511. ++it;
  1512. }
  1513. }
  1514. }
  1515. template <class P>
  1516. void TextureCache<P>::RemoveFramebuffers(std::span<const ImageViewId> removed_views) {
  1517. auto it = framebuffers.begin();
  1518. while (it != framebuffers.end()) {
  1519. if (it->first.Contains(removed_views)) {
  1520. it = framebuffers.erase(it);
  1521. } else {
  1522. ++it;
  1523. }
  1524. }
  1525. }
  1526. template <class P>
  1527. void TextureCache<P>::MarkModification(ImageBase& image) noexcept {
  1528. image.flags |= ImageFlagBits::GpuModified;
  1529. image.modification_tick = ++modification_tick;
  1530. }
  1531. template <class P>
  1532. void TextureCache<P>::SynchronizeAliases(ImageId image_id) {
  1533. boost::container::small_vector<const AliasedImage*, 1> aliased_images;
  1534. Image& image = slot_images[image_id];
  1535. bool any_rescaled = True(image.flags & ImageFlagBits::Rescaled);
  1536. bool any_blacklisted = True(image.flags & ImageFlagBits::Blacklisted);
  1537. u64 most_recent_tick = image.modification_tick;
  1538. for (const AliasedImage& aliased : image.aliased_images) {
  1539. ImageBase& aliased_image = slot_images[aliased.id];
  1540. if (image.modification_tick < aliased_image.modification_tick) {
  1541. most_recent_tick = std::max(most_recent_tick, aliased_image.modification_tick);
  1542. aliased_images.push_back(&aliased);
  1543. any_rescaled |= True(aliased_image.flags & ImageFlagBits::Rescaled);
  1544. any_blacklisted |= True(aliased_image.flags & ImageFlagBits::Blacklisted);
  1545. }
  1546. }
  1547. if (aliased_images.empty()) {
  1548. return;
  1549. }
  1550. const bool can_rescale = ImageCanRescale(image);
  1551. if (any_rescaled) {
  1552. if (can_rescale) {
  1553. ScaleUp(image);
  1554. } else {
  1555. ScaleDown(image);
  1556. if (any_blacklisted) {
  1557. image.flags |= ImageFlagBits::Blacklisted;
  1558. }
  1559. }
  1560. }
  1561. image.modification_tick = most_recent_tick;
  1562. std::ranges::sort(aliased_images, [this](const AliasedImage* lhs, const AliasedImage* rhs) {
  1563. const ImageBase& lhs_image = slot_images[lhs->id];
  1564. const ImageBase& rhs_image = slot_images[rhs->id];
  1565. return lhs_image.modification_tick < rhs_image.modification_tick;
  1566. });
  1567. for (const AliasedImage* const aliased : aliased_images) {
  1568. if (any_rescaled) {
  1569. Image& aliased_image = slot_images[aliased->id];
  1570. if (can_rescale) {
  1571. ScaleUp(aliased_image);
  1572. } else {
  1573. ScaleDown(aliased_image);
  1574. if (any_blacklisted) {
  1575. aliased_image.flags |= ImageFlagBits::Blacklisted;
  1576. }
  1577. }
  1578. }
  1579. CopyImage(image_id, aliased->id, aliased->copies);
  1580. }
  1581. }
  1582. template <class P>
  1583. void TextureCache<P>::PrepareImage(ImageId image_id, bool is_modification, bool invalidate) {
  1584. Image& image = slot_images[image_id];
  1585. if (invalidate) {
  1586. image.flags &= ~(ImageFlagBits::CpuModified | ImageFlagBits::GpuModified);
  1587. if (False(image.flags & ImageFlagBits::Tracked)) {
  1588. TrackImage(image, image_id);
  1589. }
  1590. } else {
  1591. RefreshContents(image, image_id);
  1592. SynchronizeAliases(image_id);
  1593. }
  1594. if (is_modification) {
  1595. MarkModification(image);
  1596. }
  1597. lru_cache.Touch(image.lru_index, frame_tick);
  1598. }
  1599. template <class P>
  1600. void TextureCache<P>::PrepareImageView(ImageViewId image_view_id, bool is_modification,
  1601. bool invalidate) {
  1602. if (!image_view_id) {
  1603. return;
  1604. }
  1605. const ImageViewBase& image_view = slot_image_views[image_view_id];
  1606. if (image_view.IsBuffer()) {
  1607. return;
  1608. }
  1609. PrepareImage(image_view.image_id, is_modification, invalidate);
  1610. }
  1611. template <class P>
  1612. void TextureCache<P>::CopyImage(ImageId dst_id, ImageId src_id, std::vector<ImageCopy> copies) {
  1613. Image& dst = slot_images[dst_id];
  1614. Image& src = slot_images[src_id];
  1615. const bool is_rescaled = True(src.flags & ImageFlagBits::Rescaled);
  1616. if (is_rescaled) {
  1617. ASSERT(True(dst.flags & ImageFlagBits::Rescaled));
  1618. const bool both_2d{src.info.type == ImageType::e2D && dst.info.type == ImageType::e2D};
  1619. const auto& resolution = Settings::values.resolution_info;
  1620. const auto scale_up = [&](u32 value) -> u32 {
  1621. if (value == 0) {
  1622. return 0U;
  1623. }
  1624. return std::max<u32>((value * resolution.up_scale) >> resolution.down_shift, 1U);
  1625. };
  1626. for (auto& copy : copies) {
  1627. copy.src_offset.x = scale_up(copy.src_offset.x);
  1628. copy.dst_offset.x = scale_up(copy.dst_offset.x);
  1629. copy.extent.width = scale_up(copy.extent.width);
  1630. if (both_2d) {
  1631. copy.src_offset.y = scale_up(copy.src_offset.y);
  1632. copy.dst_offset.y = scale_up(copy.dst_offset.y);
  1633. copy.extent.height = scale_up(copy.extent.height);
  1634. }
  1635. }
  1636. }
  1637. const auto dst_format_type = GetFormatType(dst.info.format);
  1638. const auto src_format_type = GetFormatType(src.info.format);
  1639. if (src_format_type == dst_format_type) {
  1640. if constexpr (HAS_EMULATED_COPIES) {
  1641. if (!runtime.CanImageBeCopied(dst, src)) {
  1642. return runtime.EmulateCopyImage(dst, src, copies);
  1643. }
  1644. }
  1645. return runtime.CopyImage(dst, src, copies);
  1646. }
  1647. UNIMPLEMENTED_IF(dst.info.type != ImageType::e2D);
  1648. UNIMPLEMENTED_IF(src.info.type != ImageType::e2D);
  1649. for (const ImageCopy& copy : copies) {
  1650. UNIMPLEMENTED_IF(copy.dst_subresource.num_layers != 1);
  1651. UNIMPLEMENTED_IF(copy.src_subresource.num_layers != 1);
  1652. UNIMPLEMENTED_IF(copy.src_offset != Offset3D{});
  1653. UNIMPLEMENTED_IF(copy.dst_offset != Offset3D{});
  1654. const SubresourceBase dst_base{
  1655. .level = copy.dst_subresource.base_level,
  1656. .layer = copy.dst_subresource.base_layer,
  1657. };
  1658. const SubresourceBase src_base{
  1659. .level = copy.src_subresource.base_level,
  1660. .layer = copy.src_subresource.base_layer,
  1661. };
  1662. const SubresourceExtent dst_extent{.levels = 1, .layers = 1};
  1663. const SubresourceExtent src_extent{.levels = 1, .layers = 1};
  1664. const SubresourceRange dst_range{.base = dst_base, .extent = dst_extent};
  1665. const SubresourceRange src_range{.base = src_base, .extent = src_extent};
  1666. const ImageViewInfo dst_view_info(ImageViewType::e2D, dst.info.format, dst_range);
  1667. const ImageViewInfo src_view_info(ImageViewType::e2D, src.info.format, src_range);
  1668. const auto [dst_framebuffer_id, dst_view_id] = RenderTargetFromImage(dst_id, dst_view_info);
  1669. Framebuffer* const dst_framebuffer = &slot_framebuffers[dst_framebuffer_id];
  1670. const ImageViewId src_view_id = FindOrEmplaceImageView(src_id, src_view_info);
  1671. ImageView& dst_view = slot_image_views[dst_view_id];
  1672. ImageView& src_view = slot_image_views[src_view_id];
  1673. [[maybe_unused]] const Extent3D expected_size{
  1674. .width = std::min(dst_view.size.width, src_view.size.width),
  1675. .height = std::min(dst_view.size.height, src_view.size.height),
  1676. .depth = std::min(dst_view.size.depth, src_view.size.depth),
  1677. };
  1678. UNIMPLEMENTED_IF(copy.extent != expected_size);
  1679. runtime.ConvertImage(dst_framebuffer, dst_view, src_view);
  1680. }
  1681. }
  1682. template <class P>
  1683. void TextureCache<P>::BindRenderTarget(ImageViewId* old_id, ImageViewId new_id) {
  1684. if (*old_id == new_id) {
  1685. return;
  1686. }
  1687. if (*old_id) {
  1688. const ImageViewBase& old_view = slot_image_views[*old_id];
  1689. if (True(old_view.flags & ImageViewFlagBits::PreemtiveDownload)) {
  1690. uncommitted_downloads.push_back(old_view.image_id);
  1691. }
  1692. }
  1693. *old_id = new_id;
  1694. }
  1695. template <class P>
  1696. std::pair<FramebufferId, ImageViewId> TextureCache<P>::RenderTargetFromImage(
  1697. ImageId image_id, const ImageViewInfo& view_info) {
  1698. const ImageViewId view_id = FindOrEmplaceImageView(image_id, view_info);
  1699. const ImageBase& image = slot_images[image_id];
  1700. const bool is_rescaled = True(image.flags & ImageFlagBits::Rescaled);
  1701. const bool is_color = GetFormatType(image.info.format) == SurfaceType::ColorTexture;
  1702. const ImageViewId color_view_id = is_color ? view_id : ImageViewId{};
  1703. const ImageViewId depth_view_id = is_color ? ImageViewId{} : view_id;
  1704. Extent3D extent = MipSize(image.info.size, view_info.range.base.level);
  1705. if (is_rescaled) {
  1706. const auto& resolution = Settings::values.resolution_info;
  1707. const auto scale_up = [&](u32 value) {
  1708. return std::max<u32>((value * resolution.up_scale) >> resolution.down_shift, 1U);
  1709. };
  1710. extent.width = scale_up(extent.width);
  1711. if (image.info.type == ImageType::e2D) {
  1712. extent.height = scale_up(extent.height);
  1713. }
  1714. }
  1715. const u32 num_samples = image.info.num_samples;
  1716. const auto [samples_x, samples_y] = SamplesLog2(num_samples);
  1717. const FramebufferId framebuffer_id = GetFramebufferId(RenderTargets{
  1718. .color_buffer_ids = {color_view_id},
  1719. .depth_buffer_id = depth_view_id,
  1720. .size = {extent.width >> samples_x, extent.height >> samples_y},
  1721. });
  1722. return {framebuffer_id, view_id};
  1723. }
  1724. template <class P>
  1725. bool TextureCache<P>::IsFullClear(ImageViewId id) {
  1726. if (!id) {
  1727. return true;
  1728. }
  1729. const ImageViewBase& image_view = slot_image_views[id];
  1730. const ImageBase& image = slot_images[image_view.image_id];
  1731. const Extent3D size = image_view.size;
  1732. const auto& regs = maxwell3d.regs;
  1733. const auto& scissor = regs.scissor_test[0];
  1734. if (image.info.resources.levels > 1 || image.info.resources.layers > 1) {
  1735. // Images with multiple resources can't be cleared in a single call
  1736. return false;
  1737. }
  1738. if (regs.clear_flags.scissor == 0) {
  1739. // If scissor testing is disabled, the clear is always full
  1740. return true;
  1741. }
  1742. // Make sure the clear covers all texels in the subresource
  1743. return scissor.min_x == 0 && scissor.min_y == 0 && scissor.max_x >= size.width &&
  1744. scissor.max_y >= size.height;
  1745. }
  1746. } // namespace VideoCommon