spectre-core 0.0.1__py3-none-any.whl

spectre_core/receivers/library/rsp1a/receiver.py

@@ -0,0 +1,68 @@
+# SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
+# This file is part of SPECTRE
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from spectre_core.receivers.base import SDRPlayReceiver
+from spectre_core.receivers.receiver_register import register_receiver
+from spectre_core.receivers.library.rsp1a.gr import fixed, sweep
+from spectre_core.file_handlers.configs import CaptureConfig
+
+@register_receiver("rsp1a")
+class Receiver(SDRPlayReceiver):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+
+    def _set_capture_methods(self) -> None:
+        self._capture_methods = {
+            "fixed": self.__fixed,
+            "sweep": self.__sweep
+        }
+        return
+    
+
+    def _set_validators(self) -> None:
+        self._validators = {
+            "fixed": self.__fixed_validator,
+            "sweep": self.__sweep_validator
+        }
+        return
+    
+
+    def _set_type_templates(self) -> None:
+        self._type_templates = {
+            "fixed": self._get_default_type_template("fixed"),
+            "sweep": self._get_default_type_template("sweep")
+        }
+
+
+    def _set_specifications(self) -> None:
+        self._specifications = {
+            "center_freq_lower_bound": 1e3, # [Hz]
+            "center_freq_upper_bound": 2e9, # [Hz]
+            "samp_rate_lower_bound": 200e3, # [Hz]
+            "samp_rate_upper_bound": 10e6, # [Hz]
+            "bandwidth_lower_bound": 200e3, # [Hz]
+            "bandwidth_upper_bound": 8e6, # [Hz]
+            "IF_gain_upper_bound": -20, # [dB]
+            "RF_gain_upper_bound": 0, # [dB]
+            "api_latency": 50 * 1e-3 # [s]
+        }
+
+
+    def __fixed(self, capture_config: CaptureConfig) -> None:
+        fixed.main(capture_config)
+    
+    
+    def __sweep(self, capture_config: CaptureConfig) -> None:
+        sweep.main(capture_config)
+    
+
+    def __fixed_validator(self, capture_config: CaptureConfig) -> None:
+        self._default_fixed_validator(capture_config)
+        self._sdrplay_validator(capture_config)
+    
+
+    def __sweep_validator(self, capture_config: CaptureConfig) -> None:
+        self._default_sweep_validator(capture_config)
+        self._sdrplay_validator(capture_config)

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

