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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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