/* ScummVM - Scumm Interpreter
 * Copyright (C) 2001/2002 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 "scummsys.h"
#include "system.h"
#include "mididrv.h"
#include "mixer.h"
#include "simon.h"

void MidiPlayer::read_from_file(void *dst, uint size) {
	if (fread(dst, size, 1, _input) != 1)
		error("Midi read error");
}

byte MidiPlayer::read_byte_from_file() {
	byte num;
	read_from_file(&num, 1);
	return num;
}

uint32 MidiPlayer::read_uint32_from_file() {
	uint32 num;
	read_from_file(&num, 4);
	return READ_BE_UINT32(&num);
}

uint16 MidiPlayer::read_uint16_from_file() {
	uint16 num;
	read_from_file(&num, 2);
	return READ_BE_UINT16(&num);
}

void MidiPlayer::read_all_songs(FILE *in) {
	uint i,num;
	
	_input = in;

	_midi_cur_song_ptr = _midi_songs;

	num = read_byte_from_file();

	for(i=0; i!=num; i++) {
		read_one_song(&_midi_songs[i]);
	}
}

void MidiPlayer::read_all_songs_old(FILE *in) {
	uint i,num;
	
	_input = in;
	_midi_cur_song_ptr = _midi_songs;

	num = 1;

	for(i=0; i!=num; i++) {
		read_one_song(&_midi_songs[i]);
	}
}

void MidiPlayer::read_mthd(Song *s, bool old) {
	Track *t;
	uint i;

	if (!old) {
		if (read_uint32_from_file() != 6)
			error("Invalid 'MThd' chunk size");
		s->midi_format = read_uint16_from_file();
		s->num_tracks = read_uint16_from_file();
		s->ppqn = read_uint16_from_file();
	} else {
		s->midi_format = 0;
		s->num_tracks = 1;
		s->ppqn = 0xc0;

		read_uint16_from_file();
		read_byte_from_file();
	}

	s->tracks = t = (Track*)calloc(s->num_tracks, sizeof(Track));
	if (t == NULL)
		error("Out of memory when allocating MIDI tracks");
	
	for(i=0; i!=s->num_tracks; i++, t++) {
		if (!old) {
			if (read_uint32_from_file() != 'MTrk')
				error("Midi track has no 'MTrk'");
	
			t->data_size = read_uint32_from_file();
		} else {
			uint32 pos = ftell(_input);
			fseek(_input, 0, SEEK_END);
			uint32 end = ftell(_input);
			fseek(_input, pos, SEEK_SET);
			t->data_size = end - pos;
		}

		t->data_ptr = (byte*)calloc(t->data_size,1);
		if (t->data_ptr == NULL)
			error("Out of memory when allocating MIDI track data");

		read_from_file(t->data_ptr, t->data_size);

		t->data_cur_size = t->data_size;
		t->data_cur_ptr = t->data_ptr;

		t->a = 0;
		t->last_cmd = 0;
		t->delay = 0;
		
		if (t->data_cur_size == 0) {
			t->a |= 1;
			continue;
		}
		
		t->delay = track_read_gamma(t);
	}
}

void MidiPlayer::read_one_song(Song *s) {
	_midi_var10 = 0;
	
	s->ppqn = 0;
	s->midi_format = 0;
	s->num_tracks = 0;
	s->tracks = NULL;

	uint32 id = read_uint32_from_file();

	switch(id) {
	case 'MThd':
		read_mthd(s, false);
		break;

	case 'GMF\x1':
		warning("Old style songs not properly supported yet");
		read_mthd(s, true);
		break;

	default:
		error("Midi song has no 'MThd'");
	}

}

uint32 MidiPlayer::track_read_gamma(Track *t) {
	uint32 sum;
	byte b;

	sum = 0;
	do {
		b = track_read_byte(t);
		sum = (sum<<7) | (b & 0x7F);
	} while (b & 0x80);

	return sum;	
}

byte MidiPlayer::track_read_byte(Track *t) {
	if (t->a & 1)
		error("Trying to read byte from MIDI stream when end reached");
	
	if (!--t->data_cur_size) {
		t->a|=1;
	}

	return *t->data_cur_ptr++;
}

void MidiPlayer::initialize() {
	int res;
	int i;

	for(i=0; i!=16; i++)
		_midi_volume_table[i] = 100;
	_midi_5 = 0;

	_md->property(MidiDriver::PROP_TIMEDIV, _midi_songs[0].ppqn);

	res = _md->open(MidiDriver::MO_STREAMING);
	if (res != 0)
		error("MidiPlayer::initializer, got %s", MidiDriver::get_error_name(res));
	
	_midi_var9 = true;
}

int MidiPlayer::fill(MidiEvent *me, int num_event) {
	uint32 best,j;
	Track *best_track,*t;
	bool did_reset;
	NoteRec midi_tmp_note_rec;
	int i = 0;

	did_reset = false;
	
	for(i=0; i!=num_event;) {
		best_track = NULL;
		best = 0xFFFFFFFF;

		/* Locate which track that's next */
		t = _midi_cur_song_ptr->tracks;
		for(j=_midi_cur_song_ptr->num_tracks; j; j--,t++) {
			if (!(t->a&1)) {
				if (t->delay < best) {
					best = t->delay;
					best_track = t;
				} 
			}
		}

		if (best_track == NULL) {
			/* reset tracks if song ended? */
			if (did_reset) {
				/* exit if song ended completely */
				return i;
			}
			did_reset = true;
			reset_tracks();
			continue;
		}

		read_next_note(best_track, &midi_tmp_note_rec);
		if (fill_helper(&midi_tmp_note_rec, me + i))
			i++;

		if (midi_tmp_note_rec.sysex_data)	free(midi_tmp_note_rec.sysex_data);
	}

	return i;
}

