/*
 * Copyright (c) 2012 Jan Vesely
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/**
 * @addtogroup audio
 * @brief HelenOS sound server.
 * @{
 */
/** @file
 */

#include <assert.h>
#include <async.h>
#include <errno.h>
#include <loc.h>
#include <str.h>
#include <str_error.h>

#include <audio_pcm_iface.h>

#include "audio_device.h"
#include "log.h"

#define BUFFER_BLOCKS 2

static int device_sink_connection_callback(audio_sink_t *sink);
static int device_source_connection_callback(audio_source_t *source);
static void device_event_callback(ipc_callid_t iid, ipc_call_t *icall, void *arg);
static int get_buffer(audio_device_t *dev);
static int release_buffer(audio_device_t *dev);
static int start_playback(audio_device_t *dev);
static int stop_playback(audio_device_t *dev);
static int start_recording(audio_device_t *dev);
static int stop_recording(audio_device_t *dev);


int audio_device_init(audio_device_t *dev, service_id_t id, const char *name)
{
	assert(dev);
	link_initialize(&dev->link);
	dev->id = id;
	dev->name = str_dup(name);
	dev->sess = loc_service_connect(EXCHANGE_SERIALIZE, id, 0);
	if (!dev->sess) {
		log_debug("Failed to connect to device \"%s\"", name);
		return ENOMEM;
	}

	audio_sink_init(&dev->sink, name, dev, device_sink_connection_callback,
	    &AUDIO_FORMAT_ANY);
	audio_source_init(&dev->source, name, dev,
	    device_source_connection_callback, NULL, &AUDIO_FORMAT_ANY);

	/* Init buffer members */
	fibril_mutex_initialize(&dev->buffer.guard);
	fibril_condvar_initialize(&dev->buffer.wc);
	dev->buffer.id = 0;
	dev->buffer.base = NULL;
	dev->buffer.position = NULL;
	dev->buffer.size = 0;

	log_verbose("Initialized device (%p) '%s' with id %u.",
	    dev, dev->name, dev->id);

	return EOK;
}
void audio_device_fini(audio_device_t *dev)
{
	//TODO implement;
}

static int device_sink_connection_callback(audio_sink_t* sink)
{
	assert(sink);
	audio_device_t *dev = sink->private_data;
	if (list_count(&sink->sources) == 1) {
		int ret = get_buffer(dev);
		if (ret != EOK) {
			log_error("Failed to get device buffer: %s",
			    str_error(ret));
			return ret;
		}
		ret = start_playback(dev);
		if (ret != EOK) {
			log_error("Failed to start playback: %s",
			    str_error(ret));
			release_buffer(dev);
			return ret;
		}
	}
	if (list_count(&sink->sources) == 0) {
		int ret = stop_playback(dev);
		if (ret != EOK) {
			log_error("Failed to start playback: %s",
			    str_error(ret));
			return ret;
		}
		dev->sink.format = AUDIO_FORMAT_ANY;
		ret = release_buffer(dev);
		if (ret != EOK) {
			log_error("Failed to release buffer: %s",
			    str_error(ret));
			return ret;
		}
	}
	return EOK;
}

static int device_source_connection_callback(audio_source_t *source)
{
	assert(source);
	audio_device_t *dev = source->private_data;
	if (source->connected_sink) {
		int ret = get_buffer(dev);
		if (ret != EOK) {
			log_error("Failed to get device buffer: %s",
			    str_error(ret));
			return ret;
		}
		ret = start_recording(dev);
		if (ret != EOK) {
			log_error("Failed to start recording: %s",
			    str_error(ret));
			release_buffer(dev);
			return ret;
		}
	} else { /* Disconnected */
		int ret = stop_recording(dev);
		if (ret != EOK) {
			log_error("Failed to start recording: %s",
			    str_error(ret));
			return ret;
		}
		source->format = AUDIO_FORMAT_ANY;
		ret = release_buffer(dev);
		if (ret != EOK) {
			log_error("Failed to release buffer: %s",
			    str_error(ret));
			return ret;
		}
	}

	return EOK;
}

