1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-12-23 18:15:45 -05:00
sdrangel/sdrgui/dsp/scopevisxy.cpp

166 lines
4.7 KiB
C++

///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2018 F4EXB //
// written by Edouard Griffiths //
// //
// 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 as 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 V3 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 <chrono>
#include <thread>
#include "scopevisxy.h"
#include "util/message.h"
#include "gui/tvscreen.h"
ScopeVisXY::ScopeVisXY(TVScreen *tvScreen) :
m_tvScreen(tvScreen),
m_scale(1.0),
m_cols(0),
m_rows(0),
m_pixelsPerFrame(480),
m_pixelCount(0),
m_alphaTrace(128),
m_alphaReset(128),
m_plotRGB(qRgb(0, 255, 0)),
m_gridRGB(qRgb(255, 255 ,255))
{
setObjectName("ScopeVisXY");
setPixelsPerFrame(m_pixelsPerFrame);
m_tvScreen->setAlphaBlend(true);
}
ScopeVisXY::~ScopeVisXY()
{
}
void ScopeVisXY::setPixelsPerFrame(int pixelsPerFrame)
{
m_pixelsPerFrame = pixelsPerFrame;
m_pixelCount = 0;
m_tvScreen->setAlphaReset();
}
void ScopeVisXY::feed(const SampleVector::const_iterator& cbegin, const SampleVector::const_iterator& end, bool positiveOnly)
{
(void) positiveOnly;
SampleVector::const_iterator begin(cbegin);
while (begin < end)
{
float x = m_scale * (begin->m_real/SDR_RX_SCALEF);
float y = m_scale * (begin->m_imag/SDR_RX_SCALEF);
int row = m_rows * ((1.0 - y) / 2.0);
int col = m_cols * ((1.0 + x) / 2.0);
row = row < 0 ? 0 : row >= m_rows ? m_rows-1 : row;
col = col < 0 ? 0 : col >= m_cols ? m_cols-1 : col;
m_tvScreen->selectRow(row);
m_tvScreen->setDataColor(col, qRed(m_plotRGB), qGreen(m_plotRGB), qBlue(m_plotRGB), m_alphaTrace);
++begin;
m_pixelCount++;
if (m_pixelCount == m_pixelsPerFrame)
{
int rows, cols;
m_tvScreen->getSize(rows, cols);
if ((rows != m_rows) || (cols != m_cols))
{
calculateGraticule(rows, cols);
m_rows = rows;
m_cols = cols;
}
m_tvScreen->renderImage(0);
m_tvScreen->update();
std::this_thread::sleep_for(std::chrono::microseconds(5000));
m_tvScreen->resetImage(m_alphaReset);
drawGraticule();
m_pixelCount = 0;
}
}
}
void ScopeVisXY::start()
{
}
void ScopeVisXY::stop()
{
}
void ScopeVisXY::handleInputMessages()
{
Message* message;
while ((message = m_inputMessageQueue.pop()) != 0)
{
if (handleMessage(*message)) {
delete message;
}
}
}
bool ScopeVisXY::handleMessage(const Message& message)
{
(void) message;
return false;
}
void ScopeVisXY::addGraticulePoint(const std::complex<float>& z) {
m_graticule.push_back(z);
}
void ScopeVisXY::clearGraticule() {
m_graticule.clear();
}
void ScopeVisXY::calculateGraticule(int rows, int cols)
{
m_graticuleRows.clear();
m_graticuleCols.clear();
std::vector<std::complex<float> >::const_iterator grIt = m_graticule.begin();
for (; grIt != m_graticule.end(); ++grIt)
{
int y = rows * ((1.0 - grIt->imag()) / 2.0);
int x = cols * ((1.0 + grIt->real()) / 2.0);
for (int d = -4; d <= 4; ++d)
{
m_graticuleRows.push_back(y+d);
m_graticuleCols.push_back(x);
m_graticuleRows.push_back(y);
m_graticuleCols.push_back(x+d);
}
}
}
void ScopeVisXY::drawGraticule()
{
std::vector<int>::const_iterator rowIt = m_graticuleRows.begin();
std::vector<int>::const_iterator colIt = m_graticuleCols.begin();
for(; (rowIt != m_graticuleRows.end()) && (colIt != m_graticuleCols.end()); ++rowIt, ++colIt)
{
m_tvScreen->selectRow(*rowIt);
m_tvScreen->setDataColor(*colIt, qRed(m_gridRGB), qGreen(m_gridRGB), qBlue(m_gridRGB));
}
}