1 files changed, 7 insertions, 0 deletions
diff --git a/ebml_derive/src/lib.rs b/ebml_derive/src/lib.rs
index cd2d194..34510b9 100644
--- a/ebml_derive/src/lib.rs
+++ b/ebml_derive/src/lib.rs
@@ -81,6 +81,10 @@ pub fn define_ebml(ts: TokenStream) -> TokenStream {
         .iter()
         .map(|Tag { name, .. }| quote! { Self::#name(v) => v.write_to(w) })
         .collect::<Vec<_>>();
+    let size_match = tags
+        .iter()
+        .map(|Tag { name, .. }| quote! { Self::#name(v) => v.size() })
+        .collect::<Vec<_>>();
     let cons_master_match = tags
         .iter()
         .filter_map(
@@ -131,6 +135,9 @@ pub fn define_ebml(ts: TokenStream) -> TokenStream {
             pub fn write(&self, w: &mut Vec<u8>) -> crate::Result<()> {
                 match self { #(#write_match),* }
             }
+            pub fn size(&self) -> usize {
+                match self { #(#size_match),* }
+            }
         }
     }
     .into()