[mirror_ubuntu-zesty-kernel.git] / drivers / input / misc / twl4030-vibra.c

/*
 * twl4030-vibra.c - TWL4030 Vibrator driver
 *
 * Copyright (C) 2008-2010 Nokia Corporation
 *
 * Written by Henrik Saari <henrik.saari@nokia.com>
 * Updates by Felipe Balbi <felipe.balbi@nokia.com>
 * Input by Jari Vanhala <ext-jari.vanhala@nokia.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * 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, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA
 *
 */

#include <linux/module.h>
#include <linux/jiffies.h>
#include <linux/platform_device.h>
#include <linux/of.h>
#include <linux/workqueue.h>
#include <linux/i2c/twl.h>
#include <linux/mfd/twl4030-audio.h>
#include <linux/input.h>
#include <linux/slab.h>

/* MODULE ID2 */
#define LEDEN		0x00

/* ForceFeedback */
#define EFFECT_DIR_180_DEG	0x8000 /* range is 0 - 0xFFFF */

struct vibra_info {
	struct device		*dev;
	struct input_dev	*input_dev;

	struct work_struct	play_work;

	bool			enabled;
	int			speed;
	int			direction;

	bool			coexist;
};

static void vibra_disable_leds(void)
{
	u8 reg;

	/* Disable LEDA & LEDB, cannot be used with vibra (PWM) */
	twl_i2c_read_u8(TWL4030_MODULE_LED, &reg, LEDEN);
	reg &= ~0x03;
	twl_i2c_write_u8(TWL4030_MODULE_LED, LEDEN, reg);
}

/* Powers H-Bridge and enables audio clk */
static void vibra_enable(struct vibra_info *info)
{
	u8 reg;

	twl4030_audio_enable_resource(TWL4030_AUDIO_RES_POWER);

	/* turn H-Bridge on */
	twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
			&reg, TWL4030_REG_VIBRA_CTL);
	twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
			 (reg | TWL4030_VIBRA_EN), TWL4030_REG_VIBRA_CTL);

	twl4030_audio_enable_resource(TWL4030_AUDIO_RES_APLL);

	info->enabled = true;
}

static void vibra_disable(struct vibra_info *info)
{
	u8 reg;

	/* Power down H-Bridge */
	twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
			&reg, TWL4030_REG_VIBRA_CTL);
	twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
			 (reg & ~TWL4030_VIBRA_EN), TWL4030_REG_VIBRA_CTL);

	twl4030_audio_disable_resource(TWL4030_AUDIO_RES_APLL);
	twl4030_audio_disable_resource(TWL4030_AUDIO_RES_POWER);

	info->enabled = false;
}

static void vibra_play_work(struct work_struct *work)
{
	struct vibra_info *info = container_of(work,
			struct vibra_info, play_work);
	int dir;
	int pwm;
	u8 reg;

	dir = info->direction;
	pwm = info->speed;

	twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
			&reg, TWL4030_REG_VIBRA_CTL);
	if (pwm && (!info->coexist || !(reg & TWL4030_VIBRA_SEL))) {

		if (!info->enabled)
			vibra_enable(info);

		/* set vibra rotation direction */
		twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
				&reg, TWL4030_REG_VIBRA_CTL);
		reg = (dir) ? (reg | TWL4030_VIBRA_DIR) :
			(reg & ~TWL4030_VIBRA_DIR);
		twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
				 reg, TWL4030_REG_VIBRA_CTL);

		/* set PWM, 1 = max, 255 = min */
		twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
				 256 - pwm, TWL4030_REG_VIBRA_SET);
	} else {
		if (info->enabled)
			vibra_disable(info);
	}
}

/*** Input/ForceFeedback ***/

static int vibra_play(struct input_dev *input, void *data,
		      struct ff_effect *effect)
{
	struct vibra_info *info = input_get_drvdata(input);

	info->speed = effect->u.rumble.strong_magnitude >> 8;
	if (!info->speed)
		info->speed = effect->u.rumble.weak_magnitude >> 9;
	info->direction = effect->direction < EFFECT_DIR_180_DEG ? 0 : 1;
	schedule_work(&info->play_work);
	return 0;
}

static void twl4030_vibra_close(struct input_dev *input)
{
	struct vibra_info *info = input_get_drvdata(input);

	cancel_work_sync(&info->play_work);

	if (info->enabled)
		vibra_disable(info);
}

/*** Module ***/
static int __maybe_unused twl4030_vibra_suspend(struct device *dev)
{
	struct platform_device *pdev = to_platform_device(dev);
	struct vibra_info *info = platform_get_drvdata(pdev);

	if (info->enabled)
		vibra_disable(info);

	return 0;
}

