/*
 * Copyright 2003-2009 (C) Raster Software Vigo (Sergio Costas)
 * This file is part of FBZX
 *
 * FBZX 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 3 of the License, or
 * (at your option) any later version.
 *
 * FBZX 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, see <http://www.gnu.org/licenses/>.
 *
 */
#include "llsound.hh"

#include <sstream>

#include "z80free/Z80free.h"
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include "computer.hh"
#include "emulator.hh"

#include <fcntl.h>
#include <sys/ioctl.h>


class LLSound *llsound;

#ifdef D_SOUND_OSS
#include <sys/soundcard.h>
int audio_fd;
#endif

#ifdef D_SOUND_ALSA
#define ALSA_PCM_NEW_HW_PARAMS_API
#define ALSA_PCM_NEW_SW_PARAMS_API
#include <alsa/asoundlib.h>
int started_sound;
snd_pcm_t * _soundDevice;
#endif

#ifdef D_SOUND_PULSE
#include <pulse/simple.h>
pa_simple *pulse_s;
#endif

LLSound::LLSound(enum e_soundtype sound_type) {

	int bucle,bucle2,ret2;

	this->sound_type = sound_type;
	ret2=this->init_sound(); // check all sound systems
	if(ret2==0) {
		sound_aborted = false;
	} else { // if fails, run without sound
		this->sound_type=SOUND_NO;
		this->init_sound();
		sound_aborted = true;
	}
	printf("Init sound\n");
	if(this->format != 0)
		this->increment = 2 * this->channels;
	else
		this->increment = this->channels;

	// sound initialization
	this->sound = new uint8_t[this->buffer_len * this->increment + 8];
	for(bucle = 0; bucle < this->buffer_len * this->increment + 4; bucle++) {
		sound[bucle]=0;
	}
	this->current_buffer = this->sound;
	this->tst_sample = 3500000 / this->freq;
	printf("Init sound 2 (%d)\n",this->tst_sample);
	printf("Set volume\n");
	set_volume(70);
	printf("Return init\n");
}

void LLSound::set_speed(bool turbo) {

	if(turbo){
		this->tst_sample = 1000000000 / this->freq;
	} else {
		this->tst_sample = 3500000 / this->freq;
	}
}

int LLSound::init_sound() {

	if (this->sound_type != SOUND_AUTOMATIC) {
		switch(this->sound_type) {
		case SOUND_NO: // No sound; simulate 8bits mono
			this->sign=0;
			this->format=0;
			this->channels = 1;
			this->freq=48000;
			this->buffer_len=4800; // will wait 1/10 second
			return 0;
		break;
#ifdef D_SOUND_PULSE
		case SOUND_PULSEAUDIO:
			printf("Trying PulseAudio\n");
			if(0==this->init_pulse()) {
				this->sound_type=SOUND_PULSEAUDIO;
				return 0;
			} else {
				printf("Failed\n");
				return -1;
			}
		break;
#endif
#ifdef D_SOUND_ALSA
		case SOUND_ALSA:
			printf("Trying ALSA sound\n");
			if(0 == this->init_alsa()) {
				this->sound_type=SOUND_ALSA;
				return 0;
			} else {
				printf("Failed\n");
				return -1;
			}
		break;
#endif
#ifdef D_SOUND_OSS
		case SOUND_OSS:
			printf("Trying OSS sound\n");
			if(0 == this->init_oss()) {
				this->sound_type=SOUND_OSS;
				return 0;
			} else {
				printf("Failed\n");
				return -1;
			}
		break;
#endif
		default:
			return -1;
		break;
		}
	}

#ifdef D_SOUND_PULSE
	printf("Trying PulseAudio\n");
	if(0 == this->init_pulse()) {
		this->sound_type=SOUND_PULSEAUDIO;
		return 0;
	}
#endif

#ifdef D_SOUND_ALSA
	printf("Trying ALSA sound\n");
	if(0 == this->init_alsa()) {
		this->sound_type=SOUND_ALSA;
		return 0;
	}
#endif

#ifdef D_SOUND_OSS
	printf("Trying OSS sound\n");
	if(0 == this->init_oss()) {
		this->sound_type=SOUND_OSS;
		return 0;
	}
#endif
	return -1;
}

#ifdef D_SOUND_PULSE
int LLSound::init_pulse() {

	pa_sample_spec ss;
	pa_buffer_attr buf;

	ss.format = PA_SAMPLE_U8;
	ss.channels = 1;
	ss.rate = 48000;

	buf.maxlength=8192;
	buf.tlength=4096;
	buf.prebuf=4096;
	buf.minreq=4096;
	buf.fragsize=4096;

	pulse_s = pa_simple_new(NULL,"fbzx",PA_STREAM_PLAYBACK,NULL,"Spectrum",&ss,NULL,&buf,NULL);
	if (pulse_s==NULL) {
		return -1;
	}
	this->sign=0;
	this->format=0;
	this->channels=1;
	this->freq=48000;
	this->buffer_len=4096;

	return 0;

}
#endif

