kopia lustrzana https://github.com/AlexandreRouma/SDRPlusPlus
90 wiersze
3.0 KiB
C++
90 wiersze
3.0 KiB
C++
#pragma once
|
|
#include "../processor.h"
|
|
#include "../channel/frequency_xlator.h"
|
|
#include "../convert/complex_to_real.h"
|
|
#include "../loop/agc.h"
|
|
#include "../convert/mono_to_stereo.h"
|
|
|
|
namespace dsp::demod {
|
|
template <class T>
|
|
class CW : public Processor<complex_t, T> {
|
|
using base_type = Processor<complex_t, T>;
|
|
public:
|
|
CW() {}
|
|
|
|
CW(stream<complex_t>* in, double tone, double agcAttack, double agcDecay, double samplerate) { init(in, tone, agcAttack, agcDecay, samplerate); }
|
|
|
|
void init(stream<complex_t>* in, double tone, double agcAttack, double agcDecay, double samplerate) {
|
|
_tone = tone;
|
|
_samplerate = samplerate;
|
|
|
|
xlator.init(NULL, tone, samplerate);
|
|
agc.init(NULL, 1.0, agcAttack, agcDecay, 10e6, 10.0, INFINITY);
|
|
|
|
if constexpr (std::is_same_v<T, float>) {
|
|
agc.out.free();
|
|
}
|
|
|
|
base_type::init(in);
|
|
}
|
|
|
|
void setTone(double tone) {
|
|
assert(base_type::_block_init);
|
|
std::lock_guard<std::recursive_mutex> lck(base_type::ctrlMtx);
|
|
_tone = tone;
|
|
xlator.setOffset(_tone, _samplerate);
|
|
}
|
|
|
|
void setAGCAttack(double attack) {
|
|
assert(base_type::_block_init);
|
|
std::lock_guard<std::recursive_mutex> lck(base_type::ctrlMtx);
|
|
agc.setAttack(attack);
|
|
}
|
|
|
|
void setAGCDecay(double decay) {
|
|
assert(base_type::_block_init);
|
|
std::lock_guard<std::recursive_mutex> lck(base_type::ctrlMtx);
|
|
agc.setDecay(decay);
|
|
}
|
|
|
|
void setSamplerate(double samplerate) {
|
|
assert(base_type::_block_init);
|
|
std::lock_guard<std::recursive_mutex> lck(base_type::ctrlMtx);
|
|
_samplerate = samplerate;
|
|
xlator.setOffset(_tone, _samplerate);
|
|
}
|
|
|
|
inline int process(int count, const complex_t* in, T* out) {
|
|
xlator.process(count, in, xlator.out.writeBuf);
|
|
if constexpr (std::is_same_v<T, float>) {
|
|
dsp::convert::ComplexToReal::process(count, xlator.out.writeBuf, out);
|
|
agc.process(count, out, out);
|
|
}
|
|
if constexpr (std::is_same_v<T, stereo_t>) {
|
|
dsp::convert::ComplexToReal::process(count, xlator.out.writeBuf, agc.out.writeBuf);
|
|
agc.process(count, agc.out.writeBuf, agc.out.writeBuf);
|
|
convert::MonoToStereo::process(count, agc.out.writeBuf, out);
|
|
}
|
|
return count;
|
|
}
|
|
|
|
int run() {
|
|
int count = base_type::_in->read();
|
|
if (count < 0) { return -1; }
|
|
|
|
process(count, base_type::_in->readBuf, base_type::out.writeBuf);
|
|
|
|
base_type::_in->flush();
|
|
if (!base_type::out.swap(count)) { return -1; }
|
|
return count;
|
|
}
|
|
|
|
private:
|
|
double _tone;
|
|
double _samplerate;
|
|
|
|
dsp::channel::FrequencyXlator xlator;
|
|
dsp::loop::AGC<float> agc;
|
|
|
|
};
|
|
} |