/* arch/arm/mach-rockchip/rk28_headset.c
 *
 * Copyright (C) 2009 Rockchip Electronics Co., Ltd.
 *
 * 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/module.h>
#include <linux/device.h>
#include <linux/fs.h>
#include <linux/interrupt.h>
#include <linux/workqueue.h>
#include <linux/irq.h>
#include <linux/delay.h>
#include <linux/types.h>
#include <linux/input.h>
#include <linux/platform_device.h>
#include <linux/mutex.h>
#include <linux/errno.h>
#include <linux/err.h>
#include <linux/hrtimer.h>
#include <linux/extcon-provider.h>
#include <linux/debugfs.h>
#include <linux/wakelock.h>
#include <linux/gpio.h>
#include <linux/of_gpio.h>
#include <asm/atomic.h>
#include <linux/pm.h>
#include <linux/i2c.h>
#include <linux/spi/spi.h>
#include "rk_headset.h"
#ifdef CONFIG_HAS_EARLYSUSPEND
#include <linux/earlysuspend.h>
#endif

/* Debug */
#if 0
#define DBG(x...) printk(x)
#else
#define DBG(x...) do { } while (0)
#endif

#define BIT_HEADSET             BIT(0)
#define BIT_HEADSET_NO_MIC      BIT(1)

#define HEADSET 0
#define HOOK 1

#define HEADSET_IN 1
#define HEADSET_OUT 0
#define HOOK_DOWN 1
#define HOOK_UP 0
#define enable 1
#define disable 0

#if defined(CONFIG_SND_RK_SOC_RK2928) || defined(CONFIG_SND_RK29_SOC_RK610)
extern void rk2928_codec_set_spk(bool on);
#endif
#ifdef CONFIG_SND_SOC_WM8994
extern int wm8994_set_status(void);
#endif

/* headset private data */
struct headset_priv {
	struct input_dev *input_dev;
	struct rk_headset_pdata *pdata;
	unsigned int headset_status : 1;
	unsigned int hook_status : 1;
	unsigned int isMic : 1;
	unsigned int isHook_irq : 1;
	int cur_headset_status;
	unsigned int irq[2];
	unsigned int irq_type[2];
	struct delayed_work h_delayed_work[2];
	struct extcon_dev *edev;
	struct mutex mutex_lock[2];
	struct timer_list headset_timer;
	unsigned char *keycodes;
};

static struct headset_priv *headset_info;

#ifdef CONFIG_MODEM_MIC_SWITCH
#define HP_MIC 0
#define MAIN_MIC 1

void Modem_Mic_switch(int value)
{
	if (value == HP_MIC)
		gpio_set_value(headset_info->pdata->mic_switch_gpio,
			       headset_info->pdata->hp_mic_io_value);
	else if (value == MAIN_MIC)
		gpio_set_value(headset_info->pdata->mic_switch_gpio,
			       headset_info->pdata->main_mic_io_value);
}

void Modem_Mic_release(void)
{
	if (headset_info->cur_headset_status == 1)
		gpio_set_value(headset_info->pdata->mic_switch_gpio,
			       headset_info->pdata->hp_mic_io_value);
	else
		gpio_set_value(headset_info->pdata->mic_switch_gpio,
			       headset_info->pdata->main_mic_io_value);
}
#endif

static int read_gpio(int gpio)
{
	int i, level;

	for (i = 0; i < 3; i++) {
		level = gpio_get_value(gpio);
		if (level < 0) {
			pr_warn("%s:get pin level again,pin=%d,i=%d\n",
				__func__, gpio, i);
			msleep(1);
			continue;
		} else
			break;
	}
	if (level < 0)
		pr_err("%s:get pin level  err!\n", __func__);
	return level;
}

static irqreturn_t headset_interrupt(int irq, void *dev_id)
{
	DBG("---headset_interrupt---\n");
	schedule_delayed_work(&headset_info->h_delayed_work[HEADSET],
			      msecs_to_jiffies(50));
	return IRQ_HANDLED;
}

static irqreturn_t hook_interrupt(int irq, void *dev_id)
{
	DBG("---Hook_interrupt---\n");
	/* disable_irq_nosync(headset_info->irq[HOOK]); */
	schedule_delayed_work(&headset_info->h_delayed_work[HOOK],
			      msecs_to_jiffies(100));
	return IRQ_HANDLED;
}

