/*

    SDL - Simple DirectMedia Layer

    Copyright (C) 1997-2006 Sam Lantinga

    This library is free software; you can redistribute it and/or

    modify it under the terms of the GNU Lesser General Public

    License as published by the Free Software Foundation; either

    version 2.1 of the License, or (at your option) any later version.

    This library 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

    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public

    License along with this library; if not, write to the Free Software

    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

    Sam Lantinga

    slouken@libsdl.org

*/

#include "SDL_config.h"

#include <CoreAudio/CoreAudio.h>

#include <AudioUnit/AudioUnit.h>

#ifdef AVAILABLE_MAC_OS_X_VERSION_10_5_AND_LATER

#include <AudioUnit/AUNTComponent.h>

#endif

#include "SDL_audio.h"

#include "../SDL_audio_c.h"

#include "../SDL_sysaudio.h"

#include "SDL_coreaudio.h"

#define DEBUG_COREAUDIO 0

typedef struct COREAUDIO_DeviceList

{

    AudioDeviceID id;

    const char *name;

} COREAUDIO_DeviceList;

static COREAUDIO_DeviceList *inputDevices = NULL;

static int inputDeviceCount = 0;

static COREAUDIO_DeviceList *outputDevices = NULL;

static int outputDeviceCount = 0;

static void

free_device_list(COREAUDIO_DeviceList ** devices, int *devCount)

{

    if (*devices) {

        int i = *devCount;

        while (i--)

            SDL_free((void *) (*devices)[i].name);

        SDL_free(*devices);

        *devices = NULL;

    }

    *devCount = 0;

}

static void

build_device_list(int iscapture, COREAUDIO_DeviceList ** devices,

                  int *devCount)

{

    Boolean outWritable = 0;

    OSStatus result = noErr;

    UInt32 size = 0;

    AudioDeviceID *devs = NULL;

    UInt32 i = 0;

    UInt32 max = 0;

    free_device_list(devices, devCount);

    result = AudioHardwareGetPropertyInfo(kAudioHardwarePropertyDevices,

                                          &size, &outWritable);

    if (result != kAudioHardwareNoError)

        return;

    devs = (AudioDeviceID *) alloca(size);

    if (devs == NULL)

        return;

    max = size / sizeof(AudioDeviceID);

    *devices = (COREAUDIO_DeviceList *) SDL_malloc(max * sizeof(**devices));

    if (*devices == NULL)

        return;

    result = AudioHardwareGetProperty(kAudioHardwarePropertyDevices,

                                      &size, devs);

    if (result != kAudioHardwareNoError)

        return;

    for (i = 0; i < max; i++) {

        CFStringRef cfstr = NULL;

        char *ptr = NULL;

        AudioDeviceID dev = devs[i];

        AudioBufferList *buflist = NULL;

        int usable = 0;

        CFIndex len = 0;

        result = AudioDeviceGetPropertyInfo(dev, 0, iscapture,

                                            kAudioDevicePropertyStreamConfiguration,

                                            &size, &outWritable);

        if (result != noErr)

            continue;

        buflist = (AudioBufferList *) SDL_malloc(size);

        if (buflist == NULL)

            continue;

        result = AudioDeviceGetProperty(dev, 0, iscapture,

                                        kAudioDevicePropertyStreamConfiguration,

                                        &size, buflist);

        if (result == noErr) {

            UInt32 j;

            for (j = 0; j < buflist->mNumberBuffers; j++) {

                if (buflist->mBuffers[j].mNumberChannels > 0) {

                    usable = 1;

                    break;

                }

            }

        }

        SDL_free(buflist);

        if (!usable)

            continue;

        size = sizeof(CFStringRef);

        result = AudioDeviceGetProperty(dev, 0, iscapture,

                                        kAudioDevicePropertyDeviceNameCFString,

                                        &size, &cfstr);

        if (result != kAudioHardwareNoError)

            continue;

        len = CFStringGetMaximumSizeForEncoding(CFStringGetLength(cfstr),

                                                kCFStringEncodingUTF8);

        ptr = (char *) SDL_malloc(len + 1);

        usable = ((ptr != NULL) &&

                  (CFStringGetCString

                   (cfstr, ptr, len + 1, kCFStringEncodingUTF8)));

        CFRelease(cfstr);

        if (usable) {

            len = strlen(ptr);

            /* Some devices have whitespace at the end...trim it. */

            while ((len > 0) && (ptr[len - 1] == ' ')) {

                len--;

            }

            usable = (len > 0);

        }

        if (!usable) {

            SDL_free(ptr);

        } else {

            ptr[len] = '\0';

#if DEBUG_COREAUDIO

            printf("COREAUDIO: Found %s device #%d: '%s' (devid %d)\n",

                   ((iscapture) ? "capture" : "output"),

                   (int) *devCount, ptr, (int) dev);

#endif

            (*devices)[*devCount].id = dev;

            (*devices)[*devCount].name = ptr;

            (*devCount)++;

        }

    }

}

