Skip to content

Commit

Permalink
More generated #[func] tests (argument/return value passing between…
Browse files Browse the repository at this point in the history
… GDScript and Rust)
  • Loading branch information
Bromeon committed Mar 22, 2023
1 parent 049f163 commit 7900641
Show file tree
Hide file tree
Showing 7 changed files with 118 additions and 56 deletions.
15 changes: 14 additions & 1 deletion godot-core/src/builtin/array.rs
Original file line number Diff line number Diff line change
Expand Up @@ -855,7 +855,7 @@ macro_rules! varray {
/// [`set_typed`](https://docs.godotengine.org/en/latest/classes/class_array.html#class-array-method-set-typed).
///
/// We ignore the `script` parameter because it has no impact on typing in Godot.
#[derive(Debug, PartialEq, Eq)]
#[derive(PartialEq, Eq)]
struct TypeInfo {
variant_type: VariantType,
class_name: StringName,
Expand All @@ -880,3 +880,16 @@ impl TypeInfo {
self.variant_type != VariantType::Nil
}
}

impl fmt::Debug for TypeInfo {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let class = self.class_name.to_string();
let class_str = if class.is_empty() {
String::new()
} else {
format!(" (class={class})")
};

write!(f, "{:?}{}", self.variant_type, class_str)
}
}
45 changes: 15 additions & 30 deletions godot-core/src/builtin/meta/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -27,13 +27,14 @@ pub trait VariantMetadata {
}

fn property_info(property_name: &str) -> PropertyInfo {
PropertyInfo::new(
Self::variant_type(),
Self::class_name(),
StringName::from(property_name),
global::PropertyHint::PROPERTY_HINT_NONE,
GodotString::new(),
)
PropertyInfo {
variant_type: Self::variant_type(),
class_name: Self::class_name(),
property_name: StringName::from(property_name),
hint: global::PropertyHint::PROPERTY_HINT_NONE,
hint_string: GodotString::new(),
usage: global::PropertyUsageFlags::PROPERTY_USAGE_DEFAULT,
}
}

