kopia lustrzana https://github.com/AlexandreRouma/SDRPlusPlus
new radio code + fixes
rodzic
2daaf00cb3
commit
2748c13142
|
@ -342,7 +342,6 @@ jobs:
|
||||||
mv sdrpp_ubuntu_groovy_amd64/sdrpp_debian_amd64.deb sdrpp_all/sdrpp_ubuntu_groovy_amd64.deb &&
|
mv sdrpp_ubuntu_groovy_amd64/sdrpp_debian_amd64.deb sdrpp_all/sdrpp_ubuntu_groovy_amd64.deb &&
|
||||||
mv sdrpp_ubuntu_hirsute_amd64/sdrpp_debian_amd64.deb sdrpp_all/sdrpp_ubuntu_hirsute_amd64.deb &&
|
mv sdrpp_ubuntu_hirsute_amd64/sdrpp_debian_amd64.deb sdrpp_all/sdrpp_ubuntu_hirsute_amd64.deb &&
|
||||||
mv sdrpp_ubuntu_impish_amd64/sdrpp_debian_amd64.deb sdrpp_all/sdrpp_ubuntu_impish_amd64.deb
|
mv sdrpp_ubuntu_impish_amd64/sdrpp_debian_amd64.deb sdrpp_all/sdrpp_ubuntu_impish_amd64.deb
|
||||||
mv sdrpp_raspios_bullseye_arm32/sdrpp_raspios_arm32.deb sdrpp_all/sdrpp_raspios_bullseye_arm32.deb
|
|
||||||
|
|
||||||
- uses: actions/upload-artifact@v2
|
- uses: actions/upload-artifact@v2
|
||||||
with:
|
with:
|
||||||
|
|
|
@ -805,9 +805,9 @@ namespace dsp {
|
||||||
generic_hier_block<StereoFMDemod>::_block_init = true;
|
generic_hier_block<StereoFMDemod>::_block_init = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
void setInput(stream<float>* input) {
|
void setInput(stream<complex_t>* input) {
|
||||||
assert(generic_hier_block<StereoFMDemod>::_block_init);
|
assert(generic_hier_block<StereoFMDemod>::_block_init);
|
||||||
r2c.setInput(input);
|
demod.setInput(input);
|
||||||
}
|
}
|
||||||
|
|
||||||
void setDeviation(float deviation) {
|
void setDeviation(float deviation) {
|
||||||
|
|
|
@ -0,0 +1,131 @@
|
||||||
|
#pragma once
|
||||||
|
#include <dsp/block.h>
|
||||||
|
#include <dsp/utils/window_functions.h>
|
||||||
|
#include <fftw3.h>
|
||||||
|
|
||||||
|
#define NR_TAP_COUNT 4096
|
||||||
|
|
||||||
|
namespace dsp {
|
||||||
|
class FFTNoiseReduction : public generic_block<FFTNoiseReduction> {
|
||||||
|
public:
|
||||||
|
FFTNoiseReduction() {}
|
||||||
|
|
||||||
|
FFTNoiseReduction(stream<float>* in) { init(in); }
|
||||||
|
|
||||||
|
~FFTNoiseReduction() {
|
||||||
|
if (!generic_block<FFTNoiseReduction>::_block_init) { return; }
|
||||||
|
generic_block<FFTNoiseReduction>::stop();
|
||||||
|
fftwf_destroy_plan(forwardPlan);
|
||||||
|
fftwf_destroy_plan(backwardPlan);
|
||||||
|
fftwf_free(delay);
|
||||||
|
fftwf_free(fft_in);
|
||||||
|
fftwf_free(fft_window);
|
||||||
|
fftwf_free(amp_buf);
|
||||||
|
fftwf_free(fft_cout);
|
||||||
|
fftwf_free(fft_fout);
|
||||||
|
}
|
||||||
|
|
||||||
|
void init(stream<float>* in) {
|
||||||
|
_in = in;
|
||||||
|
|
||||||
|
delay = (float*)fftwf_malloc(sizeof(float)*STREAM_BUFFER_SIZE);
|
||||||
|
fft_in = (float*)fftwf_malloc(sizeof(float)*NR_TAP_COUNT);
|
||||||
|
fft_window = (float*)fftwf_malloc(sizeof(float)*NR_TAP_COUNT);
|
||||||
|
amp_buf = (float*)fftwf_malloc(sizeof(float)*NR_TAP_COUNT);
|
||||||
|
fft_cout = (complex_t*)fftwf_malloc(sizeof(complex_t)*NR_TAP_COUNT);
|
||||||
|
fft_fout = (float*)fftwf_malloc(sizeof(float)*NR_TAP_COUNT);
|
||||||
|
delay_start = &delay[NR_TAP_COUNT];
|
||||||
|
|
||||||
|
memset(delay, 0, sizeof(float)*STREAM_BUFFER_SIZE);
|
||||||
|
memset(fft_in, 0, sizeof(float)*NR_TAP_COUNT);
|
||||||
|
memset(amp_buf, 0, sizeof(float)*NR_TAP_COUNT);
|
||||||
|
memset(fft_cout, 0, sizeof(complex_t)*NR_TAP_COUNT);
|
||||||
|
memset(fft_fout, 0, sizeof(float)*NR_TAP_COUNT);
|
||||||
|
|
||||||
|
for (int i = 0; i < NR_TAP_COUNT; i++) {
|
||||||
|
fft_window[i] = window_function::blackman(i, NR_TAP_COUNT - 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
forwardPlan = fftwf_plan_dft_r2c_1d(NR_TAP_COUNT, fft_in, (fftwf_complex*)fft_cout, FFTW_ESTIMATE);
|
||||||
|
backwardPlan = fftwf_plan_dft_c2r_1d(NR_TAP_COUNT, (fftwf_complex*)fft_cout, fft_fout, FFTW_ESTIMATE);
|
||||||
|
|
||||||
|
generic_block<FFTNoiseReduction>::registerInput(_in);
|
||||||
|
generic_block<FFTNoiseReduction>::registerOutput(&out);
|
||||||
|
generic_block<FFTNoiseReduction>::_block_init = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void setInput(stream<float>* in) {
|
||||||
|
assert(generic_block<FFTNoiseReduction>::_block_init);
|
||||||
|
std::lock_guard<std::mutex> lck(generic_block<FFTNoiseReduction>::ctrlMtx);
|
||||||
|
generic_block<FFTNoiseReduction>::tempStop();
|
||||||
|
generic_block<FFTNoiseReduction>::unregisterInput(_in);
|
||||||
|
_in = in;
|
||||||
|
generic_block<FFTNoiseReduction>::registerInput(_in);
|
||||||
|
generic_block<FFTNoiseReduction>::tempStart();
|
||||||
|
}
|
||||||
|
|
||||||
|
int run() {
|
||||||
|
int count = _in->read();
|
||||||
|
if (count < 0) { return -1; }
|
||||||
|
|
||||||
|
// Bypass
|
||||||
|
if (!bypass) {
|
||||||
|
memcpy(out.writeBuf, _in->readBuf, count * sizeof(float));
|
||||||
|
_in->flush();
|
||||||
|
if (!out.swap(count)) { return -1; }
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write to delay buffer
|
||||||
|
memcpy(delay_start, _in->readBuf, count * sizeof(float));
|
||||||
|
|
||||||
|
// Iterate the FFT
|
||||||
|
for (int i = 0; i < count; i++) {
|
||||||
|
// Apply windows
|
||||||
|
volk_32f_x2_multiply_32f(fft_in, &delay[i], fft_window, NR_TAP_COUNT);
|
||||||
|
|
||||||
|
// Do forward FFT
|
||||||
|
fftwf_execute(forwardPlan);
|
||||||
|
|
||||||
|
// Process bins here
|
||||||
|
volk_32fc_magnitude_32f(amp_buf, (lv_32fc_t*)fft_cout, NR_TAP_COUNT/2);
|
||||||
|
for (int j = 1; j < NR_TAP_COUNT/2; j++) {
|
||||||
|
if (log10f(amp_buf[0]) < level) {
|
||||||
|
fft_cout[j] = {0, 0};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Do reverse FFT and get first element
|
||||||
|
fftwf_execute(backwardPlan);
|
||||||
|
out.writeBuf[i] = fft_fout[NR_TAP_COUNT/2];
|
||||||
|
}
|
||||||
|
|
||||||
|
volk_32f_s32f_multiply_32f(out.writeBuf, out.writeBuf, 1.0f/(float)NR_TAP_COUNT, count);
|
||||||
|
|
||||||
|
// Copy last values to delay
|
||||||
|
memmove(delay, &delay[count], NR_TAP_COUNT * sizeof(float));
|
||||||
|
|
||||||
|
_in->flush();
|
||||||
|
if (!out.swap(count)) { return -1; }
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool bypass = true;
|
||||||
|
stream<float> out;
|
||||||
|
|
||||||
|
float level = 0.0f;
|
||||||
|
|
||||||
|
private:
|
||||||
|
stream<float>* _in;
|
||||||
|
fftwf_plan forwardPlan;
|
||||||
|
fftwf_plan backwardPlan;
|
||||||
|
float* delay;
|
||||||
|
float* fft_in;
|
||||||
|
float* fft_window;
|
||||||
|
float* amp_buf;
|
||||||
|
float* delay_start;
|
||||||
|
complex_t* fft_cout;
|
||||||
|
float* fft_fout;
|
||||||
|
|
||||||
|
};
|
||||||
|
}
|
|
@ -0,0 +1,21 @@
|
||||||
|
cmake_minimum_required(VERSION 3.13)
|
||||||
|
project(new_radio)
|
||||||
|
|
||||||
|
file(GLOB_RECURSE SRC "src/*.cpp")
|
||||||
|
|
||||||
|
add_library(new_radio SHARED ${SRC})
|
||||||
|
target_link_libraries(new_radio PRIVATE sdrpp_core)
|
||||||
|
set_target_properties(new_radio PROPERTIES PREFIX "")
|
||||||
|
|
||||||
|
target_include_directories(new_radio PRIVATE "src/")
|
||||||
|
|
||||||
|
if (MSVC)
|
||||||
|
target_compile_options(new_radio PRIVATE /O2 /Ob2 /std:c++17 /EHsc)
|
||||||
|
elseif (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
|
||||||
|
target_compile_options(new_radio PRIVATE -O3 -std=c++17 -Wno-unused-command-line-argument -undefined dynamic_lookup)
|
||||||
|
else ()
|
||||||
|
target_compile_options(new_radio PRIVATE -O3 -std=c++17)
|
||||||
|
endif ()
|
||||||
|
|
||||||
|
# Install directives
|
||||||
|
install(TARGETS new_radio DESTINATION lib/sdrpp/plugins)
|
|
@ -0,0 +1,31 @@
|
||||||
|
#pragma once
|
||||||
|
#include <dsp/stream.h>
|
||||||
|
#include <dsp/types.h>
|
||||||
|
#include <gui/widgets/waterfall.h>
|
||||||
|
#include <config.h>
|
||||||
|
#include "radio_module.h"
|
||||||
|
|
||||||
|
namespace demod {
|
||||||
|
class Demodulator {
|
||||||
|
public:
|
||||||
|
virtual ~Demodulator() {}
|
||||||
|
virtual void init(std::string name, ConfigManager* config, dsp::stream<dsp::complex_t>* input, double bandwidth) = 0;
|
||||||
|
virtual void start() = 0;
|
||||||
|
virtual void stop() = 0;
|
||||||
|
virtual void showMenu() = 0;
|
||||||
|
virtual void setBandwidth(double bandwidth) = 0;
|
||||||
|
virtual void setInput(dsp::stream<dsp::complex_t>* input) = 0;
|
||||||
|
virtual const char* getName() = 0;
|
||||||
|
virtual double getIFSampleRate() = 0;
|
||||||
|
virtual double getAFSampleRate() = 0;
|
||||||
|
virtual double getDefaultBandwidth() = 0;
|
||||||
|
virtual double getMinBandwidth() = 0;
|
||||||
|
virtual double getMaxBandwidth() = 0;
|
||||||
|
virtual double getMaxAFBandwidth() = 0;
|
||||||
|
virtual double getDefaultSnapInterval() = 0;
|
||||||
|
virtual int getVFOReference() = 0;
|
||||||
|
virtual dsp::stream<dsp::stereo_t>* getOutput() = 0;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#include "demodulators/wfm.h"
|
|
@ -0,0 +1,120 @@
|
||||||
|
#pragma once
|
||||||
|
#include "../demod.h"
|
||||||
|
#include <dsp/demodulator.h>
|
||||||
|
#include <dsp/filter.h>
|
||||||
|
|
||||||
|
namespace demod {
|
||||||
|
class WFM : public Demodulator {
|
||||||
|
public:
|
||||||
|
WFM() {}
|
||||||
|
|
||||||
|
WFM(std::string name, ConfigManager* config, dsp::stream<dsp::complex_t>* input, double bandwidth) {
|
||||||
|
init(name, config, input, bandwidth);
|
||||||
|
}
|
||||||
|
|
||||||
|
~WFM() {
|
||||||
|
stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
void init(std::string name, ConfigManager* config, dsp::stream<dsp::complex_t>* input, double bandwidth) {
|
||||||
|
this->name = name;
|
||||||
|
_config = config;
|
||||||
|
|
||||||
|
// Load config
|
||||||
|
_config->acquire();
|
||||||
|
bool modified =false;
|
||||||
|
if (!config->conf[name].contains(getName())) {
|
||||||
|
config->conf[name][getName()]["deempMode"] = 0;
|
||||||
|
config->conf[name][getName()]["stereo"] = false;
|
||||||
|
modified = true;
|
||||||
|
}
|
||||||
|
if (config->conf[name][getName()].contains("deempMode")) {
|
||||||
|
deempMode = config->conf[name][getName()]["deempMode"];
|
||||||
|
}
|
||||||
|
if (config->conf[name][getName()].contains("stereo")) {
|
||||||
|
stereo = config->conf[name][getName()]["stereo"];
|
||||||
|
}
|
||||||
|
_config->release(modified);
|
||||||
|
|
||||||
|
// Define structure
|
||||||
|
demod.init(input, getIFSampleRate(), bandwidth / 2.0f);
|
||||||
|
demodStereo.init(input, getIFSampleRate(), bandwidth / 2.0f);
|
||||||
|
//deemp.init(stereo ? demodStereo.out : &demod.out, getAFSampleRate(), 50e-6);
|
||||||
|
//deemp.bypass = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
void start() {
|
||||||
|
stereo ? demodStereo.start() : demod.start();
|
||||||
|
//deemp.start();
|
||||||
|
}
|
||||||
|
|
||||||
|
void stop() {
|
||||||
|
demod.stop();
|
||||||
|
demodStereo.stop();
|
||||||
|
//deemp.stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
void showMenu() {
|
||||||
|
if (ImGui::Checkbox(("Stereo##_radio_wfm_stereo_" + name).c_str(), &stereo)) {
|
||||||
|
setStereo(stereo);
|
||||||
|
_config->acquire();
|
||||||
|
_config->conf[name][getName()]["stereo"] = stereo;
|
||||||
|
_config->release(true);
|
||||||
|
}
|
||||||
|
//ImGui::Checkbox("Deemp bypass", &deemp.bypass);
|
||||||
|
}
|
||||||
|
|
||||||
|
void setBandwidth(double bandwidth) {
|
||||||
|
demod.setDeviation(bandwidth / 2.0f);
|
||||||
|
demodStereo.setDeviation(bandwidth / 2.0f);
|
||||||
|
}
|
||||||
|
|
||||||
|
void setInput(dsp::stream<dsp::complex_t>* input) {
|
||||||
|
demod.setInput(input);
|
||||||
|
demodStereo.setInput(input);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============= INFO =============
|
||||||
|
|
||||||
|
const char* getName() { return "WFM"; }
|
||||||
|
double getIFSampleRate() { return 250000.0; }
|
||||||
|
double getAFSampleRate() { return getIFSampleRate(); }
|
||||||
|
double getDefaultBandwidth() { return 150000.0; }
|
||||||
|
double getMinBandwidth() { return 50000.0; }
|
||||||
|
double getMaxBandwidth() { return getIFSampleRate(); }
|
||||||
|
double getMaxAFBandwidth() { return 16000.0; }
|
||||||
|
double getDefaultSnapInterval() { return 100000.0; }
|
||||||
|
int getVFOReference() { return ImGui::WaterfallVFO::REF_CENTER; }
|
||||||
|
dsp::stream<dsp::stereo_t>* getOutput() { return demodStereo.out; }
|
||||||
|
|
||||||
|
// ============= DEDICATED FUNCTIONS =============
|
||||||
|
|
||||||
|
void setStereo(bool _stereo) {
|
||||||
|
stereo = _stereo;
|
||||||
|
if (stereo) {
|
||||||
|
demod.stop();
|
||||||
|
//deemp.setInput(demodStereo.out);
|
||||||
|
demodStereo.start();
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
demodStereo.stop();
|
||||||
|
//deemp.setInput(&demod.out);
|
||||||
|
demod.start();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
dsp::FMDemod demod;
|
||||||
|
dsp::StereoFMDemod demodStereo;
|
||||||
|
//dsp::BFMDeemp deemp;
|
||||||
|
|
||||||
|
RadioModule* _rad = NULL;
|
||||||
|
ConfigManager* _config = NULL;
|
||||||
|
|
||||||
|
int deempMode;
|
||||||
|
bool stereo;
|
||||||
|
|
||||||
|
std::string name;
|
||||||
|
|
||||||
|
};
|
||||||
|
}
|
|
@ -0,0 +1,30 @@
|
||||||
|
#include "radio_module.h"
|
||||||
|
#include <options.h>
|
||||||
|
|
||||||
|
SDRPP_MOD_INFO {
|
||||||
|
/* Name: */ "new_radio",
|
||||||
|
/* Description: */ "Analog radio decoder",
|
||||||
|
/* Author: */ "Ryzerth",
|
||||||
|
/* Version: */ 2, 0, 0,
|
||||||
|
/* Max instances */ -1
|
||||||
|
};
|
||||||
|
|
||||||
|
MOD_EXPORT void _INIT_() {
|
||||||
|
json def = json({});
|
||||||
|
config.setPath(options::opts.root + "/new_radio_config.json");
|
||||||
|
config.load(def);
|
||||||
|
config.enableAutoSave();
|
||||||
|
}
|
||||||
|
|
||||||
|
MOD_EXPORT ModuleManager::Instance* _CREATE_INSTANCE_(std::string name) {
|
||||||
|
return new RadioModule(name);
|
||||||
|
}
|
||||||
|
|
||||||
|
MOD_EXPORT void _DELETE_INSTANCE_(void* instance) {
|
||||||
|
delete (RadioModule*)instance;
|
||||||
|
}
|
||||||
|
|
||||||
|
MOD_EXPORT void _END_() {
|
||||||
|
config.disableAutoSave();
|
||||||
|
config.save();
|
||||||
|
}
|
|
@ -0,0 +1,281 @@
|
||||||
|
#pragma once
|
||||||
|
#include <imgui.h>
|
||||||
|
#include <module.h>
|
||||||
|
#include <gui/gui.h>
|
||||||
|
#include <gui/style.h>
|
||||||
|
#include <signal_path/signal_path.h>
|
||||||
|
#include <config.h>
|
||||||
|
|
||||||
|
ConfigManager config;
|
||||||
|
|
||||||
|
#define CONCAT(a, b) ((std::string(a) + b).c_str())
|
||||||
|
|
||||||
|
class RadioModule;
|
||||||
|
#include "demod.h"
|
||||||
|
|
||||||
|
class RadioModule : public ModuleManager::Instance {
|
||||||
|
public:
|
||||||
|
RadioModule(std::string name) {
|
||||||
|
this->name = name;
|
||||||
|
|
||||||
|
// Initialize the config if it doesn't exist
|
||||||
|
config.acquire();
|
||||||
|
if (!config.conf.contains(name)) {
|
||||||
|
config.conf[name]["selectedDemodId"] = 1;
|
||||||
|
}
|
||||||
|
selectedDemodID = config.conf[name]["selectedDemodId"];
|
||||||
|
config.release(true);
|
||||||
|
|
||||||
|
// Create demodulator instances
|
||||||
|
demods.fill(NULL);
|
||||||
|
demods[RADIO_DEMOD_WFM] = new demod::WFM();
|
||||||
|
|
||||||
|
// Initialize the VFO
|
||||||
|
vfo = sigpath::vfoManager.createVFO(name, ImGui::WaterfallVFO::REF_CENTER, 0, 200000, 200000, 50000, 200000, false);
|
||||||
|
onUserChangedBandwidthHandler.handler = vfoUserChangedBandwidthHandler;
|
||||||
|
onUserChangedBandwidthHandler.ctx = this;
|
||||||
|
vfo->wtfVFO->onUserChangedBandwidth.bindHandler(&onUserChangedBandwidthHandler);
|
||||||
|
|
||||||
|
// Initialize the sink
|
||||||
|
srChangeHandler.ctx = this;
|
||||||
|
srChangeHandler.handler = sampleRateChangeHandler;
|
||||||
|
stream.init(&deemp.out, &srChangeHandler, audioSampleRate);
|
||||||
|
sigpath::sinkManager.registerStream(name, &stream);
|
||||||
|
|
||||||
|
// Load configuration for all demodulators
|
||||||
|
for (auto& demod : demods) {
|
||||||
|
if (!demod) { continue; }
|
||||||
|
|
||||||
|
// Default config
|
||||||
|
double bw = demod->getDefaultBandwidth();
|
||||||
|
if (!config.conf[name].contains(demod->getName())) {
|
||||||
|
config.conf[name][demod->getName()]["bandwidth"] = bw;
|
||||||
|
config.conf[name][demod->getName()]["snapInterval"] = demod->getDefaultSnapInterval();
|
||||||
|
config.conf[name][demod->getName()]["squelchLevel"] = MIN_SQUELCH;
|
||||||
|
}
|
||||||
|
bw = std::clamp<double>(bw, demod->getMinBandwidth(), demod->getMaxBandwidth());
|
||||||
|
|
||||||
|
// Initialize
|
||||||
|
demod->init(name, &config, &squelch.out, bw);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initialize DSP
|
||||||
|
squelch.init(vfo->output, MIN_SQUELCH);
|
||||||
|
win.init(24000, 24000, 48000);
|
||||||
|
resamp.init(NULL, &win, 250000, 48000);
|
||||||
|
deemp.init(&resamp.out, 48000, 50e-6);
|
||||||
|
deemp.bypass = false;
|
||||||
|
|
||||||
|
// Select the demodulator
|
||||||
|
selectDemodByID((DemodID)selectedDemodID);
|
||||||
|
|
||||||
|
// Start DSP
|
||||||
|
squelch.start();
|
||||||
|
resamp.start();
|
||||||
|
deemp.start();
|
||||||
|
stream.start();
|
||||||
|
|
||||||
|
gui::menu.registerEntry(name, menuHandler, this, NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
~RadioModule() {
|
||||||
|
gui::menu.removeEntry(name);
|
||||||
|
}
|
||||||
|
|
||||||
|
void postInit() {}
|
||||||
|
|
||||||
|
void enable() {
|
||||||
|
enabled = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void disable() {
|
||||||
|
enabled = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool isEnabled() {
|
||||||
|
return enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string name;
|
||||||
|
|
||||||
|
enum DemodID {
|
||||||
|
RADIO_DEMOD_NFM,
|
||||||
|
RADIO_DEMOD_WFM,
|
||||||
|
RADIO_DEMOD_AM,
|
||||||
|
RADIO_DEMOD_DSB,
|
||||||
|
RADIO_DEMOD_USB,
|
||||||
|
RADIO_DEMOD_CW,
|
||||||
|
RADIO_DEMOD_LSB,
|
||||||
|
RADIO_DEMOD_RAW,
|
||||||
|
_RADIO_DEMOD_COUNT,
|
||||||
|
};
|
||||||
|
|
||||||
|
private:
|
||||||
|
static void menuHandler(void* ctx) {
|
||||||
|
RadioModule* _this = (RadioModule*)ctx;
|
||||||
|
|
||||||
|
if (!_this->enabled) { style::beginDisabled(); }
|
||||||
|
|
||||||
|
float menuWidth = ImGui::GetContentRegionAvailWidth();
|
||||||
|
ImGui::BeginGroup();
|
||||||
|
|
||||||
|
// TODO: Change VFO ref in signal path
|
||||||
|
|
||||||
|
ImGui::Columns(4, CONCAT("RadioModeColumns##_", _this->name), false);
|
||||||
|
if (ImGui::RadioButton(CONCAT("NFM##_", _this->name), _this->selectedDemodID == 0) && _this->selectedDemodID != 0) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_NFM);
|
||||||
|
}
|
||||||
|
if (ImGui::RadioButton(CONCAT("WFM##_", _this->name), _this->selectedDemodID == 1) && _this->selectedDemodID != 1) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_WFM);
|
||||||
|
}
|
||||||
|
ImGui::NextColumn();
|
||||||
|
if (ImGui::RadioButton(CONCAT("AM##_", _this->name), _this->selectedDemodID == 2) && _this->selectedDemodID != 2) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_AM);
|
||||||
|
}
|
||||||
|
if (ImGui::RadioButton(CONCAT("DSB##_", _this->name), _this->selectedDemodID == 3) && _this->selectedDemodID != 3) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_DSB);
|
||||||
|
}
|
||||||
|
ImGui::NextColumn();
|
||||||
|
if (ImGui::RadioButton(CONCAT("USB##_", _this->name), _this->selectedDemodID == 4) && _this->selectedDemodID != 4) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_USB);
|
||||||
|
}
|
||||||
|
if (ImGui::RadioButton(CONCAT("CW##_", _this->name), _this->selectedDemodID == 5) && _this->selectedDemodID != 5) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_CW);
|
||||||
|
};
|
||||||
|
ImGui::NextColumn();
|
||||||
|
if (ImGui::RadioButton(CONCAT("LSB##_", _this->name), _this->selectedDemodID == 6) && _this->selectedDemodID != 6) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_LSB);
|
||||||
|
}
|
||||||
|
if (ImGui::RadioButton(CONCAT("RAW##_", _this->name), _this->selectedDemodID == 7) && _this->selectedDemodID != 7) {
|
||||||
|
_this->selectDemodByID(RADIO_DEMOD_RAW);
|
||||||
|
};
|
||||||
|
ImGui::Columns(1, CONCAT("EndRadioModeColumns##_", _this->name), false);
|
||||||
|
|
||||||
|
ImGui::EndGroup();
|
||||||
|
|
||||||
|
_this->selectedDemod->showMenu();
|
||||||
|
|
||||||
|
ImGui::LeftLabel("Squelch");
|
||||||
|
ImGui::SetNextItemWidth(menuWidth - ImGui::GetCursorPosX());
|
||||||
|
if (ImGui::SliderFloat(("##_radio_sqelch_" + _this->name).c_str(), &_this->squelchLevel, _this->MIN_SQUELCH, _this->MAX_SQUELCH, "%.3fdB")) {
|
||||||
|
_this->squelch.setLevel(_this->squelchLevel);
|
||||||
|
config.acquire();
|
||||||
|
config.conf[_this->name][_this->selectedDemod->getName()]["squelchLevel"] = _this->squelchLevel;
|
||||||
|
config.release(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!_this->enabled) { style::endDisabled(); }
|
||||||
|
}
|
||||||
|
|
||||||
|
void selectDemodByID(DemodID id) {
|
||||||
|
demod::Demodulator* demod = demods[id];
|
||||||
|
if (!demod) {
|
||||||
|
spdlog::error("Demodulator {0} not implemented", id);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
selectedDemodID = id;
|
||||||
|
selectDemod(demod);
|
||||||
|
}
|
||||||
|
|
||||||
|
void selectDemod(demod::Demodulator* demod) {
|
||||||
|
// Stopcurrently selected demodulator and select new
|
||||||
|
if (selectedDemod) { selectedDemod->stop(); }
|
||||||
|
selectedDemod = demod;
|
||||||
|
|
||||||
|
// Load config
|
||||||
|
bandwidth = selectedDemod->getDefaultBandwidth();
|
||||||
|
double minBandwidth = selectedDemod->getMinBandwidth();
|
||||||
|
double maxBandwidth = selectedDemod->getMaxBandwidth();
|
||||||
|
snapInterval = selectedDemod->getDefaultSnapInterval();
|
||||||
|
squelchLevel = MIN_SQUELCH;
|
||||||
|
if (config.conf[name][selectedDemod->getName()].contains("snapInterval")) {
|
||||||
|
bandwidth = config.conf[name][selectedDemod->getName()]["bandwidth"];
|
||||||
|
bandwidth = std::clamp<double>(bandwidth, minBandwidth, maxBandwidth);
|
||||||
|
}
|
||||||
|
if (config.conf[name][selectedDemod->getName()].contains("snapInterval")) {
|
||||||
|
snapInterval = config.conf[name][selectedDemod->getName()]["snapInterval"];
|
||||||
|
}
|
||||||
|
if (config.conf[name][selectedDemod->getName()].contains("squelchLevel")) {
|
||||||
|
squelchLevel = config.conf[name][selectedDemod->getName()]["squelchLevel"];
|
||||||
|
}
|
||||||
|
|
||||||
|
// Configure VFO
|
||||||
|
if (vfo) {
|
||||||
|
vfo->setBandwidthLimits(minBandwidth, maxBandwidth, false);
|
||||||
|
vfo->setReference(selectedDemod->getVFOReference());
|
||||||
|
vfo->setSnapInterval(snapInterval);
|
||||||
|
vfo->setSampleRate(selectedDemod->getIFSampleRate(), bandwidth);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Configure squelch
|
||||||
|
squelch.setLevel(squelchLevel);
|
||||||
|
|
||||||
|
// Configure resampler
|
||||||
|
resamp.stop();
|
||||||
|
resamp.setInput(selectedDemod->getOutput());
|
||||||
|
resamp.setInSampleRate(selectedDemod->getAFSampleRate());
|
||||||
|
setAudioSampleRate(audioSampleRate);
|
||||||
|
resamp.start();
|
||||||
|
|
||||||
|
// Start new demodulator
|
||||||
|
selectedDemod->start();
|
||||||
|
}
|
||||||
|
|
||||||
|
void setBandwidth(double bw) {
|
||||||
|
bandwidth = bw;
|
||||||
|
if (!selectedDemod) { return; }
|
||||||
|
selectedDemod->setBandwidth(bandwidth);
|
||||||
|
config.acquire();
|
||||||
|
config.conf[name][selectedDemod->getName()]["bandwidth"] = bandwidth;
|
||||||
|
config.release(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
void setAudioSampleRate(double sr) {
|
||||||
|
audioSampleRate = sr;
|
||||||
|
if (!selectedDemod) { return; }
|
||||||
|
float audioBW = std::min<float>(selectedDemod->getMaxAFBandwidth(), audioSampleRate / 2.0f);
|
||||||
|
resamp.stop();
|
||||||
|
resamp.setOutSampleRate(audioSampleRate);
|
||||||
|
win.setSampleRate(audioSampleRate * resamp.getInterpolation());
|
||||||
|
win.setCutoff(audioBW);
|
||||||
|
win.setTransWidth(audioBW);
|
||||||
|
resamp.updateWindow(&win);
|
||||||
|
resamp.start();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void vfoUserChangedBandwidthHandler(double newBw, void* ctx) {
|
||||||
|
RadioModule* _this = (RadioModule*)ctx;
|
||||||
|
_this->setBandwidth(newBw);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void sampleRateChangeHandler(float sampleRate, void* ctx) {
|
||||||
|
RadioModule* _this = (RadioModule*)ctx;
|
||||||
|
_this->setAudioSampleRate(sampleRate);
|
||||||
|
}
|
||||||
|
|
||||||
|
EventHandler<double> onUserChangedBandwidthHandler;
|
||||||
|
VFOManager::VFO* vfo;
|
||||||
|
dsp::Squelch squelch;
|
||||||
|
|
||||||
|
dsp::filter_window::BlackmanWindow win;
|
||||||
|
dsp::PolyphaseResampler<dsp::stereo_t> resamp;
|
||||||
|
dsp::BFMDeemp deemp;
|
||||||
|
|
||||||
|
EventHandler<float> srChangeHandler;
|
||||||
|
SinkManager::Stream stream;
|
||||||
|
|
||||||
|
std::array<demod::Demodulator*, _RADIO_DEMOD_COUNT> demods;
|
||||||
|
demod::Demodulator* selectedDemod = NULL;
|
||||||
|
|
||||||
|
double audioSampleRate = 48000.0;
|
||||||
|
double bandwidth;
|
||||||
|
double snapInterval;
|
||||||
|
float squelchLevel;
|
||||||
|
int selectedDemodID = 1;
|
||||||
|
|
||||||
|
const double MIN_SQUELCH = -100.0;
|
||||||
|
const double MAX_SQUELCH = 0.0;
|
||||||
|
|
||||||
|
bool enabled = true;
|
||||||
|
|
||||||
|
};
|
Ładowanie…
Reference in New Issue