sdrangel/sdrbase/dsp/spectrumvis.h

214 wiersze
6.5 KiB
C
Czysty Zwykły widok Historia

///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2015-2020 Edouard Griffiths, F4EXB //
// //
// Symbol synchronizer or symbol clock recovery mostly encapsulating //
// liquid-dsp's symsync "object" //
// //
// 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 //
// (at your option) any later version. //
// //
// 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_SPECTRUMVIS_H
#define INCLUDE_SPECTRUMVIS_H
#include <QMutex>
#include "dsp/basebandsamplesink.h"
#include "dsp/fftengine.h"
#include "dsp/fftwindow.h"
2020-05-01 00:12:30 +00:00
#include "dsp/glspectrumsettings.h"
#include "export.h"
#include "util/message.h"
2018-06-28 21:47:15 +00:00
#include "util/movingaverage2d.h"
2018-07-01 00:16:59 +00:00
#include "util/fixedaverage2d.h"
#include "util/max2d.h"
#include "websockets/wsspectrum.h"
class GLSpectrumInterface;
class MessageQueue;
2020-05-14 20:14:11 +00:00
class SDRBASE_API SpectrumVis : public BasebandSampleSink {
public:
2020-05-14 20:14:11 +00:00
class SDRBASE_API MsgConfigureSpectrumVis : public Message {
2020-05-01 00:12:30 +00:00
MESSAGE_CLASS_DECLARATION
public:
const GLSpectrumSettings& getSettings() const { return m_settings; }
bool getForce() const { return m_force; }
static MsgConfigureSpectrumVis* create(const GLSpectrumSettings& settings, bool force)
{
return new MsgConfigureSpectrumVis(settings, force);
}
private:
GLSpectrumSettings m_settings;
bool m_force;
MsgConfigureSpectrumVis(const GLSpectrumSettings& settings, bool force) :
Message(),
m_settings(settings),
m_force(force)
{ }
};
class SDRBASE_API MsgStartStop : public Message {
2020-07-05 08:03:41 +00:00
MESSAGE_CLASS_DECLARATION
public:
bool getStartStop() const { return m_startStop; }
static MsgStartStop* create(bool startStop) {
return new MsgStartStop(startStop);
}
protected:
bool m_startStop;
MsgStartStop(bool startStop) :
Message(),
m_startStop(startStop)
{ }
};
enum AvgMode
2018-07-01 00:16:59 +00:00
{
AvgModeNone,
AvgModeMovingAvg,
AvgModeFixedAvg,
AvgModeMax
2018-07-01 00:16:59 +00:00
};
2020-04-30 07:57:05 +00:00
SpectrumVis(Real scalef);
virtual ~SpectrumVis();
void setGLSpectrum(GLSpectrumInterface* glSpectrum) { m_glSpectrum = glSpectrum; }
void openWSSpectrum();
void closeWSSpectrum();
void configure(
int fftSize,
float refLevel,
float powerRange,
int overlapPercent,
unsigned int averagingNb,
AvgMode averagingMode,
FFTWindow::Function window,
bool m_linear
);
2020-04-30 07:57:05 +00:00
void setScalef(Real scalef);
void configureWSSpectrum(const QString& address, uint16_t port);
const GLSpectrumSettings& getSettings() const { return m_settings; }
Real getSpecMax() const { return m_specMax / m_powFFTDiv; }
2020-04-30 07:57:05 +00:00
virtual void feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end, bool positiveOnly);
virtual void feed(const Complex *begin, unsigned int length); //!< direct FFT feed
void feedTriggered(const SampleVector::const_iterator& triggerPoint, const SampleVector::const_iterator& end, bool positiveOnly);
virtual void start();
virtual void stop();
virtual bool handleMessage(const Message& message);
private:
class MsgConfigureScalingFactor : public Message
{
MESSAGE_CLASS_DECLARATION
public:
MsgConfigureScalingFactor(Real scalef) :
Message(),
m_scalef(scalef)
{}
Real getScalef() const { return m_scalef; }
private:
Real m_scalef;
};
2020-04-30 07:57:05 +00:00
class MsgConfigureWSpectrumOpenClose : public Message
{
MESSAGE_CLASS_DECLARATION
2020-04-30 07:57:05 +00:00
public:
MsgConfigureWSpectrumOpenClose(bool openClose) :
Message(),
m_openClose(openClose)
{}
2020-04-30 07:57:05 +00:00
Real getOpenClose() const { return m_openClose; }
private:
bool m_openClose;
};
class MsgConfigureWSpectrum : public Message
{
MESSAGE_CLASS_DECLARATION
public:
MsgConfigureWSpectrum(const QString& address, uint16_t port) :
Message(),
m_address(address),
m_port(port)
{}
const QString& getAddress() const { return m_address; }
uint16_t getPort() const { return m_port; }
private:
QString m_address;
uint16_t m_port;
};
2020-07-05 08:03:41 +00:00
bool m_running;
FFTEngine* m_fft;
FFTWindow m_window;
2020-03-12 05:27:38 +00:00
unsigned int m_fftEngineSequence;
std::vector<Complex> m_fftBuffer;
std::vector<Real> m_powerSpectrum;
2020-05-01 00:12:30 +00:00
GLSpectrumSettings m_settings;
2015-08-14 03:00:28 +00:00
std::size_t m_overlapSize;
std::size_t m_refillSize;
std::size_t m_fftBufferFill;
bool m_needMoreSamples;
Real m_scalef;
GLSpectrumInterface* m_glSpectrum;
WSSpectrum m_wsSpectrum;
2018-07-01 00:16:59 +00:00
MovingAverage2D<double> m_movingAverage;
FixedAverage2D<double> m_fixedAverage;
Max2D<double> m_max;
Real m_specMax;
2018-06-28 21:47:15 +00:00
uint64_t m_centerFrequency;
int m_sampleRate;
2018-06-28 21:47:15 +00:00
Real m_ofs;
Real m_powFFTDiv;
2018-06-28 21:47:15 +00:00
static const Real m_mult;
QMutex m_mutex;
void setRunning(bool running) { m_running = running; }
2020-05-01 00:12:30 +00:00
void applySettings(const GLSpectrumSettings& settings, bool force = false);
2020-05-01 00:16:20 +00:00
void handleConfigureDSP(uint64_t centerFrequency, int sampleRate);
void handleScalef(Real scalef);
2020-05-01 00:16:20 +00:00
void handleWSOpenClose(bool openClose);
2020-04-30 07:57:05 +00:00
void handleConfigureWSSpectrum(const QString& address, uint16_t port);
};
#endif // INCLUDE_SPECTRUMVIS_H