/*
 * audio_out_win.c
 * Copyright (C) 2000-2002 Michel Lespinasse <walken@zoy.org>
 * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
 *
 * This file is part of a52dec, a free ATSC A-52 stream decoder.
 * See http://liba52.sourceforge.net/ for updates.
 *
 * a52dec 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.
 *
 * a52dec 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 "config.h"

#ifdef LIBAO_WIN

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <windows.h>
#include <mmsystem.h>
#include <inttypes.h>

#include "a52.h"
#include "audio_out.h"
#include "audio_out_internal.h"

static void win_close (ao_instance_t * _instance);

/* each buffer has a size of 256 samples. Using 40 of them will give us about
 * 1/5 sec of buffered sound for a 48000hz stream */
#define NUMBUF 40

typedef struct win_instance_s {
    ao_instance_t ao;
    HWAVEOUT h_waveout;
    WAVEHDR waveheader[NUMBUF];
    int16_t int16_samples[NUMBUF][256*2];
    int current_buffer;
    int sample_rate;
    int set_params;
    int flags;
} win_instance_t;

static int win_setup (ao_instance_t * _instance, int sample_rate, int * flags,
		      sample_t * level, sample_t * bias)
{
    win_instance_t * instance = (win_instance_t *) _instance;

    if ((instance->set_params == 0) && (instance->sample_rate != sample_rate))
	return 1;
    instance->sample_rate = sample_rate;

    *flags = instance->flags;
    *level = 1;
    *bias = 384;

    return 0;
}

static int win_play (ao_instance_t * _instance, int flags, sample_t * _samples)
{
    win_instance_t * instance = (win_instance_t *) _instance;
    int current_buffer;
    MMRESULT result;

#ifdef LIBA52_DOUBLE
    float samples[256 * 2];
    int i;

    for (i = 0; i < 256 * 2; i++)
	samples[i] = _samples[i];
#else
    float * samples = _samples;
#endif

    flags &= A52_CHANNEL_MASK | A52_LFE;

    if (instance->set_params) {
	WAVEFORMATEX waveformat;
	MMRESULT result;
	int i;

	waveformat.wFormatTag = WAVE_FORMAT_PCM;
	waveformat.nChannels = 2;
	waveformat.nSamplesPerSec = instance->sample_rate;
	waveformat.wBitsPerSample = 16;
	waveformat.nBlockAlign = 4;
	waveformat.nAvgBytesPerSec = 4 * instance->sample_rate;

	result = waveOutOpen (&instance->h_waveout, WAVE_MAPPER, &waveformat,
			      0 /*callback*/, 0 /*data*/, CALLBACK_NULL);
	if (result != MMSYSERR_NOERROR) {
	    fprintf (stderr, "Can not open waveOut device\n");
	    return 1;
	}

	for (i = 0; i < NUMBUF; i++) {
	    instance->waveheader[i].lpData = (LPSTR)instance->int16_samples[i];
	    instance->waveheader[i].dwBufferLength = 256 * 2 * sizeof(int16_t);
	    instance->waveheader[i].dwFlags = WHDR_DONE;
	}

	instance->flags = flags;
	instance->set_params = 0;
    } else if ((flags == A52_DOLBY) && (instance->flags == A52_STEREO)) {
	fprintf (stderr, "Switching from stereo to dolby surround\n");
	instance->flags = A52_DOLBY;
    } else if ((flags == A52_STEREO) && (instance->flags == A52_DOLBY)) {
	fprintf (stderr, "Switching from dolby surround to stereo\n");
	instance->flags = A52_STEREO;
    } else if (flags != instance->flags)
	return 1;

    current_buffer = instance->current_buffer;
    instance->current_buffer = (current_buffer + 1) % NUMBUF;

    while (!(instance->waveheader[current_buffer].dwFlags & WHDR_DONE))
	Sleep (1000 * 256 / instance->sample_rate);

    result = waveOutUnprepareHeader (instance->h_waveout,
				     &instance->waveheader[current_buffer],
				     sizeof(WAVEHDR));
    if (result != MMSYSERR_NOERROR) {
	fprintf (stderr, "waveOutUnprepareHeader failed\n");
	return 1;
    }

    instance->waveheader[current_buffer].dwFlags = 0;
    result = waveOutPrepareHeader (instance->h_waveout,
				   &instance->waveheader[current_buffer],
				   sizeof(WAVEHDR));
    if (result != MMSYSERR_NOERROR) {
	fprintf (stderr, "waveOutPrepareHeader failed\n");
	return 1;
    }

    float2s16_2 (samples, instance->int16_samples[current_buffer]);

    result = waveOutWrite (instance->h_waveout,
			   &instance->waveheader[current_buffer],
			   sizeof(WAVEHDR));
    if (result != MMSYSERR_NOERROR) {
	fprintf (stderr, "waveOutWrite failed\n");
	return 1;
    }

    return 0;
}

static void win_close (ao_instance_t * _instance)
{
    win_instance_t * instance = (win_instance_t *) _instance;
    int i;

    waveOutReset (instance->h_waveout);

    for (i = 0; i < NUMBUF; i++)
	waveOutUnprepareHeader (instance->h_waveout,
				&instance->waveheader[i],
				sizeof(WAVEHDR));
    waveOutClose (instance->h_waveout);
}

static ao_instance_t * win_open (int flags)
{
    win_instance_t * instance;

    instance = malloc (sizeof (win_instance_t));
    if (instance == NULL)
	return NULL;

    instance->ao.setup = win_setup;
    instance->ao.play = win_play;
    instance->ao.close = win_close;

    instance->sample_rate = 0;
    instance->set_params = 1;
    instance->flags = flags;

    return (ao_instance_t *) instance;
}

ao_instance_t * ao_win_open (void)
{
    return win_open (A52_STEREO);
}

ao_instance_t * ao_windolby_open (void)
{
    return win_open (A52_DOLBY);
}

#endif
