Added support functions to get the underlying storage container of st…

…ructures
BlockProject3D · Aug 24, 2024 · cf2877d · cf2877d
1 parent 68d7086
commit cf2877d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/compiler/Cargo.toml b/compiler/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "bp3d-protoc"
-version = "1.0.0-rc.2.0.0"
+version = "1.0.0-rc.2.1.0"
 edition = "2021"
 authors = ["Yuri Edward <yuri6037@outlook.com>"]
 description = "BP3D protocol compiler/generator."

diff --git a/compiler/src/gen/rust/structure.template b/compiler/src/gen/rust/structure.template
@@ -3,6 +3,10 @@ impl<T> {name}<T> {{
     pub fn new(data: T) -> Self {{
         Self {{ data }}
     }}
+
+    pub fn into_inner(self) -> T {{
+        self.data
+    }}
 }}
 
 impl {name}<[u8; {byte_size}]> {{
@@ -29,6 +33,18 @@ impl<'a, T: AsMut<[u8]>> {name}<T> {{
     }}
 }}
 
+impl<T: AsRef<[u8]>> AsRef<[u8]> for {name}<T> {{
+    fn as_ref(&self) -> &[u8] {{
+        self.data.as_ref()
+    }}
+}}
+
+impl<T: AsMut<[u8]>> AsMut<[u8]> for {name}<T> {{
+    fn as_mut(&mut self) -> &mut [u8] {{
+        self.data.as_mut()
+    }}
+}}
+
 pub const SIZE_{name:to_screaming_case}: usize = {byte_size};
 #fragment pop