SDRPlusPlus/weather_sat_decoder/src/noaa_hrpt_decoder.h
2021-04-18 19:32:18 +02:00

564 lines
17 KiB
C++

#pragma once
#include <sat_decoder.h>
#include <dsp/demodulator.h>
#include <dsp/deframing.h>
#include <dsp/noaa/hrpt.h>
#include <dsp/noaa/tip.h>
#include <dsp/sink.h>
#include <gui/widgets/symbol_diagram.h>
#include <gui/widgets/line_push_image.h>
#define NOAA_HRPT_VFO_SR 3000000.0f
#define NOAA_HRPT_VFO_BW 2000000.0f
class NOAAHRPTDecoder : public SatDecoder {
public:
NOAAHRPTDecoder(VFOManager::VFO* vfo, std::string name) : avhrrRGBImage(2048, 256), avhrr1Image(2048, 256), avhrr2Image(2048, 256), avhrr3Image(2048, 256), avhrr4Image(2048, 256), avhrr5Image(2048, 256), symDiag(0.5f) {
_vfo = vfo;
_name = name;
// Core DSP
demod.init(vfo->output, NOAA_HRPT_VFO_SR, 665400.0f * 2.0f, 0.02e-3, (0.06f*0.06f) / 2.0f, 32, 0.6f, (0.01f*0.01f) / 4.0f, 0.01f, 0.005);
split.init(demod.out);
split.bindStream(&dataStream);
split.bindStream(&visStream);
reshape.init(&visStream, 1024, (NOAA_HRPT_VFO_SR/30) - 1024);
visSink.init(&reshape.out, visHandler, this);
deframe.init(&dataStream, 11090 * 10 * 2, (uint8_t*)dsp::noaa::HRPTSyncWord, 60);
manDec.init(&deframe.out, false);
packer.init(&manDec.out);
demux.init(&packer.out);
tipDemux.init(&demux.TIPOut);
hirsDemux.init(&tipDemux.HIRSOut);
// All Sinks
avhrr1Sink.init(&demux.AVHRRChan1Out, avhrr1Handler, this);
avhrr2Sink.init(&demux.AVHRRChan2Out, avhrr2Handler, this);
avhrr3Sink.init(&demux.AVHRRChan3Out, avhrr3Handler, this);
avhrr4Sink.init(&demux.AVHRRChan4Out, avhrr4Handler, this);
avhrr5Sink.init(&demux.AVHRRChan5Out, avhrr5Handler, this);
sbuvSink.init(&tipDemux.SBUVOut);
dcsSink.init(&tipDemux.DCSOut);
semSink.init(&tipDemux.SEMOut);
aipSink.init(&demux.AIPOut);
hirs1Sink.init(&hirsDemux.radChannels[0], hirs1Handler, this);
hirs2Sink.init(&hirsDemux.radChannels[1], hirs2Handler, this);
hirs3Sink.init(&hirsDemux.radChannels[2], hirs3Handler, this);
hirs4Sink.init(&hirsDemux.radChannels[3], hirs4Handler, this);
hirs5Sink.init(&hirsDemux.radChannels[4], hirs5Handler, this);
hirs6Sink.init(&hirsDemux.radChannels[5], hirs6Handler, this);
hirs7Sink.init(&hirsDemux.radChannels[6], hirs7Handler, this);
hirs8Sink.init(&hirsDemux.radChannels[7], hirs8Handler, this);
hirs9Sink.init(&hirsDemux.radChannels[8], hirs9Handler, this);
hirs10Sink.init(&hirsDemux.radChannels[9], hirs10Handler, this);
hirs11Sink.init(&hirsDemux.radChannels[10], hirs11Handler, this);
hirs12Sink.init(&hirsDemux.radChannels[11], hirs12Handler, this);
hirs13Sink.init(&hirsDemux.radChannels[12], hirs13Handler, this);
hirs14Sink.init(&hirsDemux.radChannels[13], hirs14Handler, this);
hirs15Sink.init(&hirsDemux.radChannels[14], hirs15Handler, this);
hirs16Sink.init(&hirsDemux.radChannels[15], hirs16Handler, this);
hirs17Sink.init(&hirsDemux.radChannels[16], hirs17Handler, this);
hirs18Sink.init(&hirsDemux.radChannels[17], hirs18Handler, this);
hirs19Sink.init(&hirsDemux.radChannels[18], hirs19Handler, this);
hirs20Sink.init(&hirsDemux.radChannels[19], hirs20Handler, this);
}
void select() {
_vfo->setSampleRate(NOAA_HRPT_VFO_SR, NOAA_HRPT_VFO_BW);
_vfo->setReference(ImGui::WaterfallVFO::REF_CENTER);
_vfo->setBandwidthLimits(NOAA_HRPT_VFO_BW, NOAA_HRPT_VFO_BW, true);
};
void start() {
demod.start();
split.start();
reshape.start();
visSink.start();
deframe.start();
manDec.start();
packer.start();
demux.start();
tipDemux.start();
hirsDemux.start();
avhrr1Sink.start();
avhrr2Sink.start();
avhrr3Sink.start();
avhrr4Sink.start();
avhrr5Sink.start();
sbuvSink.start();
dcsSink.start();
semSink.start();
aipSink.start();
hirs1Sink.start();
hirs2Sink.start();
hirs3Sink.start();
hirs4Sink.start();
hirs5Sink.start();
hirs6Sink.start();
hirs7Sink.start();
hirs8Sink.start();
hirs9Sink.start();
hirs10Sink.start();
hirs11Sink.start();
hirs12Sink.start();
hirs13Sink.start();
hirs14Sink.start();
hirs15Sink.start();
hirs16Sink.start();
hirs17Sink.start();
hirs18Sink.start();
hirs19Sink.start();
hirs20Sink.start();
compositeThread = std::thread(&NOAAHRPTDecoder::avhrrCompositeWorker, this);
};
void stop() {
compositeIn1.stopReader();
compositeIn1.stopWriter();
compositeIn2.stopReader();
compositeIn2.stopWriter();
demod.stop();
split.stop();
reshape.stop();
visSink.stop();
deframe.stop();
manDec.stop();
packer.stop();
demux.stop();
tipDemux.stop();
hirsDemux.stop();
avhrr1Sink.stop();
avhrr2Sink.stop();
avhrr3Sink.stop();
avhrr4Sink.stop();
avhrr5Sink.stop();
sbuvSink.stop();
dcsSink.stop();
semSink.stop();
aipSink.stop();
hirs1Sink.stop();
hirs2Sink.stop();
hirs3Sink.stop();
hirs4Sink.stop();
hirs5Sink.stop();
hirs6Sink.stop();
hirs7Sink.stop();
hirs8Sink.stop();
hirs9Sink.stop();
hirs10Sink.stop();
hirs11Sink.stop();
hirs12Sink.stop();
hirs13Sink.stop();
hirs14Sink.stop();
hirs15Sink.stop();
hirs16Sink.stop();
hirs17Sink.stop();
hirs18Sink.stop();
hirs19Sink.stop();
hirs20Sink.stop();
if (compositeThread.joinable()) {
compositeThread.join();
}
compositeIn1.clearReadStop();
compositeIn1.clearWriteStop();
compositeIn2.clearReadStop();
compositeIn2.clearWriteStop();
};
void setVFO(VFOManager::VFO* vfo) {
_vfo = vfo;
demod.setInput(_vfo->output);
};
virtual bool canRecord() {
return false;
}
// bool startRecording(std::string recPath) {
// };
// void stopRecording() {
// };
// bool isRecording() {
// };
void drawMenu(float menuWidth) {
ImGui::SetNextItemWidth(menuWidth);
symDiag.draw();
ImGui::Begin("NOAA HRPT Decoder");
ImGui::BeginTabBar("NOAAHRPTTabs");
if (ImGui::BeginTabItem("AVHRR RGB(221)")) {
ImGui::BeginChild("AVHRRRGBChild");
ImGui::SetNextItemWidth(ImGui::GetContentRegionAvailWidth());
avhrrRGBImage.draw();
ImGui::SetScrollHere(1.0f);
ImGui::EndChild();
ImGui::EndTabItem();
}
if (ImGui::BeginTabItem("AVHRR 1")) {
ImGui::BeginChild("AVHRR1Child");
ImGui::SetNextItemWidth(ImGui::GetContentRegionAvailWidth());
avhrr1Image.draw();
ImGui::SetScrollHere(1.0f);
ImGui::EndChild();
ImGui::EndTabItem();
}
if (ImGui::BeginTabItem("AVHRR 2")) {
ImGui::BeginChild("AVHRR2Child");
ImGui::SetNextItemWidth(ImGui::GetContentRegionAvailWidth());
avhrr2Image.draw();
ImGui::SetScrollHere(1.0f);
ImGui::EndChild();
ImGui::EndTabItem();
}
if (ImGui::BeginTabItem("AVHRR 3")) {
ImGui::BeginChild("AVHRR3Child");
ImGui::SetNextItemWidth(ImGui::GetContentRegionAvailWidth());
avhrr3Image.draw();
ImGui::SetScrollHere(1.0f);
ImGui::EndChild();
ImGui::EndTabItem();
}
if (ImGui::BeginTabItem("AVHRR 4")) {
ImGui::BeginChild("AVHRR4Child");
ImGui::SetNextItemWidth(ImGui::GetContentRegionAvailWidth());
avhrr4Image.draw();
ImGui::SetScrollHere(1.0f);
ImGui::EndChild();
ImGui::EndTabItem();
}
if (ImGui::BeginTabItem("AVHRR 5")) {
ImGui::BeginChild("AVHRR5Child");
ImGui::SetNextItemWidth(ImGui::GetContentRegionAvailWidth());
avhrr5Image.draw();
ImGui::SetScrollHere(1.0f);
ImGui::EndChild();
ImGui::EndTabItem();
}
if (ImGui::BeginTabItem("HIRS")) {
ImGui::BeginChild("HIRSChild");
ImGui::EndChild();
ImGui::EndTabItem();
}
ImGui::EndTabBar();
ImGui::End();
};
private:
// AVHRR Data Handlers
void avhrrCompositeWorker() {
compositeIn1.flush();
compositeIn2.flush();
while (true) {
if (compositeIn1.read() < 0) { return; }
if (compositeIn2.read() < 0) { return; }
uint8_t* buf = avhrrRGBImage.aquireNextLine();
float rg, b;
for (int i = 0; i < 2048; i++) {
b = ((float)compositeIn1.readBuf[i] * 255.0f) / 1024.0f;
rg = ((float)compositeIn2.readBuf[i] * 255.0f) / 1024.0f;
buf[(i*4)] = rg;
buf[(i*4) + 1] = rg;
buf[(i*4) + 2] = b;
buf[(i*4) + 3] = 255;
}
avhrrRGBImage.releaseNextLine();
compositeIn1.flush();
compositeIn2.flush();
}
}
static void avhrr1Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
uint8_t* buf = _this->avhrr1Image.aquireNextLine();
float val;
for (int i = 0; i < 2048; i++) {
val = ((float)data[i] * 255.0f) / 1024.0f;
buf[(i*4)] = val;
buf[(i*4) + 1] = val;
buf[(i*4) + 2] = val;
buf[(i*4) + 3] = 255;
}
_this->avhrr1Image.releaseNextLine();
memcpy(_this->compositeIn1.writeBuf, data, count * sizeof(uint16_t));
_this->compositeIn1.swap(count);
}
static void avhrr2Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
uint8_t* buf = _this->avhrr2Image.aquireNextLine();
float val;
for (int i = 0; i < 2048; i++) {
val = ((float)data[i] * 255.0f) / 1024.0f;
buf[(i*4)] = val;
buf[(i*4) + 1] = val;
buf[(i*4) + 2] = val;
buf[(i*4) + 3] = 255;
}
_this->avhrr2Image.releaseNextLine();
memcpy(_this->compositeIn2.writeBuf, data, count * sizeof(uint16_t));
_this->compositeIn2.swap(count);
}
static void avhrr3Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
uint8_t* buf = _this->avhrr3Image.aquireNextLine();
float val;
for (int i = 0; i < 2048; i++) {
val = ((float)data[i] * 255.0f) / 1024.0f;
buf[(i*4)] = val;
buf[(i*4) + 1] = val;
buf[(i*4) + 2] = val;
buf[(i*4) + 3] = 255;
}
_this->avhrr3Image.releaseNextLine();
}
static void avhrr4Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
uint8_t* buf = _this->avhrr4Image.aquireNextLine();
float val;
for (int i = 0; i < 2048; i++) {
val = ((float)data[i] * 255.0f) / 1024.0f;
buf[(i*4)] = val;
buf[(i*4) + 1] = val;
buf[(i*4) + 2] = val;
buf[(i*4) + 3] = 255;
}
_this->avhrr4Image.releaseNextLine();
}
static void avhrr5Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
uint8_t* buf = _this->avhrr5Image.aquireNextLine();
float val;
for (int i = 0; i < 2048; i++) {
val = ((float)data[i] * 255.0f) / 1024.0f;
buf[(i*4)] = val;
buf[(i*4) + 1] = val;
buf[(i*4) + 2] = val;
buf[(i*4) + 3] = 255;
}
_this->avhrr5Image.releaseNextLine();
}
// HIRS Data Handlers
static void hirs1Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs2Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs3Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs4Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs5Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs6Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs7Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs8Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs9Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs10Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs11Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs12Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs13Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs14Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs15Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs16Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs17Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs18Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs19Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void hirs20Handler(uint16_t* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
}
static void visHandler(float* data, int count, void* ctx) {
NOAAHRPTDecoder* _this = (NOAAHRPTDecoder*)ctx;
memcpy(_this->symDiag.aquireBuffer(), data, 1024 * sizeof(float));
_this->symDiag.releaseBuffer();
}
std::string _name;
std::string _recPath;
VFOManager::VFO* _vfo;
// DSP
dsp::PMDemod demod;
dsp::stream<float> visStream;
dsp::stream<float> dataStream;
dsp::Splitter<float> split;
dsp::Reshaper<float> reshape;
dsp::ManchesterDeframer deframe;
dsp::ManchesterDecoder manDec;
dsp::BitPacker packer;
dsp::noaa::HRPTDemux demux;
dsp::noaa::TIPDemux tipDemux;
dsp::noaa::HIRSDemux hirsDemux;
// AHVRR Handlers
dsp::HandlerSink<uint16_t> avhrr1Sink;
dsp::HandlerSink<uint16_t> avhrr2Sink;
dsp::HandlerSink<uint16_t> avhrr3Sink;
dsp::HandlerSink<uint16_t> avhrr4Sink;
dsp::HandlerSink<uint16_t> avhrr5Sink;
// (at the moment) Unused TIP handlers
dsp::NullSink<uint8_t> sbuvSink;
dsp::NullSink<uint8_t> dcsSink;
dsp::NullSink<uint8_t> semSink;
// (at the moment) Unused AIP handlers
dsp::NullSink<uint8_t> aipSink;
// HIRS Handlers
dsp::HandlerSink<uint16_t> hirs1Sink;
dsp::HandlerSink<uint16_t> hirs2Sink;
dsp::HandlerSink<uint16_t> hirs3Sink;
dsp::HandlerSink<uint16_t> hirs4Sink;
dsp::HandlerSink<uint16_t> hirs5Sink;
dsp::HandlerSink<uint16_t> hirs6Sink;
dsp::HandlerSink<uint16_t> hirs7Sink;
dsp::HandlerSink<uint16_t> hirs8Sink;
dsp::HandlerSink<uint16_t> hirs9Sink;
dsp::HandlerSink<uint16_t> hirs10Sink;
dsp::HandlerSink<uint16_t> hirs11Sink;
dsp::HandlerSink<uint16_t> hirs12Sink;
dsp::HandlerSink<uint16_t> hirs13Sink;
dsp::HandlerSink<uint16_t> hirs14Sink;
dsp::HandlerSink<uint16_t> hirs15Sink;
dsp::HandlerSink<uint16_t> hirs16Sink;
dsp::HandlerSink<uint16_t> hirs17Sink;
dsp::HandlerSink<uint16_t> hirs18Sink;
dsp::HandlerSink<uint16_t> hirs19Sink;
dsp::HandlerSink<uint16_t> hirs20Sink;
dsp::HandlerSink<float> visSink;
ImGui::LinePushImage avhrrRGBImage;
ImGui::LinePushImage avhrr1Image;
ImGui::LinePushImage avhrr2Image;
ImGui::LinePushImage avhrr3Image;
ImGui::LinePushImage avhrr4Image;
ImGui::LinePushImage avhrr5Image;
ImGui::SymbolDiagram symDiag;
dsp::stream<uint16_t> compositeIn1;
dsp::stream<uint16_t> compositeIn2;
std::thread compositeThread;
};