renderer_scene_sky_rd.cpp 65 KB

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