mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-12-23 01:55:48 -05:00
184 lines
5.5 KiB
C++
184 lines
5.5 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2017 Edouard Griffiths, F4EXB //
|
|
// //
|
|
// This program is free software; you can redistribute it and/or modify //
|
|
// it under the terms of the GNU General Public License as published by //
|
|
// the Free Software Foundation as version 3 of the License, or //
|
|
// //
|
|
// This program is distributed in the hope that it will be useful, //
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
|
|
// GNU General Public License V3 for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU General Public License //
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
|
|
///////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef INCLUDE_CHANALYZERNG_H
|
|
#define INCLUDE_CHANALYZERNG_H
|
|
|
|
#include <dsp/basebandsamplesink.h>
|
|
#include <QMutex>
|
|
#include <vector>
|
|
|
|
#include "dsp/interpolator.h"
|
|
#include "dsp/ncof.h"
|
|
#include "dsp/fftfilt.h"
|
|
#include "audio/audiofifo.h"
|
|
#include "util/message.h"
|
|
|
|
#define ssbFftLen 1024
|
|
|
|
class ChannelAnalyzerNG : public BasebandSampleSink {
|
|
public:
|
|
ChannelAnalyzerNG(BasebandSampleSink* m_sampleSink);
|
|
virtual ~ChannelAnalyzerNG();
|
|
|
|
void configure(MessageQueue* messageQueue,
|
|
int channelSampleRate,
|
|
Real Bandwidth,
|
|
Real LowCutoff,
|
|
int spanLog2,
|
|
bool ssb);
|
|
|
|
int getInputSampleRate() const { return m_running.m_inputSampleRate; }
|
|
int getChannelSampleRate() const { return m_running.m_channelSampleRate; }
|
|
Real getMagSq() const { return m_magsq; }
|
|
|
|
virtual void feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end, bool positiveOnly);
|
|
virtual void start();
|
|
virtual void stop();
|
|
virtual bool handleMessage(const Message& cmd);
|
|
|
|
private:
|
|
class MsgConfigureChannelAnalyzer : public Message {
|
|
MESSAGE_CLASS_DECLARATION
|
|
|
|
public:
|
|
int getChannelSampleRate() const { return m_channelSampleRate; }
|
|
Real getBandwidth() const { return m_Bandwidth; }
|
|
Real getLoCutoff() const { return m_LowCutoff; }
|
|
int getSpanLog2() const { return m_spanLog2; }
|
|
bool getSSB() const { return m_ssb; }
|
|
|
|
static MsgConfigureChannelAnalyzer* create(
|
|
int channelSampleRate,
|
|
Real Bandwidth,
|
|
Real LowCutoff,
|
|
int spanLog2,
|
|
bool ssb)
|
|
{
|
|
return new MsgConfigureChannelAnalyzer(channelSampleRate, Bandwidth, LowCutoff, spanLog2, ssb);
|
|
}
|
|
|
|
private:
|
|
int m_channelSampleRate;
|
|
Real m_Bandwidth;
|
|
Real m_LowCutoff;
|
|
int m_spanLog2;
|
|
bool m_ssb;
|
|
|
|
MsgConfigureChannelAnalyzer(
|
|
int channelSampleRate,
|
|
Real Bandwidth,
|
|
Real LowCutoff,
|
|
int spanLog2,
|
|
bool ssb) :
|
|
Message(),
|
|
m_channelSampleRate(channelSampleRate),
|
|
m_Bandwidth(Bandwidth),
|
|
m_LowCutoff(LowCutoff),
|
|
m_spanLog2(spanLog2),
|
|
m_ssb(ssb)
|
|
{ }
|
|
};
|
|
|
|
struct Config
|
|
{
|
|
int m_frequency;
|
|
int m_inputSampleRate;
|
|
int m_channelSampleRate;
|
|
Real m_Bandwidth;
|
|
Real m_LowCutoff;
|
|
int m_spanLog2;
|
|
bool m_ssb;
|
|
|
|
Config() :
|
|
m_frequency(0),
|
|
m_inputSampleRate(96000),
|
|
m_channelSampleRate(96000),
|
|
m_Bandwidth(5000),
|
|
m_LowCutoff(300),
|
|
m_spanLog2(3),
|
|
m_ssb(false)
|
|
{}
|
|
};
|
|
|
|
Config m_config;
|
|
Config m_running;
|
|
|
|
int m_undersampleCount;
|
|
fftfilt::cmplx m_sum;
|
|
bool m_usb;
|
|
Real m_magsq;
|
|
bool m_useInterpolator;
|
|
|
|
NCOF m_nco;
|
|
Interpolator m_interpolator;
|
|
Real m_interpolatorDistance;
|
|
Real m_interpolatorDistanceRemain;
|
|
|
|
fftfilt* SSBFilter;
|
|
fftfilt* DSBFilter;
|
|
|
|
BasebandSampleSink* m_sampleSink;
|
|
SampleVector m_sampleBuffer;
|
|
QMutex m_settingsMutex;
|
|
|
|
void apply(bool force = false);
|
|
|
|
void processOneSample(Complex& c, fftfilt::cmplx *sideband)
|
|
{
|
|
int n_out;
|
|
int decim = 1<<m_running.m_spanLog2;
|
|
|
|
if (m_running.m_ssb)
|
|
{
|
|
n_out = SSBFilter->runSSB(c, &sideband, m_usb);
|
|
}
|
|
else
|
|
{
|
|
n_out = DSBFilter->runDSB(c, &sideband);
|
|
}
|
|
|
|
for (int i = 0; i < n_out; i++)
|
|
{
|
|
// Downsample by 2^(m_scaleLog2 - 1) for SSB band spectrum display
|
|
// smart decimation with bit gain using float arithmetic (23 bits significand)
|
|
|
|
m_sum += sideband[i];
|
|
|
|
if (!(m_undersampleCount++ & (decim - 1))) // counter LSB bit mask for decimation by 2^(m_scaleLog2 - 1)
|
|
{
|
|
m_sum /= decim;
|
|
m_magsq = (m_sum.real() * m_sum.real() + m_sum.imag() * m_sum.imag())/ (1<<30);
|
|
|
|
if (m_running.m_ssb & !m_usb)
|
|
{ // invert spectrum for LSB
|
|
//m_sampleBuffer.push_back(Sample(m_sum.imag() * 32768.0, m_sum.real() * 32768.0));
|
|
m_sampleBuffer.push_back(Sample(m_sum.imag(), m_sum.real()));
|
|
}
|
|
else
|
|
{
|
|
//m_sampleBuffer.push_back(Sample(m_sum.real() * 32768.0, m_sum.imag() * 32768.0));
|
|
m_sampleBuffer.push_back(Sample(m_sum.real(), m_sum.imag()));
|
|
}
|
|
|
|
m_sum = 0;
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
#endif // INCLUDE_CHANALYZERNG_H
|