audiooutput.cpp

Go to the documentation of this file.

// Copyright (C) 2012 maintech GmbH, Otto-Hahn-Str. 15, 97204 Hoechberg, Germany //
// written by Christian Daniel                                                   //
//                                                                               //
// 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/>.          //

#include <string.h>
#include <QAudioFormat>
#include <QAudioDeviceInfo>
#include <QAudioOutput>
#include "audiooutput.h"
#include "audiofifo.h"
#include "audionetsink.h"

AudioOutput::AudioOutput() :
    m_mutex(QMutex::Recursive),
    m_audioOutput(0),
    m_audioNetSink(0),
    m_copyAudioToUdp(false),
    m_udpChannelMode(UDPChannelLeft),
    m_udpChannelCodec(UDPCodecL16),
    m_audioUsageCount(0),
    m_onExit(false),
    m_audioFifos()
{
}

AudioOutput::~AudioOutput()
{
//  stop();
//
//  QMutexLocker mutexLocker(&m_mutex);
//
//  for (std::list<AudioFifo*>::iterator it = m_audioFifos.begin(); it != m_audioFifos.end(); ++it)
//  {
//      delete *it;
//  }
//
//  m_audioFifos.clear();
}

bool AudioOutput::start(int device, int rate)
{

//  if (m_audioUsageCount == 0)
//  {
        QMutexLocker mutexLocker(&m_mutex);
        QAudioDeviceInfo devInfo;

        if (device < 0)
        {
            devInfo = QAudioDeviceInfo::defaultOutputDevice();
            qWarning("AudioOutput::start: using system default device %s", qPrintable(devInfo.defaultOutputDevice().deviceName()));
        }
        else
        {
            QList<QAudioDeviceInfo> devicesInfo = QAudioDeviceInfo::availableDevices(QAudio::AudioOutput);

            if (device < devicesInfo.size())
            {
                devInfo = devicesInfo[device];
                qWarning("AudioOutput::start: using audio device #%d: %s", device, qPrintable(devInfo.deviceName()));
            }
            else
            {
                devInfo = QAudioDeviceInfo::defaultOutputDevice();
                qWarning("AudioOutput::start: audio device #%d does not exist. Using system default device %s", device, qPrintable(devInfo.defaultOutputDevice().deviceName()));
            }
        }

        //QAudioDeviceInfo devInfo(QAudioDeviceInfo::defaultOutputDevice());

        m_audioFormat.setSampleRate(rate);
        m_audioFormat.setChannelCount(2);
        m_audioFormat.setSampleSize(16);
        m_audioFormat.setCodec("audio/pcm");
        m_audioFormat.setByteOrder(QAudioFormat::LittleEndian);
        m_audioFormat.setSampleType(QAudioFormat::SignedInt);

        if (!devInfo.isFormatSupported(m_audioFormat))
        {
            m_audioFormat = devInfo.nearestFormat(m_audioFormat);
            std::ostringstream os;
            os << " sampleRate: " << m_audioFormat.sampleRate()
               << " channelCount: " << m_audioFormat.channelCount()
               << " sampleSize: " << m_audioFormat.sampleSize()
               << " codec: "  << m_audioFormat.codec().toStdString()
               << " byteOrder: " <<  (m_audioFormat.byteOrder() == QAudioFormat::BigEndian ? "BE" : "LE")
               << " sampleType: " << (int) m_audioFormat.sampleType();
            qWarning("AudioOutput::start: format %d Hz 2xS16LE audio/pcm not supported. Using: %s", rate, os.str().c_str());
        }
        else
        {
            qInfo("AudioOutput::start: audio format OK");
        }

        if (m_audioFormat.sampleSize() != 16)
        {
            qWarning("AudioOutput::start: Audio device '%s' failed", qPrintable(devInfo.defaultOutputDevice().deviceName()));
            return false;
        }

        m_audioOutput = new QAudioOutput(devInfo, m_audioFormat);
        m_audioNetSink = new AudioNetSink(0, m_audioFormat.sampleRate(), false);

        QIODevice::open(QIODevice::ReadOnly);

        m_audioOutput->start(this);

        if (m_audioOutput->state() != QAudio::ActiveState)
        {
            qWarning("AudioOutput::start: cannot start");
        }
//  }
//
//  m_audioUsageCount++;

    return true;
}

void AudioOutput::stop()
{
    qDebug("AudioOutput::stop");

    QMutexLocker mutexLocker(&m_mutex);
    m_audioOutput->stop();
    QIODevice::close();
    delete m_audioNetSink;
    m_audioNetSink = 0;
    delete m_audioOutput;

//    if (m_audioUsageCount > 0)
//    {
//        m_audioUsageCount--;
//
//        if (m_audioUsageCount == 0)
//        {
//            QMutexLocker mutexLocker(&m_mutex);
//            QIODevice::close();
//
//            if (!m_onExit) {
//                delete m_audioOutput;
//            }
//        }
//    }
}

