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

*/

#include "quakedef.h"
#include "snd_main.h"


typedef struct portable_samplepair_s
{
	int sample[SND_LISTENERS];
} portable_sampleframe_t;

// LordHavoc: was 512, expanded to 2048
#define	PAINTBUFFER_SIZE 2048
portable_sampleframe_t paintbuffer[PAINTBUFFER_SIZE];


extern void SCR_CaptureVideo_SoundFrame(unsigned char *bufstereo16le, size_t length, int rate);
static void S_CaptureAVISound(size_t length)
{
	size_t i;
	unsigned char out[PAINTBUFFER_SIZE * 4];
	unsigned char* out_ptr;

	if (!cls.capturevideo.active)
		return;

	// write the sound buffer as little endian 16bit interleaved stereo
	for(i = 0, out_ptr = out; i < length; i++, out_ptr += 4)
	{
		int n0, n1;

		n0 = paintbuffer[i].sample[0];
		n0 = bound(-32768, n0, 32767);
		out_ptr[0] = (unsigned char)n0;
		out_ptr[1] = (unsigned char)(n0 >> 8);

		n1 = paintbuffer[i].sample[1];
		n1 = bound(-32768, n1, 32767);
		out_ptr[2] = (unsigned char)n1;
		out_ptr[3] = (unsigned char)(n1 >> 8);
	}
	SCR_CaptureVideo_SoundFrame(out, length, snd_renderbuffer->format.speed);
}

static unsigned int S_TransferPaintBuffer(snd_ringbuffer_t* rb, unsigned int starttime, unsigned int endtime)
{
	unsigned int partialend;

	// Lock submitbuffer
	if (!simsound && !SndSys_LockRenderBuffer())
		return 0;

	partialend = starttime;
	while (partialend < endtime)  // handle recirculating buffer issues
	{
		unsigned int startoffset, maxframes, nbframes, i;
		void *rb_ptr;
		portable_sampleframe_t *painted_ptr;
		int val;

		startoffset = partialend % rb->maxframes;
		maxframes = rb->maxframes - startoffset;
		nbframes = endtime - partialend;
		if (nbframes > maxframes)
			nbframes = maxframes;

		rb_ptr = &rb->ring[startoffset * rb->format.width * rb->format.channels];
		painted_ptr = &paintbuffer[partialend - starttime];

		if (rb->format.width == 2)  // 16bit
		{
			short *snd_out = (short*)rb_ptr;
			if (rb->format.channels == 8)  // 7.1 surround
			{
				for (i = 0;i < nbframes;i++, painted_ptr++)
				{
					*snd_out++ = bound(-32768, painted_ptr->sample[0], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[1], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[2], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[3], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[4], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[5], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[6], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[7], 32767);
				}
			}
			else if (rb->format.channels == 6)  // 5.1 surround
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					*snd_out++ = bound(-32768, painted_ptr->sample[0], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[1], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[2], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[3], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[4], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[5], 32767);
				}
			}
			else if (rb->format.channels == 4)  // 4.0 surround
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					*snd_out++ = bound(-32768, painted_ptr->sample[0], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[1], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[2], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[3], 32767);
				}
			}
			else if (rb->format.channels == 2)  // 2.0 stereo
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					*snd_out++ = bound(-32768, painted_ptr->sample[0], 32767);
					*snd_out++ = bound(-32768, painted_ptr->sample[1], 32767);
				}
			}
			else if (rb->format.channels == 1)  // 1.0 mono
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					val = (painted_ptr->sample[0] + painted_ptr->sample[1]) >> 1;
					*snd_out++ = bound(-32768, val, 32767);
				}
			}
		}
		else  // 8bit
		{
			unsigned char *snd_out = (unsigned char*)rb_ptr;
			if (rb->format.channels == 8)  // 7.1 surround
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					val = (painted_ptr->sample[0] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[1] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[2] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[3] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[4] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[5] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[6] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[7] >> 8) + 128; *snd_out++ = bound(0, val, 255);
				}
			}
			else if (rb->format.channels == 6)  // 5.1 surround
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					val = (painted_ptr->sample[0] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[1] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[2] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[3] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[4] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[5] >> 8) + 128; *snd_out++ = bound(0, val, 255);
				}
			}
			else if (rb->format.channels == 4)  // 4.0 surround
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					val = (painted_ptr->sample[0] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[1] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[2] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[3] >> 8) + 128; *snd_out++ = bound(0, val, 255);
				}
			}
			else if (rb->format.channels == 2)  // 2.0 stereo
			{
				for (i = 0; i < nbframes; i++, painted_ptr++)
				{
					val = (painted_ptr->sample[0] >> 8) + 128; *snd_out++ = bound(0, val, 255);
					val = (painted_ptr->sample[1] >> 8) + 128; *snd_out++ = bound(0, val, 255);
				}
			}
			else if (rb->format.channels == 1)  // 1.0 mono
			{
				for (i = 0;i < nbframes;i++, painted_ptr++)
				{
					val = ((painted_ptr->sample[0] + painted_ptr->sample[1]) >> 9) + 128;
					*snd_out++ = bound(0, val, 255);
				}
			}
		}

		partialend += nbframes;
	}

	rb->endframe = endtime;

	// Remove outdated samples from the ring buffer, if any
	if (rb->startframe < soundtime)
		rb->startframe = soundtime;

	if (!simsound)
		SndSys_UnlockRenderBuffer();

	return endtime - starttime;
}