static inline void

build_device_lists(void)

{

    build_device_list(0, &outputDevices, &outputDeviceCount);

    build_device_list(1, &inputDevices, &inputDeviceCount);

}

static inline void

free_device_lists(void)

{

    free_device_list(&outputDevices, &outputDeviceCount);

    free_device_list(&inputDevices, &inputDeviceCount);

}

static int

find_device_id(const char *devname, int iscapture, AudioDeviceID * id)

{

    int i = ((iscapture) ? inputDeviceCount : outputDeviceCount);

    COREAUDIO_DeviceList *devs = ((iscapture) ? inputDevices : outputDevices);

    while (i--) {

        if (SDL_strcmp(devname, devs->name) == 0) {

            *id = devs->id;

            return 1;

        }

        devs++;

    }

    return 0;

}

static int

COREAUDIO_DetectDevices(int iscapture)

{

    if (iscapture) {

        build_device_list(1, &inputDevices, &inputDeviceCount);

        return inputDeviceCount;

    } else {

        build_device_list(0, &outputDevices, &outputDeviceCount);

        return outputDeviceCount;

    }

    return 0;                   /* shouldn't ever hit this. */

}

static const char *

COREAUDIO_GetDeviceName(int index, int iscapture)

{

    if ((iscapture) && (index < inputDeviceCount)) {

        return inputDevices[index].name;

    } else if ((!iscapture) && (index < outputDeviceCount)) {

        return outputDevices[index].name;

    }

    SDL_SetError("No such device");

    return NULL;

}

static void

COREAUDIO_Deinitialize(void)

{

    free_device_lists();

}

/* The CoreAudio callback */

static OSStatus

outputCallback(void *inRefCon,

               AudioUnitRenderActionFlags * ioActionFlags,

               const AudioTimeStamp * inTimeStamp,

               UInt32 inBusNumber, UInt32 inNumberFrames,

               AudioBufferList * ioDataList)

{

    SDL_AudioDevice *this = (SDL_AudioDevice *) inRefCon;

    AudioBuffer *ioData = &ioDataList->mBuffers[0];

    UInt32 remaining, len;

    void *ptr;

    /* Is there ever more than one buffer, and what do you do with it? */

    if (ioDataList->mNumberBuffers != 1) {

        return noErr;

    }

    /* Only do anything if audio is enabled and not paused */

    if (!this->enabled || this->paused) {

        SDL_memset(ioData->mData, this->spec.silence, ioData->mDataByteSize);

        return 0;

    }

    /* No SDL conversion should be needed here, ever, since we accept

       any input format in OpenAudio, and leave the conversion to CoreAudio.

     */

    /*

       assert(!this->convert.needed);

       assert(this->spec.channels == ioData->mNumberChannels);

     */

    remaining = ioData->mDataByteSize;

    ptr = ioData->mData;

    while (remaining > 0) {

        if (this->hidden->bufferOffset >= this->hidden->bufferSize) {

            /* Generate the data */

            SDL_memset(this->hidden->buffer, this->spec.silence,

                       this->hidden->bufferSize);

            SDL_mutexP(this->mixer_lock);

            (*this->spec.callback) (this->spec.userdata, this->hidden->buffer,

                                    this->hidden->bufferSize);

            SDL_mutexV(this->mixer_lock);

            this->hidden->bufferOffset = 0;

        }

        len = this->hidden->bufferSize - this->hidden->bufferOffset;

        if (len > remaining)

            len = remaining;

        SDL_memcpy(ptr,

                   (char *) this->hidden->buffer + this->hidden->bufferOffset,

                   len);

        ptr = (char *) ptr + len;

        remaining -= len;

        this->hidden->bufferOffset += len;

    }

    return 0;

}

