/////////////////////////////////////////////////////////////////////////////////// // Copyright (C) 2016 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 . // /////////////////////////////////////////////////////////////////////////////////// #include #include #include #include "SWGChannelSettings.h" #include "SWGCWKeyerSettings.h" #include #include #include #include #include "dsp/dspengine.h" #include "dsp/pidcontroller.h" #include "device/devicesinkapi.h" #include "dsp/threadedbasebandsamplesource.h" #include "nfmmod.h" MESSAGE_CLASS_DEFINITION(NFMMod::MsgConfigureNFMMod, Message) MESSAGE_CLASS_DEFINITION(NFMMod::MsgConfigureChannelizer, Message) MESSAGE_CLASS_DEFINITION(NFMMod::MsgConfigureFileSourceName, Message) MESSAGE_CLASS_DEFINITION(NFMMod::MsgConfigureFileSourceSeek, Message) MESSAGE_CLASS_DEFINITION(NFMMod::MsgConfigureFileSourceStreamTiming, Message) MESSAGE_CLASS_DEFINITION(NFMMod::MsgReportFileSourceStreamData, Message) MESSAGE_CLASS_DEFINITION(NFMMod::MsgReportFileSourceStreamTiming, Message) const QString NFMMod::m_channelIdURI = "sdrangel.channeltx.modnfm"; const QString NFMMod::m_channelId = "NFMMod"; const int NFMMod::m_levelNbSamples = 480; // every 10ms NFMMod::NFMMod(DeviceSinkAPI *deviceAPI) : ChannelSourceAPI(m_channelIdURI), m_deviceAPI(deviceAPI), m_basebandSampleRate(48000), m_outputSampleRate(48000), m_inputFrequencyOffset(0), m_modPhasor(0.0f), m_movingAverage(40, 0), m_volumeAGC(40, 0), m_audioFifo(4800), m_settingsMutex(QMutex::Recursive), m_fileSize(0), m_recordLength(0), m_sampleRate(48000), m_levelCalcCount(0), m_peakLevel(0.0f), m_levelSum(0.0f) { setObjectName(m_channelId); m_audioBuffer.resize(1<<14); m_audioBufferFill = 0; m_movingAverage.resize(16, 0); m_volumeAGC.resize(4096, 0.003, 0); m_magsq = 0.0; m_toneNco.setFreq(1000.0, m_settings.m_audioSampleRate); m_ctcssNco.setFreq(88.5, m_settings.m_audioSampleRate); DSPEngine::instance()->addAudioSource(&m_audioFifo); // CW keyer m_cwKeyer.setSampleRate(m_settings.m_audioSampleRate); m_cwKeyer.setWPM(13); m_cwKeyer.setMode(CWKeyerSettings::CWNone); m_channelizer = new UpChannelizer(this); m_threadedChannelizer = new ThreadedBasebandSampleSource(m_channelizer, this); m_deviceAPI->addThreadedSource(m_threadedChannelizer); m_deviceAPI->addChannelAPI(this); applyChannelSettings(m_basebandSampleRate, m_outputSampleRate, m_inputFrequencyOffset, true); applySettings(m_settings, true); } NFMMod::~NFMMod() { DSPEngine::instance()->removeAudioSource(&m_audioFifo); m_deviceAPI->removeChannelAPI(this); m_deviceAPI->removeThreadedSource(m_threadedChannelizer); delete m_threadedChannelizer; delete m_channelizer; } void NFMMod::pull(Sample& sample) { if (m_settings.m_channelMute) { sample.m_real = 0.0f; sample.m_imag = 0.0f; return; } Complex ci; m_settingsMutex.lock(); if (m_interpolatorDistance > 1.0f) // decimate { modulateSample(); while (!m_interpolator.decimate(&m_interpolatorDistanceRemain, m_modSample, &ci)) { modulateSample(); } } else { if (m_interpolator.interpolate(&m_interpolatorDistanceRemain, m_modSample, &ci)) { modulateSample(); } } m_interpolatorDistanceRemain += m_interpolatorDistance; ci *= m_carrierNco.nextIQ(); // shift to carrier frequency m_settingsMutex.unlock(); Real magsq = ci.real() * ci.real() + ci.imag() * ci.imag(); magsq /= (1<<30); m_movingAverage.feed(magsq); m_magsq = m_movingAverage.average(); sample.m_real = (FixReal) ci.real(); sample.m_imag = (FixReal) ci.imag(); } void NFMMod::pullAudio(int nbSamples) { unsigned int nbSamplesAudio = nbSamples * ((Real) m_settings.m_audioSampleRate / (Real) m_basebandSampleRate); if (nbSamplesAudio > m_audioBuffer.size()) { m_audioBuffer.resize(nbSamplesAudio); } m_audioFifo.read(reinterpret_cast(&m_audioBuffer[0]), nbSamplesAudio, 10); m_audioBufferFill = 0; } void NFMMod::modulateSample() { Real t; pullAF(t); calculateLevel(t); m_audioBufferFill++; if (m_settings.m_ctcssOn) { m_modPhasor += (m_settings.m_fmDeviation / (float) m_settings.m_audioSampleRate) * (0.85f * m_bandpass.filter(t) + 0.15f * 378.0f * m_ctcssNco.next()) * (M_PI / 378.0f); } else { // 378 = 302 * 1.25; 302 = number of filter taps (established experimentally) m_modPhasor += (m_settings.m_fmDeviation / (float) m_settings.m_audioSampleRate) * m_bandpass.filter(t) * (M_PI / 378.0f); } m_modSample.real(cos(m_modPhasor) * 29204.0f); // -1 dB m_modSample.imag(sin(m_modPhasor) * 29204.0f); } void NFMMod::pullAF(Real& sample) { switch (m_settings.m_modAFInput) { case NFMModInputTone: sample = m_toneNco.next(); break; case NFMModInputFile: // sox f4exb_call.wav --encoding float --endian little f4exb_call.raw // ffplay -f f32le -ar 48k -ac 1 f4exb_call.raw if (m_ifstream.is_open()) { if (m_ifstream.eof()) { if (m_settings.m_playLoop) { m_ifstream.clear(); m_ifstream.seekg(0, std::ios::beg); } } if (m_ifstream.eof()) { sample = 0.0f; } else { m_ifstream.read(reinterpret_cast(&sample), sizeof(Real)); sample *= m_settings.m_volumeFactor; } } else { sample = 0.0f; } break; case NFMModInputAudio: sample = ((m_audioBuffer[m_audioBufferFill].l + m_audioBuffer[m_audioBufferFill].r) / 65536.0f) * m_settings.m_volumeFactor; break; case NFMModInputCWTone: Real fadeFactor; if (m_cwKeyer.getSample()) { m_cwKeyer.getCWSmoother().getFadeSample(true, fadeFactor); sample = m_toneNco.next() * fadeFactor; } else { if (m_cwKeyer.getCWSmoother().getFadeSample(false, fadeFactor)) { sample = m_toneNco.next() * fadeFactor; } else { sample = 0.0f; m_toneNco.setPhase(0); } } break; case NFMModInputNone: default: sample = 0.0f; break; } } void NFMMod::calculateLevel(Real& sample) { if (m_levelCalcCount < m_levelNbSamples) { m_peakLevel = std::max(std::fabs(m_peakLevel), sample); m_levelSum += sample * sample; m_levelCalcCount++; } else { qreal rmsLevel = sqrt(m_levelSum / m_levelNbSamples); //qDebug("NFMMod::calculateLevel: %f %f", rmsLevel, m_peakLevel); emit levelChanged(rmsLevel, m_peakLevel, m_levelNbSamples); m_peakLevel = 0.0f; m_levelSum = 0.0f; m_levelCalcCount = 0; } } void NFMMod::start() { qDebug() << "NFMMod::start: m_outputSampleRate: " << m_outputSampleRate << " m_inputFrequencyOffset: " << m_inputFrequencyOffset; m_audioFifo.clear(); applyChannelSettings(m_basebandSampleRate, m_outputSampleRate, m_inputFrequencyOffset, true); } void NFMMod::stop() { } bool NFMMod::handleMessage(const Message& cmd) { if (UpChannelizer::MsgChannelizerNotification::match(cmd)) { UpChannelizer::MsgChannelizerNotification& notif = (UpChannelizer::MsgChannelizerNotification&) cmd; qDebug() << "NFMMod::handleMessage: UpChannelizer::MsgChannelizerNotification"; applyChannelSettings(notif.getBasebandSampleRate(), notif.getSampleRate(), notif.getFrequencyOffset()); return true; } else if (MsgConfigureChannelizer::match(cmd)) { MsgConfigureChannelizer& cfg = (MsgConfigureChannelizer&) cmd; qDebug() << "NFMMod::handleMessage: MsgConfigureChannelizer:" << " getSampleRate: " << cfg.getSampleRate() << " getCenterFrequency: " << cfg.getCenterFrequency(); m_channelizer->configure(m_channelizer->getInputMessageQueue(), cfg.getSampleRate(), cfg.getCenterFrequency()); return true; } else if (MsgConfigureNFMMod::match(cmd)) { MsgConfigureNFMMod& cfg = (MsgConfigureNFMMod&) cmd; qDebug() << "NFMMod::handleMessage: MsgConfigureNFMMod"; applySettings(cfg.getSettings(), cfg.getForce()); return true; } else if (MsgConfigureFileSourceName::match(cmd)) { MsgConfigureFileSourceName& conf = (MsgConfigureFileSourceName&) cmd; m_fileName = conf.getFileName(); openFileStream(); qDebug() << "NFMMod::handleMessage: MsgConfigureFileSourceName:" << " m_fileName: " << m_fileName; return true; } else if (MsgConfigureFileSourceSeek::match(cmd)) { MsgConfigureFileSourceSeek& conf = (MsgConfigureFileSourceSeek&) cmd; int seekPercentage = conf.getPercentage(); seekFileStream(seekPercentage); qDebug() << "NFMMod::handleMessage: MsgConfigureFileSourceSeek:" << " seekPercentage: " << seekPercentage; return true; } else if (MsgConfigureFileSourceStreamTiming::match(cmd)) { std::size_t samplesCount; if (m_ifstream.eof()) { samplesCount = m_fileSize / sizeof(Real); } else { samplesCount = m_ifstream.tellg() / sizeof(Real); } MsgReportFileSourceStreamTiming *report; report = MsgReportFileSourceStreamTiming::create(samplesCount); getMessageQueueToGUI()->push(report); return true; } else { return false; } } void NFMMod::openFileStream() { if (m_ifstream.is_open()) { m_ifstream.close(); } m_ifstream.open(m_fileName.toStdString().c_str(), std::ios::binary | std::ios::ate); m_fileSize = m_ifstream.tellg(); m_ifstream.seekg(0,std::ios_base::beg); m_sampleRate = 48000; // fixed rate m_recordLength = m_fileSize / (sizeof(Real) * m_sampleRate); qDebug() << "NFMMod::openFileStream: " << m_fileName.toStdString().c_str() << " fileSize: " << m_fileSize << "bytes" << " length: " << m_recordLength << " seconds"; MsgReportFileSourceStreamData *report; report = MsgReportFileSourceStreamData::create(m_sampleRate, m_recordLength); getMessageQueueToGUI()->push(report); } void NFMMod::seekFileStream(int seekPercentage) { QMutexLocker mutexLocker(&m_settingsMutex); if (m_ifstream.is_open()) { int seekPoint = ((m_recordLength * seekPercentage) / 100) * m_sampleRate; seekPoint *= sizeof(Real); m_ifstream.clear(); m_ifstream.seekg(seekPoint, std::ios::beg); } } void NFMMod::applyChannelSettings(int basebandSampleRate, int outputSampleRate, int inputFrequencyOffset, bool force) { qDebug() << "NFMMod::applyChannelSettings:" << " basebandSampleRate: " << basebandSampleRate << " outputSampleRate: " << outputSampleRate << " inputFrequencyOffset: " << inputFrequencyOffset; if ((inputFrequencyOffset != m_inputFrequencyOffset) || (outputSampleRate != m_outputSampleRate) || force) { m_settingsMutex.lock(); m_carrierNco.setFreq(inputFrequencyOffset, outputSampleRate); m_settingsMutex.unlock(); } if ((outputSampleRate != m_outputSampleRate) || force) { m_settingsMutex.lock(); m_interpolatorDistanceRemain = 0; m_interpolatorConsumed = false; m_interpolatorDistance = (Real) m_settings.m_audioSampleRate / (Real) outputSampleRate; m_interpolator.create(48, m_settings.m_audioSampleRate, m_settings.m_rfBandwidth / 2.2, 3.0); m_settingsMutex.unlock(); } m_basebandSampleRate = basebandSampleRate; m_outputSampleRate = outputSampleRate; m_inputFrequencyOffset = inputFrequencyOffset; } void NFMMod::applySettings(const NFMModSettings& settings, bool force) { qDebug() << "NFMMod::applySettings:" << " m_inputFrequencyOffset: " << settings.m_inputFrequencyOffset << " m_rfBandwidth: " << settings.m_rfBandwidth << " m_afBandwidth: " << settings.m_afBandwidth << " m_fmDeviation: " << settings.m_fmDeviation << " m_volumeFactor: " << settings.m_volumeFactor << " m_toneFrequency: " << settings.m_toneFrequency << " m_ctcssIndex: " << settings.m_ctcssIndex << " m_ctcssOn: " << settings.m_ctcssOn << " m_channelMute: " << settings.m_channelMute << " m_playLoop: " << settings.m_playLoop << " m_modAFInout " << settings.m_modAFInput << " force: " << force; if((settings.m_rfBandwidth != m_settings.m_rfBandwidth) || (settings.m_audioSampleRate != m_settings.m_audioSampleRate) || force) { m_settingsMutex.lock(); m_interpolatorDistanceRemain = 0; m_interpolatorConsumed = false; m_interpolatorDistance = (Real) settings.m_audioSampleRate / (Real) m_outputSampleRate; m_interpolator.create(48, settings.m_audioSampleRate, settings.m_rfBandwidth / 2.2, 3.0); m_settingsMutex.unlock(); } if ((settings.m_afBandwidth != m_settings.m_afBandwidth) || (settings.m_audioSampleRate != m_settings.m_audioSampleRate) || force) { m_settingsMutex.lock(); m_lowpass.create(301, settings.m_audioSampleRate, 250.0); m_bandpass.create(301, settings.m_audioSampleRate, 300.0, settings.m_afBandwidth); m_settingsMutex.unlock(); } if ((settings.m_toneFrequency != m_settings.m_toneFrequency) || (settings.m_audioSampleRate != m_settings.m_audioSampleRate) || force) { m_settingsMutex.lock(); m_toneNco.setFreq(settings.m_toneFrequency, settings.m_audioSampleRate); m_settingsMutex.unlock(); } if ((settings.m_audioSampleRate != m_settings.m_audioSampleRate) || force) { m_cwKeyer.setSampleRate(settings.m_audioSampleRate); } if ((settings.m_ctcssIndex != m_settings.m_ctcssIndex) || (settings.m_audioSampleRate != m_settings.m_audioSampleRate) || force) { m_settingsMutex.lock(); m_ctcssNco.setFreq(NFMModSettings::getCTCSSFreq(settings.m_ctcssIndex), settings.m_audioSampleRate); m_settingsMutex.unlock(); } m_settings = settings; } QByteArray NFMMod::serialize() const { return m_settings.serialize(); } bool NFMMod::deserialize(const QByteArray& data) { bool success = true; if (!m_settings.deserialize(data)) { m_settings.resetToDefaults(); success = false; } MsgConfigureChannelizer *msgChan = MsgConfigureChannelizer::create( 48000, m_settings.m_inputFrequencyOffset); m_inputMessageQueue.push(msgChan); MsgConfigureNFMMod *msg = MsgConfigureNFMMod::create(m_settings, true); m_inputMessageQueue.push(msg); return success; } int NFMMod::webapiSettingsGet( SWGSDRangel::SWGChannelSettings& response, QString& errorMessage __attribute__((unused))) { response.setNfmModSettings(new SWGSDRangel::SWGNFMModSettings()); response.getNfmModSettings()->setAfBandwidth(m_settings.m_afBandwidth); response.getNfmModSettings()->setAudioSampleRate(m_settings.m_audioSampleRate); response.getNfmModSettings()->setChannelMute(m_settings.m_channelMute ? 1 : 0); response.getNfmModSettings()->setCtcssIndex(m_settings.m_ctcssIndex); response.getNfmModSettings()->setCtcssOn(m_settings.m_ctcssOn ? 1 : 0); response.getNfmModSettings()->setFmDeviation(m_settings.m_fmDeviation); response.getNfmModSettings()->setInputFrequencyOffset(m_settings.m_inputFrequencyOffset); response.getNfmModSettings()->setModAfInput((int) m_settings.m_modAFInput); response.getNfmModSettings()->setPlayLoop(m_settings.m_playLoop ? 1 : 0); response.getNfmModSettings()->setRfBandwidth(m_settings.m_rfBandwidth); response.getNfmModSettings()->setRgbColor(m_settings.m_rgbColor); *response.getNfmModSettings()->getTitle() = m_settings.m_title; response.getNfmModSettings()->setToneFrequency(m_settings.m_toneFrequency); response.getNfmModSettings()->setVolumeFactor(m_settings.m_volumeFactor); SWGSDRangel::SWGCWKeyerSettings *apiCwKeyerSettings = response.getNfmModSettings()->getCwKeyer(); const CWKeyerSettings& cwKeyerSettings = m_cwKeyer.getSettings(); apiCwKeyerSettings->setLoop(cwKeyerSettings.m_loop ? 1 : 0); apiCwKeyerSettings->setMode((int) cwKeyerSettings.m_mode); apiCwKeyerSettings->setSampleRate(cwKeyerSettings.m_sampleRate); apiCwKeyerSettings->setText(new QString(cwKeyerSettings.m_text)); apiCwKeyerSettings->setWpm(cwKeyerSettings.m_wpm); return 200; } int NFMMod::webapiSettingsPutPatch( bool force, const QStringList& channelSettingsKeys, SWGSDRangel::SWGChannelSettings& response, QString& errorMessage __attribute__((unused))) { NFMModSettings settings; bool frequencyOffsetChanged = false; // for (int i = 0; i < channelSettingsKeys.size(); i++) { // qDebug("NFMMod::webapiSettingsPutPatch: settingKey: %s", qPrintable(channelSettingsKeys.at(i))); // } if (channelSettingsKeys.contains("afBandwidth")) { settings.m_afBandwidth = response.getNfmModSettings()->getAfBandwidth(); } if (channelSettingsKeys.contains("audioSampleRate")) { settings.m_audioSampleRate = response.getNfmModSettings()->getAudioSampleRate(); } if (channelSettingsKeys.contains("channelMute")) { settings.m_channelMute = response.getNfmModSettings()->getChannelMute() != 0; } if (channelSettingsKeys.contains("ctcssIndex")) { settings.m_ctcssIndex = response.getNfmModSettings()->getCtcssIndex(); } if (channelSettingsKeys.contains("ctcssOn")) { settings.m_ctcssOn = response.getNfmModSettings()->getCtcssOn() != 0; } if (channelSettingsKeys.contains("fmDeviation")) { settings.m_fmDeviation = response.getNfmModSettings()->getFmDeviation(); } if (channelSettingsKeys.contains("inputFrequencyOffset")) { settings.m_inputFrequencyOffset = response.getNfmModSettings()->getInputFrequencyOffset(); frequencyOffsetChanged = true; } if (channelSettingsKeys.contains("modAFInput")) { settings.m_modAFInput = (NFMModSettings::NFMModInputAF) response.getNfmModSettings()->getModAfInput(); } if (channelSettingsKeys.contains("playLoop")) { settings.m_playLoop = response.getNfmModSettings()->getPlayLoop() != 0; } if (channelSettingsKeys.contains("rfBandwidth")) { settings.m_rfBandwidth = response.getNfmModSettings()->getRfBandwidth(); } if (channelSettingsKeys.contains("rgbColor")) { settings.m_rgbColor = response.getNfmModSettings()->getRgbColor(); } if (channelSettingsKeys.contains("title")) { settings.m_title = *response.getNfmModSettings()->getTitle(); } if (channelSettingsKeys.contains("toneFrequency")) { settings.m_toneFrequency = response.getNfmModSettings()->getToneFrequency(); } if (channelSettingsKeys.contains("volumeFactor")) { settings.m_volumeFactor = response.getNfmModSettings()->getVolumeFactor(); } if (channelSettingsKeys.contains("cwKeyer")) { SWGSDRangel::SWGCWKeyerSettings *apiCwKeyerSettings = response.getNfmModSettings()->getCwKeyer(); CWKeyerSettings cwKeyerSettings = m_cwKeyer.getSettings(); if (channelSettingsKeys.contains("cwKeyer.loop")) { cwKeyerSettings.m_loop = apiCwKeyerSettings->getLoop() != 0; } if (channelSettingsKeys.contains("cwKeyer.mode")) { cwKeyerSettings.m_mode = (CWKeyerSettings::CWMode) apiCwKeyerSettings->getMode(); } if (channelSettingsKeys.contains("cwKeyer.text")) { cwKeyerSettings.m_text = *apiCwKeyerSettings->getText(); } if (channelSettingsKeys.contains("cwKeyer.sampleRate")) { cwKeyerSettings.m_sampleRate = apiCwKeyerSettings->getSampleRate(); } if (channelSettingsKeys.contains("cwKeyer.wpm")) { cwKeyerSettings.m_wpm = apiCwKeyerSettings->getWpm(); } m_cwKeyer.setLoop(cwKeyerSettings.m_loop); m_cwKeyer.setMode(cwKeyerSettings.m_mode); m_cwKeyer.setSampleRate(cwKeyerSettings.m_sampleRate); m_cwKeyer.setText(cwKeyerSettings.m_text); m_cwKeyer.setWPM(cwKeyerSettings.m_wpm); if (m_guiMessageQueue) // forward to GUI if any { CWKeyer::MsgConfigureCWKeyer *msgCwKeyer = CWKeyer::MsgConfigureCWKeyer::create(cwKeyerSettings, force); m_guiMessageQueue->push(msgCwKeyer); } } if (frequencyOffsetChanged) { NFMMod::MsgConfigureChannelizer *msgChan = NFMMod::MsgConfigureChannelizer::create( 48000, settings.m_inputFrequencyOffset); m_inputMessageQueue.push(msgChan); } MsgConfigureNFMMod *msg = MsgConfigureNFMMod::create(settings, force); m_inputMessageQueue.push(msg); if (m_guiMessageQueue) // forward to GUI if any { MsgConfigureNFMMod *msgToGUI = MsgConfigureNFMMod::create(settings, force); m_guiMessageQueue->push(msgToGUI); } return 200; }