From f5774f921481ed8aff42956f05c8b9537c52b12e Mon Sep 17 00:00:00 2001
From: devdesk <tami@tami.org.il>
Date: Mon, 25 Dec 2023 23:09:03 +0200
Subject: [PATCH] initial

---
 decode.py        | 103 +++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  37 +++++++++++++++++
 2 files changed, 140 insertions(+)
 create mode 100644 decode.py
 create mode 100644 requirements.txt

diff --git a/decode.py b/decode.py
new file mode 100644
index 0000000..96a1cc1
--- /dev/null
+++ b/decode.py
@@ -0,0 +1,103 @@
+# coding: utf-8
+
+from os import system
+import numpy as np
+from tqdm import tqdm
+import pandas as pd
+import pcapng
+from struct  import unpack
+import matplotlib.pyplot as plt
+from PIL import Image
+from PIL import Image, ImageDraw
+import cv2
+
+
+pd.options.display.max_colwidth = 150
+
+scanner = pcapng.scanner.FileScanner(open('in.pcap', 'rb'))
+blocks = list(tqdm(scanner))
+
+
+def tryget(obj, att):
+    if hasattr(obj, att):
+        return getattr(obj, att)
+    return None
+    
+
+df = pd.DataFrame([{'index': i, 'length': tryget(obj, 'packet_len')} for i, obj in enumerate(blocks)])
+# get udp packets
+data = [blocks[i] for i in df[df.length == 6972.0].index]
+# remove udp header
+raw = [d.packet_data[0x2a:] for d in data]
+
+
+def parse(data):
+    hdr = 4 + 2 * 7
+    c1, c2, part, a, ffaa, b, c, d = unpack('>Lhhhhhhh', data[:hdr])
+    ret = locals()
+    del ret['data']
+    del ret['hdr']
+    ret['data'] = data[hdr:]
+    return ret
+    
+df = pd.DataFrame([parse(d) for d in raw])
+df2 = df[[c for c in df.columns if c != 'data']]
+
+
+def getframes(df):
+    frames = []
+    current = []
+    for i, row in df.iterrows():
+        if row.part == 0:
+            # roll over
+            if len(current) > 0:
+                frames.append(current)
+            current = []
+        current.append(row['data'])
+    if len(current) > 0:
+        frames.append(current)
+    return [b''.join(parts) for parts in frames]
+
+
+def image16(frame, width, height, pixelformat='>H'):
+    return [Image.fromarray(np.frombuffer(frame, dtype=pixelformat).reshape(width, height)) for frame in frames if len(frame) == 2 * width * height]
+
+
+def showvideo(images):
+    # wip
+    videodims = images[0].size
+    fourcc = cv2.VideoWriter_fourcc(*'avc1')    
+    video = cv2.VideoWriter("test.mp4",fourcc, 60,videodims)
+    img = Image.new('RGB', videodims, color = 'darkred')
+    #draw stuff that goes on every frame here
+    for i in range(0,60*60):
+        imtemp = img.copy()
+        # draw frame specific stuff here.
+        video.write(cv2.cvtColor(np.array(imtemp), cv2.COLOR_RGB2BGR))
+    video.release()
+
+#def shape(frames, width, height, mode='RGB'):
+#    return [Image.frombytes('RGB', (width, height), frame) for frame in frames if len(frame) == width * height * 3]
+
+
+frames = getframes(df)
+images = image16(frames, 384, 288)
+for i, img in enumerate(tqdm(images)):
+    img.save(f'{i:04}.png')
+#frame = b''.join(df.iloc[:32]['data'])
+#Image.frombytes('RGB', (288, 256), frame).show()
+start_len = len(b'T=(-1.665884e-08)*X^4+(1.347094e-05)*X^3+(-4.396264e-03)*X^2+(9.506939e-01)*X+(-6.353247e+01)\r\n                                                                                                                                                           ')
+# 250 bytes at start of frame
+
+equations = {x[:250].decode().strip() for x in df[df.part == 0]['data']}
+# seen only a single equation on all packets
+assert len(equations) == 1
+
+assert list(equations)[0] == 'T=(-1.665884e-08)*X^4+(1.347094e-05)*X^3+(-4.396264e-03)*X^2+(9.506939e-01)*X+(-6.353247e+01)'
+
+# we have 6372 packets
+assert df.shape[0] == 6372
+
+# produce a video
+system('ffmpeg -f image2 -framerate 25 -i %04d.png -s 288x384 thermal.avi')
+print('to play: ffplay thermal.avi')
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..de5d789
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,37 @@
+aenum==3.1.15
+asttokens==2.4.1
+chardet==5.2.0
+contourpy==1.2.0
+cycler==0.12.1
+decorator==5.1.1
+dictdumper==0.8.4.post2
+executing==2.0.1
+fonttools==4.47.0
+ipython==8.19.0
+jedi==0.19.1
+kiwisolver==1.4.5
+matplotlib==3.8.2
+matplotlib-inline==0.1.6
+numpy==1.26.2
+opencv-python==4.8.1.78
+packaging==23.2
+pandas==2.1.4
+parso==0.8.3
+pexpect==4.9.0
+Pillow==10.1.0
+prompt-toolkit==3.0.43
+ptyprocess==0.7.0
+pure-eval==0.2.2
+Pygments==2.17.2
+pyparsing==3.1.1
+pypcapkit==1.3.1
+python-dateutil==2.8.2
+python-pcapng==2.1.1
+pytz==2023.3.post1
+six==1.16.0
+stack-data==0.6.3
+tbtrim==0.3.1
+tqdm==4.66.1
+traitlets==5.14.0
+tzdata==2023.3
+wcwidth==0.2.12