static OSStatus

inputCallback(void *inRefCon,

              AudioUnitRenderActionFlags * ioActionFlags,

              const AudioTimeStamp * inTimeStamp,

              UInt32 inBusNumber, UInt32 inNumberFrames,

              AudioBufferList * ioData)

{

    //err = AudioUnitRender(afr->fAudioUnit, ioActionFlags, inTimeStamp, inBusNumber, inNumberFrames, afr->fAudioBuffer);

    // !!! FIXME: write me!

    return noErr;

}

static void

COREAUDIO_CloseDevice(_THIS)

{

    if (this->hidden != NULL) {

        if (this->hidden->audioUnitOpened) {

            OSStatus result = noErr;

            AURenderCallbackStruct callback;

            const AudioUnitElement output_bus = 0;

            const AudioUnitElement input_bus = 1;

            const int iscapture = this->iscapture;

            const AudioUnitElement bus =

                ((iscapture) ? input_bus : output_bus);

            const AudioUnitScope scope =

                ((iscapture) ? kAudioUnitScope_Output :

                 kAudioUnitScope_Input);

            /* stop processing the audio unit */

            result = AudioOutputUnitStop(this->hidden->audioUnit);

            /* Remove the input callback */

            SDL_memset(&callback, '\0', sizeof(AURenderCallbackStruct));

            result = AudioUnitSetProperty(this->hidden->audioUnit,

                                          kAudioUnitProperty_SetRenderCallback,

                                          scope, bus, &callback,

                                          sizeof(callback));

            CloseComponent(this->hidden->audioUnit);

            this->hidden->audioUnitOpened = 0;

        }

        SDL_free(this->hidden->buffer);

        SDL_free(this->hidden);

        this->hidden = NULL;

    }

}

#define CHECK_RESULT(msg) \

    if (result != noErr) { \

        COREAUDIO_CloseDevice(this); \

        SDL_SetError("CoreAudio error (%s): %d", msg, (int) result); \

        return 0; \

    }

static int

find_device_by_name(_THIS, const char *devname, int iscapture)

{

    AudioDeviceID devid = 0;

    OSStatus result = noErr;

    UInt32 size = 0;

    UInt32 alive = 0;

    pid_t pid = 0;

    if (devname == NULL) {

        size = sizeof(AudioDeviceID);

        const AudioHardwarePropertyID propid =

            ((iscapture) ? kAudioHardwarePropertyDefaultInputDevice :

             kAudioHardwarePropertyDefaultOutputDevice);

        result = AudioHardwareGetProperty(propid, &size, &devid);

        CHECK_RESULT("AudioHardwareGetProperty (default device)");

    } else {

        if (!find_device_id(devname, iscapture, &devid)) {

            SDL_SetError("CoreAudio: No such audio device.");

            return 0;

        }

    }

    size = sizeof(alive);

    result = AudioDeviceGetProperty(devid, 0, iscapture,

                                    kAudioDevicePropertyDeviceIsAlive,

                                    &size, &alive);

    CHECK_RESULT

        ("AudioDeviceGetProperty (kAudioDevicePropertyDeviceIsAlive)");

    if (!alive) {

        SDL_SetError("CoreAudio: requested device exists, but isn't alive.");

        return 0;

    }

    size = sizeof(pid);

    result = AudioDeviceGetProperty(devid, 0, iscapture,

                                    kAudioDevicePropertyHogMode, &size, &pid);

    /* some devices don't support this property, so errors are fine here. */

    if ((result == noErr) && (pid != -1)) {

        SDL_SetError("CoreAudio: requested device is being hogged.");

        return 0;

    }

    this->hidden->deviceID = devid;

    return 1;

}

