1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * Copyright (C) 2023 Richtek Technology Corp.
4 *
5 * Authors:
6 * ChiYuan Huang <cy_huang@richtek.com>
7 * Alice Chen <alice_chen@richtek.com>
8 */
9
10 #include <linux/bitfield.h>
11 #include <linux/bitops.h>
12 #include <linux/kernel.h>
13 #include <linux/leds.h>
14 #include <linux/led-class-multicolor.h>
15 #include <linux/linear_range.h>
16 #include <linux/mod_devicetable.h>
17 #include <linux/module.h>
18 #include <linux/mutex.h>
19 #include <linux/platform_device.h>
20 #include <linux/property.h>
21 #include <linux/regmap.h>
22 #include <linux/util_macros.h>
23
24 #include <linux/unaligned.h>
25
26 enum {
27 MT6370_LED_ISNK1 = 0,
28 MT6370_LED_ISNK2,
29 MT6370_LED_ISNK3,
30 MT6370_LED_ISNK4,
31 MT6370_MAX_LEDS
32 };
33
34 enum mt6370_led_mode {
35 MT6370_LED_PWM_MODE = 0,
36 MT6370_LED_BREATH_MODE,
37 MT6370_LED_REG_MODE,
38 MT6370_LED_MAX_MODE
39 };
40
41 enum mt6370_led_field {
42 F_RGB_EN = 0,
43 F_CHGIND_EN,
44 F_LED1_CURR,
45 F_LED2_CURR,
46 F_LED3_CURR,
47 F_LED4_CURR,
48 F_LED1_MODE,
49 F_LED2_MODE,
50 F_LED3_MODE,
51 F_LED4_MODE,
52 F_LED1_DUTY,
53 F_LED2_DUTY,
54 F_LED3_DUTY,
55 F_LED4_DUTY,
56 F_LED1_FREQ,
57 F_LED2_FREQ,
58 F_LED3_FREQ,
59 F_LED4_FREQ,
60 F_MAX_FIELDS
61 };
62
63 enum mt6370_led_ranges {
64 R_LED123_CURR = 0,
65 R_LED4_CURR,
66 R_LED_TRFON,
67 R_LED_TOFF,
68 R_MAX_RANGES
69 };
70
71 enum mt6370_pattern {
72 P_LED_TR1 = 0,
73 P_LED_TR2,
74 P_LED_TF1,
75 P_LED_TF2,
76 P_LED_TON,
77 P_LED_TOFF,
78 P_MAX_PATTERNS
79 };
80
81 #define MT6370_REG_DEV_INFO 0x100
82 #define MT6370_REG_RGB1_DIM 0x182
83 #define MT6370_REG_RGB2_DIM 0x183
84 #define MT6370_REG_RGB3_DIM 0x184
85 #define MT6370_REG_RGB_EN 0x185
86 #define MT6370_REG_RGB1_ISNK 0x186
87 #define MT6370_REG_RGB2_ISNK 0x187
88 #define MT6370_REG_RGB3_ISNK 0x188
89 #define MT6370_REG_RGB1_TR 0x189
90 #define MT6370_REG_RGB_CHRIND_DIM 0x192
91 #define MT6370_REG_RGB_CHRIND_CTRL 0x193
92 #define MT6370_REG_RGB_CHRIND_TR 0x194
93
94 #define MT6372_REG_RGB_EN 0x182
95 #define MT6372_REG_RGB1_ISNK 0x183
96 #define MT6372_REG_RGB2_ISNK 0x184
97 #define MT6372_REG_RGB3_ISNK 0x185
98 #define MT6372_REG_RGB4_ISNK 0x186
99 #define MT6372_REG_RGB1_DIM 0x187
100 #define MT6372_REG_RGB2_DIM 0x188
101 #define MT6372_REG_RGB3_DIM 0x189
102 #define MT6372_REG_RGB4_DIM 0x18A
103 #define MT6372_REG_RGB12_FREQ 0x18B
104 #define MT6372_REG_RGB34_FREQ 0x18C
105 #define MT6372_REG_RGB1_TR 0x18D
106
107 #define MT6370_VENDOR_ID_MASK GENMASK(7, 4)
108 #define MT6372_VENDOR_ID 0x9
109 #define MT6372C_VENDOR_ID 0xb
110 #define MT6370_CHEN_BIT(id) BIT(MT6370_LED_ISNK4 - id)
111 #define MT6370_VIRTUAL_MULTICOLOR 5
112 #define MC_CHANNEL_NUM 3
113 #define MT6370_PWM_DUTY (BIT(5) - 1)
114 #define MT6372_PWM_DUTY (BIT(8) - 1)
115
116 struct mt6370_led {
117 /*
118 * If the color of the LED in DT is set to
119 * - 'LED_COLOR_ID_RGB'
120 * - 'LED_COLOR_ID_MULTI'
121 * The member 'index' of this struct will be set to
122 * 'MT6370_VIRTUAL_MULTICOLOR'.
123 * If so, this LED will choose 'struct led_classdev_mc mc' to use.
124 * Instead, if the member 'index' of this struct is set to
125 * 'MT6370_LED_ISNK1' ~ 'MT6370_LED_ISNK4', then this LED will choose
126 * 'struct led_classdev isink' to use.
127 */
128 union {
129 struct led_classdev isink;
130 struct led_classdev_mc mc;
131 };
132 struct mt6370_priv *priv;
133 enum led_default_state default_state;
134 u32 index;
135 };
136
137 struct mt6370_pdata {
138 const unsigned int *tfreq;
139 unsigned int tfreq_len;
140 u16 reg_rgb1_tr;
141 s16 reg_rgb_chrind_tr;
142 u8 pwm_duty;
143 };
144
145 struct mt6370_priv {
146 /* Per LED access lock */
147 struct mutex lock;
148 struct regmap *regmap;
149 struct regmap_field *fields[F_MAX_FIELDS];
150 const struct reg_field *reg_fields;
151 const struct linear_range *ranges;
152 const struct mt6370_pdata *pdata;
153 unsigned int leds_count;
154 unsigned int leds_active;
155 struct mt6370_led leds[] __counted_by(leds_count);
156 };
157
158 static const struct reg_field common_reg_fields[F_MAX_FIELDS] = {
159 [F_RGB_EN] = REG_FIELD(MT6370_REG_RGB_EN, 4, 7),
160 [F_CHGIND_EN] = REG_FIELD(MT6370_REG_RGB_CHRIND_DIM, 7, 7),
161 [F_LED1_CURR] = REG_FIELD(MT6370_REG_RGB1_ISNK, 0, 2),
162 [F_LED2_CURR] = REG_FIELD(MT6370_REG_RGB2_ISNK, 0, 2),
163 [F_LED3_CURR] = REG_FIELD(MT6370_REG_RGB3_ISNK, 0, 2),
164 [F_LED4_CURR] = REG_FIELD(MT6370_REG_RGB_CHRIND_CTRL, 0, 1),
165 [F_LED1_MODE] = REG_FIELD(MT6370_REG_RGB1_DIM, 5, 6),
166 [F_LED2_MODE] = REG_FIELD(MT6370_REG_RGB2_DIM, 5, 6),
167 [F_LED3_MODE] = REG_FIELD(MT6370_REG_RGB3_DIM, 5, 6),
168 [F_LED4_MODE] = REG_FIELD(MT6370_REG_RGB_CHRIND_DIM, 5, 6),
169 [F_LED1_DUTY] = REG_FIELD(MT6370_REG_RGB1_DIM, 0, 4),
170 [F_LED2_DUTY] = REG_FIELD(MT6370_REG_RGB2_DIM, 0, 4),
171 [F_LED3_DUTY] = REG_FIELD(MT6370_REG_RGB3_DIM, 0, 4),
172 [F_LED4_DUTY] = REG_FIELD(MT6370_REG_RGB_CHRIND_DIM, 0, 4),
173 [F_LED1_FREQ] = REG_FIELD(MT6370_REG_RGB1_ISNK, 3, 5),
174 [F_LED2_FREQ] = REG_FIELD(MT6370_REG_RGB2_ISNK, 3, 5),
175 [F_LED3_FREQ] = REG_FIELD(MT6370_REG_RGB3_ISNK, 3, 5),
176 [F_LED4_FREQ] = REG_FIELD(MT6370_REG_RGB_CHRIND_CTRL, 2, 4),
177 };
178
179 static const struct reg_field mt6372_reg_fields[F_MAX_FIELDS] = {
180 [F_RGB_EN] = REG_FIELD(MT6372_REG_RGB_EN, 4, 7),
181 [F_CHGIND_EN] = REG_FIELD(MT6372_REG_RGB_EN, 3, 3),
182 [F_LED1_CURR] = REG_FIELD(MT6372_REG_RGB1_ISNK, 0, 3),
183 [F_LED2_CURR] = REG_FIELD(MT6372_REG_RGB2_ISNK, 0, 3),
184 [F_LED3_CURR] = REG_FIELD(MT6372_REG_RGB3_ISNK, 0, 3),
185 [F_LED4_CURR] = REG_FIELD(MT6372_REG_RGB4_ISNK, 0, 3),
186 [F_LED1_MODE] = REG_FIELD(MT6372_REG_RGB1_ISNK, 6, 7),
187 [F_LED2_MODE] = REG_FIELD(MT6372_REG_RGB2_ISNK, 6, 7),
188 [F_LED3_MODE] = REG_FIELD(MT6372_REG_RGB3_ISNK, 6, 7),
189 [F_LED4_MODE] = REG_FIELD(MT6372_REG_RGB4_ISNK, 6, 7),
190 [F_LED1_DUTY] = REG_FIELD(MT6372_REG_RGB1_DIM, 0, 7),
191 [F_LED2_DUTY] = REG_FIELD(MT6372_REG_RGB2_DIM, 0, 7),
192 [F_LED3_DUTY] = REG_FIELD(MT6372_REG_RGB3_DIM, 0, 7),
193 [F_LED4_DUTY] = REG_FIELD(MT6372_REG_RGB4_DIM, 0, 7),
194 [F_LED1_FREQ] = REG_FIELD(MT6372_REG_RGB12_FREQ, 5, 7),
195 [F_LED2_FREQ] = REG_FIELD(MT6372_REG_RGB12_FREQ, 2, 4),
196 [F_LED3_FREQ] = REG_FIELD(MT6372_REG_RGB34_FREQ, 5, 7),
197 [F_LED4_FREQ] = REG_FIELD(MT6372_REG_RGB34_FREQ, 2, 4),
198 };
199
200 /* Current unit: microamp, time unit: millisecond */
201 static const struct linear_range common_led_ranges[R_MAX_RANGES] = {
202 [R_LED123_CURR] = { 4000, 1, 6, 4000 },
203 [R_LED4_CURR] = { 2000, 1, 3, 2000 },
204 [R_LED_TRFON] = { 125, 0, 15, 200 },
205 [R_LED_TOFF] = { 250, 0, 15, 400 },
206 };
207
208 static const struct linear_range mt6372_led_ranges[R_MAX_RANGES] = {
209 [R_LED123_CURR] = { 2000, 1, 14, 2000 },
210 [R_LED4_CURR] = { 2000, 1, 14, 2000 },
211 [R_LED_TRFON] = { 125, 0, 15, 250 },
212 [R_LED_TOFF] = { 250, 0, 15, 500 },
213 };
214
215 static const unsigned int common_tfreqs[] = {
216 10000, 5000, 2000, 1000, 500, 200, 5, 1,
217 };
218
219 static const unsigned int mt6372_tfreqs[] = {
220 8000, 4000, 2000, 1000, 500, 250, 8, 4,
221 };
222
223 static const struct mt6370_pdata common_pdata = {
224 .tfreq = common_tfreqs,
225 .tfreq_len = ARRAY_SIZE(common_tfreqs),
226 .pwm_duty = MT6370_PWM_DUTY,
227 .reg_rgb1_tr = MT6370_REG_RGB1_TR,
228 .reg_rgb_chrind_tr = MT6370_REG_RGB_CHRIND_TR,
229 };
230
231 static const struct mt6370_pdata mt6372_pdata = {
232 .tfreq = mt6372_tfreqs,
233 .tfreq_len = ARRAY_SIZE(mt6372_tfreqs),
234 .pwm_duty = MT6372_PWM_DUTY,
235 .reg_rgb1_tr = MT6372_REG_RGB1_TR,
236 .reg_rgb_chrind_tr = -1,
237 };
238
mt6370_get_led_current_field(unsigned int led_no)239 static enum mt6370_led_field mt6370_get_led_current_field(unsigned int led_no)
240 {
241 switch (led_no) {
242 case MT6370_LED_ISNK1:
243 return F_LED1_CURR;
244 case MT6370_LED_ISNK2:
245 return F_LED2_CURR;
246 case MT6370_LED_ISNK3:
247 return F_LED3_CURR;
248 default:
249 return F_LED4_CURR;
250 }
251 }
252
mt6370_set_led_brightness(struct mt6370_priv * priv,unsigned int led_no,unsigned int level)253 static int mt6370_set_led_brightness(struct mt6370_priv *priv, unsigned int led_no,
254 unsigned int level)
255 {
256 enum mt6370_led_field sel_field;
257
258 sel_field = mt6370_get_led_current_field(led_no);
259
260 return regmap_field_write(priv->fields[sel_field], level);
261 }
262
mt6370_get_led_brightness(struct mt6370_priv * priv,unsigned int led_no,unsigned int * level)263 static int mt6370_get_led_brightness(struct mt6370_priv *priv, unsigned int led_no,
264 unsigned int *level)
265 {
266 enum mt6370_led_field sel_field;
267
268 sel_field = mt6370_get_led_current_field(led_no);
269
270 return regmap_field_read(priv->fields[sel_field], level);
271 }
272
mt6370_set_led_duty(struct mt6370_priv * priv,unsigned int led_no,unsigned int ton,unsigned int toff)273 static int mt6370_set_led_duty(struct mt6370_priv *priv, unsigned int led_no, unsigned int ton,
274 unsigned int toff)
275 {
276 const struct mt6370_pdata *pdata = priv->pdata;
277 enum mt6370_led_field sel_field;
278 unsigned int divisor, ratio;
279
280 divisor = pdata->pwm_duty;
281 ratio = ton * divisor / (ton + toff);
282
283 switch (led_no) {
284 case MT6370_LED_ISNK1:
285 sel_field = F_LED1_DUTY;
286 break;
287 case MT6370_LED_ISNK2:
288 sel_field = F_LED2_DUTY;
289 break;
290 case MT6370_LED_ISNK3:
291 sel_field = F_LED3_DUTY;
292 break;
293 default:
294 sel_field = F_LED4_DUTY;
295 break;
296 }
297
298 return regmap_field_write(priv->fields[sel_field], ratio);
299 }
300
mt6370_set_led_freq(struct mt6370_priv * priv,unsigned int led_no,unsigned int ton,unsigned int toff)301 static int mt6370_set_led_freq(struct mt6370_priv *priv, unsigned int led_no, unsigned int ton,
302 unsigned int toff)
303 {
304 const struct mt6370_pdata *pdata = priv->pdata;
305 enum mt6370_led_field sel_field;
306 unsigned int tfreq_len = pdata->tfreq_len;
307 unsigned int tsum, sel;
308
309 tsum = ton + toff;
310
311 if (tsum > pdata->tfreq[0] || tsum < pdata->tfreq[tfreq_len - 1])
312 return -EOPNOTSUPP;
313
314 sel = find_closest_descending(tsum, pdata->tfreq, tfreq_len);
315
316 switch (led_no) {
317 case MT6370_LED_ISNK1:
318 sel_field = F_LED1_FREQ;
319 break;
320 case MT6370_LED_ISNK2:
321 sel_field = F_LED2_FREQ;
322 break;
323 case MT6370_LED_ISNK3:
324 sel_field = F_LED3_FREQ;
325 break;
326 default:
327 sel_field = F_LED4_FREQ;
328 break;
329 }
330
331 return regmap_field_write(priv->fields[sel_field], sel);
332 }
333
mt6370_get_breath_reg_base(struct mt6370_priv * priv,unsigned int led_no,unsigned int * base)334 static void mt6370_get_breath_reg_base(struct mt6370_priv *priv, unsigned int led_no,
335 unsigned int *base)
336 {
337 const struct mt6370_pdata *pdata = priv->pdata;
338
339 if (pdata->reg_rgb_chrind_tr < 0) {
340 *base = pdata->reg_rgb1_tr + led_no * 3;
341 return;
342 }
343
344 switch (led_no) {
345 case MT6370_LED_ISNK1:
346 case MT6370_LED_ISNK2:
347 case MT6370_LED_ISNK3:
348 *base = pdata->reg_rgb1_tr + led_no * 3;
349 break;
350 default:
351 *base = pdata->reg_rgb_chrind_tr;
352 break;
353 }
354 }
355
mt6370_gen_breath_pattern(struct mt6370_priv * priv,struct led_pattern * pattern,u32 len,u8 * pattern_val,u32 val_len)356 static int mt6370_gen_breath_pattern(struct mt6370_priv *priv, struct led_pattern *pattern, u32 len,
357 u8 *pattern_val, u32 val_len)
358 {
359 enum mt6370_led_ranges sel_range;
360 struct led_pattern *curr;
361 unsigned int sel;
362 u32 val = 0;
363 int i;
364
365 if (len < P_MAX_PATTERNS && val_len < P_MAX_PATTERNS / 2)
366 return -EINVAL;
367
368 /*
369 * Pattern list
370 * tr1: byte 0, b'[7:4]
371 * tr2: byte 0, b'[3:0]
372 * tf1: byte 1, b'[7:4]
373 * tf2: byte 1, b'[3:0]
374 * ton: byte 2, b'[7:4]
375 * toff: byte 2, b'[3:0]
376 */
377 for (i = 0; i < P_MAX_PATTERNS; i++) {
378 curr = pattern + i;
379
380 sel_range = i == P_LED_TOFF ? R_LED_TOFF : R_LED_TRFON;
381
382 linear_range_get_selector_within(priv->ranges + sel_range, curr->delta_t, &sel);
383
384 if (i % 2) {
385 val |= sel;
386 } else {
387 val <<= 8;
388 val |= sel << 4;
389 }
390 }
391
392 put_unaligned_be24(val, pattern_val);
393
394 return 0;
395 }
396
mt6370_set_led_mode(struct mt6370_priv * priv,unsigned int led_no,enum mt6370_led_mode mode)397 static int mt6370_set_led_mode(struct mt6370_priv *priv, unsigned int led_no,
398 enum mt6370_led_mode mode)
399 {
400 enum mt6370_led_field sel_field;
401
402 switch (led_no) {
403 case MT6370_LED_ISNK1:
404 sel_field = F_LED1_MODE;
405 break;
406 case MT6370_LED_ISNK2:
407 sel_field = F_LED2_MODE;
408 break;
409 case MT6370_LED_ISNK3:
410 sel_field = F_LED3_MODE;
411 break;
412 default:
413 sel_field = F_LED4_MODE;
414 break;
415 }
416
417 return regmap_field_write(priv->fields[sel_field], mode);
418 }
419
mt6370_mc_brightness_set(struct led_classdev * lcdev,enum led_brightness level)420 static int mt6370_mc_brightness_set(struct led_classdev *lcdev, enum led_brightness level)
421 {
422 struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev);
423 struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc);
424 struct mt6370_priv *priv = led->priv;
425 struct mc_subled *subled;
426 unsigned int enable, disable;
427 int i, ret;
428
429 mutex_lock(&priv->lock);
430
431 led_mc_calc_color_components(mccdev, level);
432
433 ret = regmap_field_read(priv->fields[F_RGB_EN], &enable);
434 if (ret)
435 goto out_unlock;
436
437 disable = enable;
438
439 for (i = 0; i < mccdev->num_colors; i++) {
440 u32 brightness;
441
442 subled = mccdev->subled_info + i;
443 brightness = min(subled->brightness, lcdev->max_brightness);
444 disable &= ~MT6370_CHEN_BIT(subled->channel);
445
446 if (level == 0) {
447 enable &= ~MT6370_CHEN_BIT(subled->channel);
448
449 ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_REG_MODE);
450 if (ret)
451 goto out_unlock;
452
453 continue;
454 }
455
456 if (brightness == 0) {
457 enable &= ~MT6370_CHEN_BIT(subled->channel);
458 continue;
459 }
460
461 enable |= MT6370_CHEN_BIT(subled->channel);
462
463 ret = mt6370_set_led_brightness(priv, subled->channel, brightness);
464 if (ret)
465 goto out_unlock;
466 }
467
468 ret = regmap_field_write(priv->fields[F_RGB_EN], disable);
469 if (ret)
470 goto out_unlock;
471
472 ret = regmap_field_write(priv->fields[F_RGB_EN], enable);
473
474 out_unlock:
475 mutex_unlock(&priv->lock);
476
477 return ret;
478 }
479
mt6370_mc_blink_set(struct led_classdev * lcdev,unsigned long * delay_on,unsigned long * delay_off)480 static int mt6370_mc_blink_set(struct led_classdev *lcdev,
481 unsigned long *delay_on,
482 unsigned long *delay_off)
483 {
484 struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev);
485 struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc);
486 struct mt6370_priv *priv = led->priv;
487 struct mc_subled *subled;
488 unsigned int enable, disable;
489 int i, ret;
490
491 mutex_lock(&priv->lock);
492
493 if (!*delay_on && !*delay_off)
494 *delay_on = *delay_off = 500;
495
496 ret = regmap_field_read(priv->fields[F_RGB_EN], &enable);
497 if (ret)
498 goto out_unlock;
499
500 disable = enable;
501
502 for (i = 0; i < mccdev->num_colors; i++) {
503 subled = mccdev->subled_info + i;
504
505 disable &= ~MT6370_CHEN_BIT(subled->channel);
506
507 ret = mt6370_set_led_duty(priv, subled->channel, *delay_on, *delay_off);
508 if (ret)
509 goto out_unlock;
510
511 ret = mt6370_set_led_freq(priv, subled->channel, *delay_on, *delay_off);
512 if (ret)
513 goto out_unlock;
514
515 ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_PWM_MODE);
516 if (ret)
517 goto out_unlock;
518 }
519
520 /* Toggle to make pattern timing the same */
521 ret = regmap_field_write(priv->fields[F_RGB_EN], disable);
522 if (ret)
523 goto out_unlock;
524
525 ret = regmap_field_write(priv->fields[F_RGB_EN], enable);
526
527 out_unlock:
528 mutex_unlock(&priv->lock);
529
530 return ret;
531 }
532
mt6370_mc_pattern_set(struct led_classdev * lcdev,struct led_pattern * pattern,u32 len,int repeat)533 static int mt6370_mc_pattern_set(struct led_classdev *lcdev, struct led_pattern *pattern, u32 len,
534 int repeat)
535 {
536 struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev);
537 struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc);
538 struct mt6370_priv *priv = led->priv;
539 struct mc_subled *subled;
540 unsigned int reg_base, enable, disable;
541 u8 params[P_MAX_PATTERNS / 2];
542 int i, ret;
543
544 mutex_lock(&priv->lock);
545
546 ret = mt6370_gen_breath_pattern(priv, pattern, len, params, sizeof(params));
547 if (ret)
548 goto out_unlock;
549
550 ret = regmap_field_read(priv->fields[F_RGB_EN], &enable);
551 if (ret)
552 goto out_unlock;
553
554 disable = enable;
555
556 for (i = 0; i < mccdev->num_colors; i++) {
557 subled = mccdev->subled_info + i;
558
559 mt6370_get_breath_reg_base(priv, subled->channel, ®_base);
560 disable &= ~MT6370_CHEN_BIT(subled->channel);
561
562 ret = regmap_raw_write(priv->regmap, reg_base, params, sizeof(params));
563 if (ret)
564 goto out_unlock;
565
566 ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_BREATH_MODE);
567 if (ret)
568 goto out_unlock;
569 }
570
571 /* Toggle to make pattern timing be the same */
572 ret = regmap_field_write(priv->fields[F_RGB_EN], disable);
573 if (ret)
574 goto out_unlock;
575
576 ret = regmap_field_write(priv->fields[F_RGB_EN], enable);
577
578 out_unlock:
579 mutex_unlock(&priv->lock);
580
581 return ret;
582 }
583
mt6370_mc_pattern_clear(struct led_classdev * lcdev)584 static inline int mt6370_mc_pattern_clear(struct led_classdev *lcdev)
585 {
586 struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev);
587 struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc);
588 struct mt6370_priv *priv = led->priv;
589 struct mc_subled *subled;
590 int i, ret = 0;
591
592 mutex_lock(&led->priv->lock);
593
594 for (i = 0; i < mccdev->num_colors; i++) {
595 subled = mccdev->subled_info + i;
596
597 ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_REG_MODE);
598 if (ret)
599 break;
600 }
601
602 mutex_unlock(&led->priv->lock);
603
604 return ret;
605 }
606
mt6370_isnk_brightness_set(struct led_classdev * lcdev,enum led_brightness level)607 static int mt6370_isnk_brightness_set(struct led_classdev *lcdev,
608 enum led_brightness level)
609 {
610 struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink);
611 struct mt6370_priv *priv = led->priv;
612 unsigned int enable;
613 int ret;
614
615 mutex_lock(&priv->lock);
616
617 ret = regmap_field_read(priv->fields[F_RGB_EN], &enable);
618 if (ret)
619 goto out_unlock;
620
621 if (level == 0) {
622 enable &= ~MT6370_CHEN_BIT(led->index);
623
624 ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_REG_MODE);
625 if (ret)
626 goto out_unlock;
627 } else {
628 enable |= MT6370_CHEN_BIT(led->index);
629
630 ret = mt6370_set_led_brightness(priv, led->index, level);
631 if (ret)
632 goto out_unlock;
633 }
634
635 ret = regmap_field_write(priv->fields[F_RGB_EN], enable);
636
637 out_unlock:
638 mutex_unlock(&priv->lock);
639
640 return ret;
641 }
642
mt6370_isnk_blink_set(struct led_classdev * lcdev,unsigned long * delay_on,unsigned long * delay_off)643 static int mt6370_isnk_blink_set(struct led_classdev *lcdev, unsigned long *delay_on,
644 unsigned long *delay_off)
645 {
646 struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink);
647 struct mt6370_priv *priv = led->priv;
648 int ret;
649
650 mutex_lock(&priv->lock);
651
652 if (!*delay_on && !*delay_off)
653 *delay_on = *delay_off = 500;
654
655 ret = mt6370_set_led_duty(priv, led->index, *delay_on, *delay_off);
656 if (ret)
657 goto out_unlock;
658
659 ret = mt6370_set_led_freq(priv, led->index, *delay_on, *delay_off);
660 if (ret)
661 goto out_unlock;
662
663 ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_PWM_MODE);
664
665 out_unlock:
666 mutex_unlock(&priv->lock);
667
668 return ret;
669 }
670
mt6370_isnk_pattern_set(struct led_classdev * lcdev,struct led_pattern * pattern,u32 len,int repeat)671 static int mt6370_isnk_pattern_set(struct led_classdev *lcdev, struct led_pattern *pattern, u32 len,
672 int repeat)
673 {
674 struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink);
675 struct mt6370_priv *priv = led->priv;
676 unsigned int reg_base;
677 u8 params[P_MAX_PATTERNS / 2];
678 int ret;
679
680 mutex_lock(&priv->lock);
681
682 ret = mt6370_gen_breath_pattern(priv, pattern, len, params, sizeof(params));
683 if (ret)
684 goto out_unlock;
685
686 mt6370_get_breath_reg_base(priv, led->index, ®_base);
687
688 ret = regmap_raw_write(priv->regmap, reg_base, params, sizeof(params));
689 if (ret)
690 goto out_unlock;
691
692 ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_BREATH_MODE);
693
694 out_unlock:
695 mutex_unlock(&priv->lock);
696
697 return ret;
698 }
699
mt6370_isnk_pattern_clear(struct led_classdev * lcdev)700 static inline int mt6370_isnk_pattern_clear(struct led_classdev *lcdev)
701 {
702 struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink);
703 struct mt6370_priv *priv = led->priv;
704 int ret;
705
706 mutex_lock(&led->priv->lock);
707 ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_REG_MODE);
708 mutex_unlock(&led->priv->lock);
709
710 return ret;
711 }
712
mt6370_assign_multicolor_info(struct device * dev,struct mt6370_led * led,struct fwnode_handle * fwnode)713 static int mt6370_assign_multicolor_info(struct device *dev, struct mt6370_led *led,
714 struct fwnode_handle *fwnode)
715 {
716 struct mt6370_priv *priv = led->priv;
717 struct fwnode_handle *child;
718 struct mc_subled *sub_led;
719 u32 num_color = 0;
720 int ret;
721
722 sub_led = devm_kcalloc(dev, MC_CHANNEL_NUM, sizeof(*sub_led), GFP_KERNEL);
723 if (!sub_led)
724 return -ENOMEM;
725
726 fwnode_for_each_child_node(fwnode, child) {
727 u32 reg, color;
728
729 ret = fwnode_property_read_u32(child, "reg", ®);
730 if (ret || reg > MT6370_LED_ISNK3 || priv->leds_active & BIT(reg)) {
731 fwnode_handle_put(child);
732 return -EINVAL;
733 }
734
735 ret = fwnode_property_read_u32(child, "color", &color);
736 if (ret) {
737 fwnode_handle_put(child);
738 return dev_err_probe(dev, ret, "LED %d, no color specified\n", led->index);
739 }
740
741 priv->leds_active |= BIT(reg);
742 sub_led[num_color].color_index = color;
743 sub_led[num_color].channel = reg;
744 sub_led[num_color].intensity = 0;
745 num_color++;
746 }
747
748 if (num_color < 2)
749 return dev_err_probe(dev, -EINVAL,
750 "Multicolor must include 2 or more LED channels\n");
751
752 led->mc.num_colors = num_color;
753 led->mc.subled_info = sub_led;
754
755 return 0;
756 }
757
mt6370_init_led_properties(struct device * dev,struct mt6370_led * led,struct led_init_data * init_data)758 static int mt6370_init_led_properties(struct device *dev, struct mt6370_led *led,
759 struct led_init_data *init_data)
760 {
761 struct mt6370_priv *priv = led->priv;
762 struct led_classdev *lcdev;
763 enum mt6370_led_ranges sel_range;
764 u32 max_uA, max_level;
765 int ret;
766
767 if (led->index == MT6370_VIRTUAL_MULTICOLOR) {
768 ret = mt6370_assign_multicolor_info(dev, led, init_data->fwnode);
769 if (ret)
770 return ret;
771
772 lcdev = &led->mc.led_cdev;
773 lcdev->brightness_set_blocking = mt6370_mc_brightness_set;
774 lcdev->blink_set = mt6370_mc_blink_set;
775 lcdev->pattern_set = mt6370_mc_pattern_set;
776 lcdev->pattern_clear = mt6370_mc_pattern_clear;
777 } else {
778 lcdev = &led->isink;
779 lcdev->brightness_set_blocking = mt6370_isnk_brightness_set;
780 lcdev->blink_set = mt6370_isnk_blink_set;
781 lcdev->pattern_set = mt6370_isnk_pattern_set;
782 lcdev->pattern_clear = mt6370_isnk_pattern_clear;
783 }
784
785 ret = fwnode_property_read_u32(init_data->fwnode, "led-max-microamp", &max_uA);
786 if (ret) {
787 dev_warn(dev, "Not specified led-max-microamp, config to the minimum\n");
788 max_uA = 0;
789 }
790
791 if (led->index == MT6370_LED_ISNK4)
792 sel_range = R_LED4_CURR;
793 else
794 sel_range = R_LED123_CURR;
795
796 linear_range_get_selector_within(priv->ranges + sel_range, max_uA, &max_level);
797
798 lcdev->max_brightness = max_level;
799
800 led->default_state = led_init_default_state_get(init_data->fwnode);
801
802 return 0;
803 }
804
mt6370_isnk_init_default_state(struct mt6370_led * led)805 static int mt6370_isnk_init_default_state(struct mt6370_led *led)
806 {
807 struct mt6370_priv *priv = led->priv;
808 unsigned int enable, level;
809 int ret;
810
811 ret = mt6370_get_led_brightness(priv, led->index, &level);
812 if (ret)
813 return ret;
814
815 ret = regmap_field_read(priv->fields[F_RGB_EN], &enable);
816 if (ret)
817 return ret;
818
819 if (!(enable & MT6370_CHEN_BIT(led->index)))
820 level = 0;
821
822 switch (led->default_state) {
823 case LEDS_DEFSTATE_ON:
824 led->isink.brightness = led->isink.max_brightness;
825 break;
826 case LEDS_DEFSTATE_KEEP:
827 led->isink.brightness = min(level, led->isink.max_brightness);
828 break;
829 default:
830 led->isink.brightness = 0;
831 break;
832 }
833
834 return mt6370_isnk_brightness_set(&led->isink, led->isink.brightness);
835 }
836
mt6370_multicolor_led_register(struct device * dev,struct mt6370_led * led,struct led_init_data * init_data)837 static int mt6370_multicolor_led_register(struct device *dev, struct mt6370_led *led,
838 struct led_init_data *init_data)
839 {
840 int ret;
841
842 ret = mt6370_mc_brightness_set(&led->mc.led_cdev, 0);
843 if (ret)
844 return dev_err_probe(dev, ret, "Couldn't set multicolor brightness\n");
845
846 ret = devm_led_classdev_multicolor_register_ext(dev, &led->mc, init_data);
847 if (ret)
848 return dev_err_probe(dev, ret, "Couldn't register multicolor\n");
849
850 return 0;
851 }
852
mt6370_led_register(struct device * dev,struct mt6370_led * led,struct led_init_data * init_data)853 static int mt6370_led_register(struct device *dev, struct mt6370_led *led,
854 struct led_init_data *init_data)
855 {
856 struct mt6370_priv *priv = led->priv;
857 int ret;
858
859 if (led->index == MT6370_VIRTUAL_MULTICOLOR)
860 return mt6370_multicolor_led_register(dev, led, init_data);
861
862 /* If ISNK4 is declared, change its mode from HW auto to SW control */
863 if (led->index == MT6370_LED_ISNK4) {
864 ret = regmap_field_write(priv->fields[F_CHGIND_EN], 1);
865 if (ret)
866 return dev_err_probe(dev, ret, "Failed to set CHRIND to SW\n");
867 }
868
869 ret = mt6370_isnk_init_default_state(led);
870 if (ret)
871 return dev_err_probe(dev, ret, "Failed to init %d isnk state\n", led->index);
872
873 ret = devm_led_classdev_register_ext(dev, &led->isink, init_data);
874 if (ret)
875 return dev_err_probe(dev, ret, "Couldn't register isink %d\n", led->index);
876
877 return 0;
878 }
879
mt6370_check_vendor_info(struct mt6370_priv * priv)880 static int mt6370_check_vendor_info(struct mt6370_priv *priv)
881 {
882 unsigned int devinfo, vid;
883 int ret;
884
885 ret = regmap_read(priv->regmap, MT6370_REG_DEV_INFO, &devinfo);
886 if (ret)
887 return ret;
888
889 vid = FIELD_GET(MT6370_VENDOR_ID_MASK, devinfo);
890 if (vid == MT6372_VENDOR_ID || vid == MT6372C_VENDOR_ID) {
891 priv->reg_fields = mt6372_reg_fields;
892 priv->ranges = mt6372_led_ranges;
893 priv->pdata = &mt6372_pdata;
894 } else {
895 /* Common for MT6370/71 */
896 priv->reg_fields = common_reg_fields;
897 priv->ranges = common_led_ranges;
898 priv->pdata = &common_pdata;
899 }
900
901 return 0;
902 }
903
mt6370_leds_probe(struct platform_device * pdev)904 static int mt6370_leds_probe(struct platform_device *pdev)
905 {
906 struct device *dev = &pdev->dev;
907 struct mt6370_priv *priv;
908 size_t count;
909 unsigned int i = 0;
910 int ret;
911
912 count = device_get_child_node_count(dev);
913 if (!count || count > MT6370_MAX_LEDS)
914 return dev_err_probe(dev, -EINVAL,
915 "No child node or node count over max LED number %zu\n",
916 count);
917
918 priv = devm_kzalloc(dev, struct_size(priv, leds, count), GFP_KERNEL);
919 if (!priv)
920 return -ENOMEM;
921
922 priv->leds_count = count;
923 mutex_init(&priv->lock);
924
925 priv->regmap = dev_get_regmap(dev->parent, NULL);
926 if (!priv->regmap)
927 return dev_err_probe(dev, -ENODEV, "Failed to get parent regmap\n");
928
929 ret = mt6370_check_vendor_info(priv);
930 if (ret)
931 return dev_err_probe(dev, ret, "Failed to check vendor info\n");
932
933 ret = devm_regmap_field_bulk_alloc(dev, priv->regmap, priv->fields, priv->reg_fields,
934 F_MAX_FIELDS);
935 if (ret)
936 return dev_err_probe(dev, ret, "Failed to allocate regmap field\n");
937
938 device_for_each_child_node_scoped(dev, child) {
939 struct mt6370_led *led = priv->leds + i++;
940 struct led_init_data init_data = { .fwnode = child };
941 u32 reg, color;
942
943 ret = fwnode_property_read_u32(child, "reg", ®);
944 if (ret)
945 dev_err_probe(dev, ret, "Failed to parse reg property\n");
946
947 if (reg >= MT6370_MAX_LEDS)
948 return dev_err_probe(dev, -EINVAL, "Error reg property number\n");
949
950 ret = fwnode_property_read_u32(child, "color", &color);
951 if (ret)
952 return dev_err_probe(dev, ret, "Failed to parse color property\n");
953
954 if (color == LED_COLOR_ID_RGB || color == LED_COLOR_ID_MULTI)
955 reg = MT6370_VIRTUAL_MULTICOLOR;
956
957 if (priv->leds_active & BIT(reg))
958 return dev_err_probe(dev, -EINVAL, "Duplicate reg property\n");
959
960 priv->leds_active |= BIT(reg);
961
962 led->index = reg;
963 led->priv = priv;
964
965 ret = mt6370_init_led_properties(dev, led, &init_data);
966 if (ret)
967 return ret;
968
969 ret = mt6370_led_register(dev, led, &init_data);
970 if (ret)
971 return ret;
972 }
973
974 return 0;
975 }
976
977 static const struct of_device_id mt6370_rgbled_device_table[] = {
978 { .compatible = "mediatek,mt6370-indicator" },
979 {}
980 };
981 MODULE_DEVICE_TABLE(of, mt6370_rgbled_device_table);
982
983 static struct platform_driver mt6370_rgbled_driver = {
984 .driver = {
985 .name = "mt6370-indicator",
986 .of_match_table = mt6370_rgbled_device_table,
987 },
988 .probe = mt6370_leds_probe,
989 };
990 module_platform_driver(mt6370_rgbled_driver);
991
992 MODULE_AUTHOR("Alice Chen <alice_chen@richtek.com>");
993 MODULE_AUTHOR("ChiYuan Huang <cy_huang@richtek.com>");
994 MODULE_DESCRIPTION("MediaTek MT6370 RGB LED Driver");
995 MODULE_LICENSE("GPL");
996