visual_script_property_selector.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733
  1. /*************************************************************************/
  2. /* visual_script_property_selector.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2018 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2018 Godot Engine contributors (cf. AUTHORS.md) */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /*************************************************************************/
  30. #include "visual_script_property_selector.h"
  31. #include "core/os/keyboard.h"
  32. #include "editor/editor_node.h"
  33. #include "editor_scale.h"
  34. #include "modules/visual_script/visual_script.h"
  35. #include "modules/visual_script/visual_script_builtin_funcs.h"
  36. #include "modules/visual_script/visual_script_flow_control.h"
  37. #include "modules/visual_script/visual_script_func_nodes.h"
  38. #include "modules/visual_script/visual_script_nodes.h"
  39. #include "scene/main/node.h"
  40. #include "scene/main/viewport.h"
  41. void VisualScriptPropertySelector::_text_changed(const String &p_newtext) {
  42. _update_search();
  43. }
  44. void VisualScriptPropertySelector::_sbox_input(const Ref<InputEvent> &p_ie) {
  45. Ref<InputEventKey> k = p_ie;
  46. if (k.is_valid()) {
  47. switch (k->get_scancode()) {
  48. case KEY_UP:
  49. case KEY_DOWN:
  50. case KEY_PAGEUP:
  51. case KEY_PAGEDOWN: {
  52. search_options->call("_gui_input", k);
  53. search_box->accept_event();
  54. TreeItem *root = search_options->get_root();
  55. if (!root->get_children())
  56. break;
  57. TreeItem *current = search_options->get_selected();
  58. TreeItem *item = search_options->get_next_selected(root);
  59. while (item) {
  60. item->deselect(0);
  61. item = search_options->get_next_selected(item);
  62. }
  63. current->select(0);
  64. } break;
  65. }
  66. }
  67. }
  68. void VisualScriptPropertySelector::_update_search() {
  69. set_title(TTR("Search VisualScript"));
  70. search_options->clear();
  71. help_bit->set_text("");
  72. TreeItem *root = search_options->create_item();
  73. bool found = false;
  74. if (properties) {
  75. List<PropertyInfo> props;
  76. if (instance) {
  77. instance->get_property_list(&props, true);
  78. } else if (type != Variant::NIL) {
  79. Variant v;
  80. Variant::CallError ce;
  81. v = Variant::construct(type, NULL, 0, ce);
  82. v.get_property_list(&props);
  83. } else {
  84. Object *obj = ObjectDB::get_instance(script);
  85. if (Object::cast_to<Script>(obj)) {
  86. props.push_back(PropertyInfo(Variant::NIL, "Script Variables", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_CATEGORY));
  87. Object::cast_to<Script>(obj)->get_script_property_list(&props);
  88. }
  89. StringName base = base_type;
  90. while (base) {
  91. props.push_back(PropertyInfo(Variant::NIL, base, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_CATEGORY));
  92. ClassDB::get_property_list(base, &props, true);
  93. base = ClassDB::get_parent_class_nocheck(base);
  94. }
  95. }
  96. TreeItem *category = NULL;
  97. Ref<Texture> type_icons[Variant::VARIANT_MAX] = {
  98. Control::get_icon("Variant", "EditorIcons"),
  99. Control::get_icon("bool", "EditorIcons"),
  100. Control::get_icon("int", "EditorIcons"),
  101. Control::get_icon("float", "EditorIcons"),
  102. Control::get_icon("String", "EditorIcons"),
  103. Control::get_icon("Vector2", "EditorIcons"),
  104. Control::get_icon("Rect2", "EditorIcons"),
  105. Control::get_icon("Vector3", "EditorIcons"),
  106. Control::get_icon("Transform2D", "EditorIcons"),
  107. Control::get_icon("Plane", "EditorIcons"),
  108. Control::get_icon("Quat", "EditorIcons"),
  109. Control::get_icon("AABB", "EditorIcons"),
  110. Control::get_icon("Basis", "EditorIcons"),
  111. Control::get_icon("Transform", "EditorIcons"),
  112. Control::get_icon("Color", "EditorIcons"),
  113. Control::get_icon("Path", "EditorIcons"),
  114. Control::get_icon("RID", "EditorIcons"),
  115. Control::get_icon("Object", "EditorIcons"),
  116. Control::get_icon("Dictionary", "EditorIcons"),
  117. Control::get_icon("Array", "EditorIcons"),
  118. Control::get_icon("PoolByteArray", "EditorIcons"),
  119. Control::get_icon("PoolIntArray", "EditorIcons"),
  120. Control::get_icon("PoolRealArray", "EditorIcons"),
  121. Control::get_icon("PoolStringArray", "EditorIcons"),
  122. Control::get_icon("PoolVector2Array", "EditorIcons"),
  123. Control::get_icon("PoolVector3Array", "EditorIcons"),
  124. Control::get_icon("PoolColorArray", "EditorIcons")
  125. };
  126. if (!seq_connect && !visual_script_generic) {
  127. get_visual_node_names("flow_control/type_cast", Set<String>(), found, root, search_box);
  128. get_visual_node_names("functions/built_in/print", Set<String>(), found, root, search_box);
  129. get_visual_node_names("functions/by_type/" + Variant::get_type_name(type), Set<String>(), found, root, search_box);
  130. get_visual_node_names("operators/compare/", Set<String>(), found, root, search_box);
  131. if (type == Variant::INT) {
  132. get_visual_node_names("operators/bitwise/", Set<String>(), found, root, search_box);
  133. }
  134. if (type == Variant::BOOL) {
  135. get_visual_node_names("operators/logic/", Set<String>(), found, root, search_box);
  136. }
  137. if (type == Variant::BOOL || type == Variant::INT || type == Variant::REAL || type == Variant::VECTOR2 || type == Variant::VECTOR3) {
  138. get_visual_node_names("operators/math/", Set<String>(), found, root, search_box);
  139. }
  140. }
  141. for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
  142. if (E->get().usage == PROPERTY_USAGE_CATEGORY) {
  143. if (category && category->get_children() == NULL) {
  144. memdelete(category); //old category was unused
  145. }
  146. category = search_options->create_item(root);
  147. category->set_text(0, E->get().name);
  148. category->set_selectable(0, false);
  149. Ref<Texture> icon;
  150. if (E->get().name == "Script Variables") {
  151. icon = get_icon("Script", "EditorIcons");
  152. } else {
  153. icon = EditorNode::get_singleton()->get_class_icon(E->get().name);
  154. }
  155. category->set_icon(0, icon);
  156. continue;
  157. }
  158. if (!(E->get().usage & PROPERTY_USAGE_EDITOR) && !(E->get().usage & PROPERTY_USAGE_SCRIPT_VARIABLE))
  159. continue;
  160. if (type_filter.size() && type_filter.find(E->get().type) == -1)
  161. continue;
  162. // capitalize() also converts underscore to space, we'll match again both possible styles
  163. String get_text_raw = String(vformat(TTR("Get %s"), E->get().name));
  164. String get_text = get_text_raw.capitalize();
  165. String set_text_raw = String(vformat(TTR("Set %s"), E->get().name));
  166. String set_text = set_text_raw.capitalize();
  167. String input = search_box->get_text().capitalize();
  168. if (input == String() || get_text_raw.findn(input) != -1 || get_text.findn(input) != -1) {
  169. TreeItem *item = search_options->create_item(category ? category : root);
  170. item->set_text(0, get_text);
  171. item->set_metadata(0, E->get().name);
  172. item->set_icon(0, type_icons[E->get().type]);
  173. item->set_metadata(1, "get");
  174. item->set_collapsed(1);
  175. item->set_selectable(0, true);
  176. item->set_selectable(1, false);
  177. item->set_selectable(2, false);
  178. item->set_metadata(2, connecting);
  179. }
  180. if (input == String() || set_text_raw.findn(input) != -1 || set_text.findn(input) != -1) {
  181. TreeItem *item = search_options->create_item(category ? category : root);
  182. item->set_text(0, set_text);
  183. item->set_metadata(0, E->get().name);
  184. item->set_icon(0, type_icons[E->get().type]);
  185. item->set_metadata(1, "set");
  186. item->set_selectable(0, true);
  187. item->set_selectable(1, false);
  188. item->set_selectable(2, false);
  189. item->set_metadata(2, connecting);
  190. }
  191. }
  192. if (category && category->get_children() == NULL) {
  193. memdelete(category); //old category was unused
  194. }
  195. }
  196. if (seq_connect && !visual_script_generic) {
  197. String text = search_box->get_text();
  198. create_visualscript_item(String("VisualScriptCondition"), root, text, String("Condition"));
  199. create_visualscript_item(String("VisualScriptSwitch"), root, text, String("Switch"));
  200. create_visualscript_item(String("VisualScriptSequence"), root, text, String("Sequence"));
  201. create_visualscript_item(String("VisualScriptIterator"), root, text, String("Iterator"));
  202. create_visualscript_item(String("VisualScriptWhile"), root, text, String("While"));
  203. create_visualscript_item(String("VisualScriptReturn"), root, text, String("Return"));
  204. get_visual_node_names("flow_control/type_cast", Set<String>(), found, root, search_box);
  205. get_visual_node_names("functions/built_in/print", Set<String>(), found, root, search_box);
  206. }
  207. if (visual_script_generic) {
  208. get_visual_node_names("", Set<String>(), found, root, search_box);
  209. }
  210. List<MethodInfo> methods;
  211. if (type != Variant::NIL) {
  212. Variant v;
  213. Variant::CallError ce;
  214. v = Variant::construct(type, NULL, 0, ce);
  215. v.get_method_list(&methods);
  216. } else {
  217. Object *obj = ObjectDB::get_instance(script);
  218. if (Object::cast_to<Script>(obj)) {
  219. methods.push_back(MethodInfo("*Script Methods"));
  220. Object::cast_to<Script>(obj)->get_script_method_list(&methods);
  221. }
  222. StringName base = base_type;
  223. while (base) {
  224. methods.push_back(MethodInfo("*" + String(base)));
  225. ClassDB::get_method_list(base, &methods, true, true);
  226. base = ClassDB::get_parent_class_nocheck(base);
  227. }
  228. }
  229. TreeItem *category = NULL;
  230. bool script_methods = false;
  231. for (List<MethodInfo>::Element *E = methods.front(); E; E = E->next()) {
  232. if (E->get().name.begins_with("*")) {
  233. if (category && category->get_children() == NULL) {
  234. memdelete(category); //old category was unused
  235. }
  236. category = search_options->create_item(root);
  237. category->set_text(0, E->get().name.replace_first("*", ""));
  238. category->set_selectable(0, false);
  239. Ref<Texture> icon;
  240. script_methods = false;
  241. String rep = E->get().name.replace("*", "");
  242. if (E->get().name == "*Script Methods") {
  243. icon = get_icon("Script", "EditorIcons");
  244. script_methods = true;
  245. } else {
  246. icon = EditorNode::get_singleton()->get_class_icon(rep);
  247. }
  248. category->set_icon(0, icon);
  249. continue;
  250. }
  251. String name = E->get().name.get_slice(":", 0);
  252. if (!script_methods && name.begins_with("_") && !(E->get().flags & METHOD_FLAG_VIRTUAL))
  253. continue;
  254. if (virtuals_only && !(E->get().flags & METHOD_FLAG_VIRTUAL))
  255. continue;
  256. if (!virtuals_only && (E->get().flags & METHOD_FLAG_VIRTUAL))
  257. continue;
  258. MethodInfo mi = E->get();
  259. String desc = mi.name.capitalize() + " (";
  260. if (search_box->get_text() != String() &&
  261. name.findn(search_box->get_text()) == -1 &&
  262. desc.findn(search_box->get_text()) == -1)
  263. continue;
  264. TreeItem *item = search_options->create_item(category ? category : root);
  265. for (int i = 0; i < mi.arguments.size(); i++) {
  266. if (i > 0)
  267. desc += ", ";
  268. if (mi.arguments[i].type == Variant::NIL)
  269. desc += "var";
  270. else if (mi.arguments[i].name.find(":") != -1) {
  271. desc += mi.arguments[i].name.get_slice(":", 1);
  272. mi.arguments[i].name = mi.arguments[i].name.get_slice(":", 0);
  273. } else
  274. desc += Variant::get_type_name(mi.arguments[i].type);
  275. }
  276. desc += ")";
  277. item->set_text(0, desc);
  278. item->set_icon(0, get_icon("MemberMethod", "EditorIcons"));
  279. item->set_metadata(0, name);
  280. item->set_selectable(0, true);
  281. item->set_metadata(1, "method");
  282. item->set_collapsed(1);
  283. item->set_selectable(1, false);
  284. item->set_selectable(2, false);
  285. item->set_metadata(2, connecting);
  286. if (category && category->get_children() == NULL) {
  287. memdelete(category); //old category was unused
  288. }
  289. }
  290. TreeItem *selected_item = search_options->search_item_text(search_box->get_text());
  291. if (!found && selected_item != NULL) {
  292. selected_item->select(0);
  293. found = true;
  294. }
  295. if (category && category->get_children() == NULL) {
  296. memdelete(category); //old category was unused
  297. }
  298. get_ok()->set_disabled(root->get_children() == NULL);
  299. }
  300. void VisualScriptPropertySelector::create_visualscript_item(const String &name, TreeItem *const root, const String &search_input, const String &text) {
  301. if (search_input == String() || text.findn(search_input) != -1) {
  302. TreeItem *item = search_options->create_item(root);
  303. item->set_text(0, text);
  304. item->set_icon(0, get_icon("VisualScript", "EditorIcons"));
  305. item->set_metadata(0, name);
  306. item->set_metadata(1, "action");
  307. item->set_selectable(0, true);
  308. item->set_collapsed(1);
  309. item->set_selectable(1, false);
  310. item->set_selectable(2, false);
  311. item->set_metadata(2, connecting);
  312. }
  313. }
  314. void VisualScriptPropertySelector::get_visual_node_names(const String &root_filter, const Set<String> &filter, bool &found, TreeItem *const root, LineEdit *const search_box) {
  315. Map<String, TreeItem *> path_cache;
  316. List<String> fnodes;
  317. VisualScriptLanguage::singleton->get_registered_node_names(&fnodes);
  318. for (List<String>::Element *E = fnodes.front(); E; E = E->next()) {
  319. if (!E->get().begins_with(root_filter)) {
  320. continue;
  321. }
  322. Vector<String> path = E->get().split("/");
  323. bool is_filter = false;
  324. for (Set<String>::Element *F = filter.front(); F; F = F->next()) {
  325. if (path.size() >= 2 && path[1].findn(F->get()) != -1) {
  326. is_filter = true;
  327. break;
  328. }
  329. }
  330. if (is_filter) {
  331. continue;
  332. }
  333. if (search_box->get_text() != String() && E->get().findn(search_box->get_text()) == -1) {
  334. continue;
  335. }
  336. TreeItem *item = search_options->create_item(root);
  337. VisualScriptOperator *vnode_operator = Object::cast_to<VisualScriptOperator>(*VisualScriptLanguage::singleton->create_node_from_name(E->get()));
  338. String type_name;
  339. if (vnode_operator != NULL) {
  340. String type;
  341. if (path.size() >= 2) {
  342. type = path[1];
  343. }
  344. type_name = type.capitalize() + " ";
  345. }
  346. VisualScriptFunctionCall *vnode_function_call = Object::cast_to<VisualScriptFunctionCall>(*VisualScriptLanguage::singleton->create_node_from_name(E->get()));
  347. if (vnode_function_call != NULL) {
  348. String basic_type = Variant::get_type_name(vnode_function_call->get_basic_type());
  349. type_name = basic_type.capitalize() + " ";
  350. }
  351. VisualScriptBuiltinFunc *vnode_builtin_function_call = Object::cast_to<VisualScriptBuiltinFunc>(*VisualScriptLanguage::singleton->create_node_from_name(E->get()));
  352. if (vnode_builtin_function_call != NULL) {
  353. type_name = "Builtin ";
  354. }
  355. item->set_text(0, type_name + path[path.size() - 1].capitalize());
  356. item->set_icon(0, get_icon("VisualScript", "EditorIcons"));
  357. item->set_selectable(0, true);
  358. item->set_metadata(0, E->get());
  359. item->set_selectable(0, true);
  360. item->set_metadata(1, "visualscript");
  361. item->set_selectable(1, false);
  362. item->set_selectable(2, false);
  363. item->set_metadata(2, connecting);
  364. }
  365. }
  366. void VisualScriptPropertySelector::_confirmed() {
  367. TreeItem *ti = search_options->get_selected();
  368. if (!ti)
  369. return;
  370. emit_signal("selected", ti->get_metadata(0), ti->get_metadata(1), ti->get_metadata(2));
  371. hide();
  372. }
  373. void VisualScriptPropertySelector::_item_selected() {
  374. help_bit->set_text("");
  375. TreeItem *item = search_options->get_selected();
  376. if (!item)
  377. return;
  378. String name = item->get_metadata(0);
  379. String class_type;
  380. if (type) {
  381. class_type = Variant::get_type_name(type);
  382. } else {
  383. class_type = base_type;
  384. }
  385. DocData *dd = EditorHelp::get_doc_data();
  386. String text;
  387. String at_class = class_type;
  388. while (at_class != String()) {
  389. Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(at_class);
  390. if (E) {
  391. for (int i = 0; i < E->get().properties.size(); i++) {
  392. if (E->get().properties[i].name == name) {
  393. text = E->get().properties[i].description;
  394. }
  395. }
  396. }
  397. at_class = ClassDB::get_parent_class_nocheck(at_class);
  398. }
  399. at_class = class_type;
  400. while (at_class != String()) {
  401. Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(at_class);
  402. if (E) {
  403. for (int i = 0; i < E->get().methods.size(); i++) {
  404. if (E->get().methods[i].name == name) {
  405. text = E->get().methods[i].description;
  406. }
  407. }
  408. }
  409. at_class = ClassDB::get_parent_class_nocheck(at_class);
  410. }
  411. Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(class_type);
  412. if (E) {
  413. for (int i = 0; i < E->get().methods.size(); i++) {
  414. Vector<String> functions = name.rsplit("/", false, 1);
  415. if (E->get().methods[i].name == functions[functions.size() - 1]) {
  416. text = E->get().methods[i].description;
  417. }
  418. }
  419. }
  420. List<String> *names = memnew(List<String>);
  421. VisualScriptLanguage::singleton->get_registered_node_names(names);
  422. if (names->find(name) != NULL) {
  423. Ref<VisualScriptOperator> operator_node = VisualScriptLanguage::singleton->create_node_from_name(name);
  424. if (operator_node.is_valid()) {
  425. Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(operator_node->get_class_name());
  426. if (E) {
  427. text = Variant::get_operator_name(operator_node->get_operator());
  428. }
  429. }
  430. Ref<VisualScriptTypeCast> typecast_node = VisualScriptLanguage::singleton->create_node_from_name(name);
  431. if (typecast_node.is_valid()) {
  432. Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(typecast_node->get_class_name());
  433. if (E) {
  434. text = E->get().description;
  435. }
  436. }
  437. Ref<VisualScriptBuiltinFunc> builtin_node = VisualScriptLanguage::singleton->create_node_from_name(name);
  438. if (builtin_node.is_valid()) {
  439. Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(builtin_node->get_class_name());
  440. if (E) {
  441. for (int i = 0; i < E->get().constants.size(); i++) {
  442. if (E->get().constants[i].value.to_int() == int(builtin_node->get_func())) {
  443. text = E->get().constants[i].description;
  444. }
  445. }
  446. }
  447. }
  448. }
  449. memdelete(names);
  450. if (text == String())
  451. return;
  452. help_bit->set_text(text);
  453. }
  454. void VisualScriptPropertySelector::_notification(int p_what) {
  455. if (p_what == NOTIFICATION_ENTER_TREE) {
  456. connect("confirmed", this, "_confirmed");
  457. }
  458. }
  459. void VisualScriptPropertySelector::select_method_from_base_type(const String &p_base, const String &p_current, const bool p_virtuals_only, const bool p_connecting) {
  460. base_type = p_base;
  461. selected = p_current;
  462. type = Variant::NIL;
  463. script = 0;
  464. properties = false;
  465. instance = NULL;
  466. virtuals_only = p_virtuals_only;
  467. show_window(.5f);
  468. search_box->set_text("");
  469. search_box->grab_focus();
  470. connecting = p_connecting;
  471. _update_search();
  472. }
  473. void VisualScriptPropertySelector::set_type_filter(const Vector<Variant::Type> &p_type_filter) {
  474. type_filter = p_type_filter;
  475. }
  476. void VisualScriptPropertySelector::select_from_base_type(const String &p_base, const String &p_current, bool p_virtuals_only, bool p_seq_connect, const bool p_connecting) {
  477. base_type = p_base;
  478. selected = p_current;
  479. type = Variant::NIL;
  480. script = 0;
  481. properties = true;
  482. visual_script_generic = false;
  483. instance = NULL;
  484. virtuals_only = p_virtuals_only;
  485. show_window(.5f);
  486. search_box->set_text("");
  487. search_box->grab_focus();
  488. seq_connect = p_seq_connect;
  489. connecting = p_connecting;
  490. _update_search();
  491. }
  492. void VisualScriptPropertySelector::select_from_script(const Ref<Script> &p_script, const String &p_current, const bool p_connecting) {
  493. ERR_FAIL_COND(p_script.is_null());
  494. base_type = p_script->get_instance_base_type();
  495. selected = p_current;
  496. type = Variant::NIL;
  497. script = p_script->get_instance_id();
  498. properties = true;
  499. visual_script_generic = false;
  500. instance = NULL;
  501. virtuals_only = false;
  502. show_window(.5f);
  503. search_box->set_text("");
  504. search_box->grab_focus();
  505. seq_connect = false;
  506. connecting = p_connecting;
  507. _update_search();
  508. }
  509. void VisualScriptPropertySelector::select_from_basic_type(Variant::Type p_type, const String &p_current, const bool p_connecting) {
  510. ERR_FAIL_COND(p_type == Variant::NIL);
  511. base_type = "";
  512. selected = p_current;
  513. type = p_type;
  514. script = 0;
  515. properties = true;
  516. visual_script_generic = false;
  517. instance = NULL;
  518. virtuals_only = false;
  519. show_window(.5f);
  520. search_box->set_text("");
  521. search_box->grab_focus();
  522. seq_connect = false;
  523. connecting = p_connecting;
  524. _update_search();
  525. }
  526. void VisualScriptPropertySelector::select_from_action(const String &p_type, const String &p_current, const bool p_connecting) {
  527. base_type = p_type;
  528. selected = p_current;
  529. type = Variant::NIL;
  530. script = 0;
  531. properties = false;
  532. visual_script_generic = false;
  533. instance = NULL;
  534. virtuals_only = false;
  535. show_window(.5f);
  536. search_box->set_text("");
  537. search_box->grab_focus();
  538. seq_connect = true;
  539. connecting = p_connecting;
  540. _update_search();
  541. }
  542. void VisualScriptPropertySelector::select_from_instance(Object *p_instance, const String &p_current, const bool p_connecting) {
  543. base_type = "";
  544. selected = p_current;
  545. type = Variant::NIL;
  546. script = 0;
  547. properties = true;
  548. visual_script_generic = false;
  549. instance = p_instance;
  550. virtuals_only = false;
  551. show_window(.5f);
  552. search_box->set_text("");
  553. search_box->grab_focus();
  554. seq_connect = false;
  555. connecting = p_connecting;
  556. _update_search();
  557. }
  558. void VisualScriptPropertySelector::select_from_visual_script(const String &p_base, const bool p_connecting) {
  559. base_type = p_base;
  560. selected = "";
  561. type = Variant::NIL;
  562. script = 0;
  563. properties = true;
  564. visual_script_generic = true;
  565. instance = NULL;
  566. virtuals_only = false;
  567. show_window(.5f);
  568. search_box->set_text("");
  569. search_box->grab_focus();
  570. connecting = p_connecting;
  571. _update_search();
  572. }
  573. void VisualScriptPropertySelector::show_window(float p_screen_ratio) {
  574. Rect2 rect;
  575. Point2 window_size = get_viewport_rect().size;
  576. rect.size = (window_size * p_screen_ratio).floor();
  577. rect.size.x = rect.size.x / 1.25f;
  578. rect.position = ((window_size - rect.size) / 2.0f).floor();
  579. popup(rect);
  580. }
  581. void VisualScriptPropertySelector::_bind_methods() {
  582. ClassDB::bind_method(D_METHOD("_text_changed"), &VisualScriptPropertySelector::_text_changed);
  583. ClassDB::bind_method(D_METHOD("_confirmed"), &VisualScriptPropertySelector::_confirmed);
  584. ClassDB::bind_method(D_METHOD("_sbox_input"), &VisualScriptPropertySelector::_sbox_input);
  585. ClassDB::bind_method(D_METHOD("_item_selected"), &VisualScriptPropertySelector::_item_selected);
  586. ADD_SIGNAL(MethodInfo("selected", PropertyInfo(Variant::STRING, "name"), PropertyInfo(Variant::STRING, "category"), PropertyInfo(Variant::BOOL, "connecting")));
  587. }
  588. VisualScriptPropertySelector::VisualScriptPropertySelector() {
  589. VBoxContainer *vbc = memnew(VBoxContainer);
  590. add_child(vbc);
  591. //set_child_rect(vbc);
  592. search_box = memnew(LineEdit);
  593. vbc->add_margin_child(TTR("Search:"), search_box);
  594. search_box->connect("text_changed", this, "_text_changed");
  595. search_box->connect("gui_input", this, "_sbox_input");
  596. search_options = memnew(Tree);
  597. vbc->add_margin_child(TTR("Matches:"), search_options, true);
  598. get_ok()->set_text(TTR("Open"));
  599. get_ok()->set_disabled(true);
  600. register_text_enter(search_box);
  601. set_hide_on_ok(false);
  602. search_options->connect("item_activated", this, "_confirmed");
  603. search_options->connect("cell_selected", this, "_item_selected");
  604. search_options->set_hide_root(true);
  605. search_options->set_hide_folding(true);
  606. virtuals_only = false;
  607. seq_connect = false;
  608. help_bit = memnew(EditorHelpBit);
  609. vbc->add_margin_child(TTR("Description:"), help_bit);
  610. help_bit->connect("request_hide", this, "_closed");
  611. search_options->set_columns(3);
  612. search_options->set_column_expand(1, false);
  613. search_options->set_column_expand(2, false);
  614. }