static int __maybe_unused twl4030_vibra_resume(struct device *dev)
{
	vibra_disable_leds();
	return 0;
}

static SIMPLE_DEV_PM_OPS(twl4030_vibra_pm_ops,
			 twl4030_vibra_suspend, twl4030_vibra_resume);

static bool twl4030_vibra_check_coexist(struct twl4030_vibra_data *pdata,
			      struct device_node *node)
{
	if (pdata && pdata->coexist)
		return true;

	node = of_find_node_by_name(node, "codec");
	if (node) {
		of_node_put(node);
		return true;
	}

	return false;
}

static int twl4030_vibra_probe(struct platform_device *pdev)
{
	struct twl4030_vibra_data *pdata = dev_get_platdata(&pdev->dev);
	struct device_node *twl4030_core_node = pdev->dev.parent->of_node;
	struct vibra_info *info;
	int ret;

	if (!pdata && !twl4030_core_node) {
		dev_dbg(&pdev->dev, "platform_data not available\n");
		return -EINVAL;
	}

	info = devm_kzalloc(&pdev->dev, sizeof(*info), GFP_KERNEL);
	if (!info)
		return -ENOMEM;

	info->dev = &pdev->dev;
	info->coexist = twl4030_vibra_check_coexist(pdata, twl4030_core_node);
	INIT_WORK(&info->play_work, vibra_play_work);

	info->input_dev = devm_input_allocate_device(&pdev->dev);
	if (info->input_dev == NULL) {
		dev_err(&pdev->dev, "couldn't allocate input device\n");
		return -ENOMEM;
	}

	input_set_drvdata(info->input_dev, info);

	info->input_dev->name = "twl4030:vibrator";
	info->input_dev->id.version = 1;
	info->input_dev->close = twl4030_vibra_close;
	__set_bit(FF_RUMBLE, info->input_dev->ffbit);

	ret = input_ff_create_memless(info->input_dev, NULL, vibra_play);
	if (ret < 0) {
		dev_dbg(&pdev->dev, "couldn't register vibrator to FF\n");
		return ret;
	}

	ret = input_register_device(info->input_dev);
	if (ret < 0) {
		dev_dbg(&pdev->dev, "couldn't register input device\n");
		goto err_iff;
	}

	vibra_disable_leds();

	platform_set_drvdata(pdev, info);
	return 0;

err_iff:
	input_ff_destroy(info->input_dev);
	return ret;
}

static struct platform_driver twl4030_vibra_driver = {
	.probe		= twl4030_vibra_probe,
	.driver		= {
		.name	= "twl4030-vibra",
		.pm	= &twl4030_vibra_pm_ops,
	},
};
module_platform_driver(twl4030_vibra_driver);

MODULE_ALIAS("platform:twl4030-vibra");
MODULE_DESCRIPTION("TWL4030 Vibra driver");
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Nokia Corporation");
Commit	Line	Data
3dd1b394 JV	1	/*
	2	* twl4030-vibra.c - TWL4030 Vibrator driver
	3	*
	4	* Copyright (C) 2008-2010 Nokia Corporation
	5	*
	6	* Written by Henrik Saari <henrik.saari@nokia.com>
	7	* Updates by Felipe Balbi <felipe.balbi@nokia.com>
	8	* Input by Jari Vanhala <ext-jari.vanhala@nokia.com>
	9	*
	10	* This program is free software; you can redistribute it and/or modify
	11	* it under the terms of the GNU General Public License version 2 as
	12	* published by the Free Software Foundation.
	13	*
	14	* This program is distributed in the hope that it will be useful, but
	15	* WITHOUT ANY WARRANTY; without even the implied warranty of
	16	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	17	* General Public License for more details.
	18	*
	19	* You should have received a copy of the GNU General Public License
	20	* along with this program; if not, write to the Free Software
	21	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
	22	* 02110-1301 USA
	23	*
	24	*/
	25
	26	#include <linux/module.h>
	27	#include <linux/jiffies.h>
	28	#include <linux/platform_device.h>
