summaryrefslogtreecommitdiff
path: root/python/examples/kaitai/jpeg.py
diff options
context:
space:
mode:
authorXusheng <xusheng@vector35.com>2021-07-01 12:34:30 +0800
committerXusheng Li <xushengli@protonmail.com>2021-07-05 10:11:47 +0800
commit13ff200ba134b8704f37eca99c42b70dab0d17dc (patch)
tree876b3db3c53f704b31152aeaede1f2edc9d27b24 /python/examples/kaitai/jpeg.py
parent4039a800349dcd587833d3e27d9137e4649818f4 (diff)
Remove snippet and kaitai plugin. Note their individual links.
Diffstat (limited to 'python/examples/kaitai/jpeg.py')
-rw-r--r--python/examples/kaitai/jpeg.py339
1 files changed, 0 insertions, 339 deletions
diff --git a/python/examples/kaitai/jpeg.py b/python/examples/kaitai/jpeg.py
deleted file mode 100644
index 54c28bb6..00000000
--- a/python/examples/kaitai/jpeg.py
+++ /dev/null
@@ -1,339 +0,0 @@
-# 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))
-
-from .exif import Exif
-class Jpeg(KaitaiStruct):
-
- class ComponentId(Enum):
- y = 1
- cb = 2
- cr = 3
- i = 4
- q = 5
- SEQ_FIELDS = ["segments"]
- 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['segments']['start'] = self._io.pos()
- self.segments = []
- i = 0
- while not self._io.is_eof():
- if not 'arr' in self._debug['segments']:
- self._debug['segments']['arr'] = []
- self._debug['segments']['arr'].append({'start': self._io.pos()})
- _t_segments = self._root.Segment(self._io, self, self._root)
- _t_segments._read()
- self.segments.append(_t_segments)
- self._debug['segments']['arr'][len(self.segments) - 1]['end'] = self._io.pos()
- i += 1
-
- self._debug['segments']['end'] = self._io.pos()
-
- class Segment(KaitaiStruct):
-
- class MarkerEnum(Enum):
- tem = 1
- sof0 = 192
- sof1 = 193
- sof2 = 194
- sof3 = 195
- dht = 196
- sof5 = 197
- sof6 = 198
- sof7 = 199
- soi = 216
- eoi = 217
- sos = 218
- dqt = 219
- dnl = 220
- dri = 221
- dhp = 222
- app0 = 224
- app1 = 225
- app2 = 226
- app3 = 227
- app4 = 228
- app5 = 229
- app6 = 230
- app7 = 231
- app8 = 232
- app9 = 233
- app10 = 234
- app11 = 235
- app12 = 236
- app13 = 237
- app14 = 238
- app15 = 239
- com = 254
- SEQ_FIELDS = ["magic", "marker", "length", "data", "image_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['magic']['start'] = self._io.pos()
- self.magic = self._io.ensure_fixed_contents(b"\xFF")
- self._debug['magic']['end'] = self._io.pos()
- self._debug['marker']['start'] = self._io.pos()
- self.marker = KaitaiStream.resolve_enum(self._root.Segment.MarkerEnum, self._io.read_u1())
- self._debug['marker']['end'] = self._io.pos()
- if ((self.marker != self._root.Segment.MarkerEnum.soi) and (self.marker != self._root.Segment.MarkerEnum.eoi)) :
- self._debug['length']['start'] = self._io.pos()
- self.length = self._io.read_u2be()
- self._debug['length']['end'] = self._io.pos()
-
- if ((self.marker != self._root.Segment.MarkerEnum.soi) and (self.marker != self._root.Segment.MarkerEnum.eoi)) :
- self._debug['data']['start'] = self._io.pos()
- _on = self.marker
- if _on == self._root.Segment.MarkerEnum.app1:
- self._raw_data = self._io.read_bytes((self.length - 2))
- io = KaitaiStream(BytesIO(self._raw_data))
- self.data = self._root.SegmentApp1(io, self, self._root)
- self.data._read()
- elif _on == self._root.Segment.MarkerEnum.app0:
- self._raw_data = self._io.read_bytes((self.length - 2))
- io = KaitaiStream(BytesIO(self._raw_data))
- self.data = self._root.SegmentApp0(io, self, self._root)
- self.data._read()
- elif _on == self._root.Segment.MarkerEnum.sof0:
- self._raw_data = self._io.read_bytes((self.length - 2))
- io = KaitaiStream(BytesIO(self._raw_data))
- self.data = self._root.SegmentSof0(io, self, self._root)
- self.data._read()
- elif _on == self._root.Segment.MarkerEnum.sos:
- self._raw_data = self._io.read_bytes((self.length - 2))
- io = KaitaiStream(BytesIO(self._raw_data))
- self.data = self._root.SegmentSos(io, self, self._root)
- self.data._read()
- else:
- self.data = self._io.read_bytes((self.length - 2))
- self._debug['data']['end'] = self._io.pos()
-
- if self.marker == self._root.Segment.MarkerEnum.sos:
- self._debug['image_data']['start'] = self._io.pos()
- self.image_data = self._io.read_bytes_full()
- self._debug['image_data']['end'] = self._io.pos()
-
-
-
- class SegmentSos(KaitaiStruct):
- SEQ_FIELDS = ["num_components", "components", "start_spectral_selection", "end_spectral", "appr_bit_pos"]
- 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['num_components']['start'] = self._io.pos()
- self.num_components = self._io.read_u1()
- self._debug['num_components']['end'] = self._io.pos()
- self._debug['components']['start'] = self._io.pos()
- self.components = [None] * (self.num_components)
- for i in range(self.num_components):
- if not 'arr' in self._debug['components']:
- self._debug['components']['arr'] = []
- self._debug['components']['arr'].append({'start': self._io.pos()})
- _t_components = self._root.SegmentSos.Component(self._io, self, self._root)
- _t_components._read()
- self.components[i] = _t_components
- self._debug['components']['arr'][i]['end'] = self._io.pos()
-
- self._debug['components']['end'] = self._io.pos()
- self._debug['start_spectral_selection']['start'] = self._io.pos()
- self.start_spectral_selection = self._io.read_u1()
- self._debug['start_spectral_selection']['end'] = self._io.pos()
- self._debug['end_spectral']['start'] = self._io.pos()
- self.end_spectral = self._io.read_u1()
- self._debug['end_spectral']['end'] = self._io.pos()
- self._debug['appr_bit_pos']['start'] = self._io.pos()
- self.appr_bit_pos = self._io.read_u1()
- self._debug['appr_bit_pos']['end'] = self._io.pos()
-
- class Component(KaitaiStruct):
- SEQ_FIELDS = ["id", "huffman_table"]
- 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['id']['start'] = self._io.pos()
- self.id = KaitaiStream.resolve_enum(self._root.ComponentId, self._io.read_u1())
- self._debug['id']['end'] = self._io.pos()
- self._debug['huffman_table']['start'] = self._io.pos()
- self.huffman_table = self._io.read_u1()
- self._debug['huffman_table']['end'] = self._io.pos()
-
-
-
- class SegmentApp1(KaitaiStruct):
- SEQ_FIELDS = ["magic", "body"]
- 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.read_bytes_term(0, False, True, True)).decode(u"ASCII")
- self._debug['magic']['end'] = self._io.pos()
- self._debug['body']['start'] = self._io.pos()
- _on = self.magic
- if _on == u"Exif":
- self.body = self._root.ExifInJpeg(self._io, self, self._root)
- self.body._read()
- self._debug['body']['end'] = self._io.pos()
-
-
- class SegmentSof0(KaitaiStruct):
- SEQ_FIELDS = ["bits_per_sample", "image_height", "image_width", "num_components", "components"]
- 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['bits_per_sample']['start'] = self._io.pos()
- self.bits_per_sample = self._io.read_u1()
- self._debug['bits_per_sample']['end'] = self._io.pos()
- self._debug['image_height']['start'] = self._io.pos()
- self.image_height = self._io.read_u2be()
- self._debug['image_height']['end'] = self._io.pos()
- self._debug['image_width']['start'] = self._io.pos()
- self.image_width = self._io.read_u2be()
- self._debug['image_width']['end'] = self._io.pos()
- self._debug['num_components']['start'] = self._io.pos()
- self.num_components = self._io.read_u1()
- self._debug['num_components']['end'] = self._io.pos()
- self._debug['components']['start'] = self._io.pos()
- self.components = [None] * (self.num_components)
- for i in range(self.num_components):
- if not 'arr' in self._debug['components']:
- self._debug['components']['arr'] = []
- self._debug['components']['arr'].append({'start': self._io.pos()})
- _t_components = self._root.SegmentSof0.Component(self._io, self, self._root)
- _t_components._read()
- self.components[i] = _t_components
- self._debug['components']['arr'][i]['end'] = self._io.pos()
-
- self._debug['components']['end'] = self._io.pos()
-
- class Component(KaitaiStruct):
- SEQ_FIELDS = ["id", "sampling_factors", "quantization_table_id"]
- 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['id']['start'] = self._io.pos()
- self.id = KaitaiStream.resolve_enum(self._root.ComponentId, self._io.read_u1())
- self._debug['id']['end'] = self._io.pos()
- self._debug['sampling_factors']['start'] = self._io.pos()
- self.sampling_factors = self._io.read_u1()
- self._debug['sampling_factors']['end'] = self._io.pos()
- self._debug['quantization_table_id']['start'] = self._io.pos()
- self.quantization_table_id = self._io.read_u1()
- self._debug['quantization_table_id']['end'] = self._io.pos()
-
- @property
- def sampling_x(self):
- if hasattr(self, '_m_sampling_x'):
- return self._m_sampling_x if hasattr(self, '_m_sampling_x') else None
-
- self._m_sampling_x = ((self.sampling_factors & 240) >> 4)
- return self._m_sampling_x if hasattr(self, '_m_sampling_x') else None
-
- @property
- def sampling_y(self):
- if hasattr(self, '_m_sampling_y'):
- return self._m_sampling_y if hasattr(self, '_m_sampling_y') else None
-
- self._m_sampling_y = (self.sampling_factors & 15)
- return self._m_sampling_y if hasattr(self, '_m_sampling_y') else None
-
-
-
- class ExifInJpeg(KaitaiStruct):
- SEQ_FIELDS = ["extra_zero", "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['extra_zero']['start'] = self._io.pos()
- self.extra_zero = self._io.ensure_fixed_contents(b"\x00")
- self._debug['extra_zero']['end'] = self._io.pos()
- self._debug['data']['start'] = self._io.pos()
- self._raw_data = self._io.read_bytes_full()
- io = KaitaiStream(BytesIO(self._raw_data))
- self.data = Exif(io)
- self.data._read()
- self._debug['data']['end'] = self._io.pos()
-
-
- class SegmentApp0(KaitaiStruct):
-
- class DensityUnit(Enum):
- no_units = 0
- pixels_per_inch = 1
- pixels_per_cm = 2
- SEQ_FIELDS = ["magic", "version_major", "version_minor", "density_units", "density_x", "density_y", "thumbnail_x", "thumbnail_y", "thumbnail"]
- 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.read_bytes(5)).decode(u"ASCII")
- self._debug['magic']['end'] = self._io.pos()
- self._debug['version_major']['start'] = self._io.pos()
- self.version_major = self._io.read_u1()
- self._debug['version_major']['end'] = self._io.pos()
- self._debug['version_minor']['start'] = self._io.pos()
- self.version_minor = self._io.read_u1()
- self._debug['version_minor']['end'] = self._io.pos()
- self._debug['density_units']['start'] = self._io.pos()
- self.density_units = KaitaiStream.resolve_enum(self._root.SegmentApp0.DensityUnit, self._io.read_u1())
- self._debug['density_units']['end'] = self._io.pos()
- self._debug['density_x']['start'] = self._io.pos()
- self.density_x = self._io.read_u2be()
- self._debug['density_x']['end'] = self._io.pos()
- self._debug['density_y']['start'] = self._io.pos()
- self.density_y = self._io.read_u2be()
- self._debug['density_y']['end'] = self._io.pos()
- self._debug['thumbnail_x']['start'] = self._io.pos()
- self.thumbnail_x = self._io.read_u1()
- self._debug['thumbnail_x']['end'] = self._io.pos()
- self._debug['thumbnail_y']['start'] = self._io.pos()
- self.thumbnail_y = self._io.read_u1()
- self._debug['thumbnail_y']['end'] = self._io.pos()
- self._debug['thumbnail']['start'] = self._io.pos()
- self.thumbnail = self._io.read_bytes(((self.thumbnail_x * self.thumbnail_y) * 3))
- self._debug['thumbnail']['end'] = self._io.pos()
-
-
-