/**************************************************************************/ /* progress_dialog.cpp */ /**************************************************************************/ /* This file is part of: */ /* GODOT ENGINE */ /* https://godotengine.org */ /**************************************************************************/ /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */ /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */ /* */ /* 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 "progress_dialog.h" #include "core/object/message_queue.h" #include "core/os/os.h" #include "editor/editor_node.h" #include "editor/editor_scale.h" #include "main/main.h" #include "servers/display_server.h" void BackgroundProgress::_add_task(const String &p_task, const String &p_label, int p_steps) { _THREAD_SAFE_METHOD_ ERR_FAIL_COND_MSG(tasks.has(p_task), "Task '" + p_task + "' already exists."); BackgroundProgress::Task t; t.hb = memnew(HBoxContainer); Label *l = memnew(Label); l->set_text(p_label + " "); t.hb->add_child(l); t.progress = memnew(ProgressBar); t.progress->set_max(p_steps); t.progress->set_value(p_steps); Control *ec = memnew(Control); ec->set_h_size_flags(SIZE_EXPAND_FILL); ec->set_v_size_flags(SIZE_EXPAND_FILL); t.progress->set_anchors_and_offsets_preset(Control::PRESET_FULL_RECT); ec->add_child(t.progress); ec->set_custom_minimum_size(Size2(80, 5) * EDSCALE); t.hb->add_child(ec); add_child(t.hb); tasks[p_task] = t; } void BackgroundProgress::_update() { _THREAD_SAFE_METHOD_ for (const KeyValue<String, int> &E : updates) { if (tasks.has(E.key)) { _task_step(E.key, E.value); } } updates.clear(); } void BackgroundProgress::_task_step(const String &p_task, int p_step) { _THREAD_SAFE_METHOD_ ERR_FAIL_COND(!tasks.has(p_task)); Task &t = tasks[p_task]; if (p_step < 0) { t.progress->set_value(t.progress->get_value() + 1); } else { t.progress->set_value(p_step); } } void BackgroundProgress::_end_task(const String &p_task) { _THREAD_SAFE_METHOD_ ERR_FAIL_COND(!tasks.has(p_task)); Task &t = tasks[p_task]; memdelete(t.hb); tasks.erase(p_task); } void BackgroundProgress::_bind_methods() { ClassDB::bind_method("_add_task", &BackgroundProgress::_add_task); ClassDB::bind_method("_task_step", &BackgroundProgress::_task_step); ClassDB::bind_method("_end_task", &BackgroundProgress::_end_task); ClassDB::bind_method("_update", &BackgroundProgress::_update); } void BackgroundProgress::add_task(const String &p_task, const String &p_label, int p_steps) { MessageQueue::get_singleton()->push_call(this, "_add_task", p_task, p_label, p_steps); } void BackgroundProgress::task_step(const String &p_task, int p_step) { //this code is weird, but it prevents deadlock. bool no_updates = true; { _THREAD_SAFE_METHOD_ no_updates = updates.is_empty(); } if (no_updates) { MessageQueue::get_singleton()->push_call(this, "_update"); } { _THREAD_SAFE_METHOD_ updates[p_task] = p_step; } } void BackgroundProgress::end_task(const String &p_task) { MessageQueue::get_singleton()->push_call(this, "_end_task", p_task); } //////////////////////////////////////////////// ProgressDialog *ProgressDialog::singleton = nullptr; void ProgressDialog::_notification(int p_what) { if (p_what == NOTIFICATION_VISIBILITY_CHANGED) { if (!is_visible()) { Node *p = get_parent(); if (p) { p->remove_child(this); } } } } void ProgressDialog::_popup() { Size2 ms = main->get_combined_minimum_size(); ms.width = MAX(500 * EDSCALE, ms.width); Ref<StyleBox> style = main->get_theme_stylebox(SNAME("panel"), SNAME("PopupMenu")); ms += style->get_minimum_size(); main->set_offset(SIDE_LEFT, style->get_margin(SIDE_LEFT)); main->set_offset(SIDE_RIGHT, -style->get_margin(SIDE_RIGHT)); main->set_offset(SIDE_TOP, style->get_margin(SIDE_TOP)); main->set_offset(SIDE_BOTTOM, -style->get_margin(SIDE_BOTTOM)); EditorNode *ed = EditorNode::get_singleton(); if (ed && !is_inside_tree()) { Window *w = ed->get_window(); while (w && w->get_exclusive_child()) { w = w->get_exclusive_child(); } if (w && w != this) { w->add_child(this); popup_centered(ms); } } } void ProgressDialog::add_task(const String &p_task, const String &p_label, int p_steps, bool p_can_cancel) { if (MessageQueue::get_singleton()->is_flushing()) { ERR_PRINT("Do not use progress dialog (task) while flushing the message queue or using call_deferred()!"); return; } ERR_FAIL_COND_MSG(tasks.has(p_task), "Task '" + p_task + "' already exists."); ProgressDialog::Task t; t.vb = memnew(VBoxContainer); VBoxContainer *vb2 = memnew(VBoxContainer); t.vb->add_margin_child(p_label, vb2); t.progress = memnew(ProgressBar); t.progress->set_max(p_steps); t.progress->set_value(p_steps); vb2->add_child(t.progress); t.state = memnew(Label); t.state->set_clip_text(true); vb2->add_child(t.state); main->add_child(t.vb); tasks[p_task] = t; if (p_can_cancel) { cancel_hb->show(); } else { cancel_hb->hide(); } cancel_hb->move_to_front(); cancelled = false; _popup(); if (p_can_cancel) { cancel->grab_focus(); } } bool ProgressDialog::task_step(const String &p_task, const String &p_state, int p_step, bool p_force_redraw) { ERR_FAIL_COND_V(!tasks.has(p_task), cancelled); if (!p_force_redraw) { uint64_t tus = OS::get_singleton()->get_ticks_usec(); if (tus - last_progress_tick < 200000) { //200ms return cancelled; } } Task &t = tasks[p_task]; if (p_step < 0) { t.progress->set_value(t.progress->get_value() + 1); } else { t.progress->set_value(p_step); } t.state->set_text(p_state); last_progress_tick = OS::get_singleton()->get_ticks_usec(); if (cancel_hb->is_visible()) { DisplayServer::get_singleton()->process_events(); } #ifndef ANDROID_ENABLED Main::iteration(); // this will not work on a lot of platforms, so it's only meant for the editor #endif return cancelled; } void ProgressDialog::end_task(const String &p_task) { ERR_FAIL_COND(!tasks.has(p_task)); Task &t = tasks[p_task]; memdelete(t.vb); tasks.erase(p_task); if (tasks.is_empty()) { hide(); } else { _popup(); } } void ProgressDialog::_cancel_pressed() { cancelled = true; } void ProgressDialog::_bind_methods() { } ProgressDialog::ProgressDialog() { main = memnew(VBoxContainer); add_child(main); main->set_anchors_and_offsets_preset(Control::PRESET_FULL_RECT); set_exclusive(true); last_progress_tick = 0; singleton = this; cancel_hb = memnew(HBoxContainer); main->add_child(cancel_hb); cancel_hb->hide(); cancel = memnew(Button); cancel_hb->add_spacer(); cancel_hb->add_child(cancel); cancel->set_text(TTR("Cancel")); cancel_hb->add_spacer(); cancel->connect("pressed", callable_mp(this, &ProgressDialog::_cancel_pressed)); }