sdrangel/plugins/channelrx/demoddatv/datvconstellation.h

129 wiersze
4.1 KiB
C++

///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2018 F4HKW //
// for F4EXB / SDRAngel //
// using LeanSDR Framework (C) 2016 F4DAV //
// //
// 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 //
// //
// 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/>. //
///////////////////////////////////////////////////////////////////////////////////
#ifndef DATVCONSTELLATION_H
#define DATVCONSTELLATION_H
#include <sys/time.h>
#include "leansdr/framework.h"
#include "datvscreen.h"
namespace leansdr
{
static const int DEFAULT_GUI_DECIMATION = 64;
template<typename T> struct datvconstellation : runnable
{
T xymin, xymax;
unsigned long decimation;
unsigned long pixels_per_frame;
cstln_lut<256> **cstln; // Optional ptr to optional constellation
DATVScreen *m_objDATVScreen;
datvconstellation(scheduler *sch, pipebuf< complex<T> > &_in, T _xymin, T _xymax, const char *_name=NULL, DATVScreen * objDATVScreen=NULL) :
runnable(sch, _name?_name:_in.name),
xymin(_xymin),
xymax(_xymax),
decimation(DEFAULT_GUI_DECIMATION),
pixels_per_frame(1024),
cstln(NULL),
in(_in),
phase(0),
m_objDATVScreen(objDATVScreen)
{
}
void run()
{
//Symbols
while ( in.readable() >= pixels_per_frame )
{
if ( ! phase )
{
m_objDATVScreen->resetImage();
complex<T> *p = in.rd(), *pend = p+pixels_per_frame;
for ( ; p<pend; ++p )
{
if(m_objDATVScreen!=NULL)
{
m_objDATVScreen->selectRow(256*(p->re-xymin)/(xymax-xymin));
m_objDATVScreen->setDataColor(256- 256*((p->im-xymin)/(xymax-xymin)),255,0,255);
}
}
if ( cstln && (*cstln) )
{
// Plot constellation points
for ( int i=0; i<(*cstln)->nsymbols; ++i )
{
complex<signed char> *p = &(*cstln)->symbols[i];
int x = 256*(p->re-xymin)/(xymax-xymin);
int y = 256 - 256*(p->im-xymin)/(xymax-xymin);
for ( int d=-4; d<=4; ++d )
{
m_objDATVScreen->selectRow(x+d);
m_objDATVScreen->setDataColor(y,5,250,250);
m_objDATVScreen->selectRow(x);
m_objDATVScreen->setDataColor(y+d,5,250,250);
}
}
}
m_objDATVScreen->renderImage(NULL);
}
in.read(pixels_per_frame);
if ( ++phase >= decimation )
{
phase = 0;
}
}
}
//private:
pipereader< complex<T> > in;
unsigned long phase;
//gfx g;
void draw_begin()
{
//g.clear();
//g.setfg(0, 255, 0);
//g.line(g.w/2,0, g.w/2, g.h);
//g.line(0,g.h/2, g.w,g.h/2);
}
};
}
#endif // DATVCONSTELLATION_H