Skip to content

Commit a969e47

Browse files
dsnopekdsnopek
committed
Allow GDExtensions to register virtual methods and call them on scripts
1 parent 36847f6 commit a969e47

File tree

14 files changed

+310
-1
lines changed

14 files changed

+310
-1
lines changed

‎binding_generator.py

Lines changed: 132 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,136 @@ def generate_wrappers(target):
7070
f.write(txt)
7171

7272

73+
def generate_virtual_version(argcount, const=False, returns=False):
74+
s = """#define GDVIRTUAL$VER($RET m_name $ARG)\\
75+
StringName _gdvirtual_##m_name##_sn = #m_name;\\
76+
template <bool required>\\
77+
_FORCE_INLINE_ bool _gdvirtual_##m_name##_call($CALLARGS) $CONST {\\
78+
if (::godot::internal::gdextension_interface_object_has_script_method(_owner, &_gdvirtual_##m_name##_sn)) { \\
79+
GDExtensionCallError ce;\\
80+
$CALLSIARGS\\
81+
$CALLSIBEGIN::godot::internal::gdextension_interface_object_call_script_method(_owner, &_gdvirtual_##m_name##_sn, $CALLSIARGPASS, $CALLSIRETPASS, &ce);\\
82+
if (ce.error == GDEXTENSION_CALL_OK) {\\
83+
$CALLSIRET\\
84+
return true;\\
85+
}\\
86+
}\\
87+
if (required) {\\
88+
ERR_PRINT_ONCE("Required virtual method " + get_class() + "::" + #m_name + " must be overridden before calling.");\\
89+
$RVOID\\
90+
}\\
91+
return false;\\
92+
}\\
93+
_FORCE_INLINE_ bool _gdvirtual_##m_name##_overridden() const {\\
94+
return godot::internal::gdextension_interface_object_has_script_method(_owner, &_gdvirtual_##m_name##_sn); \\
95+
}\\
96+
_FORCE_INLINE_ static MethodInfo _gdvirtual_##m_name##_get_method_info() {\\
97+
MethodInfo method_info;\\
98+
method_info.name = #m_name;\\
99+
method_info.flags = $METHOD_FLAGS;\\
100+
$FILL_METHOD_INFO\\
101+
return method_info;\\
102+
}
103+
104+
"""
105+
106+
sproto = str(argcount)
107+
method_info = ""
108+
if returns:
109+
sproto += "R"
110+
s = s.replace("$RET", "m_ret,")
111+
s = s.replace("$RVOID", "(void)r_ret;") # If required, may lead to uninitialized errors
112+
method_info += "method_info.return_val = GetTypeInfo<m_ret>::get_class_info();\\\n"
113+
method_info += "\t\tmethod_info.return_val_metadata = GetTypeInfo<m_ret>::METADATA;"
114+
else:
115+
s = s.replace("$RET ", "")
116+
s = s.replace("\t\t\t$RVOID\\\n", "")
117+
118+
if const:
119+
sproto += "C"
120+
s = s.replace("$CONST", "const")
121+
s = s.replace("$METHOD_FLAGS", "METHOD_FLAG_VIRTUAL | METHOD_FLAG_CONST")
122+
else:
123+
s = s.replace("$CONST ", "")
124+
s = s.replace("$METHOD_FLAGS", "METHOD_FLAG_VIRTUAL")
125+
126+
s = s.replace("$VER", sproto)
127+
argtext = ""
128+
callargtext = ""
129+
callsiargs = ""
130+
callsiargptrs = ""
131+
if argcount > 0:
132+
argtext += ", "
133+
callsiargs = f"Variant vargs[{argcount}] = {{ "
134+
callsiargptrs = f"\t\t\tconst Variant *vargptrs[{argcount}] = {{ "
135+
for i in range(argcount):
136+
if i > 0:
137+
argtext += ", "
138+
callargtext += ", "
139+
callsiargs += ", "
140+
callsiargptrs += ", "
141+
argtext += f"m_type{i + 1}"
142+
callargtext += f"m_type{i + 1} arg{i + 1}"
143+
callsiargs += f"Variant(arg{i + 1})"
144+
callsiargptrs += f"&vargs[{i}]"
145+
if method_info:
146+
method_info += "\\\n\t\t"
147+
method_info += f"method_info.arguments.push_back(GetTypeInfo<m_type{i + 1}>::get_class_info());\\\n"
148+
method_info += f"\t\tmethod_info.arguments_metadata.push_back(GetTypeInfo<m_type{i + 1}>::METADATA);"
149+
150+
if argcount:
151+
callsiargs += " };\\\n"
152+
callsiargptrs += " };"
153+
s = s.replace("$CALLSIARGS", callsiargs + callsiargptrs)
154+
s = s.replace("$CALLSIARGPASS", f"(const GDExtensionConstVariantPtr *)vargptrs, {argcount}")
155+
else:
156+
s = s.replace("\t\t\t$CALLSIARGS\\\n", "")
157+
s = s.replace("$CALLSIARGPASS", "nullptr, 0")
158+
159+
if returns:
160+
if argcount > 0:
161+
callargtext += ", "
162+
callargtext += "m_ret &r_ret"
163+
s = s.replace("$CALLSIBEGIN", "Variant ret;\\\n\t\t\t")
164+
s = s.replace("$CALLSIRETPASS", "&ret")
165+
s = s.replace("$CALLSIRET", "r_ret = VariantCaster<m_ret>::cast(ret);")
166+
else:
167+
s = s.replace("$CALLSIBEGIN", "")
168+
s = s.replace("$CALLSIRETPASS", "nullptr")
169+
s = s.replace("\t\t\t\t$CALLSIRET\\\n", "")
170+
171+
s = s.replace(" $ARG", argtext)
172+
s = s.replace("$CALLARGS", callargtext)
173+
if method_info:
174+
s = s.replace("$FILL_METHOD_INFO", method_info)
175+
else:
176+
s = s.replace("\t\t$FILL_METHOD_INFO\\\n", method_info)
177+
178+
return s
179+
180+
181+
def generate_virtuals(target):
182+
max_versions = 12
183+
184+
txt = """/* THIS FILE IS GENERATED DO NOT EDIT */
185+
#ifndef GDEXTENSION_GDVIRTUAL_GEN_H
186+
#define GDEXTENSION_GDVIRTUAL_GEN_H
187+
188+
"""
189+
190+
for i in range(max_versions + 1):
191+
txt += f"/* {i} Arguments */\n\n"
192+
txt += generate_virtual_version(i, False, False)
193+
txt += generate_virtual_version(i, False, True)
194+
txt += generate_virtual_version(i, True, False)
195+
txt += generate_virtual_version(i, True, True)
196+
197+
txt += "#endif // GDEXTENSION_GDVIRTUAL_GEN_H\n"
198+
199+
with open(target, "w", encoding="utf-8") as f:
200+
f.write(txt)
201+
202+
73203
def get_file_list(api_filepath, output_dir, headers=False, sources=False):
74204
api = {}
75205
files = []
@@ -81,6 +211,7 @@ def get_file_list(api_filepath, output_dir, headers=False, sources=False):
81211
source_gen_folder = Path(output_dir) / "gen" / "src"
82212

