/*
Copyright (C) 1996-1997 Id Software, Inc.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/
// Quake is a trademark of Id Software, Inc., (c) 1996 Id Software, Inc. All
// rights reserved.

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/ioctl.h>
#include <sys/file.h>
#include <sys/types.h>
#include <fcntl.h>
#include <string.h>
#include <time.h>
#include <errno.h>

#if defined(__linux__)
#include <linux/cdrom.h>
#else
#include <sys/cdio.h>
#define CDROM_DATA_TRACK 4
#define CDROMPLAYTRKIND 0x5304
#endif

#include "quakedef.h"

static qboolean	cdValid = false;
static qboolean	playing = false;
static qboolean	wasPlaying = false;
static qboolean	initialized = false;
static qboolean	enabled = true;
static qboolean	playLooping = false;
static float	cdvolume;
static byte	remap[100];
static byte	playTrack;
static byte	maxTrack;

static int	cdfile = -1;
#if defined(__linux__)
static char cd_dev[64] = "/dev/cdrom";
#else
static char cd_dev[64] = "/dev/acd0";
#endif

void CDAudio_Pause(void);

static void CDAudio_Eject(void)
{
	if (cdfile == -1 || !enabled)
		return;		/* no cd init'd */
#if defined(__linux__)
	if (ioctl(cdfile, CDROMEJECT) == -1)
		Con_DPrintf("ioctl cdromeject failed\n");
#else
	if (ioctl(cdfile, CDIOCEJECT) == -1)
		Con_DPrintf("ioctl cdioeject failed\n");
#endif
}


static void CDAudio_CloseDoor(void)
{
	if (cdfile == -1 || !enabled)
		return;		/* no cd init'd */

#if defined(__linux__)
	if (ioctl(cdfile, CDROMCLOSETRAY) == -1)
		Con_DPrintf("ioctl cdromclosetray failed\n");
#else
	if (ioctl(cdfile, CDIOCCLOSE) == -1)
		Con_DPrintf("ioctl cdiocclose failed\n");
#endif
}

static int CDAudio_GetAudioDiskInfo(void)
{
#if defined(__linux__)
	struct cdrom_tochdr tochdr;

	cdValid = false;
	if (ioctl(cdfile, CDROMREADTOCHDR, &tochdr) == -1) {
		Con_DPrintf("ioctl cdromreadtochdr failed\n");
		return -1;
	}
	if (tochdr.cdth_trk0 < 1) {
		Con_DPrintf("CDAudio: no music tracks\n");
		return -1;
	}
	cdValid = true;
	maxTrack = tochdr.cdth_trk1;
#else
	struct ioc_toc_header tochdr;

	cdValid = false;

	if (ioctl(cdfile, CDIOREADTOCHEADER, &tochdr) == -1) {
		Con_DPrintf("ioctl cdioreadtocheader failed\n");
		return -1;
	}
	if (tochdr.starting_track < 1) {
		Con_DPrintf("CDAudio: no music tracks\n");
		return -1;
	}
	cdValid = true;
	maxTrack = tochdr.ending_track;
#endif

	return 0;
}


