/*

  Simple DirectMedia Layer

  Copyright (C) 1997-2012 Sam Lantinga <slouken@libsdl.org>

  This software is provided 'as-is', without any express or implied

  warranty.  In no event will the authors be held liable for any damages

  arising from the use of this software.

  Permission is granted to anyone to use this software for any purpose,

  including commercial applications, and to alter it and redistribute it

  freely, subject to the following restrictions:

  1. The origin of this software must not be misrepresented; you must not

     claim that you wrote the original software. If you use this software

     in a product, an acknowledgment in the product documentation would be

     appreciated but is not required.

  2. Altered source versions must be plainly marked as such, and must not be

     misrepresented as being the original software.

  3. This notice may not be removed or altered from any source distribution.

*/

#include "SDL_config.h"

#if SDL_AUDIO_DRIVER_WINMM

/* Allow access to a raw mixing buffer */

#include "../../core/windows/SDL_windows.h"

#include <mmsystem.h>

#include "SDL_timer.h"

#include "SDL_audio.h"

#include "../SDL_audio_c.h"

#include "SDL_winmm.h"

#define DETECT_DEV_IMPL(typ, capstyp) \

static void DetectWave##typ##Devs(SDL_AddAudioDevice addfn) { \

    const UINT devcount = wave##typ##GetNumDevs(); \

    capstyp caps; \

    UINT i; \

    for (i = 0; i < devcount; i++) { \

        if (wave##typ##GetDevCaps(i,&caps,sizeof(caps))==MMSYSERR_NOERROR) { \

            char *name = WIN_StringToUTF8(caps.szPname); \

            if (name != NULL) { \

                addfn(name); \

                SDL_free(name); \

            } \

        } \

    } \

}

DETECT_DEV_IMPL(Out, WAVEOUTCAPS)

DETECT_DEV_IMPL(In, WAVEINCAPS)

static void

WINMM_DetectDevices(int iscapture, SDL_AddAudioDevice addfn)

{

    if (iscapture) {

        DetectWaveInDevs(addfn);

    } else {

        DetectWaveOutDevs(addfn);

    }

}

static void CALLBACK

CaptureSound(HWAVEIN hwi, UINT uMsg, DWORD_PTR dwInstance,

          DWORD_PTR dwParam1, DWORD_PTR dwParam2)

{

    SDL_AudioDevice *this = (SDL_AudioDevice *) dwInstance;

    /* Only service "buffer is filled" messages */

    if (uMsg != WIM_DATA)

        return;

    /* Signal that we have a new buffer of data */

    ReleaseSemaphore(this->hidden->audio_sem, 1, NULL);

}

/* The Win32 callback for filling the WAVE device */

static void CALLBACK

FillSound(HWAVEOUT hwo, UINT uMsg, DWORD_PTR dwInstance,

          DWORD_PTR dwParam1, DWORD_PTR dwParam2)

{

    SDL_AudioDevice *this = (SDL_AudioDevice *) dwInstance;

    /* Only service "buffer done playing" messages */

    if (uMsg != WOM_DONE)

        return;

    /* Signal that we are done playing a buffer */

    ReleaseSemaphore(this->hidden->audio_sem, 1, NULL);

}

static void

SetMMerror(char *function, MMRESULT code)

{

    size_t len;

    char errbuf[MAXERRORLENGTH];

    wchar_t werrbuf[MAXERRORLENGTH];

    SDL_snprintf(errbuf, SDL_arraysize(errbuf), "%s: ", function);

    len = SDL_strlen(errbuf);

    waveOutGetErrorText(code, werrbuf, MAXERRORLENGTH - len);

    WideCharToMultiByte(CP_ACP, 0, werrbuf, -1, errbuf + len,

                        MAXERRORLENGTH - len, NULL, NULL);

    SDL_SetError("%s", errbuf);

}

static void

WINMM_WaitDevice(_THIS)

{

    /* Wait for an audio chunk to finish */

    WaitForSingleObject(this->hidden->audio_sem, INFINITE);

}

static Uint8 *

WINMM_GetDeviceBuf(_THIS)

{

    return (Uint8 *) (this->hidden->

                      wavebuf[this->hidden->next_buffer].lpData);

}

static void

WINMM_PlayDevice(_THIS)

