/*
 * 86Box	A hypervisor and IBM PC system emulator that specializes in
 *		running old operating systems and software designed for IBM
 *		PC systems and compatibles from 1981 through fairly recent
 *		system designs based on the PCI bus.
 *
 *		This file is part of the 86Box distribution.
 *
 *		Sound emulation core.
 *
 * Version:	@(#)sound.c	1.0.2	2017/06/14
 *
 * Authors:	Sarah Walker, <http://pcem-emulator.co.uk/>
 *		Miran Grca, <mgrca8@gmail.com>
 *		Copyright 2008-2017 Sarah Walker.
 *		Copyright 2016-2017 Miran Grca.
 */
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include "../ibm.h"
#include "../device.h"
#include "../timer.h"
#include "../cdrom.h"
#include "../win/plat_thread.h"
#include "sound.h"
#include "snd_opl.h"
#include "snd_adlib.h"
#include "snd_adlibgold.h"
#if 0
#include "snd_pas16.h"
#endif
#include "snd_sb.h"
#include "snd_sb_dsp.h"
#include "snd_wss.h"
#include "filters.h"


int sound_card_current = 0;
static int sound_card_last = 0;


typedef struct
{
        char name[64];
        char internal_name[24];
        device_t *device;
} SOUND_CARD;

static SOUND_CARD sound_cards[] =
{
    { "None",                  "none",		NULL			},
    { "Adlib",                 "adlib",		&adlib_device		},
    { "Adlib MCA",             "adlib_mca",	&adlib_mca_device	},
    { "Sound Blaster 1.0",     "sb",		&sb_1_device		},
    { "Sound Blaster 1.5",     "sb1.5",		&sb_15_device		},
    { "Sound Blaster MCV",     "sbmcv",		&sb_mcv_device		},
    { "Sound Blaster 2.0",     "sb2.0",		&sb_2_device		},
    { "Sound Blaster Pro v1",  "sbprov1",	&sb_pro_v1_device	},
    { "Sound Blaster Pro v2",  "sbprov2",	&sb_pro_v2_device	},
    { "Sound Blaster Pro MCV", "sbpromcv",	&sb_pro_mcv_device	},
    { "Sound Blaster 16",      "sb16",		&sb_16_device		},
    { "Sound Blaster AWE32",   "sbawe32",	&sb_awe32_device	},
    { "Adlib Gold",            "adlibgold",	&adgold_device		},
    { "Windows Sound System",  "wss",		&wss_device		},
#if 0
    { "Pro Audio Spectrum 16", "pas16",		&pas16_device		},
#endif
    { "",			"",		NULL			}
};


int sound_card_available(int card)
{
        if (sound_cards[card].device)
                return device_available(sound_cards[card].device);

        return 1;
}

char *sound_card_getname(int card)
{
        return sound_cards[card].name;
}

device_t *sound_card_getdevice(int card)
{
        return sound_cards[card].device;
}

int sound_card_has_config(int card)
{
        if (!sound_cards[card].device)
                return 0;
        return sound_cards[card].device->config ? 1 : 0;
}

char *sound_card_get_internal_name(int card)
{
        return sound_cards[card].internal_name;
}

int sound_card_get_from_internal_name(char *s)
{
	int c = 0;
	
	while (strlen(sound_cards[c].internal_name))
	{
		if (!strcmp(sound_cards[c].internal_name, s))
			return c;
		c++;
	}
	
	return 0;
}

void sound_card_init(void)
{
        if (sound_cards[sound_card_current].device)
                device_add(sound_cards[sound_card_current].device);
        sound_card_last = sound_card_current;
}

static struct
{
        void (*get_buffer)(int32_t *buffer, int len, void *p);
        void *priv;
} sound_handlers[8];

static int sound_handlers_num;

static int sound_poll_time = 0, sound_poll_latch;
int sound_pos_global = 0;

int soundon = 1;

static int16_t cd_buffer[CDROM_NUM][CD_BUFLEN * 2];
static float cd_out_buffer[CD_BUFLEN * 2];
static int16_t cd_out_buffer_int16[CD_BUFLEN * 2];
static thread_t *sound_cd_thread_h;
static event_t *sound_cd_event;
static unsigned int cd_vol_l, cd_vol_r;
static int cd_buf_update = CD_BUFLEN / SOUNDBUFLEN;

int sound_is_float = 1;