@@ -0,0 +1,110 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+#
+# SPDX-License-Identifier: GPL-3.0
+#
+# GNU Radio Python Flow Graph
+# Title: Options 0
+# 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
+from typing import Any
+
+from gnuradio import gr
+from gnuradio.filter import firdes
+from gnuradio.fft import window
+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_1_fixed(gr.top_block):
+
+    def __init__(self, 
+                 capture_config: CaptureConfig):
+        gr.top_block.__init__(self, "tuner_1_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 1 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 1 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')
+        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_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 main(capture_config: CaptureConfig, 
+         top_block_cls=tuner_1_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_1_sweep.py

@@ -0,0 +1,135 @@
+#!/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
+
+import sys
+import signal
+from argparse import ArgumentParser
+from typing import Any
+
+from gnuradio import blocks
+from gnuradio import gr
+from gnuradio.filter import firdes
+from gnuradio.fft import window
+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_1_sweep(gr.top_block):
+    def __init__(self, 
+                 capture_config: CaptureConfig):
+        gr.top_block.__init__(self, "tuner-1-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 1 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 1 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)
+        self.blocks_tag_debug_0 = blocks.tag_debug(gr.sizeof_gr_complex*1, 'freq', "freq")
+        self.blocks_tag_debug_0.set_display(True)
+
+
+        ##################################################
+        # 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 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, 
+         top_block_cls=tuner_1_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()
+

spectre_core/receivers/library/rspduo/receiver.py

@@ -0,0 +1,68 @@
+# SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
+# This file is part of SPECTRE
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from spectre_core.receivers.base import SDRPlayReceiver
+from spectre_core.receivers.receiver_register import register_receiver
+from spectre_core.receivers.library.rspduo.gr import tuner_1_fixed, tuner_1_sweep
+from spectre_core.file_handlers.configs import CaptureConfig
+
+
+@register_receiver("rspduo")
+class Receiver(SDRPlayReceiver):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+
+    def _set_capture_methods(self) -> None:
+        self._capture_methods = {
+            "tuner-1-fixed": self.__tuner_1_fixed,
+            "tuner-1-sweep": self.__tuner_1_sweep
+        }
+    
+
+    def _set_validators(self) -> None:
+        self._validators = {
+            "tuner-1-fixed": self.__tuner_1_fixed_validator,
+            "tuner-1-sweep": self.__tuner_1_sweep_validator
+        }
+        return
+    
+
+    def _set_type_templates(self) -> None:
+        self._type_templates = {
+            "tuner-1-fixed": self._get_default_type_template("fixed"),
+            "tuner-1-sweep": self._get_default_type_template("sweep"),
+        }
+
+    def _set_specifications(self) -> None:
+        self._specifications = {
+            "center_freq_lower_bound": 1e3, # [Hz]
+            "center_freq_upper_bound": 2e9, # [Hz]
+            "samp_rate_lower_bound": 200e3, # [Hz]
+            "samp_rate_upper_bound": 10e6, # [Hz]
+            "bandwidth_lower_bound": 200e3, # [Hz]
+            "bandwidth_upper_bound": 8e6, # [Hz]
+            "IF_gain_upper_bound": -20, # [dB]
+            "RF_gain_upper_bound": 0, # [dB]
+            "api_latency": 50 * 1e-3 # [s]
+        }
+
+
+    def __tuner_1_fixed(self, capture_config: CaptureConfig) -> None:
+        tuner_1_fixed.main(capture_config)
+    
+
+    def __tuner_1_sweep(self, capture_config: CaptureConfig) -> None:
+        tuner_1_sweep.main(capture_config)
+    
+
+    def __tuner_1_fixed_validator(self, capture_config: CaptureConfig) -> None:
+        self._default_fixed_validator(capture_config)
+        self._sdrplay_validator(capture_config)
+    
+    
+    def __tuner_1_sweep_validator(self, capture_config: CaptureConfig) -> None:
+        self._default_sweep_validator(capture_config)
+        self._sdrplay_validator(capture_config)
+    

spectre_core/receivers/library/test/gr/cosine_signal_1.py

@@ -0,0 +1,83 @@
+#!/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
+
+import sys
+import signal
+from typing import Any
+
+from gnuradio import analog
+from gnuradio import blocks
+from gnuradio import gr
+from gnuradio.filter import firdes
+from gnuradio.fft import window
+from gnuradio.eng_arg import eng_float, intx
+from gnuradio import eng_notation
+from gnuradio import spectre
+
+from spectre_core.cfg import CHUNKS_DIR_PATH
+from spectre_core.file_handlers.configs import CaptureConfig
+
+class cosine_signal_1(gr.top_block):
+
+    def __init__(self, 
+                 capture_config: CaptureConfig):
+        gr.top_block.__init__(self, "cosine-signal-1", catch_exceptions=True)
+
+        ##################################################
+        # Unpack capture config
+        ##################################################
+        samp_rate = capture_config['samp_rate']
+        tag = capture_config['tag']
+        chunk_size = capture_config['chunk_size']
+        frequency = capture_config['frequency']
+        amplitude = capture_config['amplitude']
+
+        ##################################################
+        # Blocks
+        ##################################################
+        self.spectre_batched_file_sink_0 = spectre.batched_file_sink(CHUNKS_DIR_PATH, tag, chunk_size, samp_rate)
+        self.blocks_throttle_0_1 = blocks.throttle(gr.sizeof_float*1, samp_rate,True)
+        self.blocks_throttle_0 = blocks.throttle(gr.sizeof_float*1, samp_rate,True)
+        self.blocks_null_source_1 = blocks.null_source(gr.sizeof_float*1)
+        self.blocks_float_to_complex_1 = blocks.float_to_complex(1)
+        self.analog_sig_source_x_0 = analog.sig_source_f(samp_rate, analog.GR_COS_WAVE, frequency, amplitude, 0, 0)
+
+
+        ##################################################
+        # Connections
+        ##################################################
+        self.connect((self.analog_sig_source_x_0, 0), (self.blocks_throttle_0, 0))
+        self.connect((self.blocks_float_to_complex_1, 0), (self.spectre_batched_file_sink_0, 0))
+        self.connect((self.blocks_null_source_1, 0), (self.blocks_throttle_0_1, 0))
+        self.connect((self.blocks_throttle_0, 0), (self.blocks_float_to_complex_1, 0))
+        self.connect((self.blocks_throttle_0_1, 0), (self.blocks_float_to_complex_1, 1))
+
+
+def main(capture_config: CaptureConfig, 
+         top_block_cls=cosine_signal_1, 
+         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/test/gr/tagged_staircase.py

@@ -0,0 +1,93 @@
+#!/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
+
+import sys
+import signal
+from argparse import ArgumentParser
+from typing import Any
+
+from gnuradio import blocks
+from gnuradio import gr
+from gnuradio.filter import firdes
+from gnuradio.fft import window
+from gnuradio.eng_arg import eng_float, intx
+from gnuradio import eng_notation
+from gnuradio import spectre
+
+from spectre_core.cfg import CHUNKS_DIR_PATH
+from spectre_core.file_handlers.configs import CaptureConfig
+
+class tagged_staircase(gr.top_block):
+
+    def __init__(self, 
+                 capture_config: CaptureConfig):
+        gr.top_block.__init__(self, "tagged-staircase", catch_exceptions=True)
+
+        ##################################################
+        # Unpack capture config
+        ##################################################
+        tag = capture_config['tag']
+        step_increment = capture_config['step_increment']
+        samp_rate = capture_config['samp_rate']
+        min_samples_per_step = capture_config['min_samples_per_step']
+        max_samples_per_step = capture_config['max_samples_per_step']
+        freq_step = capture_config['freq_step']
+        chunk_size = capture_config['chunk_size']
+        is_sweeping = True
+
+        ##################################################
+        # Blocks
+        ##################################################
+        self.spectre_tagged_staircase_0 = spectre.tagged_staircase(min_samples_per_step, 
+                                                                   max_samples_per_step, 
+                                                                   freq_step,
+                                                                   step_increment, 
+                                                                   samp_rate)
+        self.spectre_batched_file_sink_0 = spectre.batched_file_sink(CHUNKS_DIR_PATH,
+                                                                     tag, 
+                                                                     chunk_size, 
+                                                                     samp_rate, 
+                                                                     is_sweeping,
+                                                                     'rx_freq',
+                                                                     0
+                                                                     )
+        self.blocks_throttle_0 = blocks.throttle(gr.sizeof_gr_complex*1, samp_rate, True)
+
+
+        ##################################################
+        # Connections
+        ##################################################
+        self.connect((self.blocks_throttle_0, 0), (self.spectre_batched_file_sink_0, 0))
+        self.connect((self.spectre_tagged_staircase_0, 0), (self.blocks_throttle_0, 0))
+
+
+
+
+def main(capture_config: CaptureConfig, 
+         top_block_cls=tagged_staircase, 
+         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()