gdextension.cpp 49 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070
  1. /**************************************************************************/
  2. /* gdextension.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 "gdextension.h"
  31. #include "gdextension.compat.inc"
  32. #include "core/config/project_settings.h"
  33. #include "core/object/class_db.h"
  34. #include "core/object/method_bind.h"
  35. #include "gdextension_library_loader.h"
  36. #include "gdextension_manager.h"
  37. extern void gdextension_setup_interface();
  38. extern GDExtensionInterfaceFunctionPtr gdextension_get_proc_address(const char *p_name);
  39. typedef GDExtensionBool (*GDExtensionLegacyInitializationFunction)(void *p_interface, GDExtensionClassLibraryPtr p_library, GDExtensionInitialization *r_initialization);
  40. String GDExtension::get_extension_list_config_file() {
  41. return ProjectSettings::get_singleton()->get_project_data_path().path_join("extension_list.cfg");
  42. }
  43. class GDExtensionMethodBind : public MethodBind {
  44. GDExtensionClassMethodCall call_func;
  45. GDExtensionClassMethodValidatedCall validated_call_func;
  46. GDExtensionClassMethodPtrCall ptrcall_func;
  47. void *method_userdata;
  48. bool vararg;
  49. uint32_t argument_count;
  50. PropertyInfo return_value_info;
  51. GodotTypeInfo::Metadata return_value_metadata;
  52. List<PropertyInfo> arguments_info;
  53. List<GodotTypeInfo::Metadata> arguments_metadata;
  54. #ifdef TOOLS_ENABLED
  55. friend class GDExtension;
  56. StringName name;
  57. bool is_reloading = false;
  58. bool valid = true;
  59. #endif
  60. protected:
  61. virtual Variant::Type _gen_argument_type(int p_arg) const override {
  62. if (p_arg < 0) {
  63. return return_value_info.type;
  64. } else {
  65. return arguments_info.get(p_arg).type;
  66. }
  67. }
  68. virtual PropertyInfo _gen_argument_type_info(int p_arg) const override {
  69. if (p_arg < 0) {
  70. return return_value_info;
  71. } else {
  72. return arguments_info.get(p_arg);
  73. }
  74. }
  75. public:
  76. #ifdef TOOLS_ENABLED
  77. virtual bool is_valid() const override { return valid; }
  78. #endif
  79. #ifdef DEBUG_METHODS_ENABLED
  80. virtual GodotTypeInfo::Metadata get_argument_meta(int p_arg) const override {
  81. if (p_arg < 0) {
  82. return return_value_metadata;
  83. } else {
  84. return arguments_metadata.get(p_arg);
  85. }
  86. }
  87. #endif
  88. virtual Variant call(Object *p_object, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) const override {
  89. #ifdef TOOLS_ENABLED
  90. ERR_FAIL_COND_V_MSG(!valid, Variant(), vformat("Cannot call invalid GDExtension method bind '%s'. It's probably cached - you may need to restart Godot.", name));
  91. ERR_FAIL_COND_V_MSG(p_object && p_object->is_extension_placeholder(), Variant(), vformat("Cannot call GDExtension method bind '%s' on placeholder instance.", name));
  92. #endif
  93. Variant ret;
  94. GDExtensionClassInstancePtr extension_instance = is_static() ? nullptr : p_object->_get_extension_instance();
  95. GDExtensionCallError ce{ GDEXTENSION_CALL_OK, 0, 0 };
  96. call_func(method_userdata, extension_instance, reinterpret_cast<GDExtensionConstVariantPtr *>(p_args), p_arg_count, (GDExtensionVariantPtr)&ret, &ce);
  97. r_error.error = Callable::CallError::Error(ce.error);
  98. r_error.argument = ce.argument;
  99. r_error.expected = ce.expected;
  100. return ret;
  101. }
  102. virtual void validated_call(Object *p_object, const Variant **p_args, Variant *r_ret) const override {
  103. #ifdef TOOLS_ENABLED
  104. ERR_FAIL_COND_MSG(!valid, vformat("Cannot call invalid GDExtension method bind '%s'. It's probably cached - you may need to restart Godot.", name));
  105. ERR_FAIL_COND_MSG(p_object && p_object->is_extension_placeholder(), vformat("Cannot call GDExtension method bind '%s' on placeholder instance.", name));
  106. #endif
  107. ERR_FAIL_COND_MSG(vararg, "Vararg methods don't have validated call support. This is most likely an engine bug.");
  108. GDExtensionClassInstancePtr extension_instance = is_static() ? nullptr : p_object->_get_extension_instance();
  109. if (validated_call_func) {
  110. // This is added here, but it's unlikely to be provided by most extensions.
  111. validated_call_func(method_userdata, extension_instance, reinterpret_cast<GDExtensionConstVariantPtr *>(p_args), (GDExtensionVariantPtr)r_ret);
  112. } else {
  113. // If not provided, go via ptrcall, which is faster than resorting to regular call.
  114. const void **argptrs = (const void **)alloca(argument_count * sizeof(void *));
  115. for (uint32_t i = 0; i < argument_count; i++) {
  116. argptrs[i] = VariantInternal::get_opaque_pointer(p_args[i]);
  117. }
  118. void *ret_opaque = nullptr;
  119. if (r_ret) {
  120. VariantInternal::initialize(r_ret, return_value_info.type);
  121. ret_opaque = r_ret->get_type() == Variant::NIL ? r_ret : VariantInternal::get_opaque_pointer(r_ret);
  122. }
  123. ptrcall_func(method_userdata, extension_instance, reinterpret_cast<GDExtensionConstTypePtr *>(argptrs), (GDExtensionTypePtr)ret_opaque);
  124. if (r_ret && r_ret->get_type() == Variant::OBJECT) {
  125. VariantInternal::update_object_id(r_ret);
  126. }
  127. }
  128. }
  129. virtual void ptrcall(Object *p_object, const void **p_args, void *r_ret) const override {
  130. #ifdef TOOLS_ENABLED
  131. ERR_FAIL_COND_MSG(!valid, vformat("Cannot call invalid GDExtension method bind '%s'. It's probably cached - you may need to restart Godot.", name));
  132. ERR_FAIL_COND_MSG(p_object && p_object->is_extension_placeholder(), vformat("Cannot call GDExtension method bind '%s' on placeholder instance.", name));
  133. #endif
  134. ERR_FAIL_COND_MSG(vararg, "Vararg methods don't have ptrcall support. This is most likely an engine bug.");
  135. GDExtensionClassInstancePtr extension_instance = is_static() ? nullptr : p_object->_get_extension_instance();
  136. ptrcall_func(method_userdata, extension_instance, reinterpret_cast<GDExtensionConstTypePtr *>(p_args), (GDExtensionTypePtr)r_ret);
  137. }
  138. virtual bool is_vararg() const override {
  139. return vararg;
  140. }
  141. #ifdef TOOLS_ENABLED
  142. bool try_update(const GDExtensionClassMethodInfo *p_method_info) {
  143. if (is_static() != (bool)(p_method_info->method_flags & GDEXTENSION_METHOD_FLAG_STATIC)) {
  144. return false;
  145. }
  146. if (vararg != (bool)(p_method_info->method_flags & GDEXTENSION_METHOD_FLAG_VARARG)) {
  147. return false;
  148. }
  149. if (has_return() != (bool)p_method_info->has_return_value) {
  150. return false;
  151. }
  152. if (has_return() && return_value_info.type != (Variant::Type)p_method_info->return_value_info->type) {
  153. return false;
  154. }
  155. if (argument_count != p_method_info->argument_count) {
  156. return false;
  157. }
  158. List<PropertyInfo>::ConstIterator itr = arguments_info.begin();
  159. for (uint32_t i = 0; i < p_method_info->argument_count; ++itr, ++i) {
  160. if (itr->type != (Variant::Type)p_method_info->arguments_info[i].type) {
  161. return false;
  162. }
  163. }
  164. update(p_method_info);
  165. return true;
  166. }
  167. #endif
  168. void update(const GDExtensionClassMethodInfo *p_method_info) {
  169. #ifdef TOOLS_ENABLED
  170. name = *reinterpret_cast<StringName *>(p_method_info->name);
  171. #endif
  172. method_userdata = p_method_info->method_userdata;
  173. call_func = p_method_info->call_func;
  174. validated_call_func = nullptr;
  175. ptrcall_func = p_method_info->ptrcall_func;
  176. set_name(*reinterpret_cast<StringName *>(p_method_info->name));
  177. if (p_method_info->has_return_value) {
  178. return_value_info = PropertyInfo(*p_method_info->return_value_info);
  179. return_value_metadata = GodotTypeInfo::Metadata(p_method_info->return_value_metadata);
  180. }
  181. arguments_info.clear();
  182. arguments_metadata.clear();
  183. for (uint32_t i = 0; i < p_method_info->argument_count; i++) {
  184. arguments_info.push_back(PropertyInfo(p_method_info->arguments_info[i]));
  185. arguments_metadata.push_back(GodotTypeInfo::Metadata(p_method_info->arguments_metadata[i]));
  186. }
  187. set_hint_flags(p_method_info->method_flags);
  188. argument_count = p_method_info->argument_count;
  189. vararg = p_method_info->method_flags & GDEXTENSION_METHOD_FLAG_VARARG;
  190. _set_returns(p_method_info->has_return_value);
  191. _set_const(p_method_info->method_flags & GDEXTENSION_METHOD_FLAG_CONST);
  192. _set_static(p_method_info->method_flags & GDEXTENSION_METHOD_FLAG_STATIC);
  193. #ifdef DEBUG_METHODS_ENABLED
  194. _generate_argument_types(p_method_info->argument_count);
  195. #endif
  196. set_argument_count(p_method_info->argument_count);
  197. Vector<Variant> defargs;
  198. defargs.resize(p_method_info->default_argument_count);
  199. for (uint32_t i = 0; i < p_method_info->default_argument_count; i++) {
  200. defargs.write[i] = *static_cast<Variant *>(p_method_info->default_arguments[i]);
  201. }
  202. set_default_arguments(defargs);
  203. }
  204. explicit GDExtensionMethodBind(const GDExtensionClassMethodInfo *p_method_info) {
  205. update(p_method_info);
  206. }
  207. };
  208. #ifndef DISABLE_DEPRECATED
  209. void GDExtension::_register_extension_class(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringNamePtr p_parent_class_name, const GDExtensionClassCreationInfo *p_extension_funcs) {
  210. const GDExtensionClassCreationInfo4 class_info4 = {
  211. p_extension_funcs->is_virtual, // GDExtensionBool is_virtual;
  212. p_extension_funcs->is_abstract, // GDExtensionBool is_abstract;
  213. true, // GDExtensionBool is_exposed;
  214. false, // GDExtensionBool is_runtime;
  215. nullptr, // GDExtensionConstStringPtr icon_path;
  216. p_extension_funcs->set_func, // GDExtensionClassSet set_func;
  217. p_extension_funcs->get_func, // GDExtensionClassGet get_func;
  218. p_extension_funcs->get_property_list_func, // GDExtensionClassGetPropertyList get_property_list_func;
  219. nullptr, // GDExtensionClassFreePropertyList2 free_property_list_func;
  220. p_extension_funcs->property_can_revert_func, // GDExtensionClassPropertyCanRevert property_can_revert_func;
  221. p_extension_funcs->property_get_revert_func, // GDExtensionClassPropertyGetRevert property_get_revert_func;
  222. nullptr, // GDExtensionClassValidateProperty validate_property_func;
  223. nullptr, // GDExtensionClassNotification2 notification_func;
  224. p_extension_funcs->to_string_func, // GDExtensionClassToString to_string_func;
  225. p_extension_funcs->reference_func, // GDExtensionClassReference reference_func;
  226. p_extension_funcs->unreference_func, // GDExtensionClassUnreference unreference_func;
  227. nullptr, // GDExtensionClassCreateInstance2 create_instance_func; /* this one is mandatory */
  228. p_extension_funcs->free_instance_func, // GDExtensionClassFreeInstance free_instance_func; /* this one is mandatory */
  229. nullptr, // GDExtensionClassRecreateInstance recreate_instance_func;
  230. nullptr, // GDExtensionClassGetVirtual get_virtual_func;
  231. nullptr, // GDExtensionClassGetVirtualCallData get_virtual_call_data_func;
  232. nullptr, // GDExtensionClassCallVirtualWithData call_virtual_func;
  233. p_extension_funcs->class_userdata, // void *class_userdata;
  234. };
  235. const ClassCreationDeprecatedInfo legacy = {
  236. p_extension_funcs->notification_func, // GDExtensionClassNotification notification_func;
  237. p_extension_funcs->free_property_list_func, // GDExtensionClassFreePropertyList free_property_list_func;
  238. p_extension_funcs->create_instance_func, // GDExtensionClassCreateInstance create_instance_func;
  239. p_extension_funcs->get_rid_func, // GDExtensionClassGetRID get_rid;
  240. p_extension_funcs->get_virtual_func, // GDExtensionClassGetVirtual get_virtual_func;
  241. nullptr,
  242. };
  243. _register_extension_class_internal(p_library, p_class_name, p_parent_class_name, &class_info4, &legacy);
  244. }
  245. void GDExtension::_register_extension_class2(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringNamePtr p_parent_class_name, const GDExtensionClassCreationInfo2 *p_extension_funcs) {
  246. const GDExtensionClassCreationInfo4 class_info4 = {
  247. p_extension_funcs->is_virtual, // GDExtensionBool is_virtual;
  248. p_extension_funcs->is_abstract, // GDExtensionBool is_abstract;
  249. p_extension_funcs->is_exposed, // GDExtensionBool is_exposed;
  250. false, // GDExtensionBool is_runtime;
  251. nullptr, // GDExtensionConstStringPtr icon_path;
  252. p_extension_funcs->set_func, // GDExtensionClassSet set_func;
  253. p_extension_funcs->get_func, // GDExtensionClassGet get_func;
  254. p_extension_funcs->get_property_list_func, // GDExtensionClassGetPropertyList get_property_list_func;
  255. nullptr, // GDExtensionClassFreePropertyList2 free_property_list_func;
  256. p_extension_funcs->property_can_revert_func, // GDExtensionClassPropertyCanRevert property_can_revert_func;
  257. p_extension_funcs->property_get_revert_func, // GDExtensionClassPropertyGetRevert property_get_revert_func;
  258. p_extension_funcs->validate_property_func, // GDExtensionClassValidateProperty validate_property_func;
  259. p_extension_funcs->notification_func, // GDExtensionClassNotification2 notification_func;
  260. p_extension_funcs->to_string_func, // GDExtensionClassToString to_string_func;
  261. p_extension_funcs->reference_func, // GDExtensionClassReference reference_func;
  262. p_extension_funcs->unreference_func, // GDExtensionClassUnreference unreference_func;
  263. nullptr, // GDExtensionClassCreateInstance2 create_instance_func; /* this one is mandatory */
  264. p_extension_funcs->free_instance_func, // GDExtensionClassFreeInstance free_instance_func; /* this one is mandatory */
  265. p_extension_funcs->recreate_instance_func, // GDExtensionClassRecreateInstance recreate_instance_func;
  266. nullptr, // GDExtensionClassGetVirtual get_virtual_func;
  267. nullptr, // GDExtensionClassGetVirtualCallData get_virtual_call_data_func;
  268. p_extension_funcs->call_virtual_with_data_func, // GDExtensionClassCallVirtualWithData call_virtual_func;
  269. p_extension_funcs->class_userdata, // void *class_userdata;
  270. };
  271. const ClassCreationDeprecatedInfo legacy = {
  272. nullptr, // GDExtensionClassNotification notification_func;
  273. p_extension_funcs->free_property_list_func, // GDExtensionClassFreePropertyList free_property_list_func;
  274. p_extension_funcs->create_instance_func, // GDExtensionClassCreateInstance create_instance_func;
  275. p_extension_funcs->get_rid_func, // GDExtensionClassGetRID get_rid;
  276. p_extension_funcs->get_virtual_func, // GDExtensionClassGetVirtual get_virtual_func;
  277. p_extension_funcs->get_virtual_call_data_func, // GDExtensionClassGetVirtual get_virtual_func;
  278. };
  279. _register_extension_class_internal(p_library, p_class_name, p_parent_class_name, &class_info4, &legacy);
  280. }
  281. void GDExtension::_register_extension_class3(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringNamePtr p_parent_class_name, const GDExtensionClassCreationInfo3 *p_extension_funcs) {
  282. const GDExtensionClassCreationInfo4 class_info4 = {
  283. p_extension_funcs->is_virtual, // GDExtensionBool is_virtual;
  284. p_extension_funcs->is_abstract, // GDExtensionBool is_abstract;
  285. p_extension_funcs->is_exposed, // GDExtensionBool is_exposed;
  286. p_extension_funcs->is_runtime, // GDExtensionBool is_runtime;
  287. nullptr, // GDExtensionConstStringPtr icon_path;
  288. p_extension_funcs->set_func, // GDExtensionClassSet set_func;
  289. p_extension_funcs->get_func, // GDExtensionClassGet get_func;
  290. p_extension_funcs->get_property_list_func, // GDExtensionClassGetPropertyList get_property_list_func;
  291. p_extension_funcs->free_property_list_func, // GDExtensionClassFreePropertyList free_property_list_func;
  292. p_extension_funcs->property_can_revert_func, // GDExtensionClassPropertyCanRevert property_can_revert_func;
  293. p_extension_funcs->property_get_revert_func, // GDExtensionClassPropertyGetRevert property_get_revert_func;
  294. p_extension_funcs->validate_property_func, // GDExtensionClassValidateProperty validate_property_func;
  295. p_extension_funcs->notification_func, // GDExtensionClassNotification2 notification_func;
  296. p_extension_funcs->to_string_func, // GDExtensionClassToString to_string_func;
  297. p_extension_funcs->reference_func, // GDExtensionClassReference reference_func;
  298. p_extension_funcs->unreference_func, // GDExtensionClassUnreference unreference_func;
  299. nullptr, // GDExtensionClassCreateInstance2 create_instance_func; /* this one is mandatory */
  300. p_extension_funcs->free_instance_func, // GDExtensionClassFreeInstance free_instance_func; /* this one is mandatory */
  301. p_extension_funcs->recreate_instance_func, // GDExtensionClassRecreateInstance recreate_instance_func;
  302. nullptr, // GDExtensionClassGetVirtual get_virtual_func;
  303. nullptr, // GDExtensionClassGetVirtualCallData get_virtual_call_data_func;
  304. p_extension_funcs->call_virtual_with_data_func, // GDExtensionClassCallVirtualWithData call_virtual_func;
  305. p_extension_funcs->class_userdata, // void *class_userdata;
  306. };
  307. const ClassCreationDeprecatedInfo legacy = {
  308. nullptr, // GDExtensionClassNotification notification_func;
  309. nullptr, // GDExtensionClassFreePropertyList free_property_list_func;
  310. p_extension_funcs->create_instance_func, // GDExtensionClassCreateInstance2 create_instance_func;
  311. p_extension_funcs->get_rid_func, // GDExtensionClassGetRID get_rid;
  312. p_extension_funcs->get_virtual_func, // GDExtensionClassGetVirtual get_virtual_func;
  313. p_extension_funcs->get_virtual_call_data_func, // GDExtensionClassGetVirtual get_virtual_func;
  314. };
  315. _register_extension_class_internal(p_library, p_class_name, p_parent_class_name, &class_info4, &legacy);
  316. }
  317. #endif // DISABLE_DEPRECATED
  318. void GDExtension::_register_extension_class4(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringNamePtr p_parent_class_name, const GDExtensionClassCreationInfo4 *p_extension_funcs) {
  319. _register_extension_class_internal(p_library, p_class_name, p_parent_class_name, p_extension_funcs);
  320. }
  321. void GDExtension::_register_extension_class_internal(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringNamePtr p_parent_class_name, const GDExtensionClassCreationInfo4 *p_extension_funcs, const ClassCreationDeprecatedInfo *p_deprecated_funcs) {
  322. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  323. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  324. StringName parent_class_name = *reinterpret_cast<const StringName *>(p_parent_class_name);
  325. ERR_FAIL_COND_MSG(!String(class_name).is_valid_unicode_identifier(), vformat("Attempt to register extension class '%s', which is not a valid class identifier.", class_name));
  326. ERR_FAIL_COND_MSG(ClassDB::class_exists(class_name), vformat("Attempt to register extension class '%s', which appears to be already registered.", class_name));
  327. Extension *parent_extension = nullptr;
  328. if (self->extension_classes.has(parent_class_name)) {
  329. parent_extension = &self->extension_classes[parent_class_name];
  330. } else if (ClassDB::class_exists(parent_class_name)) {
  331. if (ClassDB::get_api_type(parent_class_name) == ClassDB::API_EXTENSION || ClassDB::get_api_type(parent_class_name) == ClassDB::API_EDITOR_EXTENSION) {
  332. ERR_PRINT("Unimplemented yet");
  333. //inheriting from another extension
  334. } else {
  335. //inheriting from engine class
  336. }
  337. } else {
  338. ERR_FAIL_MSG(vformat("Attempt to register an extension class '%s' using non-existing parent class '%s'.", String(class_name), String(parent_class_name)));
  339. }
  340. #ifdef TOOLS_ENABLED
  341. Extension *extension = nullptr;
  342. bool is_runtime = (bool)p_extension_funcs->is_runtime;
  343. if (self->is_reloading && self->extension_classes.has(class_name)) {
  344. extension = &self->extension_classes[class_name];
  345. if (!parent_extension && parent_class_name != extension->gdextension.parent_class_name) {
  346. ERR_FAIL_MSG(vformat("GDExtension class '%s' cannot change parent type from '%s' to '%s' on hot reload. Restart Godot for this change to take effect.", class_name, extension->gdextension.parent_class_name, parent_class_name));
  347. }
  348. if (extension->gdextension.is_runtime != is_runtime) {
  349. ERR_PRINT(vformat("GDExtension class '%s' cannot change to/from runtime class on hot reload. Restart Godot for this change to take effect.", class_name));
  350. is_runtime = extension->gdextension.is_runtime;
  351. }
  352. extension->is_reloading = false;
  353. } else {
  354. self->extension_classes[class_name] = Extension();
  355. extension = &self->extension_classes[class_name];
  356. }
  357. #else
  358. self->extension_classes[class_name] = Extension();
  359. Extension *extension = &self->extension_classes[class_name];
  360. #endif
  361. if (parent_extension) {
  362. extension->gdextension.parent = &parent_extension->gdextension;
  363. parent_extension->gdextension.children.push_back(&extension->gdextension);
  364. }
  365. if (self->reloadable && p_extension_funcs->recreate_instance_func == nullptr) {
  366. ERR_PRINT(vformat("Extension marked as reloadable, but attempted to register class '%s' which doesn't support reloading. Perhaps your language binding don't support it? Reloading disabled for this extension.", class_name));
  367. self->reloadable = false;
  368. }
  369. extension->gdextension.library = self;
  370. extension->gdextension.parent_class_name = parent_class_name;
  371. extension->gdextension.class_name = class_name;
  372. extension->gdextension.editor_class = self->level_initialized == INITIALIZATION_LEVEL_EDITOR;
  373. extension->gdextension.is_virtual = p_extension_funcs->is_virtual;
  374. extension->gdextension.is_abstract = p_extension_funcs->is_abstract;
  375. extension->gdextension.is_exposed = p_extension_funcs->is_exposed;
  376. #ifdef TOOLS_ENABLED
  377. extension->gdextension.is_runtime = is_runtime;
  378. #endif
  379. extension->gdextension.set = p_extension_funcs->set_func;
  380. extension->gdextension.get = p_extension_funcs->get_func;
  381. extension->gdextension.get_property_list = p_extension_funcs->get_property_list_func;
  382. extension->gdextension.free_property_list2 = p_extension_funcs->free_property_list_func;
  383. extension->gdextension.property_can_revert = p_extension_funcs->property_can_revert_func;
  384. extension->gdextension.property_get_revert = p_extension_funcs->property_get_revert_func;
  385. extension->gdextension.validate_property = p_extension_funcs->validate_property_func;
  386. #ifndef DISABLE_DEPRECATED
  387. if (p_deprecated_funcs) {
  388. extension->gdextension.notification = p_deprecated_funcs->notification_func;
  389. extension->gdextension.free_property_list = p_deprecated_funcs->free_property_list_func;
  390. extension->gdextension.create_instance = p_deprecated_funcs->create_instance_func;
  391. extension->gdextension.get_rid = p_deprecated_funcs->get_rid_func;
  392. extension->gdextension.get_virtual = p_deprecated_funcs->get_virtual_func;
  393. extension->gdextension.get_virtual_call_data = p_deprecated_funcs->get_virtual_call_data_func;
  394. }
  395. #endif // DISABLE_DEPRECATED
  396. extension->gdextension.notification2 = p_extension_funcs->notification_func;
  397. extension->gdextension.to_string = p_extension_funcs->to_string_func;
  398. extension->gdextension.reference = p_extension_funcs->reference_func;
  399. extension->gdextension.unreference = p_extension_funcs->unreference_func;
  400. extension->gdextension.class_userdata = p_extension_funcs->class_userdata;
  401. extension->gdextension.create_instance2 = p_extension_funcs->create_instance_func;
  402. extension->gdextension.free_instance = p_extension_funcs->free_instance_func;
  403. extension->gdextension.recreate_instance = p_extension_funcs->recreate_instance_func;
  404. extension->gdextension.get_virtual2 = p_extension_funcs->get_virtual_func;
  405. extension->gdextension.get_virtual_call_data2 = p_extension_funcs->get_virtual_call_data_func;
  406. extension->gdextension.call_virtual_with_data = p_extension_funcs->call_virtual_with_data_func;
  407. extension->gdextension.reloadable = self->reloadable;
  408. #ifdef TOOLS_ENABLED
  409. if (extension->gdextension.reloadable) {
  410. extension->gdextension.tracking_userdata = extension;
  411. extension->gdextension.track_instance = &GDExtension::_track_instance;
  412. extension->gdextension.untrack_instance = &GDExtension::_untrack_instance;
  413. } else {
  414. extension->gdextension.tracking_userdata = nullptr;
  415. extension->gdextension.track_instance = nullptr;
  416. extension->gdextension.untrack_instance = nullptr;
  417. }
  418. #endif
  419. ClassDB::register_extension_class(&extension->gdextension);
  420. if (p_extension_funcs->icon_path != nullptr) {
  421. const String icon_path = *reinterpret_cast<const String *>(p_extension_funcs->icon_path);
  422. if (!icon_path.is_empty()) {
  423. self->class_icon_paths[class_name] = icon_path;
  424. }
  425. }
  426. }
  427. void GDExtension::_register_extension_class_method(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, const GDExtensionClassMethodInfo *p_method_info) {
  428. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  429. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  430. StringName method_name = *reinterpret_cast<const StringName *>(p_method_info->name);
  431. ERR_FAIL_COND_MSG(!self->extension_classes.has(class_name), vformat("Attempt to register extension method '%s' for unexisting class '%s'.", String(method_name), class_name));
  432. #ifdef TOOLS_ENABLED
  433. Extension *extension = &self->extension_classes[class_name];
  434. GDExtensionMethodBind *method = nullptr;
  435. // If the extension is still marked as reloading, that means it failed to register again.
  436. if (extension->is_reloading) {
  437. return;
  438. }
  439. if (self->is_reloading && extension->methods.has(method_name)) {
  440. method = extension->methods[method_name];
  441. // Try to update the method bind. If it doesn't work (because it's incompatible) then
  442. // mark as invalid and create a new one.
  443. if (!method->is_reloading || !method->try_update(p_method_info)) {
  444. method->valid = false;
  445. self->invalid_methods.push_back(method);
  446. method = nullptr;
  447. }
  448. }
  449. if (method == nullptr) {
  450. method = memnew(GDExtensionMethodBind(p_method_info));
  451. method->set_instance_class(class_name);
  452. extension->methods[method_name] = method;
  453. } else {
  454. method->is_reloading = false;
  455. }
  456. #else
  457. GDExtensionMethodBind *method = memnew(GDExtensionMethodBind(p_method_info));
  458. method->set_instance_class(class_name);
  459. #endif
  460. ClassDB::bind_method_custom(class_name, method);
  461. }
  462. void GDExtension::_register_extension_class_virtual_method(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, const GDExtensionClassVirtualMethodInfo *p_method_info) {
  463. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  464. ClassDB::add_extension_class_virtual_method(class_name, p_method_info);
  465. }
  466. void GDExtension::_register_extension_class_integer_constant(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringNamePtr p_enum_name, GDExtensionConstStringNamePtr p_constant_name, GDExtensionInt p_constant_value, GDExtensionBool p_is_bitfield) {
  467. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  468. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  469. StringName enum_name = *reinterpret_cast<const StringName *>(p_enum_name);
  470. StringName constant_name = *reinterpret_cast<const StringName *>(p_constant_name);
  471. ERR_FAIL_COND_MSG(!self->extension_classes.has(class_name), vformat("Attempt to register extension constant '%s' for unexisting class '%s'.", constant_name, class_name));
  472. #ifdef TOOLS_ENABLED
  473. // If the extension is still marked as reloading, that means it failed to register again.
  474. Extension *extension = &self->extension_classes[class_name];
  475. if (extension->is_reloading) {
  476. return;
  477. }
  478. #endif
  479. ClassDB::bind_integer_constant(class_name, enum_name, constant_name, p_constant_value, p_is_bitfield);
  480. }
  481. void GDExtension::_register_extension_class_property(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, const GDExtensionPropertyInfo *p_info, GDExtensionConstStringNamePtr p_setter, GDExtensionConstStringNamePtr p_getter) {
  482. _register_extension_class_property_indexed(p_library, p_class_name, p_info, p_setter, p_getter, -1);
  483. }
  484. void GDExtension::_register_extension_class_property_indexed(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, const GDExtensionPropertyInfo *p_info, GDExtensionConstStringNamePtr p_setter, GDExtensionConstStringNamePtr p_getter, GDExtensionInt p_index) {
  485. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  486. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  487. StringName setter = *reinterpret_cast<const StringName *>(p_setter);
  488. StringName getter = *reinterpret_cast<const StringName *>(p_getter);
  489. String property_name = *reinterpret_cast<const StringName *>(p_info->name);
  490. ERR_FAIL_COND_MSG(!self->extension_classes.has(class_name), vformat("Attempt to register extension class property '%s' for unexisting class '%s'.", property_name, class_name));
  491. #ifdef TOOLS_ENABLED
  492. // If the extension is still marked as reloading, that means it failed to register again.
  493. Extension *extension = &self->extension_classes[class_name];
  494. if (extension->is_reloading) {
  495. return;
  496. }
  497. #endif
  498. PropertyInfo pinfo(*p_info);
  499. ClassDB::add_property(class_name, pinfo, setter, getter, p_index);
  500. }
  501. void GDExtension::_register_extension_class_property_group(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringPtr p_group_name, GDExtensionConstStringPtr p_prefix) {
  502. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  503. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  504. String group_name = *reinterpret_cast<const String *>(p_group_name);
  505. String prefix = *reinterpret_cast<const String *>(p_prefix);
  506. ERR_FAIL_COND_MSG(!self->extension_classes.has(class_name), vformat("Attempt to register extension class property group '%s' for unexisting class '%s'.", group_name, class_name));
  507. #ifdef TOOLS_ENABLED
  508. // If the extension is still marked as reloading, that means it failed to register again.
  509. Extension *extension = &self->extension_classes[class_name];
  510. if (extension->is_reloading) {
  511. return;
  512. }
  513. #endif
  514. ClassDB::add_property_group(class_name, group_name, prefix);
  515. }
  516. void GDExtension::_register_extension_class_property_subgroup(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringPtr p_subgroup_name, GDExtensionConstStringPtr p_prefix) {
  517. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  518. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  519. String subgroup_name = *reinterpret_cast<const String *>(p_subgroup_name);
  520. String prefix = *reinterpret_cast<const String *>(p_prefix);
  521. ERR_FAIL_COND_MSG(!self->extension_classes.has(class_name), vformat("Attempt to register extension class property subgroup '%s' for unexisting class '%s'.", subgroup_name, class_name));
  522. #ifdef TOOLS_ENABLED
  523. // If the extension is still marked as reloading, that means it failed to register again.
  524. Extension *extension = &self->extension_classes[class_name];
  525. if (extension->is_reloading) {
  526. return;
  527. }
  528. #endif
  529. ClassDB::add_property_subgroup(class_name, subgroup_name, prefix);
  530. }
  531. void GDExtension::_register_extension_class_signal(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, GDExtensionConstStringNamePtr p_signal_name, const GDExtensionPropertyInfo *p_argument_info, GDExtensionInt p_argument_count) {
  532. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  533. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  534. StringName signal_name = *reinterpret_cast<const StringName *>(p_signal_name);
  535. ERR_FAIL_COND_MSG(!self->extension_classes.has(class_name), vformat("Attempt to register extension class signal '%s' for unexisting class '%s'.", signal_name, class_name));
  536. #ifdef TOOLS_ENABLED
  537. // If the extension is still marked as reloading, that means it failed to register again.
  538. Extension *extension = &self->extension_classes[class_name];
  539. if (extension->is_reloading) {
  540. return;
  541. }
  542. #endif
  543. MethodInfo s;
  544. s.name = signal_name;
  545. for (int i = 0; i < p_argument_count; i++) {
  546. PropertyInfo arg(p_argument_info[i]);
  547. s.arguments.push_back(arg);
  548. }
  549. ClassDB::add_signal(class_name, s);
  550. }
  551. void GDExtension::_unregister_extension_class(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name) {
  552. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  553. StringName class_name = *reinterpret_cast<const StringName *>(p_class_name);
  554. ERR_FAIL_COND_MSG(!self->extension_classes.has(class_name), vformat("Attempt to unregister unexisting extension class '%s'.", class_name));
  555. Extension *ext = &self->extension_classes[class_name];
  556. #ifdef TOOLS_ENABLED
  557. if (ext->is_reloading) {
  558. self->_clear_extension(ext);
  559. }
  560. #endif
  561. ERR_FAIL_COND_MSG(ext->gdextension.children.size(), vformat("Attempt to unregister class '%s' while other extension classes inherit from it.", class_name));
  562. #ifdef TOOLS_ENABLED
  563. ClassDB::unregister_extension_class(class_name, !ext->is_reloading);
  564. #else
  565. ClassDB::unregister_extension_class(class_name);
  566. #endif
  567. if (ext->gdextension.parent != nullptr) {
  568. ext->gdextension.parent->children.erase(&ext->gdextension);
  569. }
  570. #ifdef TOOLS_ENABLED
  571. if (!ext->is_reloading) {
  572. self->extension_classes.erase(class_name);
  573. }
  574. GDExtensionEditorHelp::remove_class(class_name);
  575. #else
  576. self->extension_classes.erase(class_name);
  577. #endif
  578. }
  579. void GDExtension::_get_library_path(GDExtensionClassLibraryPtr p_library, GDExtensionUninitializedStringPtr r_path) {
  580. GDExtension *self = reinterpret_cast<GDExtension *>(p_library);
  581. Ref<GDExtensionLibraryLoader> library_loader = self->loader;
  582. String library_path;
  583. if (library_loader.is_valid()) {
  584. library_path = library_loader->library_path;
  585. }
  586. memnew_placement(r_path, String(library_path));
  587. }
  588. HashMap<StringName, GDExtensionInterfaceFunctionPtr> GDExtension::gdextension_interface_functions;
  589. void GDExtension::register_interface_function(const StringName &p_function_name, GDExtensionInterfaceFunctionPtr p_function_pointer) {
  590. ERR_FAIL_COND_MSG(gdextension_interface_functions.has(p_function_name), vformat("Attempt to register interface function '%s', which appears to be already registered.", p_function_name));
  591. gdextension_interface_functions.insert(p_function_name, p_function_pointer);
  592. }
  593. GDExtensionInterfaceFunctionPtr GDExtension::get_interface_function(const StringName &p_function_name) {
  594. GDExtensionInterfaceFunctionPtr *function = gdextension_interface_functions.getptr(p_function_name);
  595. ERR_FAIL_NULL_V_MSG(function, nullptr, vformat("Attempt to get non-existent interface function: '%s'.", String(p_function_name)));
  596. return *function;
  597. }
  598. Error GDExtension::open_library(const String &p_path, const Ref<GDExtensionLoader> &p_loader) {
  599. ERR_FAIL_COND_V_MSG(p_loader.is_null(), FAILED, "Can't open GDExtension without a loader.");
  600. loader = p_loader;
  601. Error err = loader->open_library(p_path);
  602. ERR_FAIL_COND_V_MSG(err == ERR_FILE_NOT_FOUND, err, vformat("GDExtension dynamic library not found: '%s'.", p_path));
  603. ERR_FAIL_COND_V_MSG(err != OK, err, vformat("Can't open GDExtension dynamic library: '%s'.", p_path));
  604. err = loader->initialize(&gdextension_get_proc_address, this, &initialization);
  605. if (err != OK) {
  606. // Errors already logged in initialize().
  607. loader->close_library();
  608. return err;
  609. }
  610. level_initialized = -1;
  611. return OK;
  612. }
  613. void GDExtension::close_library() {
  614. ERR_FAIL_COND(!is_library_open());
  615. loader->close_library();
  616. class_icon_paths.clear();
  617. #ifdef TOOLS_ENABLED
  618. instance_bindings.clear();
  619. #endif
  620. }
  621. bool GDExtension::is_library_open() const {
  622. return loader.is_valid() && loader->is_library_open();
  623. }
  624. GDExtension::InitializationLevel GDExtension::get_minimum_library_initialization_level() const {
  625. ERR_FAIL_COND_V(!is_library_open(), INITIALIZATION_LEVEL_CORE);
  626. return InitializationLevel(initialization.minimum_initialization_level);
  627. }
  628. void GDExtension::initialize_library(InitializationLevel p_level) {
  629. ERR_FAIL_COND(!is_library_open());
  630. ERR_FAIL_COND_MSG(p_level <= int32_t(level_initialized), vformat("Level '%d' must be higher than the current level '%d'", p_level, level_initialized));
  631. level_initialized = int32_t(p_level);
  632. ERR_FAIL_NULL(initialization.initialize);
  633. initialization.initialize(initialization.userdata, GDExtensionInitializationLevel(p_level));
  634. }
  635. void GDExtension::deinitialize_library(InitializationLevel p_level) {
  636. ERR_FAIL_COND(!is_library_open());
  637. ERR_FAIL_COND(p_level > int32_t(level_initialized));
  638. level_initialized = int32_t(p_level) - 1;
  639. ERR_FAIL_NULL(initialization.deinitialize);
  640. initialization.deinitialize(initialization.userdata, GDExtensionInitializationLevel(p_level));
  641. }
  642. void GDExtension::_bind_methods() {
  643. ClassDB::bind_method(D_METHOD("is_library_open"), &GDExtension::is_library_open);
  644. ClassDB::bind_method(D_METHOD("get_minimum_library_initialization_level"), &GDExtension::get_minimum_library_initialization_level);
  645. BIND_ENUM_CONSTANT(INITIALIZATION_LEVEL_CORE);
  646. BIND_ENUM_CONSTANT(INITIALIZATION_LEVEL_SERVERS);
  647. BIND_ENUM_CONSTANT(INITIALIZATION_LEVEL_SCENE);
  648. BIND_ENUM_CONSTANT(INITIALIZATION_LEVEL_EDITOR);
  649. }
  650. GDExtension::GDExtension() {
  651. }
  652. GDExtension::~GDExtension() {
  653. if (is_library_open()) {
  654. close_library();
  655. }
  656. #ifdef TOOLS_ENABLED
  657. // If we have any invalid method binds still laying around, we can finally free them!
  658. for (GDExtensionMethodBind *E : invalid_methods) {
  659. memdelete(E);
  660. }
  661. #endif
  662. }
  663. void GDExtension::initialize_gdextensions() {
  664. gdextension_setup_interface();
  665. #ifndef DISABLE_DEPRECATED
  666. register_interface_function("classdb_register_extension_class", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class);
  667. register_interface_function("classdb_register_extension_class2", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class2);
  668. register_interface_function("classdb_register_extension_class3", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class3);
  669. #endif // DISABLE_DEPRECATED
  670. register_interface_function("classdb_register_extension_class4", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class4);
  671. register_interface_function("classdb_register_extension_class_method", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_method);
  672. register_interface_function("classdb_register_extension_class_virtual_method", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_virtual_method);
  673. register_interface_function("classdb_register_extension_class_integer_constant", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_integer_constant);
  674. register_interface_function("classdb_register_extension_class_property", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_property);
  675. register_interface_function("classdb_register_extension_class_property_indexed", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_property_indexed);
  676. register_interface_function("classdb_register_extension_class_property_group", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_property_group);
  677. register_interface_function("classdb_register_extension_class_property_subgroup", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_property_subgroup);
  678. register_interface_function("classdb_register_extension_class_signal", (GDExtensionInterfaceFunctionPtr)&GDExtension::_register_extension_class_signal);
  679. register_interface_function("classdb_unregister_extension_class", (GDExtensionInterfaceFunctionPtr)&GDExtension::_unregister_extension_class);
  680. register_interface_function("get_library_path", (GDExtensionInterfaceFunctionPtr)&GDExtension::_get_library_path);
  681. }
  682. void GDExtension::finalize_gdextensions() {
  683. gdextension_interface_functions.clear();
  684. }
  685. Error GDExtensionResourceLoader::load_gdextension_resource(const String &p_path, Ref<GDExtension> &p_extension) {
  686. ERR_FAIL_COND_V_MSG(p_extension.is_valid() && p_extension->is_library_open(), ERR_ALREADY_IN_USE, "Cannot load GDExtension resource into already opened library.");
  687. GDExtensionManager *extension_manager = GDExtensionManager::get_singleton();
  688. GDExtensionManager::LoadStatus status = extension_manager->load_extension(p_path);
  689. if (status != GDExtensionManager::LOAD_STATUS_OK && status != GDExtensionManager::LOAD_STATUS_ALREADY_LOADED) {
  690. // Errors already logged in load_extension().
  691. return FAILED;
  692. }
  693. p_extension = extension_manager->get_extension(p_path);
  694. return OK;
  695. }
  696. Ref<Resource> GDExtensionResourceLoader::load(const String &p_path, const String &p_original_path, Error *r_error, bool p_use_sub_threads, float *r_progress, CacheMode p_cache_mode) {
  697. // We can't have two GDExtension resource object representing the same library, because
  698. // loading (or unloading) a GDExtension affects global data. So, we need reuse the same
  699. // object if one has already been loaded (even if caching is disabled at the resource
  700. // loader level).
  701. GDExtensionManager *manager = GDExtensionManager::get_singleton();
  702. if (manager->is_extension_loaded(p_path)) {
  703. return manager->get_extension(p_path);
  704. }
  705. Ref<GDExtension> lib;
  706. Error err = load_gdextension_resource(p_path, lib);
  707. if (err != OK && r_error) {
  708. // Errors already logged in load_gdextension_resource().
  709. *r_error = err;
  710. }
  711. return lib;
  712. }
  713. void GDExtensionResourceLoader::get_recognized_extensions(List<String> *p_extensions) const {
  714. p_extensions->push_back("gdextension");
  715. }
  716. bool GDExtensionResourceLoader::handles_type(const String &p_type) const {
  717. return p_type == "GDExtension";
  718. }
  719. String GDExtensionResourceLoader::get_resource_type(const String &p_path) const {
  720. String el = p_path.get_extension().to_lower();
  721. if (el == "gdextension") {
  722. return "GDExtension";
  723. }
  724. return "";
  725. }
  726. #ifdef TOOLS_ENABLED
  727. bool GDExtension::has_library_changed() const {
  728. return loader->has_library_changed();
  729. }
  730. void GDExtension::prepare_reload() {
  731. is_reloading = true;
  732. for (KeyValue<StringName, Extension> &E : extension_classes) {
  733. E.value.is_reloading = true;
  734. for (KeyValue<StringName, GDExtensionMethodBind *> &M : E.value.methods) {
  735. M.value->is_reloading = true;
  736. }
  737. for (const ObjectID &obj_id : E.value.instances) {
  738. Object *obj = ObjectDB::get_instance(obj_id);
  739. if (!obj) {
  740. continue;
  741. }
  742. // Store instance state so it can be restored after reload.
  743. List<Pair<String, Variant>> state;
  744. List<PropertyInfo> prop_list;
  745. obj->get_property_list(&prop_list);
  746. for (const PropertyInfo &P : prop_list) {
  747. if (!(P.usage & PROPERTY_USAGE_STORAGE)) {
  748. continue;
  749. }
  750. Variant value = obj->get(P.name);
  751. Variant default_value = ClassDB::class_get_default_property_value(obj->get_class_name(), P.name);
  752. if (default_value.get_type() != Variant::NIL && bool(Variant::evaluate(Variant::OP_EQUAL, value, default_value))) {
  753. continue;
  754. }
  755. if (P.type == Variant::OBJECT && value.is_zero() && !(P.usage & PROPERTY_USAGE_STORE_IF_NULL)) {
  756. continue;
  757. }
  758. state.push_back(Pair<String, Variant>(P.name, value));
  759. }
  760. E.value.instance_state[obj_id] = {
  761. state, // List<Pair<String, Variant>> properties;
  762. obj->is_extension_placeholder(), // bool is_placeholder;
  763. };
  764. }
  765. }
  766. }
  767. void GDExtension::_clear_extension(Extension *p_extension) {
  768. // Clear out hierarchy information because it may change.
  769. p_extension->gdextension.parent = nullptr;
  770. p_extension->gdextension.children.clear();
  771. // Clear all objects of any GDExtension data. It will become its native parent class
  772. // until the reload can reset the object with the new GDExtension data.
  773. for (const ObjectID &obj_id : p_extension->instances) {
  774. Object *obj = ObjectDB::get_instance(obj_id);
  775. if (!obj) {
  776. continue;
  777. }
  778. obj->clear_internal_extension();
  779. }
  780. }
  781. void GDExtension::track_instance_binding(Object *p_object) {
  782. instance_bindings.push_back(p_object->get_instance_id());
  783. }
  784. void GDExtension::untrack_instance_binding(Object *p_object) {
  785. instance_bindings.erase(p_object->get_instance_id());
  786. }
  787. void GDExtension::clear_instance_bindings() {
  788. for (ObjectID obj_id : instance_bindings) {
  789. Object *obj = ObjectDB::get_instance(obj_id);
  790. if (!obj) {
  791. continue;
  792. }
  793. obj->free_instance_binding(this);
  794. }
  795. instance_bindings.clear();
  796. }
  797. void GDExtension::finish_reload() {
  798. is_reloading = false;
  799. // Clean up any classes or methods that didn't get re-added.
  800. Vector<StringName> classes_to_remove;
  801. for (KeyValue<StringName, Extension> &E : extension_classes) {
  802. if (E.value.is_reloading) {
  803. E.value.is_reloading = false;
  804. classes_to_remove.push_back(E.key);
  805. }
  806. Vector<StringName> methods_to_remove;
  807. for (KeyValue<StringName, GDExtensionMethodBind *> &M : E.value.methods) {
  808. if (M.value->is_reloading) {
  809. M.value->valid = false;
  810. invalid_methods.push_back(M.value);
  811. M.value->is_reloading = false;
  812. methods_to_remove.push_back(M.key);
  813. }
  814. }
  815. for (const StringName &method_name : methods_to_remove) {
  816. E.value.methods.erase(method_name);
  817. }
  818. }
  819. for (const StringName &class_name : classes_to_remove) {
  820. extension_classes.erase(class_name);
  821. }
  822. // Reset any the extension on instances made from the classes that remain.
  823. for (KeyValue<StringName, Extension> &E : extension_classes) {
  824. // Loop over 'instance_state' rather than 'instance' because new instances
  825. // may have been created when re-initializing the extension.
  826. for (const KeyValue<ObjectID, Extension::InstanceState> &S : E.value.instance_state) {
  827. Object *obj = ObjectDB::get_instance(S.key);
  828. if (!obj) {
  829. continue;
  830. }
  831. if (S.value.is_placeholder) {
  832. obj->reset_internal_extension(ClassDB::get_placeholder_extension(E.value.gdextension.class_name));
  833. } else {
  834. obj->reset_internal_extension(&E.value.gdextension);
  835. }
  836. }
  837. }
  838. // Now that all the classes are back, restore the state.
  839. for (KeyValue<StringName, Extension> &E : extension_classes) {
  840. for (const KeyValue<ObjectID, Extension::InstanceState> &S : E.value.instance_state) {
  841. Object *obj = ObjectDB::get_instance(S.key);
  842. if (!obj) {
  843. continue;
  844. }
  845. for (const Pair<String, Variant> &state : S.value.properties) {
  846. obj->set(state.first, state.second);
  847. }
  848. }
  849. }
  850. // Finally, let the objects know that we are done reloading them.
  851. for (KeyValue<StringName, Extension> &E : extension_classes) {
  852. for (const KeyValue<ObjectID, Extension::InstanceState> &S : E.value.instance_state) {
  853. Object *obj = ObjectDB::get_instance(S.key);
  854. if (!obj) {
  855. continue;
  856. }
  857. obj->notification(NOTIFICATION_EXTENSION_RELOADED);
  858. }
  859. // Clear the instance state, we're done looping.
  860. E.value.instance_state.clear();
  861. }
  862. }
  863. void GDExtension::_track_instance(void *p_user_data, void *p_instance) {
  864. Extension *extension = reinterpret_cast<Extension *>(p_user_data);
  865. Object *obj = reinterpret_cast<Object *>(p_instance);
  866. extension->instances.insert(obj->get_instance_id());
  867. }
  868. void GDExtension::_untrack_instance(void *p_user_data, void *p_instance) {
  869. Extension *extension = reinterpret_cast<Extension *>(p_user_data);
  870. Object *obj = reinterpret_cast<Object *>(p_instance);
  871. extension->instances.erase(obj->get_instance_id());
  872. }
  873. Vector<StringName> GDExtensionEditorPlugins::extension_classes;
  874. GDExtensionEditorPlugins::EditorPluginRegisterFunc GDExtensionEditorPlugins::editor_node_add_plugin = nullptr;
  875. GDExtensionEditorPlugins::EditorPluginRegisterFunc GDExtensionEditorPlugins::editor_node_remove_plugin = nullptr;
  876. void GDExtensionEditorPlugins::add_extension_class(const StringName &p_class_name) {
  877. if (editor_node_add_plugin) {
  878. editor_node_add_plugin(p_class_name);
  879. } else {
  880. extension_classes.push_back(p_class_name);
  881. }
  882. }
  883. void GDExtensionEditorPlugins::remove_extension_class(const StringName &p_class_name) {
  884. if (editor_node_remove_plugin) {
  885. editor_node_remove_plugin(p_class_name);
  886. } else {
  887. extension_classes.erase(p_class_name);
  888. }
  889. }
  890. GDExtensionEditorHelp::EditorHelpLoadXmlBufferFunc GDExtensionEditorHelp::editor_help_load_xml_buffer = nullptr;
  891. GDExtensionEditorHelp::EditorHelpRemoveClassFunc GDExtensionEditorHelp::editor_help_remove_class = nullptr;
  892. void GDExtensionEditorHelp::load_xml_buffer(const uint8_t *p_buffer, int p_size) {
  893. ERR_FAIL_NULL(editor_help_load_xml_buffer);
  894. editor_help_load_xml_buffer(p_buffer, p_size);
  895. }
  896. void GDExtensionEditorHelp::remove_class(const String &p_class) {
  897. ERR_FAIL_NULL(editor_help_remove_class);
  898. editor_help_remove_class(p_class);
  899. }
  900. #endif // TOOLS_ENABLED