update armature types

1 files changed, 68 insertions, 0 deletions

diff --git a/shared/src/helper.rs b/shared/src/helper.rs
index 723f91e..14fb2bd 100644
--- a/shared/src/helper.rs
+++ b/shared/src/helper.rs
@@ -220,6 +220,63 @@ impl ReadWrite for Vec<f32> {
             .collect())
     }
 }
+impl ReadWrite for Vec<u16> {
+    fn write(&self, w: &mut dyn Write) -> Result<()> {
+        for e in self {
+            e.write(w)?;
+        }
+        Ok(())
+    }
+    fn read(r: &mut dyn Read) -> Result<Self> {
+        let mut buf = Vec::new();
+        r.read_to_end(&mut buf)?;
+        Ok(buf
+            .into_iter()
+            .array_chunks::<{ size_of::<u16>() }>()
+            .map(u16::from_be_bytes)
+            .collect())
+    }
+}
+impl ReadWrite for Vec<Affine3A> {
+    fn write(&self, w: &mut dyn Write) -> Result<()> {
+        for e in self {
+            e.write(w)?;
+        }
+        Ok(())
+    }
+    fn read(r: &mut dyn Read) -> Result<Self> {
+        let mut buf = Vec::new();
+        r.read_to_end(&mut buf)?;
+        Ok(buf
+            .into_iter()
+            .array_chunks::<{ size_of::<f32>() }>()
+            .map(f32::from_be_bytes)
+            .array_chunks::<12>()
+            .map(|m| Affine3A::from_cols_array(&m))
+            .collect())
+    }
+}
+impl ReadWrite for Vec<String> {
+    fn write(&self, w: &mut dyn Write) -> Result<()> {
+        for e in self {
+            w.write_all(&(e.len() as u16).to_le_bytes())?;
+            w.write_all(e.as_bytes())?;
+        }
+        Ok(())
+    }
+    fn read(r: &mut dyn Read) -> Result<Self> {
+        let mut buf = Vec::new();
+        r.read_to_end(&mut buf)?;
+        let mut buf = buf.as_slice();
+        let mut out = Vec::new();
+        while !buf.is_empty() {
+            let size = u16::read(&mut buf)? as usize;
+            out.push(String::read(&mut &buf[..size])?);
+            buf = &buf[size..];
+        }
+        Ok(out)
+    }
+}
 impl ReadWrite for String {
     fn write(&self, w: &mut dyn Write) -> Result<()> {
         w.write_all(self.as_bytes())?;
@@ -364,3 +421,14 @@ impl ReadWrite for u8 {
         Ok(buf[0])
     }
 }
+impl ReadWrite for u16 {
+    fn write(&self, w: &mut dyn Write) -> Result<()> {
+        w.write_all(&self.to_be_bytes())?;
+        Ok(())
+    }
+    fn read(r: &mut dyn Read) -> Result<Self> {
+        let mut buf = [0u8; 2];
+        r.read_exact(&mut buf)?;
+        Ok(u16::from_be_bytes(buf))
+    }
+}