git.gag.com Git - debian/gnuradio/blob - gr-qtgui/src/lib/qtgui_sink_c.cc

   1 /* -*- c++ -*- */
   2 /*
   3  * Copyright 2008,2009 Free Software Foundation, Inc.
   4  *
   5  * This file is part of GNU Radio
   6  *
   7  * GNU Radio is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 3, or (at your option)
  10  * any later version.
  11  *
  12  * GNU Radio is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with GNU Radio; see the file COPYING.  If not, write to
  19  * the Free Software Foundation, Inc., 51 Franklin Street,
  20  * Boston, MA 02110-1301, USA.
  21  */
  22
  23 #ifdef HAVE_CONFIG_H
  24 #include <config.h>
  25 #endif
  26
  27 #include <qtgui_sink_c.h>
  28 #include <gr_io_signature.h>
  29 #include <string.h>
  30
  31 #include <QTimer>
  32
  33 qtgui_sink_c_sptr
  34 qtgui_make_sink_c (int fftsize, int wintype,
  35                    float fmin, float fmax,
  36                    const std::string &name,
  37                    bool plotfreq, bool plotwaterfall,
  38                    bool plotwaterfall3d, bool plottime,
  39                    bool plotconst,
  40                    QWidget *parent)
  41 {
  42   return qtgui_sink_c_sptr (new qtgui_sink_c (fftsize, wintype,
  43                                               fmin, fmax, name,
  44                                               plotfreq, plotwaterfall,
  45                                               plotwaterfall3d, plottime,
  46                                               plotconst,
  47                                               parent));
  48 }
  49
  50 qtgui_sink_c::qtgui_sink_c (int fftsize, int wintype,
  51                             float fmin, float fmax,
  52                             const std::string &name,
  53                             bool plotfreq, bool plotwaterfall,
  54                             bool plotwaterfall3d, bool plottime,
  55                             bool plotconst,
  56                             QWidget *parent)
  57   : gr_block ("sink_c",
  58               gr_make_io_signature (1, -1, sizeof(gr_complex)),
  59               gr_make_io_signature (0, 0, 0)),
  60     d_fftsize(fftsize),
  61     d_wintype((gr_firdes::win_type)(wintype)),
  62     d_fmin(fmin), d_fmax(fmax), d_name(name),
  63     d_plotfreq(plotfreq), d_plotwaterfall(plotwaterfall),
  64     d_plotwaterfall3d(plotwaterfall3d), d_plottime(plottime),
  65     d_plotconst(plotconst),
  66     d_parent(parent)
  67 {
  68   d_main_gui = NULL;
  69   pthread_mutex_init(&d_pmutex, NULL);
  70   lock();
  71
  72   // Perform fftshift operation;
  73   // this is usually desired when plotting
  74   d_shift = true;
  75
  76   d_fft = new gri_fft_complex (d_fftsize, true);
  77
  78   d_fftdata = new gr_complex[d_fftsize];
  79
  80   d_index = 0;
  81   d_residbuf = new gr_complex[d_fftsize];
  82
  83   buildwindow();
  84
  85   initialize();
  86 }
  87
  88 qtgui_sink_c::~qtgui_sink_c()
  89 {
  90   delete d_object;
  91   delete [] d_fftdata;
  92   delete [] d_residbuf;
  93   delete d_fft;
  94 }
  95
  96 void qtgui_sink_c::lock()
  97 {
  98   pthread_mutex_lock(&d_pmutex);
  99 }
 100
 101 void qtgui_sink_c::unlock()
 102 {
 103   pthread_mutex_unlock(&d_pmutex);
 104 }
 105
 106
 107 void
 108 qtgui_sink_c::initialize()
 109 {
 110   if(qApp != NULL) {
 111     d_qApplication = qApp;
 112   }
 113   else {
 114     int argc;
 115     char **argv = NULL;
 116     d_qApplication = new QApplication(argc, argv);
 117   }
 118
 119   uint64_t maxBufferSize = 32768;
 120   d_main_gui = new SpectrumGUIClass(maxBufferSize, d_fftsize,
 121                                     d_fmin, d_fmax);
 122
 123   d_main_gui->SetDisplayTitle(d_name);
 124   d_main_gui->SetFFTSize(d_fftsize);
 125   d_main_gui->SetWindowType((int)d_wintype);
 126
 127   d_main_gui->OpenSpectrumWindow(d_parent,
 128                                  d_plotfreq, d_plotwaterfall,
 129                                  d_plotwaterfall3d, d_plottime,
 130                                  d_plotconst);
 131
 132   d_object = new qtgui_obj(d_qApplication);
 133   qApp->postEvent(d_object, new qtgui_event(&d_pmutex));
 134 }
 135
 136
 137 void
 138 qtgui_sink_c::exec_()
 139 {
 140   d_qApplication->exec();
 141 }
 142
 143 QWidget*
 144 qtgui_sink_c::qwidget()
 145 {
 146   return d_main_gui->qwidget();
 147 }
 148
 149 PyObject*
 150 qtgui_sink_c::pyqwidget()
 151 {
 152   PyObject *w = PyLong_FromVoidPtr((void*)d_main_gui->qwidget());
 153   PyObject *retarg = Py_BuildValue("N", w);
 154   return retarg;
 155 }
 156
 157 void
 158 qtgui_sink_c::set_frequency_range(const double centerfreq,
 159                                   const double startfreq,
 160                                   const double stopfreq)
 161 {
 162   d_main_gui->SetFrequencyRange(centerfreq, startfreq, stopfreq);
 163 }
 164
 165 void
 166 qtgui_sink_c::fft(const gr_complex *data_in, int size, gr_complex *data_out)
 167 {
 168   if (d_window.size()) {
 169     gr_complex *dst = d_fft->get_inbuf();
 170     int i;
 171     for (i = 0; i < size; i++)          // apply window
 172       dst[i] = data_in[i] * d_window[i];
 173   }
 174   else {
 175     memcpy (d_fft->get_inbuf(), data_in, sizeof(gr_complex)*size);
 176   }
 177
 178   d_fft->execute ();     // compute the fft
 179
 180   for(int i=0; i < size; i++) {
 181     d_fft->get_outbuf()[i] /= size;
 182   }
 183
 184   // copy result to our output
 185   if(d_shift) {  // apply a fft shift on the data
 186     unsigned int len = (unsigned int)(ceil(size/2.0));
 187     memcpy(&data_out[0], &d_fft->get_outbuf()[len], sizeof(gr_complex)*(size - len));
 188     memcpy(&data_out[size - len], &d_fft->get_outbuf()[0], sizeof(gr_complex)*len);
 189   }
 190   else {
 191     memcpy(data_out, d_fft->get_outbuf(), sizeof(gr_complex)*size);
 192   }
 193 }
 194
 195 void
 196 qtgui_sink_c::windowreset()
 197 {
 198   gr_firdes::win_type newwintype = (gr_firdes::win_type)d_main_gui->GetWindowType();
 199   if(d_wintype != newwintype) {
 200     d_wintype = newwintype;
 201     buildwindow();
 202   }
 203 }
 204
 205 void
 206 qtgui_sink_c::buildwindow()
 207 {
 208   d_window.clear();
 209   if(d_wintype != 0) {
 210     d_window = gr_firdes::window(d_wintype, d_fftsize, 6.76);
 211   }
 212 }
 213
 214 void
 215 qtgui_sink_c::fftresize()
 216 {
 217   int newfftsize = d_main_gui->GetFFTSize();
 218
 219   if(newfftsize != d_fftsize) {
 220
 221     // Resize the fftdata buffer; no need to preserve old data
 222     delete [] d_fftdata;
 223     d_fftdata = new gr_complex[newfftsize];
 224
 225     // Resize residbuf and replace data
 226     delete [] d_residbuf;
 227     d_residbuf = new gr_complex[newfftsize];
 228
 229     // Set new fft size and reset buffer index
 230     // (throws away any currently held data, but who cares?)
 231     d_fftsize = newfftsize;
 232     d_index = 0;
 233
 234     // Reset window to reflect new size
 235     buildwindow();
 236
 237     // Reset FFTW plan for new size
 238     delete d_fft;
 239     d_fft = new gri_fft_complex (d_fftsize, true);
 240   }
 241 }
 242
 243
 244 int
 245 qtgui_sink_c::general_work (int noutput_items,
 246                             gr_vector_int &ninput_items,
 247                             gr_vector_const_void_star &input_items,
 248                             gr_vector_void_star &output_items)
 249 {
 250   int i=0, j=0;
 251   const gr_complex *in = (const gr_complex*)input_items[0];
 252
 253   pthread_mutex_lock(&d_pmutex);
 254
 255   // Update the FFT size from the application
 256   fftresize();
 257   windowreset();
 258
 259   const timespec currentTime = get_highres_clock();
 260   const timespec lastUpdateGUITime = d_main_gui->GetLastGUIUpdateTime();
 261
 262   if(diff_timespec(currentTime, lastUpdateGUITime) > 0.25) {
 263
 264     if(d_index) {
 265       int filler = std::min(d_fftsize - d_index, noutput_items);
 266
 267       memcpy(&d_residbuf[d_index], &in[0], sizeof(gr_complex)*filler);
 268       d_index += filler;
 269       i = filler;
 270       j = filler;
 271     }
 272
 273     if(d_index == d_fftsize) {
 274       d_index = 0;
 275       fft(d_residbuf, d_fftsize, d_fftdata);
 276
 277       d_main_gui->UpdateWindow(true, d_fftdata, d_fftsize, NULL, 0,
 278                                (float*)d_residbuf, d_fftsize,
 279                                1.0/4.0, convert_to_timespec(0.0), true);
 280     }
 281
 282     for(; i < noutput_items; i+=d_fftsize) {
 283       if(noutput_items - i > d_fftsize) {
 284         j += d_fftsize;
 285         fft(&in[i], d_fftsize, d_fftdata);
 286
 287         d_main_gui->UpdateWindow(true, d_fftdata, d_fftsize, NULL, 0,
 288                                  (float*)&in[i], d_fftsize,
 289                                  1.0/4.0, convert_to_timespec(0.0), true);
 290       }
 291     }
 292
 293     if(noutput_items > j) {
 294       d_index = noutput_items - j;
 295       memcpy(d_residbuf, &in[j], sizeof(gr_complex)*d_index);
 296     }
 297   }
 298
 299   pthread_mutex_unlock(&d_pmutex);
 300
 301   consume_each(noutput_items);
 302   return noutput_items;
 303 }