void CDAudio_Play(byte track, qboolean looping)
{
#if defined(__linux__)
	struct cdrom_tocentry entry;
	struct cdrom_ti	ti;

	if (cdfile == -1 || !enabled)
		return;

	if (!cdValid) {
		CDAudio_GetAudioDiskInfo();
		if (!cdValid)
			return;
	}
	track = remap[track];

	if (track < 1 || track > maxTrack) {
		Con_DPrintf("CDAudio: Bad track number %u.\n", track);
		return;
	}
	
	// don't try to play a non-audio track
	entry.cdte_track = track;
	entry.cdte_format = CDROM_LBA;
	if (ioctl(cdfile, CDROMREADTOCENTRY, &entry) == -1) {
		Con_DPrintf("ioctl cdromreadtocentry failed\n");
		return;
	}
	if (entry.cdte_ctrl == CDROM_DATA_TRACK) {
		Con_Printf("CDAudio: track %i is not audio\n", track);
		return;
	}
	if (playing) {
		if (playTrack == track)
			return;
		CDAudio_Stop();
	}
	ti.cdti_trk0 = track;
	ti.cdti_trk1 = track;
	ti.cdti_ind0 = 0;
	ti.cdti_ind1 = 0;

	if (ioctl(cdfile, CDROMPLAYTRKIND, &ti) == -1) {
		Con_DPrintf("ioctl cdromplaytrkind failed\n");
		return;
	}
	if (ioctl(cdfile, CDROMRESUME) == -1)
		Con_DPrintf("ioctl cdromresume failed\n");
#else
	struct ioc_read_toc_entry entry;
	struct cd_toc_entry toc_buffer;
	struct ioc_play_track ti;


	if (cdfile == -1 || !enabled)
		return;

	if (!cdValid) {
		CDAudio_GetAudioDiskInfo();
		if (!cdValid)
			return;
	}
	track = remap[track];

	if (track < 1 || track > maxTrack) {
		Con_DPrintf("CDAudio: Bad track number %u.\n", track);
		return;
	}
	
	bzero((char *)&toc_buffer, sizeof(toc_buffer));
	entry.data_len = sizeof(toc_buffer);
	entry.data = &toc_buffer;
	
	// don't try to play a non-audio track
	entry.starting_track = track;
	entry.address_format = CD_MSF_FORMAT;
	if (ioctl(cdfile, CDIOREADTOCENTRYS, &entry) == -1) {
		Con_DPrintf("ioctl cdromreadtocentry failed\n");
		return;
	}
	if (toc_buffer.control == CDROM_DATA_TRACK) {
		Con_Printf("CDAudio: track %i is not audio\n", track);
		return;
	}
	if (playing) {
		if (playTrack == track)
			return;
		CDAudio_Stop();
	}
	
	ti.start_track = track;
	ti.end_track = track;
	ti.start_index = 1;
	ti.end_index = 99;

	if (ioctl(cdfile, CDIOCPLAYTRACKS, &ti) == -1) {
		Con_DPrintf("ioctl cdromplaytrkind failed\n");
		return;
	}
	if (ioctl(cdfile, CDIOCRESUME) == -1)
		Con_DPrintf("ioctl cdromresume failed\n");
#endif
	playLooping = looping;
	playTrack = track;
	playing = true;

	if (cdvolume == 0.0)
		CDAudio_Pause();
}

void CDAudio_RandomPlay(void)
{
	int		track, i = 0, free_tracks = 0, remap_track;
	float		f;
	byte           *track_bools;
#if defined(__linux__)
	struct cdrom_tocentry entry;
	struct cdrom_ti	ti;
#else
	struct ioc_read_toc_entry entry;
	struct cd_toc_entry toc_buffer;
	struct ioc_play_track ti;
#endif
	if (cdfile == -1 || !enabled)
		return;

	track_bools = (byte *) malloc(maxTrack * sizeof(byte));

	if (track_bools == 0)
		return;

	/* create array of available audio tracknumbers */

	for (; i < maxTrack; i++) {
#if defined(__linux__)
		entry.cdte_track = remap[i];
		entry.cdte_format = CDROM_LBA;
		if (ioctl(cdfile, CDROMREADTOCENTRY, &entry) == -1) {
			track_bools[i] = 0;
		} else
			track_bools[i] = (entry.cdte_ctrl != CDROM_DATA_TRACK);
#else
		bzero((char *)&toc_buffer, sizeof(toc_buffer));
		entry.data_len = sizeof(toc_buffer);
		entry.data = &toc_buffer;

		entry.starting_track = remap[i];
		entry.address_format = CD_LBA_FORMAT;
		if (ioctl(cdfile, CDIOREADTOCENTRYS, &entry) == -1) {
			track_bools[i] = 0;
		} else
			track_bools[i] = (entry.data->control != CDROM_DATA_TRACK);
#endif
		free_tracks += track_bools[i];
	}

	if (!free_tracks) {
		Con_DPrintf("CDAudio_RandomPlay: Unable to find and play a random audio track, insert an audio cd please");
		goto free_end;
	}
	/* choose random audio track */
	do {
		do {
			f = ((float)rand()) / ((float)RAND_MAX + 1.0);
			track = (int)(maxTrack * f);
		}
		while (!track_bools[track]);

		remap_track = remap[track];

		if (playing) {
			if (playTrack == remap_track) {
				goto free_end;
			}
			CDAudio_Stop();
		}
#if defined(__linux__)
		ti.cdti_trk0 = remap_track;
		ti.cdti_trk1 = remap_track;
		ti.cdti_ind0 = 0;
		ti.cdti_ind1 = 0;
#else
		ti.start_track = remap_track;
		ti.end_track = remap_track;
		ti.start_index = 0;
		ti.end_index = 0;
#endif
		if (ioctl(cdfile, CDROMPLAYTRKIND, &ti) == -1) {
			track_bools[track] = 0;
			free_tracks--;
		} else {
			playLooping = true;
			playTrack = remap_track;
			playing = true;
			break;
		}
	}
	while (free_tracks > 0);

free_end:
	free((void *)track_bools);
}

