Allow FFT writing to be disabled.

IQTLabs · May 13, 2024 · c997abc · c997abc
1 parent d03d2ee
commit c997abc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/gamutrf/grscan.py b/gamutrf/grscan.py
@@ -92,6 +92,7 @@ def __init__(
         use_external_heading=False,
         vkfft=False,
         wavelearner=None,
+        write_fft_points=False,
         write_samples=0,
     ):
         gr.top_block.__init__(self, "scan", catch_exceptions=True)
@@ -186,7 +187,8 @@ def __init__(
         self.samples_blocks = []
         self.write_samples_block = None
         if write_samples:
-            fft_dir = sample_dir
+            if write_fft_points:
+                fft_dir = sample_dir
             Path(sample_dir).mkdir(parents=True, exist_ok=True)
             self.samples_blocks.extend(
                 [

diff --git a/gamutrf/scan.py b/gamutrf/scan.py
@@ -118,7 +118,14 @@ def argument_parser():
         dest="write_samples",
         default=0,
         type=int,
-        help="if > 0, write FFT/raw samples to --sample_dir",
+        help="if > 0, write FFT points/raw samples to --sample_dir",
+    )
+    parser.add_argument(
+        "--write_fft_points",
+        dest="write_fft_points",
+        default=False,
+        action=BooleanOptionalAction,
+        help="Write FFT points to --sample_dir if write_samples > 0",
     )
     parser.add_argument(
         "--nfft",