int MidiPlayer::on_fill(void *param, MidiEvent *ev, int num) {
	MidiPlayer *mp = (MidiPlayer*)param;
	return mp->fill(ev, num);
}

bool MidiPlayer::fill_helper(NoteRec *nr, MidiEvent *me) {
	uint b;

	b = nr->delay - _midi_var10;
	_midi_var10 = nr->delay;
	
	if (nr->cmd<0xF0) {
		me->delta = b;
		me->event = nr->cmd | (nr->param_1<<8) | (nr->param_2<<16);
		
		if ((nr->cmd&0xF0) == 0xB0 && nr->param_1 == 7) {
			_midi_volume_table[nr->cmd&0xF] = nr->param_2;
			nr->param_1 = 0x76;
			me->event = nr->cmd | (nr->param_1<<8) | (nr->param_2<<16)/* | MEVT_F_CALLBACK*/;
		}
		return true;

	}
	
	if (nr->cmd==0xF0 || nr->cmd==0xF7 || nr->param_1 != 0x51) {
		return false;
	}
	
	_midi_tempo = nr->sysex_data[2] | (nr->sysex_data[1]<<8) | (nr->sysex_data[0]<<16);
	_midi_var8 = (_midi_cur_song_ptr->ppqn*60000) / _midi_tempo;

	me->delta = b;
	me->event = (MidiDriver::ME_TEMPO << 24) | _midi_tempo;

	return true;
}

void MidiPlayer::reset_tracks() {
	Track *t;
	uint i;

	_midi_var10 = 0;

	for(i=0,t=_midi_cur_song_ptr->tracks; i!=_midi_cur_song_ptr->num_tracks; i++,t++) {
		t->data_cur_size = t->data_size;
		t->data_cur_ptr = t->data_ptr;
		t->a = 0;
		t->last_cmd = 0;
		t->delay = 0;
		if (t->data_cur_size==0) {
			t->a|=1;
		} else {
			t->delay = track_read_gamma(t);
		}
	}
}

void MidiPlayer::read_next_note(Track *t, NoteRec *nr) {
	byte cmd_byte;
	uint i;

	nr->delay = 0;
	nr->cmd = 0;
	nr->param_1 = 0;
	nr->param_2 = 0;
	nr->cmd_length = 0;
	nr->sysex_data = NULL;
	
	if(t->a&1 || t->data_cur_size==0)
		error("read next note when track ended");

	/* read next midi byte, but skip any pitch bends. */
	for(;;) {
		cmd_byte = track_read_byte(t);
		if ((cmd_byte&0xF0) != 0xE0)
			break;

		track_read_byte(t);
		track_read_byte(t);
	} 
	
	if (!(cmd_byte & 0x80)) {
		/* running status? */
		if (t->last_cmd==0)
			error("Last cmd = 0");
		nr->cmd = t->last_cmd;
		nr->param_1 = cmd_byte;
		cmd_byte = nr->cmd&0xF0;
		nr->cmd_length = 2;
		if (cmd_byte!=0xC0 && cmd_byte!=0xD0) {
			nr->param_2 = track_read_byte(t);
			nr->cmd_length++;
		}
	} else if ((cmd_byte&0xF0)!=0xF0) {
		nr->cmd = cmd_byte;
		t->last_cmd = cmd_byte;
		cmd_byte &= 0xF0;
		nr->cmd_length = (cmd_byte==0xC0 || cmd_byte==0xD0) ? 2 : 3;
		
		if (t->data_cur_size < nr->cmd_length-1) {
			error("read_next_note: end of stream");
		}
		
		nr->param_1 = track_read_byte(t);
		if (nr->cmd_length==3)
			nr->param_2 = track_read_byte(t);

	} else if (cmd_byte==0xF0 || cmd_byte==0xF7) {
		nr->cmd = cmd_byte;
		nr->cmd_length = track_read_gamma(t);
		if (t->data_cur_size < nr->cmd_length)
			error("read_next_note: end of stream 2");
		nr->sysex_data = (byte*)malloc(nr->cmd_length);
		if (nr->sysex_data==NULL)
			error("read_next_note: out of memory");
		for(i=0; i!=nr->cmd_length; i++)
			nr->sysex_data[i] = track_read_byte(t);
	} else if (cmd_byte==0xFF) {
		
		nr->cmd = cmd_byte;
		nr->param_1 = track_read_byte(t);
		nr->cmd_length = track_read_gamma(t);
		if (nr->cmd_length) {
			if (t->data_cur_size < nr->cmd_length)
				error("read_next_note: end of stream 3");
			nr->sysex_data = (byte*)malloc(nr->cmd_length);
			if (nr->sysex_data==NULL)
				error("read_next_note: out of memory");
			for(i=0; i!=nr->cmd_length; i++)
				nr->sysex_data[i] = track_read_byte(t);
		}
		if (nr->param_1==0x2F)
			t->a|=1;
	} else {
		error("Invalid sysex cmd");
	}

	nr->delay = t->delay;
	if (!(t->a&1)) {
		t->delay += track_read_gamma(t);
	}
}

void MidiPlayer::shutdown() {
	_md->close();
	unload();
}

void MidiPlayer::unload() {
	uint i,j;
	Song *s;
	Track *t;
	for(i=0,s=_midi_songs; i!=8; i++,s++) {
		if (s->tracks) {
			for(j=0,t=s->tracks; j!=s->num_tracks;j++,t++) {
				if (t->data_ptr)
					free(t->data_ptr);
			}
			free(s->tracks);
			s->tracks = NULL;
		}
	}
}

void MidiPlayer::play() {
	_md->pause(false);
}


void MidiPlayer::set_driver(MidiDriver *md) {
	_md = md;
	md->set_stream_callback(this, on_fill);
}