scene_tree_dock.cpp 84 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537
  1. /*************************************************************************/
  2. /* scene_tree_dock.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2019 Godot Engine contributors (cf. AUTHORS.md) */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /*************************************************************************/
  30. #include "scene_tree_dock.h"
  31. #include "core/io/resource_saver.h"
  32. #include "core/os/keyboard.h"
  33. #include "core/project_settings.h"
  34. #include "editor/editor_node.h"
  35. #include "editor/editor_settings.h"
  36. #include "editor/multi_node_edit.h"
  37. #include "editor/plugins/animation_player_editor_plugin.h"
  38. #include "editor/plugins/canvas_item_editor_plugin.h"
  39. #include "editor/plugins/script_editor_plugin.h"
  40. #include "editor/plugins/spatial_editor_plugin.h"
  41. #include "editor/script_editor_debugger.h"
  42. #include "scene/main/viewport.h"
  43. #include "scene/resources/packed_scene.h"
  44. void SceneTreeDock::_nodes_drag_begin() {
  45. if (restore_script_editor_on_drag) {
  46. EditorNode::get_singleton()->set_visible_editor(EditorNode::EDITOR_SCRIPT);
  47. restore_script_editor_on_drag = false;
  48. }
  49. }
  50. void SceneTreeDock::_quick_open() {
  51. Vector<String> files = quick_open->get_selected_files();
  52. for (int i = 0; i < files.size(); i++) {
  53. instance(files[i]);
  54. }
  55. }
  56. void SceneTreeDock::_input(Ref<InputEvent> p_event) {
  57. Ref<InputEventMouseButton> mb = p_event;
  58. if (mb.is_valid() && !mb->is_pressed() && mb->get_button_index() == BUTTON_LEFT) {
  59. restore_script_editor_on_drag = false; //lost chance
  60. }
  61. }
  62. void SceneTreeDock::_unhandled_key_input(Ref<InputEvent> p_event) {
  63. if (get_viewport()->get_modal_stack_top())
  64. return; //ignore because of modal window
  65. if (get_focus_owner() && get_focus_owner()->is_text_field())
  66. return;
  67. if (!p_event->is_pressed() || p_event->is_echo())
  68. return;
  69. if (ED_IS_SHORTCUT("scene_tree/batch_rename", p_event)) {
  70. _tool_selected(TOOL_BATCH_RENAME);
  71. } else if (ED_IS_SHORTCUT("scene_tree/rename", p_event)) {
  72. _tool_selected(TOOL_RENAME);
  73. } else if (ED_IS_SHORTCUT("scene_tree/add_child_node", p_event)) {
  74. _tool_selected(TOOL_NEW);
  75. } else if (ED_IS_SHORTCUT("scene_tree/instance_scene", p_event)) {
  76. _tool_selected(TOOL_INSTANCE);
  77. } else if (ED_IS_SHORTCUT("scene_tree/change_node_type", p_event)) {
  78. _tool_selected(TOOL_REPLACE);
  79. } else if (ED_IS_SHORTCUT("scene_tree/duplicate", p_event)) {
  80. _tool_selected(TOOL_DUPLICATE);
  81. } else if (ED_IS_SHORTCUT("scene_tree/attach_script", p_event)) {
  82. _tool_selected(TOOL_ATTACH_SCRIPT);
  83. } else if (ED_IS_SHORTCUT("scene_tree/clear_script", p_event)) {
  84. _tool_selected(TOOL_CLEAR_SCRIPT);
  85. } else if (ED_IS_SHORTCUT("scene_tree/move_up", p_event)) {
  86. _tool_selected(TOOL_MOVE_UP);
  87. } else if (ED_IS_SHORTCUT("scene_tree/move_down", p_event)) {
  88. _tool_selected(TOOL_MOVE_DOWN);
  89. } else if (ED_IS_SHORTCUT("scene_tree/reparent", p_event)) {
  90. _tool_selected(TOOL_REPARENT);
  91. } else if (ED_IS_SHORTCUT("scene_tree/merge_from_scene", p_event)) {
  92. _tool_selected(TOOL_MERGE_FROM_SCENE);
  93. } else if (ED_IS_SHORTCUT("scene_tree/save_branch_as_scene", p_event)) {
  94. _tool_selected(TOOL_NEW_SCENE_FROM);
  95. } else if (ED_IS_SHORTCUT("scene_tree/delete_no_confirm", p_event)) {
  96. _tool_selected(TOOL_ERASE, true);
  97. } else if (ED_IS_SHORTCUT("scene_tree/copy_node_path", p_event)) {
  98. _tool_selected(TOOL_COPY_NODE_PATH);
  99. } else if (ED_IS_SHORTCUT("scene_tree/delete", p_event)) {
  100. _tool_selected(TOOL_ERASE);
  101. }
  102. }
  103. void SceneTreeDock::instance(const String &p_file) {
  104. Node *parent = scene_tree->get_selected();
  105. if (!parent) {
  106. parent = edited_scene;
  107. };
  108. if (!edited_scene) {
  109. current_option = -1;
  110. accept->set_text(TTR("No parent to instance a child at."));
  111. accept->popup_centered_minsize();
  112. return;
  113. };
  114. ERR_FAIL_COND(!parent);
  115. Vector<String> scenes;
  116. scenes.push_back(p_file);
  117. _perform_instance_scenes(scenes, parent, -1);
  118. }
  119. void SceneTreeDock::instance_scenes(const Vector<String> &p_files, Node *p_parent) {
  120. Node *parent = p_parent;
  121. if (!parent) {
  122. parent = scene_tree->get_selected();
  123. }
  124. if (!parent || !edited_scene) {
  125. accept->set_text(TTR("No parent to instance the scenes at."));
  126. accept->popup_centered_minsize();
  127. return;
  128. };
  129. _perform_instance_scenes(p_files, parent, -1);
  130. }
  131. void SceneTreeDock::_perform_instance_scenes(const Vector<String> &p_files, Node *parent, int p_pos) {
  132. ERR_FAIL_COND(!parent);
  133. Vector<Node *> instances;
  134. bool error = false;
  135. for (int i = 0; i < p_files.size(); i++) {
  136. Ref<PackedScene> sdata = ResourceLoader::load(p_files[i]);
  137. if (!sdata.is_valid()) {
  138. current_option = -1;
  139. accept->set_text(vformat(TTR("Error loading scene from %s"), p_files[i]));
  140. accept->popup_centered_minsize();
  141. error = true;
  142. break;
  143. }
  144. Node *instanced_scene = sdata->instance(PackedScene::GEN_EDIT_STATE_INSTANCE);
  145. if (!instanced_scene) {
  146. current_option = -1;
  147. accept->set_text(vformat(TTR("Error instancing scene from %s"), p_files[i]));
  148. accept->popup_centered_minsize();
  149. error = true;
  150. break;
  151. }
  152. if (edited_scene->get_filename() != "") {
  153. if (_cyclical_dependency_exists(edited_scene->get_filename(), instanced_scene)) {
  154. accept->set_text(vformat(TTR("Cannot instance the scene '%s' because the current scene exists within one of its nodes."), p_files[i]));
  155. accept->popup_centered_minsize();
  156. error = true;
  157. break;
  158. }
  159. }
  160. instanced_scene->set_filename(ProjectSettings::get_singleton()->localize_path(p_files[i]));
  161. instances.push_back(instanced_scene);
  162. }
  163. if (error) {
  164. for (int i = 0; i < instances.size(); i++) {
  165. memdelete(instances[i]);
  166. }
  167. return;
  168. }
  169. editor_data->get_undo_redo().create_action(TTR("Instance Scene(s)"));
  170. for (int i = 0; i < instances.size(); i++) {
  171. Node *instanced_scene = instances[i];
  172. editor_data->get_undo_redo().add_do_method(parent, "add_child", instanced_scene);
  173. if (p_pos >= 0) {
  174. editor_data->get_undo_redo().add_do_method(parent, "move_child", instanced_scene, p_pos + i);
  175. }
  176. editor_data->get_undo_redo().add_do_method(instanced_scene, "set_owner", edited_scene);
  177. editor_data->get_undo_redo().add_do_method(editor_selection, "clear");
  178. editor_data->get_undo_redo().add_do_method(editor_selection, "add_node", instanced_scene);
  179. editor_data->get_undo_redo().add_do_reference(instanced_scene);
  180. editor_data->get_undo_redo().add_undo_method(parent, "remove_child", instanced_scene);
  181. String new_name = parent->validate_child_name(instanced_scene);
  182. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  183. editor_data->get_undo_redo().add_do_method(sed, "live_debug_instance_node", edited_scene->get_path_to(parent), p_files[i], new_name);
  184. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_remove_node", NodePath(String(edited_scene->get_path_to(parent)).plus_file(new_name)));
  185. }
  186. editor_data->get_undo_redo().commit_action();
  187. }
  188. void SceneTreeDock::_replace_with_branch_scene(const String &p_file, Node *base) {
  189. Ref<PackedScene> sdata = ResourceLoader::load(p_file);
  190. if (!sdata.is_valid()) {
  191. accept->set_text(vformat(TTR("Error loading scene from %s"), p_file));
  192. accept->popup_centered_minsize();
  193. return;
  194. }
  195. Node *instanced_scene = sdata->instance(PackedScene::GEN_EDIT_STATE_INSTANCE);
  196. if (!instanced_scene) {
  197. accept->set_text(vformat(TTR("Error instancing scene from %s"), p_file));
  198. accept->popup_centered_minsize();
  199. return;
  200. }
  201. Node *parent = base->get_parent();
  202. int pos = base->get_index();
  203. parent->remove_child(base);
  204. parent->add_child(instanced_scene);
  205. parent->move_child(instanced_scene, pos);
  206. instanced_scene->set_owner(edited_scene);
  207. editor_selection->clear();
  208. editor_selection->add_node(instanced_scene);
  209. scene_tree->set_selected(instanced_scene);
  210. // Delete the node as late as possible because before another one is selected
  211. // an editor plugin could be referencing it to do something with it before
  212. // switching to another (or to none); and since some steps of changing the
  213. // editor state are deferred, the safest thing is to do this is as the last
  214. // step of this function and also by enqueing instead of memdelete()-ing it here
  215. base->queue_delete();
  216. }
  217. bool SceneTreeDock::_cyclical_dependency_exists(const String &p_target_scene_path, Node *p_desired_node) {
  218. int childCount = p_desired_node->get_child_count();
  219. if (p_desired_node->get_filename() == p_target_scene_path) {
  220. return true;
  221. }
  222. for (int i = 0; i < childCount; i++) {
  223. Node *child = p_desired_node->get_child(i);
  224. if (_cyclical_dependency_exists(p_target_scene_path, child)) {
  225. return true;
  226. }
  227. }
  228. return false;
  229. }
  230. void SceneTreeDock::_tool_selected(int p_tool, bool p_confirm_override) {
  231. current_option = p_tool;
  232. switch (p_tool) {
  233. case TOOL_BATCH_RENAME: {
  234. Tree *tree = scene_tree->get_scene_tree();
  235. if (tree->is_anything_selected()) {
  236. rename_dialog->popup_centered();
  237. }
  238. } break;
  239. case TOOL_RENAME: {
  240. Tree *tree = scene_tree->get_scene_tree();
  241. if (tree->is_anything_selected()) {
  242. tree->grab_focus();
  243. tree->edit_selected();
  244. }
  245. } break;
  246. case TOOL_NEW: {
  247. String preferred = "";
  248. Node *current_edited_scene_root = EditorNode::get_singleton()->get_edited_scene();
  249. if (current_edited_scene_root) {
  250. if (ClassDB::is_parent_class(current_edited_scene_root->get_class_name(), "Node2D"))
  251. preferred = "Node2D";
  252. else if (ClassDB::is_parent_class(current_edited_scene_root->get_class_name(), "Spatial"))
  253. preferred = "Spatial";
  254. }
  255. create_dialog->set_preferred_search_result_type(preferred);
  256. create_dialog->popup_create(true);
  257. } break;
  258. case TOOL_INSTANCE: {
  259. Node *scene = edited_scene;
  260. if (!scene) {
  261. EditorNode::get_singleton()->new_inherited_scene();
  262. break;
  263. }
  264. quick_open->popup_dialog("PackedScene", true);
  265. quick_open->set_title(TTR("Instance Child Scene"));
  266. } break;
  267. case TOOL_REPLACE: {
  268. create_dialog->popup_create(false, true);
  269. } break;
  270. case TOOL_ATTACH_SCRIPT: {
  271. Node *selected = scene_tree->get_selected();
  272. if (!selected)
  273. break;
  274. Ref<Script> existing = selected->get_script();
  275. String path = selected->get_filename();
  276. if (path == "") {
  277. String root_path = editor_data->get_edited_scene_root()->get_filename();
  278. if (root_path == "") {
  279. path = String("res://").plus_file(selected->get_name());
  280. } else {
  281. path = root_path.get_base_dir().plus_file(selected->get_name());
  282. }
  283. }
  284. String inherits = selected->get_class();
  285. if (existing.is_valid()) {
  286. for (int i = 0; i < ScriptServer::get_language_count(); i++) {
  287. ScriptLanguage *l = ScriptServer::get_language(i);
  288. if (l->get_type() == existing->get_class()) {
  289. String name = l->get_global_class_name(existing->get_path());
  290. if (ScriptServer::is_global_class(name) && EDITOR_GET("interface/editors/derive_script_globals_by_name").operator bool()) {
  291. inherits = name;
  292. } else if (l->can_inherit_from_file()) {
  293. inherits = "\"" + existing->get_path() + "\"";
  294. }
  295. break;
  296. }
  297. }
  298. }
  299. script_create_dialog->config(inherits, path);
  300. script_create_dialog->popup_centered();
  301. } break;
  302. case TOOL_CLEAR_SCRIPT: {
  303. List<Node *> selection = editor_selection->get_selected_node_list();
  304. if (selection.empty())
  305. return;
  306. editor_data->get_undo_redo().create_action(TTR("Clear Script"));
  307. editor_data->get_undo_redo().add_do_method(editor, "push_item", (Script *)NULL);
  308. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  309. Ref<Script> existing = E->get()->get_script();
  310. if (existing.is_valid()) {
  311. const RefPtr empty;
  312. editor_data->get_undo_redo().add_do_method(E->get(), "set_script", empty);
  313. editor_data->get_undo_redo().add_undo_method(E->get(), "set_script", existing);
  314. }
  315. }
  316. editor_data->get_undo_redo().add_do_method(this, "_update_script_button");
  317. editor_data->get_undo_redo().add_undo_method(this, "_update_script_button");
  318. editor_data->get_undo_redo().commit_action();
  319. } break;
  320. case TOOL_MOVE_UP:
  321. case TOOL_MOVE_DOWN: {
  322. if (!scene_tree->get_selected())
  323. break;
  324. if (scene_tree->get_selected() == edited_scene) {
  325. current_option = -1;
  326. accept->set_text(TTR("This operation can't be done on the tree root."));
  327. accept->popup_centered_minsize();
  328. break;
  329. }
  330. if (!_validate_no_foreign())
  331. break;
  332. bool MOVING_DOWN = (p_tool == TOOL_MOVE_DOWN);
  333. bool MOVING_UP = !MOVING_DOWN;
  334. Node *common_parent = scene_tree->get_selected()->get_parent();
  335. List<Node *> selection = editor_selection->get_selected_node_list();
  336. selection.sort_custom<Node::Comparator>(); // sort by index
  337. if (MOVING_DOWN)
  338. selection.invert();
  339. int lowest_id = common_parent->get_child_count() - 1;
  340. int highest_id = 0;
  341. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  342. int index = E->get()->get_index();
  343. if (index > highest_id) highest_id = index;
  344. if (index < lowest_id) lowest_id = index;
  345. if (E->get()->get_parent() != common_parent)
  346. common_parent = NULL;
  347. }
  348. if (!common_parent || (MOVING_DOWN && highest_id >= common_parent->get_child_count() - MOVING_DOWN) || (MOVING_UP && lowest_id == 0))
  349. break; // one or more nodes can not be moved
  350. if (selection.size() == 1) editor_data->get_undo_redo().create_action(TTR("Move Node In Parent"));
  351. if (selection.size() > 1) editor_data->get_undo_redo().create_action(TTR("Move Nodes In Parent"));
  352. for (int i = 0; i < selection.size(); i++) {
  353. Node *top_node = selection[i];
  354. Node *bottom_node = selection[selection.size() - 1 - i];
  355. ERR_FAIL_COND(!top_node->get_parent());
  356. ERR_FAIL_COND(!bottom_node->get_parent());
  357. int bottom_node_pos = bottom_node->get_index();
  358. int top_node_pos_next = top_node->get_index() + (MOVING_DOWN ? 1 : -1);
  359. editor_data->get_undo_redo().add_do_method(top_node->get_parent(), "move_child", top_node, top_node_pos_next);
  360. editor_data->get_undo_redo().add_undo_method(bottom_node->get_parent(), "move_child", bottom_node, bottom_node_pos);
  361. }
  362. editor_data->get_undo_redo().commit_action();
  363. } break;
  364. case TOOL_DUPLICATE: {
  365. if (!edited_scene)
  366. break;
  367. if (editor_selection->is_selected(edited_scene)) {
  368. current_option = -1;
  369. accept->set_text(TTR("This operation can't be done on the tree root."));
  370. accept->popup_centered_minsize();
  371. break;
  372. }
  373. if (!_validate_no_foreign())
  374. break;
  375. List<Node *> selection = editor_selection->get_selected_node_list();
  376. if (selection.size() == 0)
  377. break;
  378. editor_data->get_undo_redo().create_action(TTR("Duplicate Node(s)"));
  379. editor_data->get_undo_redo().add_do_method(editor_selection, "clear");
  380. Node *dupsingle = NULL;
  381. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  382. Node *node = E->get();
  383. Node *parent = node->get_parent();
  384. List<Node *> owned;
  385. node->get_owned_by(node->get_owner(), &owned);
  386. Map<const Node *, Node *> duplimap;
  387. Node *dup = node->duplicate_from_editor(duplimap);
  388. ERR_CONTINUE(!dup);
  389. if (selection.size() == 1)
  390. dupsingle = dup;
  391. dup->set_name(parent->validate_child_name(dup));
  392. editor_data->get_undo_redo().add_do_method(parent, "add_child_below_node", node, dup);
  393. for (List<Node *>::Element *F = owned.front(); F; F = F->next()) {
  394. if (!duplimap.has(F->get())) {
  395. continue;
  396. }
  397. Node *d = duplimap[F->get()];
  398. editor_data->get_undo_redo().add_do_method(d, "set_owner", node->get_owner());
  399. }
  400. editor_data->get_undo_redo().add_do_method(editor_selection, "add_node", dup);
  401. editor_data->get_undo_redo().add_undo_method(parent, "remove_child", dup);
  402. editor_data->get_undo_redo().add_do_reference(dup);
  403. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  404. editor_data->get_undo_redo().add_do_method(sed, "live_debug_duplicate_node", edited_scene->get_path_to(node), dup->get_name());
  405. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_remove_node", NodePath(String(edited_scene->get_path_to(parent)).plus_file(dup->get_name())));
  406. }
  407. editor_data->get_undo_redo().commit_action();
  408. if (dupsingle)
  409. editor->push_item(dupsingle);
  410. } break;
  411. case TOOL_REPARENT: {
  412. if (!scene_tree->get_selected())
  413. break;
  414. if (editor_selection->is_selected(edited_scene)) {
  415. current_option = -1;
  416. accept->set_text(TTR("This operation can't be done on the tree root."));
  417. accept->popup_centered_minsize();
  418. break;
  419. }
  420. if (!_validate_no_foreign())
  421. break;
  422. List<Node *> nodes = editor_selection->get_selected_node_list();
  423. Set<Node *> nodeset;
  424. for (List<Node *>::Element *E = nodes.front(); E; E = E->next()) {
  425. nodeset.insert(E->get());
  426. }
  427. reparent_dialog->popup_centered_ratio();
  428. reparent_dialog->set_current(nodeset);
  429. } break;
  430. case TOOL_MAKE_ROOT: {
  431. List<Node *> nodes = editor_selection->get_selected_node_list();
  432. ERR_FAIL_COND(nodes.size() != 1);
  433. Node *node = nodes.front()->get();
  434. Node *root = get_tree()->get_edited_scene_root();
  435. if (node == root)
  436. return;
  437. //check that from node to root, all owners are right
  438. if (root->get_scene_inherited_state().is_valid()) {
  439. accept->set_text(TTR("Can't reparent nodes in inherited scenes, order of nodes can't change."));
  440. accept->popup_centered_minsize();
  441. return;
  442. }
  443. if (node->get_owner() != root) {
  444. accept->set_text(TTR("Node must belong to the edited scene to become root."));
  445. accept->popup_centered_minsize();
  446. return;
  447. }
  448. if (node->get_filename() != String()) {
  449. accept->set_text(TTR("Instantiated scenes can't become root"));
  450. accept->popup_centered_minsize();
  451. return;
  452. }
  453. editor_data->get_undo_redo().create_action("Make node as Root");
  454. editor_data->get_undo_redo().add_do_method(node->get_parent(), "remove_child", node);
  455. editor_data->get_undo_redo().add_do_method(root->get_parent(), "remove_child", root);
  456. editor_data->get_undo_redo().add_do_method(node, "add_child", root);
  457. editor_data->get_undo_redo().add_do_method(editor, "set_edited_scene", node);
  458. editor_data->get_undo_redo().add_do_method(node, "set_filename", root->get_filename());
  459. editor_data->get_undo_redo().add_do_method(root, "set_filename", String());
  460. editor_data->get_undo_redo().add_do_method(node, "set_owner", (Object *)NULL);
  461. editor_data->get_undo_redo().add_do_method(root, "set_owner", node);
  462. _node_replace_owner(root, root, node, MODE_DO);
  463. editor_data->get_undo_redo().add_undo_method(root, "set_filename", root->get_filename());
  464. editor_data->get_undo_redo().add_undo_method(node, "set_filename", String());
  465. editor_data->get_undo_redo().add_undo_method(node, "remove_child", root);
  466. editor_data->get_undo_redo().add_undo_method(editor, "set_edited_scene", root);
  467. editor_data->get_undo_redo().add_undo_method(node->get_parent(), "add_child", node);
  468. editor_data->get_undo_redo().add_undo_method(root, "set_owner", (Object *)NULL);
  469. editor_data->get_undo_redo().add_undo_method(node, "set_owner", root);
  470. _node_replace_owner(root, root, root, MODE_UNDO);
  471. editor_data->get_undo_redo().add_do_method(scene_tree, "update_tree");
  472. editor_data->get_undo_redo().add_undo_method(scene_tree, "update_tree");
  473. editor_data->get_undo_redo().add_undo_reference(root);
  474. editor_data->get_undo_redo().commit_action();
  475. } break;
  476. case TOOL_MULTI_EDIT: {
  477. Node *root = EditorNode::get_singleton()->get_edited_scene();
  478. if (!root)
  479. break;
  480. Ref<MultiNodeEdit> mne = memnew(MultiNodeEdit);
  481. for (const Map<Node *, Object *>::Element *E = EditorNode::get_singleton()->get_editor_selection()->get_selection().front(); E; E = E->next()) {
  482. mne->add_node(root->get_path_to(E->key()));
  483. }
  484. EditorNode::get_singleton()->push_item(mne.ptr());
  485. } break;
  486. case TOOL_ERASE: {
  487. List<Node *> remove_list = editor_selection->get_selected_node_list();
  488. if (remove_list.empty())
  489. return;
  490. if (!_validate_no_foreign())
  491. break;
  492. if (p_confirm_override) {
  493. _delete_confirm();
  494. } else {
  495. delete_dialog->set_text(TTR("Delete Node(s)?"));
  496. delete_dialog->popup_centered_minsize();
  497. }
  498. } break;
  499. case TOOL_MERGE_FROM_SCENE: {
  500. EditorNode::get_singleton()->merge_from_scene();
  501. } break;
  502. case TOOL_NEW_SCENE_FROM: {
  503. Node *scene = editor_data->get_edited_scene_root();
  504. if (!scene) {
  505. accept->set_text(TTR("This operation can't be done without a scene."));
  506. accept->popup_centered_minsize();
  507. break;
  508. }
  509. List<Node *> selection = editor_selection->get_selected_node_list();
  510. if (selection.size() != 1) {
  511. accept->set_text(TTR("This operation requires a single selected node."));
  512. accept->popup_centered_minsize();
  513. break;
  514. }
  515. Node *tocopy = selection.front()->get();
  516. if (tocopy == scene) {
  517. accept->set_text(TTR("Can not perform with the root node."));
  518. accept->popup_centered_minsize();
  519. break;
  520. }
  521. if (tocopy != editor_data->get_edited_scene_root() && tocopy->get_filename() != "") {
  522. accept->set_text(TTR("This operation can't be done on instanced scenes."));
  523. accept->popup_centered_minsize();
  524. break;
  525. }
  526. new_scene_from_dialog->set_mode(EditorFileDialog::MODE_SAVE_FILE);
  527. List<String> extensions;
  528. Ref<PackedScene> sd = memnew(PackedScene);
  529. ResourceSaver::get_recognized_extensions(sd, &extensions);
  530. new_scene_from_dialog->clear_filters();
  531. for (int i = 0; i < extensions.size(); i++) {
  532. new_scene_from_dialog->add_filter("*." + extensions[i] + " ; " + extensions[i].to_upper());
  533. }
  534. String existing;
  535. if (extensions.size()) {
  536. String root_name(tocopy->get_name());
  537. existing = root_name + "." + extensions.front()->get().to_lower();
  538. }
  539. new_scene_from_dialog->set_current_path(existing);
  540. new_scene_from_dialog->popup_centered_ratio();
  541. new_scene_from_dialog->set_title(TTR("Save New Scene As..."));
  542. } break;
  543. case TOOL_COPY_NODE_PATH: {
  544. List<Node *> selection = editor_selection->get_selected_node_list();
  545. List<Node *>::Element *e = selection.front();
  546. if (e) {
  547. Node *node = e->get();
  548. if (node) {
  549. Node *root = EditorNode::get_singleton()->get_edited_scene();
  550. NodePath path = root->get_path().rel_path_to(node->get_path());
  551. OS::get_singleton()->set_clipboard(path);
  552. }
  553. }
  554. } break;
  555. case TOOL_OPEN_DOCUMENTATION: {
  556. List<Node *> selection = editor_selection->get_selected_node_list();
  557. for (int i = 0; i < selection.size(); i++) {
  558. ScriptEditor::get_singleton()->goto_help("class_name:" + selection[i]->get_class());
  559. }
  560. EditorNode::get_singleton()->set_visible_editor(EditorNode::EDITOR_SCRIPT);
  561. } break;
  562. case TOOL_SCENE_EDITABLE_CHILDREN: {
  563. List<Node *> selection = editor_selection->get_selected_node_list();
  564. List<Node *>::Element *e = selection.front();
  565. if (e) {
  566. Node *node = e->get();
  567. if (node) {
  568. bool editable = EditorNode::get_singleton()->get_edited_scene()->is_editable_instance(node);
  569. if (editable) {
  570. editable_instance_remove_dialog->set_text(TTR("Disabling \"editable_instance\" will cause all properties of the node to be reverted to their default."));
  571. editable_instance_remove_dialog->popup_centered_minsize();
  572. break;
  573. }
  574. _toggle_editable_children();
  575. }
  576. }
  577. } break;
  578. case TOOL_SCENE_USE_PLACEHOLDER: {
  579. List<Node *> selection = editor_selection->get_selected_node_list();
  580. List<Node *>::Element *e = selection.front();
  581. if (e) {
  582. Node *node = e->get();
  583. if (node) {
  584. bool placeholder = node->get_scene_instance_load_placeholder();
  585. placeholder = !placeholder;
  586. int editable_item_idx = menu->get_item_idx_from_text(TTR("Editable Children"));
  587. int placeholder_item_idx = menu->get_item_idx_from_text(TTR("Load As Placeholder"));
  588. if (placeholder)
  589. EditorNode::get_singleton()->get_edited_scene()->set_editable_instance(node, false);
  590. node->set_scene_instance_load_placeholder(placeholder);
  591. menu->set_item_checked(editable_item_idx, false);
  592. menu->set_item_checked(placeholder_item_idx, placeholder);
  593. scene_tree->update_tree();
  594. }
  595. }
  596. } break;
  597. case TOOL_SCENE_MAKE_LOCAL: {
  598. List<Node *> selection = editor_selection->get_selected_node_list();
  599. List<Node *>::Element *e = selection.front();
  600. if (e) {
  601. Node *node = e->get();
  602. if (node) {
  603. Node *root = EditorNode::get_singleton()->get_edited_scene();
  604. UndoRedo *undo_redo = &editor_data->get_undo_redo();
  605. if (!root)
  606. break;
  607. ERR_FAIL_COND(node->get_filename() == String());
  608. undo_redo->create_action(TTR("Make Local"));
  609. undo_redo->add_do_method(node, "set_filename", "");
  610. undo_redo->add_undo_method(node, "set_filename", node->get_filename());
  611. _node_replace_owner(node, node, root);
  612. undo_redo->add_do_method(scene_tree, "update_tree");
  613. undo_redo->add_undo_method(scene_tree, "update_tree");
  614. undo_redo->commit_action();
  615. }
  616. }
  617. } break;
  618. case TOOL_SCENE_OPEN: {
  619. List<Node *> selection = editor_selection->get_selected_node_list();
  620. List<Node *>::Element *e = selection.front();
  621. if (e) {
  622. Node *node = e->get();
  623. if (node) {
  624. scene_tree->emit_signal("open", node->get_filename());
  625. }
  626. }
  627. } break;
  628. case TOOL_SCENE_CLEAR_INHERITANCE: {
  629. clear_inherit_confirm->popup_centered_minsize();
  630. } break;
  631. case TOOL_SCENE_CLEAR_INHERITANCE_CONFIRM: {
  632. List<Node *> selection = editor_selection->get_selected_node_list();
  633. List<Node *>::Element *e = selection.front();
  634. if (e) {
  635. Node *node = e->get();
  636. if (node) {
  637. node->set_scene_inherited_state(Ref<SceneState>());
  638. scene_tree->update_tree();
  639. EditorNode::get_singleton()->get_inspector()->update_tree();
  640. }
  641. }
  642. } break;
  643. case TOOL_SCENE_OPEN_INHERITED: {
  644. List<Node *> selection = editor_selection->get_selected_node_list();
  645. List<Node *>::Element *e = selection.front();
  646. if (e) {
  647. Node *node = e->get();
  648. if (node) {
  649. if (node && node->get_scene_inherited_state().is_valid()) {
  650. scene_tree->emit_signal("open", node->get_scene_inherited_state()->get_path());
  651. }
  652. }
  653. }
  654. } break;
  655. case TOOL_CREATE_2D_SCENE:
  656. case TOOL_CREATE_3D_SCENE:
  657. case TOOL_CREATE_USER_INTERFACE:
  658. case TOOL_CREATE_FAVORITE: {
  659. Node *new_node = NULL;
  660. if (TOOL_CREATE_FAVORITE == p_tool) {
  661. String name = selected_favorite_root.get_slicec(' ', 0);
  662. if (ScriptServer::is_global_class(name)) {
  663. new_node = Object::cast_to<Node>(ClassDB::instance(ScriptServer::get_global_class_base(name)));
  664. Ref<Script> script = ResourceLoader::load(ScriptServer::get_global_class_path(name), "Script");
  665. if (new_node && script.is_valid()) {
  666. new_node->set_script(script.get_ref_ptr());
  667. new_node->set_name(name);
  668. }
  669. } else {
  670. new_node = Object::cast_to<Node>(ClassDB::instance(selected_favorite_root));
  671. }
  672. if (!new_node) {
  673. ERR_EXPLAIN("Creating root from favorite '" + selected_favorite_root + "' failed. Creating 'Node' instead.");
  674. new_node = memnew(Node);
  675. }
  676. } else {
  677. switch (p_tool) {
  678. case TOOL_CREATE_2D_SCENE: new_node = memnew(Node2D); break;
  679. case TOOL_CREATE_3D_SCENE: new_node = memnew(Spatial); break;
  680. case TOOL_CREATE_USER_INTERFACE: {
  681. Control *node = memnew(Control);
  682. node->set_anchors_and_margins_preset(PRESET_WIDE); //more useful for resizable UIs.
  683. new_node = node;
  684. } break;
  685. }
  686. }
  687. editor_data->get_undo_redo().create_action("New Scene Root");
  688. editor_data->get_undo_redo().add_do_method(editor, "set_edited_scene", new_node);
  689. editor_data->get_undo_redo().add_do_method(scene_tree, "update_tree");
  690. editor_data->get_undo_redo().add_do_reference(new_node);
  691. editor_data->get_undo_redo().add_undo_method(editor, "set_edited_scene", (Object *)NULL);
  692. editor_data->get_undo_redo().commit_action();
  693. editor->edit_node(new_node);
  694. editor_selection->clear();
  695. editor_selection->add_node(new_node);
  696. } break;
  697. default: {
  698. if (p_tool >= EDIT_SUBRESOURCE_BASE) {
  699. int idx = p_tool - EDIT_SUBRESOURCE_BASE;
  700. ERR_FAIL_INDEX(idx, subresources.size());
  701. Object *obj = ObjectDB::get_instance(subresources[idx]);
  702. ERR_FAIL_COND(!obj);
  703. editor->push_item(obj);
  704. }
  705. }
  706. }
  707. }
  708. void SceneTreeDock::_notification(int p_what) {
  709. switch (p_what) {
  710. case NOTIFICATION_READY: {
  711. if (!first_enter)
  712. break;
  713. first_enter = false;
  714. CanvasItemEditorPlugin *canvas_item_plugin = Object::cast_to<CanvasItemEditorPlugin>(editor_data->get_editor("2D"));
  715. if (canvas_item_plugin) {
  716. canvas_item_plugin->get_canvas_item_editor()->connect("item_lock_status_changed", scene_tree, "_update_tree");
  717. canvas_item_plugin->get_canvas_item_editor()->connect("item_group_status_changed", scene_tree, "_update_tree");
  718. scene_tree->connect("node_changed", canvas_item_plugin->get_canvas_item_editor()->get_viewport_control(), "update");
  719. }
  720. SpatialEditorPlugin *spatial_editor_plugin = Object::cast_to<SpatialEditorPlugin>(editor_data->get_editor("3D"));
  721. spatial_editor_plugin->get_spatial_editor()->connect("item_lock_status_changed", scene_tree, "_update_tree");
  722. button_add->set_icon(get_icon("Add", "EditorIcons"));
  723. button_instance->set_icon(get_icon("Instance", "EditorIcons"));
  724. button_create_script->set_icon(get_icon("ScriptCreate", "EditorIcons"));
  725. button_clear_script->set_icon(get_icon("ScriptRemove", "EditorIcons"));
  726. filter->set_right_icon(get_icon("Search", "EditorIcons"));
  727. filter->set_clear_button_enabled(true);
  728. EditorNode::get_singleton()->get_editor_selection()->connect("selection_changed", this, "_selection_changed");
  729. // create_root_dialog
  730. HBoxContainer *top_row = memnew(HBoxContainer);
  731. top_row->set_name("NodeShortcutsTopRow");
  732. top_row->set_h_size_flags(SIZE_EXPAND_FILL);
  733. top_row->add_child(memnew(Label(TTR("Create Root Node:"))));
  734. top_row->add_spacer();
  735. ToolButton *node_shortcuts_toggle = memnew(ToolButton);
  736. node_shortcuts_toggle->set_name("NodeShortcutsToggle");
  737. node_shortcuts_toggle->set_icon(get_icon("Favorites", "EditorIcons"));
  738. node_shortcuts_toggle->set_toggle_mode(true);
  739. node_shortcuts_toggle->set_pressed(EDITOR_GET("_use_favorites_root_selection"));
  740. node_shortcuts_toggle->set_anchors_and_margins_preset(Control::PRESET_CENTER_RIGHT);
  741. node_shortcuts_toggle->connect("pressed", this, "_update_create_root_dialog");
  742. top_row->add_child(node_shortcuts_toggle);
  743. create_root_dialog->add_child(top_row);
  744. VBoxContainer *node_shortcuts = memnew(VBoxContainer);
  745. node_shortcuts->set_name("NodeShortcuts");
  746. VBoxContainer *beginner_node_shortcuts = memnew(VBoxContainer);
  747. beginner_node_shortcuts->set_name("BeginnerNodeShortcuts");
  748. node_shortcuts->add_child(beginner_node_shortcuts);
  749. Button *button_2d = memnew(Button);
  750. beginner_node_shortcuts->add_child(button_2d);
  751. button_2d->set_text(TTR("2D Scene"));
  752. button_2d->set_icon(get_icon("Node2D", "EditorIcons"));
  753. button_2d->connect("pressed", this, "_tool_selected", make_binds(TOOL_CREATE_2D_SCENE, false));
  754. Button *button_3d = memnew(Button);
  755. beginner_node_shortcuts->add_child(button_3d);
  756. button_3d->set_text(TTR("3D Scene"));
  757. button_3d->set_icon(get_icon("Spatial", "EditorIcons"));
  758. button_3d->connect("pressed", this, "_tool_selected", make_binds(TOOL_CREATE_3D_SCENE, false));
  759. Button *button_ui = memnew(Button);
  760. beginner_node_shortcuts->add_child(button_ui);
  761. button_ui->set_text(TTR("User Interface"));
  762. button_ui->set_icon(get_icon("Control", "EditorIcons"));
  763. button_ui->connect("pressed", this, "_tool_selected", make_binds(TOOL_CREATE_USER_INTERFACE, false));
  764. VBoxContainer *favorite_node_shortcuts = memnew(VBoxContainer);
  765. favorite_node_shortcuts->set_name("FavoriteNodeShortcuts");
  766. node_shortcuts->add_child(favorite_node_shortcuts);
  767. Button *button_custom = memnew(Button);
  768. node_shortcuts->add_child(button_custom);
  769. button_custom->set_text(TTR("Custom Node"));
  770. button_custom->set_icon(get_icon("Add", "EditorIcons"));
  771. button_custom->connect("pressed", this, "_tool_selected", make_binds(TOOL_NEW, false));
  772. node_shortcuts->add_spacer();
  773. create_root_dialog->add_child(node_shortcuts);
  774. _update_create_root_dialog();
  775. } break;
  776. case NOTIFICATION_ENTER_TREE: {
  777. clear_inherit_confirm->connect("confirmed", this, "_tool_selected", varray(TOOL_SCENE_CLEAR_INHERITANCE_CONFIRM));
  778. } break;
  779. case NOTIFICATION_EXIT_TREE: {
  780. clear_inherit_confirm->disconnect("confirmed", this, "_tool_selected");
  781. } break;
  782. case EditorSettings::NOTIFICATION_EDITOR_SETTINGS_CHANGED: {
  783. button_add->set_icon(get_icon("Add", "EditorIcons"));
  784. button_instance->set_icon(get_icon("Instance", "EditorIcons"));
  785. button_create_script->set_icon(get_icon("ScriptCreate", "EditorIcons"));
  786. button_clear_script->set_icon(get_icon("ScriptRemove", "EditorIcons"));
  787. filter->set_right_icon(get_icon("Search", "EditorIcons"));
  788. filter->set_clear_button_enabled(true);
  789. } break;
  790. case NOTIFICATION_PROCESS: {
  791. bool show_create_root = bool(EDITOR_GET("interface/editors/show_scene_tree_root_selection")) && get_tree()->get_edited_scene_root() == NULL;
  792. if (show_create_root != create_root_dialog->is_visible_in_tree()) {
  793. if (show_create_root) {
  794. create_root_dialog->show();
  795. scene_tree->hide();
  796. } else {
  797. create_root_dialog->hide();
  798. scene_tree->show();
  799. }
  800. }
  801. } break;
  802. }
  803. }
  804. void SceneTreeDock::_node_replace_owner(Node *p_base, Node *p_node, Node *p_root, ReplaceOwnerMode p_mode) {
  805. if (p_node->get_owner() == p_base && p_node != p_root) {
  806. UndoRedo *undo_redo = &editor_data->get_undo_redo();
  807. switch (p_mode) {
  808. case MODE_BIDI: {
  809. undo_redo->add_do_method(p_node, "set_owner", p_root);
  810. undo_redo->add_undo_method(p_node, "set_owner", p_base);
  811. } break;
  812. case MODE_DO: {
  813. undo_redo->add_do_method(p_node, "set_owner", p_root);
  814. } break;
  815. case MODE_UNDO: {
  816. undo_redo->add_undo_method(p_node, "set_owner", p_root);
  817. } break;
  818. }
  819. }
  820. for (int i = 0; i < p_node->get_child_count(); i++) {
  821. _node_replace_owner(p_base, p_node->get_child(i), p_root, p_mode);
  822. }
  823. }
  824. void SceneTreeDock::_load_request(const String &p_path) {
  825. editor->open_request(p_path);
  826. }
  827. void SceneTreeDock::_script_open_request(const Ref<Script> &p_script) {
  828. editor->edit_resource(p_script);
  829. }
  830. void SceneTreeDock::_node_selected() {
  831. Node *node = scene_tree->get_selected();
  832. if (!node) {
  833. editor->push_item(NULL);
  834. return;
  835. }
  836. if (ScriptEditor::get_singleton()->is_visible_in_tree()) {
  837. restore_script_editor_on_drag = true;
  838. }
  839. editor->push_item(node);
  840. }
  841. void SceneTreeDock::_node_renamed() {
  842. _node_selected();
  843. }
  844. void SceneTreeDock::_set_owners(Node *p_owner, const Array &p_nodes) {
  845. for (int i = 0; i < p_nodes.size(); i++) {
  846. Node *n = Object::cast_to<Node>(p_nodes[i]);
  847. if (!n)
  848. continue;
  849. n->set_owner(p_owner);
  850. }
  851. }
  852. void SceneTreeDock::_fill_path_renames(Vector<StringName> base_path, Vector<StringName> new_base_path, Node *p_node, List<Pair<NodePath, NodePath> > *p_renames) {
  853. base_path.push_back(p_node->get_name());
  854. if (new_base_path.size())
  855. new_base_path.push_back(p_node->get_name());
  856. NodePath from(base_path, true);
  857. NodePath to;
  858. if (new_base_path.size())
  859. to = NodePath(new_base_path, true);
  860. Pair<NodePath, NodePath> npp;
  861. npp.first = from;
  862. npp.second = to;
  863. p_renames->push_back(npp);
  864. for (int i = 0; i < p_node->get_child_count(); i++) {
  865. _fill_path_renames(base_path, new_base_path, p_node->get_child(i), p_renames);
  866. }
  867. }
  868. void SceneTreeDock::fill_path_renames(Node *p_node, Node *p_new_parent, List<Pair<NodePath, NodePath> > *p_renames) {
  869. if (!bool(EDITOR_DEF("editors/animation/autorename_animation_tracks", true)))
  870. return;
  871. Vector<StringName> base_path;
  872. Node *n = p_node->get_parent();
  873. while (n) {
  874. base_path.push_back(n->get_name());
  875. n = n->get_parent();
  876. }
  877. base_path.invert();
  878. Vector<StringName> new_base_path;
  879. if (p_new_parent) {
  880. n = p_new_parent;
  881. while (n) {
  882. new_base_path.push_back(n->get_name());
  883. n = n->get_parent();
  884. }
  885. new_base_path.invert();
  886. }
  887. _fill_path_renames(base_path, new_base_path, p_node, p_renames);
  888. }
  889. void SceneTreeDock::perform_node_renames(Node *p_base, List<Pair<NodePath, NodePath> > *p_renames, Map<Ref<Animation>, Set<int> > *r_rem_anims) {
  890. Map<Ref<Animation>, Set<int> > rem_anims;
  891. if (!r_rem_anims)
  892. r_rem_anims = &rem_anims;
  893. if (!p_base) {
  894. p_base = edited_scene;
  895. }
  896. if (!p_base)
  897. return;
  898. // Renaming node paths used in script instances
  899. if (p_base->get_script_instance()) {
  900. ScriptInstance *si = p_base->get_script_instance();
  901. if (si) {
  902. List<PropertyInfo> properties;
  903. si->get_property_list(&properties);
  904. for (List<PropertyInfo>::Element *E = properties.front(); E; E = E->next()) {
  905. String propertyname = E->get().name;
  906. Variant p = p_base->get(propertyname);
  907. if (p.get_type() == Variant::NODE_PATH) {
  908. // Goes through all paths to check if its matching
  909. for (List<Pair<NodePath, NodePath> >::Element *F = p_renames->front(); F; F = F->next()) {
  910. NodePath root_path = p_base->get_path();
  911. NodePath rel_path_old = root_path.rel_path_to(F->get().first);
  912. NodePath rel_path_new = F->get().second;
  913. // if not empty, get new relative path
  914. if (F->get().second != NodePath()) {
  915. rel_path_new = root_path.rel_path_to(F->get().second);
  916. }
  917. // if old path detected, then it needs to be replaced with the new one
  918. if (p == rel_path_old) {
  919. editor_data->get_undo_redo().add_do_property(p_base, propertyname, rel_path_new);
  920. editor_data->get_undo_redo().add_undo_property(p_base, propertyname, rel_path_old);
  921. p_base->set(propertyname, rel_path_new);
  922. break;
  923. }
  924. }
  925. }
  926. }
  927. }
  928. }
  929. bool autorename_animation_tracks = bool(EDITOR_DEF("editors/animation/autorename_animation_tracks", true));
  930. if (autorename_animation_tracks && Object::cast_to<AnimationPlayer>(p_base)) {
  931. AnimationPlayer *ap = Object::cast_to<AnimationPlayer>(p_base);
  932. List<StringName> anims;
  933. ap->get_animation_list(&anims);
  934. Node *root = ap->get_node(ap->get_root());
  935. if (root) {
  936. NodePath root_path = root->get_path();
  937. NodePath new_root_path = root_path;
  938. for (List<Pair<NodePath, NodePath> >::Element *E = p_renames->front(); E; E = E->next()) {
  939. if (E->get().first == root_path) {
  940. new_root_path = E->get().second;
  941. break;
  942. }
  943. }
  944. if (new_root_path != NodePath()) {
  945. //will not be erased
  946. for (List<StringName>::Element *E = anims.front(); E; E = E->next()) {
  947. Ref<Animation> anim = ap->get_animation(E->get());
  948. if (!r_rem_anims->has(anim)) {
  949. r_rem_anims->insert(anim, Set<int>());
  950. Set<int> &ran = r_rem_anims->find(anim)->get();
  951. for (int i = 0; i < anim->get_track_count(); i++)
  952. ran.insert(i);
  953. }
  954. Set<int> &ran = r_rem_anims->find(anim)->get();
  955. if (anim.is_null())
  956. continue;
  957. for (int i = 0; i < anim->get_track_count(); i++) {
  958. NodePath track_np = anim->track_get_path(i);
  959. Node *n = root->get_node(track_np);
  960. if (!n) {
  961. continue;
  962. }
  963. NodePath old_np = n->get_path();
  964. if (!ran.has(i))
  965. continue; //channel was removed
  966. for (List<Pair<NodePath, NodePath> >::Element *F = p_renames->front(); F; F = F->next()) {
  967. if (F->get().first == old_np) {
  968. if (F->get().second == NodePath()) {
  969. //will be erased
  970. int idx = 0;
  971. Set<int>::Element *EI = ran.front();
  972. ERR_FAIL_COND(!EI); //bug
  973. while (EI->get() != i) {
  974. idx++;
  975. EI = EI->next();
  976. ERR_FAIL_COND(!EI); //another bug
  977. }
  978. editor_data->get_undo_redo().add_do_method(anim.ptr(), "remove_track", idx);
  979. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "add_track", anim->track_get_type(i), idx);
  980. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "track_set_path", idx, track_np);
  981. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "track_set_interpolation_type", idx, anim->track_get_interpolation_type(i));
  982. for (int j = 0; j < anim->track_get_key_count(i); j++) {
  983. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "track_insert_key", idx, anim->track_get_key_time(i, j), anim->track_get_key_value(i, j), anim->track_get_key_transition(i, j));
  984. }
  985. ran.erase(i); //byebye channel
  986. } else {
  987. //will be renamed
  988. NodePath rel_path = new_root_path.rel_path_to(F->get().second);
  989. NodePath new_path = NodePath(rel_path.get_names(), track_np.get_subnames(), false);
  990. if (new_path == track_np)
  991. continue; //bleh
  992. editor_data->get_undo_redo().add_do_method(anim.ptr(), "track_set_path", i, new_path);
  993. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "track_set_path", i, track_np);
  994. }
  995. }
  996. }
  997. }
  998. }
  999. }
  1000. }
  1001. }
  1002. for (int i = 0; i < p_base->get_child_count(); i++)
  1003. perform_node_renames(p_base->get_child(i), p_renames, r_rem_anims);
  1004. }
  1005. void SceneTreeDock::_node_prerenamed(Node *p_node, const String &p_new_name) {
  1006. List<Pair<NodePath, NodePath> > path_renames;
  1007. Vector<StringName> base_path;
  1008. Node *n = p_node->get_parent();
  1009. while (n) {
  1010. base_path.push_back(n->get_name());
  1011. n = n->get_parent();
  1012. }
  1013. base_path.invert();
  1014. Vector<StringName> new_base_path = base_path;
  1015. base_path.push_back(p_node->get_name());
  1016. new_base_path.push_back(p_new_name);
  1017. Pair<NodePath, NodePath> npp;
  1018. npp.first = NodePath(base_path, true);
  1019. npp.second = NodePath(new_base_path, true);
  1020. path_renames.push_back(npp);
  1021. for (int i = 0; i < p_node->get_child_count(); i++)
  1022. _fill_path_renames(base_path, new_base_path, p_node->get_child(i), &path_renames);
  1023. perform_node_renames(NULL, &path_renames);
  1024. }
  1025. bool SceneTreeDock::_validate_no_foreign() {
  1026. List<Node *> selection = editor_selection->get_selected_node_list();
  1027. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1028. if (E->get() != edited_scene && E->get()->get_owner() != edited_scene) {
  1029. accept->set_text(TTR("Can't operate on nodes from a foreign scene!"));
  1030. accept->popup_centered_minsize();
  1031. return false;
  1032. }
  1033. // When edited_scene inherits from another one the root Node will be the parent Scene,
  1034. // we don't want to consider that Node a foreign one otherwise we would not be able to
  1035. // delete it
  1036. if (edited_scene->get_scene_inherited_state().is_valid() && edited_scene == E->get()) {
  1037. continue;
  1038. }
  1039. if (edited_scene->get_scene_inherited_state().is_valid() && edited_scene->get_scene_inherited_state()->find_node_by_path(edited_scene->get_path_to(E->get())) >= 0) {
  1040. accept->set_text(TTR("Can't operate on nodes the current scene inherits from!"));
  1041. accept->popup_centered_minsize();
  1042. return false;
  1043. }
  1044. }
  1045. return true;
  1046. }
  1047. void SceneTreeDock::_node_reparent(NodePath p_path, bool p_keep_global_xform) {
  1048. Node *new_parent = scene_root->get_node(p_path);
  1049. ERR_FAIL_COND(!new_parent);
  1050. List<Node *> selection = editor_selection->get_selected_node_list();
  1051. if (selection.empty())
  1052. return; //nothing to reparent
  1053. Vector<Node *> nodes;
  1054. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1055. nodes.push_back(E->get());
  1056. }
  1057. _do_reparent(new_parent, -1, nodes, p_keep_global_xform);
  1058. }
  1059. void SceneTreeDock::_do_reparent(Node *p_new_parent, int p_position_in_parent, Vector<Node *> p_nodes, bool p_keep_global_xform) {
  1060. Node *new_parent = p_new_parent;
  1061. ERR_FAIL_COND(!new_parent);
  1062. Node *validate = new_parent;
  1063. while (validate) {
  1064. if (p_nodes.find(validate) != -1) {
  1065. ERR_EXPLAIN("Selection changed at some point.. can't reparent");
  1066. ERR_FAIL();
  1067. return;
  1068. }
  1069. validate = validate->get_parent();
  1070. }
  1071. //ok all valid
  1072. if (p_nodes.size() == 0)
  1073. return; //nothing to reparent
  1074. //sort by tree order, so re-adding is easy
  1075. p_nodes.sort_custom<Node::Comparator>();
  1076. editor_data->get_undo_redo().create_action(TTR("Reparent Node"));
  1077. List<Pair<NodePath, NodePath> > path_renames;
  1078. Vector<StringName> former_names;
  1079. int inc = 0;
  1080. for (int ni = 0; ni < p_nodes.size(); ni++) {
  1081. //no undo for now, sorry
  1082. Node *node = p_nodes[ni];
  1083. fill_path_renames(node, new_parent, &path_renames);
  1084. former_names.push_back(node->get_name());
  1085. List<Node *> owned;
  1086. node->get_owned_by(node->get_owner(), &owned);
  1087. Array owners;
  1088. for (List<Node *>::Element *E = owned.front(); E; E = E->next()) {
  1089. owners.push_back(E->get());
  1090. }
  1091. if (new_parent == node->get_parent() && node->get_index() < p_position_in_parent + ni) {
  1092. //if child will generate a gap when moved, adjust
  1093. inc--;
  1094. }
  1095. editor_data->get_undo_redo().add_do_method(node->get_parent(), "remove_child", node);
  1096. editor_data->get_undo_redo().add_do_method(new_parent, "add_child", node);
  1097. if (p_position_in_parent >= 0)
  1098. editor_data->get_undo_redo().add_do_method(new_parent, "move_child", node, p_position_in_parent + inc);
  1099. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  1100. String old_name = former_names[ni];
  1101. String new_name = new_parent->validate_child_name(node);
  1102. // name was modified, fix the path renames
  1103. if (old_name.casecmp_to(new_name) != 0) {
  1104. // Fix the to name to have the new name
  1105. NodePath old_new_name = path_renames[ni].second;
  1106. NodePath new_path;
  1107. Vector<StringName> unfixed_new_names = old_new_name.get_names();
  1108. Vector<StringName> fixed_new_names;
  1109. // Get last name and replace with fixed new name
  1110. for (int a = 0; a < (unfixed_new_names.size() - 1); a++) {
  1111. fixed_new_names.push_back(unfixed_new_names[a]);
  1112. }
  1113. fixed_new_names.push_back(new_name);
  1114. NodePath fixed_node_path = NodePath(fixed_new_names, true);
  1115. path_renames[ni].second = fixed_node_path;
  1116. }
  1117. editor_data->get_undo_redo().add_do_method(sed, "live_debug_reparent_node", edited_scene->get_path_to(node), edited_scene->get_path_to(new_parent), new_name, p_position_in_parent + inc);
  1118. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_reparent_node", NodePath(String(edited_scene->get_path_to(new_parent)).plus_file(new_name)), edited_scene->get_path_to(node->get_parent()), node->get_name(), node->get_index());
  1119. if (p_keep_global_xform) {
  1120. if (Object::cast_to<Node2D>(node))
  1121. editor_data->get_undo_redo().add_do_method(node, "set_global_transform", Object::cast_to<Node2D>(node)->get_global_transform());
  1122. if (Object::cast_to<Spatial>(node))
  1123. editor_data->get_undo_redo().add_do_method(node, "set_global_transform", Object::cast_to<Spatial>(node)->get_global_transform());
  1124. if (Object::cast_to<Control>(node))
  1125. editor_data->get_undo_redo().add_do_method(node, "set_global_position", Object::cast_to<Control>(node)->get_global_position());
  1126. }
  1127. editor_data->get_undo_redo().add_do_method(this, "_set_owners", edited_scene, owners);
  1128. if (AnimationPlayerEditor::singleton->get_track_editor()->get_root() == node)
  1129. editor_data->get_undo_redo().add_do_method(AnimationPlayerEditor::singleton->get_track_editor(), "set_root", node);
  1130. editor_data->get_undo_redo().add_undo_method(new_parent, "remove_child", node);
  1131. editor_data->get_undo_redo().add_undo_method(node, "set_name", former_names[ni]);
  1132. inc++;
  1133. }
  1134. //add and move in a second step.. (so old order is preserved)
  1135. for (int ni = 0; ni < p_nodes.size(); ni++) {
  1136. Node *node = p_nodes[ni];
  1137. List<Node *> owned;
  1138. node->get_owned_by(node->get_owner(), &owned);
  1139. Array owners;
  1140. for (List<Node *>::Element *E = owned.front(); E; E = E->next()) {
  1141. owners.push_back(E->get());
  1142. }
  1143. int child_pos = node->get_position_in_parent();
  1144. editor_data->get_undo_redo().add_undo_method(node->get_parent(), "add_child", node);
  1145. editor_data->get_undo_redo().add_undo_method(node->get_parent(), "move_child", node, child_pos);
  1146. editor_data->get_undo_redo().add_undo_method(this, "_set_owners", edited_scene, owners);
  1147. if (AnimationPlayerEditor::singleton->get_track_editor()->get_root() == node)
  1148. editor_data->get_undo_redo().add_undo_method(AnimationPlayerEditor::singleton->get_track_editor(), "set_root", node);
  1149. if (p_keep_global_xform) {
  1150. if (Object::cast_to<Node2D>(node))
  1151. editor_data->get_undo_redo().add_undo_method(node, "set_transform", Object::cast_to<Node2D>(node)->get_transform());
  1152. if (Object::cast_to<Spatial>(node))
  1153. editor_data->get_undo_redo().add_undo_method(node, "set_transform", Object::cast_to<Spatial>(node)->get_transform());
  1154. if (Object::cast_to<Control>(node))
  1155. editor_data->get_undo_redo().add_undo_method(node, "set_position", Object::cast_to<Control>(node)->get_position());
  1156. }
  1157. }
  1158. perform_node_renames(NULL, &path_renames);
  1159. editor_data->get_undo_redo().commit_action();
  1160. }
  1161. void SceneTreeDock::_script_created(Ref<Script> p_script) {
  1162. List<Node *> selected = editor_selection->get_selected_node_list();
  1163. if (selected.empty())
  1164. return;
  1165. editor_data->get_undo_redo().create_action(TTR("Attach Script"));
  1166. for (List<Node *>::Element *E = selected.front(); E; E = E->next()) {
  1167. Ref<Script> existing = E->get()->get_script();
  1168. editor_data->get_undo_redo().add_do_method(E->get(), "set_script", p_script.get_ref_ptr());
  1169. editor_data->get_undo_redo().add_undo_method(E->get(), "set_script", existing);
  1170. }
  1171. editor_data->get_undo_redo().commit_action();
  1172. editor->push_item(p_script.operator->());
  1173. _update_script_button();
  1174. }
  1175. void SceneTreeDock::_toggle_editable_children() {
  1176. List<Node *> selection = editor_selection->get_selected_node_list();
  1177. List<Node *>::Element *e = selection.front();
  1178. if (e) {
  1179. Node *node = e->get();
  1180. if (node) {
  1181. bool editable = EditorNode::get_singleton()->get_edited_scene()->is_editable_instance(node);
  1182. int editable_item_idx = menu->get_item_idx_from_text(TTR("Editable Children"));
  1183. int placeholder_item_idx = menu->get_item_idx_from_text(TTR("Load As Placeholder"));
  1184. editable = !editable;
  1185. EditorNode::get_singleton()->get_edited_scene()->set_editable_instance(node, editable);
  1186. menu->set_item_checked(editable_item_idx, editable);
  1187. if (editable) {
  1188. node->set_scene_instance_load_placeholder(false);
  1189. menu->set_item_checked(placeholder_item_idx, false);
  1190. }
  1191. SpatialEditor::get_singleton()->update_all_gizmos(node);
  1192. scene_tree->update_tree();
  1193. }
  1194. }
  1195. }
  1196. void SceneTreeDock::_delete_confirm() {
  1197. List<Node *> remove_list = editor_selection->get_selected_node_list();
  1198. if (remove_list.empty())
  1199. return;
  1200. editor->get_editor_plugins_over()->make_visible(false);
  1201. editor_data->get_undo_redo().create_action(TTR("Remove Node(s)"));
  1202. bool entire_scene = false;
  1203. for (List<Node *>::Element *E = remove_list.front(); E; E = E->next()) {
  1204. if (E->get() == edited_scene) {
  1205. entire_scene = true;
  1206. }
  1207. }
  1208. if (entire_scene) {
  1209. editor_data->get_undo_redo().add_do_method(editor, "set_edited_scene", (Object *)NULL);
  1210. editor_data->get_undo_redo().add_undo_method(editor, "set_edited_scene", edited_scene);
  1211. editor_data->get_undo_redo().add_undo_method(edited_scene, "set_owner", edited_scene->get_owner());
  1212. editor_data->get_undo_redo().add_undo_method(scene_tree, "update_tree");
  1213. editor_data->get_undo_redo().add_undo_reference(edited_scene);
  1214. } else {
  1215. remove_list.sort_custom<Node::Comparator>(); //sort nodes to keep positions
  1216. List<Pair<NodePath, NodePath> > path_renames;
  1217. //delete from animation
  1218. for (List<Node *>::Element *E = remove_list.front(); E; E = E->next()) {
  1219. Node *n = E->get();
  1220. if (!n->is_inside_tree() || !n->get_parent())
  1221. continue;
  1222. fill_path_renames(n, NULL, &path_renames);
  1223. }
  1224. perform_node_renames(NULL, &path_renames);
  1225. //delete for read
  1226. for (List<Node *>::Element *E = remove_list.front(); E; E = E->next()) {
  1227. Node *n = E->get();
  1228. if (!n->is_inside_tree() || !n->get_parent())
  1229. continue;
  1230. List<Node *> owned;
  1231. n->get_owned_by(n->get_owner(), &owned);
  1232. Array owners;
  1233. for (List<Node *>::Element *F = owned.front(); F; F = F->next()) {
  1234. owners.push_back(F->get());
  1235. }
  1236. editor_data->get_undo_redo().add_do_method(n->get_parent(), "remove_child", n);
  1237. editor_data->get_undo_redo().add_undo_method(n->get_parent(), "add_child", n);
  1238. editor_data->get_undo_redo().add_undo_method(n->get_parent(), "move_child", n, n->get_index());
  1239. if (AnimationPlayerEditor::singleton->get_track_editor()->get_root() == n)
  1240. editor_data->get_undo_redo().add_undo_method(AnimationPlayerEditor::singleton->get_track_editor(), "set_root", n);
  1241. editor_data->get_undo_redo().add_undo_method(this, "_set_owners", edited_scene, owners);
  1242. editor_data->get_undo_redo().add_undo_reference(n);
  1243. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  1244. editor_data->get_undo_redo().add_do_method(sed, "live_debug_remove_and_keep_node", edited_scene->get_path_to(n), n->get_instance_id());
  1245. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_restore_node", n->get_instance_id(), edited_scene->get_path_to(n->get_parent()), n->get_index());
  1246. }
  1247. }
  1248. editor_data->get_undo_redo().commit_action();
  1249. // hack, force 2d editor viewport to refresh after deletion
  1250. if (CanvasItemEditor *editor = CanvasItemEditor::get_singleton())
  1251. editor->get_viewport_control()->update();
  1252. editor->push_item(NULL);
  1253. // Fixes the EditorHistory from still offering deleted notes
  1254. EditorHistory *editor_history = EditorNode::get_singleton()->get_editor_history();
  1255. editor_history->cleanup_history();
  1256. EditorNode::get_singleton()->get_inspector_dock()->call("_prepare_history");
  1257. }
  1258. void SceneTreeDock::_update_script_button() {
  1259. if (EditorNode::get_singleton()->get_editor_selection()->get_selection().size() == 0) {
  1260. button_create_script->hide();
  1261. button_clear_script->hide();
  1262. } else if (EditorNode::get_singleton()->get_editor_selection()->get_selection().size() == 1) {
  1263. Node *n = EditorNode::get_singleton()->get_editor_selection()->get_selected_node_list()[0];
  1264. if (n->get_script().is_null()) {
  1265. button_create_script->show();
  1266. button_clear_script->hide();
  1267. } else {
  1268. button_create_script->hide();
  1269. button_clear_script->show();
  1270. }
  1271. } else {
  1272. button_create_script->show();
  1273. List<Node *> selection = EditorNode::get_singleton()->get_editor_selection()->get_selected_node_list();
  1274. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1275. Node *n = E->get();
  1276. if (!n->get_script().is_null()) {
  1277. button_clear_script->show();
  1278. return;
  1279. }
  1280. }
  1281. button_clear_script->hide();
  1282. }
  1283. }
  1284. void SceneTreeDock::_selection_changed() {
  1285. int selection_size = EditorNode::get_singleton()->get_editor_selection()->get_selection().size();
  1286. if (selection_size > 1) {
  1287. //automatically turn on multi-edit
  1288. _tool_selected(TOOL_MULTI_EDIT);
  1289. } else if (selection_size == 1) {
  1290. editor->push_item(EditorNode::get_singleton()->get_editor_selection()->get_selected_node_list()[0]);
  1291. } else {
  1292. editor->push_item(NULL);
  1293. }
  1294. _update_script_button();
  1295. }
  1296. void SceneTreeDock::_create() {
  1297. if (current_option == TOOL_NEW) {
  1298. Node *parent = NULL;
  1299. if (edited_scene) {
  1300. // If root exists in edited scene
  1301. parent = scene_tree->get_selected();
  1302. if (!parent)
  1303. parent = edited_scene;
  1304. } else {
  1305. // If no root exist in edited scene
  1306. parent = scene_root;
  1307. ERR_FAIL_COND(!parent);
  1308. }
  1309. Object *c = create_dialog->instance_selected();
  1310. ERR_FAIL_COND(!c);
  1311. Node *child = Object::cast_to<Node>(c);
  1312. ERR_FAIL_COND(!child);
  1313. editor_data->get_undo_redo().create_action(TTR("Create Node"));
  1314. if (edited_scene) {
  1315. editor_data->get_undo_redo().add_do_method(parent, "add_child", child);
  1316. editor_data->get_undo_redo().add_do_method(child, "set_owner", edited_scene);
  1317. editor_data->get_undo_redo().add_do_method(editor_selection, "clear");
  1318. editor_data->get_undo_redo().add_do_method(editor_selection, "add_node", child);
  1319. editor_data->get_undo_redo().add_do_reference(child);
  1320. editor_data->get_undo_redo().add_undo_method(parent, "remove_child", child);
  1321. String new_name = parent->validate_child_name(child);
  1322. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  1323. editor_data->get_undo_redo().add_do_method(sed, "live_debug_create_node", edited_scene->get_path_to(parent), child->get_class(), new_name);
  1324. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_remove_node", NodePath(String(edited_scene->get_path_to(parent)).plus_file(new_name)));
  1325. } else {
  1326. editor_data->get_undo_redo().add_do_method(editor, "set_edited_scene", child);
  1327. editor_data->get_undo_redo().add_do_method(scene_tree, "update_tree");
  1328. editor_data->get_undo_redo().add_do_reference(child);
  1329. editor_data->get_undo_redo().add_undo_method(editor, "set_edited_scene", (Object *)NULL);
  1330. }
  1331. editor_data->get_undo_redo().commit_action();
  1332. editor->push_item(c);
  1333. editor_selection->clear();
  1334. editor_selection->add_node(child);
  1335. if (Object::cast_to<Control>(c)) {
  1336. //make editor more comfortable, so some controls don't appear super shrunk
  1337. Control *ct = Object::cast_to<Control>(c);
  1338. Size2 ms = ct->get_minimum_size();
  1339. if (ms.width < 4)
  1340. ms.width = 40;
  1341. if (ms.height < 4)
  1342. ms.height = 40;
  1343. ct->set_size(ms);
  1344. }
  1345. } else if (current_option == TOOL_REPLACE) {
  1346. List<Node *> selection = editor_selection->get_selected_node_list();
  1347. ERR_FAIL_COND(selection.size() <= 0);
  1348. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1349. Node *n = E->get();
  1350. ERR_FAIL_COND(!n);
  1351. Object *c = create_dialog->instance_selected();
  1352. ERR_FAIL_COND(!c);
  1353. Node *newnode = Object::cast_to<Node>(c);
  1354. ERR_FAIL_COND(!newnode);
  1355. replace_node(n, newnode);
  1356. }
  1357. }
  1358. }
  1359. void SceneTreeDock::replace_node(Node *p_node, Node *p_by_node, bool p_keep_properties) {
  1360. Node *n = p_node;
  1361. Node *newnode = p_by_node;
  1362. if (p_keep_properties) {
  1363. Node *default_oldnode = Object::cast_to<Node>(ClassDB::instance(n->get_class()));
  1364. List<PropertyInfo> pinfo;
  1365. n->get_property_list(&pinfo);
  1366. for (List<PropertyInfo>::Element *E = pinfo.front(); E; E = E->next()) {
  1367. if (!(E->get().usage & PROPERTY_USAGE_STORAGE))
  1368. continue;
  1369. if (E->get().name == "__meta__")
  1370. continue;
  1371. if (default_oldnode->get(E->get().name) != n->get(E->get().name)) {
  1372. newnode->set(E->get().name, n->get(E->get().name));
  1373. }
  1374. }
  1375. memdelete(default_oldnode);
  1376. }
  1377. editor->push_item(NULL);
  1378. //reconnect signals
  1379. List<MethodInfo> sl;
  1380. n->get_signal_list(&sl);
  1381. for (List<MethodInfo>::Element *E = sl.front(); E; E = E->next()) {
  1382. List<Object::Connection> cl;
  1383. n->get_signal_connection_list(E->get().name, &cl);
  1384. for (List<Object::Connection>::Element *F = cl.front(); F; F = F->next()) {
  1385. Object::Connection &c = F->get();
  1386. if (!(c.flags & Object::CONNECT_PERSIST))
  1387. continue;
  1388. newnode->connect(c.signal, c.target, c.method, varray(), Object::CONNECT_PERSIST);
  1389. }
  1390. }
  1391. String newname = n->get_name();
  1392. List<Node *> to_erase;
  1393. for (int i = 0; i < n->get_child_count(); i++) {
  1394. if (n->get_child(i)->get_owner() == NULL && n->is_owned_by_parent()) {
  1395. to_erase.push_back(n->get_child(i));
  1396. }
  1397. }
  1398. n->replace_by(newnode, true);
  1399. if (n == edited_scene) {
  1400. edited_scene = newnode;
  1401. editor->set_edited_scene(newnode);
  1402. newnode->set_editable_instances(n->get_editable_instances());
  1403. }
  1404. //small hack to make collisionshapes and other kind of nodes to work
  1405. for (int i = 0; i < newnode->get_child_count(); i++) {
  1406. Node *c = newnode->get_child(i);
  1407. c->call("set_transform", c->call("get_transform"));
  1408. }
  1409. editor_data->get_undo_redo().clear_history();
  1410. newnode->set_name(newname);
  1411. editor->push_item(newnode);
  1412. memdelete(n);
  1413. while (to_erase.front()) {
  1414. memdelete(to_erase.front()->get());
  1415. to_erase.pop_front();
  1416. }
  1417. }
  1418. void SceneTreeDock::set_edited_scene(Node *p_scene) {
  1419. edited_scene = p_scene;
  1420. }
  1421. void SceneTreeDock::set_selected(Node *p_node, bool p_emit_selected) {
  1422. scene_tree->set_selected(p_node, p_emit_selected);
  1423. }
  1424. void SceneTreeDock::import_subscene() {
  1425. import_subscene_dialog->popup_centered_ratio();
  1426. }
  1427. void SceneTreeDock::_import_subscene() {
  1428. Node *parent = scene_tree->get_selected();
  1429. if (!parent) {
  1430. parent = editor_data->get_edited_scene_root();
  1431. ERR_FAIL_COND(!parent);
  1432. }
  1433. import_subscene_dialog->move(parent, edited_scene);
  1434. editor_data->get_undo_redo().clear_history(); //no undo for now..
  1435. }
  1436. void SceneTreeDock::_new_scene_from(String p_file) {
  1437. List<Node *> selection = editor_selection->get_selected_node_list();
  1438. if (selection.size() != 1) {
  1439. accept->set_text(TTR("This operation requires a single selected node."));
  1440. accept->popup_centered_minsize();
  1441. return;
  1442. }
  1443. if (EditorNode::get_singleton()->is_scene_open(p_file)) {
  1444. accept->set_text(TTR("Can't overwrite scene that is still open!"));
  1445. accept->popup_centered_minsize();
  1446. return;
  1447. }
  1448. Node *base = selection.front()->get();
  1449. Map<Node *, Node *> reown;
  1450. reown[editor_data->get_edited_scene_root()] = base;
  1451. Node *copy = base->duplicate_and_reown(reown);
  1452. if (copy) {
  1453. Ref<PackedScene> sdata = memnew(PackedScene);
  1454. Error err = sdata->pack(copy);
  1455. memdelete(copy);
  1456. if (err != OK) {
  1457. accept->set_text(TTR("Couldn't save new scene. Likely dependencies (instances) couldn't be satisfied."));
  1458. accept->popup_centered_minsize();
  1459. return;
  1460. }
  1461. int flg = 0;
  1462. if (EditorSettings::get_singleton()->get("filesystem/on_save/compress_binary_resources"))
  1463. flg |= ResourceSaver::FLAG_COMPRESS;
  1464. err = ResourceSaver::save(p_file, sdata, flg);
  1465. if (err != OK) {
  1466. accept->set_text(TTR("Error saving scene."));
  1467. accept->popup_centered_minsize();
  1468. return;
  1469. }
  1470. _replace_with_branch_scene(p_file, base);
  1471. } else {
  1472. accept->set_text(TTR("Error duplicating scene to save it."));
  1473. accept->popup_centered_minsize();
  1474. return;
  1475. }
  1476. }
  1477. static bool _is_node_visible(Node *p_node) {
  1478. if (!p_node->get_owner())
  1479. return false;
  1480. if (p_node->get_owner() != EditorNode::get_singleton()->get_edited_scene() && !EditorNode::get_singleton()->get_edited_scene()->is_editable_instance(p_node->get_owner()))
  1481. return false;
  1482. return true;
  1483. }
  1484. static bool _has_visible_children(Node *p_node) {
  1485. bool collapsed = p_node->is_displayed_folded();
  1486. if (collapsed)
  1487. return false;
  1488. for (int i = 0; i < p_node->get_child_count(); i++) {
  1489. Node *child = p_node->get_child(i);
  1490. if (!_is_node_visible(child))
  1491. continue;
  1492. return true;
  1493. }
  1494. return false;
  1495. }
  1496. void SceneTreeDock::_normalize_drop(Node *&to_node, int &to_pos, int p_type) {
  1497. to_pos = -1;
  1498. if (p_type == -1) {
  1499. //drop at above selected node
  1500. if (to_node == EditorNode::get_singleton()->get_edited_scene()) {
  1501. to_node = NULL;
  1502. ERR_EXPLAIN("Cannot perform drop above the root node!");
  1503. ERR_FAIL();
  1504. }
  1505. to_pos = to_node->get_index();
  1506. to_node = to_node->get_parent();
  1507. } else if (p_type == 1) {
  1508. //drop at below selected node
  1509. if (to_node == EditorNode::get_singleton()->get_edited_scene()) {
  1510. //if at lower sibling of root node
  1511. to_pos = 0; //just insert at beginning of root node
  1512. return;
  1513. }
  1514. Node *lower_sibling = NULL;
  1515. if (_has_visible_children(to_node)) {
  1516. to_pos = 0;
  1517. } else {
  1518. for (int i = to_node->get_index() + 1; i < to_node->get_parent()->get_child_count(); i++) {
  1519. Node *c = to_node->get_parent()->get_child(i);
  1520. if (_is_node_visible(c)) {
  1521. lower_sibling = c;
  1522. break;
  1523. }
  1524. }
  1525. if (lower_sibling) {
  1526. to_pos = lower_sibling->get_index();
  1527. }
  1528. to_node = to_node->get_parent();
  1529. }
  1530. }
  1531. }
  1532. void SceneTreeDock::_files_dropped(Vector<String> p_files, NodePath p_to, int p_type) {
  1533. Node *node = get_node(p_to);
  1534. ERR_FAIL_COND(!node);
  1535. int to_pos = -1;
  1536. _normalize_drop(node, to_pos, p_type);
  1537. _perform_instance_scenes(p_files, node, to_pos);
  1538. }
  1539. void SceneTreeDock::_script_dropped(String p_file, NodePath p_to) {
  1540. Ref<Script> scr = ResourceLoader::load(p_file);
  1541. ERR_FAIL_COND(!scr.is_valid());
  1542. Node *n = get_node(p_to);
  1543. if (n) {
  1544. editor_data->get_undo_redo().create_action(TTR("Attach Script"));
  1545. editor_data->get_undo_redo().add_do_method(n, "set_script", scr);
  1546. editor_data->get_undo_redo().add_undo_method(n, "set_script", n->get_script());
  1547. editor_data->get_undo_redo().add_do_method(this, "_update_script_button");
  1548. editor_data->get_undo_redo().add_undo_method(this, "_update_script_button");
  1549. editor_data->get_undo_redo().commit_action();
  1550. }
  1551. }
  1552. void SceneTreeDock::_nodes_dragged(Array p_nodes, NodePath p_to, int p_type) {
  1553. Vector<Node *> nodes;
  1554. Node *to_node;
  1555. for (int i = 0; i < p_nodes.size(); i++) {
  1556. Node *n = get_node((p_nodes[i]));
  1557. if (n) {
  1558. nodes.push_back(n);
  1559. }
  1560. }
  1561. if (nodes.size() == 0)
  1562. return;
  1563. to_node = get_node(p_to);
  1564. if (!to_node)
  1565. return;
  1566. int to_pos = -1;
  1567. _normalize_drop(to_node, to_pos, p_type);
  1568. _do_reparent(to_node, to_pos, nodes, true);
  1569. }
  1570. void SceneTreeDock::_add_children_to_popup(Object *p_obj, int p_depth) {
  1571. if (p_depth > 8)
  1572. return;
  1573. List<PropertyInfo> pinfo;
  1574. p_obj->get_property_list(&pinfo);
  1575. for (List<PropertyInfo>::Element *E = pinfo.front(); E; E = E->next()) {
  1576. if (!(E->get().usage & PROPERTY_USAGE_EDITOR))
  1577. continue;
  1578. if (E->get().hint != PROPERTY_HINT_RESOURCE_TYPE)
  1579. continue;
  1580. Variant value = p_obj->get(E->get().name);
  1581. if (value.get_type() != Variant::OBJECT)
  1582. continue;
  1583. Object *obj = value;
  1584. if (!obj)
  1585. continue;
  1586. Ref<Texture> icon = EditorNode::get_singleton()->get_object_icon(obj);
  1587. if (menu->get_item_count() == 0) {
  1588. menu->add_submenu_item(TTR("Sub-Resources"), "Sub-Resources");
  1589. }
  1590. int index = menu_subresources->get_item_count();
  1591. menu_subresources->add_icon_item(icon, E->get().name.capitalize(), EDIT_SUBRESOURCE_BASE + subresources.size());
  1592. menu_subresources->set_item_h_offset(index, p_depth * 10 * EDSCALE);
  1593. subresources.push_back(obj->get_instance_id());
  1594. _add_children_to_popup(obj, p_depth + 1);
  1595. }
  1596. }
  1597. void SceneTreeDock::_tree_rmb(const Vector2 &p_menu_pos) {
  1598. if (!EditorNode::get_singleton()->get_edited_scene()) {
  1599. menu->clear();
  1600. menu->add_icon_shortcut(get_icon("Add", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/add_child_node"), TOOL_NEW);
  1601. menu->add_icon_shortcut(get_icon("Instance", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/instance_scene"), TOOL_INSTANCE);
  1602. menu->set_size(Size2(1, 1));
  1603. menu->set_position(p_menu_pos);
  1604. menu->popup();
  1605. return;
  1606. }
  1607. List<Node *> selection = editor_selection->get_selected_node_list();
  1608. if (selection.size() == 0)
  1609. return;
  1610. menu->clear();
  1611. Ref<Script> existing_script;
  1612. if (selection.size() == 1) {
  1613. Node *selected = selection[0];
  1614. subresources.clear();
  1615. menu_subresources->clear();
  1616. menu_subresources->set_size(Size2(1, 1));
  1617. _add_children_to_popup(selection.front()->get(), 0);
  1618. if (menu->get_item_count() > 0)
  1619. menu->add_separator();
  1620. menu->add_icon_shortcut(get_icon("Add", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/add_child_node"), TOOL_NEW);
  1621. menu->add_icon_shortcut(get_icon("Instance", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/instance_scene"), TOOL_INSTANCE);
  1622. menu->add_separator();
  1623. existing_script = selected->get_script();
  1624. }
  1625. if (!existing_script.is_valid()) {
  1626. menu->add_icon_shortcut(get_icon("ScriptCreate", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/attach_script"), TOOL_ATTACH_SCRIPT);
  1627. }
  1628. if (selection.size() > 1 || existing_script.is_valid()) {
  1629. menu->add_icon_shortcut(get_icon("ScriptRemove", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/clear_script"), TOOL_CLEAR_SCRIPT);
  1630. menu->add_icon_shortcut(get_icon("ScriptExtend", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/extend_script"), TOOL_ATTACH_SCRIPT);
  1631. }
  1632. menu->add_separator();
  1633. if (selection.size() == 1) {
  1634. menu->add_icon_shortcut(get_icon("Rename", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/rename"), TOOL_RENAME);
  1635. }
  1636. menu->add_icon_shortcut(get_icon("Reload", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/change_node_type"), TOOL_REPLACE);
  1637. if (scene_tree->get_selected() != edited_scene) {
  1638. menu->add_separator();
  1639. menu->add_icon_shortcut(get_icon("MoveUp", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/move_up"), TOOL_MOVE_UP);
  1640. menu->add_icon_shortcut(get_icon("MoveDown", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/move_down"), TOOL_MOVE_DOWN);
  1641. menu->add_icon_shortcut(get_icon("Duplicate", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/duplicate"), TOOL_DUPLICATE);
  1642. menu->add_icon_shortcut(get_icon("Reparent", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/reparent"), TOOL_REPARENT);
  1643. }
  1644. if (selection.size() == 1) {
  1645. menu->add_icon_shortcut(get_icon("NewRoot", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/make_root"), TOOL_MAKE_ROOT);
  1646. menu->add_separator();
  1647. menu->add_icon_shortcut(get_icon("Blend", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/merge_from_scene"), TOOL_MERGE_FROM_SCENE);
  1648. menu->add_icon_shortcut(get_icon("CreateNewSceneFrom", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/save_branch_as_scene"), TOOL_NEW_SCENE_FROM);
  1649. menu->add_separator();
  1650. menu->add_icon_shortcut(get_icon("CopyNodePath", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/copy_node_path"), TOOL_COPY_NODE_PATH);
  1651. bool is_external = (selection[0]->get_filename() != "");
  1652. if (is_external) {
  1653. bool is_inherited = selection[0]->get_scene_inherited_state() != NULL;
  1654. bool is_top_level = selection[0]->get_owner() == NULL;
  1655. if (is_inherited && is_top_level) {
  1656. menu->add_separator();
  1657. menu->add_item(TTR("Clear Inheritance"), TOOL_SCENE_CLEAR_INHERITANCE);
  1658. menu->add_icon_item(get_icon("Load", "EditorIcons"), TTR("Open in Editor"), TOOL_SCENE_OPEN_INHERITED);
  1659. } else if (!is_top_level) {
  1660. menu->add_separator();
  1661. bool editable = EditorNode::get_singleton()->get_edited_scene()->is_editable_instance(selection[0]);
  1662. bool placeholder = selection[0]->get_scene_instance_load_placeholder();
  1663. menu->add_check_item(TTR("Editable Children"), TOOL_SCENE_EDITABLE_CHILDREN);
  1664. menu->add_check_item(TTR("Load As Placeholder"), TOOL_SCENE_USE_PLACEHOLDER);
  1665. menu->add_item(TTR("Make Local"), TOOL_SCENE_MAKE_LOCAL);
  1666. menu->add_icon_item(get_icon("Load", "EditorIcons"), TTR("Open in Editor"), TOOL_SCENE_OPEN);
  1667. menu->set_item_checked(menu->get_item_idx_from_text(TTR("Editable Children")), editable);
  1668. menu->set_item_checked(menu->get_item_idx_from_text(TTR("Load As Placeholder")), placeholder);
  1669. }
  1670. }
  1671. }
  1672. if (selection.size() > 1) {
  1673. //this is not a commonly used action, it makes no sense for it to be where it was nor always present.
  1674. menu->add_separator();
  1675. menu->add_icon_shortcut(get_icon("Rename", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/batch_rename"), TOOL_BATCH_RENAME);
  1676. }
  1677. menu->add_separator();
  1678. menu->add_icon_item(get_icon("Help", "EditorIcons"), TTR("Open documentation"), TOOL_OPEN_DOCUMENTATION);
  1679. menu->add_separator();
  1680. menu->add_icon_shortcut(get_icon("Remove", "EditorIcons"), ED_SHORTCUT("scene_tree/delete", TTR("Delete Node(s)"), KEY_DELETE), TOOL_ERASE);
  1681. menu->set_size(Size2(1, 1));
  1682. menu->set_position(p_menu_pos);
  1683. menu->popup();
  1684. }
  1685. void SceneTreeDock::_filter_changed(const String &p_filter) {
  1686. scene_tree->set_filter(p_filter);
  1687. }
  1688. String SceneTreeDock::get_filter() {
  1689. return filter->get_text();
  1690. }
  1691. void SceneTreeDock::set_filter(const String &p_filter) {
  1692. filter->set_text(p_filter);
  1693. scene_tree->set_filter(p_filter);
  1694. }
  1695. void SceneTreeDock::_focus_node() {
  1696. Node *node = scene_tree->get_selected();
  1697. ERR_FAIL_COND(!node);
  1698. if (node->is_class("CanvasItem")) {
  1699. CanvasItemEditorPlugin *editor = Object::cast_to<CanvasItemEditorPlugin>(editor_data->get_editor("2D"));
  1700. editor->get_canvas_item_editor()->focus_selection();
  1701. } else {
  1702. SpatialEditorPlugin *editor = Object::cast_to<SpatialEditorPlugin>(editor_data->get_editor("3D"));
  1703. editor->get_spatial_editor()->get_editor_viewport(0)->focus_selection();
  1704. }
  1705. }
  1706. void SceneTreeDock::open_script_dialog(Node *p_for_node) {
  1707. scene_tree->set_selected(p_for_node, false);
  1708. _tool_selected(TOOL_ATTACH_SCRIPT);
  1709. }
  1710. void SceneTreeDock::add_remote_tree_editor(Control *p_remote) {
  1711. ERR_FAIL_COND(remote_tree != NULL);
  1712. add_child(p_remote);
  1713. remote_tree = p_remote;
  1714. remote_tree->hide();
  1715. }
  1716. void SceneTreeDock::show_remote_tree() {
  1717. _remote_tree_selected();
  1718. }
  1719. void SceneTreeDock::hide_remote_tree() {
  1720. _local_tree_selected();
  1721. }
  1722. void SceneTreeDock::show_tab_buttons() {
  1723. button_hb->show();
  1724. }
  1725. void SceneTreeDock::hide_tab_buttons() {
  1726. button_hb->hide();
  1727. }
  1728. void SceneTreeDock::_remote_tree_selected() {
  1729. scene_tree->hide();
  1730. if (remote_tree)
  1731. remote_tree->show();
  1732. edit_remote->set_pressed(true);
  1733. edit_local->set_pressed(false);
  1734. emit_signal("remote_tree_selected");
  1735. }
  1736. void SceneTreeDock::_local_tree_selected() {
  1737. scene_tree->show();
  1738. if (remote_tree)
  1739. remote_tree->hide();
  1740. edit_remote->set_pressed(false);
  1741. edit_local->set_pressed(true);
  1742. }
  1743. void SceneTreeDock::_update_create_root_dialog() {
  1744. BaseButton *toggle = Object::cast_to<BaseButton>(create_root_dialog->get_node(String("NodeShortcutsTopRow/NodeShortcutsToggle")));
  1745. Node *node_shortcuts = create_root_dialog->get_node(String("NodeShortcuts"));
  1746. if (!toggle || !node_shortcuts)
  1747. return;
  1748. Control *beginner_nodes = Object::cast_to<Control>(node_shortcuts->get_node(String("BeginnerNodeShortcuts")));
  1749. Control *favorite_nodes = Object::cast_to<Control>(node_shortcuts->get_node(String("FavoriteNodeShortcuts")));
  1750. if (!beginner_nodes || !favorite_nodes)
  1751. return;
  1752. EditorSettings::get_singleton()->set_setting("_use_favorites_root_selection", toggle->is_pressed());
  1753. EditorSettings::get_singleton()->save();
  1754. if (toggle->is_pressed()) {
  1755. for (int i = 0; i < favorite_nodes->get_child_count(); i++) {
  1756. favorite_nodes->get_child(i)->queue_delete();
  1757. }
  1758. FileAccess *f = FileAccess::open(EditorSettings::get_singleton()->get_project_settings_dir().plus_file("favorites.Node"), FileAccess::READ);
  1759. if (f) {
  1760. while (!f->eof_reached()) {
  1761. String l = f->get_line().strip_edges();
  1762. if (l != String()) {
  1763. Button *button = memnew(Button);
  1764. favorite_nodes->add_child(button);
  1765. button->set_text(TTR(l));
  1766. String name = l.get_slicec(' ', 0);
  1767. if (ScriptServer::is_global_class(name))
  1768. name = ScriptServer::get_global_class_base(name);
  1769. button->set_icon(EditorNode::get_singleton()->get_class_icon(name));
  1770. button->connect("pressed", this, "_favorite_root_selected", make_binds(l));
  1771. }
  1772. }
  1773. memdelete(f);
  1774. }
  1775. if (!favorite_nodes->is_visible_in_tree()) {
  1776. favorite_nodes->show();
  1777. beginner_nodes->hide();
  1778. }
  1779. } else {
  1780. if (!beginner_nodes->is_visible_in_tree()) {
  1781. beginner_nodes->show();
  1782. favorite_nodes->hide();
  1783. }
  1784. }
  1785. }
  1786. void SceneTreeDock::_favorite_root_selected(const String &p_class) {
  1787. selected_favorite_root = p_class;
  1788. _tool_selected(TOOL_CREATE_FAVORITE, false);
  1789. }
  1790. void SceneTreeDock::_bind_methods() {
  1791. ClassDB::bind_method(D_METHOD("_tool_selected"), &SceneTreeDock::_tool_selected, DEFVAL(false));
  1792. ClassDB::bind_method(D_METHOD("_create"), &SceneTreeDock::_create);
  1793. ClassDB::bind_method(D_METHOD("_node_reparent"), &SceneTreeDock::_node_reparent);
  1794. ClassDB::bind_method(D_METHOD("_set_owners"), &SceneTreeDock::_set_owners);
  1795. ClassDB::bind_method(D_METHOD("_node_selected"), &SceneTreeDock::_node_selected);
  1796. ClassDB::bind_method(D_METHOD("_node_renamed"), &SceneTreeDock::_node_renamed);
  1797. ClassDB::bind_method(D_METHOD("_script_created"), &SceneTreeDock::_script_created);
  1798. ClassDB::bind_method(D_METHOD("_load_request"), &SceneTreeDock::_load_request);
  1799. ClassDB::bind_method(D_METHOD("_script_open_request"), &SceneTreeDock::_script_open_request);
  1800. ClassDB::bind_method(D_METHOD("_unhandled_key_input"), &SceneTreeDock::_unhandled_key_input);
  1801. ClassDB::bind_method(D_METHOD("_input"), &SceneTreeDock::_input);
  1802. ClassDB::bind_method(D_METHOD("_nodes_drag_begin"), &SceneTreeDock::_nodes_drag_begin);
  1803. ClassDB::bind_method(D_METHOD("_delete_confirm"), &SceneTreeDock::_delete_confirm);
  1804. ClassDB::bind_method(D_METHOD("_toggle_editable_children"), &SceneTreeDock::_toggle_editable_children);
  1805. ClassDB::bind_method(D_METHOD("_node_prerenamed"), &SceneTreeDock::_node_prerenamed);
  1806. ClassDB::bind_method(D_METHOD("_import_subscene"), &SceneTreeDock::_import_subscene);
  1807. ClassDB::bind_method(D_METHOD("_selection_changed"), &SceneTreeDock::_selection_changed);
  1808. ClassDB::bind_method(D_METHOD("_new_scene_from"), &SceneTreeDock::_new_scene_from);
  1809. ClassDB::bind_method(D_METHOD("_nodes_dragged"), &SceneTreeDock::_nodes_dragged);
  1810. ClassDB::bind_method(D_METHOD("_files_dropped"), &SceneTreeDock::_files_dropped);
  1811. ClassDB::bind_method(D_METHOD("_quick_open"), &SceneTreeDock::_quick_open);
  1812. ClassDB::bind_method(D_METHOD("_script_dropped"), &SceneTreeDock::_script_dropped);
  1813. ClassDB::bind_method(D_METHOD("_tree_rmb"), &SceneTreeDock::_tree_rmb);
  1814. ClassDB::bind_method(D_METHOD("_filter_changed"), &SceneTreeDock::_filter_changed);
  1815. ClassDB::bind_method(D_METHOD("_focus_node"), &SceneTreeDock::_focus_node);
  1816. ClassDB::bind_method(D_METHOD("_remote_tree_selected"), &SceneTreeDock::_remote_tree_selected);
  1817. ClassDB::bind_method(D_METHOD("_local_tree_selected"), &SceneTreeDock::_local_tree_selected);
  1818. ClassDB::bind_method(D_METHOD("_update_script_button"), &SceneTreeDock::_update_script_button);
  1819. ClassDB::bind_method(D_METHOD("_favorite_root_selected"), &SceneTreeDock::_favorite_root_selected);
  1820. ClassDB::bind_method(D_METHOD("_update_create_root_dialog"), &SceneTreeDock::_update_create_root_dialog);
  1821. ClassDB::bind_method(D_METHOD("instance"), &SceneTreeDock::instance);
  1822. ADD_SIGNAL(MethodInfo("remote_tree_selected"));
  1823. }
  1824. SceneTreeDock::SceneTreeDock(EditorNode *p_editor, Node *p_scene_root, EditorSelection *p_editor_selection, EditorData &p_editor_data) {
  1825. set_name("Scene");
  1826. editor = p_editor;
  1827. edited_scene = NULL;
  1828. editor_data = &p_editor_data;
  1829. editor_selection = p_editor_selection;
  1830. scene_root = p_scene_root;
  1831. VBoxContainer *vbc = this;
  1832. HBoxContainer *filter_hbc = memnew(HBoxContainer);
  1833. filter_hbc->add_constant_override("separate", 0);
  1834. ToolButton *tb;
  1835. ED_SHORTCUT("scene_tree/rename", TTR("Rename"));
  1836. ED_SHORTCUT("scene_tree/batch_rename", TTR("Batch Rename"), KEY_MASK_CMD | KEY_F2);
  1837. ED_SHORTCUT("scene_tree/add_child_node", TTR("Add Child Node"), KEY_MASK_CMD | KEY_A);
  1838. ED_SHORTCUT("scene_tree/instance_scene", TTR("Instance Child Scene"));
  1839. ED_SHORTCUT("scene_tree/change_node_type", TTR("Change Type"));
  1840. ED_SHORTCUT("scene_tree/attach_script", TTR("Attach Script"));
  1841. ED_SHORTCUT("scene_tree/extend_script", TTR("Extend Script"));
  1842. ED_SHORTCUT("scene_tree/clear_script", TTR("Clear Script"));
  1843. ED_SHORTCUT("scene_tree/move_up", TTR("Move Up"), KEY_MASK_CMD | KEY_UP);
  1844. ED_SHORTCUT("scene_tree/move_down", TTR("Move Down"), KEY_MASK_CMD | KEY_DOWN);
  1845. ED_SHORTCUT("scene_tree/duplicate", TTR("Duplicate"), KEY_MASK_CMD | KEY_D);
  1846. ED_SHORTCUT("scene_tree/reparent", TTR("Reparent"));
  1847. ED_SHORTCUT("scene_tree/make_root", TTR("Make Scene Root"));
  1848. ED_SHORTCUT("scene_tree/merge_from_scene", TTR("Merge From Scene"));
  1849. ED_SHORTCUT("scene_tree/save_branch_as_scene", TTR("Save Branch as Scene"));
  1850. ED_SHORTCUT("scene_tree/copy_node_path", TTR("Copy Node Path"), KEY_MASK_CMD | KEY_C);
  1851. ED_SHORTCUT("scene_tree/delete_no_confirm", TTR("Delete (No Confirm)"), KEY_MASK_SHIFT | KEY_DELETE);
  1852. ED_SHORTCUT("scene_tree/delete", TTR("Delete"), KEY_DELETE);
  1853. tb = memnew(ToolButton);
  1854. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_NEW, false));
  1855. tb->set_tooltip(TTR("Add/Create a New Node"));
  1856. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/add_child_node"));
  1857. filter_hbc->add_child(tb);
  1858. button_add = tb;
  1859. tb = memnew(ToolButton);
  1860. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_INSTANCE, false));
  1861. tb->set_tooltip(TTR("Instance a scene file as a Node. Creates an inherited scene if no root node exists."));
  1862. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/instance_scene"));
  1863. filter_hbc->add_child(tb);
  1864. button_instance = tb;
  1865. vbc->add_child(filter_hbc);
  1866. filter = memnew(LineEdit);
  1867. filter->set_h_size_flags(SIZE_EXPAND_FILL);
  1868. filter->set_placeholder(TTR("Filter nodes"));
  1869. filter_hbc->add_child(filter);
  1870. filter->add_constant_override("minimum_spaces", 0);
  1871. filter->connect("text_changed", this, "_filter_changed");
  1872. tb = memnew(ToolButton);
  1873. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_ATTACH_SCRIPT, false));
  1874. tb->set_tooltip(TTR("Attach a new or existing script for the selected node."));
  1875. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/attach_script"));
  1876. filter_hbc->add_child(tb);
  1877. tb->hide();
  1878. button_create_script = tb;
  1879. tb = memnew(ToolButton);
  1880. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_CLEAR_SCRIPT, false));
  1881. tb->set_tooltip(TTR("Clear a script for the selected node."));
  1882. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/clear_script"));
  1883. filter_hbc->add_child(tb);
  1884. button_clear_script = tb;
  1885. tb->hide();
  1886. button_hb = memnew(HBoxContainer);
  1887. vbc->add_child(button_hb);
  1888. edit_remote = memnew(ToolButton);
  1889. button_hb->add_child(edit_remote);
  1890. edit_remote->set_h_size_flags(SIZE_EXPAND_FILL);
  1891. edit_remote->set_text(TTR("Remote"));
  1892. edit_remote->set_toggle_mode(true);
  1893. edit_remote->connect("pressed", this, "_remote_tree_selected");
  1894. edit_local = memnew(ToolButton);
  1895. button_hb->add_child(edit_local);
  1896. edit_local->set_h_size_flags(SIZE_EXPAND_FILL);
  1897. edit_local->set_text(TTR("Local"));
  1898. edit_local->set_toggle_mode(true);
  1899. edit_local->connect("pressed", this, "_local_tree_selected");
  1900. remote_tree = NULL;
  1901. button_hb->hide();
  1902. create_root_dialog = memnew(VBoxContainer);
  1903. vbc->add_child(create_root_dialog);
  1904. create_root_dialog->hide();
  1905. scene_tree = memnew(SceneTreeEditor(false, true, true));
  1906. vbc->add_child(scene_tree);
  1907. scene_tree->set_v_size_flags(SIZE_EXPAND | SIZE_FILL);
  1908. scene_tree->connect("rmb_pressed", this, "_tree_rmb");
  1909. scene_tree->connect("node_selected", this, "_node_selected", varray(), CONNECT_DEFERRED);
  1910. scene_tree->connect("node_renamed", this, "_node_renamed", varray(), CONNECT_DEFERRED);
  1911. scene_tree->connect("node_prerename", this, "_node_prerenamed");
  1912. scene_tree->connect("open", this, "_load_request");
  1913. scene_tree->connect("open_script", this, "_script_open_request");
  1914. scene_tree->connect("nodes_rearranged", this, "_nodes_dragged");
  1915. scene_tree->connect("files_dropped", this, "_files_dropped");
  1916. scene_tree->connect("script_dropped", this, "_script_dropped");
  1917. scene_tree->connect("nodes_dragged", this, "_nodes_drag_begin");
  1918. scene_tree->get_scene_tree()->connect("item_double_clicked", this, "_focus_node");
  1919. scene_tree->set_undo_redo(&editor_data->get_undo_redo());
  1920. scene_tree->set_editor_selection(editor_selection);
  1921. create_dialog = memnew(CreateDialog);
  1922. create_dialog->set_base_type("Node");
  1923. add_child(create_dialog);
  1924. create_dialog->connect("create", this, "_create");
  1925. create_dialog->connect("favorites_updated", this, "_update_create_root_dialog");
  1926. rename_dialog = memnew(RenameDialog(scene_tree, &editor_data->get_undo_redo()));
  1927. add_child(rename_dialog);
  1928. script_create_dialog = memnew(ScriptCreateDialog);
  1929. add_child(script_create_dialog);
  1930. script_create_dialog->connect("script_created", this, "_script_created");
  1931. reparent_dialog = memnew(ReparentDialog);
  1932. add_child(reparent_dialog);
  1933. reparent_dialog->connect("reparent", this, "_node_reparent");
  1934. accept = memnew(AcceptDialog);
  1935. add_child(accept);
  1936. quick_open = memnew(EditorQuickOpen);
  1937. add_child(quick_open);
  1938. quick_open->connect("quick_open", this, "_quick_open");
  1939. set_process_unhandled_key_input(true);
  1940. delete_dialog = memnew(ConfirmationDialog);
  1941. add_child(delete_dialog);
  1942. delete_dialog->connect("confirmed", this, "_delete_confirm");
  1943. editable_instance_remove_dialog = memnew(ConfirmationDialog);
  1944. add_child(editable_instance_remove_dialog);
  1945. editable_instance_remove_dialog->connect("confirmed", this, "_toggle_editable_children");
  1946. import_subscene_dialog = memnew(EditorSubScene);
  1947. add_child(import_subscene_dialog);
  1948. import_subscene_dialog->connect("subscene_selected", this, "_import_subscene");
  1949. new_scene_from_dialog = memnew(EditorFileDialog);
  1950. new_scene_from_dialog->set_mode(EditorFileDialog::MODE_SAVE_FILE);
  1951. add_child(new_scene_from_dialog);
  1952. new_scene_from_dialog->connect("file_selected", this, "_new_scene_from");
  1953. menu = memnew(PopupMenu);
  1954. add_child(menu);
  1955. menu->connect("id_pressed", this, "_tool_selected");
  1956. menu->set_hide_on_window_lose_focus(true);
  1957. menu_subresources = memnew(PopupMenu);
  1958. menu_subresources->set_name("Sub-Resources");
  1959. menu_subresources->connect("id_pressed", this, "_tool_selected");
  1960. menu->add_child(menu_subresources);
  1961. first_enter = true;
  1962. restore_script_editor_on_drag = false;
  1963. clear_inherit_confirm = memnew(ConfirmationDialog);
  1964. clear_inherit_confirm->set_text(TTR("Clear Inheritance? (No Undo!)"));
  1965. clear_inherit_confirm->get_ok()->set_text(TTR("Clear"));
  1966. add_child(clear_inherit_confirm);
  1967. set_process_input(true);
  1968. set_process(true);
  1969. EDITOR_DEF("interface/editors/show_scene_tree_root_selection", true);
  1970. EDITOR_DEF("_use_favorites_root_selection", false);
  1971. }