fn param_metadata() -> sys::GDExtensionClassMethodArgumentMetadata {
Expand All @@ -52,33 +53,17 @@ impl<T: VariantMetadata> VariantMetadata for Option<T> {
/// Rusty abstraction of sys::GDExtensionPropertyInfo
/// Keeps the actual allocated values (the sys equivalent only keeps pointers, which fall out of scope)
#[derive(Debug)]
// Note: is not #[non_exhaustive], so adding fields is a breaking change. Mostly used internally at the moment though.
pub struct PropertyInfo {
variant_type: VariantType,
class_name: ClassName,
property_name: StringName,
hint: global::PropertyHint,
hint_string: GodotString,
usage: global::PropertyUsageFlags,
pub variant_type: VariantType,
pub class_name: ClassName,
pub property_name: StringName,
pub hint: global::PropertyHint,
pub hint_string: GodotString,
pub usage: global::PropertyUsageFlags,
}

impl PropertyInfo {
pub fn new(
variant_type: VariantType,
class_name: ClassName,
property_name: StringName,
hint: global::PropertyHint,
hint_string: GodotString,
) -> Self {
Self {
variant_type,
class_name,
property_name,
hint,
hint_string,
usage: global::PropertyUsageFlags::PROPERTY_USAGE_DEFAULT,
}
}

/// Converts to the FFI type. Keep this object allocated while using that!
pub fn property_sys(&self) -> sys::GDExtensionPropertyInfo {
use crate::obj::EngineEnum as _;
Expand Down
2 changes: 2 additions & 0 deletions godot-core/src/builtin/node_path.rs
Original file line number Diff line number Diff line change
Expand Up @@ -80,5 +80,7 @@ impl_builtin_traits! {
Default => node_path_construct_default;
Clone => node_path_construct_copy;
Drop => node_path_destroy;
Eq => node_path_operator_equal;
// Ord => node_path_operator_less;
}
}
18 changes: 17 additions & 1 deletion godot-core/src/builtin/variant/impls.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,9 @@
*/

use super::*;
use crate::builtin::meta::VariantMetadata;
use crate::builtin::meta::{PropertyInfo, VariantMetadata};
use crate::builtin::*;
use crate::engine::global;
use crate::obj::EngineEnum;
use godot_ffi as sys;
use sys::GodotFfi;
Expand Down Expand Up @@ -224,6 +225,21 @@ impl VariantMetadata for Variant {
// Arrays use the `NIL` type to indicate that they are untyped.
VariantType::Nil
}

fn property_info(property_name: &str) -> PropertyInfo {
PropertyInfo {
variant_type: Self::variant_type(),
class_name: Self::class_name(),
property_name: StringName::from(property_name),
hint: global::PropertyHint::PROPERTY_HINT_NONE,
hint_string: GodotString::new(),
usage: global::PropertyUsageFlags::PROPERTY_USAGE_NIL_IS_VARIANT,
}
}

fn param_metadata() -> sys::GDExtensionClassMethodArgumentMetadata {
sys::GDEXTENSION_METHOD_ARGUMENT_METADATA_INT_IS_INT8
}
}

impl<T: EngineEnum> ToVariant for T {
Expand Down
16 changes: 8 additions & 8 deletions godot-macros/src/derive_godot_class.rs
Original file line number Diff line number Diff line change
Expand Up @@ -377,14 +377,14 @@ fn make_exports_impl(class_name: &Ident, fields: &Fields) -> TokenStream {
use ::godot::builtin::meta::VariantMetadata;

let class_name = ::godot::builtin::StringName::from(#class_name::CLASS_NAME);

let property_info = ::godot::builtin::meta::PropertyInfo::new(
<#field_type>::variant_type(),
::godot::builtin::meta::ClassName::of::<#class_name>(),
::godot::builtin::StringName::from(#field_name),
::godot::engine::global::PropertyHint::#hint_type,
::godot::builtin::GodotString::from(#description),
);
let property_info = ::godot::builtin::meta::PropertyInfo {
variant_type: <#field_type>::variant_type(),
class_name: ::godot::builtin::meta::ClassName::of::<#class_name>(),
property_name: ::godot::builtin::StringName::from(#field_name),
hint: ::godot::engine::global::PropertyHint::#hint_type,
hint_string: ::godot::builtin::GodotString::from(#description),
usage: ::godot::engine::global::PropertyUsageFlags::PROPERTY_USAGE_DEFAULT,
};
let property_info_sys = property_info.property_sys();

let getter_name = ::godot::builtin::StringName::from(#getter_name);
Expand Down
27 changes: 22 additions & 5 deletions itest/godot/input/GenFfiTests.template.gd
Original file line number Diff line number Diff line change
Expand Up @@ -4,14 +4,31 @@

extends TestSuite

# Note: GDScript only uses ptrcalls if it has the full type information available at "compile" (parse) time.
# That includes all arguments (including receiver) as well as function signature (parameters + return type).
# Otherwise, GDScript will use varcall. Both are tested below.
# It is thus important that `ffi` is initialized using = for varcalls, and using := for ptrcalls.

#(
func test_varcall_IDENT():
var ffi = GenFfi.new()

var from_rust = ffi.return_IDENT()
assert_that(ffi.accept_IDENT(from_rust))
var from_rust: Variant = ffi.return_IDENT()
assert_that(ffi.accept_IDENT(from_rust), "ffi.accept_IDENT(from_rust)")

var from_gdscript: Variant = VAL
var mirrored: Variant = ffi.mirror_IDENT(from_gdscript)
assert_eq(mirrored, from_gdscript, "mirrored == from_gdscript")
#)

#(
func test_ptrcall_IDENT():
var ffi := GenFfi.new()

var from_rust: TYPE = ffi.return_IDENT()
assert_that(ffi.accept_IDENT(from_rust), "ffi.accept_IDENT(from_rust)")

var from_gdscript = VAL
var mirrored = ffi.mirror_IDENT(from_gdscript)
assert_eq(mirrored, from_gdscript)
var from_gdscript: TYPE = VAL
var mirrored: TYPE = ffi.mirror_IDENT(from_gdscript)
assert_eq(mirrored, from_gdscript, "mirrored == from_gdscript")
#)
51 changes: 40 additions & 11 deletions itest/rust/build.rs
Original file line number Diff line number Diff line change
Expand Up @@ -11,22 +11,33 @@ use std::path::Path;

type IoResult = std::io::Result<()>;

macro_rules! push {
($inputs:ident; $gdscript_ty:ident, $rust_ty:ty, $val:expr) => {
push!($inputs; $gdscript_ty, $rust_ty, $val, $val);
};

($inputs:ident; $gdscript_ty:ident, $rust_ty:ty, $gdscript_val:expr, $rust_val:expr) => {
/// Push with GDScript expr in string
macro_rules! pushs {
($inputs:ident; $GDScriptTy:expr, $RustTy:ty, $gdscript_val:expr, $rust_val:expr) => {
$inputs.push(Input {
ident: stringify!($rust_ty).to_ascii_lowercase(),
gdscript_ty: stringify!($gdscript_ty),
gdscript_val: stringify!($gdscript_val),
rust_ty: quote! { $rust_ty },
ident: stringify!($RustTy)
.to_ascii_lowercase()
.replace("<", "_")
.replace(">", ""),
gdscript_ty: stringify!($GDScriptTy),
gdscript_val: $gdscript_val,
rust_ty: quote! { $RustTy },
rust_val: quote! { $rust_val },
});
};
}

/// Push simple GDScript expression, outside string
macro_rules! push {
($inputs:ident; $GDScriptTy:expr, $RustTy:ty, $val:expr) => {
push!($inputs; $GDScriptTy, $RustTy, $val, $val);
};

($inputs:ident; $GDScriptTy:expr, $RustTy:ty, $gdscript_val:expr, $rust_val:expr) => {
pushs!($inputs; $GDScriptTy, $RustTy, stringify!($gdscript_val), $rust_val);
};
}

// Edit this to change involved types
fn collect_inputs() -> Vec<Input> {
let mut inputs = vec![];
Expand All @@ -41,15 +52,33 @@ fn collect_inputs() -> Vec<Input> {
push!(inputs; bool, bool, true);
push!(inputs; Color, Color, Color(0.7, 0.5, 0.3, 0.2), Color::from_rgba(0.7, 0.5, 0.3, 0.2));
push!(inputs; String, GodotString, "hello", "hello".into());
push!(inputs; StringName, StringName, &"hello", "hello".into());
pushs!(inputs; NodePath, NodePath, r#"^"hello""#, "hello".into());
push!(inputs; Vector2, Vector2, Vector2(12.5, -3.5), Vector2::new(12.5, -3.5));
push!(inputs; Vector3, Vector3, Vector3(117.5, 100.0, -323.25), Vector3::new(117.5, 100.0, -323.25));
push!(inputs; Vector4, Vector4, Vector4(-18.5, 24.75, -1.25, 777.875), Vector4::new(-18.5, 24.75, -1.25, 777.875));
push!(inputs; Vector2i, Vector2i, Vector2i(-2147483648, 2147483647), Vector2i::new(-2147483648, 2147483647));
push!(inputs; Vector3i, Vector3i, Vector3i(-1, -2147483648, 2147483647), Vector3i::new(-1, -2147483648, 2147483647));
//push!(inputs; Variant, Variant, 123, 123i64.to_variant());

// Data structures
// TODO enable below, when GDScript has typed array literals, or find a hack with eval/lambdas
/*pushs!(inputs; Array[int], Array<i32>,
"(func() -> Array[int]: [-7, 12, 40])()",
array![-7, 12, 40]
);*/

push!(inputs; Array, VariantArray,
[-7, "godot", false, Vector2i(-77, 88)],
varray![-7, "godot", false, Vector2i::new(-77, 88)]);

pushs!(inputs; Dictionary, Dictionary,
r#"{"key": 83, -3: Vector2(1, 2), 0.03: true}"#,
dict! { "key": 83, (-3): Vector2::new(1.0, 2.0), 0.03: true }
);

// Composite
push!(inputs; int, InstanceId, -1, InstanceId::from_nonzero(0xFFFFFFFFFFFFFFF));
push!(inputs; Variant, Variant, 123, 123i64.to_variant());

inputs
}
Expand Down

0 comments on commit 7900641

Please sign in to comment.