spectre-core 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

spectre_core/plotting/panel_stack.py

@@ -15,7 +15,9 @@ from spectre_core.plotting.factory import get_panel
 from spectre_core.plotting.library.spectrogram.panel import Panel as SpectrogramPanel
 
 class PanelStack:
-    def __init__(self, time_type: str, figsize: Tuple[int, int] = (10, 10)):
+    def __init__(self, 
+                 time_type: str = "seconds", 
+                 figsize: Tuple[int, int] = (10, 10)):
         self._time_type = time_type
         self._figsize = figsize
         self._panels: List[BasePanel] = []

spectre_core/{watchdog → post_processing}/__init__.py

@@ -3,4 +3,4 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 
 # dynamically import all event handlers
-import spectre_core.watchdog.library
+import spectre_core.post_processing.library

spectre_core/post_processing/base.py

@@ -0,0 +1,132 @@
+# SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
+# This file is part of SPECTRE
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from logging import getLogger
+_LOGGER = getLogger(__name__)
+
+from queue import Queue
+from typing import Optional
+from abc import ABC, abstractmethod
+from math import floor
+
+from watchdog.events import (
+    FileSystemEventHandler, 
+    FileCreatedEvent, 
+)
+
+from spectre_core.chunks.factory import get_chunk_from_tag
+from spectre_core.file_handlers.configs import CaptureConfig
+from spectre_core.spectrograms.spectrogram import Spectrogram
+from spectre_core.spectrograms.transform import join_spectrograms
+from spectre_core.spectrograms.transform import (
+    time_average, 
+    frequency_average
+)
+
+
+class BaseEventHandler(ABC, FileSystemEventHandler):
+    def __init__(self, 
+                 tag: str):
+        self._tag = tag
+
+        self._Chunk = get_chunk_from_tag(tag)
+
+        self._capture_config = CaptureConfig(tag)
+
+        self._watch_extension = self._capture_config.get("watch_extension")
+        if self._watch_extension is None:
+            raise KeyError("The watch extension has not been specified in the capture config")
+
+        # attribute to store the next file to be processed 
+        # (specifically, the absolute file path of the file)
+        self._queued_file: Optional[str] = None
+
+        # spectrogram cache stores spectrograms in memory
+        # such that they can be periodically written to files
+        # according to the joining time.
+        self._spectrogram: Optional[Spectrogram] = None
+        
+
+
+    @abstractmethod
+    def process(self, 
+                absolute_file_path: str) -> None:
+        """Process the file stored at the input absolute file path.
+        
+        To be implemented by derived classes.
+        """
+
+
+    def on_created(self, 
+                   event: FileCreatedEvent):
+        """Process a newly created batch file, only once the next batch is created.
+        
+        Since we assume that the batches are non-overlapping in time, this guarantees
+        we avoid post processing a file while it is being written to. Files are processed
+        sequentially, in the order they are created.
+        """
+
+        # the 'src_path' attribute holds the absolute path of the newly created file
+        absolute_file_path = event.src_path
+        
+        # only 'notice' a file if it ends with the appropriate extension
+        # as defined in the capture config
+        if absolute_file_path.endswith(self._watch_extension):
+            _LOGGER.info(f"Noticed {absolute_file_path}")
+            
+            # If there exists a queued file, try and process it
+            if self._queued_file is not None:
+                try:
+                    self.process(self._queued_file)
+                except Exception:
+                    _LOGGER.error(f"An error has occured while processing {self._queued_file}",
+                                  exc_info=True)
+                     # flush any internally stored spectrogram on error to avoid lost data
+                    self._flush_spectrogram()
+                    # re-raise the exception to the main thread
+                    raise
+            
+            # Queue the current file for processing next
+            _LOGGER.info(f"Queueing {absolute_file_path} for post processing")
+            self._queued_file = absolute_file_path
+
+
+    def _average_in_time(self, 
+                         spectrogram: Spectrogram) -> Spectrogram:
+        _LOGGER.info("Averaging spectrogram in time")
+        requested_time_resolution = self._capture_config['time_resolution'] # [s]
+        if requested_time_resolution is None:
+            raise KeyError(f"Time resolution has not been specified in the capture config")
+        average_over = floor(requested_time_resolution/spectrogram.time_resolution) if requested_time_resolution > spectrogram.time_resolution else 1
+        return time_average(spectrogram, average_over)
+    
+    
+    def _average_in_frequency(self, 
+                              spectrogram: Spectrogram) -> Spectrogram:
+        _LOGGER.info("Averaging spectrogram in frequency")
+        frequency_resolution = self._capture_config['frequency_resolution'] # [Hz]
+        if frequency_resolution is None:
+            raise KeyError(f"Frequency resolution has not been specified in the capture config")
+        average_over = floor(frequency_resolution/spectrogram.frequency_resolution) if frequency_resolution > spectrogram.frequency_resolution else 1
+        return frequency_average(spectrogram, average_over)
+    
+
+    def _join_spectrogram(self, 
+                          spectrogram: Spectrogram) -> None:
+        _LOGGER.info("Joining spectrogram")
+        if self._spectrogram is None:
+            self._spectrogram = spectrogram
+        else:
+            self._spectrogram = join_spectrograms([self._spectrogram, spectrogram])
+
+        if self._spectrogram.time_range >= self._capture_config['joining_time']:
+            self._flush_spectrogram()
+    
+
+    def _flush_spectrogram(self) -> None:
+        if self._spectrogram:
+            _LOGGER.info(f"Flushing spectrogram to file with chunk start time {self._spectrogram.chunk_start_time}")
+            self._spectrogram.save()
+            _LOGGER.info("Flush successful, resetting spectrogram cache")
+            self._spectrogram = None # reset the cache

