sky.cpp 71 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717
  1. /**************************************************************************/
  2. /* sky.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 "sky.h"
  31. #include "core/config/project_settings.h"
  32. #include "core/math/math_defs.h"
  33. #include "servers/rendering/renderer_rd/effects/copy_effects.h"
  34. #include "servers/rendering/renderer_rd/framebuffer_cache_rd.h"
  35. #include "servers/rendering/renderer_rd/renderer_compositor_rd.h"
  36. #include "servers/rendering/renderer_rd/renderer_scene_render_rd.h"
  37. #include "servers/rendering/renderer_rd/storage_rd/material_storage.h"
  38. #include "servers/rendering/renderer_rd/storage_rd/render_scene_buffers_rd.h"
  39. #include "servers/rendering/renderer_rd/storage_rd/texture_storage.h"
  40. #include "servers/rendering/renderer_rd/uniform_set_cache_rd.h"
  41. #include "servers/rendering/rendering_server_default.h"
  42. #include "servers/rendering/rendering_server_globals.h"
  43. using namespace RendererRD;
  44. #define RB_SCOPE_SKY SNAME("sky_buffers")
  45. #define RB_HALF_TEXTURE SNAME("half_texture")
  46. #define RB_QUARTER_TEXTURE SNAME("quarter_texture")
  47. ////////////////////////////////////////////////////////////////////////////////
  48. // SKY SHADER
  49. void SkyRD::SkyShaderData::set_code(const String &p_code) {
  50. //compile
  51. code = p_code;
  52. valid = false;
  53. ubo_size = 0;
  54. uniforms.clear();
  55. if (code.is_empty()) {
  56. return; //just invalid, but no error
  57. }
  58. ShaderCompiler::GeneratedCode gen_code;
  59. ShaderCompiler::IdentifierActions actions;
  60. actions.entry_point_stages["sky"] = ShaderCompiler::STAGE_FRAGMENT;
  61. uses_time = false;
  62. uses_half_res = false;
  63. uses_quarter_res = false;
  64. uses_position = false;
  65. uses_light = false;
  66. actions.render_mode_flags["use_half_res_pass"] = &uses_half_res;
  67. actions.render_mode_flags["use_quarter_res_pass"] = &uses_quarter_res;
  68. actions.usage_flag_pointers["TIME"] = &uses_time;
  69. actions.usage_flag_pointers["POSITION"] = &uses_position;
  70. actions.usage_flag_pointers["LIGHT0_ENABLED"] = &uses_light;
  71. actions.usage_flag_pointers["LIGHT0_ENERGY"] = &uses_light;
  72. actions.usage_flag_pointers["LIGHT0_DIRECTION"] = &uses_light;
  73. actions.usage_flag_pointers["LIGHT0_COLOR"] = &uses_light;
  74. actions.usage_flag_pointers["LIGHT0_SIZE"] = &uses_light;
  75. actions.usage_flag_pointers["LIGHT1_ENABLED"] = &uses_light;
  76. actions.usage_flag_pointers["LIGHT1_ENERGY"] = &uses_light;
  77. actions.usage_flag_pointers["LIGHT1_DIRECTION"] = &uses_light;
  78. actions.usage_flag_pointers["LIGHT1_COLOR"] = &uses_light;
  79. actions.usage_flag_pointers["LIGHT1_SIZE"] = &uses_light;
  80. actions.usage_flag_pointers["LIGHT2_ENABLED"] = &uses_light;
  81. actions.usage_flag_pointers["LIGHT2_ENERGY"] = &uses_light;
  82. actions.usage_flag_pointers["LIGHT2_DIRECTION"] = &uses_light;
  83. actions.usage_flag_pointers["LIGHT2_COLOR"] = &uses_light;
  84. actions.usage_flag_pointers["LIGHT2_SIZE"] = &uses_light;
  85. actions.usage_flag_pointers["LIGHT3_ENABLED"] = &uses_light;
  86. actions.usage_flag_pointers["LIGHT3_ENERGY"] = &uses_light;
  87. actions.usage_flag_pointers["LIGHT3_DIRECTION"] = &uses_light;
  88. actions.usage_flag_pointers["LIGHT3_COLOR"] = &uses_light;
  89. actions.usage_flag_pointers["LIGHT3_SIZE"] = &uses_light;
  90. actions.uniforms = &uniforms;
  91. // !BAS! Contemplate making `SkyShader sky` accessible from this struct or even part of this struct.
  92. RendererSceneRenderRD *scene_singleton = static_cast<RendererSceneRenderRD *>(RendererSceneRenderRD::singleton);
  93. Error err = scene_singleton->sky.sky_shader.compiler.compile(RS::SHADER_SKY, code, &actions, path, gen_code);
  94. ERR_FAIL_COND_MSG(err != OK, "Shader compilation failed.");
  95. if (version.is_null()) {
  96. version = scene_singleton->sky.sky_shader.shader.version_create();
  97. }
  98. #if 0
  99. print_line("**compiling shader:");
  100. print_line("**defines:\n");
  101. for (int i = 0; i < gen_code.defines.size(); i++) {
  102. print_line(gen_code.defines[i]);
  103. }
  104. HashMap<String, String>::Iterator el = gen_code.code.begin();
  105. while (el) {
  106. print_line("\n**code " + el->key + ":\n" + el->value);
  107. ++el;
  108. }
  109. print_line("\n**uniforms:\n" + gen_code.uniforms);
  110. print_line("\n**vertex_globals:\n" + gen_code.stage_globals[ShaderCompiler::STAGE_VERTEX]);
  111. print_line("\n**fragment_globals:\n" + gen_code.stage_globals[ShaderCompiler::STAGE_FRAGMENT]);
  112. #endif
  113. scene_singleton->sky.sky_shader.shader.version_set_code(version, gen_code.code, gen_code.uniforms, gen_code.stage_globals[ShaderCompiler::STAGE_VERTEX], gen_code.stage_globals[ShaderCompiler::STAGE_FRAGMENT], gen_code.defines);
  114. ERR_FAIL_COND(!scene_singleton->sky.sky_shader.shader.version_is_valid(version));
  115. ubo_size = gen_code.uniform_total_size;
  116. ubo_offsets = gen_code.uniform_offsets;
  117. texture_uniforms = gen_code.texture_uniforms;
  118. //update pipelines
  119. for (int i = 0; i < SKY_VERSION_MAX; i++) {
  120. RD::PipelineDepthStencilState depth_stencil_state;
  121. depth_stencil_state.enable_depth_test = true;
  122. depth_stencil_state.depth_compare_operator = RD::COMPARE_OP_GREATER_OR_EQUAL;
  123. if (scene_singleton->sky.sky_shader.shader.is_variant_enabled(i)) {
  124. RID shader_variant = scene_singleton->sky.sky_shader.shader.version_get_shader(version, i);
  125. pipelines[i].setup(shader_variant, RD::RENDER_PRIMITIVE_TRIANGLES, RD::PipelineRasterizationState(), RD::PipelineMultisampleState(), depth_stencil_state, RD::PipelineColorBlendState::create_disabled(), 0);
  126. } else {
  127. pipelines[i].clear();
  128. }
  129. }
  130. valid = true;
  131. }
  132. bool SkyRD::SkyShaderData::is_animated() const {
  133. return false;
  134. }
  135. bool SkyRD::SkyShaderData::casts_shadows() const {
  136. return false;
  137. }
  138. RS::ShaderNativeSourceCode SkyRD::SkyShaderData::get_native_source_code() const {
  139. RendererSceneRenderRD *scene_singleton = static_cast<RendererSceneRenderRD *>(RendererSceneRenderRD::singleton);
  140. return scene_singleton->sky.sky_shader.shader.version_get_native_source_code(version);
  141. }
  142. SkyRD::SkyShaderData::~SkyShaderData() {
  143. RendererSceneRenderRD *scene_singleton = static_cast<RendererSceneRenderRD *>(RendererSceneRenderRD::singleton);
  144. ERR_FAIL_NULL(scene_singleton);
  145. //pipeline variants will clear themselves if shader is gone
  146. if (version.is_valid()) {
  147. scene_singleton->sky.sky_shader.shader.version_free(version);
  148. }
  149. }
  150. ////////////////////////////////////////////////////////////////////////////////
  151. // Sky material
  152. bool SkyRD::SkyMaterialData::update_parameters(const HashMap<StringName, Variant> &p_parameters, bool p_uniform_dirty, bool p_textures_dirty) {
  153. RendererSceneRenderRD *scene_singleton = static_cast<RendererSceneRenderRD *>(RendererSceneRenderRD::singleton);
  154. uniform_set_updated = true;
  155. return update_parameters_uniform_set(p_parameters, p_uniform_dirty, p_textures_dirty, shader_data->uniforms, shader_data->ubo_offsets.ptr(), shader_data->texture_uniforms, shader_data->default_texture_params, shader_data->ubo_size, uniform_set, scene_singleton->sky.sky_shader.shader.version_get_shader(shader_data->version, 0), SKY_SET_MATERIAL, true, true);
  156. }
  157. SkyRD::SkyMaterialData::~SkyMaterialData() {
  158. free_parameters_uniform_set(uniform_set);
  159. }
  160. ////////////////////////////////////////////////////////////////////////////////
  161. // Render sky
  162. static _FORCE_INLINE_ void store_transform_3x3(const Basis &p_basis, float *p_array) {
  163. p_array[0] = p_basis.rows[0][0];
  164. p_array[1] = p_basis.rows[1][0];
  165. p_array[2] = p_basis.rows[2][0];
  166. p_array[3] = 0;
  167. p_array[4] = p_basis.rows[0][1];
  168. p_array[5] = p_basis.rows[1][1];
  169. p_array[6] = p_basis.rows[2][1];
  170. p_array[7] = 0;
  171. p_array[8] = p_basis.rows[0][2];
  172. p_array[9] = p_basis.rows[1][2];
  173. p_array[10] = p_basis.rows[2][2];
  174. p_array[11] = 0;
  175. }
  176. void SkyRD::_render_sky(RD::DrawListID p_list, float p_time, RID p_fb, PipelineCacheRD *p_pipeline, RID p_uniform_set, RID p_texture_set, const Projection &p_projection, const Basis &p_orientation, const Vector3 &p_position, float p_luminance_multiplier, float p_brightness_multiplier) {
  177. SkyPushConstant sky_push_constant;
  178. memset(&sky_push_constant, 0, sizeof(SkyPushConstant));
  179. // We only need key components of our projection matrix
  180. sky_push_constant.projection[0] = p_projection.columns[2][0];
  181. sky_push_constant.projection[1] = p_projection.columns[0][0];
  182. sky_push_constant.projection[2] = p_projection.columns[2][1];
  183. sky_push_constant.projection[3] = p_projection.columns[1][1];
  184. sky_push_constant.position[0] = p_position.x;
  185. sky_push_constant.position[1] = p_position.y;
  186. sky_push_constant.position[2] = p_position.z;
  187. sky_push_constant.time = p_time;
  188. sky_push_constant.luminance_multiplier = p_luminance_multiplier;
  189. sky_push_constant.brightness_multiplier = p_brightness_multiplier;
  190. store_transform_3x3(p_orientation, sky_push_constant.orientation);
  191. RenderingDevice::FramebufferFormatID fb_format = RD::get_singleton()->framebuffer_get_format(p_fb);
  192. RD::DrawListID draw_list = p_list;
  193. RD::get_singleton()->draw_list_bind_render_pipeline(draw_list, p_pipeline->get_render_pipeline(RD::INVALID_ID, fb_format, false, RD::get_singleton()->draw_list_get_current_pass()));
  194. // Update uniform sets.
  195. {
  196. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, sky_scene_state.uniform_set, SKY_SET_UNIFORMS);
  197. if (p_uniform_set.is_valid() && RD::get_singleton()->uniform_set_is_valid(p_uniform_set)) { // Material may not have a uniform set.
  198. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, p_uniform_set, SKY_SET_MATERIAL);
  199. }
  200. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, p_texture_set, SKY_SET_TEXTURES);
  201. // Fog uniform set can be invalidated before drawing, so validate at draw time
  202. if (sky_scene_state.fog_uniform_set.is_valid() && RD::get_singleton()->uniform_set_is_valid(sky_scene_state.fog_uniform_set)) {
  203. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, sky_scene_state.fog_uniform_set, SKY_SET_FOG);
  204. } else {
  205. RD::get_singleton()->draw_list_bind_uniform_set(draw_list, sky_scene_state.default_fog_uniform_set, SKY_SET_FOG);
  206. }
  207. }
  208. RD::get_singleton()->draw_list_set_push_constant(draw_list, &sky_push_constant, sizeof(SkyPushConstant));
  209. RD::get_singleton()->draw_list_draw(draw_list, false, 1u, 3u);
  210. }
  211. ////////////////////////////////////////////////////////////////////////////////
  212. // ReflectionData
  213. void SkyRD::ReflectionData::clear_reflection_data() {
  214. layers.clear();
  215. radiance_base_cubemap = RID();
  216. if (downsampled_radiance_cubemap.is_valid()) {
  217. RD::get_singleton()->free(downsampled_radiance_cubemap);
  218. }
  219. downsampled_radiance_cubemap = RID();
  220. downsampled_layer.mipmaps.clear();
  221. coefficient_buffer = RID();
  222. }
  223. void SkyRD::ReflectionData::update_reflection_data(int p_size, int p_mipmaps, bool p_use_array, RID p_base_cube, int p_base_layer, bool p_low_quality, int p_roughness_layers, RD::DataFormat p_texture_format) {
  224. //recreate radiance and all data
  225. int mipmaps = p_mipmaps;
  226. uint32_t w = p_size, h = p_size;
  227. bool render_buffers_can_be_storage = RendererSceneRenderRD::get_singleton()->_render_buffers_can_be_storage();
  228. if (p_use_array) {
  229. int num_layers = p_low_quality ? 8 : p_roughness_layers;
  230. for (int i = 0; i < num_layers; i++) {
  231. ReflectionData::Layer layer;
  232. uint32_t mmw = w;
  233. uint32_t mmh = h;
  234. layer.mipmaps.resize(mipmaps);
  235. layer.views.resize(mipmaps);
  236. for (int j = 0; j < mipmaps; j++) {
  237. ReflectionData::Layer::Mipmap &mm = layer.mipmaps.write[j];
  238. mm.size.width = mmw;
  239. mm.size.height = mmh;
  240. for (int k = 0; k < 6; k++) {
  241. mm.views[k] = RD::get_singleton()->texture_create_shared_from_slice(RD::TextureView(), p_base_cube, p_base_layer + i * 6 + k, j);
  242. Vector<RID> fbtex;
  243. fbtex.push_back(mm.views[k]);
  244. mm.framebuffers[k] = RD::get_singleton()->framebuffer_create(fbtex);
  245. }
  246. layer.views.write[j] = RD::get_singleton()->texture_create_shared_from_slice(RD::TextureView(), p_base_cube, p_base_layer + i * 6, j, 1, RD::TEXTURE_SLICE_CUBEMAP);
  247. mmw = MAX(1u, mmw >> 1);
  248. mmh = MAX(1u, mmh >> 1);
  249. }
  250. layers.push_back(layer);
  251. }
  252. } else {
  253. mipmaps = p_low_quality ? 8 : mipmaps;
  254. //regular cubemap, lower quality (aliasing, less memory)
  255. ReflectionData::Layer layer;
  256. uint32_t mmw = w;
  257. uint32_t mmh = h;
  258. layer.mipmaps.resize(mipmaps);
  259. layer.views.resize(mipmaps);
  260. for (int j = 0; j < mipmaps; j++) {
  261. ReflectionData::Layer::Mipmap &mm = layer.mipmaps.write[j];
  262. mm.size.width = mmw;
  263. mm.size.height = mmh;
  264. for (int k = 0; k < 6; k++) {
  265. mm.views[k] = RD::get_singleton()->texture_create_shared_from_slice(RD::TextureView(), p_base_cube, p_base_layer + k, j);
  266. Vector<RID> fbtex;
  267. fbtex.push_back(mm.views[k]);
  268. mm.framebuffers[k] = RD::get_singleton()->framebuffer_create(fbtex);
  269. }
  270. layer.views.write[j] = RD::get_singleton()->texture_create_shared_from_slice(RD::TextureView(), p_base_cube, p_base_layer, j, 1, RD::TEXTURE_SLICE_CUBEMAP);
  271. mmw = MAX(1u, mmw >> 1);
  272. mmh = MAX(1u, mmh >> 1);
  273. }
  274. layers.push_back(layer);
  275. }
  276. radiance_base_cubemap = RD::get_singleton()->texture_create_shared_from_slice(RD::TextureView(), p_base_cube, p_base_layer, 0, 1, RD::TEXTURE_SLICE_CUBEMAP);
  277. RD::get_singleton()->set_resource_name(radiance_base_cubemap, "radiance base cubemap");
  278. RD::TextureFormat tf;
  279. tf.format = p_texture_format;
  280. tf.width = p_low_quality ? 64 : p_size >> 1; // Always 64x64 when using REALTIME.
  281. tf.height = p_low_quality ? 64 : p_size >> 1;
  282. tf.texture_type = RD::TEXTURE_TYPE_CUBE;
  283. tf.array_layers = 6;
  284. tf.mipmaps = p_low_quality ? 7 : mipmaps - 1;
  285. tf.usage_bits = RD::TEXTURE_USAGE_SAMPLING_BIT | RD::TEXTURE_USAGE_COLOR_ATTACHMENT_BIT;
  286. if (render_buffers_can_be_storage) {
  287. tf.usage_bits |= RD::TEXTURE_USAGE_STORAGE_BIT;
  288. }
  289. downsampled_radiance_cubemap = RD::get_singleton()->texture_create(tf, RD::TextureView());
  290. RD::get_singleton()->set_resource_name(downsampled_radiance_cubemap, "downsampled radiance cubemap");
  291. {
  292. uint32_t mmw = tf.width;
  293. uint32_t mmh = tf.height;
  294. downsampled_layer.mipmaps.resize(tf.mipmaps);
  295. for (int j = 0; j < downsampled_layer.mipmaps.size(); j++) {
  296. ReflectionData::DownsampleLayer::Mipmap &mm = downsampled_layer.mipmaps.write[j];
  297. mm.size.width = mmw;
  298. mm.size.height = mmh;
  299. mm.view = RD::get_singleton()->texture_create_shared_from_slice(RD::TextureView(), downsampled_radiance_cubemap, 0, j, 1, RD::TEXTURE_SLICE_CUBEMAP);
  300. RD::get_singleton()->set_resource_name(mm.view, "Downsampled Radiance Cubemap Mip " + itos(j) + " ");
  301. if (!render_buffers_can_be_storage) {
  302. // we need a framebuffer for each side of our cubemap
  303. for (int k = 0; k < 6; k++) {
  304. mm.views[k] = RD::get_singleton()->texture_create_shared_from_slice(RD::TextureView(), downsampled_radiance_cubemap, k, j);
  305. RD::get_singleton()->set_resource_name(mm.view, "Downsampled Radiance Cubemap Mip: " + itos(j) + " Face: " + itos(k) + " ");
  306. Vector<RID> fbtex;
  307. fbtex.push_back(mm.views[k]);
  308. mm.framebuffers[k] = RD::get_singleton()->framebuffer_create(fbtex);
  309. }
  310. }
  311. mmw = MAX(1u, mmw >> 1);
  312. mmh = MAX(1u, mmh >> 1);
  313. }
  314. }
  315. }
  316. void SkyRD::ReflectionData::create_reflection_fast_filter(bool p_use_arrays) {
  317. RendererRD::CopyEffects *copy_effects = RendererRD::CopyEffects::get_singleton();
  318. ERR_FAIL_NULL_MSG(copy_effects, "Effects haven't been initialized");
  319. bool prefer_raster_effects = copy_effects->get_prefer_raster_effects();
  320. if (prefer_raster_effects) {
  321. RD::get_singleton()->draw_command_begin_label("Downsample radiance map");
  322. for (int k = 0; k < 6; k++) {
  323. copy_effects->cubemap_downsample_raster(radiance_base_cubemap, downsampled_layer.mipmaps[0].framebuffers[k], k, downsampled_layer.mipmaps[0].size);
  324. }
  325. for (int i = 1; i < downsampled_layer.mipmaps.size(); i++) {
  326. for (int k = 0; k < 6; k++) {
  327. copy_effects->cubemap_downsample_raster(downsampled_layer.mipmaps[i - 1].view, downsampled_layer.mipmaps[i].framebuffers[k], k, downsampled_layer.mipmaps[i].size);
  328. }
  329. }
  330. RD::get_singleton()->draw_command_end_label(); // Downsample Radiance
  331. if (p_use_arrays) {
  332. RD::get_singleton()->draw_command_begin_label("filter radiance map into array heads");
  333. for (int i = 0; i < layers.size(); i++) {
  334. for (int k = 0; k < 6; k++) {
  335. copy_effects->cubemap_filter_raster(downsampled_radiance_cubemap, layers[i].mipmaps[0].framebuffers[k], k, i);
  336. }
  337. }
  338. } else {
  339. RD::get_singleton()->draw_command_begin_label("filter radiance map into mipmaps directly");
  340. for (int j = 0; j < layers[0].mipmaps.size(); j++) {
  341. for (int k = 0; k < 6; k++) {
  342. copy_effects->cubemap_filter_raster(downsampled_radiance_cubemap, layers[0].mipmaps[j].framebuffers[k], k, j);
  343. }
  344. }
  345. }
  346. RD::get_singleton()->draw_command_end_label(); // Filter radiance
  347. } else {
  348. RD::get_singleton()->draw_command_begin_label("Downsample radiance map");
  349. copy_effects->cubemap_downsample(radiance_base_cubemap, downsampled_layer.mipmaps[0].view, downsampled_layer.mipmaps[0].size);
  350. for (int i = 1; i < downsampled_layer.mipmaps.size(); i++) {
  351. copy_effects->cubemap_downsample(downsampled_layer.mipmaps[i - 1].view, downsampled_layer.mipmaps[i].view, downsampled_layer.mipmaps[i].size);
  352. }
  353. RD::get_singleton()->draw_command_end_label(); // Downsample Radiance
  354. Vector<RID> views;
  355. if (p_use_arrays) {
  356. for (int i = 1; i < layers.size(); i++) {
  357. views.push_back(layers[i].views[0]);
  358. }
  359. } else {
  360. for (int i = 1; i < layers[0].views.size(); i++) {
  361. views.push_back(layers[0].views[i]);
  362. }
  363. }
  364. RD::get_singleton()->draw_command_begin_label("Fast filter radiance");
  365. copy_effects->cubemap_filter(downsampled_radiance_cubemap, views, p_use_arrays);
  366. RD::get_singleton()->draw_command_end_label(); // Filter radiance
  367. }
  368. }
  369. void SkyRD::ReflectionData::create_reflection_importance_sample(bool p_use_arrays, int p_cube_side, int p_base_layer, uint32_t p_sky_ggx_samples_quality) {
  370. RendererRD::CopyEffects *copy_effects = RendererRD::CopyEffects::get_singleton();
  371. ERR_FAIL_NULL_MSG(copy_effects, "Effects haven't been initialized");
  372. bool prefer_raster_effects = copy_effects->get_prefer_raster_effects();
  373. if (prefer_raster_effects) {
  374. if (p_base_layer == 1) {
  375. RD::get_singleton()->draw_command_begin_label("Downsample radiance map");
  376. for (int k = 0; k < 6; k++) {
  377. copy_effects->cubemap_downsample_raster(radiance_base_cubemap, downsampled_layer.mipmaps[0].framebuffers[k], k, downsampled_layer.mipmaps[0].size);
  378. }
  379. for (int i = 1; i < downsampled_layer.mipmaps.size(); i++) {
  380. for (int k = 0; k < 6; k++) {
  381. copy_effects->cubemap_downsample_raster(downsampled_layer.mipmaps[i - 1].view, downsampled_layer.mipmaps[i].framebuffers[k], k, downsampled_layer.mipmaps[i].size);
  382. }
  383. }
  384. RD::get_singleton()->draw_command_end_label(); // Downsample Radiance
  385. }
  386. RD::get_singleton()->draw_command_begin_label("High Quality filter radiance");
  387. if (p_use_arrays) {
  388. for (int k = 0; k < 6; k++) {
  389. copy_effects->cubemap_roughness_raster(
  390. downsampled_radiance_cubemap,
  391. layers[p_base_layer].mipmaps[0].framebuffers[k],
  392. k,
  393. p_sky_ggx_samples_quality,
  394. float(p_base_layer) / (layers.size() - 1.0),
  395. layers[p_base_layer].mipmaps[0].size.x);
  396. }
  397. } else {
  398. for (int k = 0; k < 6; k++) {
  399. copy_effects->cubemap_roughness_raster(
  400. downsampled_radiance_cubemap,
  401. layers[0].mipmaps[p_base_layer].framebuffers[k],
  402. k,
  403. p_sky_ggx_samples_quality,
  404. float(p_base_layer) / (layers[0].mipmaps.size() - 1.0),
  405. layers[0].mipmaps[p_base_layer].size.x);
  406. }
  407. }
  408. } else {
  409. if (p_base_layer == 1) {
  410. RD::get_singleton()->draw_command_begin_label("Downsample radiance map");
  411. copy_effects->cubemap_downsample(radiance_base_cubemap, downsampled_layer.mipmaps[0].view, downsampled_layer.mipmaps[0].size);
  412. for (int i = 1; i < downsampled_layer.mipmaps.size(); i++) {
  413. copy_effects->cubemap_downsample(downsampled_layer.mipmaps[i - 1].view, downsampled_layer.mipmaps[i].view, downsampled_layer.mipmaps[i].size);
  414. }
  415. RD::get_singleton()->draw_command_end_label(); // Downsample Radiance
  416. }
  417. RD::get_singleton()->draw_command_begin_label("High Quality filter radiance");
  418. if (p_use_arrays) {
  419. copy_effects->cubemap_roughness(downsampled_radiance_cubemap, layers[p_base_layer].views[0], p_cube_side, p_sky_ggx_samples_quality, float(p_base_layer) / (layers.size() - 1.0), layers[p_base_layer].mipmaps[0].size.x);
  420. } else {
  421. copy_effects->cubemap_roughness(
  422. downsampled_radiance_cubemap,
  423. layers[0].views[p_base_layer],
  424. p_cube_side,
  425. p_sky_ggx_samples_quality,
  426. float(p_base_layer) / (layers[0].mipmaps.size() - 1.0),
  427. layers[0].mipmaps[p_base_layer].size.x);
  428. }
  429. }
  430. RD::get_singleton()->draw_command_end_label(); // Filter radiance
  431. }
  432. void SkyRD::ReflectionData::update_reflection_mipmaps(int p_start, int p_end) {
  433. RendererRD::CopyEffects *copy_effects = RendererRD::CopyEffects::get_singleton();
  434. ERR_FAIL_NULL_MSG(copy_effects, "Effects haven't been initialized");
  435. bool prefer_raster_effects = copy_effects->get_prefer_raster_effects();
  436. RD::get_singleton()->draw_command_begin_label("Update Radiance Cubemap Array Mipmaps");
  437. for (int i = p_start; i < p_end; i++) {
  438. for (int j = 0; j < layers[i].views.size() - 1; j++) {
  439. RID view = layers[i].views[j];
  440. Size2i size = layers[i].mipmaps[j + 1].size;
  441. if (prefer_raster_effects) {
  442. for (int k = 0; k < 6; k++) {
  443. RID framebuffer = layers[i].mipmaps[j + 1].framebuffers[k];
  444. copy_effects->cubemap_downsample_raster(view, framebuffer, k, size);
  445. }
  446. } else {
  447. RID texture = layers[i].views[j + 1];
  448. copy_effects->cubemap_downsample(view, texture, size);
  449. }
  450. }
  451. }
  452. RD::get_singleton()->draw_command_end_label();
  453. }
  454. ////////////////////////////////////////////////////////////////////////////////
  455. // SkyRD::Sky
  456. void SkyRD::Sky::free() {
  457. if (radiance.is_valid()) {
  458. RD::get_singleton()->free(radiance);
  459. radiance = RID();
  460. }
  461. reflection.clear_reflection_data();
  462. if (uniform_buffer.is_valid()) {
  463. RD::get_singleton()->free(uniform_buffer);
  464. uniform_buffer = RID();
  465. }
  466. if (material.is_valid()) {
  467. material = RID();
  468. }
  469. }
  470. RID SkyRD::Sky::get_textures(SkyTextureSetVersion p_version, RID p_default_shader_rd, Ref<RenderSceneBuffersRD> p_render_buffers) {
  471. RendererRD::TextureStorage *texture_storage = RendererRD::TextureStorage::get_singleton();
  472. thread_local LocalVector<RD::Uniform> uniforms;
  473. uniforms.clear();
  474. {
  475. RD::Uniform u;
  476. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  477. u.binding = 0;
  478. if (radiance.is_valid() && p_version <= SKY_TEXTURE_SET_QUARTER_RES) {
  479. u.append_id(radiance);
  480. } else {
  481. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_CUBEMAP_BLACK));
  482. }
  483. uniforms.push_back(u);
  484. }
  485. {
  486. RD::Uniform u;
  487. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  488. u.binding = 1; // half res
  489. if (p_version >= SKY_TEXTURE_SET_CUBEMAP) {
  490. if (reflection.layers.size() && reflection.layers[0].views.size() >= 2 && reflection.layers[0].views[1].is_valid() && p_version != SKY_TEXTURE_SET_CUBEMAP_HALF_RES) {
  491. u.append_id(reflection.layers[0].views[1]);
  492. } else {
  493. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_CUBEMAP_BLACK));
  494. }
  495. } else {
  496. RID half_texture = p_render_buffers->has_texture(RB_SCOPE_SKY, RB_HALF_TEXTURE) ? p_render_buffers->get_texture(RB_SCOPE_SKY, RB_HALF_TEXTURE) : RID();
  497. if (half_texture.is_valid() && p_version != SKY_TEXTURE_SET_HALF_RES) {
  498. u.append_id(half_texture);
  499. } else {
  500. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_WHITE));
  501. }
  502. }
  503. uniforms.push_back(u);
  504. }
  505. {
  506. RD::Uniform u;
  507. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  508. u.binding = 2; // quarter res
  509. if (p_version >= SKY_TEXTURE_SET_CUBEMAP) {
  510. if (reflection.layers.size() && reflection.layers[0].views.size() >= 3 && reflection.layers[0].views[2].is_valid() && p_version != SKY_TEXTURE_SET_CUBEMAP_QUARTER_RES) {
  511. u.append_id(reflection.layers[0].views[2]);
  512. } else {
  513. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_CUBEMAP_BLACK));
  514. }
  515. } else {
  516. RID quarter_texture = p_render_buffers->has_texture(RB_SCOPE_SKY, RB_QUARTER_TEXTURE) ? p_render_buffers->get_texture(RB_SCOPE_SKY, RB_QUARTER_TEXTURE) : RID();
  517. if (quarter_texture.is_valid() && p_version != SKY_TEXTURE_SET_QUARTER_RES) {
  518. u.append_id(quarter_texture);
  519. } else {
  520. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_WHITE));
  521. }
  522. }
  523. uniforms.push_back(u);
  524. }
  525. return UniformSetCacheRD::get_singleton()->get_cache_vec(p_default_shader_rd, SKY_SET_TEXTURES, uniforms);
  526. }
  527. bool SkyRD::Sky::set_radiance_size(int p_radiance_size) {
  528. ERR_FAIL_COND_V(p_radiance_size < 32 || p_radiance_size > 2048, false);
  529. if (radiance_size == p_radiance_size) {
  530. return false;
  531. }
  532. radiance_size = p_radiance_size;
  533. if (mode == RS::SKY_MODE_REALTIME && radiance_size != 256) {
  534. WARN_PRINT("Realtime Skies can only use a radiance size of 256. Radiance size will be set to 256 internally.");
  535. radiance_size = 256;
  536. }
  537. if (radiance.is_valid()) {
  538. RD::get_singleton()->free(radiance);
  539. radiance = RID();
  540. }
  541. reflection.clear_reflection_data();
  542. return true;
  543. }
  544. bool SkyRD::Sky::set_mode(RS::SkyMode p_mode) {
  545. if (mode == p_mode) {
  546. return false;
  547. }
  548. mode = p_mode;
  549. if (mode == RS::SKY_MODE_REALTIME && radiance_size != 256) {
  550. WARN_PRINT("Realtime Skies can only use a radiance size of 256. Radiance size will be set to 256 internally.");
  551. set_radiance_size(256);
  552. }
  553. if (radiance.is_valid()) {
  554. RD::get_singleton()->free(radiance);
  555. radiance = RID();
  556. }
  557. reflection.clear_reflection_data();
  558. return true;
  559. }
  560. bool SkyRD::Sky::set_material(RID p_material) {
  561. if (material == p_material) {
  562. return false;
  563. }
  564. material = p_material;
  565. return true;
  566. }
  567. Ref<Image> SkyRD::Sky::bake_panorama(float p_energy, int p_roughness_layers, const Size2i &p_size) {
  568. if (radiance.is_valid()) {
  569. RendererRD::CopyEffects *copy_effects = RendererRD::CopyEffects::get_singleton();
  570. RD::TextureFormat tf;
  571. tf.format = RD::DATA_FORMAT_R32G32B32A32_SFLOAT; // Could be RGBA16
  572. tf.width = p_size.width;
  573. tf.height = p_size.height;
  574. tf.usage_bits = RD::TEXTURE_USAGE_STORAGE_BIT | RD::TEXTURE_USAGE_CAN_COPY_FROM_BIT;
  575. RID rad_tex = RD::get_singleton()->texture_create(tf, RD::TextureView());
  576. copy_effects->copy_cubemap_to_panorama(radiance, rad_tex, p_size, p_roughness_layers, reflection.layers.size() > 1);
  577. Vector<uint8_t> data = RD::get_singleton()->texture_get_data(rad_tex, 0);
  578. RD::get_singleton()->free(rad_tex);
  579. Ref<Image> img = Image::create_from_data(p_size.width, p_size.height, false, Image::FORMAT_RGBAF, data);
  580. for (int i = 0; i < p_size.width; i++) {
  581. for (int j = 0; j < p_size.height; j++) {
  582. Color c = img->get_pixel(i, j);
  583. c.r *= p_energy;
  584. c.g *= p_energy;
  585. c.b *= p_energy;
  586. img->set_pixel(i, j, c);
  587. }
  588. }
  589. return img;
  590. }
  591. return Ref<Image>();
  592. }
  593. ////////////////////////////////////////////////////////////////////////////////
  594. // SkyRD
  595. RendererRD::MaterialStorage::ShaderData *SkyRD::_create_sky_shader_func() {
  596. SkyShaderData *shader_data = memnew(SkyShaderData);
  597. return shader_data;
  598. }
  599. RendererRD::MaterialStorage::ShaderData *SkyRD::_create_sky_shader_funcs() {
  600. // !BAS! Why isn't _create_sky_shader_func not just static too?
  601. return static_cast<RendererSceneRenderRD *>(RendererSceneRenderRD::singleton)->sky._create_sky_shader_func();
  602. }
  603. RendererRD::MaterialStorage::MaterialData *SkyRD::_create_sky_material_func(SkyShaderData *p_shader) {
  604. SkyMaterialData *material_data = memnew(SkyMaterialData);
  605. material_data->shader_data = p_shader;
  606. //update will happen later anyway so do nothing.
  607. return material_data;
  608. }
  609. RendererRD::MaterialStorage::MaterialData *SkyRD::_create_sky_material_funcs(RendererRD::MaterialStorage::ShaderData *p_shader) {
  610. // !BAS! same here, we could just make _create_sky_material_func static?
  611. return static_cast<RendererSceneRenderRD *>(RendererSceneRenderRD::singleton)->sky._create_sky_material_func(static_cast<SkyShaderData *>(p_shader));
  612. }
  613. SkyRD::SkyRD() {
  614. roughness_layers = GLOBAL_GET("rendering/reflections/sky_reflections/roughness_layers");
  615. sky_ggx_samples_quality = GLOBAL_GET("rendering/reflections/sky_reflections/ggx_samples");
  616. sky_use_cubemap_array = GLOBAL_GET("rendering/reflections/sky_reflections/texture_array_reflections");
  617. }
  618. void SkyRD::init() {
  619. RendererRD::TextureStorage *texture_storage = RendererRD::TextureStorage::get_singleton();
  620. RendererRD::MaterialStorage *material_storage = RendererRD::MaterialStorage::get_singleton();
  621. {
  622. // Start with the directional lights for the sky
  623. sky_scene_state.max_directional_lights = 4;
  624. uint32_t directional_light_buffer_size = sky_scene_state.max_directional_lights * sizeof(SkyDirectionalLightData);
  625. sky_scene_state.directional_lights = memnew_arr(SkyDirectionalLightData, sky_scene_state.max_directional_lights);
  626. sky_scene_state.last_frame_directional_lights = memnew_arr(SkyDirectionalLightData, sky_scene_state.max_directional_lights);
  627. sky_scene_state.last_frame_directional_light_count = sky_scene_state.max_directional_lights + 1;
  628. sky_scene_state.directional_light_buffer = RD::get_singleton()->uniform_buffer_create(directional_light_buffer_size);
  629. String defines = "\n#define MAX_DIRECTIONAL_LIGHT_DATA_STRUCTS " + itos(sky_scene_state.max_directional_lights) + "\n";
  630. defines += "\n#define SAMPLERS_BINDING_FIRST_INDEX " + itos(SAMPLERS_BINDING_FIRST_INDEX) + "\n";
  631. // Initialize sky
  632. Vector<String> sky_modes;
  633. sky_modes.push_back(""); // Full size
  634. sky_modes.push_back("\n#define USE_HALF_RES_PASS\n"); // Half Res
  635. sky_modes.push_back("\n#define USE_QUARTER_RES_PASS\n"); // Quarter res
  636. sky_modes.push_back("\n#define USE_CUBEMAP_PASS\n"); // Cubemap
  637. sky_modes.push_back("\n#define USE_CUBEMAP_PASS\n#define USE_HALF_RES_PASS\n"); // Half Res Cubemap
  638. sky_modes.push_back("\n#define USE_CUBEMAP_PASS\n#define USE_QUARTER_RES_PASS\n"); // Quarter res Cubemap
  639. sky_modes.push_back("\n#define USE_MULTIVIEW\n"); // Full size multiview
  640. sky_modes.push_back("\n#define USE_HALF_RES_PASS\n#define USE_MULTIVIEW\n"); // Half Res multiview
  641. sky_modes.push_back("\n#define USE_QUARTER_RES_PASS\n#define USE_MULTIVIEW\n"); // Quarter res multiview
  642. sky_shader.shader.initialize(sky_modes, defines);
  643. if (!RendererCompositorRD::get_singleton()->is_xr_enabled()) {
  644. sky_shader.shader.set_variant_enabled(SKY_VERSION_BACKGROUND_MULTIVIEW, false);
  645. sky_shader.shader.set_variant_enabled(SKY_VERSION_HALF_RES_MULTIVIEW, false);
  646. sky_shader.shader.set_variant_enabled(SKY_VERSION_QUARTER_RES_MULTIVIEW, false);
  647. }
  648. }
  649. // register our shader funds
  650. material_storage->shader_set_data_request_function(RendererRD::MaterialStorage::SHADER_TYPE_SKY, _create_sky_shader_funcs);
  651. material_storage->material_set_data_request_function(RendererRD::MaterialStorage::SHADER_TYPE_SKY, _create_sky_material_funcs);
  652. {
  653. ShaderCompiler::DefaultIdentifierActions actions;
  654. actions.renames["COLOR"] = "color";
  655. actions.renames["ALPHA"] = "alpha";
  656. actions.renames["EYEDIR"] = "cube_normal";
  657. actions.renames["POSITION"] = "params.position";
  658. actions.renames["SKY_COORDS"] = "panorama_coords";
  659. actions.renames["SCREEN_UV"] = "uv";
  660. actions.renames["FRAGCOORD"] = "gl_FragCoord";
  661. actions.renames["TIME"] = "params.time";
  662. actions.renames["PI"] = _MKSTR(Math_PI);
  663. actions.renames["TAU"] = _MKSTR(Math_TAU);
  664. actions.renames["E"] = _MKSTR(Math_E);
  665. actions.renames["HALF_RES_COLOR"] = "half_res_color";
  666. actions.renames["QUARTER_RES_COLOR"] = "quarter_res_color";
  667. actions.renames["RADIANCE"] = "radiance";
  668. actions.renames["FOG"] = "custom_fog";
  669. actions.renames["LIGHT0_ENABLED"] = "directional_lights.data[0].enabled";
  670. actions.renames["LIGHT0_DIRECTION"] = "directional_lights.data[0].direction_energy.xyz";
  671. actions.renames["LIGHT0_ENERGY"] = "directional_lights.data[0].direction_energy.w";
  672. actions.renames["LIGHT0_COLOR"] = "directional_lights.data[0].color_size.xyz";
  673. actions.renames["LIGHT0_SIZE"] = "directional_lights.data[0].color_size.w";
  674. actions.renames["LIGHT1_ENABLED"] = "directional_lights.data[1].enabled";
  675. actions.renames["LIGHT1_DIRECTION"] = "directional_lights.data[1].direction_energy.xyz";
  676. actions.renames["LIGHT1_ENERGY"] = "directional_lights.data[1].direction_energy.w";
  677. actions.renames["LIGHT1_COLOR"] = "directional_lights.data[1].color_size.xyz";
  678. actions.renames["LIGHT1_SIZE"] = "directional_lights.data[1].color_size.w";
  679. actions.renames["LIGHT2_ENABLED"] = "directional_lights.data[2].enabled";
  680. actions.renames["LIGHT2_DIRECTION"] = "directional_lights.data[2].direction_energy.xyz";
  681. actions.renames["LIGHT2_ENERGY"] = "directional_lights.data[2].direction_energy.w";
  682. actions.renames["LIGHT2_COLOR"] = "directional_lights.data[2].color_size.xyz";
  683. actions.renames["LIGHT2_SIZE"] = "directional_lights.data[2].color_size.w";
  684. actions.renames["LIGHT3_ENABLED"] = "directional_lights.data[3].enabled";
  685. actions.renames["LIGHT3_DIRECTION"] = "directional_lights.data[3].direction_energy.xyz";
  686. actions.renames["LIGHT3_ENERGY"] = "directional_lights.data[3].direction_energy.w";
  687. actions.renames["LIGHT3_COLOR"] = "directional_lights.data[3].color_size.xyz";
  688. actions.renames["LIGHT3_SIZE"] = "directional_lights.data[3].color_size.w";
  689. actions.renames["AT_CUBEMAP_PASS"] = "AT_CUBEMAP_PASS";
  690. actions.renames["AT_HALF_RES_PASS"] = "AT_HALF_RES_PASS";
  691. actions.renames["AT_QUARTER_RES_PASS"] = "AT_QUARTER_RES_PASS";
  692. actions.custom_samplers["RADIANCE"] = "SAMPLER_LINEAR_WITH_MIPMAPS_CLAMP";
  693. actions.usage_defines["HALF_RES_COLOR"] = "\n#define USES_HALF_RES_COLOR\n";
  694. actions.usage_defines["QUARTER_RES_COLOR"] = "\n#define USES_QUARTER_RES_COLOR\n";
  695. actions.render_mode_defines["disable_fog"] = "#define DISABLE_FOG\n";
  696. actions.render_mode_defines["use_debanding"] = "#define USE_DEBANDING\n";
  697. actions.base_texture_binding_index = 1;
  698. actions.texture_layout_set = 1;
  699. actions.base_uniform_string = "material.";
  700. actions.base_varying_index = 10;
  701. actions.default_filter = ShaderLanguage::FILTER_LINEAR_MIPMAP;
  702. actions.default_repeat = ShaderLanguage::REPEAT_ENABLE;
  703. actions.global_buffer_array_variable = "global_shader_uniforms.data";
  704. sky_shader.compiler.initialize(actions);
  705. }
  706. {
  707. // default material and shader for sky shader
  708. sky_shader.default_shader = material_storage->shader_allocate();
  709. material_storage->shader_initialize(sky_shader.default_shader);
  710. material_storage->shader_set_code(sky_shader.default_shader, R"(
  711. // Default sky shader.
  712. shader_type sky;
  713. void sky() {
  714. COLOR = vec3(0.0);
  715. }
  716. )");
  717. sky_shader.default_material = material_storage->material_allocate();
  718. material_storage->material_initialize(sky_shader.default_material);
  719. material_storage->material_set_shader(sky_shader.default_material, sky_shader.default_shader);
  720. SkyMaterialData *md = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_shader.default_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  721. sky_shader.default_shader_rd = sky_shader.shader.version_get_shader(md->shader_data->version, SKY_VERSION_BACKGROUND);
  722. sky_scene_state.uniform_buffer = RD::get_singleton()->uniform_buffer_create(sizeof(SkySceneState::UBO));
  723. Vector<RD::Uniform> uniforms;
  724. {
  725. RD::Uniform u;
  726. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  727. u.binding = 1;
  728. u.append_id(RendererRD::MaterialStorage::get_singleton()->global_shader_uniforms_get_storage_buffer());
  729. uniforms.push_back(u);
  730. }
  731. {
  732. RD::Uniform u;
  733. u.binding = 2;
  734. u.uniform_type = RD::UNIFORM_TYPE_UNIFORM_BUFFER;
  735. u.append_id(sky_scene_state.uniform_buffer);
  736. uniforms.push_back(u);
  737. }
  738. {
  739. RD::Uniform u;
  740. u.binding = 3;
  741. u.uniform_type = RD::UNIFORM_TYPE_UNIFORM_BUFFER;
  742. u.append_id(sky_scene_state.directional_light_buffer);
  743. uniforms.push_back(u);
  744. }
  745. material_storage->samplers_rd_get_default().append_uniforms(uniforms, SAMPLERS_BINDING_FIRST_INDEX);
  746. sky_scene_state.uniform_set = RD::get_singleton()->uniform_set_create(uniforms, sky_shader.default_shader_rd, SKY_SET_UNIFORMS);
  747. }
  748. {
  749. Vector<RD::Uniform> uniforms;
  750. {
  751. RD::Uniform u;
  752. u.binding = 0;
  753. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  754. RID vfog = texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_3D_WHITE);
  755. u.append_id(vfog);
  756. uniforms.push_back(u);
  757. }
  758. sky_scene_state.default_fog_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, sky_shader.default_shader_rd, SKY_SET_FOG);
  759. }
  760. {
  761. // Need defaults for using fog with clear color
  762. sky_scene_state.fog_shader = material_storage->shader_allocate();
  763. material_storage->shader_initialize(sky_scene_state.fog_shader);
  764. material_storage->shader_set_code(sky_scene_state.fog_shader, R"(
  765. // Default clear color sky shader.
  766. shader_type sky;
  767. uniform vec4 clear_color;
  768. void sky() {
  769. COLOR = clear_color.rgb;
  770. }
  771. )");
  772. sky_scene_state.fog_material = material_storage->material_allocate();
  773. material_storage->material_initialize(sky_scene_state.fog_material);
  774. material_storage->material_set_shader(sky_scene_state.fog_material, sky_scene_state.fog_shader);
  775. Vector<RD::Uniform> uniforms;
  776. {
  777. RD::Uniform u;
  778. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  779. u.binding = 0;
  780. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_CUBEMAP_BLACK));
  781. uniforms.push_back(u);
  782. }
  783. {
  784. RD::Uniform u;
  785. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  786. u.binding = 1;
  787. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_WHITE));
  788. uniforms.push_back(u);
  789. }
  790. {
  791. RD::Uniform u;
  792. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  793. u.binding = 2;
  794. u.append_id(texture_storage->texture_rd_get_default(RendererRD::TextureStorage::DEFAULT_RD_TEXTURE_WHITE));
  795. uniforms.push_back(u);
  796. }
  797. sky_scene_state.fog_only_texture_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, sky_shader.default_shader_rd, SKY_SET_TEXTURES);
  798. }
  799. }
  800. void SkyRD::set_texture_format(RD::DataFormat p_texture_format) {
  801. texture_format = p_texture_format;
  802. }
  803. SkyRD::~SkyRD() {
  804. // cleanup anything created in init...
  805. RendererRD::MaterialStorage *material_storage = RendererRD::MaterialStorage::get_singleton();
  806. SkyMaterialData *md = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_shader.default_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  807. sky_shader.shader.version_free(md->shader_data->version);
  808. RD::get_singleton()->free(sky_scene_state.directional_light_buffer);
  809. RD::get_singleton()->free(sky_scene_state.uniform_buffer);
  810. memdelete_arr(sky_scene_state.directional_lights);
  811. memdelete_arr(sky_scene_state.last_frame_directional_lights);
  812. material_storage->shader_free(sky_shader.default_shader);
  813. material_storage->material_free(sky_shader.default_material);
  814. material_storage->shader_free(sky_scene_state.fog_shader);
  815. material_storage->material_free(sky_scene_state.fog_material);
  816. if (RD::get_singleton()->uniform_set_is_valid(sky_scene_state.uniform_set)) {
  817. RD::get_singleton()->free(sky_scene_state.uniform_set);
  818. }
  819. if (RD::get_singleton()->uniform_set_is_valid(sky_scene_state.default_fog_uniform_set)) {
  820. RD::get_singleton()->free(sky_scene_state.default_fog_uniform_set);
  821. }
  822. if (RD::get_singleton()->uniform_set_is_valid(sky_scene_state.fog_only_texture_uniform_set)) {
  823. RD::get_singleton()->free(sky_scene_state.fog_only_texture_uniform_set);
  824. }
  825. }
  826. void SkyRD::setup_sky(const RenderDataRD *p_render_data, const Size2i p_screen_size) {
  827. RendererRD::LightStorage *light_storage = RendererRD::LightStorage::get_singleton();
  828. RendererRD::MaterialStorage *material_storage = RendererRD::MaterialStorage::get_singleton();
  829. ERR_FAIL_COND(p_render_data->environment.is_null());
  830. ERR_FAIL_COND(p_render_data->render_buffers.is_null());
  831. // make sure we support our view count
  832. ERR_FAIL_COND(p_render_data->scene_data->view_count == 0);
  833. ERR_FAIL_COND(p_render_data->scene_data->view_count > RendererSceneRender::MAX_RENDER_VIEWS);
  834. SkyMaterialData *material = nullptr;
  835. Sky *sky = get_sky(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_render_data->environment));
  836. RID sky_material;
  837. SkyShaderData *shader_data = nullptr;
  838. if (sky) {
  839. sky_material = sky_get_material(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_render_data->environment));
  840. if (sky_material.is_valid()) {
  841. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  842. if (!material || !material->shader_data->valid) {
  843. material = nullptr;
  844. }
  845. }
  846. }
  847. if (!material) {
  848. sky_material = sky_shader.default_material;
  849. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  850. }
  851. ERR_FAIL_NULL(material);
  852. shader_data = material->shader_data;
  853. ERR_FAIL_NULL(shader_data);
  854. material->set_as_used();
  855. if (sky) {
  856. // Save our screen size; our buffers will already have been cleared.
  857. sky->screen_size.x = p_screen_size.x < 4 ? 4 : p_screen_size.x;
  858. sky->screen_size.y = p_screen_size.y < 4 ? 4 : p_screen_size.y;
  859. // Trigger updating radiance buffers.
  860. if (sky->radiance.is_null()) {
  861. invalidate_sky(sky);
  862. update_dirty_skys();
  863. }
  864. if (shader_data->uses_time && p_render_data->scene_data->time - sky->prev_time > 0.00001) {
  865. sky->prev_time = p_render_data->scene_data->time;
  866. sky->reflection.dirty = true;
  867. RenderingServerDefault::redraw_request();
  868. }
  869. if (material != sky->prev_material) {
  870. sky->prev_material = material;
  871. sky->reflection.dirty = true;
  872. }
  873. if (material->uniform_set_updated) {
  874. material->uniform_set_updated = false;
  875. sky->reflection.dirty = true;
  876. }
  877. if (!p_render_data->scene_data->cam_transform.origin.is_equal_approx(sky->prev_position) && shader_data->uses_position) {
  878. sky->prev_position = p_render_data->scene_data->cam_transform.origin;
  879. sky->reflection.dirty = true;
  880. }
  881. }
  882. sky_scene_state.ubo.directional_light_count = 0;
  883. if (shader_data->uses_light) {
  884. const PagedArray<RID> &lights = *p_render_data->lights;
  885. // Run through the list of lights in the scene and pick out the Directional Lights.
  886. // This can't be done in RenderSceneRenderRD::_setup lights because that needs to be called
  887. // after the depth prepass, but this runs before the depth prepass.
  888. for (int i = 0; i < (int)lights.size(); i++) {
  889. if (!light_storage->owns_light_instance(lights[i])) {
  890. continue;
  891. }
  892. RID base = light_storage->light_instance_get_base_light(lights[i]);
  893. ERR_CONTINUE(base.is_null());
  894. RS::LightType type = light_storage->light_get_type(base);
  895. if (type == RS::LIGHT_DIRECTIONAL && light_storage->light_directional_get_sky_mode(base) != RS::LIGHT_DIRECTIONAL_SKY_MODE_LIGHT_ONLY) {
  896. SkyDirectionalLightData &sky_light_data = sky_scene_state.directional_lights[sky_scene_state.ubo.directional_light_count];
  897. Transform3D light_transform = light_storage->light_instance_get_base_transform(lights[i]);
  898. Vector3 world_direction = light_transform.basis.xform(Vector3(0, 0, 1)).normalized();
  899. sky_light_data.direction[0] = world_direction.x;
  900. sky_light_data.direction[1] = world_direction.y;
  901. sky_light_data.direction[2] = world_direction.z;
  902. float sign = light_storage->light_is_negative(base) ? -1 : 1;
  903. sky_light_data.energy = sign * light_storage->light_get_param(base, RS::LIGHT_PARAM_ENERGY);
  904. if (RendererSceneRenderRD::get_singleton()->is_using_physical_light_units()) {
  905. sky_light_data.energy *= light_storage->light_get_param(base, RS::LIGHT_PARAM_INTENSITY);
  906. }
  907. if (p_render_data->camera_attributes.is_valid()) {
  908. sky_light_data.energy *= RSG::camera_attributes->camera_attributes_get_exposure_normalization_factor(p_render_data->camera_attributes);
  909. }
  910. Color linear_col = light_storage->light_get_color(base).srgb_to_linear();
  911. sky_light_data.color[0] = linear_col.r;
  912. sky_light_data.color[1] = linear_col.g;
  913. sky_light_data.color[2] = linear_col.b;
  914. sky_light_data.enabled = true;
  915. float angular_diameter = light_storage->light_get_param(base, RS::LIGHT_PARAM_SIZE);
  916. sky_light_data.size = Math::deg_to_rad(angular_diameter);
  917. sky_scene_state.ubo.directional_light_count++;
  918. if (sky_scene_state.ubo.directional_light_count >= sky_scene_state.max_directional_lights) {
  919. break;
  920. }
  921. }
  922. }
  923. // Check whether the directional_light_buffer changes.
  924. bool light_data_dirty = false;
  925. // Light buffer is dirty if we have fewer or more lights.
  926. // If we have fewer lights, make sure that old lights are disabled.
  927. if (sky_scene_state.ubo.directional_light_count != sky_scene_state.last_frame_directional_light_count) {
  928. light_data_dirty = true;
  929. for (uint32_t i = sky_scene_state.ubo.directional_light_count; i < sky_scene_state.max_directional_lights; i++) {
  930. sky_scene_state.directional_lights[i].enabled = false;
  931. sky_scene_state.last_frame_directional_lights[i].enabled = false;
  932. }
  933. }
  934. if (!light_data_dirty) {
  935. for (uint32_t i = 0; i < sky_scene_state.ubo.directional_light_count; i++) {
  936. if (sky_scene_state.directional_lights[i].direction[0] != sky_scene_state.last_frame_directional_lights[i].direction[0] ||
  937. sky_scene_state.directional_lights[i].direction[1] != sky_scene_state.last_frame_directional_lights[i].direction[1] ||
  938. sky_scene_state.directional_lights[i].direction[2] != sky_scene_state.last_frame_directional_lights[i].direction[2] ||
  939. sky_scene_state.directional_lights[i].energy != sky_scene_state.last_frame_directional_lights[i].energy ||
  940. sky_scene_state.directional_lights[i].color[0] != sky_scene_state.last_frame_directional_lights[i].color[0] ||
  941. sky_scene_state.directional_lights[i].color[1] != sky_scene_state.last_frame_directional_lights[i].color[1] ||
  942. sky_scene_state.directional_lights[i].color[2] != sky_scene_state.last_frame_directional_lights[i].color[2] ||
  943. sky_scene_state.directional_lights[i].enabled != sky_scene_state.last_frame_directional_lights[i].enabled ||
  944. sky_scene_state.directional_lights[i].size != sky_scene_state.last_frame_directional_lights[i].size) {
  945. light_data_dirty = true;
  946. break;
  947. }
  948. }
  949. }
  950. if (light_data_dirty) {
  951. RD::get_singleton()->buffer_update(sky_scene_state.directional_light_buffer, 0, sizeof(SkyDirectionalLightData) * sky_scene_state.max_directional_lights, sky_scene_state.directional_lights);
  952. SkyDirectionalLightData *temp = sky_scene_state.last_frame_directional_lights;
  953. sky_scene_state.last_frame_directional_lights = sky_scene_state.directional_lights;
  954. sky_scene_state.directional_lights = temp;
  955. sky_scene_state.last_frame_directional_light_count = sky_scene_state.ubo.directional_light_count;
  956. if (sky) {
  957. sky->reflection.dirty = true;
  958. }
  959. }
  960. }
  961. // Setup fog variables.
  962. sky_scene_state.ubo.volumetric_fog_enabled = false;
  963. if (p_render_data->render_buffers->has_custom_data(RB_SCOPE_FOG)) {
  964. Ref<RendererRD::Fog::VolumetricFog> fog = p_render_data->render_buffers->get_custom_data(RB_SCOPE_FOG);
  965. sky_scene_state.ubo.volumetric_fog_enabled = true;
  966. float fog_end = fog->length;
  967. if (fog_end > 0.0) {
  968. sky_scene_state.ubo.volumetric_fog_inv_length = 1.0 / fog_end;
  969. } else {
  970. sky_scene_state.ubo.volumetric_fog_inv_length = 1.0;
  971. }
  972. float fog_detail_spread = fog->spread; // Reverse lookup.
  973. if (fog_detail_spread > 0.0) {
  974. sky_scene_state.ubo.volumetric_fog_detail_spread = 1.0 / fog_detail_spread;
  975. } else {
  976. sky_scene_state.ubo.volumetric_fog_detail_spread = 1.0;
  977. }
  978. sky_scene_state.fog_uniform_set = fog->sky_uniform_set;
  979. }
  980. sky_scene_state.view_count = p_render_data->scene_data->view_count;
  981. sky_scene_state.cam_transform = p_render_data->scene_data->cam_transform;
  982. Projection correction;
  983. correction.set_depth_correction(p_render_data->scene_data->flip_y, true);
  984. correction.add_jitter_offset(p_render_data->scene_data->taa_jitter);
  985. Projection projection = p_render_data->scene_data->cam_projection;
  986. if (p_render_data->scene_data->cam_frustum) {
  987. // We don't use a full projection matrix for the sky, this is enough to make up for it.
  988. projection[2].y = -projection[2].y;
  989. }
  990. float custom_fov = RendererSceneRenderRD::get_singleton()->environment_get_sky_custom_fov(p_render_data->environment);
  991. if (custom_fov && sky_scene_state.view_count == 1) {
  992. // With custom fov we don't support stereo...
  993. float near_plane = projection.get_z_near();
  994. float far_plane = projection.get_z_far();
  995. float aspect = projection.get_aspect();
  996. projection.set_perspective(custom_fov, aspect, near_plane, far_plane);
  997. }
  998. sky_scene_state.cam_projection = correction * projection;
  999. // Our info in our UBO is only used if we're rendering stereo.
  1000. for (uint32_t i = 0; i < p_render_data->scene_data->view_count; i++) {
  1001. Projection view_inv_projection = (correction * p_render_data->scene_data->view_projection[i]).inverse();
  1002. if (p_render_data->scene_data->view_count > 1) {
  1003. // Reprojection is used when we need to have things in combined space.
  1004. RendererRD::MaterialStorage::store_camera(p_render_data->scene_data->cam_projection * view_inv_projection, sky_scene_state.ubo.combined_reprojection[i]);
  1005. } else {
  1006. // This is unused so just reset to identity.
  1007. Projection ident;
  1008. RendererRD::MaterialStorage::store_camera(ident, sky_scene_state.ubo.combined_reprojection[i]);
  1009. }
  1010. RendererRD::MaterialStorage::store_camera(view_inv_projection, sky_scene_state.ubo.view_inv_projections[i]);
  1011. sky_scene_state.ubo.view_eye_offsets[i][0] = p_render_data->scene_data->view_eye_offset[i].x;
  1012. sky_scene_state.ubo.view_eye_offsets[i][1] = p_render_data->scene_data->view_eye_offset[i].y;
  1013. sky_scene_state.ubo.view_eye_offsets[i][2] = p_render_data->scene_data->view_eye_offset[i].z;
  1014. sky_scene_state.ubo.view_eye_offsets[i][3] = 0.0;
  1015. }
  1016. sky_scene_state.ubo.z_far = p_render_data->scene_data->view_projection[0].get_z_far(); // Should be the same for all projection.
  1017. sky_scene_state.ubo.fog_enabled = RendererSceneRenderRD::get_singleton()->environment_get_fog_enabled(p_render_data->environment);
  1018. sky_scene_state.ubo.fog_density = RendererSceneRenderRD::get_singleton()->environment_get_fog_density(p_render_data->environment);
  1019. sky_scene_state.ubo.fog_aerial_perspective = RendererSceneRenderRD::get_singleton()->environment_get_fog_aerial_perspective(p_render_data->environment);
  1020. Color fog_color = RendererSceneRenderRD::get_singleton()->environment_get_fog_light_color(p_render_data->environment).srgb_to_linear();
  1021. float fog_energy = RendererSceneRenderRD::get_singleton()->environment_get_fog_light_energy(p_render_data->environment);
  1022. sky_scene_state.ubo.fog_light_color[0] = fog_color.r * fog_energy;
  1023. sky_scene_state.ubo.fog_light_color[1] = fog_color.g * fog_energy;
  1024. sky_scene_state.ubo.fog_light_color[2] = fog_color.b * fog_energy;
  1025. sky_scene_state.ubo.fog_sun_scatter = RendererSceneRenderRD::get_singleton()->environment_get_fog_sun_scatter(p_render_data->environment);
  1026. sky_scene_state.ubo.fog_sky_affect = RendererSceneRenderRD::get_singleton()->environment_get_fog_sky_affect(p_render_data->environment);
  1027. sky_scene_state.ubo.volumetric_fog_sky_affect = RendererSceneRenderRD::get_singleton()->environment_get_volumetric_fog_sky_affect(p_render_data->environment);
  1028. RD::get_singleton()->buffer_update(sky_scene_state.uniform_buffer, 0, sizeof(SkySceneState::UBO), &sky_scene_state.ubo);
  1029. }
  1030. void SkyRD::update_radiance_buffers(Ref<RenderSceneBuffersRD> p_render_buffers, RID p_env, const Vector3 &p_global_pos, double p_time, float p_luminance_multiplier, float p_brightness_multiplier) {
  1031. ERR_FAIL_COND(p_render_buffers.is_null());
  1032. RendererRD::MaterialStorage *material_storage = RendererRD::MaterialStorage::get_singleton();
  1033. ERR_FAIL_COND(p_env.is_null());
  1034. Sky *sky = get_sky(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_env));
  1035. ERR_FAIL_NULL(sky);
  1036. RID sky_material = sky_get_material(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_env));
  1037. SkyMaterialData *material = nullptr;
  1038. if (sky_material.is_valid()) {
  1039. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1040. if (!material || !material->shader_data->valid) {
  1041. material = nullptr;
  1042. }
  1043. }
  1044. if (!material) {
  1045. sky_material = sky_shader.default_material;
  1046. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1047. }
  1048. ERR_FAIL_NULL(material);
  1049. SkyShaderData *shader_data = material->shader_data;
  1050. ERR_FAIL_NULL(shader_data);
  1051. bool update_single_frame = sky->mode == RS::SKY_MODE_REALTIME || sky->mode == RS::SKY_MODE_QUALITY;
  1052. RS::SkyMode sky_mode = sky->mode;
  1053. if (sky_mode == RS::SKY_MODE_AUTOMATIC) {
  1054. if ((shader_data->uses_time || shader_data->uses_position) && sky->radiance_size == 256) {
  1055. update_single_frame = true;
  1056. sky_mode = RS::SKY_MODE_REALTIME;
  1057. } else if (shader_data->uses_light || shader_data->ubo_size > 0) {
  1058. update_single_frame = false;
  1059. sky_mode = RS::SKY_MODE_INCREMENTAL;
  1060. } else {
  1061. update_single_frame = true;
  1062. sky_mode = RS::SKY_MODE_QUALITY;
  1063. }
  1064. }
  1065. if (sky->processing_layer == 0 && sky_mode == RS::SKY_MODE_INCREMENTAL) {
  1066. // On the first frame after creating sky, rebuild in single frame
  1067. update_single_frame = true;
  1068. sky_mode = RS::SKY_MODE_QUALITY;
  1069. }
  1070. int max_processing_layer = sky_use_cubemap_array ? sky->reflection.layers.size() : sky->reflection.layers[0].mipmaps.size();
  1071. // Update radiance cubemap
  1072. if (sky->reflection.dirty && (sky->processing_layer >= max_processing_layer || update_single_frame)) {
  1073. static const Vector3 view_normals[6] = {
  1074. Vector3(+1, 0, 0),
  1075. Vector3(-1, 0, 0),
  1076. Vector3(0, +1, 0),
  1077. Vector3(0, -1, 0),
  1078. Vector3(0, 0, +1),
  1079. Vector3(0, 0, -1)
  1080. };
  1081. static const Vector3 view_up[6] = {
  1082. Vector3(0, -1, 0),
  1083. Vector3(0, -1, 0),
  1084. Vector3(0, 0, +1),
  1085. Vector3(0, 0, -1),
  1086. Vector3(0, -1, 0),
  1087. Vector3(0, -1, 0)
  1088. };
  1089. Projection cm;
  1090. cm.set_perspective(90, 1, 0.01, 10.0);
  1091. Projection correction;
  1092. correction.set_depth_correction(false);
  1093. cm = correction * cm;
  1094. // Note, we ignore environment_get_sky_orientation here as this is applied when we do our lookup in our scene shader.
  1095. if (shader_data->uses_quarter_res && roughness_layers >= 3) {
  1096. RD::get_singleton()->draw_command_begin_label("Render Sky to Quarter Res Cubemap");
  1097. PipelineCacheRD *pipeline = &shader_data->pipelines[SKY_VERSION_CUBEMAP_QUARTER_RES];
  1098. Vector<Color> clear_colors;
  1099. clear_colors.push_back(Color(0.0, 0.0, 0.0));
  1100. RD::DrawListID cubemap_draw_list;
  1101. for (int i = 0; i < 6; i++) {
  1102. Basis local_view = Basis::looking_at(view_normals[i], view_up[i]);
  1103. RID texture_uniform_set = sky->get_textures(SKY_TEXTURE_SET_CUBEMAP_QUARTER_RES, sky_shader.default_shader_rd, p_render_buffers);
  1104. cubemap_draw_list = RD::get_singleton()->draw_list_begin(sky->reflection.layers[0].mipmaps[2].framebuffers[i]);
  1105. _render_sky(cubemap_draw_list, p_time, sky->reflection.layers[0].mipmaps[2].framebuffers[i], pipeline, material->uniform_set, texture_uniform_set, cm, local_view, p_global_pos, p_luminance_multiplier, p_brightness_multiplier);
  1106. RD::get_singleton()->draw_list_end();
  1107. }
  1108. RD::get_singleton()->draw_command_end_label();
  1109. } else if (shader_data->uses_quarter_res && roughness_layers < 3) {
  1110. ERR_PRINT_ED("Cannot use quarter res buffer in sky shader when roughness layers is less than 3. Please increase rendering/reflections/sky_reflections/roughness_layers.");
  1111. }
  1112. if (shader_data->uses_half_res && roughness_layers >= 2) {
  1113. RD::get_singleton()->draw_command_begin_label("Render Sky to Half Res Cubemap");
  1114. PipelineCacheRD *pipeline = &shader_data->pipelines[SKY_VERSION_CUBEMAP_HALF_RES];
  1115. Vector<Color> clear_colors;
  1116. clear_colors.push_back(Color(0.0, 0.0, 0.0));
  1117. RD::DrawListID cubemap_draw_list;
  1118. for (int i = 0; i < 6; i++) {
  1119. Basis local_view = Basis::looking_at(view_normals[i], view_up[i]);
  1120. RID texture_uniform_set = sky->get_textures(SKY_TEXTURE_SET_CUBEMAP_HALF_RES, sky_shader.default_shader_rd, p_render_buffers);
  1121. cubemap_draw_list = RD::get_singleton()->draw_list_begin(sky->reflection.layers[0].mipmaps[1].framebuffers[i]);
  1122. _render_sky(cubemap_draw_list, p_time, sky->reflection.layers[0].mipmaps[1].framebuffers[i], pipeline, material->uniform_set, texture_uniform_set, cm, local_view, p_global_pos, p_luminance_multiplier, p_brightness_multiplier);
  1123. RD::get_singleton()->draw_list_end();
  1124. }
  1125. RD::get_singleton()->draw_command_end_label();
  1126. } else if (shader_data->uses_half_res && roughness_layers < 2) {
  1127. ERR_PRINT_ED("Cannot use half res buffer in sky shader when roughness layers is less than 2. Please increase rendering/reflections/sky_reflections/roughness_layers.");
  1128. }
  1129. RD::DrawListID cubemap_draw_list;
  1130. PipelineCacheRD *pipeline = &shader_data->pipelines[SKY_VERSION_CUBEMAP];
  1131. RD::get_singleton()->draw_command_begin_label("Render Sky Cubemap");
  1132. for (int i = 0; i < 6; i++) {
  1133. Basis local_view = Basis::looking_at(view_normals[i], view_up[i]);
  1134. RID texture_uniform_set = sky->get_textures(SKY_TEXTURE_SET_CUBEMAP, sky_shader.default_shader_rd, p_render_buffers);
  1135. cubemap_draw_list = RD::get_singleton()->draw_list_begin(sky->reflection.layers[0].mipmaps[0].framebuffers[i], RD::DRAW_DEFAULT_ALL, Vector<Color>(), 1.0f, 0, Rect2(), RDD::BreadcrumbMarker::SKY_PASS | uint32_t(i));
  1136. _render_sky(cubemap_draw_list, p_time, sky->reflection.layers[0].mipmaps[0].framebuffers[i], pipeline, material->uniform_set, texture_uniform_set, cm, local_view, p_global_pos, p_luminance_multiplier, p_brightness_multiplier);
  1137. RD::get_singleton()->draw_list_end();
  1138. }
  1139. RD::get_singleton()->draw_command_end_label();
  1140. if (sky_mode == RS::SKY_MODE_REALTIME) {
  1141. sky->reflection.create_reflection_fast_filter(sky_use_cubemap_array);
  1142. if (sky_use_cubemap_array) {
  1143. sky->reflection.update_reflection_mipmaps(0, sky->reflection.layers.size());
  1144. }
  1145. } else {
  1146. if (update_single_frame) {
  1147. for (int i = 1; i < max_processing_layer; i++) {
  1148. sky->reflection.create_reflection_importance_sample(sky_use_cubemap_array, 10, i, sky_ggx_samples_quality);
  1149. }
  1150. if (sky_use_cubemap_array) {
  1151. sky->reflection.update_reflection_mipmaps(0, sky->reflection.layers.size());
  1152. }
  1153. } else {
  1154. if (sky_use_cubemap_array) {
  1155. // Multi-Frame so just update the first array level
  1156. sky->reflection.update_reflection_mipmaps(0, 1);
  1157. }
  1158. }
  1159. sky->processing_layer = 1;
  1160. }
  1161. sky->baked_exposure = p_luminance_multiplier;
  1162. sky->reflection.dirty = false;
  1163. } else {
  1164. if (sky_mode == RS::SKY_MODE_INCREMENTAL && sky->processing_layer < max_processing_layer) {
  1165. sky->reflection.create_reflection_importance_sample(sky_use_cubemap_array, 10, sky->processing_layer, sky_ggx_samples_quality);
  1166. if (sky_use_cubemap_array) {
  1167. sky->reflection.update_reflection_mipmaps(sky->processing_layer, sky->processing_layer + 1);
  1168. }
  1169. sky->processing_layer++;
  1170. }
  1171. }
  1172. }
  1173. void SkyRD::update_res_buffers(Ref<RenderSceneBuffersRD> p_render_buffers, RID p_env, double p_time, float p_luminance_multiplier, float p_brightness_multiplier) {
  1174. ERR_FAIL_COND(p_render_buffers.is_null());
  1175. RendererRD::MaterialStorage *material_storage = RendererRD::MaterialStorage::get_singleton();
  1176. ERR_FAIL_COND(p_env.is_null());
  1177. Sky *sky = get_sky(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_env));
  1178. SkyMaterialData *material = nullptr;
  1179. RID sky_material;
  1180. RS::EnvironmentBG background = RendererSceneRenderRD::get_singleton()->environment_get_background(p_env);
  1181. if (!(background == RS::ENV_BG_CLEAR_COLOR || background == RS::ENV_BG_COLOR) || sky) {
  1182. ERR_FAIL_NULL(sky);
  1183. sky_material = sky_get_material(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_env));
  1184. if (sky_material.is_valid()) {
  1185. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1186. if (!material || !material->shader_data->valid) {
  1187. material = nullptr;
  1188. }
  1189. }
  1190. if (!material) {
  1191. sky_material = sky_shader.default_material;
  1192. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1193. }
  1194. }
  1195. if (background == RS::ENV_BG_CLEAR_COLOR || background == RS::ENV_BG_COLOR) {
  1196. sky_material = sky_scene_state.fog_material;
  1197. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1198. }
  1199. ERR_FAIL_NULL(material);
  1200. SkyShaderData *shader_data = material->shader_data;
  1201. ERR_FAIL_NULL(shader_data);
  1202. if (!shader_data->uses_quarter_res && !shader_data->uses_half_res) {
  1203. return;
  1204. }
  1205. material->set_as_used();
  1206. RENDER_TIMESTAMP("Setup Sky Resolution Buffers");
  1207. RD::get_singleton()->draw_command_begin_label("Setup Sky Resolution Buffers");
  1208. Basis sky_transform = RendererSceneRenderRD::get_singleton()->environment_get_sky_orientation(p_env);
  1209. sky_transform.invert();
  1210. // Camera
  1211. Projection projection = sky_scene_state.cam_projection;
  1212. sky_transform = sky_transform * sky_scene_state.cam_transform.basis;
  1213. if (shader_data->uses_quarter_res) {
  1214. PipelineCacheRD *pipeline = &shader_data->pipelines[sky_scene_state.view_count > 1 ? SKY_VERSION_QUARTER_RES_MULTIVIEW : SKY_VERSION_QUARTER_RES];
  1215. // Grab texture and framebuffer from cache, create if needed...
  1216. uint32_t usage_bits = RD::TEXTURE_USAGE_SAMPLING_BIT | RD::TEXTURE_USAGE_COLOR_ATTACHMENT_BIT;
  1217. Size2i quarter_size = sky->screen_size / 4;
  1218. RID texture = p_render_buffers->create_texture(RB_SCOPE_SKY, RB_QUARTER_TEXTURE, texture_format, usage_bits, RD::TEXTURE_SAMPLES_1, quarter_size);
  1219. RID framebuffer = FramebufferCacheRD::get_singleton()->get_cache_multiview(sky_scene_state.view_count, texture);
  1220. RID texture_uniform_set = sky->get_textures(SKY_TEXTURE_SET_QUARTER_RES, sky_shader.default_shader_rd, p_render_buffers);
  1221. Vector<Color> clear_colors;
  1222. clear_colors.push_back(Color(0.0, 0.0, 0.0));
  1223. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(framebuffer, RD::DRAW_CLEAR_ALL, clear_colors);
  1224. _render_sky(draw_list, p_time, framebuffer, pipeline, material->uniform_set, texture_uniform_set, projection, sky_transform, sky_scene_state.cam_transform.origin, p_luminance_multiplier, p_brightness_multiplier);
  1225. RD::get_singleton()->draw_list_end();
  1226. }
  1227. if (shader_data->uses_half_res) {
  1228. PipelineCacheRD *pipeline = &shader_data->pipelines[sky_scene_state.view_count > 1 ? SKY_VERSION_HALF_RES_MULTIVIEW : SKY_VERSION_HALF_RES];
  1229. // Grab texture and framebuffer from cache, create if needed...
  1230. uint32_t usage_bits = RD::TEXTURE_USAGE_SAMPLING_BIT | RD::TEXTURE_USAGE_COLOR_ATTACHMENT_BIT;
  1231. Size2i half_size = sky->screen_size / 2;
  1232. RID texture = p_render_buffers->create_texture(RB_SCOPE_SKY, RB_HALF_TEXTURE, texture_format, usage_bits, RD::TEXTURE_SAMPLES_1, half_size);
  1233. RID framebuffer = FramebufferCacheRD::get_singleton()->get_cache_multiview(sky_scene_state.view_count, texture);
  1234. RID texture_uniform_set = sky->get_textures(SKY_TEXTURE_SET_HALF_RES, sky_shader.default_shader_rd, p_render_buffers);
  1235. Vector<Color> clear_colors;
  1236. clear_colors.push_back(Color(0.0, 0.0, 0.0));
  1237. RD::DrawListID draw_list = RD::get_singleton()->draw_list_begin(framebuffer, RD::DRAW_CLEAR_ALL, clear_colors);
  1238. _render_sky(draw_list, p_time, framebuffer, pipeline, material->uniform_set, texture_uniform_set, projection, sky_transform, sky_scene_state.cam_transform.origin, p_luminance_multiplier, p_brightness_multiplier);
  1239. RD::get_singleton()->draw_list_end();
  1240. }
  1241. RD::get_singleton()->draw_command_end_label(); // Setup Sky resolution buffers
  1242. }
  1243. void SkyRD::draw_sky(RD::DrawListID p_draw_list, Ref<RenderSceneBuffersRD> p_render_buffers, RID p_env, RID p_fb, double p_time, float p_luminance_multiplier, float p_brightness_multiplier) {
  1244. ERR_FAIL_COND(p_render_buffers.is_null());
  1245. RendererRD::MaterialStorage *material_storage = RendererRD::MaterialStorage::get_singleton();
  1246. ERR_FAIL_COND(p_env.is_null());
  1247. Sky *sky = get_sky(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_env));
  1248. SkyMaterialData *material = nullptr;
  1249. RID sky_material;
  1250. RS::EnvironmentBG background = RendererSceneRenderRD::get_singleton()->environment_get_background(p_env);
  1251. if (!(background == RS::ENV_BG_CLEAR_COLOR || background == RS::ENV_BG_COLOR) || sky) {
  1252. ERR_FAIL_NULL(sky);
  1253. sky_material = sky_get_material(RendererSceneRenderRD::get_singleton()->environment_get_sky(p_env));
  1254. if (sky_material.is_valid()) {
  1255. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1256. if (!material || !material->shader_data->valid) {
  1257. material = nullptr;
  1258. }
  1259. }
  1260. if (!material) {
  1261. sky_material = sky_shader.default_material;
  1262. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1263. }
  1264. }
  1265. if (background == RS::ENV_BG_CLEAR_COLOR || background == RS::ENV_BG_COLOR) {
  1266. sky_material = sky_scene_state.fog_material;
  1267. material = static_cast<SkyMaterialData *>(material_storage->material_get_data(sky_material, RendererRD::MaterialStorage::SHADER_TYPE_SKY));
  1268. }
  1269. ERR_FAIL_NULL(material);
  1270. SkyShaderData *shader_data = material->shader_data;
  1271. ERR_FAIL_NULL(shader_data);
  1272. material->set_as_used();
  1273. Basis sky_transform = RendererSceneRenderRD::get_singleton()->environment_get_sky_orientation(p_env);
  1274. sky_transform.invert();
  1275. // Camera
  1276. Projection projection = sky_scene_state.cam_projection;
  1277. sky_transform = sky_transform * sky_scene_state.cam_transform.basis;
  1278. PipelineCacheRD *pipeline = &shader_data->pipelines[sky_scene_state.view_count > 1 ? SKY_VERSION_BACKGROUND_MULTIVIEW : SKY_VERSION_BACKGROUND];
  1279. RID texture_uniform_set;
  1280. if (sky) {
  1281. texture_uniform_set = sky->get_textures(SKY_TEXTURE_SET_BACKGROUND, sky_shader.default_shader_rd, p_render_buffers);
  1282. } else {
  1283. texture_uniform_set = sky_scene_state.fog_only_texture_uniform_set;
  1284. }
  1285. _render_sky(p_draw_list, p_time, p_fb, pipeline, material->uniform_set, texture_uniform_set, projection, sky_transform, sky_scene_state.cam_transform.origin, p_luminance_multiplier, p_brightness_multiplier);
  1286. }
  1287. void SkyRD::invalidate_sky(Sky *p_sky) {
  1288. if (!p_sky->dirty) {
  1289. p_sky->dirty = true;
  1290. p_sky->dirty_list = dirty_sky_list;
  1291. dirty_sky_list = p_sky;
  1292. }
  1293. }
  1294. void SkyRD::update_dirty_skys() {
  1295. Sky *sky = dirty_sky_list;
  1296. while (sky) {
  1297. //update sky configuration if texture is missing
  1298. // TODO See if we can move this into `update_radiance_buffers` and remove our dirty_sky logic.
  1299. // As this is basically a duplicate of the logic in reflection probes we could move this logic
  1300. // into RenderSceneBuffersRD and use that from both places.
  1301. if (sky->radiance.is_null()) {
  1302. int mipmaps = Image::get_image_required_mipmaps(sky->radiance_size, sky->radiance_size, Image::FORMAT_RGBAH) + 1;
  1303. uint32_t w = sky->radiance_size, h = sky->radiance_size;
  1304. int layers = roughness_layers;
  1305. if (sky->mode == RS::SKY_MODE_REALTIME) {
  1306. layers = 8;
  1307. if (roughness_layers != 8) {
  1308. WARN_PRINT("When using the Real-Time sky update mode (or Automatic with a sky shader using \"TIME\"), \"rendering/reflections/sky_reflections/roughness_layers\" should be set to 8 in the project settings for best quality reflections.");
  1309. }
  1310. }
  1311. if (sky_use_cubemap_array) {
  1312. //array (higher quality, 6 times more memory)
  1313. RD::TextureFormat tf;
  1314. tf.array_layers = layers * 6;
  1315. tf.format = texture_format;
  1316. tf.texture_type = RD::TEXTURE_TYPE_CUBE_ARRAY;
  1317. tf.mipmaps = mipmaps;
  1318. tf.width = w;
  1319. tf.height = h;
  1320. tf.usage_bits = RD::TEXTURE_USAGE_COLOR_ATTACHMENT_BIT | RD::TEXTURE_USAGE_SAMPLING_BIT;
  1321. if (RendererSceneRenderRD::get_singleton()->_render_buffers_can_be_storage()) {
  1322. tf.usage_bits |= RD::TEXTURE_USAGE_STORAGE_BIT;
  1323. }
  1324. sky->radiance = RD::get_singleton()->texture_create(tf, RD::TextureView());
  1325. sky->reflection.update_reflection_data(sky->radiance_size, mipmaps, true, sky->radiance, 0, sky->mode == RS::SKY_MODE_REALTIME, roughness_layers, texture_format);
  1326. } else {
  1327. //regular cubemap, lower quality (aliasing, less memory)
  1328. RD::TextureFormat tf;
  1329. tf.array_layers = 6;
  1330. tf.format = texture_format;
  1331. tf.texture_type = RD::TEXTURE_TYPE_CUBE;
  1332. tf.mipmaps = MIN(mipmaps, layers);
  1333. tf.width = w;
  1334. tf.height = h;
  1335. tf.usage_bits = RD::TEXTURE_USAGE_COLOR_ATTACHMENT_BIT | RD::TEXTURE_USAGE_SAMPLING_BIT;
  1336. if (RendererSceneRenderRD::get_singleton()->_render_buffers_can_be_storage()) {
  1337. tf.usage_bits |= RD::TEXTURE_USAGE_STORAGE_BIT;
  1338. }
  1339. sky->radiance = RD::get_singleton()->texture_create(tf, RD::TextureView());
  1340. sky->reflection.update_reflection_data(sky->radiance_size, MIN(mipmaps, layers), false, sky->radiance, 0, sky->mode == RS::SKY_MODE_REALTIME, roughness_layers, texture_format);
  1341. }
  1342. }
  1343. sky->reflection.dirty = true;
  1344. sky->processing_layer = 0;
  1345. Sky *next = sky->dirty_list;
  1346. sky->dirty_list = nullptr;
  1347. sky->dirty = false;
  1348. sky = next;
  1349. }
  1350. dirty_sky_list = nullptr;
  1351. }
  1352. RID SkyRD::sky_get_material(RID p_sky) const {
  1353. Sky *sky = get_sky(p_sky);
  1354. ERR_FAIL_NULL_V(sky, RID());
  1355. return sky->material;
  1356. }
  1357. float SkyRD::sky_get_baked_exposure(RID p_sky) const {
  1358. Sky *sky = get_sky(p_sky);
  1359. ERR_FAIL_NULL_V(sky, 1.0);
  1360. return sky->baked_exposure;
  1361. }
  1362. RID SkyRD::allocate_sky_rid() {
  1363. return sky_owner.allocate_rid();
  1364. }
  1365. void SkyRD::initialize_sky_rid(RID p_rid) {
  1366. sky_owner.initialize_rid(p_rid, Sky());
  1367. }
  1368. SkyRD::Sky *SkyRD::get_sky(RID p_sky) const {
  1369. return sky_owner.get_or_null(p_sky);
  1370. }
  1371. void SkyRD::free_sky(RID p_sky) {
  1372. Sky *sky = get_sky(p_sky);
  1373. ERR_FAIL_NULL(sky);
  1374. sky->free();
  1375. sky_owner.free(p_sky);
  1376. }
  1377. void SkyRD::sky_set_radiance_size(RID p_sky, int p_radiance_size) {
  1378. Sky *sky = get_sky(p_sky);
  1379. ERR_FAIL_NULL(sky);
  1380. if (sky->set_radiance_size(p_radiance_size)) {
  1381. invalidate_sky(sky);
  1382. }
  1383. }
  1384. void SkyRD::sky_set_mode(RID p_sky, RS::SkyMode p_mode) {
  1385. Sky *sky = get_sky(p_sky);
  1386. ERR_FAIL_NULL(sky);
  1387. if (sky->set_mode(p_mode)) {
  1388. invalidate_sky(sky);
  1389. }
  1390. }
  1391. void SkyRD::sky_set_material(RID p_sky, RID p_material) {
  1392. Sky *sky = get_sky(p_sky);
  1393. ERR_FAIL_NULL(sky);
  1394. if (sky->set_material(p_material)) {
  1395. invalidate_sky(sky);
  1396. }
  1397. }
  1398. Ref<Image> SkyRD::sky_bake_panorama(RID p_sky, float p_energy, bool p_bake_irradiance, const Size2i &p_size) {
  1399. Sky *sky = get_sky(p_sky);
  1400. ERR_FAIL_NULL_V(sky, Ref<Image>());
  1401. update_dirty_skys();
  1402. return sky->bake_panorama(p_energy, p_bake_irradiance ? roughness_layers : 0, p_size);
  1403. }
  1404. RID SkyRD::sky_get_radiance_texture_rd(RID p_sky) const {
  1405. Sky *sky = get_sky(p_sky);
  1406. ERR_FAIL_NULL_V(sky, RID());
  1407. return sky->radiance;
  1408. }