void sound_set_cd_volume(unsigned int vol_l, unsigned int vol_r)
{
        cd_vol_l = vol_l;
        cd_vol_r = vol_r;
}

static void sound_cd_thread(void *param)
{
	int i = 0;

	float cd_buffer_temp[2] = {0.0, 0.0};
	float cd_buffer_temp2[2] = {0.0, 0.0};

	int c, has_audio;

	while (1)
	{
		thread_wait_event(sound_cd_event, -1);
		if (!soundon)
		{
			return;
		}
		for (c = 0; c < CD_BUFLEN*2; c += 2)
		{
			if (sound_is_float)
			{
				cd_out_buffer[c] = 0.0;
				cd_out_buffer[c+1] = 0.0;
			}
			else
			{
				cd_out_buffer_int16[c] = 0;
				cd_out_buffer_int16[c+1] = 0;
			}
		}
		for (i = 0; i < CDROM_NUM; i++)
		{
			has_audio = 0;
			if (cdrom_drives[i].handler->audio_callback)
			{
				cdrom_drives[i].handler->audio_callback(i, cd_buffer[i], CD_BUFLEN*2);
				has_audio = (cdrom_drives[i].bus_type && cdrom_drives[i].sound_on);
			}
			if (soundon && has_audio)
			{
				int32_t audio_vol_l = cdrom_mode_sense_get_volume(i, 0);
				int32_t audio_vol_r = cdrom_mode_sense_get_volume(i, 1);
				int channel_select[2];

				channel_select[0] = cdrom_mode_sense_get_channel(i, 0);
				channel_select[1] = cdrom_mode_sense_get_channel(i, 1);

				for (c = 0; c < CD_BUFLEN*2; c += 2)
				{
					/* First, transfer the CD audio data to the temporary buffer. */
					cd_buffer_temp[0] = (float) cd_buffer[i][c];
					cd_buffer_temp[1] = (float) cd_buffer[i][c+1];
					
					/* Then, adjust input from drive according to ATAPI/SCSI volume. */
					cd_buffer_temp[0] *= (float) audio_vol_l;
					cd_buffer_temp[0] /= 511.0;
					cd_buffer_temp[1] *= (float) audio_vol_r;
					cd_buffer_temp[1] /= 511.0;

					/*Apply ATAPI channel select*/
					cd_buffer_temp2[0] = cd_buffer_temp2[1] = 0.0;
					if (channel_select[0] & 1)
					{
						cd_buffer_temp2[0] += cd_buffer_temp[0];
					}
					if (channel_select[0] & 2)
					{
						cd_buffer_temp2[1] += cd_buffer_temp[0];
					}
					if (channel_select[1] & 1)
					{
						cd_buffer_temp2[0] += cd_buffer_temp[1];
					}
					if (channel_select[1] & 2)
					{
						cd_buffer_temp2[1] += cd_buffer_temp[1];
					}

					/*Apply sound card CD volume*/
					cd_buffer_temp2[0] *= (float) cd_vol_l;
					cd_buffer_temp2[0] /= 65535.0;

					cd_buffer_temp2[1] *= (float) cd_vol_r;
					cd_buffer_temp2[1] /= 65535.0;

					if (sound_is_float)
					{
						cd_out_buffer[c] += (cd_buffer_temp2[0] / 32768.0);
						cd_out_buffer[c+1] += (cd_buffer_temp2[1] / 32768.0);
					}
					else
					{
						if (cd_buffer_temp2[0] > 32767)
							cd_buffer_temp2[0] = 32767;
						if (cd_buffer_temp2[0] < -32768)
							cd_buffer_temp2[0] = -32768;
						if (cd_buffer_temp2[1] > 32767)
							cd_buffer_temp2[1] = 32767;
						if (cd_buffer_temp2[1] < -32768)
							cd_buffer_temp2[1] = -32768;

						cd_out_buffer_int16[c] += cd_buffer_temp2[0];
						cd_out_buffer_int16[c+1] += cd_buffer_temp2[1];
					}
				}
			}
		}
		if (sound_is_float)
		{
			givealbuffer_cd(cd_out_buffer);
		}
		else
		{
			givealbuffer_cd_int16(cd_out_buffer_int16);
		}
	}
}

static int32_t *outbuffer;
static float *outbuffer_ex;
static int16_t *outbuffer_ex_int16;

static int cd_thread_enable = 0;

