kaitai: adjustments and rearrangements for python2/3 compat

1 files changed, 170 insertions, 0 deletions

diff --git a/python/examples/kaitai/pcx.py b/python/examples/kaitai/pcx.py
new file mode 100644
index 00000000..2b6123d5
--- /dev/null
+++ b/python/examples/kaitai/pcx.py
@@ -0,0 +1,170 @@
+# 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 Pcx(KaitaiStruct):
+
+    class Versions(Enum):
+        v2_5 = 0
+        v2_8_with_palette = 2
+        v2_8_without_palette = 3
+        paintbrush_for_windows = 4
+        v3_0 = 5
+
+    class Encodings(Enum):
+        rle = 1
+    SEQ_FIELDS = ["hdr"]
+    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['hdr']['start'] = self._io.pos()
+        self._raw_hdr = self._io.read_bytes(128)
+        io = KaitaiStream(BytesIO(self._raw_hdr))
+        self.hdr = self._root.Header(io, self, self._root)
+        self.hdr._read()
+        self._debug['hdr']['end'] = self._io.pos()
+
+    class Header(KaitaiStruct):
+        """
+        .. seealso::
+           - "ZSoft .PCX FILE HEADER FORMAT" - http://web.archive.org/web/20100206055706/http://www.qzx.com/pc-gpe/pcx.txt
+        """
+        SEQ_FIELDS = ["magic", "version", "encoding", "bits_per_pixel", "img_x_min", "img_y_min", "img_x_max", "img_y_max", "hdpi", "vdpi", "palette_16", "reserved", "num_planes", "bytes_per_line", "palette_info", "h_screen_size", "v_screen_size"]
+        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['magic']['start'] = self._io.pos()
+            self.magic = self._io.ensure_fixed_contents(b"\x0A")
+            self._debug['magic']['end'] = self._io.pos()
+            self._debug['version']['start'] = self._io.pos()
+            self.version = KaitaiStream.resolve_enum(self._root.Versions, self._io.read_u1())
+            self._debug['version']['end'] = self._io.pos()
+            self._debug['encoding']['start'] = self._io.pos()
+            self.encoding = KaitaiStream.resolve_enum(self._root.Encodings, self._io.read_u1())
+            self._debug['encoding']['end'] = self._io.pos()
+            self._debug['bits_per_pixel']['start'] = self._io.pos()
+            self.bits_per_pixel = self._io.read_u1()
+            self._debug['bits_per_pixel']['end'] = self._io.pos()
+            self._debug['img_x_min']['start'] = self._io.pos()
+            self.img_x_min = self._io.read_u2le()
+            self._debug['img_x_min']['end'] = self._io.pos()
+            self._debug['img_y_min']['start'] = self._io.pos()
+            self.img_y_min = self._io.read_u2le()
+            self._debug['img_y_min']['end'] = self._io.pos()
+            self._debug['img_x_max']['start'] = self._io.pos()
+            self.img_x_max = self._io.read_u2le()
+            self._debug['img_x_max']['end'] = self._io.pos()
+            self._debug['img_y_max']['start'] = self._io.pos()
+            self.img_y_max = self._io.read_u2le()
+            self._debug['img_y_max']['end'] = self._io.pos()
+            self._debug['hdpi']['start'] = self._io.pos()
+            self.hdpi = self._io.read_u2le()
+            self._debug['hdpi']['end'] = self._io.pos()
+            self._debug['vdpi']['start'] = self._io.pos()
+            self.vdpi = self._io.read_u2le()
+            self._debug['vdpi']['end'] = self._io.pos()
+            self._debug['palette_16']['start'] = self._io.pos()
+            self.palette_16 = self._io.read_bytes(48)
+            self._debug['palette_16']['end'] = self._io.pos()
+            self._debug['reserved']['start'] = self._io.pos()
+            self.reserved = self._io.ensure_fixed_contents(b"\x00")
+            self._debug['reserved']['end'] = self._io.pos()
+            self._debug['num_planes']['start'] = self._io.pos()
+            self.num_planes = self._io.read_u1()
+            self._debug['num_planes']['end'] = self._io.pos()
+            self._debug['bytes_per_line']['start'] = self._io.pos()
+            self.bytes_per_line = self._io.read_u2le()
+            self._debug['bytes_per_line']['end'] = self._io.pos()
+            self._debug['palette_info']['start'] = self._io.pos()
+            self.palette_info = self._io.read_u2le()
+            self._debug['palette_info']['end'] = self._io.pos()
+            self._debug['h_screen_size']['start'] = self._io.pos()
+            self.h_screen_size = self._io.read_u2le()
+            self._debug['h_screen_size']['end'] = self._io.pos()
+            self._debug['v_screen_size']['start'] = self._io.pos()
+            self.v_screen_size = self._io.read_u2le()
+            self._debug['v_screen_size']['end'] = self._io.pos()
+
+
+    class TPalette256(KaitaiStruct):
+        SEQ_FIELDS = ["magic", "colors"]
+        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['magic']['start'] = self._io.pos()
+            self.magic = self._io.ensure_fixed_contents(b"\x0C")
+            self._debug['magic']['end'] = self._io.pos()
+            self._debug['colors']['start'] = self._io.pos()
+            self.colors = [None] * (256)
+            for i in range(256):
+                if not 'arr' in self._debug['colors']:
+                    self._debug['colors']['arr'] = []
+                self._debug['colors']['arr'].append({'start': self._io.pos()})
+                _t_colors = self._root.Rgb(self._io, self, self._root)
+                _t_colors._read()
+                self.colors[i] = _t_colors
+                self._debug['colors']['arr'][i]['end'] = self._io.pos()
+
+            self._debug['colors']['end'] = self._io.pos()
+
+
+    class Rgb(KaitaiStruct):
+        SEQ_FIELDS = ["r", "g", "b"]
+        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['r']['start'] = self._io.pos()
+            self.r = self._io.read_u1()
+            self._debug['r']['end'] = self._io.pos()
+            self._debug['g']['start'] = self._io.pos()
+            self.g = self._io.read_u1()
+            self._debug['g']['end'] = self._io.pos()
+            self._debug['b']['start'] = self._io.pos()
+            self.b = self._io.read_u1()
+            self._debug['b']['end'] = self._io.pos()
+
+
+    @property
+    def palette_256(self):
+        """
+        .. seealso::
+           - "VGA 256 Color Palette Information" - http://web.archive.org/web/20100206055706/http://www.qzx.com/pc-gpe/pcx.txt
+        """
+        if hasattr(self, '_m_palette_256'):
+            return self._m_palette_256 if hasattr(self, '_m_palette_256') else None
+
+        if  ((self.hdr.version == self._root.Versions.v3_0) and (self.hdr.bits_per_pixel == 8) and (self.hdr.num_planes == 1)) :
+            _pos = self._io.pos()
+            self._io.seek((self._io.size() - 769))
+            self._debug['_m_palette_256']['start'] = self._io.pos()
+            self._m_palette_256 = self._root.TPalette256(self._io, self, self._root)
+            self._m_palette_256._read()
+            self._debug['_m_palette_256']['end'] = self._io.pos()
+            self._io.seek(_pos)
+
+        return self._m_palette_256 if hasattr(self, '_m_palette_256') else None
+
+