//
// libtgvoip is free and unencumbered public domain software.
// For more information, see http://unlicense.org or the UNLICENSE file
// you should have received with this source code distribution.
//

#include "AudioInputAudioUnit.h"
#include "../../logging.h"
#include "AudioUnitIO.h"
#include <cassert>
#include <cstdio>
#include <cstdlib>

#define BUFFER_SIZE 960

using namespace tgvoip;
using namespace tgvoip::audio;

AudioInputAudioUnit::AudioInputAudioUnit(std::string deviceID, AudioUnitIO* io)
{
    remainingDataSize = 0;
    isRecording = false;
    this->io = io;
#if TARGET_OS_OSX
    io->SetCurrentDevice(true, deviceID);
#endif
}

AudioInputAudioUnit::~AudioInputAudioUnit()
{
}

void AudioInputAudioUnit::Start()
{
    isRecording = true;
    io->EnableInput(true);
}

void AudioInputAudioUnit::Stop()
{
    isRecording = false;
    io->EnableInput(false);
}

void AudioInputAudioUnit::HandleBufferCallback(AudioBufferList* ioData)
{
    int i;
    for (i = 0; i < ioData->mNumberBuffers; i++)
    {
        AudioBuffer buf = ioData->mBuffers[i];
#if TARGET_OS_OSX
        assert(remainingDataSize + buf.mDataByteSize / 2 < 10240);
        float* src = reinterpret_cast<float*>(buf.mData);
        std::int16_t* dst = reinterpret_cast<std::int16_t*>(remainingData + remainingDataSize);
        for (int j = 0; j < buf.mDataByteSize / 4; j++)
        {
            dst[j] = (std::int16_t)(src[j] * INT16_MAX);
        }
        remainingDataSize += buf.mDataByteSize / 2;
#else
        assert(remainingDataSize + buf.mDataByteSize < 10240);
        std::memcpy(remainingData + remainingDataSize, buf.mData, buf.mDataByteSize);
        remainingDataSize += buf.mDataByteSize;
#endif
        while (remainingDataSize >= BUFFER_SIZE * 2)
        {
            InvokeCallback(remainingData, BUFFER_SIZE * 2);
            remainingDataSize -= BUFFER_SIZE * 2;
            if (remainingDataSize > 0)
            {
                memmove(remainingData, remainingData + (BUFFER_SIZE * 2), remainingDataSize);
            }
        }
    }
}

#if TARGET_OS_OSX
void AudioInputAudioUnit::SetCurrentDevice(std::string deviceID)
{
    io->SetCurrentDevice(true, deviceID);
}
#endif