def pyobj_to_gdobj(pyobj, steal_gdobj=True):
if pyobj is None:
return ffi.NULL
elif isinstance(pyobj, bool):
return godot_bool_alloc(1 if pyobj else 0)
elif isinstance(pyobj, int):
return godot_int_alloc(pyobj)
elif isinstance(pyobj, float):
return godot_real_alloc(pyobj)
elif isinstance(pyobj, str):
gdobj = godot_string_alloc(initialized=False)
lib.godot_string_new_with_wide_string(gdobj, pyobj, -1)
return gdobj
elif isinstance(pyobj, BaseBuiltinWithGDObjOwnership):
if steal_gdobj:
return pyobj._gd_ptr
else:
return pyobj._copy_gdobj(pyobj._gd_ptr)
elif isinstance(pyobj, BaseBuiltin):
return pyobj._gd_ptr
elif isinstance(pyobj, BaseObject):
# TODO: copy ptr box ?
return pyobj._gd_ptr
else:
raise TypeError("Cannot convert Python object `%s` into Godot object." % pyobj)
def new_uninitialized_gdobj(gdtype):
# TODO: use dict to optimize this ?
# It seems Godot encode Variant as type nil...
if gdtype == lib.GODOT_VARIANT_TYPE_NIL:
return godot_variant_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_BOOL:
return godot_bool_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_INT:
return godot_int_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_REAL:
return godot_real_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_STRING:
return godot_string_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_VECTOR2:
return godot_vector2_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_RECT2:
return godot_rect2_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_VECTOR3:
return godot_vector3_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_TRANSFORM2D:
return godot_transform2d_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_PLANE:
return godot_plane_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_QUAT:
return godot_quat_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_AABB:
return godot_aabb_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_BASIS:
return godot_basis_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_TRANSFORM:
return godot_transform_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_COLOR:
return godot_color_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_NODE_PATH:
return godot_node_path_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_RID:
return godot_rid_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_OBJECT:
return godot_object_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_DICTIONARY:
return godot_dictionary_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_ARRAY:
return godot_array_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_POOL_BYTE_ARRAY:
return godot_pool_byte_array_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_POOL_INT_ARRAY:
return godot_pool_int_array_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_POOL_REAL_ARRAY:
return godot_pool_real_array_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_POOL_STRING_ARRAY:
return godot_pool_string_array_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_POOL_VECTOR2_ARRAY:
return godot_pool_vector2_array_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_POOL_VECTOR3_ARRAY:
return godot_pool_vector3_array_alloc()
elif gdtype == lib.GODOT_VARIANT_TYPE_POOL_COLOR_ARRAY:
return godot_pool_color_array_alloc()
else:
raise TypeError(
"Unknown Variant type `%s` (this should never happen !)" % gdtype)
def get_parent_class(cls, classname):
ret = godot_string_alloc()
lib.godot_string_new(ret)
gd_classname = godot_string_from_pyobj(classname)
args = ffi.new("godot_string**", gd_classname)
lib.godot_method_bind_ptrcall(cls._meth_get_parent_class,
cls._instance, ffi.cast("void**",
args), ret)
raw_str = lib.godot_string_wide_str(ret)
return ffi.string(raw_str)
def godot_string_from_pyobj(pystr):
gdstr = godot_string_alloc(initialized=False)
lib.godot_string_new_with_wide_string(gdstr, pystr, len(pystr))
return gdstr
def pyobj_to_variant(pyobj, p_gdvar=None, for_ffi_return=False):
"""
`initialized=False` means we MUST manually init this by hand no matter what
`for_ffi_return=True` means the returned variant won't have it destructor called
once it is garbage collected by python.
This is typically what we want when we pass the variant object by copy as a
return value to Godot (which is then in charge of calling the destructor itself).
"""
if not p_gdvar:
if for_ffi_return:
p_gdvar = ffi.new("godot_variant*")
else:
p_gdvar = godot_variant_alloc(initialized=False)
try:
if pyobj is None:
lib.godot_variant_new_nil(p_gdvar)
elif (isinstance(pyobj, bool)):
lib.godot_variant_new_bool(p_gdvar, pyobj)
elif (isinstance(pyobj, int)):
lib.godot_variant_new_int(p_gdvar, pyobj)
elif (isinstance(pyobj, float)):
lib.godot_variant_new_real(p_gdvar, pyobj)
elif (isinstance(pyobj, str)):
gdstr = godot_string_alloc(initialized=False)
lib.godot_string_new_with_wide_string(gdstr, pyobj, len(pyobj))
lib.godot_variant_new_string(p_gdvar, gdstr)
elif isinstance(pyobj, BaseBuiltin):
if pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_VECTOR2:
lib.godot_variant_new_vector2(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_RECT2:
lib.godot_variant_new_rect2(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_VECTOR3:
lib.godot_variant_new_vector3(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_TRANSFORM2D:
lib.godot_variant_new_transform2d(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_PLANE:
lib.godot_variant_new_plane(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_QUAT:
lib.godot_variant_new_quat(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_AABB:
lib.godot_variant_new_aabb(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_BASIS:
lib.godot_variant_new_basis(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_TRANSFORM:
lib.godot_variant_new_transform(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_COLOR:
lib.godot_variant_new_color(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_NODE_PATH:
lib.godot_variant_new_node_path(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_RID:
lib.godot_variant_new_rid(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_OBJECT:
lib.godot_variant_new_object(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_DICTIONARY:
lib.godot_variant_new_dictionary(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_ARRAY:
lib.godot_variant_new_array(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_POOL_BYTE_ARRAY:
lib.godot_variant_new_pool_byte_array(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_POOL_INT_ARRAY:
lib.godot_variant_new_pool_int_array(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_POOL_REAL_ARRAY:
lib.godot_variant_new_pool_real_array(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_POOL_STRING_ARRAY:
lib.godot_variant_new_pool_string_array(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_POOL_VECTOR2_ARRAY:
lib.godot_variant_new_pool_vector2_array(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_POOL_VECTOR3_ARRAY:
lib.godot_variant_new_pool_vector3_array(p_gdvar, pyobj._gd_ptr)
elif pyobj.GD_TYPE == lib.GODOT_VARIANT_TYPE_POOL_COLOR_ARRAY:
lib.godot_variant_new_pool_color_array(p_gdvar, pyobj._gd_ptr)
elif isinstance(pyobj, BaseObject):
lib.godot_variant_new_object(p_gdvar, pyobj._gd_ptr)
else:
raise TypeError("Cannot convert `%s` to Godot's Variant" % pyobj)
except BaseException:
# Must init the variant anyway to avoid segfault in it destructor
lib.godot_variant_new_nil(p_gdvar)
raise
return p_gdvar
