1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-12-23 18:15:45 -05:00
sdrangel/plugins/channelrx/heatmap/heatmapgui.cpp
2024-07-10 23:06:38 +02:00

1647 lines
55 KiB
C++

///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2023 Jon Beniston, M7RCE <jon@beniston.com> //
// Copyright (C) 2023 Mohamed <mohamedadlyi@github.com> //
// //
// 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 <limits>
#include <ctype.h>
#include <QMainWindow>
#include <QDebug>
#include <QMessageBox>
#include <QAction>
#include <QClipboard>
#include <QBuffer>
#include "heatmapgui.h"
#include "device/deviceuiset.h"
#include "device/deviceapi.h"
#include "dsp/dspengine.h"
#include "dsp/dspcommands.h"
#include "ui_heatmapgui.h"
#include "plugin/pluginapi.h"
#include "util/astronomy.h"
#include "util/csv.h"
#include "util/colormap.h"
#include "util/db.h"
#include "util/units.h"
#include "gui/basicchannelsettingsdialog.h"
#include "gui/dialpopup.h"
#include "gui/dialogpositioner.h"
#include "dsp/dspengine.h"
#include "maincore.h"
#include "heatmap.h"
#include "SWGMapItem.h"
HeatMapGUI* HeatMapGUI::create(PluginAPI* pluginAPI, DeviceUISet *deviceUISet, BasebandSampleSink *rxChannel)
{
HeatMapGUI* gui = new HeatMapGUI(pluginAPI, deviceUISet, rxChannel);
return gui;
}
void HeatMapGUI::destroy()
{
delete this;
}
void HeatMapGUI::resetToDefaults()
{
m_settings.resetToDefaults();
displaySettings();
applySettings(true);
}
QByteArray HeatMapGUI::serialize() const
{
return m_settings.serialize();
}
bool HeatMapGUI::deserialize(const QByteArray& data)
{
if(m_settings.deserialize(data)) {
displaySettings();
applySettings(true);
return true;
} else {
resetToDefaults();
return false;
}
}
bool HeatMapGUI::handleMessage(const Message& message)
{
if (HeatMap::MsgConfigureHeatMap::match(message))
{
qDebug("HeatMapGUI::handleMessage: HeatMap::MsgConfigureHeatMap");
const HeatMap::MsgConfigureHeatMap& cfg = (HeatMap::MsgConfigureHeatMap&) message;
m_settings = cfg.getSettings();
blockApplySettings(true);
ui->scopeGUI->updateSettings();
m_channelMarker.updateSettings(static_cast<const ChannelMarker*>(m_settings.m_channelMarker));
displaySettings();
blockApplySettings(false);
return true;
}
else if (DSPSignalNotification::match(message))
{
DSPSignalNotification& notif = (DSPSignalNotification&) message;
m_deviceCenterFrequency = notif.getCenterFrequency();
m_basebandSampleRate = notif.getSampleRate();
ui->deltaFrequency->setValueRange(false, 7, -m_basebandSampleRate/2, m_basebandSampleRate/2);
ui->deltaFrequencyLabel->setToolTip(tr("Range %1 %L2 Hz").arg(QChar(0xB1)).arg(m_basebandSampleRate/2));
updateAbsoluteCenterFrequency();
return true;
}
return false;
}
void HeatMapGUI::handleInputMessages()
{
Message* message;
while ((message = getInputMessageQueue()->pop()) != 0)
{
if (handleMessage(*message))
{
delete message;
}
}
}
void HeatMapGUI::channelMarkerChangedByCursor()
{
ui->deltaFrequency->setValue(m_channelMarker.getCenterFrequency());
m_settings.m_inputFrequencyOffset = m_channelMarker.getCenterFrequency();
applySettings();
}
void HeatMapGUI::channelMarkerHighlightedByCursor()
{
setHighlighted(m_channelMarker.getHighlighted());
}
void HeatMapGUI::on_deltaFrequency_changed(qint64 value)
{
m_channelMarker.setCenterFrequency(value);
m_settings.m_inputFrequencyOffset = m_channelMarker.getCenterFrequency();
updateAbsoluteCenterFrequency();
applySettings();
}
void HeatMapGUI::on_rfBW_valueChanged(int value)
{
float bw = value * 100.0f;
ui->rfBWText->setText(QString("%1k").arg(value / 10.0, 0, 'f', 1));
m_channelMarker.setBandwidth(bw);
m_settings.m_rfBandwidth = bw;
applySettings();
}
void HeatMapGUI::on_minPower_valueChanged(double value)
{
m_settings.m_minPower = (float)value;
plotMap();
if (m_powerYAxis) {
m_powerYAxis->setMin(m_settings.m_minPower);
}
applySettings();
}
void HeatMapGUI::on_maxPower_valueChanged(double value)
{
m_settings.m_maxPower = (float)value;
plotMap();
if (m_powerYAxis) {
m_powerYAxis->setMax(m_settings.m_maxPower);
}
applySettings();
}
void HeatMapGUI::on_colorMap_currentIndexChanged(int index)
{
if (index >= 0)
{
m_settings.m_colorMapName = ui->colorMap->currentText();
m_colorMap = ColorMap::getColorMap(m_settings.m_colorMapName);
}
plotMap();
applySettings();
}
void HeatMapGUI::on_pulseTH_valueChanged(int value)
{
m_settings.m_pulseThreshold = (float)value;
ui->pulseTHText->setText(QString::number(value));
applySettings();
}
const QStringList HeatMapGUI::m_averagePeriodTexts = {
"10us", "100us", "1ms", "10ms", "100ms", "1s", "10s"
};
void HeatMapGUI::on_averagePeriod_valueChanged(int value)
{
m_settings.m_averagePeriodUS = (int)std::pow(10.0f, (float)value);
ui->averagePeriodText->setText(m_averagePeriodTexts[value-1]);
applySettings();
}
const QStringList HeatMapGUI::m_sampleRateTexts = {
"100", "1k", "10k", "100k", "1M", "10M"
};
void HeatMapGUI::on_sampleRate_valueChanged(int value)
{
m_settings.m_sampleRate = (int)std::pow(10.0f, (float)value);
ui->sampleRateText->setText(m_sampleRateTexts[value-2]); // value range is [2,7]
ui->averagePeriod->setMinimum(std::max(1, static_cast<int> (m_averagePeriodTexts.size()) - value));
ui->rfBW->setMaximum(m_settings.m_sampleRate/100);
m_scopeVis->setLiveRate(m_settings.m_sampleRate);
applySettings();
}
void HeatMapGUI::on_mode_currentIndexChanged(int index)
{
if (index >= 0)
{
m_settings.m_mode = (HeatMapSettings::Mode)index;
bool none = m_settings.m_mode == HeatMapSettings::None;
ui->writeImage->setEnabled(!none);
ui->writeCSV->setEnabled(!none);
ui->readCSV->setEnabled(!none);
ui->colorMapLabel->setEnabled(!none);
ui->colorMap->setEnabled(!none);
if (none)
{
deleteFromMap();
}
else
{
if (m_image.isNull()) {
createImage(m_width, m_height);
}
plotMap();
}
applySettings();
}
}
void HeatMapGUI::displayPowerChart()
{
if (m_settings.m_displayChart)
{
ui->chartContainer->setVisible(true);
plotPowerVsTimeChart();
}
else
{
ui->chartContainer->setVisible(false);
QChart *emptyChart = new QChart();
emptyChart->setTheme(QChart::ChartThemeDark);
ui->powerChart->setChart(emptyChart); // Can't set to nullptr
delete m_powerChart;
m_powerChart = emptyChart;
m_powerAverageSeries = nullptr; // Deleted when chart deleted?
}
}
void HeatMapGUI::on_displayChart_clicked(bool checked)
{
m_settings.m_displayChart = checked;
displayPowerChart();
applySettings();
}
void HeatMapGUI::on_clearHeatMap_clicked()
{
// Clear heat map
m_heatMap->resetMagLevels();
clearPower();
plotMap();
if (m_powerAverageSeries)
{
m_powerAverageSeries->clear();
m_powerMaxPeakSeries->clear();
m_powerMinPeakSeries->clear();
m_powerPulseAverageSeries->clear();
m_powerPathLossSeries->clear();
}
}
bool HeatMapGUI::pixelValid(int x, int y) const
{
return (y >= 0) && (x >= 0) && (y < m_height) && (x < m_width);
}
void HeatMapGUI::coordsToPixel(double latitude, double longitude, int& x, int& y) const
{
y = m_height - (latitude - m_south) / m_degreesLatPerPixel;
x = (longitude - m_west) / m_degreesLonPerPixel;
}
void HeatMapGUI::pixelToCoords(int x, int y, double& latitude, double& longitude) const
{
latitude = m_north - y * m_degreesLatPerPixel;
longitude = m_west + x * m_degreesLonPerPixel;
}
void HeatMapGUI::on_writeCSV_clicked()
{
m_csvFileDialog.setAcceptMode(QFileDialog::AcceptSave);
m_csvFileDialog.setNameFilter("*.csv");
if (m_csvFileDialog.exec())
{
QStringList fileNames = m_csvFileDialog.selectedFiles();
if (fileNames.size() > 0)
{
QFile file(fileNames[0]);
if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
{
QMessageBox::critical(this, "Heat Map", QString("Failed to open file %1").arg(fileNames[0]));
return;
}
QTextStream out(&file);
out.setRealNumberPrecision(9); // https://bugreports.qt.io/browse/QTBUG-110775 - set to 9, so we get at least 6 fractional digits
out << "Latitude,Longitude," << ui->mode->currentText() << " Power (dB)\n";
float *power = getCurrentModePowerData();
for (int y = 0; y < m_height; y++)
{
for (int x = 0; x < m_width; x++)
{
float pow = power[y*m_width+x];
if (!std::isnan(pow))
{
double latitude, longitude;
pixelToCoords(x, y, latitude, longitude);
out << latitude << "," << longitude << "," << pow << "\n";
}
}
}
}
}
}
void HeatMapGUI::on_readCSV_clicked()
{
m_csvFileDialog.setAcceptMode(QFileDialog::AcceptOpen);
m_csvFileDialog.setNameFilter("*.csv");
if (m_csvFileDialog.exec())
{
QStringList fileNames = m_csvFileDialog.selectedFiles();
if (fileNames.size() > 0)
{
QFile file(fileNames[0]);
if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
{
QMessageBox::critical(this, "Heat Map", QString("Failed to open file %1").arg(fileNames[0]));
return;
}
QTextStream in(&file);
QString powerColName = ui->mode->currentText() + " Power (dB)";
QString error;
QStringList colNames = {"Latitude", "Longitude", powerColName};
QHash<QString, int> colIndexes = CSV::readHeader(in, colNames, error);
if (error.isEmpty())
{
clearPower(getCurrentModePowerData());
clearImage();
int latitudeCol = colIndexes.value("Latitude");
int longitudeCol = colIndexes.value("Longitude");
int powerCol = colIndexes.value(powerColName);
QStringList cols;
while (CSV::readRow(in, &cols))
{
if (cols.size() >= 3)
{
double latitude = cols[latitudeCol].toDouble();
double longitude = cols[longitudeCol].toDouble();
float power = cols[powerCol].toFloat();
updatePower(latitude, longitude, power);
}
}
}
else
{
QString actualColNames = colIndexes.keys().join(" ");
QString expectedColNames = colNames.join(" ");
QMessageBox::critical(this, "Heat Map", QString("Failed to read expected header in CSV file. %1 != %2").arg(actualColNames).arg(expectedColNames));
return;
}
}
}
}
void HeatMapGUI::on_writeImage_clicked()
{
m_imageFileDialog.setAcceptMode(QFileDialog::AcceptSave);
m_imageFileDialog.setNameFilter("*.png *.jpg *.jpeg *.bmp *.ppm *.xbm *.xpm");
if (m_imageFileDialog.exec())
{
QStringList fileNames = m_imageFileDialog.selectedFiles();
if (fileNames.size() > 0)
{
if (!m_image.save(fileNames[0]))
{
QMessageBox::critical(this, "Heat Map", QString("Failed to save image to %1").arg(fileNames[0]));
return;
}
}
}
}
void HeatMapGUI::displayTXPosition(bool enabled)
{
ui->txLatitudeLabel->setEnabled(enabled);
ui->txLatitude->setEnabled(enabled);
ui->txLongitude->setEnabled(enabled);
ui->txLongitudeLabel->setEnabled(enabled);
ui->txPower->setEnabled(enabled);
ui->txPowerLabel->setEnabled(enabled);
ui->txPowerUnits->setEnabled(enabled);
ui->txPositionSet->setEnabled(enabled);
ui->rangeLabel->setEnabled(enabled);
ui->range->setEnabled(enabled);
ui->rangeUnits->setEnabled(enabled);
ui->pathLossLabel->setEnabled(enabled);
ui->pathLoss->setEnabled(enabled);
ui->pathLossUnits->setEnabled(enabled);
if (enabled) {
sendTxToMap();
} else {
deleteTxFromMap();
}
}
void HeatMapGUI::on_txPosition_clicked(bool checked)
{
m_settings.m_txPosValid = checked;
displayTXPosition(checked);
applySettings();
}
void HeatMapGUI::on_txLatitude_editingFinished()
{
m_settings.m_txLatitude = ui->txLatitude->text().toFloat();
updateRange();
sendTxToMap();
applySettings();
}
void HeatMapGUI::on_txLongitude_editingFinished()
{
m_settings.m_txLongitude = ui->txLongitude->text().toFloat();
updateRange();
sendTxToMap();
applySettings();
}
void HeatMapGUI::on_txPower_valueChanged(double value)
{
m_settings.m_txPower = (float)value;
sendTxToMap();
applySettings();
}
void HeatMapGUI::on_txPositionSet_clicked(bool checked)
{
(void) checked;
ui->txLatitude->setText(QString::number(m_latitude));
ui->txLongitude->setText(QString::number(m_longitude));
m_settings.m_txLatitude = m_latitude;
m_settings.m_txLongitude = m_longitude;
updateRange();
sendTxToMap();
applySettings();
}
void HeatMapGUI::onWidgetRolled(QWidget* widget, bool rollDown)
{
(void) widget;
(void) rollDown;
getRollupContents()->saveState(m_rollupState);
applySettings();
}
void HeatMapGUI::onMenuDialogCalled(const QPoint &p)
{
if (m_contextMenuType == ContextMenuChannelSettings)
{
BasicChannelSettingsDialog dialog(&m_channelMarker, this);
dialog.setUseReverseAPI(m_settings.m_useReverseAPI);
dialog.setReverseAPIAddress(m_settings.m_reverseAPIAddress);
dialog.setReverseAPIPort(m_settings.m_reverseAPIPort);
dialog.setReverseAPIDeviceIndex(m_settings.m_reverseAPIDeviceIndex);
dialog.setReverseAPIChannelIndex(m_settings.m_reverseAPIChannelIndex);
dialog.setDefaultTitle(m_displayedName);
if (m_deviceUISet->m_deviceMIMOEngine)
{
dialog.setNumberOfStreams(m_heatMap->getNumberOfDeviceStreams());
dialog.setStreamIndex(m_settings.m_streamIndex);
}
dialog.move(p);
new DialogPositioner(&dialog, true);
dialog.exec();
m_settings.m_rgbColor = m_channelMarker.getColor().rgb();
m_settings.m_title = m_channelMarker.getTitle();
m_settings.m_useReverseAPI = dialog.useReverseAPI();
m_settings.m_reverseAPIAddress = dialog.getReverseAPIAddress();
m_settings.m_reverseAPIPort = dialog.getReverseAPIPort();
m_settings.m_reverseAPIDeviceIndex = dialog.getReverseAPIDeviceIndex();
m_settings.m_reverseAPIChannelIndex = dialog.getReverseAPIChannelIndex();
setWindowTitle(m_settings.m_title);
setTitle(m_channelMarker.getTitle());
setTitleColor(m_settings.m_rgbColor);
if (m_deviceUISet->m_deviceMIMOEngine)
{
m_settings.m_streamIndex = dialog.getSelectedStreamIndex();
m_channelMarker.clearStreamIndexes();
m_channelMarker.addStreamIndex(m_settings.m_streamIndex);
updateIndexLabel();
}
applySettings();
}
resetContextMenuType();
}
HeatMapGUI::HeatMapGUI(PluginAPI* pluginAPI, DeviceUISet *deviceUISet, BasebandSampleSink *rxChannel, QWidget* parent) :
ChannelGUI(parent),
ui(new Ui::HeatMapGUI),
m_pluginAPI(pluginAPI),
m_deviceUISet(deviceUISet),
m_channelMarker(this),
m_deviceCenterFrequency(0),
m_basebandSampleRate(1),
m_doApplySettings(true),
m_tickCount(0),
m_powerChart(nullptr),
m_powerAverageSeries(nullptr),
m_powerMaxPeakSeries(nullptr),
m_powerMinPeakSeries(nullptr),
m_powerPulseAverageSeries(nullptr),
m_powerPathLossSeries(nullptr)
{
setAttribute(Qt::WA_DeleteOnClose, true);
m_helpURL = "plugins/channelrx/heatmap/readme.md";
RollupContents *rollupContents = getRollupContents();
ui->setupUi(rollupContents);
setSizePolicy(rollupContents->sizePolicy());
rollupContents->arrangeRollups();
connect(rollupContents, SIGNAL(widgetRolled(QWidget*,bool)), this, SLOT(onWidgetRolled(QWidget*,bool)));
connect(this, SIGNAL(customContextMenuRequested(const QPoint &)), this, SLOT(onMenuDialogCalled(const QPoint &)));
m_heatMap = reinterpret_cast<HeatMap*>(rxChannel);
m_heatMap->setMessageQueueToGUI(getInputMessageQueue());
#if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
// Disable 256MB limit on image size
QImageReader::setAllocationLimit(0);
#endif
connect(&MainCore::instance()->getMasterTimer(), SIGNAL(timeout()), this, SLOT(tick())); // 50 ms
m_scopeVis = m_heatMap->getScopeSink();
m_scopeVis->setGLScope(ui->glScope);
ui->glScope->connectTimer(MainCore::instance()->getMasterTimer());
ui->scopeGUI->setBuddies(m_scopeVis->getInputMessageQueue(), m_scopeVis, ui->glScope);
// Scope settings to display magdB
ui->scopeGUI->setPreTrigger(1);
GLScopeSettings::TraceData traceDataI, traceDataQ;
traceDataI.m_projectionType = Projector::ProjectionMagDB;
traceDataI.m_amp = 1.0; // for 0 -100 dB
traceDataI.m_ofs = 0.0;
ui->scopeGUI->changeTrace(0, traceDataI);
ui->scopeGUI->setDisplayMode(GLScopeSettings::DisplayX);
ui->scopeGUI->focusOnTrace(0); // re-focus to take changes into account in the GUI
GLScopeSettings::TriggerData triggerData;
triggerData.m_triggerLevel = 0.1;
triggerData.m_triggerLevelCoarse = 10;
triggerData.m_triggerPositiveEdge = true;
ui->scopeGUI->changeTrigger(0, triggerData);
ui->scopeGUI->focusOnTrigger(0); // re-focus to take changes into account in the GUI
m_scopeVis->setLiveRate(m_settings.m_sampleRate);
ui->deltaFrequencyLabel->setText(QString("%1f").arg(QChar(0x94, 0x03)));
ui->deltaFrequency->setColorMapper(ColorMapper(ColorMapper::GrayGold));
ui->deltaFrequency->setValueRange(false, 7, -9999999, 9999999);
ui->channelPowerMeter->setColorTheme(LevelMeterSignalDB::ColorGreenAndBlue);
m_channelMarker.blockSignals(true);
m_channelMarker.setColor(Qt::yellow);
m_channelMarker.setBandwidth(m_settings.m_rfBandwidth);
m_channelMarker.setCenterFrequency(m_settings.m_inputFrequencyOffset);
m_channelMarker.setTitle("Heat Map");
m_channelMarker.blockSignals(false);
m_channelMarker.setVisible(true); // activate signal on the last setting only
setTitleColor(m_channelMarker.getColor());
m_settings.setChannelMarker(&m_channelMarker);
m_settings.setScopeGUI(ui->scopeGUI);
m_settings.setRollupState(&m_rollupState);
m_deviceUISet->addChannelMarker(&m_channelMarker);
connect(&m_channelMarker, SIGNAL(changedByCursor()), this, SLOT(channelMarkerChangedByCursor()));
connect(&m_channelMarker, SIGNAL(highlightedByCursor()), this, SLOT(channelMarkerHighlightedByCursor()));
connect(getInputMessageQueue(), SIGNAL(messageEnqueued()), this, SLOT(handleInputMessages()));
m_latitude = MainCore::instance()->getSettings().getLatitude();
m_longitude = MainCore::instance()->getSettings().getLongitude();
m_altitude = MainCore::instance()->getSettings().getAltitude();
ui->latitude->setText(QString::number(m_latitude));
ui->longitude->setText(QString::number(m_longitude));
QStringList colorMapNames = ColorMap::getColorMapNames();
for (const auto& color : colorMapNames) {
ui->colorMap->addItem(color);
}
m_colorMap = ColorMap::getColorMap(m_settings.m_colorMapName);
m_pen.setColor(Qt::black);
m_painter.setPen(m_pen);
createMap();
// Get updated when position changes
connect(&MainCore::instance()->getSettings(), &MainSettings::preferenceChanged, this, &HeatMapGUI::preferenceChanged);
ui->scopeContainer->setVisible(false);
displaySettings();
makeUIConnections();
applySettings(true);
DialPopup::addPopupsToChildDials(this);
m_resizer.enableChildMouseTracking();
plotPowerVsTimeChart();
}
HeatMapGUI::~HeatMapGUI()
{
disconnect(&MainCore::instance()->getMasterTimer(), SIGNAL(timeout()), this, SLOT(tick()));
deleteFromMap();
deleteTxFromMap();
deleteMap();
delete ui;
}
void HeatMapGUI::blockApplySettings(bool block)
{
m_doApplySettings = !block;
}
void HeatMapGUI::applySettings(bool force)
{
if (m_doApplySettings)
{
HeatMap::MsgConfigureHeatMap* message = HeatMap::MsgConfigureHeatMap::create(m_settings, force);
m_heatMap->getInputMessageQueue()->push(message);
}
}
void HeatMapGUI::displaySettings()
{
m_channelMarker.blockSignals(true);
m_channelMarker.setBandwidth(m_settings.m_rfBandwidth);
m_channelMarker.setCenterFrequency(m_settings.m_inputFrequencyOffset);
m_channelMarker.setTitle(m_settings.m_title);
m_channelMarker.blockSignals(false);
m_channelMarker.setColor(m_settings.m_rgbColor); // activate signal on the last setting only
setTitleColor(m_settings.m_rgbColor);
setWindowTitle(m_channelMarker.getTitle());
setTitle(m_channelMarker.getTitle());
blockApplySettings(true);
ui->deltaFrequency->setValue(m_channelMarker.getCenterFrequency());
ui->rfBWText->setText(QString("%1k").arg(m_settings.m_rfBandwidth / 1000.0, 0, 'f', 1));
ui->rfBW->setValue(m_settings.m_rfBandwidth / 100.0);
ui->minPower->setValue(m_settings.m_minPower);
ui->maxPower->setValue(m_settings.m_maxPower);
ui->colorMap->setCurrentText(m_settings.m_colorMapName);
m_colorMap = ColorMap::getColorMap(m_settings.m_colorMapName);
ui->mode->setCurrentIndex((int)m_settings.m_mode);
ui->pulseTH->setValue(m_settings.m_pulseThreshold);
ui->pulseTHText->setText(QString::number((int)m_settings.m_pulseThreshold));
int value = (int)std::log10(m_settings.m_averagePeriodUS);
ui->averagePeriod->setValue(value);
ui->averagePeriodText->setText(m_averagePeriodTexts[value-1]);
value = (int)std::log10(m_settings.m_sampleRate);
ui->sampleRate->setValue(value);
int idx = std::min(std::max(0, value-2), (int)m_sampleRateTexts.size() - 1);
ui->sampleRateText->setText(m_sampleRateTexts[idx]);
ui->averagePeriod->setMinimum(std::max(1, static_cast<int> (m_averagePeriodTexts.size()) - value));
ui->txPosition->setChecked(m_settings.m_txPosValid);
displayTXPosition(m_settings.m_txPosValid);
ui->txLatitude->setText(QString::number(m_settings.m_txLatitude));
ui->txLongitude->setText(QString::number(m_settings.m_txLongitude));
ui->txPower->setValue(m_settings.m_txPower);
ui->displayChart->setChecked(m_settings.m_displayChart);
displayPowerChart();
ui->displayAverage->setChecked(m_settings.m_displayAverage);
ui->displayMax->setChecked(m_settings.m_displayMax);
ui->displayMin->setChecked(m_settings.m_displayMin);
ui->displayPulseAverage->setChecked(m_settings.m_displayPulseAverage);
ui->displayPathLoss->setChecked(m_settings.m_displayPathLoss);
ui->displayMins->setValue(m_settings.m_displayMins);
ui->recordAverage->setChecked(m_settings.m_recordAverage);
ui->recordMax->setChecked(m_settings.m_recordMax);
ui->recordMin->setChecked(m_settings.m_recordMin);
ui->recordPulseAverage->setChecked(m_settings.m_recordPulseAverage);
ui->recordPathLoss->setChecked(m_settings.m_recordPathLoss);
m_scopeVis->setLiveRate(m_settings.m_sampleRate);
updateIndexLabel();
updateRange();
getRollupContents()->restoreState(m_rollupState);
updateAbsoluteCenterFrequency();
blockApplySettings(false);
}
void HeatMapGUI::leaveEvent(QEvent* event)
{
m_channelMarker.setHighlighted(false);
ChannelGUI::leaveEvent(event);
}
void HeatMapGUI::enterEvent(EnterEventType* event)
{
m_channelMarker.setHighlighted(true);
ChannelGUI::enterEvent(event);
}
void HeatMapGUI::tick()
{
// Update power meter widget
double magsqAvg, magsqpeak;
int nbMagsqSamples;
m_heatMap->getMagSqLevels(magsqAvg, magsqpeak, nbMagsqSamples);
double powDbAvg = CalcDb::dbPower(magsqAvg);
double powDbMaxPeak = CalcDb::dbPower(magsqpeak);
ui->channelPowerMeter->levelChanged(
(100.0f + powDbAvg) / 100.0f,
(100.0f + powDbMaxPeak) / 100.0f,
nbMagsqSamples);
if (m_tickCount % 4 == 0) {
ui->channelPower->setText(QString::number(powDbAvg, 'f', 1));
}
// Get power measurements and plot on map
if ((m_y >= 0) && (m_x >= 0) && (m_y < m_height) && (m_x < m_width))
{
double magAvg, magPulseAvg, magMaxPeak, magMinPeak;
m_heatMap->getMagLevels(magAvg, magPulseAvg, magMaxPeak, magMinPeak);
double powDbPulseAvg, powDbMinPeak, powDbPathLoss;
powDbAvg = std::numeric_limits<double>::quiet_NaN();
powDbPulseAvg = std::numeric_limits<double>::quiet_NaN();
powDbMaxPeak = std::numeric_limits<double>::quiet_NaN();
powDbMinPeak = std::numeric_limits<double>::quiet_NaN();
powDbPathLoss = std::numeric_limits<double>::quiet_NaN();
int idx = m_y * m_width + m_x;
if (!std::isnan(magAvg))
{
powDbAvg = CalcDb::dbPower(magAvg * magAvg);
if (m_powerAverage) {
m_powerAverage[idx] = powDbAvg;
}
if (m_tickCount % 4 == 0) {
ui->average->setText(QString::number(powDbAvg, 'f', 1));
}
}
else
{
ui->average->setText("");
}
if (!std::isnan(magPulseAvg))
{
powDbPulseAvg = CalcDb::dbPower(magPulseAvg * magPulseAvg);
if (m_powerPulseAverage) {
m_powerPulseAverage[idx] = powDbPulseAvg;
}
if (m_tickCount % 4 == 0) {
ui->pulseAverage->setText(QString::number(powDbPulseAvg, 'f', 1));
}
}
else
{
ui->pulseAverage->setText("");
}
if (magMaxPeak != -std::numeric_limits<double>::max())
{
powDbMaxPeak = CalcDb::dbPower(magMaxPeak * magMaxPeak);
if (m_powerMaxPeak) {
m_powerMaxPeak[idx] = powDbMaxPeak;
}
if (m_tickCount % 4 == 0) {
ui->maxPeak->setText(QString::number(powDbMaxPeak, 'f', 1));
}
}
else
{
ui->maxPeak->setText("");
}
if (magMinPeak != std::numeric_limits<double>::max())
{
powDbMinPeak = CalcDb::dbPower(magMinPeak * magMinPeak);
if (m_powerMinPeak) {
m_powerMinPeak[idx] = powDbMinPeak;
}
if (m_tickCount % 4 == 0) {
ui->minPeak->setText(QString::number(powDbMinPeak, 'f', 1));
}
}
else
{
ui->minPeak->setText("");
}
double range = calcRange(m_latitude, m_longitude);
double frequency = m_deviceCenterFrequency + m_settings.m_inputFrequencyOffset;
powDbPathLoss = m_settings.m_txPower - calcFreeSpacePathLoss(range, frequency);
if (m_powerPathLoss) {
m_powerPathLoss[idx] = powDbPathLoss;
}
if (m_heatMap->getDeviceAPI()->state(0) == DeviceAPI::StRunning)
{
addToPowerSeries(QDateTime::currentDateTime(), powDbAvg, powDbPulseAvg, powDbMaxPeak, powDbMinPeak, powDbPathLoss);
if (m_settings.m_mode != HeatMapSettings::None)
{
// Plot newest measurement on map
float *power = getCurrentModePowerData();
if (power)
{
double powDb = power[idx];
if (!std::isnan(powDb)) {
plotPixel(m_x, m_y, powDb);
}
}
}
if (m_tickCount % 15 == 0)
{
trimPowerSeries(QDateTime::currentDateTime().addSecs(-10*60));
// Updating axis range causes chart to be redrawn, so don't call for every sample
updateAxis();
}
}
}
if (m_tickCount % 25 == 0) {
sendToMap();
}
m_tickCount++;
}
float *HeatMapGUI::getCurrentModePowerData()
{
switch (m_settings.m_mode)
{
case HeatMapSettings::None:
case HeatMapSettings::Average:
return m_powerAverage;
case HeatMapSettings::PulseAverage:
return m_powerPulseAverage;
case HeatMapSettings::Max:
return m_powerMaxPeak;
case HeatMapSettings::Min:
return m_powerMinPeak;
case HeatMapSettings::PathLoss:
return m_powerPathLoss;
default:
return nullptr;
}
}
void HeatMapGUI::updatePower(double latitude, double longitude, float power)
{
int x, y;
coordsToPixel(latitude, longitude, x, y);
if (!pixelValid(x, y))
{
resizeMap(x, y);
coordsToPixel(latitude, longitude, x, y);
}
float *powerArray = getCurrentModePowerData();
powerArray[y*m_width+x] = power;
plotPixel(x, y, power);
}
void HeatMapGUI::plotMap()
{
if ((m_settings.m_mode != HeatMapSettings::None) && !m_image.isNull())
{
clearImage();
float *data = getCurrentModePowerData();
if (data) {
plotMap(data);
}
sendToMap();
}
}
void HeatMapGUI::clearPower()
{
clearPower(m_powerAverage);
clearPower(m_powerPulseAverage);
clearPower(m_powerMaxPeak);
clearPower(m_powerMinPeak);
clearPower(m_powerPathLoss);
}
void HeatMapGUI::clearPower(float *power)
{
clearPower(power, m_width * m_height);
}
void HeatMapGUI::clearPower(float *power, int size)
{
if (power) {
std::fill_n(power, size, std::numeric_limits<float>::quiet_NaN());
}
}
void HeatMapGUI::createImage(int width, int height)
{
if (!m_image.isNull()) {
m_painter.end();
}
try
{
if (m_settings.m_mode != HeatMapSettings::None)
{
qDebug() << "HeatMapGUI::createImage" << width << "*" << height;
m_image = QImage(width, height, QImage::Format_ARGB32);
m_painter.begin(&m_image);
}
else
{
m_image = QImage();
}
}
catch (std::bad_alloc&)
{
m_image = QImage();
QMessageBox::critical(this, "Heat Map", QString("Failed to allocate memory (width=%1 height=%2)").arg(m_width).arg(m_height));
}
}
void HeatMapGUI::clearImage()
{
m_image.fill(Qt::transparent);
//m_pen.setColor(Qt::black);
//m_painter.drawEllipse(QPoint(m_width/2, m_height/2), m_width/2-1, m_height/2-1);
}
void HeatMapGUI::plotMap(float *power)
{
for (int y = 0; y < m_height; y++)
{
for (int x = 0; x < m_width; x++)
{
float pow = power[y * m_width + x];
if (!std::isnan(pow)) {
plotPixel(x, y, pow);
}
}
}
}
void HeatMapGUI::plotPixel(int x, int y, double power)
{
if (m_image.isNull()) {
return;
}
// Normalise to [0,1]
float powNorm = (power - m_settings.m_minPower) / (m_settings.m_maxPower - m_settings.m_minPower);
if (powNorm < 0) {
return; // Don't plot below min, so we can easily see where we're below the min limit
}
powNorm = std::max(0.0f, powNorm);
powNorm = std::min(1.0f, powNorm);
int index = std::round(powNorm * 255);
QColor color = QColor::fromRgbF(m_colorMap[index*3], m_colorMap[index*3+1], m_colorMap[index*3+2]);
m_pen.setColor(color);
m_painter.setPen(m_pen);
m_painter.drawPoint(QPoint(x, y));
}
void HeatMapGUI::updateRange()
{
if (m_settings.m_txPosValid)
{
// Calculate range
qreal range = calcRange(m_latitude, m_longitude);
if (range < 1000)
{
ui->range->setText(QString::number(std::round(range)));
ui->rangeUnits->setText("m");
}
else
{
ui->range->setText(QString::number(range / 1000.0, 'f', 1));
ui->rangeUnits->setText("km");
}
double frequency = m_deviceCenterFrequency + m_settings.m_inputFrequencyOffset;
double loss = calcFreeSpacePathLoss(range, frequency);
ui->pathLoss->setText(QString::number(loss, 'f', 1));
}
else
{
ui->range->setText("");
ui->pathLoss->setText("");
}
}
qreal HeatMapGUI::calcRange(double latitude, double longitude)
{
QGeoCoordinate pos(latitude, longitude);
QGeoCoordinate tx(m_settings.m_txLatitude, m_settings.m_txLongitude);
return tx.distanceTo(pos);
}
double HeatMapGUI::calcFreeSpacePathLoss(double range, double frequency)
{
// In dB
if ((range == 0.0) || (frequency == 0.0)) {
return 0.0;
} else {
return 20.0 * log10(range) + 20 * log10(frequency) + 20 * log10(4.0 * M_PI / Astronomy::m_speedOfLight);
}
}
void HeatMapGUI::makeUIConnections()
{
QObject::connect(ui->deltaFrequency, &ValueDialZ::changed, this, &HeatMapGUI::on_deltaFrequency_changed);
QObject::connect(ui->rfBW, &QSlider::valueChanged, this, &HeatMapGUI::on_rfBW_valueChanged);
QObject::connect(ui->minPower, QOverload<double>::of(&QDoubleSpinBox::valueChanged), this, &HeatMapGUI::on_minPower_valueChanged);
QObject::connect(ui->maxPower, QOverload<double>::of(&QDoubleSpinBox::valueChanged), this, &HeatMapGUI::on_maxPower_valueChanged);
QObject::connect(ui->colorMap, QOverload<int>::of(&QComboBox::currentIndexChanged), this, &HeatMapGUI::on_colorMap_currentIndexChanged);
QObject::connect(ui->pulseTH, QOverload<int>::of(&QDial::valueChanged), this, &HeatMapGUI::on_pulseTH_valueChanged);
QObject::connect(ui->averagePeriod, QOverload<int>::of(&QDial::valueChanged), this, &HeatMapGUI::on_averagePeriod_valueChanged);
QObject::connect(ui->sampleRate, QOverload<int>::of(&QDial::valueChanged), this, &HeatMapGUI::on_sampleRate_valueChanged);
QObject::connect(ui->mode, QOverload<int>::of(&QComboBox::currentIndexChanged), this, &HeatMapGUI::on_mode_currentIndexChanged);
QObject::connect(ui->displayChart, &QPushButton::clicked, this, &HeatMapGUI::on_displayChart_clicked);
QObject::connect(ui->clearHeatMap, &QPushButton::clicked, this, &HeatMapGUI::on_clearHeatMap_clicked);
QObject::connect(ui->writeCSV, &QPushButton::clicked, this, &HeatMapGUI::on_writeCSV_clicked);
QObject::connect(ui->readCSV, &QPushButton::clicked, this, &HeatMapGUI::on_readCSV_clicked);
QObject::connect(ui->writeImage, &QPushButton::clicked, this, &HeatMapGUI::on_writeImage_clicked);
QObject::connect(ui->txPosition, &QPushButton::clicked, this, &HeatMapGUI::on_txPosition_clicked);
QObject::connect(ui->txLatitude, &QLineEdit::editingFinished, this, &HeatMapGUI::on_txLatitude_editingFinished);
QObject::connect(ui->txLongitude, &QLineEdit::editingFinished, this, &HeatMapGUI::on_txLongitude_editingFinished);
QObject::connect(ui->txPower, QOverload<double>::of(&QDoubleSpinBox::valueChanged), this, &HeatMapGUI::on_txPower_valueChanged);
QObject::connect(ui->txPositionSet, &QPushButton::clicked, this, &HeatMapGUI::on_txPositionSet_clicked);
QObject::connect(ui->displayAverage, &QCheckBox::clicked, this, &HeatMapGUI::on_displayAverage_clicked);
QObject::connect(ui->displayMax, &QCheckBox::clicked, this, &HeatMapGUI::on_displayMax_clicked);
QObject::connect(ui->displayMin, &QCheckBox::clicked, this, &HeatMapGUI::on_displayMin_clicked);
QObject::connect(ui->displayPulseAverage, &QCheckBox::clicked, this, &HeatMapGUI::on_displayPulseAverage_clicked);
QObject::connect(ui->displayPathLoss, &QCheckBox::clicked, this, &HeatMapGUI::on_displayPathLoss_clicked);
QObject::connect(ui->displayMins, QOverload<int>::of(&QSpinBox::valueChanged), this, &HeatMapGUI::on_displayMins_valueChanged);
QObject::connect(ui->recordAverage, &QCheckBox::clicked, this, &HeatMapGUI::on_recordAverage_clicked);
QObject::connect(ui->recordMax, &QCheckBox::clicked, this, &HeatMapGUI::on_recordMax_clicked);
QObject::connect(ui->recordMin, &QCheckBox::clicked, this, &HeatMapGUI::on_recordMin_clicked);
QObject::connect(ui->recordPulseAverage, &QCheckBox::clicked, this, &HeatMapGUI::on_recordPulseAverage_clicked);
QObject::connect(ui->recordPathLoss, &QCheckBox::clicked, this, &HeatMapGUI::on_recordPathLoss_clicked);
}
void HeatMapGUI::updateAbsoluteCenterFrequency()
{
setStatusFrequency(m_deviceCenterFrequency + m_settings.m_inputFrequencyOffset);
}
void HeatMapGUI::sendToMap()
{
if (m_settings.m_mode != HeatMapSettings::None)
{
// Send to Map feature
QList<ObjectPipe*> mapPipes;
MainCore::instance()->getMessagePipes().getMessagePipes(m_heatMap, "mapitems", mapPipes);
if (mapPipes.size() > 0)
{
// Encode image as base64 PNG
QByteArray ba;
QBuffer buffer(&ba);
buffer.open(QIODevice::WriteOnly);
m_image.save(&buffer, "PNG");
QByteArray data = ba.toBase64();
for (const auto& pipe : mapPipes)
{
MessageQueue *messageQueue = qobject_cast<MessageQueue*>(pipe->m_element);
SWGSDRangel::SWGMapItem *swgMapItem = new SWGSDRangel::SWGMapItem();
swgMapItem->setName(new QString("Heat Map"));
swgMapItem->setImage(new QString(data));
swgMapItem->setAltitude(0);
swgMapItem->setType(1);
swgMapItem->setImageTileEast(m_east);
swgMapItem->setImageTileWest(m_west);
swgMapItem->setImageTileNorth(m_north);
swgMapItem->setImageTileSouth(m_south);
swgMapItem->setImageZoomLevel((float)m_zoomLevel);
MainCore::MsgMapItem *msg = MainCore::MsgMapItem::create(m_heatMap, swgMapItem);
messageQueue->push(msg);
}
}
}
}
void HeatMapGUI::deleteFromMap()
{
QList<ObjectPipe*> mapPipes;
MainCore::instance()->getMessagePipes().getMessagePipes(m_heatMap, "mapitems", mapPipes);
for (const auto& pipe : mapPipes)
{
MessageQueue *messageQueue = qobject_cast<MessageQueue*>(pipe->m_element);
SWGSDRangel::SWGMapItem *swgMapItem = new SWGSDRangel::SWGMapItem();
swgMapItem->setName(new QString("Heat Map"));
swgMapItem->setImage(new QString()); // Set image to "" to delete it
swgMapItem->setType(1);
MainCore::MsgMapItem *msg = MainCore::MsgMapItem::create(m_heatMap, swgMapItem);
messageQueue->push(msg);
}
}
void HeatMapGUI::sendTxToMap()
{
// Send to Map feature
QList<ObjectPipe*> mapPipes;
MainCore::instance()->getMessagePipes().getMessagePipes(m_heatMap, "mapitems", mapPipes);
if (mapPipes.size() > 0)
{
QString text = QString("Heat Map Transmitter\nPower: %1 dB").arg(m_settings.m_txPower);
for (const auto& pipe : mapPipes)
{
MessageQueue *messageQueue = qobject_cast<MessageQueue*>(pipe->m_element);
SWGSDRangel::SWGMapItem *swgMapItem = new SWGSDRangel::SWGMapItem();
swgMapItem->setName(new QString("TX"));
swgMapItem->setLatitude(m_settings.m_txLatitude);
swgMapItem->setLongitude(m_settings.m_txLongitude);
swgMapItem->setAltitude(0);
swgMapItem->setImage(new QString("antenna.png"));
swgMapItem->setText(new QString(text));
swgMapItem->setModel(new QString("antenna.glb"));
swgMapItem->setFixedPosition(true);
swgMapItem->setLabel(new QString("TX"));
swgMapItem->setLabelAltitudeOffset(4.5);
swgMapItem->setAltitudeReference(1);
swgMapItem->setType(0);
MainCore::MsgMapItem *msg = MainCore::MsgMapItem::create(m_heatMap, swgMapItem);
messageQueue->push(msg);
}
}
}
void HeatMapGUI::deleteTxFromMap()
{
QList<ObjectPipe*> mapPipes;
MainCore::instance()->getMessagePipes().getMessagePipes(m_heatMap, "mapitems", mapPipes);
for (const auto& pipe : mapPipes)
{
MessageQueue *messageQueue = qobject_cast<MessageQueue*>(pipe->m_element);
SWGSDRangel::SWGMapItem *swgMapItem = new SWGSDRangel::SWGMapItem();
swgMapItem->setName(new QString("TX"));
swgMapItem->setImage(new QString()); // Set image to "" to delete it
swgMapItem->setType(0);
MainCore::MsgMapItem *msg = MainCore::MsgMapItem::create(m_heatMap, swgMapItem);
messageQueue->push(msg);
}
}
void HeatMapGUI::preferenceChanged(int elementType)
{
Preferences::ElementType pref = (Preferences::ElementType)elementType;
if ((pref == Preferences::Latitude) || (pref == Preferences::Longitude) || (pref == Preferences::Altitude))
{
// Get new position
m_latitude = MainCore::instance()->getSettings().getLatitude();
m_longitude = MainCore::instance()->getSettings().getLongitude();
m_altitude = MainCore::instance()->getSettings().getAltitude();
// Display new position in GUI
ui->latitude->setText(QString::number(m_latitude));
ui->longitude->setText(QString::number(m_longitude));
updateRange();
// Map position to pixel
int x, y;
coordsToPixel(m_latitude, m_longitude, x, y);
if ((x != m_x) || (y != m_y))
{
m_x = x;
m_y = y;
m_heatMap->resetMagLevels();
if (!pixelValid(x, y)) {
resizeMap(x, y);
}
}
}
}
void HeatMapGUI::createMap()
{
// https://wiki.openstreetmap.org/wiki/Zoom_levels
double earthCircumference = 40075016.686;
double scale = cos(Units::degreesToRadians(m_latitude));
m_resolution = earthCircumference * scale / pow(2.0, m_zoomLevel+8.0); // metres per pixel
m_width = m_blockSize;
m_height = m_blockSize;
m_degreesLonPerPixel = m_resolution / scale / (earthCircumference / 360.0);
m_degreesLatPerPixel = m_resolution / (earthCircumference / 360.0);
int size = m_width * m_height;
try
{
if (m_settings.m_recordAverage) {
m_powerAverage = new float[size];
} else {
m_powerAverage = nullptr;
}
if (m_settings.m_recordPulseAverage) {
m_powerPulseAverage = new float[size];
} else {
m_powerPulseAverage = nullptr;
}
if (m_settings.m_recordMax) {
m_powerMaxPeak = new float[size];
} else {
m_powerMaxPeak = nullptr;
}
if (m_settings.m_recordMin) {
m_powerMinPeak = new float[size];
} else {
m_powerMinPeak = nullptr;
}
if (m_settings.m_recordPathLoss) {
m_powerPathLoss = new float[size];
} else {
m_powerPathLoss = nullptr;
}
m_north = m_latitude + m_degreesLatPerPixel * m_height / 2;
m_south = m_latitude - m_degreesLatPerPixel * m_height / 2;
m_east = m_longitude + m_degreesLonPerPixel * m_width / 2;
m_west = m_longitude - m_degreesLonPerPixel * m_width / 2;
m_x = m_width / 2;
m_y = m_height / 2;
createImage(m_width, m_height);
}
catch (std::bad_alloc&)
{
deleteMap();
QMessageBox::critical(this, "Heat Map", QString("Failed to allocate memory (width=%1 height=%2)").arg(m_width).arg(m_height));
}
on_clearHeatMap_clicked();
}
void HeatMapGUI::deleteMap()
{
deleteFromMap();
delete[] m_powerAverage;
m_powerAverage = nullptr;
delete[] m_powerPulseAverage;
m_powerPulseAverage = nullptr;
delete[] m_powerMaxPeak;
m_powerMaxPeak = nullptr;
delete[] m_powerMinPeak;
m_powerMinPeak = nullptr;
delete[] m_powerPathLoss;
m_powerPathLoss = nullptr;
if (!m_image.isNull()) {
m_painter.end();
}
}
void HeatMapGUI::resizeMap(int x, int y)
{
if ((x <= -m_blockSize) || (x >= m_width+m_blockSize) || (y <= -m_blockSize) || (y >= m_height+m_blockSize))
{
// Position has moved a long way - restart from scratch to avoid map being too big
qDebug() << "HeatMapGUI::resizeMap: Position has moved significantly. Recreating map";
deleteMap();
createMap();
}
else
{
// Expand map
int newWidth = m_width;
int newHeight = m_height;
int xOffset = 0;
int yOffset = 0;
if (x < 0)
{
newWidth += m_blockSize;
xOffset = m_blockSize;
m_west -= m_blockSize * m_degreesLonPerPixel;
}
if (x >= m_width)
{
newWidth += m_blockSize;
m_east += m_blockSize * m_degreesLonPerPixel;
}
if (y < 0)
{
newHeight += m_blockSize;
yOffset = m_blockSize * newWidth;
m_north += m_blockSize * m_degreesLatPerPixel;
}
if (y >= m_height)
{
newHeight += m_blockSize;
m_south -= m_blockSize * m_degreesLatPerPixel;
}
float *powerAverage = nullptr;
float *powerPulseAverage = nullptr;
float *powerMaxPeak = nullptr;
float *powerMinPeak = nullptr;
float *powerPathLoss = nullptr;
int newSize = newWidth * newHeight;
qDebug() << "HeatMapGUI::resizeMap:" << m_width << "*" << m_height << "to" << newWidth << "*" << newHeight;
try
{
// Allocate new memory
if (m_settings.m_recordAverage) {
powerAverage = new float[newSize];
}
if (m_settings.m_recordPulseAverage) {
powerPulseAverage = new float[newSize];
}
if (m_settings.m_recordMax) {
powerMaxPeak = new float[newSize];
}
if (m_settings.m_recordMin) {
powerMinPeak = new float[newSize];
}
if (m_settings.m_recordPathLoss) {
powerPathLoss = new float[newSize];
}
clearPower(powerAverage, newSize);
clearPower(powerPulseAverage, newSize);
clearPower(powerMaxPeak, newSize);
clearPower(powerMinPeak, newSize);
clearPower(powerPathLoss, newSize);
// Copy across old data
for (int j = 0; j < m_height; j++)
{
int srcStart = j * m_width;
int srcEnd = (j + 1) * m_width;
int destStart = j * newWidth + yOffset + xOffset;
//qDebug() << srcStart << srcEnd << destStart;
if (powerAverage && m_powerAverage) {
std::copy(m_powerAverage + srcStart, m_powerAverage + srcEnd, powerAverage + destStart);
}
if (powerPulseAverage && m_powerPulseAverage) {
std::copy(m_powerPulseAverage + srcStart, m_powerPulseAverage + srcEnd, powerPulseAverage + destStart);
}
if (powerMaxPeak && m_powerMaxPeak) {
std::copy(m_powerMaxPeak + srcStart, m_powerMaxPeak + srcEnd, powerMaxPeak + destStart);
}
if (powerMinPeak && m_powerMinPeak) {
std::copy(m_powerMinPeak + srcStart, m_powerMinPeak + srcEnd, powerMinPeak + destStart);
}
if (powerPathLoss && m_powerPathLoss) {
std::copy(m_powerPathLoss + srcStart, m_powerPathLoss + srcEnd, powerPathLoss + destStart);
}
}
createImage(newWidth, newHeight);
m_width = newWidth;
m_height = newHeight;
// Delete old memory
delete[] m_powerAverage;
delete[] m_powerPulseAverage;
delete[] m_powerMaxPeak;
delete[] m_powerMinPeak;
m_powerAverage = powerAverage;
m_powerPulseAverage = powerPulseAverage;
m_powerMaxPeak = powerMaxPeak;
m_powerMinPeak = powerMinPeak;
m_powerPathLoss = powerPathLoss;
plotMap();
}
catch (std::bad_alloc&)
{
// Delete partially allocated memory
delete[] powerAverage;
delete[] powerPulseAverage;
delete[] powerMaxPeak;
delete[] powerMinPeak;
delete[] powerPathLoss;
QMessageBox::critical(this, "Heat Map", QString("Failed to allocate memory (width=%1 height=%2)").arg(newWidth).arg(newHeight));
}
}
}
void HeatMapGUI::plotPowerVsTimeChart()
{
QChart *oldChart = m_powerChart;
m_powerChart = new QChart();
m_powerChart->layout()->setContentsMargins(0, 0, 0, 0);
m_powerChart->setMargins(QMargins(1, 1, 1, 1));
m_powerChart->setTheme(QChart::ChartThemeDark);
m_powerChart->legend()->setAlignment(Qt::AlignBottom);
m_powerChart->legend()->setVisible(true);
// Create measurement data series
m_powerAverageSeries = new QLineSeries();
m_powerAverageSeries->setVisible(m_settings.m_displayAverage);
m_powerAverageSeries->setName("Average");
m_powerMaxPeakSeries = new QLineSeries();
m_powerMaxPeakSeries->setVisible(m_settings.m_displayMax);
m_powerMaxPeakSeries->setName("Max");
m_powerMinPeakSeries = new QLineSeries();
m_powerMinPeakSeries->setVisible(m_settings.m_displayMin);
m_powerMinPeakSeries->setName("Min");
m_powerPulseAverageSeries = new QLineSeries();
m_powerPulseAverageSeries->setVisible(m_settings.m_displayPulseAverage);
m_powerPulseAverageSeries->setName("Pulse Average");
m_powerPathLossSeries = new QLineSeries();
m_powerPathLossSeries->setVisible(m_settings.m_displayPathLoss);
m_powerPathLossSeries->setName("Path Loss");
// Create X axis
m_powerXAxis = new QDateTimeAxis();
QString dateTimeFormat = "hh:mm:ss";
m_powerXAxis->setFormat(dateTimeFormat);
m_powerXAxis->setTitleText("Time");
// Create Y axis
m_powerYAxis = new QValueAxis();
m_powerYAxis->setRange(m_settings.m_minPower, m_settings.m_maxPower);
m_powerYAxis->setTitleText("Power (dB)");
m_powerChart->addAxis(m_powerXAxis, Qt::AlignBottom);
m_powerChart->addAxis(m_powerYAxis, Qt::AlignLeft);
m_powerChart->addSeries(m_powerAverageSeries);
m_powerAverageSeries->attachAxis(m_powerXAxis);
m_powerAverageSeries->attachAxis(m_powerYAxis);
m_powerChart->addSeries(m_powerMaxPeakSeries);
m_powerMaxPeakSeries->attachAxis(m_powerXAxis);
m_powerMaxPeakSeries->attachAxis(m_powerYAxis);
m_powerChart->addSeries(m_powerMinPeakSeries);
m_powerMinPeakSeries->attachAxis(m_powerXAxis);
m_powerMinPeakSeries->attachAxis(m_powerYAxis);
m_powerChart->addSeries(m_powerPulseAverageSeries);
m_powerPulseAverageSeries->attachAxis(m_powerXAxis);
m_powerPulseAverageSeries->attachAxis(m_powerYAxis);
m_powerChart->addSeries(m_powerPathLossSeries);
m_powerPathLossSeries->attachAxis(m_powerXAxis);
m_powerPathLossSeries->attachAxis(m_powerYAxis);
ui->powerChart->setChart(m_powerChart);
delete oldChart;
}
void HeatMapGUI::addToPowerSeries(QDateTime dateTime, double average, double pulseAverage, double max, double min, double pathLoss)
{
if (m_powerAverageSeries)
{
qint64 msecs = dateTime.toMSecsSinceEpoch();
if (!std::isnan(average)) {
m_powerAverageSeries->append(msecs, average);
}
if (!std::isnan(pulseAverage)) {
m_powerPulseAverageSeries->append(msecs, pulseAverage);
}
if (!std::isnan(max)) {
m_powerMaxPeakSeries->append(msecs, max);
}
if (!std::isnan(min)) {
m_powerMinPeakSeries->append(msecs, min);
}
if (!std::isnan(pathLoss)) {
m_powerPathLossSeries->append(msecs, pathLoss);
}
}
}
void HeatMapGUI::updateAxis()
{
if (!m_powerAverageSeries || !m_powerPathLossSeries || (m_powerPathLossSeries->count() <= 1)) {
return;
}
QDateTime current = QDateTime::currentDateTime();
QDateTime first = QDateTime::fromMSecsSinceEpoch(m_powerPathLossSeries->at(0).x());
QDateTime min = current.addSecs(-60 * m_settings.m_displayMins);
if (first > min) {
min = first;
}
m_powerXAxis->setRange(min, current);
}
void HeatMapGUI::trimPowerSeries(QLineSeries *series, QDateTime dateTime)
{
qint64 msecs = dateTime.toMSecsSinceEpoch();
for (int i = 0; i < series->count(); i++)
{
const QPointF& p = series->at(i);
if (p.x() >= msecs)
{
if (i > 0) {
series->removePoints(0, i);
}
break;
}
}
}
// Remove all points in series before the specified date and time
void HeatMapGUI::trimPowerSeries(QDateTime dateTime)
{
if (m_powerAverageSeries)
{
trimPowerSeries(m_powerAverageSeries, dateTime);
trimPowerSeries(m_powerMaxPeakSeries, dateTime);
trimPowerSeries(m_powerMinPeakSeries, dateTime);
trimPowerSeries(m_powerPulseAverageSeries, dateTime);
trimPowerSeries(m_powerPathLossSeries, dateTime);
}
}
void HeatMapGUI::on_displayAverage_clicked(bool checked)
{
m_settings.m_displayAverage = checked;
m_powerAverageSeries->setVisible(checked);
applySettings();
}
void HeatMapGUI::on_displayMax_clicked(bool checked)
{
m_settings.m_displayMax = checked;
m_powerMaxPeakSeries->setVisible(checked);
applySettings();
}
void HeatMapGUI::on_displayMin_clicked(bool checked)
{
m_settings.m_displayMin = checked;
m_powerMinPeakSeries->setVisible(checked);
applySettings();
}
void HeatMapGUI::on_displayPulseAverage_clicked(bool checked)
{
m_settings.m_displayPulseAverage = checked;
m_powerPulseAverageSeries->setVisible(checked);
applySettings();
}
void HeatMapGUI::on_displayPathLoss_clicked(bool checked)
{
m_settings.m_displayPathLoss = checked;
m_powerPathLossSeries->setVisible(checked);
applySettings();
}
void HeatMapGUI::on_displayMins_valueChanged(int value)
{
m_settings.m_displayMins = value;
updateAxis();
applySettings();
}
void HeatMapGUI::on_recordAverage_clicked(bool checked)
{
m_settings.m_recordAverage = checked;
resizeMap(0, 0);
applySettings();
}
void HeatMapGUI::on_recordMax_clicked(bool checked)
{
m_settings.m_recordMax = checked;
resizeMap(0, 0);
applySettings();
}
void HeatMapGUI::on_recordMin_clicked(bool checked)
{
m_settings.m_recordMin = checked;
resizeMap(0, 0);
applySettings();
}
void HeatMapGUI::on_recordPulseAverage_clicked(bool checked)
{
m_settings.m_recordPulseAverage = checked;
resizeMap(0, 0);
applySettings();
}
void HeatMapGUI::on_recordPathLoss_clicked(bool checked)
{
m_settings.m_recordPathLoss = checked;
resizeMap(0, 0);
applySettings();
}