64b9e4d8	29	#include <linux/of.h>
3dd1b394 JV	30	#include <linux/workqueue.h>
3dd1b394 JV	31	#include <linux/i2c/twl.h>
57fe7251	32	#include <linux/mfd/twl4030-audio.h>
3dd1b394	33	#include <linux/input.h>
5a0e3ad6	34	#include <linux/slab.h>
3dd1b394 JV	35
	36	/* MODULE ID2 */
	37	#define LEDEN 0x00
	38
	39	/* ForceFeedback */
	40	#define EFFECT_DIR_180_DEG 0x8000 /* range is 0 - 0xFFFF */
	41
	42	struct vibra_info {
	43	struct device *dev;
	44	struct input_dev *input_dev;
	45
3dd1b394 JV	46	struct work_struct play_work;
	47
	48	bool enabled;
	49	int speed;
	50	int direction;
	51
	52	bool coexist;
	53	};
	54
	55	static void vibra_disable_leds(void)
	56	{
	57	u8 reg;
	58
	59	/* Disable LEDA & LEDB, cannot be used with vibra (PWM) */
	60	twl_i2c_read_u8(TWL4030_MODULE_LED, &reg, LEDEN);
	61	reg &= ~0x03;
	62	twl_i2c_write_u8(TWL4030_MODULE_LED, LEDEN, reg);
	63	}
	64
	65	/* Powers H-Bridge and enables audio clk */
	66	static void vibra_enable(struct vibra_info *info)
	67	{
	68	u8 reg;
	69
57fe7251	70	twl4030_audio_enable_resource(TWL4030_AUDIO_RES_POWER);
3dd1b394 JV	71
	72	/* turn H-Bridge on */
	73	twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
	74	&reg, TWL4030_REG_VIBRA_CTL);
	75	twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
	76	(reg \| TWL4030_VIBRA_EN), TWL4030_REG_VIBRA_CTL);
	77
57fe7251	78	twl4030_audio_enable_resource(TWL4030_AUDIO_RES_APLL);
3dd1b394 JV	79
	80	info->enabled = true;
	81	}
	82
	83	static void vibra_disable(struct vibra_info *info)
	84	{
	85	u8 reg;
	86
	87	/* Power down H-Bridge */
	88	twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
	89	&reg, TWL4030_REG_VIBRA_CTL);
	90	twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
	91	(reg & ~TWL4030_VIBRA_EN), TWL4030_REG_VIBRA_CTL);
	92
57fe7251 PU	93	twl4030_audio_disable_resource(TWL4030_AUDIO_RES_APLL);
57fe7251 PU	94	twl4030_audio_disable_resource(TWL4030_AUDIO_RES_POWER);
3dd1b394 JV	95
	96	info->enabled = false;
	97	}
	98
	99	static void vibra_play_work(struct work_struct *work)
	100	{
	101	struct vibra_info *info = container_of(work,
	102	struct vibra_info, play_work);
	103	int dir;
	104	int pwm;
	105	u8 reg;
	106
	107	dir = info->direction;
	108	pwm = info->speed;
	109
	110	twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
	111	&reg, TWL4030_REG_VIBRA_CTL);
	112	if (pwm && (!info->coexist \|\| !(reg & TWL4030_VIBRA_SEL))) {
	113
	114	if (!info->enabled)
	115	vibra_enable(info);
	116
	117	/* set vibra rotation direction */
	118	twl_i2c_read_u8(TWL4030_MODULE_AUDIO_VOICE,
	119	&reg, TWL4030_REG_VIBRA_CTL);
	120	reg = (dir) ? (reg \| TWL4030_VIBRA_DIR) :
	121	(reg & ~TWL4030_VIBRA_DIR);
	122	twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
	123	reg, TWL4030_REG_VIBRA_CTL);
	124
	125	/* set PWM, 1 = max, 255 = min */
	126	twl_i2c_write_u8(TWL4030_MODULE_AUDIO_VOICE,
	127	256 - pwm, TWL4030_REG_VIBRA_SET);
	128	} else {
	129	if (info->enabled)
	130	vibra_disable(info);
	131	}
	132	}
	133
	134	/* Input/ForceFeedback */
	135
	136	static int vibra_play(struct input_dev input, void data,
	137	struct ff_effect *effect)
	138	{
	139	struct vibra_info *info = input_get_drvdata(input);
	140
	141	info->speed = effect->u.rumble.strong_magnitude >> 8;
	142	if (!info->speed)
	143	info->speed = effect->u.rumble.weak_magnitude >> 9;
	144	info->direction = effect->direction < EFFECT_DIR_180_DEG ? 0 : 1;
4a31ba37	145	schedule_work(&info->play_work);
3dd1b394 JV	146	return 0;
	147	}
	148
	149	static void twl4030_vibra_close(struct input_dev *input)
	150	{
	151	struct vibra_info *info = input_get_drvdata(input);
	152
	153	cancel_work_sync(&info->play_work);
3dd1b394 JV	154
	155	if (info->enabled)
	156	vibra_disable(info);
	157	}
	158
	159	/* Module */
