tools/hci_trace_to_pcap.py: Support colour input.

jimmo · jimmo · commit 159144630a58 · 2023-09-21T16:39:54.000+10:00
Filter out escape sequences to set the color. Also allow the "TX:" and
"RX:" used in some places.

This work was funded through GitHub Sponsors.

Signed-off-by: Jim Mussared &lt;jim.mussared@gmail.com&gt;
diff --git a/tools/hci_trace_to_pcap.py b/tools/hci_trace_to_pcap.py
@@ -65,8 +65,15 @@
 
 with open(sys.argv[1], "r") as f:
     for line in f:
-        line = line.strip()
-        m = re.match("([<>]) \\[ *([0-9]+)\\] ([A-Fa-f0-9:]+)", line)
+        # Remove color formatting.
+        line = line.strip().encode()
+        if line[0] == ord("\x1b"):
+            line = line[7:]
+        if len(line) >= 4 and line[-4] == ord("\x1b"):
+            line = line[:-4]
+        line = line.decode()
+
+        m = re.match("([<>]) \\[ *([0-9]+)\\] (?:[TR]X: )?([A-Fa-f0-9:]+)", line)
         if not m:
             continue
 
