Mirrors/inspectrum
2
0
Fork 0
mirror of https://github.com/miek/inspectrum.git synced 2026-03-03 06:54:17 +01:00
Code Issues Packages Projects Releases Wiki Activity

Rename WaveformView -> PlotView

Browse Source
This commit is contained in:
Mike Walters
2016-01-24 22:54:54 +00:00
parent dbd45d0519
commit d1a580f0b9
5 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

154
plotview.cpp Normal file
View File

@@ -0,0 +1,154 @@
/*
* Copyright (C) 2015-2016, Mike Walters <mike@flomp.net>
*
* This file is part of inspectrum.
*
* This program 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 of the License, or
* (at your option) any later version.
*
* This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "plotview.h"
#include <QDebug>
#include <QPainter>
#include <gnuradio/top_block.h>
#include <gnuradio/analog/quadrature_demod_cf.h>
#include <gnuradio/blocks/multiply_const_cc.h>
#include <gnuradio/filter/firdes.h>
#include <gnuradio/filter/freq_xlating_fir_filter_ccf.h>
#include "grsamplebuffer.h"
#include "memory_sink.h"
#include "memory_source.h"
PlotView::PlotView()
{
for (int i = 0; i < 128; i++) {
colormap[i] = qRgb(i/2, i*1.5, i*1.5);
}
for (int i = 128; i < 256; i++) {
colormap[i] = qRgb(64 + (i-128)*1.5, 192+(i-128)/2, 192+(i-128)/2);
}
}
void PlotView::refreshSources()
{
sampleSources.clear();
gr::top_block_sptr iq_tb = gr::make_top_block("multiply");
auto iq_mem_source = gr::blocks::memory_source::make(8);
auto iq_mem_sink = gr::blocks::memory_sink::make(8);
auto multiply = gr::blocks::multiply_const_cc::make(20);
if (selection) {
float centre = (selectionFreq.first + selectionFreq.second) / 2;
float cutoff = std::abs(selectionFreq.first - centre);
auto lp_taps = gr::filter::firdes::low_pass(1.0, 1.0, cutoff, cutoff / 2);
auto filter = gr::filter::freq_xlating_fir_filter_ccf::make(1, lp_taps, centre, 1.0);
iq_tb->connect(iq_mem_source, 0, filter, 0);
iq_tb->connect(filter, 0, multiply, 0);
iq_tb->connect(multiply, 0, iq_mem_sink, 0);
} else {
iq_tb->connect(iq_mem_source, 0, multiply, 0);
iq_tb->connect(multiply, 0, iq_mem_sink, 0);
}
gr::top_block_sptr quad_demod_tb = gr::make_top_block("quad_demod");
auto quad_demod_mem_source = gr::blocks::memory_source::make(8);
auto quad_demod_mem_sink = gr::blocks::memory_sink::make(4);
auto quad_demod = gr::analog::quadrature_demod_cf::make(5);
quad_demod_tb->connect(quad_demod_mem_source, 0, quad_demod, 0);
quad_demod_tb->connect(quad_demod, 0, quad_demod_mem_sink, 0);
sampleSources.emplace_back(new GRSampleBuffer<std::complex<float>, std::complex<float>>(mainSampleSource, iq_tb, iq_mem_source, iq_mem_sink));
sampleSources.emplace_back(new GRSampleBuffer<std::complex<float>, float>(dynamic_cast<SampleSource<std::complex<float>>*>(sampleSources[0].get()), quad_demod_tb, quad_demod_mem_source, quad_demod_mem_sink));
update();
}
void PlotView::inputSourceChanged(AbstractSampleSource *src)
{
auto derived = dynamic_cast<SampleSource<std::complex<float>>*>(src);
if (derived == nullptr)
throw new std::runtime_error("SampleSource doesn't provide correct type for GRSampleBuffer");
mainSampleSource = derived;
refreshSources();
}
void PlotView::viewChanged(off_t firstSample, off_t lastSample)
{
this->firstSample = firstSample;
this->lastSample = lastSample;
update();
}
void PlotView::selectionChanged(std::pair<off_t, off_t> selectionTime, std::pair<float, float> selectionFreq)
{
this->selectionTime = selectionTime;
this->selectionFreq = selectionFreq;
selection = true;
refreshSources();
}
void PlotView::selectionCleared()
{
selection = false;
refreshSources();
}
void PlotView::paintEvent(QPaintEvent *event)
{
if (lastSample - firstSample <= 0) return;
QRect rect = QRect(0, 0, width(), height());
QPainter painter(this);
painter.fillRect(rect, Qt::black);
// Split space equally between waveforms for now
int waveHeight = height() / sampleSources.size();
int wave = 0;
for (auto&& sampleSource : sampleSources) {
QRect waveRect = QRect(0, wave * waveHeight, width(), waveHeight);
off_t length = lastSample - firstSample;
if (auto src = dynamic_cast<SampleSource<std::complex<float>>*>(sampleSource.get())) {
auto samples = src->getSamples(firstSample, length);
painter.setPen(Qt::red);
plot(&painter, waveRect, reinterpret_cast<float*>(samples.get()), length, 2);
painter.setPen(Qt::blue);
plot(&painter, waveRect, reinterpret_cast<float*>(samples.get())+1, length, 2);
} else if (auto src = dynamic_cast<SampleSource<float>*>(sampleSource.get())) {
auto samples = src->getSamples(firstSample, length);
painter.setPen(Qt::green);
plot(&painter, waveRect, samples.get(), length, 1);
}
wave++;
}
}
void PlotView::plot(QPainter *painter, QRect &rect, float *samples, off_t count, int step = 1)
{
int xprev = 0;
int yprev = 0;
for (off_t i = 0; i < count; i++) {
float sample = samples[i*step];
int x = (float)i / count * rect.width();
int y = (sample * rect.height()/2) + rect.height()/2;
if (x < 0) x = 0;
if (y < 0) y = 0;
if (x >= rect.width()-1) x = rect.width()-2;
if (y >= rect.height()-1) y = rect.height()-2;
painter->drawLine(xprev + rect.x(), yprev + rect.y(), x + rect.x(), y + rect.y());
xprev = x;
yprev = y;
}
}