static int

prepare_audiounit(_THIS, const char *devname, int iscapture,

                  const AudioStreamBasicDescription * strdesc)

{

    OSStatus result = noErr;

    AURenderCallbackStruct callback;

    ComponentDescription desc;

    Component comp = NULL;

    UInt32 enableIO = 0;

    const AudioUnitElement output_bus = 0;

    const AudioUnitElement input_bus = 1;

    const AudioUnitElement bus = ((iscapture) ? input_bus : output_bus);

    const AudioUnitScope scope = ((iscapture) ? kAudioUnitScope_Output :

                                  kAudioUnitScope_Input);

    if (!find_device_by_name(this, devname, iscapture)) {

        SDL_SetError("Couldn't find requested CoreAudio device");

        return 0;

    }

    SDL_memset(&desc, '\0', sizeof(ComponentDescription));

    desc.componentType = kAudioUnitType_Output;

    desc.componentSubType = kAudioUnitSubType_HALOutput;

    desc.componentManufacturer = kAudioUnitManufacturer_Apple;

    comp = FindNextComponent(NULL, &desc);

    if (comp == NULL) {

        SDL_SetError("Couldn't find requested CoreAudio component");

        return 0;

    }

    /* Open & initialize the audio unit */

    result = OpenAComponent(comp, &this->hidden->audioUnit);

    CHECK_RESULT("OpenAComponent");

    this->hidden->audioUnitOpened = 1;

    // !!! FIXME: this is wrong?

    enableIO = ((iscapture) ? 1 : 0);

    result = AudioUnitSetProperty(this->hidden->audioUnit,

                                  kAudioOutputUnitProperty_EnableIO,

                                  kAudioUnitScope_Input, input_bus,

                                  &enableIO, sizeof(enableIO));

    CHECK_RESULT("AudioUnitSetProperty (kAudioUnitProperty_EnableIO input)");

    // !!! FIXME: this is wrong?

    enableIO = ((iscapture) ? 0 : 1);

    result = AudioUnitSetProperty(this->hidden->audioUnit,

                                  kAudioOutputUnitProperty_EnableIO,

                                  kAudioUnitScope_Output, output_bus,

                                  &enableIO, sizeof(enableIO));

    CHECK_RESULT("AudioUnitSetProperty (kAudioUnitProperty_EnableIO output)");

    result = AudioUnitSetProperty(this->hidden->audioUnit,

                                  kAudioOutputUnitProperty_CurrentDevice,

                                  kAudioUnitScope_Global, 0,

                                  &this->hidden->deviceID,

                                  sizeof(AudioDeviceID));

    CHECK_RESULT

        ("AudioUnitSetProperty (kAudioOutputUnitProperty_CurrentDevice)");

    /* Set the data format of the audio unit. */

    result = AudioUnitSetProperty(this->hidden->audioUnit,

                                  kAudioUnitProperty_StreamFormat,

                                  scope, bus, strdesc, sizeof(*strdesc));

    CHECK_RESULT("AudioUnitSetProperty (kAudioUnitProperty_StreamFormat)");

    /* Set the audio callback */

    SDL_memset(&callback, '\0', sizeof(AURenderCallbackStruct));

    callback.inputProc = ((iscapture) ? inputCallback : outputCallback);

    callback.inputProcRefCon = this;

    result = AudioUnitSetProperty(this->hidden->audioUnit,

                                  kAudioUnitProperty_SetRenderCallback,

                                  scope, bus, &callback, sizeof(callback));

    CHECK_RESULT

        ("AudioUnitSetProperty (kAudioUnitProperty_SetInputCallback)");

    /* Calculate the final parameters for this audio specification */

    SDL_CalculateAudioSpec(&this->spec);

    /* Allocate a sample buffer */

    this->hidden->bufferOffset = this->hidden->bufferSize = this->spec.size;

    this->hidden->buffer = SDL_malloc(this->hidden->bufferSize);

    result = AudioUnitInitialize(this->hidden->audioUnit);

    CHECK_RESULT("AudioUnitInitialize");

    /* Finally, start processing of the audio unit */

    result = AudioOutputUnitStart(this->hidden->audioUnit);

    CHECK_RESULT("AudioOutputUnitStart");

    /* We're running! */

    return 1;

}

