/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtMultimedia module of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

//

//  W A R N I N G

//  -------------

//

// This file is not part of the Qt API.  It exists for the convenience

// of other Qt classes.  This header file may change from version to

// version without notice, or even be removed.

//

// We mean it.

//

#include <QtCore/qendian.h>

#include <QtCore/qtimer.h>

#include <QtCore/qdebug.h>

#include <QtMultimedia/qaudioinput.h>

#include "qaudio_mac_p.h"

#include "qaudioinput_mac_p.h"

#include "qaudiodeviceinfo_mac_p.h"

QT_BEGIN_NAMESPACE

namespace QtMultimediaInternal

{

static const int default_buffer_size = 4 * 1024;

class QAudioBufferList

{

public:

    QAudioBufferList(AudioStreamBasicDescription const& streamFormat):

        owner(false),

        sf(streamFormat)

    {

        const bool isInterleaved = (sf.mFormatFlags & kAudioFormatFlagIsNonInterleaved) == 0;

        const int numberOfBuffers = isInterleaved ? 1 : sf.mChannelsPerFrame;

        dataSize = 0;

        bfs = reinterpret_cast<AudioBufferList*>(qMalloc(sizeof(AudioBufferList) +

                                                                (sizeof(AudioBuffer) * numberOfBuffers)));

        bfs->mNumberBuffers = numberOfBuffers;

        for (int i = 0; i < numberOfBuffers; ++i) {

            bfs->mBuffers[i].mNumberChannels = isInterleaved ? numberOfBuffers : 1;

            bfs->mBuffers[i].mDataByteSize = 0;

            bfs->mBuffers[i].mData = 0;

        }

    }

    QAudioBufferList(AudioStreamBasicDescription const& streamFormat, char* buffer, int bufferSize):

        owner(false),

        sf(streamFormat),

        bfs(0)

    {

        dataSize = bufferSize;

        bfs = reinterpret_cast<AudioBufferList*>(qMalloc(sizeof(AudioBufferList) + sizeof(AudioBuffer)));

        bfs->mNumberBuffers = 1;

        bfs->mBuffers[0].mNumberChannels = 1;

        bfs->mBuffers[0].mDataByteSize = dataSize;

        bfs->mBuffers[0].mData = buffer;

    }

    QAudioBufferList(AudioStreamBasicDescription const& streamFormat, int framesToBuffer):

        owner(true),

        sf(streamFormat),

        bfs(0)

    {

        const bool isInterleaved = (sf.mFormatFlags & kAudioFormatFlagIsNonInterleaved) == 0;

        const int numberOfBuffers = isInterleaved ? 1 : sf.mChannelsPerFrame;

        dataSize = framesToBuffer * sf.mBytesPerFrame;

        bfs = reinterpret_cast<AudioBufferList*>(qMalloc(sizeof(AudioBufferList) +

                                                                (sizeof(AudioBuffer) * numberOfBuffers)));

        bfs->mNumberBuffers = numberOfBuffers;

        for (int i = 0; i < numberOfBuffers; ++i) {

            bfs->mBuffers[i].mNumberChannels = isInterleaved ? numberOfBuffers : 1;

            bfs->mBuffers[i].mDataByteSize = dataSize;

            bfs->mBuffers[i].mData = qMalloc(dataSize);

        }

    }

    ~QAudioBufferList()

    {

        if (owner) {

            for (UInt32 i = 0; i < bfs->mNumberBuffers; ++i)

                qFree(bfs->mBuffers[i].mData);

        }

        qFree(bfs);

    }

    AudioBufferList* audioBufferList() const

    {

        return bfs;

    }

    char* data(int buffer = 0) const

    {

        return static_cast<char*>(bfs->mBuffers[buffer].mData);

    }

    qint64 bufferSize(int buffer = 0) const

    {

        return bfs->mBuffers[buffer].mDataByteSize;

    }

    int frameCount(int buffer = 0) const

    {

        return bfs->mBuffers[buffer].mDataByteSize / sf.mBytesPerFrame;

    }

    int packetCount(int buffer = 0) const

    {

        return bfs->mBuffers[buffer].mDataByteSize / sf.mBytesPerPacket;

    }

    int packetSize() const

    {

        return sf.mBytesPerPacket;

    }

    void reset()

    {

        for (UInt32 i = 0; i < bfs->mNumberBuffers; ++i) {

            bfs->mBuffers[i].mDataByteSize = dataSize;

            bfs->mBuffers[i].mData = 0;

        }

    }

private:

    bool    owner;

    int     dataSize;

    AudioStreamBasicDescription sf;

    AudioBufferList* bfs;

};

class QAudioPacketFeeder

{

public:

    QAudioPacketFeeder(QAudioBufferList* abl):

        audioBufferList(abl)

    {

        totalPackets = audioBufferList->packetCount();

        position = 0;

    }

    bool feed(AudioBufferList& dst, UInt32& packetCount)

    {

        if (position == totalPackets) {

            dst.mBuffers[0].mDataByteSize = 0;

            packetCount = 0;

            return false;

        }

        if (totalPackets - position < packetCount)

            packetCount = totalPackets - position;

        dst.mBuffers[0].mDataByteSize = packetCount * audioBufferList->packetSize();

        dst.mBuffers[0].mData = audioBufferList->data() + (position * audioBufferList->packetSize());

        position += packetCount;

        return true;

    }

    bool empty() const

    {

        return position == totalPackets;

    }

private:

    UInt32 totalPackets;

    UInt32 position;

    QAudioBufferList*   audioBufferList;

};

class QAudioInputBuffer : public QObject

{

    Q_OBJECT

public:

    QAudioInputBuffer(int bufferSize,

                        int maxPeriodSize,

                        AudioStreamBasicDescription const& inputFormat,

                        AudioStreamBasicDescription const& outputFormat,

                        QObject* parent):

        QObject(parent),

        m_deviceError(false),

        m_audioConverter(0),

        m_inputFormat(inputFormat),

        m_outputFormat(outputFormat)

    {

        m_maxPeriodSize = maxPeriodSize;

        m_periodTime = m_maxPeriodSize / m_outputFormat.mBytesPerFrame * 1000 / m_outputFormat.mSampleRate;

        m_buffer = new QAudioRingBuffer(bufferSize + (bufferSize % maxPeriodSize == 0 ? 0 : maxPeriodSize - (bufferSize % maxPeriodSize)));

        m_inputBufferList = new QAudioBufferList(m_inputFormat);

        m_flushTimer = new QTimer(this);

        connect(m_flushTimer, SIGNAL(timeout()), SLOT(flushBuffer()));

        if (toQAudioFormat(inputFormat) != toQAudioFormat(outputFormat)) {

            if (AudioConverterNew(&m_inputFormat, &m_outputFormat, &m_audioConverter) != noErr) {

                qWarning() << "QAudioInput: Unable to create an Audio Converter";

                m_audioConverter = 0;

            }

        }

    }

    ~QAudioInputBuffer()

    {

        delete m_buffer;

    }

    qint64 renderFromDevice(AudioUnit audioUnit,

                             AudioUnitRenderActionFlags* ioActionFlags,

                             const AudioTimeStamp* inTimeStamp,

                             UInt32 inBusNumber,

                             UInt32 inNumberFrames)

    {

        const bool  pullMode = m_device == 0;

        OSStatus    err;

        qint64      framesRendered = 0;

        m_inputBufferList->reset();

        err = AudioUnitRender(audioUnit,

                                ioActionFlags,

                                inTimeStamp,

                                inBusNumber,

                                inNumberFrames,

                                m_inputBufferList->audioBufferList());

        if (m_audioConverter != 0) {

            QAudioPacketFeeder  feeder(m_inputBufferList);

            int     copied = 0;

            const int available = m_buffer->free();

            while (err == noErr && !feeder.empty()) {

                QAudioRingBuffer::Region region = m_buffer->acquireWriteRegion(available);

                if (region.second == 0)

                    break;

                AudioBufferList     output;

                output.mNumberBuffers = 1;

                output.mBuffers[0].mNumberChannels = 1;

                output.mBuffers[0].mDataByteSize = region.second;

                output.mBuffers[0].mData = region.first;

                UInt32  packetSize = region.second / m_outputFormat.mBytesPerPacket;

                err = AudioConverterFillComplexBuffer(m_audioConverter,

                                                      converterCallback,

                                                      &feeder,

                                                      &packetSize,

                                                      &output,

                                                      0);

                region.second = output.mBuffers[0].mDataByteSize;

                copied += region.second;

                m_buffer->releaseWriteRegion(region);

            }

            framesRendered += copied / m_outputFormat.mBytesPerFrame;

        }

        else {

            const int available = m_inputBufferList->bufferSize();

            bool    wecan = true;

            int     copied = 0;

            while (wecan && copied < available) {

                QAudioRingBuffer::Region region = m_buffer->acquireWriteRegion(available - copied);

                if (region.second > 0) {

                    memcpy(region.first, m_inputBufferList->data() + copied, region.second);

                    copied += region.second;

                }

                else

                    wecan = false;

                m_buffer->releaseWriteRegion(region);

            }

            framesRendered = copied / m_outputFormat.mBytesPerFrame;

        }

        if (pullMode && framesRendered > 0)

            emit readyRead();

        return framesRendered;

    }

