123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292 |
- #include "editor_main_screen.h"
- #include "core/io/config_file.h"
- #include "editor/editor_node.h"
- #include "editor/editor_settings.h"
- #include "editor/editor_string_names.h"
- #include "editor/plugins/editor_plugin.h"
- #include "scene/gui/box_container.h"
- #include "scene/gui/button.h"
- void EditorMainScreen::_notification(int p_what) {
- switch (p_what) {
- case NOTIFICATION_READY: {
- if (EDITOR_3D < buttons.size() && buttons[EDITOR_3D]->is_visible()) {
-
- select(EDITOR_3D);
- return;
- }
-
-
- for (int i = 0; i < buttons.size(); i++) {
- Button *editor_button = buttons[i];
- if (editor_button->is_visible()) {
- select(i);
- return;
- }
- }
- select(-1);
- } break;
- case NOTIFICATION_THEME_CHANGED: {
- for (int i = 0; i < buttons.size(); i++) {
- Button *tb = buttons[i];
- EditorPlugin *p_editor = editor_table[i];
- Ref<Texture2D> icon = p_editor->get_icon();
- if (icon.is_valid()) {
- tb->set_button_icon(icon);
- } else if (has_theme_icon(p_editor->get_name(), EditorStringName(EditorIcons))) {
- tb->set_button_icon(get_theme_icon(p_editor->get_name(), EditorStringName(EditorIcons)));
- }
- }
- } break;
- }
- }
- void EditorMainScreen::set_button_container(HBoxContainer *p_button_hb) {
- button_hb = p_button_hb;
- }
- void EditorMainScreen::save_layout_to_config(Ref<ConfigFile> p_config_file, const String &p_section) const {
- int selected_main_editor_idx = -1;
- for (int i = 0; i < buttons.size(); i++) {
- if (buttons[i]->is_pressed()) {
- selected_main_editor_idx = i;
- break;
- }
- }
- if (selected_main_editor_idx != -1) {
- p_config_file->set_value(p_section, "selected_main_editor_idx", selected_main_editor_idx);
- } else {
- p_config_file->set_value(p_section, "selected_main_editor_idx", Variant());
- }
- }
- void EditorMainScreen::load_layout_from_config(Ref<ConfigFile> p_config_file, const String &p_section) {
- int selected_main_editor_idx = p_config_file->get_value(p_section, "selected_main_editor_idx", -1);
- if (selected_main_editor_idx >= 0 && selected_main_editor_idx < buttons.size()) {
- callable_mp(this, &EditorMainScreen::select).call_deferred(selected_main_editor_idx);
- }
- }
- void EditorMainScreen::set_button_enabled(int p_index, bool p_enabled) {
- ERR_FAIL_INDEX(p_index, buttons.size());
- buttons[p_index]->set_visible(p_enabled);
- if (!p_enabled && buttons[p_index]->is_pressed()) {
- select(EDITOR_2D);
- }
- }
- bool EditorMainScreen::is_button_enabled(int p_index) const {
- ERR_FAIL_INDEX_V(p_index, buttons.size(), false);
- return buttons[p_index]->is_visible();
- }
- int EditorMainScreen::_get_current_main_editor() const {
- for (int i = 0; i < editor_table.size(); i++) {
- if (editor_table[i] == selected_plugin) {
- return i;
- }
- }
- return 0;
- }
- void EditorMainScreen::select_next() {
- int editor = _get_current_main_editor();
- do {
- if (editor == editor_table.size() - 1) {
- editor = 0;
- } else {
- editor++;
- }
- } while (!buttons[editor]->is_visible());
- select(editor);
- }
- void EditorMainScreen::select_prev() {
- int editor = _get_current_main_editor();
- do {
- if (editor == 0) {
- editor = editor_table.size() - 1;
- } else {
- editor--;
- }
- } while (!buttons[editor]->is_visible());
- select(editor);
- }
- void EditorMainScreen::select_by_name(const String &p_name) {
- ERR_FAIL_COND(p_name.is_empty());
- for (int i = 0; i < buttons.size(); i++) {
- if (buttons[i]->get_text() == p_name) {
- select(i);
- return;
- }
- }
- ERR_FAIL_MSG("The editor name '" + p_name + "' was not found.");
- }
- void EditorMainScreen::select(int p_index) {
- if (EditorNode::get_singleton()->is_changing_scene()) {
- return;
- }
- ERR_FAIL_INDEX(p_index, editor_table.size());
- if (!buttons[p_index]->is_visible()) {
- return;
- }
- for (int i = 0; i < buttons.size(); i++) {
- buttons[i]->set_pressed_no_signal(i == p_index);
- }
- EditorPlugin *new_editor = editor_table[p_index];
- ERR_FAIL_NULL(new_editor);
- if (selected_plugin == new_editor) {
- return;
- }
- if (selected_plugin) {
- selected_plugin->make_visible(false);
- }
- selected_plugin = new_editor;
- selected_plugin->make_visible(true);
- selected_plugin->selected_notify();
- EditorData &editor_data = EditorNode::get_editor_data();
- int plugin_count = editor_data.get_editor_plugin_count();
- for (int i = 0; i < plugin_count; i++) {
- editor_data.get_editor_plugin(i)->notify_main_screen_changed(selected_plugin->get_name());
- }
- EditorNode::get_singleton()->update_distraction_free_mode();
- }
- int EditorMainScreen::get_selected_index() const {
- for (int i = 0; i < editor_table.size(); i++) {
- if (selected_plugin == editor_table[i]) {
- return i;
- }
- }
- return -1;
- }
- int EditorMainScreen::get_plugin_index(EditorPlugin *p_editor) const {
- int screen = -1;
- for (int i = 0; i < editor_table.size(); i++) {
- if (p_editor == editor_table[i]) {
- screen = i;
- break;
- }
- }
- return screen;
- }
- EditorPlugin *EditorMainScreen::get_selected_plugin() const {
- return selected_plugin;
- }
- VBoxContainer *EditorMainScreen::get_control() const {
- return main_screen_vbox;
- }
- void EditorMainScreen::add_main_plugin(EditorPlugin *p_editor) {
- Button *tb = memnew(Button);
- tb->set_toggle_mode(true);
- tb->set_theme_type_variation("MainScreenButton");
- tb->set_name(p_editor->get_name());
- tb->set_text(p_editor->get_name());
- Ref<Texture2D> icon = p_editor->get_icon();
- if (icon.is_null() && has_theme_icon(p_editor->get_name(), EditorStringName(EditorIcons))) {
- icon = get_editor_theme_icon(p_editor->get_name());
- }
- if (icon.is_valid()) {
- tb->set_button_icon(icon);
-
- icon->connect_changed(callable_mp((Control *)tb, &Control::update_minimum_size));
- }
- tb->connect(SceneStringName(pressed), callable_mp(this, &EditorMainScreen::select).bind(buttons.size()));
- buttons.push_back(tb);
- button_hb->add_child(tb);
- editor_table.push_back(p_editor);
- }
- void EditorMainScreen::remove_main_plugin(EditorPlugin *p_editor) {
-
-
- for (int i = buttons.size() - 1; i >= 0; i--) {
- if (p_editor->get_name() == buttons[i]->get_text()) {
- if (buttons[i]->is_pressed()) {
- select(EDITOR_SCRIPT);
- }
- memdelete(buttons[i]);
- buttons.remove_at(i);
- break;
- } else {
- buttons[i]->disconnect(SceneStringName(pressed), callable_mp(this, &EditorMainScreen::select));
- buttons[i]->connect(SceneStringName(pressed), callable_mp(this, &EditorMainScreen::select).bind(i - 1));
- }
- }
- if (selected_plugin == p_editor) {
- selected_plugin = nullptr;
- }
- editor_table.erase(p_editor);
- }
- EditorMainScreen::EditorMainScreen() {
- main_screen_vbox = memnew(VBoxContainer);
- main_screen_vbox->set_name("MainScreen");
- main_screen_vbox->set_v_size_flags(Control::SIZE_EXPAND_FILL);
- main_screen_vbox->add_theme_constant_override("separation", 0);
- add_child(main_screen_vbox);
- }
|