spectre_core/{watchdog → post_processing}/factory.py

@@ -2,17 +2,17 @@
 # This file is part of SPECTRE
 # SPDX-License-Identifier: GPL-3.0-or-later
 
-from spectre_core.watchdog.event_handler_register import event_handler_map
-from spectre_core.watchdog.base import BaseEventHandler
+from spectre_core.post_processing.event_handler_register import event_handler_map
+from spectre_core.post_processing.base import BaseEventHandler
 from spectre_core.file_handlers.configs import CaptureConfig
 from spectre_core.exceptions import EventHandlerNotFoundError
 
 def get_event_handler(event_handler_key: str) -> BaseEventHandler:
-    # try and fetch the capture config mount
     EventHandler = event_handler_map.get(event_handler_key)
     if EventHandler is None:
         valid_event_handler_keys = list(event_handler_map.keys())
-        raise EventHandlerNotFoundError(f"No event handler found for the event handler key: {event_handler_key}. Please specify one of the following event handler keys {valid_event_handler_keys}")
+        raise EventHandlerNotFoundError((f"No event handler found for the event handler key '{event_handler_key}'. "
+                                         f"Please specify one of the following event handler keys: {valid_event_handler_keys}"))
     return EventHandler
 
 

spectre_core/{watchdog → post_processing}/library/fixed/event_handler.py

@@ -7,8 +7,8 @@ _LOGGER = getLogger(__name__)
 
 import os
 
-from spectre_core.watchdog.base import BaseEventHandler
-from spectre_core.watchdog.event_handler_register import register_event_handler
+from spectre_core.post_processing.base import BaseEventHandler
+from spectre_core.post_processing.event_handler_register import register_event_handler
 
 @register_event_handler("fixed")
 class EventHandler(BaseEventHandler):
@@ -16,20 +16,19 @@ class EventHandler(BaseEventHandler):
         super().__init__(*args, **kwargs)
 
 