static int

COREAUDIO_OpenDevice(_THIS, const char *devname, int iscapture)

{

    AudioStreamBasicDescription strdesc;

    SDL_AudioFormat test_format = SDL_FirstAudioFormat(this->spec.format);

    int valid_datatype = 0;

    /* Initialize all variables that we clean on shutdown */

    this->hidden = (struct SDL_PrivateAudioData *)

        SDL_malloc((sizeof *this->hidden));

    if (this->hidden == NULL) {

        SDL_OutOfMemory();

        return (0);

    }

    SDL_memset(this->hidden, 0, (sizeof *this->hidden));

    /* Setup a AudioStreamBasicDescription with the requested format */

    SDL_memset(&strdesc, '\0', sizeof(AudioStreamBasicDescription));

    strdesc.mFormatID = kAudioFormatLinearPCM;

    strdesc.mFormatFlags = kLinearPCMFormatFlagIsPacked;

    strdesc.mChannelsPerFrame = this->spec.channels;

    strdesc.mSampleRate = this->spec.freq;

    strdesc.mFramesPerPacket = 1;

    while ((!valid_datatype) && (test_format)) {

        this->spec.format = test_format;

        /* Just a list of valid SDL formats, so people don't pass junk here. */

        switch (test_format) {

        case AUDIO_U8:

        case AUDIO_S8:

        case AUDIO_U16LSB:

        case AUDIO_S16LSB:

        case AUDIO_U16MSB:

        case AUDIO_S16MSB:

        case AUDIO_S32LSB:

        case AUDIO_S32MSB:

        case AUDIO_F32LSB:

        case AUDIO_F32MSB:

            valid_datatype = 1;

            strdesc.mBitsPerChannel = SDL_AUDIO_BITSIZE(this->spec.format);

            if (SDL_AUDIO_ISBIGENDIAN(this->spec.format))

                strdesc.mFormatFlags |= kLinearPCMFormatFlagIsBigEndian;

            if (SDL_AUDIO_ISFLOAT(this->spec.format))

                strdesc.mFormatFlags |= kLinearPCMFormatFlagIsFloat;

            else if (SDL_AUDIO_ISSIGNED(this->spec.format))

                strdesc.mFormatFlags |= kLinearPCMFormatFlagIsSignedInteger;

            break;

        }

    }

    if (!valid_datatype) {      /* shouldn't happen, but just in case... */

        COREAUDIO_CloseDevice(this);

        SDL_SetError("Unsupported audio format");

        return 0;

    }

    strdesc.mBytesPerFrame =

        strdesc.mBitsPerChannel * strdesc.mChannelsPerFrame / 8;

    strdesc.mBytesPerPacket =

        strdesc.mBytesPerFrame * strdesc.mFramesPerPacket;

    if (!prepare_audiounit(this, devname, iscapture, &strdesc)) {

        COREAUDIO_CloseDevice(this);

        return 0;               /* prepare_audiounit() will call SDL_SetError()... */

    }

    return 1;                   /* good to go. */

}

static int

COREAUDIO_Init(SDL_AudioDriverImpl * impl)

{

    /* Set the function pointers */

    impl->DetectDevices = COREAUDIO_DetectDevices;

    impl->GetDeviceName = COREAUDIO_GetDeviceName;

    impl->OpenDevice = COREAUDIO_OpenDevice;

    impl->CloseDevice = COREAUDIO_CloseDevice;

    impl->Deinitialize = COREAUDIO_Deinitialize;

    impl->ProvidesOwnCallbackThread = 1;

    build_device_lists();       /* do an initial check for devices... */

    return 1;

}

AudioBootStrap COREAUDIO_bootstrap = {

    "coreaudio", "Mac OS X CoreAudio", COREAUDIO_Init, 0

};

/* vi: set ts=4 sw=4 expandtab: */