diff options
Diffstat (limited to 'gr-qtgui/lib/waterfalldisplayform.cc')
| -rw-r--r-- | gr-qtgui/lib/waterfalldisplayform.cc | 205 |
1 files changed, 205 insertions, 0 deletions
diff --git a/gr-qtgui/lib/waterfalldisplayform.cc b/gr-qtgui/lib/waterfalldisplayform.cc new file mode 100644 index 0000000000..1a69e97d10 --- /dev/null +++ b/gr-qtgui/lib/waterfalldisplayform.cc @@ -0,0 +1,205 @@ +/* -*- c++ -*- */ +/* + * Copyright 2012 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. + */ + +#include <cmath> +#include <QColorDialog> +#include <QMessageBox> +#include <waterfalldisplayform.h> +#include <iostream> + +WaterfallDisplayForm::WaterfallDisplayForm(int nplots, QWidget* parent) + : DisplayForm(nplots, parent) +{ + _intValidator = new QIntValidator(this); + _intValidator->setBottom(0); + + _layout = new QGridLayout(this); + _displayPlot = new WaterfallDisplayPlot(nplots, this); + _layout->addWidget(_displayPlot, 0, 0); + setLayout(_layout); + + _numRealDataPoints = 1024; + _fftsize = 1024; + _fftavg = 1.0; + + _min_val = 1000; + _max_val = -1000; + + QAction *autoscale_act = new QAction("Auto Scale", this); + autoscale_act->setStatusTip(tr("Autoscale intensity range")); + connect(autoscale_act, SIGNAL(triggered()), this, SLOT(AutoScale())); + + FFTSizeMenu *sizemenu = new FFTSizeMenu(this); + FFTAverageMenu *avgmenu = new FFTAverageMenu(this); + FFTWindowMenu *winmenu = new FFTWindowMenu(this); + ColorMapMenu *colormenu = new ColorMapMenu(this); + _menu->addMenu(sizemenu); + _menu->addMenu(avgmenu); + _menu->addMenu(winmenu); + _menu->addMenu(colormenu); + _menu->addAction(autoscale_act); + connect(sizemenu, SIGNAL(whichTrigger(int)), + this, SLOT(SetFFTSize(const int))); + connect(avgmenu, SIGNAL(whichTrigger(float)), + this, SLOT(SetFFTAverage(const float))); + connect(winmenu, SIGNAL(whichTrigger(gr::filter::firdes::win_type)), + this, SLOT(SetFFTWindowType(const gr::filter::firdes::win_type))); + connect(colormenu, SIGNAL(whichTrigger(const int, const QColor&, const QColor&)), + this, SLOT(SetColorMap(const int, const QColor&, const QColor&))); + + Reset(); + + connect(_displayPlot, SIGNAL(plotPointSelected(const QPointF)), + this, SLOT(onPlotPointSelected(const QPointF))); +} + +WaterfallDisplayForm::~WaterfallDisplayForm() +{ + // Qt deletes children when parent is deleted + + // Don't worry about deleting Display Plots - they are deleted when parents are deleted + delete _intValidator; +} + +WaterfallDisplayPlot* +WaterfallDisplayForm::getPlot() +{ + return ((WaterfallDisplayPlot*)_displayPlot); +} + +void +WaterfallDisplayForm::newData(const QEvent *updateEvent) +{ + WaterfallUpdateEvent *event = (WaterfallUpdateEvent*)updateEvent; + const std::vector<double*> dataPoints = event->getPoints(); + const uint64_t numDataPoints = event->getNumDataPoints(); + const gruel::high_res_timer_type dataTimestamp = event->getDataTimestamp(); + + _min_val = 1000; + _max_val = -1000; + for(size_t i=0; i < dataPoints.size(); i++) { + double *min_val = std::min_element(&dataPoints[i][0], &dataPoints[i][numDataPoints-1]); + double *max_val = std::max_element(&dataPoints[i][0], &dataPoints[i][numDataPoints-1]); + if(*min_val < _min_val) + _min_val = *min_val; + if(*max_val > _max_val) + _max_val = *max_val; + } + + getPlot()->PlotNewData(dataPoints, numDataPoints, + d_update_time, dataTimestamp, 0); +} + +void +WaterfallDisplayForm::customEvent( QEvent * e) +{ + if(e->type() == WaterfallUpdateEvent::Type()) { + newData(e); + } +} + +int +WaterfallDisplayForm::GetFFTSize() const +{ + return _fftsize; +} + +float +WaterfallDisplayForm::GetFFTAverage() const +{ + return _fftavg; +} + +gr::filter::firdes::win_type +WaterfallDisplayForm::GetFFTWindowType() const +{ + return _fftwintype; +} + +void +WaterfallDisplayForm::SetFFTSize(const int newsize) +{ + _fftsize = newsize; +} + +void +WaterfallDisplayForm::SetFFTAverage(const float newavg) +{ + _fftavg = newavg; +} + +void +WaterfallDisplayForm::SetFFTWindowType(const gr::filter::firdes::win_type newwin) +{ + _fftwintype = newwin; +} + +void +WaterfallDisplayForm::SetFrequencyRange(const double newCenterFrequency, + const double newStartFrequency, + const double newStopFrequency) +{ + double fdiff = std::max(fabs(newStartFrequency), fabs(newStopFrequency)); + + if(fdiff > 0) { + std::string strunits[4] = {"Hz", "kHz", "MHz", "GHz"}; + double units10 = floor(log10(fdiff)); + double units3 = std::max(floor(units10 / 3.0), 0.0); + double units = pow(10, (units10-fmod(units10, 3.0))); + int iunit = static_cast<int>(units3); + + _startFrequency = newStartFrequency; + _stopFrequency = newStopFrequency; + double centerFrequency = newCenterFrequency; + + getPlot()->SetFrequencyRange(_startFrequency, + _stopFrequency, + centerFrequency, + true, + units, strunits[iunit]); + } +} + +void +WaterfallDisplayForm::SetColorMap(const int newType, + const QColor lowColor, + const QColor highColor) +{ + getPlot()->SetIntensityColorMapType(0, newType, + lowColor, highColor); +} + +void +WaterfallDisplayForm::SetIntensityRange(const double minIntensity, + const double maxIntensity) +{ + getPlot()->SetIntensityRange(minIntensity, maxIntensity); +} + +void +WaterfallDisplayForm::AutoScale() +{ + double min_int = _min_val - 5; + double max_int = _max_val + 10; + + getPlot()->SetIntensityRange(min_int, max_int); +} |