static void headsetobserve_work(struct work_struct *work)
{
	int level = 0;
	int level2 = 0;
	struct rk_headset_pdata *pdata = headset_info->pdata;
	static unsigned int old_status = 0;

	printk("---headsetobserve_work---\n");
	mutex_lock(&headset_info->mutex_lock[HEADSET]);
	level = read_gpio(pdata->headset_gpio);
	if (level < 0)
		goto out;
	msleep(100);
	level2 = read_gpio(pdata->headset_gpio);
	if (level2 < 0)
		goto out;
	if (level2 != level)
		goto out;
	old_status = headset_info->headset_status;
	if (pdata->headset_insert_type == HEADSET_IN_HIGH)
		headset_info->headset_status = level ? HEADSET_IN : HEADSET_OUT;
	else
		headset_info->headset_status = level ? HEADSET_OUT : HEADSET_IN;

	if (old_status == headset_info->headset_status) {
		pr_warn("old_status == headset_info->headset_status\n");
		goto out;
	}
	DBG("(headset in is %s)headset status is %s\n",
	    pdata->headset_insert_type ? "high level" : "low level",
	    headset_info->headset_status ? "in" : "out");
	if (headset_info->headset_status == HEADSET_IN) {
		headset_info->cur_headset_status = BIT_HEADSET_NO_MIC;
		if (pdata->headset_insert_type == HEADSET_IN_HIGH)
			irq_set_irq_type(headset_info->irq[HEADSET],
					 IRQF_TRIGGER_FALLING);
		else
			irq_set_irq_type(headset_info->irq[HEADSET],
					 IRQF_TRIGGER_RISING);
		if (pdata->hook_gpio) {
			/* Start the timer, wait for switch to the headphone channel */
			del_timer(&headset_info->headset_timer);
			headset_info->headset_timer.expires = jiffies + 100;
			add_timer(&headset_info->headset_timer);
			goto out;
		}
	} else if (headset_info->headset_status == HEADSET_OUT) {
		headset_info->hook_status = HOOK_UP;
		if (headset_info->isHook_irq == enable) {
			DBG("disable headset_hook irq\n");
			headset_info->isHook_irq = disable;
			disable_irq(headset_info->irq[HOOK]);
		}
		headset_info->cur_headset_status = 0;
		if (pdata->headset_insert_type == HEADSET_IN_HIGH)
			irq_set_irq_type(headset_info->irq[HEADSET],
					 IRQF_TRIGGER_RISING);
		else
			irq_set_irq_type(headset_info->irq[HEADSET],
					 IRQF_TRIGGER_FALLING);
	}
	if (headset_info->cur_headset_status)
		extcon_set_state_sync(headset_info->edev, EXTCON_JACK_HEADPHONE,
				      true);
	else
		extcon_set_state_sync(headset_info->edev, EXTCON_JACK_HEADPHONE,
				      false);
	DBG("headset_info->cur_headset_status = %d\n",
	    headset_info->cur_headset_status);
out:
	mutex_unlock(&headset_info->mutex_lock[HEADSET]);
}

