mirror of https://github.com/f4exb/sdrangel.git
135 lines
4.5 KiB
C++
135 lines
4.5 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 <vector>
|
|
|
|
#include "leansdr/framework.h"
|
|
#include "gui/tvscreen.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
|
|
TVScreen *m_objDATVScreen;
|
|
pipereader<complex<T> > in;
|
|
unsigned long phase;
|
|
std::vector<int> cstln_rows;
|
|
std::vector<int> cstln_cols;
|
|
|
|
datvconstellation(scheduler *sch, pipebuf<complex<T> > &_in, T _xymin, T _xymax, const char *_name = 0, TVScreen *objDATVScreen = 0) :
|
|
runnable(sch, _name ? _name : _in.name),
|
|
xymin(_xymin),
|
|
xymax(_xymax),
|
|
decimation(DEFAULT_GUI_DECIMATION),
|
|
pixels_per_frame(1024),
|
|
cstln(0),
|
|
m_objDATVScreen(objDATVScreen),
|
|
in(_in),
|
|
phase(0)
|
|
{
|
|
}
|
|
|
|
void run()
|
|
{
|
|
//Symbols
|
|
|
|
while (in.readable() >= pixels_per_frame)
|
|
{
|
|
if ((!phase) && m_objDATVScreen)
|
|
{
|
|
m_objDATVScreen->resetImage();
|
|
|
|
complex<T> *p = in.rd(), *pend = p + pixels_per_frame;
|
|
|
|
for (; p < pend; ++p)
|
|
{
|
|
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
|
|
|
|
std::vector<int>::const_iterator row_it = cstln_rows.begin();
|
|
std::vector<int>::const_iterator col_it = cstln_cols.begin();
|
|
|
|
for (; (row_it != cstln_rows.end()) && (col_it != cstln_cols.end()); ++row_it, ++col_it)
|
|
{
|
|
m_objDATVScreen->selectRow(*row_it);
|
|
m_objDATVScreen->setDataColor(*col_it, 250, 250, 5);
|
|
}
|
|
}
|
|
|
|
m_objDATVScreen->renderImage(0);
|
|
}
|
|
|
|
in.read(pixels_per_frame);
|
|
|
|
if (++phase >= decimation)
|
|
{
|
|
phase = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
void draw_begin()
|
|
{
|
|
}
|
|
|
|
void calculate_cstln_points()
|
|
{
|
|
if (!(*cstln)) {
|
|
return;
|
|
}
|
|
|
|
cstln_rows.clear();
|
|
cstln_cols.clear();
|
|
|
|
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)
|
|
{
|
|
cstln_rows.push_back(x + d);
|
|
cstln_cols.push_back(y);
|
|
cstln_rows.push_back(x);
|
|
cstln_cols.push_back(y + d);
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif // DATVCONSTELLATION_H
|