/*************************************************************************/ /* connections_dialog.cpp */ /*************************************************************************/ /* This file is part of: */ /* GODOT ENGINE */ /* http://www.godotengine.org */ /*************************************************************************/ /* Copyright (c) 2007-2017 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 "connections_dialog.h" #include "scene/gui/label.h" #include "print_string.h" #include "editor_settings.h" #include "editor_node.h" #include "plugins/script_editor_plugin.h" class ConnectDialogBinds : public Object { OBJ_TYPE( ConnectDialogBinds, Object ); public: Vector params; bool _set(const StringName& p_name, const Variant& p_value) { String name=p_name; if (name.begins_with("bind/")) { int which = name.get_slice("/",1).to_int()-1; ERR_FAIL_INDEX_V(which,params.size(),false); params[which]=p_value; } else return false; return true; } bool _get(const StringName& p_name,Variant &r_ret) const { String name=p_name; if (name.begins_with("bind/")) { int which = name.get_slice("/",1).to_int()-1; ERR_FAIL_INDEX_V(which,params.size(),false); r_ret = params[which]; } else return false; return true; } void _get_property_list(List *p_list) const { for(int i=0;ipush_back( PropertyInfo( params[i].get_type(), "bind/"+itos(i+1)) ); } } void notify_changed() { _change_notify(); } ConnectDialogBinds() { } }; void ConnectDialog::_notification(int p_what) { if (p_what==NOTIFICATION_DRAW) { //RID ci = get_canvas_item(); //get_stylebox("panel","PopupMenu")->draw(ci,Rect2(Point2(),get_size())); } if (p_what==NOTIFICATION_ENTER_TREE) { bind_editor->edit(cdbinds); } } void ConnectDialog::_tree_node_selected() { //dst_method_list->get_popup()->clear(); Node *current=tree->get_selected(); if (!current) { make_callback->hide(); return; } if (current->get_script().is_null()) make_callback->hide(); else make_callback->show(); #if 0 List methods; current->get_method_list(&methods); for (List::Element *E=methods.front();E;E=E->next()) { if (E->get().name.length() && E->get().name[0]=='_') continue; // hidden method, not show! if (ObjectTypeDB::has_method(node->get_type(),"Node") || ObjectTypeDB::has_method(node->get_type(),"Control",true)) continue; //avoid too much unnecesary stuff String method=E->get().name+"("; for(int i=0;iget().arguments.size();i++) { if (i!=0) method+=", "; method+=Variant::get_type_name(E->get().arguments[i].type); if (E->get().arguments[i].name.length()) { method+=" "; method+=E->get().arguments[i].name; } } method+=")"; //dst_method_list->get_popup()->add_item(method); } #endif dst_path->set_text(node->get_path_to(current)); } void ConnectDialog::_dst_method_list_selected(int p_idx) { //dst_method->set_text( dst_method_list->get_popup()->get_item_text(p_idx)); } void ConnectDialog::edit(Node *p_node) { node=p_node; //dst_method_list->get_popup()->clear(); tree->set_selected(NULL); tree->set_marked(node,true); dst_path->set_text(""); dst_method->set_text(""); deferred->set_pressed(false); oneshot->set_pressed(false); cdbinds->params.clear(); cdbinds->notify_changed(); } void ConnectDialog::ok_pressed() { if (dst_method->get_text()=="") { error->set_text(TTR("Method in target Node must be specified!")); error->popup_centered_minsize(); return; } Node* target = tree->get_selected(); if (target->get_script().is_null()) { if (!target->has_method(dst_method->get_text())) { error->set_text(TTR("Target method not found! Specify a valid method or attach a script to target Node.")); error->popup_centered_minsize(); return; } } emit_signal("connected"); hide(); } void ConnectDialog::_cancel_pressed() { hide(); } NodePath ConnectDialog::get_dst_path() const { return dst_path->get_text(); } bool ConnectDialog::get_deferred() const { return deferred->is_pressed(); } bool ConnectDialog::get_oneshot() const { return oneshot->is_pressed(); } StringName ConnectDialog::get_dst_method() const { String txt=dst_method->get_text(); if (txt.find("(")!=-1) txt=txt.left( txt.find("(")).strip_edges(); return txt; } Vector ConnectDialog::get_binds() const { return cdbinds->params; } void ConnectDialog::_add_bind() { if (cdbinds->params.size() >= VARIANT_ARG_MAX) return; Variant::Type vt = (Variant::Type)type_list->get_item_ID(type_list->get_selected()); Variant value; switch(vt) { case Variant::BOOL: value = false ; break; case Variant::INT: value = 0; break; case Variant::REAL: value = 0.0; break; case Variant::STRING: value = ""; break; case Variant::VECTOR2: value = Vector2(); break; case Variant::RECT2: value = Rect2(); break; case Variant::VECTOR3: value = Vector3(); break; case Variant::PLANE: value = Plane(); break; case Variant::QUAT: value = Quat(); break; case Variant::_AABB: value = AABB(); break; case Variant::MATRIX3: value = Matrix3(); break; case Variant::TRANSFORM: value = Transform(); break; case Variant::COLOR: value = Color(); break; case Variant::IMAGE: value = Image(); break; default: { ERR_FAIL(); } break; } ERR_FAIL_COND(value.get_type()==Variant::NIL); cdbinds->params.push_back(value); cdbinds->notify_changed(); } void ConnectDialog::_remove_bind() { String st = bind_editor->get_selected_path(); if (st=="") return; int idx = st.get_slice("/",1).to_int()-1; ERR_FAIL_INDEX(idx,cdbinds->params.size()); cdbinds->params.remove(idx); cdbinds->notify_changed(); } void ConnectDialog::set_dst_node(Node* p_node) { tree->set_selected(p_node); } void ConnectDialog::set_dst_method(const StringName& p_method) { dst_method->set_text(p_method); } void ConnectDialog::_bind_methods() { //ObjectTypeDB::bind_method("_ok",&ConnectDialog::_ok_pressed); ObjectTypeDB::bind_method("_cancel",&ConnectDialog::_cancel_pressed); //ObjectTypeDB::bind_method("_dst_method_list_selected",&ConnectDialog::_dst_method_list_selected); ObjectTypeDB::bind_method("_tree_node_selected",&ConnectDialog::_tree_node_selected); ObjectTypeDB::bind_method("_add_bind",&ConnectDialog::_add_bind); ObjectTypeDB::bind_method("_remove_bind",&ConnectDialog::_remove_bind); ADD_SIGNAL( MethodInfo("connected") ); } ConnectDialog::ConnectDialog() { VBoxContainer *vbc = memnew( VBoxContainer ); add_child(vbc); set_child_rect(vbc); HBoxContainer *main_hb = memnew( HBoxContainer ); vbc->add_child(main_hb); main_hb->set_v_size_flags(SIZE_EXPAND_FILL); VBoxContainer *vbc_left = memnew( VBoxContainer ); main_hb->add_child(vbc_left); vbc_left->set_h_size_flags(SIZE_EXPAND_FILL); tree = memnew(SceneTreeEditor(false)); tree->get_scene_tree()->connect("item_activated",this,"_ok"); vbc_left->add_margin_child(TTR("Connect To Node:"),tree,true); VBoxContainer *vbc_right = memnew( VBoxContainer ); main_hb->add_child(vbc_right); vbc_right->set_h_size_flags(SIZE_EXPAND_FILL); HBoxContainer *add_bind_hb = memnew( HBoxContainer ); type_list = memnew( OptionButton ); type_list->set_h_size_flags(SIZE_EXPAND_FILL); add_bind_hb->add_child(type_list); type_list->add_item("bool",Variant::BOOL); type_list->add_item("int",Variant::INT); type_list->add_item("real",Variant::REAL); type_list->add_item("string",Variant::STRING); //type_list->add_separator(); type_list->add_item("Vector2",Variant::VECTOR2); type_list->add_item("Rect2",Variant::RECT2); type_list->add_item("Vector3",Variant::VECTOR3); type_list->add_item("Plane",Variant::PLANE); type_list->add_item("Quat",Variant::QUAT); type_list->add_item("AABB",Variant::_AABB); type_list->add_item("Matrix3",Variant::MATRIX3); type_list->add_item("Transform",Variant::TRANSFORM); //type_list->add_separator(); type_list->add_item("Color",Variant::COLOR); type_list->add_item("Image",Variant::IMAGE); type_list->select(0); Button *add_bind = memnew( Button ); add_bind->set_text(TTR("Add")); add_bind_hb->add_child(add_bind); add_bind->connect("pressed",this,"_add_bind"); Button *del_bind = memnew( Button ); del_bind->set_text(TTR("Remove")); add_bind_hb->add_child(del_bind); del_bind->connect("pressed",this,"_remove_bind"); vbc_right->add_margin_child(TTR("Add Extra Call Argument:"),add_bind_hb); bind_editor = memnew( PropertyEditor ); bind_editor->hide_top_label(); vbc_right->add_margin_child(TTR("Extra Call Arguments:"),bind_editor,true); dst_path = memnew(LineEdit); vbc->add_margin_child(TTR("Path to Node:"),dst_path); HBoxContainer *dstm_hb = memnew( HBoxContainer ); vbc->add_margin_child("Method In Node:",dstm_hb); dst_method = memnew(LineEdit); dst_method->set_h_size_flags(SIZE_EXPAND_FILL); dstm_hb->add_child(dst_method); /*dst_method_list = memnew( MenuButton ); dst_method_list->set_text("List.."); dst_method_list->set_anchor( MARGIN_RIGHT, ANCHOR_END ); dst_method_list->set_anchor( MARGIN_LEFT, ANCHOR_END ); dst_method_list->set_anchor( MARGIN_TOP, ANCHOR_END ); dst_method_list->set_anchor( MARGIN_BOTTOM, ANCHOR_END ); dst_method_list->set_begin( Point2( 70,59) ); dst_method_list->set_end( Point2( 15,39 ) ); */ //add_child(dst_method_list); make_callback = memnew( CheckButton ); make_callback->set_toggle_mode(true); make_callback->set_pressed( EDITOR_DEF("text_editor/create_signal_callbacks",true)); make_callback->set_text(TTR("Make Function")); dstm_hb->add_child(make_callback); deferred = memnew( CheckButton ); deferred->set_text(TTR("Deferred")); dstm_hb->add_child(deferred); oneshot = memnew( CheckButton ); oneshot->set_text(TTR("Oneshot")); dstm_hb->add_child(oneshot); /* realtime = memnew( CheckButton ); realtime->set_anchor( MARGIN_TOP, ANCHOR_END ); realtime->set_anchor( MARGIN_BOTTOM, ANCHOR_END ); realtime->set_anchor( MARGIN_RIGHT, ANCHOR_END ); realtime->set_begin( Point2( 120, button_margin-10 ) ); realtime->set_end( Point2( 80, margin ) ); realtime->set_text("Realtime"); add_child(realtime); */ // dst_method_list->get_popup()->connect("item_pressed", this,"_dst_method_list_selected"); tree->connect("node_selected", this,"_tree_node_selected"); set_as_toplevel(true); cdbinds = memnew( ConnectDialogBinds ); error = memnew( ConfirmationDialog ); add_child(error); error->get_ok()->set_text(TTR("Close")); get_ok()->set_text(TTR("Connect")); // error->get_cancel()->set_text("Close"); } ConnectDialog::~ConnectDialog() { memdelete( cdbinds ); } void ConnectionsDock::_notification(int p_what) { if (p_what==NOTIFICATION_DRAW) { //RID ci = get_canvas_item(); //get_stylebox("panel","PopupMenu")->draw(ci,Rect2(Point2(),get_size())); } } void ConnectionsDock::_close() { hide(); } void ConnectionsDock::_connect() { TreeItem *it = tree->get_selected(); ERR_FAIL_COND(!it); String signal=it->get_metadata(0).operator Dictionary()["name"]; NodePath dst_path=connect_dialog->get_dst_path(); Node *target = node->get_node(dst_path); ERR_FAIL_COND(!target); StringName dst_method=connect_dialog->get_dst_method(); bool defer=connect_dialog->get_deferred(); bool oshot=connect_dialog->get_oneshot(); Vector binds = connect_dialog->get_binds(); StringArray args = it->get_metadata(0).operator Dictionary()["args"]; int flags = CONNECT_PERSIST | (defer?CONNECT_DEFERRED:0) | (oshot?CONNECT_ONESHOT:0); undo_redo->create_action(vformat(TTR("Connect '%s' to '%s'"),signal,String(dst_method))); undo_redo->add_do_method(node,"connect",signal,target,dst_method,binds,flags); undo_redo->add_undo_method(node,"disconnect",signal,target,dst_method); undo_redo->add_do_method(this,"update_tree"); undo_redo->add_undo_method(this,"update_tree"); undo_redo->add_do_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(),"update_tree"); //to force redraw of scene tree undo_redo->add_undo_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(),"update_tree"); //to force redraw of scene tree undo_redo->commit_action(); if (connect_dialog->get_make_callback()) { print_line("request connect"); editor->emit_signal("script_add_function_request",target,dst_method,args); hide(); } update_tree(); } void ConnectionsDock::_connect_pressed() { TreeItem *item = tree->get_selected(); if (!item) { //no idea how this happened, but disable connect_button->set_disabled(true); return; } if (item->get_parent()==tree->get_root() || item->get_parent()->get_parent()==tree->get_root()) { //a signal - connect String signal=item->get_metadata(0).operator Dictionary()["name"]; String signalname=signal; String midname=node->get_name(); for(int i=0;i='a' && c<='z') || (c>='A' && c<='Z') || (c>='0' && c<='9') || c=='_') { //all good } else if (c==' ') { c='_'; } else { midname.remove(i); i--; continue; } midname[i]=c; } connect_dialog->edit(node); connect_dialog->popup_centered_ratio(); connect_dialog->set_title(TTR("Connecting Signal:")+" "+signalname); connect_dialog->set_dst_method("_on_"+midname+"_"+signal); connect_dialog->set_dst_node(node->get_owner()?node->get_owner():node); } else { //a slot- disconnect Connection c=item->get_metadata(0); ERR_FAIL_COND(c.source!=node); //shouldn't happen but...bugcheck undo_redo->create_action(TTR("Create Subscription")); undo_redo->add_do_method(node,"disconnect",c.signal,c.target,c.method); undo_redo->add_undo_method(node,"connect",c.signal,c.target,c.method,Vector(),c.flags); undo_redo->add_do_method(this,"update_tree"); undo_redo->add_undo_method(this,"update_tree"); undo_redo->add_do_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(),"update_tree"); //to force redraw of scene tree undo_redo->add_undo_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(),"update_tree"); //to force redraw of scene tree undo_redo->commit_action(); c.source->disconnect(c.signal,c.target,c.method); update_tree(); } } /* void ConnectionsDock::_remove() { if (!tree->get_selected()) return; TreeItem *selected=tree->get_selected(); if (!selected) return; Dictionary meta=selected->get_metadata(0); remove_confirm->set_text(String()+"Remove Connection \""+meta["from_event"].operator String()+"\" ?"); remove_confirm->popup_centered(Size2(340,80)); } */ /* void ConnectionsDock::_remove_confirm() { if (!tree->get_selected()) return; TreeItem *selected=tree->get_selected(); if (!selected) return; Dictionary meta=selected->get_metadata(0); undo_redo->create_action("Remove Subscription"); undo_redo->add_do_method(node,"unsubscribe_path_event",meta["from_event"].operator String(),meta["from_path"].operator NodePath(),meta["to_method"].operator String()); undo_redo->add_undo_method(node,"subscribe_path_event_persist",meta["from_event"].operator String(),meta["from_path"].operator NodePath(),meta["to_method"].operator String(),Array(),false); undo_redo->add_do_method(this,"update_tree"); undo_redo->add_undo_method(this,"update_tree"); undo_redo->commit_action(); } */ struct _ConnectionsDockMethodInfoSort { _FORCE_INLINE_ bool operator()(const MethodInfo& a, const MethodInfo& b) const { return a.name < b.name; } }; void ConnectionsDock::update_tree() { tree->clear(); if (!node) return; TreeItem *root=tree->create_item(); List node_signals; node->get_signal_list(&node_signals); //node_signals.sort_custom<_ConnectionsDockMethodInfoSort>(); bool did_script=false; StringName base = node->get_type(); while(base) { List node_signals; Ref icon; String name; if (!did_script) { Ref