/*
 * Copyright (C) 2012 Paul Kocialkowski <contact@paulk.fr>
 * 
 * This is based on Galaxy Nexus audio.primary.tuna implementation:
 * Copyright 2011, The Android Open-Source 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 3 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, see <http://www.gnu.org/licenses/>.
 */

#define LOG_TAG "TinyALSA-Audio Input"

#include <stdlib.h>
#include <errno.h>
#include <pthread.h>
#include <stdint.h>
#include <sys/time.h>

#include <cutils/str_parms.h>
#include <cutils/log.h>

#define EFFECT_UUID_NULL EFFECT_UUID_NULL_IN
#define EFFECT_UUID_NULL_STR EFFECT_UUID_NULL_STR_IN
#include "audio_hw.h"
#include "mixer.h"

/*
 * Functions
 */

int audio_in_pcm_open(struct tinyalsa_audio_stream_in *stream_in)
{
	struct pcm *pcm = NULL;
	struct pcm_config pcm_config;

	if(stream_in == NULL)
		return -1;

	memset(&pcm_config, 0, sizeof(pcm_config));
	pcm_config.channels = popcount(stream_in->mixer_props->channels);
	pcm_config.rate = stream_in->mixer_props->rate;
	switch(stream_in->mixer_props->format) {
		case AUDIO_FORMAT_PCM_16_BIT:
			pcm_config.format = PCM_FORMAT_S16_LE;
			break;
		case AUDIO_FORMAT_PCM_32_BIT:
			pcm_config.format = PCM_FORMAT_S32_LE;
			break;
		default:
			LOGE("Invalid format: 0x%x", stream_in->mixer_props->format);
			return -1;
	}
	pcm_config.period_size = stream_in->mixer_props->period_size;
	pcm_config.period_count = stream_in->mixer_props->period_count;

	pcm = pcm_open(stream_in->mixer_props->card,
		stream_in->mixer_props->device, PCM_IN, &pcm_config);

	if(pcm == NULL || !pcm_is_ready(pcm)) {
		LOGE("Unable to open pcm device: %s", pcm_get_error(pcm));
		return -1;
	}

	stream_in->pcm = pcm;

	return 0;
}

void audio_in_pcm_close(struct tinyalsa_audio_stream_in *stream_in)
{
	if(stream_in->pcm == NULL)
		return;

	pcm_close(stream_in->pcm);
	stream_in->pcm = NULL;
}

int audio_in_set_route(struct tinyalsa_audio_stream_in *stream_in,
	audio_devices_t device)
{
	int rc;

	if(stream_in == NULL)
		return -1;

	stream_in->device_current = device;

	if(device == 0)
		return stream_in->stream.common.standby((struct audio_stream *) stream_in);


	tinyalsa_mixer_set_device(stream_in->device->mixer, stream_in->device_current);

#ifdef YAMAHA_MC1N2_AUDIO
	yamaha_mc1n2_audio_set_route(stream_in->device->mc1n2_pdata, device);
#endif

	return 0;
}

static uint32_t audio_in_get_sample_rate(const struct audio_stream *stream)
{
	struct tinyalsa_audio_stream_in *stream_in;

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	return stream_in->rate;
}

static int audio_in_set_sample_rate(struct audio_stream *stream, uint32_t rate)
{
	struct tinyalsa_audio_stream_in *stream_in;

	LOGD("%s(%p, %d)", __func__, stream, rate);

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	// FIXME: If rate is different, change resampler
	stream_in->rate = rate;

	return 0;
}

static size_t audio_in_get_buffer_size(const struct audio_stream *stream)
{
	struct tinyalsa_audio_stream_in *stream_in;
	size_t size;

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	size = (stream_in->mixer_props->period_size * stream_in->rate) /
		stream_in->rate;
	size = ((size + 15) / 16) * 16;
	size = size * audio_stream_frame_size((struct audio_stream *) stream);

	return size;
}

static uint32_t audio_in_get_channels(const struct audio_stream *stream)
{
	struct tinyalsa_audio_stream_in *stream_in;

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	return stream_in->channels;
}

static int audio_in_get_format(const struct audio_stream *stream)
{
	struct tinyalsa_audio_stream_in *stream_in;

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	return stream_in->format;
}