97a652a8	160	static int __maybe_unused twl4030_vibra_suspend(struct device *dev)
3dd1b394 JV	161	{
	162	struct platform_device *pdev = to_platform_device(dev);
	163	struct vibra_info *info = platform_get_drvdata(pdev);
	164
	165	if (info->enabled)
	166	vibra_disable(info);
	167
	168	return 0;
	169	}
	170
97a652a8	171	static int __maybe_unused twl4030_vibra_resume(struct device *dev)
3dd1b394 JV	172	{
	173	vibra_disable_leds();
	174	return 0;
	175	}
	176
	177	static SIMPLE_DEV_PM_OPS(twl4030_vibra_pm_ops,
	178	twl4030_vibra_suspend, twl4030_vibra_resume);
3dd1b394	179
64b9e4d8 PU	180	static bool twl4030_vibra_check_coexist(struct twl4030_vibra_data *pdata,
	181	struct device_node *node)
	182	{
	183	if (pdata && pdata->coexist)
	184	return true;
	185
e661d0a0 MB	186	node = of_find_node_by_name(node, "codec");
e661d0a0 MB	187	if (node) {
a9e1d3c0	188	of_node_put(node);
64b9e4d8	189	return true;
a9e1d3c0	190	}
64b9e4d8 PU	191
	192	return false;
	193	}
	194
5298cc4c	195	static int twl4030_vibra_probe(struct platform_device *pdev)
3dd1b394	196	{
c838cb3d	197	struct twl4030_vibra_data *pdata = dev_get_platdata(&pdev->dev);
64b9e4d8	198	struct device_node *twl4030_core_node = pdev->dev.parent->of_node;
3dd1b394 JV	199	struct vibra_info *info;
	200	int ret;
	201
64b9e4d8	202	if (!pdata && !twl4030_core_node) {
3dd1b394 JV	203	dev_dbg(&pdev->dev, "platform_data not available\n");
	204	return -EINVAL;
	205	}
	206
c3ead16e	207	info = devm_kzalloc(&pdev->dev, sizeof(*info), GFP_KERNEL);
3dd1b394 JV	208	if (!info)
	209	return -ENOMEM;
	210
	211	info->dev = &pdev->dev;
64b9e4d8	212	info->coexist = twl4030_vibra_check_coexist(pdata, twl4030_core_node);
3dd1b394 JV	213	INIT_WORK(&info->play_work, vibra_play_work);
3dd1b394 JV	214
c3ead16e	215	info->input_dev = devm_input_allocate_device(&pdev->dev);
3dd1b394 JV	216	if (info->input_dev == NULL) {
3dd1b394 JV	217	dev_err(&pdev->dev, "couldn't allocate input device\n");
c3ead16e	218	return -ENOMEM;
3dd1b394 JV	219	}
	220
	221	input_set_drvdata(info->input_dev, info);
	222
	223	info->input_dev->name = "twl4030:vibrator";
	224	info->input_dev->id.version = 1;
3dd1b394 JV	225	info->input_dev->close = twl4030_vibra_close;
	226	__set_bit(FF_RUMBLE, info->input_dev->ffbit);
	227
	228	ret = input_ff_create_memless(info->input_dev, NULL, vibra_play);
	229	if (ret < 0) {
	230	dev_dbg(&pdev->dev, "couldn't register vibrator to FF\n");
c3ead16e	231	return ret;
3dd1b394 JV	232	}
	233
	234	ret = input_register_device(info->input_dev);
	235	if (ret < 0) {
	236	dev_dbg(&pdev->dev, "couldn't register input device\n");
	237	goto err_iff;
	238	}
	239
	240	vibra_disable_leds();
	241
	242	platform_set_drvdata(pdev, info);
	243	return 0;
	244
	245	err_iff:
	246	input_ff_destroy(info->input_dev);
3dd1b394 JV	247	return ret;
	248	}
	249
3dd1b394 JV	250	static struct platform_driver twl4030_vibra_driver = {
3dd1b394 JV	251	.probe = twl4030_vibra_probe,
3dd1b394	252	.driver = {
f0fba2ad	253	.name = "twl4030-vibra",
3dd1b394	254	.pm = &twl4030_vibra_pm_ops,
3dd1b394 JV	255	},
3dd1b394 JV	256	};
840a746b	257	module_platform_driver(twl4030_vibra_driver);
3dd1b394	258
f0fba2ad	259	MODULE_ALIAS("platform:twl4030-vibra");
3dd1b394 JV	260	MODULE_DESCRIPTION("TWL4030 Vibra driver");
	261	MODULE_LICENSE("GPL");
	262	MODULE_AUTHOR("Nokia Corporation");