variant_construct.cpp 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781
  1. /*************************************************************************/
  2. /* variant_construct.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2020 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2020 Godot Engine contributors (cf. AUTHORS.md). */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /*************************************************************************/
  30. #include "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. _FORCE_INLINE_ void sarray_add_str(Vector<String> &arr) {
  40. }
  41. _FORCE_INLINE_ void sarray_add_str(Vector<String> &arr, const String &p_str) {
  42. arr.push_back(p_str);
  43. }
  44. template <class... P>
  45. _FORCE_INLINE_ void sarray_add_str(Vector<String> &arr, const String &p_str, P... p_args) {
  46. arr.push_back(p_str);
  47. sarray_add_str(arr, p_args...);
  48. }
  49. template <class... P>
  50. _FORCE_INLINE_ Vector<String> sarray(P... p_args) {
  51. Vector<String> arr;
  52. sarray_add_str(arr, p_args...);
  53. return arr;
  54. }
  55. template <class T, class... P>
  56. class VariantConstructor {
  57. template <size_t... Is>
  58. static _FORCE_INLINE_ void construct_helper(T &base, const Variant **p_args, Callable::CallError &r_error, IndexSequence<Is...>) {
  59. r_error.error = Callable::CallError::CALL_OK;
  60. #ifdef DEBUG_METHODS_ENABLED
  61. base = T(VariantCasterAndValidate<P>::cast(p_args, Is, r_error)...);
  62. #else
  63. base = T(VariantCaster<P>::cast(*p_args[Is])...);
  64. #endif
  65. }
  66. template <size_t... Is>
  67. static _FORCE_INLINE_ void validated_construct_helper(T &base, const Variant **p_args, IndexSequence<Is...>) {
  68. base = T((*VariantGetInternalPtr<P>::get_ptr(p_args[Is]))...);
  69. }
  70. template <size_t... Is>
  71. static _FORCE_INLINE_ void ptr_construct_helper(void *base, const void **p_args, IndexSequence<Is...>) {
  72. PtrToArg<T>::encode(T(PtrToArg<P>::convert(p_args[Is])...), base);
  73. }
  74. public:
  75. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  76. r_error.error = Callable::CallError::CALL_OK;
  77. VariantTypeChanger<T>::change(&r_ret);
  78. construct_helper(*VariantGetInternalPtr<T>::get_ptr(&r_ret), p_args, r_error, BuildIndexSequence<sizeof...(P)>{});
  79. }
  80. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  81. VariantTypeChanger<T>::change(&r_ret);
  82. validated_construct_helper(*VariantGetInternalPtr<T>::get_ptr(&r_ret), p_args, BuildIndexSequence<sizeof...(P)>{});
  83. }
  84. static void ptr_construct(void *base, const void **p_args) {
  85. ptr_construct_helper(base, p_args, BuildIndexSequence<sizeof...(P)>{});
  86. }
  87. static int get_argument_count() {
  88. return sizeof...(P);
  89. }
  90. static Variant::Type get_argument_type(int p_arg) {
  91. return call_get_argument_type<P...>(p_arg);
  92. }
  93. static Variant::Type get_base_type() {
  94. return GetTypeInfo<T>::VARIANT_TYPE;
  95. }
  96. };
  97. class VariantConstructorObject {
  98. public:
  99. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  100. VariantInternal::clear(&r_ret);
  101. if (p_args[0]->get_type() == Variant::NIL) {
  102. VariantInternal::object_assign_null(&r_ret);
  103. r_error.error = Callable::CallError::CALL_OK;
  104. } else if (p_args[0]->get_type() == Variant::OBJECT) {
  105. VariantInternal::object_assign(&r_ret, p_args[0]);
  106. r_error.error = Callable::CallError::CALL_OK;
  107. } else {
  108. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  109. r_error.argument = 0;
  110. r_error.expected = Variant::OBJECT;
  111. }
  112. }
  113. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  114. VariantInternal::clear(&r_ret);
  115. VariantInternal::object_assign(&r_ret, p_args[0]);
  116. }
  117. static void ptr_construct(void *base, const void **p_args) {
  118. PtrToArg<Object *>::encode(PtrToArg<Object *>::convert(p_args[0]), base);
  119. }
  120. static int get_argument_count() {
  121. return 1;
  122. }
  123. static Variant::Type get_argument_type(int p_arg) {
  124. return Variant::OBJECT;
  125. }
  126. static Variant::Type get_base_type() {
  127. return Variant::OBJECT;
  128. }
  129. };
  130. class VariantConstructorNilObject {
  131. public:
  132. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  133. if (p_args[0]->get_type() != Variant::NIL) {
  134. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  135. r_error.argument = 0;
  136. r_error.expected = Variant::NIL;
  137. }
  138. VariantInternal::clear(&r_ret);
  139. VariantInternal::object_assign_null(&r_ret);
  140. }
  141. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  142. VariantInternal::clear(&r_ret);
  143. VariantInternal::object_assign_null(&r_ret);
  144. }
  145. static void ptr_construct(void *base, const void **p_args) {
  146. PtrToArg<Object *>::encode(nullptr, base);
  147. }
  148. static int get_argument_count() {
  149. return 1;
  150. }
  151. static Variant::Type get_argument_type(int p_arg) {
  152. return Variant::NIL;
  153. }
  154. static Variant::Type get_base_type() {
  155. return Variant::OBJECT;
  156. }
  157. };
  158. class VariantConstructorCallableArgs {
  159. public:
  160. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  161. ObjectID object_id;
  162. StringName method;
  163. if (p_args[0]->get_type() == Variant::NIL) {
  164. // leave as is
  165. } else if (p_args[0]->get_type() == Variant::OBJECT) {
  166. object_id = VariantInternal::get_object_id(p_args[0]);
  167. } else {
  168. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  169. r_error.argument = 0;
  170. r_error.expected = Variant::OBJECT;
  171. return;
  172. }
  173. if (p_args[1]->get_type() == Variant::STRING_NAME) {
  174. method = *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]);
  175. } else if (p_args[1]->get_type() == Variant::STRING) {
  176. method = *VariantGetInternalPtr<String>::get_ptr(p_args[1]);
  177. } else {
  178. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  179. r_error.argument = 1;
  180. r_error.expected = Variant::STRING_NAME;
  181. return;
  182. }
  183. VariantTypeChanger<Callable>::change(&r_ret);
  184. *VariantGetInternalPtr<Callable>::get_ptr(&r_ret) = Callable(object_id, method);
  185. }
  186. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  187. VariantTypeChanger<Callable>::change(&r_ret);
  188. *VariantGetInternalPtr<Callable>::get_ptr(&r_ret) = Callable(VariantInternal::get_object_id(p_args[0]), *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]));
  189. }
  190. static void ptr_construct(void *base, const void **p_args) {
  191. PtrToArg<Callable>::encode(Callable(PtrToArg<Object *>::convert(p_args[0]), PtrToArg<StringName>::convert(p_args[1])), base);
  192. }
  193. static int get_argument_count() {
  194. return 2;
  195. }
  196. static Variant::Type get_argument_type(int p_arg) {
  197. if (p_arg == 0) {
  198. return Variant::OBJECT;
  199. } else {
  200. return Variant::STRING_NAME;
  201. }
  202. }
  203. static Variant::Type get_base_type() {
  204. return Variant::CALLABLE;
  205. }
  206. };
  207. class VariantConstructorSignalArgs {
  208. public:
  209. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  210. ObjectID object_id;
  211. StringName method;
  212. if (p_args[0]->get_type() == Variant::NIL) {
  213. // leave as is
  214. } else if (p_args[0]->get_type() == Variant::OBJECT) {
  215. object_id = VariantInternal::get_object_id(p_args[0]);
  216. } else {
  217. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  218. r_error.argument = 0;
  219. r_error.expected = Variant::OBJECT;
  220. return;
  221. }
  222. if (p_args[1]->get_type() == Variant::STRING_NAME) {
  223. method = *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]);
  224. } else if (p_args[1]->get_type() == Variant::STRING) {
  225. method = *VariantGetInternalPtr<String>::get_ptr(p_args[1]);
  226. } else {
  227. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  228. r_error.argument = 1;
  229. r_error.expected = Variant::STRING_NAME;
  230. return;
  231. }
  232. VariantTypeChanger<Signal>::change(&r_ret);
  233. *VariantGetInternalPtr<Signal>::get_ptr(&r_ret) = Signal(object_id, method);
  234. }
  235. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  236. VariantTypeChanger<Signal>::change(&r_ret);
  237. *VariantGetInternalPtr<Signal>::get_ptr(&r_ret) = Signal(VariantInternal::get_object_id(p_args[0]), *VariantGetInternalPtr<StringName>::get_ptr(p_args[1]));
  238. }
  239. static void ptr_construct(void *base, const void **p_args) {
  240. PtrToArg<Signal>::encode(Signal(PtrToArg<Object *>::convert(p_args[0]), PtrToArg<StringName>::convert(p_args[1])), base);
  241. }
  242. static int get_argument_count() {
  243. return 2;
  244. }
  245. static Variant::Type get_argument_type(int p_arg) {
  246. if (p_arg == 0) {
  247. return Variant::OBJECT;
  248. } else {
  249. return Variant::STRING_NAME;
  250. }
  251. }
  252. static Variant::Type get_base_type() {
  253. return Variant::CALLABLE;
  254. }
  255. };
  256. template <class T>
  257. class VariantConstructorToArray {
  258. public:
  259. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  260. if (p_args[0]->get_type() != GetTypeInfo<T>::VARIANT_TYPE) {
  261. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  262. r_error.argument = 0;
  263. r_error.expected = GetTypeInfo<T>::VARIANT_TYPE;
  264. return;
  265. }
  266. VariantTypeChanger<Array>::change(&r_ret);
  267. Array &dst_arr = *VariantGetInternalPtr<Array>::get_ptr(&r_ret);
  268. const T &src_arr = *VariantGetInternalPtr<T>::get_ptr(p_args[0]);
  269. int size = src_arr.size();
  270. dst_arr.resize(size);
  271. for (int i = 0; i < size; i++) {
  272. dst_arr[i] = src_arr[i];
  273. }
  274. }
  275. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  276. VariantTypeChanger<Array>::change(&r_ret);
  277. Array &dst_arr = *VariantGetInternalPtr<Array>::get_ptr(&r_ret);
  278. const T &src_arr = *VariantGetInternalPtr<T>::get_ptr(p_args[0]);
  279. int size = src_arr.size();
  280. dst_arr.resize(size);
  281. for (int i = 0; i < size; i++) {
  282. dst_arr[i] = src_arr[i];
  283. }
  284. }
  285. static void ptr_construct(void *base, const void **p_args) {
  286. Array dst_arr;
  287. T src_arr = PtrToArg<T>::convert(p_args[0]);
  288. int size = src_arr.size();
  289. dst_arr.resize(size);
  290. for (int i = 0; i < size; i++) {
  291. dst_arr[i] = src_arr[i];
  292. }
  293. PtrToArg<Array>::encode(dst_arr, base);
  294. }
  295. static int get_argument_count() {
  296. return 1;
  297. }
  298. static Variant::Type get_argument_type(int p_arg) {
  299. return GetTypeInfo<T>::VARIANT_TYPE;
  300. }
  301. static Variant::Type get_base_type() {
  302. return Variant::ARRAY;
  303. }
  304. };
  305. template <class T>
  306. class VariantConstructorFromArray {
  307. public:
  308. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  309. if (p_args[0]->get_type() != Variant::ARRAY) {
  310. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  311. r_error.argument = 0;
  312. r_error.expected = Variant::ARRAY;
  313. return;
  314. }
  315. VariantTypeChanger<T>::change(&r_ret);
  316. const Array &src_arr = *VariantGetInternalPtr<Array>::get_ptr(p_args[0]);
  317. T &dst_arr = *VariantGetInternalPtr<T>::get_ptr(&r_ret);
  318. int size = src_arr.size();
  319. dst_arr.resize(size);
  320. for (int i = 0; i < size; i++) {
  321. dst_arr.write[i] = src_arr[i];
  322. }
  323. }
  324. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  325. VariantTypeChanger<T>::change(&r_ret);
  326. const Array &src_arr = *VariantGetInternalPtr<Array>::get_ptr(p_args[0]);
  327. T &dst_arr = *VariantGetInternalPtr<T>::get_ptr(&r_ret);
  328. int size = src_arr.size();
  329. dst_arr.resize(size);
  330. for (int i = 0; i < size; i++) {
  331. dst_arr.write[i] = src_arr[i];
  332. }
  333. }
  334. static void ptr_construct(void *base, const void **p_args) {
  335. Array src_arr = PtrToArg<Array>::convert(p_args[0]);
  336. T dst_arr;
  337. int size = src_arr.size();
  338. dst_arr.resize(size);
  339. for (int i = 0; i < size; i++) {
  340. dst_arr.write[i] = src_arr[i];
  341. }
  342. PtrToArg<T>::encode(dst_arr, base);
  343. }
  344. static int get_argument_count() {
  345. return 1;
  346. }
  347. static Variant::Type get_argument_type(int p_arg) {
  348. return Variant::ARRAY;
  349. }
  350. static Variant::Type get_base_type() {
  351. return GetTypeInfo<T>::VARIANT_TYPE;
  352. }
  353. };
  354. class VariantConstructorNil {
  355. public:
  356. static void construct(Variant &r_ret, const Variant **p_args, Callable::CallError &r_error) {
  357. if (p_args[0]->get_type() != Variant::NIL) {
  358. r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT;
  359. r_error.argument = 0;
  360. r_error.expected = Variant::NIL;
  361. return;
  362. }
  363. r_error.error = Callable::CallError::CALL_OK;
  364. VariantInternal::clear(&r_ret);
  365. }
  366. static void validated_construct(Variant &r_ret, const Variant **p_args) {
  367. VariantInternal::clear(&r_ret);
  368. }
  369. static void ptr_construct(void *base, const void **p_args) {
  370. PtrToArg<Variant>::encode(Variant(), base);
  371. }
  372. static int get_argument_count() {
  373. return 1;
  374. }
  375. static Variant::Type get_argument_type(int p_arg) {
  376. return Variant::NIL;
  377. }
  378. static Variant::Type get_base_type() {
  379. return Variant::NIL;
  380. }
  381. };
  382. struct VariantConstructData {
  383. void (*construct)(Variant &r_base, const Variant **p_args, Callable::CallError &r_error);
  384. Variant::ValidatedConstructor validated_construct;
  385. Variant::PTRConstructor ptr_construct;
  386. Variant::Type (*get_argument_type)(int);
  387. int argument_count;
  388. Vector<String> arg_names;
  389. };
  390. static LocalVector<VariantConstructData> construct_data[Variant::VARIANT_MAX];
  391. static void variant_initialize_nil(Variant *v) {
  392. VariantInternal::clear(v);
  393. }
  394. template <class T>
  395. static void variant_initialize(Variant *v) {
  396. VariantTypeChanger<T>::change(v);
  397. }
  398. template <class T>
  399. static void variant_initialize_zero(Variant *v) {
  400. VariantTypeChanger<T>::change(v);
  401. *VariantGetInternalPtr<T>::get_ptr(v) = 0;
  402. }
  403. static void variant_initialize_false(Variant *v) {
  404. VariantTypeChanger<bool>::change(v);
  405. *VariantGetInternalPtr<bool>::get_ptr(v) = false;
  406. }
  407. static void variant_initialize_obj(Variant *v) {
  408. VariantInternal::clear(v);
  409. VariantInternal::object_assign_null(v);
  410. }
  411. typedef void (*VariantInitializeFunc)(Variant *v);
  412. static VariantInitializeFunc initialize_funcs[Variant::VARIANT_MAX];
  413. template <class T>
  414. static void add_constructor(const Vector<String> &arg_names) {
  415. ERR_FAIL_COND_MSG(arg_names.size() != T::get_argument_count(), "Argument names size mismatch for " + Variant::get_type_name(T::get_base_type()) + ".");
  416. VariantConstructData cd;
  417. cd.construct = T::construct;
  418. cd.validated_construct = T::validated_construct;
  419. cd.ptr_construct = T::ptr_construct;
  420. cd.get_argument_type = T::get_argument_type;
  421. cd.argument_count = T::get_argument_count();
  422. cd.arg_names = arg_names;
  423. construct_data[T::get_base_type()].push_back(cd);
  424. }
  425. void Variant::_register_variant_constructors() {
  426. add_constructor<VariantConstructorNil>(sarray("from"));
  427. add_constructor<VariantConstructor<bool, bool>>(sarray("from"));
  428. add_constructor<VariantConstructor<bool, int64_t>>(sarray("from"));
  429. add_constructor<VariantConstructor<bool, double>>(sarray("from"));
  430. add_constructor<VariantConstructor<int64_t, int64_t>>(sarray("from"));
  431. add_constructor<VariantConstructor<int64_t, double>>(sarray("from"));
  432. add_constructor<VariantConstructor<double, double>>(sarray("from"));
  433. add_constructor<VariantConstructor<double, int64_t>>(sarray("from"));
  434. add_constructor<VariantConstructor<String, String>>(sarray("from"));
  435. add_constructor<VariantConstructor<String, StringName>>(sarray("from"));
  436. add_constructor<VariantConstructor<String, NodePath>>(sarray("from"));
  437. add_constructor<VariantConstructor<Vector2, Vector2>>(sarray("from"));
  438. add_constructor<VariantConstructor<Vector2, Vector2i>>(sarray("from"));
  439. add_constructor<VariantConstructor<Vector2, double, double>>(sarray("x", "y"));
  440. add_constructor<VariantConstructor<Vector2i, Vector2i>>(sarray("from"));
  441. add_constructor<VariantConstructor<Vector2i, Vector2>>(sarray("from"));
  442. add_constructor<VariantConstructor<Vector2i, int64_t, int64_t>>(sarray("x", "y"));
  443. add_constructor<VariantConstructor<Rect2, Rect2>>(sarray("from"));
  444. add_constructor<VariantConstructor<Rect2, Rect2i>>(sarray("from"));
  445. add_constructor<VariantConstructor<Rect2, Vector2, Vector2>>(sarray("position", "size"));
  446. add_constructor<VariantConstructor<Rect2, double, double, double, double>>(sarray("x", "y", "width", "height"));
  447. add_constructor<VariantConstructor<Rect2i, Rect2i>>(sarray("from"));
  448. add_constructor<VariantConstructor<Rect2i, Rect2>>(sarray("from"));
  449. add_constructor<VariantConstructor<Rect2i, Vector2i, Vector2i>>(sarray("position", "size"));
  450. add_constructor<VariantConstructor<Rect2i, int64_t, int64_t, int64_t, int64_t>>(sarray("x", "y", "width", "height"));
  451. add_constructor<VariantConstructor<Vector3, Vector3>>(sarray("from"));
  452. add_constructor<VariantConstructor<Vector3, Vector3i>>(sarray("from"));
  453. add_constructor<VariantConstructor<Vector3, double, double, double>>(sarray("x", "y", "z"));
  454. add_constructor<VariantConstructor<Vector3i, Vector3i>>(sarray("from"));
  455. add_constructor<VariantConstructor<Vector3i, Vector3>>(sarray("from"));
  456. add_constructor<VariantConstructor<Vector3i, int64_t, int64_t, int64_t>>(sarray("x", "y", "z"));
  457. add_constructor<VariantConstructor<Transform2D, Transform2D>>(sarray("from"));
  458. add_constructor<VariantConstructor<Transform2D, Vector2, Vector2, Vector2>>(sarray("x", "y", "origin"));
  459. add_constructor<VariantConstructor<Plane, Plane>>(sarray("from"));
  460. add_constructor<VariantConstructor<Plane, Vector3, double>>(sarray("normal", "d"));
  461. add_constructor<VariantConstructor<Plane, Vector3, Vector3>>(sarray("point", "normal"));
  462. add_constructor<VariantConstructor<Plane, Vector3, Vector3, Vector3>>(sarray("point1", "point2", "point3"));
  463. add_constructor<VariantConstructor<Quat, Quat>>(sarray("from"));
  464. add_constructor<VariantConstructor<Quat, Basis>>(sarray("from"));
  465. add_constructor<VariantConstructor<Quat, Vector3>>(sarray("euler"));
  466. add_constructor<VariantConstructor<Quat, Vector3, double>>(sarray("axis", "angle"));
  467. add_constructor<VariantConstructor<Quat, Vector3, Vector3>>(sarray("arc_from", "arc_to"));
  468. add_constructor<VariantConstructor<::AABB, ::AABB>>(sarray("from"));
  469. add_constructor<VariantConstructor<::AABB, Vector3, Vector3>>(sarray("position", "size"));
  470. add_constructor<VariantConstructor<Basis, Basis>>(sarray("from"));
  471. add_constructor<VariantConstructor<Basis, Quat>>(sarray("from"));
  472. add_constructor<VariantConstructor<Basis, Vector3, Vector3, Vector3>>(sarray("x", "y", "z"));
  473. add_constructor<VariantConstructor<Transform, Transform>>(sarray("from"));
  474. add_constructor<VariantConstructor<Transform, Basis, Vector3>>(sarray("basis", "origin"));
  475. add_constructor<VariantConstructor<Color, Color>>(sarray("from"));
  476. add_constructor<VariantConstructor<StringName, StringName>>(sarray("from"));
  477. add_constructor<VariantConstructor<StringName, String>>(sarray("from"));
  478. add_constructor<VariantConstructor<NodePath, NodePath>>(sarray("from"));
  479. add_constructor<VariantConstructor<NodePath, String>>(sarray("from"));
  480. add_constructor<VariantConstructor<::RID, ::RID>>(sarray("from"));
  481. add_constructor<VariantConstructorObject>(sarray("from"));
  482. add_constructor<VariantConstructorNilObject>(sarray("from"));
  483. add_constructor<VariantConstructor<Callable, Callable>>(sarray("from"));
  484. add_constructor<VariantConstructorCallableArgs>(sarray("object", "method"));
  485. add_constructor<VariantConstructor<Signal, Signal>>(sarray("from"));
  486. add_constructor<VariantConstructorSignalArgs>(sarray("object", "signal"));
  487. add_constructor<VariantConstructor<Dictionary, Dictionary>>(sarray("from"));
  488. add_constructor<VariantConstructor<Array, Array>>(sarray("from"));
  489. add_constructor<VariantConstructorToArray<PackedByteArray>>(sarray("from"));
  490. add_constructor<VariantConstructorToArray<PackedInt32Array>>(sarray("from"));
  491. add_constructor<VariantConstructorToArray<PackedInt64Array>>(sarray("from"));
  492. add_constructor<VariantConstructorToArray<PackedFloat32Array>>(sarray("from"));
  493. add_constructor<VariantConstructorToArray<PackedFloat64Array>>(sarray("from"));
  494. add_constructor<VariantConstructorToArray<PackedStringArray>>(sarray("from"));
  495. add_constructor<VariantConstructorToArray<PackedVector2Array>>(sarray("from"));
  496. add_constructor<VariantConstructorToArray<PackedVector3Array>>(sarray("from"));
  497. add_constructor<VariantConstructorToArray<PackedColorArray>>(sarray("from"));
  498. add_constructor<VariantConstructor<PackedByteArray, PackedByteArray>>(sarray("from"));
  499. add_constructor<VariantConstructorFromArray<PackedByteArray>>(sarray("from"));
  500. add_constructor<VariantConstructor<PackedInt32Array, PackedInt32Array>>(sarray("from"));
  501. add_constructor<VariantConstructorFromArray<PackedInt32Array>>(sarray("from"));
  502. add_constructor<VariantConstructor<PackedInt64Array, PackedInt64Array>>(sarray("from"));
  503. add_constructor<VariantConstructorFromArray<PackedInt64Array>>(sarray("from"));
  504. add_constructor<VariantConstructor<PackedFloat32Array, PackedFloat32Array>>(sarray("from"));
  505. add_constructor<VariantConstructorFromArray<PackedFloat32Array>>(sarray("from"));
  506. add_constructor<VariantConstructor<PackedFloat64Array, PackedFloat64Array>>(sarray("from"));
  507. add_constructor<VariantConstructorFromArray<PackedFloat64Array>>(sarray("from"));
  508. add_constructor<VariantConstructor<PackedStringArray, PackedStringArray>>(sarray("from"));
  509. add_constructor<VariantConstructorFromArray<PackedStringArray>>(sarray("from"));
  510. add_constructor<VariantConstructor<PackedVector2Array, PackedVector2Array>>(sarray("from"));
  511. add_constructor<VariantConstructorFromArray<PackedVector2Array>>(sarray("from"));
  512. add_constructor<VariantConstructor<PackedVector3Array, PackedVector3Array>>(sarray("from"));
  513. add_constructor<VariantConstructorFromArray<PackedVector3Array>>(sarray("from"));
  514. add_constructor<VariantConstructor<PackedColorArray, PackedColorArray>>(sarray("from"));
  515. add_constructor<VariantConstructorFromArray<PackedColorArray>>(sarray("from"));
  516. initialize_funcs[NIL] = variant_initialize_nil;
  517. // atomic types
  518. initialize_funcs[BOOL] = variant_initialize_false;
  519. initialize_funcs[INT] = variant_initialize_zero<int64_t>;
  520. initialize_funcs[FLOAT] = variant_initialize_zero<double>;
  521. initialize_funcs[STRING] = variant_initialize<String>;
  522. // math types
  523. initialize_funcs[VECTOR2] = variant_initialize<Vector2>;
  524. initialize_funcs[VECTOR2I] = variant_initialize<Vector2i>;
  525. initialize_funcs[RECT2] = variant_initialize<Rect2>;
  526. initialize_funcs[RECT2I] = variant_initialize<Rect2i>;
  527. initialize_funcs[VECTOR3] = variant_initialize<Vector3>;
  528. initialize_funcs[VECTOR3I] = variant_initialize<Vector3i>;
  529. initialize_funcs[TRANSFORM2D] = variant_initialize<Transform2D>;
  530. initialize_funcs[PLANE] = variant_initialize<Plane>;
  531. initialize_funcs[QUAT] = variant_initialize<Quat>;
  532. initialize_funcs[AABB] = variant_initialize<::AABB>;
  533. initialize_funcs[BASIS] = variant_initialize<Basis>;
  534. initialize_funcs[TRANSFORM] = variant_initialize<Transform>;
  535. // misc types
  536. initialize_funcs[COLOR] = variant_initialize<Color>;
  537. initialize_funcs[STRING_NAME] = variant_initialize<StringName>;
  538. initialize_funcs[NODE_PATH] = variant_initialize<NodePath>;
  539. initialize_funcs[RID] = variant_initialize<::RID>;
  540. initialize_funcs[OBJECT] = variant_initialize_obj;
  541. initialize_funcs[CALLABLE] = variant_initialize<Callable>;
  542. initialize_funcs[SIGNAL] = variant_initialize<Signal>;
  543. initialize_funcs[DICTIONARY] = variant_initialize<Dictionary>;
  544. initialize_funcs[ARRAY] = variant_initialize<Array>;
  545. // typed arrays
  546. initialize_funcs[PACKED_BYTE_ARRAY] = variant_initialize<PackedByteArray>;
  547. initialize_funcs[PACKED_INT32_ARRAY] = variant_initialize<PackedInt32Array>;
  548. initialize_funcs[PACKED_INT64_ARRAY] = variant_initialize<PackedInt64Array>;
  549. initialize_funcs[PACKED_FLOAT32_ARRAY] = variant_initialize<PackedFloat32Array>;
  550. initialize_funcs[PACKED_FLOAT64_ARRAY] = variant_initialize<PackedFloat64Array>;
  551. initialize_funcs[PACKED_STRING_ARRAY] = variant_initialize<PackedStringArray>;
  552. initialize_funcs[PACKED_VECTOR2_ARRAY] = variant_initialize<PackedVector2Array>;
  553. initialize_funcs[PACKED_VECTOR3_ARRAY] = variant_initialize<PackedVector3Array>;
  554. initialize_funcs[PACKED_COLOR_ARRAY] = variant_initialize<PackedColorArray>;
  555. }
  556. void Variant::_unregister_variant_constructors() {
  557. for (int i = 0; i < Variant::VARIANT_MAX; i++) {
  558. construct_data[i].clear();
  559. }
  560. }
  561. void Variant::construct(Variant::Type p_type, Variant &base, const Variant **p_args, int p_argcount, Callable::CallError &r_error) {
  562. if (p_argcount == 0) {
  563. initialize_funcs[p_type](&base);
  564. r_error.error = Callable::CallError::CALL_OK;
  565. } else {
  566. uint32_t s = construct_data[p_type].size();
  567. for (uint32_t i = 0; i < s; i++) {
  568. int argc = construct_data[p_type][i].argument_count;
  569. if (argc != p_argcount) {
  570. continue;
  571. }
  572. bool args_match = true;
  573. for (int j = 0; j < argc; j++) {
  574. if (!Variant::can_convert_strict(p_args[j]->get_type(), construct_data[p_type][i].get_argument_type(j))) {
  575. args_match = false;
  576. break;
  577. }
  578. }
  579. if (!args_match) {
  580. continue;
  581. }
  582. construct_data[p_type][i].construct(base, p_args, r_error);
  583. return;
  584. }
  585. r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD;
  586. }
  587. }
  588. int Variant::get_constructor_count(Variant::Type p_type) {
  589. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, -1);
  590. return construct_data[p_type].size();
  591. }
  592. Variant::ValidatedConstructor Variant::get_validated_constructor(Variant::Type p_type, int p_constructor) {
  593. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, nullptr);
  594. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), nullptr);
  595. return construct_data[p_type][p_constructor].validated_construct;
  596. }
  597. Variant::PTRConstructor Variant::get_ptr_constructor(Variant::Type p_type, int p_constructor) {
  598. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, nullptr);
  599. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), nullptr);
  600. return construct_data[p_type][p_constructor].ptr_construct;
  601. }
  602. int Variant::get_constructor_argument_count(Variant::Type p_type, int p_constructor) {
  603. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, -1);
  604. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), -1);
  605. return construct_data[p_type][p_constructor].argument_count;
  606. }
  607. Variant::Type Variant::get_constructor_argument_type(Variant::Type p_type, int p_constructor, int p_argument) {
  608. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, Variant::VARIANT_MAX);
  609. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), Variant::VARIANT_MAX);
  610. return construct_data[p_type][p_constructor].get_argument_type(p_argument);
  611. }
  612. String Variant::get_constructor_argument_name(Variant::Type p_type, int p_constructor, int p_argument) {
  613. ERR_FAIL_INDEX_V(p_type, Variant::VARIANT_MAX, String());
  614. ERR_FAIL_INDEX_V(p_constructor, (int)construct_data[p_type].size(), String());
  615. return construct_data[p_type][p_constructor].arg_names[p_argument];
  616. }
  617. void VariantInternal::object_assign(Variant *v, const Variant *o) {
  618. if (o->_get_obj().obj && o->_get_obj().id.is_reference()) {
  619. Reference *reference = static_cast<Reference *>(o->_get_obj().obj);
  620. if (!reference->reference()) {
  621. v->_get_obj().obj = nullptr;
  622. v->_get_obj().id = ObjectID();
  623. return;
  624. }
  625. }
  626. v->_get_obj().obj = const_cast<Object *>(o->_get_obj().obj);
  627. v->_get_obj().id = o->_get_obj().id;
  628. }
  629. void Variant::get_constructor_list(Type p_type, List<MethodInfo> *r_list) {
  630. ERR_FAIL_INDEX(p_type, Variant::VARIANT_MAX);
  631. MethodInfo mi;
  632. mi.return_val.type = p_type;
  633. mi.name = get_type_name(p_type);
  634. for (int i = 0; i < get_constructor_count(p_type); i++) {
  635. int ac = get_constructor_argument_count(p_type, i);
  636. mi.arguments.clear();
  637. for (int j = 0; j < ac; j++) {
  638. PropertyInfo arg;
  639. arg.name = get_constructor_argument_name(p_type, i, j);
  640. arg.type = get_constructor_argument_type(p_type, i, j);
  641. mi.arguments.push_back(arg);
  642. }
  643. r_list->push_back(mi);
  644. }
  645. }