From 85670726fdf840da22d3ab1fe55de4162e9289df Mon Sep 17 00:00:00 2001
From: "K. S. Ernest (iFire) Lee" <ernest.lee@chibifire.com>
Date: Sat, 12 May 2018 20:34:35 -0700
Subject: Improve VisualScript UX

* Prototype faster function call ux.
* Work on general search ux.
* Able to create nodes from search.
* Show class for variables but not methods.
* Get actions search working.
* Descriptions now show for both methods and properties.
* Enable zooming on mouse wheel up and down.
* Make the drag trigger on right mouse button.
* Search now shows for action visual script nodes.
* Able to search visual node names.
* Search works better.
* Change zooming scale to hide artifacts better.
* Remove zoom changes
* Select from base should check properties too like the other functions.
* Seq_connect flag is needed to set sequence lines correctly.
* Remove comment
* Code cleanup with function names and arguments.
* Use brief description for search descriptions.
* Clean and fix bug with input nodes connecting with sequence lines.
* Add a warning and fix some edge conditions with sequence into data lines and vice versa.
* Don't search functions when pulling from a sequence node.
* Don't show actions when pulling from a data line.
* Set set and get properties.
* Convert visual script operators to the correct type
* Create a function preset finds only functions.
* Singletons can now find functions.
* Add shift-a for generic search.
* Add brief descriptions for Visual Script nodes.
* Search boxes can now filter names.
* Add bigger hit zones to node connect.
* For the drop zones, make all the rect2 areas the same size.
* Function names in visual script node should be lower case so that search works better.
* Use the convention of capitalize() for set, set, visual script nodes and methods.
* Make search more general. Ignore "_" and make case-insensitive. Also made the search window smaller and remove extra info from search
* Make type_cast use the connecting node's type and remove use of found variable.
* Fix case where you call an instance's call function where it becomes an invalid call.
* Make get_visual_node_names use a set of filters, move action creations and fix bug with sequence node connections.
* Make the window bigger.
* Make connect_data and connect_seq more robust.
* Add icons to search items.
* Add vs constructors in shift-a menu.
* Operators, builtins and constructors show type name. Fix several problems with port connections.
* In shift-a mode search everything.
* Code cleanup
* Work on autocompleting the type.
  * Use type guess in action creation.
  * Check if type hint string exists in object variables when generating the visual script search.
  * Add the hint to SceneTree.
  * Add original type detection.
  * Make type casting great again. This puts the type casted base type as the data output type string hint.
  * Pass the type in a VisualScriptFunctionCall too.
  * Set the base type correctly in VisualScriptFunctionGet and VisualScriptFunctionSet using hint string.
  * Make sure the instance is passed in VisualScriptPropertySet.
  * Restore search on the node's type.
* Remove dependencies from graph_edit.
* Remove dependencies from property_selector and name the class visual_script_property_selector.
* Extract hot zones into a function.
* Move hot_zones constants into default theme.
* Bigger capture zones.
* Clean messy port_grab_distance variables.
* Remove RMB functionality.
* Remove memory leak on showing visual script descriptions.
* Read the port_grab_distance constants on enter tree and theme changed.
---
 .../visual_script_property_selector.cpp            | 705 +++++++++++++++++++++
 1 file changed, 705 insertions(+)
 create mode 100644 modules/visual_script/visual_script_property_selector.cpp

(limited to 'modules/visual_script/visual_script_property_selector.cpp')

