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