-    def process(self, file_path: str):
-        _LOGGER.info(f"Processing: {file_path}")
-        file_name = os.path.basename(file_path)
-        chunk_start_time, _ = os.path.splitext(file_name)[0].split('_')
+    def process(self, 
+                absolute_file_path: str):
+        _LOGGER.info(f"Processing: {absolute_file_path}")
+        file_name = os.path.basename(absolute_file_path)
+        base_file_name, _ = os.path.splitext(file_name)
+        chunk_start_time, _ = base_file_name.split('_')
         chunk = self._Chunk(chunk_start_time, self._tag)
 
         _LOGGER.info("Creating spectrogram")
         spectrogram = chunk.build_spectrogram()
 
-        _LOGGER.info("Averaging spectrogram")
         spectrogram = self._average_in_time(spectrogram)
         spectrogram = self._average_in_frequency(spectrogram)
-
-        _LOGGER.info("Joining spectrogram")
         self._join_spectrogram(spectrogram)
 
         bin_chunk = chunk.get_file('bin')

spectre_core/{watchdog → post_processing}/library/sweep/event_handler.py

@@ -8,8 +8,8 @@ _LOGGER = getLogger(__name__)
 import os
 
 from spectre_core.chunks.base import BaseChunk
-from spectre_core.watchdog.base import BaseEventHandler
-from spectre_core.watchdog.event_handler_register import register_event_handler
+from spectre_core.post_processing.base import BaseEventHandler
+from spectre_core.post_processing.event_handler_register import register_event_handler
 
 @register_event_handler("sweep")
 class EventHandler(BaseEventHandler):
@@ -19,23 +19,22 @@ class EventHandler(BaseEventHandler):
         self.previous_chunk: BaseChunk = None # cache for previous chunk
         
 
-    def process(self, file_path: str):
-        _LOGGER.info(f"Processing: {file_path}")
-        file_name = os.path.basename(file_path)
-        chunk_start_time, _ = os.path.splitext(file_name)[0].split('_')
+    def process(self, 
+                absolute_file_path: str):
+        _LOGGER.info(f"Processing: {absolute_file_path}")
+        file_name = os.path.basename(absolute_file_path)
+        base_file_name, _ = os.path.splitext(file_name)
+        chunk_start_time, _ = base_file_name.split('_')
         chunk = self._Chunk(chunk_start_time, self._tag)
 
         _LOGGER.info("Creating spectrogram")
         spectrogram = chunk.build_spectrogram(previous_chunk = self.previous_chunk)
 
-        _LOGGER.info("Averaging spectrogram")
         spectrogram = self._average_in_time(spectrogram)
         spectrogram = self._average_in_frequency(spectrogram)
-
-        _LOGGER.info("Joining spectrogram")
         self._join_spectrogram(spectrogram)
 
-        # if the previous chunk has not yet been set, it means we were processing the first chunk
+        # if the previous chunk has not yet been set, it means we are processing the first chunk
         # so we don't need to handle the previous chunk
         if self.previous_chunk is None:
             # instead, only set it for the next time this method is called

spectre_core/post_processing/post_processor.py

@@ -0,0 +1,40 @@
+# SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
+# This file is part of SPECTRE
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from logging import getLogger
+_LOGGER = getLogger(__name__)
+
+from watchdog.observers import Observer
+from watchdog.events import FileCreatedEvent
+
+from spectre_core.post_processing.factory import get_event_handler_from_tag
+from spectre_core.cfg import CHUNKS_DIR_PATH
+
+class PostProcessor:
+    def __init__(self, 
+                 tag: str):
+        
+        self._observer = Observer()
+
+        EventHandler = get_event_handler_from_tag(tag)
+        self._event_handler = EventHandler(tag)
+
+
+    def start(self):
+        """Start an observer to process newly created files in the chunks directory"""
+        self._observer.schedule(self._event_handler, 
+                                CHUNKS_DIR_PATH, 
+                                recursive=True,
+                                event_filter=[FileCreatedEvent])
+        
+        try:
+            _LOGGER.info("Starting the post processing thread...") 
+            self._observer.start()
+            self._observer.join()
+        except KeyboardInterrupt:
+            _LOGGER.warning(("Keyboard interrupt detected. Signalling "
+                             "the post processing thread to stop"))
+            self._observer.stop()
+            _LOGGER.warning(("Post processing thread has been successfully stopped"))
+

