/* ScummVM - Scumm Interpreter
 * Copyright (C) 2002-2003 The ScummVM project
 *
 * 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.
 *
 * $Header$
 *
 */

#include <stdafx.h>
#include "channel.h"
#include "chunk.h"
#include "chunk_type.h"

#include <assert.h>
#include <string.h>

ImuseChannel::ImuseChannel(int32 track, int32 freq) : 
	_track(track), 
	_tbuffer(0), 
	_tbufferSize(0), 
	_sbuffer(0), 
	_sbufferSize(0), 
	_frequency(freq), 
	_dataSize(-1),
	_inData(false) {
}

ImuseChannel::~ImuseChannel() {
	if(_tbuffer) {
		delete []_tbuffer; 
	}
	if(_sbuffer) {
		warning("_sbuffer should be 0 !!!");
		delete []_sbuffer; 
	}
}

bool ImuseChannel::isTerminated() const {
	return (_dataSize <= 0 && _sbuffer == 0);
}

bool ImuseChannel::setParameters(int32 nb, int32 size, int32 flags, int32 unk1) {
	// flags: 0 - 8 bits
	// values:
	// 1 - Voice
	// 2 - Background music
	// 0, 3-511 - SFX and volume
	// FIXME: this should be better
	if ((flags != 1) && (flags != 2) && ((flags >> 2) != 0)) {
		_volume = 300 - ((flags >> 3) << 2);
	}
	else {
		_volume = 127;
	}
	return true;
}

bool ImuseChannel::checkParameters(int32 index, int32 nbframes, int32 size, int32 track_flags, int32 unk1) {
	return true;
}

bool ImuseChannel::appendData(Chunk &b, int32 size) {
	if(_dataSize == -1) { // First call
		assert(size > 8);
		Chunk::type imus_type = b.getDword(); imus_type = SWAP_BYTES(imus_type);
		uint32 imus_size = b.getDword(); imus_size = SWAP_BYTES(imus_size);
		if(imus_type != TYPE_iMUS)
			error("Invalid Chunk for imuse_channel");
		size -= 8;
		_tbufferSize = size;
		assert(_tbufferSize);
		_tbuffer = new byte[_tbufferSize];
		if(!_tbuffer)
			error("imuse_channel failed to allocate memory");
		b.read(_tbuffer, size);
		_dataSize = -2; // even if _in_data does not get set, this won't be called again
	} else {
		if(_tbuffer) { // remaining from last call
			byte *old = _tbuffer;
			int32 new_size = size + _tbufferSize;
			_tbuffer = new byte[new_size];
			if(!_tbuffer)
				error("imuse_channel failed to allocate memory");
			memcpy(_tbuffer, old, _tbufferSize);
			delete []old;
			b.read(_tbuffer + _tbufferSize, size);
			_tbufferSize += size;
		} else {
			_tbufferSize = size;
			_tbuffer = new byte[_tbufferSize];
			if(!_tbuffer)
				error("imuse_channel failed to allocate memory");
			b.read(_tbuffer, size);
		}
	}
	return processBuffer();
}

bool ImuseChannel::handleFormat(Chunk &src) {
	if(src.getSize() != 20) error("invalid size for FRMT Chunk");
	uint32 imuse_start = src.getDword();
	imuse_start = SWAP_BYTES(imuse_start);
	src.seek(4);
	_bitsize = src.getDword();
	_bitsize = SWAP_BYTES(_bitsize);
	_rate = src.getDword();
	_rate = SWAP_BYTES(_rate);
	_channels = src.getDword();
	_channels = SWAP_BYTES(_channels);
	assert(_channels == 1 || _channels == 2);
	return true;
}

bool ImuseChannel::handleText(Chunk &src) {
	return true;
}

bool ImuseChannel::handleRegion(Chunk &src) {
	if(src.getSize() != 8)
		error("invalid size for REGN Chunk");
	return true;
}

bool ImuseChannel::handleStop(Chunk &src) {
	if(src.getSize() != 4)
		error("invalid size for STOP Chunk");
	return true;
}