void sound_realloc_buffers(void)
{
	closeal();
        initalmain(0,NULL);
        inital();

	if (outbuffer_ex != NULL)
	{
		free(outbuffer_ex);
	}

	if (outbuffer_ex_int16 != NULL)
	{
		free(outbuffer_ex_int16);
	}

	if (sound_is_float)
	{
	        outbuffer_ex = malloc(SOUNDBUFLEN * 2 * sizeof(float));
	}
	else
	{
	        outbuffer_ex_int16 = malloc(SOUNDBUFLEN * 2 * sizeof(int16_t));
	}
}

void sound_init(void)
{
	int i = 0;
	int available_cdrom_drives = 0;

        initalmain(0,NULL);
        inital();

	outbuffer_ex = NULL;
	outbuffer_ex_int16 = NULL;

        outbuffer = malloc(SOUNDBUFLEN * 2 * sizeof(int32_t));

	sound_realloc_buffers();

	for (i = 0; i < CDROM_NUM; i++)
	{
		if (cdrom_drives[i].bus_type != CDROM_BUS_DISABLED)
		{
			available_cdrom_drives++;
		}
	}        

	if (available_cdrom_drives)
	{
		sound_cd_event = thread_create_event();
		sound_cd_thread_h = thread_create(sound_cd_thread, NULL);
	}

	cd_thread_enable = available_cdrom_drives ? 1 : 0;
}

void sound_add_handler(void (*get_buffer)(int32_t *buffer, int len, void *p), void *p)
{
        sound_handlers[sound_handlers_num].get_buffer = get_buffer;
        sound_handlers[sound_handlers_num].priv = p;
        sound_handlers_num++;
}

void sound_poll(void *priv)
{
        sound_poll_time += sound_poll_latch;
        
        sound_pos_global++;
        if (sound_pos_global == SOUNDBUFLEN)
        {
                int c;

                memset(outbuffer, 0, SOUNDBUFLEN * 2 * sizeof(int32_t));

                for (c = 0; c < sound_handlers_num; c++)
                        sound_handlers[c].get_buffer(outbuffer, SOUNDBUFLEN, sound_handlers[c].priv);


		for (c = 0; c < SOUNDBUFLEN * 2; c++)
		{
			if (sound_is_float)
			{
				outbuffer_ex[c] = ((float) outbuffer[c]) / 32768.0;
			}
			else
			{
				if (outbuffer[c] > 32767)
					outbuffer[c] = 32767;
				if (outbuffer[c] < -32768)
					outbuffer[c] = -32768;

				outbuffer_ex_int16[c] = outbuffer[c];
			}
		}
        
                if (soundon)
		{
			if (sound_is_float)
			{
				givealbuffer(outbuffer_ex);
			}
			else
			{
				givealbuffer_int16(outbuffer_ex_int16);
			}
		}
        
		if (cd_thread_enable)
		{
	                cd_buf_update--;
        	        if (!cd_buf_update)
	                {
        	                cd_buf_update = (48000 / SOUNDBUFLEN) / (CD_FREQ / CD_BUFLEN);
                	        thread_set_event(sound_cd_event);
	                }
		}

                sound_pos_global = 0;
        }
}

void sound_speed_changed(void)
{
        sound_poll_latch = (int)((double)TIMER_USEC * (1000000.0 / 48000.0));
}

void sound_reset(void)
{
	int i = 0;

        timer_add(sound_poll, &sound_poll_time, TIMER_ALWAYS_ENABLED, NULL);

        sound_handlers_num = 0;
        
        sound_set_cd_volume(65535, 65535);

	for (i = 0; i < CDROM_NUM; i++)
	{
		if (cdrom_drives[i].handler->audio_stop)
		{
	        	cdrom_drives[i].handler->audio_stop(i);
		}
	}
}

void sound_cd_thread_reset(void)
{
	int i = 0;
	int available_cdrom_drives = 0;

	for (i = 0; i < CDROM_NUM; i++)
	{
		if (cdrom_drives[i].bus_type != CDROM_BUS_DISABLED)
		{
			available_cdrom_drives++;
		}
	}

	if (available_cdrom_drives && !cd_thread_enable)
	{
		sound_cd_event = thread_create_event();
		sound_cd_thread_h = thread_create(sound_cd_thread, NULL);
	}
	else if (!available_cdrom_drives && cd_thread_enable)
	{
		thread_destroy_event(sound_cd_event);
		thread_kill(sound_cd_thread_h);
		sound_cd_thread_h = NULL;
	}

	cd_thread_enable = available_cdrom_drives ? 1 : 0;
}