scene_debugger.cpp 88 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797
  1. /**************************************************************************/
  2. /* scene_debugger.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 "scene_debugger.h"
  31. #include "core/debugger/debugger_marshalls.h"
  32. #include "core/debugger/engine_debugger.h"
  33. #include "core/io/marshalls.h"
  34. #include "core/math/math_fieldwise.h"
  35. #include "core/object/script_language.h"
  36. #include "core/templates/local_vector.h"
  37. #include "scene/gui/popup_menu.h"
  38. #include "scene/main/canvas_layer.h"
  39. #include "scene/main/scene_tree.h"
  40. #include "scene/main/window.h"
  41. #include "scene/resources/packed_scene.h"
  42. #include "scene/theme/theme_db.h"
  43. #include "servers/audio_server.h"
  44. #ifndef PHYSICS_2D_DISABLED
  45. #include "scene/2d/physics/collision_object_2d.h"
  46. #include "scene/2d/physics/collision_polygon_2d.h"
  47. #include "scene/2d/physics/collision_shape_2d.h"
  48. #endif // PHYSICS_2D_DISABLED
  49. #ifndef _3D_DISABLED
  50. #include "scene/3d/camera_3d.h"
  51. #ifndef PHYSICS_3D_DISABLED
  52. #include "scene/3d/physics/collision_object_3d.h"
  53. #include "scene/3d/physics/collision_shape_3d.h"
  54. #endif // PHYSICS_3D_DISABLED
  55. #include "scene/3d/visual_instance_3d.h"
  56. #include "scene/resources/3d/convex_polygon_shape_3d.h"
  57. #include "scene/resources/surface_tool.h"
  58. #endif // _3D_DISABLED
  59. SceneDebugger::SceneDebugger() {
  60. singleton = this;
  61. #ifdef DEBUG_ENABLED
  62. LiveEditor::singleton = memnew(LiveEditor);
  63. RuntimeNodeSelect::singleton = memnew(RuntimeNodeSelect);
  64. EngineDebugger::register_message_capture("scene", EngineDebugger::Capture(nullptr, SceneDebugger::parse_message));
  65. #endif // DEBUG_ENABLED
  66. }
  67. SceneDebugger::~SceneDebugger() {
  68. #ifdef DEBUG_ENABLED
  69. if (LiveEditor::singleton) {
  70. EngineDebugger::unregister_message_capture("scene");
  71. memdelete(LiveEditor::singleton);
  72. LiveEditor::singleton = nullptr;
  73. }
  74. if (RuntimeNodeSelect::singleton) {
  75. memdelete(RuntimeNodeSelect::singleton);
  76. RuntimeNodeSelect::singleton = nullptr;
  77. }
  78. #endif // DEBUG_ENABLED
  79. singleton = nullptr;
  80. }
  81. void SceneDebugger::initialize() {
  82. if (EngineDebugger::is_active()) {
  83. memnew(SceneDebugger);
  84. }
  85. }
  86. void SceneDebugger::deinitialize() {
  87. if (singleton) {
  88. memdelete(singleton);
  89. }
  90. }
  91. #ifdef DEBUG_ENABLED
  92. void SceneDebugger::_handle_input(const Ref<InputEvent> &p_event, const Ref<Shortcut> &p_shortcut) {
  93. Ref<InputEventKey> k = p_event;
  94. if (p_shortcut.is_valid() && k.is_valid() && k->is_pressed() && !k->is_echo() && p_shortcut->matches_event(k)) {
  95. EngineDebugger::get_singleton()->send_message("request_quit", Array());
  96. }
  97. }
  98. Error SceneDebugger::parse_message(void *p_user, const String &p_msg, const Array &p_args, bool &r_captured) {
  99. SceneTree *scene_tree = SceneTree::get_singleton();
  100. if (!scene_tree) {
  101. return ERR_UNCONFIGURED;
  102. }
  103. LiveEditor *live_editor = LiveEditor::get_singleton();
  104. if (!live_editor) {
  105. return ERR_UNCONFIGURED;
  106. }
  107. RuntimeNodeSelect *runtime_node_select = RuntimeNodeSelect::get_singleton();
  108. if (!runtime_node_select) {
  109. return ERR_UNCONFIGURED;
  110. }
  111. r_captured = true;
  112. if (p_msg == "setup_scene") {
  113. SceneTree::get_singleton()->get_root()->connect(SceneStringName(window_input), callable_mp_static(SceneDebugger::_handle_input).bind(DebuggerMarshalls::deserialize_key_shortcut(p_args)));
  114. } else if (p_msg == "request_scene_tree") { /// Scene Tree
  115. live_editor->_send_tree();
  116. } else if (p_msg == "save_node") {
  117. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  118. _save_node(p_args[0], p_args[1]);
  119. Array arr;
  120. arr.append(p_args[1]);
  121. EngineDebugger::get_singleton()->send_message("filesystem:update_file", { arr });
  122. } else if (p_msg == "inspect_objects") { /// Object Inspect
  123. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  124. Vector<ObjectID> ids;
  125. for (const Variant &id : (Array)p_args[0]) {
  126. ids.append(ObjectID(id.operator uint64_t()));
  127. }
  128. _send_object_ids(ids, p_args[1]);
  129. } else if (p_msg == "clear_selection") {
  130. runtime_node_select->_clear_selection();
  131. } else if (p_msg == "suspend_changed") {
  132. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  133. bool suspended = p_args[0];
  134. scene_tree->set_suspend(suspended);
  135. runtime_node_select->_update_input_state();
  136. } else if (p_msg == "next_frame") {
  137. _next_frame();
  138. } else if (p_msg == "debug_mute_audio") { // Enable/disable audio.
  139. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  140. bool do_mute = p_args[0];
  141. AudioServer::get_singleton()->set_debug_mute(do_mute);
  142. } else if (p_msg == "override_cameras") { /// Camera
  143. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  144. bool enable = p_args[0];
  145. bool from_editor = p_args[1];
  146. scene_tree->get_root()->enable_canvas_transform_override(enable);
  147. #ifndef _3D_DISABLED
  148. scene_tree->get_root()->enable_camera_3d_override(enable);
  149. #endif // _3D_DISABLED
  150. runtime_node_select->_set_camera_override_enabled(enable && !from_editor);
  151. } else if (p_msg == "transform_camera_2d") {
  152. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  153. Transform2D transform = p_args[0];
  154. scene_tree->get_root()->set_canvas_transform_override(transform);
  155. runtime_node_select->_queue_selection_update();
  156. #ifndef _3D_DISABLED
  157. } else if (p_msg == "transform_camera_3d") {
  158. ERR_FAIL_COND_V(p_args.size() < 5, ERR_INVALID_DATA);
  159. Transform3D transform = p_args[0];
  160. bool is_perspective = p_args[1];
  161. float size_or_fov = p_args[2];
  162. float depth_near = p_args[3];
  163. float depth_far = p_args[4];
  164. if (is_perspective) {
  165. scene_tree->get_root()->set_camera_3d_override_perspective(size_or_fov, depth_near, depth_far);
  166. } else {
  167. scene_tree->get_root()->set_camera_3d_override_orthogonal(size_or_fov, depth_near, depth_far);
  168. }
  169. scene_tree->get_root()->set_camera_3d_override_transform(transform);
  170. runtime_node_select->_queue_selection_update();
  171. #endif // _3D_DISABLED
  172. } else if (p_msg == "set_object_property") {
  173. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  174. _set_object_property(p_args[0], p_args[1], p_args[2]);
  175. runtime_node_select->_queue_selection_update();
  176. } else if (p_msg == "set_object_property_field") {
  177. ERR_FAIL_COND_V(p_args.size() < 4, ERR_INVALID_DATA);
  178. _set_object_property(p_args[0], p_args[1], p_args[2], p_args[3]);
  179. runtime_node_select->_queue_selection_update();
  180. } else if (p_msg == "reload_cached_files") {
  181. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  182. PackedStringArray files = p_args[0];
  183. reload_cached_files(files);
  184. } else if (p_msg.begins_with("live_")) { /// Live Edit
  185. if (p_msg == "live_set_root") {
  186. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  187. live_editor->_root_func(p_args[0], p_args[1]);
  188. } else if (p_msg == "live_node_path") {
  189. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  190. live_editor->_node_path_func(p_args[0], p_args[1]);
  191. } else if (p_msg == "live_res_path") {
  192. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  193. live_editor->_res_path_func(p_args[0], p_args[1]);
  194. } else if (p_msg == "live_node_prop_res") {
  195. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  196. live_editor->_node_set_res_func(p_args[0], p_args[1], p_args[2]);
  197. } else if (p_msg == "live_node_prop") {
  198. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  199. live_editor->_node_set_func(p_args[0], p_args[1], p_args[2]);
  200. } else if (p_msg == "live_res_prop_res") {
  201. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  202. live_editor->_res_set_res_func(p_args[0], p_args[1], p_args[2]);
  203. } else if (p_msg == "live_res_prop") {
  204. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  205. live_editor->_res_set_func(p_args[0], p_args[1], p_args[2]);
  206. } else if (p_msg == "live_node_call") {
  207. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  208. LocalVector<Variant> args;
  209. LocalVector<Variant *> argptrs;
  210. args.resize(p_args.size() - 2);
  211. argptrs.resize(args.size());
  212. for (uint32_t i = 0; i < args.size(); i++) {
  213. args[i] = p_args[i + 2];
  214. argptrs[i] = &args[i];
  215. }
  216. live_editor->_node_call_func(p_args[0], p_args[1], argptrs.size() ? (const Variant **)argptrs.ptr() : nullptr, argptrs.size());
  217. } else if (p_msg == "live_res_call") {
  218. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  219. LocalVector<Variant> args;
  220. LocalVector<Variant *> argptrs;
  221. args.resize(p_args.size() - 2);
  222. argptrs.resize(args.size());
  223. for (uint32_t i = 0; i < args.size(); i++) {
  224. args[i] = p_args[i + 2];
  225. argptrs[i] = &args[i];
  226. }
  227. live_editor->_res_call_func(p_args[0], p_args[1], argptrs.size() ? (const Variant **)argptrs.ptr() : nullptr, argptrs.size());
  228. } else if (p_msg == "live_create_node") {
  229. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  230. live_editor->_create_node_func(p_args[0], p_args[1], p_args[2]);
  231. } else if (p_msg == "live_instantiate_node") {
  232. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  233. live_editor->_instance_node_func(p_args[0], p_args[1], p_args[2]);
  234. } else if (p_msg == "live_remove_node") {
  235. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  236. live_editor->_remove_node_func(p_args[0]);
  237. runtime_node_select->_queue_selection_update();
  238. } else if (p_msg == "live_remove_and_keep_node") {
  239. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  240. live_editor->_remove_and_keep_node_func(p_args[0], p_args[1]);
  241. runtime_node_select->_queue_selection_update();
  242. } else if (p_msg == "live_restore_node") {
  243. ERR_FAIL_COND_V(p_args.size() < 3, ERR_INVALID_DATA);
  244. live_editor->_restore_node_func(p_args[0], p_args[1], p_args[2]);
  245. } else if (p_msg == "live_duplicate_node") {
  246. ERR_FAIL_COND_V(p_args.size() < 2, ERR_INVALID_DATA);
  247. live_editor->_duplicate_node_func(p_args[0], p_args[1]);
  248. } else if (p_msg == "live_reparent_node") {
  249. ERR_FAIL_COND_V(p_args.size() < 4, ERR_INVALID_DATA);
  250. live_editor->_reparent_node_func(p_args[0], p_args[1], p_args[2], p_args[3]);
  251. } else {
  252. return ERR_SKIP;
  253. }
  254. } else if (p_msg.begins_with("runtime_node_select_")) { /// Runtime Node Selection
  255. if (p_msg == "runtime_node_select_setup") {
  256. ERR_FAIL_COND_V(p_args.is_empty() || p_args[0].get_type() != Variant::DICTIONARY, ERR_INVALID_DATA);
  257. runtime_node_select->_setup(p_args[0]);
  258. } else if (p_msg == "runtime_node_select_set_type") {
  259. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  260. RuntimeNodeSelect::NodeType type = (RuntimeNodeSelect::NodeType)(int)p_args[0];
  261. runtime_node_select->_node_set_type(type);
  262. } else if (p_msg == "runtime_node_select_set_mode") {
  263. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  264. RuntimeNodeSelect::SelectMode mode = (RuntimeNodeSelect::SelectMode)(int)p_args[0];
  265. runtime_node_select->_select_set_mode(mode);
  266. } else if (p_msg == "runtime_node_select_set_visible") {
  267. ERR_FAIL_COND_V(p_args.is_empty(), ERR_INVALID_DATA);
  268. bool visible = p_args[0];
  269. runtime_node_select->_set_selection_visible(visible);
  270. } else if (p_msg == "runtime_node_select_reset_camera_2d") {
  271. runtime_node_select->_reset_camera_2d();
  272. #ifndef _3D_DISABLED
  273. } else if (p_msg == "runtime_node_select_reset_camera_3d") {
  274. runtime_node_select->_reset_camera_3d();
  275. #endif // _3D_DISABLED
  276. } else {
  277. return ERR_SKIP;
  278. }
  279. } else {
  280. r_captured = false;
  281. }
  282. return OK;
  283. }
  284. void SceneDebugger::_save_node(ObjectID id, const String &p_path) {
  285. Node *node = ObjectDB::get_instance<Node>(id);
  286. ERR_FAIL_NULL(node);
  287. #ifdef TOOLS_ENABLED
  288. HashMap<const Node *, Node *> duplimap;
  289. Node *copy = node->duplicate_from_editor(duplimap);
  290. #else
  291. Node *copy = node->duplicate();
  292. #endif // TOOLS_ENABLED
  293. // Handle Unique Nodes.
  294. for (int i = 0; i < copy->get_child_count(false); i++) {
  295. _set_node_owner_recursive(copy->get_child(i, false), copy);
  296. }
  297. // Root node cannot ever be unique name in its own Scene!
  298. copy->set_unique_name_in_owner(false);
  299. Ref<PackedScene> ps = memnew(PackedScene);
  300. ps->pack(copy);
  301. ResourceSaver::save(ps, p_path);
  302. memdelete(copy);
  303. }
  304. void SceneDebugger::_set_node_owner_recursive(Node *p_node, Node *p_owner) {
  305. if (!p_node->get_owner()) {
  306. p_node->set_owner(p_owner);
  307. }
  308. for (int i = 0; i < p_node->get_child_count(false); i++) {
  309. _set_node_owner_recursive(p_node->get_child(i, false), p_owner);
  310. }
  311. }
  312. void SceneDebugger::_send_object_ids(const Vector<ObjectID> &p_ids, bool p_update_selection) {
  313. Vector<ObjectID> ids = p_ids;
  314. if (ids.size() > RuntimeNodeSelect::get_singleton()->max_selection) {
  315. ids.resize(RuntimeNodeSelect::get_singleton()->max_selection);
  316. EngineDebugger::get_singleton()->send_message("show_selection_limit_warning", Array());
  317. }
  318. LocalVector<Node *> nodes;
  319. Array objs;
  320. bool objs_missing = false;
  321. for (const ObjectID &id : ids) {
  322. SceneDebuggerObject obj(id);
  323. if (obj.id.is_null()) {
  324. objs_missing = true;
  325. continue;
  326. }
  327. if (p_update_selection) {
  328. if (Node *node = ObjectDB::get_instance<Node>(id)) {
  329. nodes.push_back(node);
  330. }
  331. }
  332. Array arr;
  333. obj.serialize(arr);
  334. objs.append(arr);
  335. }
  336. if (p_update_selection) {
  337. RuntimeNodeSelect::get_singleton()->_set_selected_nodes(nodes);
  338. }
  339. if (objs_missing) {
  340. Array invalid_selection;
  341. for (const ObjectID &id : ids) {
  342. invalid_selection.append(id);
  343. }
  344. Array arr;
  345. arr.append(invalid_selection);
  346. EngineDebugger::get_singleton()->send_message("remote_selection_invalidated", arr);
  347. EngineDebugger::get_singleton()->send_message(objs.is_empty() ? "remote_nothing_selected" : "remote_objects_selected", objs);
  348. } else {
  349. EngineDebugger::get_singleton()->send_message(p_update_selection ? "remote_objects_selected" : "scene:inspect_objects", objs);
  350. }
  351. }
  352. void SceneDebugger::_set_object_property(ObjectID p_id, const String &p_property, const Variant &p_value, const String &p_field) {
  353. Object *obj = ObjectDB::get_instance(p_id);
  354. if (!obj) {
  355. return;
  356. }
  357. String prop_name = p_property;
  358. if (p_property.begins_with("Members/")) {
  359. Vector<String> ss = p_property.split("/");
  360. prop_name = ss[ss.size() - 1];
  361. }
  362. Variant value;
  363. if (p_field.is_empty()) {
  364. // Whole value.
  365. value = p_value;
  366. } else {
  367. // Only one field.
  368. value = fieldwise_assign(obj->get(prop_name), p_value, p_field);
  369. }
  370. obj->set(prop_name, value);
  371. }
  372. void SceneDebugger::_next_frame() {
  373. SceneTree *scene_tree = SceneTree::get_singleton();
  374. if (!scene_tree->is_suspended()) {
  375. return;
  376. }
  377. scene_tree->set_suspend(false);
  378. RenderingServer::get_singleton()->connect("frame_post_draw", callable_mp(scene_tree, &SceneTree::set_suspend).bind(true), Object::CONNECT_ONE_SHOT);
  379. }
  380. void SceneDebugger::add_to_cache(const String &p_filename, Node *p_node) {
  381. LiveEditor *debugger = LiveEditor::get_singleton();
  382. if (!debugger) {
  383. return;
  384. }
  385. if (EngineDebugger::get_script_debugger() && !p_filename.is_empty()) {
  386. debugger->live_scene_edit_cache[p_filename].insert(p_node);
  387. }
  388. }
  389. void SceneDebugger::remove_from_cache(const String &p_filename, Node *p_node) {
  390. LiveEditor *debugger = LiveEditor::get_singleton();
  391. if (!debugger) {
  392. return;
  393. }
  394. HashMap<String, HashSet<Node *>> &edit_cache = debugger->live_scene_edit_cache;
  395. HashMap<String, HashSet<Node *>>::Iterator E = edit_cache.find(p_filename);
  396. if (E) {
  397. E->value.erase(p_node);
  398. if (E->value.size() == 0) {
  399. edit_cache.remove(E);
  400. }
  401. }
  402. HashMap<Node *, HashMap<ObjectID, Node *>> &remove_list = debugger->live_edit_remove_list;
  403. HashMap<Node *, HashMap<ObjectID, Node *>>::Iterator F = remove_list.find(p_node);
  404. if (F) {
  405. for (const KeyValue<ObjectID, Node *> &G : F->value) {
  406. memdelete(G.value);
  407. }
  408. remove_list.remove(F);
  409. }
  410. }
  411. void SceneDebugger::reload_cached_files(const PackedStringArray &p_files) {
  412. for (const String &file : p_files) {
  413. Ref<Resource> res = ResourceCache::get_ref(file);
  414. if (res.is_valid()) {
  415. res->reload_from_file();
  416. }
  417. }
  418. }
  419. /// SceneDebuggerObject
  420. SceneDebuggerObject::SceneDebuggerObject(ObjectID p_id) {
  421. id = ObjectID();
  422. Object *obj = ObjectDB::get_instance(p_id);
  423. if (!obj) {
  424. return;
  425. }
  426. id = p_id;
  427. class_name = obj->get_class();
  428. if (ScriptInstance *si = obj->get_script_instance()) {
  429. // Read script instance constants and variables
  430. if (!si->get_script().is_null()) {
  431. Script *s = si->get_script().ptr();
  432. _parse_script_properties(s, si);
  433. }
  434. }
  435. if (Node *node = Object::cast_to<Node>(obj)) {
  436. // For debugging multiplayer.
  437. {
  438. PropertyInfo pi(Variant::INT, String("Node/multiplayer_authority"), PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_READ_ONLY);
  439. properties.push_back(SceneDebuggerProperty(pi, node->get_multiplayer_authority()));
  440. }
  441. // Add specialized NodePath info (if inside tree).
  442. if (node->is_inside_tree()) {
  443. PropertyInfo pi(Variant::NODE_PATH, String("Node/path"));
  444. properties.push_back(SceneDebuggerProperty(pi, node->get_path()));
  445. } else { // Can't ask for path if a node is not in tree.
  446. PropertyInfo pi(Variant::STRING, String("Node/path"));
  447. properties.push_back(SceneDebuggerProperty(pi, "[Orphan]"));
  448. }
  449. } else if (Script *s = Object::cast_to<Script>(obj)) {
  450. // Add script constants (no instance).
  451. _parse_script_properties(s, nullptr);
  452. }
  453. // Add base object properties.
  454. List<PropertyInfo> pinfo;
  455. obj->get_property_list(&pinfo, true);
  456. for (const PropertyInfo &E : pinfo) {
  457. if (E.usage & (PROPERTY_USAGE_EDITOR | PROPERTY_USAGE_CATEGORY)) {
  458. properties.push_back(SceneDebuggerProperty(E, obj->get(E.name)));
  459. }
  460. }
  461. }
  462. void SceneDebuggerObject::_parse_script_properties(Script *p_script, ScriptInstance *p_instance) {
  463. typedef HashMap<const Script *, HashSet<StringName>> ScriptMemberMap;
  464. typedef HashMap<const Script *, HashMap<StringName, Variant>> ScriptConstantsMap;
  465. ScriptMemberMap members;
  466. if (p_instance) {
  467. members[p_script] = HashSet<StringName>();
  468. p_script->get_members(&(members[p_script]));
  469. }
  470. ScriptConstantsMap constants;
  471. constants[p_script] = HashMap<StringName, Variant>();
  472. p_script->get_constants(&(constants[p_script]));
  473. Ref<Script> base = p_script->get_base_script();
  474. while (base.is_valid()) {
  475. if (p_instance) {
  476. members[base.ptr()] = HashSet<StringName>();
  477. base->get_members(&(members[base.ptr()]));
  478. }
  479. constants[base.ptr()] = HashMap<StringName, Variant>();
  480. base->get_constants(&(constants[base.ptr()]));
  481. base = base->get_base_script();
  482. }
  483. // Members
  484. for (KeyValue<const Script *, HashSet<StringName>> sm : members) {
  485. for (const StringName &E : sm.value) {
  486. Variant m;
  487. if (p_instance->get(E, m)) {
  488. String script_path = sm.key == p_script ? "" : sm.key->get_path().get_file() + "/";
  489. PropertyInfo pi(m.get_type(), "Members/" + script_path + E);
  490. properties.push_back(SceneDebuggerProperty(pi, m));
  491. }
  492. }
  493. }
  494. // Constants
  495. for (KeyValue<const Script *, HashMap<StringName, Variant>> &sc : constants) {
  496. for (const KeyValue<StringName, Variant> &E : sc.value) {
  497. String script_path = sc.key == p_script ? "" : sc.key->get_path().get_file() + "/";
  498. if (E.value.get_type() == Variant::OBJECT) {
  499. Variant inst_id = ((Object *)E.value)->get_instance_id();
  500. PropertyInfo pi(inst_id.get_type(), "Constants/" + E.key, PROPERTY_HINT_OBJECT_ID, "Object");
  501. properties.push_back(SceneDebuggerProperty(pi, inst_id));
  502. } else {
  503. PropertyInfo pi(E.value.get_type(), "Constants/" + script_path + E.key);
  504. properties.push_back(SceneDebuggerProperty(pi, E.value));
  505. }
  506. }
  507. }
  508. }
  509. void SceneDebuggerObject::serialize(Array &r_arr, int p_max_size) {
  510. Array send_props;
  511. for (SceneDebuggerObject::SceneDebuggerProperty &property : properties) {
  512. const PropertyInfo &pi = property.first;
  513. Variant &var = property.second;
  514. Ref<Resource> res = var;
  515. Array prop = { pi.name, pi.type };
  516. PropertyHint hint = pi.hint;
  517. String hint_string = pi.hint_string;
  518. if (res.is_valid() && !res->get_path().is_empty()) {
  519. var = res->get_path();
  520. } else { //only send information that can be sent..
  521. int len = 0; //test how big is this to encode
  522. encode_variant(var, nullptr, len);
  523. if (len > p_max_size) { //limit to max size
  524. hint = PROPERTY_HINT_OBJECT_TOO_BIG;
  525. hint_string = "";
  526. var = Variant();
  527. }
  528. }
  529. prop.push_back(hint);
  530. prop.push_back(hint_string);
  531. prop.push_back(pi.usage);
  532. prop.push_back(var);
  533. send_props.push_back(prop);
  534. }
  535. r_arr.push_back(uint64_t(id));
  536. r_arr.push_back(class_name);
  537. r_arr.push_back(send_props);
  538. }
  539. void SceneDebuggerObject::deserialize(const Array &p_arr) {
  540. #define CHECK_TYPE(p_what, p_type) ERR_FAIL_COND(p_what.get_type() != Variant::p_type);
  541. ERR_FAIL_COND(p_arr.size() < 3);
  542. CHECK_TYPE(p_arr[0], INT);
  543. CHECK_TYPE(p_arr[1], STRING);
  544. CHECK_TYPE(p_arr[2], ARRAY);
  545. id = uint64_t(p_arr[0]);
  546. class_name = p_arr[1];
  547. Array props = p_arr[2];
  548. for (int i = 0; i < props.size(); i++) {
  549. CHECK_TYPE(props[i], ARRAY);
  550. Array prop = props[i];
  551. ERR_FAIL_COND(prop.size() != 6);
  552. CHECK_TYPE(prop[0], STRING);
  553. CHECK_TYPE(prop[1], INT);
  554. CHECK_TYPE(prop[2], INT);
  555. CHECK_TYPE(prop[3], STRING);
  556. CHECK_TYPE(prop[4], INT);
  557. PropertyInfo pinfo;
  558. pinfo.name = prop[0];
  559. pinfo.type = Variant::Type(int(prop[1]));
  560. pinfo.hint = PropertyHint(int(prop[2]));
  561. pinfo.hint_string = prop[3];
  562. pinfo.usage = PropertyUsageFlags(int(prop[4]));
  563. Variant var = prop[5];
  564. if (pinfo.type == Variant::OBJECT) {
  565. if (var.is_zero()) {
  566. var = Ref<Resource>();
  567. } else if (var.get_type() == Variant::OBJECT) {
  568. if (((Object *)var)->is_class("EncodedObjectAsID")) {
  569. var = Object::cast_to<EncodedObjectAsID>(var)->get_object_id();
  570. pinfo.type = var.get_type();
  571. pinfo.hint = PROPERTY_HINT_OBJECT_ID;
  572. pinfo.hint_string = "Object";
  573. }
  574. }
  575. }
  576. properties.push_back(SceneDebuggerProperty(pinfo, var));
  577. }
  578. }
  579. /// SceneDebuggerTree
  580. SceneDebuggerTree::SceneDebuggerTree(Node *p_root) {
  581. // Flatten tree into list, depth first, use stack to avoid recursion.
  582. List<Node *> stack;
  583. stack.push_back(p_root);
  584. bool is_root = true;
  585. const StringName &is_visible_sn = SNAME("is_visible");
  586. const StringName &is_visible_in_tree_sn = SNAME("is_visible_in_tree");
  587. while (stack.size()) {
  588. Node *n = stack.front()->get();
  589. stack.pop_front();
  590. int count = n->get_child_count();
  591. for (int i = 0; i < count; i++) {
  592. stack.push_front(n->get_child(count - i - 1));
  593. }
  594. int view_flags = 0;
  595. if (is_root) {
  596. // Prevent root window visibility from being changed.
  597. is_root = false;
  598. } else if (n->has_method(is_visible_sn)) {
  599. const Variant visible = n->call(is_visible_sn);
  600. if (visible.get_type() == Variant::BOOL) {
  601. view_flags = RemoteNode::VIEW_HAS_VISIBLE_METHOD;
  602. view_flags |= uint8_t(visible) * RemoteNode::VIEW_VISIBLE;
  603. }
  604. if (n->has_method(is_visible_in_tree_sn)) {
  605. const Variant visible_in_tree = n->call(is_visible_in_tree_sn);
  606. if (visible_in_tree.get_type() == Variant::BOOL) {
  607. view_flags |= uint8_t(visible_in_tree) * RemoteNode::VIEW_VISIBLE_IN_TREE;
  608. }
  609. }
  610. }
  611. String class_name;
  612. ScriptInstance *script_instance = n->get_script_instance();
  613. if (script_instance) {
  614. Ref<Script> script = script_instance->get_script();
  615. if (script.is_valid()) {
  616. class_name = script->get_global_name();
  617. if (class_name.is_empty()) {
  618. // If there is no class_name in this script we just take the script path.
  619. class_name = script->get_path();
  620. }
  621. }
  622. }
  623. nodes.push_back(RemoteNode(count, n->get_name(), class_name.is_empty() ? n->get_class() : class_name, n->get_instance_id(), n->get_scene_file_path(), view_flags));
  624. }
  625. }
  626. void SceneDebuggerTree::serialize(Array &p_arr) {
  627. for (const RemoteNode &n : nodes) {
  628. p_arr.push_back(n.child_count);
  629. p_arr.push_back(n.name);
  630. p_arr.push_back(n.type_name);
  631. p_arr.push_back(n.id);
  632. p_arr.push_back(n.scene_file_path);
  633. p_arr.push_back(n.view_flags);
  634. }
  635. }
  636. void SceneDebuggerTree::deserialize(const Array &p_arr) {
  637. int idx = 0;
  638. while (p_arr.size() > idx) {
  639. ERR_FAIL_COND(p_arr.size() < 6);
  640. CHECK_TYPE(p_arr[idx], INT); // child_count.
  641. CHECK_TYPE(p_arr[idx + 1], STRING); // name.
  642. CHECK_TYPE(p_arr[idx + 2], STRING); // type_name.
  643. CHECK_TYPE(p_arr[idx + 3], INT); // id.
  644. CHECK_TYPE(p_arr[idx + 4], STRING); // scene_file_path.
  645. CHECK_TYPE(p_arr[idx + 5], INT); // view_flags.
  646. nodes.push_back(RemoteNode(p_arr[idx], p_arr[idx + 1], p_arr[idx + 2], p_arr[idx + 3], p_arr[idx + 4], p_arr[idx + 5]));
  647. idx += 6;
  648. }
  649. }
  650. /// LiveEditor
  651. LiveEditor *LiveEditor::get_singleton() {
  652. return singleton;
  653. }
  654. void LiveEditor::_send_tree() {
  655. SceneTree *scene_tree = SceneTree::get_singleton();
  656. if (!scene_tree) {
  657. return;
  658. }
  659. Array arr;
  660. // Encoded as a flat list depth first.
  661. SceneDebuggerTree tree(scene_tree->root);
  662. tree.serialize(arr);
  663. EngineDebugger::get_singleton()->send_message("scene:scene_tree", arr);
  664. }
  665. void LiveEditor::_node_path_func(const NodePath &p_path, int p_id) {
  666. live_edit_node_path_cache[p_id] = p_path;
  667. }
  668. void LiveEditor::_res_path_func(const String &p_path, int p_id) {
  669. live_edit_resource_cache[p_id] = p_path;
  670. }
  671. void LiveEditor::_node_set_func(int p_id, const StringName &p_prop, const Variant &p_value) {
  672. SceneTree *scene_tree = SceneTree::get_singleton();
  673. if (!scene_tree) {
  674. return;
  675. }
  676. if (!live_edit_node_path_cache.has(p_id)) {
  677. return;
  678. }
  679. NodePath np = live_edit_node_path_cache[p_id];
  680. Node *base = nullptr;
  681. if (scene_tree->root->has_node(live_edit_root)) {
  682. base = scene_tree->root->get_node(live_edit_root);
  683. }
  684. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  685. if (!E) {
  686. return; //scene not editable
  687. }
  688. for (Node *F : E->value) {
  689. Node *n = F;
  690. if (base && !base->is_ancestor_of(n)) {
  691. continue;
  692. }
  693. if (!n->has_node(np)) {
  694. continue;
  695. }
  696. Node *n2 = n->get_node(np);
  697. // Do not change transform of edited scene root, unless it's the scene being played.
  698. // See GH-86659 for additional context.
  699. bool keep_transform = (n2 == n) && (n2->get_parent() != scene_tree->root);
  700. Variant orig_tf;
  701. if (keep_transform) {
  702. if (n2->is_class("Node3D")) {
  703. orig_tf = n2->call("get_transform");
  704. } else if (n2->is_class("CanvasItem")) {
  705. orig_tf = n2->call("_edit_get_state");
  706. }
  707. }
  708. n2->set(p_prop, p_value);
  709. if (keep_transform) {
  710. if (n2->is_class("Node3D")) {
  711. Variant new_tf = n2->call("get_transform");
  712. if (new_tf != orig_tf) {
  713. n2->call("set_transform", orig_tf);
  714. }
  715. } else if (n2->is_class("CanvasItem")) {
  716. Variant new_tf = n2->call("_edit_get_state");
  717. if (new_tf != orig_tf) {
  718. n2->call("_edit_set_state", orig_tf);
  719. }
  720. }
  721. }
  722. }
  723. }
  724. void LiveEditor::_node_set_res_func(int p_id, const StringName &p_prop, const String &p_value) {
  725. Ref<Resource> r = ResourceLoader::load(p_value);
  726. if (r.is_null()) {
  727. return;
  728. }
  729. _node_set_func(p_id, p_prop, r);
  730. }
  731. void LiveEditor::_node_call_func(int p_id, const StringName &p_method, const Variant **p_args, int p_argcount) {
  732. SceneTree *scene_tree = SceneTree::get_singleton();
  733. if (!scene_tree) {
  734. return;
  735. }
  736. if (!live_edit_node_path_cache.has(p_id)) {
  737. return;
  738. }
  739. NodePath np = live_edit_node_path_cache[p_id];
  740. Node *base = nullptr;
  741. if (scene_tree->root->has_node(live_edit_root)) {
  742. base = scene_tree->root->get_node(live_edit_root);
  743. }
  744. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  745. if (!E) {
  746. return; //scene not editable
  747. }
  748. for (Node *F : E->value) {
  749. Node *n = F;
  750. if (base && !base->is_ancestor_of(n)) {
  751. continue;
  752. }
  753. if (!n->has_node(np)) {
  754. continue;
  755. }
  756. Node *n2 = n->get_node(np);
  757. // Do not change transform of edited scene root, unless it's the scene being played.
  758. // See GH-86659 for additional context.
  759. bool keep_transform = (n2 == n) && (n2->get_parent() != scene_tree->root);
  760. Variant orig_tf;
  761. if (keep_transform) {
  762. if (n2->is_class("Node3D")) {
  763. orig_tf = n2->call("get_transform");
  764. } else if (n2->is_class("CanvasItem")) {
  765. orig_tf = n2->call("_edit_get_state");
  766. }
  767. }
  768. Callable::CallError ce;
  769. n2->callp(p_method, p_args, p_argcount, ce);
  770. if (keep_transform) {
  771. if (n2->is_class("Node3D")) {
  772. Variant new_tf = n2->call("get_transform");
  773. if (new_tf != orig_tf) {
  774. n2->call("set_transform", orig_tf);
  775. }
  776. } else if (n2->is_class("CanvasItem")) {
  777. Variant new_tf = n2->call("_edit_get_state");
  778. if (new_tf != orig_tf) {
  779. n2->call("_edit_set_state", orig_tf);
  780. }
  781. }
  782. }
  783. }
  784. }
  785. void LiveEditor::_res_set_func(int p_id, const StringName &p_prop, const Variant &p_value) {
  786. if (!live_edit_resource_cache.has(p_id)) {
  787. return;
  788. }
  789. String resp = live_edit_resource_cache[p_id];
  790. if (!ResourceCache::has(resp)) {
  791. return;
  792. }
  793. Ref<Resource> r = ResourceCache::get_ref(resp);
  794. if (r.is_null()) {
  795. return;
  796. }
  797. r->set(p_prop, p_value);
  798. }
  799. void LiveEditor::_res_set_res_func(int p_id, const StringName &p_prop, const String &p_value) {
  800. Ref<Resource> r = ResourceLoader::load(p_value);
  801. if (r.is_null()) {
  802. return;
  803. }
  804. _res_set_func(p_id, p_prop, r);
  805. }
  806. void LiveEditor::_res_call_func(int p_id, const StringName &p_method, const Variant **p_args, int p_argcount) {
  807. if (!live_edit_resource_cache.has(p_id)) {
  808. return;
  809. }
  810. String resp = live_edit_resource_cache[p_id];
  811. if (!ResourceCache::has(resp)) {
  812. return;
  813. }
  814. Ref<Resource> r = ResourceCache::get_ref(resp);
  815. if (r.is_null()) {
  816. return;
  817. }
  818. Callable::CallError ce;
  819. r->callp(p_method, p_args, p_argcount, ce);
  820. }
  821. void LiveEditor::_root_func(const NodePath &p_scene_path, const String &p_scene_from) {
  822. live_edit_root = p_scene_path;
  823. live_edit_scene = p_scene_from;
  824. }
  825. void LiveEditor::_create_node_func(const NodePath &p_parent, const String &p_type, const String &p_name) {
  826. SceneTree *scene_tree = SceneTree::get_singleton();
  827. if (!scene_tree) {
  828. return;
  829. }
  830. Node *base = nullptr;
  831. if (scene_tree->root->has_node(live_edit_root)) {
  832. base = scene_tree->root->get_node(live_edit_root);
  833. }
  834. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  835. if (!E) {
  836. return; //scene not editable
  837. }
  838. for (Node *F : E->value) {
  839. Node *n = F;
  840. if (base && !base->is_ancestor_of(n)) {
  841. continue;
  842. }
  843. if (!n->has_node(p_parent)) {
  844. continue;
  845. }
  846. Node *n2 = n->get_node(p_parent);
  847. Node *no = Object::cast_to<Node>(ClassDB::instantiate(p_type));
  848. if (!no) {
  849. continue;
  850. }
  851. no->set_name(p_name);
  852. n2->add_child(no);
  853. }
  854. }
  855. void LiveEditor::_instance_node_func(const NodePath &p_parent, const String &p_path, const String &p_name) {
  856. SceneTree *scene_tree = SceneTree::get_singleton();
  857. if (!scene_tree) {
  858. return;
  859. }
  860. Ref<PackedScene> ps = ResourceLoader::load(p_path);
  861. if (ps.is_null()) {
  862. return;
  863. }
  864. Node *base = nullptr;
  865. if (scene_tree->root->has_node(live_edit_root)) {
  866. base = scene_tree->root->get_node(live_edit_root);
  867. }
  868. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  869. if (!E) {
  870. return; //scene not editable
  871. }
  872. for (Node *F : E->value) {
  873. Node *n = F;
  874. if (base && !base->is_ancestor_of(n)) {
  875. continue;
  876. }
  877. if (!n->has_node(p_parent)) {
  878. continue;
  879. }
  880. Node *n2 = n->get_node(p_parent);
  881. Node *no = ps->instantiate();
  882. if (!no) {
  883. continue;
  884. }
  885. no->set_name(p_name);
  886. n2->add_child(no);
  887. }
  888. }
  889. void LiveEditor::_remove_node_func(const NodePath &p_at) {
  890. SceneTree *scene_tree = SceneTree::get_singleton();
  891. if (!scene_tree) {
  892. return;
  893. }
  894. Node *base = nullptr;
  895. if (scene_tree->root->has_node(live_edit_root)) {
  896. base = scene_tree->root->get_node(live_edit_root);
  897. }
  898. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  899. if (!E) {
  900. return; //scene not editable
  901. }
  902. Vector<Node *> to_delete;
  903. for (HashSet<Node *>::Iterator F = E->value.begin(); F; ++F) {
  904. Node *n = *F;
  905. if (base && !base->is_ancestor_of(n)) {
  906. continue;
  907. }
  908. if (!n->has_node(p_at)) {
  909. continue;
  910. }
  911. Node *n2 = n->get_node(p_at);
  912. to_delete.push_back(n2);
  913. }
  914. for (int i = 0; i < to_delete.size(); i++) {
  915. memdelete(to_delete[i]);
  916. }
  917. }
  918. void LiveEditor::_remove_and_keep_node_func(const NodePath &p_at, ObjectID p_keep_id) {
  919. SceneTree *scene_tree = SceneTree::get_singleton();
  920. if (!scene_tree) {
  921. return;
  922. }
  923. Node *base = nullptr;
  924. if (scene_tree->root->has_node(live_edit_root)) {
  925. base = scene_tree->root->get_node(live_edit_root);
  926. }
  927. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  928. if (!E) {
  929. return; //scene not editable
  930. }
  931. Vector<Node *> to_remove;
  932. for (HashSet<Node *>::Iterator F = E->value.begin(); F; ++F) {
  933. Node *n = *F;
  934. if (base && !base->is_ancestor_of(n)) {
  935. continue;
  936. }
  937. if (!n->has_node(p_at)) {
  938. continue;
  939. }
  940. to_remove.push_back(n);
  941. }
  942. for (int i = 0; i < to_remove.size(); i++) {
  943. Node *n = to_remove[i];
  944. Node *n2 = n->get_node(p_at);
  945. n2->get_parent()->remove_child(n2);
  946. live_edit_remove_list[n][p_keep_id] = n2;
  947. }
  948. }
  949. void LiveEditor::_restore_node_func(ObjectID p_id, const NodePath &p_at, int p_at_pos) {
  950. SceneTree *scene_tree = SceneTree::get_singleton();
  951. if (!scene_tree) {
  952. return;
  953. }
  954. Node *base = nullptr;
  955. if (scene_tree->root->has_node(live_edit_root)) {
  956. base = scene_tree->root->get_node(live_edit_root);
  957. }
  958. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  959. if (!E) {
  960. return; //scene not editable
  961. }
  962. for (HashSet<Node *>::Iterator F = E->value.begin(); F;) {
  963. HashSet<Node *>::Iterator N = F;
  964. ++N;
  965. Node *n = *F;
  966. if (base && !base->is_ancestor_of(n)) {
  967. continue;
  968. }
  969. if (!n->has_node(p_at)) {
  970. continue;
  971. }
  972. Node *n2 = n->get_node(p_at);
  973. HashMap<Node *, HashMap<ObjectID, Node *>>::Iterator EN = live_edit_remove_list.find(n);
  974. if (!EN) {
  975. continue;
  976. }
  977. HashMap<ObjectID, Node *>::Iterator FN = EN->value.find(p_id);
  978. if (!FN) {
  979. continue;
  980. }
  981. n2->add_child(FN->value);
  982. EN->value.remove(FN);
  983. if (EN->value.size() == 0) {
  984. live_edit_remove_list.remove(EN);
  985. }
  986. F = N;
  987. }
  988. }
  989. void LiveEditor::_duplicate_node_func(const NodePath &p_at, const String &p_new_name) {
  990. SceneTree *scene_tree = SceneTree::get_singleton();
  991. if (!scene_tree) {
  992. return;
  993. }
  994. Node *base = nullptr;
  995. if (scene_tree->root->has_node(live_edit_root)) {
  996. base = scene_tree->root->get_node(live_edit_root);
  997. }
  998. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  999. if (!E) {
  1000. return; //scene not editable
  1001. }
  1002. for (Node *F : E->value) {
  1003. Node *n = F;
  1004. if (base && !base->is_ancestor_of(n)) {
  1005. continue;
  1006. }
  1007. if (!n->has_node(p_at)) {
  1008. continue;
  1009. }
  1010. Node *n2 = n->get_node(p_at);
  1011. Node *dup = n2->duplicate(Node::DUPLICATE_SIGNALS | Node::DUPLICATE_GROUPS | Node::DUPLICATE_SCRIPTS);
  1012. if (!dup) {
  1013. continue;
  1014. }
  1015. dup->set_name(p_new_name);
  1016. n2->get_parent()->add_child(dup);
  1017. }
  1018. }
  1019. void LiveEditor::_reparent_node_func(const NodePath &p_at, const NodePath &p_new_place, const String &p_new_name, int p_at_pos) {
  1020. SceneTree *scene_tree = SceneTree::get_singleton();
  1021. if (!scene_tree) {
  1022. return;
  1023. }
  1024. Node *base = nullptr;
  1025. if (scene_tree->root->has_node(live_edit_root)) {
  1026. base = scene_tree->root->get_node(live_edit_root);
  1027. }
  1028. HashMap<String, HashSet<Node *>>::Iterator E = live_scene_edit_cache.find(live_edit_scene);
  1029. if (!E) {
  1030. return; //scene not editable
  1031. }
  1032. for (Node *F : E->value) {
  1033. Node *n = F;
  1034. if (base && !base->is_ancestor_of(n)) {
  1035. continue;
  1036. }
  1037. if (!n->has_node(p_at)) {
  1038. continue;
  1039. }
  1040. Node *nfrom = n->get_node(p_at);
  1041. if (!n->has_node(p_new_place)) {
  1042. continue;
  1043. }
  1044. Node *nto = n->get_node(p_new_place);
  1045. nfrom->get_parent()->remove_child(nfrom);
  1046. nfrom->set_name(p_new_name);
  1047. nto->add_child(nfrom);
  1048. if (p_at_pos >= 0) {
  1049. nto->move_child(nfrom, p_at_pos);
  1050. }
  1051. }
  1052. }
  1053. /// RuntimeNodeSelect
  1054. RuntimeNodeSelect *RuntimeNodeSelect::get_singleton() {
  1055. return singleton;
  1056. }
  1057. RuntimeNodeSelect::~RuntimeNodeSelect() {
  1058. if (selection_list && !selection_list->is_visible()) {
  1059. memdelete(selection_list);
  1060. }
  1061. if (draw_canvas.is_valid()) {
  1062. RS::get_singleton()->free(sel_drag_ci);
  1063. RS::get_singleton()->free(sbox_2d_ci);
  1064. RS::get_singleton()->free(draw_canvas);
  1065. }
  1066. }
  1067. void RuntimeNodeSelect::_setup(const Dictionary &p_settings) {
  1068. Window *root = SceneTree::get_singleton()->get_root();
  1069. ERR_FAIL_COND(root->is_connected(SceneStringName(window_input), callable_mp(this, &RuntimeNodeSelect::_root_window_input)));
  1070. root->connect(SceneStringName(window_input), callable_mp(this, &RuntimeNodeSelect::_root_window_input));
  1071. root->connect("size_changed", callable_mp(this, &RuntimeNodeSelect::_queue_selection_update), CONNECT_DEFERRED);
  1072. max_selection = p_settings.get("debugger/max_node_selection", 1);
  1073. panner.instantiate();
  1074. panner->set_callbacks(callable_mp(this, &RuntimeNodeSelect::_pan_callback), callable_mp(this, &RuntimeNodeSelect::_zoom_callback));
  1075. ViewPanner::ControlScheme panning_scheme = (ViewPanner::ControlScheme)p_settings.get("editors/panning/2d_editor_panning_scheme", 0).operator int();
  1076. bool simple_panning = p_settings.get("editors/panning/simple_panning", false);
  1077. int pan_speed = p_settings.get("editors/panning/2d_editor_pan_speed", 20);
  1078. Array keys = p_settings.get("canvas_item_editor/pan_view", Array()).operator Array();
  1079. panner->setup(panning_scheme, DebuggerMarshalls::deserialize_key_shortcut(keys), simple_panning);
  1080. panner->setup_warped_panning(root, p_settings.get("editors/panning/warped_mouse_panning", true));
  1081. panner->set_scroll_speed(pan_speed);
  1082. sel_2d_grab_dist = p_settings.get("editors/polygon_editor/point_grab_radius", 0);
  1083. selection_area_fill = p_settings.get("box_selection_fill_color", Color());
  1084. selection_area_outline = p_settings.get("box_selection_stroke_color", Color());
  1085. draw_canvas = RS::get_singleton()->canvas_create();
  1086. sel_drag_ci = RS::get_singleton()->canvas_item_create();
  1087. /// 2D Selection Box Generation
  1088. sbox_2d_ci = RS::get_singleton()->canvas_item_create();
  1089. RS::get_singleton()->viewport_attach_canvas(root->get_viewport_rid(), draw_canvas);
  1090. RS::get_singleton()->canvas_item_set_parent(sel_drag_ci, draw_canvas);
  1091. RS::get_singleton()->canvas_item_set_parent(sbox_2d_ci, draw_canvas);
  1092. #ifndef _3D_DISABLED
  1093. cursor = Cursor();
  1094. camera_fov = p_settings.get("editors/3d/default_fov", 70);
  1095. camera_znear = p_settings.get("editors/3d/default_z_near", 0.05);
  1096. camera_zfar = p_settings.get("editors/3d/default_z_far", 4'000);
  1097. invert_x_axis = p_settings.get("editors/3d/navigation/invert_x_axis", false);
  1098. invert_y_axis = p_settings.get("editors/3d/navigation/invert_y_axis", false);
  1099. warped_mouse_panning_3d = p_settings.get("editors/3d/navigation/warped_mouse_panning", true);
  1100. freelook_base_speed = p_settings.get("editors/3d/freelook/freelook_base_speed", 5);
  1101. freelook_sensitivity = Math::deg_to_rad((real_t)p_settings.get("editors/3d/freelook/freelook_sensitivity", 0.25));
  1102. orbit_sensitivity = Math::deg_to_rad((real_t)p_settings.get("editors/3d/navigation_feel/orbit_sensitivity", 0.004));
  1103. translation_sensitivity = p_settings.get("editors/3d/navigation_feel/translation_sensitivity", 1);
  1104. /// 3D Selection Box Generation
  1105. // Copied from the Node3DEditor implementation.
  1106. sbox_3d_color = p_settings.get("editors/3d/selection_box_color", Color());
  1107. // Use two AABBs to create the illusion of a slightly thicker line.
  1108. AABB aabb(Vector3(), Vector3(1, 1, 1));
  1109. // Create a x-ray (visible through solid surfaces) and standard version of the selection box.
  1110. // Both will be drawn at the same position, but with different opacity.
  1111. // This lets the user see where the selection is while still having a sense of depth.
  1112. Ref<SurfaceTool> st = memnew(SurfaceTool);
  1113. Ref<SurfaceTool> st_xray = memnew(SurfaceTool);
  1114. st->begin(Mesh::PRIMITIVE_LINES);
  1115. st_xray->begin(Mesh::PRIMITIVE_LINES);
  1116. for (int i = 0; i < 12; i++) {
  1117. Vector3 a, b;
  1118. aabb.get_edge(i, a, b);
  1119. st->add_vertex(a);
  1120. st->add_vertex(b);
  1121. st_xray->add_vertex(a);
  1122. st_xray->add_vertex(b);
  1123. }
  1124. Ref<StandardMaterial3D> mat = memnew(StandardMaterial3D);
  1125. mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED);
  1126. mat->set_flag(StandardMaterial3D::FLAG_DISABLE_FOG, true);
  1127. mat->set_albedo(sbox_3d_color);
  1128. mat->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA);
  1129. st->set_material(mat);
  1130. sbox_3d_mesh = st->commit();
  1131. Ref<StandardMaterial3D> mat_xray = memnew(StandardMaterial3D);
  1132. mat_xray->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED);
  1133. mat_xray->set_flag(StandardMaterial3D::FLAG_DISABLE_FOG, true);
  1134. mat_xray->set_flag(StandardMaterial3D::FLAG_DISABLE_DEPTH_TEST, true);
  1135. mat_xray->set_albedo(sbox_3d_color * Color(1, 1, 1, 0.15));
  1136. mat_xray->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA);
  1137. st_xray->set_material(mat_xray);
  1138. sbox_3d_mesh_xray = st_xray->commit();
  1139. #endif // _3D_DISABLED
  1140. SceneTree::get_singleton()->connect("process_frame", callable_mp(this, &RuntimeNodeSelect::_process_frame));
  1141. SceneTree::get_singleton()->connect("physics_frame", callable_mp(this, &RuntimeNodeSelect::_physics_frame));
  1142. // This function will be called before the root enters the tree at first when the Game view is passing its settings to
  1143. // the debugger, so queue the update for after it enters.
  1144. root->connect(SceneStringName(tree_entered), callable_mp(this, &RuntimeNodeSelect::_update_input_state), Object::CONNECT_ONE_SHOT);
  1145. }
  1146. void RuntimeNodeSelect::_node_set_type(NodeType p_type) {
  1147. node_select_type = p_type;
  1148. _update_input_state();
  1149. }
  1150. void RuntimeNodeSelect::_select_set_mode(SelectMode p_mode) {
  1151. node_select_mode = p_mode;
  1152. }
  1153. void RuntimeNodeSelect::_set_camera_override_enabled(bool p_enabled) {
  1154. camera_override = p_enabled;
  1155. if (p_enabled) {
  1156. _update_view_2d();
  1157. }
  1158. #ifndef _3D_DISABLED
  1159. if (camera_first_override) {
  1160. _reset_camera_2d();
  1161. _reset_camera_3d();
  1162. camera_first_override = false;
  1163. } else if (p_enabled) {
  1164. _update_view_2d();
  1165. SceneTree::get_singleton()->get_root()->set_camera_3d_override_transform(_get_cursor_transform());
  1166. SceneTree::get_singleton()->get_root()->set_camera_3d_override_perspective(camera_fov * cursor.fov_scale, camera_znear, camera_zfar);
  1167. }
  1168. #endif // _3D_DISABLED
  1169. }
  1170. void RuntimeNodeSelect::_root_window_input(const Ref<InputEvent> &p_event) {
  1171. Window *root = SceneTree::get_singleton()->get_root();
  1172. if (node_select_type == NODE_TYPE_NONE || (selection_list && selection_list->is_visible())) {
  1173. // Workaround for platforms that don't allow subwindows.
  1174. if (selection_list && selection_list->is_visible() && selection_list->is_embedded()) {
  1175. root->set_disable_input_override(false);
  1176. selection_list->push_input(p_event);
  1177. callable_mp(root->get_viewport(), &Viewport::set_disable_input_override).call_deferred(true);
  1178. }
  1179. return;
  1180. }
  1181. bool is_dragging_camera = false;
  1182. if (camera_override) {
  1183. if (node_select_type == NODE_TYPE_2D) {
  1184. is_dragging_camera = panner->gui_input(p_event, Rect2(Vector2(), root->get_visible_rect().get_size()));
  1185. } else if (node_select_type == NODE_TYPE_3D && selection_drag_state == SELECTION_DRAG_NONE) {
  1186. #ifndef _3D_DISABLED
  1187. if (_handle_3d_input(p_event)) {
  1188. return;
  1189. }
  1190. #endif // _3D_DISABLED
  1191. }
  1192. }
  1193. Ref<InputEventMouseButton> b = p_event;
  1194. if (selection_drag_state == SELECTION_DRAG_MOVE) {
  1195. Ref<InputEventMouseMotion> m = p_event;
  1196. if (m.is_valid()) {
  1197. _update_selection_drag(root->get_screen_transform().affine_inverse().xform(m->get_position()));
  1198. return;
  1199. } else if (b.is_valid()) {
  1200. // Account for actions like zooming.
  1201. _update_selection_drag(root->get_screen_transform().affine_inverse().xform(b->get_position()));
  1202. }
  1203. }
  1204. if (b.is_null()) {
  1205. return;
  1206. }
  1207. // Ignore mouse wheel inputs.
  1208. if (b->get_button_index() != MouseButton::LEFT && b->get_button_index() != MouseButton::RIGHT) {
  1209. return;
  1210. }
  1211. if (selection_drag_state == SELECTION_DRAG_MOVE && !b->is_pressed() && b->get_button_index() == MouseButton::LEFT) {
  1212. selection_drag_state = SELECTION_DRAG_END;
  1213. selection_drag_area = selection_drag_area.abs();
  1214. _update_selection_drag();
  1215. // Trigger a selection in the position on release.
  1216. if (multi_shortcut_pressed) {
  1217. selection_position = root->get_screen_transform().affine_inverse().xform(b->get_position());
  1218. }
  1219. }
  1220. if (!is_dragging_camera && b->is_pressed()) {
  1221. multi_shortcut_pressed = b->is_shift_pressed();
  1222. list_shortcut_pressed = node_select_mode == SELECT_MODE_SINGLE && b->get_button_index() == MouseButton::RIGHT && b->is_alt_pressed();
  1223. if (list_shortcut_pressed || b->get_button_index() == MouseButton::LEFT) {
  1224. selection_position = root->get_screen_transform().affine_inverse().xform(b->get_position());
  1225. }
  1226. }
  1227. }
  1228. void RuntimeNodeSelect::_items_popup_index_pressed(int p_index, PopupMenu *p_popup) {
  1229. Object *obj = p_popup->get_item_metadata(p_index).get_validated_object();
  1230. if (obj) {
  1231. Vector<Node *> node;
  1232. node.append(Object::cast_to<Node>(obj));
  1233. _send_ids(node);
  1234. }
  1235. }
  1236. void RuntimeNodeSelect::_update_input_state() {
  1237. SceneTree *scene_tree = SceneTree::get_singleton();
  1238. // This function can be called at the very beginning, when the root hasn't entered the tree yet.
  1239. // So check first to avoid a crash.
  1240. if (!scene_tree->get_root()->is_inside_tree()) {
  1241. return;
  1242. }
  1243. bool disable_input = scene_tree->is_suspended() || node_select_type != RuntimeNodeSelect::NODE_TYPE_NONE;
  1244. Input::get_singleton()->set_disable_input(disable_input);
  1245. Input::get_singleton()->set_mouse_mode_override_enabled(disable_input);
  1246. scene_tree->get_root()->set_disable_input_override(disable_input);
  1247. }
  1248. void RuntimeNodeSelect::_process_frame() {
  1249. #ifndef _3D_DISABLED
  1250. if (camera_freelook) {
  1251. Transform3D transform = _get_cursor_transform();
  1252. Vector3 forward = transform.basis.xform(Vector3(0, 0, -1));
  1253. const Vector3 right = transform.basis.xform(Vector3(1, 0, 0));
  1254. Vector3 up = transform.basis.xform(Vector3(0, 1, 0));
  1255. Vector3 direction;
  1256. Input *input = Input::get_singleton();
  1257. bool was_input_disabled = input->is_input_disabled();
  1258. if (was_input_disabled) {
  1259. input->set_disable_input(false);
  1260. }
  1261. if (input->is_physical_key_pressed(Key::A)) {
  1262. direction -= right;
  1263. }
  1264. if (input->is_physical_key_pressed(Key::D)) {
  1265. direction += right;
  1266. }
  1267. if (input->is_physical_key_pressed(Key::W)) {
  1268. direction += forward;
  1269. }
  1270. if (input->is_physical_key_pressed(Key::S)) {
  1271. direction -= forward;
  1272. }
  1273. if (input->is_physical_key_pressed(Key::E)) {
  1274. direction += up;
  1275. }
  1276. if (input->is_physical_key_pressed(Key::Q)) {
  1277. direction -= up;
  1278. }
  1279. real_t speed = freelook_base_speed;
  1280. if (input->is_physical_key_pressed(Key::SHIFT)) {
  1281. speed *= 3.0;
  1282. }
  1283. if (input->is_physical_key_pressed(Key::ALT)) {
  1284. speed *= 0.333333;
  1285. }
  1286. if (was_input_disabled) {
  1287. input->set_disable_input(true);
  1288. }
  1289. if (direction != Vector3()) {
  1290. // Calculate the process time manually, as the time scale is frozen.
  1291. const double process_time = (1.0 / Engine::get_singleton()->get_frames_per_second()) * Engine::get_singleton()->get_unfrozen_time_scale();
  1292. const Vector3 motion = direction * speed * process_time;
  1293. cursor.pos += motion;
  1294. cursor.eye_pos += motion;
  1295. SceneTree::get_singleton()->get_root()->set_camera_3d_override_transform(_get_cursor_transform());
  1296. }
  1297. }
  1298. #endif // _3D_DISABLED
  1299. if (selection_update_queued || !SceneTree::get_singleton()->is_suspended()) {
  1300. selection_update_queued = false;
  1301. if (has_selection) {
  1302. _update_selection();
  1303. }
  1304. }
  1305. }
  1306. void RuntimeNodeSelect::_physics_frame() {
  1307. if (selection_drag_state != SELECTION_DRAG_END && (selection_drag_state == SELECTION_DRAG_MOVE || Math::is_inf(selection_position.x))) {
  1308. return;
  1309. }
  1310. Window *root = SceneTree::get_singleton()->get_root();
  1311. bool selection_drag_valid = selection_drag_state == SELECTION_DRAG_END && selection_drag_area.get_area() > SELECTION_MIN_AREA;
  1312. Vector<SelectResult> items;
  1313. if (node_select_type == NODE_TYPE_2D) {
  1314. if (selection_drag_valid) {
  1315. for (int i = 0; i < root->get_child_count(); i++) {
  1316. _find_canvas_items_at_rect(selection_drag_area, root->get_child(i), items);
  1317. }
  1318. } else if (!Math::is_inf(selection_position.x)) {
  1319. for (int i = 0; i < root->get_child_count(); i++) {
  1320. _find_canvas_items_at_pos(selection_position, root->get_child(i), items);
  1321. }
  1322. }
  1323. // Remove possible duplicates.
  1324. for (int i = 0; i < items.size(); i++) {
  1325. Node *item = items[i].item;
  1326. for (int j = 0; j < i; j++) {
  1327. if (items[j].item == item) {
  1328. items.remove_at(i);
  1329. i--;
  1330. break;
  1331. }
  1332. }
  1333. }
  1334. } else if (node_select_type == NODE_TYPE_3D) {
  1335. #ifndef _3D_DISABLED
  1336. if (selection_drag_valid) {
  1337. _find_3d_items_at_rect(selection_drag_area, items);
  1338. } else {
  1339. _find_3d_items_at_pos(selection_position, items);
  1340. }
  1341. #endif // _3D_DISABLED
  1342. }
  1343. items.sort();
  1344. switch (selection_drag_state) {
  1345. case SELECTION_DRAG_END: {
  1346. selection_position = Point2(INFINITY, INFINITY);
  1347. selection_drag_state = SELECTION_DRAG_NONE;
  1348. if (selection_drag_area.get_area() > SELECTION_MIN_AREA) {
  1349. if (!items.is_empty()) {
  1350. Vector<Node *> nodes;
  1351. for (const SelectResult item : items) {
  1352. nodes.append(item.item);
  1353. }
  1354. _send_ids(nodes, false);
  1355. }
  1356. _update_selection_drag();
  1357. return;
  1358. }
  1359. _update_selection_drag();
  1360. } break;
  1361. case SELECTION_DRAG_NONE: {
  1362. if (node_select_mode == SELECT_MODE_LIST) {
  1363. break;
  1364. }
  1365. if (multi_shortcut_pressed) {
  1366. // Allow forcing box selection when an item was clicked.
  1367. selection_drag_state = SELECTION_DRAG_MOVE;
  1368. } else if (items.is_empty()) {
  1369. #ifdef _3D_DISABLED
  1370. if (!selected_ci_nodes.is_empty()) {
  1371. #else
  1372. if (!selected_ci_nodes.is_empty() || !selected_3d_nodes.is_empty()) {
  1373. #endif // _3D_DISABLED
  1374. EngineDebugger::get_singleton()->send_message("remote_nothing_selected", Array());
  1375. _clear_selection();
  1376. }
  1377. selection_drag_state = SELECTION_DRAG_MOVE;
  1378. } else {
  1379. break;
  1380. }
  1381. [[fallthrough]];
  1382. }
  1383. case SELECTION_DRAG_MOVE: {
  1384. selection_drag_area.position = selection_position;
  1385. // Stop selection on click, so it can happen on release if the selection area doesn't pass the threshold.
  1386. if (multi_shortcut_pressed) {
  1387. return;
  1388. }
  1389. }
  1390. }
  1391. if (items.is_empty()) {
  1392. selection_position = Point2(INFINITY, INFINITY);
  1393. return;
  1394. }
  1395. if ((!list_shortcut_pressed && node_select_mode == SELECT_MODE_SINGLE) || items.size() == 1) {
  1396. selection_position = Point2(INFINITY, INFINITY);
  1397. Vector<Node *> node;
  1398. node.append(items[0].item);
  1399. _send_ids(node);
  1400. return;
  1401. }
  1402. if (!selection_list && (list_shortcut_pressed || node_select_mode == SELECT_MODE_LIST)) {
  1403. _open_selection_list(items, selection_position);
  1404. }
  1405. selection_position = Point2(INFINITY, INFINITY);
  1406. }
  1407. void RuntimeNodeSelect::_send_ids(const Vector<Node *> &p_picked_nodes, bool p_invert_new_selections) {
  1408. ERR_FAIL_COND(p_picked_nodes.is_empty());
  1409. Vector<Node *> picked_nodes = p_picked_nodes;
  1410. Array message;
  1411. if (!multi_shortcut_pressed) {
  1412. if (picked_nodes.size() > max_selection) {
  1413. picked_nodes.resize(max_selection);
  1414. EngineDebugger::get_singleton()->send_message("show_selection_limit_warning", Array());
  1415. }
  1416. for (const Node *node : picked_nodes) {
  1417. SceneDebuggerObject obj(node->get_instance_id());
  1418. Array arr;
  1419. obj.serialize(arr);
  1420. message.append(arr);
  1421. }
  1422. EngineDebugger::get_singleton()->send_message("remote_objects_selected", message);
  1423. _set_selected_nodes(picked_nodes);
  1424. return;
  1425. }
  1426. int limit = max_selection - selected_ci_nodes.size();
  1427. #ifndef _3D_DISABLED
  1428. limit -= selected_3d_nodes.size();
  1429. #endif // _3D_DISABLED
  1430. if (limit <= 0) {
  1431. return;
  1432. }
  1433. if (picked_nodes.size() > limit) {
  1434. picked_nodes.resize(limit);
  1435. EngineDebugger::get_singleton()->send_message("show_selection_limit_warning", Array());
  1436. }
  1437. LocalVector<Node *> nodes;
  1438. LocalVector<ObjectID> ids;
  1439. for (Node *node : picked_nodes) {
  1440. ObjectID id = node->get_instance_id();
  1441. if (CanvasItem *ci = Object::cast_to<CanvasItem>(node)) {
  1442. if (selected_ci_nodes.has(id)) {
  1443. if (p_invert_new_selections) {
  1444. selected_ci_nodes.erase(id);
  1445. }
  1446. } else {
  1447. ids.push_back(id);
  1448. nodes.push_back(ci);
  1449. }
  1450. } else {
  1451. #ifndef _3D_DISABLED
  1452. if (Node3D *node3d = Object::cast_to<Node3D>(node)) {
  1453. if (selected_3d_nodes.has(id)) {
  1454. if (p_invert_new_selections) {
  1455. selected_3d_nodes.erase(id);
  1456. }
  1457. } else {
  1458. ids.push_back(id);
  1459. nodes.push_back(node3d);
  1460. }
  1461. }
  1462. #endif // _3D_DISABLED
  1463. }
  1464. }
  1465. for (ObjectID id : selected_ci_nodes) {
  1466. ids.push_back(id);
  1467. nodes.push_back(ObjectDB::get_instance<Node>(id));
  1468. }
  1469. #ifndef _3D_DISABLED
  1470. for (const KeyValue<ObjectID, Ref<SelectionBox3D>> &KV : selected_3d_nodes) {
  1471. ids.push_back(KV.key);
  1472. nodes.push_back(ObjectDB::get_instance<Node>(KV.key));
  1473. }
  1474. #endif // _3D_DISABLED
  1475. if (ids.size() > (unsigned)max_selection) {
  1476. ids.resize(max_selection);
  1477. EngineDebugger::get_singleton()->send_message("show_selection_limit_warning", Array());
  1478. }
  1479. if (ids.is_empty()) {
  1480. EngineDebugger::get_singleton()->send_message("remote_nothing_selected", message);
  1481. } else {
  1482. for (const ObjectID &id : ids) {
  1483. SceneDebuggerObject obj(id);
  1484. Array arr;
  1485. obj.serialize(arr);
  1486. message.append(arr);
  1487. }
  1488. EngineDebugger::get_singleton()->send_message("remote_objects_selected", message);
  1489. }
  1490. _set_selected_nodes(nodes);
  1491. }
  1492. void RuntimeNodeSelect::_set_selected_nodes(const Vector<Node *> &p_nodes) {
  1493. if (p_nodes.is_empty()) {
  1494. _clear_selection();
  1495. return;
  1496. }
  1497. bool changed = false;
  1498. LocalVector<ObjectID> nodes_ci;
  1499. #ifndef _3D_DISABLED
  1500. HashMap<ObjectID, Ref<SelectionBox3D>> nodes_3d;
  1501. #endif // _3D_DISABLED
  1502. for (Node *node : p_nodes) {
  1503. ObjectID id = node->get_instance_id();
  1504. if (Object::cast_to<CanvasItem>(node)) {
  1505. if (!changed || !selected_ci_nodes.has(id)) {
  1506. changed = true;
  1507. }
  1508. nodes_ci.push_back(id);
  1509. } else {
  1510. #ifndef _3D_DISABLED
  1511. Node3D *node_3d = Object::cast_to<Node3D>(node);
  1512. if (!node_3d || !node_3d->is_inside_world()) {
  1513. continue;
  1514. }
  1515. if (!changed || !selected_3d_nodes.has(id)) {
  1516. changed = true;
  1517. }
  1518. if (selected_3d_nodes.has(id)) {
  1519. // Assign an already available visual instance.
  1520. nodes_3d[id] = selected_3d_nodes.get(id);
  1521. continue;
  1522. }
  1523. if (sbox_3d_mesh.is_null() || sbox_3d_mesh_xray.is_null()) {
  1524. continue;
  1525. }
  1526. Ref<SelectionBox3D> sb;
  1527. sb.instantiate();
  1528. nodes_3d[id] = sb;
  1529. RID scenario = node_3d->get_world_3d()->get_scenario();
  1530. sb->instance = RS::get_singleton()->instance_create2(sbox_3d_mesh->get_rid(), scenario);
  1531. sb->instance_ofs = RS::get_singleton()->instance_create2(sbox_3d_mesh->get_rid(), scenario);
  1532. RS::get_singleton()->instance_geometry_set_cast_shadows_setting(sb->instance, RS::SHADOW_CASTING_SETTING_OFF);
  1533. RS::get_singleton()->instance_geometry_set_cast_shadows_setting(sb->instance_ofs, RS::SHADOW_CASTING_SETTING_OFF);
  1534. RS::get_singleton()->instance_geometry_set_flag(sb->instance, RS::INSTANCE_FLAG_IGNORE_OCCLUSION_CULLING, true);
  1535. RS::get_singleton()->instance_geometry_set_flag(sb->instance, RS::INSTANCE_FLAG_USE_BAKED_LIGHT, false);
  1536. RS::get_singleton()->instance_geometry_set_flag(sb->instance_ofs, RS::INSTANCE_FLAG_IGNORE_OCCLUSION_CULLING, true);
  1537. RS::get_singleton()->instance_geometry_set_flag(sb->instance_ofs, RS::INSTANCE_FLAG_USE_BAKED_LIGHT, false);
  1538. sb->instance_xray = RS::get_singleton()->instance_create2(sbox_3d_mesh_xray->get_rid(), scenario);
  1539. sb->instance_xray_ofs = RS::get_singleton()->instance_create2(sbox_3d_mesh_xray->get_rid(), scenario);
  1540. RS::get_singleton()->instance_geometry_set_cast_shadows_setting(sb->instance_xray, RS::SHADOW_CASTING_SETTING_OFF);
  1541. RS::get_singleton()->instance_geometry_set_cast_shadows_setting(sb->instance_xray_ofs, RS::SHADOW_CASTING_SETTING_OFF);
  1542. RS::get_singleton()->instance_geometry_set_flag(sb->instance_xray, RS::INSTANCE_FLAG_IGNORE_OCCLUSION_CULLING, true);
  1543. RS::get_singleton()->instance_geometry_set_flag(sb->instance_xray, RS::INSTANCE_FLAG_USE_BAKED_LIGHT, false);
  1544. RS::get_singleton()->instance_geometry_set_flag(sb->instance_xray_ofs, RS::INSTANCE_FLAG_IGNORE_OCCLUSION_CULLING, true);
  1545. RS::get_singleton()->instance_geometry_set_flag(sb->instance_xray_ofs, RS::INSTANCE_FLAG_USE_BAKED_LIGHT, false);
  1546. #endif // _3D_DISABLED
  1547. }
  1548. }
  1549. #ifdef _3D_DISABLED
  1550. if (!changed && nodes_ci.size() == selected_ci_nodes.size()) {
  1551. return;
  1552. }
  1553. #else
  1554. if (!changed && nodes_ci.size() == selected_ci_nodes.size() && nodes_3d.size() == selected_3d_nodes.size()) {
  1555. return;
  1556. }
  1557. #endif // _3D_DISABLED
  1558. _clear_selection();
  1559. selected_ci_nodes = nodes_ci;
  1560. has_selection = !nodes_ci.is_empty();
  1561. #ifndef _3D_DISABLED
  1562. if (!nodes_3d.is_empty()) {
  1563. selected_3d_nodes = nodes_3d;
  1564. has_selection = true;
  1565. }
  1566. #endif // _3D_DISABLED
  1567. _queue_selection_update();
  1568. }
  1569. void RuntimeNodeSelect::_queue_selection_update() {
  1570. if (has_selection && selection_visible) {
  1571. if (SceneTree::get_singleton()->is_suspended()) {
  1572. _update_selection();
  1573. } else {
  1574. selection_update_queued = true;
  1575. }
  1576. }
  1577. }
  1578. void RuntimeNodeSelect::_update_selection() {
  1579. Window *root = SceneTree::get_singleton()->get_root();
  1580. RS::get_singleton()->canvas_item_clear(sbox_2d_ci);
  1581. RS::get_singleton()->canvas_item_set_visible(sbox_2d_ci, selection_visible);
  1582. for (LocalVector<ObjectID>::Iterator E = selected_ci_nodes.begin(); E != selected_ci_nodes.end(); ++E) {
  1583. ObjectID id = *E;
  1584. CanvasItem *ci = ObjectDB::get_instance<CanvasItem>(id);
  1585. if (!ci) {
  1586. selected_ci_nodes.erase(id);
  1587. --E;
  1588. continue;
  1589. }
  1590. Transform2D xform;
  1591. // Cameras (overridden or not) don't affect `CanvasLayer`s.
  1592. if (root->is_canvas_transform_override_enabled() && !(ci->get_canvas_layer_node() && !ci->get_canvas_layer_node()->is_following_viewport())) {
  1593. xform = root->get_canvas_transform_override() * ci->get_global_transform();
  1594. } else {
  1595. xform = ci->get_global_transform_with_canvas();
  1596. }
  1597. // Fallback.
  1598. Rect2 rect = Rect2(Vector2(), Vector2(10, 10));
  1599. if (ci->_edit_use_rect()) {
  1600. rect = ci->_edit_get_rect();
  1601. } else {
  1602. #ifndef PHYSICS_2D_DISABLED
  1603. CollisionShape2D *collision_shape = Object::cast_to<CollisionShape2D>(ci);
  1604. if (collision_shape) {
  1605. Ref<Shape2D> shape = collision_shape->get_shape();
  1606. if (shape.is_valid()) {
  1607. rect = shape->get_rect();
  1608. }
  1609. }
  1610. #endif // PHYSICS_2D_DISABLED
  1611. }
  1612. const Vector2 endpoints[4] = {
  1613. xform.xform(rect.position),
  1614. xform.xform(rect.position + Point2(rect.size.x, 0)),
  1615. xform.xform(rect.position + rect.size),
  1616. xform.xform(rect.position + Point2(0, rect.size.y))
  1617. };
  1618. const Color selection_color_2d = Color(1, 0.6, 0.4, 0.7);
  1619. for (int i = 0; i < 4; i++) {
  1620. RS::get_singleton()->canvas_item_add_line(sbox_2d_ci, endpoints[i], endpoints[(i + 1) % 4], selection_color_2d, 2);
  1621. }
  1622. }
  1623. #ifndef _3D_DISABLED
  1624. for (HashMap<ObjectID, Ref<SelectionBox3D>>::ConstIterator KV = selected_3d_nodes.begin(); KV != selected_3d_nodes.end(); ++KV) {
  1625. ObjectID id = KV->key;
  1626. Node3D *node_3d = ObjectDB::get_instance<Node3D>(id);
  1627. if (!node_3d) {
  1628. selected_3d_nodes.erase(id);
  1629. --KV;
  1630. continue;
  1631. }
  1632. // Fallback.
  1633. AABB bounds(Vector3(-0.5, -0.5, -0.5), Vector3(1, 1, 1));
  1634. VisualInstance3D *visual_instance = Object::cast_to<VisualInstance3D>(node_3d);
  1635. if (visual_instance) {
  1636. bounds = visual_instance->get_aabb();
  1637. } else {
  1638. #ifndef PHYSICS_2D_DISABLED
  1639. CollisionShape3D *collision_shape = Object::cast_to<CollisionShape3D>(node_3d);
  1640. if (collision_shape) {
  1641. Ref<Shape3D> shape = collision_shape->get_shape();
  1642. if (shape.is_valid()) {
  1643. bounds = shape->get_debug_mesh()->get_aabb();
  1644. }
  1645. }
  1646. #endif // PHYSICS_2D_DISABLED
  1647. }
  1648. Transform3D xform_to_top_level_parent_space = node_3d->get_global_transform().affine_inverse() * node_3d->get_global_transform();
  1649. bounds = xform_to_top_level_parent_space.xform(bounds);
  1650. Transform3D t = node_3d->get_global_transform();
  1651. Ref<SelectionBox3D> sb = KV->value;
  1652. if (t == sb->transform && bounds == sb->bounds) {
  1653. continue; // Nothing changed.
  1654. }
  1655. sb->transform = t;
  1656. sb->bounds = bounds;
  1657. Transform3D t_offset = t;
  1658. // Apply AABB scaling before item's global transform.
  1659. {
  1660. const Vector3 offset(0.005, 0.005, 0.005);
  1661. Basis aabb_s;
  1662. aabb_s.scale(bounds.size + offset);
  1663. t.translate_local(bounds.position - offset / 2);
  1664. t.basis = t.basis * aabb_s;
  1665. }
  1666. {
  1667. const Vector3 offset(0.01, 0.01, 0.01);
  1668. Basis aabb_s;
  1669. aabb_s.scale(bounds.size + offset);
  1670. t_offset.translate_local(bounds.position - offset / 2);
  1671. t_offset.basis = t_offset.basis * aabb_s;
  1672. }
  1673. RS::get_singleton()->instance_set_visible(sb->instance, selection_visible);
  1674. RS::get_singleton()->instance_set_visible(sb->instance_ofs, selection_visible);
  1675. RS::get_singleton()->instance_set_visible(sb->instance_xray, selection_visible);
  1676. RS::get_singleton()->instance_set_visible(sb->instance_xray_ofs, selection_visible);
  1677. RS::get_singleton()->instance_set_transform(sb->instance, t);
  1678. RS::get_singleton()->instance_set_transform(sb->instance_ofs, t_offset);
  1679. RS::get_singleton()->instance_set_transform(sb->instance_xray, t);
  1680. RS::get_singleton()->instance_set_transform(sb->instance_xray_ofs, t_offset);
  1681. RS::get_singleton()->instance_reset_physics_interpolation(sb->instance);
  1682. RS::get_singleton()->instance_reset_physics_interpolation(sb->instance_ofs);
  1683. RS::get_singleton()->instance_reset_physics_interpolation(sb->instance_xray);
  1684. RS::get_singleton()->instance_reset_physics_interpolation(sb->instance_xray_ofs);
  1685. }
  1686. #endif // _3D_DISABLED
  1687. }
  1688. void RuntimeNodeSelect::_clear_selection() {
  1689. selected_ci_nodes.clear();
  1690. if (draw_canvas.is_valid()) {
  1691. RS::get_singleton()->canvas_item_clear(sbox_2d_ci);
  1692. }
  1693. #ifndef _3D_DISABLED
  1694. selected_3d_nodes.clear();
  1695. #endif // _3D_DISABLED
  1696. has_selection = false;
  1697. }
  1698. void RuntimeNodeSelect::_update_selection_drag(const Point2 &p_end_pos) {
  1699. RS::get_singleton()->canvas_item_clear(sel_drag_ci);
  1700. if (selection_drag_state != SELECTION_DRAG_MOVE) {
  1701. return;
  1702. }
  1703. selection_drag_area.size = p_end_pos - selection_drag_area.position;
  1704. if (selection_drag_state == SELECTION_DRAG_END) {
  1705. return;
  1706. }
  1707. Window *root = SceneTree::get_singleton()->get_root();
  1708. Rect2 selection_drawing;
  1709. int thickness;
  1710. if (root->is_canvas_transform_override_enabled()) {
  1711. Transform2D xform = root->get_canvas_transform_override();
  1712. RS::get_singleton()->canvas_item_set_transform(sel_drag_ci, xform);
  1713. RS::get_singleton()->canvas_item_reset_physics_interpolation(sel_drag_ci);
  1714. selection_drawing.position = xform.affine_inverse().xform(selection_drag_area.position);
  1715. selection_drawing.size = xform.affine_inverse().xform(p_end_pos);
  1716. thickness = MAX(1, Math::ceil(1 / view_2d_zoom));
  1717. } else {
  1718. RS::get_singleton()->canvas_item_set_transform(sel_drag_ci, Transform2D());
  1719. RS::get_singleton()->canvas_item_reset_physics_interpolation(sel_drag_ci);
  1720. selection_drawing.position = selection_drag_area.position;
  1721. selection_drawing.size = p_end_pos;
  1722. thickness = 1;
  1723. }
  1724. selection_drawing.size -= selection_drawing.position;
  1725. selection_drawing = selection_drawing.abs();
  1726. const Vector2 endpoints[4] = {
  1727. selection_drawing.position,
  1728. selection_drawing.position + Point2(selection_drawing.size.x, 0),
  1729. selection_drawing.position + selection_drawing.size,
  1730. selection_drawing.position + Point2(0, selection_drawing.size.y)
  1731. };
  1732. // Draw fill.
  1733. RS::get_singleton()->canvas_item_add_rect(sel_drag_ci, selection_drawing, selection_area_fill);
  1734. // Draw outline.
  1735. for (int i = 0; i < 4; i++) {
  1736. RS::get_singleton()->canvas_item_add_line(sel_drag_ci, endpoints[i], endpoints[(i + 1) % 4], selection_area_outline, thickness);
  1737. }
  1738. }
  1739. void RuntimeNodeSelect::_open_selection_list(const Vector<SelectResult> &p_items, const Point2 &p_pos) {
  1740. Window *root = SceneTree::get_singleton()->get_root();
  1741. selection_list = memnew(PopupMenu);
  1742. selection_list->set_theme(ThemeDB::get_singleton()->get_default_theme());
  1743. selection_list->set_auto_translate_mode(Node::AUTO_TRANSLATE_MODE_DISABLED);
  1744. selection_list->set_force_native(true);
  1745. selection_list->connect("index_pressed", callable_mp(this, &RuntimeNodeSelect::_items_popup_index_pressed).bind(selection_list));
  1746. selection_list->connect("popup_hide", callable_mp(this, &RuntimeNodeSelect::_close_selection_list));
  1747. root->add_child(selection_list);
  1748. for (const SelectResult &I : p_items) {
  1749. selection_list->add_item(I.item->get_name());
  1750. selection_list->set_item_metadata(-1, I.item);
  1751. }
  1752. selection_list->set_position(selection_list->is_embedded() ? p_pos : (Input::get_singleton()->get_mouse_position() + root->get_position()));
  1753. selection_list->reset_size();
  1754. selection_list->popup();
  1755. // FIXME: Ugly hack that stops the popup from hiding when the button is released.
  1756. selection_list->call_deferred(SNAME("set_position"), selection_list->get_position() + Point2(1, 0));
  1757. }
  1758. void RuntimeNodeSelect::_close_selection_list() {
  1759. selection_list->queue_free();
  1760. selection_list = nullptr;
  1761. }
  1762. void RuntimeNodeSelect::_set_selection_visible(bool p_visible) {
  1763. selection_visible = p_visible;
  1764. if (has_selection) {
  1765. _update_selection();
  1766. }
  1767. }
  1768. // Copied and trimmed from the CanvasItemEditor implementation.
  1769. void RuntimeNodeSelect::_find_canvas_items_at_pos(const Point2 &p_pos, Node *p_node, Vector<SelectResult> &r_items, const Transform2D &p_parent_xform, const Transform2D &p_canvas_xform) {
  1770. if (!p_node || Object::cast_to<Viewport>(p_node)) {
  1771. return;
  1772. }
  1773. CanvasItem *ci = Object::cast_to<CanvasItem>(p_node);
  1774. for (int i = p_node->get_child_count() - 1; i >= 0; i--) {
  1775. if (ci) {
  1776. if (!ci->is_set_as_top_level()) {
  1777. _find_canvas_items_at_pos(p_pos, p_node->get_child(i), r_items, p_parent_xform * ci->get_transform(), p_canvas_xform);
  1778. } else {
  1779. _find_canvas_items_at_pos(p_pos, p_node->get_child(i), r_items, ci->get_transform(), p_canvas_xform);
  1780. }
  1781. } else {
  1782. CanvasLayer *cl = Object::cast_to<CanvasLayer>(p_node);
  1783. _find_canvas_items_at_pos(p_pos, p_node->get_child(i), r_items, Transform2D(), cl ? cl->get_transform() : p_canvas_xform);
  1784. }
  1785. }
  1786. if (!ci || !ci->is_visible_in_tree()) {
  1787. return;
  1788. }
  1789. Transform2D xform = p_canvas_xform;
  1790. if (!ci->is_set_as_top_level()) {
  1791. xform *= p_parent_xform;
  1792. }
  1793. Window *root = SceneTree::get_singleton()->get_root();
  1794. Point2 pos;
  1795. // Cameras (overridden or not) don't affect `CanvasLayer`s.
  1796. if (!ci->get_canvas_layer_node() || ci->get_canvas_layer_node()->is_following_viewport()) {
  1797. pos = (root->is_canvas_transform_override_enabled() ? root->get_canvas_transform_override() : root->get_canvas_transform()).affine_inverse().xform(p_pos);
  1798. } else {
  1799. pos = p_pos;
  1800. }
  1801. xform = (xform * ci->get_transform()).affine_inverse();
  1802. const real_t local_grab_distance = xform.basis_xform(Vector2(sel_2d_grab_dist, 0)).length() / view_2d_zoom;
  1803. if (ci->_edit_is_selected_on_click(xform.xform(pos), local_grab_distance)) {
  1804. SelectResult res;
  1805. res.item = ci;
  1806. res.order = ci->get_effective_z_index() + ci->get_canvas_layer();
  1807. r_items.push_back(res);
  1808. #ifndef PHYSICS_2D_DISABLED
  1809. // If it's a shape, get the collision object it's from.
  1810. // FIXME: If the collision object has multiple shapes, only the topmost will be above it in the list.
  1811. if (Object::cast_to<CollisionShape2D>(ci) || Object::cast_to<CollisionPolygon2D>(ci)) {
  1812. CollisionObject2D *collision_object = Object::cast_to<CollisionObject2D>(ci->get_parent());
  1813. if (collision_object) {
  1814. SelectResult res_col;
  1815. res_col.item = ci->get_parent();
  1816. res_col.order = collision_object->get_z_index() + ci->get_canvas_layer();
  1817. r_items.push_back(res_col);
  1818. }
  1819. }
  1820. #endif // PHYSICS_2D_DISABLED
  1821. }
  1822. }
  1823. // Copied and trimmed from the CanvasItemEditor implementation.
  1824. void RuntimeNodeSelect::_find_canvas_items_at_rect(const Rect2 &p_rect, Node *p_node, Vector<SelectResult> &r_items, const Transform2D &p_parent_xform, const Transform2D &p_canvas_xform) {
  1825. if (!p_node || Object::cast_to<Viewport>(p_node)) {
  1826. return;
  1827. }
  1828. CanvasItem *ci = Object::cast_to<CanvasItem>(p_node);
  1829. for (int i = p_node->get_child_count() - 1; i >= 0; i--) {
  1830. if (ci) {
  1831. if (!ci->is_set_as_top_level()) {
  1832. _find_canvas_items_at_rect(p_rect, p_node->get_child(i), r_items, p_parent_xform * ci->get_transform(), p_canvas_xform);
  1833. } else {
  1834. _find_canvas_items_at_rect(p_rect, p_node->get_child(i), r_items, ci->get_transform(), p_canvas_xform);
  1835. }
  1836. } else {
  1837. CanvasLayer *cl = Object::cast_to<CanvasLayer>(p_node);
  1838. _find_canvas_items_at_rect(p_rect, p_node->get_child(i), r_items, Transform2D(), cl ? cl->get_transform() : p_canvas_xform);
  1839. }
  1840. }
  1841. if (!ci || !ci->is_visible_in_tree()) {
  1842. return;
  1843. }
  1844. Transform2D xform = p_canvas_xform;
  1845. if (!ci->is_set_as_top_level()) {
  1846. xform *= p_parent_xform;
  1847. }
  1848. Window *root = SceneTree::get_singleton()->get_root();
  1849. Rect2 rect;
  1850. // Cameras (overridden or not) don't affect `CanvasLayer`s.
  1851. if (!ci->get_canvas_layer_node() || ci->get_canvas_layer_node()->is_following_viewport()) {
  1852. rect = (root->is_canvas_transform_override_enabled() ? root->get_canvas_transform_override() : root->get_canvas_transform()).affine_inverse().xform(p_rect);
  1853. } else {
  1854. rect = p_rect;
  1855. }
  1856. rect = (xform * ci->get_transform()).affine_inverse().xform(rect);
  1857. bool selected = false;
  1858. if (ci->_edit_use_rect()) {
  1859. Rect2 ci_rect = ci->_edit_get_rect();
  1860. if (rect.has_point(ci_rect.position) &&
  1861. rect.has_point(ci_rect.position + Vector2(ci_rect.size.x, 0)) &&
  1862. rect.has_point(ci_rect.position + Vector2(ci_rect.size.x, ci_rect.size.y)) &&
  1863. rect.has_point(ci_rect.position + Vector2(0, ci_rect.size.y))) {
  1864. selected = true;
  1865. }
  1866. } else if (rect.has_point(Point2())) {
  1867. selected = true;
  1868. }
  1869. if (selected) {
  1870. SelectResult res;
  1871. res.item = ci;
  1872. res.order = ci->get_effective_z_index() + ci->get_canvas_layer();
  1873. r_items.push_back(res);
  1874. }
  1875. }
  1876. void RuntimeNodeSelect::_pan_callback(Vector2 p_scroll_vec, Ref<InputEvent> p_event) {
  1877. Vector2 scroll = SceneTree::get_singleton()->get_root()->get_screen_transform().affine_inverse().xform(p_scroll_vec);
  1878. view_2d_offset.x -= scroll.x / view_2d_zoom;
  1879. view_2d_offset.y -= scroll.y / view_2d_zoom;
  1880. _update_view_2d();
  1881. }
  1882. // A very shallow copy of the same function inside CanvasItemEditor.
  1883. void RuntimeNodeSelect::_zoom_callback(float p_zoom_factor, Vector2 p_origin, Ref<InputEvent> p_event) {
  1884. real_t prev_zoom = view_2d_zoom;
  1885. view_2d_zoom = CLAMP(view_2d_zoom * p_zoom_factor, VIEW_2D_MIN_ZOOM, VIEW_2D_MAX_ZOOM);
  1886. Vector2 pos = SceneTree::get_singleton()->get_root()->get_screen_transform().affine_inverse().xform(p_origin);
  1887. view_2d_offset += pos / prev_zoom - pos / view_2d_zoom;
  1888. // We want to align in-scene pixels to screen pixels, this prevents blurry rendering
  1889. // of small details (texts, lines).
  1890. // This correction adds a jitter movement when zooming, so we correct only when the
  1891. // zoom factor is an integer. (in the other cases, all pixels won't be aligned anyway)
  1892. const real_t closest_zoom_factor = Math::round(view_2d_zoom);
  1893. if (Math::is_zero_approx(view_2d_zoom - closest_zoom_factor)) {
  1894. // Make sure scene pixel at view_offset is aligned on a screen pixel.
  1895. Vector2 view_offset_int = view_2d_offset.floor();
  1896. Vector2 view_offset_frac = view_2d_offset - view_offset_int;
  1897. view_2d_offset = view_offset_int + (view_offset_frac * closest_zoom_factor).round() / closest_zoom_factor;
  1898. }
  1899. _update_view_2d();
  1900. }
  1901. void RuntimeNodeSelect::_reset_camera_2d() {
  1902. view_2d_offset = -SceneTree::get_singleton()->get_root()->get_canvas_transform().get_origin();
  1903. view_2d_zoom = 1;
  1904. _update_view_2d();
  1905. }
  1906. void RuntimeNodeSelect::_update_view_2d() {
  1907. Transform2D transform = Transform2D();
  1908. transform.scale_basis(Size2(view_2d_zoom, view_2d_zoom));
  1909. transform.columns[2] = -view_2d_offset * view_2d_zoom;
  1910. SceneTree::get_singleton()->get_root()->set_canvas_transform_override(transform);
  1911. _queue_selection_update();
  1912. }
  1913. #ifndef _3D_DISABLED
  1914. void RuntimeNodeSelect::_find_3d_items_at_pos(const Point2 &p_pos, Vector<SelectResult> &r_items) {
  1915. Window *root = SceneTree::get_singleton()->get_root();
  1916. Vector3 ray, pos, to;
  1917. if (root->is_camera_3d_override_enabled()) {
  1918. ray = root->camera_3d_override_project_ray_normal(p_pos);
  1919. pos = root->camera_3d_override_project_ray_origin(p_pos);
  1920. to = pos + ray * root->get_camera_3d_override_properties()["z_far"];
  1921. } else {
  1922. Camera3D *camera = root->get_camera_3d();
  1923. if (!camera) {
  1924. return;
  1925. }
  1926. ray = camera->project_ray_normal(p_pos);
  1927. pos = camera->project_ray_origin(p_pos);
  1928. to = pos + ray * camera->get_far();
  1929. }
  1930. #ifndef PHYSICS_3D_DISABLED
  1931. // Start with physical objects.
  1932. PhysicsDirectSpaceState3D *ss = root->get_world_3d()->get_direct_space_state();
  1933. PhysicsDirectSpaceState3D::RayResult result;
  1934. HashSet<RID> excluded;
  1935. PhysicsDirectSpaceState3D::RayParameters ray_params;
  1936. ray_params.from = pos;
  1937. ray_params.to = to;
  1938. ray_params.collide_with_areas = true;
  1939. while (true) {
  1940. ray_params.exclude = excluded;
  1941. if (ss->intersect_ray(ray_params, result)) {
  1942. SelectResult res;
  1943. res.item = Object::cast_to<Node>(result.collider);
  1944. res.order = -pos.distance_to(result.position);
  1945. // Fetch collision shapes.
  1946. CollisionObject3D *collision = Object::cast_to<CollisionObject3D>(result.collider);
  1947. if (collision) {
  1948. List<uint32_t> owners;
  1949. collision->get_shape_owners(&owners);
  1950. for (uint32_t &I : owners) {
  1951. SelectResult res_shape;
  1952. res_shape.item = Object::cast_to<Node>(collision->shape_owner_get_owner(I));
  1953. res_shape.order = res.order;
  1954. r_items.push_back(res_shape);
  1955. }
  1956. }
  1957. r_items.push_back(res);
  1958. excluded.insert(result.rid);
  1959. } else {
  1960. break;
  1961. }
  1962. }
  1963. #endif // PHYSICS_3D_DISABLED
  1964. // Then go for the meshes.
  1965. Vector<ObjectID> items = RS::get_singleton()->instances_cull_ray(pos, to, root->get_world_3d()->get_scenario());
  1966. for (int i = 0; i < items.size(); i++) {
  1967. Object *obj = ObjectDB::get_instance(items[i]);
  1968. GeometryInstance3D *geo_instance = Object::cast_to<GeometryInstance3D>(obj);
  1969. if (geo_instance) {
  1970. Ref<TriangleMesh> mesh_collision = geo_instance->generate_triangle_mesh();
  1971. if (mesh_collision.is_valid()) {
  1972. Transform3D gt = geo_instance->get_global_transform();
  1973. Transform3D ai = gt.affine_inverse();
  1974. Vector3 point, normal;
  1975. if (mesh_collision->intersect_ray(ai.xform(pos), ai.basis.xform(ray).normalized(), point, normal)) {
  1976. SelectResult res;
  1977. res.item = Object::cast_to<Node>(obj);
  1978. res.order = -pos.distance_to(gt.xform(point));
  1979. r_items.push_back(res);
  1980. continue;
  1981. }
  1982. }
  1983. }
  1984. items.remove_at(i);
  1985. i--;
  1986. }
  1987. }
  1988. void RuntimeNodeSelect::_find_3d_items_at_rect(const Rect2 &p_rect, Vector<SelectResult> &r_items) {
  1989. Window *root = SceneTree::get_singleton()->get_root();
  1990. Camera3D *camera = root->get_camera_3d();
  1991. if (!camera) {
  1992. return;
  1993. }
  1994. bool cam_override = root->is_camera_3d_override_enabled();
  1995. Vector3 cam_pos;
  1996. Vector3 dist_pos;
  1997. if (cam_override) {
  1998. cam_pos = root->get_camera_3d_override_transform().origin;
  1999. dist_pos = root->camera_3d_override_project_ray_origin(p_rect.position + p_rect.size / 2);
  2000. } else {
  2001. cam_pos = camera->get_global_position();
  2002. dist_pos = camera->project_ray_origin(p_rect.position + p_rect.size / 2);
  2003. }
  2004. real_t znear, zfar = 0;
  2005. real_t zofs = 5.0;
  2006. if (cam_override) {
  2007. HashMap<StringName, real_t> override_props = root->get_camera_3d_override_properties();
  2008. znear = override_props["z_near"];
  2009. zfar = override_props["z_far"];
  2010. zofs -= znear;
  2011. } else {
  2012. znear = camera->get_near();
  2013. zfar = camera->get_far();
  2014. zofs -= znear;
  2015. }
  2016. zofs = MAX(0.0, zofs);
  2017. const Point2 pos_end = p_rect.position + p_rect.size;
  2018. Vector3 box[4] = {
  2019. Vector3(
  2020. MIN(p_rect.position.x, pos_end.x),
  2021. MIN(p_rect.position.y, pos_end.y),
  2022. zofs),
  2023. Vector3(
  2024. MAX(p_rect.position.x, pos_end.x),
  2025. MIN(p_rect.position.y, pos_end.y),
  2026. zofs),
  2027. Vector3(
  2028. MAX(p_rect.position.x, pos_end.x),
  2029. MAX(p_rect.position.y, pos_end.y),
  2030. zofs),
  2031. Vector3(
  2032. MIN(p_rect.position.x, pos_end.x),
  2033. MAX(p_rect.position.y, pos_end.y),
  2034. zofs)
  2035. };
  2036. Vector<Plane> frustum;
  2037. for (int i = 0; i < 4; i++) {
  2038. Vector3 a = _get_screen_to_space(box[i]);
  2039. Vector3 b = _get_screen_to_space(box[(i + 1) % 4]);
  2040. frustum.push_back(Plane(a, b, cam_pos));
  2041. }
  2042. Plane near_plane;
  2043. // Get the camera normal.
  2044. if (cam_override) {
  2045. near_plane = Plane(root->get_camera_3d_override_transform().basis.get_column(2), cam_pos);
  2046. } else {
  2047. near_plane = Plane(camera->get_global_transform().basis.get_column(2), cam_pos);
  2048. }
  2049. near_plane.d -= znear;
  2050. frustum.push_back(near_plane);
  2051. Plane far_plane = -near_plane;
  2052. far_plane.d += zfar;
  2053. frustum.push_back(far_plane);
  2054. #ifndef PHYSICS_3D_DISABLED
  2055. Vector<Vector3> points = Geometry3D::compute_convex_mesh_points(&frustum[0], frustum.size());
  2056. Ref<ConvexPolygonShape3D> shape;
  2057. shape.instantiate();
  2058. shape->set_points(points);
  2059. // Start with physical objects.
  2060. PhysicsDirectSpaceState3D *ss = root->get_world_3d()->get_direct_space_state();
  2061. PhysicsDirectSpaceState3D::ShapeResult result;
  2062. PhysicsDirectSpaceState3D::ShapeParameters shape_params;
  2063. shape_params.shape_rid = shape->get_rid();
  2064. shape_params.collide_with_areas = true;
  2065. if (ss->intersect_shape(shape_params, &result, 32)) {
  2066. SelectResult res;
  2067. res.item = Object::cast_to<Node>(result.collider);
  2068. res.order = -dist_pos.distance_to(Object::cast_to<Node3D>(res.item)->get_global_transform().origin);
  2069. // Fetch collision shapes.
  2070. CollisionObject3D *collision = Object::cast_to<CollisionObject3D>(result.collider);
  2071. if (collision) {
  2072. List<uint32_t> owners;
  2073. collision->get_shape_owners(&owners);
  2074. for (uint32_t &I : owners) {
  2075. SelectResult res_shape;
  2076. res_shape.item = Object::cast_to<Node>(collision->shape_owner_get_owner(I));
  2077. res_shape.order = res.order;
  2078. r_items.push_back(res_shape);
  2079. }
  2080. }
  2081. r_items.push_back(res);
  2082. }
  2083. #endif // PHYSICS_3D_DISABLED
  2084. // Then go for the meshes.
  2085. Vector<ObjectID> items = RS::get_singleton()->instances_cull_convex(frustum, root->get_world_3d()->get_scenario());
  2086. for (int i = 0; i < items.size(); i++) {
  2087. Object *obj = ObjectDB::get_instance(items[i]);
  2088. GeometryInstance3D *geo_instance = Object::cast_to<GeometryInstance3D>(obj);
  2089. if (geo_instance) {
  2090. Ref<TriangleMesh> mesh_collision = geo_instance->generate_triangle_mesh();
  2091. if (mesh_collision.is_valid()) {
  2092. Transform3D gt = geo_instance->get_global_transform();
  2093. Vector3 mesh_scale = gt.get_basis().get_scale();
  2094. gt.orthonormalize();
  2095. Transform3D it = gt.affine_inverse();
  2096. Vector<Plane> transformed_frustum;
  2097. int plane_count = frustum.size();
  2098. transformed_frustum.resize(plane_count);
  2099. for (int j = 0; j < plane_count; j++) {
  2100. transformed_frustum.write[j] = it.xform(frustum[j]);
  2101. }
  2102. Vector<Vector3> convex_points = Geometry3D::compute_convex_mesh_points(transformed_frustum.ptr(), plane_count);
  2103. if (mesh_collision->inside_convex_shape(transformed_frustum.ptr(), transformed_frustum.size(), convex_points.ptr(), convex_points.size(), mesh_scale)) {
  2104. SelectResult res;
  2105. res.item = Object::cast_to<Node>(obj);
  2106. res.order = -dist_pos.distance_to(gt.origin);
  2107. r_items.push_back(res);
  2108. continue;
  2109. }
  2110. }
  2111. }
  2112. items.remove_at(i);
  2113. i--;
  2114. }
  2115. }
  2116. Vector3 RuntimeNodeSelect::_get_screen_to_space(const Vector3 &p_vector3) {
  2117. Window *root = SceneTree::get_singleton()->get_root();
  2118. Camera3D *camera = root->get_camera_3d();
  2119. Size2 size = root->get_size();
  2120. real_t znear = 0;
  2121. Projection cm;
  2122. Transform3D camera_transform;
  2123. if (root->is_camera_3d_override_enabled()) {
  2124. HashMap<StringName, real_t> override_props = root->get_camera_3d_override_properties();
  2125. znear = override_props["z_near"];
  2126. cm.set_perspective(override_props["fov"], size.aspect(), znear + p_vector3.z, override_props["z_far"]);
  2127. camera_transform.translate_local(cursor.pos);
  2128. camera_transform.basis.rotate(Vector3(1, 0, 0), -cursor.x_rot);
  2129. camera_transform.basis.rotate(Vector3(0, 1, 0), -cursor.y_rot);
  2130. camera_transform.translate_local(0, 0, cursor.distance);
  2131. } else {
  2132. znear = camera->get_near();
  2133. cm.set_perspective(camera->get_fov(), size.aspect(), znear + p_vector3.z, camera->get_far());
  2134. camera_transform = camera->get_camera_transform();
  2135. }
  2136. Vector2 screen_he = cm.get_viewport_half_extents();
  2137. return camera_transform.xform(Vector3(((p_vector3.x / size.width) * 2.0 - 1.0) * screen_he.x, ((1.0 - (p_vector3.y / size.height)) * 2.0 - 1.0) * screen_he.y, -(znear + p_vector3.z)));
  2138. }
  2139. bool RuntimeNodeSelect::_handle_3d_input(const Ref<InputEvent> &p_event) {
  2140. Ref<InputEventMouseButton> b = p_event;
  2141. if (b.is_valid()) {
  2142. const real_t zoom_factor = 1.08 * b->get_factor();
  2143. switch (b->get_button_index()) {
  2144. case MouseButton::WHEEL_UP: {
  2145. if (!camera_freelook) {
  2146. _cursor_scale_distance(1.0 / zoom_factor);
  2147. } else {
  2148. _scale_freelook_speed(zoom_factor);
  2149. }
  2150. return true;
  2151. } break;
  2152. case MouseButton::WHEEL_DOWN: {
  2153. if (!camera_freelook) {
  2154. _cursor_scale_distance(zoom_factor);
  2155. } else {
  2156. _scale_freelook_speed(1.0 / zoom_factor);
  2157. }
  2158. return true;
  2159. } break;
  2160. case MouseButton::RIGHT: {
  2161. _set_camera_freelook_enabled(b->is_pressed());
  2162. return true;
  2163. } break;
  2164. default: {
  2165. }
  2166. }
  2167. }
  2168. Ref<InputEventMouseMotion> m = p_event;
  2169. if (m.is_valid()) {
  2170. if (camera_freelook) {
  2171. _cursor_look(m);
  2172. } else if (m->get_button_mask().has_flag(MouseButtonMask::MIDDLE)) {
  2173. if (m->is_shift_pressed()) {
  2174. _cursor_pan(m);
  2175. } else {
  2176. _cursor_orbit(m);
  2177. }
  2178. }
  2179. return true;
  2180. }
  2181. Ref<InputEventKey> k = p_event;
  2182. if (k.is_valid()) {
  2183. if (k->get_physical_keycode() == Key::ESCAPE) {
  2184. _set_camera_freelook_enabled(false);
  2185. return true;
  2186. } else if (k->is_ctrl_pressed()) {
  2187. switch (k->get_physical_keycode()) {
  2188. case Key::EQUAL: {
  2189. cursor.fov_scale = CLAMP(cursor.fov_scale - 0.05, CAMERA_MIN_FOV_SCALE, CAMERA_MAX_FOV_SCALE);
  2190. SceneTree::get_singleton()->get_root()->set_camera_3d_override_perspective(camera_fov * cursor.fov_scale, camera_znear, camera_zfar);
  2191. return true;
  2192. } break;
  2193. case Key::MINUS: {
  2194. cursor.fov_scale = CLAMP(cursor.fov_scale + 0.05, CAMERA_MIN_FOV_SCALE, CAMERA_MAX_FOV_SCALE);
  2195. SceneTree::get_singleton()->get_root()->set_camera_3d_override_perspective(camera_fov * cursor.fov_scale, camera_znear, camera_zfar);
  2196. return true;
  2197. } break;
  2198. case Key::KEY_0: {
  2199. cursor.fov_scale = 1;
  2200. SceneTree::get_singleton()->get_root()->set_camera_3d_override_perspective(camera_fov, camera_znear, camera_zfar);
  2201. return true;
  2202. } break;
  2203. default: {
  2204. }
  2205. }
  2206. }
  2207. }
  2208. // TODO: Handle magnify and pan input gestures.
  2209. return false;
  2210. }
  2211. void RuntimeNodeSelect::_set_camera_freelook_enabled(bool p_enabled) {
  2212. camera_freelook = p_enabled;
  2213. if (p_enabled) {
  2214. // Make sure eye_pos is synced, because freelook referential is eye pos rather than orbit pos
  2215. Vector3 forward = _get_cursor_transform().basis.xform(Vector3(0, 0, -1));
  2216. cursor.eye_pos = cursor.pos - cursor.distance * forward;
  2217. previous_mouse_position = SceneTree::get_singleton()->get_root()->get_mouse_position();
  2218. // Hide mouse like in an FPS (warping doesn't work).
  2219. Input::get_singleton()->set_mouse_mode_override(Input::MOUSE_MODE_CAPTURED);
  2220. } else {
  2221. // Restore mouse.
  2222. Input::get_singleton()->set_mouse_mode_override(Input::MOUSE_MODE_VISIBLE);
  2223. // Restore the previous mouse position when leaving freelook mode.
  2224. // This is done because leaving `Input.MOUSE_MODE_CAPTURED` will center the cursor
  2225. // due to OS limitations.
  2226. Input::get_singleton()->warp_mouse(previous_mouse_position);
  2227. }
  2228. }
  2229. void RuntimeNodeSelect::_cursor_scale_distance(real_t p_scale) {
  2230. real_t min_distance = MAX(camera_znear * 4, VIEW_3D_MIN_ZOOM);
  2231. real_t max_distance = MIN(camera_zfar / 4, VIEW_3D_MAX_ZOOM);
  2232. cursor.distance = CLAMP(cursor.distance * p_scale, min_distance, max_distance);
  2233. SceneTree::get_singleton()->get_root()->set_camera_3d_override_transform(_get_cursor_transform());
  2234. }
  2235. void RuntimeNodeSelect::_scale_freelook_speed(real_t p_scale) {
  2236. real_t min_speed = MAX(camera_znear * 4, VIEW_3D_MIN_ZOOM);
  2237. real_t max_speed = MIN(camera_zfar / 4, VIEW_3D_MAX_ZOOM);
  2238. if (unlikely(min_speed > max_speed)) {
  2239. freelook_base_speed = (min_speed + max_speed) / 2;
  2240. } else {
  2241. freelook_base_speed = CLAMP(freelook_base_speed * p_scale, min_speed, max_speed);
  2242. }
  2243. }
  2244. void RuntimeNodeSelect::_cursor_look(Ref<InputEventWithModifiers> p_event) {
  2245. Window *root = SceneTree::get_singleton()->get_root();
  2246. const Vector2 relative = _get_warped_mouse_motion(p_event, Rect2(Vector2(), root->get_size()));
  2247. const Transform3D prev_camera_transform = _get_cursor_transform();
  2248. if (invert_y_axis) {
  2249. cursor.x_rot -= relative.y * freelook_sensitivity;
  2250. } else {
  2251. cursor.x_rot += relative.y * freelook_sensitivity;
  2252. }
  2253. // Clamp the Y rotation to roughly -90..90 degrees so the user can't look upside-down and end up disoriented.
  2254. cursor.x_rot = CLAMP(cursor.x_rot, -1.57, 1.57);
  2255. cursor.y_rot += relative.x * freelook_sensitivity;
  2256. // Look is like the opposite of Orbit: the focus point rotates around the camera.
  2257. Transform3D camera_transform = _get_cursor_transform();
  2258. Vector3 pos = camera_transform.xform(Vector3(0, 0, 0));
  2259. Vector3 prev_pos = prev_camera_transform.xform(Vector3(0, 0, 0));
  2260. Vector3 diff = prev_pos - pos;
  2261. cursor.pos += diff;
  2262. SceneTree::get_singleton()->get_root()->set_camera_3d_override_transform(_get_cursor_transform());
  2263. }
  2264. void RuntimeNodeSelect::_cursor_pan(Ref<InputEventWithModifiers> p_event) {
  2265. Window *root = SceneTree::get_singleton()->get_root();
  2266. // Reduce all sides of the area by 1, so warping works when windows are maximized/fullscreen.
  2267. const Vector2 relative = _get_warped_mouse_motion(p_event, Rect2(Vector2(1, 1), root->get_size() - Vector2(2, 2)));
  2268. const real_t pan_speed = translation_sensitivity / 150.0;
  2269. Transform3D camera_transform;
  2270. camera_transform.translate_local(cursor.pos);
  2271. camera_transform.basis.rotate(Vector3(1, 0, 0), -cursor.x_rot);
  2272. camera_transform.basis.rotate(Vector3(0, 1, 0), -cursor.y_rot);
  2273. Vector3 translation(1 * -relative.x * pan_speed, relative.y * pan_speed, 0);
  2274. translation *= cursor.distance / 4;
  2275. camera_transform.translate_local(translation);
  2276. cursor.pos = camera_transform.origin;
  2277. SceneTree::get_singleton()->get_root()->set_camera_3d_override_transform(_get_cursor_transform());
  2278. }
  2279. void RuntimeNodeSelect::_cursor_orbit(Ref<InputEventWithModifiers> p_event) {
  2280. Window *root = SceneTree::get_singleton()->get_root();
  2281. // Reduce all sides of the area by 1, so warping works when windows are maximized/fullscreen.
  2282. const Vector2 relative = _get_warped_mouse_motion(p_event, Rect2(Vector2(1, 1), root->get_size() - Vector2(2, 2)));
  2283. if (invert_y_axis) {
  2284. cursor.x_rot -= relative.y * orbit_sensitivity;
  2285. } else {
  2286. cursor.x_rot += relative.y * orbit_sensitivity;
  2287. }
  2288. // Clamp the Y rotation to roughly -90..90 degrees so the user can't look upside-down and end up disoriented.
  2289. cursor.x_rot = CLAMP(cursor.x_rot, -1.57, 1.57);
  2290. if (invert_x_axis) {
  2291. cursor.y_rot -= relative.x * orbit_sensitivity;
  2292. } else {
  2293. cursor.y_rot += relative.x * orbit_sensitivity;
  2294. }
  2295. SceneTree::get_singleton()->get_root()->set_camera_3d_override_transform(_get_cursor_transform());
  2296. }
  2297. Point2 RuntimeNodeSelect::_get_warped_mouse_motion(const Ref<InputEventMouseMotion> &p_event, Rect2 p_area) const {
  2298. ERR_FAIL_COND_V(p_event.is_null(), Point2());
  2299. if (warped_mouse_panning_3d) {
  2300. return Input::get_singleton()->warp_mouse_motion(p_event, p_area);
  2301. }
  2302. return p_event->get_relative();
  2303. }
  2304. Transform3D RuntimeNodeSelect::_get_cursor_transform() {
  2305. Transform3D camera_transform;
  2306. camera_transform.translate_local(cursor.pos);
  2307. camera_transform.basis.rotate(Vector3(1, 0, 0), -cursor.x_rot);
  2308. camera_transform.basis.rotate(Vector3(0, 1, 0), -cursor.y_rot);
  2309. camera_transform.translate_local(0, 0, cursor.distance);
  2310. return camera_transform;
  2311. }
  2312. void RuntimeNodeSelect::_reset_camera_3d() {
  2313. camera_first_override = true;
  2314. cursor = Cursor();
  2315. Window *root = SceneTree::get_singleton()->get_root();
  2316. Camera3D *camera = root->get_camera_3d();
  2317. if (camera) {
  2318. Transform3D transform = camera->get_global_transform();
  2319. transform.translate_local(0, 0, -cursor.distance);
  2320. cursor.pos = transform.origin;
  2321. cursor.x_rot = -camera->get_global_rotation().x;
  2322. cursor.y_rot = -camera->get_global_rotation().y;
  2323. cursor.fov_scale = CLAMP(camera->get_fov() / camera_fov, CAMERA_MIN_FOV_SCALE, CAMERA_MAX_FOV_SCALE);
  2324. } else {
  2325. cursor.fov_scale = 1.0;
  2326. }
  2327. SceneTree::get_singleton()->get_root()->set_camera_3d_override_transform(_get_cursor_transform());
  2328. SceneTree::get_singleton()->get_root()->set_camera_3d_override_perspective(camera_fov * cursor.fov_scale, camera_znear, camera_zfar);
  2329. }
  2330. #endif // _3D_DISABLED
  2331. #endif // DEBUG_ENABLED