From f82a35702cb85da7ace298960024b773a2da6d85 Mon Sep 17 00:00:00 2001 From: HampusM Date: Wed, 17 Jun 2026 15:29:22 +0200 Subject: feat(engine): add tagged union info to enum reflection --- engine-macros/src/reflection/enum_impl.rs | 25 +++++++++++-------------- 1 file changed, 11 insertions(+), 14 deletions(-) (limited to 'engine-macros/src/reflection/enum_impl.rs') diff --git a/engine-macros/src/reflection/enum_impl.rs b/engine-macros/src/reflection/enum_impl.rs index 611da6d..5dbb176 100644 --- a/engine-macros/src/reflection/enum_impl.rs +++ b/engine-macros/src/reflection/enum_impl.rs @@ -219,23 +219,21 @@ pub fn generate(input: syn::ItemEnum) -> proc_macro2::TokenStream &engine_crate_path, ); - let discriminant_layout = if is_unit_only { - quote! { - std::alloc::Layout::new::() - } - } else { - quote! { - std::alloc::Layout::new::<#mod_name::Discriminant>() - } - }; - - let fields_layout = if is_unit_only { + let tagged_union = if is_unit_only { quote! { None } } else { quote! { - Some(std::alloc::Layout::new::<#mod_name::Fields>()) + Some(#engine_crate_path::reflection::EnumTaggedUnion { + discriminant_layout: std::alloc::Layout::new::<#mod_name::Discriminant>(), + discriminant_byte_offset: + std::mem::offset_of!(#mod_name::Equivalent #type_generics, tag), + fields_layout: + std::alloc::Layout::new::<#mod_name::Fields #type_generics>(), + fields_byte_offset: + std::mem::offset_of!(#mod_name::Equivalent #type_generics, payload), + }) } }; @@ -256,8 +254,7 @@ pub fn generate(input: syn::ItemEnum) -> proc_macro2::TokenStream #engine_crate_path::reflection::Enum { variants: &[#(#variants),*], is_unit_only: #is_unit_only, - discriminant_layout: #discriminant_layout, - fields_layout: #fields_layout, + tagged_union: #tagged_union, get_default_value: || { #get_default_value_fn } -- cgit v1.2.3-18-g5258