static void device_event_callback(ipc_callid_t iid, ipc_call_t *icall, void *arg)
{
	/* Answer initial request */
	async_answer_0(iid, EOK);
	audio_device_t *dev = arg;
	assert(dev);
	while (1) {
		ipc_call_t call;
		ipc_callid_t callid = async_get_call(&call);
		async_answer_0(callid, EOK);
		if (IPC_GET_IMETHOD(call) != IPC_FIRST_USER_METHOD) {
			log_debug("Unknown event.\n");
			continue;
		}
		// Assume playback for now
		if (dev->buffer.position) {
			dev->buffer.position += dev->buffer.size / BUFFER_BLOCKS;
		}
		if (!dev->buffer.position ||
		    dev->buffer.position >= dev->buffer.base + dev->buffer.size)
		{
			dev->buffer.position = dev->buffer.base;
		}
		audio_sink_mix_inputs(
		    &dev->sink, dev->buffer.base,
		    dev->buffer.size / BUFFER_BLOCKS);

	}
	//TODO implement
}

static int get_buffer(audio_device_t *dev)
{
	assert(dev);
	if (!dev->sess) {
		log_debug("No connection to device");
		return EIO;
	}
	if (dev->buffer.base) {
		log_debug("We already have a buffer");
		return EBUSY;
	}

	dev->buffer.size = 0;

	async_exch_t *exch = async_exchange_begin(dev->sess);
	const int ret = audio_pcm_get_buffer(exch, &dev->buffer.base,
	    &dev->buffer.size, &dev->buffer.id, device_event_callback, dev);
	async_exchange_end(exch);
	return ret;
}

#define CHECK_BUFFER_AND_CONNECTION() \
do { \
	assert(dev); \
	if (!dev->sess) { \
		log_debug("No connection to device"); \
		return EIO; \
	} \
	if (!dev->buffer.base) { \
		log_debug("We don't have a buffer"); \
		return ENOENT; \
	} \
} while (0)


static int release_buffer(audio_device_t *dev)
{
	CHECK_BUFFER_AND_CONNECTION();

	async_exch_t *exch = async_exchange_begin(dev->sess);
	const int ret = audio_pcm_release_buffer(exch, dev->buffer.id);
	async_exchange_end(exch);
	if (ret == EOK) {
		dev->buffer.base = NULL;
		dev->buffer.size = 0;
		dev->buffer.position = NULL;
	}
	return ret;
}

static int start_playback(audio_device_t *dev)
{
	CHECK_BUFFER_AND_CONNECTION();

	/* Fill the buffer first */
	audio_sink_mix_inputs(&dev->sink, dev->buffer.base, dev->buffer.size);

	async_exch_t *exch = async_exchange_begin(dev->sess);
	const int ret = audio_pcm_start_playback(exch, dev->buffer.id,
	    BUFFER_BLOCKS, dev->sink.format.channels,
	    dev->sink.format.sampling_rate, dev->sink.format.sample_format);
	async_exchange_end(exch);
	return ret;
}

static int stop_playback(audio_device_t *dev)
{
	CHECK_BUFFER_AND_CONNECTION();

	async_exch_t *exch = async_exchange_begin(dev->sess);
	const int ret = audio_pcm_stop_playback(exch, dev->buffer.id);
	async_exchange_end(exch);
	return ret;
}

static int start_recording(audio_device_t *dev)
{
	CHECK_BUFFER_AND_CONNECTION();

	async_exch_t *exch = async_exchange_begin(dev->sess);
	const int ret = audio_pcm_start_record(exch, dev->buffer.id,
	    BUFFER_BLOCKS, dev->sink.format.channels,
	    dev->sink.format.sampling_rate, dev->sink.format.sample_format);
	async_exchange_end(exch);
	return ret;
}

static int stop_recording(audio_device_t *dev)
{
	CHECK_BUFFER_AND_CONNECTION();

	async_exch_t *exch = async_exchange_begin(dev->sess);
	const int ret = audio_pcm_stop_record(exch, dev->buffer.id);
	async_exchange_end(exch);
	return ret;
}

/**
 * @}
 */