static void hook_work(struct work_struct *work)
{
	int level = 0;
	struct rk_headset_pdata *pdata = headset_info->pdata;
	static unsigned int old_status = HOOK_UP;

	mutex_lock(&headset_info->mutex_lock[HOOK]);
	if (headset_info->headset_status == HEADSET_OUT) {
		DBG("Headset is out\n");
		goto RE_ERROR;
	}
	level = read_gpio(pdata->hook_gpio);
	if (level < 0)
		goto RE_ERROR;
	old_status = headset_info->hook_status;
	DBG("Hook_work -- level = %d\n", level);
	if (level == 0)
		headset_info->hook_status =
			pdata->hook_down_type == HOOK_DOWN_HIGH ? HOOK_UP :
								  HOOK_DOWN;
	else if (level > 0)
		headset_info->hook_status =
			pdata->hook_down_type == HOOK_DOWN_HIGH ? HOOK_DOWN :
								  HOOK_UP;
	if (old_status == headset_info->hook_status) {
		DBG("old_status == headset_info->hook_status\n");
		goto RE_ERROR;
	}
	DBG("Hook_work -- level = %d  hook status is %s\n", level,
	    headset_info->hook_status ? "key down" : "key up");
	if (headset_info->hook_status == HOOK_DOWN) {
		if (pdata->hook_down_type == HOOK_DOWN_HIGH)
			irq_set_irq_type(headset_info->irq[HOOK],
					 IRQF_TRIGGER_FALLING);
		else
			irq_set_irq_type(headset_info->irq[HOOK],
					 IRQF_TRIGGER_RISING);
	} else {
		if (pdata->hook_down_type == HOOK_DOWN_HIGH)
			irq_set_irq_type(headset_info->irq[HOOK],
					 IRQF_TRIGGER_RISING);
		else
			irq_set_irq_type(headset_info->irq[HOOK],
					 IRQF_TRIGGER_FALLING);
	}
	input_report_key(headset_info->input_dev, HOOK_KEY_CODE,
			 headset_info->hook_status);
	input_sync(headset_info->input_dev);
RE_ERROR:
	mutex_unlock(&headset_info->mutex_lock[HOOK]);
}

static void headset_timer_callback(struct timer_list *t)
{
	struct headset_priv *headset = from_timer(headset, t, headset_timer);
	struct rk_headset_pdata *pdata = headset->pdata;
	int level = 0;

	pr_info("headset_timer_callback, headset->headset_status %d\n",
		headset->headset_status);
	if (headset->headset_status == HEADSET_OUT) {
		pr_info("Headset is out\n");
		goto out;
	}
	level = read_gpio(pdata->hook_gpio);
	if (level < 0)
		goto out;
	if ((level > 0 && pdata->hook_down_type == HOOK_DOWN_LOW) ||
	    (level == 0 && pdata->hook_down_type == HOOK_DOWN_HIGH)) {
		headset->isMic = 1;
		enable_irq(headset_info->irq[HOOK]);
		headset->isHook_irq = enable;
		headset_info->hook_status = HOOK_UP;
		if (pdata->hook_down_type == HOOK_DOWN_HIGH)
			irq_set_irq_type(headset_info->irq[HOOK],
					 IRQF_TRIGGER_RISING);
		else
			irq_set_irq_type(headset_info->irq[HOOK],
					 IRQF_TRIGGER_FALLING);
	} else {
		headset->isMic = 0;
	}
	pr_info("headset->isMic = %d\n", headset->isMic);
	headset_info->cur_headset_status = headset_info->isMic ? 1 : 2;
	if (headset->isMic == 1)
		extcon_set_state_sync(headset_info->edev,
				      EXTCON_JACK_MICROPHONE, true);
	else
		extcon_set_state_sync(headset_info->edev,
				      EXTCON_JACK_MICROPHONE, false);
	DBG("headset_info->cur_headset_status = %d\n",
	    headset_info->cur_headset_status);
out:
	return;
}

#ifdef CONFIG_HAS_EARLYSUSPEND
static void headset_early_resume(struct early_suspend *h)
{
	schedule_delayed_work(&headset_info->h_delayed_work[HEADSET],
			      msecs_to_jiffies(10));
	//DBG(">>>>>headset_early_resume\n");
}

static struct early_suspend hs_early_suspend;
#endif

static int rk_hskey_open(struct input_dev *dev)
{
	//struct rk28_adckey *adckey = input_get_drvdata(dev);
	//DBG("===========rk_Hskey_open===========\n");
	return 0;
}

static void rk_hskey_close(struct input_dev *dev)
{
	//DBG("===========rk_Hskey_close===========\n");
	//struct rk28_adckey *adckey = input_get_drvdata(dev);
}

static const unsigned int headset_cable[] = {
	EXTCON_JACK_MICROPHONE,
	EXTCON_JACK_HEADPHONE,
	EXTCON_NONE,
};

int rk_headset_probe(struct platform_device *pdev,
		     struct rk_headset_pdata *pdata)
{
	int ret = 0;
	struct headset_priv *headset;

