mirror of
https://github.com/saitohirga/WSJT-X.git
synced 2024-10-31 15:47:10 -04:00
d91aed5aee
procedure; better window functions for some FFTs, resulting in better decoder performance; User-selectable colors for backgrounds of decoded messages. NB: more testing is desirable! git-svn-id: svn+ssh://svn.code.sf.net/p/wsjt/wsjt/branches/wsjtx@4951 ab8295b8-cf94-4d9e-aec4-7959e3be5d79
484 lines
12 KiB
C++
484 lines
12 KiB
C++
#include "plotter.h"
|
|
#include <math.h>
|
|
#include <QDebug>
|
|
|
|
#include "moc_plotter.cpp"
|
|
|
|
#define MAX_SCREENSIZE 2048
|
|
|
|
|
|
CPlotter::CPlotter(QWidget *parent) : //CPlotter Constructor
|
|
QFrame(parent)
|
|
{
|
|
setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
|
|
setFocusPolicy(Qt::StrongFocus);
|
|
setAttribute(Qt::WA_PaintOnScreen,false);
|
|
setAutoFillBackground(false);
|
|
setAttribute(Qt::WA_OpaquePaintEvent, false);
|
|
setAttribute(Qt::WA_NoSystemBackground, true);
|
|
|
|
m_startFreq = 0;
|
|
m_nSpan=2; //used for FFT bins/pixel
|
|
m_fSpan=(float)m_nSpan;
|
|
m_hdivs = HORZ_DIVS;
|
|
m_FreqUnits = 1;
|
|
m_Running = false;
|
|
m_paintEventBusy=false;
|
|
m_WaterfallPixmap = QPixmap(0,0);
|
|
m_2DPixmap = QPixmap(0,0);
|
|
m_ScalePixmap = QPixmap(0,0);
|
|
m_OverlayPixmap = QPixmap(0,0);
|
|
m_Size = QSize(0,0);
|
|
m_rxFreq = 1020;
|
|
m_line = 0;
|
|
m_fSample = 12000;
|
|
m_nsps=6912;
|
|
m_dBStepSize=10;
|
|
m_Percent2DScreen = 30; //percent of screen used for 2D display
|
|
m_txFreq=0;
|
|
}
|
|
|
|
CPlotter::~CPlotter() { } // Destructor
|
|
|
|
QSize CPlotter::minimumSizeHint() const
|
|
{
|
|
return QSize(50, 50);
|
|
}
|
|
|
|
QSize CPlotter::sizeHint() const
|
|
{
|
|
return QSize(180, 180);
|
|
}
|
|
|
|
void CPlotter::resizeEvent(QResizeEvent* ) //resizeEvent()
|
|
{
|
|
if(!size().isValid()) return;
|
|
if( m_Size != size() ) { //if changed, resize pixmaps to new screensize
|
|
m_Size = size();
|
|
m_w = m_Size.width();
|
|
m_h = m_Size.height();
|
|
m_h1 = (100-m_Percent2DScreen)*(m_Size.height())/100;
|
|
m_h2 = (m_Percent2DScreen)*(m_Size.height())/100;
|
|
|
|
m_2DPixmap = QPixmap(m_Size.width(), m_h2);
|
|
m_2DPixmap.fill(Qt::black);
|
|
m_WaterfallPixmap = QPixmap(m_Size.width(), m_h1);
|
|
m_OverlayPixmap = QPixmap(m_Size.width(), m_h2);
|
|
m_OverlayPixmap.fill(Qt::black);
|
|
|
|
m_WaterfallPixmap.fill(Qt::black);
|
|
m_2DPixmap.fill(Qt::black);
|
|
m_ScalePixmap = QPixmap(m_w,30);
|
|
m_ScalePixmap.fill(Qt::white);
|
|
}
|
|
DrawOverlay();
|
|
}
|
|
|
|
void CPlotter::paintEvent(QPaintEvent *) // paintEvent()
|
|
{
|
|
if(m_paintEventBusy) return;
|
|
m_paintEventBusy=true;
|
|
QPainter painter(this);
|
|
painter.drawPixmap(0,0,m_ScalePixmap);
|
|
painter.drawPixmap(0,30,m_WaterfallPixmap);
|
|
painter.drawPixmap(0,m_h1,m_2DPixmap);
|
|
m_paintEventBusy=false;
|
|
}
|
|
|
|
void CPlotter::draw(float swide[]) //draw()
|
|
{
|
|
int j,j0,y2;
|
|
float y;
|
|
|
|
double gain = pow(10.0,0.05*m_plotGain);
|
|
|
|
//move current data down one line (must do this before attaching a QPainter object)
|
|
m_WaterfallPixmap.scroll(0,1,0,0,m_w,m_h1);
|
|
QPainter painter1(&m_WaterfallPixmap);
|
|
m_2DPixmap = m_OverlayPixmap.copy(0,0,m_w,m_h2);
|
|
QPainter painter2D(&m_2DPixmap);
|
|
|
|
if(m_bLinearAvg) {
|
|
painter2D.setPen(Qt::yellow);
|
|
} else {
|
|
painter2D.setPen(Qt::green);
|
|
}
|
|
|
|
QPoint LineBuf[MAX_SCREENSIZE];
|
|
j=0;
|
|
j0=int(m_startFreq/m_fftBinWidth + 0.5);
|
|
int iz=XfromFreq(5000.0);
|
|
int jz=iz*m_binsPerPixel;
|
|
m_fMax=FreqfromX(iz);
|
|
|
|
flat4_(swide,&iz,&m_Flatten);
|
|
flat4_(&jt9com_.savg[j0],&jz,&m_Flatten);
|
|
|
|
for(int i=0; i<iz; i++) {
|
|
y=swide[i];
|
|
int y1 = 10.0*gain*y + 10*m_plotZero +40;
|
|
if (y1<0) y1=0;
|
|
if (y1>254) y1=254;
|
|
if (swide[i]>1.e29) y1=255;
|
|
painter1.setPen(m_ColorTbl[y1]);
|
|
painter1.drawPoint(i,0);
|
|
|
|
y2=0;
|
|
if(m_bCurrent) y2 = gain*y - 15;
|
|
|
|
if(m_bCumulative) {
|
|
float sum=0.0;
|
|
int j=j0+m_binsPerPixel*i;
|
|
for(int k=0; k<m_binsPerPixel; k++) {
|
|
sum+=jt9com_.savg[j++];
|
|
}
|
|
y2=1.3*gain*sum/m_binsPerPixel + m_plotZero - 15;
|
|
if(m_Flatten==0) y2 += 40;
|
|
}
|
|
|
|
if(m_bLinearAvg) {
|
|
float sum=0.0;
|
|
int j=j0+m_binsPerPixel*i;
|
|
for(int k=0; k<m_binsPerPixel; k++) {
|
|
sum+=jt9w_.syellow[j++];
|
|
}
|
|
y2=gain*sum/m_binsPerPixel * (m_h/50.0) - 20.0;
|
|
}
|
|
|
|
if(i==iz-1) painter2D.drawPolyline(LineBuf,j);
|
|
LineBuf[j].setX(i);
|
|
LineBuf[j].setY(m_h-(y2+0.8*m_h));
|
|
j++;
|
|
}
|
|
|
|
if(swide[0]>1.0e29) m_line=0;
|
|
m_line++;
|
|
if(m_line == 13) {
|
|
UTCstr();
|
|
painter1.setPen(Qt::white);
|
|
painter1.drawText(5,10,m_sutc);
|
|
}
|
|
update(); //trigger a new paintEvent
|
|
}
|
|
|
|
void CPlotter::UTCstr()
|
|
{
|
|
int ihr,imin;
|
|
if(jt9com_.ndiskdat != 0) {
|
|
ihr=jt9com_.nutc/100;
|
|
imin=jt9com_.nutc % 100;
|
|
} else {
|
|
qint64 ms = QDateTime::currentMSecsSinceEpoch() % 86400000;
|
|
imin=ms/60000;
|
|
ihr=imin/60;
|
|
imin=imin % 60;
|
|
imin=imin - (imin % (m_TRperiod/60));
|
|
}
|
|
sprintf(m_sutc,"%2.2d:%2.2d",ihr,imin);
|
|
}
|
|
|
|
void CPlotter::DrawOverlay() //DrawOverlay()
|
|
{
|
|
if(m_OverlayPixmap.isNull()) return;
|
|
if(m_WaterfallPixmap.isNull()) return;
|
|
int w = m_WaterfallPixmap.width();
|
|
int x,y,x1,x2;
|
|
// int nHzDiv[11]={0,50,100,200,200,200,500,500,500,500,500};
|
|
float pixperdiv;
|
|
|
|
double df = m_binsPerPixel*m_fftBinWidth;
|
|
QRect rect;
|
|
{
|
|
QPainter painter(&m_OverlayPixmap);
|
|
painter.initFrom(this);
|
|
QLinearGradient gradient(0, 0, 0 ,m_h2); //fill background with gradient
|
|
gradient.setColorAt(1, Qt::black);
|
|
gradient.setColorAt(0, Qt::darkBlue);
|
|
painter.setBrush(gradient);
|
|
painter.drawRect(0, 0, m_w, m_h2);
|
|
painter.setBrush(Qt::SolidPattern);
|
|
|
|
pixperdiv = m_freqPerDiv/df;
|
|
y = m_h2 - m_h2/VERT_DIVS;
|
|
m_hdivs = w*df/m_freqPerDiv + 1.9999;
|
|
float xx0=float(m_startFreq)/float(m_freqPerDiv);
|
|
xx0=xx0-int(xx0);
|
|
int x0=xx0*pixperdiv+0.5;
|
|
for( int i=1; i<m_hdivs; i++) { //draw vertical grids
|
|
x = (int)((float)i*pixperdiv ) - x0;
|
|
if(x >= 0 and x<=m_w) {
|
|
painter.setPen(QPen(Qt::white, 1,Qt::DotLine));
|
|
painter.drawLine(x, 0, x , y);
|
|
painter.drawLine(x, m_h2-5, x , m_h2);
|
|
}
|
|
}
|
|
|
|
pixperdiv = (float)m_h2 / (float)VERT_DIVS;
|
|
painter.setPen(QPen(Qt::white, 1,Qt::DotLine));
|
|
for( int i=1; i<VERT_DIVS; i++) { //draw horizontal grids
|
|
y = (int)( (float)i*pixperdiv );
|
|
painter.drawLine(0, y, w, y);
|
|
}
|
|
}
|
|
|
|
QRect rect0;
|
|
QPainter painter0(&m_ScalePixmap);
|
|
painter0.initFrom(this);
|
|
|
|
//create Font to use for scales
|
|
QFont Font("Arial");
|
|
Font.setPointSize(12);
|
|
QFontMetrics metrics(Font);
|
|
Font.setWeight(QFont::Normal);
|
|
painter0.setFont(Font);
|
|
painter0.setPen(Qt::black);
|
|
|
|
if(m_binsPerPixel < 1) m_binsPerPixel=1;
|
|
m_fSpan = w*df;
|
|
int n=m_fSpan/10;
|
|
m_freqPerDiv=10;
|
|
if(n>25) m_freqPerDiv=50;
|
|
if(n>70) m_freqPerDiv=100;
|
|
if(n>140) m_freqPerDiv=200;
|
|
if(n>310) m_freqPerDiv=500;
|
|
m_hdivs = w*df/m_freqPerDiv + 0.9999;
|
|
m_ScalePixmap.fill(Qt::white);
|
|
painter0.drawRect(0, 0, w, 30);
|
|
MakeFrequencyStrs();
|
|
|
|
//draw tick marks on upper scale
|
|
pixperdiv = m_freqPerDiv/df;
|
|
for( int i=0; i<m_hdivs; i++) { //major ticks
|
|
x = (int)((m_xOffset+i)*pixperdiv );
|
|
painter0.drawLine(x,18,x,30);
|
|
}
|
|
int minor=5;
|
|
if(m_freqPerDiv==200) minor=4;
|
|
for( int i=1; i<minor*m_hdivs; i++) { //minor ticks
|
|
x = i*pixperdiv/minor;
|
|
painter0.drawLine(x,24,x,30);
|
|
}
|
|
|
|
//draw frequency values
|
|
for( int i=0; i<=m_hdivs; i++) {
|
|
x = (int)((m_xOffset+i)*pixperdiv - pixperdiv/2);
|
|
rect0.setRect(x,0, (int)pixperdiv, 20);
|
|
painter0.drawText(rect0, Qt::AlignHCenter|Qt::AlignVCenter,m_HDivText[i]);
|
|
}
|
|
|
|
float bw=9.0*12000.0/m_nsps;
|
|
if(m_modeTx=="JT65") bw=66.0*11025.0/4096.0;
|
|
|
|
QPen pen0(Qt::green, 3); //Mark Rx Freq with green
|
|
painter0.setPen(pen0);
|
|
x1=XfromFreq(m_rxFreq);
|
|
x2=XfromFreq(m_rxFreq+bw);
|
|
painter0.drawLine(x1,24,x1,30);
|
|
painter0.drawLine(x1,28,x2,28);
|
|
painter0.drawLine(x2,24,x2,30);
|
|
|
|
if(m_mode=="JT9+JT65") {
|
|
QPen pen2(Qt::blue, 3); //Mark the JT65 | JT9 divider
|
|
painter0.setPen(pen2);
|
|
x1=XfromFreq(m_fMin);
|
|
if(x1<2) x1=2;
|
|
x2=x1+30;
|
|
painter0.drawLine(x1,8,x1,28);
|
|
}
|
|
|
|
QPen pen1(Qt::red, 3); //Mark Tx freq with red
|
|
painter0.setPen(pen1);
|
|
x1=XfromFreq(m_txFreq);
|
|
x2=XfromFreq(m_txFreq+bw);
|
|
painter0.drawLine(x1,17,x1,21);
|
|
painter0.drawLine(x1,17,x2,17);
|
|
painter0.drawLine(x2,17,x2,21);
|
|
|
|
if(m_dialFreq>10.13 and m_dialFreq< 10.15) {
|
|
float f1=1.0e6*(10.1401 - m_dialFreq);
|
|
float f2=f1+200.0;
|
|
x1=XfromFreq(f1);
|
|
x2=XfromFreq(f2);
|
|
if(x1<=m_w and x2>=0) {
|
|
QPen pen1(QColor(255,165,0),3); //Mark WSPR sub-band orange
|
|
painter0.setPen(pen1);
|
|
painter0.drawLine(x1,9,x2,9);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CPlotter::MakeFrequencyStrs() //MakeFrequencyStrs
|
|
{
|
|
int f=(m_startFreq+m_freqPerDiv-1)/m_freqPerDiv;
|
|
f*=m_freqPerDiv;
|
|
m_xOffset=float(f-m_startFreq)/m_freqPerDiv;
|
|
for(int i=0; i<=m_hdivs; i++) {
|
|
m_HDivText[i].setNum(f);
|
|
f+=m_freqPerDiv;
|
|
}
|
|
}
|
|
|
|
int CPlotter::XfromFreq(float f) //XfromFreq()
|
|
{
|
|
// float w = m_WaterfallPixmap.width();
|
|
int x = int(m_w * (f - m_startFreq)/m_fSpan + 0.5);
|
|
if(x<0 ) return 0;
|
|
if(x>m_w) return m_w;
|
|
return x;
|
|
}
|
|
|
|
float CPlotter::FreqfromX(int x) //FreqfromX()
|
|
{
|
|
return float(m_startFreq + x*m_binsPerPixel*m_fftBinWidth);
|
|
}
|
|
|
|
void CPlotter::SetRunningState(bool running) //SetRunningState()
|
|
{
|
|
m_Running = running;
|
|
}
|
|
|
|
void CPlotter::setPlotZero(int plotZero) //setPlotZero()
|
|
{
|
|
m_plotZero=plotZero;
|
|
}
|
|
|
|
int CPlotter::getPlotZero() //getPlotZero()
|
|
{
|
|
return m_plotZero;
|
|
}
|
|
|
|
void CPlotter::setPlotGain(int plotGain) //setPlotGain()
|
|
{
|
|
m_plotGain=plotGain;
|
|
}
|
|
|
|
int CPlotter::getPlotGain() //getPlotGain()
|
|
{
|
|
return m_plotGain;
|
|
}
|
|
|
|
void CPlotter::setStartFreq(int f) //SetStartFreq()
|
|
{
|
|
m_startFreq=f;
|
|
resizeEvent(NULL);
|
|
update();
|
|
}
|
|
|
|
int CPlotter::startFreq() //startFreq()
|
|
{
|
|
return m_startFreq;
|
|
}
|
|
|
|
int CPlotter::plotWidth(){return m_WaterfallPixmap.width();}
|
|
void CPlotter::UpdateOverlay() {DrawOverlay();}
|
|
void CPlotter::setDataFromDisk(bool b) {m_dataFromDisk=b;}
|
|
|
|
void CPlotter::setRxRange(int fMin)
|
|
{
|
|
m_fMin=fMin;
|
|
}
|
|
|
|
void CPlotter::setBinsPerPixel(int n) // set nbpp
|
|
{
|
|
m_binsPerPixel = n;
|
|
DrawOverlay(); //Redraw scales and ticks
|
|
update(); //trigger a new paintEvent}
|
|
}
|
|
|
|
int CPlotter::binsPerPixel() // get nbpp
|
|
{
|
|
return m_binsPerPixel;
|
|
}
|
|
|
|
void CPlotter::setRxFreq (int x)
|
|
{
|
|
m_rxFreq = x; // x is freq in Hz
|
|
DrawOverlay();
|
|
update();
|
|
}
|
|
|
|
int CPlotter::rxFreq() {return m_rxFreq;} //get rxFreq
|
|
|
|
void CPlotter::mousePressEvent(QMouseEvent *event) //mousePressEvent
|
|
{
|
|
int x=event->x();
|
|
if(x<0) x=0;
|
|
if(x>m_Size.width()) x=m_Size.width();
|
|
bool ctrl = (event->modifiers() & Qt::ControlModifier);
|
|
int freq = int(FreqfromX(x)+0.5);
|
|
int tx_freq = m_txFreq;
|
|
if (ctrl or m_lockTxFreq) tx_freq = freq;
|
|
|
|
emit setFreq1 (freq, tx_freq);
|
|
|
|
int n=1;
|
|
if(ctrl) n+=100;
|
|
emit freezeDecode1(n);
|
|
}
|
|
|
|
void CPlotter::mouseDoubleClickEvent(QMouseEvent *event) //mouse2click
|
|
{
|
|
bool ctrl = (event->modifiers() & Qt::ControlModifier);
|
|
int n=2;
|
|
if(ctrl) n+=100;
|
|
emit freezeDecode1(n);
|
|
}
|
|
|
|
void CPlotter::setNSpan(int n) //setNSpan()
|
|
{
|
|
m_nSpan=n;
|
|
}
|
|
|
|
double CPlotter::fGreen()
|
|
{
|
|
return m_fGreen;
|
|
}
|
|
|
|
void CPlotter::setNsps(int ntrperiod, int nsps) //setNSpan()
|
|
{
|
|
m_TRperiod=ntrperiod;
|
|
m_nsps=nsps;
|
|
m_fftBinWidth=1500.0/2048.0;
|
|
if(m_nsps==15360) m_fftBinWidth=1500.0/2048.0;
|
|
if(m_nsps==40960) m_fftBinWidth=1500.0/6144.0;
|
|
if(m_nsps==82944) m_fftBinWidth=1500.0/12288.0;
|
|
if(m_nsps==252000) m_fftBinWidth=1500.0/32768.0;
|
|
DrawOverlay(); //Redraw scales and ticks
|
|
update(); //trigger a new paintEvent}
|
|
}
|
|
|
|
void CPlotter::setTxFreq(int n) //setTol()
|
|
{
|
|
m_txFreq=n;
|
|
DrawOverlay();
|
|
update();
|
|
}
|
|
|
|
void CPlotter::setMode(QString mode)
|
|
{
|
|
m_mode=mode;
|
|
}
|
|
|
|
void CPlotter::setModeTx(QString modeTx)
|
|
{
|
|
m_modeTx=modeTx;
|
|
}
|
|
|
|
int CPlotter::getFmax()
|
|
{
|
|
return m_fMax;
|
|
}
|
|
|
|
void CPlotter::setDialFreq(double d)
|
|
{
|
|
m_dialFreq=d;
|
|
DrawOverlay();
|
|
update();
|
|
}
|
|
|
|
void CPlotter::setFlatten(bool b)
|
|
{
|
|
m_Flatten=0;
|
|
if(b) m_Flatten=1;
|
|
}
|