{

    /* Queue it up */

    waveOutWrite(this->hidden->hout,

                 &this->hidden->wavebuf[this->hidden->next_buffer],

                 sizeof(this->hidden->wavebuf[0]));

    this->hidden->next_buffer = (this->hidden->next_buffer + 1) % NUM_BUFFERS;

}

static void

WINMM_WaitDone(_THIS)

{

    int i, left;

    do {

        left = NUM_BUFFERS;

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

            if (this->hidden->wavebuf[i].dwFlags & WHDR_DONE) {

                --left;

            }

        }

        if (left > 0) {

            SDL_Delay(100);

        }

    } while (left > 0);

}

static void

WINMM_CloseDevice(_THIS)

{

    /* Close up audio */

    if (this->hidden != NULL) {

        int i;

        if (this->hidden->audio_sem) {

            CloseHandle(this->hidden->audio_sem);

            this->hidden->audio_sem = 0;

        }

        if (this->hidden->hin) {

            waveInClose(this->hidden->hin);

            this->hidden->hin = 0;

        }

        if (this->hidden->hout) {

            waveOutClose(this->hidden->hout);

            this->hidden->hout = 0;

        }

        /* Clean up mixing buffers */

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

            if (this->hidden->wavebuf[i].dwUser != 0xFFFF) {

                waveOutUnprepareHeader(this->hidden->hout,

                                       &this->hidden->wavebuf[i],

                                       sizeof(this->hidden->wavebuf[i]));

                this->hidden->wavebuf[i].dwUser = 0xFFFF;

            }

        }

        if (this->hidden->mixbuf != NULL) {

            /* Free raw mixing buffer */

            SDL_free(this->hidden->mixbuf);

            this->hidden->mixbuf = NULL;

        }

        SDL_free(this->hidden);

        this->hidden = NULL;

    }

}

static int

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

{

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

    int valid_datatype = 0;

    MMRESULT result;

    WAVEFORMATEX waveformat;

    UINT_PTR devId = WAVE_MAPPER;  /* WAVE_MAPPER == choose system's default */

    char *utf8 = NULL;

    int i;

    if (devname != NULL) {  /* specific device requested? */

        if (iscapture) {

            const int devcount = (int) waveInGetNumDevs();

            WAVEINCAPS caps;

            for (i = 0; (i < devcount) && (devId == WAVE_MAPPER); i++) {

                result = waveInGetDevCaps(i, &caps, sizeof (caps));

                if (result != MMSYSERR_NOERROR)

                    continue;

                else if ((utf8 = WIN_StringToUTF8(caps.szPname)) == NULL)

                    continue;

                else if (SDL_strcmp(devname, utf8) == 0)

                    devId = (UINT_PTR) i;

                SDL_free(utf8);

            }

        } else {

            const int devcount = (int) waveOutGetNumDevs();

            WAVEOUTCAPS caps;

            for (i = 0; (i < devcount) && (devId == WAVE_MAPPER); i++) {

                result = waveOutGetDevCaps(i, &caps, sizeof (caps));

                if (result != MMSYSERR_NOERROR)

                    continue;

                else if ((utf8 = WIN_StringToUTF8(caps.szPname)) == NULL)

                    continue;

                else if (SDL_strcmp(devname, utf8) == 0)

                    devId = (UINT_PTR) i;

                SDL_free(utf8);

            }

        }

        if (devId == WAVE_MAPPER) {

            SDL_SetError("Requested device not found");

            return 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));

    /* Initialize the wavebuf structures for closing */

    for (i = 0; i < NUM_BUFFERS; ++i)

        this->hidden->wavebuf[i].dwUser = 0xFFFF;

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

        valid_datatype = 1;

        this->spec.format = test_format;

        switch (test_format) {

        case AUDIO_U8:

        case AUDIO_S16:

        case AUDIO_S32:

            break;              /* valid. */

        default:

            valid_datatype = 0;

            test_format = SDL_NextAudioFormat();

            break;

        }

    }

    if (!valid_datatype) {

        WINMM_CloseDevice(this);

        SDL_SetError("Unsupported audio format");

        return 0;

    }

    /* Set basic WAVE format parameters */

    SDL_memset(&waveformat, '\0', sizeof(waveformat));

    waveformat.wFormatTag = WAVE_FORMAT_PCM;

    waveformat.wBitsPerSample = SDL_AUDIO_BITSIZE(this->spec.format);

    if (this->spec.channels > 2)

        this->spec.channels = 2;        /* !!! FIXME: is this right? */

    waveformat.nChannels = this->spec.channels;

    waveformat.nSamplesPerSec = this->spec.freq;

    waveformat.nBlockAlign =

        waveformat.nChannels * (waveformat.wBitsPerSample / 8);

    waveformat.nAvgBytesPerSec =

        waveformat.nSamplesPerSec * waveformat.nBlockAlign;

    /* Check the buffer size -- minimum of 1/4 second (word aligned) */

    if (this->spec.samples < (this->spec.freq / 4))

        this->spec.samples = ((this->spec.freq / 4) + 3) & ~3;

    /* Update the fragment size as size in bytes */

    SDL_CalculateAudioSpec(&this->spec);

    /* Open the audio device */

    if (iscapture) {

        result = waveInOpen(&this->hidden->hin, devId, &waveformat,

                             (DWORD_PTR) CaptureSound, (DWORD_PTR) this,

                             CALLBACK_FUNCTION);

    } else {

        result = waveOutOpen(&this->hidden->hout, devId, &waveformat,

                             (DWORD_PTR) FillSound, (DWORD_PTR) this,

                             CALLBACK_FUNCTION);

    }

    if (result != MMSYSERR_NOERROR) {

        WINMM_CloseDevice(this);

        SetMMerror("waveOutOpen()", result);

        return 0;

    }

