From 79d7df78dc6ce3461ef1b3498f244e9207140ed9 Mon Sep 17 00:00:00 2001 From: ab2qz Date: Thu, 1 Jul 2010 12:21:15 +0000 Subject: [PATCH] added Balint's pcm decoder git-svn-id: http://op25.osmocom.org/svn/trunk@215 65a5c917-d112-43f1-993d-58c26a4786be --- python/pcm_fsk4_oscope.py | 443 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 443 insertions(+) create mode 100755 python/pcm_fsk4_oscope.py diff --git a/python/pcm_fsk4_oscope.py b/python/pcm_fsk4_oscope.py new file mode 100755 index 0000000..8466eeb --- /dev/null +++ b/python/pcm_fsk4_oscope.py @@ -0,0 +1,443 @@ +#!/usr/bin/env python +# +# Copyright 2004,2005,2006,2007 Free Software Foundation, Inc. +# +# This file is part of GNU Radio +# +# GNU Radio is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 3, or (at your option) +# any later version. +# +# GNU Radio is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with GNU Radio; see the file COPYING. If not, write to +# the Free Software Foundation, Inc., 51 Franklin Street, +# Boston, MA 02110-1301, USA. +# + +# Original 'usrp_fsk4_oscope.py' modified to work with a normal sound card, WAV files and complex data files. +# By balint256 on http://tech.groups.yahoo.com/group/op25-dev/ +# +# Channel filter is only block that decimates +# Therefore same decimation value must be used as when channel filter was used in previous run so filters are identical +# Sensitive to gain when using discriminator tap from N-FM demod + +from gnuradio import gr, gru, optfir +from gnuradio import audio # Soundcard input +from gnuradio import eng_notation +from gnuradio.eng_option import eng_option +from gnuradio.wxgui import stdgui2, scopesink2, form, slider +from optparse import OptionParser +import wx +import sys +from gnuradio import fsk4 +from math import pi + +import threading + +class app_top_block(stdgui2.std_top_block): + def __init__( self, frame, panel, vbox, argv): + stdgui2.std_top_block.__init__( self, frame, panel, vbox, argv) + + self.frame = frame + self.panel = panel + + self.offset = 0.0 # Channel frequency offset + + parser = OptionParser(option_class=eng_option) + + parser.add_option("-p", "--protocol", type="int", default=1, + help="set protocol: 0 = RDLAP 19.2kbps; 1 = APCO25 (default)") + parser.add_option("-g", "--gain", type="eng_float", default=1.0, + help="set linear input gain (default: %default)") + parser.add_option("-x", "--freq-translation", type="eng_float", default=0.0, + help="initial channel frequency translation") + + parser.add_option("-n", "--frame-decim", type="int", default=1, + help="set oscope frame decimation factor to n [default=1]") + parser.add_option("-v", "--v-scale", type="eng_float", default=5000, + help="set oscope initial V/div to SCALE [default=%default]") + parser.add_option("-t", "--t-scale", type="eng_float", default=49e-6, + help="set oscope initial s/div to SCALE [default=50us]") + + parser.add_option("-I", "--audio-input", type="string", default="", + help="pcm input device name. E.g., hw:0,0 or /dev/dsp") + parser.add_option("-r", "--sample-rate", type="eng_float", default=48000, + help="set sample rate to RATE (default: %default)") + + parser.add_option("-d", "--channel-decim", type="int", default=None, + help="set channel decimation factor to n [default depends on protocol]") + + parser.add_option("-w", "--wav-file", type="string", default=None, + help="WAV input path") + parser.add_option("-f", "--data-file", type="string", default=None, + help="Data input path") + parser.add_option("-B", "--base-band", action="store_true", default=False) + parser.add_option("-R", "--repeat", action="store_true", default=False) + + parser.add_option("-o", "--wav-out", type="string", default=None, + help="WAV output path") + parser.add_option("-G", "--wav-out-gain", type="eng_float", default=0.05, + help="set WAV output gain (default: %default)") + + parser.add_option("-F", "--data-out", type="string", default=None, + help="Data output path") + + parser.add_option("-C", "--carrier-freq", type="eng_float", default=None, + help="set data output carrier frequency to FREQ", metavar="FREQ") + + (options, args) = parser.parse_args() + if len(args) != 0: + parser.print_help() + sys.exit(1) + + self.options = options + + if options.wav_file is not None: + #try: + self.input_file = gr.wavfile_source(options.wav_file, options.repeat) + #except: + # print "WAV file not found or not a WAV file" + # sys.exit(1) + + print "WAV input: %i Hz, %i bits, %i channels" % (self.input_file.sample_rate(), self.input_file.bits_per_sample(), self.input_file.channels()) + + self.sample_rate = self.input_file.sample_rate() + self.input_stream = gr.throttle(gr.sizeof_float, self.sample_rate) + self.connect(self.input_file, self.input_stream) + self.src = gr.multiply_const_ff(options.gain) + elif options.data_file is not None: + if options.base_band: + sample_size = gr.sizeof_float + print "Data file is baseband (float)" + self.src = gr.multiply_const_ff(options.gain) + else: + sample_size = gr.sizeof_gr_complex + print "Data file is IF (complex)" + self.src = gr.multiply_const_cc(options.gain) + + self.input_file = gr.file_source(sample_size, options.data_file, options.repeat) + self.sample_rate = options.sample_rate # E.g. 250000 + + print "Data file sampling rate = " + str(self.sample_rate) + + self.input_stream = gr.throttle(sample_size, self.sample_rate) + self.connect(self.input_file, self.input_stream) + else: + self.sample_rate = options.sample_rate + print "Soundcard sampling rate = " + str(self.sample_rate) + self.input_stream = audio.source(self.sample_rate, options.audio_input) # float samples + self.src = gr.multiply_const_ff(options.gain) + + print "Fixed input gain = " + str(options.gain) + self.connect(self.input_stream, self.src) + + if options.wav_out is not None: + output_rate = int(self.sample_rate) + if options.channel_decim is not None: + output_rate /= options.channel_decim + self.wav_out = gr.wavfile_sink(options.wav_out, 1, output_rate, 16) + print "Opened WAV output file: " + options.wav_out + " at rate: " + str(output_rate) + else: + self.wav_out = None + + if options.data_out is not None: + if options.carrier_freq is None: + self.data_out = gr.file_sink(gr.sizeof_float, options.data_out) + print "Opened float data output file: " + options.data_out + else: + self.data_out = gr.file_sink(gr.sizeof_gr_complex, options.data_out) + print "Opened complex data output file: " + options.data_out + else: + self.data_out = None + + self.num_inputs = 1 + + input_rate = self.sample_rate + #title='IF', + #size=(1024,800), + if (options.data_file is not None) and (options.base_band == False): + self.scope = scopesink2.scope_sink_c(panel, + sample_rate=input_rate, + frame_decim=options.frame_decim, + v_scale=options.v_scale, + t_scale=options.t_scale, + num_inputs=self.num_inputs) + else: + self.scope = scopesink2.scope_sink_f(panel, + sample_rate=input_rate, + frame_decim=options.frame_decim, + v_scale=options.v_scale, + t_scale=options.t_scale, + num_inputs=self.num_inputs) + + #self.di = gr.deinterleave(gr.sizeof_float) + #self.di = gr.complex_to_float(1) + #self.di = gr.complex_to_imag() + #self.dr = gr.complex_to_real() + #self.connect(self.src,self.dr) + #self.connect(self.src,self.di) + #self.null = gr.null_sink(gr.sizeof_double) + #self.connect(self.src,self.di) + #self.connect((self.di,0),(self.scope,0)) + #self.connect((self.di,1),(self.scope,1)) + #self.connect(self.dr,(self.scope,0)) + #self.connect(self.di,(self.scope,1)) + + self.msgq = gr.msg_queue(2) # queue that holds a maximum of 2 messages + self.queue_watcher = queue_watcher(self.msgq, self.adjust_freq_norm) + + #------------------------------------------------------------------------------- + + if options.protocol == 0: + # ---------- RD-LAP 19.2 kbps (9600 ksps), 25kHz channel, + print "RD-LAP selected" + self.symbol_rate = 9600 # symbol rate; at 2 bits/symbol this corresponds to 19.2kbps + + if options.channel_decim is None: + self.channel_decimation = 10 # decimation (final rate should be at least several symbol rate) + else: + self.channel_decimation = options.channel_decim + + self.max_frequency_offset = 12000.0 # coarse carrier tracker leash, ~ half a channel either way + self.symbol_deviation = 1200.0 # this is frequency offset from center of channel to +1 / -1 symbols + + self.input_sample_rate = self.sample_rate + + self.protocol_processing = fsk4.rdlap_f(self.msgq, 0) # desired protocol processing block selected here + + self.channel_rate = self.input_sample_rate / self.channel_decimation + + # channel selection filter characteristics + channel_taps = optfir.low_pass(1.0, # Filter gain + self.input_sample_rate, # Sample rate + 10000, # One-sided modulation bandwidth + 12000, # One-sided channel bandwidth + 0.1, # Passband ripple + 60) # Stopband attenuation + + # symbol shaping filter characteristics + symbol_coeffs = gr.firdes.root_raised_cosine (1.0, # gain + self.channel_rate , # sampling rate + self.symbol_rate, # symbol rate + 0.2, # alpha + 500) # taps + + if options.protocol == 1: + # ---------- APCO-25 C4FM Test Data + print "APCO selected" + self.symbol_rate = 4800 # symbol rate + + if options.channel_decim is None: + self.channel_decimation = 20 # decimation + else: + self.channel_decimation = options.channel_decim + + self.max_frequency_offset = 6000.0 # coarse carrier tracker leash + self.symbol_deviation = 600.0 # this is frequency offset from center of channel to +1 / -1 symbols + + self.input_sample_rate = self.sample_rate + + self.protocol_processing = fsk4.apco25_f(self.msgq, 0) + self.channel_rate = self.input_sample_rate / self.channel_decimation + + # channel selection filter + if (options.data_file is not None) and (options.base_band == False): + channel_taps = optfir.low_pass(1.0, # Filter gain + self.input_sample_rate, # Sample rate + 5000, # One-sided modulation bandwidth + 6500, # One-sided channel bandwidth + 0.2, # Passband ripple (was 0.1) + 60) # Stopband attenuation + else: + channel_taps = None + + # symbol shaping filter + symbol_coeffs = gr.firdes.root_raised_cosine (1.0, # gain + self.channel_rate, # sampling rate + self.symbol_rate, # symbol rate + 0.2, # alpha + 500) # taps + + # ----------------- End of setup block + + print "Input rate = " + str(self.input_sample_rate) + print "Channel decimation = " + str(self.channel_decimation) + print "Channel rate = " + str(self.channel_rate) + + if channel_taps is not None: + self.chan = gr.freq_xlating_fir_filter_ccf(self.channel_decimation, # Decimation rate + channel_taps, # Filter taps + 0.0, # Offset frequency + self.input_sample_rate) # Sample rate + if (options.freq_translation != 0): + print "Channel center frequency = " + str(options.freq_translation) + self.chan.set_center_freq(options.freq_translation) + else: + self.chan = None + + if options.carrier_freq is not None: + print "Carrier frequency = " + str(options.carrier_freq) + self.sig_carrier = gr.sig_source_c(self.channel_rate, gr.GR_COS_WAVE, options.carrier_freq, 1, 0) + self.carrier_mul = gr.multiply_vcc(1) # cc(gr.sizeof_gr_complex) + + self.connect(self.sig_carrier, (self.carrier_mul, 0)) + self.connect(self.chan, (self.carrier_mul, 1)) + + self.sig_i_carrier = gr.sig_source_f(self.channel_rate, gr.GR_COS_WAVE, options.carrier_freq, 1, 0) + self.sig_q_carrier = gr.sig_source_f(self.channel_rate, gr.GR_COS_WAVE, options.carrier_freq, 1, (pi / 2.0)) + self.carrier_i_mul = gr.multiply_ff(1) + self.carrier_q_mul = gr.multiply_ff(1) + self.iq_to_float = gr.complex_to_float(1) + self.carrier_iq_add = gr.add_ff(1) + + self.connect(self.carrier_mul, self.iq_to_float) + self.connect((self.iq_to_float, 0), (self.carrier_i_mul, 0)) + self.connect((self.iq_to_float, 1), (self.carrier_q_mul, 0)) + self.connect(self.sig_i_carrier, (self.carrier_i_mul, 1)) + self.connect(self.sig_q_carrier, (self.carrier_q_mul, 1)) + self.connect(self.carrier_i_mul, (self.carrier_iq_add, 0)) + self.connect(self.carrier_q_mul, (self.carrier_iq_add, 1)) + else: + self.sig_carrier = None + self.carrier_mul = None + + #lf_channel_taps = optfir.low_pass(1.0, # Filter gain + # self.input_sample_rate, # Sample rate + # 2500, # One-sided modulation bandwidth + # 3250, # One-sided channel bandwidth + # 0.1, # Passband ripple (0.1) + # 60, # Stopband attenuation + # 9) # Extra taps (default 2, which doesn't work at 48kHz) + #self.lf = gr.fir_filter_fff(self.channel_decimation, lf_channel_taps) + + self.scope2 = scopesink2.scope_sink_f(panel, sample_rate=self.symbol_rate, + frame_decim=1, + v_scale=2, + t_scale=0.025, + num_inputs=self.num_inputs) + + # also note: this specifies the nominal frequency deviation for the 4-level fsk signal + self.fm_demod_gain = self.channel_rate / (2.0 * pi * self.symbol_deviation) + self.fm_demod = gr.quadrature_demod_cf(self.fm_demod_gain) + + symbol_decim = 1 + self.symbol_filter = gr.fir_filter_fff(symbol_decim, symbol_coeffs) + + # eventually specify: sample rate, symbol rate + self.demod_fsk4 = fsk4.demod_ff(self.msgq, self.channel_rate, self.symbol_rate) + + if (self.chan is not None): + self.connect(self.src, self.chan, self.fm_demod, self.symbol_filter, self.demod_fsk4, self.protocol_processing) + + if options.wav_out is not None: + print "WAV output gain = " + str(options.wav_out_gain) + self.scaled_wav_data = gr.multiply_const_ff(float(options.wav_out_gain)) + self.connect(self.scaled_wav_data, self.wav_out) + + if self.carrier_mul is None: + self.connect(self.fm_demod, self.scaled_wav_data) + else: + self.connect(self.carrier_iq_add, self.scaled_wav_data) + + if self.data_out is not None: + if self.carrier_mul is None: + self.connect(self.fm_demod, self.data_out) + else: + self.connect(self.carrier_mul, self.data_out) + + # During signal, -4..4 + #self.connect(self.fm_demod, self.scope2) + else: + self.connect(self.src, self.symbol_filter, self.demod_fsk4, self.protocol_processing) + + self.connect(self.src, self.scope) + #self.connect(self.lf, self.scope) + + self.connect(self.demod_fsk4, self.scope2) + #self.connect(self.symbol_filter, self.scope2) + + # --------------- End of most of the 4L-FSK hack & slash + + self._build_gui(vbox) + + # set initial values + + if options.gain is None: + options.gain = 0 + + self.set_gain(options.gain) + + + def adjust_freq_norm(self, frequency_offset): + self.offset -= frequency_offset * self.symbol_deviation + if self.offset > self.max_frequency_offset: + self.offset = self.max_frequency_offset + if self.offset < -self.max_frequency_offset: + self.offset = -self.max_frequency_offset + print "Channel frequency offset message processed, now at (Hz):", int(self.offset) + if (self.chan is not None): + self.chan.set_center_freq(self.offset) + + + def _set_status_msg(self, msg): + self.frame.GetStatusBar().SetStatusText(msg, 0) + + def _build_gui(self, vbox): + vbox.Add(self.scope.win, 10, wx.EXPAND) + vbox.Add(self.scope2.win, 10, wx.EXPAND) + + # add control area at the bottom + self.myform = myform = form.form() + hbox = wx.BoxSizer(wx.HORIZONTAL) + hbox.Add((5,0), 0, 0) + g = [0,1000] # Faux gain range + myform['gain'] = form.slider_field(parent=self.panel, sizer=hbox, label="Gain", + weight=3, + min=int(g[0]), max=int(g[1]), + callback=self.gui_set_gain) + self.myform['gain'].set_value(int(self.options.gain * 10.0)) + + def gui_set_gain(self, gain): + self.set_gain(float(gain) / 10.0) + + def set_gain(self, gain): + #self.myform['gain'].set_value(int(gain * 10.0)) # update displayed value + #self.subdev.set_gain(gain) + print "Gain now = " + str(gain) + self.src.set_k(gain) + + +#----------------------------------------------------------------------- +# Queue watcher. Dispatches measured delay to callback. +#----------------------------------------------------------------------- +class queue_watcher (threading.Thread): + def __init__ (self, msgq, callback, **kwds): + threading.Thread.__init__ (self, **kwds) + self.setDaemon (1) + self.msgq = msgq + self.callback = callback + self.keep_running = True + self.start () + + def run (self): + while (self.keep_running): + msg = self.msgq.delete_head() # blocking read of message queue + frequency_correction = msg.arg1() + self.callback(frequency_correction) + + +def main (): + app = stdgui2.stdapp( app_top_block, "PCM O'scope + 4LFSK", nstatus=1 ) + wav_out = app.TopWindow.top_block().wav_out + app.MainLoop() + if wav_out is not None: + wav_out.close() + +if __name__ == '__main__': + main ()