    qint64 readBytes(char* data, qint64 len)

    {

        bool    wecan = true;

        qint64  bytesCopied = 0;

        len -= len % m_maxPeriodSize;

        while (wecan && bytesCopied < len) {

            QAudioRingBuffer::Region region = m_buffer->acquireReadRegion(len - bytesCopied);

            if (region.second > 0) {

                memcpy(data + bytesCopied, region.first, region.second);

                bytesCopied += region.second;

            }

            else

                wecan = false;

            m_buffer->releaseReadRegion(region);

        }

        return bytesCopied;

    }

    void setFlushDevice(QIODevice* device)

    {

        if (m_device != device)

            m_device = device;

    }

    void startFlushTimer()

    {

        if (m_device != 0) {

            m_flushTimer->start((m_buffer->size() - (m_maxPeriodSize * 2)) / m_maxPeriodSize * m_periodTime);

        }

    }

    void stopFlushTimer()

    {

        m_flushTimer->stop();

    }

    void flush(bool all = false)

    {

        if (m_device == 0)

            return;

        const int used = m_buffer->used();

        const int readSize = all ? used : used - (used % m_maxPeriodSize);

        if (readSize > 0) {

            bool    wecan = true;

            int     flushed = 0;

            while (!m_deviceError && wecan && flushed < readSize) {

                QAudioRingBuffer::Region region = m_buffer->acquireReadRegion(readSize - flushed);

                if (region.second > 0) {

                    int bytesWritten = m_device->write(region.first, region.second);

                    if (bytesWritten < 0) {

                        stopFlushTimer();

                        m_deviceError = true;

                    }

                    else {

                        region.second = bytesWritten;

                        flushed += bytesWritten;

                        wecan = bytesWritten != 0;

                    }

                }

                else

                    wecan = false;

                m_buffer->releaseReadRegion(region);

            }

        }

    }

    void reset()

    {

        m_buffer->reset();

        m_deviceError = false;

    }

    int available() const

    {

        return m_buffer->free();

    }

    int used() const

    {

        return m_buffer->used();

    }

signals:

    void readyRead();

private slots:

    void flushBuffer()

    {

        flush();

    }

private:

    bool        m_deviceError;

    int         m_maxPeriodSize;

    int         m_periodTime;

    QIODevice*  m_device;

    QTimer*     m_flushTimer;

    QAudioRingBuffer*   m_buffer;

    QAudioBufferList*   m_inputBufferList;

    AudioConverterRef   m_audioConverter;

    AudioStreamBasicDescription m_inputFormat;

    AudioStreamBasicDescription m_outputFormat;

    const static OSStatus as_empty = 'qtem';

    // Converter callback

    static OSStatus converterCallback(AudioConverterRef inAudioConverter,

                                UInt32* ioNumberDataPackets,

                                AudioBufferList* ioData,

                                AudioStreamPacketDescription** outDataPacketDescription,

                                void* inUserData)

    {

        Q_UNUSED(inAudioConverter);

        Q_UNUSED(outDataPacketDescription);

        QAudioPacketFeeder* feeder = static_cast<QAudioPacketFeeder*>(inUserData);

        if (!feeder->feed(*ioData, *ioNumberDataPackets))

            return as_empty;

        return noErr;

    }

};

class MacInputDevice : public QIODevice

{

    Q_OBJECT

public:

    MacInputDevice(QAudioInputBuffer* audioBuffer, QObject* parent):

        QIODevice(parent),

        m_audioBuffer(audioBuffer)

    {

        open(QIODevice::ReadOnly | QIODevice::Unbuffered);

        connect(m_audioBuffer, SIGNAL(readyRead()), SIGNAL(readyRead()));

    }

    qint64 readData(char* data, qint64 len)

    {

        return m_audioBuffer->readBytes(data, len);

    }

    qint64 writeData(const char* data, qint64 len)

    {

        Q_UNUSED(data);

        Q_UNUSED(len);

        return 0;

    }

    bool isSequential() const

    {

        return true;

    }

private:

    QAudioInputBuffer*   m_audioBuffer;

};

}

QAudioInputPrivate::QAudioInputPrivate(const QByteArray& device, QAudioFormat const& format):

    audioFormat(format)

{

    QDataStream ds(device);

    quint32 did, mode;

    ds >> did >> mode;

    if (QAudio::Mode(mode) == QAudio::AudioOutput)

        errorCode = QAudio::OpenError;

    else {

        audioDeviceInfo = new QAudioDeviceInfoInternal(device, QAudio::AudioInput);

        isOpen = false;