diff --git a/modules/visual_script/visual_script_property_selector.cpp b/modules/visual_script/visual_script_property_selector.cpp
new file mode 100644
index 0000000000..994ea1f791
--- /dev/null
+++ b/modules/visual_script/visual_script_property_selector.cpp
@@ -0,0 +1,705 @@
+/*************************************************************************/
+/*  visual_script_property_selector.cpp                                  */
+/*************************************************************************/
+/*                       This file is part of:                           */
+/*                           GODOT ENGINE                                */
+/*                      https://godotengine.org                          */
+/*************************************************************************/
+/* Copyright (c) 2007-2018 Juan Linietsky, Ariel Manzur.                 */
+/* Copyright (c) 2014-2018 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 "visual_script_property_selector.h"
+
+#include "editor_scale.h"
+#include "modules/visual_script/visual_script.h"
+#include "modules/visual_script/visual_script_builtin_funcs.h"
+#include "modules/visual_script/visual_script_flow_control.h"
+#include "modules/visual_script/visual_script_func_nodes.h"
+#include "modules/visual_script/visual_script_nodes.h"
+#include "os/keyboard.h"
+#include "scene/main/node.h"
+#include "scene/main/viewport.h"
+
+void VisualScriptPropertySelector::_text_changed(const String &p_newtext) {
+	_update_search();
+}
+
+void VisualScriptPropertySelector::_sbox_input(const Ref<InputEvent> &p_ie) {
+
+	Ref<InputEventKey> k = p_ie;
+
+	if (k.is_valid()) {
+
+		switch (k->get_scancode()) {
+			case KEY_UP:
+			case KEY_DOWN:
+			case KEY_PAGEUP:
+			case KEY_PAGEDOWN: {
+
+				search_options->call("_gui_input", k);
+				search_box->accept_event();
+
+				TreeItem *root = search_options->get_root();
+				if (!root->get_children())
+					break;
+
+				TreeItem *current = search_options->get_selected();
+
+				TreeItem *item = search_options->get_next_selected(root);
+				while (item) {
+					item->deselect(0);
+					item = search_options->get_next_selected(item);
+				}
+
+				current->select(0);
+
+			} break;
+		}
+	}
+}
+
+void VisualScriptPropertySelector::_update_search() {
+	set_title(TTR("Search VisualScript"));
+
+	search_options->clear();
+	help_bit->set_text("");
+
+	TreeItem *root = search_options->create_item();
+	bool found = false;
+
+	if (properties) {
+
+		List<PropertyInfo> props;
+
+		if (instance) {
+			instance->get_property_list(&props, true);
+		} else if (type != Variant::NIL) {
+			Variant v;
+			Variant::CallError ce;
+			v = Variant::construct(type, NULL, 0, ce);
+
+			v.get_property_list(&props);
+		} else {
+
+			Object *obj = ObjectDB::get_instance(script);
+			if (Object::cast_to<Script>(obj)) {
+
+				props.push_back(PropertyInfo(Variant::NIL, "Script Variables", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_CATEGORY));
+				Object::cast_to<Script>(obj)->get_script_property_list(&props);
+			}
+
+			StringName base = base_type;
+			while (base) {
+				props.push_back(PropertyInfo(Variant::NIL, base, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_CATEGORY));
+				ClassDB::get_property_list(base, &props, true);
+				base = ClassDB::get_parent_class_nocheck(base);
+			}
+		}
+
+		TreeItem *category = NULL;
+
+		Ref<Texture> type_icons[Variant::VARIANT_MAX] = {
+			Control::get_icon("Variant", "EditorIcons"),
+			Control::get_icon("bool", "EditorIcons"),
+			Control::get_icon("int", "EditorIcons"),
+			Control::get_icon("float", "EditorIcons"),
+			Control::get_icon("String", "EditorIcons"),
+			Control::get_icon("Vector2", "EditorIcons"),
+			Control::get_icon("Rect2", "EditorIcons"),
+			Control::get_icon("Vector3", "EditorIcons"),
+			Control::get_icon("Transform2D", "EditorIcons"),
+			Control::get_icon("Plane", "EditorIcons"),
+			Control::get_icon("Quat", "EditorIcons"),
+			Control::get_icon("AABB", "EditorIcons"),
+			Control::get_icon("Basis", "EditorIcons"),
+			Control::get_icon("Transform", "EditorIcons"),
+			Control::get_icon("Color", "EditorIcons"),
+			Control::get_icon("Path", "EditorIcons"),
+			Control::get_icon("RID", "EditorIcons"),
+			Control::get_icon("Object", "EditorIcons"),
+			Control::get_icon("Dictionary", "EditorIcons"),
+			Control::get_icon("Array", "EditorIcons"),
+			Control::get_icon("PoolByteArray", "EditorIcons"),
+			Control::get_icon("PoolIntArray", "EditorIcons"),
+			Control::get_icon("PoolRealArray", "EditorIcons"),
+			Control::get_icon("PoolStringArray", "EditorIcons"),
+			Control::get_icon("PoolVector2Array", "EditorIcons"),
+			Control::get_icon("PoolVector3Array", "EditorIcons"),
+			Control::get_icon("PoolColorArray", "EditorIcons")
+		};
+
+		if (!seq_connect && visual_script_generic == false) {
+			get_visual_node_names("flow_control/type_cast", Set<String>(), found, root, search_box);
+			get_visual_node_names("functions/built_in/print", Set<String>(), found, root, search_box);
+			get_visual_node_names("functions/by_type/" + Variant::get_type_name(type), Set<String>(), found, root, search_box);
+			get_visual_node_names("operators/compare/", Set<String>(), found, root, search_box);
+			if (type == Variant::INT) {
+				get_visual_node_names("operators/bitwise/", Set<String>(), found, root, search_box);
+			}
+			if (type == Variant::BOOL) {
+				get_visual_node_names("operators/logic/", Set<String>(), found, root, search_box);
+			}
+			if (type == Variant::BOOL || type == Variant::INT || type == Variant::REAL || type == Variant::VECTOR2 || type == Variant::VECTOR3) {
+				get_visual_node_names("operators/math/", Set<String>(), found, root, search_box);
+			}
+		}
+
+		for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
+			if (E->get().usage == PROPERTY_USAGE_CATEGORY) {
+				if (category && category->get_children() == NULL) {
+					memdelete(category); //old category was unused
+				}
+				category = search_options->create_item(root);
+				category->set_text(0, E->get().name);
+				category->set_selectable(0, false);
+
+				Ref<Texture> icon;
+				if (E->get().name == "Script Variables") {
+					icon = get_icon("Script", "EditorIcons");
+				} else if (has_icon(E->get().name, "EditorIcons")) {
+					icon = get_icon(E->get().name, "EditorIcons");
+				} else {
+					icon = get_icon("Object", "EditorIcons");
+				}
+				category->set_icon(0, icon);
+				continue;
+			}
+
+			if (!(E->get().usage & PROPERTY_USAGE_EDITOR) && !(E->get().usage & PROPERTY_USAGE_SCRIPT_VARIABLE))
+				continue;
+
+			if (type_filter.size() && type_filter.find(E->get().type) == -1)
+				continue;
+
+			String get_text_raw = String(TTR("Get")) + String(" ") + E->get().name;
+			String get_text = get_text_raw.capitalize();
+
+			String set_text_raw = String(TTR("Set ")) + String(" ") + E->get().name;
+			String set_text = set_text_raw.capitalize();
+			String input = search_box->get_text().capitalize();
+			if (input == String() ||
+					get_text_raw.findn(input) != -1 ||
+					get_text.findn(input) != -1) {
+				TreeItem *item = search_options->create_item(category ? category : root);
+				item->set_text(0, get_text);
+				item->set_metadata(0, E->get().name);
+				item->set_icon(0, type_icons[E->get().type]);
+				item->set_metadata(1, "get");
+				item->set_collapsed(1);
+				item->set_selectable(1, false);
+				item->set_selectable(0, true);
+			}
+
+			if (input == String() ||
+					set_text_raw.findn(input) != -1 &&
+							set_text.findn(input) != -1) {
+				TreeItem *item = search_options->create_item(category ? category : root);
+				item->set_text(0, set_text);
+				item->set_metadata(0, E->get().name);
+				item->set_icon(0, type_icons[E->get().type]);
+				item->set_metadata(1, "set");
+				item->set_selectable(1, false);
+				item->set_selectable(0, true);
+			}
+		}
+	}
+
+	if (seq_connect == true && visual_script_generic == false) {
+		String text = search_box->get_text();
+		create_visualscript_item(String("VisualScriptCondition"), root, text, String("Condition"));
+		create_visualscript_item(String("VisualScriptSwitch"), root, text, String("Switch"));
+		create_visualscript_item(String("VisualScriptSequence"), root, text, String("Sequence"));
+		create_visualscript_item(String("VisualScriptIterator"), root, text, String("Iterator"));
+		create_visualscript_item(String("VisualScriptWhile"), root, text, String("While"));
+		create_visualscript_item(String("VisualScriptReturn"), root, text, String("Return"));
+		get_visual_node_names("flow_control/type_cast", Set<String>(), found, root, search_box);
+		get_visual_node_names("functions/built_in/print", Set<String>(), found, root, search_box);
+	}
+
+	if (visual_script_generic) {
+		get_visual_node_names("", Set<String>(), found, root, search_box);
+	}
+
+	List<MethodInfo> methods;
+
+	if (type != Variant::NIL) {
+		Variant v;
+		Variant::CallError ce;
+		v = Variant::construct(type, NULL, 0, ce);
+		v.get_method_list(&methods);
+	} else {
+
+		Object *obj = ObjectDB::get_instance(script);
+		if (Object::cast_to<Script>(obj)) {
+
+			methods.push_back(MethodInfo("*Script Methods"));
+			Object::cast_to<Script>(obj)->get_script_method_list(&methods);
+		}
+
+		StringName base = base_type;
+		while (base) {
+			methods.push_back(MethodInfo("*" + String(base)));
+			ClassDB::get_method_list(base, &methods, true, true);
+			base = ClassDB::get_parent_class_nocheck(base);
+		}
+	}
+	TreeItem *category = NULL;
+	bool script_methods = false;
+
+	for (List<MethodInfo>::Element *E = methods.front(); E; E = E->next()) {
+		if (E->get().name.begins_with("*")) {
+			if (category && category->get_children() == NULL) {
+				memdelete(category); //old category was unused
+			}
+			category = search_options->create_item(root);
+			category->set_text(0, E->get().name.replace_first("*", ""));
+			category->set_selectable(0, false);
+
+			Ref<Texture> icon;
+			script_methods = false;
+			print_line("name: " + E->get().name);
+			String rep = E->get().name.replace("*", "");
+			if (E->get().name == "*Script Methods") {
+				icon = get_icon("Script", "EditorIcons");
+				script_methods = true;
+			} else if (has_icon(rep, "EditorIcons")) {
+				icon = get_icon(rep, "EditorIcons");
+			} else {
+				icon = get_icon("Object", "EditorIcons");
+			}
+			category->set_icon(0, icon);
+
+			continue;
+		}
+
+		String name = E->get().name.get_slice(":", 0);
+		if (!script_methods && name.begins_with("_") && !(E->get().flags & METHOD_FLAG_VIRTUAL))
+			continue;
+
+		if (virtuals_only && !(E->get().flags & METHOD_FLAG_VIRTUAL))
+			continue;
+
+		if (!virtuals_only && (E->get().flags & METHOD_FLAG_VIRTUAL))
+			continue;
+
+		MethodInfo mi = E->get();
+		String desc = mi.name.capitalize() + " (";
+
+		if (search_box->get_text() != String() &&
+				name.findn(search_box->get_text()) == -1 &&
+				desc.findn(search_box->get_text()) == -1)
+			continue;
+
+		TreeItem *item = search_options->create_item(category ? category : root);
+
+		for (int i = 0; i < mi.arguments.size(); i++) {
+
+			if (i > 0)
+				desc += ", ";
+
+			if (mi.arguments[i].type == Variant::NIL)
+				desc += "var";
+			else if (mi.arguments[i].name.find(":") != -1) {
+				desc += mi.arguments[i].name.get_slice(":", 1);
+				mi.arguments[i].name = mi.arguments[i].name.get_slice(":", 0);
+			} else
+				desc += Variant::get_type_name(mi.arguments[i].type);
+		}
+
+		desc += ")";
+
+		item->set_text(0, desc);
+		item->set_icon(0, get_icon("MemberMethod", "EditorIcons"));
+		item->set_metadata(0, name);
+		item->set_selectable(0, true);
+
+		item->set_metadata(1, "method");
+		item->set_collapsed(1);
+		item->set_selectable(1, false);
+
+		if (category && category->get_children() == NULL) {
+			memdelete(category); //old category was unused
+		}
+	}
+
+	TreeItem *selected_item = search_options->search_item_text(search_box->get_text());
+	if (!found && selected_item != NULL) {
+		selected_item->select(0);
+		found = true;
+	}
+
+	get_ok()->set_disabled(root->get_children() == NULL);
+}
+
+void VisualScriptPropertySelector::create_visualscript_item(const String &name, TreeItem *const root, const String &search_input, const String &text) {
+	if (search_input == String() || text.findn(search_input) != -1) {
+		TreeItem *item = search_options->create_item(root);
+		item->set_text(0, text);
+		item->set_icon(0, get_icon("VisualScript", "EditorIcons"));
+		item->set_metadata(0, name);
+		item->set_metadata(1, "action");
+		item->set_selectable(0, true);
+		item->set_collapsed(1);
+		item->set_selectable(1, false);
+	}
+}
+
+void VisualScriptPropertySelector::get_visual_node_names(const String &root_filter, const Set<String> &filter, bool &found, TreeItem *const root, LineEdit *const search_box) {
+	Map<String, TreeItem *> path_cache;
+
+	List<String> fnodes;
+	VisualScriptLanguage::singleton->get_registered_node_names(&fnodes);
+
+	for (List<String>::Element *E = fnodes.front(); E; E = E->next()) {
+		if (!E->get().begins_with(root_filter)) {
+			continue;
+		}
+		Vector<String> path = E->get().split("/");
+		bool is_filter = false;
+		for (Set<String>::Element *E = filter.front(); E; E = E->next()) {
+			if (path.size() >= 2 && path[1].findn(E->get()) != -1) {
+				is_filter = true;
+				break;
+			}
+		}
+		if (is_filter == true) {
+			continue;
+		}
+
+		if (search_box->get_text() != String() && E->get().findn(search_box->get_text()) == -1) {
+			continue;
+		}
+		TreeItem *item = search_options->create_item(root);
+		VisualScriptOperator *vnode_operator = Object::cast_to<VisualScriptOperator>(*VisualScriptLanguage::singleton->create_node_from_name(E->get()));
+		String type_name;
+		if (vnode_operator != NULL) {
+			String type;
+			if (path.size() >= 2) {
+				type = path[1];
+			}
+			type_name = type.capitalize() + " ";
+		}
+		VisualScriptFunctionCall *vnode_function_call = Object::cast_to<VisualScriptFunctionCall>(*VisualScriptLanguage::singleton->create_node_from_name(E->get()));
+		if (vnode_function_call != NULL) {
+			String basic_type = Variant::get_type_name(vnode_function_call->get_basic_type());
+			type_name = basic_type.capitalize() + " ";
+		}
+		VisualScriptBuiltinFunc *vnode_builtin_function_call = Object::cast_to<VisualScriptBuiltinFunc>(*VisualScriptLanguage::singleton->create_node_from_name(E->get()));
+		if (vnode_builtin_function_call != NULL) {
+			type_name = "Builtin ";
+		}
+		item->set_text(0, type_name + path[path.size() - 1].capitalize());
+		item->set_icon(0, get_icon("VisualScript", "EditorIcons"));
+		item->set_selectable(0, true);
+		item->set_metadata(0, E->get());
+		item->set_selectable(0, true);
+		item->set_metadata(1, "visualscript");
+		item->set_selectable(1, false);
+	}
+}
+
+void VisualScriptPropertySelector::_confirmed() {
+
+	TreeItem *ti = search_options->get_selected();
+	if (!ti)
+		return;
+	emit_signal("selected", ti->get_metadata(0), ti->get_metadata(1));
+	hide();
+}
+
+void VisualScriptPropertySelector::_item_selected() {
+
+	help_bit->set_text("");
+
+	TreeItem *item = search_options->get_selected();
+	if (!item)
+		return;
+	String name = item->get_metadata(0);
+
+	String class_type;
+	if (type) {
+		class_type = Variant::get_type_name(type);
+
+	} else {
+		class_type = base_type;
+	}
+
+	DocData *dd = EditorHelp::get_doc_data();
+	String text;
+
+	String at_class = class_type;
+
+	while (at_class != String()) {
+
+		Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(at_class);
+		if (E) {
+			for (int i = 0; i < E->get().properties.size(); i++) {
+				if (E->get().properties[i].name == name) {
+					text = E->get().properties[i].description;
+				}
+			}
+		}
+
+		at_class = ClassDB::get_parent_class_nocheck(at_class);
+	}
+	at_class = class_type;
+
+	while (at_class != String()) {
+
+		Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(at_class);
+		if (E) {
+			for (int i = 0; i < E->get().methods.size(); i++) {
+				if (E->get().methods[i].name == name) {
+					text = E->get().methods[i].description;
+				}
+			}
+		}
+
+		at_class = ClassDB::get_parent_class_nocheck(at_class);
+	}
+	Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(class_type);
+	if (E) {
+		for (int i = 0; i < E->get().methods.size(); i++) {
+			Vector<String> functions = name.rsplit("/", false, 1);
+			if (E->get().methods[i].name == functions[functions.size() - 1]) {
+				text = E->get().methods[i].description;
+			}
+		}
+	}
+
+	List<String> *names = memnew(List<String>);
+	VisualScriptLanguage::singleton->get_registered_node_names(names);
+	if (names->find(name) != NULL) {
+		Ref<VisualScriptOperator> operator_node = VisualScriptLanguage::singleton->create_node_from_name(name);
+		if (operator_node.is_valid()) {
+			Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(operator_node->get_class_name());
+			if (E) {
+				text = Variant::get_operator_name(operator_node->get_operator());
+			}
+		}
+		Ref<VisualScriptTypeCast> typecast_node = VisualScriptLanguage::singleton->create_node_from_name(name);
+		if (typecast_node.is_valid()) {
+			Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(typecast_node->get_class_name());
+			if (E) {
+				text = E->get().description;
+			}
+		}
+
+		Ref<VisualScriptBuiltinFunc> builtin_node = VisualScriptLanguage::singleton->create_node_from_name(name);
+		if (builtin_node.is_valid()) {
+			Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(builtin_node->get_class_name());
+			if (E) {
+				for (int i = 0; i < E->get().constants.size(); i++) {
+					if (E->get().constants[i].value.to_int() == int(builtin_node->get_func())) {
+						text = E->get().constants[i].description;
+					}
+				}
+			}
+		}
+	}
+
+	memdelete(names);
+
+	if (text == String())
+		return;
+
+	help_bit->set_text(text);
+}
+
+void VisualScriptPropertySelector::_notification(int p_what) {
+
+	if (p_what == NOTIFICATION_ENTER_TREE) {
+
+		connect("confirmed", this, "_confirmed");
+	}
+}
+
+void VisualScriptPropertySelector::select_method_from_base_type(const String &p_base, const String &p_current, bool p_virtuals_only) {
+
+	base_type = p_base;
+	selected = p_current;
+	type = Variant::NIL;
+	script = 0;
+	properties = false;
+	instance = NULL;
+	virtuals_only = p_virtuals_only;
+
+	show_window(.5f);
+	search_box->set_text("");
+	search_box->grab_focus();
+	_update_search();
+}
+
+void VisualScriptPropertySelector::set_type_filter(const Vector<Variant::Type> &p_type_filter) {
+	type_filter = p_type_filter;
+}
+
+void VisualScriptPropertySelector::select_from_base_type(const String &p_base, const String &p_current /*= ""*/, bool p_virtuals_only /*= false*/, bool p_seq_connect /*= false*/) {
+
+	base_type = p_base;
+	selected = p_current;
+	type = Variant::NIL;
+	script = 0;
+	properties = true;
+	instance = NULL;
+	virtuals_only = p_virtuals_only;
+
+	show_window(.5f);
+	search_box->set_text("");
+	search_box->grab_focus();
+	seq_connect = p_seq_connect;
+
+	_update_search();
+}
+
+void VisualScriptPropertySelector::select_from_script(const Ref<Script> &p_script, const String &p_current /*= ""*/) {
+	ERR_FAIL_COND(p_script.is_null());
+
+	base_type = p_script->get_instance_base_type();
+	selected = p_current;
+	type = Variant::NIL;
+	script = p_script->get_instance_id();
+	properties = true;
+	instance = NULL;
+	virtuals_only = false;
+
+	show_window(.5f);
+	search_box->set_text("");
+	search_box->grab_focus();
+	seq_connect = false;
+
+	_update_search();
+}
+
+void VisualScriptPropertySelector::select_from_basic_type(Variant::Type p_type, const String &p_current /*= ""*/) {
+	ERR_FAIL_COND(p_type == Variant::NIL);
+	base_type = "";
+	selected = p_current;
+	type = p_type;
+	script = 0;
+	properties = true;
+	instance = NULL;
+	virtuals_only = false;
+
+	show_window(.5f);
+	search_box->set_text("");
+	search_box->grab_focus();
+	seq_connect = false;
+
+	_update_search();
+}
+
+void VisualScriptPropertySelector::select_from_action(const String &p_type, const String &p_current /*= ""*/) {
+	base_type = p_type;
+	selected = p_current;
+	type = Variant::NIL;
+	script = 0;
+	properties = false;
+	instance = NULL;
+	virtuals_only = false;
+
+	show_window(.5f);
+	search_box->set_text("");
+	search_box->grab_focus();
+	seq_connect = true;
+	_update_search();
+}
+
+void VisualScriptPropertySelector::select_from_instance(Object *p_instance, const String &p_current /*= ""*/) {
+	base_type = "";
+	selected = p_current;
+	type = Variant::NIL;
+	script = 0;
+	properties = true;
+	instance = p_instance;
+	virtuals_only = false;
+
+	show_window(.5f);
+	search_box->set_text("");
+	search_box->grab_focus();
+	seq_connect = false;
+
+	_update_search();
+}
+
+void VisualScriptPropertySelector::select_from_visual_script(const String &p_base) {
+	base_type = p_base;
+	selected = "";
+	type = Variant::NIL;
+	script = 0;
+	properties = true;
+	visual_script_generic = true;
+	instance = NULL;
+	virtuals_only = false;
+	show_window(.5f);
+	search_box->set_text("");
+	search_box->grab_focus();
+
+	_update_search();
+}
+
+void VisualScriptPropertySelector::show_window(float p_screen_ratio) {
+	Rect2 rect;
+	Point2 window_size = get_viewport_rect().size;
+	rect.size = (window_size * p_screen_ratio).floor();
+	rect.size.x = rect.size.x / 1.25f;
+	rect.position = ((window_size - rect.size) / 2.0f).floor();
+	popup(rect);
+}
+
+void VisualScriptPropertySelector::_bind_methods() {
+
+	ClassDB::bind_method(D_METHOD("_text_changed"), &VisualScriptPropertySelector::_text_changed);
+	ClassDB::bind_method(D_METHOD("_confirmed"), &VisualScriptPropertySelector::_confirmed);
+	ClassDB::bind_method(D_METHOD("_sbox_input"), &VisualScriptPropertySelector::_sbox_input);
+	ClassDB::bind_method(D_METHOD("_item_selected"), &VisualScriptPropertySelector::_item_selected);
+
+	ADD_SIGNAL(MethodInfo("selected", PropertyInfo(Variant::STRING, "name")));
+}
+
+VisualScriptPropertySelector::VisualScriptPropertySelector() {
+
+	VBoxContainer *vbc = memnew(VBoxContainer);
+	add_child(vbc);
+	//set_child_rect(vbc);
+	search_box = memnew(LineEdit);
+	vbc->add_margin_child(TTR("Search:"), search_box);
+	search_box->connect("text_changed", this, "_text_changed");
+	search_box->connect("gui_input", this, "_sbox_input");
+	search_options = memnew(Tree);
+	vbc->add_margin_child(TTR("Matches:"), search_options, true);
+	get_ok()->set_text(TTR("Open"));
+	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);
+	search_options->set_hide_folding(true);
+	virtuals_only = false;
+	help_bit = memnew(EditorHelpBit);
+	vbc->add_margin_child(TTR("Description:"), help_bit);
+	help_bit->connect("request_hide", this, "_closed");
+	search_options->set_columns(2);
+	search_options->set_column_expand(1, false);
+}
-- 
cgit v1.2.3