Initial commit: gather scripts from different hosts

plot_wf.py

@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+
+import sys
+
+from datetime import datetime
+
+import numpy as np
+import matplotlib.pyplot as pp
+from matplotlib.dates import MinuteLocator, HourLocator, DateFormatter, drange, date2num
+from matplotlib.ticker import MultipleLocator
+
+accumulation = sys.argv[1]
+resolution = int(sys.argv[2])
+logname = sys.argv[3]
+outname = sys.argv[4]
+
+print(f"Accumulation: {accumulation}")
+print(f"Resolution: {resolution} seconds")
+print(f"Input Log: {logname}")
+print(f"Output Image: {outname}")
+
+accumulate = {
+        "avg": np.mean,
+        "peak": np.max,
+    }[accumulation]
+
+data = -1000 * np.ones((86400//resolution, 1024), dtype=float)
+counts = np.zeros(86400//resolution, dtype=int)
+
+basetime = None
+last_lineidx = 0
+accumulated_data = []
+
+with open(logname, 'r') as logfile:
+    for line in logfile:
+        parts = line.split(";", maxsplit=1)
+        timestamp = float(parts[0])
+
+        if not basetime:
+            basetime = np.floor(timestamp / 86400) * 86400
+
+        lineidx = int((timestamp - basetime) // resolution)
+        if lineidx != last_lineidx:
+            if len(accumulated_data) != 0:
+                # apply the accumulation function
+                data[last_lineidx] = accumulate(np.array(accumulated_data).astype(float), 0)
+
+            accumulated_data = []
+            last_lineidx = lineidx
+
+        tmp_data = np.fromstring(parts[1], sep=";", dtype=int)
+        if tmp_data.size == 1024:
+            accumulated_data.append(tmp_data)
+        else:
+            print(f"Warning: line ignored due to wrong number of elements: has {tmp_data.size}, expected 1024.")
+
+x = np.linspace(0, 30, data.shape[1])
+y = np.linspace(basetime, basetime + resolution * data.shape[0])
+
+extent = [
+        0, # MHz
+        30, # MHz
+        date2num(datetime.utcfromtimestamp(basetime + resolution * data.shape[0])),
+        date2num(datetime.utcfromtimestamp(basetime))]
+
+fig_x_scale = 0.85
+fig_y_scale = 0.90
+fig_x_off   = 0.05
+fig_y_off   = 0.05
+
+dpi = pp.rcParams['figure.dpi'] #get the default dpi value
+fig_size = (data.shape[1]/dpi/fig_x_scale, data.shape[0]/dpi/fig_y_scale) # convert pixels to DPI
+
+fix, ax = pp.subplots(figsize=fig_size)
+
+image = ax.imshow(data, vmin=-100, vmax=0, cmap='inferno',
+                  extent=extent)
+
+ax.set_position(pos=[fig_x_off, fig_y_off, fig_x_scale, fig_y_scale])
+ax.set_title('Spektrogramm vom ' + datetime.fromtimestamp(basetime).strftime('%Y-%m-%d'))
+
+xrange = extent[1] - extent[0]
+yrange = extent[2] - extent[3]
+
+ax.set_aspect((data.shape[0] / yrange) / (data.shape[1] / xrange))
+
+ax.yaxis_date()
+
+ax.yaxis.set_major_locator(HourLocator(range(0, 25, 1)))
+ax.yaxis.set_minor_locator(MinuteLocator(range(0, 60, 20)))
+ax.yaxis.set_major_formatter(DateFormatter('%H:%M'))
+ax.yaxis.set_minor_formatter(DateFormatter('%M'))
+
+ax.xaxis.set_major_locator(MultipleLocator(1.0))
+ax.set_xlabel('Frequenz [MHz]')
+
+ax_top = ax.twiny()
+ax_top.xaxis.set_major_locator(MultipleLocator(1.0))
+ax_top.set_xlabel('Frequenz [MHz]')
+ax_top.set_xlim(ax.get_xlim())
+ax_top.set_position(pos=[fig_x_off, fig_y_off, fig_x_scale, fig_y_scale])
+
+cax = pp.axes([fig_x_scale+fig_x_off+0.02, fig_y_off, 0.03, fig_y_scale])
+pp.colorbar(cax=cax, mappable=image)
+cax.set_title('dBm')
+
+pp.savefig(outname)