diff options
Diffstat (limited to 'src/infiray_irg.py')
| -rw-r--r-- | src/infiray_irg.py | 23 |
1 files changed, 18 insertions, 5 deletions
diff --git a/src/infiray_irg.py b/src/infiray_irg.py index 9d573f6..80d04b3 100644 --- a/src/infiray_irg.py +++ b/src/infiray_irg.py @@ -8,11 +8,14 @@ def load(data): nonlocal data out, data = data[:n], data[n:] if len(out) < n: - raise ValueError('file is truncated') + raise ValueError(f'File is truncated, tried to read {n} bytes, but only {len(out)} bytes remain.') return out header = consume(128) - if header[:2] != bytes([0xca, 0xac]) or header[-2:] != bytes([0xac, 0xca]): + if header[:2] not in [ + bytes([0xca, 0xac]), + bytes([0xba, 0xab]) + ] or header[-2:] != bytes([0xac, 0xca]): raise ValueError('Header magic not found.') _unk0, coarse_section_length, y_res, x_res,\ @@ -28,9 +31,19 @@ def load(data): raise ValueError('Resolution mismatch in header') coarse_img = np.frombuffer(consume(coarse_section_length), dtype=np.uint8).reshape((y_res, x_res)) - fine_img = np.frombuffer(consume(x_res*y_res*2), dtype=np.int16).reshape((y_res, x_res)) - fine_img = (fine_img / 16) - 273 - vis_jpg = Image.open(io.BytesIO(consume(jpeg_length))) + if header[:2] == bytes([0xca, 0xac]): + # 1/16th Kelvin steps + fine_img = np.frombuffer(consume(x_res*y_res*2), dtype=np.uint16).reshape((y_res, x_res)) + fine_img = (fine_img / 16) - 273 + + vis_jpg = Image.open(io.BytesIO(consume(jpeg_length))) + + else: # 0xbaac variant + # 0.1 Kelvin steps + fine_img = np.frombuffer(consume(x_res*y_res*2), dtype=np.uint16).reshape((y_res, x_res)) + fine_img = fine_img / 10 - 273 + + vis_jpg = Image.open(io.BytesIO(data)) return coarse_img, fine_img, vis_jpg |