static int audio_in_set_format(struct audio_stream *stream, int format)
{
	struct tinyalsa_audio_stream_in *stream_in;

	LOGD("%s(%p, %d)", __func__, stream, format);

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	// FIXME: If format is different, change resampler
	stream_in->format = format;

	return 0;
}

static int audio_in_standby(struct audio_stream *stream)
{
	struct tinyalsa_audio_stream_in *stream_in;
	int rc;

	LOGD("%s(%p)", __func__, stream);

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	if(stream_in->pcm != NULL)
		audio_in_pcm_close(stream_in);

#ifdef YAMAHA_MC1N2_AUDIO
	if(!stream_in->standby) {
		rc = yamaha_mc1n2_audio_input_stop(stream_in->device->mc1n2_pdata);
		if(rc < 0) {
			LOGE("Failed to set Yamaha-MC1N2-Audio route");
		}
	}
#endif

	stream_in->standby = 1;

	return 0;
}

static int audio_in_dump(const struct audio_stream *stream, int fd)
{
	LOGD("%s(%p, %d)", __func__, stream, fd);

	return 0;
}

static int audio_in_set_parameters(struct audio_stream *stream, const char *kvpairs)
{
	struct tinyalsa_audio_stream_in *stream_in;
	struct str_parms *parms;
	char value_string[32] = { 0 };
	int value;
	int rc;

	LOGD("%s(%p, %s)", __func__, stream, kvpairs);

	if(stream == NULL || kvpairs == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	if(stream_in->device->mixer == NULL)
		return -1;

	parms = str_parms_create_str(kvpairs);
	if(parms == NULL)
		return -1;

	rc = str_parms_get_str(parms, AUDIO_PARAMETER_STREAM_ROUTING, value_string, sizeof(value_string));
	if(rc < 0)
		goto error_params;

	value = atoi(value_string);

	if(stream_in->device_current != (audio_devices_t) value)
		audio_in_set_route(stream_in, (audio_devices_t) value);

	str_parms_destroy(parms);

	return 0;

error_params:
	str_parms_destroy(parms);

	return -1;
}

static char *audio_in_get_parameters(const struct audio_stream *stream,
	const char *keys)
{
	LOGD("%s(%p, %s)", __func__, stream, keys);

	return strdup("");
}

static int audio_in_set_gain(struct audio_stream_in *stream, float gain)
{
	struct tinyalsa_audio_stream_in *stream_in;

	LOGD("%s(%p, %f)", __func__, stream, gain);

	if(stream == NULL)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	if(stream_in->device == NULL || stream_in->device->mixer == NULL)
		return -1;

	tinyalsa_mixer_set_input_gain(stream_in->device->mixer,
		stream_in->device_current, gain);

	return 0;
}

static ssize_t audio_in_read(struct audio_stream_in *stream,
	void *buffer, size_t bytes)
{
	struct tinyalsa_audio_stream_in *stream_in;
	int rc;

	if(stream == NULL || buffer == NULL || bytes <= 0)
		return -1;

	stream_in = (struct tinyalsa_audio_stream_in *) stream;

	if(stream_in->standby) {
#ifdef YAMAHA_MC1N2_AUDIO
		rc = yamaha_mc1n2_audio_input_start(stream_in->device->mc1n2_pdata);
		if(rc < 0) {
			LOGE("Failed to set Yamaha-MC1N2-Audio route");
		}
#endif

		rc = audio_in_pcm_open(stream_in);
		if(rc < 0) {
			LOGE("Unable to open pcm device");
			return -1;
		}

		stream_in->standby = 0;
	}

	if(stream_in->pcm == NULL || !pcm_is_ready(stream_in->pcm)) {
		LOGE("pcm device is not ready");
		return -1;
	}

	rc = pcm_read(stream_in->pcm, (void *) buffer, (int) bytes);
	if(rc != 0) {
		LOGE("pcm read failed!");
		return -1;
	}

	if(stream_in->device != NULL && stream_in->device->mic_mute)
		memset(buffer, 0, bytes);

	return bytes;
}

static uint32_t audio_in_get_input_frames_lost(struct audio_stream_in *stream)
{
	return 0;
}

static int audio_in_add_audio_effect(const struct audio_stream *stream, effect_handle_t effect)
{
	LOGD("%s(%p, %p)", __func__, stream, effect);

	return 0;
}

static int audio_in_remove_audio_effect(const struct audio_stream *stream, effect_handle_t effect)
{
	LOGD("%s(%p, %p)", __func__, stream, effect);

	return 0;
}

/*
 * Interface
 */

void audio_hw_close_input_stream(struct audio_hw_device *dev,
	struct audio_stream_in *stream)
{
	struct tinyalsa_audio_device *tinyalsa_audio_device;

	LOGD("%s(%p)", __func__, stream);

	if(stream != NULL)
		free(stream);

	if(dev == NULL)
		return;

	tinyalsa_audio_device = (struct tinyalsa_audio_device *) dev;

	tinyalsa_mixer_set_input_state(tinyalsa_audio_device->mixer, 0);

	tinyalsa_audio_device->stream_in = NULL;
}

int audio_hw_open_input_stream(struct audio_hw_device *dev,
	uint32_t devices, int *format, uint32_t *channels, uint32_t *sample_rate,
	audio_in_acoustics_t acoustics, struct audio_stream_in **stream_in)
{
	struct tinyalsa_audio_device *tinyalsa_audio_device;
	struct tinyalsa_audio_stream_in *tinyalsa_audio_stream_in;
	struct audio_stream_in *stream;
	int rc;

	LOGD("%s(%p, %d, %p, %p, %p, %d, %p)",
		__func__, dev, devices, format, channels, sample_rate, acoustics, stream_in);

	if(dev == NULL)
		return -EINVAL;

	tinyalsa_audio_device = (struct tinyalsa_audio_device *) dev;
	tinyalsa_audio_stream_in = calloc(1, sizeof(struct tinyalsa_audio_stream_in));

	if(tinyalsa_audio_stream_in == NULL)
		return -ENOMEM;

	tinyalsa_audio_stream_in->device = tinyalsa_audio_device;
	stream = &(tinyalsa_audio_stream_in->stream);

	stream->common.get_sample_rate = audio_in_get_sample_rate;
	stream->common.set_sample_rate = audio_in_set_sample_rate;
	stream->common.get_buffer_size = audio_in_get_buffer_size;
	stream->common.get_channels = audio_in_get_channels;
	stream->common.get_format = audio_in_get_format;
	stream->common.set_format = audio_in_set_format;
	stream->common.standby = audio_in_standby;
	stream->common.dump = audio_in_dump;
	stream->common.set_parameters = audio_in_set_parameters;
	stream->common.get_parameters = audio_in_get_parameters;
	stream->common.add_audio_effect = audio_in_add_audio_effect;
	stream->common.remove_audio_effect = audio_in_remove_audio_effect;
	stream->set_gain = audio_in_set_gain;
	stream->read = audio_in_read;
	stream->get_input_frames_lost = audio_in_get_input_frames_lost;

	if(tinyalsa_audio_device->mixer == NULL)
		goto error_stream;

	tinyalsa_audio_stream_in->mixer_props =
		tinyalsa_mixer_get_input_props(tinyalsa_audio_device->mixer);

	if(tinyalsa_audio_stream_in->mixer_props == NULL)
		goto error_stream;

	tinyalsa_audio_stream_in->rate =
		tinyalsa_audio_stream_in->mixer_props->rate;
	tinyalsa_audio_stream_in->channels =
		tinyalsa_audio_stream_in->mixer_props->channels;
	tinyalsa_audio_stream_in->format =
		tinyalsa_audio_stream_in->mixer_props->format;

	*sample_rate = (uint32_t) tinyalsa_audio_stream_in->rate;
	*channels = (uint32_t) tinyalsa_audio_stream_in->channels;
	*format = (uint32_t) tinyalsa_audio_stream_in->format;

	rc = tinyalsa_mixer_set_input_state(tinyalsa_audio_device->mixer, 1);
	if(rc < 0) {
		LOGE("Unable to set input state");
		goto error_stream;
	}

	audio_in_set_route(tinyalsa_audio_stream_in, devices);

	rc = audio_in_pcm_open(tinyalsa_audio_stream_in);
	if(rc < 0) {
		LOGE("Unable to open pcm device");
		goto error_stream;
	}

	audio_in_pcm_close(tinyalsa_audio_stream_in);

	tinyalsa_audio_stream_in->standby = 1;

	*stream_in = stream;

	return 0;

error_stream:
	*stream_in = NULL;

	return -1;
}