/*
===============================================================================

CHANNEL MIXING

===============================================================================
*/

static qboolean SND_PaintChannel (channel_t *ch, unsigned int count)
{
	int snd_vol, vol[SND_LISTENERS];
	const snd_buffer_t *sb;
	unsigned int i, sb_offset;

	// If this channel manages its own volume
	if (ch->flags & CHANNELFLAG_FULLVOLUME)
		snd_vol = 256;
	else
		snd_vol = (int)(volume.value * 256);

	// calculate mixing volumes based on channel volumes and volume cvar
	// also limit the volumes to values that won't clip
	for (i = 0;i < SND_LISTENERS;i++)
	{
		vol[i] = ch->listener_volume[i] * snd_vol;
		vol[i] = bound(0, vol[i], 65536);
	}

	// if volumes are all zero, just return
	for (i = 0;i < SND_LISTENERS;i++)
		if (vol[i])
			break;
	if (i == SND_LISTENERS)
		return false;

	sb_offset = ch->pos;
	sb = ch->sfx->fetcher->getsb (ch->sfx->fetcher_data, &ch->fetcher_data, &sb_offset, count);
	if (sb == NULL)
	{
		Con_DPrintf("SND_PaintChannel: ERROR: can't get sound buffer from sfx \"%s\"\n",
					ch->sfx->name); // , count); // or add this? FIXME
		return false;
	}
	else
	{
#if SND_LISTENERS != 8
#		error the following code only supports up to 8 channels, update it
#endif
		if (sb->format.width == 1)
		{
			const signed char *samples = (signed char*)sb->samples + (ch->pos - sb_offset) * sb->format.channels;

			// Stereo sound support
			if (sb->format.channels == 2)
			{
				if (vol[6] + vol[7] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 8;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 8;
						paintbuffer[i].sample[3] += (samples[1] * vol[3]) >> 8;
						paintbuffer[i].sample[4] += ((samples[0] + samples[1]) * vol[4]) >> 9;
						paintbuffer[i].sample[5] += ((samples[0] + samples[1]) * vol[5]) >> 9;
						paintbuffer[i].sample[6] += (samples[0] * vol[6]) >> 8;
						paintbuffer[i].sample[7] += (samples[1] * vol[7]) >> 8;
						samples += 2;
					}
				}
				else if (vol[4] + vol[5] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 8;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 8;
						paintbuffer[i].sample[3] += (samples[1] * vol[3]) >> 8;
						paintbuffer[i].sample[4] += ((samples[0] + samples[1]) * vol[4]) >> 9;
						paintbuffer[i].sample[5] += ((samples[0] + samples[1]) * vol[5]) >> 9;
						samples += 2;
					}
				}
				else if (vol[2] + vol[3] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 8;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 8;
						paintbuffer[i].sample[3] += (samples[1] * vol[3]) >> 8;
						samples += 2;
					}
				}
				else if (vol[0] + vol[1] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 8;
						samples += 2;
					}
				}
			}
			else if (sb->format.channels == 1)
			{
				if (vol[6] + vol[7] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 8;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 8;
						paintbuffer[i].sample[3] += (samples[0] * vol[3]) >> 8;
						paintbuffer[i].sample[4] += (samples[0] * vol[4]) >> 8;
						paintbuffer[i].sample[5] += (samples[0] * vol[5]) >> 8;
						paintbuffer[i].sample[6] += (samples[0] * vol[6]) >> 8;
						paintbuffer[i].sample[7] += (samples[0] * vol[7]) >> 8;
						samples += 1;
					}
				}
				else if (vol[4] + vol[5] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 8;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 8;
						paintbuffer[i].sample[3] += (samples[0] * vol[3]) >> 8;
						paintbuffer[i].sample[4] += (samples[0] * vol[4]) >> 8;
						paintbuffer[i].sample[5] += (samples[0] * vol[5]) >> 8;
						samples += 1;
					}
				}
				else if (vol[2] + vol[3] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 8;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 8;
						paintbuffer[i].sample[3] += (samples[0] * vol[3]) >> 8;
						samples += 1;
					}
				}
				else if (vol[0] + vol[1] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 8;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 8;
						samples += 1;
					}
				}
			}
			else
				return false; // unsupported number of channels in sound
		}
		else if (sb->format.width == 2)
		{
			const signed short *samples = (signed short*)sb->samples + (ch->pos - sb_offset) * sb->format.channels;

			// Stereo sound support
			if (sb->format.channels == 2)
			{
				if (vol[6] + vol[7] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 16;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 16;
						paintbuffer[i].sample[3] += (samples[1] * vol[3]) >> 16;
						paintbuffer[i].sample[4] += ((samples[0] + samples[1]) * vol[4]) >> 17;
						paintbuffer[i].sample[5] += ((samples[0] + samples[1]) * vol[5]) >> 17;
						paintbuffer[i].sample[6] += (samples[0] * vol[6]) >> 16;
						paintbuffer[i].sample[7] += (samples[1] * vol[7]) >> 16;
						samples += 2;
					}
				}
				else if (vol[4] + vol[5] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 16;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 16;
						paintbuffer[i].sample[3] += (samples[1] * vol[3]) >> 16;
						paintbuffer[i].sample[4] += ((samples[0] + samples[1]) * vol[4]) >> 17;
						paintbuffer[i].sample[5] += ((samples[0] + samples[1]) * vol[5]) >> 17;
						samples += 2;
					}
				}
				else if (vol[2] + vol[3] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 16;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 16;
						paintbuffer[i].sample[3] += (samples[1] * vol[3]) >> 16;
						samples += 2;
					}
				}
				else if (vol[0] + vol[1] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[1] * vol[1]) >> 16;
						samples += 2;
					}
				}
			}
			else if (sb->format.channels == 1)
			{
				if (vol[6] + vol[7] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 16;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 16;
						paintbuffer[i].sample[3] += (samples[0] * vol[3]) >> 16;
						paintbuffer[i].sample[4] += (samples[0] * vol[4]) >> 16;
						paintbuffer[i].sample[5] += (samples[0] * vol[5]) >> 16;
						paintbuffer[i].sample[6] += (samples[0] * vol[6]) >> 16;
						paintbuffer[i].sample[7] += (samples[0] * vol[7]) >> 16;
						samples += 1;
					}
				}
				else if (vol[4] + vol[5] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 16;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 16;
						paintbuffer[i].sample[3] += (samples[0] * vol[3]) >> 16;
						paintbuffer[i].sample[4] += (samples[0] * vol[4]) >> 16;
						paintbuffer[i].sample[5] += (samples[0] * vol[5]) >> 16;
						samples += 1;
					}
				}
				else if (vol[2] + vol[3] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 16;
						paintbuffer[i].sample[2] += (samples[0] * vol[2]) >> 16;
						paintbuffer[i].sample[3] += (samples[0] * vol[3]) >> 16;
						samples += 1;
					}
				}
				else if (vol[0] + vol[1] > 0)
				{
					for (i = 0;i < count;i++)
					{
						paintbuffer[i].sample[0] += (samples[0] * vol[0]) >> 16;
						paintbuffer[i].sample[1] += (samples[0] * vol[1]) >> 16;
						samples += 1;
					}
				}
			}
			else
				return false; // unsupported number of channels in sound
		}
	}
	return true;
}

