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