1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-11-25 17:28:50 -05:00

Audio output device recording: implemented mono recording option

This commit is contained in:
f4exb 2022-11-11 23:25:04 +01:00
parent 1ca47ddcc9
commit 67ee05c7ef
4 changed files with 55 additions and 8 deletions

View File

@ -218,6 +218,15 @@ void AudioOutputDevice::setUdpChannelFormat(UDPChannelCodec udpChannelCodec, boo
if (m_audioNetSink) {
m_audioNetSink->setParameters((AudioNetSink::Codec) m_udpChannelCodec, stereo, sampleRate);
}
if (m_wavFileRecord)
{
if (m_wavFileRecord->isRecording()) {
m_wavFileRecord->stopRecording();
}
m_wavFileRecord->setMono(!stereo);
}
}
void AudioOutputDevice::setUdpDecimation(uint32_t decimation)
@ -403,12 +412,12 @@ qint64 AudioOutputDevice::readData(char* data, qint64 maxLen)
{
if (m_recordSilenceNbSamples <= 0)
{
m_wavFileRecord->write(sl, sr);
writeSampleToFile(sl, sr);
m_recordSilenceCount = 0;
}
else if (m_recordSilenceCount < m_recordSilenceNbSamples)
{
m_wavFileRecord->write(sl, sr);
writeSampleToFile(sl, sr);
m_recordSilenceCount++;
}
else
@ -422,7 +431,7 @@ qint64 AudioOutputDevice::readData(char* data, qint64 maxLen)
m_wavFileRecord->startRecording();
}
m_wavFileRecord->write(sl, sr);
writeSampleToFile(sl, sr);
m_recordSilenceCount = 0;
}
}
@ -431,6 +440,26 @@ qint64 AudioOutputDevice::readData(char* data, qint64 maxLen)
return samplesPerBuffer * 4;
}
void AudioOutputDevice::writeSampleToFile(qint16 lSample, qint16 rSample)
{
switch (m_udpChannelMode)
{
case UDPChannelStereo:
m_wavFileRecord->write(lSample, rSample);
break;
case UDPChannelMixed:
m_wavFileRecord->writeMono((lSample+rSample)/2);
break;
case UDPChannelRight:
m_wavFileRecord->writeMono(rSample);
break;
case UDPChannelLeft:
default:
m_wavFileRecord->writeMono(lSample);
break;
}
}
qint64 AudioOutputDevice::writeData(const char* data, qint64 len)
{
Q_UNUSED(data);

View File

@ -102,6 +102,7 @@ private:
//virtual bool open(OpenMode mode);
virtual qint64 readData(char* data, qint64 maxLen);
virtual qint64 writeData(const char* data, qint64 len);
void writeSampleToFile(qint16 lSample, qint16 rSample);
friend class AudioOutputPipe;
};

View File

@ -37,7 +37,8 @@ WavFileRecord::WavFileRecord(quint32 sampleRate, quint64 centerFrequency) :
m_recordOn(false),
m_recordStart(false),
m_byteCount(0),
m_msShift(0)
m_msShift(0),
m_nbChannels(2)
{
setObjectName("WavFileRecord");
}
@ -49,7 +50,8 @@ WavFileRecord::WavFileRecord(const QString& fileBase) :
m_centerFrequency(0),
m_recordOn(false),
m_recordStart(false),
m_byteCount(0)
m_byteCount(0),
m_nbChannels(2)
{
setObjectName("WavFileRecord");
}
@ -124,6 +126,18 @@ void WavFileRecord::write(qint16 lSample, qint16 rSample)
m_byteCount += 4;
}
void WavFileRecord::writeMono(qint16 sample)
{
if (m_recordStart)
{
writeHeader();
m_recordStart = false;
}
m_sampleFile.write(reinterpret_cast<const char*>(&sample), 2);
m_byteCount += 2;
}
void WavFileRecord::start()
{
}
@ -229,11 +243,11 @@ void WavFileRecord::writeHeader()
header.m_fmtHeader.m_id[3] = ' ';
header.m_fmtHeader.m_size = 16;
header.m_audioFormat = 1; // Linear PCM
header.m_numChannels = 2; // I/Q
header.m_numChannels = m_nbChannels; // 2 for I/Q
header.m_sampleRate = m_sampleRate;
// We always use 16-bits regardless of SDR_RX_SAMP_SZ
header.m_byteRate = m_sampleRate * 2 * 16 / 8;
header.m_blockAlign = 2 * 16 / 8;
header.m_byteRate = m_sampleRate * m_nbChannels * 16 / 8;
header.m_blockAlign = m_nbChannels * 16 / 8;
header.m_bitsPerSample = 16;
header.m_auxiHeader.m_id[0] = 'a';

View File

@ -93,11 +93,13 @@ public:
void setMsShift(qint64 shift) override { m_msShift = shift; }
virtual int getBytesPerSample() override { return 4; };
const QString& getCurrentFileName() override { return m_currentFileName; }
void setMono(bool mono) { m_nbChannels = mono ? 1 : 2; }
void genUniqueFileName(uint deviceUID, int istream = -1);
virtual void feed(const SampleVector::const_iterator& begin, const SampleVector::const_iterator& end, bool positiveOnly) override;
void write(qint16 lSample, qint16 rSample); //!< write a single sample
void writeMono(qint16 sample); //!< write a single mono sample
virtual void start() override;
virtual void stop() override;
virtual bool handleMessage(const Message& message) override;
@ -124,6 +126,7 @@ private:
QString m_currentFileName;
quint64 m_byteCount;
qint64 m_msShift;
int m_nbChannels;
void writeHeader();
};