bool ImuseChannel::handleMap(Chunk &map) {
	while(!map.eof()) {
		Chunk *sub = map.subBlock();
		switch(sub->getType()) {
			case TYPE_FRMT:
				handleFormat(*sub);
				break;
			case TYPE_TEXT:
				handleText(*sub);
				break;
			case TYPE_REGN:
				handleRegion(*sub);
				break;
			case TYPE_STOP:
				handleStop(*sub);
				break;
			default:
				error("Unknown iMUS subChunk found : %s, %d", Chunk::ChunkString(sub->getType()), sub->getSize());
		}
		delete sub;
	}
	return true;
}

void ImuseChannel::decode() {
	int remaining_size = _sbufferSize % 3;
	if(remaining_size) {
		_srbufferSize -= remaining_size;
		assert(_inData);
		if(_tbuffer == 0) {
			_tbuffer = new byte[remaining_size];
			memcpy(_tbuffer, _sbuffer + _sbufferSize - remaining_size, remaining_size);
			_tbufferSize = remaining_size;
			_sbufferSize -= remaining_size;
		} else {
			debug(2, "impossible ! : %p, %d, %d, %p(%d), %p(%d, %d)",
				this, _dataSize, _inData, _tbuffer, _tbufferSize, _sbuffer, _sbufferSize, _srbufferSize);
			byte *old = _tbuffer;
			int new_size = remaining_size + _tbufferSize;
			_tbuffer = new byte[new_size];
			if(!_tbuffer)  error("imuse_channel failed to allocate memory");
			memcpy(_tbuffer, old, _tbufferSize);
			delete []old;
			memcpy(_tbuffer + _tbufferSize, _sbuffer + _sbufferSize - remaining_size, remaining_size);
			_tbufferSize += remaining_size;
		}
	}
	int loop_size = _sbufferSize / 3;
	int new_size = loop_size * 2;
	byte *keep, *decoded;
	uint32 value;
	keep = decoded = new byte[new_size * 2];
	assert(keep);
	unsigned char * source = _sbuffer;
		while(loop_size--) {
		byte v1 =  *source++;
		byte v2 =  *source++;
		byte v3 =  *source++;
		value = ((((v2 & 0x0f) << 8) | v1) << 4) - 0x8000;
		*decoded++ = (byte)((value >> 8) & 0xff);
		*decoded++ = (byte)(value & 0xff);
		value = ((((v2 & 0xf0) << 4) | v3) << 4) - 0x8000;
		*decoded++ = (byte)((value >> 8) & 0xff);
		*decoded++ = (byte)(value & 0xff);
	}
	delete []_sbuffer;
	_sbuffer = (byte *)keep;
	_sbufferSize = new_size * sizeof(int16);
}

bool ImuseChannel::handleSubTags(int32 &offset) {
	if(_tbufferSize - offset >= 8) {
		Chunk::type type = READ_BE_UINT32(_tbuffer + offset);
		uint32 size = READ_BE_UINT32(_tbuffer + offset + 4);
		uint32 available_size = _tbufferSize - offset;
		switch(type) {
			case TYPE_MAP_: 
				_inData = false;
				if(available_size >= (size + 8)) {
					ContChunk c((byte *)_tbuffer + offset);
					handleMap(c);
				}
				break;
			case TYPE_DATA: // Sound data !!!
				_inData = true;
				_dataSize = size;
				offset += 8;
				{
					int reqsize = 1;
					if(_channels == 2)
						reqsize *= 2;
					if(_bitsize == 16)
						reqsize *= 2;
					else if(_bitsize == 12) {
						if(reqsize > 1)
							reqsize = reqsize * 3 / 2;
						else reqsize = 3;
					}
					if((size % reqsize) != 0) {
						debug(2, "Invalid iMUS sound data size : (%d %% %d) != 0, correcting...", size, reqsize);
						size += 3 - (size % reqsize);
					}
				}
				return false;
			default:
				error("unknown Chunk in iMUS track : %s ", Chunk::ChunkString(type));
		}
		offset += size + 8;
		return true;
	}
	return false;
}

