/*
    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"

#ifdef SDL_CDROM_WIN32

/* Functions for system-level CD-ROM audio control */

#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <mmsystem.h>

#include "SDL_cdrom.h"
#include "../SDL_syscdrom.h"

/* This really broken?? */
#define BROKEN_MCI_PAUSE        /* Pausing actually stops play -- Doh! */

/* The maximum number of CD-ROM drives we'll detect (Don't change!) */
#define MAX_DRIVES	26

/* A list of available CD-ROM drives */
static char *SDL_cdlist[MAX_DRIVES];
static MCIDEVICEID SDL_mciID[MAX_DRIVES];
#ifdef BROKEN_MCI_PAUSE
static int SDL_paused[MAX_DRIVES];
#endif
static int SDL_CD_end_position;

/* The system-dependent CD control functions */
static const char *SDL_SYS_CDName(int drive);
static int SDL_SYS_CDOpen(int drive);
static int SDL_SYS_CDGetTOC(SDL_CD * cdrom);
static CDstatus SDL_SYS_CDStatus(SDL_CD * cdrom, int *position);
static int SDL_SYS_CDPlay(SDL_CD * cdrom, int start, int length);
static int SDL_SYS_CDPause(SDL_CD * cdrom);
static int SDL_SYS_CDResume(SDL_CD * cdrom);
static int SDL_SYS_CDStop(SDL_CD * cdrom);
static int SDL_SYS_CDEject(SDL_CD * cdrom);
static void SDL_SYS_CDClose(SDL_CD * cdrom);


/* Add a CD-ROM drive to our list of valid drives */
static void
AddDrive(char *drive)
{
    int i;

    if (SDL_numcds < MAX_DRIVES) {
        /* Add this drive to our list */
        i = SDL_numcds;
        SDL_cdlist[i] = SDL_strdup(drive);
        if (SDL_cdlist[i] == NULL) {
            SDL_OutOfMemory();
            return;
        }
        ++SDL_numcds;
#ifdef CDROM_DEBUG
        fprintf(stderr, "Added CD-ROM drive: %s\n", drive);
#endif
    }
}

int
SDL_SYS_CDInit(void)
{
    /* checklist: Drive 'A' - 'Z' */
    int i;
    char drive[4];

    /* Fill in our driver capabilities */
    SDL_CDcaps.Name = SDL_SYS_CDName;
    SDL_CDcaps.Open = SDL_SYS_CDOpen;
    SDL_CDcaps.GetTOC = SDL_SYS_CDGetTOC;
    SDL_CDcaps.Status = SDL_SYS_CDStatus;
    SDL_CDcaps.Play = SDL_SYS_CDPlay;
    SDL_CDcaps.Pause = SDL_SYS_CDPause;
    SDL_CDcaps.Resume = SDL_SYS_CDResume;
    SDL_CDcaps.Stop = SDL_SYS_CDStop;
    SDL_CDcaps.Eject = SDL_SYS_CDEject;
    SDL_CDcaps.Close = SDL_SYS_CDClose;

    /* Scan the system for CD-ROM drives */
    for (i = 'A'; i <= 'Z'; ++i) {
        SDL_snprintf(drive, SDL_arraysize(drive), "%c:\\", i);
        if (GetDriveType(drive) == DRIVE_CDROM) {
            AddDrive(drive);
        }
    }
    SDL_memset(SDL_mciID, 0, sizeof(SDL_mciID));
    return (0);
}

/* General ioctl() CD-ROM command function */
static int
SDL_SYS_CDioctl(int id, UINT msg, DWORD flags, void *arg)
{
    MCIERROR mci_error;

    mci_error = mciSendCommand(SDL_mciID[id], msg, flags, (DWORD_PTR) arg);
    if (mci_error) {
        char error[256];

        mciGetErrorString(mci_error, error, 256);
        SDL_SetError("mciSendCommand() error: %s", error);
    }
    return (!mci_error ? 0 : -1);
}

static const char *
SDL_SYS_CDName(int drive)
{
    return (SDL_cdlist[drive]);
}

static int
SDL_SYS_CDOpen(int drive)
{
    MCI_OPEN_PARMS mci_open;
    MCI_SET_PARMS mci_set;
    char device[3];
    DWORD flags;

    /* Open the requested device */
    mci_open.lpstrDeviceType = (LPCSTR) MCI_DEVTYPE_CD_AUDIO;
    device[0] = *SDL_cdlist[drive];
    device[1] = ':';
    device[2] = '\0';
    mci_open.lpstrElementName = device;
    flags =
        (MCI_OPEN_TYPE | MCI_OPEN_SHAREABLE | MCI_OPEN_TYPE_ID |
         MCI_OPEN_ELEMENT);
    if (SDL_SYS_CDioctl(0, MCI_OPEN, flags, &mci_open) < 0) {
        flags &= ~MCI_OPEN_SHAREABLE;
        if (SDL_SYS_CDioctl(0, MCI_OPEN, flags, &mci_open) < 0) {
            return (-1);
        }
    }
    SDL_mciID[drive] = mci_open.wDeviceID;

    /* Set the minute-second-frame time format */
    mci_set.dwTimeFormat = MCI_FORMAT_MSF;
    SDL_SYS_CDioctl(drive, MCI_SET, MCI_SET_TIME_FORMAT, &mci_set);

#ifdef BROKEN_MCI_PAUSE
    SDL_paused[drive] = 0;
#endif
    return (drive);
}

static int
SDL_SYS_CDGetTOC(SDL_CD * cdrom)
{
    MCI_STATUS_PARMS mci_status;
    int i, okay;
    DWORD flags;

    okay = 0;
    mci_status.dwItem = MCI_STATUS_NUMBER_OF_TRACKS;
    flags = MCI_STATUS_ITEM | MCI_WAIT;
    if (SDL_SYS_CDioctl(cdrom->id, MCI_STATUS, flags, &mci_status) == 0) {
        cdrom->numtracks = mci_status.dwReturn;
        if (cdrom->numtracks > SDL_MAX_TRACKS) {
            cdrom->numtracks = SDL_MAX_TRACKS;
        }
        /* Read all the track TOC entries */
        flags = MCI_STATUS_ITEM | MCI_TRACK | MCI_WAIT;
        for (i = 0; i < cdrom->numtracks; ++i) {
            cdrom->track[i].id = i + 1;
            mci_status.dwTrack = cdrom->track[i].id;
#ifdef MCI_CDA_STATUS_TYPE_TRACK
            mci_status.dwItem = MCI_CDA_STATUS_TYPE_TRACK;
            if (SDL_SYS_CDioctl(cdrom->id, MCI_STATUS, flags,
                                &mci_status) < 0) {
                break;
            }
            if (mci_status.dwReturn == MCI_CDA_TRACK_AUDIO) {
                cdrom->track[i].type = SDL_AUDIO_TRACK;
            } else {
                cdrom->track[i].type = SDL_DATA_TRACK;
            }
#else
            cdrom->track[i].type = SDL_AUDIO_TRACK;
#endif
            mci_status.dwItem = MCI_STATUS_POSITION;
            if (SDL_SYS_CDioctl(cdrom->id, MCI_STATUS, flags,
                                &mci_status) < 0) {
                break;
            }
            cdrom->track[i].offset =
                MSF_TO_FRAMES(MCI_MSF_MINUTE(mci_status.dwReturn),
                              MCI_MSF_SECOND(mci_status.dwReturn),
                              MCI_MSF_FRAME(mci_status.dwReturn));
            cdrom->track[i].length = 0;
            if (i > 0) {
                cdrom->track[i - 1].length =
                    cdrom->track[i].offset - cdrom->track[i - 1].offset;
            }
        }
        if (i == cdrom->numtracks) {
            mci_status.dwTrack = cdrom->track[i - 1].id;
            mci_status.dwItem = MCI_STATUS_LENGTH;
            if (SDL_SYS_CDioctl(cdrom->id, MCI_STATUS, flags,
                                &mci_status) == 0) {
                cdrom->track[i - 1].length =
                    MSF_TO_FRAMES(MCI_MSF_MINUTE(mci_status.dwReturn),
                                  MCI_MSF_SECOND(mci_status.dwReturn),
                                  MCI_MSF_FRAME(mci_status.dwReturn));
                /* compute lead-out offset */
                cdrom->track[i].offset = cdrom->track[i - 1].offset +
                    cdrom->track[i - 1].length;
                cdrom->track[i].length = 0;
                okay = 1;
            }
        }
    }
    return (okay ? 0 : -1);
}

