mirror of https://github.com/hmatuschek/libsdr
Fixed minors...
Hannes Matuschek
12 years ago
parent
5a95ac8dab
commit
cad5e766a9
@ -0,0 +1,191 @@
|
|||||||
|
#include "baseband.hh"
|
||||||
|
#include "autocast.hh"
|
||||||
|
#include "demod.hh"
|
||||||
|
#include "portaudio.hh"
|
||||||
|
#include "wavfile.hh"
|
||||||
|
#include "gui/gui.hh"
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
#include <QApplication>
|
||||||
|
#include <QMainWindow>
|
||||||
|
#include <QThread>
|
||||||
|
|
||||||
|
using namespace sdr;
|
||||||
|
|
||||||
|
static void __sigint_handler(int signo) {
|
||||||
|
// On SIGINT -> stop queue properly
|
||||||
|
Queue::get().stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class Scalar>
|
||||||
|
class BPSK31: public Sink< std::complex<Scalar> >, public Source
|
||||||
|
{
|
||||||
|
typedef typename Traits<Scalar>::SScalar SScalar;
|
||||||
|
|
||||||
|
public:
|
||||||
|
BPSK31()
|
||||||
|
: Sink< std::complex<Scalar> >(), Source(), _lut(256)
|
||||||
|
{
|
||||||
|
_shift = 8*(sizeof(int16_t)-sizeof(Scalar));
|
||||||
|
// Initialize LUT
|
||||||
|
for (size_t i=0; i<256; i++) {
|
||||||
|
_lut[i] = std::exp(std::complex<double>(0., (-2*M_PI*i)/256))*(1<<8);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Destructor. */
|
||||||
|
virtual ~BPSK31() {
|
||||||
|
// pass...
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Configures the FM demodulator. */
|
||||||
|
virtual void config(const Config &src_cfg) {
|
||||||
|
// Requires type & buffer size
|
||||||
|
if (!src_cfg.hasType() || !src_cfg.hasBufferSize()) { return; }
|
||||||
|
// Check if buffer type matches template
|
||||||
|
if (Config::typeId< std::complex<Scalar> >() != src_cfg.type()) {
|
||||||
|
ConfigError err;
|
||||||
|
err << "Can not configure BPSK31: Invalid type " << src_cfg.type()
|
||||||
|
<< ", expected " << Config::typeId< std::complex<Scalar> >();
|
||||||
|
throw err;
|
||||||
|
}
|
||||||
|
// Unreference buffer if non-empty
|
||||||
|
if (! _buffer.isEmpty()) { _buffer.unref(); }
|
||||||
|
// Allocate buffer
|
||||||
|
_buffer = Buffer<int16_t>(16);
|
||||||
|
|
||||||
|
_lut_idx = 0;
|
||||||
|
_lut_incr = ((256.*256.*2144.)/src_cfg.sampleRate());
|
||||||
|
_pll_incr = 0;
|
||||||
|
double tau = 1./20;
|
||||||
|
_alpha = 2*M_PI/(2*M_PI+src_cfg.sampleRate()*tau);
|
||||||
|
_lp_phase = 0;
|
||||||
|
// Results into a 4*31.25 Hz sample
|
||||||
|
_subsample = (256.*src_cfg.sampleRate()/(4*32.25));
|
||||||
|
_sample_count = 0;
|
||||||
|
_dec_count = 0;
|
||||||
|
|
||||||
|
LogMessage msg(LOG_DEBUG);
|
||||||
|
msg << "Configured BPSK31 node:" << std::endl
|
||||||
|
<< " sample-rate: " << src_cfg.sampleRate() << std::endl
|
||||||
|
<< " in-type / out-type: " << src_cfg.type()
|
||||||
|
<< " / " << Config::typeId<int16_t>() << std::endl
|
||||||
|
<< " in-place: " << (_can_overwrite ? "true" : "false") << std::endl
|
||||||
|
<< " output scale: 2^" << _shift << std::endl
|
||||||
|
<< " PLL tau/alpha: " << tau << "/" << _alpha;
|
||||||
|
Logger::get().log(msg);
|
||||||
|
|
||||||
|
// Propergate config
|
||||||
|
this->setConfig(Config(Config::typeId<int16_t>(), 4*31.25, 16, 1));
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Performs the FM demodulation. */
|
||||||
|
virtual void process(const Buffer<std::complex<Scalar> > &buffer, bool allow_overwrite)
|
||||||
|
{
|
||||||
|
if (0 == buffer.size()) { return; }
|
||||||
|
_process(buffer, _buffer);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
/** The actual demodulation. */
|
||||||
|
void _process(const Buffer< std::complex<Scalar> > &in, const Buffer<int16_t> &out)
|
||||||
|
{
|
||||||
|
// Calc abs phase values
|
||||||
|
for (size_t i=1; i<in.size(); i++) {
|
||||||
|
std::complex<double> val =
|
||||||
|
std::complex<double>(in[i].real(), in[i].imag()) * _lut[_lut_idx>>8];
|
||||||
|
_lut_idx += (_lut_incr+_pll_incr);
|
||||||
|
_sample_count += 256;
|
||||||
|
// Get phase difference
|
||||||
|
double phase = std::atan2(double(val.imag()), double(val.real()))/M_PI;
|
||||||
|
// low-pass
|
||||||
|
_lp_phase = (1.-_alpha)*_lp_phase + _alpha*phase;
|
||||||
|
// Correct PLL frequency by LP-ed phase
|
||||||
|
_pll_incr += _lp_phase*10;
|
||||||
|
std::cout << "Fcorr=" << 22050. * double(_pll_incr)/(1<<16)
|
||||||
|
<< "; Phi=" << 180*phase << " (" << 180*_lp_phase << ")" << std::endl;
|
||||||
|
if (_sample_count >= _subsample) {
|
||||||
|
out[_dec_count] = (_lp_phase*(1<<14));
|
||||||
|
_sample_count -= _subsample; _dec_count++;
|
||||||
|
// If 8 bits has been collected:
|
||||||
|
if (8 == _dec_count) {
|
||||||
|
// propergate resulting 8 bits
|
||||||
|
this->send(out.head(8));
|
||||||
|
_dec_count = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
int _shift;
|
||||||
|
/** If true, in-place demodulation is poissible. */
|
||||||
|
bool _can_overwrite;
|
||||||
|
Buffer< std::complex<SScalar> > _lut;
|
||||||
|
/** The current LUT index. It is defined in multiples of 256 for heiher precision. */
|
||||||
|
uint16_t _lut_idx;
|
||||||
|
/** The LUT increment is defined in multiples of 256 for higher precision. */
|
||||||
|
int32_t _lut_incr;
|
||||||
|
/** The PLL increment correction is defined in multiples of 256 for higher precision. */
|
||||||
|
double _pll_incr;
|
||||||
|
double _alpha;
|
||||||
|
double _lp_phase;
|
||||||
|
size_t _subsample;
|
||||||
|
size_t _sample_count;
|
||||||
|
size_t _dec_count;
|
||||||
|
/** The output buffer, unused if demodulation is performed in-place. */
|
||||||
|
Buffer<int16_t> _buffer;
|
||||||
|
};
|
||||||
|
|
||||||
|
int main(int argc, char *argv[]) {
|
||||||
|
if (2 != argc) {
|
||||||
|
std::cerr << "Usage: sdr_psk31 FILENAME" << std::endl;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
sdr::Logger::get().addHandler(
|
||||||
|
new sdr::StreamLogHandler(std::cerr, sdr::LOG_DEBUG));
|
||||||
|
|
||||||
|
// Register handler:
|
||||||
|
signal(SIGINT, __sigint_handler);
|
||||||
|
|
||||||
|
QApplication app(argc, argv);
|
||||||
|
|
||||||
|
QMainWindow *win = new QMainWindow();
|
||||||
|
gui::Spectrum *spec = new gui::Spectrum(2, 1024, 5);
|
||||||
|
gui::SpectrumView *spec_view = new gui::SpectrumView(spec);
|
||||||
|
spec_view->setMindB(-60);
|
||||||
|
win->setCentralWidget(spec_view);
|
||||||
|
win->setMinimumSize(640, 240);
|
||||||
|
win->show();
|
||||||
|
|
||||||
|
PortAudio::init();
|
||||||
|
Queue &queue = Queue::get();
|
||||||
|
|
||||||
|
//WavSource src(argv[1]);
|
||||||
|
SigGen<int16_t> src(22050., 1024);
|
||||||
|
src.addSine(2144);
|
||||||
|
AutoCast< std::complex<int16_t> > cast;
|
||||||
|
IQBaseBand<int16_t> baseband(0.0, 2144., 400.0, 15, 1, 8000.0);
|
||||||
|
BPSK31<int16_t> demod;
|
||||||
|
PortSink sink;
|
||||||
|
|
||||||
|
src.connect(&cast, true);
|
||||||
|
cast.connect(&baseband, true);
|
||||||
|
baseband.connect(spec);
|
||||||
|
baseband.connect(&sink);
|
||||||
|
baseband.connect(&demod);
|
||||||
|
|
||||||
|
//queue.addIdle(&src, &WavSource::next);
|
||||||
|
queue.addIdle(&src, &SigGen<int16_t>::next);
|
||||||
|
|
||||||
|
queue.start();
|
||||||
|
app.exec();
|
||||||
|
queue.stop();
|
||||||
|
queue.wait();
|
||||||
|
|
||||||
|
PortAudio::terminate();
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue