/////////////////////////////////////////////////////////////////////////////////// // Copyright (C) 2015-2018 Edouard Griffiths, F4EXB. // // Copyright (C) 2021 Jon Beniston, M7RCE // // // // 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 . // /////////////////////////////////////////////////////////////////////////////////// #include "aptdemod.h" #include #include #include #include #include #include #include #include #include "SWGChannelSettings.h" #include "SWGAPTDemodSettings.h" #include "SWGChannelReport.h" #include "SWGChannelActions.h" #include "SWGMapItem.h" #include "SWGAPTDemodActions.h" #include "dsp/dspengine.h" #include "dsp/dspcommands.h" #include "device/deviceapi.h" #include "feature/feature.h" #include "util/db.h" #include "maincore.h" MESSAGE_CLASS_DEFINITION(APTDemod::MsgConfigureAPTDemod, Message) MESSAGE_CLASS_DEFINITION(APTDemod::MsgPixels, Message) MESSAGE_CLASS_DEFINITION(APTDemod::MsgImage, Message) MESSAGE_CLASS_DEFINITION(APTDemod::MsgResetDecoder, Message) const char * const APTDemod::m_channelIdURI = "sdrangel.channel.aptdemod"; const char * const APTDemod::m_channelId = "APTDemod"; APTDemod::APTDemod(DeviceAPI *deviceAPI) : ChannelAPI(m_channelIdURI, ChannelAPI::StreamSingleSink), m_deviceAPI(deviceAPI), m_basebandSampleRate(0) { setObjectName(m_channelId); m_basebandSink = new APTDemodBaseband(this); m_basebandSink->setMessageQueueToChannel(getInputMessageQueue()); m_basebandSink->moveToThread(&m_thread); applySettings(m_settings, true); m_deviceAPI->addChannelSink(this); m_deviceAPI->addChannelSinkAPI(this); m_networkManager = new QNetworkAccessManager(); connect(m_networkManager, SIGNAL(finished(QNetworkReply*)), this, SLOT(networkManagerFinished(QNetworkReply*))); for (int y = 0; y < APT_MAX_HEIGHT; y++) { m_image.prow[y] = new float[APT_PROW_WIDTH]; m_tempImage.prow[y] = new float[APT_PROW_WIDTH]; } resetDecoder(); } APTDemod::~APTDemod() { qDebug("APTDemod::~APTDemod"); disconnect(m_networkManager, SIGNAL(finished(QNetworkReply*)), this, SLOT(networkManagerFinished(QNetworkReply*))); delete m_networkManager; m_deviceAPI->removeChannelSinkAPI(this); m_deviceAPI->removeChannelSink(this); if (m_basebandSink->isRunning()) { stop(); } delete m_basebandSink; for (int y = 0; y < APT_MAX_HEIGHT; y++) { delete m_image.prow[y]; delete m_tempImage.prow[y]; } } uint32_t APTDemod::getNumberOfDeviceStreams() const { return m_deviceAPI->getNbSourceStreams(); } void APTDemod::feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end, bool firstOfBurst) { (void) firstOfBurst; m_basebandSink->feed(begin, end); } void APTDemod::start() { qDebug("APTDemod::start"); m_basebandSink->reset(); m_basebandSink->startWork(); m_thread.start(); DSPSignalNotification *dspMsg = new DSPSignalNotification(m_basebandSampleRate, m_centerFrequency); m_basebandSink->getInputMessageQueue()->push(dspMsg); APTDemodBaseband::MsgConfigureAPTDemodBaseband *msg = APTDemodBaseband::MsgConfigureAPTDemodBaseband::create(m_settings, true); m_basebandSink->getInputMessageQueue()->push(msg); } void APTDemod::stop() { qDebug("APTDemod::stop"); m_basebandSink->stopWork(); m_thread.quit(); m_thread.wait(); } bool APTDemod::matchSatellite(const QString satelliteName) { return m_settings.m_satelliteTrackerControl && ( (satelliteName == m_settings.m_satelliteName) || ( (m_settings.m_satelliteName == "All") && ( (satelliteName == "NOAA 15") || (satelliteName == "NOAA 18") || (satelliteName == "NOAA 19")))); } bool APTDemod::handleMessage(const Message& cmd) { if (MsgConfigureAPTDemod::match(cmd)) { MsgConfigureAPTDemod& cfg = (MsgConfigureAPTDemod&) cmd; qDebug() << "APTDemod::handleMessage: MsgConfigureAPTDemod"; applySettings(cfg.getSettings(), cfg.getForce()); return true; } else if (DSPSignalNotification::match(cmd)) { DSPSignalNotification& notif = (DSPSignalNotification&) cmd; m_basebandSampleRate = notif.getSampleRate(); m_centerFrequency = notif.getCenterFrequency(); // Forward to the sink DSPSignalNotification* rep = new DSPSignalNotification(notif); // make a copy qDebug() << "APTDemod::handleMessage: DSPSignalNotification"; m_basebandSink->getInputMessageQueue()->push(rep); // Forward to GUI if any if (m_guiMessageQueue) m_guiMessageQueue->push(new DSPSignalNotification(notif)); return true; } else if (APTDemod::MsgPixels::match(cmd)) { const APTDemod::MsgPixels& pixelsMsg = (APTDemod::MsgPixels&) cmd; const float *pixels = pixelsMsg.getPixels(); processPixels(pixels); return true; } else if (APTDemod::MsgResetDecoder::match(cmd)) { resetDecoder(); // Forward to sink m_basebandSink->getInputMessageQueue()->push(APTDemod::MsgResetDecoder::create()); return true; } else { return false; } } void APTDemod::applySettings(const APTDemodSettings& settings, bool force) { bool callProcessImage = false; qDebug() << "APTDemod::applySettings:" << " m_cropNoise: " << settings.m_cropNoise << " m_denoise: " << settings.m_denoise << " m_linearEqualise: " << settings.m_linearEqualise << " m_histogramEqualise: " << settings.m_histogramEqualise << " m_precipitationOverlay: " << settings.m_precipitationOverlay << " m_flip: " << settings.m_flip << " m_channels: " << settings.m_channels << " m_decodeEnabled: " << settings.m_decodeEnabled << " m_autoSave: " << settings.m_autoSave << " m_autoSavePath: " << settings.m_autoSavePath << " m_autoSaveMinScanLines: " << settings.m_autoSaveMinScanLines << " m_streamIndex: " << settings.m_streamIndex << " m_useReverseAPI: " << settings.m_useReverseAPI << " m_reverseAPIAddress: " << settings.m_reverseAPIAddress << " m_reverseAPIPort: " << settings.m_reverseAPIPort << " m_reverseAPIDeviceIndex: " << settings.m_reverseAPIDeviceIndex << " m_reverseAPIChannelIndex: " << settings.m_reverseAPIChannelIndex << " force: " << force; QList reverseAPIKeys; if ((settings.m_inputFrequencyOffset != m_settings.m_inputFrequencyOffset) || force) { reverseAPIKeys.append("inputFrequencyOffset"); } if ((settings.m_rfBandwidth != m_settings.m_rfBandwidth) || force) { reverseAPIKeys.append("rfBandwidth"); } if ((settings.m_fmDeviation != m_settings.m_fmDeviation) || force) { reverseAPIKeys.append("fmDeviation"); } if ((settings.m_denoise != m_settings.m_denoise) || force) { reverseAPIKeys.append("denoise"); } if ((settings.m_linearEqualise != m_settings.m_linearEqualise) || force) { reverseAPIKeys.append("linearEqualise"); } if ((settings.m_histogramEqualise != m_settings.m_histogramEqualise) || force) { reverseAPIKeys.append("histogramEqualise"); } if ((settings.m_precipitationOverlay != m_settings.m_precipitationOverlay) || force) { reverseAPIKeys.append("precipitationOverlay"); } if ((settings.m_flip != m_settings.m_flip) || force) { reverseAPIKeys.append("flip"); } if ((settings.m_channels != m_settings.m_channels) || force) { reverseAPIKeys.append("channels"); } if ((settings.m_decodeEnabled != m_settings.m_decodeEnabled) || force) { reverseAPIKeys.append("decodeEnabled"); } if ((settings.m_autoSave != m_settings.m_autoSave) || force) { reverseAPIKeys.append("autoSave"); } if ((settings.m_autoSavePath != m_settings.m_autoSavePath) || force) { reverseAPIKeys.append("autoSavePath"); } if ((settings.m_autoSaveMinScanLines != m_settings.m_autoSaveMinScanLines) || force) { reverseAPIKeys.append("autoSaveMinScanLines"); } if (m_settings.m_streamIndex != settings.m_streamIndex) { if (m_deviceAPI->getSampleMIMO()) // change of stream is possible for MIMO devices only { m_deviceAPI->removeChannelSinkAPI(this); m_deviceAPI->removeChannelSink(this, m_settings.m_streamIndex); m_deviceAPI->addChannelSink(this, settings.m_streamIndex); m_deviceAPI->addChannelSinkAPI(this); } reverseAPIKeys.append("streamIndex"); } APTDemodBaseband::MsgConfigureAPTDemodBaseband *msg = APTDemodBaseband::MsgConfigureAPTDemodBaseband::create(settings, force); m_basebandSink->getInputMessageQueue()->push(msg); if (settings.m_useReverseAPI) { bool fullUpdate = ((m_settings.m_useReverseAPI != settings.m_useReverseAPI) && settings.m_useReverseAPI) || (m_settings.m_reverseAPIAddress != settings.m_reverseAPIAddress) || (m_settings.m_reverseAPIPort != settings.m_reverseAPIPort) || (m_settings.m_reverseAPIDeviceIndex != settings.m_reverseAPIDeviceIndex) || (m_settings.m_reverseAPIChannelIndex != settings.m_reverseAPIChannelIndex); webapiReverseSendSettings(reverseAPIKeys, settings, fullUpdate || force); } if ((settings.m_cropNoise != m_settings.m_cropNoise) || (settings.m_denoise != m_settings.m_denoise) || (settings.m_linearEqualise != m_settings.m_linearEqualise) || (settings.m_histogramEqualise != m_settings.m_histogramEqualise) || (settings.m_precipitationOverlay != m_settings.m_precipitationOverlay) || (settings.m_flip != m_settings.m_flip) || (settings.m_channels != m_settings.m_channels)) { // Call after settings have been applied callProcessImage = true; } m_settings = settings; if (callProcessImage) sendImageToGUI(); } QByteArray APTDemod::serialize() const { return m_settings.serialize(); } bool APTDemod::deserialize(const QByteArray& data) { if (m_settings.deserialize(data)) { MsgConfigureAPTDemod *msg = MsgConfigureAPTDemod::create(m_settings, true); m_inputMessageQueue.push(msg); return true; } else { m_settings.resetToDefaults(); MsgConfigureAPTDemod *msg = MsgConfigureAPTDemod::create(m_settings, true); m_inputMessageQueue.push(msg); return false; } } int APTDemod::webapiSettingsGet( SWGSDRangel::SWGChannelSettings& response, QString& errorMessage) { (void) errorMessage; response.setAptDemodSettings(new SWGSDRangel::SWGAPTDemodSettings()); response.getAptDemodSettings()->init(); webapiFormatChannelSettings(response, m_settings); return 200; } int APTDemod::webapiSettingsPutPatch( bool force, const QStringList& channelSettingsKeys, SWGSDRangel::SWGChannelSettings& response, QString& errorMessage) { (void) errorMessage; APTDemodSettings settings = m_settings; webapiUpdateChannelSettings(settings, channelSettingsKeys, response); MsgConfigureAPTDemod *msg = MsgConfigureAPTDemod::create(settings, force); m_inputMessageQueue.push(msg); qDebug("APTDemod::webapiSettingsPutPatch: forward to GUI: %p", m_guiMessageQueue); if (m_guiMessageQueue) // forward to GUI if any { MsgConfigureAPTDemod *msgToGUI = MsgConfigureAPTDemod::create(settings, force); m_guiMessageQueue->push(msgToGUI); } webapiFormatChannelSettings(response, settings); return 200; } void APTDemod::webapiUpdateChannelSettings( APTDemodSettings& settings, const QStringList& channelSettingsKeys, SWGSDRangel::SWGChannelSettings& response) { if (channelSettingsKeys.contains("inputFrequencyOffset")) { settings.m_inputFrequencyOffset = response.getAptDemodSettings()->getInputFrequencyOffset(); } if (channelSettingsKeys.contains("fmDeviation")) { settings.m_fmDeviation = response.getAptDemodSettings()->getFmDeviation(); } if (channelSettingsKeys.contains("rfBandwidth")) { settings.m_rfBandwidth = response.getAptDemodSettings()->getRfBandwidth(); } if (channelSettingsKeys.contains("cropNoise")) { settings.m_cropNoise = response.getAptDemodSettings()->getCropNoise(); } if (channelSettingsKeys.contains("denoise")) { settings.m_denoise = response.getAptDemodSettings()->getDenoise(); } if (channelSettingsKeys.contains("linearEqualise")) { settings.m_linearEqualise = response.getAptDemodSettings()->getLinearEqualise(); } if (channelSettingsKeys.contains("histogramEqualise")) { settings.m_histogramEqualise = response.getAptDemodSettings()->getHistogramEqualise(); } if (channelSettingsKeys.contains("precipitationOverlay")) { settings.m_precipitationOverlay = response.getAptDemodSettings()->getPrecipitationOverlay(); } if (channelSettingsKeys.contains("flip")) { settings.m_flip = response.getAptDemodSettings()->getFlip(); } if (channelSettingsKeys.contains("channels")) { settings.m_channels = (APTDemodSettings::ChannelSelection)response.getAptDemodSettings()->getChannels(); } if (channelSettingsKeys.contains("decodeEnabled")) { settings.m_decodeEnabled = response.getAptDemodSettings()->getDecodeEnabled(); } if (channelSettingsKeys.contains("autoSave")) { settings.m_autoSave = response.getAptDemodSettings()->getAutoSave(); } if (channelSettingsKeys.contains("autoSavePath")) { settings.m_autoSavePath = *response.getAptDemodSettings()->getAutoSavePath(); } if (channelSettingsKeys.contains("autoSaveMinScanLines")) { settings.m_autoSaveMinScanLines = response.getAptDemodSettings()->getAutoSaveMinScanLines(); } if (channelSettingsKeys.contains("rgbColor")) { settings.m_rgbColor = response.getAptDemodSettings()->getRgbColor(); } if (channelSettingsKeys.contains("title")) { settings.m_title = *response.getAptDemodSettings()->getTitle(); } if (channelSettingsKeys.contains("streamIndex")) { settings.m_streamIndex = response.getAptDemodSettings()->getStreamIndex(); } if (channelSettingsKeys.contains("useReverseAPI")) { settings.m_useReverseAPI = response.getAptDemodSettings()->getUseReverseApi() != 0; } if (channelSettingsKeys.contains("reverseAPIAddress")) { settings.m_reverseAPIAddress = *response.getAptDemodSettings()->getReverseApiAddress(); } if (channelSettingsKeys.contains("reverseAPIPort")) { settings.m_reverseAPIPort = response.getAptDemodSettings()->getReverseApiPort(); } if (channelSettingsKeys.contains("reverseAPIDeviceIndex")) { settings.m_reverseAPIDeviceIndex = response.getAptDemodSettings()->getReverseApiDeviceIndex(); } if (channelSettingsKeys.contains("reverseAPIChannelIndex")) { settings.m_reverseAPIChannelIndex = response.getAptDemodSettings()->getReverseApiChannelIndex(); } } void APTDemod::webapiFormatChannelSettings(SWGSDRangel::SWGChannelSettings& response, const APTDemodSettings& settings) { response.getAptDemodSettings()->setInputFrequencyOffset(settings.m_inputFrequencyOffset); response.getAptDemodSettings()->setRfBandwidth(settings.m_rfBandwidth); response.getAptDemodSettings()->setFmDeviation(settings.m_fmDeviation); response.getAptDemodSettings()->setCropNoise(settings.m_cropNoise); response.getAptDemodSettings()->setCropNoise(settings.m_denoise); response.getAptDemodSettings()->setLinearEqualise(settings.m_linearEqualise); response.getAptDemodSettings()->setHistogramEqualise(settings.m_histogramEqualise); response.getAptDemodSettings()->setPrecipitationOverlay(settings.m_precipitationOverlay); response.getAptDemodSettings()->setFlip(settings.m_flip); response.getAptDemodSettings()->setChannels((int)settings.m_channels); response.getAptDemodSettings()->setDecodeEnabled(settings.m_decodeEnabled); response.getAptDemodSettings()->setAutoSave(settings.m_autoSave); response.getAptDemodSettings()->setAutoSavePath(new QString(settings.m_autoSavePath)); response.getAptDemodSettings()->setAutoSaveMinScanLines(settings.m_autoSaveMinScanLines); response.getAptDemodSettings()->setRgbColor(settings.m_rgbColor); if (response.getAptDemodSettings()->getTitle()) { *response.getAptDemodSettings()->getTitle() = settings.m_title; } else { response.getAptDemodSettings()->setTitle(new QString(settings.m_title)); } response.getAptDemodSettings()->setStreamIndex(settings.m_streamIndex); response.getAptDemodSettings()->setUseReverseApi(settings.m_useReverseAPI ? 1 : 0); if (response.getAptDemodSettings()->getReverseApiAddress()) { *response.getAptDemodSettings()->getReverseApiAddress() = settings.m_reverseAPIAddress; } else { response.getAptDemodSettings()->setReverseApiAddress(new QString(settings.m_reverseAPIAddress)); } response.getAptDemodSettings()->setReverseApiPort(settings.m_reverseAPIPort); response.getAptDemodSettings()->setReverseApiDeviceIndex(settings.m_reverseAPIDeviceIndex); response.getAptDemodSettings()->setReverseApiChannelIndex(settings.m_reverseAPIChannelIndex); } void APTDemod::webapiReverseSendSettings(QList& channelSettingsKeys, const APTDemodSettings& settings, bool force) { SWGSDRangel::SWGChannelSettings *swgChannelSettings = new SWGSDRangel::SWGChannelSettings(); webapiFormatChannelSettings(channelSettingsKeys, swgChannelSettings, settings, force); QString channelSettingsURL = QString("http://%1:%2/sdrangel/deviceset/%3/channel/%4/settings") .arg(settings.m_reverseAPIAddress) .arg(settings.m_reverseAPIPort) .arg(settings.m_reverseAPIDeviceIndex) .arg(settings.m_reverseAPIChannelIndex); m_networkRequest.setUrl(QUrl(channelSettingsURL)); m_networkRequest.setHeader(QNetworkRequest::ContentTypeHeader, "application/json"); QBuffer *buffer = new QBuffer(); buffer->open((QBuffer::ReadWrite)); buffer->write(swgChannelSettings->asJson().toUtf8()); buffer->seek(0); // Always use PATCH to avoid passing reverse API settings QNetworkReply *reply = m_networkManager->sendCustomRequest(m_networkRequest, "PATCH", buffer); buffer->setParent(reply); delete swgChannelSettings; } void APTDemod::webapiFormatChannelSettings( QList& channelSettingsKeys, SWGSDRangel::SWGChannelSettings *swgChannelSettings, const APTDemodSettings& settings, bool force ) { swgChannelSettings->setDirection(0); // Single sink (Rx) swgChannelSettings->setOriginatorChannelIndex(getIndexInDeviceSet()); swgChannelSettings->setOriginatorDeviceSetIndex(getDeviceSetIndex()); swgChannelSettings->setChannelType(new QString("APTDemod")); swgChannelSettings->setAptDemodSettings(new SWGSDRangel::SWGAPTDemodSettings()); SWGSDRangel::SWGAPTDemodSettings *swgAPTDemodSettings = swgChannelSettings->getAptDemodSettings(); // transfer data that has been modified. When force is on transfer all data except reverse API data if (channelSettingsKeys.contains("inputFrequencyOffset") || force) { swgAPTDemodSettings->setInputFrequencyOffset(settings.m_inputFrequencyOffset); } if (channelSettingsKeys.contains("rfBandwidth") || force) { swgAPTDemodSettings->setRfBandwidth(settings.m_rfBandwidth); } if (channelSettingsKeys.contains("fmDeviation") || force) { swgAPTDemodSettings->setFmDeviation(settings.m_fmDeviation); } if (channelSettingsKeys.contains("cropNoise") || force) { swgAPTDemodSettings->setCropNoise(settings.m_cropNoise); } if (channelSettingsKeys.contains("denoise") || force) { swgAPTDemodSettings->setDenoise(settings.m_denoise); } if (channelSettingsKeys.contains("linearEqualise") || force) { swgAPTDemodSettings->setLinearEqualise(settings.m_linearEqualise); } if (channelSettingsKeys.contains("histogramEqualise") || force) { swgAPTDemodSettings->setHistogramEqualise(settings.m_histogramEqualise); } if (channelSettingsKeys.contains("precipitationOverlay") || force) { swgAPTDemodSettings->setPrecipitationOverlay(settings.m_precipitationOverlay); } if (channelSettingsKeys.contains("flip") || force) { swgAPTDemodSettings->setFlip(settings.m_flip); } if (channelSettingsKeys.contains("channels") || force) { swgAPTDemodSettings->setChannels((int)settings.m_channels); } if (channelSettingsKeys.contains("decodeEnabled") || force) { swgAPTDemodSettings->setDecodeEnabled(settings.m_decodeEnabled); } if (channelSettingsKeys.contains("m_autoSave") || force) { swgAPTDemodSettings->setAutoSave(settings.m_autoSave); } if (channelSettingsKeys.contains("m_autoSavePath") || force) { swgAPTDemodSettings->setAutoSavePath(new QString(settings.m_autoSavePath)); } if (channelSettingsKeys.contains("m_autoSaveMinScanLines") || force) { swgAPTDemodSettings->setAutoSaveMinScanLines(settings.m_autoSaveMinScanLines); } if (channelSettingsKeys.contains("rgbColor") || force) { swgAPTDemodSettings->setRgbColor(settings.m_rgbColor); } if (channelSettingsKeys.contains("title") || force) { swgAPTDemodSettings->setTitle(new QString(settings.m_title)); } if (channelSettingsKeys.contains("streamIndex") || force) { swgAPTDemodSettings->setStreamIndex(settings.m_streamIndex); } } int APTDemod::webapiActionsPost( const QStringList& channelActionsKeys, SWGSDRangel::SWGChannelActions& query, QString& errorMessage) { SWGSDRangel::SWGAPTDemodActions *swgAPTDemodActions = query.getAptDemodActions(); if (swgAPTDemodActions) { if (channelActionsKeys.contains("aos")) { qDebug() << "Aos action"; SWGSDRangel::SWGAPTDemodActions_aos* aos = swgAPTDemodActions->getAos(); QString *satelliteName = aos->getSatelliteName(); if (satelliteName != nullptr) { qDebug() << "sat " << *satelliteName; if (matchSatellite(*satelliteName)) { qDebug() << "Matched sat"; // Reset for new pass resetDecoder(); m_basebandSink->getInputMessageQueue()->push(APTDemod::MsgResetDecoder::create()); // Save satellite name m_satelliteName = *satelliteName; // Enable decoder and set direction of pass APTDemodSettings settings = m_settings; settings.m_decodeEnabled = true; settings.m_flip = !aos->getNorthToSouthPass(); qDebug() << "Sending settings"; m_inputMessageQueue.push(MsgConfigureAPTDemod::create(settings, false)); if (m_guiMessageQueue) m_guiMessageQueue->push(MsgConfigureAPTDemod::create(settings, false)); } return 202; } else { errorMessage = "Missing satellite name"; return 400; } } else if (channelActionsKeys.contains("los")) { SWGSDRangel::SWGAPTDemodActions_los* los = swgAPTDemodActions->getLos(); QString *satelliteName = los->getSatelliteName(); if (satelliteName != nullptr) { if (matchSatellite(*satelliteName)) { // Save image if (m_settings.m_autoSave) saveImageToDisk(); // Disable decoder APTDemodSettings settings = m_settings; settings.m_decodeEnabled = false; m_inputMessageQueue.push(MsgConfigureAPTDemod::create(settings, false)); if (m_guiMessageQueue) m_guiMessageQueue->push(MsgConfigureAPTDemod::create(settings, false)); } return 202; } else { errorMessage = "Missing satellite name"; return 400; } } else { errorMessage = "Unknown action"; return 400; } } else { errorMessage = "Missing APTDemodActions in query"; return 400; } } void APTDemod::networkManagerFinished(QNetworkReply *reply) { QNetworkReply::NetworkError replyError = reply->error(); if (replyError) { qWarning() << "APTDemod::networkManagerFinished:" << " error(" << (int) replyError << "): " << replyError << ": " << reply->errorString(); } else { QString answer = reply->readAll(); answer.chop(1); // remove last \n qDebug("APTDemod::networkManagerFinished: reply:\n%s", answer.toStdString().c_str()); } reply->deleteLater(); } void APTDemod::resetDecoder() { m_image.nrow = 0; m_tempImage.nrow = 0; m_greyImage = QImage(APT_IMG_WIDTH, APT_MAX_HEIGHT, QImage::Format_Grayscale8); m_greyImage.fill(0); m_colourImage = QImage(APT_IMG_WIDTH, APT_MAX_HEIGHT, QImage::Format_RGB888); m_colourImage.fill(0); m_satelliteName = ""; } void APTDemod::processPixels(const float *pixels) { memcpy(m_image.prow[m_image.nrow], pixels, sizeof(float) * APT_PROW_WIDTH); m_image.nrow++; sendImageToGUI(); } static void copyImage(apt_image_t *dst, apt_image_t *src) { dst->nrow = src->nrow; dst->zenith = src->zenith; dst->chA = src->chA; dst->chB = src->chB; for (int i = 0; i < src->nrow; i++) memcpy(dst->prow[i], src->prow[i], sizeof(float) * APT_PROW_WIDTH); } static uchar roundAndClip(float p) { int q = (int)round(p); if (q > 255) q = 255; else if (q < 0) q = 0; return q; } QImage APTDemod::extractImage(QImage image) { if (m_settings.m_channels == APTDemodSettings::BOTH_CHANNELS) return image.copy(0, 0, APT_IMG_WIDTH, m_tempImage.nrow); else if (m_settings.m_channels == APTDemodSettings::CHANNEL_A) return image.copy(APT_CHA_OFFSET, 0, APT_CH_WIDTH, m_tempImage.nrow); else return image.copy(APT_CHB_OFFSET, 0, APT_CH_WIDTH, m_tempImage.nrow); } QImage APTDemod::processImage(QStringList& imageTypes) { copyImage(&m_tempImage, &m_image); // Calibrate channels according to wavelength if (m_tempImage.nrow >= APT_CALIBRATION_ROWS) { m_tempImage.chA = apt_calibrate(m_tempImage.prow, m_tempImage.nrow, APT_CHA_OFFSET, APT_CH_WIDTH); m_tempImage.chB = apt_calibrate(m_tempImage.prow, m_tempImage.nrow, APT_CHB_OFFSET, APT_CH_WIDTH); QStringList channelTypes({ "", // Unknown "Visible (0.58-0.68 um)", "Near-IR (0.725-1.0 um)", "Near-IR (1.58-1.64 um)", "Mid-infrared (3.55-3.93 um)", "Thermal-infrared (10.3-11.3 um)", "Thermal-infrared (11.5-12.5 um)" }); imageTypes.append(channelTypes[m_tempImage.chA]); imageTypes.append(channelTypes[m_tempImage.chB]); } // Crop noise due to low elevation at top and bottom of image if (m_settings.m_cropNoise) m_tempImage.zenith -= apt_cropNoise(&m_tempImage); // Denoise filter if (m_settings.m_denoise) { apt_denoise(m_tempImage.prow, m_tempImage.nrow, APT_CHA_OFFSET, APT_CH_WIDTH); apt_denoise(m_tempImage.prow, m_tempImage.nrow, APT_CHB_OFFSET, APT_CH_WIDTH); } // Flip image if satellite pass is North to South if (m_settings.m_flip) { apt_flipImage(&m_tempImage, APT_CH_WIDTH, APT_CHA_OFFSET); apt_flipImage(&m_tempImage, APT_CH_WIDTH, APT_CHB_OFFSET); } // Linear equalise to improve contrast if (m_settings.m_linearEqualise) { apt_linearEnhance(m_tempImage.prow, m_tempImage.nrow, APT_CHA_OFFSET, APT_CH_WIDTH); apt_linearEnhance(m_tempImage.prow, m_tempImage.nrow, APT_CHB_OFFSET, APT_CH_WIDTH); } // Histogram equalise to improve contrast if (m_settings.m_histogramEqualise) { apt_histogramEqualise(m_tempImage.prow, m_tempImage.nrow, APT_CHA_OFFSET, APT_CH_WIDTH); apt_histogramEqualise(m_tempImage.prow, m_tempImage.nrow, APT_CHB_OFFSET, APT_CH_WIDTH); } if (m_settings.m_precipitationOverlay) { // Overlay precipitation for (int r = 0; r < m_tempImage.nrow; r++) { uchar *l = m_colourImage.scanLine(r); for (int i = 0; i < APT_IMG_WIDTH; i++) { float p = m_tempImage.prow[r][i]; if ((i >= APT_CHB_OFFSET) && (i < APT_CHB_OFFSET + APT_CH_WIDTH) && (p >= 198)) { apt_rgb_t rgb = apt_applyPalette(apt_PrecipPalette, p - 198); // Negative float values get converted to positive uchars here l[i*3] = (uchar)rgb.r; l[i*3+1] = (uchar)rgb.g; l[i*3+2] = (uchar)rgb.b; int a = i - APT_CHB_OFFSET + APT_CHA_OFFSET; l[a*3] = (uchar)rgb.r; l[a*3+1] = (uchar)rgb.g; l[a*3+2] = (uchar)rgb.b; } else { uchar q = roundAndClip(p); l[i*3] = q; l[i*3+1] = q; l[i*3+2] = q; } } } return extractImage(m_colourImage); } else { for (int r = 0; r < m_tempImage.nrow; r++) { uchar *l = m_greyImage.scanLine(r); for (int i = 0; i < APT_IMG_WIDTH; i++) { float p = m_tempImage.prow[r][i]; l[i] = roundAndClip(p); } } return extractImage(m_greyImage); } } void APTDemod::sendImageToGUI() { // Send image to GUI if (getMessageQueueToGUI()) { QStringList imageTypes; QImage image = processImage(imageTypes); getMessageQueueToGUI()->push(APTDemod::MsgImage::create(image, imageTypes, m_satelliteName)); } } void APTDemod::saveImageToDisk() { QStringList imageTypes; QImage image = processImage(imageTypes); if (image.height() >= m_settings.m_autoSaveMinScanLines) { QString filename; QDateTime datetime = QDateTime::currentDateTime(); filename = QString("apt_%1_%2.png").arg(m_satelliteName).arg(datetime.toString("yyyyMMdd_hhmm")); if (!m_settings.m_autoSavePath.isEmpty()) { if (m_settings.m_autoSavePath.endsWith('/')) filename = m_settings.m_autoSavePath + filename; else filename = m_settings.m_autoSavePath + '/' + filename; } if (!image.save(filename)) qCritical() << "Failed to save APT image to: " << filename; } }