/* Get CD-ROM status */
static CDstatus
SDL_SYS_CDStatus(SDL_CD * cdrom, int *position)
{
    CDstatus status;
    MCI_STATUS_PARMS mci_status;
    DWORD flags;

    flags = MCI_STATUS_ITEM | MCI_WAIT;
    mci_status.dwItem = MCI_STATUS_MODE;
    if (SDL_SYS_CDioctl(cdrom->id, MCI_STATUS, flags, &mci_status) < 0) {
        status = CD_ERROR;
    } else {
        switch (mci_status.dwReturn) {
        case MCI_MODE_NOT_READY:
        case MCI_MODE_OPEN:
            status = CD_TRAYEMPTY;
            break;
        case MCI_MODE_STOP:
#ifdef BROKEN_MCI_PAUSE
            if (SDL_paused[cdrom->id]) {
                status = CD_PAUSED;
            } else {
                status = CD_STOPPED;
            }
#else
            status = CD_STOPPED;
#endif /* BROKEN_MCI_PAUSE */
            break;
        case MCI_MODE_PLAY:
#ifdef BROKEN_MCI_PAUSE
            if (SDL_paused[cdrom->id]) {
                status = CD_PAUSED;
            } else {
                status = CD_PLAYING;
            }
#else
            status = CD_PLAYING;
#endif /* BROKEN_MCI_PAUSE */
            break;
        case MCI_MODE_PAUSE:
            status = CD_PAUSED;
            break;
        default:
            status = CD_ERROR;
            break;
        }
    }
    if (position) {
        if (status == CD_PLAYING || (status == CD_PAUSED)) {
            mci_status.dwItem = MCI_STATUS_POSITION;
            if (SDL_SYS_CDioctl(cdrom->id, MCI_STATUS, flags,
                                &mci_status) == 0) {
                *position =
                    MSF_TO_FRAMES(MCI_MSF_MINUTE(mci_status.dwReturn),
                                  MCI_MSF_SECOND(mci_status.dwReturn),
                                  MCI_MSF_FRAME(mci_status.dwReturn));
            } else {
                *position = 0;
            }
        } else {
            *position = 0;
        }
    }
    return (status);
}

/* Start play */
static int
SDL_SYS_CDPlay(SDL_CD * cdrom, int start, int length)
{
    MCI_PLAY_PARMS mci_play;
    int m, s, f;
    DWORD flags;

    flags = MCI_FROM | MCI_TO | MCI_NOTIFY;
    mci_play.dwCallback = 0;
    FRAMES_TO_MSF(start, &m, &s, &f);
    mci_play.dwFrom = MCI_MAKE_MSF(m, s, f);
    FRAMES_TO_MSF(start + length, &m, &s, &f);
    mci_play.dwTo = MCI_MAKE_MSF(m, s, f);
    SDL_CD_end_position = mci_play.dwTo;
    return (SDL_SYS_CDioctl(cdrom->id, MCI_PLAY, flags, &mci_play));
}

/* Pause play */
static int
SDL_SYS_CDPause(SDL_CD * cdrom)
{
#ifdef BROKEN_MCI_PAUSE
    SDL_paused[cdrom->id] = 1;
#endif
    return (SDL_SYS_CDioctl(cdrom->id, MCI_PAUSE, MCI_WAIT, NULL));
}

/* Resume play */
static int
SDL_SYS_CDResume(SDL_CD * cdrom)
{
#ifdef BROKEN_MCI_PAUSE
    MCI_STATUS_PARMS mci_status;
    int okay;
    int flags;

    okay = 0;
    /* Play from the current play position to the end position set earlier */
    flags = MCI_STATUS_ITEM | MCI_WAIT;
    mci_status.dwItem = MCI_STATUS_POSITION;
    if (SDL_SYS_CDioctl(cdrom->id, MCI_STATUS, flags, &mci_status) == 0) {
        MCI_PLAY_PARMS mci_play;

        flags = MCI_FROM | MCI_TO | MCI_NOTIFY;
        mci_play.dwCallback = 0;
        mci_play.dwFrom = mci_status.dwReturn;
        mci_play.dwTo = SDL_CD_end_position;
        if (SDL_SYS_CDioctl(cdrom->id, MCI_PLAY, flags, &mci_play) == 0) {
            okay = 1;
            SDL_paused[cdrom->id] = 0;
        }
    }
    return (okay ? 0 : -1);
#else
    return (SDL_SYS_CDioctl(cdrom->id, MCI_RESUME, MCI_WAIT, NULL));
#endif /* BROKEN_MCI_PAUSE */
}

/* Stop play */
static int
SDL_SYS_CDStop(SDL_CD * cdrom)
{
    return (SDL_SYS_CDioctl(cdrom->id, MCI_STOP, MCI_WAIT, NULL));
}

/* Eject the CD-ROM */
static int
SDL_SYS_CDEject(SDL_CD * cdrom)
{
    return (SDL_SYS_CDioctl(cdrom->id, MCI_SET, MCI_SET_DOOR_OPEN, NULL));
}

/* Close the CD-ROM handle */
static void
SDL_SYS_CDClose(SDL_CD * cdrom)
{
    SDL_SYS_CDioctl(cdrom->id, MCI_CLOSE, MCI_WAIT, NULL);
}

void
SDL_SYS_CDQuit(void)
{
    int i;

    if (SDL_numcds > 0) {
        for (i = 0; i < SDL_numcds; ++i) {
            SDL_free(SDL_cdlist[i]);
        }
        SDL_numcds = 0;
    }
}

#endif /* SDL_CDROM_WIN32 */
/* vi: set ts=4 sw=4 expandtab: */