void CDAudio_Stop(void)
{
	if (cdfile == -1 || !enabled)
		return;

	if (!playing)
		return;

#if defined(__linux__)
	if (ioctl(cdfile, CDROMSTOP) == -1)
		Con_DPrintf("ioctl cdromstop failed (%d)\n", errno);
#else
	if (ioctl(cdfile, CDIOCSTOP) == -1)
		Con_DPrintf("ioctl cdiocstop failed (%d)\n", errno);
#endif
	wasPlaying = false;
	playing = false;
}

void CDAudio_Pause(void)
{
	if (cdfile == -1 || !enabled)
		return;

	if (!playing)
		return;
                
#if defined(__linux__)
	if (ioctl(cdfile, CDROMPAUSE) == -1)
		Con_DPrintf("ioctl cdrompause failed\n");
#else
	if (ioctl(cdfile, CDIOCPAUSE) == -1)
		Con_DPrintf("ioctl cdiocpause failed\n");
#endif
	wasPlaying = playing;
	playing = false;
}

void CDAudio_Resume(void)
{
	if (cdfile == -1 || !enabled)
		return;

	if (!cdValid)
		return;

	if (!wasPlaying)
		return;

#if defined(__linux__)
	if (ioctl(cdfile, CDROMRESUME) == -1)
		Con_DPrintf("ioctl cdromresume failed\n");
#else
	if (ioctl(cdfile, CDIOCRESUME) == -1)
		Con_DPrintf("ioctl cdiocpause failed\n");
#endif
	playing = true;
}

static void CD_f(void)
{
	char	*command;
	int		ret;
	int		n;

	if (Cmd_Argc() < 2)
		return;

	command = Cmd_Argv(1);

	if (Q_strcasecmp(command, "on") == 0) {
		enabled = true;
		return;
	}
	if (Q_strcasecmp(command, "off") == 0) {
		if (playing)
			CDAudio_Stop();
		enabled = false;
		return;
	}
	if (Q_strcasecmp(command, "reset") == 0) {
		enabled = true;
		if (playing)
			CDAudio_Stop();
		for (n = 0; n < 100; n++)
			remap[n] = n;
		CDAudio_GetAudioDiskInfo();
		return;
	}
	if (Q_strcasecmp(command, "remap") == 0) {
		ret = Cmd_Argc() - 2;
		if (ret <= 0) {
			for (n = 1; n < 100; n++)
				if (remap[n] != n)
					Con_Printf("  %u -> %u\n", n, remap[n]);
			return;
		}
		for (n = 1; n <= ret; n++)
			remap[n] = Q_atoi(Cmd_Argv(n + 1));
		return;
	}
	if (Q_strcasecmp(command, "close") == 0) {
		CDAudio_CloseDoor();
		return;
	}
	if (!cdValid) {
		CDAudio_GetAudioDiskInfo();
		if (!cdValid) {
			Con_Printf("No CD in player.\n");
			return;
		}
	}
	if (Q_strcasecmp(command, "play") == 0) {
		CDAudio_Play((byte) Q_atoi(Cmd_Argv(2)), false);
		return;
	}
	if (Q_strcasecmp(command, "loop") == 0) {
		CDAudio_Play((byte) Q_atoi(Cmd_Argv(2)), true);
		return;
	}
	if (Q_strcasecmp(command, "stop") == 0) {
		CDAudio_Stop();
		return;
	}
	if (Q_strcasecmp(command, "pause") == 0) {
		CDAudio_Pause();
		return;
	}
	if (Q_strcasecmp(command, "resume") == 0) {
		CDAudio_Resume();
		return;
	}
	if (Q_strcasecmp(command, "eject") == 0) {
		if (playing)
			CDAudio_Stop();
		CDAudio_Eject();
		cdValid = false;
		return;
	}
	if (!Q_strcasecmp(command, "random")) {
		CDAudio_RandomPlay();
		return;
	}
	if (Q_strcasecmp(command, "info") == 0) {
		Con_Printf("%u tracks\n", maxTrack);
		if (playing)
			Con_Printf("Currently %s track %u\n",
			           playLooping ? "looping" : "playing", playTrack);
		else if (wasPlaying)
			Con_Printf("Paused %s track %u\n",
			           playLooping ? "looping" : "playing", playTrack);
		Con_Printf("Volume is %f\n", cdvolume);
		return;
	}
}

