kaitai: adjustments and rearrangements for python2/3 compat

1 files changed, 291 insertions, 0 deletions

diff --git a/python/examples/kaitai/avi.py b/python/examples/kaitai/avi.py
new file mode 100644
index 00000000..872a4f97
--- /dev/null
+++ b/python/examples/kaitai/avi.py
@@ -0,0 +1,291 @@
+# This is a generated file! Please edit source .ksy file and use kaitai-struct-compiler to rebuild
+
+from pkg_resources import parse_version
+from .kaitaistruct import __version__ as ks_version, KaitaiStruct, KaitaiStream, BytesIO
+from enum import Enum
+import collections
+
+
+if parse_version(ks_version) < parse_version('0.7'):
+    raise Exception("Incompatible Kaitai Struct Python API: 0.7 or later is required, but you have %s" % (ks_version))
+
+class Avi(KaitaiStruct):
+    """
+    .. seealso::
+       Source - https://msdn.microsoft.com/en-us/library/ms779636.aspx
+    """
+
+    class ChunkType(Enum):
+        idx1 = 829973609
+        junk = 1263424842
+        info = 1330007625
+        isft = 1413894985
+        list = 1414744396
+        strf = 1718776947
+        avih = 1751742049
+        strh = 1752331379
+        movi = 1769369453
+        hdrl = 1819436136
+        strl = 1819440243
+
+    class StreamType(Enum):
+        mids = 1935960429
+        vids = 1935960438
+        auds = 1935963489
+        txts = 1937012852
+
+    class HandlerType(Enum):
+        mp3 = 85
+        ac3 = 8192
+        dts = 8193
+        cvid = 1684633187
+        xvid = 1684633208
+    SEQ_FIELDS = ["magic1", "file_size", "magic2", "data"]
+    def __init__(self, _io, _parent=None, _root=None):
+        self._io = _io
+        self._parent = _parent
+        self._root = _root if _root else self
+        self._debug = collections.defaultdict(dict)
+
+    def _read(self):
+        self._debug['magic1']['start'] = self._io.pos()
+        self.magic1 = self._io.ensure_fixed_contents(b"\x52\x49\x46\x46")
+        self._debug['magic1']['end'] = self._io.pos()
+        self._debug['file_size']['start'] = self._io.pos()
+        self.file_size = self._io.read_u4le()
+        self._debug['file_size']['end'] = self._io.pos()
+        self._debug['magic2']['start'] = self._io.pos()
+        self.magic2 = self._io.ensure_fixed_contents(b"\x41\x56\x49\x20")
+        self._debug['magic2']['end'] = self._io.pos()
+        self._debug['data']['start'] = self._io.pos()
+        self._raw_data = self._io.read_bytes((self.file_size - 4))
+        io = KaitaiStream(BytesIO(self._raw_data))
+        self.data = self._root.Blocks(io, self, self._root)
+        self.data._read()
+        self._debug['data']['end'] = self._io.pos()
+
+    class ListBody(KaitaiStruct):
+        SEQ_FIELDS = ["list_type", "data"]
+        def __init__(self, _io, _parent=None, _root=None):
+            self._io = _io
+            self._parent = _parent
+            self._root = _root if _root else self
+            self._debug = collections.defaultdict(dict)
+
+        def _read(self):
+            self._debug['list_type']['start'] = self._io.pos()
+            self.list_type = KaitaiStream.resolve_enum(self._root.ChunkType, self._io.read_u4le())
+            self._debug['list_type']['end'] = self._io.pos()
+            self._debug['data']['start'] = self._io.pos()
+            self.data = self._root.Blocks(self._io, self, self._root)
+            self.data._read()
+            self._debug['data']['end'] = self._io.pos()
+
+
+    class Rect(KaitaiStruct):
+        SEQ_FIELDS = ["left", "top", "right", "bottom"]
+        def __init__(self, _io, _parent=None, _root=None):
+            self._io = _io
+            self._parent = _parent
+            self._root = _root if _root else self
+            self._debug = collections.defaultdict(dict)
+
+        def _read(self):
+            self._debug['left']['start'] = self._io.pos()
+            self.left = self._io.read_s2le()
+            self._debug['left']['end'] = self._io.pos()
+            self._debug['top']['start'] = self._io.pos()
+            self.top = self._io.read_s2le()
+            self._debug['top']['end'] = self._io.pos()
+            self._debug['right']['start'] = self._io.pos()
+            self.right = self._io.read_s2le()
+            self._debug['right']['end'] = self._io.pos()
+            self._debug['bottom']['start'] = self._io.pos()
+            self.bottom = self._io.read_s2le()
+            self._debug['bottom']['end'] = self._io.pos()
+
+
+    class Blocks(KaitaiStruct):
+        SEQ_FIELDS = ["entries"]
+        def __init__(self, _io, _parent=None, _root=None):
+            self._io = _io
+            self._parent = _parent
+            self._root = _root if _root else self
+            self._debug = collections.defaultdict(dict)
+
+        def _read(self):
+            self._debug['entries']['start'] = self._io.pos()
+            self.entries = []
+            i = 0
+            while not self._io.is_eof():
+                if not 'arr' in self._debug['entries']:
+                    self._debug['entries']['arr'] = []
+                self._debug['entries']['arr'].append({'start': self._io.pos()})
+                _t_entries = self._root.Block(self._io, self, self._root)
+                _t_entries._read()
+                self.entries.append(_t_entries)
+                self._debug['entries']['arr'][len(self.entries) - 1]['end'] = self._io.pos()
+                i += 1
+
+            self._debug['entries']['end'] = self._io.pos()
+
+
+    class AvihBody(KaitaiStruct):
+        """Main header of an AVI file, defined as AVIMAINHEADER structure.
+        
+        .. seealso::
+           Source - https://msdn.microsoft.com/en-us/library/ms779632.aspx
+        """
+        SEQ_FIELDS = ["micro_sec_per_frame", "max_bytes_per_sec", "padding_granularity", "flags", "total_frames", "initial_frames", "streams", "suggested_buffer_size", "width", "height", "reserved"]
+        def __init__(self, _io, _parent=None, _root=None):
+            self._io = _io
+            self._parent = _parent
+            self._root = _root if _root else self
+            self._debug = collections.defaultdict(dict)
+
+        def _read(self):
+            self._debug['micro_sec_per_frame']['start'] = self._io.pos()
+            self.micro_sec_per_frame = self._io.read_u4le()
+            self._debug['micro_sec_per_frame']['end'] = self._io.pos()
+            self._debug['max_bytes_per_sec']['start'] = self._io.pos()
+            self.max_bytes_per_sec = self._io.read_u4le()
+            self._debug['max_bytes_per_sec']['end'] = self._io.pos()
+            self._debug['padding_granularity']['start'] = self._io.pos()
+            self.padding_granularity = self._io.read_u4le()
+            self._debug['padding_granularity']['end'] = self._io.pos()
+            self._debug['flags']['start'] = self._io.pos()
+            self.flags = self._io.read_u4le()
+            self._debug['flags']['end'] = self._io.pos()
+            self._debug['total_frames']['start'] = self._io.pos()
+            self.total_frames = self._io.read_u4le()
+            self._debug['total_frames']['end'] = self._io.pos()
+            self._debug['initial_frames']['start'] = self._io.pos()
+            self.initial_frames = self._io.read_u4le()
+            self._debug['initial_frames']['end'] = self._io.pos()
+            self._debug['streams']['start'] = self._io.pos()
+            self.streams = self._io.read_u4le()
+            self._debug['streams']['end'] = self._io.pos()
+            self._debug['suggested_buffer_size']['start'] = self._io.pos()
+            self.suggested_buffer_size = self._io.read_u4le()
+            self._debug['suggested_buffer_size']['end'] = self._io.pos()
+            self._debug['width']['start'] = self._io.pos()
+            self.width = self._io.read_u4le()
+            self._debug['width']['end'] = self._io.pos()
+            self._debug['height']['start'] = self._io.pos()
+            self.height = self._io.read_u4le()
+            self._debug['height']['end'] = self._io.pos()
+            self._debug['reserved']['start'] = self._io.pos()
+            self.reserved = self._io.read_bytes(16)
+            self._debug['reserved']['end'] = self._io.pos()
+
+
+    class Block(KaitaiStruct):
+        SEQ_FIELDS = ["four_cc", "block_size", "data"]
+        def __init__(self, _io, _parent=None, _root=None):
+            self._io = _io
+            self._parent = _parent
+            self._root = _root if _root else self
+            self._debug = collections.defaultdict(dict)
+
+        def _read(self):
+            self._debug['four_cc']['start'] = self._io.pos()
+            self.four_cc = KaitaiStream.resolve_enum(self._root.ChunkType, self._io.read_u4le())
+            self._debug['four_cc']['end'] = self._io.pos()
+            self._debug['block_size']['start'] = self._io.pos()
+            self.block_size = self._io.read_u4le()
+            self._debug['block_size']['end'] = self._io.pos()
+            self._debug['data']['start'] = self._io.pos()
+            _on = self.four_cc
+            if _on == self._root.ChunkType.list:
+                self._raw_data = self._io.read_bytes(self.block_size)
+                io = KaitaiStream(BytesIO(self._raw_data))
+                self.data = self._root.ListBody(io, self, self._root)
+                self.data._read()
+            elif _on == self._root.ChunkType.avih:
+                self._raw_data = self._io.read_bytes(self.block_size)
+                io = KaitaiStream(BytesIO(self._raw_data))
+                self.data = self._root.AvihBody(io, self, self._root)
+                self.data._read()
+            elif _on == self._root.ChunkType.strh:
+                self._raw_data = self._io.read_bytes(self.block_size)
+                io = KaitaiStream(BytesIO(self._raw_data))
+                self.data = self._root.StrhBody(io, self, self._root)
+                self.data._read()
+            else:
+                self.data = self._io.read_bytes(self.block_size)
+            self._debug['data']['end'] = self._io.pos()
+
+
+    class StrhBody(KaitaiStruct):
+        """Stream header (one header per stream), defined as AVISTREAMHEADER structure.
+        
+        .. seealso::
+           Source - https://msdn.microsoft.com/en-us/library/ms779638.aspx
+        """
+        SEQ_FIELDS = ["fcc_type", "fcc_handler", "flags", "priority", "language", "initial_frames", "scale", "rate", "start", "length", "suggested_buffer_size", "quality", "sample_size", "frame"]
+        def __init__(self, _io, _parent=None, _root=None):
+            self._io = _io
+            self._parent = _parent
+            self._root = _root if _root else self
+            self._debug = collections.defaultdict(dict)
+
+        def _read(self):
+            self._debug['fcc_type']['start'] = self._io.pos()
+            self.fcc_type = KaitaiStream.resolve_enum(self._root.StreamType, self._io.read_u4le())
+            self._debug['fcc_type']['end'] = self._io.pos()
+            self._debug['fcc_handler']['start'] = self._io.pos()
+            self.fcc_handler = KaitaiStream.resolve_enum(self._root.HandlerType, self._io.read_u4le())
+            self._debug['fcc_handler']['end'] = self._io.pos()
+            self._debug['flags']['start'] = self._io.pos()
+            self.flags = self._io.read_u4le()
+            self._debug['flags']['end'] = self._io.pos()
+            self._debug['priority']['start'] = self._io.pos()
+            self.priority = self._io.read_u2le()
+            self._debug['priority']['end'] = self._io.pos()
+            self._debug['language']['start'] = self._io.pos()
+            self.language = self._io.read_u2le()
+            self._debug['language']['end'] = self._io.pos()
+            self._debug['initial_frames']['start'] = self._io.pos()
+            self.initial_frames = self._io.read_u4le()
+            self._debug['initial_frames']['end'] = self._io.pos()
+            self._debug['scale']['start'] = self._io.pos()
+            self.scale = self._io.read_u4le()
+            self._debug['scale']['end'] = self._io.pos()
+            self._debug['rate']['start'] = self._io.pos()
+            self.rate = self._io.read_u4le()
+            self._debug['rate']['end'] = self._io.pos()
+            self._debug['start']['start'] = self._io.pos()
+            self.start = self._io.read_u4le()
+            self._debug['start']['end'] = self._io.pos()
+            self._debug['length']['start'] = self._io.pos()
+            self.length = self._io.read_u4le()
+            self._debug['length']['end'] = self._io.pos()
+            self._debug['suggested_buffer_size']['start'] = self._io.pos()
+            self.suggested_buffer_size = self._io.read_u4le()
+            self._debug['suggested_buffer_size']['end'] = self._io.pos()
+            self._debug['quality']['start'] = self._io.pos()
+            self.quality = self._io.read_u4le()
+            self._debug['quality']['end'] = self._io.pos()
+            self._debug['sample_size']['start'] = self._io.pos()
+            self.sample_size = self._io.read_u4le()
+            self._debug['sample_size']['end'] = self._io.pos()
+            self._debug['frame']['start'] = self._io.pos()
+            self.frame = self._root.Rect(self._io, self, self._root)
+            self.frame._read()
+            self._debug['frame']['end'] = self._io.pos()
+
+
+    class StrfBody(KaitaiStruct):
+        """Stream format description."""
+        SEQ_FIELDS = []
+        def __init__(self, _io, _parent=None, _root=None):
+            self._io = _io
+            self._parent = _parent
+            self._root = _root if _root else self
+            self._debug = collections.defaultdict(dict)
+
+        def _read(self):
+            pass
+
+
+