spectre_core/receivers/base.py

@@ -242,6 +242,7 @@ class SPECTREReceiver(BaseReceiver):
                 "hop": int, # STFFT window hops by so many samples
                 "chunk_key": str, # maps to the corresponding chunk class
                 "event_handler_key": str, # maps to the event handler used in post processing
+                "watch_extension": str, # event handlers watch for files with this extension
             },
             "sweep": {
                 "min_freq": float, # [Hz]
@@ -262,6 +263,7 @@ class SPECTREReceiver(BaseReceiver):
                 "hop": int, # keyword arguments for the scipy STFFT class
                 "chunk_key": str, # maps to the corresponding chunk class
                 "event_handler_key": str, # maps to the event handler used in post processing
+                "watch_extension": str, # event handlers watch for files with this extension
             }
         }
     
@@ -291,17 +293,18 @@ class SPECTREReceiver(BaseReceiver):
         window_size = capture_config["window_size"]
         hop = capture_config["hop"]
         chunk_key = capture_config["chunk_key"]
-        event_handler_key = capture_config[ "event_handler_key"]
+        event_handler_key = capture_config["event_handler_key"]
+        watch_extension = capture_config["watch_extension"]
 
         validators.center_freq_strictly_positive(min_freq)
         validators.center_freq_strictly_positive(max_freq)
         validators.samp_rate_strictly_positive(samp_rate)
         validators.bandwidth_strictly_positive(bandwidth)
         validators.nyquist_criterion(samp_rate, 
-                                              bandwidth)
+                                     bandwidth)
         validators.chunk_size_strictly_positive(chunk_size)
         validators.time_resolution(time_resolution, 
-                                             chunk_size) 
+                                   chunk_size) 
         validators.window(window_type, 
                           window_kwargs, 
                           window_size,
@@ -326,6 +329,8 @@ class SPECTREReceiver(BaseReceiver):
                                          samp_rate)
         validators.num_samples_per_step(samples_per_step, 
                                         window_size)
+        validators.watch_extension(watch_extension,
+                                   "bin")
 
         # if the api latency is defined, raise a warning if the step interval is of the same order
         api_latency = self.specifications.get("api_latency")
@@ -350,6 +355,7 @@ class SPECTREReceiver(BaseReceiver):
         hop = capture_config["hop"]
         chunk_key = capture_config["chunk_key"]
         event_handler_key = capture_config["event_handler_key"]
+        watch_extension = capture_config["watch_extension"]
 
         validators.center_freq_strictly_positive(center_freq)
         validators.samp_rate_strictly_positive(samp_rate)
@@ -369,7 +375,8 @@ class SPECTREReceiver(BaseReceiver):
                                      "fixed")
         validators.gain_is_negative(IF_gain)
         validators.gain_is_negative(RF_gain)
-    
+        validators.watch_extension(watch_extension,
+                                "bin")
 
 # parent class for shared methods and attributes of SDRPlay receivers
 class SDRPlayReceiver(SPECTREReceiver):

spectre_core/receivers/library/rspduo/gr/tuner_1_fixed.py

@@ -5,7 +5,7 @@
 # SPDX-License-Identifier: GPL-3.0
 #
 # GNU Radio Python Flow Graph
-# Title: Options 0
+# Title: tuner_1_fixed
 # GNU Radio version: 3.10.1.1
 
 # SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
@@ -50,7 +50,11 @@ class tuner_1_fixed(gr.top_block):
         ##################################################
         # Blocks
         ##################################################
