particles_storage.cpp 79 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007
  1. /**************************************************************************/
  2. /* particles_storage.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 "particles_storage.h"
  31. #include "servers/rendering/renderer_rd/renderer_compositor_rd.h"
  32. #include "servers/rendering/rendering_server_globals.h"
  33. #include "texture_storage.h"
  34. using namespace RendererRD;
  35. ParticlesStorage *ParticlesStorage::singleton = nullptr;
  36. ParticlesStorage *ParticlesStorage::get_singleton() {
  37. return singleton;
  38. }
  39. ParticlesStorage::ParticlesStorage() {
  40. singleton = this;
  41. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  42. /* Effects */
  43. sort_effects = memnew(SortEffects);
  44. /* Particles */
  45. {
  46. String defines = "#define SAMPLERS_BINDING_FIRST_INDEX " + itos(SAMPLERS_BINDING_FIRST_INDEX) + "\n";
  47. // Initialize particles
  48. Vector<String> particles_modes;
  49. particles_modes.push_back("");
  50. particles_shader.shader.initialize(particles_modes, defines);
  51. }
  52. MaterialStorage::get_singleton()->shader_set_data_request_function(MaterialStorage::SHADER_TYPE_PARTICLES, _create_particles_shader_funcs);
  53. MaterialStorage::get_singleton()->material_set_data_request_function(MaterialStorage::SHADER_TYPE_PARTICLES, _create_particles_material_funcs);
  54. {
  55. ShaderCompiler::DefaultIdentifierActions actions;
  56. actions.renames["COLOR"] = "PARTICLE.color";
  57. actions.renames["VELOCITY"] = "PARTICLE.velocity";
  58. actions.renames["MASS"] = "mass";
  59. actions.renames["ACTIVE"] = "particle_active";
  60. actions.renames["RESTART"] = "restart";
  61. actions.renames["CUSTOM"] = "PARTICLE.custom";
  62. actions.renames["AMOUNT_RATIO"] = "FRAME.amount_ratio";
  63. for (int i = 0; i < ParticlesShader::MAX_USERDATAS; i++) {
  64. String udname = "USERDATA" + itos(i + 1);
  65. actions.renames[udname] = "PARTICLE.userdata" + itos(i + 1);
  66. actions.usage_defines[udname] = "#define USERDATA" + itos(i + 1) + "_USED\n";
  67. }
  68. actions.renames["TRANSFORM"] = "PARTICLE.xform";
  69. actions.renames["TIME"] = "frame_history.data[0].time";
  70. actions.renames["PI"] = String::num(Math::PI);
  71. actions.renames["TAU"] = String::num(Math::TAU);
  72. actions.renames["E"] = String::num(Math::E);
  73. actions.renames["LIFETIME"] = "params.lifetime";
  74. actions.renames["DELTA"] = "local_delta";
  75. actions.renames["NUMBER"] = "particle_number";
  76. actions.renames["INDEX"] = "index";
  77. //actions.renames["GRAVITY"] = "current_gravity";
  78. actions.renames["EMISSION_TRANSFORM"] = "FRAME.emission_transform";
  79. actions.renames["EMITTER_VELOCITY"] = "FRAME.emitter_velocity";
  80. actions.renames["INTERPOLATE_TO_END"] = "FRAME.interp_to_end";
  81. actions.renames["RANDOM_SEED"] = "FRAME.random_seed";
  82. actions.renames["FLAG_EMIT_POSITION"] = "EMISSION_FLAG_HAS_POSITION";
  83. actions.renames["FLAG_EMIT_ROT_SCALE"] = "EMISSION_FLAG_HAS_ROTATION_SCALE";
  84. actions.renames["FLAG_EMIT_VELOCITY"] = "EMISSION_FLAG_HAS_VELOCITY";
  85. actions.renames["FLAG_EMIT_COLOR"] = "EMISSION_FLAG_HAS_COLOR";
  86. actions.renames["FLAG_EMIT_CUSTOM"] = "EMISSION_FLAG_HAS_CUSTOM";
  87. actions.renames["RESTART_POSITION"] = "restart_position";
  88. actions.renames["RESTART_ROT_SCALE"] = "restart_rotation_scale";
  89. actions.renames["RESTART_VELOCITY"] = "restart_velocity";
  90. actions.renames["RESTART_COLOR"] = "restart_color";
  91. actions.renames["RESTART_CUSTOM"] = "restart_custom";
  92. actions.renames["emit_subparticle"] = "emit_subparticle";
  93. actions.renames["COLLIDED"] = "collided";
  94. actions.renames["COLLISION_NORMAL"] = "collision_normal";
  95. actions.renames["COLLISION_DEPTH"] = "collision_depth";
  96. actions.renames["ATTRACTOR_FORCE"] = "attractor_force";
  97. actions.render_mode_defines["disable_force"] = "#define DISABLE_FORCE\n";
  98. actions.render_mode_defines["disable_velocity"] = "#define DISABLE_VELOCITY\n";
  99. actions.render_mode_defines["keep_data"] = "#define ENABLE_KEEP_DATA\n";
  100. actions.render_mode_defines["collision_use_scale"] = "#define USE_COLLISION_SCALE\n";
  101. actions.base_texture_binding_index = 1;
  102. actions.texture_layout_set = 3;
  103. actions.base_uniform_string = "material.";
  104. actions.base_varying_index = 10;
  105. actions.default_filter = ShaderLanguage::FILTER_LINEAR_MIPMAP;
  106. actions.default_repeat = ShaderLanguage::REPEAT_ENABLE;
  107. actions.global_buffer_array_variable = "global_shader_uniforms.data";
  108. particles_shader.compiler.initialize(actions);
  109. }
  110. {
  111. // default material and shader for particles shader
  112. particles_shader.default_shader = material_storage->shader_allocate();
  113. material_storage->shader_initialize(particles_shader.default_shader);
  114. material_storage->shader_set_code(particles_shader.default_shader, R"(
  115. // Default particles shader.
  116. shader_type particles;
  117. void process() {
  118. COLOR = vec4(1.0);
  119. }
  120. )");
  121. particles_shader.default_material = material_storage->material_allocate();
  122. material_storage->material_initialize(particles_shader.default_material);
  123. material_storage->material_set_shader(particles_shader.default_material, particles_shader.default_shader);
  124. ParticleProcessMaterialData *md = static_cast<ParticleProcessMaterialData *>(material_storage->material_get_data(particles_shader.default_material, MaterialStorage::SHADER_TYPE_PARTICLES));
  125. particles_shader.default_shader_rd = particles_shader.shader.version_get_shader(md->shader_data->version, 0);
  126. Vector<RD::Uniform> uniforms;
  127. {
  128. RD::Uniform u;
  129. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  130. u.binding = 2;
  131. u.append_id(material_storage->global_shader_uniforms_get_storage_buffer());
  132. uniforms.push_back(u);
  133. }
  134. material_storage->samplers_rd_get_default().append_uniforms(uniforms, SAMPLERS_BINDING_FIRST_INDEX);
  135. particles_shader.base_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, particles_shader.default_shader_rd, BASE_UNIFORM_SET);
  136. }
  137. {
  138. Vector<String> copy_modes;
  139. for (int i = 0; i <= ParticlesShader::MAX_USERDATAS; i++) {
  140. if (i == 0) {
  141. copy_modes.push_back("\n#define MODE_FILL_INSTANCES\n");
  142. copy_modes.push_back("\n#define MODE_FILL_SORT_BUFFER\n#define USE_SORT_BUFFER\n");
  143. copy_modes.push_back("\n#define MODE_FILL_INSTANCES\n#define USE_SORT_BUFFER\n");
  144. } else {
  145. copy_modes.push_back("\n#define MODE_FILL_INSTANCES\n#define USERDATA_COUNT " + itos(i) + "\n");
  146. copy_modes.push_back("\n#define MODE_FILL_SORT_BUFFER\n#define USE_SORT_BUFFER\n#define USERDATA_COUNT " + itos(i) + "\n");
  147. copy_modes.push_back("\n#define MODE_FILL_INSTANCES\n#define USE_SORT_BUFFER\n#define USERDATA_COUNT " + itos(i) + "\n");
  148. }
  149. }
  150. particles_shader.copy_shader.initialize(copy_modes);
  151. particles_shader.copy_shader_version = particles_shader.copy_shader.version_create();
  152. for (int i = 0; i <= ParticlesShader::MAX_USERDATAS; i++) {
  153. for (int j = 0; j < ParticlesShader::COPY_MODE_MAX; j++) {
  154. particles_shader.copy_pipelines[i * ParticlesShader::COPY_MODE_MAX + j] = RD::get_singleton()->compute_pipeline_create(particles_shader.copy_shader.version_get_shader(particles_shader.copy_shader_version, i * ParticlesShader::COPY_MODE_MAX + j));
  155. }
  156. }
  157. }
  158. }
  159. ParticlesStorage::~ParticlesStorage() {
  160. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  161. particles_shader.copy_shader.version_free(particles_shader.copy_shader_version);
  162. material_storage->material_free(particles_shader.default_material);
  163. material_storage->shader_free(particles_shader.default_shader);
  164. if (sort_effects) {
  165. memdelete(sort_effects);
  166. sort_effects = nullptr;
  167. }
  168. singleton = nullptr;
  169. }
  170. bool ParticlesStorage::free(RID p_rid) {
  171. if (owns_particles(p_rid)) {
  172. particles_free(p_rid);
  173. return true;
  174. } else if (owns_particles_collision(p_rid)) {
  175. particles_collision_free(p_rid);
  176. return true;
  177. } else if (owns_particles_collision_instance(p_rid)) {
  178. particles_collision_instance_free(p_rid);
  179. return true;
  180. }
  181. return false;
  182. }
  183. /* PARTICLES */
  184. RID ParticlesStorage::particles_allocate() {
  185. return particles_owner.allocate_rid();
  186. }
  187. void ParticlesStorage::particles_initialize(RID p_rid) {
  188. particles_owner.initialize_rid(p_rid);
  189. }
  190. void ParticlesStorage::particles_free(RID p_rid) {
  191. Particles *particles = particles_owner.get_or_null(p_rid);
  192. particles->dependency.deleted_notify(p_rid);
  193. particles->update_list.remove_from_list();
  194. _particles_free_data(particles);
  195. particles_owner.free(p_rid);
  196. }
  197. void ParticlesStorage::particles_set_mode(RID p_particles, RS::ParticlesMode p_mode) {
  198. Particles *particles = particles_owner.get_or_null(p_particles);
  199. ERR_FAIL_NULL(particles);
  200. if (particles->mode == p_mode) {
  201. return;
  202. }
  203. _particles_free_data(particles);
  204. particles->mode = p_mode;
  205. }
  206. void ParticlesStorage::particles_set_emitting(RID p_particles, bool p_emitting) {
  207. Particles *particles = particles_owner.get_or_null(p_particles);
  208. ERR_FAIL_NULL(particles);
  209. particles->emitting = p_emitting;
  210. }
  211. bool ParticlesStorage::particles_get_emitting(RID p_particles) {
  212. Particles *particles = particles_owner.get_or_null(p_particles);
  213. ERR_FAIL_NULL_V(particles, false);
  214. return particles->emitting;
  215. }
  216. void ParticlesStorage::_particles_free_data(Particles *particles) {
  217. if (particles->particle_buffer.is_valid()) {
  218. RD::get_singleton()->free(particles->particle_buffer);
  219. particles->particle_buffer = RID();
  220. RD::get_singleton()->free(particles->particle_instance_buffer);
  221. particles->particle_instance_buffer = RID();
  222. }
  223. particles->userdata_count = 0;
  224. if (particles->frame_params_buffer.is_valid()) {
  225. RD::get_singleton()->free(particles->frame_params_buffer);
  226. particles->frame_params_buffer = RID();
  227. }
  228. particles->particles_transforms_buffer_uniform_set = RID();
  229. if (RD::get_singleton()->uniform_set_is_valid(particles->trail_bind_pose_uniform_set)) {
  230. RD::get_singleton()->free(particles->trail_bind_pose_uniform_set);
  231. }
  232. particles->trail_bind_pose_uniform_set = RID();
  233. if (particles->trail_bind_pose_buffer.is_valid()) {
  234. RD::get_singleton()->free(particles->trail_bind_pose_buffer);
  235. particles->trail_bind_pose_buffer = RID();
  236. }
  237. if (RD::get_singleton()->uniform_set_is_valid(particles->collision_textures_uniform_set)) {
  238. RD::get_singleton()->free(particles->collision_textures_uniform_set);
  239. }
  240. particles->collision_textures_uniform_set = RID();
  241. if (particles->particles_sort_buffer.is_valid()) {
  242. RD::get_singleton()->free(particles->particles_sort_buffer);
  243. particles->particles_sort_buffer = RID();
  244. particles->particles_sort_uniform_set = RID();
  245. }
  246. if (particles->emission_buffer != nullptr) {
  247. particles->emission_buffer = nullptr;
  248. particles->emission_buffer_data.clear();
  249. RD::get_singleton()->free(particles->emission_storage_buffer);
  250. particles->emission_storage_buffer = RID();
  251. }
  252. if (particles->unused_emission_storage_buffer.is_valid()) {
  253. RD::get_singleton()->free(particles->unused_emission_storage_buffer);
  254. particles->unused_emission_storage_buffer = RID();
  255. }
  256. if (particles->unused_trail_storage_buffer.is_valid()) {
  257. RD::get_singleton()->free(particles->unused_trail_storage_buffer);
  258. particles->unused_trail_storage_buffer = RID();
  259. }
  260. if (RD::get_singleton()->uniform_set_is_valid(particles->particles_material_uniform_set)) {
  261. //will need to be re-created
  262. RD::get_singleton()->free(particles->particles_material_uniform_set);
  263. }
  264. particles->particles_material_uniform_set = RID();
  265. }
  266. void ParticlesStorage::particles_set_amount(RID p_particles, int p_amount) {
  267. Particles *particles = particles_owner.get_or_null(p_particles);
  268. ERR_FAIL_NULL(particles);
  269. if (particles->amount == p_amount) {
  270. return;
  271. }
  272. _particles_free_data(particles);
  273. particles->amount = p_amount;
  274. particles->prev_ticks = 0;
  275. particles->phase = 0;
  276. particles->prev_phase = 0;
  277. particles->clear = true;
  278. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_PARTICLES);
  279. }
  280. void ParticlesStorage::particles_set_amount_ratio(RID p_particles, float p_amount_ratio) {
  281. Particles *particles = particles_owner.get_or_null(p_particles);
  282. ERR_FAIL_NULL(particles);
  283. particles->amount_ratio = p_amount_ratio;
  284. }
  285. void ParticlesStorage::particles_set_lifetime(RID p_particles, double p_lifetime) {
  286. Particles *particles = particles_owner.get_or_null(p_particles);
  287. ERR_FAIL_NULL(particles);
  288. particles->lifetime = p_lifetime;
  289. }
  290. void ParticlesStorage::particles_set_one_shot(RID p_particles, bool p_one_shot) {
  291. Particles *particles = particles_owner.get_or_null(p_particles);
  292. ERR_FAIL_NULL(particles);
  293. particles->one_shot = p_one_shot;
  294. }
  295. void ParticlesStorage::particles_set_pre_process_time(RID p_particles, double p_time) {
  296. Particles *particles = particles_owner.get_or_null(p_particles);
  297. ERR_FAIL_NULL(particles);
  298. particles->pre_process_time = p_time;
  299. }
  300. void ParticlesStorage::particles_request_process_time(RID p_particles, real_t p_request_process_time) {
  301. Particles *particles = particles_owner.get_or_null(p_particles);
  302. ERR_FAIL_NULL(particles);
  303. particles->request_process_time = p_request_process_time;
  304. }
  305. void ParticlesStorage::particles_set_explosiveness_ratio(RID p_particles, real_t p_ratio) {
  306. Particles *particles = particles_owner.get_or_null(p_particles);
  307. ERR_FAIL_NULL(particles);
  308. particles->explosiveness = p_ratio;
  309. }
  310. void ParticlesStorage::particles_set_randomness_ratio(RID p_particles, real_t p_ratio) {
  311. Particles *particles = particles_owner.get_or_null(p_particles);
  312. ERR_FAIL_NULL(particles);
  313. particles->randomness = p_ratio;
  314. }
  315. void ParticlesStorage::particles_set_custom_aabb(RID p_particles, const AABB &p_aabb) {
  316. Particles *particles = particles_owner.get_or_null(p_particles);
  317. ERR_FAIL_NULL(particles);
  318. particles->custom_aabb = p_aabb;
  319. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_AABB);
  320. }
  321. void ParticlesStorage::particles_set_speed_scale(RID p_particles, double p_scale) {
  322. Particles *particles = particles_owner.get_or_null(p_particles);
  323. ERR_FAIL_NULL(particles);
  324. particles->speed_scale = p_scale;
  325. }
  326. void ParticlesStorage::particles_set_use_local_coordinates(RID p_particles, bool p_enable) {
  327. Particles *particles = particles_owner.get_or_null(p_particles);
  328. ERR_FAIL_NULL(particles);
  329. particles->use_local_coords = p_enable;
  330. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_PARTICLES);
  331. }
  332. void ParticlesStorage::particles_set_fixed_fps(RID p_particles, int p_fps) {
  333. Particles *particles = particles_owner.get_or_null(p_particles);
  334. ERR_FAIL_NULL(particles);
  335. particles->fixed_fps = p_fps;
  336. _particles_free_data(particles);
  337. particles->prev_ticks = 0;
  338. particles->phase = 0;
  339. particles->prev_phase = 0;
  340. particles->clear = true;
  341. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_PARTICLES);
  342. }
  343. void ParticlesStorage::particles_set_interpolate(RID p_particles, bool p_enable) {
  344. Particles *particles = particles_owner.get_or_null(p_particles);
  345. ERR_FAIL_NULL(particles);
  346. particles->interpolate = p_enable;
  347. }
  348. void ParticlesStorage::particles_set_fractional_delta(RID p_particles, bool p_enable) {
  349. Particles *particles = particles_owner.get_or_null(p_particles);
  350. ERR_FAIL_NULL(particles);
  351. particles->fractional_delta = p_enable;
  352. }
  353. void ParticlesStorage::particles_set_trails(RID p_particles, bool p_enable, double p_length) {
  354. Particles *particles = particles_owner.get_or_null(p_particles);
  355. ERR_FAIL_NULL(particles);
  356. ERR_FAIL_COND(p_length < 0.01 - CMP_EPSILON);
  357. p_length = MIN(10.0, p_length);
  358. particles->trails_enabled = p_enable;
  359. particles->trail_lifetime = p_length;
  360. _particles_free_data(particles);
  361. particles->prev_ticks = 0;
  362. particles->phase = 0;
  363. particles->prev_phase = 0;
  364. particles->clear = true;
  365. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_PARTICLES);
  366. }
  367. void ParticlesStorage::particles_set_trail_bind_poses(RID p_particles, const Vector<Transform3D> &p_bind_poses) {
  368. Particles *particles = particles_owner.get_or_null(p_particles);
  369. ERR_FAIL_NULL(particles);
  370. if (particles->trail_bind_pose_buffer.is_valid() && particles->trail_bind_poses.size() != p_bind_poses.size()) {
  371. _particles_free_data(particles);
  372. particles->prev_ticks = 0;
  373. particles->phase = 0;
  374. particles->prev_phase = 0;
  375. particles->clear = true;
  376. }
  377. particles->trail_bind_poses = p_bind_poses;
  378. particles->trail_bind_poses_dirty = true;
  379. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_PARTICLES);
  380. }
  381. void ParticlesStorage::particles_set_collision_base_size(RID p_particles, real_t p_size) {
  382. Particles *particles = particles_owner.get_or_null(p_particles);
  383. ERR_FAIL_NULL(particles);
  384. particles->collision_base_size = p_size;
  385. }
  386. void ParticlesStorage::particles_set_transform_align(RID p_particles, RS::ParticlesTransformAlign p_transform_align) {
  387. Particles *particles = particles_owner.get_or_null(p_particles);
  388. ERR_FAIL_NULL(particles);
  389. particles->transform_align = p_transform_align;
  390. }
  391. void ParticlesStorage::particles_set_process_material(RID p_particles, RID p_material) {
  392. Particles *particles = particles_owner.get_or_null(p_particles);
  393. ERR_FAIL_NULL(particles);
  394. particles->process_material = p_material;
  395. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_PARTICLES); //the instance buffer may have changed
  396. }
  397. RID ParticlesStorage::particles_get_process_material(RID p_particles) const {
  398. Particles *particles = particles_owner.get_or_null(p_particles);
  399. ERR_FAIL_NULL_V(particles, RID());
  400. return particles->process_material;
  401. }
  402. void ParticlesStorage::particles_set_draw_order(RID p_particles, RS::ParticlesDrawOrder p_order) {
  403. Particles *particles = particles_owner.get_or_null(p_particles);
  404. ERR_FAIL_NULL(particles);
  405. particles->draw_order = p_order;
  406. }
  407. void ParticlesStorage::particles_set_draw_passes(RID p_particles, int p_passes) {
  408. Particles *particles = particles_owner.get_or_null(p_particles);
  409. ERR_FAIL_NULL(particles);
  410. particles->draw_passes.resize(p_passes);
  411. }
  412. void ParticlesStorage::particles_set_draw_pass_mesh(RID p_particles, int p_pass, RID p_mesh) {
  413. Particles *particles = particles_owner.get_or_null(p_particles);
  414. ERR_FAIL_NULL(particles);
  415. ERR_FAIL_INDEX(p_pass, particles->draw_passes.size());
  416. particles->draw_passes.write[p_pass] = p_mesh;
  417. }
  418. void ParticlesStorage::particles_restart(RID p_particles) {
  419. Particles *particles = particles_owner.get_or_null(p_particles);
  420. ERR_FAIL_NULL(particles);
  421. particles->restart_request = true;
  422. }
  423. void ParticlesStorage::particles_set_seed(RID p_particles, uint32_t p_seed) {
  424. Particles *particles = particles_owner.get_or_null(p_particles);
  425. ERR_FAIL_NULL(particles);
  426. particles->random_seed = p_seed;
  427. }
  428. void ParticlesStorage::_particles_allocate_emission_buffer(Particles *particles) {
  429. ERR_FAIL_COND(particles->emission_buffer != nullptr);
  430. particles->emission_buffer_data.resize(sizeof(ParticleEmissionBuffer::Data) * particles->amount + sizeof(uint32_t) * 4);
  431. memset(particles->emission_buffer_data.ptrw(), 0, particles->emission_buffer_data.size());
  432. particles->emission_buffer = reinterpret_cast<ParticleEmissionBuffer *>(particles->emission_buffer_data.ptrw());
  433. particles->emission_buffer->particle_max = particles->amount;
  434. particles->emission_storage_buffer = RD::get_singleton()->storage_buffer_create(particles->emission_buffer_data.size(), particles->emission_buffer_data);
  435. if (RD::get_singleton()->uniform_set_is_valid(particles->particles_material_uniform_set)) {
  436. //will need to be re-created
  437. RD::get_singleton()->free(particles->particles_material_uniform_set);
  438. particles->particles_material_uniform_set = RID();
  439. }
  440. }
  441. void ParticlesStorage::_particles_ensure_unused_emission_buffer(Particles *particles) {
  442. if (particles->unused_emission_storage_buffer.is_null()) {
  443. // For rendering devices that do not support empty arrays (like C++),
  444. // we need to size the buffer with at least 1 element.
  445. particles->unused_emission_storage_buffer = RD::get_singleton()->storage_buffer_create(sizeof(ParticleEmissionBuffer));
  446. }
  447. }
  448. void ParticlesStorage::_particles_ensure_unused_trail_buffer(Particles *particles) {
  449. if (particles->unused_trail_storage_buffer.is_null()) {
  450. particles->unused_trail_storage_buffer = RD::get_singleton()->storage_buffer_create(16 * sizeof(float)); // Size of mat4.
  451. }
  452. }
  453. void ParticlesStorage::particles_set_subemitter(RID p_particles, RID p_subemitter_particles) {
  454. Particles *particles = particles_owner.get_or_null(p_particles);
  455. ERR_FAIL_NULL(particles);
  456. ERR_FAIL_COND(p_particles == p_subemitter_particles);
  457. particles->sub_emitter = p_subemitter_particles;
  458. if (RD::get_singleton()->uniform_set_is_valid(particles->particles_material_uniform_set)) {
  459. RD::get_singleton()->free(particles->particles_material_uniform_set);
  460. particles->particles_material_uniform_set = RID(); //clear and force to re create sub emitting
  461. }
  462. }
  463. void ParticlesStorage::particles_emit(RID p_particles, const Transform3D &p_transform, const Vector3 &p_velocity, const Color &p_color, const Color &p_custom, uint32_t p_emit_flags) {
  464. Particles *particles = particles_owner.get_or_null(p_particles);
  465. ERR_FAIL_NULL(particles);
  466. ERR_FAIL_COND(particles->amount == 0);
  467. if (particles->emitting) {
  468. particles->clear = true;
  469. particles->emitting = false;
  470. }
  471. if (particles->emission_buffer == nullptr) {
  472. _particles_allocate_emission_buffer(particles);
  473. }
  474. particles->inactive = false;
  475. particles->inactive_time = 0;
  476. int32_t idx = particles->emission_buffer->particle_count;
  477. if (idx < particles->emission_buffer->particle_max) {
  478. RendererRD::MaterialStorage::store_transform(p_transform, particles->emission_buffer->data[idx].xform);
  479. particles->emission_buffer->data[idx].velocity[0] = p_velocity.x;
  480. particles->emission_buffer->data[idx].velocity[1] = p_velocity.y;
  481. particles->emission_buffer->data[idx].velocity[2] = p_velocity.z;
  482. particles->emission_buffer->data[idx].custom[0] = p_custom.r;
  483. particles->emission_buffer->data[idx].custom[1] = p_custom.g;
  484. particles->emission_buffer->data[idx].custom[2] = p_custom.b;
  485. particles->emission_buffer->data[idx].custom[3] = p_custom.a;
  486. particles->emission_buffer->data[idx].color[0] = p_color.r;
  487. particles->emission_buffer->data[idx].color[1] = p_color.g;
  488. particles->emission_buffer->data[idx].color[2] = p_color.b;
  489. particles->emission_buffer->data[idx].color[3] = p_color.a;
  490. particles->emission_buffer->data[idx].flags = p_emit_flags;
  491. particles->emission_buffer->particle_count++;
  492. }
  493. }
  494. void ParticlesStorage::particles_request_process(RID p_particles) {
  495. Particles *particles = particles_owner.get_or_null(p_particles);
  496. ERR_FAIL_NULL(particles);
  497. if (!particles->dirty) {
  498. particles->dirty = true;
  499. if (!particles->update_list.in_list()) {
  500. particle_update_list.add(&particles->update_list);
  501. }
  502. }
  503. }
  504. AABB ParticlesStorage::particles_get_current_aabb(RID p_particles) {
  505. const Particles *particles = particles_owner.get_or_null(p_particles);
  506. ERR_FAIL_NULL_V(particles, AABB());
  507. int total_amount = particles->amount;
  508. if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  509. total_amount *= particles->trail_bind_poses.size();
  510. }
  511. uint32_t particle_data_size = sizeof(ParticleData) + sizeof(float) * 4 * particles->userdata_count;
  512. Vector<uint8_t> buffer = RD::get_singleton()->buffer_get_data(particles->particle_buffer);
  513. ERR_FAIL_COND_V(buffer.size() != (int)(total_amount * particle_data_size), AABB());
  514. Transform3D inv = particles->emission_transform.affine_inverse();
  515. AABB aabb;
  516. if (buffer.size()) {
  517. bool first = true;
  518. const uint8_t *data_ptr = (const uint8_t *)buffer.ptr();
  519. for (int i = 0; i < total_amount; i++) {
  520. const ParticleData &particle_data = *(const ParticleData *)&data_ptr[particle_data_size * i];
  521. if (particle_data.active) {
  522. Vector3 pos = Vector3(particle_data.xform[12], particle_data.xform[13], particle_data.xform[14]);
  523. if (!particles->use_local_coords) {
  524. pos = inv.xform(pos);
  525. }
  526. if (first) {
  527. aabb.position = pos;
  528. first = false;
  529. } else {
  530. aabb.expand_to(pos);
  531. }
  532. }
  533. }
  534. }
  535. float longest_axis_size = 0;
  536. for (int i = 0; i < particles->draw_passes.size(); i++) {
  537. if (particles->draw_passes[i].is_valid()) {
  538. AABB maabb = MeshStorage::get_singleton()->mesh_get_aabb(particles->draw_passes[i], RID());
  539. longest_axis_size = MAX(maabb.get_longest_axis_size(), longest_axis_size);
  540. }
  541. }
  542. aabb.grow_by(longest_axis_size);
  543. return aabb;
  544. }
  545. AABB ParticlesStorage::particles_get_aabb(RID p_particles) const {
  546. const Particles *particles = particles_owner.get_or_null(p_particles);
  547. ERR_FAIL_NULL_V(particles, AABB());
  548. return particles->custom_aabb;
  549. }
  550. void ParticlesStorage::particles_set_emission_transform(RID p_particles, const Transform3D &p_transform) {
  551. Particles *particles = particles_owner.get_or_null(p_particles);
  552. ERR_FAIL_NULL(particles);
  553. particles->emission_transform = p_transform;
  554. }
  555. void ParticlesStorage::particles_set_emitter_velocity(RID p_particles, const Vector3 &p_velocity) {
  556. Particles *particles = particles_owner.get_or_null(p_particles);
  557. ERR_FAIL_NULL(particles);
  558. particles->emitter_velocity = p_velocity;
  559. }
  560. void ParticlesStorage::particles_set_interp_to_end(RID p_particles, float p_interp) {
  561. Particles *particles = particles_owner.get_or_null(p_particles);
  562. ERR_FAIL_NULL(particles);
  563. particles->interp_to_end = p_interp;
  564. }
  565. int ParticlesStorage::particles_get_draw_passes(RID p_particles) const {
  566. const Particles *particles = particles_owner.get_or_null(p_particles);
  567. ERR_FAIL_NULL_V(particles, 0);
  568. return particles->draw_passes.size();
  569. }
  570. RID ParticlesStorage::particles_get_draw_pass_mesh(RID p_particles, int p_pass) const {
  571. const Particles *particles = particles_owner.get_or_null(p_particles);
  572. ERR_FAIL_NULL_V(particles, RID());
  573. ERR_FAIL_INDEX_V(p_pass, particles->draw_passes.size(), RID());
  574. return particles->draw_passes[p_pass];
  575. }
  576. void ParticlesStorage::particles_update_dependency(RID p_particles, DependencyTracker *p_instance) {
  577. Particles *particles = particles_owner.get_or_null(p_particles);
  578. ERR_FAIL_NULL(particles);
  579. p_instance->update_dependency(&particles->dependency);
  580. }
  581. void ParticlesStorage::particles_get_instance_buffer_motion_vectors_offsets(RID p_particles, uint32_t &r_current_offset, uint32_t &r_prev_offset) {
  582. Particles *particles = particles_owner.get_or_null(p_particles);
  583. ERR_FAIL_NULL(particles);
  584. r_current_offset = particles->instance_motion_vectors_current_offset;
  585. r_prev_offset = particles->instance_motion_vectors_previous_offset;
  586. }
  587. void ParticlesStorage::particles_add_collision(RID p_particles, RID p_particles_collision_instance) {
  588. Particles *particles = particles_owner.get_or_null(p_particles);
  589. ERR_FAIL_NULL(particles);
  590. particles->collisions.insert(p_particles_collision_instance);
  591. }
  592. void ParticlesStorage::particles_remove_collision(RID p_particles, RID p_particles_collision_instance) {
  593. Particles *particles = particles_owner.get_or_null(p_particles);
  594. ERR_FAIL_NULL(particles);
  595. particles->collisions.erase(p_particles_collision_instance);
  596. }
  597. void ParticlesStorage::particles_set_canvas_sdf_collision(RID p_particles, bool p_enable, const Transform2D &p_xform, const Rect2 &p_to_screen, RID p_texture) {
  598. Particles *particles = particles_owner.get_or_null(p_particles);
  599. ERR_FAIL_NULL(particles);
  600. particles->has_sdf_collision = p_enable;
  601. particles->sdf_collision_transform = p_xform;
  602. particles->sdf_collision_to_screen = p_to_screen;
  603. particles->sdf_collision_texture = p_texture;
  604. }
  605. void ParticlesStorage::_particles_process(Particles *p_particles, double p_delta) {
  606. TextureStorage *texture_storage = TextureStorage::get_singleton();
  607. MaterialStorage *material_storage = MaterialStorage::get_singleton();
  608. if (p_particles->particles_material_uniform_set.is_null() || !RD::get_singleton()->uniform_set_is_valid(p_particles->particles_material_uniform_set)) {
  609. thread_local LocalVector<RD::Uniform> uniforms;
  610. uniforms.clear();
  611. {
  612. RD::Uniform u;
  613. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  614. u.binding = 0;
  615. u.append_id(p_particles->frame_params_buffer);
  616. uniforms.push_back(u);
  617. }
  618. {
  619. RD::Uniform u;
  620. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  621. u.binding = 1;
  622. u.append_id(p_particles->particle_buffer);
  623. uniforms.push_back(u);
  624. }
  625. {
  626. RD::Uniform u;
  627. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  628. u.binding = 2;
  629. if (p_particles->emission_storage_buffer.is_valid()) {
  630. u.append_id(p_particles->emission_storage_buffer);
  631. } else {
  632. _particles_ensure_unused_emission_buffer(p_particles);
  633. u.append_id(p_particles->unused_emission_storage_buffer);
  634. }
  635. uniforms.push_back(u);
  636. }
  637. {
  638. RD::Uniform u;
  639. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  640. u.binding = 3;
  641. Particles *sub_emitter = particles_owner.get_or_null(p_particles->sub_emitter);
  642. if (sub_emitter) {
  643. if (sub_emitter->emission_buffer == nullptr) { //no emission buffer, allocate emission buffer
  644. _particles_allocate_emission_buffer(sub_emitter);
  645. }
  646. u.append_id(sub_emitter->emission_storage_buffer);
  647. } else {
  648. _particles_ensure_unused_emission_buffer(p_particles);
  649. u.append_id(p_particles->unused_emission_storage_buffer);
  650. }
  651. uniforms.push_back(u);
  652. }
  653. p_particles->particles_material_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, particles_shader.default_shader_rd, 1);
  654. }
  655. double new_phase = Math::fmod((double)p_particles->phase + (p_delta / p_particles->lifetime), 1.0);
  656. //move back history (if there is any)
  657. for (uint32_t i = p_particles->frame_history.size() - 1; i > 0; i--) {
  658. p_particles->frame_history[i] = p_particles->frame_history[i - 1];
  659. }
  660. //update current frame
  661. ParticlesFrameParams &frame_params = p_particles->frame_history[0];
  662. if (p_particles->clear) {
  663. p_particles->cycle_number = 0;
  664. } else if (new_phase < p_particles->phase) {
  665. if (p_particles->one_shot) {
  666. p_particles->emitting = false;
  667. }
  668. p_particles->cycle_number++;
  669. }
  670. frame_params.emitting = p_particles->emitting;
  671. frame_params.system_phase = new_phase;
  672. frame_params.prev_system_phase = p_particles->phase;
  673. p_particles->phase = new_phase;
  674. frame_params.time = RendererCompositorRD::get_singleton()->get_total_time();
  675. frame_params.delta = p_delta;
  676. frame_params.random_seed = p_particles->random_seed;
  677. frame_params.explosiveness = p_particles->explosiveness;
  678. frame_params.randomness = p_particles->randomness;
  679. if (p_particles->use_local_coords) {
  680. RendererRD::MaterialStorage::store_transform(Transform3D(), frame_params.emission_transform);
  681. } else {
  682. RendererRD::MaterialStorage::store_transform(p_particles->emission_transform, frame_params.emission_transform);
  683. }
  684. frame_params.cycle = p_particles->cycle_number;
  685. frame_params.frame = p_particles->frame_counter++;
  686. frame_params.amount_ratio = p_particles->amount_ratio;
  687. frame_params.pad1 = 0;
  688. frame_params.pad2 = 0;
  689. frame_params.emitter_velocity[0] = p_particles->emitter_velocity.x;
  690. frame_params.emitter_velocity[1] = p_particles->emitter_velocity.y;
  691. frame_params.emitter_velocity[2] = p_particles->emitter_velocity.z;
  692. frame_params.interp_to_end = p_particles->interp_to_end;
  693. { //collision and attractors
  694. frame_params.collider_count = 0;
  695. frame_params.attractor_count = 0;
  696. frame_params.particle_size = p_particles->collision_base_size;
  697. RID collision_3d_textures[ParticlesFrameParams::MAX_3D_TEXTURES];
  698. RID collision_heightmap_texture;
  699. Transform3D to_particles;
  700. if (p_particles->use_local_coords) {
  701. to_particles = p_particles->emission_transform.affine_inverse();
  702. }
  703. if (p_particles->has_sdf_collision && RD::get_singleton()->texture_is_valid(p_particles->sdf_collision_texture)) {
  704. //2D collision
  705. Transform2D xform = p_particles->sdf_collision_transform; //will use dotproduct manually so invert beforehand
  706. if (!p_particles->use_local_coords) {
  707. Transform2D emission;
  708. emission.columns[0] = Vector2(p_particles->emission_transform.basis.get_column(0).x, p_particles->emission_transform.basis.get_column(0).y);
  709. emission.columns[1] = Vector2(p_particles->emission_transform.basis.get_column(1).x, p_particles->emission_transform.basis.get_column(1).y);
  710. emission.set_origin(Vector2(p_particles->emission_transform.origin.x, p_particles->emission_transform.origin.y));
  711. xform = xform * emission.affine_inverse();
  712. }
  713. Transform2D revert = xform.affine_inverse();
  714. frame_params.collider_count = 1;
  715. frame_params.colliders[0].transform[0] = xform.columns[0][0];
  716. frame_params.colliders[0].transform[1] = xform.columns[0][1];
  717. frame_params.colliders[0].transform[2] = 0;
  718. frame_params.colliders[0].transform[3] = xform.columns[2][0];
  719. frame_params.colliders[0].transform[4] = xform.columns[1][0];
  720. frame_params.colliders[0].transform[5] = xform.columns[1][1];
  721. frame_params.colliders[0].transform[6] = 0;
  722. frame_params.colliders[0].transform[7] = xform.columns[2][1];
  723. frame_params.colliders[0].transform[8] = revert.columns[0][0];
  724. frame_params.colliders[0].transform[9] = revert.columns[0][1];
  725. frame_params.colliders[0].transform[10] = 0;
  726. frame_params.colliders[0].transform[11] = revert.columns[2][0];
  727. frame_params.colliders[0].transform[12] = revert.columns[1][0];
  728. frame_params.colliders[0].transform[13] = revert.columns[1][1];
  729. frame_params.colliders[0].transform[14] = 0;
  730. frame_params.colliders[0].transform[15] = revert.columns[2][1];
  731. frame_params.colliders[0].extents[0] = p_particles->sdf_collision_to_screen.size.x;
  732. frame_params.colliders[0].extents[1] = p_particles->sdf_collision_to_screen.size.y;
  733. frame_params.colliders[0].extents[2] = p_particles->sdf_collision_to_screen.position.x;
  734. frame_params.colliders[0].scale = p_particles->sdf_collision_to_screen.position.y;
  735. frame_params.colliders[0].texture_index = 0;
  736. frame_params.colliders[0].type = ParticlesFrameParams::COLLISION_TYPE_2D_SDF;
  737. collision_heightmap_texture = p_particles->sdf_collision_texture;
  738. //replace in all other history frames where used because parameters are no longer valid if screen moves
  739. for (ParticlesFrameParams &params : p_particles->frame_history) {
  740. if (params.collider_count > 0 && params.colliders[0].type == ParticlesFrameParams::COLLISION_TYPE_2D_SDF) {
  741. params.colliders[0] = frame_params.colliders[0];
  742. }
  743. }
  744. }
  745. uint32_t collision_3d_textures_used = 0;
  746. for (const RID &E : p_particles->collisions) {
  747. ParticlesCollisionInstance *pci = particles_collision_instance_owner.get_or_null(E);
  748. if (!pci || !pci->active) {
  749. continue;
  750. }
  751. ParticlesCollision *pc = particles_collision_owner.get_or_null(pci->collision);
  752. ERR_CONTINUE(!pc);
  753. Transform3D to_collider = pci->transform;
  754. if (p_particles->use_local_coords) {
  755. to_collider = to_particles * to_collider;
  756. }
  757. Vector3 scale = to_collider.basis.get_scale();
  758. to_collider.basis.orthonormalize();
  759. if (pc->type <= RS::PARTICLES_COLLISION_TYPE_VECTOR_FIELD_ATTRACT) {
  760. //attractor
  761. if (frame_params.attractor_count >= ParticlesFrameParams::MAX_ATTRACTORS) {
  762. continue;
  763. }
  764. ParticlesFrameParams::Attractor &attr = frame_params.attractors[frame_params.attractor_count];
  765. RendererRD::MaterialStorage::store_transform(to_collider, attr.transform);
  766. attr.strength = pc->attractor_strength;
  767. attr.attenuation = pc->attractor_attenuation;
  768. attr.directionality = pc->attractor_directionality;
  769. switch (pc->type) {
  770. case RS::PARTICLES_COLLISION_TYPE_SPHERE_ATTRACT: {
  771. attr.type = ParticlesFrameParams::ATTRACTOR_TYPE_SPHERE;
  772. float radius = pc->radius;
  773. radius *= (scale.x + scale.y + scale.z) / 3.0;
  774. attr.extents[0] = radius;
  775. attr.extents[1] = radius;
  776. attr.extents[2] = radius;
  777. } break;
  778. case RS::PARTICLES_COLLISION_TYPE_BOX_ATTRACT: {
  779. attr.type = ParticlesFrameParams::ATTRACTOR_TYPE_BOX;
  780. Vector3 extents = pc->extents * scale;
  781. attr.extents[0] = extents.x;
  782. attr.extents[1] = extents.y;
  783. attr.extents[2] = extents.z;
  784. } break;
  785. case RS::PARTICLES_COLLISION_TYPE_VECTOR_FIELD_ATTRACT: {
  786. if (collision_3d_textures_used >= ParticlesFrameParams::MAX_3D_TEXTURES) {
  787. continue;
  788. }
  789. attr.type = ParticlesFrameParams::ATTRACTOR_TYPE_VECTOR_FIELD;
  790. Vector3 extents = pc->extents * scale;
  791. attr.extents[0] = extents.x;
  792. attr.extents[1] = extents.y;
  793. attr.extents[2] = extents.z;
  794. attr.texture_index = collision_3d_textures_used;
  795. collision_3d_textures[collision_3d_textures_used] = pc->field_texture;
  796. collision_3d_textures_used++;
  797. } break;
  798. default: {
  799. }
  800. }
  801. frame_params.attractor_count++;
  802. } else {
  803. //collider
  804. if (frame_params.collider_count >= ParticlesFrameParams::MAX_COLLIDERS) {
  805. continue;
  806. }
  807. ParticlesFrameParams::Collider &col = frame_params.colliders[frame_params.collider_count];
  808. RendererRD::MaterialStorage::store_transform(to_collider, col.transform);
  809. switch (pc->type) {
  810. case RS::PARTICLES_COLLISION_TYPE_SPHERE_COLLIDE: {
  811. col.type = ParticlesFrameParams::COLLISION_TYPE_SPHERE;
  812. float radius = pc->radius;
  813. radius *= (scale.x + scale.y + scale.z) / 3.0;
  814. col.extents[0] = radius;
  815. col.extents[1] = radius;
  816. col.extents[2] = radius;
  817. } break;
  818. case RS::PARTICLES_COLLISION_TYPE_BOX_COLLIDE: {
  819. col.type = ParticlesFrameParams::COLLISION_TYPE_BOX;
  820. Vector3 extents = pc->extents * scale;
  821. col.extents[0] = extents.x;
  822. col.extents[1] = extents.y;
  823. col.extents[2] = extents.z;
  824. } break;
  825. case RS::PARTICLES_COLLISION_TYPE_SDF_COLLIDE: {
  826. if (collision_3d_textures_used >= ParticlesFrameParams::MAX_3D_TEXTURES) {
  827. continue;
  828. }
  829. col.type = ParticlesFrameParams::COLLISION_TYPE_SDF;
  830. Vector3 extents = pc->extents * scale;
  831. col.extents[0] = extents.x;
  832. col.extents[1] = extents.y;
  833. col.extents[2] = extents.z;
  834. col.texture_index = collision_3d_textures_used;
  835. col.scale = (scale.x + scale.y + scale.z) * 0.333333333333; //non uniform scale non supported
  836. collision_3d_textures[collision_3d_textures_used] = pc->field_texture;
  837. collision_3d_textures_used++;
  838. } break;
  839. case RS::PARTICLES_COLLISION_TYPE_HEIGHTFIELD_COLLIDE: {
  840. if (collision_heightmap_texture != RID()) { //already taken
  841. continue;
  842. }
  843. col.type = ParticlesFrameParams::COLLISION_TYPE_HEIGHT_FIELD;
  844. Vector3 extents = pc->extents * scale;
  845. col.extents[0] = extents.x;
  846. col.extents[1] = extents.y;
  847. col.extents[2] = extents.z;
  848. collision_heightmap_texture = pc->heightfield_texture;
  849. } break;
  850. default: {
  851. }
  852. }
  853. frame_params.collider_count++;
  854. }
  855. }
  856. bool different = false;
  857. if (collision_3d_textures_used == p_particles->collision_3d_textures_used) {
  858. for (int i = 0; i < ParticlesFrameParams::MAX_3D_TEXTURES; i++) {
  859. if (p_particles->collision_3d_textures[i] != collision_3d_textures[i]) {
  860. different = true;
  861. break;
  862. }
  863. }
  864. }
  865. if (collision_heightmap_texture != p_particles->collision_heightmap_texture) {
  866. different = true;
  867. }
  868. bool uniform_set_valid = RD::get_singleton()->uniform_set_is_valid(p_particles->collision_textures_uniform_set);
  869. if (different || !uniform_set_valid) {
  870. if (uniform_set_valid) {
  871. RD::get_singleton()->free(p_particles->collision_textures_uniform_set);
  872. }
  873. thread_local LocalVector<RD::Uniform> uniforms;
  874. uniforms.clear();
  875. {
  876. RD::Uniform u;
  877. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  878. u.binding = 0;
  879. for (uint32_t i = 0; i < ParticlesFrameParams::MAX_3D_TEXTURES; i++) {
  880. RID rd_tex;
  881. if (i < collision_3d_textures_used) {
  882. if (TextureStorage::get_singleton()->texture_get_type(collision_3d_textures[i]) == TextureStorage::TYPE_3D) {
  883. rd_tex = TextureStorage::get_singleton()->texture_get_rd_texture(collision_3d_textures[i]);
  884. }
  885. }
  886. if (rd_tex == RID()) {
  887. rd_tex = texture_storage->texture_rd_get_default(TextureStorage::DEFAULT_RD_TEXTURE_3D_WHITE);
  888. }
  889. u.append_id(rd_tex);
  890. }
  891. uniforms.push_back(u);
  892. }
  893. {
  894. RD::Uniform u;
  895. u.uniform_type = RD::UNIFORM_TYPE_TEXTURE;
  896. u.binding = 1;
  897. if (collision_heightmap_texture.is_valid()) {
  898. u.append_id(collision_heightmap_texture);
  899. } else {
  900. u.append_id(texture_storage->texture_rd_get_default(TextureStorage::DEFAULT_RD_TEXTURE_BLACK));
  901. }
  902. uniforms.push_back(u);
  903. }
  904. p_particles->collision_textures_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, particles_shader.default_shader_rd, 2);
  905. p_particles->collision_heightmap_texture = collision_heightmap_texture;
  906. }
  907. }
  908. ParticlesShader::PushConstant push_constant;
  909. int process_amount = p_particles->amount;
  910. if (p_particles->trails_enabled && p_particles->trail_bind_poses.size() > 1) {
  911. process_amount *= p_particles->trail_bind_poses.size();
  912. }
  913. push_constant.clear = p_particles->clear;
  914. push_constant.total_particles = p_particles->amount;
  915. push_constant.lifetime = p_particles->lifetime;
  916. push_constant.trail_size = p_particles->trail_params.size();
  917. push_constant.use_fractional_delta = p_particles->fractional_delta;
  918. push_constant.sub_emitter_mode = !p_particles->emitting && p_particles->emission_buffer && (p_particles->emission_buffer->particle_count > 0 || p_particles->force_sub_emit);
  919. push_constant.trail_pass = false;
  920. p_particles->force_sub_emit = false; //reset
  921. Particles *sub_emitter = particles_owner.get_or_null(p_particles->sub_emitter);
  922. if (sub_emitter && sub_emitter->emission_storage_buffer.is_valid()) {
  923. // print_line("updating subemitter buffer");
  924. int32_t zero[4] = { 0, sub_emitter->amount, 0, 0 };
  925. RD::get_singleton()->buffer_update(sub_emitter->emission_storage_buffer, 0, sizeof(uint32_t) * 4, zero);
  926. push_constant.can_emit = true;
  927. if (sub_emitter->emitting) {
  928. sub_emitter->emitting = false;
  929. sub_emitter->clear = true; //will need to clear if it was emitting, sorry
  930. }
  931. //make sure the sub emitter processes particles too
  932. sub_emitter->inactive = false;
  933. sub_emitter->inactive_time = 0;
  934. sub_emitter->force_sub_emit = true;
  935. } else {
  936. push_constant.can_emit = false;
  937. }
  938. if (p_particles->emission_buffer && p_particles->emission_buffer->particle_count) {
  939. RD::get_singleton()->buffer_update(p_particles->emission_storage_buffer, 0, sizeof(uint32_t) * 4 + sizeof(ParticleEmissionBuffer::Data) * p_particles->emission_buffer->particle_count, p_particles->emission_buffer);
  940. p_particles->emission_buffer->particle_count = 0;
  941. }
  942. p_particles->clear = false;
  943. if (p_particles->trail_params.size() > 1) {
  944. //fill the trail params
  945. for (uint32_t i = 0; i < p_particles->trail_params.size(); i++) {
  946. uint32_t src_idx = i * p_particles->frame_history.size() / p_particles->trail_params.size();
  947. if (p_particles->speed_scale <= 0.0) {
  948. // Stop trails.
  949. src_idx = 0;
  950. }
  951. p_particles->trail_params[i] = p_particles->frame_history[src_idx];
  952. }
  953. } else {
  954. p_particles->trail_params[0] = p_particles->frame_history[0];
  955. }
  956. RD::get_singleton()->buffer_update(p_particles->frame_params_buffer, 0, sizeof(ParticlesFrameParams) * p_particles->trail_params.size(), p_particles->trail_params.ptr());
  957. ParticleProcessMaterialData *m = static_cast<ParticleProcessMaterialData *>(material_storage->material_get_data(p_particles->process_material, MaterialStorage::SHADER_TYPE_PARTICLES));
  958. if (!m) {
  959. m = static_cast<ParticleProcessMaterialData *>(material_storage->material_get_data(particles_shader.default_material, MaterialStorage::SHADER_TYPE_PARTICLES));
  960. }
  961. ERR_FAIL_NULL(m);
  962. p_particles->has_collision_cache = m->shader_data->uses_collision;
  963. //todo should maybe compute all particle systems together?
  964. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  965. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, m->shader_data->pipeline);
  966. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles_shader.base_uniform_set, BASE_UNIFORM_SET);
  967. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, p_particles->particles_material_uniform_set, MATERIAL_UNIFORM_SET);
  968. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, p_particles->collision_textures_uniform_set, COLLISION_TEXTURTES_UNIFORM_SET);
  969. if (m->uniform_set.is_valid() && RD::get_singleton()->uniform_set_is_valid(m->uniform_set)) {
  970. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, m->uniform_set, 3);
  971. m->set_as_used();
  972. }
  973. RD::get_singleton()->compute_list_set_push_constant(compute_list, &push_constant, sizeof(ParticlesShader::PushConstant));
  974. if (p_particles->trails_enabled && p_particles->trail_bind_poses.size() > 1) {
  975. //trails requires two passes in order to catch particle starts
  976. RD::get_singleton()->compute_list_dispatch_threads(compute_list, process_amount / p_particles->trail_bind_poses.size(), 1, 1);
  977. RD::get_singleton()->compute_list_add_barrier(compute_list);
  978. push_constant.trail_pass = true;
  979. RD::get_singleton()->compute_list_set_push_constant(compute_list, &push_constant, sizeof(ParticlesShader::PushConstant));
  980. RD::get_singleton()->compute_list_dispatch_threads(compute_list, process_amount - p_particles->amount, 1, 1);
  981. } else {
  982. RD::get_singleton()->compute_list_dispatch_threads(compute_list, process_amount, 1, 1);
  983. }
  984. RD::get_singleton()->compute_list_end();
  985. }
  986. void ParticlesStorage::particles_set_view_axis(RID p_particles, const Vector3 &p_axis, const Vector3 &p_up_axis) {
  987. Particles *particles = particles_owner.get_or_null(p_particles);
  988. ERR_FAIL_NULL(particles);
  989. if (particles->draw_order != RS::PARTICLES_DRAW_ORDER_VIEW_DEPTH && particles->transform_align != RS::PARTICLES_TRANSFORM_ALIGN_Z_BILLBOARD && particles->transform_align != RS::PARTICLES_TRANSFORM_ALIGN_Z_BILLBOARD_Y_TO_VELOCITY) {
  990. return;
  991. }
  992. if (particles->particle_buffer.is_null() || particles->trail_bind_pose_uniform_set.is_null()) {
  993. return; //particles have not processed yet
  994. }
  995. bool do_sort = particles->draw_order == RS::PARTICLES_DRAW_ORDER_VIEW_DEPTH;
  996. //copy to sort buffer
  997. if (do_sort && particles->particles_sort_buffer == RID()) {
  998. uint32_t size = particles->amount;
  999. if (size & 1) {
  1000. size++; //make multiple of 16
  1001. }
  1002. size *= sizeof(float) * 2;
  1003. particles->particles_sort_buffer = RD::get_singleton()->storage_buffer_create(size);
  1004. {
  1005. Vector<RD::Uniform> uniforms;
  1006. {
  1007. RD::Uniform u;
  1008. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  1009. u.binding = 0;
  1010. u.append_id(particles->particles_sort_buffer);
  1011. uniforms.push_back(u);
  1012. }
  1013. particles->particles_sort_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, particles_shader.copy_shader.version_get_shader(particles_shader.copy_shader_version, ParticlesShader::COPY_MODE_FILL_SORT_BUFFER), 1);
  1014. }
  1015. }
  1016. ParticlesShader::CopyPushConstant copy_push_constant;
  1017. if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  1018. int fixed_fps = 60.0;
  1019. if (particles->fixed_fps > 0) {
  1020. fixed_fps = particles->fixed_fps;
  1021. }
  1022. copy_push_constant.trail_size = particles->trail_bind_poses.size();
  1023. copy_push_constant.trail_total = particles->frame_history.size();
  1024. copy_push_constant.frame_delta = 1.0 / fixed_fps;
  1025. } else {
  1026. copy_push_constant.trail_size = 1;
  1027. copy_push_constant.trail_total = 1;
  1028. copy_push_constant.frame_delta = 0.0;
  1029. }
  1030. copy_push_constant.order_by_lifetime = (particles->draw_order == RS::PARTICLES_DRAW_ORDER_LIFETIME || particles->draw_order == RS::PARTICLES_DRAW_ORDER_REVERSE_LIFETIME);
  1031. copy_push_constant.lifetime_split = (MIN(int(particles->amount * particles->phase), particles->amount - 1) + 1) % particles->amount;
  1032. copy_push_constant.lifetime_reverse = particles->draw_order == RS::PARTICLES_DRAW_ORDER_REVERSE_LIFETIME;
  1033. copy_push_constant.motion_vectors_current_offset = particles->instance_motion_vectors_current_offset;
  1034. copy_push_constant.frame_remainder = particles->interpolate ? particles->frame_remainder : 0.0;
  1035. copy_push_constant.total_particles = particles->amount;
  1036. copy_push_constant.copy_mode_2d = false;
  1037. Vector3 axis = -p_axis; // cameras look to z negative
  1038. if (particles->use_local_coords) {
  1039. axis = particles->emission_transform.basis.xform_inv(axis).normalized();
  1040. }
  1041. copy_push_constant.sort_direction[0] = axis.x;
  1042. copy_push_constant.sort_direction[1] = axis.y;
  1043. copy_push_constant.sort_direction[2] = axis.z;
  1044. copy_push_constant.align_up[0] = p_up_axis.x;
  1045. copy_push_constant.align_up[1] = p_up_axis.y;
  1046. copy_push_constant.align_up[2] = p_up_axis.z;
  1047. copy_push_constant.align_mode = particles->transform_align;
  1048. if (do_sort) {
  1049. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  1050. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, particles_shader.copy_pipelines[ParticlesShader::COPY_MODE_FILL_SORT_BUFFER + particles->userdata_count * ParticlesShader::COPY_MODE_MAX]);
  1051. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->particles_copy_uniform_set, 0);
  1052. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->particles_sort_uniform_set, 1);
  1053. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->trail_bind_pose_uniform_set, 2);
  1054. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy_push_constant, sizeof(ParticlesShader::CopyPushConstant));
  1055. RD::get_singleton()->compute_list_dispatch_threads(compute_list, particles->amount, 1, 1);
  1056. RD::get_singleton()->compute_list_end();
  1057. sort_effects->sort_buffer(particles->particles_sort_uniform_set, particles->amount);
  1058. }
  1059. if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  1060. copy_push_constant.total_particles *= particles->trail_bind_poses.size();
  1061. }
  1062. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  1063. uint32_t copy_pipeline = do_sort ? ParticlesShader::COPY_MODE_FILL_INSTANCES_WITH_SORT_BUFFER : ParticlesShader::COPY_MODE_FILL_INSTANCES;
  1064. copy_pipeline += particles->userdata_count * ParticlesShader::COPY_MODE_MAX;
  1065. copy_push_constant.copy_mode_2d = particles->mode == RS::PARTICLES_MODE_2D ? 1 : 0;
  1066. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, particles_shader.copy_pipelines[copy_pipeline]);
  1067. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->particles_copy_uniform_set, 0);
  1068. if (do_sort) {
  1069. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->particles_sort_uniform_set, 1);
  1070. }
  1071. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->trail_bind_pose_uniform_set, 2);
  1072. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy_push_constant, sizeof(ParticlesShader::CopyPushConstant));
  1073. RD::get_singleton()->compute_list_dispatch_threads(compute_list, copy_push_constant.total_particles, 1, 1);
  1074. RD::get_singleton()->compute_list_end();
  1075. }
  1076. void ParticlesStorage::_particles_update_buffers(Particles *particles) {
  1077. uint32_t userdata_count = 0;
  1078. if (particles->process_material.is_valid()) {
  1079. ParticleProcessMaterialData *material_data = static_cast<ParticleProcessMaterialData *>(MaterialStorage::get_singleton()->material_get_data(particles->process_material, MaterialStorage::SHADER_TYPE_PARTICLES));
  1080. if (material_data && material_data->shader_data->version.is_valid() && material_data->shader_data->valid) {
  1081. userdata_count = material_data->shader_data->userdata_count;
  1082. }
  1083. }
  1084. bool uses_motion_vectors = RSG::viewport->get_num_viewports_with_motion_vectors() > 0 || (RendererCompositorStorage::get_singleton()->get_num_compositor_effects_with_motion_vectors() > 0);
  1085. bool index_draw_order = particles->draw_order == RS::ParticlesDrawOrder::PARTICLES_DRAW_ORDER_INDEX;
  1086. bool enable_motion_vectors = uses_motion_vectors && index_draw_order && !particles->instance_motion_vectors_enabled;
  1087. bool only_instances_changed = false;
  1088. if (userdata_count != particles->userdata_count) {
  1089. // Mismatch userdata, re-create all buffers.
  1090. _particles_free_data(particles);
  1091. } else if (enable_motion_vectors) {
  1092. // Only motion vectors are required, release the transforms buffer and uniform set.
  1093. if (particles->particle_instance_buffer.is_valid()) {
  1094. RD::get_singleton()->free(particles->particle_instance_buffer);
  1095. particles->particle_instance_buffer = RID();
  1096. }
  1097. particles->particles_transforms_buffer_uniform_set = RID();
  1098. only_instances_changed = true;
  1099. } else if (!particles->particle_buffer.is_null()) {
  1100. // No operation is required because a buffer already exists, return early.
  1101. return;
  1102. }
  1103. if (particles->amount > 0) {
  1104. int total_amount = particles->amount;
  1105. if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  1106. total_amount *= particles->trail_bind_poses.size();
  1107. }
  1108. uint32_t xform_size = particles->mode == RS::PARTICLES_MODE_2D ? 2 : 3;
  1109. if (particles->particle_buffer.is_null()) {
  1110. particles->particle_buffer = RD::get_singleton()->storage_buffer_create((sizeof(ParticleData) + userdata_count * sizeof(float) * 4) * total_amount);
  1111. particles->userdata_count = userdata_count;
  1112. }
  1113. PackedByteArray data;
  1114. uint32_t particle_instance_buffer_size = total_amount * (xform_size + 1 + 1) * sizeof(float) * 4;
  1115. if (uses_motion_vectors) {
  1116. particle_instance_buffer_size *= 2;
  1117. particles->instance_motion_vectors_enabled = true;
  1118. }
  1119. data.resize_initialized(particle_instance_buffer_size);
  1120. particles->particle_instance_buffer = RD::get_singleton()->storage_buffer_create(particle_instance_buffer_size, data);
  1121. {
  1122. Vector<RD::Uniform> uniforms;
  1123. {
  1124. RD::Uniform u;
  1125. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  1126. u.binding = 1;
  1127. u.append_id(particles->particle_buffer);
  1128. uniforms.push_back(u);
  1129. }
  1130. {
  1131. RD::Uniform u;
  1132. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  1133. u.binding = 2;
  1134. u.append_id(particles->particle_instance_buffer);
  1135. uniforms.push_back(u);
  1136. }
  1137. particles->particles_copy_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, particles_shader.copy_shader.version_get_shader(particles_shader.copy_shader_version, 0), 0);
  1138. }
  1139. particles->instance_motion_vectors_current_offset = 0;
  1140. particles->instance_motion_vectors_previous_offset = 0;
  1141. particles->instance_motion_vectors_last_change = -1;
  1142. if (only_instances_changed) {
  1143. // Notify the renderer the instances uniform must be retrieved again, as it's the only element that has been changed because motion vectors were enabled.
  1144. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_PARTICLES_INSTANCES);
  1145. }
  1146. }
  1147. }
  1148. void ParticlesStorage::update_particles() {
  1149. if (!particle_update_list.first()) {
  1150. return;
  1151. }
  1152. RENDER_TIMESTAMP("Update GPUParticles");
  1153. uint32_t frame = RSG::rasterizer->get_frame_number();
  1154. bool uses_motion_vectors = RSG::viewport->get_num_viewports_with_motion_vectors() > 0 || (RendererCompositorStorage::get_singleton()->get_num_compositor_effects_with_motion_vectors() > 0);
  1155. while (particle_update_list.first()) {
  1156. //use transform feedback to process particles
  1157. Particles *particles = particle_update_list.first()->self();
  1158. particles->update_list.remove_from_list();
  1159. particles->dirty = false;
  1160. _particles_update_buffers(particles);
  1161. if (particles->restart_request) {
  1162. particles->prev_ticks = 0;
  1163. particles->phase = 0;
  1164. particles->prev_phase = 0;
  1165. particles->clear = true;
  1166. particles->restart_request = false;
  1167. }
  1168. if (particles->inactive && !particles->emitting) {
  1169. //go next
  1170. continue;
  1171. }
  1172. if (particles->emitting) {
  1173. if (particles->inactive) {
  1174. //restart system from scratch
  1175. particles->prev_ticks = 0;
  1176. particles->phase = 0;
  1177. particles->prev_phase = 0;
  1178. particles->clear = true;
  1179. }
  1180. particles->inactive = false;
  1181. particles->inactive_time = 0;
  1182. } else {
  1183. particles->inactive_time += particles->speed_scale * RendererCompositorRD::get_singleton()->get_frame_delta_time();
  1184. if (particles->inactive_time > particles->lifetime * 1.2) {
  1185. particles->inactive = true;
  1186. continue;
  1187. }
  1188. }
  1189. // TODO: Should use display refresh rate for all this.
  1190. float screen_hz = 60;
  1191. int fixed_fps = 0;
  1192. if (particles->fixed_fps > 0) {
  1193. fixed_fps = particles->fixed_fps;
  1194. } else if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  1195. fixed_fps = screen_hz;
  1196. }
  1197. {
  1198. //update trails
  1199. int history_size = 1;
  1200. int trail_steps = 1;
  1201. if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  1202. history_size = MAX(1, int(particles->trail_lifetime * fixed_fps));
  1203. trail_steps = particles->trail_bind_poses.size();
  1204. }
  1205. if (uint32_t(history_size) != particles->frame_history.size()) {
  1206. particles->frame_history.resize(history_size);
  1207. memset(particles->frame_history.ptr(), 0, sizeof(ParticlesFrameParams) * history_size);
  1208. // Set the frame number so that we are able to distinguish an uninitialized
  1209. // frame from the true frame number zero. See issue #88712 for details.
  1210. for (int i = 0; i < history_size; i++) {
  1211. particles->frame_history[i].frame = UINT32_MAX;
  1212. }
  1213. }
  1214. if (uint32_t(trail_steps) != particles->trail_params.size() || particles->frame_params_buffer.is_null()) {
  1215. particles->trail_params.resize(trail_steps);
  1216. if (particles->frame_params_buffer.is_valid()) {
  1217. RD::get_singleton()->free(particles->frame_params_buffer);
  1218. }
  1219. particles->frame_params_buffer = RD::get_singleton()->storage_buffer_create(sizeof(ParticlesFrameParams) * trail_steps);
  1220. }
  1221. if (particles->trail_bind_poses.size() > 1 && particles->trail_bind_pose_buffer.is_null()) {
  1222. particles->trail_bind_pose_buffer = RD::get_singleton()->storage_buffer_create(sizeof(float) * 16 * particles->trail_bind_poses.size());
  1223. particles->trail_bind_poses_dirty = true;
  1224. }
  1225. if (particles->trail_bind_pose_uniform_set.is_null()) {
  1226. Vector<RD::Uniform> uniforms;
  1227. {
  1228. RD::Uniform u;
  1229. u.uniform_type = RD::UNIFORM_TYPE_STORAGE_BUFFER;
  1230. u.binding = 0;
  1231. if (particles->trail_bind_pose_buffer.is_valid()) {
  1232. u.append_id(particles->trail_bind_pose_buffer);
  1233. } else {
  1234. _particles_ensure_unused_trail_buffer(particles);
  1235. u.append_id(particles->unused_trail_storage_buffer);
  1236. }
  1237. uniforms.push_back(u);
  1238. }
  1239. particles->trail_bind_pose_uniform_set = RD::get_singleton()->uniform_set_create(uniforms, particles_shader.copy_shader.version_get_shader(particles_shader.copy_shader_version, 0), 2);
  1240. }
  1241. if (particles->trail_bind_pose_buffer.is_valid() && particles->trail_bind_poses_dirty) {
  1242. if (particles_shader.pose_update_buffer.size() < uint32_t(particles->trail_bind_poses.size()) * 16) {
  1243. particles_shader.pose_update_buffer.resize(particles->trail_bind_poses.size() * 16);
  1244. }
  1245. for (int i = 0; i < particles->trail_bind_poses.size(); i++) {
  1246. RendererRD::MaterialStorage::store_transform(particles->trail_bind_poses[i], &particles_shader.pose_update_buffer[i * 16]);
  1247. }
  1248. RD::get_singleton()->buffer_update(particles->trail_bind_pose_buffer, 0, particles->trail_bind_poses.size() * 16 * sizeof(float), particles_shader.pose_update_buffer.ptr());
  1249. }
  1250. }
  1251. double todo = particles->request_process_time;
  1252. if (particles->clear) {
  1253. todo += particles->pre_process_time;
  1254. }
  1255. if (todo > 0.0) {
  1256. double frame_time;
  1257. if (fixed_fps > 0) {
  1258. frame_time = 1.0 / fixed_fps;
  1259. } else {
  1260. frame_time = 1.0 / 30.0;
  1261. }
  1262. float tmp_scale = particles->speed_scale;
  1263. // We need this otherwise the speed scale of the particle system influences the TODO.
  1264. particles->speed_scale = 1.0;
  1265. while (todo >= 0) {
  1266. _particles_process(particles, frame_time);
  1267. todo -= frame_time;
  1268. }
  1269. particles->request_process_time = 0.0;
  1270. particles->speed_scale = tmp_scale;
  1271. }
  1272. double time_scale = MAX(particles->speed_scale, 0.0);
  1273. if (fixed_fps > 0) {
  1274. double frame_time = 1.0 / fixed_fps;
  1275. double delta = RendererCompositorRD::get_singleton()->get_frame_delta_time();
  1276. if (delta > 0.1) { //avoid recursive stalls if fps goes below 10
  1277. delta = 0.1;
  1278. } else if (delta <= 0.0) { //unlikely but..
  1279. delta = 0.001;
  1280. }
  1281. todo = particles->frame_remainder + delta * time_scale;
  1282. while (todo >= frame_time || particles->clear) {
  1283. _particles_process(particles, frame_time);
  1284. todo -= frame_time;
  1285. }
  1286. particles->frame_remainder = todo;
  1287. } else {
  1288. _particles_process(particles, RendererCompositorRD::get_singleton()->get_frame_delta_time() * time_scale);
  1289. }
  1290. // Ensure that memory is initialized (the code above should ensure that _particles_process is always called at least once upon clearing).
  1291. DEV_ASSERT(!particles->clear);
  1292. int total_amount = particles->amount;
  1293. if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  1294. total_amount *= particles->trail_bind_poses.size();
  1295. }
  1296. // Swap offsets for motion vectors. Motion vectors can only be used when the draw order keeps the indices consistent across frames.
  1297. bool index_draw_order = particles->draw_order == RS::ParticlesDrawOrder::PARTICLES_DRAW_ORDER_INDEX;
  1298. particles->instance_motion_vectors_previous_offset = particles->instance_motion_vectors_current_offset;
  1299. if (uses_motion_vectors && index_draw_order && particles->instance_motion_vectors_enabled && (frame - particles->instance_motion_vectors_last_change) == 1) {
  1300. particles->instance_motion_vectors_current_offset = total_amount - particles->instance_motion_vectors_current_offset;
  1301. }
  1302. particles->instance_motion_vectors_last_change = frame;
  1303. // Copy particles to instance buffer.
  1304. if (particles->draw_order != RS::PARTICLES_DRAW_ORDER_VIEW_DEPTH && particles->transform_align != RS::PARTICLES_TRANSFORM_ALIGN_Z_BILLBOARD && particles->transform_align != RS::PARTICLES_TRANSFORM_ALIGN_Z_BILLBOARD_Y_TO_VELOCITY) {
  1305. //does not need view dependent operation, do copy here
  1306. ParticlesShader::CopyPushConstant copy_push_constant;
  1307. // Affect 2D only.
  1308. if (particles->use_local_coords) {
  1309. // In local mode, particle positions are calculated locally (relative to the node position)
  1310. // and they're also drawn locally.
  1311. // It works as expected, so we just pass an identity transform.
  1312. RendererRD::MaterialStorage::store_transform(Transform3D(), copy_push_constant.inv_emission_transform);
  1313. } else {
  1314. // In global mode, particle positions are calculated globally (relative to the canvas origin)
  1315. // but they're drawn locally.
  1316. // So, we need to pass the inverse of the emission transform to bring the
  1317. // particles to local coordinates before drawing.
  1318. Transform3D inv = particles->emission_transform.affine_inverse();
  1319. RendererRD::MaterialStorage::store_transform(inv, copy_push_constant.inv_emission_transform);
  1320. }
  1321. copy_push_constant.total_particles = total_amount;
  1322. copy_push_constant.frame_remainder = particles->interpolate ? particles->frame_remainder : 0.0;
  1323. copy_push_constant.align_mode = particles->transform_align;
  1324. copy_push_constant.align_up[0] = 0;
  1325. copy_push_constant.align_up[1] = 0;
  1326. copy_push_constant.align_up[2] = 0;
  1327. if (particles->trails_enabled && particles->trail_bind_poses.size() > 1) {
  1328. copy_push_constant.trail_size = particles->trail_bind_poses.size();
  1329. copy_push_constant.trail_total = particles->frame_history.size();
  1330. copy_push_constant.frame_delta = 1.0 / fixed_fps;
  1331. } else {
  1332. copy_push_constant.trail_size = 1;
  1333. copy_push_constant.trail_total = 1;
  1334. copy_push_constant.frame_delta = 0.0;
  1335. }
  1336. copy_push_constant.order_by_lifetime = (particles->draw_order == RS::PARTICLES_DRAW_ORDER_LIFETIME || particles->draw_order == RS::PARTICLES_DRAW_ORDER_REVERSE_LIFETIME);
  1337. copy_push_constant.lifetime_split = (MIN(int(particles->amount * particles->phase), particles->amount - 1) + 1) % particles->amount;
  1338. copy_push_constant.lifetime_reverse = particles->draw_order == RS::PARTICLES_DRAW_ORDER_REVERSE_LIFETIME;
  1339. copy_push_constant.motion_vectors_current_offset = particles->instance_motion_vectors_current_offset;
  1340. RD::ComputeListID compute_list = RD::get_singleton()->compute_list_begin();
  1341. copy_push_constant.copy_mode_2d = particles->mode == RS::PARTICLES_MODE_2D ? 1 : 0;
  1342. RD::get_singleton()->compute_list_bind_compute_pipeline(compute_list, particles_shader.copy_pipelines[ParticlesShader::COPY_MODE_FILL_INSTANCES + particles->userdata_count * ParticlesShader::COPY_MODE_MAX]);
  1343. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->particles_copy_uniform_set, 0);
  1344. RD::get_singleton()->compute_list_bind_uniform_set(compute_list, particles->trail_bind_pose_uniform_set, 2);
  1345. RD::get_singleton()->compute_list_set_push_constant(compute_list, &copy_push_constant, sizeof(ParticlesShader::CopyPushConstant));
  1346. RD::get_singleton()->compute_list_dispatch_threads(compute_list, total_amount, 1, 1);
  1347. RD::get_singleton()->compute_list_end();
  1348. }
  1349. particles->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_AABB);
  1350. }
  1351. }
  1352. Dependency *ParticlesStorage::particles_get_dependency(RID p_particles) const {
  1353. Particles *particles = particles_owner.get_or_null(p_particles);
  1354. ERR_FAIL_NULL_V(particles, nullptr);
  1355. return &particles->dependency;
  1356. }
  1357. bool ParticlesStorage::particles_is_inactive(RID p_particles) const {
  1358. const Particles *particles = particles_owner.get_or_null(p_particles);
  1359. ERR_FAIL_NULL_V(particles, false);
  1360. return !particles->emitting && particles->inactive;
  1361. }
  1362. /* Particles SHADER */
  1363. void ParticlesStorage::ParticlesShaderData::set_code(const String &p_code) {
  1364. ParticlesStorage *particles_storage = ParticlesStorage::get_singleton();
  1365. //compile
  1366. code = p_code;
  1367. valid = false;
  1368. ubo_size = 0;
  1369. uniforms.clear();
  1370. uses_collision = false;
  1371. if (code.is_empty()) {
  1372. return; //just invalid, but no error
  1373. }
  1374. ShaderCompiler::GeneratedCode gen_code;
  1375. ShaderCompiler::IdentifierActions actions;
  1376. actions.entry_point_stages["start"] = ShaderCompiler::STAGE_COMPUTE;
  1377. actions.entry_point_stages["process"] = ShaderCompiler::STAGE_COMPUTE;
  1378. /*
  1379. uses_time = false;
  1380. actions.render_mode_flags["use_half_res_pass"] = &uses_half_res;
  1381. actions.render_mode_flags["use_quarter_res_pass"] = &uses_quarter_res;
  1382. actions.usage_flag_pointers["TIME"] = &uses_time;
  1383. */
  1384. actions.usage_flag_pointers["COLLIDED"] = &uses_collision;
  1385. userdata_count = 0;
  1386. for (uint32_t i = 0; i < ParticlesShader::MAX_USERDATAS; i++) {
  1387. userdatas_used[i] = false;
  1388. actions.usage_flag_pointers["USERDATA" + itos(i + 1)] = &userdatas_used[i];
  1389. }
  1390. actions.uniforms = &uniforms;
  1391. Error err = particles_storage->particles_shader.compiler.compile(RS::SHADER_PARTICLES, code, &actions, path, gen_code);
  1392. ERR_FAIL_COND_MSG(err != OK, "Shader compilation failed.");
  1393. if (version.is_null()) {
  1394. version = particles_storage->particles_shader.shader.version_create();
  1395. }
  1396. for (uint32_t i = 0; i < ParticlesShader::MAX_USERDATAS; i++) {
  1397. if (userdatas_used[i]) {
  1398. userdata_count++;
  1399. }
  1400. }
  1401. particles_storage->particles_shader.shader.version_set_compute_code(version, gen_code.code, gen_code.uniforms, gen_code.stage_globals[ShaderCompiler::STAGE_COMPUTE], gen_code.defines);
  1402. ERR_FAIL_COND(!particles_storage->particles_shader.shader.version_is_valid(version));
  1403. ubo_size = gen_code.uniform_total_size;
  1404. ubo_offsets = gen_code.uniform_offsets;
  1405. texture_uniforms = gen_code.texture_uniforms;
  1406. //update pipelines
  1407. pipeline = RD::get_singleton()->compute_pipeline_create(particles_storage->particles_shader.shader.version_get_shader(version, 0));
  1408. valid = true;
  1409. }
  1410. bool ParticlesStorage::ParticlesShaderData::is_animated() const {
  1411. return false;
  1412. }
  1413. bool ParticlesStorage::ParticlesShaderData::casts_shadows() const {
  1414. return false;
  1415. }
  1416. RS::ShaderNativeSourceCode ParticlesStorage::ParticlesShaderData::get_native_source_code() const {
  1417. return ParticlesStorage::get_singleton()->particles_shader.shader.version_get_native_source_code(version);
  1418. }
  1419. Pair<ShaderRD *, RID> ParticlesStorage::ParticlesShaderData::get_native_shader_and_version() const {
  1420. return { &ParticlesStorage::get_singleton()->particles_shader.shader, version };
  1421. }
  1422. ParticlesStorage::ParticlesShaderData::~ParticlesShaderData() {
  1423. //pipeline variants will clear themselves if shader is gone
  1424. if (version.is_valid()) {
  1425. ParticlesStorage::get_singleton()->particles_shader.shader.version_free(version);
  1426. }
  1427. }
  1428. MaterialStorage::ShaderData *ParticlesStorage::_create_particles_shader_func() {
  1429. ParticlesShaderData *shader_data = memnew(ParticlesShaderData);
  1430. return shader_data;
  1431. }
  1432. bool ParticlesStorage::ParticleProcessMaterialData::update_parameters(const HashMap<StringName, Variant> &p_parameters, bool p_uniform_dirty, bool p_textures_dirty) {
  1433. 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, ParticlesStorage::get_singleton()->particles_shader.shader.version_get_shader(shader_data->version, 0), 3, true, false);
  1434. }
  1435. ParticlesStorage::ParticleProcessMaterialData::~ParticleProcessMaterialData() {
  1436. free_parameters_uniform_set(uniform_set);
  1437. }
  1438. MaterialStorage::MaterialData *ParticlesStorage::_create_particles_material_func(ParticlesShaderData *p_shader) {
  1439. ParticleProcessMaterialData *material_data = memnew(ParticleProcessMaterialData);
  1440. material_data->shader_data = p_shader;
  1441. //update will happen later anyway so do nothing.
  1442. return material_data;
  1443. }
  1444. ////////
  1445. /* PARTICLES COLLISION API */
  1446. RID ParticlesStorage::particles_collision_allocate() {
  1447. return particles_collision_owner.allocate_rid();
  1448. }
  1449. void ParticlesStorage::particles_collision_initialize(RID p_rid) {
  1450. particles_collision_owner.initialize_rid(p_rid, ParticlesCollision());
  1451. }
  1452. void ParticlesStorage::particles_collision_free(RID p_rid) {
  1453. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_rid);
  1454. if (particles_collision->heightfield_texture.is_valid()) {
  1455. RD::get_singleton()->free(particles_collision->heightfield_texture);
  1456. }
  1457. particles_collision->dependency.deleted_notify(p_rid);
  1458. particles_collision_owner.free(p_rid);
  1459. }
  1460. RID ParticlesStorage::particles_collision_get_heightfield_framebuffer(RID p_particles_collision) const {
  1461. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1462. ERR_FAIL_NULL_V(particles_collision, RID());
  1463. ERR_FAIL_COND_V(particles_collision->type != RS::PARTICLES_COLLISION_TYPE_HEIGHTFIELD_COLLIDE, RID());
  1464. if (particles_collision->heightfield_texture == RID()) {
  1465. //create
  1466. const int resolutions[RS::PARTICLES_COLLISION_HEIGHTFIELD_RESOLUTION_MAX] = { 256, 512, 1024, 2048, 4096, 8192 };
  1467. Size2i size;
  1468. if (particles_collision->extents.x > particles_collision->extents.z) {
  1469. size.x = resolutions[particles_collision->heightfield_resolution];
  1470. size.y = int32_t(particles_collision->extents.z / particles_collision->extents.x * size.x);
  1471. } else {
  1472. size.y = resolutions[particles_collision->heightfield_resolution];
  1473. size.x = int32_t(particles_collision->extents.x / particles_collision->extents.z * size.y);
  1474. }
  1475. RD::TextureFormat tf;
  1476. tf.format = RD::DATA_FORMAT_D32_SFLOAT;
  1477. tf.width = size.x;
  1478. tf.height = size.y;
  1479. tf.texture_type = RD::TEXTURE_TYPE_2D;
  1480. tf.usage_bits = RD::TEXTURE_USAGE_SAMPLING_BIT | RD::TEXTURE_USAGE_DEPTH_STENCIL_ATTACHMENT_BIT;
  1481. particles_collision->heightfield_texture = RD::get_singleton()->texture_create(tf, RD::TextureView());
  1482. Vector<RID> fb_tex;
  1483. fb_tex.push_back(particles_collision->heightfield_texture);
  1484. particles_collision->heightfield_fb = RD::get_singleton()->framebuffer_create(fb_tex);
  1485. particles_collision->heightfield_fb_size = size;
  1486. }
  1487. return particles_collision->heightfield_fb;
  1488. }
  1489. void ParticlesStorage::particles_collision_set_collision_type(RID p_particles_collision, RS::ParticlesCollisionType p_type) {
  1490. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1491. ERR_FAIL_NULL(particles_collision);
  1492. if (p_type == particles_collision->type) {
  1493. return;
  1494. }
  1495. if (particles_collision->heightfield_texture.is_valid()) {
  1496. RD::get_singleton()->free(particles_collision->heightfield_texture);
  1497. particles_collision->heightfield_texture = RID();
  1498. }
  1499. particles_collision->type = p_type;
  1500. particles_collision->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_AABB);
  1501. }
  1502. void ParticlesStorage::particles_collision_set_cull_mask(RID p_particles_collision, uint32_t p_cull_mask) {
  1503. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1504. ERR_FAIL_NULL(particles_collision);
  1505. particles_collision->cull_mask = p_cull_mask;
  1506. particles_collision->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_CULL_MASK);
  1507. }
  1508. uint32_t ParticlesStorage::particles_collision_get_cull_mask(RID p_particles_collision) const {
  1509. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1510. ERR_FAIL_NULL_V(particles_collision, 0);
  1511. return particles_collision->cull_mask;
  1512. }
  1513. uint32_t ParticlesStorage::particles_collision_get_height_field_mask(RID p_particles_collision) const {
  1514. const ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1515. ERR_FAIL_NULL_V(particles_collision, false);
  1516. return particles_collision->heightfield_mask;
  1517. }
  1518. void ParticlesStorage::particles_collision_set_height_field_mask(RID p_particles_collision, uint32_t p_heightfield_mask) {
  1519. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1520. ERR_FAIL_NULL(particles_collision);
  1521. particles_collision->heightfield_mask = p_heightfield_mask;
  1522. }
  1523. void ParticlesStorage::particles_collision_set_sphere_radius(RID p_particles_collision, real_t p_radius) {
  1524. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1525. ERR_FAIL_NULL(particles_collision);
  1526. particles_collision->radius = p_radius;
  1527. particles_collision->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_AABB);
  1528. }
  1529. void ParticlesStorage::particles_collision_set_box_extents(RID p_particles_collision, const Vector3 &p_extents) {
  1530. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1531. ERR_FAIL_NULL(particles_collision);
  1532. particles_collision->extents = p_extents;
  1533. particles_collision->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_AABB);
  1534. }
  1535. void ParticlesStorage::particles_collision_set_attractor_strength(RID p_particles_collision, real_t p_strength) {
  1536. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1537. ERR_FAIL_NULL(particles_collision);
  1538. particles_collision->attractor_strength = p_strength;
  1539. }
  1540. void ParticlesStorage::particles_collision_set_attractor_directionality(RID p_particles_collision, real_t p_directionality) {
  1541. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1542. ERR_FAIL_NULL(particles_collision);
  1543. particles_collision->attractor_directionality = p_directionality;
  1544. }
  1545. void ParticlesStorage::particles_collision_set_attractor_attenuation(RID p_particles_collision, real_t p_curve) {
  1546. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1547. ERR_FAIL_NULL(particles_collision);
  1548. particles_collision->attractor_attenuation = p_curve;
  1549. }
  1550. void ParticlesStorage::particles_collision_set_field_texture(RID p_particles_collision, RID p_texture) {
  1551. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1552. ERR_FAIL_NULL(particles_collision);
  1553. particles_collision->field_texture = p_texture;
  1554. }
  1555. void ParticlesStorage::particles_collision_height_field_update(RID p_particles_collision) {
  1556. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1557. ERR_FAIL_NULL(particles_collision);
  1558. particles_collision->dependency.changed_notify(Dependency::DEPENDENCY_CHANGED_AABB);
  1559. }
  1560. void ParticlesStorage::particles_collision_set_height_field_resolution(RID p_particles_collision, RS::ParticlesCollisionHeightfieldResolution p_resolution) {
  1561. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1562. ERR_FAIL_NULL(particles_collision);
  1563. ERR_FAIL_INDEX(p_resolution, RS::PARTICLES_COLLISION_HEIGHTFIELD_RESOLUTION_MAX);
  1564. if (particles_collision->heightfield_resolution == p_resolution) {
  1565. return;
  1566. }
  1567. particles_collision->heightfield_resolution = p_resolution;
  1568. if (particles_collision->heightfield_texture.is_valid()) {
  1569. RD::get_singleton()->free(particles_collision->heightfield_texture);
  1570. particles_collision->heightfield_texture = RID();
  1571. }
  1572. }
  1573. AABB ParticlesStorage::particles_collision_get_aabb(RID p_particles_collision) const {
  1574. ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1575. ERR_FAIL_NULL_V(particles_collision, AABB());
  1576. switch (particles_collision->type) {
  1577. case RS::PARTICLES_COLLISION_TYPE_SPHERE_ATTRACT:
  1578. case RS::PARTICLES_COLLISION_TYPE_SPHERE_COLLIDE: {
  1579. AABB aabb;
  1580. aabb.position = -Vector3(1, 1, 1) * particles_collision->radius;
  1581. aabb.size = Vector3(2, 2, 2) * particles_collision->radius;
  1582. return aabb;
  1583. }
  1584. default: {
  1585. AABB aabb;
  1586. aabb.position = -particles_collision->extents;
  1587. aabb.size = particles_collision->extents * 2;
  1588. return aabb;
  1589. }
  1590. }
  1591. }
  1592. Vector3 ParticlesStorage::particles_collision_get_extents(RID p_particles_collision) const {
  1593. const ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1594. ERR_FAIL_NULL_V(particles_collision, Vector3());
  1595. return particles_collision->extents;
  1596. }
  1597. bool ParticlesStorage::particles_collision_is_heightfield(RID p_particles_collision) const {
  1598. const ParticlesCollision *particles_collision = particles_collision_owner.get_or_null(p_particles_collision);
  1599. ERR_FAIL_NULL_V(particles_collision, false);
  1600. return particles_collision->type == RS::PARTICLES_COLLISION_TYPE_HEIGHTFIELD_COLLIDE;
  1601. }
  1602. Dependency *ParticlesStorage::particles_collision_get_dependency(RID p_particles_collision) const {
  1603. ParticlesCollision *pc = particles_collision_owner.get_or_null(p_particles_collision);
  1604. ERR_FAIL_NULL_V(pc, nullptr);
  1605. return &pc->dependency;
  1606. }
  1607. /* Particles collision instance */
  1608. RID ParticlesStorage::particles_collision_instance_create(RID p_collision) {
  1609. ParticlesCollisionInstance pci;
  1610. pci.collision = p_collision;
  1611. return particles_collision_instance_owner.make_rid(pci);
  1612. }
  1613. void ParticlesStorage::particles_collision_instance_free(RID p_rid) {
  1614. particles_collision_instance_owner.free(p_rid);
  1615. }
  1616. void ParticlesStorage::particles_collision_instance_set_transform(RID p_collision_instance, const Transform3D &p_transform) {
  1617. ParticlesCollisionInstance *pci = particles_collision_instance_owner.get_or_null(p_collision_instance);
  1618. ERR_FAIL_NULL(pci);
  1619. pci->transform = p_transform;
  1620. }
  1621. void ParticlesStorage::particles_collision_instance_set_active(RID p_collision_instance, bool p_active) {
  1622. ParticlesCollisionInstance *pci = particles_collision_instance_owner.get_or_null(p_collision_instance);
  1623. ERR_FAIL_NULL(pci);
  1624. pci->active = p_active;
  1625. }