#ifdef D_SOUND_ALSA
int LLSound::init_alsa() {

	int err;
	snd_pcm_hw_params_t *hw_params;

	unsigned int resample,samplerate;
	snd_pcm_uframes_t bufferSize;

	err = snd_pcm_open( &_soundDevice, "plughw:0,0", SND_PCM_STREAM_PLAYBACK, 0 );
	if (err<0) {
		return -1;
	}

	if ((err = snd_pcm_hw_params_malloc (&hw_params)) < 0) {
		snd_pcm_close (_soundDevice);
		return -2;
	}

	if ((err = snd_pcm_hw_params_any (_soundDevice, hw_params)) < 0) {
		snd_pcm_close (_soundDevice);
		return -2;
	}


	if ((err = snd_pcm_hw_params_set_access (_soundDevice, hw_params,SND_PCM_ACCESS_RW_INTERLEAVED)) < 0) {
		snd_pcm_close (_soundDevice);
		return -3;
	}

	if (snd_pcm_hw_params_set_format (_soundDevice, hw_params,SND_PCM_FORMAT_U8) >= 0) {
		this->sign=0;
	    this->format=0;
	} else if (snd_pcm_hw_params_set_format (_soundDevice, hw_params,SND_PCM_FORMAT_S8) >= 0) {
		this->sign=-128;
	    this->format=0;
	} else if (snd_pcm_hw_params_set_format (_soundDevice, hw_params,SND_PCM_FORMAT_U16_LE) >= 0) {
		this->sign=0;
	    this->format=1;
	} else if (snd_pcm_hw_params_set_format (_soundDevice, hw_params,SND_PCM_FORMAT_S16_LE) >= 0) {
		this->sign=-128;
	    this->format=1;
	} else if (snd_pcm_hw_params_set_format (_soundDevice, hw_params,SND_PCM_FORMAT_U16_BE) >= 0) {
		this->sign=0;
	    this->format=2;
	} else if (snd_pcm_hw_params_set_format (_soundDevice, hw_params,SND_PCM_FORMAT_S16_BE) >= 0) {
		this->sign=-128;
	    this->format=2;
	} else {
		snd_pcm_close (_soundDevice);
		return -3;
	}

	// Disable resampling.
	resample = 0;
	err = snd_pcm_hw_params_set_rate_resample(_soundDevice, hw_params, resample);
	if (err < 0) {
		snd_pcm_close (_soundDevice);
		return -3;
	}

	if ((err = snd_pcm_hw_params_set_channels (_soundDevice, hw_params, 1)) >= 0) {
		this->channels=1;
	} else if ((err = snd_pcm_hw_params_set_channels (_soundDevice, hw_params, 2)) >= 0) {
		this->channels=2;
	} else {
		snd_pcm_close (_soundDevice);
		return -3;
	}

	samplerate=48000;
	if ((err = snd_pcm_hw_params_set_rate_near (_soundDevice, hw_params, &samplerate, 0)) < 0) {
		snd_pcm_close (_soundDevice);
		return -3;
	}

	bufferSize=4096;
	if (snd_pcm_hw_params_set_buffer_size_near(_soundDevice, hw_params, &bufferSize) < 0) {
		fprintf(stderr, "Error setting buffersize.\n");
		return -3;
	}

	this->freq=samplerate;

	err = snd_pcm_hw_params (_soundDevice, hw_params);
	if (err<0) {
		return -3;
	}
	//snd_pcm_hw_params_get_buffer_size( hw_params, &bufferSize );

	this->buffer_len=bufferSize;

	started_sound=0;
	return 0;

	err = snd_pcm_prepare (_soundDevice);
	if (err<0) {
		return -5;
	}
	return 0;
}
#endif

