/*
 * Copyright (C) 2009 Google, Inc.
 * Copyright (C) 2009 HTC Corporation
 * Copyright (c) 2010, The Linux Foundation. All rights reserved.
 *
 * This software is licensed under the terms of the GNU General Public
 * License version 2, as published by the Free Software Foundation, and
 * may be copied, distributed, and modified under those terms.
 *
 * 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.
 *
 */

#include <linux/fs.h>
#include <linux/module.h>
#include <linux/miscdevice.h>
#include <linux/mutex.h>
#include <linux/sched.h>
#include <linux/wait.h>
#include <linux/uaccess.h>

#include <linux/msm_audio_amrnb.h>
#include <mach/msm_qdsp6_audiov2.h>
#include "dal_audio.h"
#include "dal_audio_format.h"
#include <mach/debug_mm.h>


struct amrnb {
	struct mutex lock;
	struct msm_audio_amrnb_enc_config_v2 cfg;
	struct msm_audio_stream_config str_cfg;
	struct audio_client *audio_client;
};


static long q6_amrnb_in_ioctl(struct file *file, unsigned int cmd,
				unsigned long arg)
{
	struct amrnb *amrnb = file->private_data;
	struct adsp_open_command rpc;
	int rc = 0;

	if (cmd == AUDIO_GET_STATS) {
		struct msm_audio_stats stats;
		memset(&stats, 0, sizeof(stats));
		if (copy_to_user((void *) arg, &stats, sizeof(stats)))
			return -EFAULT;
		return 0;
	}

	mutex_lock(&amrnb->lock);
	switch (cmd) {
	case AUDIO_START:
		if (amrnb->audio_client) {
			rc = -EBUSY;
			break;
		} else {
			amrnb->audio_client = q6audio_open(AUDIO_FLAG_READ,
						amrnb->str_cfg.buffer_size);

			if (!amrnb->audio_client) {
				kfree(amrnb);
				rc = -ENOMEM;
				break;
			}
		}

		tx_clk_freq = 8000;

		memset(&rpc, 0, sizeof(rpc));

		rpc.format_block.standard.format = ADSP_AUDIO_FORMAT_AMRNB_FS;
		rpc.format_block.standard.channels = 1;
		rpc.format_block.standard.bits_per_sample = 16;
		rpc.format_block.standard.sampling_rate = 8000;
		rpc.format_block.standard.is_signed = 1;
		rpc.format_block.standard.is_interleaved = 0;

		rpc.hdr.opcode = ADSP_AUDIO_IOCTL_CMD_OPEN_READ;
		rpc.device = ADSP_AUDIO_DEVICE_ID_DEFAULT;
		rpc.stream_context = ADSP_AUDIO_DEVICE_CONTEXT_RECORD;
		rpc.buf_max_size = amrnb->str_cfg.buffer_size;
		rpc.config.amr.mode = amrnb->cfg.band_mode;
		rpc.config.amr.dtx_mode = amrnb->cfg.dtx_enable;
		rpc.config.amr.enable = 1;
		q6audio_start(amrnb->audio_client, &rpc, sizeof(rpc));
		break;
	case AUDIO_STOP:
		break;
	case AUDIO_FLUSH:
		break;
	case AUDIO_SET_VOLUME:
		break;
	case AUDIO_GET_STREAM_CONFIG:
		if (copy_to_user((void *)arg, &amrnb->str_cfg,
			sizeof(struct msm_audio_stream_config)))
			rc = -EFAULT;
		break;
	case AUDIO_SET_STREAM_CONFIG:
		if (copy_from_user(&amrnb->str_cfg, (void *)arg,
			sizeof(struct msm_audio_stream_config))) {
			rc = -EFAULT;
			break;
		}

		if (amrnb->str_cfg.buffer_size < 768) {
			pr_err("[%s:%s] Buffer size too small\n", __MM_FILE__,
					__func__);
			rc = -EINVAL;
			break;
		}

		if (amrnb->str_cfg.buffer_count != 2)
			pr_info("[%s:%s] Buffer count set to 2\n", __MM_FILE__,
					__func__);
		break;
	case AUDIO_SET_AMRNB_ENC_CONFIG:
		if (copy_from_user(&amrnb->cfg, (void *) arg,
			sizeof(struct msm_audio_amrnb_enc_config_v2)))
			rc = -EFAULT;
		break;
	case AUDIO_GET_AMRNB_ENC_CONFIG:
		if (copy_to_user((void *) arg, &amrnb->cfg,
				 sizeof(struct msm_audio_amrnb_enc_config_v2)))
			rc = -EFAULT;
		break;

	default:
		rc = -EINVAL;
	}

	mutex_unlock(&amrnb->lock);
	return rc;
}

