more impl

Author: molpopgen

examples/manual_metadata_encoding.rs

@@ -1,3 +1,5 @@
+use core::str;
+
 use tskit::metadata::MetadataRoundtrip;
 
 struct MutationMetadata {
@@ -37,12 +39,28 @@ struct IndividualMetadata {
 
 impl MetadataRoundtrip for IndividualMetadata {
     fn encode(&self) -> Result<Vec<u8>, tskit::metadata::MetadataError> {
-        todo!()
+        let mut rv = vec![];
+        rv.extend_from_slice(self.name.len().to_be_bytes().as_slice());
+        rv.extend_from_slice(self.name.as_bytes());
+        println!("{} {:?}", self.name, self.name.as_bytes());
+        rv.extend_from_slice(self.phenotypes.len().to_be_bytes().as_slice());
+        for i in self.phenotypes.iter() {
+            rv.extend_from_slice(i.to_be_bytes().as_slice());
+        }
+        Ok(rv)
     }
     fn decode(md: &[u8]) -> Result<Self, tskit::metadata::MetadataError>
     where
         Self: Sized,
     {
+        let size: [u8; std::mem::size_of::<usize>()] =
+            md[0..std::mem::size_of::<usize>()].try_into().unwrap();
+        let size = usize::from_be_bytes(size);
+        let name =
+            str::from_utf8(&md[std::mem::size_of::<usize>()..size + std::mem::size_of::<usize>()])
+                .unwrap()
+                .to_string();
+        assert_eq!(name, "Jerome");
         todo!()
     }
 }
@@ -102,7 +120,8 @@ fn test_mutation_metadata_roundtrip() {
 #[test]
 fn test_individual_metadata_roundtrip() {
     let md = IndividualMetadata {
-        name: "Jerome".to_string(), phenotypes: vec![10,9]
+        name: "Jerome".to_string(),
+        phenotypes: vec![10, 9],
     };
     let encoded = md.encode().unwrap();
     let decoded = IndividualMetadata::decode(&encoded).unwrap();