copy_effects.cpp 68 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297
  1. /**************************************************************************/
  2. /* copy_effects.cpp */
  3. /**************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /**************************************************************************/
  8. /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
  9. /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /**************************************************************************/
  30. #include "copy_effects.h"
  31. #include "core/config/project_settings.h"
  32. #include "servers/rendering/renderer_rd/renderer_compositor_rd.h"
  33. #include "servers/rendering/renderer_rd/storage_rd/material_storage.h"
  34. #include "servers/rendering/renderer_rd/uniform_set_cache_rd.h"
  35. #include "thirdparty/misc/cubemap_coeffs.h"
  36. using namespace RendererRD;
  37. CopyEffects *CopyEffects::singleton = nullptr;
  38. CopyEffects *CopyEffects::get_singleton() {
  39. return singleton;
  40. }
  41. CopyEffects::CopyEffects(bool p_prefer_raster_effects) {
  42. singleton = this;
  43. prefer_raster_effects = p_prefer_raster_effects;
  44. if (prefer_raster_effects) {
  45. // init blur shader (on compute use copy shader)
  46. Vector<String> blur_modes;
  47. blur_modes.push_back("\n#define MODE_MIPMAP\n"); // BLUR_MIPMAP
  48. blur_modes.push_back("\n#define MODE_GAUSSIAN_BLUR\n"); // BLUR_MODE_GAUSSIAN_BLUR
  49. blur_modes.push_back("\n#define MODE_GAUSSIAN_GLOW\n"); // BLUR_MODE_GAUSSIAN_GLOW
  50. blur_modes.push_back("\n#define MODE_GAUSSIAN_GLOW\n#define GLOW_USE_AUTO_EXPOSURE\n"); // BLUR_MODE_GAUSSIAN_GLOW_AUTO_EXPOSURE
  51. blur_modes.push_back("\n#define MODE_COPY\n"); // BLUR_MODE_COPY
  52. blur_modes.push_back("\n#define MODE_SET_COLOR\n"); // BLUR_MODE_SET_COLOR
  53. blur_raster.shader.initialize(blur_modes);
  54. memset(&blur_raster.push_constant, 0, sizeof(BlurRasterPushConstant));
  55. blur_raster.shader_version = blur_raster.shader.version_create();
  56. for (int i = 0; i < BLUR_MODE_MAX; i++) {
  57. blur_raster.pipelines[i].setup(blur_raster.shader.version_get_shader(blur_raster.shader_version, i), RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), RD::PipelineDepthStencilState(), RD::PipelineColorBlendState::create_disabled(), 0);
  58. }
  59. } else {
  60. // not used in clustered
  61. for (int i = 0; i < BLUR_MODE_MAX; i++) {
  62. blur_raster.pipelines[i].clear();
  63. }
  64. }
  65. {
  66. Vector<String> copy_modes;
  67. copy_modes.push_back("\n#define MODE_GAUSSIAN_BLUR\n");
  68. copy_modes.push_back("\n#define MODE_GAUSSIAN_BLUR\n#define DST_IMAGE_8BIT\n");
  69. copy_modes.push_back("\n#define MODE_GAUSSIAN_BLUR\n#define MODE_GLOW\n");
  70. copy_modes.push_back("\n#define MODE_GAUSSIAN_BLUR\n#define MODE_GLOW\n#define GLOW_USE_AUTO_EXPOSURE\n");
  71. copy_modes.push_back("\n#define MODE_SIMPLE_COPY\n");
  72. copy_modes.push_back("\n#define MODE_SIMPLE_COPY\n#define DST_IMAGE_8BIT\n");
  73. copy_modes.push_back("\n#define MODE_SIMPLE_COPY_DEPTH\n");
  74. copy_modes.push_back("\n#define MODE_SET_COLOR\n");
  75. copy_modes.push_back("\n#define MODE_SET_COLOR\n#define DST_IMAGE_8BIT\n");
  76. copy_modes.push_back("\n#define MODE_MIPMAP\n");
  77. copy_modes.push_back("\n#define MODE_LINEARIZE_DEPTH_COPY\n");
  78. copy_modes.push_back("\n#define MODE_CUBEMAP_TO_PANORAMA\n");
  79. copy_modes.push_back("\n#define MODE_CUBEMAP_ARRAY_TO_PANORAMA\n");
  80. copy.shader.initialize(copy_modes);
  81. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  82. copy.shader_version = copy.shader.version_create();
  83. for (int i = 0; i < COPY_MODE_MAX; i++) {
  84. if (copy.shader.is_variant_enabled(i)) {
  85. copy.pipelines[i] = RD::get_singleton()->compute_pipeline_create(copy.shader.version_get_shader(copy.shader_version, i));
  86. }
  87. }
  88. }
  89. {
  90. Vector<String> copy_modes;
  91. copy_modes.push_back("\n"); // COPY_TO_FB_COPY
  92. copy_modes.push_back("\n#define MODE_PANORAMA_TO_DP\n"); // COPY_TO_FB_COPY_PANORAMA_TO_DP
  93. copy_modes.push_back("\n#define MODE_TWO_SOURCES\n"); // COPY_TO_FB_COPY2
  94. copy_modes.push_back("\n#define MODE_SET_COLOR\n"); // COPY_TO_FB_SET_COLOR
  95. copy_modes.push_back("\n#define MULTIVIEW\n"); // COPY_TO_FB_MULTIVIEW
  96. copy_modes.push_back("\n#define MULTIVIEW\n#define MODE_TWO_SOURCES\n"); // COPY_TO_FB_MULTIVIEW_WITH_DEPTH
  97. copy_to_fb.shader.initialize(copy_modes);
  98. if (!RendererCompositorRD::get_singleton()->is_xr_enabled()) {
  99. copy_to_fb.shader.set_variant_enabled(COPY_TO_FB_MULTIVIEW, false);
  100. copy_to_fb.shader.set_variant_enabled(COPY_TO_FB_MULTIVIEW_WITH_DEPTH, false);
  101. }
  102. copy_to_fb.shader_version = copy_to_fb.shader.version_create();
  103. //use additive
  104. for (int i = 0; i < COPY_TO_FB_MAX; i++) {
  105. if (copy_to_fb.shader.is_variant_enabled(i)) {
  106. copy_to_fb.pipelines[i].setup(copy_to_fb.shader.version_get_shader(copy_to_fb.shader_version, i), RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), RD::PipelineDepthStencilState(), RD::PipelineColorBlendState::create_disabled(), 0);
  107. } else {
  108. copy_to_fb.pipelines[i].clear();
  109. }
  110. }
  111. }
  112. {
  113. // Initialize copier
  114. Vector<String> copy_modes;
  115. copy_modes.push_back("\n");
  116. cube_to_dp.shader.initialize(copy_modes);
  117. cube_to_dp.shader_version = cube_to_dp.shader.version_create();
  118. RID shader = cube_to_dp.shader.version_get_shader(cube_to_dp.shader_version, 0);
  119. RD::PipelineDepthStencilState dss;
  120. dss.enable_depth_test = true;
  121. dss.depth_compare_operator = RD::COMPARE_OP_ALWAYS;
  122. dss.enable_depth_write = true;
  123. cube_to_dp.pipeline.setup(shader, RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), dss, RD::PipelineColorBlendState(), 0);
  124. }
  125. {
  126. //Initialize cubemap downsampler
  127. Vector<String> cubemap_downsampler_modes;
  128. cubemap_downsampler_modes.push_back("");
  129. if (prefer_raster_effects) {
  130. cubemap_downsampler.raster_shader.initialize(cubemap_downsampler_modes);
  131. cubemap_downsampler.shader_version = cubemap_downsampler.raster_shader.version_create();
  132. cubemap_downsampler.raster_pipeline.setup(cubemap_downsampler.raster_shader.version_get_shader(cubemap_downsampler.shader_version, 0), RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), RD::PipelineDepthStencilState(), RD::PipelineColorBlendState::create_disabled(), 0);
  133. } else {
  134. cubemap_downsampler.compute_shader.initialize(cubemap_downsampler_modes);
  135. cubemap_downsampler.shader_version = cubemap_downsampler.compute_shader.version_create();
  136. cubemap_downsampler.compute_pipeline = RD::get_singleton()->compute_pipeline_create(cubemap_downsampler.compute_shader.version_get_shader(cubemap_downsampler.shader_version, 0));
  137. cubemap_downsampler.raster_pipeline.clear();
  138. }
  139. }
  140. {
  141. // Initialize cubemap filter
  142. filter.use_high_quality = GLOBAL_GET("rendering/reflections/sky_reflections/fast_filter_high_quality");
  143. Vector<String> cubemap_filter_modes;
  144. cubemap_filter_modes.push_back("\n#define USE_HIGH_QUALITY\n");
  145. cubemap_filter_modes.push_back("\n#define USE_LOW_QUALITY\n");
  146. cubemap_filter_modes.push_back("\n#define USE_HIGH_QUALITY\n#define USE_TEXTURE_ARRAY\n");
  147. cubemap_filter_modes.push_back("\n#define USE_LOW_QUALITY\n#define USE_TEXTURE_ARRAY\n");
  148. if (filter.use_high_quality) {
  149. filter.coefficient_buffer = RD::get_singleton()->storage_buffer_create(sizeof(high_quality_coeffs));
  150. RD::get_singleton()->buffer_update(filter.coefficient_buffer, 0, sizeof(high_quality_coeffs), &high_quality_coeffs[0]);
  151. } else {
  152. filter.coefficient_buffer = RD::get_singleton()->storage_buffer_create(sizeof(low_quality_coeffs));
  153. RD::get_singleton()->buffer_update(filter.coefficient_buffer, 0, sizeof(low_quality_coeffs), &low_quality_coeffs[0]);
  154. }
  155. if (prefer_raster_effects) {
  156. filter.raster_shader.initialize(cubemap_filter_modes);
  157. // array variants are not supported in raster
  158. filter.raster_shader.set_variant_enabled(FILTER_MODE_HIGH_QUALITY_ARRAY, false);
  159. filter.raster_shader.set_variant_enabled(FILTER_MODE_LOW_QUALITY_ARRAY, false);
  160. filter.shader_version = filter.raster_shader.version_create();
  161. for (int i = 0; i < FILTER_MODE_MAX; i++) {
  162. if (filter.raster_shader.is_variant_enabled(i)) {
  163. filter.raster_pipelines[i].setup(filter.raster_shader.version_get_shader(filter.shader_version, i), RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), RD::PipelineDepthStencilState(), RD::PipelineColorBlendState::create_disabled(), 0);
  164. } else {
  165. filter.raster_pipelines[i].clear();
  166. }
  167. }
  168. Vector<RD::Uniform> uniforms;
  169. {
  170. RD::Uniform u;
  171. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  172. u.binding = 0;
  173. u.append_id(filter.coefficient_buffer);
  174. uniforms.push_back(u);
  175. }
  176. filter.uniform_set = RD::get_singleton()->uniform_set_create(uniforms, filter.raster_shader.version_get_shader(filter.shader_version, filter.use_high_quality ? 0 : 1), 1);
  177. } else {
  178. filter.compute_shader.initialize(cubemap_filter_modes);
  179. filter.shader_version = filter.compute_shader.version_create();
  180. for (int i = 0; i < FILTER_MODE_MAX; i++) {
  181. filter.compute_pipelines[i] = RD::get_singleton()->compute_pipeline_create(filter.compute_shader.version_get_shader(filter.shader_version, i));
  182. filter.raster_pipelines[i].clear();
  183. }
  184. Vector<RD::Uniform> uniforms;
  185. {
  186. RD::Uniform u;
  187. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  188. u.binding = 0;
  189. u.append_id(filter.coefficient_buffer);
  190. uniforms.push_back(u);
  191. }
  192. filter.uniform_set = RD::get_singleton()->uniform_set_create(uniforms, filter.compute_shader.version_get_shader(filter.shader_version, filter.use_high_quality ? 0 : 1), 1);
  193. }
  194. }
  195. {
  196. // Initialize roughness
  197. Vector<String> cubemap_roughness_modes;
  198. cubemap_roughness_modes.push_back("");
  199. if (prefer_raster_effects) {
  200. roughness.raster_shader.initialize(cubemap_roughness_modes);
  201. roughness.shader_version = roughness.raster_shader.version_create();
  202. roughness.raster_pipeline.setup(roughness.raster_shader.version_get_shader(roughness.shader_version, 0), RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), RD::PipelineDepthStencilState(), RD::PipelineColorBlendState::create_disabled(), 0);
  203. } else {
  204. roughness.compute_shader.initialize(cubemap_roughness_modes);
  205. roughness.shader_version = roughness.compute_shader.version_create();
  206. roughness.compute_pipeline = RD::get_singleton()->compute_pipeline_create(roughness.compute_shader.version_get_shader(roughness.shader_version, 0));
  207. roughness.raster_pipeline.clear();
  208. }
  209. }
  210. {
  211. Vector<String> specular_modes;
  212. specular_modes.push_back("\n#define MODE_MERGE\n"); // SPECULAR_MERGE_ADD
  213. specular_modes.push_back("\n#define MODE_MERGE\n#define MODE_SSR\n"); // SPECULAR_MERGE_SSR
  214. specular_modes.push_back("\n"); // SPECULAR_MERGE_ADDITIVE_ADD
  215. specular_modes.push_back("\n#define MODE_SSR\n"); // SPECULAR_MERGE_ADDITIVE_SSR
  216. specular_modes.push_back("\n#define USE_MULTIVIEW\n#define MODE_MERGE\n"); // SPECULAR_MERGE_ADD_MULTIVIEW
  217. specular_modes.push_back("\n#define USE_MULTIVIEW\n#define MODE_MERGE\n#define MODE_SSR\n"); // SPECULAR_MERGE_SSR_MULTIVIEW
  218. specular_modes.push_back("\n#define USE_MULTIVIEW\n"); // SPECULAR_MERGE_ADDITIVE_ADD_MULTIVIEW
  219. specular_modes.push_back("\n#define USE_MULTIVIEW\n#define MODE_SSR\n"); // SPECULAR_MERGE_ADDITIVE_SSR_MULTIVIEW
  220. specular_merge.shader.initialize(specular_modes);
  221. if (!RendererCompositorRD::get_singleton()->is_xr_enabled()) {
  222. specular_merge.shader.set_variant_enabled(SPECULAR_MERGE_ADD_MULTIVIEW, false);
  223. specular_merge.shader.set_variant_enabled(SPECULAR_MERGE_SSR_MULTIVIEW, false);
  224. specular_merge.shader.set_variant_enabled(SPECULAR_MERGE_ADDITIVE_ADD_MULTIVIEW, false);
  225. specular_merge.shader.set_variant_enabled(SPECULAR_MERGE_ADDITIVE_SSR_MULTIVIEW, false);
  226. }
  227. specular_merge.shader_version = specular_merge.shader.version_create();
  228. //use additive
  229. RD::PipelineColorBlendState::Attachment ba;
  230. ba.enable_blend = true;
  231. ba.src_color_blend_factor = RD::BLEND_FACTOR_ONE;
  232. ba.dst_color_blend_factor = RD::BLEND_FACTOR_ONE;
  233. ba.src_alpha_blend_factor = RD::BLEND_FACTOR_ZERO;
  234. ba.dst_alpha_blend_factor = RD::BLEND_FACTOR_ZERO;
  235. ba.color_blend_op = RD::BLEND_OP_ADD;
  236. ba.alpha_blend_op = RD::BLEND_OP_ADD;
  237. RD::PipelineColorBlendState blend_additive;
  238. blend_additive.attachments.push_back(ba);
  239. for (int i = 0; i < SPECULAR_MERGE_MAX; i++) {
  240. if (specular_merge.shader.is_variant_enabled(i)) {
  241. RD::PipelineColorBlendState blend_state;
  242. if (i == SPECULAR_MERGE_ADDITIVE_ADD || i == SPECULAR_MERGE_ADDITIVE_SSR || i == SPECULAR_MERGE_ADDITIVE_ADD_MULTIVIEW || i == SPECULAR_MERGE_ADDITIVE_SSR_MULTIVIEW) {
  243. blend_state = blend_additive;
  244. } else {
  245. blend_state = RD::PipelineColorBlendState::create_disabled();
  246. }
  247. specular_merge.pipelines[i].setup(specular_merge.shader.version_get_shader(specular_merge.shader_version, i), RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), RD::PipelineDepthStencilState(), blend_state, 0);
  248. }
  249. }
  250. }
  251. }
  252. CopyEffects::~CopyEffects() {
  253. if (prefer_raster_effects) {
  254. blur_raster.shader.version_free(blur_raster.shader_version);
  255. cubemap_downsampler.raster_shader.version_free(cubemap_downsampler.shader_version);
  256. filter.raster_shader.version_free(filter.shader_version);
  257. roughness.raster_shader.version_free(roughness.shader_version);
  258. } else {
  259. cubemap_downsampler.compute_shader.version_free(cubemap_downsampler.shader_version);
  260. filter.compute_shader.version_free(filter.shader_version);
  261. roughness.compute_shader.version_free(roughness.shader_version);
  262. }
  263. copy.shader.version_free(copy.shader_version);
  264. specular_merge.shader.version_free(specular_merge.shader_version);
  265. RD::get_singleton()->free(filter.coefficient_buffer);
  266. if (RD::get_singleton()->uniform_set_is_valid(filter.image_uniform_set)) {
  267. RD::get_singleton()->free(filter.image_uniform_set);
  268. }
  269. if (RD::get_singleton()->uniform_set_is_valid(filter.uniform_set)) {
  270. RD::get_singleton()->free(filter.uniform_set);
  271. }
  272. copy_to_fb.shader.version_free(copy_to_fb.shader_version);
  273. cube_to_dp.shader.version_free(cube_to_dp.shader_version);
  274. singleton = nullptr;
  275. }
  276. void CopyEffects::copy_to_rect(RID p_source_rd_texture, RID p_dest_texture, const Rect2i &p_rect, bool p_flip_y, bool p_force_luminance, bool p_all_source, bool p_8_bit_dst, bool p_alpha_to_one) {
  277. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  278. ERR_FAIL_NULL(uniform_set_cache);
  279. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  280. ERR_FAIL_NULL(material_storage);
  281. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  282. if (p_flip_y) {
  283. copy.push_constant.flags |= COPY_FLAG_FLIP_Y;
  284. }
  285. if (p_force_luminance) {
  286. copy.push_constant.flags |= COPY_FLAG_FORCE_LUMINANCE;
  287. }
  288. if (p_all_source) {
  289. copy.push_constant.flags |= COPY_FLAG_ALL_SOURCE;
  290. }
  291. if (p_alpha_to_one) {
  292. copy.push_constant.flags |= COPY_FLAG_ALPHA_TO_ONE;
  293. }
  294. copy.push_constant.section[0] = p_rect.position.x;
  295. copy.push_constant.section[1] = p_rect.position.y;
  296. copy.push_constant.section[2] = p_rect.size.width;
  297. copy.push_constant.section[3] = p_rect.size.height;
  298. copy.push_constant.target[0] = p_rect.position.x;
  299. copy.push_constant.target[1] = p_rect.position.y;
  300. // setup our uniforms
  301. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  302. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  303. RD::Uniform u_dest_texture(RD::UNIFORM_TYPE_IMAGE, 0, p_dest_texture);
  304. CopyMode mode = p_8_bit_dst ? COPY_MODE_SIMPLY_COPY_8BIT : COPY_MODE_SIMPLY_COPY;
  305. RID shader = copy.shader.version_get_shader(copy.shader_version, mode);
  306. ERR_FAIL_COND(shader.is_null());
  307. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  308. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[mode]);
  309. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  310. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_dest_texture), 3);
  311. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  312. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_rect.size.width, p_rect.size.height, 1);
  313. RD::get_singleton()->compute_list_end();
  314. }
  315. void CopyEffects::copy_cubemap_to_panorama(RID p_source_cube, RID p_dest_panorama, const Size2i &p_panorama_size, float p_lod, bool p_is_array) {
  316. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  317. ERR_FAIL_NULL(uniform_set_cache);
  318. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  319. ERR_FAIL_NULL(material_storage);
  320. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  321. copy.push_constant.section[0] = 0;
  322. copy.push_constant.section[1] = 0;
  323. copy.push_constant.section[2] = p_panorama_size.width;
  324. copy.push_constant.section[3] = p_panorama_size.height;
  325. copy.push_constant.target[0] = 0;
  326. copy.push_constant.target[1] = 0;
  327. copy.push_constant.camera_z_far = p_lod;
  328. // setup our uniforms
  329. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  330. RD::Uniform u_source_cube(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_cube }));
  331. RD::Uniform u_dest_panorama(RD::UNIFORM_TYPE_IMAGE, 0, p_dest_panorama);
  332. CopyMode mode = p_is_array ? COPY_MODE_CUBE_ARRAY_TO_PANORAMA : COPY_MODE_CUBE_TO_PANORAMA;
  333. RID shader = copy.shader.version_get_shader(copy.shader_version, mode);
  334. ERR_FAIL_COND(shader.is_null());
  335. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  336. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[mode]);
  337. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_cube), 0);
  338. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_dest_panorama), 3);
  339. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  340. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_panorama_size.width, p_panorama_size.height, 1);
  341. RD::get_singleton()->compute_list_end();
  342. }
  343. void CopyEffects::copy_depth_to_rect(RID p_source_rd_texture, RID p_dest_texture, const Rect2i &p_rect, bool p_flip_y) {
  344. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  345. ERR_FAIL_NULL(uniform_set_cache);
  346. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  347. ERR_FAIL_NULL(material_storage);
  348. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  349. if (p_flip_y) {
  350. copy.push_constant.flags |= COPY_FLAG_FLIP_Y;
  351. }
  352. copy.push_constant.section[0] = 0;
  353. copy.push_constant.section[1] = 0;
  354. copy.push_constant.section[2] = p_rect.size.width;
  355. copy.push_constant.section[3] = p_rect.size.height;
  356. copy.push_constant.target[0] = p_rect.position.x;
  357. copy.push_constant.target[1] = p_rect.position.y;
  358. // setup our uniforms
  359. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  360. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  361. RD::Uniform u_dest_texture(RD::UNIFORM_TYPE_IMAGE, 0, p_dest_texture);
  362. CopyMode mode = COPY_MODE_SIMPLY_COPY_DEPTH;
  363. RID shader = copy.shader.version_get_shader(copy.shader_version, mode);
  364. ERR_FAIL_COND(shader.is_null());
  365. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  366. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[mode]);
  367. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  368. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_dest_texture), 3);
  369. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  370. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_rect.size.width, p_rect.size.height, 1);
  371. RD::get_singleton()->compute_list_end();
  372. }
  373. void CopyEffects::copy_depth_to_rect_and_linearize(RID p_source_rd_texture, RID p_dest_texture, const Rect2i &p_rect, bool p_flip_y, float p_z_near, float p_z_far) {
  374. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  375. ERR_FAIL_NULL(uniform_set_cache);
  376. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  377. ERR_FAIL_NULL(material_storage);
  378. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  379. if (p_flip_y) {
  380. copy.push_constant.flags |= COPY_FLAG_FLIP_Y;
  381. }
  382. copy.push_constant.section[0] = 0;
  383. copy.push_constant.section[1] = 0;
  384. copy.push_constant.section[2] = p_rect.size.width;
  385. copy.push_constant.section[3] = p_rect.size.height;
  386. copy.push_constant.target[0] = p_rect.position.x;
  387. copy.push_constant.target[1] = p_rect.position.y;
  388. copy.push_constant.camera_z_far = p_z_far;
  389. copy.push_constant.camera_z_near = p_z_near;
  390. // setup our uniforms
  391. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  392. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  393. RD::Uniform u_dest_texture(RD::UNIFORM_TYPE_IMAGE, 0, p_dest_texture);
  394. CopyMode mode = COPY_MODE_LINEARIZE_DEPTH;
  395. RID shader = copy.shader.version_get_shader(copy.shader_version, mode);
  396. ERR_FAIL_COND(shader.is_null());
  397. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  398. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[mode]);
  399. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  400. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_dest_texture), 3);
  401. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  402. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_rect.size.width, p_rect.size.height, 1);
  403. RD::get_singleton()->compute_list_end();
  404. }
  405. void CopyEffects::copy_to_atlas_fb(RID p_source_rd_texture, RID p_dest_framebuffer, const Rect2 &p_uv_rect, RD::DrawListID p_draw_list, bool p_flip_y, bool p_panorama) {
  406. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  407. ERR_FAIL_NULL(uniform_set_cache);
  408. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  409. ERR_FAIL_NULL(material_storage);
  410. memset(&copy_to_fb.push_constant, 0, sizeof(CopyToFbPushConstant));
  411. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_USE_SECTION;
  412. copy_to_fb.push_constant.section[0] = p_uv_rect.position.x;
  413. copy_to_fb.push_constant.section[1] = p_uv_rect.position.y;
  414. copy_to_fb.push_constant.section[2] = p_uv_rect.size.x;
  415. copy_to_fb.push_constant.section[3] = p_uv_rect.size.y;
  416. if (p_flip_y) {
  417. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_FLIP_Y;
  418. }
  419. copy_to_fb.push_constant.luminance_multiplier = 1.0;
  420. // setup our uniforms
  421. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  422. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  423. CopyToFBMode mode = p_panorama ? COPY_TO_FB_COPY_PANORAMA_TO_DP : COPY_TO_FB_COPY;
  424. RID shader = copy_to_fb.shader.version_get_shader(copy_to_fb.shader_version, mode);
  425. ERR_FAIL_COND(shader.is_null());
  426. RD::DrawListID draw_list = p_draw_list;
  427. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, copy_to_fb.pipelines[mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dest_framebuffer)));
  428. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  429. RD::get_singleton()->draw_list_bind_index_array(draw_list, material_storage->get_quad_index_array());
  430. RD::get_singleton()->draw_list_set_push_constant(draw_list, &copy_to_fb.push_constant, sizeof(CopyToFbPushConstant));
  431. RD::get_singleton()->draw_list_draw(draw_list, true);
  432. }
  433. void CopyEffects::copy_to_fb_rect(RID p_source_rd_texture, RID p_dest_framebuffer, const Rect2i &p_rect, bool p_flip_y, bool p_force_luminance, bool p_alpha_to_zero, bool p_srgb, RID p_secondary, bool p_multiview, bool p_alpha_to_one, bool p_linear) {
  434. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  435. ERR_FAIL_NULL(uniform_set_cache);
  436. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  437. ERR_FAIL_NULL(material_storage);
  438. memset(&copy_to_fb.push_constant, 0, sizeof(CopyToFbPushConstant));
  439. copy_to_fb.push_constant.luminance_multiplier = 1.0;
  440. if (p_flip_y) {
  441. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_FLIP_Y;
  442. }
  443. if (p_force_luminance) {
  444. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_FORCE_LUMINANCE;
  445. }
  446. if (p_alpha_to_zero) {
  447. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_ALPHA_TO_ZERO;
  448. }
  449. if (p_srgb) {
  450. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_SRGB;
  451. }
  452. if (p_alpha_to_one) {
  453. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_ALPHA_TO_ONE;
  454. }
  455. if (p_linear) {
  456. // Used for copying to a linear buffer. In the mobile renderer we divide the contents of the linear buffer
  457. // to allow for a wider effective range.
  458. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_LINEAR;
  459. copy_to_fb.push_constant.luminance_multiplier = prefer_raster_effects ? 2.0 : 1.0;
  460. }
  461. // setup our uniforms
  462. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  463. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  464. CopyToFBMode mode;
  465. if (p_multiview) {
  466. mode = p_secondary.is_valid() ? COPY_TO_FB_MULTIVIEW_WITH_DEPTH : COPY_TO_FB_MULTIVIEW;
  467. } else {
  468. mode = p_secondary.is_valid() ? COPY_TO_FB_COPY2 : COPY_TO_FB_COPY;
  469. }
  470. RID shader = copy_to_fb.shader.version_get_shader(copy_to_fb.shader_version, mode);
  471. ERR_FAIL_COND(shader.is_null());
  472. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(p_dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD, Vector<Color>(), 1.0, 0, p_rect);
  473. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, copy_to_fb.pipelines[mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dest_framebuffer)));
  474. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  475. if (p_secondary.is_valid()) {
  476. // TODO may need to do this differently when reading from depth buffer for multiview
  477. RD::Uniform u_secondary(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_secondary }));
  478. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 1, u_secondary), 1);
  479. }
  480. RD::get_singleton()->draw_list_bind_index_array(draw_list, material_storage->get_quad_index_array());
  481. RD::get_singleton()->draw_list_set_push_constant(draw_list, &copy_to_fb.push_constant, sizeof(CopyToFbPushConstant));
  482. RD::get_singleton()->draw_list_draw(draw_list, true);
  483. RD::get_singleton()->draw_list_end();
  484. }
  485. void CopyEffects::copy_to_drawlist(RD::DrawListID p_draw_list, RD::FramebufferFormatID p_fb_format, RID p_source_rd_texture, bool p_linear) {
  486. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  487. ERR_FAIL_NULL(uniform_set_cache);
  488. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  489. ERR_FAIL_NULL(material_storage);
  490. memset(&copy_to_fb.push_constant, 0, sizeof(CopyToFbPushConstant));
  491. copy_to_fb.push_constant.luminance_multiplier = 1.0;
  492. if (p_linear) {
  493. // Used for copying to a linear buffer. In the mobile renderer we divide the contents of the linear buffer
  494. // to allow for a wider effective range.
  495. copy_to_fb.push_constant.flags |= COPY_TO_FB_FLAG_LINEAR;
  496. copy_to_fb.push_constant.luminance_multiplier = prefer_raster_effects ? 2.0 : 1.0;
  497. }
  498. // setup our uniforms
  499. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  500. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  501. // Multiview not supported here!
  502. CopyToFBMode mode = COPY_TO_FB_COPY;
  503. RID shader = copy_to_fb.shader.version_get_shader(copy_to_fb.shader_version, mode);
  504. ERR_FAIL_COND(shader.is_null());
  505. RD::get_singleton()->draw_list_bind_render_pipeline(p_draw_list, copy_to_fb.pipelines[mode].get_render_pipeline(RD::INVALID_ID, p_fb_format));
  506. RD::get_singleton()->draw_list_bind_uniform_set(p_draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  507. RD::get_singleton()->draw_list_bind_index_array(p_draw_list, material_storage->get_quad_index_array());
  508. RD::get_singleton()->draw_list_set_push_constant(p_draw_list, &copy_to_fb.push_constant, sizeof(CopyToFbPushConstant));
  509. RD::get_singleton()->draw_list_draw(p_draw_list, true);
  510. }
  511. void CopyEffects::copy_raster(RID p_source_texture, RID p_dest_framebuffer) {
  512. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use the raster version of the copy with the clustered renderer.");
  513. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  514. ERR_FAIL_NULL(uniform_set_cache);
  515. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  516. ERR_FAIL_NULL(material_storage);
  517. memset(&blur_raster.push_constant, 0, sizeof(BlurRasterPushConstant));
  518. // setup our uniforms
  519. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  520. RD::Uniform u_source_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_texture }));
  521. RID shader = blur_raster.shader.version_get_shader(blur_raster.shader_version, BLUR_MODE_COPY);
  522. ERR_FAIL_COND(shader.is_null());
  523. // Just copy it back (we use our blur raster shader here)..
  524. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(p_dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  525. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, blur_raster.pipelines[BLUR_MODE_COPY].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dest_framebuffer)));
  526. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_texture), 0);
  527. RD::get_singleton()->draw_list_bind_index_array(draw_list, material_storage->get_quad_index_array());
  528. RD::get_singleton()->draw_list_set_push_constant(draw_list, &blur_raster.push_constant, sizeof(BlurRasterPushConstant));
  529. RD::get_singleton()->draw_list_draw(draw_list, true);
  530. RD::get_singleton()->draw_list_end();
  531. }
  532. void CopyEffects::gaussian_blur(RID p_source_rd_texture, RID p_texture, const Rect2i &p_region, const Size2i &p_size, bool p_8bit_dst) {
  533. ERR_FAIL_COND_MSG(prefer_raster_effects, "Can't use the compute version of the gaussian blur with the mobile renderer.");
  534. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  535. ERR_FAIL_NULL(uniform_set_cache);
  536. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  537. ERR_FAIL_NULL(material_storage);
  538. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  539. copy.push_constant.section[0] = p_region.position.x;
  540. copy.push_constant.section[1] = p_region.position.y;
  541. copy.push_constant.target[0] = p_region.position.x;
  542. copy.push_constant.target[1] = p_region.position.y;
  543. copy.push_constant.section[2] = p_size.width;
  544. copy.push_constant.section[3] = p_size.height;
  545. // setup our uniforms
  546. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  547. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  548. RD::Uniform u_texture(RD::UNIFORM_TYPE_IMAGE, 0, p_texture);
  549. CopyMode mode = p_8bit_dst ? COPY_MODE_GAUSSIAN_COPY_8BIT : COPY_MODE_GAUSSIAN_COPY;
  550. RID shader = copy.shader.version_get_shader(copy.shader_version, mode);
  551. ERR_FAIL_COND(shader.is_null());
  552. RD::DrawListID compute_list = RD::get_singleton()->compute_list_begin();
  553. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[mode]);
  554. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  555. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_texture), 3);
  556. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  557. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_region.size.width, p_region.size.height, 1);
  558. RD::get_singleton()->compute_list_end();
  559. }
  560. void CopyEffects::gaussian_blur_raster(RID p_source_rd_texture, RID p_dest_texture, const Rect2i &p_region, const Size2i &p_size) {
  561. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use the raster version of the gaussian blur with the clustered renderer.");
  562. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  563. ERR_FAIL_NULL(uniform_set_cache);
  564. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  565. ERR_FAIL_NULL(material_storage);
  566. RID dest_framebuffer = FramebufferCacheRD::get_singleton()->get_cache(p_dest_texture);
  567. memset(&blur_raster.push_constant, 0, sizeof(BlurRasterPushConstant));
  568. BlurRasterMode blur_mode = BLUR_MODE_GAUSSIAN_BLUR;
  569. blur_raster.push_constant.pixel_size[0] = 1.0 / float(p_size.x);
  570. blur_raster.push_constant.pixel_size[1] = 1.0 / float(p_size.y);
  571. // setup our uniforms
  572. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  573. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  574. RID shader = blur_raster.shader.version_get_shader(blur_raster.shader_version, blur_mode);
  575. ERR_FAIL_COND(shader.is_null());
  576. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  577. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, blur_raster.pipelines[blur_mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(dest_framebuffer)));
  578. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  579. RD::get_singleton()->draw_list_set_push_constant(draw_list, &blur_raster.push_constant, sizeof(BlurRasterPushConstant));
  580. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  581. RD::get_singleton()->draw_list_end();
  582. }
  583. void CopyEffects::gaussian_glow(RID p_source_rd_texture, RID p_back_texture, const Size2i &p_size, float p_strength, bool p_first_pass, float p_luminance_cap, float p_exposure, float p_bloom, float p_hdr_bleed_threshold, float p_hdr_bleed_scale, RID p_auto_exposure, float p_auto_exposure_scale) {
  584. ERR_FAIL_COND_MSG(prefer_raster_effects, "Can't use the compute version of the gaussian glow with the mobile renderer.");
  585. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  586. ERR_FAIL_NULL(uniform_set_cache);
  587. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  588. ERR_FAIL_NULL(material_storage);
  589. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  590. CopyMode copy_mode = p_first_pass && p_auto_exposure.is_valid() ? COPY_MODE_GAUSSIAN_GLOW_AUTO_EXPOSURE : COPY_MODE_GAUSSIAN_GLOW;
  591. uint32_t base_flags = 0;
  592. copy.push_constant.section[2] = p_size.x;
  593. copy.push_constant.section[3] = p_size.y;
  594. copy.push_constant.glow_strength = p_strength;
  595. copy.push_constant.glow_bloom = p_bloom;
  596. copy.push_constant.glow_hdr_threshold = p_hdr_bleed_threshold;
  597. copy.push_constant.glow_hdr_scale = p_hdr_bleed_scale;
  598. copy.push_constant.glow_exposure = p_exposure;
  599. copy.push_constant.glow_white = 0; //actually unused
  600. copy.push_constant.glow_luminance_cap = p_luminance_cap;
  601. copy.push_constant.glow_auto_exposure_scale = p_auto_exposure_scale; //unused also
  602. // setup our uniforms
  603. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  604. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  605. RD::Uniform u_back_texture(RD::UNIFORM_TYPE_IMAGE, 0, p_back_texture);
  606. RID shader = copy.shader.version_get_shader(copy.shader_version, copy_mode);
  607. ERR_FAIL_COND(shader.is_null());
  608. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  609. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[copy_mode]);
  610. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  611. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_back_texture), 3);
  612. if (p_auto_exposure.is_valid() && p_first_pass) {
  613. RD::Uniform u_auto_exposure(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_auto_exposure }));
  614. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 1, u_auto_exposure), 1);
  615. }
  616. copy.push_constant.flags = base_flags | (p_first_pass ? COPY_FLAG_GLOW_FIRST_PASS : 0);
  617. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  618. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_size.width, p_size.height, 1);
  619. RD::get_singleton()->compute_list_end();
  620. }
  621. void CopyEffects::gaussian_glow_raster(RID p_source_rd_texture, RID p_half_texture, RID p_dest_texture, float p_luminance_multiplier, const Size2i &p_size, float p_strength, bool p_first_pass, float p_luminance_cap, float p_exposure, float p_bloom, float p_hdr_bleed_threshold, float p_hdr_bleed_scale, RID p_auto_exposure, float p_auto_exposure_scale) {
  622. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use the raster version of the gaussian glow with the clustered renderer.");
  623. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  624. ERR_FAIL_NULL(uniform_set_cache);
  625. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  626. ERR_FAIL_NULL(material_storage);
  627. RID half_framebuffer = FramebufferCacheRD::get_singleton()->get_cache(p_half_texture);
  628. RID dest_framebuffer = FramebufferCacheRD::get_singleton()->get_cache(p_dest_texture);
  629. memset(&blur_raster.push_constant, 0, sizeof(BlurRasterPushConstant));
  630. BlurRasterMode blur_mode = p_first_pass && p_auto_exposure.is_valid() ? BLUR_MODE_GAUSSIAN_GLOW_AUTO_EXPOSURE : BLUR_MODE_GAUSSIAN_GLOW;
  631. uint32_t base_flags = 0;
  632. blur_raster.push_constant.pixel_size[0] = 1.0 / float(p_size.x);
  633. blur_raster.push_constant.pixel_size[1] = 1.0 / float(p_size.y);
  634. blur_raster.push_constant.glow_strength = p_strength;
  635. blur_raster.push_constant.glow_bloom = p_bloom;
  636. blur_raster.push_constant.glow_hdr_threshold = p_hdr_bleed_threshold;
  637. blur_raster.push_constant.glow_hdr_scale = p_hdr_bleed_scale;
  638. blur_raster.push_constant.glow_exposure = p_exposure;
  639. blur_raster.push_constant.glow_white = 0; //actually unused
  640. blur_raster.push_constant.glow_luminance_cap = p_luminance_cap;
  641. blur_raster.push_constant.glow_auto_exposure_scale = p_auto_exposure_scale; //unused also
  642. blur_raster.push_constant.luminance_multiplier = p_luminance_multiplier;
  643. // setup our uniforms
  644. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  645. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  646. RD::Uniform u_half_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_half_texture }));
  647. RID shader = blur_raster.shader.version_get_shader(blur_raster.shader_version, blur_mode);
  648. ERR_FAIL_COND(shader.is_null());
  649. //HORIZONTAL
  650. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(half_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  651. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, blur_raster.pipelines[blur_mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(half_framebuffer)));
  652. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  653. if (p_auto_exposure.is_valid() && p_first_pass) {
  654. RD::Uniform u_auto_exposure(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_auto_exposure }));
  655. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 1, u_auto_exposure), 1);
  656. }
  657. blur_raster.push_constant.flags = base_flags | BLUR_FLAG_HORIZONTAL | (p_first_pass ? BLUR_FLAG_GLOW_FIRST_PASS : 0);
  658. RD::get_singleton()->draw_list_set_push_constant(draw_list, &blur_raster.push_constant, sizeof(BlurRasterPushConstant));
  659. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  660. RD::get_singleton()->draw_list_end();
  661. blur_mode = BLUR_MODE_GAUSSIAN_GLOW;
  662. shader = blur_raster.shader.version_get_shader(blur_raster.shader_version, blur_mode);
  663. ERR_FAIL_COND(shader.is_null());
  664. //VERTICAL
  665. draw_list = RD::get_singleton()->draw_list_begin(dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  666. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, blur_raster.pipelines[blur_mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(dest_framebuffer)));
  667. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_half_texture), 0);
  668. blur_raster.push_constant.flags = base_flags;
  669. RD::get_singleton()->draw_list_set_push_constant(draw_list, &blur_raster.push_constant, sizeof(BlurRasterPushConstant));
  670. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  671. RD::get_singleton()->draw_list_end();
  672. }
  673. void CopyEffects::make_mipmap(RID p_source_rd_texture, RID p_dest_texture, const Size2i &p_size) {
  674. ERR_FAIL_COND_MSG(prefer_raster_effects, "Can't use the compute version of the make_mipmap shader with the mobile renderer.");
  675. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  676. ERR_FAIL_NULL(uniform_set_cache);
  677. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  678. ERR_FAIL_NULL(material_storage);
  679. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  680. copy.push_constant.section[0] = 0;
  681. copy.push_constant.section[1] = 0;
  682. copy.push_constant.section[2] = p_size.width;
  683. copy.push_constant.section[3] = p_size.height;
  684. // setup our uniforms
  685. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  686. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  687. RD::Uniform u_dest_texture(RD::UNIFORM_TYPE_IMAGE, 0, p_dest_texture);
  688. CopyMode mode = COPY_MODE_MIPMAP;
  689. RID shader = copy.shader.version_get_shader(copy.shader_version, mode);
  690. ERR_FAIL_COND(shader.is_null());
  691. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  692. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[mode]);
  693. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  694. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_dest_texture), 3);
  695. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  696. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_size.width, p_size.height, 1);
  697. RD::get_singleton()->compute_list_end();
  698. }
  699. void CopyEffects::make_mipmap_raster(RID p_source_rd_texture, RID p_dest_texture, const Size2i &p_size) {
  700. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use the raster version of mipmap with the clustered renderer.");
  701. RID dest_framebuffer = FramebufferCacheRD::get_singleton()->get_cache(p_dest_texture);
  702. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  703. ERR_FAIL_NULL(uniform_set_cache);
  704. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  705. ERR_FAIL_NULL(material_storage);
  706. memset(&blur_raster.push_constant, 0, sizeof(BlurRasterPushConstant));
  707. BlurRasterMode mode = BLUR_MIPMAP;
  708. blur_raster.push_constant.pixel_size[0] = 1.0 / float(p_size.x);
  709. blur_raster.push_constant.pixel_size[1] = 1.0 / float(p_size.y);
  710. // setup our uniforms
  711. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  712. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  713. RID shader = blur_raster.shader.version_get_shader(blur_raster.shader_version, mode);
  714. ERR_FAIL_COND(shader.is_null());
  715. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  716. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, blur_raster.pipelines[mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(dest_framebuffer)));
  717. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  718. RD::get_singleton()->draw_list_bind_index_array(draw_list, material_storage->get_quad_index_array());
  719. RD::get_singleton()->draw_list_set_push_constant(draw_list, &blur_raster.push_constant, sizeof(BlurRasterPushConstant));
  720. RD::get_singleton()->draw_list_draw(draw_list, true);
  721. RD::get_singleton()->draw_list_end();
  722. }
  723. void CopyEffects::set_color(RID p_dest_texture, const Color &p_color, const Rect2i &p_region, bool p_8bit_dst) {
  724. ERR_FAIL_COND_MSG(prefer_raster_effects, "Can't use the compute version of the set_color shader with the mobile renderer.");
  725. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  726. ERR_FAIL_NULL(uniform_set_cache);
  727. memset(&copy.push_constant, 0, sizeof(CopyPushConstant));
  728. copy.push_constant.section[0] = 0;
  729. copy.push_constant.section[1] = 0;
  730. copy.push_constant.section[2] = p_region.size.width;
  731. copy.push_constant.section[3] = p_region.size.height;
  732. copy.push_constant.target[0] = p_region.position.x;
  733. copy.push_constant.target[1] = p_region.position.y;
  734. copy.push_constant.set_color[0] = p_color.r;
  735. copy.push_constant.set_color[1] = p_color.g;
  736. copy.push_constant.set_color[2] = p_color.b;
  737. copy.push_constant.set_color[3] = p_color.a;
  738. // setup our uniforms
  739. RD::Uniform u_dest_texture(RD::UNIFORM_TYPE_IMAGE, 0, p_dest_texture);
  740. CopyMode mode = p_8bit_dst ? COPY_MODE_SET_COLOR_8BIT : COPY_MODE_SET_COLOR;
  741. RID shader = copy.shader.version_get_shader(copy.shader_version, mode);
  742. ERR_FAIL_COND(shader.is_null());
  743. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  744. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, copy.pipelines[mode]);
  745. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 3, u_dest_texture), 3);
  746. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy.push_constant, sizeof(CopyPushConstant));
  747. RD::get_singleton()->compute_list_dispatch_threads(compute_list, p_region.size.width, p_region.size.height, 1);
  748. RD::get_singleton()->compute_list_end();
  749. }
  750. void CopyEffects::set_color_raster(RID p_dest_texture, const Color &p_color, const Rect2i &p_region) {
  751. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use the raster version of the set_color shader with the clustered renderer.");
  752. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  753. ERR_FAIL_NULL(uniform_set_cache);
  754. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  755. ERR_FAIL_NULL(material_storage);
  756. memset(&copy_to_fb.push_constant, 0, sizeof(CopyToFbPushConstant));
  757. copy_to_fb.push_constant.set_color[0] = p_color.r;
  758. copy_to_fb.push_constant.set_color[1] = p_color.g;
  759. copy_to_fb.push_constant.set_color[2] = p_color.b;
  760. copy_to_fb.push_constant.set_color[3] = p_color.a;
  761. RID dest_framebuffer = FramebufferCacheRD::get_singleton()->get_cache(p_dest_texture);
  762. CopyToFBMode mode = COPY_TO_FB_SET_COLOR;
  763. RID shader = copy_to_fb.shader.version_get_shader(copy_to_fb.shader_version, mode);
  764. ERR_FAIL_COND(shader.is_null());
  765. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD, Vector<Color>(), 1.0, 0, p_region);
  766. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, copy_to_fb.pipelines[mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(dest_framebuffer)));
  767. RD::get_singleton()->draw_list_bind_index_array(draw_list, material_storage->get_quad_index_array());
  768. RD::get_singleton()->draw_list_set_push_constant(draw_list, &copy_to_fb.push_constant, sizeof(CopyToFbPushConstant));
  769. RD::get_singleton()->draw_list_draw(draw_list, true);
  770. RD::get_singleton()->draw_list_end();
  771. }
  772. void CopyEffects::copy_cubemap_to_dp(RID p_source_rd_texture, RID p_dst_framebuffer, const Rect2 &p_rect, const Vector2 &p_dst_size, float p_z_near, float p_z_far, bool p_dp_flip, BitField<RD::BarrierMask> p_post_barrier) {
  773. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  774. ERR_FAIL_NULL(uniform_set_cache);
  775. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  776. ERR_FAIL_NULL(material_storage);
  777. CopyToDPPushConstant push_constant;
  778. push_constant.screen_rect[0] = p_rect.position.x;
  779. push_constant.screen_rect[1] = p_rect.position.y;
  780. push_constant.screen_rect[2] = p_rect.size.width;
  781. push_constant.screen_rect[3] = p_rect.size.height;
  782. push_constant.z_far = p_z_far;
  783. push_constant.z_near = p_z_near;
  784. push_constant.texel_size[0] = 1.0f / p_dst_size.x;
  785. push_constant.texel_size[1] = 1.0f / p_dst_size.y;
  786. push_constant.texel_size[0] *= p_dp_flip ? -1.0f : 1.0f; // Encode dp flip as x size sign
  787. // setup our uniforms
  788. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  789. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  790. RID shader = cube_to_dp.shader.version_get_shader(cube_to_dp.shader_version, 0);
  791. ERR_FAIL_COND(shader.is_null());
  792. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(p_dst_framebuffer, RD::INITIAL_ACTION_DROP, RD::FINAL_ACTION_DISCARD, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ);
  793. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, cube_to_dp.pipeline.get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dst_framebuffer)));
  794. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  795. RD::get_singleton()->draw_list_bind_index_array(draw_list, material_storage->get_quad_index_array());
  796. RD::get_singleton()->draw_list_set_push_constant(draw_list, &push_constant, sizeof(CopyToDPPushConstant));
  797. RD::get_singleton()->draw_list_draw(draw_list, true);
  798. RD::get_singleton()->draw_list_end(p_post_barrier);
  799. }
  800. void CopyEffects::cubemap_downsample(RID p_source_cubemap, RID p_dest_cubemap, const Size2i &p_size) {
  801. ERR_FAIL_COND_MSG(prefer_raster_effects, "Can't use compute based cubemap downsample with the mobile renderer.");
  802. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  803. ERR_FAIL_NULL(uniform_set_cache);
  804. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  805. ERR_FAIL_NULL(material_storage);
  806. cubemap_downsampler.push_constant.face_size = p_size.x;
  807. cubemap_downsampler.push_constant.face_id = 0; // we render all 6 sides to each layer in one call
  808. // setup our uniforms
  809. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  810. RD::Uniform u_source_cubemap(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_cubemap }));
  811. RD::Uniform u_dest_cubemap(RD::UNIFORM_TYPE_IMAGE, 0, Vector<RID>({ p_dest_cubemap }));
  812. RID shader = cubemap_downsampler.compute_shader.version_get_shader(cubemap_downsampler.shader_version, 0);
  813. ERR_FAIL_COND(shader.is_null());
  814. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  815. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, cubemap_downsampler.compute_pipeline);
  816. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_cubemap), 0);
  817. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 1, u_dest_cubemap), 1);
  818. int x_groups = (p_size.x - 1) / 8 + 1;
  819. int y_groups = (p_size.y - 1) / 8 + 1;
  820. RD::get_singleton()->compute_list_set_push_constant(compute_list, &cubemap_downsampler.push_constant, sizeof(CubemapDownsamplerPushConstant));
  821. RD::get_singleton()->compute_list_dispatch(compute_list, x_groups, y_groups, 6); // one z_group for each face
  822. RD::get_singleton()->compute_list_end();
  823. }
  824. void CopyEffects::cubemap_downsample_raster(RID p_source_cubemap, RID p_dest_framebuffer, uint32_t p_face_id, const Size2i &p_size) {
  825. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use raster based cubemap downsample with the clustered renderer.");
  826. ERR_FAIL_COND_MSG(p_face_id >= 6, "Raster implementation of cubemap downsample must process one side at a time.");
  827. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  828. ERR_FAIL_NULL(uniform_set_cache);
  829. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  830. ERR_FAIL_NULL(material_storage);
  831. cubemap_downsampler.push_constant.face_size = p_size.x;
  832. cubemap_downsampler.push_constant.face_id = p_face_id;
  833. // setup our uniforms
  834. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  835. RD::Uniform u_source_cubemap(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_cubemap }));
  836. RID shader = cubemap_downsampler.raster_shader.version_get_shader(cubemap_downsampler.shader_version, 0);
  837. ERR_FAIL_COND(shader.is_null());
  838. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(p_dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  839. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, cubemap_downsampler.raster_pipeline.get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dest_framebuffer)));
  840. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_cubemap), 0);
  841. RD::get_singleton()->draw_list_set_push_constant(draw_list, &cubemap_downsampler.push_constant, sizeof(CubemapDownsamplerPushConstant));
  842. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  843. RD::get_singleton()->draw_list_end();
  844. }
  845. void CopyEffects::cubemap_filter(RID p_source_cubemap, Vector<RID> p_dest_cubemap, bool p_use_array) {
  846. ERR_FAIL_COND_MSG(prefer_raster_effects, "Can't use compute based cubemap filter with the mobile renderer.");
  847. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  848. ERR_FAIL_NULL(uniform_set_cache);
  849. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  850. ERR_FAIL_NULL(material_storage);
  851. Vector<RD::Uniform> uniforms;
  852. for (int i = 0; i < p_dest_cubemap.size(); i++) {
  853. RD::Uniform u;
  854. u.uniform_type = RD::UNIFORM_TYPE_IMAGE;
  855. u.binding = i;
  856. u.append_id(p_dest_cubemap[i]);
  857. uniforms.push_back(u);
  858. }
  859. if (RD::get_singleton()->uniform_set_is_valid(filter.image_uniform_set)) {
  860. RD::get_singleton()->free(filter.image_uniform_set);
  861. }
  862. filter.image_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, filter.compute_shader.version_get_shader(filter.shader_version, 0), 2);
  863. // setup our uniforms
  864. RID default_mipmap_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR_WITH_MIPMAPS, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  865. RD::Uniform u_source_cubemap(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_mipmap_sampler, p_source_cubemap }));
  866. int mode = p_use_array ? FILTER_MODE_HIGH_QUALITY_ARRAY : FILTER_MODE_HIGH_QUALITY;
  867. mode = filter.use_high_quality ? mode : mode + 1;
  868. RID shader = filter.compute_shader.version_get_shader(filter.shader_version, mode);
  869. ERR_FAIL_COND(shader.is_null());
  870. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  871. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, filter.compute_pipelines[mode]);
  872. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_cubemap), 0);
  873. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, filter.uniform_set, 1);
  874. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, filter.image_uniform_set, 2);
  875. int x_groups = p_use_array ? 1792 : 342; // (128 * 128 * 7) / 64 : (128*128 + 64*64 + 32*32 + 16*16 + 8*8 + 4*4 + 2*2) / 64
  876. RD::get_singleton()->compute_list_dispatch(compute_list, x_groups, 6, 1); // one y_group for each face
  877. RD::get_singleton()->compute_list_end();
  878. }
  879. void CopyEffects::cubemap_filter_raster(RID p_source_cubemap, RID p_dest_framebuffer, uint32_t p_face_id, uint32_t p_mip_level) {
  880. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use raster based cubemap filter with the clustered renderer.");
  881. ERR_FAIL_COND_MSG(p_face_id >= 6, "Raster implementation of cubemap filter must process one side at a time.");
  882. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  883. ERR_FAIL_NULL(uniform_set_cache);
  884. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  885. ERR_FAIL_NULL(material_storage);
  886. // TODO implement!
  887. CubemapFilterRasterPushConstant push_constant;
  888. push_constant.mip_level = p_mip_level;
  889. push_constant.face_id = p_face_id;
  890. // setup our uniforms
  891. RID default_mipmap_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR_WITH_MIPMAPS, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  892. RD::Uniform u_source_cubemap(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_mipmap_sampler, p_source_cubemap }));
  893. CubemapFilterMode mode = filter.use_high_quality ? FILTER_MODE_HIGH_QUALITY : FILTER_MODE_LOW_QUALITY;
  894. RID shader = filter.raster_shader.version_get_shader(filter.shader_version, mode);
  895. ERR_FAIL_COND(shader.is_null());
  896. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(p_dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  897. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, filter.raster_pipelines[mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dest_framebuffer)));
  898. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_cubemap), 0);
  899. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, filter.uniform_set, 1);
  900. RD::get_singleton()->draw_list_set_push_constant(draw_list, &push_constant, sizeof(CubemapFilterRasterPushConstant));
  901. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  902. RD::get_singleton()->draw_list_end();
  903. }
  904. void CopyEffects::cubemap_roughness(RID p_source_rd_texture, RID p_dest_texture, uint32_t p_face_id, uint32_t p_sample_count, float p_roughness, float p_size) {
  905. ERR_FAIL_COND_MSG(prefer_raster_effects, "Can't use compute based cubemap roughness with the mobile renderer.");
  906. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  907. ERR_FAIL_NULL(uniform_set_cache);
  908. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  909. ERR_FAIL_NULL(material_storage);
  910. memset(&roughness.push_constant, 0, sizeof(CubemapRoughnessPushConstant));
  911. roughness.push_constant.face_id = p_face_id > 9 ? 0 : p_face_id;
  912. // Remap to perceptual-roughness^2 to create more detail in lower mips and match the mapping of cubemap_filter.
  913. roughness.push_constant.roughness = p_roughness * p_roughness;
  914. roughness.push_constant.sample_count = p_sample_count;
  915. roughness.push_constant.use_direct_write = p_roughness == 0.0;
  916. roughness.push_constant.face_size = p_size;
  917. // setup our uniforms
  918. RID default_mipmap_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR_WITH_MIPMAPS, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  919. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_mipmap_sampler, p_source_rd_texture }));
  920. RD::Uniform u_dest_texture(RD::UNIFORM_TYPE_IMAGE, 0, Vector<RID>({ p_dest_texture }));
  921. RID shader = roughness.compute_shader.version_get_shader(roughness.shader_version, 0);
  922. ERR_FAIL_COND(shader.is_null());
  923. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  924. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, roughness.compute_pipeline);
  925. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  926. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, uniform_set_cache->get_cache(shader, 1, u_dest_texture), 1);
  927. RD::get_singleton()->compute_list_set_push_constant(compute_list, &roughness.push_constant, sizeof(CubemapRoughnessPushConstant));
  928. int x_groups = (p_size - 1) / 8 + 1;
  929. int y_groups = (p_size - 1) / 8 + 1;
  930. RD::get_singleton()->compute_list_dispatch(compute_list, x_groups, y_groups, p_face_id > 9 ? 6 : 1);
  931. RD::get_singleton()->compute_list_end();
  932. }
  933. void CopyEffects::cubemap_roughness_raster(RID p_source_rd_texture, RID p_dest_framebuffer, uint32_t p_face_id, uint32_t p_sample_count, float p_roughness, float p_size) {
  934. ERR_FAIL_COND_MSG(!prefer_raster_effects, "Can't use raster based cubemap roughness with the clustered renderer.");
  935. ERR_FAIL_COND_MSG(p_face_id >= 6, "Raster implementation of cubemap roughness must process one side at a time.");
  936. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  937. ERR_FAIL_NULL(uniform_set_cache);
  938. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  939. ERR_FAIL_NULL(material_storage);
  940. memset(&roughness.push_constant, 0, sizeof(CubemapRoughnessPushConstant));
  941. roughness.push_constant.face_id = p_face_id;
  942. roughness.push_constant.roughness = p_roughness * p_roughness; // Shader expects roughness, not perceptual roughness, so multiply before passing in.
  943. roughness.push_constant.sample_count = p_sample_count;
  944. roughness.push_constant.use_direct_write = p_roughness == 0.0;
  945. roughness.push_constant.face_size = p_size;
  946. // Setup our uniforms.
  947. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR_WITH_MIPMAPS, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  948. RD::Uniform u_source_rd_texture(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_source_rd_texture }));
  949. RID shader = roughness.raster_shader.version_get_shader(roughness.shader_version, 0);
  950. ERR_FAIL_COND(shader.is_null());
  951. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(p_dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_DISCARD);
  952. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, roughness.raster_pipeline.get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dest_framebuffer)));
  953. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_source_rd_texture), 0);
  954. RD::get_singleton()->draw_list_set_push_constant(draw_list, &roughness.push_constant, sizeof(CubemapRoughnessPushConstant));
  955. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  956. RD::get_singleton()->draw_list_end();
  957. }
  958. void CopyEffects::merge_specular(RID p_dest_framebuffer, RID p_specular, RID p_base, RID p_reflection, uint32_t p_view_count) {
  959. UniformSetCacheRD *uniform_set_cache = UniformSetCacheRD::get_singleton();
  960. ERR_FAIL_NULL(uniform_set_cache);
  961. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  962. ERR_FAIL_NULL(material_storage);
  963. RID default_sampler = material_storage->sampler_rd_get_default(RS::CANVAS_ITEM_TEXTURE_FILTER_LINEAR, RS::CANVAS_ITEM_TEXTURE_REPEAT_DISABLED);
  964. RD::get_singleton()->draw_command_begin_label("Merge specular");
  965. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(p_dest_framebuffer, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, RD::INITIAL_ACTION_KEEP, RD::FINAL_ACTION_READ, Vector<Color>());
  966. int mode;
  967. if (p_reflection.is_valid()) {
  968. if (p_base.is_valid()) {
  969. mode = SPECULAR_MERGE_SSR;
  970. } else {
  971. mode = SPECULAR_MERGE_ADDITIVE_SSR;
  972. }
  973. } else {
  974. if (p_base.is_valid()) {
  975. mode = SPECULAR_MERGE_ADD;
  976. } else {
  977. mode = SPECULAR_MERGE_ADDITIVE_ADD;
  978. }
  979. }
  980. if (p_view_count > 1) {
  981. mode += SPECULAR_MERGE_ADD_MULTIVIEW;
  982. }
  983. RID shader = specular_merge.shader.version_get_shader(specular_merge.shader_version, mode);
  984. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, specular_merge.pipelines[mode].get_render_pipeline(RD::INVALID_ID, RD::get_singleton()->framebuffer_get_format(p_dest_framebuffer)));
  985. if (p_base.is_valid()) {
  986. RD::Uniform u_base(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_base }));
  987. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 2, u_base), 2);
  988. }
  989. RD::Uniform u_specular(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_specular }));
  990. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 0, u_specular), 0);
  991. if (p_reflection.is_valid()) {
  992. RD::Uniform u_reflection(RD::UNIFORM_TYPE_SAMPLER_WITH_TEXTURE, 0, Vector<RID>({ default_sampler, p_reflection }));
  993. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, uniform_set_cache->get_cache(shader, 1, u_reflection), 1);
  994. }
  995. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  996. RD::get_singleton()->draw_list_end();
  997. RD::get_singleton()->draw_command_end_label();
  998. }