1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * twl6040-vibra.c - TWL6040 Vibrator driver
4 *
5 * Author: Jorge Eduardo Candelaria <jorge.candelaria@ti.com>
6 * Author: Misael Lopez Cruz <misael.lopez@ti.com>
7 *
8 * Copyright: (C) 2011 Texas Instruments, Inc.
9 *
10 * Based on twl4030-vibra.c by Henrik Saari <henrik.saari@nokia.com>
11 * Felipe Balbi <felipe.balbi@nokia.com>
12 * Jari Vanhala <ext-javi.vanhala@nokia.com>
13 */
14 #include <linux/module.h>
15 #include <linux/platform_device.h>
16 #include <linux/of.h>
17 #include <linux/workqueue.h>
18 #include <linux/input.h>
19 #include <linux/mfd/twl6040.h>
20 #include <linux/slab.h>
21 #include <linux/delay.h>
22 #include <linux/regulator/consumer.h>
23
24 #define EFFECT_DIR_180_DEG 0x8000
25
26 /* Recommended modulation index 85% */
27 #define TWL6040_VIBRA_MOD 85
28
29 #define TWL6040_NUM_SUPPLIES 2
30
31 struct vibra_info {
32 struct device *dev;
33 struct input_dev *input_dev;
34 struct work_struct play_work;
35
36 int irq;
37
38 bool enabled;
39 int weak_speed;
40 int strong_speed;
41 int direction;
42
43 unsigned int vibldrv_res;
44 unsigned int vibrdrv_res;
45 unsigned int viblmotor_res;
46 unsigned int vibrmotor_res;
47
48 struct regulator_bulk_data supplies[TWL6040_NUM_SUPPLIES];
49
50 struct twl6040 *twl6040;
51 };
52
twl6040_vib_irq_handler(int irq,void * data)53 static irqreturn_t twl6040_vib_irq_handler(int irq, void *data)
54 {
55 struct vibra_info *info = data;
56 struct twl6040 *twl6040 = info->twl6040;
57 u8 status;
58
59 status = twl6040_reg_read(twl6040, TWL6040_REG_STATUS);
60 if (status & TWL6040_VIBLOCDET) {
61 dev_warn(info->dev, "Left Vibrator overcurrent detected\n");
62 twl6040_clear_bits(twl6040, TWL6040_REG_VIBCTLL,
63 TWL6040_VIBENA);
64 }
65 if (status & TWL6040_VIBROCDET) {
66 dev_warn(info->dev, "Right Vibrator overcurrent detected\n");
67 twl6040_clear_bits(twl6040, TWL6040_REG_VIBCTLR,
68 TWL6040_VIBENA);
69 }
70
71 return IRQ_HANDLED;
72 }
73
twl6040_vibra_enable(struct vibra_info * info)74 static void twl6040_vibra_enable(struct vibra_info *info)
75 {
76 struct twl6040 *twl6040 = info->twl6040;
77 int ret;
78
79 ret = regulator_bulk_enable(ARRAY_SIZE(info->supplies), info->supplies);
80 if (ret) {
81 dev_err(info->dev, "failed to enable regulators %d\n", ret);
82 return;
83 }
84
85 twl6040_power(info->twl6040, 1);
86 if (twl6040_get_revid(twl6040) <= TWL6040_REV_ES1_1) {
87 /*
88 * ERRATA: Disable overcurrent protection for at least
89 * 3ms when enabling vibrator drivers to avoid false
90 * overcurrent detection
91 */
92 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL,
93 TWL6040_VIBENA | TWL6040_VIBCTRL);
94 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR,
95 TWL6040_VIBENA | TWL6040_VIBCTRL);
96 usleep_range(3000, 3500);
97 }
98
99 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL,
100 TWL6040_VIBENA);
101 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR,
102 TWL6040_VIBENA);
103
104 info->enabled = true;
105 }
106
twl6040_vibra_disable(struct vibra_info * info)107 static void twl6040_vibra_disable(struct vibra_info *info)
108 {
109 struct twl6040 *twl6040 = info->twl6040;
110
111 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLL, 0x00);
112 twl6040_reg_write(twl6040, TWL6040_REG_VIBCTLR, 0x00);
113 twl6040_power(info->twl6040, 0);
114
115 regulator_bulk_disable(ARRAY_SIZE(info->supplies), info->supplies);
116
117 info->enabled = false;
118 }
119
twl6040_vibra_code(int vddvib,int vibdrv_res,int motor_res,int speed,int direction)120 static u8 twl6040_vibra_code(int vddvib, int vibdrv_res, int motor_res,
121 int speed, int direction)
122 {
123 int vpk, max_code;
124 u8 vibdat;
125
126 /* output swing */
127 vpk = (vddvib * motor_res * TWL6040_VIBRA_MOD) /
128 (100 * (vibdrv_res + motor_res));
129
130 /* 50mV per VIBDAT code step */
131 max_code = vpk / 50;
132 if (max_code > TWL6040_VIBDAT_MAX)
133 max_code = TWL6040_VIBDAT_MAX;
134
135 /* scale speed to max allowed code */
136 vibdat = (u8)((speed * max_code) / USHRT_MAX);
137
138 /* 2's complement for direction > 180 degrees */
139 vibdat *= direction;
140
141 return vibdat;
142 }
143
twl6040_vibra_set_effect(struct vibra_info * info)144 static void twl6040_vibra_set_effect(struct vibra_info *info)
145 {
146 struct twl6040 *twl6040 = info->twl6040;
147 u8 vibdatl, vibdatr;
148 int volt;
149
150 /* weak motor */
151 volt = regulator_get_voltage(info->supplies[0].consumer) / 1000;
152 vibdatl = twl6040_vibra_code(volt, info->vibldrv_res,
153 info->viblmotor_res,
154 info->weak_speed, info->direction);
155
156 /* strong motor */
157 volt = regulator_get_voltage(info->supplies[1].consumer) / 1000;
158 vibdatr = twl6040_vibra_code(volt, info->vibrdrv_res,
159 info->vibrmotor_res,
160 info->strong_speed, info->direction);
161
162 twl6040_reg_write(twl6040, TWL6040_REG_VIBDATL, vibdatl);
163 twl6040_reg_write(twl6040, TWL6040_REG_VIBDATR, vibdatr);
164 }
165
vibra_play_work(struct work_struct * work)166 static void vibra_play_work(struct work_struct *work)
167 {
168 struct vibra_info *info = container_of(work,
169 struct vibra_info, play_work);
170 int ret;
171
172 /* Do not allow effect, while the routing is set to use audio */
173 ret = twl6040_get_vibralr_status(info->twl6040);
174 if (ret & TWL6040_VIBSEL) {
175 dev_info(info->dev, "Vibra is configured for audio\n");
176 return;
177 }
178
179 if (info->weak_speed || info->strong_speed) {
180 if (!info->enabled)
181 twl6040_vibra_enable(info);
182
183 twl6040_vibra_set_effect(info);
184 } else if (info->enabled)
185 twl6040_vibra_disable(info);
186
187 }
188
vibra_play(struct input_dev * input,void * data,struct ff_effect * effect)189 static int vibra_play(struct input_dev *input, void *data,
190 struct ff_effect *effect)
191 {
192 struct vibra_info *info = input_get_drvdata(input);
193
194 info->weak_speed = effect->u.rumble.weak_magnitude;
195 info->strong_speed = effect->u.rumble.strong_magnitude;
196 info->direction = effect->direction < EFFECT_DIR_180_DEG ? 1 : -1;
197
198 schedule_work(&info->play_work);
199
200 return 0;
201 }
202
twl6040_vibra_close(struct input_dev * input)203 static void twl6040_vibra_close(struct input_dev *input)
204 {
205 struct vibra_info *info = input_get_drvdata(input);
206
207 cancel_work_sync(&info->play_work);
208
209 if (info->enabled)
210 twl6040_vibra_disable(info);
211 }
212
twl6040_vibra_suspend(struct device * dev)213 static int twl6040_vibra_suspend(struct device *dev)
214 {
215 struct platform_device *pdev = to_platform_device(dev);
216 struct vibra_info *info = platform_get_drvdata(pdev);
217
218 cancel_work_sync(&info->play_work);
219
220 if (info->enabled)
221 twl6040_vibra_disable(info);
222
223 return 0;
224 }
225
226 static DEFINE_SIMPLE_DEV_PM_OPS(twl6040_vibra_pm_ops,
227 twl6040_vibra_suspend, NULL);
228
twl6040_vibra_probe(struct platform_device * pdev)229 static int twl6040_vibra_probe(struct platform_device *pdev)
230 {
231 struct device *twl6040_core_dev = pdev->dev.parent;
232 struct vibra_info *info;
233 int vddvibl_uV = 0;
234 int vddvibr_uV = 0;
235 int error;
236
237 struct device_node *twl6040_core_node __free(device_node) =
238 of_get_child_by_name(twl6040_core_dev->of_node, "vibra");
239 if (!twl6040_core_node) {
240 dev_err(&pdev->dev, "parent of node is missing?\n");
241 return -EINVAL;
242 }
243
244 info = devm_kzalloc(&pdev->dev, sizeof(*info), GFP_KERNEL);
245 if (!info) {
246 dev_err(&pdev->dev, "couldn't allocate memory\n");
247 return -ENOMEM;
248 }
249
250 info->dev = &pdev->dev;
251
252 info->twl6040 = dev_get_drvdata(pdev->dev.parent);
253
254 of_property_read_u32(twl6040_core_node, "ti,vibldrv-res",
255 &info->vibldrv_res);
256 of_property_read_u32(twl6040_core_node, "ti,vibrdrv-res",
257 &info->vibrdrv_res);
258 of_property_read_u32(twl6040_core_node, "ti,viblmotor-res",
259 &info->viblmotor_res);
260 of_property_read_u32(twl6040_core_node, "ti,vibrmotor-res",
261 &info->vibrmotor_res);
262 of_property_read_u32(twl6040_core_node, "ti,vddvibl-uV", &vddvibl_uV);
263 of_property_read_u32(twl6040_core_node, "ti,vddvibr-uV", &vddvibr_uV);
264
265 if ((!info->vibldrv_res && !info->viblmotor_res) ||
266 (!info->vibrdrv_res && !info->vibrmotor_res)) {
267 dev_err(info->dev, "invalid vibra driver/motor resistance\n");
268 return -EINVAL;
269 }
270
271 info->irq = platform_get_irq(pdev, 0);
272 if (info->irq < 0)
273 return -EINVAL;
274
275 error = devm_request_threaded_irq(&pdev->dev, info->irq, NULL,
276 twl6040_vib_irq_handler,
277 IRQF_ONESHOT,
278 "twl6040_irq_vib", info);
279 if (error) {
280 dev_err(info->dev, "VIB IRQ request failed: %d\n", error);
281 return error;
282 }
283
284 info->supplies[0].supply = "vddvibl";
285 info->supplies[1].supply = "vddvibr";
286 /*
287 * When booted with Device tree the regulators are attached to the
288 * parent device (twl6040 MFD core)
289 */
290 error = devm_regulator_bulk_get(twl6040_core_dev,
291 ARRAY_SIZE(info->supplies),
292 info->supplies);
293 if (error) {
294 dev_err(info->dev, "couldn't get regulators %d\n", error);
295 return error;
296 }
297
298 if (vddvibl_uV) {
299 error = regulator_set_voltage(info->supplies[0].consumer,
300 vddvibl_uV, vddvibl_uV);
301 if (error) {
302 dev_err(info->dev, "failed to set VDDVIBL volt %d\n",
303 error);
304 return error;
305 }
306 }
307
308 if (vddvibr_uV) {
309 error = regulator_set_voltage(info->supplies[1].consumer,
310 vddvibr_uV, vddvibr_uV);
311 if (error) {
312 dev_err(info->dev, "failed to set VDDVIBR volt %d\n",
313 error);
314 return error;
315 }
316 }
317
318 INIT_WORK(&info->play_work, vibra_play_work);
319
320 info->input_dev = devm_input_allocate_device(&pdev->dev);
321 if (!info->input_dev) {
322 dev_err(info->dev, "couldn't allocate input device\n");
323 return -ENOMEM;
324 }
325
326 input_set_drvdata(info->input_dev, info);
327
328 info->input_dev->name = "twl6040:vibrator";
329 info->input_dev->id.version = 1;
330 info->input_dev->close = twl6040_vibra_close;
331 __set_bit(FF_RUMBLE, info->input_dev->ffbit);
332
333 error = input_ff_create_memless(info->input_dev, NULL, vibra_play);
334 if (error) {
335 dev_err(info->dev, "couldn't register vibrator to FF\n");
336 return error;
337 }
338
339 error = input_register_device(info->input_dev);
340 if (error) {
341 dev_err(info->dev, "couldn't register input device\n");
342 return error;
343 }
344
345 platform_set_drvdata(pdev, info);
346
347 return 0;
348 }
349
350 static struct platform_driver twl6040_vibra_driver = {
351 .probe = twl6040_vibra_probe,
352 .driver = {
353 .name = "twl6040-vibra",
354 .pm = pm_sleep_ptr(&twl6040_vibra_pm_ops),
355 },
356 };
357 module_platform_driver(twl6040_vibra_driver);
358
359 MODULE_ALIAS("platform:twl6040-vibra");
360 MODULE_DESCRIPTION("TWL6040 Vibra driver");
361 MODULE_LICENSE("GPL");
362 MODULE_AUTHOR("Jorge Eduardo Candelaria <jorge.candelaria@ti.com>");
363 MODULE_AUTHOR("Misael Lopez Cruz <misael.lopez@ti.com>");
364