#!/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 blocks
from gnuradio import filter
from gnuradio.filter import firdes
from gnuradio import gr
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




class RSPdx_testHFD(gr.top_block):

    def __init__(self):
        gr.top_block.__init__(self, "Not titled yet", catch_exceptions=True)

        ##################################################
        # Variables
        ##################################################
        self.samp_rate_0 = samp_rate_0 = 2000000
        self.samp_rate = samp_rate = 2000000
        self.dcm = dcm = 1
        self.bandw = bandw = 2000000

        ##################################################
        # Blocks
        ##################################################
        self.sdrplay3_rspdx_0 = sdrplay3.rspdx(
            '',
            stream_args=sdrplay3.stream_args(
                output_type='fc32',
                channels_size=1
            ),
        )
        self.sdrplay3_rspdx_0.set_sample_rate(samp_rate, False)
        self.sdrplay3_rspdx_0.set_center_freq(5500000, False)
        self.sdrplay3_rspdx_0.set_bandwidth(1536e3)
        self.sdrplay3_rspdx_0.set_antenna('Antenna A')
        self.sdrplay3_rspdx_0.set_gain_mode(False)
        self.sdrplay3_rspdx_0.set_gain(-(30), 'IF', False)
        self.sdrplay3_rspdx_0.set_gain(-(0), 'RF', False)
        self.sdrplay3_rspdx_0.set_freq_corr(0)
        self.sdrplay3_rspdx_0.set_dc_offset_mode(False)
        self.sdrplay3_rspdx_0.set_iq_balance_mode(False)
        self.sdrplay3_rspdx_0.set_agc_setpoint(-30)
        self.sdrplay3_rspdx_0.set_hdr_mode(False)
        self.sdrplay3_rspdx_0.set_rf_notch_filter(False)
        self.sdrplay3_rspdx_0.set_dab_notch_filter(False)
        self.sdrplay3_rspdx_0.set_biasT(False)
        self.sdrplay3_rspdx_0.set_debug_mode(False)
        self.sdrplay3_rspdx_0.set_sample_sequence_gaps_check(False)
        self.sdrplay3_rspdx_0.set_show_gain_changes(False)
        self.low_pass_filter_1_0 = filter.fir_filter_ccf(
            20000,
            firdes.low_pass(
                30,
                samp_rate/dcm,
                10,
                5,
                window.WIN_HAMMING,
                6.76))
        self.low_pass_filter_1 = filter.fir_filter_ccf(
            20000,
            firdes.low_pass(
                30,
                samp_rate/dcm,
                10,
                5,
                window.WIN_HAMMING,
                6.76))
        self.freq_xlating_fir_filter_xxx_1 = filter.freq_xlating_fir_filter_ccc(1, [32], 555000, samp_rate/1)
        self.freq_xlating_fir_filter_xxx_0 = filter.freq_xlating_fir_filter_ccc(1, [32], -494000, samp_rate/1)
        self.blocks_file_sink_1 = blocks.file_sink(gr.sizeof_gr_complex*1, '/home/hfd/data/data_6055.bin', True)
        self.blocks_file_sink_1.set_unbuffered(False)
        self.blocks_file_sink_0 = blocks.file_sink(gr.sizeof_gr_complex*1, '/home/hfd/data/data_5006.bin', True)
        self.blocks_file_sink_0.set_unbuffered(False)


        ##################################################
        # Connections
        ##################################################
        self.connect((self.freq_xlating_fir_filter_xxx_0, 0), (self.low_pass_filter_1, 0))
        self.connect((self.freq_xlating_fir_filter_xxx_1, 0), (self.low_pass_filter_1_0, 0))
        self.connect((self.low_pass_filter_1, 0), (self.blocks_file_sink_0, 0))
        self.connect((self.low_pass_filter_1_0, 0), (self.blocks_file_sink_1, 0))
        self.connect((self.sdrplay3_rspdx_0, 0), (self.freq_xlating_fir_filter_xxx_0, 0))
        self.connect((self.sdrplay3_rspdx_0, 0), (self.freq_xlating_fir_filter_xxx_1, 0))


    def get_samp_rate_0(self):
        return self.samp_rate_0

    def set_samp_rate_0(self, samp_rate_0):
        self.samp_rate_0 = samp_rate_0

    def get_samp_rate(self):
        return self.samp_rate

    def set_samp_rate(self, samp_rate):
        self.samp_rate = samp_rate
        self.low_pass_filter_1.set_taps(firdes.low_pass(30, self.samp_rate/self.dcm, 10, 5, window.WIN_HAMMING, 6.76))
        self.low_pass_filter_1_0.set_taps(firdes.low_pass(30, self.samp_rate/self.dcm, 10, 5, window.WIN_HAMMING, 6.76))
        self.sdrplay3_rspdx_0.set_sample_rate(self.samp_rate, False)

    def get_dcm(self):
        return self.dcm

    def set_dcm(self, dcm):
        self.dcm = dcm
        self.low_pass_filter_1.set_taps(firdes.low_pass(30, self.samp_rate/self.dcm, 10, 5, window.WIN_HAMMING, 6.76))
        self.low_pass_filter_1_0.set_taps(firdes.low_pass(30, self.samp_rate/self.dcm, 10, 5, window.WIN_HAMMING, 6.76))

    def get_bandw(self):
        return self.bandw

    def set_bandw(self, bandw):
        self.bandw = bandw




def main(top_block_cls=RSPdx_testHFD, options=None):
    tb = top_block_cls()

    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()


if __name__ == '__main__':
    main()