83213
files.append(str((core_gen_folder / "ext_wrappers.gen.inc").as_posix()))
214+
files.append(str((core_gen_folder / "gdvirtual.gen.inc").as_posix()))
84215

85216
for builtin_class in api["builtin_classes"]:
86217
if is_pod_type(builtin_class["name"]):
@@ -204,6 +335,7 @@ def generate_builtin_bindings(api, output_dir, build_config):
204335
source_gen_folder.mkdir(parents=True, exist_ok=True)
205336

206337
generate_wrappers(core_gen_folder / "ext_wrappers.gen.inc")
338+
generate_virtuals(core_gen_folder / "gdvirtual.gen.inc")
207339

208340
# Store types beforehand.
209341
for builtin_api in api["builtin_classes"]:

‎gdextension/gdextension_interface.h

Lines changed: 54 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -381,6 +381,18 @@ typedef struct {
381381
GDExtensionVariantPtr *default_arguments;
382382
} GDExtensionClassMethodInfo;
383383

384+
typedef struct {
385+
GDExtensionStringNamePtr name;
386+
uint32_t method_flags; // Bitfield of `GDExtensionClassMethodFlags`.
387+
388+
GDExtensionPropertyInfo return_value;
389+
GDExtensionClassMethodArgumentMetadata return_value_metadata;
390+
391+
uint32_t argument_count;
392+
GDExtensionPropertyInfo *arguments_info;
393+
GDExtensionClassMethodArgumentMetadata *arguments_metadata;
394+
} GDExtensionClassVirtualMethodInfo;
395+
384396
typedef void (*GDExtensionCallableCustomCall)(void *callable_userdata, const GDExtensionConstVariantPtr *p_args, GDExtensionInt p_argument_count, GDExtensionVariantPtr r_return, GDExtensionCallError *r_error);
385397
typedef GDExtensionBool (*GDExtensionCallableCustomIsValid)(void *callable_userdata);
386398
typedef void (*GDExtensionCallableCustomFree)(void *callable_userdata);
@@ -2268,6 +2280,34 @@ typedef GDExtensionObjectPtr (*GDExtensionInterfaceObjectGetInstanceFromId)(GDOb
22682280
*/
22692281
typedef GDObjectInstanceID (*GDExtensionInterfaceObjectGetInstanceId)(GDExtensionConstObjectPtr p_object);
22702282

2283+
/**
2284+
* @name object_has_script_method
2285+
* @since 4.3
2286+
*
2287+
* Checks if this object has a script with the given method.
2288+
*
2289+
* @param p_object A pointer to the Object.
2290+
* @param p_method A pointer to a StringName identifying the method.
2291+
*
2292+
* @returns true if the method exists.
2293+
*/
2294+
typedef GDExtensionBool (*GDExtensionInterfaceObjectHasScriptMethod)(GDExtensionConstObjectPtr p_object, GDExtensionConstStringNamePtr p_method);
2295+
2296+
/**
2297+
* @name object_call_script_method
2298+
* @since 4.3
2299+
*
2300+
* Call the given script method on this object.
2301+
*
2302+
* @param p_object A pointer to the Object.
2303+
* @param p_method A pointer to a StringName identifying the method.
2304+
* @param p_args A pointer to a C array of Variant.
2305+
* @param p_argument_count The number of arguments.
2306+
* @param r_return A pointer a Variant which will be assigned the return value.
2307+
* @param r_error A pointer the structure which will hold error information.
2308+
*/
2309+
typedef void (*GDExtensionInterfaceObjectCallScriptMethod)(GDExtensionObjectPtr p_object, GDExtensionConstStringNamePtr p_method, const GDExtensionConstVariantPtr *p_args, GDExtensionInt p_argument_count, GDExtensionUninitializedVariantPtr r_return, GDExtensionCallError *r_error);
2310+
22712311
/* INTERFACE: Reference */
22722312

22732313
/**
@@ -2483,6 +2523,20 @@ typedef void (*GDExtensionInterfaceClassdbRegisterExtensionClass2)(GDExtensionCl
24832523
*/
24842524
typedef void (*GDExtensionInterfaceClassdbRegisterExtensionClassMethod)(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, const GDExtensionClassMethodInfo *p_method_info);
24852525

2526+
/**
2527+
* @name classdb_register_extension_class_virtual_method
2528+
* @since 4.3
2529+
*
2530+
* Registers a virtual method on an extension class in ClassDB, that can be implemented by scripts or other extensions.
2531+
*
2532+
* Provided struct can be safely freed once the function returns.
2533+
*
2534+
* @param p_library A pointer the library received by the GDExtension's entry point function.
2535+
* @param p_class_name A pointer to a StringName with the class name.
2536+
* @param p_method_info A pointer to a GDExtensionClassMethodInfo struct.
2537+
*/
2538+
typedef void (*GDExtensionInterfaceClassdbRegisterExtensionClassVirtualMethod)(GDExtensionClassLibraryPtr p_library, GDExtensionConstStringNamePtr p_class_name, const GDExtensionClassVirtualMethodInfo *p_method_info);
2539+
24862540
/**
24872541
* @name classdb_register_extension_class_integer_constant
24882542
* @since 4.1

‎include/godot_cpp/classes/wrapped.hpp

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
#include <godot_cpp/core/property_info.hpp>
3737

3838
#include <godot_cpp/templates/list.hpp>
39+
#include <godot_cpp/templates/vector.hpp>
3940

4041
#include <godot_cpp/godot.hpp>
4142

@@ -107,6 +108,26 @@ class Wrapped {
107108
GodotObject *_owner = nullptr;
108109
};
109110

111+
_FORCE_INLINE_ void snarray_add_str(Vector<StringName> &arr) {
112+
}
113+
114+
_FORCE_INLINE_ void snarray_add_str(Vector<StringName> &arr, const StringName &p_str) {
115+
arr.push_back(p_str);
116+
}
117+
118+
template <class... P>
119+
_FORCE_INLINE_ void snarray_add_str(Vector<StringName> &arr, const StringName &p_str, P... p_args) {
120+
arr.push_back(p_str);
121+
snarray_add_str(arr, p_args...);
122+
}
123+
124+
template <class... P>
125+
_FORCE_INLINE_ Vector<StringName> snarray(P... p_args) {
126+
Vector<StringName> arr;
127+
snarray_add_str(arr, p_args...);
128+
return arr;
129+
}
130+
110131
namespace internal {
111132

112133
GDExtensionPropertyInfo *create_c_property_list(const ::godot::List<::godot::PropertyInfo> &plist_cpp, uint32_t *r_size);
@@ -445,4 +466,14 @@ public:
445466
// Don't use this for your classes, use GDCLASS() instead.
446467
#define GDEXTENSION_CLASS(m_class, m_inherits) GDEXTENSION_CLASS_ALIAS(m_class, m_class, m_inherits)
447468

469+
#define GDVIRTUAL_CALL(m_name, ...) _gdvirtual_##m_name##_call<false>(__VA_ARGS__)
470+
#define GDVIRTUAL_CALL_PTR(m_obj, m_name, ...) m_obj->_gdvirtual_##m_name##_call<false>(__VA_ARGS__)
471+
472+
#define GDVIRTUAL_REQUIRED_CALL(m_name, ...) _gdvirtual_##m_name##_call<true>(__VA_ARGS__)
473+
#define GDVIRTUAL_REQUIRED_CALL_PTR(m_obj, m_name, ...) m_obj->_gdvirtual_##m_name##_call<true>(__VA_ARGS__)
474+
475+
#define GDVIRTUAL_BIND(m_name, ...) ::godot::ClassDB::add_virtual_method(get_class_static(), _gdvirtual_##m_name##_get_method_info(), ::godot::snarray(__VA_ARGS__));
476+
#define GDVIRTUAL_IS_OVERRIDDEN(m_name) _gdvirtual_##m_name##_overridden()
477+
#define GDVIRTUAL_IS_OVERRIDDEN_PTR(m_obj, m_name) m_obj->_gdvirtual_##m_name##_overridden()
478+
448479
#endif // GODOT_WRAPPED_HPP

‎include/godot_cpp/core/class_db.hpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,10 @@ class ClassDB {
165165
static void add_property(const StringName &p_class, const PropertyInfo &p_pinfo, const StringName &p_setter, const StringName &p_getter, int p_index = -1);
166166
static void add_signal(const StringName &p_class, const MethodInfo &p_signal);
167167
static void bind_integer_constant(const StringName &p_class_name, const StringName &p_enum_name, const StringName &p_constant_name, GDExtensionInt p_constant_value, bool p_is_bitfield = false);
168+
// Binds an implementation of a virtual method defined in Godot.
168169
static void bind_virtual_method(const StringName &p_class, const StringName &p_method, GDExtensionClassCallVirtual p_call);
170+
// Add a new virtual method that can be implemented by scripts.
171+
static void add_virtual_method(const StringName &p_class, const MethodInfo &p_method, const Vector<StringName> &p_arg_names = Vector<StringName>());
169172

170173
static MethodBind *get_method(const StringName &p_class, const StringName &p_method);
171174

‎include/godot_cpp/core/object.hpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,8 @@ struct MethodInfo {
6868
int id = 0;
6969
std::vector<PropertyInfo> arguments;
7070
std::vector<Variant> default_arguments;
71+
GDExtensionClassMethodArgumentMetadata return_val_metadata;
72+
std::vector<GDExtensionClassMethodArgumentMetadata> arguments_metadata;
7173

7274
inline bool operator==(const MethodInfo &p_method) const { return id == p_method.id; }
7375
inline bool operator<(const MethodInfo &p_method) const { return id == p_method.id ? (name < p_method.name) : (id < p_method.id); }

‎include/godot_cpp/core/property_info.hpp

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,17 @@ struct PropertyInfo {
8080
p_info->usage = usage;
8181
*(reinterpret_cast<StringName *>(p_info->class_name)) = class_name;
8282
}
83+
84+
GDExtensionPropertyInfo _to_gdextension() const {
85+
return {
86+
(GDExtensionVariantType)type,
87+
name._native_ptr(),
88+
class_name._native_ptr(),
89+
hint,
90+
hint_string._native_ptr(),
91+
usage,
92+
};
93+
}
8394
};
8495

8596
} // namespace godot

‎include/godot_cpp/godot.hpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,8 @@ extern "C" GDExtensionInterfaceObjectGetClassName gdextension_interface_object_g
165165
extern "C" GDExtensionInterfaceObjectCastTo gdextension_interface_object_cast_to;
166166
extern "C" GDExtensionInterfaceObjectGetInstanceFromId gdextension_interface_object_get_instance_from_id;
167167
extern "C" GDExtensionInterfaceObjectGetInstanceId gdextension_interface_object_get_instance_id;
168+
extern "C" GDExtensionInterfaceObjectHasScriptMethod gdextension_interface_object_has_script_method;
169+
extern "C" GDExtensionInterfaceObjectCallScriptMethod gdextension_interface_object_call_script_method;
168170
extern "C" GDExtensionInterfaceCallableCustomCreate gdextension_interface_callable_custom_create;
169171
extern "C" GDExtensionInterfaceCallableCustomGetUserData gdextension_interface_callable_custom_get_userdata;
170172
extern "C" GDExtensionInterfaceRefGetObject gdextension_interface_ref_get_object;
@@ -177,6 +179,7 @@ extern "C" GDExtensionInterfaceClassdbGetMethodBind gdextension_interface_classd
177179
extern "C" GDExtensionInterfaceClassdbGetClassTag gdextension_interface_classdb_get_class_tag;
178180
extern "C" GDExtensionInterfaceClassdbRegisterExtensionClass2 gdextension_interface_classdb_register_extension_class2;
179181
extern "C" GDExtensionInterfaceClassdbRegisterExtensionClassMethod gdextension_interface_classdb_register_extension_class_method;
182+
extern "C" GDExtensionInterfaceClassdbRegisterExtensionClassVirtualMethod gdextension_interface_classdb_register_extension_class_virtual_method;
180183
extern "C" GDExtensionInterfaceClassdbRegisterExtensionClassIntegerConstant gdextension_interface_classdb_register_extension_class_integer_constant;
181184
extern "C" GDExtensionInterfaceClassdbRegisterExtensionClassProperty gdextension_interface_classdb_register_extension_class_property;
182185
extern "C" GDExtensionInterfaceClassdbRegisterExtensionClassPropertyIndexed gdextension_interface_classdb_register_extension_class_property_indexed;

‎src/core/class_db.cpp

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232

3333
#include <godot_cpp/core/error_macros.hpp>
3434
#include <godot_cpp/godot.hpp>
35+
#include <godot_cpp/templates/vector.hpp>
3536

3637
#include <godot_cpp/core/memory.hpp>
3738

@@ -337,6 +338,46 @@ void ClassDB::bind_virtual_method(const StringName &p_class, const StringName &p
337338
type.virtual_methods[p_method] = p_call;
338339
}
339340

341+
void ClassDB::add_virtual_method(const StringName &p_class, const MethodInfo &p_method, const Vector<StringName> &p_arg_names) {
342+
std::unordered_map<StringName, ClassInfo>::iterator type_it = classes.find(p_class);
343+
ERR_FAIL_COND_MSG(type_it == classes.end(), String("Class '{0}' doesn't exist.").format(Array::make(p_class)));
344+
345+
GDExtensionClassVirtualMethodInfo mi;
346+
mi.name = (GDExtensionStringNamePtr)&p_method.name;
347+
mi.method_flags = p_method.flags;
348+
mi.return_value = p_method.return_val._to_gdextension();
349+
mi.return_value_metadata = p_method.return_val_metadata;
350+
mi.argument_count = p_method.arguments.size();
351+
if (mi.argument_count > 0) {
352+
mi.arguments_info = (GDExtensionPropertyInfo *)memalloc(sizeof(GDExtensionPropertyInfo) * mi.argument_count);
353+
mi.arguments_metadata = (GDExtensionClassMethodArgumentMetadata *)memalloc(sizeof(GDExtensionClassMethodArgumentMetadata) * mi.argument_count);
354+
for (int i = 0; i < mi.argument_count; i++) {
355+
mi.arguments_info[i] = p_method.arguments[i]._to_gdextension();
356+
mi.arguments_metadata[i] = p_method.arguments_metadata[i];
357+
}
358+
} else {
359+
mi.arguments_info = nullptr;
360+
mi.arguments_metadata = nullptr;
361+
}
362+
363+
if (p_arg_names.size() != mi.argument_count) {
364+
WARN_PRINT("Mismatch argument name count for virtual method: " + String(p_class) + "::" + p_method.name);
365+
} else {
366+
for (int i = 0; i < p_arg_names.size(); i++) {
367+
mi.arguments_info[i].name = (GDExtensionStringNamePtr)&p_arg_names[i];
368+
}
369+
}
370+
371+
internal::gdextension_interface_classdb_register_extension_class_virtual_method(internal::library, &p_class, &mi);
372+
373+
if (mi.arguments_info) {
374+
memfree(mi.arguments_info);
375+
}
376+
if (mi.arguments_metadata) {
377+
memfree(mi.arguments_metadata);
378+
}
379+
}
380+
340381
void ClassDB::initialize_class(const ClassInfo &p_cl) {
341382
}
342383

0 commit comments

Comments
 (0)