/////////////////////////////////////////////////////////////////////////////////// // Copyright (C) 2019 Edouard Griffiths, F4EXB // // Copyright (C) 2021 Jon Beniston, M7RCE // // // // 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 . // /////////////////////////////////////////////////////////////////////////////////// #ifndef INCLUDE_AISDEMODSINK_H #define INCLUDE_AISDEMODSINK_H #include #include "dsp/channelsamplesink.h" #include "dsp/phasediscri.h" #include "dsp/nco.h" #include "dsp/interpolator.h" #include "dsp/firfilter.h" #include "dsp/gaussian.h" #include "util/movingaverage.h" #include "util/doublebufferfifo.h" #include "util/messagequeue.h" #include "util/crc.h" #include "aisdemodsettings.h" #include #include #include #define AISDEMOD_MAX_BYTES (3+1+126+2+1+1) class ChannelAPI; class AISDemod; class ScopeVis; class AISDemodSink : public ChannelSampleSink { public: AISDemodSink(AISDemod *aisDemod); ~AISDemodSink(); virtual void feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end); void setScopeSink(ScopeVis* scopeSink) { m_scopeSink = scopeSink; } void applyChannelSettings(int channelSampleRate, int channelFrequencyOffset, bool force = false); void applySettings(const AISDemodSettings& settings, bool force = false); void setMessageQueueToChannel(MessageQueue *messageQueue) { m_messageQueueToChannel = messageQueue; } void setChannel(ChannelAPI *channel) { m_channel = channel; } double getMagSq() const { return m_magsq; } void getMagSqLevels(double& avg, double& peak, int& nbSamples) { if (m_magsqCount > 0) { m_magsq = m_magsqSum / m_magsqCount; m_magSqLevelStore.m_magsq = m_magsq; m_magSqLevelStore.m_magsqPeak = m_magsqPeak; } avg = m_magSqLevelStore.m_magsq; peak = m_magSqLevelStore.m_magsqPeak; nbSamples = m_magsqCount == 0 ? 1 : m_magsqCount; m_magsqSum = 0.0f; m_magsqPeak = 0.0f; m_magsqCount = 0; } private: struct MagSqLevelsStore { MagSqLevelsStore() : m_magsq(1e-12), m_magsqPeak(1e-12) {} double m_magsq; double m_magsqPeak; }; ScopeVis* m_scopeSink; // Scope GUI to display baseband waveform AISDemod *m_aisDemod; AISDemodSettings m_settings; ChannelAPI *m_channel; int m_channelSampleRate; int m_channelFrequencyOffset; int m_samplesPerSymbol; // Number of samples per symbol NCO m_nco; Interpolator m_interpolator; Real m_interpolatorDistance; Real m_interpolatorDistanceRemain; double m_magsq; double m_magsqSum; double m_magsqPeak; int m_magsqCount; MagSqLevelsStore m_magSqLevelStore; MessageQueue *m_messageQueueToChannel; MovingAverageUtil m_movingAverage; Lowpass m_lowpass; // RF input filter PhaseDiscriminators m_phaseDiscri; // FM demodulator Gaussian m_pulseShape; // Pulse shaping filter Real *m_rxBuf; // Receive sample buffer, large enough for one max length messsage int m_rxBufLength; // Size in elements in m_rxBuf int m_rxBufIdx; // Index in to circular buffer int m_rxBufCnt; // Number of valid samples in buffer Real *m_train; // Training sequence to look for int m_correlationLength; unsigned char m_bytes[AISDEMOD_MAX_BYTES]; crc16x25 m_crc; QVector m_demodBuffer; int m_demodBufferFill; ComplexVector m_sampleBuffer[AISDemodSettings::m_scopeStreams]; static const int m_sampleBufferSize = AISDemodSettings::AISDEMOD_CHANNEL_SAMPLE_RATE / 20; int m_sampleBufferIndex; void processOneSample(Complex &ci); MessageQueue *getMessageQueueToChannel() { return m_messageQueueToChannel; } void sampleToScope(Complex sample, Real magsq, Real fmDemod, Real filt, Real rxBuf, Real corr, Real thresholdMet, Real dcOffset, Real crcValid); }; #endif // INCLUDE_AISDEMODSINK_H