void CDAudio_Update(void)
{
#if defined(__linux__)
	struct cdrom_subchnl subchnl;
#else
	struct ioc_read_subchannel subchnl;
	struct cd_sub_channel_info data;
#endif
	static time_t	lastchk;

	if (cdfile == -1 || !enabled)
		return;

	if (bgmvolume.value != cdvolume) {
		if (cdvolume) {
			Cvar_SetValue ("bgmvolume", 0.0);
			cdvolume = bgmvolume.value;
			CDAudio_Pause ();
		}
		else {
			Cvar_SetValue ("bgmvolume", 1.0);
			cdvolume = bgmvolume.value;
			CDAudio_Resume ();
		}
	}
	
	if (playing && lastchk < time(NULL)) {
		lastchk = time(NULL) + 2;	/* two seconds between chks */
#if defined(__linux__)
		subchnl.cdsc_format = CDROM_MSF;
		if (ioctl(cdfile, CDROMSUBCHNL, &subchnl) == -1) {
			Con_DPrintf("ioctl cdromsubchnl failed\n");
			playing = false;
			return;
		}
		if (subchnl.cdsc_audiostatus != CDROM_AUDIO_PLAY &&
		    subchnl.cdsc_audiostatus != CDROM_AUDIO_PAUSED) {
			playing = false;
			if (playLooping)
				CDAudio_Play(playTrack, true);
		}
#else
		subchnl.address_format = CD_MSF_FORMAT;
		subchnl.data_format = CD_CURRENT_POSITION;
		subchnl.data_len = sizeof(data);
		subchnl.track = playTrack;
		subchnl.data = &data;
		if (ioctl(cdfile, CDIOCREADSUBCHANNEL, &subchnl) == -1) {
			Con_DPrintf("ioctl cdiocreadsubchannel failed\n");
			playing = false;
			return;
		}
		if (subchnl.data->header.audio_status != CD_AS_PLAY_IN_PROGRESS &&
		    subchnl.data->header.audio_status != CD_AS_PLAY_PAUSED) {
			playing = false;
			if (playLooping)
				CDAudio_Play(playTrack, true);
		}
#endif
	}
}

int CDAudio_Init(void)
{
	int	i;

	if (cls.state == ca_dedicated)
		return -1;

	if (COM_CheckParm("-nocdaudio"))
		return -1;

	if ((i = COM_CheckParm("-cddev")) != 0 && i < com_argc - 1)
		Q_strncpyz (cd_dev, com_argv[i + 1], sizeof(cd_dev));

	if ((cdfile = open(cd_dev, O_RDONLY | O_NONBLOCK | O_EXCL)) == -1) {
		Con_Printf ("CDAudio_Init: open of \"%s\" failed (%i)\n", cd_dev, errno);
		cdfile = -1;
		return -1;
	}
	
	for (i = 0; i < 100; i++)
		remap[i] = i;
	initialized = true;
	enabled = true;

	if (CDAudio_GetAudioDiskInfo()) {
		Con_Printf("CDAudio_Init: No CD in player.\n");
		cdValid = false;
	}
	
	Cmd_AddCommand("cd", CD_f);

	Con_Printf("CD Audio Initialized\n");

	return 0;
}

void CDAudio_Shutdown(void)
{
	if (!initialized)
		return;
	CDAudio_Stop();
	close(cdfile);
	cdfile = -1;
	initialized = false;
}