1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28#include <linux/module.h>
29#include <linux/platform_device.h>
30#include <linux/of.h>
31#include <linux/workqueue.h>
32#include <linux/input.h>
33#include <linux/mfd/twl6040.h>
34#include <linux/slab.h>
35#include <linux/delay.h>
36#include <linux/regulator/consumer.h>
37
38#define EFFECT_DIR_180_DEG 0x8000
39
40
41#define TWL6040_VIBRA_MOD 85
42
43#define TWL6040_NUM_SUPPLIES 2
44
45struct vibra_info {
46 struct device *dev;
47 struct input_dev *input_dev;
48 struct workqueue_struct *workqueue;
49 struct work_struct play_work;
50 struct mutex mutex;
51 int irq;
52
53 bool enabled;
54 int weak_speed;
55 int strong_speed;
56 int direction;
57
58 unsigned int vibldrv_res;
59 unsigned int vibrdrv_res;
60 unsigned int viblmotor_res;
61 unsigned int vibrmotor_res;
62
63 struct regulator_bulk_data supplies[TWL6040_NUM_SUPPLIES];
64
65 struct twl6040 *twl6040;
66};
67
68static irqreturn_t twl6040_vib_irq_handler(int irq, void *data)
69{
70 struct vibra_info *info = data;
71 struct twl6040 *twl6040 = info->twl6040;
72 u8 status;
73
74 status = twl6040_reg_read(twl6040, TWL6040_REG_STATUS);
75 if (status & TWL6040_VIBLOCDET) {
76 dev_warn(info->dev, "Left Vibrator overcurrent detected\n");
77 twl6040_clear_bits(twl6040, TWL6040_REG_VIBCTLL,
78 TWL6040_VIBENA);
79 }
80 if (status & TWL6040_VIBROCDET) {
81 dev_warn(info->dev, "Right Vibrator overcurrent detected\n");
82 twl6040_clear_bits(twl6040, TWL6040_REG_VIBCTLR,
83 TWL6040_VIBENA);
84 }
85
86 return IRQ_HANDLED;
87}
88
89static void twl6040_vibra_enable(struct vibra_info *info)
90{
91 struct twl6040 *twl6040 = info->twl6040;
92 int ret;
93
94 ret = regulator_bulk_enable(ARRAY_SIZE(info->supplies), info->supplies);
95 if (ret) {
96 dev_err(info->dev, "failed to enable regulators %d\n", ret);
97 return;
98 }
99
100 twl6040_power(info->twl6040, 1);
101 if (twl6040_get_revid(twl6040) <= TWL6040_REV_ES1_1) {
102
103
104
105
106
107 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL,
108 TWL6040_VIBENA | TWL6040_VIBCTRL);
109 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR,
110 TWL6040_VIBENA | TWL6040_VIBCTRL);
111 usleep_range(3000, 3500);
112 }
113
114 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL,
115 TWL6040_VIBENA);
116 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR,
117 TWL6040_VIBENA);
118
119 info->enabled = true;
120}
121
122static void twl6040_vibra_disable(struct vibra_info *info)
123{
124 struct twl6040 *twl6040 = info->twl6040;
125
126 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL, 0x00);
127 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR, 0x00);
128 twl6040_power(info->twl6040, 0);
129
130 regulator_bulk_disable(ARRAY_SIZE(info->supplies), info->supplies);
131
132 info->enabled = false;
133}
134
135static u8 twl6040_vibra_code(int vddvib, int vibdrv_res, int motor_res,
136 int speed, int direction)
137{
138 int vpk, max_code;
139 u8 vibdat;
140
141
142 vpk = (vddvib * motor_res * TWL6040_VIBRA_MOD) /
143 (100 * (vibdrv_res + motor_res));
144
145
146 max_code = vpk / 50;
147 if (max_code > TWL6040_VIBDAT_MAX)
148 max_code = TWL6040_VIBDAT_MAX;
149
150
151 vibdat = (u8)((speed * max_code) / USHRT_MAX);
152
153
154 vibdat *= direction;
155
156 return vibdat;
157}
158
159static void twl6040_vibra_set_effect(struct vibra_info *info)
160{
161 struct twl6040 *twl6040 = info->twl6040;
162 u8 vibdatl, vibdatr;
163 int volt;
164
165
166 volt = regulator_get_voltage(info->supplies[0].consumer) / 1000;
167 vibdatl = twl6040_vibra_code(volt, info->vibldrv_res,
168 info->viblmotor_res,
169 info->weak_speed, info->direction);
170
171
172 volt = regulator_get_voltage(info->supplies[1].consumer) / 1000;
173 vibdatr = twl6040_vibra_code(volt, info->vibrdrv_res,
174 info->vibrmotor_res,
175 info->strong_speed, info->direction);
176
177 twl6040_reg_write(twl6040, TWL6040_REG_VIBDATL, vibdatl);
178 twl6040_reg_write(twl6040, TWL6040_REG_VIBDATR, vibdatr);
179}
180
181static void vibra_play_work(struct work_struct *work)
182{
183 struct vibra_info *info = container_of(work,
184 struct vibra_info, play_work);
185
186 mutex_lock(&info->mutex);
187
188 if (info->weak_speed || info->strong_speed) {
189 if (!info->enabled)
190 twl6040_vibra_enable(info);
191
192 twl6040_vibra_set_effect(info);
193 } else if (info->enabled)
194 twl6040_vibra_disable(info);
195
196 mutex_unlock(&info->mutex);
197}
198
199static int vibra_play(struct input_dev *input, void *data,
200 struct ff_effect *effect)
201{
202 struct vibra_info *info = input_get_drvdata(input);
203 int ret;
204
205
206 ret = twl6040_get_vibralr_status(info->twl6040);
207 if (ret & TWL6040_VIBSEL) {
208 dev_info(&input->dev, "Vibra is configured for audio\n");
209 return -EBUSY;
210 }
211
212 info->weak_speed = effect->u.rumble.weak_magnitude;
213 info->strong_speed = effect->u.rumble.strong_magnitude;
214 info->direction = effect->direction < EFFECT_DIR_180_DEG ? 1 : -1;
215
216 ret = queue_work(info->workqueue, &info->play_work);
217 if (!ret) {
218 dev_info(&input->dev, "work is already on queue\n");
219 return ret;
220 }
221
222 return 0;
223}
224
225static void twl6040_vibra_close(struct input_dev *input)
226{
227 struct vibra_info *info = input_get_drvdata(input);
228
229 cancel_work_sync(&info->play_work);
230
231 mutex_lock(&info->mutex);
232
233 if (info->enabled)
234 twl6040_vibra_disable(info);
235
236 mutex_unlock(&info->mutex);
237}
238
239#ifdef CONFIG_PM_SLEEP
240static int twl6040_vibra_suspend(struct device *dev)
241{
242 struct platform_device *pdev = to_platform_device(dev);
243 struct vibra_info *info = platform_get_drvdata(pdev);
244
245 mutex_lock(&info->mutex);
246
247 if (info->enabled)
248 twl6040_vibra_disable(info);
249
250 mutex_unlock(&info->mutex);
251
252 return 0;
253}
254#endif
255
256static SIMPLE_DEV_PM_OPS(twl6040_vibra_pm_ops, twl6040_vibra_suspend, NULL);
257
258static int twl6040_vibra_probe(struct platform_device *pdev)
259{
260 struct twl6040_vibra_data *pdata = pdev->dev.platform_data;
261 struct device *twl6040_core_dev = pdev->dev.parent;
262 struct device_node *twl6040_core_node = NULL;
263 struct vibra_info *info;
264 int vddvibl_uV = 0;
265 int vddvibr_uV = 0;
266 int ret;
267
268#ifdef CONFIG_OF
269 twl6040_core_node = of_find_node_by_name(twl6040_core_dev->of_node,
270 "vibra");
271#endif
272
273 if (!pdata && !twl6040_core_node) {
274 dev_err(&pdev->dev, "platform_data not available\n");
275 return -EINVAL;
276 }
277
278 info = devm_kzalloc(&pdev->dev, sizeof(*info), GFP_KERNEL);
279 if (!info) {
280 dev_err(&pdev->dev, "couldn't allocate memory\n");
281 return -ENOMEM;
282 }
283
284 info->dev = &pdev->dev;
285
286 info->twl6040 = dev_get_drvdata(pdev->dev.parent);
287 if (pdata) {
288 info->vibldrv_res = pdata->vibldrv_res;
289 info->vibrdrv_res = pdata->vibrdrv_res;
290 info->viblmotor_res = pdata->viblmotor_res;
291 info->vibrmotor_res = pdata->vibrmotor_res;
292 vddvibl_uV = pdata->vddvibl_uV;
293 vddvibr_uV = pdata->vddvibr_uV;
294 } else {
295 of_property_read_u32(twl6040_core_node, "ti,vibldrv-res",
296 &info->vibldrv_res);
297 of_property_read_u32(twl6040_core_node, "ti,vibrdrv-res",
298 &info->vibrdrv_res);
299 of_property_read_u32(twl6040_core_node, "ti,viblmotor-res",
300 &info->viblmotor_res);
301 of_property_read_u32(twl6040_core_node, "ti,vibrmotor-res",
302 &info->vibrmotor_res);
303 of_property_read_u32(twl6040_core_node, "ti,vddvibl-uV",
304 &vddvibl_uV);
305 of_property_read_u32(twl6040_core_node, "ti,vddvibr-uV",
306 &vddvibr_uV);
307 }
308
309 if ((!info->vibldrv_res && !info->viblmotor_res) ||
310 (!info->vibrdrv_res && !info->vibrmotor_res)) {
311 dev_err(info->dev, "invalid vibra driver/motor resistance\n");
312 return -EINVAL;
313 }
314
315 info->irq = platform_get_irq(pdev, 0);
316 if (info->irq < 0) {
317 dev_err(info->dev, "invalid irq\n");
318 return -EINVAL;
319 }
320
321 mutex_init(&info->mutex);
322
323 ret = devm_request_threaded_irq(&pdev->dev, info->irq, NULL,
324 twl6040_vib_irq_handler, 0,
325 "twl6040_irq_vib", info);
326 if (ret) {
327 dev_err(info->dev, "VIB IRQ request failed: %d\n", ret);
328 return ret;
329 }
330
331 info->supplies[0].supply = "vddvibl";
332 info->supplies[1].supply = "vddvibr";
333
334
335
336
337 ret = regulator_bulk_get(pdata ? info->dev : twl6040_core_dev,
338 ARRAY_SIZE(info->supplies), info->supplies);
339 if (ret) {
340 dev_err(info->dev, "couldn't get regulators %d\n", ret);
341 return ret;
342 }
343
344 if (vddvibl_uV) {
345 ret = regulator_set_voltage(info->supplies[0].consumer,
346 vddvibl_uV, vddvibl_uV);
347 if (ret) {
348 dev_err(info->dev, "failed to set VDDVIBL volt %d\n",
349 ret);
350 goto err_regulator;
351 }
352 }
353
354 if (vddvibr_uV) {
355 ret = regulator_set_voltage(info->supplies[1].consumer,
356 vddvibr_uV, vddvibr_uV);
357 if (ret) {
358 dev_err(info->dev, "failed to set VDDVIBR volt %d\n",
359 ret);
360 goto err_regulator;
361 }
362 }
363
364 INIT_WORK(&info->play_work, vibra_play_work);
365
366 info->input_dev = input_allocate_device();
367 if (info->input_dev == NULL) {
368 dev_err(info->dev, "couldn't allocate input device\n");
369 ret = -ENOMEM;
370 goto err_regulator;
371 }
372
373 input_set_drvdata(info->input_dev, info);
374
375 info->input_dev->name = "twl6040:vibrator";
376 info->input_dev->id.version = 1;
377 info->input_dev->dev.parent = pdev->dev.parent;
378 info->input_dev->close = twl6040_vibra_close;
379 __set_bit(FF_RUMBLE, info->input_dev->ffbit);
380
381 ret = input_ff_create_memless(info->input_dev, NULL, vibra_play);
382 if (ret < 0) {
383 dev_err(info->dev, "couldn't register vibrator to FF\n");
384 goto err_ialloc;
385 }
386
387 ret = input_register_device(info->input_dev);
388 if (ret < 0) {
389 dev_err(info->dev, "couldn't register input device\n");
390 goto err_iff;
391 }
392
393 platform_set_drvdata(pdev, info);
394
395 return 0;
396
397err_iff:
398 input_ff_destroy(info->input_dev);
399err_ialloc:
400 input_free_device(info->input_dev);
401err_regulator:
402 regulator_bulk_free(ARRAY_SIZE(info->supplies), info->supplies);
403 return ret;
404}
405
406static int twl6040_vibra_remove(struct platform_device *pdev)
407{
408 struct vibra_info *info = platform_get_drvdata(pdev);
409
410 input_unregister_device(info->input_dev);
411 regulator_bulk_free(ARRAY_SIZE(info->supplies), info->supplies);
412
413 return 0;
414}
415
416static struct platform_driver twl6040_vibra_driver = {
417 .probe = twl6040_vibra_probe,
418 .remove = twl6040_vibra_remove,
419 .driver = {
420 .name = "twl6040-vibra",
421 .owner = THIS_MODULE,
422 .pm = &twl6040_vibra_pm_ops,
423 },
424};
425module_platform_driver(twl6040_vibra_driver);
426
427MODULE_ALIAS("platform:twl6040-vibra");
428MODULE_DESCRIPTION("TWL6040 Vibra driver");
429MODULE_LICENSE("GPL");
430MODULE_AUTHOR("Jorge Eduardo Candelaria <jorge.candelaria@ti.com>");
431MODULE_AUTHOR("Misael Lopez Cruz <misael.lopez@ti.com>");
432