#ifdef SOUND_DEBUG

    /* Check the sound device we retrieved */

    {

        WAVEOUTCAPS caps;

        result = waveOutGetDevCaps((UINT) this->hidden->hout,

                                   &caps, sizeof(caps));

        if (result != MMSYSERR_NOERROR) {

            WINMM_CloseDevice(this);

            SetMMerror("waveOutGetDevCaps()", result);

            return 0;

        }

        printf("Audio device: %s\n", caps.szPname);

    }

#endif

    /* Create the audio buffer semaphore */

    this->hidden->audio_sem =

        CreateSemaphore(NULL, NUM_BUFFERS - 1, NUM_BUFFERS, NULL);

    if (this->hidden->audio_sem == NULL) {

        WINMM_CloseDevice(this);

        SDL_SetError("Couldn't create semaphore");

        return 0;

    }

    /* Create the sound buffers */

    this->hidden->mixbuf =

        (Uint8 *) SDL_malloc(NUM_BUFFERS * this->spec.size);

    if (this->hidden->mixbuf == NULL) {

        WINMM_CloseDevice(this);

        SDL_OutOfMemory();

        return 0;

    }

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

        SDL_memset(&this->hidden->wavebuf[i], 0,

                   sizeof(this->hidden->wavebuf[i]));

        this->hidden->wavebuf[i].dwBufferLength = this->spec.size;

        this->hidden->wavebuf[i].dwFlags = WHDR_DONE;

        this->hidden->wavebuf[i].lpData =

            (LPSTR) & this->hidden->mixbuf[i * this->spec.size];

        result = waveOutPrepareHeader(this->hidden->hout,

                                      &this->hidden->wavebuf[i],

                                      sizeof(this->hidden->wavebuf[i]));

        if (result != MMSYSERR_NOERROR) {

            WINMM_CloseDevice(this);

            SetMMerror("waveOutPrepareHeader()", result);

            return 0;

        }

    }

    return 1;                   /* Ready to go! */

}

static int

WINMM_Init(SDL_AudioDriverImpl * impl)

{

    /* Set the function pointers */

    impl->DetectDevices = WINMM_DetectDevices;

    impl->OpenDevice = WINMM_OpenDevice;

    impl->PlayDevice = WINMM_PlayDevice;

    impl->WaitDevice = WINMM_WaitDevice;

    impl->WaitDone = WINMM_WaitDone;

    impl->GetDeviceBuf = WINMM_GetDeviceBuf;

    impl->CloseDevice = WINMM_CloseDevice;

    return 1;   /* this audio target is available. */

}

AudioBootStrap WINMM_bootstrap = {

    "winmm", "Windows Waveform Audio", WINMM_Init, 0

};

#endif /* SDL_AUDIO_DRIVER_WINMM */

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