Add methods for serialization

Author: UserUnknownFactor

kaitaistruct.py

@@ -118,6 +118,9 @@ def size(self):
     packer_u4le = struct.Struct('<I')
     packer_u8le = struct.Struct('<Q')
 
+    # ------------------------------------------------------------------------
+    # Reading
+    # ------------------------------------------------------------------------
     # ------------------------------------------------------------------------
     # Signed
     # ------------------------------------------------------------------------
@@ -213,6 +216,95 @@ def read_f4le(self):
     def read_f8le(self):
         return KaitaiStream.packer_f8le.unpack(self.read_bytes(8))[0]
 
+    # ------------------------------------------------------------------------
+    # Writing
+    # ------------------------------------------------------------------------
+    # ------------------------------------------------------------------------
+    # Signed
+    # ------------------------------------------------------------------------
+
+    def write_s1(self, data):
+        return self.write_bytes(KaitaiStream.packer_s1.pack(data))
+
+    # ........................................................................
+    # Big-endian
+    # ........................................................................
+
+    def write_s2be(self, data):
+        return self.write_bytes(KaitaiStream.packer_s2be.pack(data))
+
+    def write_s4be(self, data):
+        return self.write_bytes(KaitaiStream.packer_s4be.pack(data))
+
+    def write_s8be(self, data):
+        return self.write_bytes(KaitaiStream.packer_s8be.pack(data))
+
+    # ........................................................................
+    # Little-endian
+    # ........................................................................
+
+    def write_s2le(self, data):
+        return self.write_bytes(KaitaiStream.packer_s2le.pack(data))
+
+    def write_s4le(self, data):
+        return self.write_bytes(KaitaiStream.packer_s4le.pack(data))
+
+    def write_s8le(self, data):
+        return self.write_bytes(KaitaiStream.packer_s8le.pack(data))
+
+    # ------------------------------------------------------------------------
+    # Unsigned
+    # ------------------------------------------------------------------------
+
+    def write_u1(self, data):
+        return self.write_bytes(KaitaiStream.packer_u1.pack(data))
+
+    # ........................................................................
+    # Big-endian
+    # ........................................................................
+
+    def write_u2be(self, data):
+        return self.write_bytes(KaitaiStream.packer_u2be.pack(data))
+
+    def write_u4be(self, data):
+        return self.write_bytes(KaitaiStream.packer_u4be.pack(data))
+
+    def write_u8be(self, data):
+        return self.write_bytes(KaitaiStream.packer_u8be.pack(data))
+
+    # ........................................................................
+    # Little-endian
+    # ........................................................................
+
+    def write_u2le(self, data):
+        return self.write_bytes(KaitaiStream.packer_u2le.pack(data))
+
+    def write_u4le(self, data):
+        return self.write_bytes(KaitaiStream.packer_u4le.pack(data))
+
+    def write_u8le(self, data):
+        return self.write_bytes(KaitaiStream.packer_u8le.pack(data))
+
+    # ........................................................................
+    # Big-endian
+    # ........................................................................
+
+    def write_f4be(self, data):
+        return self.write_bytes(KaitaiStream.packer_f4be.pack(data))
+
+    def write_f8be(self, data):
+        return self.write_bytes(KaitaiStream.packer_f8be.pack(data))
+
+    # ........................................................................
+    # Little-endian
+    # ........................................................................
+
+    def write_f4le(self, data):
+        return self.write_bytes(KaitaiStream.packer_f4le.pack(data))
+
+    def write_f8le(self, data):
+        return self.write_bytes(KaitaiStream.packer_f8le.pack(data))
+
     # ========================================================================
     # Unaligned bit values
     # ========================================================================
@@ -293,6 +385,21 @@ def read_bytes(self, n):
             )
         return r
 
+    def write_bytes(self, data, pad=0, padding=b'\0'):
+        if data is None:
+            return
+        nb = len(data)
+        if nb == 0 and pad < 1:
+            return
+        if nb > 0 and self._io.write(data) != nb:
+            raise Exception(
+                "not all of %d bytes written" %
+                (nb,)
+            )
+        if pad > 0 and padding is not None:
+            self._io.write(padding * pad)
+        return
+
     def read_bytes_full(self):
         return self._io.read()