	headset = devm_kzalloc(&pdev->dev, sizeof(*headset), GFP_KERNEL);
	if (!headset) {
		dev_err(&pdev->dev, "failed to allocate driver data\n");
		return -ENOMEM;
	}
	headset_info = headset;
	headset->pdata = pdata;
	headset->headset_status = HEADSET_OUT;
	headset->hook_status = HOOK_UP;
	headset->isHook_irq = disable;
	headset->cur_headset_status = 0;
	headset->edev = devm_extcon_dev_allocate(&pdev->dev, headset_cable);
	if (IS_ERR(headset->edev)) {
		dev_err(&pdev->dev, "failed to allocate extcon device\n");
		ret = -ENOMEM;
		goto failed;
	}
	ret = devm_extcon_dev_register(&pdev->dev, headset->edev);
	if (ret < 0) {
		dev_err(&pdev->dev, "extcon_dev_register() failed: %d\n", ret);
		goto failed;
	}
	mutex_init(&headset->mutex_lock[HEADSET]);
	mutex_init(&headset->mutex_lock[HOOK]);
	INIT_DELAYED_WORK(&headset->h_delayed_work[HEADSET],
			  headsetobserve_work);
	INIT_DELAYED_WORK(&headset->h_delayed_work[HOOK], hook_work);
	headset->isMic = 0;
	timer_setup(&headset->headset_timer, headset_timer_callback, 0);
	/* Create and register the input driver */
	headset->input_dev = devm_input_allocate_device(&pdev->dev);
	if (!headset->input_dev) {
		dev_err(&pdev->dev, "failed to allocate input device\n");
		ret = -ENOMEM;
		goto failed;
	}
	headset->input_dev->name = pdev->name;
	headset->input_dev->open = rk_hskey_open;
	headset->input_dev->close = rk_hskey_close;
	headset->input_dev->dev.parent = &pdev->dev;
	/* input_dev->phys = KEY_PHYS_NAME; */
	headset->input_dev->id.vendor = 0x0001;
	headset->input_dev->id.product = 0x0001;
	headset->input_dev->id.version = 0x0100;
	/* Register the input device */
	ret = input_register_device(headset->input_dev);
	if (ret) {
		dev_err(&pdev->dev, "failed to register input device\n");
		goto failed;
	}
	input_set_capability(headset->input_dev, EV_KEY, HOOK_KEY_CODE);
#ifdef CONFIG_HAS_EARLYSUSPEND
	hs_early_suspend.suspend = NULL;
	hs_early_suspend.resume = headset_early_resume;
	hs_early_suspend.level = ~0x0;
	register_early_suspend(&hs_early_suspend);
#endif
	if (pdata->headset_gpio) {
		headset->irq[HEADSET] = gpio_to_irq(pdata->headset_gpio);
		if (pdata->headset_insert_type == HEADSET_IN_HIGH)
			headset->irq_type[HEADSET] = IRQF_TRIGGER_RISING;
		else
			headset->irq_type[HEADSET] = IRQF_TRIGGER_FALLING;
		ret = devm_request_irq(&pdev->dev, headset->irq[HEADSET],
				       headset_interrupt,
				       headset->irq_type[HEADSET],
				       "headset_input", NULL);
		if (ret)
			goto failed;
		if (pdata->headset_wakeup)
			enable_irq_wake(headset->irq[HEADSET]);
	} else {
		dev_err(&pdev->dev, "failed init headset, please full headset_gpio function in board\n");
		ret = -EEXIST;
		goto failed;
	}
	if (pdata->hook_gpio) {
		headset->irq[HOOK] = gpio_to_irq(pdata->hook_gpio);
		headset->irq_type[HOOK] =
			pdata->hook_down_type == HOOK_DOWN_HIGH ?
				IRQF_TRIGGER_RISING :
				IRQF_TRIGGER_FALLING;
		ret = devm_request_irq(&pdev->dev, headset->irq[HOOK],
				       hook_interrupt, headset->irq_type[HOOK],
				       "headset_hook", NULL);
		if (ret)
			goto failed;
		disable_irq(headset->irq[HOOK]);
	}
	schedule_delayed_work(&headset->h_delayed_work[HEADSET],
			      msecs_to_jiffies(500));
	return 0;
failed:
	dev_err(&pdev->dev, "failed to headset probe ret=%d\n", ret);
	return ret;
}
EXPORT_SYMBOL_GPL(rk_headset_probe);

MODULE_LICENSE("GPL");