gdscript_utility_functions.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732
  1. /*************************************************************************/
  2. /* gdscript_utility_functions.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2022 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 "gdscript_utility_functions.h"
  31. #include "core/io/resource_loader.h"
  32. #include "core/object/class_db.h"
  33. #include "core/object/method_bind.h"
  34. #include "core/object/object.h"
  35. #include "core/templates/oa_hash_map.h"
  36. #include "core/templates/vector.h"
  37. #include "gdscript.h"
  38. #ifdef DEBUG_ENABLED
  39. #define VALIDATE_ARG_COUNT(m_count) \
  40. if (p_arg_count < m_count) { \
  41. r_error.error = Callable::CallError::CALL_ERROR_TOO_FEW_ARGUMENTS; \
  42. r_error.argument = m_count; \
  43. r_error.expected = m_count; \
  44. *r_ret = Variant(); \
  45. return; \
  46. } \
  47. if (p_arg_count > m_count) { \
  48. r_error.error = Callable::CallError::CALL_ERROR_TOO_MANY_ARGUMENTS; \
  49. r_error.argument = m_count; \
  50. r_error.expected = m_count; \
  51. *r_ret = Variant(); \
  52. return; \
  53. }
  54. #define VALIDATE_ARG_INT(m_arg) \
  55. if (p_args[m_arg]->get_type() != Variant::INT) { \
  56. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT; \
  57. r_error.argument = m_arg; \
  58. r_error.expected = Variant::INT; \
  59. *r_ret = Variant(); \
  60. return; \
  61. }
  62. #define VALIDATE_ARG_NUM(m_arg) \
  63. if (!p_args[m_arg]->is_num()) { \
  64. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT; \
  65. r_error.argument = m_arg; \
  66. r_error.expected = Variant::FLOAT; \
  67. *r_ret = Variant(); \
  68. return; \
  69. }
  70. #else
  71. #define VALIDATE_ARG_COUNT(m_count)
  72. #define VALIDATE_ARG_INT(m_arg)
  73. #define VALIDATE_ARG_NUM(m_arg)
  74. #endif
  75. struct GDScriptUtilityFunctionsDefinitions {
  76. static inline void convert(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  77. VALIDATE_ARG_COUNT(2);
  78. VALIDATE_ARG_INT(1);
  79. int type = *p_args[1];
  80. if (type < 0 || type >= Variant::VARIANT_MAX) {
  81. *r_ret = RTR("Invalid type argument to convert(), use TYPE_* constants.");
  82. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  83. r_error.argument = 0;
  84. r_error.expected = Variant::INT;
  85. return;
  86. } else {
  87. Variant::construct(Variant::Type(type), *r_ret, p_args, 1, r_error);
  88. }
  89. }
  90. static inline void type_exists(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  91. VALIDATE_ARG_COUNT(1);
  92. *r_ret = ClassDB::class_exists(*p_args[0]);
  93. }
  94. static inline void _char(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  95. VALIDATE_ARG_COUNT(1);
  96. VALIDATE_ARG_INT(0);
  97. char32_t result[2] = { *p_args[0], 0 };
  98. *r_ret = String(result);
  99. }
  100. static inline void str(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  101. if (p_arg_count < 1) {
  102. r_error.error = Callable::CallError::CALL_ERROR_TOO_FEW_ARGUMENTS;
  103. r_error.argument = 1;
  104. r_error.expected = 1;
  105. *r_ret = Variant();
  106. return;
  107. }
  108. String str;
  109. for (int i = 0; i < p_arg_count; i++) {
  110. String os = p_args[i]->operator String();
  111. if (i == 0) {
  112. str = os;
  113. } else {
  114. str += os;
  115. }
  116. }
  117. *r_ret = str;
  118. }
  119. static inline void range(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  120. switch (p_arg_count) {
  121. case 0: {
  122. r_error.error = Callable::CallError::CALL_ERROR_TOO_FEW_ARGUMENTS;
  123. r_error.argument = 1;
  124. r_error.expected = 1;
  125. *r_ret = Variant();
  126. } break;
  127. case 1: {
  128. VALIDATE_ARG_NUM(0);
  129. int count = *p_args[0];
  130. Array arr;
  131. if (count <= 0) {
  132. *r_ret = arr;
  133. return;
  134. }
  135. Error err = arr.resize(count);
  136. if (err != OK) {
  137. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  138. *r_ret = Variant();
  139. return;
  140. }
  141. for (int i = 0; i < count; i++) {
  142. arr[i] = i;
  143. }
  144. *r_ret = arr;
  145. } break;
  146. case 2: {
  147. VALIDATE_ARG_NUM(0);
  148. VALIDATE_ARG_NUM(1);
  149. int from = *p_args[0];
  150. int to = *p_args[1];
  151. Array arr;
  152. if (from >= to) {
  153. *r_ret = arr;
  154. return;
  155. }
  156. Error err = arr.resize(to - from);
  157. if (err != OK) {
  158. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  159. *r_ret = Variant();
  160. return;
  161. }
  162. for (int i = from; i < to; i++) {
  163. arr[i - from] = i;
  164. }
  165. *r_ret = arr;
  166. } break;
  167. case 3: {
  168. VALIDATE_ARG_NUM(0);
  169. VALIDATE_ARG_NUM(1);
  170. VALIDATE_ARG_NUM(2);
  171. int from = *p_args[0];
  172. int to = *p_args[1];
  173. int incr = *p_args[2];
  174. if (incr == 0) {
  175. *r_ret = RTR("Step argument is zero!");
  176. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  177. return;
  178. }
  179. Array arr;
  180. if (from >= to && incr > 0) {
  181. *r_ret = arr;
  182. return;
  183. }
  184. if (from <= to && incr < 0) {
  185. *r_ret = arr;
  186. return;
  187. }
  188. // Calculate how many.
  189. int count = 0;
  190. if (incr > 0) {
  191. count = ((to - from - 1) / incr) + 1;
  192. } else {
  193. count = ((from - to - 1) / -incr) + 1;
  194. }
  195. Error err = arr.resize(count);
  196. if (err != OK) {
  197. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  198. *r_ret = Variant();
  199. return;
  200. }
  201. if (incr > 0) {
  202. int idx = 0;
  203. for (int i = from; i < to; i += incr) {
  204. arr[idx++] = i;
  205. }
  206. } else {
  207. int idx = 0;
  208. for (int i = from; i > to; i += incr) {
  209. arr[idx++] = i;
  210. }
  211. }
  212. *r_ret = arr;
  213. } break;
  214. default: {
  215. r_error.error = Callable::CallError::CALL_ERROR_TOO_MANY_ARGUMENTS;
  216. r_error.argument = 3;
  217. r_error.expected = 3;
  218. *r_ret = Variant();
  219. } break;
  220. }
  221. }
  222. static inline void load(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  223. VALIDATE_ARG_COUNT(1);
  224. if (p_args[0]->get_type() != Variant::STRING) {
  225. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  226. r_error.argument = 0;
  227. r_error.expected = Variant::STRING;
  228. *r_ret = Variant();
  229. } else {
  230. *r_ret = ResourceLoader::load(*p_args[0]);
  231. }
  232. }
  233. static inline void inst2dict(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  234. VALIDATE_ARG_COUNT(1);
  235. if (p_args[0]->get_type() == Variant::NIL) {
  236. *r_ret = Variant();
  237. } else if (p_args[0]->get_type() != Variant::OBJECT) {
  238. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  239. r_error.argument = 0;
  240. *r_ret = Variant();
  241. } else {
  242. Object *obj = *p_args[0];
  243. if (!obj) {
  244. *r_ret = Variant();
  245. } else if (!obj->get_script_instance() || obj->get_script_instance()->get_language() != GDScriptLanguage::get_singleton()) {
  246. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  247. r_error.argument = 0;
  248. r_error.expected = Variant::DICTIONARY;
  249. *r_ret = RTR("Not a script with an instance");
  250. return;
  251. } else {
  252. GDScriptInstance *ins = static_cast<GDScriptInstance *>(obj->get_script_instance());
  253. Ref<GDScript> base = ins->get_script();
  254. if (base.is_null()) {
  255. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  256. r_error.argument = 0;
  257. r_error.expected = Variant::DICTIONARY;
  258. *r_ret = RTR("Not based on a script");
  259. return;
  260. }
  261. GDScript *p = base.ptr();
  262. Vector<StringName> sname;
  263. while (p->_owner) {
  264. sname.push_back(p->name);
  265. p = p->_owner;
  266. }
  267. sname.reverse();
  268. if (!p->path.is_resource_file()) {
  269. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  270. r_error.argument = 0;
  271. r_error.expected = Variant::DICTIONARY;
  272. *r_ret = Variant();
  273. *r_ret = RTR("Not based on a resource file");
  274. return;
  275. }
  276. NodePath cp(sname, Vector<StringName>(), false);
  277. Dictionary d;
  278. d["@subpath"] = cp;
  279. d["@path"] = p->get_path();
  280. for (const KeyValue<StringName, GDScript::MemberInfo> &E : base->member_indices) {
  281. if (!d.has(E.key)) {
  282. d[E.key] = ins->members[E.value.index];
  283. }
  284. }
  285. *r_ret = d;
  286. }
  287. }
  288. }
  289. static inline void dict2inst(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  290. VALIDATE_ARG_COUNT(1);
  291. if (p_args[0]->get_type() != Variant::DICTIONARY) {
  292. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  293. r_error.argument = 0;
  294. r_error.expected = Variant::DICTIONARY;
  295. *r_ret = Variant();
  296. return;
  297. }
  298. Dictionary d = *p_args[0];
  299. if (!d.has("@path")) {
  300. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  301. r_error.argument = 0;
  302. r_error.expected = Variant::OBJECT;
  303. *r_ret = RTR("Invalid instance dictionary format (missing @path)");
  304. return;
  305. }
  306. Ref<Script> scr = ResourceLoader::load(d["@path"]);
  307. if (!scr.is_valid()) {
  308. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  309. r_error.argument = 0;
  310. r_error.expected = Variant::OBJECT;
  311. *r_ret = RTR("Invalid instance dictionary format (can't load script at @path)");
  312. return;
  313. }
  314. Ref<GDScript> gdscr = scr;
  315. if (!gdscr.is_valid()) {
  316. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  317. r_error.argument = 0;
  318. r_error.expected = Variant::OBJECT;
  319. *r_ret = Variant();
  320. *r_ret = RTR("Invalid instance dictionary format (invalid script at @path)");
  321. return;
  322. }
  323. NodePath sub;
  324. if (d.has("@subpath")) {
  325. sub = d["@subpath"];
  326. }
  327. for (int i = 0; i < sub.get_name_count(); i++) {
  328. gdscr = gdscr->subclasses[sub.get_name(i)];
  329. if (!gdscr.is_valid()) {
  330. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  331. r_error.argument = 0;
  332. r_error.expected = Variant::OBJECT;
  333. *r_ret = Variant();
  334. *r_ret = RTR("Invalid instance dictionary (invalid subclasses)");
  335. return;
  336. }
  337. }
  338. *r_ret = gdscr->_new(nullptr, -1 /*skip initializer*/, r_error);
  339. if (r_error.error != Callable::CallError::CALL_OK) {
  340. *r_ret = Variant();
  341. return;
  342. }
  343. GDScriptInstance *ins = static_cast<GDScriptInstance *>(static_cast<Object *>(*r_ret)->get_script_instance());
  344. Ref<GDScript> gd_ref = ins->get_script();
  345. for (KeyValue<StringName, GDScript::MemberInfo> &E : gd_ref->member_indices) {
  346. if (d.has(E.key)) {
  347. ins->members.write[E.value.index] = d[E.key];
  348. }
  349. }
  350. }
  351. static inline void Color8(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  352. if (p_arg_count < 3) {
  353. r_error.error = Callable::CallError::CALL_ERROR_TOO_FEW_ARGUMENTS;
  354. r_error.argument = 3;
  355. *r_ret = Variant();
  356. return;
  357. }
  358. if (p_arg_count > 4) {
  359. r_error.error = Callable::CallError::CALL_ERROR_TOO_MANY_ARGUMENTS;
  360. r_error.argument = 4;
  361. *r_ret = Variant();
  362. return;
  363. }
  364. VALIDATE_ARG_INT(0);
  365. VALIDATE_ARG_INT(1);
  366. VALIDATE_ARG_INT(2);
  367. Color color((int64_t)*p_args[0] / 255.0f, (int64_t)*p_args[1] / 255.0f, (int64_t)*p_args[2] / 255.0f);
  368. if (p_arg_count == 4) {
  369. VALIDATE_ARG_INT(3);
  370. color.a = (int64_t)*p_args[3] / 255.0f;
  371. }
  372. *r_ret = color;
  373. }
  374. static inline void print_debug(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  375. String s;
  376. for (int i = 0; i < p_arg_count; i++) {
  377. s += p_args[i]->operator String();
  378. }
  379. if (Thread::get_caller_id() == Thread::get_main_id()) {
  380. ScriptLanguage *script = GDScriptLanguage::get_singleton();
  381. if (script->debug_get_stack_level_count() > 0) {
  382. s += "\n At: " + script->debug_get_stack_level_source(0) + ":" + itos(script->debug_get_stack_level_line(0)) + ":" + script->debug_get_stack_level_function(0) + "()";
  383. }
  384. } else {
  385. s += "\n At: Cannot retrieve debug info outside the main thread. Thread ID: " + itos(Thread::get_caller_id());
  386. }
  387. print_line(s);
  388. *r_ret = Variant();
  389. }
  390. static inline void print_stack(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  391. VALIDATE_ARG_COUNT(0);
  392. if (Thread::get_caller_id() != Thread::get_main_id()) {
  393. print_line("Cannot retrieve debug info outside the main thread. Thread ID: " + itos(Thread::get_caller_id()));
  394. return;
  395. }
  396. ScriptLanguage *script = GDScriptLanguage::get_singleton();
  397. for (int i = 0; i < script->debug_get_stack_level_count(); i++) {
  398. print_line("Frame " + itos(i) + " - " + script->debug_get_stack_level_source(i) + ":" + itos(script->debug_get_stack_level_line(i)) + " in function '" + script->debug_get_stack_level_function(i) + "'");
  399. };
  400. *r_ret = Variant();
  401. }
  402. static inline void get_stack(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  403. VALIDATE_ARG_COUNT(0);
  404. if (Thread::get_caller_id() != Thread::get_main_id()) {
  405. *r_ret = Array();
  406. return;
  407. }
  408. ScriptLanguage *script = GDScriptLanguage::get_singleton();
  409. Array ret;
  410. for (int i = 0; i < script->debug_get_stack_level_count(); i++) {
  411. Dictionary frame;
  412. frame["source"] = script->debug_get_stack_level_source(i);
  413. frame["function"] = script->debug_get_stack_level_function(i);
  414. frame["line"] = script->debug_get_stack_level_line(i);
  415. ret.push_back(frame);
  416. };
  417. *r_ret = ret;
  418. }
  419. static inline void len(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) {
  420. VALIDATE_ARG_COUNT(1);
  421. switch (p_args[0]->get_type()) {
  422. case Variant::STRING: {
  423. String d = *p_args[0];
  424. *r_ret = d.length();
  425. } break;
  426. case Variant::DICTIONARY: {
  427. Dictionary d = *p_args[0];
  428. *r_ret = d.size();
  429. } break;
  430. case Variant::ARRAY: {
  431. Array d = *p_args[0];
  432. *r_ret = d.size();
  433. } break;
  434. case Variant::PACKED_BYTE_ARRAY: {
  435. Vector<uint8_t> d = *p_args[0];
  436. *r_ret = d.size();
  437. } break;
  438. case Variant::PACKED_INT32_ARRAY: {
  439. Vector<int32_t> d = *p_args[0];
  440. *r_ret = d.size();
  441. } break;
  442. case Variant::PACKED_INT64_ARRAY: {
  443. Vector<int64_t> d = *p_args[0];
  444. *r_ret = d.size();
  445. } break;
  446. case Variant::PACKED_FLOAT32_ARRAY: {
  447. Vector<float> d = *p_args[0];
  448. *r_ret = d.size();
  449. } break;
  450. case Variant::PACKED_FLOAT64_ARRAY: {
  451. Vector<double> d = *p_args[0];
  452. *r_ret = d.size();
  453. } break;
  454. case Variant::PACKED_STRING_ARRAY: {
  455. Vector<String> d = *p_args[0];
  456. *r_ret = d.size();
  457. } break;
  458. case Variant::PACKED_VECTOR2_ARRAY: {
  459. Vector<Vector2> d = *p_args[0];
  460. *r_ret = d.size();
  461. } break;
  462. case Variant::PACKED_VECTOR3_ARRAY: {
  463. Vector<Vector3> d = *p_args[0];
  464. *r_ret = d.size();
  465. } break;
  466. case Variant::PACKED_COLOR_ARRAY: {
  467. Vector<Color> d = *p_args[0];
  468. *r_ret = d.size();
  469. } break;
  470. default: {
  471. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  472. r_error.argument = 0;
  473. r_error.expected = Variant::NIL;
  474. *r_ret = vformat(RTR("Value of type '%s' can't provide a length."), Variant::get_type_name(p_args[0]->get_type()));
  475. }
  476. }
  477. }
  478. };
  479. struct GDScriptUtilityFunctionInfo {
  480. GDScriptUtilityFunctions::FunctionPtr function = nullptr;
  481. MethodInfo info;
  482. bool is_constant = false;
  483. };
  484. static OAHashMap<StringName, GDScriptUtilityFunctionInfo> utility_function_table;
  485. static List<StringName> utility_function_name_table;
  486. static void _register_function(const String &p_name, const MethodInfo &p_method_info, GDScriptUtilityFunctions::FunctionPtr p_function, bool p_is_const) {
  487. StringName sname(p_name);
  488. ERR_FAIL_COND(utility_function_table.has(sname));
  489. GDScriptUtilityFunctionInfo function;
  490. function.function = p_function;
  491. function.info = p_method_info;
  492. function.is_constant = p_is_const;
  493. utility_function_table.insert(sname, function);
  494. utility_function_name_table.push_back(sname);
  495. }
  496. #define REGISTER_FUNC(m_func, m_is_const, m_return_type, ...) \
  497. { \
  498. String name(#m_func); \
  499. if (name.begins_with("_")) { \
  500. name = name.substr(1, name.length() - 1); \
  501. } \
  502. MethodInfo info = MethodInfo(name, __VA_ARGS__); \
  503. info.return_val.type = m_return_type; \
  504. _register_function(name, info, GDScriptUtilityFunctionsDefinitions::m_func, m_is_const); \
  505. }
  506. #define REGISTER_FUNC_NO_ARGS(m_func, m_is_const, m_return_type) \
  507. { \
  508. String name(#m_func); \
  509. if (name.begins_with("_")) { \
  510. name = name.substr(1, name.length() - 1); \
  511. } \
  512. MethodInfo info = MethodInfo(name); \
  513. info.return_val.type = m_return_type; \
  514. _register_function(name, info, GDScriptUtilityFunctionsDefinitions::m_func, m_is_const); \
  515. }
  516. #define REGISTER_VARARG_FUNC(m_func, m_is_const, m_return_type) \
  517. { \
  518. String name(#m_func); \
  519. if (name.begins_with("_")) { \
  520. name = name.substr(1, name.length() - 1); \
  521. } \
  522. MethodInfo info = MethodInfo(name); \
  523. info.return_val.type = m_return_type; \
  524. info.flags |= METHOD_FLAG_VARARG; \
  525. _register_function(name, info, GDScriptUtilityFunctionsDefinitions::m_func, m_is_const); \
  526. }
  527. #define REGISTER_VARIANT_FUNC(m_func, m_is_const, ...) \
  528. { \
  529. String name(#m_func); \
  530. if (name.begins_with("_")) { \
  531. name = name.substr(1, name.length() - 1); \
  532. } \
  533. MethodInfo info = MethodInfo(name, __VA_ARGS__); \
  534. info.return_val.type = Variant::NIL; \
  535. info.return_val.usage |= PROPERTY_USAGE_NIL_IS_VARIANT; \
  536. _register_function(name, info, GDScriptUtilityFunctionsDefinitions::m_func, m_is_const); \
  537. }
  538. #define REGISTER_CLASS_FUNC(m_func, m_is_const, m_return_type, ...) \
  539. { \
  540. String name(#m_func); \
  541. if (name.begins_with("_")) { \
  542. name = name.substr(1, name.length() - 1); \
  543. } \
  544. MethodInfo info = MethodInfo(name, __VA_ARGS__); \
  545. info.return_val.type = Variant::OBJECT; \
  546. info.return_val.hint = PROPERTY_HINT_RESOURCE_TYPE; \
  547. info.return_val.class_name = m_return_type; \
  548. _register_function(name, info, GDScriptUtilityFunctionsDefinitions::m_func, m_is_const); \
  549. }
  550. #define REGISTER_FUNC_DEF(m_func, m_is_const, m_default, m_return_type, ...) \
  551. { \
  552. String name(#m_func); \
  553. if (name.begins_with("_")) { \
  554. name = name.substr(1, name.length() - 1); \
  555. } \
  556. MethodInfo info = MethodInfo(name, __VA_ARGS__); \
  557. info.return_val.type = m_return_type; \
  558. info.default_arguments.push_back(m_default); \
  559. _register_function(name, info, GDScriptUtilityFunctionsDefinitions::m_func, m_is_const); \
  560. }
  561. #define ARG(m_name, m_type) \
  562. PropertyInfo(m_type, m_name)
  563. #define VARARG(m_name) \
  564. PropertyInfo(Variant::NIL, m_name, PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_NIL_IS_VARIANT)
  565. void GDScriptUtilityFunctions::register_functions() {
  566. REGISTER_VARIANT_FUNC(convert, true, VARARG("what"), ARG("type", Variant::INT));
  567. REGISTER_FUNC(type_exists, true, Variant::BOOL, ARG("type", Variant::STRING_NAME));
  568. REGISTER_FUNC(_char, true, Variant::STRING, ARG("char", Variant::INT));
  569. REGISTER_VARARG_FUNC(str, true, Variant::STRING);
  570. REGISTER_VARARG_FUNC(range, false, Variant::ARRAY);
  571. REGISTER_CLASS_FUNC(load, false, "Resource", ARG("path", Variant::STRING));
  572. REGISTER_FUNC(inst2dict, false, Variant::DICTIONARY, ARG("instance", Variant::OBJECT));
  573. REGISTER_FUNC(dict2inst, false, Variant::OBJECT, ARG("dictionary", Variant::DICTIONARY));
  574. REGISTER_FUNC_DEF(Color8, true, 255, Variant::COLOR, ARG("r8", Variant::INT), ARG("g8", Variant::INT), ARG("b8", Variant::INT), ARG("a8", Variant::INT));
  575. REGISTER_VARARG_FUNC(print_debug, false, Variant::NIL);
  576. REGISTER_FUNC_NO_ARGS(print_stack, false, Variant::NIL);
  577. REGISTER_FUNC_NO_ARGS(get_stack, false, Variant::ARRAY);
  578. REGISTER_FUNC(len, true, Variant::INT, VARARG("var"));
  579. }
  580. void GDScriptUtilityFunctions::unregister_functions() {
  581. utility_function_name_table.clear();
  582. utility_function_table.clear();
  583. }
  584. GDScriptUtilityFunctions::FunctionPtr GDScriptUtilityFunctions::get_function(const StringName &p_function) {
  585. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  586. ERR_FAIL_COND_V(!info, nullptr);
  587. return info->function;
  588. }
  589. bool GDScriptUtilityFunctions::has_function_return_value(const StringName &p_function) {
  590. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  591. ERR_FAIL_COND_V(!info, false);
  592. return info->info.return_val.type != Variant::NIL || bool(info->info.return_val.usage & PROPERTY_USAGE_NIL_IS_VARIANT);
  593. }
  594. Variant::Type GDScriptUtilityFunctions::get_function_return_type(const StringName &p_function) {
  595. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  596. ERR_FAIL_COND_V(!info, Variant::NIL);
  597. return info->info.return_val.type;
  598. }
  599. StringName GDScriptUtilityFunctions::get_function_return_class(const StringName &p_function) {
  600. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  601. ERR_FAIL_COND_V(!info, StringName());
  602. return info->info.return_val.class_name;
  603. }
  604. Variant::Type GDScriptUtilityFunctions::get_function_argument_type(const StringName &p_function, int p_arg) {
  605. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  606. ERR_FAIL_COND_V(!info, Variant::NIL);
  607. ERR_FAIL_COND_V(p_arg >= info->info.arguments.size(), Variant::NIL);
  608. return info->info.arguments[p_arg].type;
  609. }
  610. int GDScriptUtilityFunctions::get_function_argument_count(const StringName &p_function, int p_arg) {
  611. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  612. ERR_FAIL_COND_V(!info, 0);
  613. return info->info.arguments.size();
  614. }
  615. bool GDScriptUtilityFunctions::is_function_vararg(const StringName &p_function) {
  616. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  617. ERR_FAIL_COND_V(!info, false);
  618. return (bool)(info->info.flags & METHOD_FLAG_VARARG);
  619. }
  620. bool GDScriptUtilityFunctions::is_function_constant(const StringName &p_function) {
  621. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  622. ERR_FAIL_COND_V(!info, false);
  623. return info->is_constant;
  624. }
  625. bool GDScriptUtilityFunctions::function_exists(const StringName &p_function) {
  626. return utility_function_table.has(p_function);
  627. }
  628. void GDScriptUtilityFunctions::get_function_list(List<StringName> *r_functions) {
  629. for (const StringName &E : utility_function_name_table) {
  630. r_functions->push_back(E);
  631. }
  632. }
  633. MethodInfo GDScriptUtilityFunctions::get_function_info(const StringName &p_function) {
  634. GDScriptUtilityFunctionInfo *info = utility_function_table.lookup_ptr(p_function);
  635. ERR_FAIL_COND_V(!info, MethodInfo());
  636. return info->info;
  637. }