variant_construct.cpp 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874
  1. /*************************************************************************/
  2. /* variant_construct.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2021 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2021 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 "variant.h"
  31. #include "core/core_string_names.h"
  32. #include "core/crypto/crypto_core.h"
  33. #include "core/debugger/engine_debugger.h"
  34. #include "core/io/compression.h"
  35. #include "core/object/class_db.h"
  36. #include "core/os/os.h"
  37. #include "core/templates/local_vector.h"
  38. #include "core/templates/oa_hash_map.h"
  39. template <class T>
  40. struct PtrConstruct {};
  41. #define MAKE_PTRCONSTRUCT(m_type) \
  42. template <> \
  43. struct PtrConstruct<m_type> { \
  44. _FORCE_INLINE_ static void construct(const m_type &p_value, void *p_ptr) { \
  45. memnew_placement(p_ptr, m_type(p_value)); \
  46. } \
  47. };
  48. MAKE_PTRCONSTRUCT(bool);
  49. MAKE_PTRCONSTRUCT(int64_t);
  50. MAKE_PTRCONSTRUCT(double);
  51. MAKE_PTRCONSTRUCT(String);
  52. MAKE_PTRCONSTRUCT(Vector2);
  53. MAKE_PTRCONSTRUCT(Vector2i);
  54. MAKE_PTRCONSTRUCT(Rect2);
  55. MAKE_PTRCONSTRUCT(Rect2i);
  56. MAKE_PTRCONSTRUCT(Vector3);
  57. MAKE_PTRCONSTRUCT(Vector3i);
  58. MAKE_PTRCONSTRUCT(Transform2D);
  59. MAKE_PTRCONSTRUCT(Plane);
  60. MAKE_PTRCONSTRUCT(Quat);
  61. MAKE_PTRCONSTRUCT(AABB);
  62. MAKE_PTRCONSTRUCT(Basis);
  63. MAKE_PTRCONSTRUCT(Transform3D);
  64. MAKE_PTRCONSTRUCT(Color);
  65. MAKE_PTRCONSTRUCT(StringName);
  66. MAKE_PTRCONSTRUCT(NodePath);
  67. MAKE_PTRCONSTRUCT(RID);
  68. template <>
  69. struct PtrConstruct<Object *> {
  70. _FORCE_INLINE_ static void construct(Object *p_value, void *p_ptr) {
  71. *((Object **)p_ptr) = p_value;
  72. }
  73. };
  74. MAKE_PTRCONSTRUCT(Callable);
  75. MAKE_PTRCONSTRUCT(Signal);
  76. MAKE_PTRCONSTRUCT(Dictionary);
  77. MAKE_PTRCONSTRUCT(Array);
  78. MAKE_PTRCONSTRUCT(PackedByteArray);
  79. MAKE_PTRCONSTRUCT(PackedInt32Array);
  80. MAKE_PTRCONSTRUCT(PackedInt64Array);
  81. MAKE_PTRCONSTRUCT(PackedFloat32Array);
  82. MAKE_PTRCONSTRUCT(PackedFloat64Array);
  83. MAKE_PTRCONSTRUCT(PackedStringArray);
  84. MAKE_PTRCONSTRUCT(PackedVector2Array);
  85. MAKE_PTRCONSTRUCT(PackedVector3Array);
  86. MAKE_PTRCONSTRUCT(PackedColorArray);
  87. MAKE_PTRCONSTRUCT(Variant);
  88. template <class T, class... P>
  89. class VariantConstructor {
  90. template <size_t... Is>
  91. static _FORCE_INLINE_ void construct_helper(T &base, const Variant **p_args, Callable::CallError &r_error, IndexSequence<Is...>) {
  92. r_error.error = Callable::CallError::CALL_OK;
  93. #ifdef DEBUG_METHODS_ENABLED
  94. base = T(VariantCasterAndValidate<P>::cast(p_args, Is, r_error)...);
  95. #else
  96. base = T(VariantCaster<P>::cast(*p_args[Is])...);
  97. #endif
  98. }
  99. template <size_t... Is>
  100. static _FORCE_INLINE_ void validated_construct_helper(T &base, const Variant **p_args, IndexSequence<Is...>) {
  101. base = T((*VariantGetInternalPtr<P>::get_ptr(p_args[Is]))...);
  102. }
  103. template <size_t... Is>
  104. static _FORCE_INLINE_ void ptr_construct_helper(void *base, const void **p_args, IndexSequence<Is...>) {
  105. PtrConstruct<T>::construct(T(PtrToArg<P>::convert(p_args[Is])...), base);
  106. }
  107. public:
  108. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  109. r_error.error = Callable::CallError::CALL_OK;
  110. VariantTypeChanger<T>::change(&r_ret);
  111. construct_helper(*VariantGetInternalPtr<T>::get_ptr(&r_ret), p_args, r_error, BuildIndexSequence<sizeof...(P)>{});
  112. }
  113. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  114. VariantTypeChanger<T>::change(r_ret);
  115. validated_construct_helper(*VariantGetInternalPtr<T>::get_ptr(r_ret), p_args, BuildIndexSequence<sizeof...(P)>{});
  116. }
  117. static void ptr_construct(void *base, const void **p_args) {
  118. ptr_construct_helper(base, p_args, BuildIndexSequence<sizeof...(P)>{});
  119. }
  120. static int get_argument_count() {
  121. return sizeof...(P);
  122. }
  123. static Variant::Type get_argument_type(int p_arg) {
  124. return call_get_argument_type<P...>(p_arg);
  125. }
  126. static Variant::Type get_base_type() {
  127. return GetTypeInfo<T>::VARIANT_TYPE;
  128. }
  129. };
  130. class VariantConstructorObject {
  131. public:
  132. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  133. VariantInternal::clear(&r_ret);
  134. if (p_args[0]->get_type() == Variant::NIL) {
  135. VariantInternal::object_assign_null(&r_ret);
  136. r_error.error = Callable::CallError::CALL_OK;
  137. } else if (p_args[0]->get_type() == Variant::OBJECT) {
  138. VariantInternal::object_assign(&r_ret, p_args[0]);
  139. r_error.error = Callable::CallError::CALL_OK;
  140. } else {
  141. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  142. r_error.argument = 0;
  143. r_error.expected = Variant::OBJECT;
  144. }
  145. }
  146. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  147. VariantInternal::clear(r_ret);
  148. VariantInternal::object_assign(r_ret, p_args[0]);
  149. }
  150. static void ptr_construct(void *base, const void **p_args) {
  151. PtrConstruct<Object *>::construct(PtrToArg<Object *>::convert(p_args[0]), base);
  152. }
  153. static int get_argument_count() {
  154. return 1;
  155. }
  156. static Variant::Type get_argument_type(int p_arg) {
  157. return Variant::OBJECT;
  158. }
  159. static Variant::Type get_base_type() {
  160. return Variant::OBJECT;
  161. }
  162. };
  163. class VariantConstructorNilObject {
  164. public:
  165. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  166. if (p_args[0]->get_type() != Variant::NIL) {
  167. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  168. r_error.argument = 0;
  169. r_error.expected = Variant::NIL;
  170. }
  171. VariantInternal::clear(&r_ret);
  172. VariantInternal::object_assign_null(&r_ret);
  173. }
  174. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  175. VariantInternal::clear(r_ret);
  176. VariantInternal::object_assign_null(r_ret);
  177. }
  178. static void ptr_construct(void *base, const void **p_args) {
  179. PtrConstruct<Object *>::construct(nullptr, base);
  180. }
  181. static int get_argument_count() {
  182. return 1;
  183. }
  184. static Variant::Type get_argument_type(int p_arg) {
  185. return Variant::NIL;
  186. }
  187. static Variant::Type get_base_type() {
  188. return Variant::OBJECT;
  189. }
  190. };
  191. class VariantConstructorCallableArgs {
  192. public:
  193. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  194. ObjectID object_id;
  195. StringName method;
  196. if (p_args[0]->get_type() == Variant::NIL) {
  197. // leave as is
  198. } else if (p_args[0]->get_type() == Variant::OBJECT) {
  199. object_id = VariantInternal::get_object_id(p_args[0]);
  200. } else {
  201. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  202. r_error.argument = 0;
  203. r_error.expected = Variant::OBJECT;
  204. return;
  205. }
  206. if (p_args[1]->get_type() == Variant::STRING_NAME) {
  207. method = *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]);
  208. } else if (p_args[1]->get_type() == Variant::STRING) {
  209. method = *VariantGetInternalPtr<String>::get_ptr(p_args[1]);
  210. } else {
  211. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  212. r_error.argument = 1;
  213. r_error.expected = Variant::STRING_NAME;
  214. return;
  215. }
  216. VariantTypeChanger<Callable>::change(&r_ret);
  217. *VariantGetInternalPtr<Callable>::get_ptr(&r_ret) = Callable(object_id, method);
  218. }
  219. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  220. VariantTypeChanger<Callable>::change(r_ret);
  221. *VariantGetInternalPtr<Callable>::get_ptr(r_ret) = Callable(VariantInternal::get_object_id(p_args[0]), *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]));
  222. }
  223. static void ptr_construct(void *base, const void **p_args) {
  224. PtrConstruct<Callable>::construct(Callable(PtrToArg<Object *>::convert(p_args[0]), PtrToArg<StringName>::convert(p_args[1])), base);
  225. }
  226. static int get_argument_count() {
  227. return 2;
  228. }
  229. static Variant::Type get_argument_type(int p_arg) {
  230. if (p_arg == 0) {
  231. return Variant::OBJECT;
  232. } else {
  233. return Variant::STRING_NAME;
  234. }
  235. }
  236. static Variant::Type get_base_type() {
  237. return Variant::CALLABLE;
  238. }
  239. };
  240. class VariantConstructorSignalArgs {
  241. public:
  242. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  243. ObjectID object_id;
  244. StringName method;
  245. if (p_args[0]->get_type() == Variant::NIL) {
  246. // leave as is
  247. } else if (p_args[0]->get_type() == Variant::OBJECT) {
  248. object_id = VariantInternal::get_object_id(p_args[0]);
  249. } else {
  250. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  251. r_error.argument = 0;
  252. r_error.expected = Variant::OBJECT;
  253. return;
  254. }
  255. if (p_args[1]->get_type() == Variant::STRING_NAME) {
  256. method = *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]);
  257. } else if (p_args[1]->get_type() == Variant::STRING) {
  258. method = *VariantGetInternalPtr<String>::get_ptr(p_args[1]);
  259. } else {
  260. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  261. r_error.argument = 1;
  262. r_error.expected = Variant::STRING_NAME;
  263. return;
  264. }
  265. VariantTypeChanger<Signal>::change(&r_ret);
  266. *VariantGetInternalPtr<Signal>::get_ptr(&r_ret) = Signal(object_id, method);
  267. }
  268. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  269. VariantTypeChanger<Signal>::change(r_ret);
  270. *VariantGetInternalPtr<Signal>::get_ptr(r_ret) = Signal(VariantInternal::get_object_id(p_args[0]), *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]));
  271. }
  272. static void ptr_construct(void *base, const void **p_args) {
  273. PtrConstruct<Signal>::construct(Signal(PtrToArg<Object *>::convert(p_args[0]), PtrToArg<StringName>::convert(p_args[1])), base);
  274. }
  275. static int get_argument_count() {
  276. return 2;
  277. }
  278. static Variant::Type get_argument_type(int p_arg) {
  279. if (p_arg == 0) {
  280. return Variant::OBJECT;
  281. } else {
  282. return Variant::STRING_NAME;
  283. }
  284. }
  285. static Variant::Type get_base_type() {
  286. return Variant::SIGNAL;
  287. }
  288. };
  289. template <class T>
  290. class VariantConstructorToArray {
  291. public:
  292. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  293. if (p_args[0]->get_type() != GetTypeInfo<T>::VARIANT_TYPE) {
  294. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  295. r_error.argument = 0;
  296. r_error.expected = GetTypeInfo<T>::VARIANT_TYPE;
  297. return;
  298. }
  299. VariantTypeChanger<Array>::change(&r_ret);
  300. Array &dst_arr = *VariantGetInternalPtr<Array>::get_ptr(&r_ret);
  301. const T &src_arr = *VariantGetInternalPtr<T>::get_ptr(p_args[0]);
  302. int size = src_arr.size();
  303. dst_arr.resize(size);
  304. for (int i = 0; i < size; i++) {
  305. dst_arr[i] = src_arr[i];
  306. }
  307. }
  308. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  309. VariantTypeChanger<Array>::change(r_ret);
  310. Array &dst_arr = *VariantGetInternalPtr<Array>::get_ptr(r_ret);
  311. const T &src_arr = *VariantGetInternalPtr<T>::get_ptr(p_args[0]);
  312. int size = src_arr.size();
  313. dst_arr.resize(size);
  314. for (int i = 0; i < size; i++) {
  315. dst_arr[i] = src_arr[i];
  316. }
  317. }
  318. static void ptr_construct(void *base, const void **p_args) {
  319. Array dst_arr;
  320. T src_arr = PtrToArg<T>::convert(p_args[0]);
  321. int size = src_arr.size();
  322. dst_arr.resize(size);
  323. for (int i = 0; i < size; i++) {
  324. dst_arr[i] = src_arr[i];
  325. }
  326. PtrConstruct<Array>::construct(dst_arr, base);
  327. }
  328. static int get_argument_count() {
  329. return 1;
  330. }
  331. static Variant::Type get_argument_type(int p_arg) {
  332. return GetTypeInfo<T>::VARIANT_TYPE;
  333. }
  334. static Variant::Type get_base_type() {
  335. return Variant::ARRAY;
  336. }
  337. };
  338. template <class T>
  339. class VariantConstructorFromArray {
  340. public:
  341. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  342. if (p_args[0]->get_type() != Variant::ARRAY) {
  343. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  344. r_error.argument = 0;
  345. r_error.expected = Variant::ARRAY;
  346. return;
  347. }
  348. VariantTypeChanger<T>::change(&r_ret);
  349. const Array &src_arr = *VariantGetInternalPtr<Array>::get_ptr(p_args[0]);
  350. T &dst_arr = *VariantGetInternalPtr<T>::get_ptr(&r_ret);
  351. int size = src_arr.size();
  352. dst_arr.resize(size);
  353. for (int i = 0; i < size; i++) {
  354. dst_arr.write[i] = src_arr[i];
  355. }
  356. }
  357. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  358. VariantTypeChanger<T>::change(r_ret);
  359. const Array &src_arr = *VariantGetInternalPtr<Array>::get_ptr(p_args[0]);
  360. T &dst_arr = *VariantGetInternalPtr<T>::get_ptr(r_ret);
  361. int size = src_arr.size();
  362. dst_arr.resize(size);
  363. for (int i = 0; i < size; i++) {
  364. dst_arr.write[i] = src_arr[i];
  365. }
  366. }
  367. static void ptr_construct(void *base, const void **p_args) {
  368. Array src_arr = PtrToArg<Array>::convert(p_args[0]);
  369. T dst_arr;
  370. int size = src_arr.size();
  371. dst_arr.resize(size);
  372. for (int i = 0; i < size; i++) {
  373. dst_arr.write[i] = src_arr[i];
  374. }
  375. PtrConstruct<T>::construct(dst_arr, base);
  376. }
  377. static int get_argument_count() {
  378. return 1;
  379. }
  380. static Variant::Type get_argument_type(int p_arg) {
  381. return Variant::ARRAY;
  382. }
  383. static Variant::Type get_base_type() {
  384. return GetTypeInfo<T>::VARIANT_TYPE;
  385. }
  386. };
  387. class VariantConstructorNil {
  388. public:
  389. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  390. if (p_args[0]->get_type() != Variant::NIL) {
  391. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  392. r_error.argument = 0;
  393. r_error.expected = Variant::NIL;
  394. return;
  395. }
  396. r_error.error = Callable::CallError::CALL_OK;
  397. VariantInternal::clear(&r_ret);
  398. }
  399. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  400. VariantInternal::clear(r_ret);
  401. }
  402. static void ptr_construct(void *base, const void **p_args) {
  403. PtrConstruct<Variant>::construct(Variant(), base);
  404. }
  405. static int get_argument_count() {
  406. return 1;
  407. }
  408. static Variant::Type get_argument_type(int p_arg) {
  409. return Variant::NIL;
  410. }
  411. static Variant::Type get_base_type() {
  412. return Variant::NIL;
  413. }
  414. };
  415. template <class T>
  416. class VariantConstructNoArgs {
  417. public:
  418. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  419. VariantTypeChanger<T>::change_and_reset(&r_ret);
  420. r_error.error = Callable::CallError::CALL_OK;
  421. }
  422. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  423. VariantTypeChanger<T>::change_and_reset(r_ret);
  424. }
  425. static void ptr_construct(void *base, const void **p_args) {
  426. PtrConstruct<T>::construct(T(), base);
  427. }
  428. static int get_argument_count() {
  429. return 0;
  430. }
  431. static Variant::Type get_argument_type(int p_arg) {
  432. return Variant::NIL;
  433. }
  434. static Variant::Type get_base_type() {
  435. return GetTypeInfo<T>::VARIANT_TYPE;
  436. }
  437. };
  438. class VariantConstructNoArgsNil {
  439. public:
  440. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  441. VariantInternal::clear(&r_ret);
  442. r_error.error = Callable::CallError::CALL_OK;
  443. }
  444. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  445. VariantInternal::clear(r_ret);
  446. }
  447. static void ptr_construct(void *base, const void **p_args) {
  448. ERR_FAIL_MSG("can't ptrcall nil constructor");
  449. }
  450. static int get_argument_count() {
  451. return 0;
  452. }
  453. static Variant::Type get_argument_type(int p_arg) {
  454. return Variant::NIL;
  455. }
  456. static Variant::Type get_base_type() {
  457. return Variant::NIL;
  458. }
  459. };
  460. class VariantConstructNoArgsObject {
  461. public:
  462. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  463. VariantInternal::clear(&r_ret);
  464. VariantInternal::object_assign_null(&r_ret);
  465. r_error.error = Callable::CallError::CALL_OK;
  466. }
  467. static void validated_construct(Variant *r_ret, const Variant **p_args) {
  468. VariantInternal::clear(r_ret);
  469. VariantInternal::object_assign_null(r_ret);
  470. }
  471. static void ptr_construct(void *base, const void **p_args) {
  472. PtrConstruct<Object *>::construct(nullptr, base);
  473. }
  474. static int get_argument_count() {
  475. return 0;
  476. }
  477. static Variant::Type get_argument_type(int p_arg) {
  478. return Variant::NIL;
  479. }
  480. static Variant::Type get_base_type() {
  481. return Variant::OBJECT;
  482. }
  483. };
  484. struct VariantConstructData {
  485. void (*construct)(Variant &r_base, const Variant **p_args, Callable::CallError &r_error);
  486. Variant::ValidatedConstructor validated_construct;
  487. Variant::PTRConstructor ptr_construct;
  488. Variant::Type (*get_argument_type)(int);
  489. int argument_count;
  490. Vector<String> arg_names;
  491. };
  492. static LocalVector<VariantConstructData> construct_data[Variant::VARIANT_MAX];
  493. template <class T>
  494. static void add_constructor(const Vector<String> &arg_names) {
  495. ERR_FAIL_COND_MSG(arg_names.size() != T::get_argument_count(), "Argument names size mismatch for " + Variant::get_type_name(T::get_base_type()) + ".");
  496. VariantConstructData cd;
  497. cd.construct = T::construct;
  498. cd.validated_construct = T::validated_construct;
  499. cd.ptr_construct = T::ptr_construct;
  500. cd.get_argument_type = T::get_argument_type;
  501. cd.argument_count = T::get_argument_count();
  502. cd.arg_names = arg_names;
  503. construct_data[T::get_base_type()].push_back(cd);
  504. }
  505. void Variant::_register_variant_constructors() {
  506. add_constructor<VariantConstructNoArgsNil>(sarray());
  507. add_constructor<VariantConstructorNil>(sarray("from"));
  508. add_constructor<VariantConstructNoArgs<bool>>(sarray());
  509. add_constructor<VariantConstructor<bool, bool>>(sarray("from"));
  510. add_constructor<VariantConstructor<bool, int64_t>>(sarray("from"));
  511. add_constructor<VariantConstructor<bool, double>>(sarray("from"));
  512. add_constructor<VariantConstructNoArgs<int64_t>>(sarray());
  513. add_constructor<VariantConstructor<int64_t, int64_t>>(sarray("from"));
  514. add_constructor<VariantConstructor<int64_t, double>>(sarray("from"));
  515. add_constructor<VariantConstructor<int64_t, bool>>(sarray("from"));
  516. add_constructor<VariantConstructNoArgs<double>>(sarray());
  517. add_constructor<VariantConstructor<double, double>>(sarray("from"));
  518. add_constructor<VariantConstructor<double, int64_t>>(sarray("from"));
  519. add_constructor<VariantConstructor<double, bool>>(sarray("from"));
  520. add_constructor<VariantConstructNoArgs<String>>(sarray());
  521. add_constructor<VariantConstructor<String, String>>(sarray("from"));
  522. add_constructor<VariantConstructor<String, StringName>>(sarray("from"));
  523. add_constructor<VariantConstructor<String, NodePath>>(sarray("from"));
  524. add_constructor<VariantConstructNoArgs<Vector2>>(sarray());
  525. add_constructor<VariantConstructor<Vector2, Vector2>>(sarray("from"));
  526. add_constructor<VariantConstructor<Vector2, Vector2i>>(sarray("from"));
  527. add_constructor<VariantConstructor<Vector2, double, double>>(sarray("x", "y"));
  528. add_constructor<VariantConstructNoArgs<Vector2i>>(sarray());
  529. add_constructor<VariantConstructor<Vector2i, Vector2i>>(sarray("from"));
  530. add_constructor<VariantConstructor<Vector2i, Vector2>>(sarray("from"));
  531. add_constructor<VariantConstructor<Vector2i, int64_t, int64_t>>(sarray("x", "y"));
  532. add_constructor<VariantConstructNoArgs<Rect2>>(sarray());
  533. add_constructor<VariantConstructor<Rect2, Rect2>>(sarray("from"));
  534. add_constructor<VariantConstructor<Rect2, Rect2i>>(sarray("from"));
  535. add_constructor<VariantConstructor<Rect2, Vector2, Vector2>>(sarray("position", "size"));
  536. add_constructor<VariantConstructor<Rect2, double, double, double, double>>(sarray("x", "y", "width", "height"));
  537. add_constructor<VariantConstructNoArgs<Rect2i>>(sarray());
  538. add_constructor<VariantConstructor<Rect2i, Rect2i>>(sarray("from"));
  539. add_constructor<VariantConstructor<Rect2i, Rect2>>(sarray("from"));
  540. add_constructor<VariantConstructor<Rect2i, Vector2i, Vector2i>>(sarray("position", "size"));
  541. add_constructor<VariantConstructor<Rect2i, int64_t, int64_t, int64_t, int64_t>>(sarray("x", "y", "width", "height"));
  542. add_constructor<VariantConstructNoArgs<Vector3>>(sarray());
  543. add_constructor<VariantConstructor<Vector3, Vector3>>(sarray("from"));
  544. add_constructor<VariantConstructor<Vector3, Vector3i>>(sarray("from"));
  545. add_constructor<VariantConstructor<Vector3, double, double, double>>(sarray("x", "y", "z"));
  546. add_constructor<VariantConstructNoArgs<Vector3i>>(sarray());
  547. add_constructor<VariantConstructor<Vector3i, Vector3i>>(sarray("from"));
  548. add_constructor<VariantConstructor<Vector3i, Vector3>>(sarray("from"));
  549. add_constructor<VariantConstructor<Vector3i, int64_t, int64_t, int64_t>>(sarray("x", "y", "z"));
  550. add_constructor<VariantConstructNoArgs<Transform2D>>(sarray());
  551. add_constructor<VariantConstructor<Transform2D, Transform2D>>(sarray("from"));
  552. add_constructor<VariantConstructor<Transform2D, float, Vector2>>(sarray("rotation", "position"));
  553. add_constructor<VariantConstructor<Transform2D, Vector2, Vector2, Vector2>>(sarray("x_axis", "y_axis", "origin"));
  554. add_constructor<VariantConstructNoArgs<Plane>>(sarray());
  555. add_constructor<VariantConstructor<Plane, Plane>>(sarray("from"));
  556. add_constructor<VariantConstructor<Plane, Vector3, double>>(sarray("normal", "d"));
  557. add_constructor<VariantConstructor<Plane, Vector3, Vector3>>(sarray("point", "normal"));
  558. add_constructor<VariantConstructor<Plane, Vector3, Vector3, Vector3>>(sarray("point1", "point2", "point3"));
  559. add_constructor<VariantConstructor<Plane, double, double, double, double>>(sarray("a", "b", "c", "d"));
  560. add_constructor<VariantConstructNoArgs<Quat>>(sarray());
  561. add_constructor<VariantConstructor<Quat, Quat>>(sarray("from"));
  562. add_constructor<VariantConstructor<Quat, Basis>>(sarray("from"));
  563. add_constructor<VariantConstructor<Quat, Vector3>>(sarray("euler"));
  564. add_constructor<VariantConstructor<Quat, Vector3, double>>(sarray("axis", "angle"));
  565. add_constructor<VariantConstructor<Quat, Vector3, Vector3>>(sarray("arc_from", "arc_to"));
  566. add_constructor<VariantConstructor<Quat, double, double, double, double>>(sarray("x", "y", "z", "w"));
  567. add_constructor<VariantConstructNoArgs<::AABB>>(sarray());
  568. add_constructor<VariantConstructor<::AABB, ::AABB>>(sarray("from"));
  569. add_constructor<VariantConstructor<::AABB, Vector3, Vector3>>(sarray("position", "size"));
  570. add_constructor<VariantConstructNoArgs<Basis>>(sarray());
  571. add_constructor<VariantConstructor<Basis, Basis>>(sarray("from"));
  572. add_constructor<VariantConstructor<Basis, Quat>>(sarray("from"));
  573. add_constructor<VariantConstructor<Basis, Vector3>>(sarray("euler"));
  574. add_constructor<VariantConstructor<Basis, Vector3, double>>(sarray("axis", "phi"));
  575. add_constructor<VariantConstructor<Basis, Vector3, Vector3, Vector3>>(sarray("x_axis", "y_axis", "z_axis"));
  576. add_constructor<VariantConstructNoArgs<Transform3D>>(sarray());
  577. add_constructor<VariantConstructor<Transform3D, Transform3D>>(sarray("from"));
  578. add_constructor<VariantConstructor<Transform3D, Basis, Vector3>>(sarray("basis", "origin"));
  579. add_constructor<VariantConstructor<Transform3D, Vector3, Vector3, Vector3, Vector3>>(sarray("x_axis", "y_axis", "z_axis", "origin"));
  580. add_constructor<VariantConstructNoArgs<Color>>(sarray());
  581. add_constructor<VariantConstructor<Color, Color>>(sarray("from"));
  582. add_constructor<VariantConstructor<Color, Color, double>>(sarray("from", "alpha"));
  583. add_constructor<VariantConstructor<Color, double, double, double>>(sarray("r", "g", "b"));
  584. add_constructor<VariantConstructor<Color, double, double, double, double>>(sarray("r", "g", "b", "a"));
  585. add_constructor<VariantConstructor<Color, String>>(sarray("code"));
  586. add_constructor<VariantConstructor<Color, String, double>>(sarray("code", "alpha"));
  587. add_constructor<VariantConstructNoArgs<StringName>>(sarray());
  588. add_constructor<VariantConstructor<StringName, StringName>>(sarray("from"));
  589. add_constructor<VariantConstructor<StringName, String>>(sarray("from"));
  590. add_constructor<VariantConstructNoArgs<NodePath>>(sarray());
  591. add_constructor<VariantConstructor<NodePath, NodePath>>(sarray("from"));
  592. add_constructor<VariantConstructor<NodePath, String>>(sarray("from"));
  593. add_constructor<VariantConstructNoArgs<::RID>>(sarray());
  594. add_constructor<VariantConstructor<::RID, ::RID>>(sarray("from"));
  595. add_constructor<VariantConstructNoArgsObject>(sarray());
  596. add_constructor<VariantConstructorObject>(sarray("from"));
  597. add_constructor<VariantConstructorNilObject>(sarray("from"));
  598. add_constructor<VariantConstructNoArgs<Callable>>(sarray());
  599. add_constructor<VariantConstructor<Callable, Callable>>(sarray("from"));
  600. add_constructor<VariantConstructorCallableArgs>(sarray("object", "method"));
  601. add_constructor<VariantConstructNoArgs<Signal>>(sarray());
  602. add_constructor<VariantConstructor<Signal, Signal>>(sarray("from"));
  603. add_constructor<VariantConstructorSignalArgs>(sarray("object", "signal"));
  604. add_constructor<VariantConstructNoArgs<Dictionary>>(sarray());
  605. add_constructor<VariantConstructor<Dictionary, Dictionary>>(sarray("from"));
  606. add_constructor<VariantConstructNoArgs<Array>>(sarray());
  607. add_constructor<VariantConstructor<Array, Array>>(sarray("from"));
  608. add_constructor<VariantConstructorToArray<PackedByteArray>>(sarray("from"));
  609. add_constructor<VariantConstructorToArray<PackedInt32Array>>(sarray("from"));
  610. add_constructor<VariantConstructorToArray<PackedInt64Array>>(sarray("from"));
  611. add_constructor<VariantConstructorToArray<PackedFloat32Array>>(sarray("from"));
  612. add_constructor<VariantConstructorToArray<PackedFloat64Array>>(sarray("from"));
  613. add_constructor<VariantConstructorToArray<PackedStringArray>>(sarray("from"));
  614. add_constructor<VariantConstructorToArray<PackedVector2Array>>(sarray("from"));
  615. add_constructor<VariantConstructorToArray<PackedVector3Array>>(sarray("from"));
  616. add_constructor<VariantConstructorToArray<PackedColorArray>>(sarray("from"));
  617. add_constructor<VariantConstructNoArgs<PackedByteArray>>(sarray());
  618. add_constructor<VariantConstructor<PackedByteArray, PackedByteArray>>(sarray("from"));
  619. add_constructor<VariantConstructorFromArray<PackedByteArray>>(sarray("from"));
  620. add_constructor<VariantConstructNoArgs<PackedInt32Array>>(sarray());
  621. add_constructor<VariantConstructor<PackedInt32Array, PackedInt32Array>>(sarray("from"));
  622. add_constructor<VariantConstructorFromArray<PackedInt32Array>>(sarray("from"));
  623. add_constructor<VariantConstructNoArgs<PackedInt64Array>>(sarray());
  624. add_constructor<VariantConstructor<PackedInt64Array, PackedInt64Array>>(sarray("from"));
  625. add_constructor<VariantConstructorFromArray<PackedInt64Array>>(sarray("from"));
  626. add_constructor<VariantConstructNoArgs<PackedFloat32Array>>(sarray());
  627. add_constructor<VariantConstructor<PackedFloat32Array, PackedFloat32Array>>(sarray("from"));
  628. add_constructor<VariantConstructorFromArray<PackedFloat32Array>>(sarray("from"));
  629. add_constructor<VariantConstructNoArgs<PackedFloat64Array>>(sarray());
  630. add_constructor<VariantConstructor<PackedFloat64Array, PackedFloat64Array>>(sarray("from"));
  631. add_constructor<VariantConstructorFromArray<PackedFloat64Array>>(sarray("from"));
  632. add_constructor<VariantConstructNoArgs<PackedStringArray>>(sarray());
  633. add_constructor<VariantConstructor<PackedStringArray, PackedStringArray>>(sarray("from"));
  634. add_constructor<VariantConstructorFromArray<PackedStringArray>>(sarray("from"));
  635. add_constructor<VariantConstructNoArgs<PackedVector2Array>>(sarray());
  636. add_constructor<VariantConstructor<PackedVector2Array, PackedVector2Array>>(sarray("from"));
  637. add_constructor<VariantConstructorFromArray<PackedVector2Array>>(sarray("from"));
  638. add_constructor<VariantConstructNoArgs<PackedVector3Array>>(sarray());
  639. add_constructor<VariantConstructor<PackedVector3Array, PackedVector3Array>>(sarray("from"));
  640. add_constructor<VariantConstructorFromArray<PackedVector3Array>>(sarray("from"));
  641. add_constructor<VariantConstructNoArgs<PackedColorArray>>(sarray());
  642. add_constructor<VariantConstructor<PackedColorArray, PackedColorArray>>(sarray("from"));
  643. add_constructor<VariantConstructorFromArray<PackedColorArray>>(sarray("from"));
  644. }
  645. void Variant::_unregister_variant_constructors() {
  646. for (int i = 0; i < Variant::VARIANT_MAX; i++) {
  647. construct_data[i].clear();
  648. }
  649. }
  650. void Variant::construct(Variant::Type p_type, Variant &base, const Variant **p_args, int p_argcount, Callable::CallError &r_error) {
  651. ERR_FAIL_INDEX(p_type, Variant::VARIANT_MAX);
  652. uint32_t s = construct_data[p_type].size();
  653. for (uint32_t i = 0; i < s; i++) {
  654. int argc = construct_data[p_type][i].argument_count;
  655. if (argc != p_argcount) {
  656. continue;
  657. }
  658. bool args_match = true;
  659. for (int j = 0; j < argc; j++) {
  660. if (!Variant::can_convert_strict(p_args[j]->get_type(), construct_data[p_type][i].get_argument_type(j))) {
  661. args_match = false;
  662. break;
  663. }
  664. }
  665. if (!args_match) {
  666. continue;
  667. }
  668. construct_data[p_type][i].construct(base, p_args, r_error);
  669. return;
  670. }
  671. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  672. }
  673. int Variant::get_constructor_count(Variant::Type p_type) {
  674. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, -1);
  675. return construct_data[p_type].size();
  676. }
  677. Variant::ValidatedConstructor Variant::get_validated_constructor(Variant::Type p_type, int p_constructor) {
  678. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, nullptr);
  679. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), nullptr);
  680. return construct_data[p_type][p_constructor].validated_construct;
  681. }
  682. Variant::PTRConstructor Variant::get_ptr_constructor(Variant::Type p_type, int p_constructor) {
  683. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, nullptr);
  684. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), nullptr);
  685. return construct_data[p_type][p_constructor].ptr_construct;
  686. }
  687. int Variant::get_constructor_argument_count(Variant::Type p_type, int p_constructor) {
  688. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, -1);
  689. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), -1);
  690. return construct_data[p_type][p_constructor].argument_count;
  691. }
  692. Variant::Type Variant::get_constructor_argument_type(Variant::Type p_type, int p_constructor, int p_argument) {
  693. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, Variant::VARIANT_MAX);
  694. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), Variant::VARIANT_MAX);
  695. return construct_data[p_type][p_constructor].get_argument_type(p_argument);
  696. }
  697. String Variant::get_constructor_argument_name(Variant::Type p_type, int p_constructor, int p_argument) {
  698. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, String());
  699. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), String());
  700. return construct_data[p_type][p_constructor].arg_names[p_argument];
  701. }
  702. void VariantInternal::object_assign(Variant *v, const Object *o) {
  703. if (o) {
  704. if (o->is_reference()) {
  705. Reference *reference = const_cast<Reference *>(static_cast<const Reference *>(o));
  706. if (!reference->init_ref()) {
  707. v->_get_obj().obj = nullptr;
  708. v->_get_obj().id = ObjectID();
  709. return;
  710. }
  711. }
  712. v->_get_obj().obj = const_cast<Object *>(o);
  713. v->_get_obj().id = o->get_instance_id();
  714. } else {
  715. v->_get_obj().obj = nullptr;
  716. v->_get_obj().id = ObjectID();
  717. }
  718. }
  719. void Variant::get_constructor_list(Type p_type, List<MethodInfo> *r_list) {
  720. ERR_FAIL_INDEX(p_type, Variant::VARIANT_MAX);
  721. MethodInfo mi;
  722. mi.return_val.type = p_type;
  723. mi.name = get_type_name(p_type);
  724. for (int i = 0; i < get_constructor_count(p_type); i++) {
  725. int ac = get_constructor_argument_count(p_type, i);
  726. mi.arguments.clear();
  727. for (int j = 0; j < ac; j++) {
  728. PropertyInfo arg;
  729. arg.name = get_constructor_argument_name(p_type, i, j);
  730. arg.type = get_constructor_argument_type(p_type, i, j);
  731. mi.arguments.push_back(arg);
  732. }
  733. r_list->push_back(mi);
  734. }
  735. }