I'm a little lost again
This commit is contained in:
@@ -22,8 +22,9 @@ static constexpr double SCALE_FACTOR = 32767.0;
|
||||
class PSKModulator {
|
||||
public:
|
||||
PSKModulator(const double _sample_rate, const bool _is_frequency_hopping, const size_t _num_taps)
|
||||
: sample_rate(validateSampleRate(_sample_rate)), gain(1.0/sqrt(2.0)), is_frequency_hopping(_is_frequency_hopping), samples_per_symbol(static_cast<size_t>(sample_rate / SYMBOL_RATE)), phase_detector(symbolMap), srrc_filter(48, _sample_rate, SYMBOL_RATE, ROLLOFF_FACTOR) {
|
||||
: sample_rate(validateSampleRate(_sample_rate)), gain(1.0/sqrt(2.0)), is_frequency_hopping(_is_frequency_hopping), samples_per_symbol(static_cast<size_t>(sample_rate / SYMBOL_RATE)), srrc_filter(48, _sample_rate, SYMBOL_RATE, ROLLOFF_FACTOR) {
|
||||
initializeSymbolMap();
|
||||
phase_detector = PhaseDetector(symbolMap);
|
||||
}
|
||||
|
||||
std::vector<int16_t> modulate(const std::vector<uint8_t>& symbols) {
|
||||
@@ -78,7 +79,7 @@ public:
|
||||
|
||||
std::vector<uint8_t> demodulate(const std::vector<int16_t> passband_signal) {
|
||||
// Carrier recovery. initialize the Costas loop.
|
||||
CostasLoop costas_loop(sample_rate, symbolMap);
|
||||
CostasLoop costas_loop(CARRIER_FREQ, sample_rate, symbolMap, 5.0);
|
||||
|
||||
// Convert passband signal to doubles.
|
||||
std::vector<double> normalized_passband(passband_signal.size());
|
||||
@@ -91,8 +92,17 @@ public:
|
||||
|
||||
// Phase detection and symbol formation
|
||||
std::vector<uint8_t> baseband_symbols;
|
||||
for (size_t i = 0; i < baseband_IQ.size(); i++) {
|
||||
baseband_symbols.emplace_back(phase_detector.getSymbol(baseband_IQ[i]));
|
||||
size_t samples_per_symbol = sample_rate / SYMBOL_RATE;
|
||||
|
||||
for (size_t i = 0; i < baseband_IQ.size(); i += samples_per_symbol) {
|
||||
std::complex<double> symbol_avg(0.0, 0.0);
|
||||
for (size_t j = 0; j < samples_per_symbol; ++j) {
|
||||
symbol_avg += baseband_IQ[i + j];
|
||||
}
|
||||
symbol_avg /= static_cast<double>(samples_per_symbol);
|
||||
|
||||
// Detect symbol from averaged signal
|
||||
baseband_symbols.emplace_back(phase_detector.getSymbol(symbol_avg));
|
||||
}
|
||||
|
||||
return baseband_symbols;
|
||||
|
||||
@@ -7,6 +7,7 @@
|
||||
|
||||
class PhaseDetector {
|
||||
public:
|
||||
PhaseDetector() {}
|
||||
PhaseDetector(const std::vector<std::complex<double>>& _symbolMap) : symbolMap(_symbolMap) {}
|
||||
|
||||
uint8_t getSymbol(const std::complex<double>& input) {
|
||||
@@ -38,8 +39,8 @@ private:
|
||||
|
||||
class CostasLoop {
|
||||
public:
|
||||
CostasLoop(const double _sample_rate, const std::vector<std::complex<double>>& _symbolMap)
|
||||
: sample_rate(_sample_rate), k_factor(-5 / _sample_rate),
|
||||
CostasLoop(const double _carrier_freq, const double _sample_rate, const std::vector<std::complex<double>>& _symbolMap, const double _vco_gain)
|
||||
: carrier_freq(_carrier_freq), sample_rate(_sample_rate), vco_gain(_vco_gain), k_factor(-1 / (_sample_rate * _vco_gain)),
|
||||
prev_in_iir(0), prev_out_iir(0), prev_in_vco(0), feedback(1.0, 0.0),
|
||||
error_total(0), out_iir_total(0), in_vco_total(0),
|
||||
srrc_filter(SRRCFilter(48, _sample_rate, 2400, 0.35)) {}
|
||||
@@ -80,15 +81,19 @@ public:
|
||||
prev_in_vco = in_vco;
|
||||
|
||||
// Generate feedback signal for next iteration
|
||||
double feedback_real = std::cos(k_factor * in_vco);
|
||||
double feedback_imag = -std::sin(k_factor * in_vco);
|
||||
double feedback_real = std::cos(current_phase);
|
||||
double feedback_imag = -std::sin(current_phase);
|
||||
feedback = std::complex<double>(feedback_real, feedback_imag);
|
||||
|
||||
current_phase += 2 * M_PI * carrier_freq / sample_rate + k_factor * in_vco;
|
||||
if (current_phase > 2 * M_PI) current_phase -= 2 * M_PI;
|
||||
}
|
||||
|
||||
return output_signal;
|
||||
}
|
||||
|
||||
private:
|
||||
double carrier_freq;
|
||||
double sample_rate;
|
||||
double k_factor;
|
||||
double prev_in_iir;
|
||||
@@ -99,6 +104,7 @@ private:
|
||||
double out_iir_total;
|
||||
double in_vco_total;
|
||||
SRRCFilter srrc_filter;
|
||||
double vco_gain;
|
||||
|
||||
std::complex<double> limiter(const std::complex<double>& sample) const {
|
||||
double limited_I = std::clamp(sample.real(), -1.0, 1.0);
|
||||
|
||||
Reference in New Issue
Block a user