scene_tree_dock.cpp 83 KB

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