feat(engine): add tagged union info to enum reflection

1 files changed, 21 insertions, 2 deletions

diff --git a/engine-reflection/src/lib.rs b/engine-reflection/src/lib.rs
index 4e5e8d8..3bd6f97 100644
--- a/engine-reflection/src/lib.rs
+++ b/engine-reflection/src/lib.rs
@@ -119,8 +119,7 @@ pub struct Enum
     /// The enum only contains unit variants.
     pub is_unit_only: bool,
 
-    pub discriminant_layout: Layout,
-    pub fields_layout: Option<Layout>,
+    pub tagged_union: Option<EnumTaggedUnion>,
 
     pub get_default_value: fn() -> Option<DefaultValueFn>,
 }
@@ -141,6 +140,15 @@ impl Enum
 }
 
 #[derive(Debug, Clone)]
+pub struct EnumTaggedUnion
+{
+    pub discriminant_layout: Layout,
+    pub discriminant_byte_offset: usize,
+    pub fields_layout: Layout,
+    pub fields_byte_offset: usize,
+}
+
+#[derive(Debug, Clone)]
 pub struct EnumVariant
 {
     pub name: &'static str,
@@ -164,6 +172,17 @@ pub enum EnumVariantFields
     },
 }
 
+impl EnumVariantFields
+{
+    pub fn fields(&self) -> &'static [Field]
+    {
+        match self {
+            Self::Named { fields } => fields,
+            Self::Unnamed { fields } => fields,
+        }
+    }
+}
+
 /// The discriminant of a enum variant.
 #[derive(Debug, Clone, PartialEq, Eq, Hash)]
 pub struct EnumDiscriminant