2
0

scene_tree_dock.cpp 85 KB

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