void AudioOutput::addFifo(AudioFifo* audioFifo)
{
    QMutexLocker mutexLocker(&m_mutex);

    m_audioFifos.push_back(audioFifo);
}

void AudioOutput::removeFifo(AudioFifo* audioFifo)
{
    QMutexLocker mutexLocker(&m_mutex);

    m_audioFifos.remove(audioFifo);
}

/*
bool AudioOutput::open(OpenMode mode)
{
    Q_UNUSED(mode);
    return false;
}*/

void AudioOutput::setUdpDestination(const QString& address, uint16_t port)
{
    if (m_audioNetSink) {
        m_audioNetSink->setDestination(address, port);
    }
}

void AudioOutput::setUdpCopyToUDP(bool copyToUDP)
{
    m_copyAudioToUdp = copyToUDP;
}

void AudioOutput::setUdpUseRTP(bool useRTP)
{
    if (m_audioNetSink) {
        m_audioNetSink->selectType(useRTP ? AudioNetSink::SinkRTP : AudioNetSink::SinkUDP);
    }
}

void AudioOutput::setUdpChannelMode(UDPChannelMode udpChannelMode)
{
    m_udpChannelMode = udpChannelMode;
}

void AudioOutput::setUdpChannelFormat(UDPChannelCodec udpChannelCodec, bool stereo, int sampleRate)
{
    m_udpChannelCodec = udpChannelCodec;

    if (m_audioNetSink) {
        m_audioNetSink->setParameters((AudioNetSink::Codec) m_udpChannelCodec, stereo, sampleRate);
    }
}

void AudioOutput::setUdpDecimation(uint32_t decimation)
{
    if (m_audioNetSink) {
        m_audioNetSink->setDecimation(decimation);
    }
}

qint64 AudioOutput::readData(char* data, qint64 maxLen)
{
    //qDebug("AudioOutput::readData: %lld", maxLen);

    // Study this mutex on OSX, for now deadlocks possible
    // Removed as it may indeed cause lockups and is in fact useless.
//#ifndef __APPLE__
//    QMutexLocker mutexLocker(&m_mutex);
//#endif

    unsigned int samplesPerBuffer = maxLen / 4;

    if (samplesPerBuffer == 0)
    {
        return 0;
    }

    if (m_mixBuffer.size() < samplesPerBuffer * 2)
    {
        m_mixBuffer.resize(samplesPerBuffer * 2); // allocate 2 qint32 per sample (stereo)

        if (m_mixBuffer.size() != samplesPerBuffer * 2)
        {
            return 0;
        }
    }

    memset(&m_mixBuffer[0], 0x00, 2 * samplesPerBuffer * sizeof(m_mixBuffer[0])); // start with silence

    // sum up a block from all fifos

    for (std::list<AudioFifo*>::iterator it = m_audioFifos.begin(); it != m_audioFifos.end(); ++it)
    {
        // use outputBuffer as temp - yes, one memcpy could be saved
        unsigned int samples = (*it)->read((quint8*) data, samplesPerBuffer);
        const qint16* src = (const qint16*) data;
        std::vector<qint32>::iterator dst = m_mixBuffer.begin();

//      if (samples != framesPerBuffer)
//      {
//            qDebug("AudioOutput::readData: read %d samples vs %d requested", samples, framesPerBuffer);
//      }

        for (unsigned int i = 0; i < samples; i++)
        {
            *dst += *src;
            ++src;
            ++dst;
            *dst += *src;
            ++src;
            ++dst;
        }
    }
    // convert to int16

    //std::vector<qint32>::const_iterator src = m_mixBuffer.begin(); // Valgrind optim
    qint16* dst = (qint16*) data;
    qint32 sl, sr;

    for (unsigned int i = 0; i < samplesPerBuffer; i++)
    {
        // left channel

        //s = *src++; // Valgrind optim
        sl = m_mixBuffer[2*i];

        if(sl < -32768)
        {
            sl = -32768;
        }
        else if (sl > 32767)
        {
            sl = 32767;
        }

        *dst++ = sl;

        // right channel

        //s = *src++; // Valgrind optim
        sr = m_mixBuffer[2*i + 1];

        if(sr < -32768)
        {
            sr = -32768;
        }
        else if (sr > 32767)
        {
            sr = 32767;
        }

        *dst++ = sr;

        if ((m_copyAudioToUdp) && (m_audioNetSink))
        {
            switch (m_udpChannelMode)
            {
            case UDPChannelStereo:
                m_audioNetSink->write(sl, sr);
                break;
            case UDPChannelMixed:
                m_audioNetSink->write((sl+sr)/2);
                break;
            case UDPChannelRight:
                m_audioNetSink->write(sr);
                break;
            case UDPChannelLeft:
            default:
                m_audioNetSink->write(sl);
                break;
            }
        }
    }

    return samplesPerBuffer * 4;
}

qint64 AudioOutput::writeData(const char* data, qint64 len)
{
    Q_UNUSED(data);
    Q_UNUSED(len);
    return 0;
}