123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887 |
- /*************************************************************************/
- /* create_dialog.cpp */
- /*************************************************************************/
- /* This file is part of: */
- /* GODOT ENGINE */
- /* https://godotengine.org */
- /*************************************************************************/
- /* Copyright (c) 2007-2020 Juan Linietsky, Ariel Manzur. */
- /* Copyright (c) 2014-2020 Godot Engine contributors (cf. AUTHORS.md). */
- /* */
- /* Permission is hereby granted, free of charge, to any person obtaining */
- /* a copy of this software and associated documentation files (the */
- /* "Software"), to deal in the Software without restriction, including */
- /* without limitation the rights to use, copy, modify, merge, publish, */
- /* distribute, sublicense, and/or sell copies of the Software, and to */
- /* permit persons to whom the Software is furnished to do so, subject to */
- /* the following conditions: */
- /* */
- /* The above copyright notice and this permission notice shall be */
- /* included in all copies or substantial portions of the Software. */
- /* */
- /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
- /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
- /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
- /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
- /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
- /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
- /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
- /*************************************************************************/
- #include "create_dialog.h"
- #include "editor_node.h"
- #include "object_type_db.h"
- #include "print_string.h"
- #include "scene/gui/box_container.h"
- #if 1
- #include "editor_help.h"
- #include "editor_settings.h"
- #include "os/keyboard.h"
- void CreateDialog::popup(bool p_dontclear) {
- recent->clear();
- FileAccess *f = FileAccess::open(EditorSettings::get_singleton()->get_project_settings_path().plus_file("create_recent." + base_type), FileAccess::READ);
- if (f) {
- TreeItem *root = recent->create_item();
- while (!f->eof_reached()) {
- String l = f->get_line().strip_edges();
- if (l != String()) {
- TreeItem *ti = recent->create_item(root);
- ti->set_text(0, l);
- if (has_icon(l, "EditorIcons")) {
- ti->set_icon(0, get_icon(l, "EditorIcons"));
- } else {
- ti->set_icon(0, get_icon("Object", "EditorIcons"));
- }
- }
- }
- memdelete(f);
- }
- favorites->clear();
- f = FileAccess::open(EditorSettings::get_singleton()->get_project_settings_path().plus_file("favorites." + base_type), FileAccess::READ);
- favorite_list.clear();
- if (f) {
- while (!f->eof_reached()) {
- String l = f->get_line().strip_edges();
- if (l != String()) {
- favorite_list.push_back(l);
- }
- }
- memdelete(f);
- } else {
- #if 0
- // I think this was way too confusing
- if (base_type=="Node") {
- //harcode some favorites :D
- favorite_list.push_back("Panel");
- favorite_list.push_back("Button");
- favorite_list.push_back("Label");
- favorite_list.push_back("LineEdit");
- favorite_list.push_back("Node2D");
- favorite_list.push_back("Sprite");
- favorite_list.push_back("Camera2D");
- favorite_list.push_back("Area2D");
- favorite_list.push_back("CollisionShape2D");
- favorite_list.push_back("Spatial");
- favorite_list.push_back("Camera");
- favorite_list.push_back("Area");
- favorite_list.push_back("CollisionShape");
- favorite_list.push_back("TestCube");
- favorite_list.push_back("AnimationPlayer");
- }
- #endif
- }
- _update_favorite_list();
- popup_centered_ratio();
- if (p_dontclear)
- search_box->select_all();
- else {
- search_box->clear();
- }
- search_box->grab_focus();
- _update_search();
- }
- void CreateDialog::_text_changed(const String &p_newtext) {
- _update_search();
- }
- void CreateDialog::_sbox_input(const InputEvent &p_ie) {
- if (p_ie.type == InputEvent::KEY && (p_ie.key.scancode == KEY_UP ||
- p_ie.key.scancode == KEY_DOWN ||
- p_ie.key.scancode == KEY_PAGEUP ||
- p_ie.key.scancode == KEY_PAGEDOWN)) {
- search_options->call("_input_event", p_ie);
- search_box->accept_event();
- }
- }
- void CreateDialog::add_type(const String &p_type, HashMap<String, TreeItem *> &p_types, TreeItem *p_root, TreeItem **to_select) {
- if (p_types.has(p_type))
- return;
- if (!ObjectTypeDB::is_type(p_type, base_type) || p_type == base_type)
- return;
- String inherits = ObjectTypeDB::type_inherits_from(p_type);
- TreeItem *parent = p_root;
- if (inherits.length()) {
- if (!p_types.has(inherits)) {
- add_type(inherits, p_types, p_root, to_select);
- }
- if (p_types.has(inherits))
- parent = p_types[inherits];
- }
- TreeItem *item = search_options->create_item(parent);
- item->set_text(0, p_type);
- if (!ObjectTypeDB::can_instance(p_type)) {
- item->set_custom_color(0, Color(0.5, 0.5, 0.5));
- item->set_selectable(0, false);
- } else {
- if ((!*to_select && (search_box->get_text().is_subsequence_ofi(p_type))) || search_box->get_text() == p_type) {
- *to_select = item;
- }
- }
- if (bool(EditorSettings::get_singleton()->get("scenetree_editor/start_create_dialog_fully_expanded"))) {
- item->set_collapsed(false);
- } else {
- // don't collapse search results
- bool collapse = (search_box->get_text() == "");
- // don't collapse the root node
- collapse &= (item != p_root);
- // don't collapse abstract nodes on the first tree level
- collapse &= ((parent != p_root) || (ObjectTypeDB::can_instance(p_type)));
- item->set_collapsed(collapse);
- }
- const String &description = EditorHelp::get_doc_data()->class_list[p_type].brief_description;
- item->set_tooltip(0, description);
- if (has_icon(p_type, "EditorIcons")) {
- item->set_icon(0, get_icon(p_type, "EditorIcons"));
- }
- p_types[p_type] = item;
- }
- void CreateDialog::_update_search() {
- search_options->clear();
- favorite->set_disabled(true);
- help_bit->set_text("");
- /*
- TreeItem *root = search_options->create_item();
- _parse_fs(EditorFileSystem::get_singleton()->get_filesystem());
- */
- List<StringName> type_list;
- ObjectTypeDB::get_type_list(&type_list);
- HashMap<String, TreeItem *> types;
- TreeItem *root = search_options->create_item();
- root->set_text(0, base_type);
- if (has_icon(base_type, "EditorIcons")) {
- root->set_icon(0, get_icon(base_type, "EditorIcons"));
- }
- List<StringName>::Element *I = type_list.front();
- TreeItem *to_select = search_box->get_text() == base_type ? root : NULL;
- for (; I; I = I->next()) {
- String type = I->get();
- if (base_type == "Node" && type.begins_with("Editor"))
- continue; // do not show editor nodes
- if (!ObjectTypeDB::can_instance(type))
- continue; // cant create what can't be instanced
- if (search_box->get_text() == "") {
- add_type(type, types, root, &to_select);
- } else {
- bool found = false;
- String type = I->get();
- while (type != "" && ObjectTypeDB::is_type(type, base_type) && type != base_type) {
- if (search_box->get_text().is_subsequence_ofi(type)) {
- found = true;
- break;
- }
- type = ObjectTypeDB::type_inherits_from(type);
- }
- if (found)
- add_type(I->get(), types, root, &to_select);
- }
- if (EditorNode::get_editor_data().get_custom_types().has(type) && ObjectTypeDB::is_type(type, base_type)) {
- //there are custom types based on this... cool.
- //print_line("there are custom types");
- const Vector<EditorData::CustomType> &ct = EditorNode::get_editor_data().get_custom_types()[type];
- for (int i = 0; i < ct.size(); i++) {
- bool show = search_box->get_text().is_subsequence_ofi(ct[i].name);
- if (!show)
- continue;
- if (!types.has(type))
- add_type(type, types, root, &to_select);
- TreeItem *ti;
- if (types.has(type))
- ti = types[type];
- else
- ti = search_options->get_root();
- TreeItem *item = search_options->create_item(ti);
- item->set_metadata(0, type);
- item->set_text(0, ct[i].name);
- if (ct[i].icon.is_valid()) {
- item->set_icon(0, ct[i].icon);
- }
- if (!to_select || ct[i].name == search_box->get_text()) {
- to_select = item;
- }
- }
- }
- }
- if (to_select) {
- to_select->select(0);
- favorite->set_disabled(false);
- favorite->set_pressed(favorite_list.find(to_select->get_text(0)) != -1);
- }
- get_ok()->set_disabled(root->get_children() == NULL);
- }
- void CreateDialog::_confirmed() {
- TreeItem *ti = search_options->get_selected();
- if (!ti)
- return;
- FileAccess *f = FileAccess::open(EditorSettings::get_singleton()->get_project_settings_path().plus_file("create_recent." + base_type), FileAccess::WRITE);
- if (f) {
- f->store_line(get_selected_type());
- TreeItem *t = recent->get_root();
- if (t)
- t = t->get_children();
- int count = 0;
- while (t) {
- if (t->get_text(0) != get_selected_type()) {
- f->store_line(t->get_text(0));
- }
- if (count > 32) {
- //limit it to 32 entries..
- break;
- }
- t = t->get_next();
- count++;
- }
- memdelete(f);
- }
- emit_signal("create");
- hide();
- }
- void CreateDialog::_notification(int p_what) {
- if (p_what == NOTIFICATION_ENTER_TREE) {
- connect("confirmed", this, "_confirmed");
- favorite->set_icon(get_icon("Favorites", "EditorIcons"));
- }
- if (p_what == NOTIFICATION_EXIT_TREE) {
- disconnect("confirmed", this, "_confirmed");
- }
- if (p_what == NOTIFICATION_VISIBILITY_CHANGED) {
- if (is_visible()) {
- search_box->call_deferred("grab_focus"); // still not visible
- search_box->select_all();
- }
- }
- }
- void CreateDialog::set_base_type(const String &p_base) {
- base_type = p_base;
- set_title(TTR("Create New") + " " + p_base);
- _update_search();
- }
- String CreateDialog::get_selected_type() {
- TreeItem *selected = search_options->get_selected();
- if (selected)
- return selected->get_text(0);
- else
- return String();
- }
- Object *CreateDialog::instance_selected() {
- TreeItem *selected = search_options->get_selected();
- if (selected) {
- String custom = selected->get_metadata(0);
- if (custom != String()) {
- if (EditorNode::get_editor_data().get_custom_types().has(custom)) {
- for (int i = 0; i < EditorNode::get_editor_data().get_custom_types()[custom].size(); i++) {
- if (EditorNode::get_editor_data().get_custom_types()[custom][i].name == selected->get_text(0)) {
- Ref<Texture> icon = EditorNode::get_editor_data().get_custom_types()[custom][i].icon;
- Ref<Script> script = EditorNode::get_editor_data().get_custom_types()[custom][i].script;
- String name = selected->get_text(0);
- Object *ob = ObjectTypeDB::instance(custom);
- ERR_FAIL_COND_V(!ob, NULL);
- if (ob->is_type("Node")) {
- ob->call("set_name", name);
- }
- ob->set_script(script.get_ref_ptr());
- if (icon.is_valid())
- ob->set_meta("_editor_icon", icon);
- return ob;
- }
- }
- }
- } else {
- return ObjectTypeDB::instance(selected->get_text(0));
- }
- }
- return NULL;
- }
- String CreateDialog::get_base_type() const {
- return base_type;
- }
- void CreateDialog::_item_selected() {
- TreeItem *item = search_options->get_selected();
- if (!item)
- return;
- String name = item->get_text(0);
- favorite->set_disabled(false);
- favorite->set_pressed(favorite_list.find(name) != -1);
- if (!EditorHelp::get_doc_data()->class_list.has(name))
- return;
- help_bit->set_text(EditorHelp::get_doc_data()->class_list[name].brief_description);
- }
- void CreateDialog::_favorite_toggled() {
- TreeItem *item = search_options->get_selected();
- if (!item)
- return;
- String name = item->get_text(0);
- if (favorite_list.find(name) == -1) {
- favorite_list.push_back(name);
- favorite->set_pressed(true);
- } else {
- favorite_list.erase(name);
- favorite->set_pressed(false);
- }
- _save_favorite_list();
- _update_favorite_list();
- }
- void CreateDialog::_save_favorite_list() {
- FileAccess *f = FileAccess::open(EditorSettings::get_singleton()->get_project_settings_path().plus_file("favorites." + base_type), FileAccess::WRITE);
- if (f) {
- for (int i = 0; i < favorite_list.size(); i++) {
- f->store_line(favorite_list[i]);
- }
- memdelete(f);
- }
- }
- void CreateDialog::_update_favorite_list() {
- favorites->clear();
- TreeItem *root = favorites->create_item();
- for (int i = 0; i < favorite_list.size(); i++) {
- TreeItem *ti = favorites->create_item(root);
- String l = favorite_list[i];
- ti->set_text(0, l);
- if (has_icon(l, "EditorIcons")) {
- ti->set_icon(0, get_icon(l, "EditorIcons"));
- } else {
- ti->set_icon(0, get_icon("Object", "EditorIcons"));
- }
- }
- }
- void CreateDialog::_history_selected() {
- TreeItem *item = recent->get_selected();
- if (!item)
- return;
- search_box->set_text(item->get_text(0));
- _update_search();
- }
- void CreateDialog::_favorite_selected() {
- TreeItem *item = favorites->get_selected();
- if (!item)
- return;
- search_box->set_text(item->get_text(0));
- _update_search();
- }
- void CreateDialog::_history_activated() {
- _history_selected();
- _confirmed();
- }
- void CreateDialog::_favorite_activated() {
- _favorite_selected();
- _confirmed();
- }
- Variant CreateDialog::get_drag_data_fw(const Point2 &p_point, Control *p_from) {
- TreeItem *ti = favorites->get_item_at_pos(p_point);
- if (ti) {
- Dictionary d;
- d["type"] = "create_favorite_drag";
- d["class"] = ti->get_text(0);
- ToolButton *tb = memnew(ToolButton);
- tb->set_icon(ti->get_icon(0));
- tb->set_text(ti->get_text(0));
- set_drag_preview(tb);
- return d;
- }
- return Variant();
- }
- bool CreateDialog::can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) const {
- Dictionary d = p_data;
- if (d.has("type") && String(d["type"]) == "create_favorite_drag") {
- favorites->set_drop_mode_flags(Tree::DROP_MODE_INBETWEEN);
- return true;
- }
- return false;
- }
- void CreateDialog::drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) {
- Dictionary d = p_data;
- TreeItem *ti = favorites->get_item_at_pos(p_point);
- if (!ti)
- return;
- String drop_at = ti->get_text(0);
- int ds = favorites->get_drop_section_at_pos(p_point);
- int drop_idx = favorite_list.find(drop_at);
- if (drop_idx < 0)
- return;
- String type = d["class"];
- int from_idx = favorite_list.find(type);
- if (from_idx < 0)
- return;
- if (drop_idx == from_idx) {
- ds = -1; //cause it will be gone
- } else if (drop_idx > from_idx) {
- drop_idx--;
- }
- favorite_list.remove(from_idx);
- if (ds < 0) {
- favorite_list.insert(drop_idx, type);
- } else {
- if (drop_idx >= favorite_list.size() - 1) {
- favorite_list.push_back(type);
- } else {
- favorite_list.insert(drop_idx + 1, type);
- }
- }
- _save_favorite_list();
- _update_favorite_list();
- }
- void CreateDialog::_bind_methods() {
- ObjectTypeDB::bind_method(_MD("_text_changed"), &CreateDialog::_text_changed);
- ObjectTypeDB::bind_method(_MD("_confirmed"), &CreateDialog::_confirmed);
- ObjectTypeDB::bind_method(_MD("_sbox_input"), &CreateDialog::_sbox_input);
- ObjectTypeDB::bind_method(_MD("_item_selected"), &CreateDialog::_item_selected);
- ObjectTypeDB::bind_method(_MD("_favorite_toggled"), &CreateDialog::_favorite_toggled);
- ObjectTypeDB::bind_method(_MD("_history_selected"), &CreateDialog::_history_selected);
- ObjectTypeDB::bind_method(_MD("_favorite_selected"), &CreateDialog::_favorite_selected);
- ObjectTypeDB::bind_method(_MD("_history_activated"), &CreateDialog::_history_activated);
- ObjectTypeDB::bind_method(_MD("_favorite_activated"), &CreateDialog::_favorite_activated);
- ObjectTypeDB::bind_method("get_drag_data_fw", &CreateDialog::get_drag_data_fw);
- ObjectTypeDB::bind_method("can_drop_data_fw", &CreateDialog::can_drop_data_fw);
- ObjectTypeDB::bind_method("drop_data_fw", &CreateDialog::drop_data_fw);
- ADD_SIGNAL(MethodInfo("create"));
- }
- CreateDialog::CreateDialog() {
- HSplitContainer *hbc = memnew(HSplitContainer);
- add_child(hbc);
- set_child_rect(hbc);
- VBoxContainer *lvbc = memnew(VBoxContainer);
- hbc->add_child(lvbc);
- lvbc->set_custom_minimum_size(Size2(150, 0) * EDSCALE);
- favorites = memnew(Tree);
- lvbc->add_margin_child(TTR("Favorites:"), favorites, true);
- favorites->set_hide_root(true);
- favorites->set_hide_folding(true);
- favorites->connect("cell_selected", this, "_favorite_selected");
- favorites->connect("item_activated", this, "_favorite_activated");
- favorites->set_drag_forwarding(this);
- recent = memnew(Tree);
- lvbc->add_margin_child(TTR("Recent:"), recent, true);
- recent->set_hide_root(true);
- recent->set_hide_folding(true);
- recent->connect("cell_selected", this, "_history_selected");
- recent->connect("item_activated", this, "_history_activated");
- VBoxContainer *vbc = memnew(VBoxContainer);
- hbc->add_child(vbc);
- vbc->set_h_size_flags(SIZE_EXPAND_FILL);
- HBoxContainer *search_hb = memnew(HBoxContainer);
- search_box = memnew(LineEdit);
- search_box->set_h_size_flags(SIZE_EXPAND_FILL);
- search_hb->add_child(search_box);
- favorite = memnew(Button);
- favorite->set_toggle_mode(true);
- search_hb->add_child(favorite);
- favorite->connect("pressed", this, "_favorite_toggled");
- vbc->add_margin_child(TTR("Search:"), search_hb);
- search_box->connect("text_changed", this, "_text_changed");
- search_box->connect("input_event", this, "_sbox_input");
- search_options = memnew(Tree);
- vbc->add_margin_child(TTR("Matches:"), search_options, true);
- get_ok()->set_text(TTR("Create"));
- get_ok()->set_disabled(true);
- register_text_enter(search_box);
- set_hide_on_ok(false);
- search_options->connect("item_activated", this, "_confirmed");
- search_options->connect("cell_selected", this, "_item_selected");
- // search_options->set_hide_root(true);
- base_type = "Object";
- help_bit = memnew(EditorHelpBit);
- vbc->add_margin_child(TTR("Description:"), help_bit);
- help_bit->connect("request_hide", this, "_closed");
- }
- #else
- //old create dialog, disabled
- void CreateDialog::_notification(int p_what) {
- if (p_what == NOTIFICATION_READY) {
- connect("confirmed", this, "_create");
- update_tree();
- }
- if (p_what == NOTIFICATION_DRAW) {
- //RID ci = get_canvas_item();
- //get_stylebox("panel","PopupMenu")->draw(ci,Rect2(Point2(),get_size()));
- }
- }
- void CreateDialog::_create() {
- if (tree->get_selected())
- emit_signal("create");
- hide();
- }
- void CreateDialog::_cancel() {
- hide();
- }
- void CreateDialog::_text_changed(String p_text) {
- update_tree();
- }
- void CreateDialog::add_type(const String &p_type, HashMap<String, TreeItem *> &p_types, TreeItem *p_root) {
- if (p_types.has(p_type))
- return;
- if (!ObjectTypeDB::is_type(p_type, base) || p_type == base)
- return;
- String inherits = ObjectTypeDB::type_inherits_from(p_type);
- TreeItem *parent = p_root;
- if (inherits.length()) {
- if (!p_types.has(inherits)) {
- add_type(inherits, p_types, p_root);
- }
- if (p_types.has(inherits))
- parent = p_types[inherits];
- }
- TreeItem *item = tree->create_item(parent);
- item->set_text(0, p_type);
- if (!ObjectTypeDB::can_instance(p_type)) {
- item->set_custom_color(0, Color(0.5, 0.5, 0.5));
- item->set_selectable(0, false);
- }
- if (has_icon(p_type, "EditorIcons")) {
- item->set_icon(0, get_icon(p_type, "EditorIcons"));
- }
- p_types[p_type] = item;
- }
- void CreateDialog::update_tree() {
- tree->clear();
- List<String> type_list;
- ObjectTypeDB::get_type_list(&type_list);
- HashMap<String, TreeItem *> types;
- TreeItem *root = tree->create_item();
- root->set_text(0, base);
- List<String>::Element *I = type_list.front();
- for (; I; I = I->next()) {
- String type = I->get();
- if (!ObjectTypeDB::can_instance(type))
- continue; // cant create what can't be instanced
- if (filter->get_text() == "")
- add_type(type, types, root);
- else {
- bool found = false;
- String type = I->get();
- while (type != "" && ObjectTypeDB::is_type(type, base) && type != base) {
- if (type.findn(filter->get_text()) != -1) {
- found = true;
- break;
- }
- type = ObjectTypeDB::type_inherits_from(type);
- }
- if (found)
- add_type(I->get(), types, root);
- }
- if (EditorNode::get_editor_data().get_custom_types().has(type)) {
- //there are custom types based on this... cool.
- const Vector<EditorData::CustomType> &ct = EditorNode::get_editor_data().get_custom_types()[type];
- for (int i = 0; i < ct.size(); i++) {
- bool show = filter->get_text() == "" || ct[i].name.findn(filter->get_text()) != -1;
- if (!show)
- continue;
- if (!types.has(type))
- add_type(type, types, root);
- TreeItem *ti;
- if (types.has(type))
- ti = types[type];
- else
- ti = tree->get_root();
- TreeItem *item = tree->create_item(ti);
- item->set_metadata(0, type);
- item->set_text(0, ct[i].name);
- if (ct[i].icon.is_valid()) {
- item->set_icon(0, ct[i].icon);
- }
- }
- }
- }
- }
- Object *CreateDialog::instance_selected() {
- if (!tree->get_selected())
- return NULL;
- String base = String(tree->get_selected()->get_metadata(0));
- if (base != "") {
- String name = tree->get_selected()->get_text(0);
- if (EditorNode::get_editor_data().get_custom_types().has(base)) {
- const Vector<EditorData::CustomType> &ct = EditorNode::get_editor_data().get_custom_types()[base];
- for (int i = 0; i < ct.size(); i++) {
- if (ct[i].name == name) {
- Object *obj = ObjectTypeDB::instance(base);
- ERR_FAIL_COND_V(!obj, NULL);
- obj->set_script(ct[i].script.get_ref_ptr());
- if (ct[i].icon.is_valid())
- obj->set_meta("_editor_icon", ct[i].icon);
- return obj;
- }
- }
- }
- ERR_FAIL_V(NULL);
- }
- return ObjectTypeDB::instance(tree->get_selected()->get_text(0));
- }
- void CreateDialog::_bind_methods() {
- ObjectTypeDB::bind_method("_create", &CreateDialog::_create);
- ObjectTypeDB::bind_method("_cancel", &CreateDialog::_cancel);
- ObjectTypeDB::bind_method("_text_changed", &CreateDialog::_text_changed);
- ADD_SIGNAL(MethodInfo("create"));
- }
- void CreateDialog::set_base_type(const String &p_base) {
- set_title(vformat("Create %s Type", p_base));
- if (base == p_base)
- return;
- base = p_base;
- if (is_inside_scene())
- update_tree();
- }
- String CreateDialog::get_base_type() const {
- return base;
- }
- CreateDialog::CreateDialog() {
- VBoxContainer *vbc = memnew(VBoxContainer);
- add_child(vbc);
- set_child_rect(vbc);
- get_ok()->set_text("Create");
- tree = memnew(Tree);
- vbc->add_margin_child("Type:", tree, true);
- //tree->set_hide_root(true);
- filter = memnew(LineEdit);
- vbc->add_margin_child("Filter:", filter);
- base = "Node";
- set_as_toplevel(true);
- tree->connect("item_activated", this, "_create");
- filter->connect("text_changed", this, "_text_changed");
- }
- CreateDialog::~CreateDialog() {
- }
- #endif
|