Merge pull request #1303 from anarkiwi/classify

Fully class-ify WaterfallPlot
IQTLabs · Jun 7, 2024 · 346b2a0 · 346b2a0
2 parents d5dc55d + d176d03
commit 346b2a0
Show file tree

Hide file tree

Showing 2 changed files with 655 additions and 649 deletions.
diff --git a/gamutrfwaterfall/gamutrfwaterfall/waterfall.py b/gamutrfwaterfall/gamutrfwaterfall/waterfall.py
@@ -90,6 +90,8 @@ def sig_handler(_sig=None, _frame=None):
     if not scan_configs:
         return
 
+    plot = None
+
     while zmqr.healthy() and running:
         if need_reconfig:
             config = make_config(
@@ -117,7 +119,7 @@ def sig_handler(_sig=None, _frame=None):
                 config.fft_len,
                 config.freq_resolution,
             )
-            plot = WaterfallPlot(config, base_save_path, peak_finder)
+            plot = WaterfallPlot(config, base_save_path, peak_finder, 1)
             results = [
                 (scan_configs, frame_resample(scan_df, config.freq_resolution * 1e6))
             ]