scene_tree_dock.cpp 64 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004
  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-2017 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2017 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/animation_editor.h"
  35. #include "editor/editor_node.h"
  36. #include "editor/editor_settings.h"
  37. #include "editor/multi_node_edit.h"
  38. #include "editor/plugins/animation_player_editor_plugin.h"
  39. #include "editor/plugins/canvas_item_editor_plugin.h"
  40. #include "editor/plugins/script_editor_plugin.h"
  41. #include "editor/plugins/spatial_editor_plugin.h"
  42. #include "editor/script_editor_debugger.h"
  43. #include "scene/main/viewport.h"
  44. #include "scene/resources/packed_scene.h"
  45. void SceneTreeDock::_nodes_drag_begin() {
  46. if (restore_script_editor_on_drag) {
  47. EditorNode::get_singleton()->set_visible_editor(EditorNode::EDITOR_SCRIPT);
  48. restore_script_editor_on_drag = false;
  49. }
  50. }
  51. void SceneTreeDock::_input(Ref<InputEvent> p_event) {
  52. Ref<InputEventMouseButton> mb = p_event;
  53. if (mb.is_valid() && !mb->is_pressed() && mb->get_button_index() == BUTTON_LEFT) {
  54. restore_script_editor_on_drag = false; //lost chance
  55. }
  56. }
  57. void SceneTreeDock::_unhandled_key_input(Ref<InputEvent> p_event) {
  58. if (get_viewport()->get_modal_stack_top())
  59. return; //ignore because of modal window
  60. if (!p_event->is_pressed() || p_event->is_echo())
  61. return;
  62. if (ED_IS_SHORTCUT("scene_tree/add_child_node", p_event)) {
  63. _tool_selected(TOOL_NEW);
  64. } else if (ED_IS_SHORTCUT("scene_tree/instance_scene", p_event)) {
  65. _tool_selected(TOOL_INSTANCE);
  66. } else if (ED_IS_SHORTCUT("scene_tree/change_node_type", p_event)) {
  67. _tool_selected(TOOL_REPLACE);
  68. } else if (ED_IS_SHORTCUT("scene_tree/duplicate", p_event)) {
  69. _tool_selected(TOOL_DUPLICATE);
  70. } else if (ED_IS_SHORTCUT("scene_tree/attach_script", p_event)) {
  71. _tool_selected(TOOL_ATTACH_SCRIPT);
  72. } else if (ED_IS_SHORTCUT("scene_tree/clear_script", p_event)) {
  73. _tool_selected(TOOL_CLEAR_SCRIPT);
  74. } else if (ED_IS_SHORTCUT("scene_tree/move_up", p_event)) {
  75. _tool_selected(TOOL_MOVE_UP);
  76. } else if (ED_IS_SHORTCUT("scene_tree/move_down", p_event)) {
  77. _tool_selected(TOOL_MOVE_DOWN);
  78. } else if (ED_IS_SHORTCUT("scene_tree/reparent", p_event)) {
  79. _tool_selected(TOOL_REPARENT);
  80. } else if (ED_IS_SHORTCUT("scene_tree/merge_from_scene", p_event)) {
  81. _tool_selected(TOOL_MERGE_FROM_SCENE);
  82. } else if (ED_IS_SHORTCUT("scene_tree/save_branch_as_scene", p_event)) {
  83. _tool_selected(TOOL_NEW_SCENE_FROM);
  84. } else if (ED_IS_SHORTCUT("scene_tree/delete_no_confirm", p_event)) {
  85. _tool_selected(TOOL_ERASE, true);
  86. } else if (ED_IS_SHORTCUT("scene_tree/copy_node_path", p_event)) {
  87. _tool_selected(TOOL_COPY_NODE_PATH);
  88. } else if (ED_IS_SHORTCUT("scene_tree/delete", p_event)) {
  89. _tool_selected(TOOL_ERASE);
  90. }
  91. }
  92. void SceneTreeDock::instance(const String &p_file) {
  93. Node *parent = scene_tree->get_selected();
  94. if (!parent || !edited_scene) {
  95. current_option = -1;
  96. accept->get_ok()->set_text(TTR("OK :("));
  97. accept->set_text(TTR("No parent to instance a child at."));
  98. accept->popup_centered_minsize();
  99. return;
  100. };
  101. ERR_FAIL_COND(!parent);
  102. Vector<String> scenes;
  103. scenes.push_back(p_file);
  104. _perform_instance_scenes(scenes, parent, -1);
  105. }
  106. void SceneTreeDock::instance_scenes(const Vector<String> &p_files, Node *p_parent) {
  107. Node *parent = p_parent;
  108. if (!parent) {
  109. parent = scene_tree->get_selected();
  110. }
  111. if (!parent || !edited_scene) {
  112. accept->get_ok()->set_text(TTR("OK"));
  113. accept->set_text(TTR("No parent to instance the scenes at."));
  114. accept->popup_centered_minsize();
  115. return;
  116. };
  117. _perform_instance_scenes(p_files, parent, -1);
  118. }
  119. void SceneTreeDock::_perform_instance_scenes(const Vector<String> &p_files, Node *parent, int p_pos) {
  120. ERR_FAIL_COND(!parent);
  121. Vector<Node *> instances;
  122. bool error = false;
  123. for (int i = 0; i < p_files.size(); i++) {
  124. Ref<PackedScene> sdata = ResourceLoader::load(p_files[i]);
  125. if (!sdata.is_valid()) {
  126. current_option = -1;
  127. accept->get_ok()->set_text(TTR("Ugh"));
  128. accept->set_text(vformat(TTR("Error loading scene from %s"), p_files[i]));
  129. accept->popup_centered_minsize();
  130. error = true;
  131. break;
  132. }
  133. Node *instanced_scene = sdata->instance(PackedScene::GEN_EDIT_STATE_INSTANCE);
  134. if (!instanced_scene) {
  135. current_option = -1;
  136. accept->get_ok()->set_text(TTR("Ugh"));
  137. accept->set_text(vformat(TTR("Error instancing scene from %s"), p_files[i]));
  138. accept->popup_centered_minsize();
  139. error = true;
  140. break;
  141. }
  142. if (edited_scene->get_filename() != "") {
  143. if (_cyclical_dependency_exists(edited_scene->get_filename(), instanced_scene)) {
  144. accept->get_ok()->set_text(TTR("Ok"));
  145. accept->set_text(vformat(TTR("Cannot instance the scene '%s' because the current scene exists within one of its nodes."), p_files[i]));
  146. accept->popup_centered_minsize();
  147. error = true;
  148. break;
  149. }
  150. }
  151. instanced_scene->set_filename(ProjectSettings::get_singleton()->localize_path(p_files[i]));
  152. instances.push_back(instanced_scene);
  153. }
  154. if (error) {
  155. for (int i = 0; i < instances.size(); i++) {
  156. memdelete(instances[i]);
  157. }
  158. return;
  159. }
  160. editor_data->get_undo_redo().create_action(TTR("Instance Scene(s)"));
  161. for (int i = 0; i < instances.size(); i++) {
  162. Node *instanced_scene = instances[i];
  163. editor_data->get_undo_redo().add_do_method(parent, "add_child", instanced_scene);
  164. if (p_pos >= 0) {
  165. editor_data->get_undo_redo().add_do_method(parent, "move_child", instanced_scene, p_pos + i);
  166. }
  167. editor_data->get_undo_redo().add_do_method(instanced_scene, "set_owner", edited_scene);
  168. editor_data->get_undo_redo().add_do_method(editor_selection, "clear");
  169. editor_data->get_undo_redo().add_do_method(editor_selection, "add_node", instanced_scene);
  170. editor_data->get_undo_redo().add_do_reference(instanced_scene);
  171. editor_data->get_undo_redo().add_undo_method(parent, "remove_child", instanced_scene);
  172. String new_name = parent->validate_child_name(instanced_scene);
  173. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  174. editor_data->get_undo_redo().add_do_method(sed, "live_debug_instance_node", edited_scene->get_path_to(parent), p_files[i], new_name);
  175. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_remove_node", NodePath(String(edited_scene->get_path_to(parent)) + "/" + new_name));
  176. }
  177. editor_data->get_undo_redo().commit_action();
  178. }
  179. void SceneTreeDock::_replace_with_branch_scene(const String &p_file, Node *base) {
  180. Ref<PackedScene> sdata = ResourceLoader::load(p_file);
  181. if (!sdata.is_valid()) {
  182. accept->get_ok()->set_text(TTR("Ugh"));
  183. accept->set_text(vformat(TTR("Error loading scene from %s"), p_file));
  184. accept->popup_centered_minsize();
  185. return;
  186. }
  187. Node *instanced_scene = sdata->instance(PackedScene::GEN_EDIT_STATE_INSTANCE);
  188. if (!instanced_scene) {
  189. accept->get_ok()->set_text(TTR("Ugh"));
  190. accept->set_text(vformat(TTR("Error instancing scene from %s"), p_file));
  191. accept->popup_centered_minsize();
  192. return;
  193. }
  194. Node *parent = base->get_parent();
  195. int pos = base->get_index();
  196. memdelete(base);
  197. parent->add_child(instanced_scene);
  198. parent->move_child(instanced_scene, pos);
  199. instanced_scene->set_owner(edited_scene);
  200. editor_selection->clear();
  201. editor_selection->add_node(instanced_scene);
  202. scene_tree->set_selected(instanced_scene);
  203. }
  204. bool SceneTreeDock::_cyclical_dependency_exists(const String &p_target_scene_path, Node *p_desired_node) {
  205. int childCount = p_desired_node->get_child_count();
  206. if (p_desired_node->get_filename() == p_target_scene_path) {
  207. return true;
  208. }
  209. for (int i = 0; i < childCount; i++) {
  210. Node *child = p_desired_node->get_child(i);
  211. if (_cyclical_dependency_exists(p_target_scene_path, child)) {
  212. return true;
  213. }
  214. }
  215. return false;
  216. }
  217. void SceneTreeDock::_tool_selected(int p_tool, bool p_confirm_override) {
  218. current_option = p_tool;
  219. switch (p_tool) {
  220. case TOOL_NEW: {
  221. create_dialog->popup_create(true);
  222. } break;
  223. case TOOL_INSTANCE: {
  224. Node *scene = edited_scene;
  225. if (!scene) {
  226. EditorNode::get_singleton()->new_inherited_scene();
  227. }
  228. file->set_mode(EditorFileDialog::MODE_OPEN_FILE);
  229. List<String> extensions;
  230. ResourceLoader::get_recognized_extensions_for_type("PackedScene", &extensions);
  231. file->clear_filters();
  232. for (int i = 0; i < extensions.size(); i++) {
  233. file->add_filter("*." + extensions[i] + " ; " + extensions[i].to_upper());
  234. }
  235. file->popup_centered_ratio();
  236. } break;
  237. case TOOL_REPLACE: {
  238. create_dialog->popup_create(false);
  239. } break;
  240. case TOOL_ATTACH_SCRIPT: {
  241. Node *selected = scene_tree->get_selected();
  242. if (!selected)
  243. break;
  244. Ref<Script> existing = selected->get_script();
  245. if (existing.is_valid())
  246. editor->push_item(existing.ptr());
  247. else {
  248. String path = selected->get_filename();
  249. script_create_dialog->config(selected->get_class(), path);
  250. script_create_dialog->popup_centered();
  251. }
  252. } break;
  253. case TOOL_CLEAR_SCRIPT: {
  254. Node *selected = scene_tree->get_selected();
  255. if (!selected)
  256. break;
  257. Ref<Script> existing = selected->get_script();
  258. if (existing.is_valid()) {
  259. const RefPtr empty;
  260. selected->set_script(empty);
  261. button_create_script->show();
  262. button_clear_script->hide();
  263. }
  264. } break;
  265. case TOOL_MOVE_UP:
  266. case TOOL_MOVE_DOWN: {
  267. if (!scene_tree->get_selected())
  268. break;
  269. if (scene_tree->get_selected() == edited_scene) {
  270. current_option = -1;
  271. accept->get_ok()->set_text(TTR("I see.."));
  272. accept->set_text(TTR("This operation can't be done on the tree root."));
  273. accept->popup_centered_minsize();
  274. break;
  275. }
  276. if (!_validate_no_foreign())
  277. break;
  278. bool MOVING_DOWN = (p_tool == TOOL_MOVE_DOWN);
  279. bool MOVING_UP = !MOVING_DOWN;
  280. Node *common_parent = scene_tree->get_selected()->get_parent();
  281. List<Node *> selection = editor_selection->get_selected_node_list();
  282. selection.sort_custom<Node::Comparator>(); // sort by index
  283. if (MOVING_DOWN)
  284. selection.invert();
  285. int lowest_id = common_parent->get_child_count() - 1;
  286. int highest_id = 0;
  287. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  288. int index = E->get()->get_index();
  289. if (index > highest_id) highest_id = index;
  290. if (index < lowest_id) lowest_id = index;
  291. if (E->get()->get_parent() != common_parent)
  292. common_parent = NULL;
  293. }
  294. if (!common_parent || (MOVING_DOWN && highest_id >= common_parent->get_child_count() - MOVING_DOWN) || (MOVING_UP && lowest_id == 0))
  295. break; // one or more nodes can not be moved
  296. if (selection.size() == 1) editor_data->get_undo_redo().create_action(TTR("Move Node In Parent"));
  297. if (selection.size() > 1) editor_data->get_undo_redo().create_action(TTR("Move Nodes In Parent"));
  298. for (int i = 0; i < selection.size(); i++) {
  299. Node *top_node = selection[i];
  300. Node *bottom_node = selection[selection.size() - 1 - i];
  301. ERR_FAIL_COND(!top_node->get_parent());
  302. ERR_FAIL_COND(!bottom_node->get_parent());
  303. int bottom_node_pos = bottom_node->get_index();
  304. int top_node_pos_next = top_node->get_index() + (MOVING_DOWN ? 1 : -1);
  305. editor_data->get_undo_redo().add_do_method(top_node->get_parent(), "move_child", top_node, top_node_pos_next);
  306. editor_data->get_undo_redo().add_undo_method(bottom_node->get_parent(), "move_child", bottom_node, bottom_node_pos);
  307. }
  308. editor_data->get_undo_redo().commit_action();
  309. } break;
  310. case TOOL_DUPLICATE: {
  311. if (!edited_scene)
  312. break;
  313. if (editor_selection->is_selected(edited_scene)) {
  314. current_option = -1;
  315. //accept->get_cancel()->hide();
  316. accept->get_ok()->set_text(TTR("I see.."));
  317. accept->set_text(TTR("This operation can't be done on the tree root."));
  318. accept->popup_centered_minsize();
  319. break;
  320. }
  321. if (!_validate_no_foreign())
  322. break;
  323. List<Node *> selection = editor_selection->get_selected_node_list();
  324. if (selection.size() == 0)
  325. break;
  326. editor_data->get_undo_redo().create_action(TTR("Duplicate Node(s)"));
  327. editor_data->get_undo_redo().add_do_method(editor_selection, "clear");
  328. Node *dupsingle = NULL;
  329. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  330. Node *node = E->get();
  331. Node *parent = node->get_parent();
  332. List<Node *> owned;
  333. node->get_owned_by(node->get_owner(), &owned);
  334. Map<Node *, Node *> duplimap;
  335. Node *dup = _duplicate(node, duplimap);
  336. ERR_CONTINUE(!dup);
  337. if (selection.size() == 1)
  338. dupsingle = dup;
  339. dup->set_name(parent->validate_child_name(dup));
  340. editor_data->get_undo_redo().add_do_method(parent, "_add_child_below_node", node, dup);
  341. for (List<Node *>::Element *F = owned.front(); F; F = F->next()) {
  342. if (!duplimap.has(F->get())) {
  343. continue;
  344. }
  345. Node *d = duplimap[F->get()];
  346. editor_data->get_undo_redo().add_do_method(d, "set_owner", node->get_owner());
  347. }
  348. editor_data->get_undo_redo().add_do_method(editor_selection, "add_node", dup);
  349. editor_data->get_undo_redo().add_undo_method(parent, "remove_child", dup);
  350. editor_data->get_undo_redo().add_do_reference(dup);
  351. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  352. editor_data->get_undo_redo().add_do_method(sed, "live_debug_duplicate_node", edited_scene->get_path_to(node), dup->get_name());
  353. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_remove_node", NodePath(String(edited_scene->get_path_to(parent)) + "/" + dup->get_name()));
  354. }
  355. editor_data->get_undo_redo().commit_action();
  356. if (dupsingle)
  357. editor->push_item(dupsingle);
  358. } break;
  359. case TOOL_REPARENT: {
  360. if (!scene_tree->get_selected())
  361. break;
  362. if (editor_selection->is_selected(edited_scene)) {
  363. current_option = -1;
  364. accept->get_ok()->set_text(TTR("I see.."));
  365. accept->set_text(TTR("This operation can't be done on the tree root."));
  366. accept->popup_centered_minsize();
  367. break;
  368. }
  369. if (!_validate_no_foreign())
  370. break;
  371. List<Node *> nodes = editor_selection->get_selected_node_list();
  372. Set<Node *> nodeset;
  373. for (List<Node *>::Element *E = nodes.front(); E; E = E->next()) {
  374. nodeset.insert(E->get());
  375. }
  376. reparent_dialog->popup_centered_ratio();
  377. reparent_dialog->set_current(nodeset);
  378. } break;
  379. case TOOL_MULTI_EDIT: {
  380. Node *root = EditorNode::get_singleton()->get_edited_scene();
  381. if (!root)
  382. break;
  383. Ref<MultiNodeEdit> mne = memnew(MultiNodeEdit);
  384. for (const Map<Node *, Object *>::Element *E = EditorNode::get_singleton()->get_editor_selection()->get_selection().front(); E; E = E->next()) {
  385. mne->add_node(root->get_path_to(E->key()));
  386. }
  387. EditorNode::get_singleton()->push_item(mne.ptr());
  388. } break;
  389. case TOOL_ERASE: {
  390. List<Node *> remove_list = editor_selection->get_selected_node_list();
  391. if (remove_list.empty())
  392. return;
  393. if (!_validate_no_foreign())
  394. break;
  395. if (p_confirm_override) {
  396. _delete_confirm();
  397. // hack, force 2d editor viewport to refresh after deletion
  398. if (CanvasItemEditor *editor = CanvasItemEditor::get_singleton())
  399. editor->get_viewport_control()->update();
  400. } else {
  401. delete_dialog->set_text(TTR("Delete Node(s)?"));
  402. delete_dialog->popup_centered_minsize();
  403. }
  404. } break;
  405. case TOOL_MERGE_FROM_SCENE: {
  406. EditorNode::get_singleton()->merge_from_scene();
  407. } break;
  408. case TOOL_NEW_SCENE_FROM: {
  409. Node *scene = editor_data->get_edited_scene_root();
  410. if (!scene) {
  411. accept->get_ok()->set_text(TTR("I see.."));
  412. accept->set_text(TTR("This operation can't be done without a scene."));
  413. accept->popup_centered_minsize();
  414. break;
  415. }
  416. List<Node *> selection = editor_selection->get_selected_node_list();
  417. if (selection.size() != 1) {
  418. accept->get_ok()->set_text(TTR("I see.."));
  419. accept->set_text(TTR("This operation requires a single selected node."));
  420. accept->popup_centered_minsize();
  421. break;
  422. }
  423. Node *tocopy = selection.front()->get();
  424. if (tocopy == scene) {
  425. accept->get_ok()->set_text(TTR("I see.."));
  426. accept->set_text(TTR("Can not perform with the root node."));
  427. accept->popup_centered_minsize();
  428. break;
  429. }
  430. if (tocopy != editor_data->get_edited_scene_root() && tocopy->get_filename() != "") {
  431. accept->get_ok()->set_text(TTR("I see.."));
  432. accept->set_text(TTR("This operation can't be done on instanced scenes."));
  433. accept->popup_centered_minsize();
  434. break;
  435. }
  436. new_scene_from_dialog->set_mode(EditorFileDialog::MODE_SAVE_FILE);
  437. List<String> extensions;
  438. Ref<PackedScene> sd = memnew(PackedScene);
  439. ResourceSaver::get_recognized_extensions(sd, &extensions);
  440. new_scene_from_dialog->clear_filters();
  441. for (int i = 0; i < extensions.size(); i++) {
  442. new_scene_from_dialog->add_filter("*." + extensions[i] + " ; " + extensions[i].to_upper());
  443. }
  444. String existing;
  445. if (extensions.size()) {
  446. String root_name(tocopy->get_name());
  447. existing = root_name + "." + extensions.front()->get().to_lower();
  448. }
  449. new_scene_from_dialog->set_current_path(existing);
  450. new_scene_from_dialog->popup_centered_ratio();
  451. new_scene_from_dialog->set_title(TTR("Save New Scene As.."));
  452. } break;
  453. case TOOL_COPY_NODE_PATH: {
  454. List<Node *> selection = editor_selection->get_selected_node_list();
  455. if (List<Node *>::Element *e = selection.front()) {
  456. if (Node *node = e->get()) {
  457. Node *root = EditorNode::get_singleton()->get_edited_scene();
  458. NodePath path = root->get_path().rel_path_to(node->get_path());
  459. OS::get_singleton()->set_clipboard(path);
  460. }
  461. }
  462. } break;
  463. case TOOL_SCENE_EDITABLE_CHILDREN: {
  464. List<Node *> selection = editor_selection->get_selected_node_list();
  465. if (List<Node *>::Element *e = selection.front()) {
  466. if (Node *node = e->get()) {
  467. bool editable = EditorNode::get_singleton()->get_edited_scene()->is_editable_instance(node);
  468. int editable_item_idx = menu->get_item_idx_from_text(TTR("Editable Children"));
  469. int placeholder_item_idx = menu->get_item_idx_from_text(TTR("Load As Placeholder"));
  470. editable = !editable;
  471. EditorNode::get_singleton()->get_edited_scene()->set_editable_instance(node, editable);
  472. menu->set_item_checked(editable_item_idx, editable);
  473. if (editable) {
  474. node->set_scene_instance_load_placeholder(false);
  475. menu->set_item_checked(placeholder_item_idx, false);
  476. }
  477. scene_tree->update_tree();
  478. }
  479. }
  480. } break;
  481. case TOOL_SCENE_USE_PLACEHOLDER: {
  482. List<Node *> selection = editor_selection->get_selected_node_list();
  483. if (List<Node *>::Element *e = selection.front()) {
  484. if (Node *node = e->get()) {
  485. bool placeholder = node->get_scene_instance_load_placeholder();
  486. placeholder = !placeholder;
  487. int editable_item_idx = menu->get_item_idx_from_text(TTR("Editable Children"));
  488. int placeholder_item_idx = menu->get_item_idx_from_text(TTR("Load As Placeholder"));
  489. if (placeholder)
  490. EditorNode::get_singleton()->get_edited_scene()->set_editable_instance(node, false);
  491. node->set_scene_instance_load_placeholder(placeholder);
  492. menu->set_item_checked(editable_item_idx, false);
  493. menu->set_item_checked(placeholder_item_idx, placeholder);
  494. scene_tree->update_tree();
  495. }
  496. }
  497. } break;
  498. case TOOL_SCENE_CLEAR_INSTANCING: {
  499. List<Node *> selection = editor_selection->get_selected_node_list();
  500. if (List<Node *>::Element *e = selection.front()) {
  501. if (Node *node = e->get()) {
  502. Node *root = EditorNode::get_singleton()->get_edited_scene();
  503. UndoRedo *undo_redo = &editor_data->get_undo_redo();
  504. if (!root)
  505. break;
  506. ERR_FAIL_COND(node->get_filename() == String());
  507. undo_redo->create_action(TTR("Discard Instancing"));
  508. undo_redo->add_do_method(node, "set_filename", "");
  509. undo_redo->add_undo_method(node, "set_filename", node->get_filename());
  510. _node_replace_owner(node, node, root);
  511. undo_redo->add_do_method(scene_tree, "update_tree");
  512. undo_redo->add_undo_method(scene_tree, "update_tree");
  513. undo_redo->commit_action();
  514. }
  515. }
  516. } break;
  517. case TOOL_SCENE_OPEN: {
  518. List<Node *> selection = editor_selection->get_selected_node_list();
  519. if (List<Node *>::Element *e = selection.front()) {
  520. if (Node *node = e->get()) {
  521. scene_tree->emit_signal("open", node->get_filename());
  522. }
  523. }
  524. } break;
  525. case TOOL_SCENE_CLEAR_INHERITANCE: {
  526. clear_inherit_confirm->popup_centered_minsize();
  527. } break;
  528. case TOOL_SCENE_CLEAR_INHERITANCE_CONFIRM: {
  529. List<Node *> selection = editor_selection->get_selected_node_list();
  530. if (List<Node *>::Element *e = selection.front()) {
  531. if (Node *node = e->get()) {
  532. node->set_scene_inherited_state(Ref<SceneState>());
  533. scene_tree->update_tree();
  534. EditorNode::get_singleton()->get_property_editor()->update_tree();
  535. }
  536. }
  537. } break;
  538. case TOOL_SCENE_OPEN_INHERITED: {
  539. List<Node *> selection = editor_selection->get_selected_node_list();
  540. if (List<Node *>::Element *e = selection.front()) {
  541. if (Node *node = e->get()) {
  542. if (node && node->get_scene_inherited_state().is_valid()) {
  543. scene_tree->emit_signal("open", node->get_scene_inherited_state()->get_path());
  544. }
  545. }
  546. }
  547. } break;
  548. default: {
  549. if (p_tool >= EDIT_SUBRESOURCE_BASE) {
  550. int idx = p_tool - EDIT_SUBRESOURCE_BASE;
  551. ERR_FAIL_INDEX(idx, subresources.size());
  552. Object *obj = ObjectDB::get_instance(subresources[idx]);
  553. ERR_FAIL_COND(!obj);
  554. editor->push_item(obj);
  555. }
  556. }
  557. }
  558. }
  559. void SceneTreeDock::_notification(int p_what) {
  560. switch (p_what) {
  561. case NOTIFICATION_READY: {
  562. if (!first_enter)
  563. break;
  564. first_enter = false;
  565. CanvasItemEditorPlugin *canvas_item_plugin = Object::cast_to<CanvasItemEditorPlugin>(editor_data->get_editor("2D"));
  566. if (canvas_item_plugin) {
  567. canvas_item_plugin->get_canvas_item_editor()->connect("item_lock_status_changed", scene_tree, "_update_tree");
  568. canvas_item_plugin->get_canvas_item_editor()->connect("item_group_status_changed", scene_tree, "_update_tree");
  569. scene_tree->connect("node_changed", canvas_item_plugin->get_canvas_item_editor()->get_viewport_control(), "update");
  570. }
  571. button_add->set_icon(get_icon("Add", "EditorIcons"));
  572. button_instance->set_icon(get_icon("Instance", "EditorIcons"));
  573. button_create_script->set_icon(get_icon("ScriptCreate", "EditorIcons"));
  574. button_clear_script->set_icon(get_icon("ScriptRemove", "EditorIcons"));
  575. filter->add_icon_override("right_icon", get_icon("Search", "EditorIcons"));
  576. EditorNode::get_singleton()->get_editor_selection()->connect("selection_changed", this, "_selection_changed");
  577. } break;
  578. case NOTIFICATION_ENTER_TREE: {
  579. clear_inherit_confirm->connect("confirmed", this, "_tool_selected", varray(TOOL_SCENE_CLEAR_INHERITANCE_CONFIRM));
  580. } break;
  581. case NOTIFICATION_EXIT_TREE: {
  582. clear_inherit_confirm->disconnect("confirmed", this, "_tool_selected");
  583. } break;
  584. case EditorSettings::NOTIFICATION_EDITOR_SETTINGS_CHANGED: {
  585. button_add->set_icon(get_icon("Add", "EditorIcons"));
  586. button_instance->set_icon(get_icon("Instance", "EditorIcons"));
  587. button_create_script->set_icon(get_icon("ScriptCreate", "EditorIcons"));
  588. button_clear_script->set_icon(get_icon("ScriptRemove", "EditorIcons"));
  589. filter->add_icon_override("right_icon", get_icon("Search", "EditorIcons"));
  590. } break;
  591. }
  592. }
  593. void SceneTreeDock::_node_replace_owner(Node *p_base, Node *p_node, Node *p_root) {
  594. if (p_base != p_node) {
  595. if (p_node->get_owner() == p_base) {
  596. UndoRedo *undo_redo = &editor_data->get_undo_redo();
  597. undo_redo->add_do_method(p_node, "set_owner", p_root);
  598. undo_redo->add_undo_method(p_node, "set_owner", p_base);
  599. }
  600. }
  601. for (int i = 0; i < p_node->get_child_count(); i++) {
  602. _node_replace_owner(p_base, p_node->get_child(i), p_root);
  603. }
  604. }
  605. void SceneTreeDock::_load_request(const String &p_path) {
  606. editor->open_request(p_path);
  607. }
  608. void SceneTreeDock::_script_open_request(const Ref<Script> &p_script) {
  609. editor->edit_resource(p_script);
  610. }
  611. void SceneTreeDock::_node_selected() {
  612. Node *node = scene_tree->get_selected();
  613. if (!node) {
  614. editor->push_item(NULL);
  615. return;
  616. }
  617. if (ScriptEditor::get_singleton()->is_visible_in_tree()) {
  618. restore_script_editor_on_drag = true;
  619. }
  620. editor->push_item(node);
  621. }
  622. void SceneTreeDock::_node_renamed() {
  623. _node_selected();
  624. }
  625. Node *SceneTreeDock::_duplicate(Node *p_node, Map<Node *, Node *> &duplimap) {
  626. Node *node = NULL;
  627. if (p_node->get_filename() != "") { //an instance
  628. Ref<PackedScene> sd = ResourceLoader::load(p_node->get_filename());
  629. ERR_FAIL_COND_V(!sd.is_valid(), NULL);
  630. node = sd->instance(PackedScene::GEN_EDIT_STATE_INSTANCE);
  631. ERR_FAIL_COND_V(!node, NULL);
  632. node->set_scene_instance_load_placeholder(p_node->get_scene_instance_load_placeholder());
  633. } else {
  634. Object *obj = ClassDB::instance(p_node->get_class());
  635. ERR_FAIL_COND_V(!obj, NULL);
  636. node = Object::cast_to<Node>(obj);
  637. if (!node)
  638. memdelete(obj);
  639. ERR_FAIL_COND_V(!node, NULL);
  640. }
  641. List<PropertyInfo> plist;
  642. p_node->get_property_list(&plist);
  643. for (List<PropertyInfo>::Element *E = plist.front(); E; E = E->next()) {
  644. if (!(E->get().usage & PROPERTY_USAGE_STORAGE))
  645. continue;
  646. String name = E->get().name;
  647. node->set(name, p_node->get(name));
  648. }
  649. List<Node::GroupInfo> group_info;
  650. p_node->get_groups(&group_info);
  651. for (List<Node::GroupInfo>::Element *E = group_info.front(); E; E = E->next()) {
  652. if (E->get().persistent)
  653. node->add_to_group(E->get().name, true);
  654. }
  655. node->set_name(p_node->get_name());
  656. duplimap[p_node] = node;
  657. for (int i = 0; i < p_node->get_child_count(); i++) {
  658. Node *child = p_node->get_child(i);
  659. if (p_node->get_owner() != child->get_owner())
  660. continue; //don't bother with not in-scene nodes.
  661. Node *dup = _duplicate(child, duplimap);
  662. if (!dup) {
  663. memdelete(node);
  664. return NULL;
  665. }
  666. node->add_child(dup);
  667. }
  668. return node;
  669. }
  670. void SceneTreeDock::_set_owners(Node *p_owner, const Array &p_nodes) {
  671. for (int i = 0; i < p_nodes.size(); i++) {
  672. Node *n = Object::cast_to<Node>(p_nodes[i]);
  673. if (!n)
  674. continue;
  675. n->set_owner(p_owner);
  676. }
  677. }
  678. void SceneTreeDock::_fill_path_renames(Vector<StringName> base_path, Vector<StringName> new_base_path, Node *p_node, List<Pair<NodePath, NodePath> > *p_renames) {
  679. base_path.push_back(p_node->get_name());
  680. if (new_base_path.size())
  681. new_base_path.push_back(p_node->get_name());
  682. NodePath from(base_path, true);
  683. NodePath to;
  684. if (new_base_path.size())
  685. to = NodePath(new_base_path, true);
  686. Pair<NodePath, NodePath> npp;
  687. npp.first = from;
  688. npp.second = to;
  689. p_renames->push_back(npp);
  690. for (int i = 0; i < p_node->get_child_count(); i++) {
  691. _fill_path_renames(base_path, new_base_path, p_node->get_child(i), p_renames);
  692. }
  693. }
  694. void SceneTreeDock::fill_path_renames(Node *p_node, Node *p_new_parent, List<Pair<NodePath, NodePath> > *p_renames) {
  695. if (!bool(EDITOR_DEF("editors/animation/autorename_animation_tracks", true)))
  696. return;
  697. Vector<StringName> base_path;
  698. Node *n = p_node->get_parent();
  699. while (n) {
  700. base_path.push_back(n->get_name());
  701. n = n->get_parent();
  702. }
  703. base_path.invert();
  704. Vector<StringName> new_base_path;
  705. if (p_new_parent) {
  706. n = p_new_parent;
  707. while (n) {
  708. new_base_path.push_back(n->get_name());
  709. n = n->get_parent();
  710. }
  711. new_base_path.invert();
  712. }
  713. _fill_path_renames(base_path, new_base_path, p_node, p_renames);
  714. }
  715. void SceneTreeDock::perform_node_renames(Node *p_base, List<Pair<NodePath, NodePath> > *p_renames, Map<Ref<Animation>, Set<int> > *r_rem_anims) {
  716. Map<Ref<Animation>, Set<int> > rem_anims;
  717. if (!r_rem_anims)
  718. r_rem_anims = &rem_anims;
  719. if (!bool(EDITOR_DEF("editors/animation/autorename_animation_tracks", true)))
  720. return;
  721. if (!p_base) {
  722. p_base = edited_scene;
  723. }
  724. if (!p_base)
  725. return;
  726. if (Object::cast_to<AnimationPlayer>(p_base)) {
  727. AnimationPlayer *ap = Object::cast_to<AnimationPlayer>(p_base);
  728. List<StringName> anims;
  729. ap->get_animation_list(&anims);
  730. Node *root = ap->get_node(ap->get_root());
  731. if (root) {
  732. NodePath root_path = root->get_path();
  733. NodePath new_root_path = root_path;
  734. for (List<Pair<NodePath, NodePath> >::Element *E = p_renames->front(); E; E = E->next()) {
  735. if (E->get().first == root_path) {
  736. new_root_path = E->get().second;
  737. break;
  738. }
  739. }
  740. if (new_root_path != NodePath()) {
  741. //will not be erased
  742. for (List<StringName>::Element *E = anims.front(); E; E = E->next()) {
  743. Ref<Animation> anim = ap->get_animation(E->get());
  744. if (!r_rem_anims->has(anim)) {
  745. r_rem_anims->insert(anim, Set<int>());
  746. Set<int> &ran = r_rem_anims->find(anim)->get();
  747. for (int i = 0; i < anim->get_track_count(); i++)
  748. ran.insert(i);
  749. }
  750. Set<int> &ran = r_rem_anims->find(anim)->get();
  751. if (anim.is_null())
  752. continue;
  753. for (int i = 0; i < anim->get_track_count(); i++) {
  754. NodePath track_np = anim->track_get_path(i);
  755. Node *n = root->get_node(track_np);
  756. if (!n) {
  757. continue;
  758. }
  759. NodePath old_np = n->get_path();
  760. if (!ran.has(i))
  761. continue; //channel was removed
  762. for (List<Pair<NodePath, NodePath> >::Element *E = p_renames->front(); E; E = E->next()) {
  763. if (E->get().first == old_np) {
  764. if (E->get().second == NodePath()) {
  765. //will be erased
  766. int idx = 0;
  767. Set<int>::Element *EI = ran.front();
  768. ERR_FAIL_COND(!EI); //bug
  769. while (EI->get() != i) {
  770. idx++;
  771. EI = EI->next();
  772. ERR_FAIL_COND(!EI); //another bug
  773. }
  774. editor_data->get_undo_redo().add_do_method(anim.ptr(), "remove_track", idx);
  775. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "add_track", anim->track_get_type(i), idx);
  776. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "track_set_path", idx, track_np);
  777. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "track_set_interpolation_type", idx, anim->track_get_interpolation_type(i));
  778. for (int j = 0; j < anim->track_get_key_count(i); j++) {
  779. 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));
  780. }
  781. ran.erase(i); //byebye channel
  782. } else {
  783. //will be renamed
  784. NodePath rel_path = new_root_path.rel_path_to(E->get().second);
  785. NodePath new_path = NodePath(rel_path.get_names(), track_np.get_subnames(), false, track_np.get_property());
  786. if (new_path == track_np)
  787. continue; //bleh
  788. editor_data->get_undo_redo().add_do_method(anim.ptr(), "track_set_path", i, new_path);
  789. editor_data->get_undo_redo().add_undo_method(anim.ptr(), "track_set_path", i, track_np);
  790. }
  791. }
  792. }
  793. }
  794. }
  795. }
  796. }
  797. }
  798. for (int i = 0; i < p_base->get_child_count(); i++)
  799. perform_node_renames(p_base->get_child(i), p_renames, r_rem_anims);
  800. }
  801. void SceneTreeDock::_node_prerenamed(Node *p_node, const String &p_new_name) {
  802. List<Pair<NodePath, NodePath> > path_renames;
  803. Vector<StringName> base_path;
  804. Node *n = p_node->get_parent();
  805. while (n) {
  806. base_path.push_back(n->get_name());
  807. n = n->get_parent();
  808. }
  809. base_path.invert();
  810. Vector<StringName> new_base_path = base_path;
  811. base_path.push_back(p_node->get_name());
  812. new_base_path.push_back(p_new_name);
  813. Pair<NodePath, NodePath> npp;
  814. npp.first = NodePath(base_path, true);
  815. npp.second = NodePath(new_base_path, true);
  816. path_renames.push_back(npp);
  817. for (int i = 0; i < p_node->get_child_count(); i++)
  818. _fill_path_renames(base_path, new_base_path, p_node->get_child(i), &path_renames);
  819. perform_node_renames(NULL, &path_renames);
  820. }
  821. bool SceneTreeDock::_validate_no_foreign() {
  822. List<Node *> selection = editor_selection->get_selected_node_list();
  823. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  824. if (E->get() != edited_scene && E->get()->get_owner() != edited_scene) {
  825. accept->get_ok()->set_text(TTR("Makes Sense!"));
  826. accept->set_text(TTR("Can't operate on nodes from a foreign scene!"));
  827. accept->popup_centered_minsize();
  828. return false;
  829. }
  830. 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) {
  831. accept->get_ok()->set_text(TTR("Makes Sense!"));
  832. accept->set_text(TTR("Can't operate on nodes the current scene inherits from!"));
  833. accept->popup_centered_minsize();
  834. return false;
  835. }
  836. }
  837. return true;
  838. }
  839. void SceneTreeDock::_node_reparent(NodePath p_path, bool p_keep_global_xform) {
  840. Node *new_parent = scene_root->get_node(p_path);
  841. ERR_FAIL_COND(!new_parent);
  842. List<Node *> selection = editor_selection->get_selected_node_list();
  843. if (selection.empty())
  844. return; //nothing to reparent
  845. Vector<Node *> nodes;
  846. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  847. nodes.push_back(E->get());
  848. }
  849. _do_reparent(new_parent, -1, nodes, p_keep_global_xform);
  850. }
  851. void SceneTreeDock::_do_reparent(Node *p_new_parent, int p_position_in_parent, Vector<Node *> p_nodes, bool p_keep_global_xform) {
  852. Node *new_parent = p_new_parent;
  853. ERR_FAIL_COND(!new_parent);
  854. Node *validate = new_parent;
  855. while (validate) {
  856. if (p_nodes.find(validate) != -1) {
  857. ERR_EXPLAIN("Selection changed at some point.. can't reparent");
  858. ERR_FAIL();
  859. return;
  860. }
  861. validate = validate->get_parent();
  862. }
  863. //ok all valid
  864. if (p_nodes.size() == 0)
  865. return; //nothing to reparent
  866. //sort by tree order, so re-adding is easy
  867. p_nodes.sort_custom<Node::Comparator>();
  868. editor_data->get_undo_redo().create_action(TTR("Reparent Node"));
  869. List<Pair<NodePath, NodePath> > path_renames;
  870. Vector<StringName> former_names;
  871. int inc = 0;
  872. for (int ni = 0; ni < p_nodes.size(); ni++) {
  873. //no undo for now, sorry
  874. Node *node = p_nodes[ni];
  875. fill_path_renames(node, new_parent, &path_renames);
  876. former_names.push_back(node->get_name());
  877. List<Node *> owned;
  878. node->get_owned_by(node->get_owner(), &owned);
  879. Array owners;
  880. for (List<Node *>::Element *E = owned.front(); E; E = E->next()) {
  881. owners.push_back(E->get());
  882. }
  883. if (new_parent == node->get_parent() && node->get_index() < p_position_in_parent + ni) {
  884. //if child will generate a gap when moved, adjust
  885. inc--;
  886. }
  887. editor_data->get_undo_redo().add_do_method(node->get_parent(), "remove_child", node);
  888. editor_data->get_undo_redo().add_do_method(new_parent, "add_child", node);
  889. if (p_position_in_parent >= 0)
  890. editor_data->get_undo_redo().add_do_method(new_parent, "move_child", node, p_position_in_parent + inc);
  891. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  892. String new_name = new_parent->validate_child_name(node);
  893. 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, -1);
  894. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_reparent_node", NodePath(String(edited_scene->get_path_to(new_parent)) + "/" + new_name), edited_scene->get_path_to(node->get_parent()), node->get_name(), node->get_index());
  895. if (p_keep_global_xform) {
  896. if (Object::cast_to<Node2D>(node))
  897. editor_data->get_undo_redo().add_do_method(node, "set_global_transform", Object::cast_to<Node2D>(node)->get_global_transform());
  898. if (Object::cast_to<Spatial>(node))
  899. editor_data->get_undo_redo().add_do_method(node, "set_global_transform", Object::cast_to<Spatial>(node)->get_global_transform());
  900. if (Object::cast_to<Control>(node))
  901. editor_data->get_undo_redo().add_do_method(node, "set_global_position", Object::cast_to<Control>(node)->get_global_position());
  902. }
  903. editor_data->get_undo_redo().add_do_method(this, "_set_owners", edited_scene, owners);
  904. if (AnimationPlayerEditor::singleton->get_key_editor()->get_root() == node)
  905. editor_data->get_undo_redo().add_do_method(AnimationPlayerEditor::singleton->get_key_editor(), "set_root", node);
  906. editor_data->get_undo_redo().add_undo_method(new_parent, "remove_child", node);
  907. editor_data->get_undo_redo().add_undo_method(node, "set_name", former_names[ni]);
  908. inc++;
  909. }
  910. //add and move in a second step.. (so old order is preserved)
  911. for (int ni = 0; ni < p_nodes.size(); ni++) {
  912. Node *node = p_nodes[ni];
  913. List<Node *> owned;
  914. node->get_owned_by(node->get_owner(), &owned);
  915. Array owners;
  916. for (List<Node *>::Element *E = owned.front(); E; E = E->next()) {
  917. owners.push_back(E->get());
  918. }
  919. int child_pos = node->get_position_in_parent();
  920. editor_data->get_undo_redo().add_undo_method(node->get_parent(), "add_child", node);
  921. editor_data->get_undo_redo().add_undo_method(node->get_parent(), "move_child", node, child_pos);
  922. editor_data->get_undo_redo().add_undo_method(this, "_set_owners", edited_scene, owners);
  923. if (AnimationPlayerEditor::singleton->get_key_editor()->get_root() == node)
  924. editor_data->get_undo_redo().add_undo_method(AnimationPlayerEditor::singleton->get_key_editor(), "set_root", node);
  925. if (p_keep_global_xform) {
  926. if (Object::cast_to<Node2D>(node))
  927. editor_data->get_undo_redo().add_undo_method(node, "set_transform", Object::cast_to<Node2D>(node)->get_transform());
  928. if (Object::cast_to<Spatial>(node))
  929. editor_data->get_undo_redo().add_undo_method(node, "set_transform", Object::cast_to<Spatial>(node)->get_transform());
  930. if (Object::cast_to<Control>(node))
  931. editor_data->get_undo_redo().add_undo_method(node, "set_position", Object::cast_to<Control>(node)->get_position());
  932. }
  933. }
  934. perform_node_renames(NULL, &path_renames);
  935. editor_data->get_undo_redo().commit_action();
  936. }
  937. void SceneTreeDock::_script_created(Ref<Script> p_script) {
  938. Node *selected = scene_tree->get_selected();
  939. if (!selected)
  940. return;
  941. selected->set_script(p_script.get_ref_ptr());
  942. editor->push_item(p_script.operator->());
  943. button_create_script->hide();
  944. button_clear_script->show();
  945. }
  946. void SceneTreeDock::_delete_confirm() {
  947. List<Node *> remove_list = editor_selection->get_selected_node_list();
  948. if (remove_list.empty())
  949. return;
  950. editor->get_editor_plugins_over()->make_visible(false);
  951. editor_data->get_undo_redo().create_action(TTR("Remove Node(s)"));
  952. bool entire_scene = false;
  953. for (List<Node *>::Element *E = remove_list.front(); E; E = E->next()) {
  954. if (E->get() == edited_scene) {
  955. entire_scene = true;
  956. }
  957. }
  958. if (entire_scene) {
  959. editor_data->get_undo_redo().add_do_method(editor, "set_edited_scene", (Object *)NULL);
  960. editor_data->get_undo_redo().add_undo_method(editor, "set_edited_scene", edited_scene);
  961. editor_data->get_undo_redo().add_undo_method(edited_scene, "set_owner", edited_scene->get_owner());
  962. editor_data->get_undo_redo().add_undo_reference(edited_scene);
  963. } else {
  964. remove_list.sort_custom<Node::Comparator>(); //sort nodes to keep positions
  965. List<Pair<NodePath, NodePath> > path_renames;
  966. //delete from animation
  967. for (List<Node *>::Element *E = remove_list.front(); E; E = E->next()) {
  968. Node *n = E->get();
  969. if (!n->is_inside_tree() || !n->get_parent())
  970. continue;
  971. fill_path_renames(n, NULL, &path_renames);
  972. }
  973. perform_node_renames(NULL, &path_renames);
  974. //delete for read
  975. for (List<Node *>::Element *E = remove_list.front(); E; E = E->next()) {
  976. Node *n = E->get();
  977. if (!n->is_inside_tree() || !n->get_parent())
  978. continue;
  979. List<Node *> owned;
  980. n->get_owned_by(n->get_owner(), &owned);
  981. Array owners;
  982. for (List<Node *>::Element *E = owned.front(); E; E = E->next()) {
  983. owners.push_back(E->get());
  984. }
  985. editor_data->get_undo_redo().add_do_method(n->get_parent(), "remove_child", n);
  986. editor_data->get_undo_redo().add_undo_method(n->get_parent(), "add_child", n);
  987. editor_data->get_undo_redo().add_undo_method(n->get_parent(), "move_child", n, n->get_index());
  988. if (AnimationPlayerEditor::singleton->get_key_editor()->get_root() == n)
  989. editor_data->get_undo_redo().add_undo_method(AnimationPlayerEditor::singleton->get_key_editor(), "set_root", n);
  990. editor_data->get_undo_redo().add_undo_method(this, "_set_owners", edited_scene, owners);
  991. editor_data->get_undo_redo().add_undo_reference(n);
  992. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  993. 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());
  994. 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());
  995. }
  996. }
  997. editor_data->get_undo_redo().commit_action();
  998. }
  999. void SceneTreeDock::_selection_changed() {
  1000. int selection_size = EditorNode::get_singleton()->get_editor_selection()->get_selection().size();
  1001. if (selection_size > 1) {
  1002. //automatically turn on multi-edit
  1003. _tool_selected(TOOL_MULTI_EDIT);
  1004. }
  1005. if (selection_size == 1) {
  1006. if (EditorNode::get_singleton()->get_editor_selection()->get_selection().front()->key()->get_script().is_null()) {
  1007. button_create_script->show();
  1008. button_clear_script->hide();
  1009. } else {
  1010. button_create_script->hide();
  1011. button_clear_script->show();
  1012. }
  1013. } else {
  1014. button_create_script->hide();
  1015. button_clear_script->hide();
  1016. }
  1017. }
  1018. void SceneTreeDock::_create() {
  1019. if (current_option == TOOL_NEW) {
  1020. Node *parent = NULL;
  1021. if (edited_scene) {
  1022. // If root exists in edited scene
  1023. parent = scene_tree->get_selected();
  1024. if (!parent)
  1025. parent = edited_scene;
  1026. } else {
  1027. // If no root exist in edited scene
  1028. parent = scene_root;
  1029. ERR_FAIL_COND(!parent);
  1030. }
  1031. Object *c = create_dialog->instance_selected();
  1032. ERR_FAIL_COND(!c);
  1033. Node *child = Object::cast_to<Node>(c);
  1034. ERR_FAIL_COND(!child);
  1035. editor_data->get_undo_redo().create_action(TTR("Create Node"));
  1036. if (edited_scene) {
  1037. editor_data->get_undo_redo().add_do_method(parent, "add_child", child);
  1038. editor_data->get_undo_redo().add_do_method(child, "set_owner", edited_scene);
  1039. editor_data->get_undo_redo().add_do_method(editor_selection, "clear");
  1040. editor_data->get_undo_redo().add_do_method(editor_selection, "add_node", child);
  1041. editor_data->get_undo_redo().add_do_reference(child);
  1042. editor_data->get_undo_redo().add_undo_method(parent, "remove_child", child);
  1043. String new_name = parent->validate_child_name(child);
  1044. ScriptEditorDebugger *sed = ScriptEditor::get_singleton()->get_debugger();
  1045. editor_data->get_undo_redo().add_do_method(sed, "live_debug_create_node", edited_scene->get_path_to(parent), child->get_class(), new_name);
  1046. editor_data->get_undo_redo().add_undo_method(sed, "live_debug_remove_node", NodePath(String(edited_scene->get_path_to(parent)) + "/" + new_name));
  1047. } else {
  1048. editor_data->get_undo_redo().add_do_method(editor, "set_edited_scene", child);
  1049. editor_data->get_undo_redo().add_do_reference(child);
  1050. editor_data->get_undo_redo().add_undo_method(editor, "set_edited_scene", (Object *)NULL);
  1051. }
  1052. editor_data->get_undo_redo().commit_action();
  1053. editor->push_item(c);
  1054. if (Object::cast_to<Control>(c)) {
  1055. //make editor more comfortable, so some controls don't appear super shrunk
  1056. Control *ct = Object::cast_to<Control>(c);
  1057. Size2 ms = ct->get_minimum_size();
  1058. if (ms.width < 4)
  1059. ms.width = 40;
  1060. if (ms.height < 4)
  1061. ms.height = 40;
  1062. ct->set_size(ms);
  1063. }
  1064. } else if (current_option == TOOL_REPLACE) {
  1065. Node *n = scene_tree->get_selected();
  1066. ERR_FAIL_COND(!n);
  1067. Object *c = create_dialog->instance_selected();
  1068. ERR_FAIL_COND(!c);
  1069. Node *newnode = Object::cast_to<Node>(c);
  1070. ERR_FAIL_COND(!newnode);
  1071. List<PropertyInfo> pinfo;
  1072. n->get_property_list(&pinfo);
  1073. for (List<PropertyInfo>::Element *E = pinfo.front(); E; E = E->next()) {
  1074. if (!(E->get().usage & PROPERTY_USAGE_STORAGE))
  1075. continue;
  1076. newnode->set(E->get().name, n->get(E->get().name));
  1077. }
  1078. editor->push_item(NULL);
  1079. //reconnect signals
  1080. List<MethodInfo> sl;
  1081. n->get_signal_list(&sl);
  1082. for (List<MethodInfo>::Element *E = sl.front(); E; E = E->next()) {
  1083. List<Object::Connection> cl;
  1084. n->get_signal_connection_list(E->get().name, &cl);
  1085. for (List<Object::Connection>::Element *F = cl.front(); F; F = F->next()) {
  1086. Object::Connection &c = F->get();
  1087. if (!(c.flags & Object::CONNECT_PERSIST))
  1088. continue;
  1089. newnode->connect(c.signal, c.target, c.method, varray(), Object::CONNECT_PERSIST);
  1090. }
  1091. }
  1092. String newname = n->get_name();
  1093. List<Node *> to_erase;
  1094. for (int i = 0; i < n->get_child_count(); i++) {
  1095. if (n->get_child(i)->get_owner() == NULL && n->is_owned_by_parent()) {
  1096. to_erase.push_back(n->get_child(i));
  1097. }
  1098. }
  1099. n->replace_by(newnode, true);
  1100. if (n == edited_scene) {
  1101. edited_scene = newnode;
  1102. editor->set_edited_scene(newnode);
  1103. newnode->set_editable_instances(n->get_editable_instances());
  1104. }
  1105. //small hack to make collisionshapes and other kind of nodes to work
  1106. for (int i = 0; i < newnode->get_child_count(); i++) {
  1107. Node *c = newnode->get_child(i);
  1108. c->call("set_transform", c->call("get_transform"));
  1109. }
  1110. editor_data->get_undo_redo().clear_history();
  1111. newnode->set_name(newname);
  1112. editor->push_item(newnode);
  1113. memdelete(n);
  1114. while (to_erase.front()) {
  1115. memdelete(to_erase.front()->get());
  1116. to_erase.pop_front();
  1117. }
  1118. }
  1119. }
  1120. void SceneTreeDock::set_edited_scene(Node *p_scene) {
  1121. edited_scene = p_scene;
  1122. }
  1123. void SceneTreeDock::set_selected(Node *p_node, bool p_emit_selected) {
  1124. scene_tree->set_selected(p_node, p_emit_selected);
  1125. }
  1126. void SceneTreeDock::import_subscene() {
  1127. import_subscene_dialog->popup_centered_ratio();
  1128. }
  1129. void SceneTreeDock::_import_subscene() {
  1130. Node *parent = scene_tree->get_selected();
  1131. if (!parent) {
  1132. parent = editor_data->get_edited_scene_root();
  1133. ERR_FAIL_COND(!parent);
  1134. }
  1135. import_subscene_dialog->move(parent, edited_scene);
  1136. editor_data->get_undo_redo().clear_history(); //no undo for now..
  1137. }
  1138. void SceneTreeDock::_new_scene_from(String p_file) {
  1139. List<Node *> selection = editor_selection->get_selected_node_list();
  1140. if (selection.size() != 1) {
  1141. accept->get_ok()->set_text(TTR("I see.."));
  1142. accept->set_text(TTR("This operation requires a single selected node."));
  1143. accept->popup_centered_minsize();
  1144. return;
  1145. }
  1146. Node *base = selection.front()->get();
  1147. Map<Node *, Node *> reown;
  1148. reown[editor_data->get_edited_scene_root()] = base;
  1149. Node *copy = base->duplicate_and_reown(reown);
  1150. if (copy) {
  1151. Ref<PackedScene> sdata = memnew(PackedScene);
  1152. Error err = sdata->pack(copy);
  1153. memdelete(copy);
  1154. if (err != OK) {
  1155. accept->get_ok()->set_text(TTR("I see.."));
  1156. accept->set_text(TTR("Couldn't save new scene. Likely dependencies (instances) couldn't be satisfied."));
  1157. accept->popup_centered_minsize();
  1158. return;
  1159. }
  1160. int flg = 0;
  1161. if (EditorSettings::get_singleton()->get("filesystem/on_save/compress_binary_resources"))
  1162. flg |= ResourceSaver::FLAG_COMPRESS;
  1163. err = ResourceSaver::save(p_file, sdata, flg);
  1164. if (err != OK) {
  1165. accept->get_ok()->set_text(TTR("I see.."));
  1166. accept->set_text(TTR("Error saving scene."));
  1167. accept->popup_centered_minsize();
  1168. return;
  1169. }
  1170. _replace_with_branch_scene(p_file, base);
  1171. } else {
  1172. accept->get_ok()->set_text(TTR("I see.."));
  1173. accept->set_text(TTR("Error duplicating scene to save it."));
  1174. accept->popup_centered_minsize();
  1175. return;
  1176. }
  1177. }
  1178. static bool _is_node_visible(Node *p_node) {
  1179. if (!p_node->get_owner())
  1180. return false;
  1181. if (p_node->get_owner() != EditorNode::get_singleton()->get_edited_scene() && !EditorNode::get_singleton()->get_edited_scene()->is_editable_instance(p_node->get_owner()))
  1182. return false;
  1183. return true;
  1184. }
  1185. static bool _has_visible_children(Node *p_node) {
  1186. bool collapsed = p_node->is_displayed_folded();
  1187. if (collapsed)
  1188. return false;
  1189. for (int i = 0; i < p_node->get_child_count(); i++) {
  1190. Node *child = p_node->get_child(i);
  1191. if (!_is_node_visible(child))
  1192. continue;
  1193. return true;
  1194. }
  1195. return false;
  1196. }
  1197. static Node *_find_last_visible(Node *p_node) {
  1198. Node *last = NULL;
  1199. bool collapsed = p_node->is_displayed_folded();
  1200. if (!collapsed) {
  1201. for (int i = 0; i < p_node->get_child_count(); i++) {
  1202. if (_is_node_visible(p_node->get_child(i))) {
  1203. last = p_node->get_child(i);
  1204. }
  1205. }
  1206. }
  1207. if (last) {
  1208. Node *lastc = _find_last_visible(last);
  1209. if (lastc)
  1210. last = lastc;
  1211. } else {
  1212. last = p_node;
  1213. }
  1214. return last;
  1215. }
  1216. void SceneTreeDock::_normalize_drop(Node *&to_node, int &to_pos, int p_type) {
  1217. to_pos = -1;
  1218. if (p_type == -1) {
  1219. //drop at above selected node
  1220. if (to_node == EditorNode::get_singleton()->get_edited_scene()) {
  1221. to_node = NULL;
  1222. ERR_EXPLAIN("Cannot perform drop above the root node!");
  1223. ERR_FAIL();
  1224. }
  1225. to_pos = to_node->get_index();
  1226. to_node = to_node->get_parent();
  1227. } else if (p_type == 1) {
  1228. //drop at below selected node
  1229. if (to_node == EditorNode::get_singleton()->get_edited_scene()) {
  1230. //if at lower sibling of root node
  1231. to_pos = 0; //just insert at beginning of root node
  1232. return;
  1233. }
  1234. Node *lower_sibling = NULL;
  1235. if (_has_visible_children(to_node)) {
  1236. to_pos = 0;
  1237. } else {
  1238. for (int i = to_node->get_index() + 1; i < to_node->get_parent()->get_child_count(); i++) {
  1239. Node *c = to_node->get_parent()->get_child(i);
  1240. if (_is_node_visible(c)) {
  1241. lower_sibling = c;
  1242. break;
  1243. }
  1244. }
  1245. if (lower_sibling) {
  1246. to_pos = lower_sibling->get_index();
  1247. }
  1248. to_node = to_node->get_parent();
  1249. }
  1250. }
  1251. }
  1252. void SceneTreeDock::_files_dropped(Vector<String> p_files, NodePath p_to, int p_type) {
  1253. Node *node = get_node(p_to);
  1254. ERR_FAIL_COND(!node);
  1255. int to_pos = -1;
  1256. _normalize_drop(node, to_pos, p_type);
  1257. _perform_instance_scenes(p_files, node, to_pos);
  1258. }
  1259. void SceneTreeDock::_script_dropped(String p_file, NodePath p_to) {
  1260. Ref<Script> scr = ResourceLoader::load(p_file);
  1261. ERR_FAIL_COND(!scr.is_valid());
  1262. Node *n = get_node(p_to);
  1263. if (n) {
  1264. n->set_script(scr.get_ref_ptr());
  1265. }
  1266. }
  1267. void SceneTreeDock::_nodes_dragged(Array p_nodes, NodePath p_to, int p_type) {
  1268. Vector<Node *> nodes;
  1269. Node *to_node;
  1270. for (int i = 0; i < p_nodes.size(); i++) {
  1271. Node *n = get_node((p_nodes[i]));
  1272. if (n) {
  1273. nodes.push_back(n);
  1274. }
  1275. }
  1276. if (nodes.size() == 0)
  1277. return;
  1278. to_node = get_node(p_to);
  1279. if (!to_node)
  1280. return;
  1281. int to_pos = -1;
  1282. _normalize_drop(to_node, to_pos, p_type);
  1283. _do_reparent(to_node, to_pos, nodes, true);
  1284. }
  1285. void SceneTreeDock::_add_children_to_popup(Object *p_obj, int p_depth) {
  1286. if (p_depth > 8)
  1287. return;
  1288. List<PropertyInfo> pinfo;
  1289. p_obj->get_property_list(&pinfo);
  1290. for (List<PropertyInfo>::Element *E = pinfo.front(); E; E = E->next()) {
  1291. if (!(E->get().usage & PROPERTY_USAGE_EDITOR))
  1292. continue;
  1293. if (E->get().hint != PROPERTY_HINT_RESOURCE_TYPE)
  1294. continue;
  1295. Variant value = p_obj->get(E->get().name);
  1296. if (value.get_type() != Variant::OBJECT)
  1297. continue;
  1298. Object *obj = value;
  1299. if (!obj)
  1300. continue;
  1301. Ref<Texture> icon;
  1302. if (has_icon(obj->get_class(), "EditorIcons"))
  1303. icon = get_icon(obj->get_class(), "EditorIcons");
  1304. else
  1305. icon = get_icon("Object", "EditorIcons");
  1306. if (menu->get_item_count() == 0) {
  1307. menu->add_item(TTR("Sub-Resources:"));
  1308. menu->set_item_disabled(0, true);
  1309. }
  1310. int index = menu->get_item_count();
  1311. menu->add_icon_item(icon, E->get().name.capitalize(), EDIT_SUBRESOURCE_BASE + subresources.size());
  1312. menu->set_item_h_offset(index, p_depth * 10 * EDSCALE);
  1313. subresources.push_back(obj->get_instance_id());
  1314. _add_children_to_popup(obj, p_depth + 1);
  1315. }
  1316. }
  1317. void SceneTreeDock::_tree_rmb(const Vector2 &p_menu_pos) {
  1318. if (!EditorNode::get_singleton()->get_edited_scene()) {
  1319. menu->clear();
  1320. menu->add_icon_shortcut(get_icon("Add", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/add_child_node"), TOOL_NEW);
  1321. menu->add_icon_shortcut(get_icon("Instance", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/instance_scene"), TOOL_INSTANCE);
  1322. menu->set_size(Size2(1, 1));
  1323. menu->set_position(p_menu_pos);
  1324. menu->popup();
  1325. return;
  1326. }
  1327. List<Node *> selection = editor_selection->get_selected_node_list();
  1328. if (selection.size() == 0)
  1329. return;
  1330. menu->clear();
  1331. if (selection.size() == 1) {
  1332. subresources.clear();
  1333. _add_children_to_popup(selection.front()->get(), 0);
  1334. if (menu->get_item_count() > 0)
  1335. menu->add_separator();
  1336. menu->add_icon_shortcut(get_icon("Add", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/add_child_node"), TOOL_NEW);
  1337. menu->add_icon_shortcut(get_icon("Instance", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/instance_scene"), TOOL_INSTANCE);
  1338. menu->add_separator();
  1339. menu->add_icon_shortcut(get_icon("Reload", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/change_node_type"), TOOL_REPLACE);
  1340. menu->add_separator();
  1341. menu->add_icon_shortcut(get_icon("ScriptCreate", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/attach_script"), TOOL_ATTACH_SCRIPT);
  1342. menu->add_icon_shortcut(get_icon("ScriptRemove", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/clear_script"), TOOL_CLEAR_SCRIPT);
  1343. menu->add_separator();
  1344. }
  1345. menu->add_icon_shortcut(get_icon("MoveUp", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/move_up"), TOOL_MOVE_UP);
  1346. menu->add_icon_shortcut(get_icon("MoveDown", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/move_down"), TOOL_MOVE_DOWN);
  1347. menu->add_icon_shortcut(get_icon("Duplicate", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/duplicate"), TOOL_DUPLICATE);
  1348. menu->add_icon_shortcut(get_icon("Reparent", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/reparent"), TOOL_REPARENT);
  1349. if (selection.size() == 1) {
  1350. menu->add_separator();
  1351. menu->add_icon_shortcut(get_icon("Blend", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/merge_from_scene"), TOOL_MERGE_FROM_SCENE);
  1352. menu->add_icon_shortcut(get_icon("CreateNewSceneFrom", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/save_branch_as_scene"), TOOL_NEW_SCENE_FROM);
  1353. menu->add_separator();
  1354. menu->add_icon_shortcut(get_icon("CopyNodePath", "EditorIcons"), ED_GET_SHORTCUT("scene_tree/copy_node_path"), TOOL_COPY_NODE_PATH);
  1355. bool is_external = (selection[0]->get_filename() != "");
  1356. if (is_external) {
  1357. bool is_inherited = selection[0]->get_scene_inherited_state() != NULL;
  1358. bool is_top_level = selection[0]->get_owner() == NULL;
  1359. if (is_inherited && is_top_level) {
  1360. menu->add_separator();
  1361. menu->add_item(TTR("Clear Inheritance"), TOOL_SCENE_CLEAR_INHERITANCE);
  1362. menu->add_icon_item(get_icon("Load", "EditorIcons"), TTR("Open in Editor"), TOOL_SCENE_OPEN_INHERITED);
  1363. } else if (!is_top_level) {
  1364. menu->add_separator();
  1365. bool editable = EditorNode::get_singleton()->get_edited_scene()->is_editable_instance(selection[0]);
  1366. bool placeholder = selection[0]->get_scene_instance_load_placeholder();
  1367. menu->add_check_item(TTR("Editable Children"), TOOL_SCENE_EDITABLE_CHILDREN);
  1368. menu->add_check_item(TTR("Load As Placeholder"), TOOL_SCENE_USE_PLACEHOLDER);
  1369. menu->add_item(TTR("Discard Instancing"), TOOL_SCENE_CLEAR_INSTANCING);
  1370. menu->add_icon_item(get_icon("Load", "EditorIcons"), TTR("Open in Editor"), TOOL_SCENE_OPEN);
  1371. menu->set_item_checked(menu->get_item_idx_from_text(TTR("Editable Children")), editable);
  1372. menu->set_item_checked(menu->get_item_idx_from_text(TTR("Load As Placeholder")), placeholder);
  1373. }
  1374. }
  1375. }
  1376. menu->add_separator();
  1377. menu->add_icon_shortcut(get_icon("Remove", "EditorIcons"), ED_SHORTCUT("scene_tree/delete", TTR("Delete Node(s)"), KEY_DELETE), TOOL_ERASE);
  1378. menu->set_size(Size2(1, 1));
  1379. menu->set_position(p_menu_pos);
  1380. menu->popup();
  1381. }
  1382. void SceneTreeDock::_filter_changed(const String &p_filter) {
  1383. scene_tree->set_filter(p_filter);
  1384. }
  1385. String SceneTreeDock::get_filter() {
  1386. return filter->get_text();
  1387. }
  1388. void SceneTreeDock::set_filter(const String &p_filter) {
  1389. filter->set_text(p_filter);
  1390. scene_tree->set_filter(p_filter);
  1391. }
  1392. void SceneTreeDock::_focus_node() {
  1393. Node *node = scene_tree->get_selected();
  1394. ERR_FAIL_COND(!node);
  1395. if (node->is_class("CanvasItem")) {
  1396. CanvasItemEditorPlugin *editor = Object::cast_to<CanvasItemEditorPlugin>(editor_data->get_editor("2D"));
  1397. editor->get_canvas_item_editor()->focus_selection();
  1398. } else {
  1399. SpatialEditorPlugin *editor = Object::cast_to<SpatialEditorPlugin>(editor_data->get_editor("3D"));
  1400. editor->get_spatial_editor()->get_editor_viewport(0)->focus_selection();
  1401. }
  1402. }
  1403. void SceneTreeDock::open_script_dialog(Node *p_for_node) {
  1404. scene_tree->set_selected(p_for_node, false);
  1405. _tool_selected(TOOL_ATTACH_SCRIPT);
  1406. }
  1407. void SceneTreeDock::_bind_methods() {
  1408. ClassDB::bind_method(D_METHOD("_tool_selected"), &SceneTreeDock::_tool_selected, DEFVAL(false));
  1409. ClassDB::bind_method(D_METHOD("_create"), &SceneTreeDock::_create);
  1410. ClassDB::bind_method(D_METHOD("_node_reparent"), &SceneTreeDock::_node_reparent);
  1411. ClassDB::bind_method(D_METHOD("_set_owners"), &SceneTreeDock::_set_owners);
  1412. ClassDB::bind_method(D_METHOD("_node_selected"), &SceneTreeDock::_node_selected);
  1413. ClassDB::bind_method(D_METHOD("_node_renamed"), &SceneTreeDock::_node_renamed);
  1414. ClassDB::bind_method(D_METHOD("_script_created"), &SceneTreeDock::_script_created);
  1415. ClassDB::bind_method(D_METHOD("_load_request"), &SceneTreeDock::_load_request);
  1416. ClassDB::bind_method(D_METHOD("_script_open_request"), &SceneTreeDock::_script_open_request);
  1417. ClassDB::bind_method(D_METHOD("_unhandled_key_input"), &SceneTreeDock::_unhandled_key_input);
  1418. ClassDB::bind_method(D_METHOD("_input"), &SceneTreeDock::_input);
  1419. ClassDB::bind_method(D_METHOD("_nodes_drag_begin"), &SceneTreeDock::_nodes_drag_begin);
  1420. ClassDB::bind_method(D_METHOD("_delete_confirm"), &SceneTreeDock::_delete_confirm);
  1421. ClassDB::bind_method(D_METHOD("_node_prerenamed"), &SceneTreeDock::_node_prerenamed);
  1422. ClassDB::bind_method(D_METHOD("_import_subscene"), &SceneTreeDock::_import_subscene);
  1423. ClassDB::bind_method(D_METHOD("_selection_changed"), &SceneTreeDock::_selection_changed);
  1424. ClassDB::bind_method(D_METHOD("_new_scene_from"), &SceneTreeDock::_new_scene_from);
  1425. ClassDB::bind_method(D_METHOD("_nodes_dragged"), &SceneTreeDock::_nodes_dragged);
  1426. ClassDB::bind_method(D_METHOD("_files_dropped"), &SceneTreeDock::_files_dropped);
  1427. ClassDB::bind_method(D_METHOD("_script_dropped"), &SceneTreeDock::_script_dropped);
  1428. ClassDB::bind_method(D_METHOD("_tree_rmb"), &SceneTreeDock::_tree_rmb);
  1429. ClassDB::bind_method(D_METHOD("_filter_changed"), &SceneTreeDock::_filter_changed);
  1430. ClassDB::bind_method(D_METHOD("_focus_node"), &SceneTreeDock::_focus_node);
  1431. ClassDB::bind_method(D_METHOD("instance"), &SceneTreeDock::instance);
  1432. }
  1433. SceneTreeDock::SceneTreeDock(EditorNode *p_editor, Node *p_scene_root, EditorSelection *p_editor_selection, EditorData &p_editor_data) {
  1434. editor = p_editor;
  1435. edited_scene = NULL;
  1436. editor_data = &p_editor_data;
  1437. editor_selection = p_editor_selection;
  1438. scene_root = p_scene_root;
  1439. VBoxContainer *vbc = this;
  1440. HBoxContainer *filter_hbc = memnew(HBoxContainer);
  1441. filter_hbc->add_constant_override("separate", 0);
  1442. ToolButton *tb;
  1443. ED_SHORTCUT("scene_tree/add_child_node", TTR("Add Child Node"), KEY_MASK_CMD | KEY_A);
  1444. ED_SHORTCUT("scene_tree/instance_scene", TTR("Instance Child Scene"));
  1445. ED_SHORTCUT("scene_tree/change_node_type", TTR("Change Type"));
  1446. ED_SHORTCUT("scene_tree/attach_script", TTR("Attach Script"));
  1447. ED_SHORTCUT("scene_tree/clear_script", TTR("Clear Script"));
  1448. ED_SHORTCUT("scene_tree/move_up", TTR("Move Up"), KEY_MASK_CMD | KEY_UP);
  1449. ED_SHORTCUT("scene_tree/move_down", TTR("Move Down"), KEY_MASK_CMD | KEY_DOWN);
  1450. ED_SHORTCUT("scene_tree/duplicate", TTR("Duplicate"), KEY_MASK_CMD | KEY_D);
  1451. ED_SHORTCUT("scene_tree/reparent", TTR("Reparent"));
  1452. ED_SHORTCUT("scene_tree/merge_from_scene", TTR("Merge From Scene"));
  1453. ED_SHORTCUT("scene_tree/save_branch_as_scene", TTR("Save Branch as Scene"));
  1454. ED_SHORTCUT("scene_tree/copy_node_path", TTR("Copy Node Path"), KEY_MASK_CMD | KEY_C);
  1455. ED_SHORTCUT("scene_tree/delete_no_confirm", TTR("Delete (No Confirm)"), KEY_MASK_SHIFT | KEY_DELETE);
  1456. ED_SHORTCUT("scene_tree/delete", TTR("Delete"), KEY_DELETE);
  1457. tb = memnew(ToolButton);
  1458. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_NEW, false));
  1459. tb->set_tooltip(TTR("Add/Create a New Node"));
  1460. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/add_child_node"));
  1461. filter_hbc->add_child(tb);
  1462. button_add = tb;
  1463. tb = memnew(ToolButton);
  1464. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_INSTANCE, false));
  1465. tb->set_tooltip(TTR("Instance a scene file as a Node. Creates an inherited scene if no root node exists."));
  1466. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/instance_scene"));
  1467. filter_hbc->add_child(tb);
  1468. button_instance = tb;
  1469. vbc->add_child(filter_hbc);
  1470. filter = memnew(LineEdit);
  1471. filter->set_h_size_flags(SIZE_EXPAND_FILL);
  1472. filter->set_placeholder(TTR("Filter nodes"));
  1473. filter_hbc->add_child(filter);
  1474. filter->connect("text_changed", this, "_filter_changed");
  1475. tb = memnew(ToolButton);
  1476. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_ATTACH_SCRIPT, false));
  1477. tb->set_tooltip(TTR("Attach a new or existing script for the selected node."));
  1478. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/attach_script"));
  1479. filter_hbc->add_child(tb);
  1480. tb->hide();
  1481. button_create_script = tb;
  1482. tb = memnew(ToolButton);
  1483. tb->connect("pressed", this, "_tool_selected", make_binds(TOOL_CLEAR_SCRIPT, false));
  1484. tb->set_tooltip(TTR("Clear a script for the selected node."));
  1485. tb->set_shortcut(ED_GET_SHORTCUT("scene_tree/clear_script"));
  1486. filter_hbc->add_child(tb);
  1487. button_clear_script = tb;
  1488. tb->hide();
  1489. scene_tree = memnew(SceneTreeEditor(false, true, true));
  1490. vbc->add_child(scene_tree);
  1491. scene_tree->set_v_size_flags(SIZE_EXPAND | SIZE_FILL);
  1492. scene_tree->connect("rmb_pressed", this, "_tree_rmb");
  1493. scene_tree->connect("node_selected", this, "_node_selected", varray(), CONNECT_DEFERRED);
  1494. scene_tree->connect("node_renamed", this, "_node_renamed", varray(), CONNECT_DEFERRED);
  1495. scene_tree->connect("node_prerename", this, "_node_prerenamed");
  1496. scene_tree->connect("open", this, "_load_request");
  1497. scene_tree->connect("open_script", this, "_script_open_request");
  1498. scene_tree->connect("nodes_rearranged", this, "_nodes_dragged");
  1499. scene_tree->connect("files_dropped", this, "_files_dropped");
  1500. scene_tree->connect("script_dropped", this, "_script_dropped");
  1501. scene_tree->connect("nodes_dragged", this, "_nodes_drag_begin");
  1502. scene_tree->get_scene_tree()->connect("item_double_clicked", this, "_focus_node");
  1503. scene_tree->set_undo_redo(&editor_data->get_undo_redo());
  1504. scene_tree->set_editor_selection(editor_selection);
  1505. create_dialog = memnew(CreateDialog);
  1506. create_dialog->set_base_type("Node");
  1507. add_child(create_dialog);
  1508. create_dialog->connect("create", this, "_create");
  1509. script_create_dialog = memnew(ScriptCreateDialog);
  1510. add_child(script_create_dialog);
  1511. script_create_dialog->connect("script_created", this, "_script_created");
  1512. reparent_dialog = memnew(ReparentDialog);
  1513. add_child(reparent_dialog);
  1514. reparent_dialog->connect("reparent", this, "_node_reparent");
  1515. accept = memnew(AcceptDialog);
  1516. add_child(accept);
  1517. file = memnew(EditorFileDialog);
  1518. add_child(file);
  1519. file->connect("file_selected", this, "instance");
  1520. set_process_unhandled_key_input(true);
  1521. delete_dialog = memnew(ConfirmationDialog);
  1522. add_child(delete_dialog);
  1523. delete_dialog->connect("confirmed", this, "_delete_confirm");
  1524. import_subscene_dialog = memnew(EditorSubScene);
  1525. add_child(import_subscene_dialog);
  1526. import_subscene_dialog->connect("subscene_selected", this, "_import_subscene");
  1527. new_scene_from_dialog = memnew(EditorFileDialog);
  1528. new_scene_from_dialog->set_mode(EditorFileDialog::MODE_SAVE_FILE);
  1529. add_child(new_scene_from_dialog);
  1530. new_scene_from_dialog->connect("file_selected", this, "_new_scene_from");
  1531. menu = memnew(PopupMenu);
  1532. add_child(menu);
  1533. menu->connect("id_pressed", this, "_tool_selected");
  1534. first_enter = true;
  1535. restore_script_editor_on_drag = false;
  1536. clear_inherit_confirm = memnew(ConfirmationDialog);
  1537. clear_inherit_confirm->set_text(TTR("Clear Inheritance? (No Undo!)"));
  1538. clear_inherit_confirm->get_ok()->set_text(TTR("Clear!"));
  1539. add_child(clear_inherit_confirm);
  1540. vbc->add_constant_override("separation", 4);
  1541. set_process_input(true);
  1542. }