diff options
| author | Tom Rondeau <trondeau@vt.edu> | 2013-01-19 16:24:22 -0500 |
|---|---|---|
| committer | Tom Rondeau <trondeau@vt.edu> | 2013-01-20 12:13:44 -0500 |
| commit | 26228d88fc785a7f24639e6216ac01bc1ba48c5c (patch) | |
| tree | 2b01eb0461d87abc7ce90d64619c28b982abf58f /gr-qtgui/apps/gr_spectrogram_plot_c | |
| parent | 182904c9c0a07383ca2363b237233c66e736961c (diff) | |
qtgui: updated constellation and spectrogram plotting tools.
Diffstat (limited to 'gr-qtgui/apps/gr_spectrogram_plot_c')
| -rwxr-xr-x | gr-qtgui/apps/gr_spectrogram_plot_c | 149 |
1 files changed, 24 insertions, 125 deletions
diff --git a/gr-qtgui/apps/gr_spectrogram_plot_c b/gr-qtgui/apps/gr_spectrogram_plot_c index de2bd09f39..c8b904742b 100755 --- a/gr-qtgui/apps/gr_spectrogram_plot_c +++ b/gr-qtgui/apps/gr_spectrogram_plot_c @@ -1,6 +1,6 @@ #!/usr/bin/env python # -# Copyright 2012 Free Software Foundation, Inc. +# Copyright 2012,2013 Free Software Foundation, Inc. # # This file is part of GNU Radio # @@ -21,10 +21,12 @@ # from gnuradio import gr -from gnuradio import blocks -from gnuradio.eng_option import eng_option -from optparse import OptionParser -import os, sys +import scipy + +try: + import gnuradio.qtgui.plot_spectrogram_base as plot_base +except ImportError: + import plot_spectrogram_base as plot_base try: from gnuradio import qtgui @@ -34,138 +36,35 @@ except ImportError: print "Error: Program requires PyQt4 and gr-qtgui." sys.exit(1) -try: - import scipy -except ImportError: - print "Error: Scipy required (www.scipy.org)." - sys.exit(1) - -try: - from gnuradio.qtgui.plot_form import * -except ImportError: - from plot_form import * - -def read_samples_and_pad(filename, start, in_size, min_size): - # Read in_size number of samples from file - fhandle = open(filename, 'r') - fhandle.seek(start*gr.sizeof_gr_complex, 0) - data = scipy.fromfile(fhandle, dtype=scipy.complex64, count=in_size) - data = data.tolist() - fhandle.close() - - # If we have to, append 0's to create min_size samples of data - if(len(data) < min_size): - data += (min_size - len(data)) * [complex(0,0)] - - return data - -class my_top_block(gr.top_block): +class spectrogram_plot_c(plot_base.plot_base): def __init__(self, filelist, fc, samp_rate, psdsize, start, nsamples, max_nsamples, avg=1.0): - gr.top_block.__init__(self) - - self._filelist = filelist - self._center_freq = fc - self._samp_rate = samp_rate - self._psd_size = psdsize - self._start = start - self._max_nsamps = max_nsamples - self._nsigs = len(self._filelist) - self._avg = avg - - if(nsamples is None): - self._nsamps = max_nsamples - else: - self._nsamps = nsamples + plot_base.plot_base.__init__(self, filelist, fc, samp_rate, + psdsize, start, nsamples, + max_nsamples, avg) - self.qapp = QtGui.QApplication(sys.argv) - - self.skip = gr.skiphead(gr.sizeof_gr_complex, self._start) - self.add = blocks.add_cc() + self.read_samples = plot_base.read_samples_c + self.dsize = gr.sizeof_gr_complex + self.src_type = gr.vector_source_c self.gui_snk = qtgui.waterfall_sink_c(self._psd_size, gr.firdes.WIN_BLACKMAN_hARRIS, self._center_freq, self._samp_rate, - "GNU Radio Spectrogram Plot") - - n = 0 - self.srcs = list() - for f in filelist: - data = read_samples_and_pad(f, self._start, - self._nsamps, self._psd_size) - self.srcs.append(gr.vector_source_c(data)) - n += 1 - - self.connect(self.add, self.skip) - self.connect(self.skip, (self.gui_snk, 0)) - - for i,s in enumerate(self.srcs): - self.connect(s, (self.add, i)) - - self.gui_snk.set_update_time(0); - self.gui_snk.set_fft_average(self._avg) - - # Get Python Qt references - pyQt = self.gui_snk.pyqwidget() - self.pyWin = sip.wrapinstance(pyQt, QtGui.QWidget) - - def get_gui(self): - return self.pyWin - - def reset(self, newstart, newnsamps): - self.stop() - self.wait() - self.gui_snk.clear_data() - - self._start = newstart - self._nsamps = newnsamps - - for s,f in zip(self.srcs, self._filelist): - data = read_samples_and_pad(f, self._start, - self._nsamps, self._psd_size) - s.set_data(data) - - self.start() + "GNU Radio Spectrogram Plot", 1) + self.setup() def main(): description = "Plots the spectrogram (waterfall) of a list of files. Files are a binary list of complex floats." - parser = OptionParser(option_class=eng_option, description=description, - conflict_handler="resolve") - parser.add_option("-N", "--nsamples", type="int", default=None, - help="Set the number of samples to display [default=prints entire file]") - parser.add_option("-S", "--start", type="int", default=0, - help="Starting sample number [default=%default]") - parser.add_option("-L", "--psd-size", type="int", default=2048, - help="Set the FFT size of the PSD [default=%default]") - parser.add_option("-f", "--center-frequency", type="eng_float", default=0.0, - help="Set the center frequency of the signal [default=%default]") - parser.add_option("-r", "--sample-rate", type="eng_float", default=1.0, - help="Set the sample rate of the signal [default=%default]") - parser.add_option("-a", "--average", type="float", default=1.0, - help="Set amount of averaging (smaller=more averaging) [default=%default]") - (options, args) = parser.parse_args() - - if(len(args) < 1): - parser.print_help() - sys.exit(0) + (options, args) = plot_base.setup_options(description) filelist = list(args) + max_nsamples = plot_base.find_max_nsamples(filelist) - nsamples = options.nsamples - - # Find the smallest number of samples in all files and use that as - # a maximum value possible. - filesizes = [] - for f in filelist: - if(os.path.exists(f)): - filesizes.append(os.path.getsize(f) / gr.sizeof_gr_complex) - max_nsamples = min(filesizes) - - tb = my_top_block(filelist, - options.center_frequency, options.sample_rate, - options.psd_size, - options.start, nsamples, max_nsamples, - options.average); + tb = spectrogram_plot_c(filelist, + options.center_frequency, options.sample_rate, + options.psd_size, + options.start, options.nsamples, max_nsamples, + options.average); - main_box = dialog_box(tb, 'GNU Radio Spectrogram Plot') + main_box = plot_base.dialog_box(tb, 'GNU Radio Spectrogram Plot', False) main_box.show() tb.run() |