static int q6_amrnb_in_open(struct inode *inode, struct file *file)
{
	struct amrnb *amrnb;
	amrnb = kmalloc(sizeof(struct amrnb), GFP_KERNEL);
	if (amrnb == NULL) {
		pr_err("[%s:%s] Could not allocate memory for amrnb driver\n",
				__MM_FILE__, __func__);
		return -ENOMEM;
	}

	mutex_init(&amrnb->lock);
	file->private_data = amrnb;
	amrnb->audio_client = NULL;
	amrnb->str_cfg.buffer_size = 768;
	amrnb->str_cfg.buffer_count = 2;
	amrnb->cfg.band_mode = ADSP_AUDIO_AMR_MR475;
	amrnb->cfg.dtx_enable  = ADSP_AUDIO_AMR_DTX_MODE_ON_AUTO;
	amrnb->cfg.frame_format  = ADSP_AUDIO_FORMAT_AMRNB_FS;
	return 0;
}

static ssize_t q6_amrnb_in_read(struct file *file, char __user *buf,
			   size_t count, loff_t *pos)
{
	struct audio_client *ac;
	struct audio_buffer *ab;
	const char __user *start = buf;
	struct amrnb *amrnb = file->private_data;
	int xfer = 0;
	int res;

	mutex_lock(&amrnb->lock);
	ac = amrnb->audio_client;
	if (!ac) {
		res = -ENODEV;
		goto fail;
	}
	while (count > xfer) {
		ab = ac->buf + ac->cpu_buf;

		if (ab->used)
			wait_event(ac->wait, (ab->used == 0));

		xfer = ab->actual_size;

		if (copy_to_user(buf, ab->data, xfer)) {
			res = -EFAULT;
			goto fail;
		}

		buf += xfer;
		count -= xfer;

		ab->used = 1;
		q6audio_read(ac, ab);
		ac->cpu_buf ^= 1;
	}

	res = buf - start;
fail:
	mutex_unlock(&amrnb->lock);

	return res;
}

static int q6_amrnb_in_release(struct inode *inode, struct file *file)
{
	int rc = 0;
	struct amrnb *amrnb = file->private_data;

	mutex_lock(&amrnb->lock);
	if (amrnb->audio_client)
		rc = q6audio_close(amrnb->audio_client);
	mutex_unlock(&amrnb->lock);
	kfree(amrnb);
	return rc;
}

static const struct file_operations q6_amrnb_in_fops = {
	.owner		= THIS_MODULE,
	.open		= q6_amrnb_in_open,
	.read		= q6_amrnb_in_read,
	.release	= q6_amrnb_in_release,
	.unlocked_ioctl	= q6_amrnb_in_ioctl,
};

struct miscdevice q6_amrnb_in_misc = {
	.minor	= MISC_DYNAMIC_MINOR,
	.name	= "msm_amr_in",
	.fops	= &q6_amrnb_in_fops,
};

static int __init q6_amrnb_in_init(void)
{
	return misc_register(&q6_amrnb_in_misc);
}

device_initcall(q6_amrnb_in_init);