visual_script_property_selector.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726
  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-2020 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2020 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/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. StringName base = base_type;
  75. List<StringName> base_list;
  76. while (base) {
  77. base_list.push_back(base);
  78. base = ClassDB::get_parent_class_nocheck(base);
  79. }
  80. for (List<StringName>::Element *E = base_list.front(); E; E = E->next()) {
  81. List<MethodInfo> methods;
  82. List<PropertyInfo> props;
  83. TreeItem *category = NULL;
  84. Ref<Texture2D> type_icons[Variant::VARIANT_MAX] = {
  85. Control::get_icon("Variant", "EditorIcons"),
  86. Control::get_icon("bool", "EditorIcons"),
  87. Control::get_icon("int", "EditorIcons"),
  88. Control::get_icon("float", "EditorIcons"),
  89. Control::get_icon("String", "EditorIcons"),
  90. Control::get_icon("Vector2", "EditorIcons"),
  91. Control::get_icon("Rect2", "EditorIcons"),
  92. Control::get_icon("Vector3", "EditorIcons"),
  93. Control::get_icon("Transform2D", "EditorIcons"),
  94. Control::get_icon("Plane", "EditorIcons"),
  95. Control::get_icon("Quat", "EditorIcons"),
  96. Control::get_icon("AABB", "EditorIcons"),
  97. Control::get_icon("Basis", "EditorIcons"),
  98. Control::get_icon("Transform", "EditorIcons"),
  99. Control::get_icon("Color", "EditorIcons"),
  100. Control::get_icon("Path", "EditorIcons"),
  101. Control::get_icon("RID", "EditorIcons"),
  102. Control::get_icon("Object", "EditorIcons"),
  103. Control::get_icon("Dictionary", "EditorIcons"),
  104. Control::get_icon("Array", "EditorIcons"),
  105. Control::get_icon("PackedByteArray", "EditorIcons"),
  106. Control::get_icon("PackedIntArray", "EditorIcons"),
  107. Control::get_icon("PackedRealArray", "EditorIcons"),
  108. Control::get_icon("PackedStringArray", "EditorIcons"),
  109. Control::get_icon("PackedVector2Array", "EditorIcons"),
  110. Control::get_icon("PackedVector3Array", "EditorIcons"),
  111. Control::get_icon("PackedColorArray", "EditorIcons")
  112. };
  113. {
  114. String b = String(E->get());
  115. category = search_options->create_item(root);
  116. if (category) {
  117. category->set_text(0, b.replace_first("*", ""));
  118. category->set_selectable(0, false);
  119. Ref<Texture2D> icon;
  120. String rep = b.replace("*", "");
  121. icon = EditorNode::get_singleton()->get_class_icon(rep);
  122. category->set_icon(0, icon);
  123. }
  124. }
  125. if (properties || seq_connect) {
  126. if (instance) {
  127. instance->get_property_list(&props, true);
  128. } else {
  129. Object *obj = ObjectDB::get_instance(script);
  130. if (Object::cast_to<Script>(obj)) {
  131. Object::cast_to<Script>(obj)->get_script_property_list(&props);
  132. } else {
  133. ClassDB::get_property_list(E->get(), &props, true);
  134. }
  135. }
  136. for (List<PropertyInfo>::Element *F = props.front(); F; F = F->next()) {
  137. if (!(F->get().usage & PROPERTY_USAGE_EDITOR) && !(F->get().usage & PROPERTY_USAGE_SCRIPT_VARIABLE))
  138. continue;
  139. if (type_filter.size() && type_filter.find(F->get().type) == -1)
  140. continue;
  141. // capitalize() also converts underscore to space, we'll match again both possible styles
  142. String get_text_raw = String(vformat(TTR("Get %s"), F->get().name));
  143. String get_text = get_text_raw.capitalize();
  144. String set_text_raw = String(vformat(TTR("Set %s"), F->get().name));
  145. String set_text = set_text_raw.capitalize();
  146. String input = search_box->get_text().capitalize();
  147. if (input == String() || get_text_raw.findn(input) != -1 || get_text.findn(input) != -1) {
  148. TreeItem *item = search_options->create_item(category ? category : root);
  149. item->set_text(0, get_text);
  150. item->set_metadata(0, F->get().name);
  151. item->set_icon(0, type_icons[F->get().type]);
  152. item->set_metadata(1, "get");
  153. item->set_collapsed(1);
  154. item->set_selectable(0, true);
  155. item->set_selectable(1, false);
  156. item->set_selectable(2, false);
  157. item->set_metadata(2, connecting);
  158. }
  159. if (input == String() || set_text_raw.findn(input) != -1 || set_text.findn(input) != -1) {
  160. TreeItem *item = search_options->create_item(category ? category : root);
  161. item->set_text(0, set_text);
  162. item->set_metadata(0, F->get().name);
  163. item->set_icon(0, type_icons[F->get().type]);
  164. item->set_metadata(1, "set");
  165. item->set_selectable(0, true);
  166. item->set_selectable(1, false);
  167. item->set_selectable(2, false);
  168. item->set_metadata(2, connecting);
  169. }
  170. }
  171. }
  172. {
  173. if (type != Variant::NIL) {
  174. Variant v;
  175. Callable::CallError ce;
  176. v = Variant::construct(type, NULL, 0, ce);
  177. v.get_method_list(&methods);
  178. } else {
  179. Object *obj = ObjectDB::get_instance(script);
  180. if (Object::cast_to<Script>(obj)) {
  181. Object::cast_to<Script>(obj)->get_script_method_list(&methods);
  182. }
  183. ClassDB::get_method_list(E->get(), &methods, true, true);
  184. }
  185. }
  186. for (List<MethodInfo>::Element *M = methods.front(); M; M = M->next()) {
  187. String name = M->get().name.get_slice(":", 0);
  188. if (name.begins_with("_") && !(M->get().flags & METHOD_FLAG_VIRTUAL))
  189. continue;
  190. if (virtuals_only && !(M->get().flags & METHOD_FLAG_VIRTUAL))
  191. continue;
  192. if (!virtuals_only && (M->get().flags & METHOD_FLAG_VIRTUAL))
  193. continue;
  194. MethodInfo mi = M->get();
  195. String desc_arguments;
  196. if (mi.arguments.size() > 0) {
  197. desc_arguments = "(";
  198. for (int i = 0; i < mi.arguments.size(); i++) {
  199. if (i > 0) {
  200. desc_arguments += ", ";
  201. }
  202. if (mi.arguments[i].type == Variant::NIL) {
  203. desc_arguments += "var";
  204. } else if (mi.arguments[i].name.find(":") != -1) {
  205. desc_arguments += mi.arguments[i].name.get_slice(":", 1);
  206. mi.arguments[i].name = mi.arguments[i].name.get_slice(":", 0);
  207. } else {
  208. desc_arguments += Variant::get_type_name(mi.arguments[i].type);
  209. }
  210. }
  211. desc_arguments += ")";
  212. }
  213. String desc_raw = mi.name + desc_arguments;
  214. String desc = desc_raw.capitalize().replace("( ", "(");
  215. if (search_box->get_text() != String() &&
  216. name.findn(search_box->get_text()) == -1 &&
  217. desc.findn(search_box->get_text()) == -1 &&
  218. desc_raw.findn(search_box->get_text()) == -1) {
  219. continue;
  220. }
  221. TreeItem *item = search_options->create_item(category ? category : root);
  222. item->set_text(0, desc);
  223. item->set_icon(0, get_icon("MemberMethod", "EditorIcons"));
  224. item->set_metadata(0, name);
  225. item->set_selectable(0, true);
  226. item->set_metadata(1, "method");
  227. item->set_collapsed(1);
  228. item->set_selectable(1, false);
  229. item->set_selectable(2, false);
  230. item->set_metadata(2, connecting);
  231. }
  232. if (category && category->get_children() == NULL) {
  233. memdelete(category); //old category was unused
  234. }
  235. }
  236. if (properties) {
  237. if (!seq_connect && !visual_script_generic) {
  238. get_visual_node_names("flow_control/type_cast", Set<String>(), found, root, search_box);
  239. get_visual_node_names("functions/built_in/print", Set<String>(), found, root, search_box);
  240. get_visual_node_names("functions/by_type/" + Variant::get_type_name(type), Set<String>(), found, root, search_box);
  241. get_visual_node_names("functions/deconstruct/" + Variant::get_type_name(type), Set<String>(), found, root, search_box);
  242. get_visual_node_names("operators/compare/", Set<String>(), found, root, search_box);
  243. if (type == Variant::INT) {
  244. get_visual_node_names("operators/bitwise/", Set<String>(), found, root, search_box);
  245. }
  246. if (type == Variant::BOOL) {
  247. get_visual_node_names("operators/logic/", Set<String>(), found, root, search_box);
  248. }
  249. if (type == Variant::BOOL || type == Variant::INT || type == Variant::REAL || type == Variant::VECTOR2 || type == Variant::VECTOR3) {
  250. get_visual_node_names("operators/math/", Set<String>(), found, root, search_box);
  251. }
  252. }
  253. }
  254. if (seq_connect && !visual_script_generic) {
  255. String text = search_box->get_text();
  256. create_visualscript_item(String("VisualScriptCondition"), root, text, String("Condition"));
  257. create_visualscript_item(String("VisualScriptSwitch"), root, text, String("Switch"));
  258. create_visualscript_item(String("VisualScriptSequence"), root, text, String("Sequence"));
  259. create_visualscript_item(String("VisualScriptIterator"), root, text, String("Iterator"));
  260. create_visualscript_item(String("VisualScriptWhile"), root, text, String("While"));
  261. create_visualscript_item(String("VisualScriptReturn"), root, text, String("Return"));
  262. get_visual_node_names("flow_control/type_cast", Set<String>(), found, root, search_box);
  263. get_visual_node_names("functions/built_in/print", Set<String>(), found, root, search_box);
  264. }
  265. if ((properties || seq_connect) && visual_script_generic) {
  266. get_visual_node_names("", Set<String>(), found, root, search_box);
  267. }
  268. TreeItem *selected_item = search_options->search_item_text(search_box->get_text());
  269. if (!found && selected_item != NULL) {
  270. selected_item->select(0);
  271. found = true;
  272. }
  273. get_ok()->set_disabled(root->get_children() == NULL);
  274. }
  275. void VisualScriptPropertySelector::create_visualscript_item(const String &name, TreeItem *const root, const String &search_input, const String &text) {
  276. if (search_input == String() || text.findn(search_input) != -1) {
  277. TreeItem *item = search_options->create_item(root);
  278. item->set_text(0, text);
  279. item->set_icon(0, get_icon("VisualScript", "EditorIcons"));
  280. item->set_metadata(0, name);
  281. item->set_metadata(1, "action");
  282. item->set_selectable(0, true);
  283. item->set_collapsed(1);
  284. item->set_selectable(1, false);
  285. item->set_selectable(2, false);
  286. item->set_metadata(2, connecting);
  287. }
  288. }
  289. void VisualScriptPropertySelector::get_visual_node_names(const String &root_filter, const Set<String> &p_modifiers, bool &found, TreeItem *const root, LineEdit *const search_box) {
  290. Map<String, TreeItem *> path_cache;
  291. List<String> fnodes;
  292. VisualScriptLanguage::singleton->get_registered_node_names(&fnodes);
  293. for (List<String>::Element *E = fnodes.front(); E; E = E->next()) {
  294. if (!E->get().begins_with(root_filter)) {
  295. continue;
  296. }
  297. Vector<String> path = E->get().split("/");
  298. // check if the name has the filter
  299. bool in_filter = false;
  300. Vector<String> tx_filters = search_box->get_text().split(" ");
  301. for (int i = 0; i < tx_filters.size(); i++) {
  302. if (tx_filters[i] == "") {
  303. in_filter = true;
  304. } else {
  305. in_filter = false;
  306. }
  307. if (E->get().findn(tx_filters[i]) != -1) {
  308. in_filter = true;
  309. break;
  310. }
  311. }
  312. if (!in_filter) {
  313. continue;
  314. }
  315. bool in_modifier = p_modifiers.empty();
  316. for (Set<String>::Element *F = p_modifiers.front(); F && in_modifier; F = F->next()) {
  317. if (E->get().findn(F->get()) != -1)
  318. in_modifier = true;
  319. }
  320. if (!in_modifier) {
  321. continue;
  322. }
  323. TreeItem *item = search_options->create_item(root);
  324. Ref<VisualScriptNode> vnode = VisualScriptLanguage::singleton->create_node_from_name(E->get());
  325. Ref<VisualScriptOperator> vnode_operator = vnode;
  326. String type_name;
  327. if (vnode_operator.is_valid()) {
  328. String type;
  329. if (path.size() >= 2) {
  330. type = path[1];
  331. }
  332. type_name = type.capitalize() + " ";
  333. }
  334. Ref<VisualScriptFunctionCall> vnode_function_call = vnode;
  335. if (vnode_function_call.is_valid()) {
  336. String basic_type = Variant::get_type_name(vnode_function_call->get_basic_type());
  337. type_name = basic_type.capitalize() + " ";
  338. }
  339. Ref<VisualScriptConstructor> vnode_constructor = vnode;
  340. if (vnode_constructor.is_valid()) {
  341. type_name = "Construct ";
  342. }
  343. Ref<VisualScriptDeconstruct> vnode_deconstruct = vnode;
  344. if (vnode_deconstruct.is_valid()) {
  345. type_name = "Deconstruct ";
  346. }
  347. Vector<String> desc = path[path.size() - 1].replace("(", " ").replace(")", " ").replace(",", " ").split(" ");
  348. for (int i = 0; i < desc.size(); i++) {
  349. desc.write[i] = desc[i].capitalize();
  350. if (desc[i].ends_with(",")) {
  351. desc.write[i] = desc[i].replace(",", ", ");
  352. }
  353. }
  354. item->set_text(0, type_name + String("").join(desc));
  355. item->set_icon(0, get_icon("VisualScript", "EditorIcons"));
  356. item->set_selectable(0, true);
  357. item->set_metadata(0, E->get());
  358. item->set_selectable(0, true);
  359. item->set_metadata(1, "visualscript");
  360. item->set_selectable(1, false);
  361. item->set_selectable(2, false);
  362. item->set_metadata(2, connecting);
  363. }
  364. }
  365. void VisualScriptPropertySelector::_confirmed() {
  366. TreeItem *ti = search_options->get_selected();
  367. if (!ti)
  368. return;
  369. emit_signal("selected", ti->get_metadata(0), ti->get_metadata(1), ti->get_metadata(2));
  370. hide();
  371. }
  372. void VisualScriptPropertySelector::_item_selected() {
  373. help_bit->set_text("");
  374. TreeItem *item = search_options->get_selected();
  375. if (!item)
  376. return;
  377. String name = item->get_metadata(0);
  378. String class_type;
  379. if (type != Variant::NIL) {
  380. class_type = Variant::get_type_name(type);
  381. } else {
  382. class_type = base_type;
  383. }
  384. DocData *dd = EditorHelp::get_doc_data();
  385. String text;
  386. String at_class = class_type;
  387. while (at_class != String()) {
  388. Map<String, DocData::ClassDoc>::Element *E = dd->class_list.find(at_class);
  389. if (E) {
  390. for (int i = 0; i < E->get().properties.size(); i++) {
  391. if (E->get().properties[i].name == name) {
  392. text = E->get().properties[i].description;
  393. }
  394. }
  395. }
  396. at_class = ClassDB::get_parent_class_nocheck(at_class);
  397. }
  398. at_class = class_type;
  399. while (at_class != String()) {
  400. Map<String, DocData::ClassDoc>::Element *C = dd->class_list.find(at_class);
  401. if (C) {
  402. for (int i = 0; i < C->get().methods.size(); i++) {
  403. if (C->get().methods[i].name == name) {
  404. text = C->get().methods[i].description;
  405. }
  406. }
  407. }
  408. at_class = ClassDB::get_parent_class_nocheck(at_class);
  409. }
  410. Map<String, DocData::ClassDoc>::Element *T = dd->class_list.find(class_type);
  411. if (T) {
  412. for (int i = 0; i < T->get().methods.size(); i++) {
  413. Vector<String> functions = name.rsplit("/", false, 1);
  414. if (T->get().methods[i].name == functions[functions.size() - 1]) {
  415. text = T->get().methods[i].description;
  416. }
  417. }
  418. }
  419. List<String> *names = memnew(List<String>);
  420. VisualScriptLanguage::singleton->get_registered_node_names(names);
  421. if (names->find(name) != NULL) {
  422. Ref<VisualScriptOperator> operator_node = VisualScriptLanguage::singleton->create_node_from_name(name);
  423. if (operator_node.is_valid()) {
  424. Map<String, DocData::ClassDoc>::Element *F = dd->class_list.find(operator_node->get_class_name());
  425. if (F) {
  426. text = Variant::get_operator_name(operator_node->get_operator());
  427. }
  428. }
  429. Ref<VisualScriptTypeCast> typecast_node = VisualScriptLanguage::singleton->create_node_from_name(name);
  430. if (typecast_node.is_valid()) {
  431. Map<String, DocData::ClassDoc>::Element *F = dd->class_list.find(typecast_node->get_class_name());
  432. if (F) {
  433. text = F->get().description;
  434. }
  435. }
  436. Ref<VisualScriptBuiltinFunc> builtin_node = VisualScriptLanguage::singleton->create_node_from_name(name);
  437. if (builtin_node.is_valid()) {
  438. Map<String, DocData::ClassDoc>::Element *F = dd->class_list.find(builtin_node->get_class_name());
  439. if (F) {
  440. for (int i = 0; i < F->get().constants.size(); i++) {
  441. if (F->get().constants[i].value.to_int() == int(builtin_node->get_func())) {
  442. text = F->get().constants[i].description;
  443. }
  444. }
  445. }
  446. }
  447. }
  448. memdelete(names);
  449. if (text == String())
  450. return;
  451. help_bit->set_text(text);
  452. }
  453. void VisualScriptPropertySelector::_notification(int p_what) {
  454. if (p_what == NOTIFICATION_ENTER_TREE) {
  455. connect_compat("confirmed", this, "_confirmed");
  456. }
  457. }
  458. void VisualScriptPropertySelector::select_method_from_base_type(const String &p_base, const String &p_current, const bool p_virtuals_only, const bool p_connecting, bool clear_text) {
  459. base_type = p_base;
  460. selected = p_current;
  461. type = Variant::NIL;
  462. properties = false;
  463. instance = NULL;
  464. virtuals_only = p_virtuals_only;
  465. show_window(.5f);
  466. if (clear_text)
  467. search_box->set_text("");
  468. else
  469. search_box->select_all();
  470. search_box->grab_focus();
  471. connecting = p_connecting;
  472. _update_search();
  473. }
  474. void VisualScriptPropertySelector::set_type_filter(const Vector<Variant::Type> &p_type_filter) {
  475. type_filter = p_type_filter;
  476. }
  477. 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, bool clear_text) {
  478. base_type = p_base;
  479. selected = p_current;
  480. type = Variant::NIL;
  481. properties = true;
  482. visual_script_generic = false;
  483. instance = NULL;
  484. virtuals_only = p_virtuals_only;
  485. show_window(.5f);
  486. if (clear_text)
  487. search_box->set_text("");
  488. else
  489. search_box->select_all();
  490. search_box->grab_focus();
  491. seq_connect = p_seq_connect;
  492. connecting = p_connecting;
  493. _update_search();
  494. }
  495. void VisualScriptPropertySelector::select_from_script(const Ref<Script> &p_script, const String &p_current, const bool p_connecting, bool clear_text) {
  496. ERR_FAIL_COND(p_script.is_null());
  497. base_type = p_script->get_instance_base_type();
  498. selected = p_current;
  499. type = Variant::NIL;
  500. script = p_script->get_instance_id();
  501. properties = true;
  502. visual_script_generic = false;
  503. instance = NULL;
  504. virtuals_only = false;
  505. show_window(.5f);
  506. if (clear_text)
  507. search_box->set_text("");
  508. else
  509. search_box->select_all();
  510. search_box->grab_focus();
  511. seq_connect = false;
  512. connecting = p_connecting;
  513. _update_search();
  514. }
  515. void VisualScriptPropertySelector::select_from_basic_type(Variant::Type p_type, const String &p_current, const bool p_connecting, bool clear_text) {
  516. ERR_FAIL_COND(p_type == Variant::NIL);
  517. base_type = "";
  518. selected = p_current;
  519. type = p_type;
  520. properties = true;
  521. visual_script_generic = false;
  522. instance = NULL;
  523. virtuals_only = false;
  524. show_window(.5f);
  525. if (clear_text)
  526. search_box->set_text("");
  527. else
  528. search_box->select_all();
  529. search_box->grab_focus();
  530. seq_connect = false;
  531. connecting = p_connecting;
  532. _update_search();
  533. }
  534. void VisualScriptPropertySelector::select_from_action(const String &p_type, const String &p_current, const bool p_connecting, bool clear_text) {
  535. base_type = p_type;
  536. selected = p_current;
  537. type = Variant::NIL;
  538. properties = false;
  539. visual_script_generic = false;
  540. instance = NULL;
  541. virtuals_only = false;
  542. show_window(.5f);
  543. if (clear_text)
  544. search_box->set_text("");
  545. else
  546. search_box->select_all();
  547. search_box->grab_focus();
  548. seq_connect = true;
  549. connecting = p_connecting;
  550. _update_search();
  551. }
  552. void VisualScriptPropertySelector::select_from_instance(Object *p_instance, const String &p_current, const bool p_connecting, const String &p_basetype, bool clear_text) {
  553. base_type = p_basetype;
  554. selected = p_current;
  555. type = Variant::NIL;
  556. properties = true;
  557. visual_script_generic = false;
  558. instance = p_instance;
  559. virtuals_only = false;
  560. show_window(.5f);
  561. if (clear_text)
  562. search_box->set_text("");
  563. else
  564. search_box->select_all();
  565. search_box->grab_focus();
  566. seq_connect = false;
  567. connecting = p_connecting;
  568. _update_search();
  569. }
  570. void VisualScriptPropertySelector::select_from_visual_script(const String &p_base, const bool p_connecting, bool clear_text) {
  571. base_type = p_base;
  572. selected = "";
  573. type = Variant::NIL;
  574. properties = true;
  575. visual_script_generic = true;
  576. instance = NULL;
  577. virtuals_only = false;
  578. show_window(.5f);
  579. if (clear_text)
  580. search_box->set_text("");
  581. else
  582. search_box->select_all();
  583. search_box->grab_focus();
  584. connecting = p_connecting;
  585. _update_search();
  586. }
  587. void VisualScriptPropertySelector::show_window(float p_screen_ratio) {
  588. Rect2 rect;
  589. Point2 window_size = get_viewport_rect().size;
  590. rect.size = (window_size * p_screen_ratio).floor();
  591. rect.size.x = rect.size.x / 2.2f;
  592. rect.position = ((window_size - rect.size) / 2.0f).floor();
  593. popup(rect);
  594. }
  595. void VisualScriptPropertySelector::_bind_methods() {
  596. ClassDB::bind_method(D_METHOD("_text_changed"), &VisualScriptPropertySelector::_text_changed);
  597. ClassDB::bind_method(D_METHOD("_confirmed"), &VisualScriptPropertySelector::_confirmed);
  598. ClassDB::bind_method(D_METHOD("_sbox_input"), &VisualScriptPropertySelector::_sbox_input);
  599. ClassDB::bind_method(D_METHOD("_item_selected"), &VisualScriptPropertySelector::_item_selected);
  600. ADD_SIGNAL(MethodInfo("selected", PropertyInfo(Variant::STRING, "name"), PropertyInfo(Variant::STRING, "category"), PropertyInfo(Variant::BOOL, "connecting")));
  601. }
  602. VisualScriptPropertySelector::VisualScriptPropertySelector() {
  603. VBoxContainer *vbc = memnew(VBoxContainer);
  604. add_child(vbc);
  605. //set_child_rect(vbc);
  606. search_box = memnew(LineEdit);
  607. vbc->add_margin_child(TTR("Search:"), search_box);
  608. search_box->connect_compat("text_changed", this, "_text_changed");
  609. search_box->connect_compat("gui_input", this, "_sbox_input");
  610. search_options = memnew(Tree);
  611. vbc->add_margin_child(TTR("Matches:"), search_options, true);
  612. get_ok()->set_text(TTR("Open"));
  613. get_ok()->set_disabled(true);
  614. register_text_enter(search_box);
  615. set_hide_on_ok(false);
  616. search_options->connect_compat("item_activated", this, "_confirmed");
  617. search_options->connect_compat("cell_selected", this, "_item_selected");
  618. search_options->set_hide_root(true);
  619. search_options->set_hide_folding(true);
  620. virtuals_only = false;
  621. seq_connect = false;
  622. help_bit = memnew(EditorHelpBit);
  623. vbc->add_margin_child(TTR("Description:"), help_bit);
  624. help_bit->connect_compat("request_hide", this, "_closed");
  625. search_options->set_columns(3);
  626. search_options->set_column_expand(1, false);
  627. search_options->set_column_expand(2, false);
  628. }