#ifdef D_SOUND_OSS
int LLSound::init_oss() {

	int parameter;
	int parameter2;
	int bytes;
	int retval;
	audio_buf_info bi;

	audio_fd=open("/dev/dsp",O_WRONLY); // open DSP
	if(audio_fd==-1) {
		return (-1);
	}

	this->sign=0;
	this->format=0;
	bytes=1;
	this->channels=1;
	this->freq=22050;
	this->buffer_len=2048/(this->channels*bytes);

	parameter=0x0002000C ; // two buffers with 4096 bytes each one

	if(ioctl(audio_fd,SNDCTL_DSP_SETFRAGMENT, &parameter)==-1)
		return (-6);

	// set format

	if(ioctl(audio_fd,SNDCTL_DSP_GETFMTS, &parameter2)==-1)
		return (-2);
	parameter = 2; // we want mono audio
	if(ioctl(audio_fd,SNDCTL_DSP_CHANNELS, &parameter)==-1)
		return (-4);

	this->channels = parameter;

	// Priority: U8, S8, U16LE, S16LE, U16BE, U16LE

	if(parameter2 & AFMT_S16_BE) {
		parameter = AFMT_S16_BE;
	}
	if(parameter2 & AFMT_U16_BE){
		parameter = AFMT_U16_BE;
	}
	if(parameter2 & AFMT_S16_LE) {
		parameter = AFMT_S16_LE;
	}
	if(parameter2 & AFMT_U16_LE) {
		parameter = AFMT_U16_LE;
	}
	if(parameter2 & AFMT_S8) {
		parameter = AFMT_S8;
	}
	if(parameter2 & AFMT_U8) {
		parameter = AFMT_U8;
	}

	bytes=0; //8 bits

	retval=ioctl(audio_fd,SNDCTL_DSP_SETFMT,&parameter);
	if(retval != 0) {
		return (-3);
	}

	switch(parameter) {
		case AFMT_U8:
	    	this->sign=0;
	    	this->format=0;
	    	bytes=1;
    	break;
		case AFMT_S8:
	    	this->sign=-128;
    		this->format=0;
    		bytes=1;
    	break;
		case AFMT_U16_LE:
		    this->sign=0;
	    	this->format=1;
			bytes=2;
    	break;
		case AFMT_S16_LE:
	    	this->sign=-128;
    		this->format=1;
    		bytes=2;
    	break;
		case AFMT_U16_BE:
	    	this->sign=0;
    		this->format=2;
    		bytes=2;
    	break;
		case AFMT_S16_BE:
		    this->sign=-128;
	    	this->format=2;
	    	bytes=2;
    	break;
	}

	parameter=48000; // we want, by default, 48000 samples per second
	if(ioctl(audio_fd,SNDCTL_DSP_SPEED, &parameter)==-1)
		return (-5);
	this->freq = parameter;

	if(ioctl(audio_fd,SNDCTL_DSP_GETOSPACE, &bi)==-1)
		return (-6);

	parameter=bi.bytes/4;
	this->buffer_len = parameter/(this->channels * bytes);
	return(0);

}
#endif


void LLSound::remove_dc(unsigned char *sound_buffer,int size){
	static unsigned char origsamples[FILTER_SAMPLES];
	int i;
	int j;
	int v;
	int m;
	unsigned char sample;

	m = 0;
	for (i=0;i<size;i++) {
		m += sound_buffer[i];
	}
	m /= size;
	for (i=0;i<size;i++) {
		sample = sound_buffer[i];
		v = (int) sample;
		/*v -= m;
		v += 128;*/
		for(j=0;j<FILTER_SAMPLES;j++) {
			v += (int) origsamples[j];
			if (j != 0) {
				origsamples[j-1] = origsamples[j];
			}
		}
		v /= (FILTER_SAMPLES + 1);
		origsamples[FILTER_SAMPLES - 1] = sample;
		sound_buffer[i] = (unsigned char) v;
	}
}

void LLSound::play() {


	static int retval,retval2;
	this->current_buffer = this->sound;
	//remove_dc(this->current_buffer,this->buffer_len);

	switch(this->sound_type) {
	case SOUND_NO: // no sound
		usleep(75000); // wait 1/20 second
		return;
	break;
#ifdef D_SOUND_OSS
	case SOUND_OSS: // OSS
		retval=write(audio_fd, sound, this->buffer_len * this->increment);
		return;
	break;
#endif
#ifdef D_SOUND_ALSA
	case SOUND_ALSA: // ALSA
		if (started_sound==0) {
			snd_pcm_prepare (_soundDevice);
			started_sound=1;
		}
		retval=snd_pcm_writei(_soundDevice,sound,this->buffer_len);
		if (retval<0) {
			retval=snd_pcm_prepare(_soundDevice);
		}
		return;
	break;
#endif
#ifdef D_SOUND_PULSE
	case SOUND_PULSEAUDIO: // PulseAudio
		// Remove the DC component to avoid losing the sound when multiplexing with other sources
		this->remove_dc(this->sound, this->buffer_len);
		retval=pa_simple_write(pulse_s,this->sound,this->buffer_len,&retval2);
		return;
	break;
#endif
	default:
	break;
	}
}


LLSound::~LLSound() {

	switch(this->sound_type) {
	case SOUND_NO:
	break;
#ifdef D_SOUND_OSS
	case SOUND_OSS:
		close(audio_fd);
	break;
#endif
#ifdef D_SOUND_ALSA
	case SOUND_ALSA:
		snd_pcm_drain (_soundDevice);
		snd_pcm_close (_soundDevice);
	break;
#endif
#ifdef D_SOUND_PULSE
	case SOUND_PULSEAUDIO:
		pa_simple_free(pulse_s);
	break;
#endif
	default:
	break;
	}
}

void LLSound::increase_volume() {

	if (this->volume > 3) {
		this->set_volume (this->volume - 4);
	}
	this->show_volume();
}

void LLSound::decrease_volume() {

	this->set_volume (this->volume + 4);
	this->show_volume();
}

void LLSound::show_volume() {

	std::ostringstream stringStream;
	stringStream << " Volume: " << this->volume/4 << " ";
	osd->set_message(stringStream.str(),1000);

}

void LLSound::set_volume (uint8_t volume) {

	if (volume > 64)
		this->volume = 64;
	else
		this->volume = volume;
}