bool ImuseChannel::processBuffer() {
	// see comments in saud_channel::processBuffer for an explanation
	assert(_tbuffer != 0);
	assert(_tbufferSize != 0);
	assert(_sbuffer == 0);
	assert(_sbufferSize == 0);
	
	if(_inData) {
		if(_dataSize < _tbufferSize) {			
			int32 offset= _dataSize;
			while(handleSubTags(offset));
			_sbufferSize = _dataSize;
			_sbuffer = _tbuffer;
			if(offset < _tbufferSize) { // there is still some unprocessed data
				int32 new_size = _tbufferSize - offset;
				_tbuffer = new byte[new_size];
				if(!_tbuffer) error("imuse_channel failed to allocate memory");
				memcpy(_tbuffer, _sbuffer + offset, new_size);
				_tbufferSize = new_size;
			} else {
				_tbuffer = 0;
				_tbufferSize = 0;
			}
			if(_sbufferSize == 0) {
				// this never happened yet, but who knows
				delete []_sbuffer; 
				_sbuffer = 0;
			}
		} else {
			// easy, swap the buffer
			_sbufferSize = _tbufferSize;
			_sbuffer = _tbuffer;
			_tbufferSize = 0;
			_tbuffer = 0;
		}
	} else {
		int32 offset = 0;
		while(handleSubTags(offset));
		if(_inData) {
			_sbufferSize = _tbufferSize - offset;
			assert(_sbufferSize);
			_sbuffer = new byte[_sbufferSize];
			if(!_sbuffer) error("imuse_channel failed to allocate memory");
			memcpy(_sbuffer, _tbuffer + offset, _sbufferSize);
			delete []_tbuffer;
			_tbuffer = 0;
			_tbufferSize = 0;
		} else {
			if(offset) { // maybe I should assert() this to avoid a lock...
				byte * old = _tbuffer;
				int32 new_size = _tbufferSize - offset;
				_tbuffer = new byte[new_size];
				if(!_tbuffer) error("imuse_channel failed to allocate memory");
				memcpy(_tbuffer, old + offset, new_size);
				_tbufferSize = new_size;
				delete []old;
			}
		}
	}
	_srbufferSize = _sbufferSize;
	if(_sbuffer && _bitsize == 12) decode();
	return true;
}

int32 ImuseChannel::availableSoundData(void) const {
	int32 ret = _sbufferSize;
	if(_channels == 2) ret /= 2;
	if(_bitsize > 8) ret /= 2;
	return ret;
}

void ImuseChannel::getSoundData(int16 *snd, int32 size) {
	if(_dataSize <= 0 || _bitsize <= 8) error("invalid call to imuse_channel::read_sound_data()");
	if(_channels == 2) size *= 2;
	byte * buf = (byte*)snd;
	if(_rate == 11025) {
		for(int32 i = 0; i < size; i++) {
			byte sample1 = *(_sbuffer + i * 2);
			byte sample2 = *(_sbuffer + i * 2 + 1);
			uint16 sample = (uint16)(((int16)((sample1 << 8) | sample2) * _volume) >> 8);
			buf[i * 4 + 0] = (byte)(sample >> 8);
			buf[i * 4 + 1] = (byte)(sample & 0xff);
			buf[i * 4 + 2] = buf[i * 4 + 0];
			buf[i * 4 + 3] = buf[i * 4 + 1];
		}
	} else {
		for(int32 i = 0; i < size; i++){
			byte sample1 = *(_sbuffer + i * 2);
			byte sample2 = *(_sbuffer + i * 2 + 1);
			uint16 sample = (uint16)(((int16)((sample1 << 8) | sample2) * _volume) >> 8);
			buf[i * 2 + 0] = (byte)(sample >> 8);
			buf[i * 2 + 1] = (byte)(sample & 0xff);
		}
	}
	delete []_sbuffer;
	assert(_sbufferSize == 2 * size);
	_sbuffer = 0;
	_sbufferSize = 0;
	_dataSize -= _srbufferSize;
}

void ImuseChannel::getSoundData(int8 *snd, int32 size) {
	if(_dataSize <= 0 || _bitsize > 8) error("invalid call to imuse_channel::read_sound_data()");
	if(_channels == 2) size *= 2;
	if(_rate == 11025) {
		for(int32 i = 0; i < size; i++) {
			snd[i * 2] = (int8)(((int8)(_sbuffer[i] ^ 0x80) * _volume) >> 8) ^ 0x80;
			snd[i * 2 + 1] = snd[i * 2];
		}
	} else {
		for(int32 i = 0; i < size; i++){
			snd[i] = (int8)(((int8)(_sbuffer[i] ^ 0x80) * _volume) >> 8) ^ 0x80;
		}
	}
	delete []_sbuffer;
	_sbuffer = 0;
	_sbufferSize = 0;
	_dataSize -= _srbufferSize;
}