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,45 @@ def read_bytes(self, n):
             )
         return r
 
+    def write_bytes(self, data):
+        if data is None:
+            return
+        nb = len(data)
+        if nb > 0 and self._io.write(data) != nb:
+            raise ValidationFailedError(
+                "not all of %d bytes written" %
+                (nb,), self._io, __file__
+            )
+
+    def write_terminator(self, term_byte):
+        self._io.write(KaitaiStream.packer_u1.pack(term_byte))
+
+    def write_padding(self, actual_size, size, pad_byte):
+        pad = size - actual_size
+        if pad > 0:
+            if pad_byte is not None:
+                self._io.write(KaitaiStream.packer_u1.pack(pad_byte) * pad)
+            else:
+                raise KaitaiStructError("no padding filler provided", __file__)
+
+    def write_bytes_term(self, data, size, term_byte, pad_byte):
+        if data is None or not size:
+            return
+        nb = len(data)
+        if nb < size:
+            if nb > 0 and self._io.write(data) != nb:
+                raise ValidationFailedError(
+                    "not all of %d bytes written" %
+                    (nb,), self._io, __file__
+                )
+        else:
+            raise ValidationFailedError(
+                "Writing %d bytes, but %d bytes (including terminator) were given" %
+                (size, nb + 1), self._io, __file__
+            )
+        self.write_terminator(term_byte)
+        self.write_padding(nb, size - 1, pad_byte)
+
     def read_bytes_full(self):
         return self._io.read()
 
@@ -302,7 +433,7 @@ def read_bytes_term(self, term, include_term, consume_term, eos_error):
             c = self._io.read(1)
             if c == b'':
                 if eos_error:
-                    raise Exception(
+                    raise EOFError(
                         "end of stream reached, but no terminator %d found" %
                         (term,)
                     )
@@ -320,9 +451,9 @@ def read_bytes_term(self, term, include_term, consume_term, eos_error):
     def ensure_fixed_contents(self, expected):
         actual = self._io.read(len(expected))
         if actual != expected:
-            raise Exception(
+            raise ValidationFailedError(
                 "unexpected fixed contents: got %r, was waiting for %r" %
-                (actual, expected)
+                (actual, expected), self._io, __file__
             )
         return actual
 
@@ -375,9 +506,9 @@ def process_xor_many(data, key):
     @staticmethod
     def process_rotate_left(data, amount, group_size):
         if group_size != 1:
-            raise Exception(
+            raise KaitaiStructError(
                 "unable to rotate group of %d bytes yet" %
-                (group_size,)
+                (group_size,), __file__
             )
 
         mask = group_size * 8 - 1