gdscript_utility_functions.cpp 30 KB

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