void S_PaintChannels (snd_ringbuffer_t* rb, unsigned int starttime, unsigned int endtime)
{
	unsigned int paintedtime;

	paintedtime = starttime;
	while (paintedtime < endtime)
	{
		unsigned int partialend, i, framecount;
		channel_t *ch;

		// if paintbuffer is too small
		if (endtime > paintedtime + PAINTBUFFER_SIZE)
			partialend = paintedtime + PAINTBUFFER_SIZE;
		else
			partialend = endtime;

		// clear the paint buffer
		memset (paintbuffer, 0, (partialend - paintedtime) * sizeof (paintbuffer[0]));

		// paint in the channels.
		// channels with zero volumes still advance in time but don't paint.
		ch = channels;
		for (i = 0; i < total_channels ; i++, ch++)
		{
			sfx_t *sfx;
			unsigned int ltime;
			unsigned int count;

			sfx = ch->sfx;
			if (sfx == NULL)
				continue;
			if (!S_LoadSound (sfx, true))
				continue;
			if (ch->flags & CHANNELFLAG_PAUSED)
				continue;

			ltime = paintedtime;
			if (ch->pos < 0)
			{
				count = -ch->pos;
				count = min(count, partialend - ltime);
				ch->pos += count;
				ltime += count;
			}

			while (ltime < partialend)
			{
				// paint up to end of buffer or of input, whichever is lower
				count = sfx->total_length - ch->pos;
				count = bound(0, count, partialend - ltime);
				if (count)
				{
					SND_PaintChannel (ch, count);
					ch->pos += count;
					ltime += count;
				}

				// if at end of sfx, loop or stop the channel
				if (ch->pos >= (int)sfx->total_length)
				{
					if (sfx->loopstart < sfx->total_length)
						ch->pos = sfx->loopstart;
					else if (ch->flags & CHANNELFLAG_FORCELOOP)
						ch->pos = 0;
					else
					{
						S_StopChannel (ch - channels);
						break;
					}
				}
			}
		}

		S_CaptureAVISound (partialend - paintedtime);
		framecount = S_TransferPaintBuffer (rb, paintedtime, partialend);
		paintedtime += framecount;

		// If there was not enough free space in the sound buffer, stop here
		if (paintedtime != partialend)
		{
			Con_DPrintf(">> S_PaintChannels: Not enough free space in the sound buffer ( %u != %u)\n", paintedtime, partialend);
			break;
		}
	}
}