From cfbee2bdcaa445bc2daa9851c2b259a1149c4d2c Mon Sep 17 00:00:00 2001
From: Willem Melching <willem.melching@gmail.com>
Date: Tue, 21 Sep 2021 10:14:16 -0700
Subject: [PATCH] Script to extract thumbnails from segment (#22294)

* Script to extract thumnails from segment

* fix name

* whitespace
old-commit-hash: 8a836b7e1eb65b4a59a5be25f8b7dce9e2bedd0b
---
 selfdrive/camerad/test/.gitignore             |  1 +
 .../test/get_thumbnails_for_segment.py        | 28 +++++++++++++++++++
 2 files changed, 29 insertions(+)
 create mode 100644 selfdrive/camerad/test/.gitignore
 create mode 100755 selfdrive/camerad/test/get_thumbnails_for_segment.py

diff --git a/selfdrive/camerad/test/.gitignore b/selfdrive/camerad/test/.gitignore
new file mode 100644
index 0000000000..44cd0b2730
--- /dev/null
+++ b/selfdrive/camerad/test/.gitignore
@@ -0,0 +1 @@
+jpegs/
diff --git a/selfdrive/camerad/test/get_thumbnails_for_segment.py b/selfdrive/camerad/test/get_thumbnails_for_segment.py
new file mode 100755
index 0000000000..2ffb4bf650
--- /dev/null
+++ b/selfdrive/camerad/test/get_thumbnails_for_segment.py
@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+
+import os
+
+from tqdm import tqdm
+
+from common.file_helpers import mkdirs_exists_ok
+from tools.lib.logreader import LogReader
+from tools.lib.route import Route
+
+import argparse
+
+if __name__ == "__main__":
+  parser = argparse.ArgumentParser()
+  parser.add_argument("route", help="The route name")
+  parser.add_argument("segment", type=int,  help="The index of the segment")
+  args = parser.parse_args()
+
+  out_path = os.path.join("jpegs", f"{args.route.replace('|', '_')}_{args.segment}")
+  mkdirs_exists_ok(out_path)
+
+  r = Route(args.route)
+  lr = list(LogReader(r.qlog_paths()[args.segment]))
+
+  for msg in tqdm(lr):
+      if msg.which() == 'thumbnail':
+          with open(os.path.join(out_path, f"{msg.thumbnail.frameId}.jpg"), 'wb') as f:
+              f.write(msg.thumbnail.thumbnail)