texture_cache.h 78 KB

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