-        self.spectre_batched_file_sink_0 = spectre.batched_file_sink(CHUNKS_DIR_PATH, tag, chunk_size, samp_rate, is_sweeping)
+        self.spectre_batched_file_sink_0 = spectre.batched_file_sink(CHUNKS_DIR_PATH, 
+                                                                     tag, 
+                                                                     chunk_size, 
+                                                                     samp_rate, 
+                                                                     is_sweeping)
         self.sdrplay3_rspduo_0 = sdrplay3.rspduo(
             '',
             rspduo_mode="Single Tuner",

spectre_core/receivers/library/rspduo/gr/tuner_1_sweep.py

@@ -5,9 +5,13 @@
 # SPDX-License-Identifier: GPL-3.0
 #
 # GNU Radio Python Flow Graph
-# Title: Not titled yet
+# Title: tuner_1_sweep
 # GNU Radio version: 3.10.1.1
 
+# SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
+# This file is part of SPECTRE
+# SPDX-License-Identifier: GPL-3.0-or-later
+
 import sys
 import signal
 from argparse import ArgumentParser
@@ -31,7 +35,7 @@ from spectre_core.file_handlers.configs import CaptureConfig
 class tuner_1_sweep(gr.top_block):
     def __init__(self, 
                  capture_config: CaptureConfig):
-        gr.top_block.__init__(self, "tuner-1-sweep", catch_exceptions=True)
+        gr.top_block.__init__(self, "tuner_1_sweep", catch_exceptions=True)
 
         ##################################################
         # Unpack capture config
@@ -93,8 +97,8 @@ class tuner_1_sweep(gr.top_block):
         self.sdrplay3_rspduo_0.set_debug_mode(False)
         self.sdrplay3_rspduo_0.set_sample_sequence_gaps_check(False)
         self.sdrplay3_rspduo_0.set_show_gain_changes(False)
-        self.blocks_tag_debug_0 = blocks.tag_debug(gr.sizeof_gr_complex*1, 'freq', "freq")
-        self.blocks_tag_debug_0.set_display(True)
+        # self.blocks_tag_debug_0 = blocks.tag_debug(gr.sizeof_gr_complex*1, 'freq', "freq")
+        # self.blocks_tag_debug_0.set_display(True)
 
 
         ##################################################
@@ -105,14 +109,6 @@ class tuner_1_sweep(gr.top_block):
         self.connect((self.sdrplay3_rspduo_0, 0), (self.spectre_sweep_driver_0, 0))
 
 
-    def get_samp_rate(self):
-        return self.samp_rate
-
-    def set_samp_rate(self, samp_rate):
-        self.samp_rate = samp_rate
-        self.sdrplay3_rspduo_0.set_sample_rate(self.samp_rate, True)
-
-
 
 
 def main(capture_config: CaptureConfig, 

spectre_core/receivers/library/rspduo/gr/tuner_2_fixed.py

@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+#
+# SPDX-License-Identifier: GPL-3.0
+#
+# GNU Radio Python Flow Graph
+# Title: Not titled yet
+# GNU Radio version: 3.10.1.1
+
+# SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
+# This file is part of SPECTRE
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from gnuradio import gr
+from gnuradio.filter import firdes
+from gnuradio.fft import window
+import sys
+import signal
+from argparse import ArgumentParser
+from gnuradio.eng_arg import eng_float, intx
+from gnuradio import eng_notation
+from gnuradio import sdrplay3
+from gnuradio import spectre
+
+from spectre_core.cfg import CHUNKS_DIR_PATH
+from spectre_core.file_handlers.configs import CaptureConfig
+
+
+class tuner_2_fixed(gr.top_block):
+
+    def __init__(self,
+                 capture_config: CaptureConfig):
+        gr.top_block.__init__(self, "tuner_2_fixed", catch_exceptions=True)
+
+        ##################################################
+        # Unpack capture config
+        ##################################################
+        samp_rate = capture_config['samp_rate']
+        tag = capture_config['tag']
+        chunk_size = capture_config['chunk_size']
+        center_freq = capture_config['center_freq']
+        bandwidth = capture_config['bandwidth']
+        IF_gain = capture_config['IF_gain']
+        RF_gain = capture_config['RF_gain']
+        is_sweeping = False
+
+        ##################################################
+        # Blocks
+        ##################################################
+        self.spectre_batched_file_sink_0 = spectre.batched_file_sink(CHUNKS_DIR_PATH, 
+                                                                     tag, 
+                                                                     chunk_size, 
+                                                                     samp_rate, 
+                                                                     is_sweeping)
+        self.sdrplay3_rspduo_0 = sdrplay3.rspduo(
+            '',
+            rspduo_mode="Single Tuner",
+            antenna="Tuner 2 50 ohm",
+            stream_args=sdrplay3.stream_args(
+                output_type='fc32',
+                channels_size=1
+            ),
+        )
+        self.sdrplay3_rspduo_0.set_sample_rate(samp_rate)
+        self.sdrplay3_rspduo_0.set_center_freq(center_freq)
+        self.sdrplay3_rspduo_0.set_bandwidth(bandwidth)
+        self.sdrplay3_rspduo_0.set_antenna("Tuner 2 50 ohm")
+        self.sdrplay3_rspduo_0.set_gain_mode(False)
+        self.sdrplay3_rspduo_0.set_gain(IF_gain, 'IF')
+        self.sdrplay3_rspduo_0.set_gain(RF_gain, 'RF', False)
+        self.sdrplay3_rspduo_0.set_freq_corr(0)
+        self.sdrplay3_rspduo_0.set_dc_offset_mode(False)
+        self.sdrplay3_rspduo_0.set_iq_balance_mode(False)
+        self.sdrplay3_rspduo_0.set_agc_setpoint(-30)
+        self.sdrplay3_rspduo_0.set_rf_notch_filter(False)
+        self.sdrplay3_rspduo_0.set_dab_notch_filter(False)
+        self.sdrplay3_rspduo_0.set_am_notch_filter(False)
+        self.sdrplay3_rspduo_0.set_biasT(False)
+        self.sdrplay3_rspduo_0.set_stream_tags(False)
+        self.sdrplay3_rspduo_0.set_debug_mode(False)
+        self.sdrplay3_rspduo_0.set_sample_sequence_gaps_check(False)
+        self.sdrplay3_rspduo_0.set_show_gain_changes(False)
+
+
+        ##################################################
+        # Connections
+        ##################################################
+        self.connect((self.sdrplay3_rspduo_0, 0), (self.spectre_batched_file_sink_0, 0))
+
+
+    def get_samp_rate(self):
+        return self.samp_rate
+
+    def set_samp_rate(self, samp_rate):
+        self.samp_rate = samp_rate
+        self.sdrplay3_rspduo_0.set_sample_rate(self.samp_rate, False)
+
+
+
+
+def main(capture_config: CaptureConfig, 
+         top_block_cls=tuner_2_fixed, 
+         options=None):
+
+    tb = top_block_cls(capture_config)
+
+    def sig_handler(sig=None, frame=None):
+        tb.stop()
+        tb.wait()
+
+        sys.exit(0)
+
+    signal.signal(signal.SIGINT, sig_handler)
+    signal.signal(signal.SIGTERM, sig_handler)
+
+    tb.start()
+
+    tb.wait()
+

spectre_core/receivers/library/rspduo/gr/tuner_2_sweep.py

@@ -0,0 +1,119 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+#
+# SPDX-License-Identifier: GPL-3.0
+#
+# GNU Radio Python Flow Graph
+# Title: Not titled yet
+# GNU Radio version: 3.10.1.1
+
+from gnuradio import gr
+from gnuradio.filter import firdes
+from gnuradio.fft import window
+import sys
+import signal
+from argparse import ArgumentParser
+from gnuradio.eng_arg import eng_float, intx
+from gnuradio import eng_notation
+from gnuradio import sdrplay3
+from gnuradio import spectre
+
+from spectre_core.cfg import CHUNKS_DIR_PATH
+from spectre_core.file_handlers.configs import CaptureConfig
+
+
+class tuner_2_sweep(gr.top_block):
+    def __init__(self, 
+                 capture_config: CaptureConfig):
+        gr.top_block.__init__(self, "tuner_2_sweep", catch_exceptions=True)
+
+        ##################################################
+        # Unpack capture config
+        ##################################################
+        samp_rate = capture_config['samp_rate']
+        bandwidth = capture_config['bandwidth']
+        min_freq = capture_config['min_freq']
+        max_freq = capture_config['max_freq']
+        freq_step = capture_config['freq_step']
+        samples_per_step = capture_config['samples_per_step']
+        IF_gain = capture_config['IF_gain']
+        RF_gain = capture_config['RF_gain']
+        chunk_size = capture_config['chunk_size']
+        start_freq = min_freq + samp_rate/2
+        tag = capture_config['tag']
+
+        ##################################################
+        # Blocks
+        ##################################################
+        self.spectre_sweep_driver_0 = spectre.sweep_driver(min_freq, 
+                                                           max_freq, 
+                                                           freq_step, 
+                                                           samp_rate, 
+                                                           samples_per_step,
+                                                           'freq')
+        self.spectre_batched_file_sink_0 = spectre.batched_file_sink(CHUNKS_DIR_PATH, 
+                                                                     tag, 
+                                                                     chunk_size, 
+                                                                     samp_rate, 
+                                                                     True, 
+                                                                     'freq', 
+                                                                     start_freq)
+        self.sdrplay3_rspduo_0 = sdrplay3.rspduo(
+            '',
+            rspduo_mode="Single Tuner",
+            antenna="Tuner 2 50 ohm",
+            stream_args=sdrplay3.stream_args(
+                output_type='fc32',
+                channels_size=1
+            ),
+        )
+        self.sdrplay3_rspduo_0.set_sample_rate(samp_rate, True)
+        self.sdrplay3_rspduo_0.set_center_freq(start_freq, True)
+        self.sdrplay3_rspduo_0.set_bandwidth(bandwidth)
+        self.sdrplay3_rspduo_0.set_antenna("Tuner 2 50 ohm")
+        self.sdrplay3_rspduo_0.set_gain_mode(False)
+        self.sdrplay3_rspduo_0.set_gain(IF_gain, 'IF', True)
+        self.sdrplay3_rspduo_0.set_gain(RF_gain, 'RF', True)
+        self.sdrplay3_rspduo_0.set_freq_corr(0)
+        self.sdrplay3_rspduo_0.set_dc_offset_mode(False)
+        self.sdrplay3_rspduo_0.set_iq_balance_mode(False)
+        self.sdrplay3_rspduo_0.set_agc_setpoint(-30)
+        self.sdrplay3_rspduo_0.set_rf_notch_filter(False)
+        self.sdrplay3_rspduo_0.set_dab_notch_filter(True)
+        self.sdrplay3_rspduo_0.set_am_notch_filter(False)
+        self.sdrplay3_rspduo_0.set_biasT(False)
+        self.sdrplay3_rspduo_0.set_stream_tags(True)
+        self.sdrplay3_rspduo_0.set_debug_mode(False)
+        self.sdrplay3_rspduo_0.set_sample_sequence_gaps_check(False)
+        self.sdrplay3_rspduo_0.set_show_gain_changes(False)
+
+
+        ##################################################
+        # Connections
+        ##################################################
+        self.msg_connect((self.spectre_sweep_driver_0, 'freq'), (self.sdrplay3_rspduo_0, 'freq'))
+        self.connect((self.sdrplay3_rspduo_0, 0), (self.spectre_batched_file_sink_0, 0))
+        self.connect((self.sdrplay3_rspduo_0, 0), (self.spectre_sweep_driver_0, 0))
+
+
+def main(capture_config: CaptureConfig, 
+         top_block_cls=tuner_2_sweep, 
+         options=None):
+    tb = top_block_cls(capture_config)
+
+    def sig_handler(sig=None, frame=None):
+        tb.stop()
+        tb.wait()
+
+        sys.exit(0)
+
+    signal.signal(signal.SIGINT, sig_handler)
+    signal.signal(signal.SIGTERM, sig_handler)
+
+    tb.start(1024)
+
+    tb.wait()
+
+if __name__ == '__main__':
+    main()