// SPDX-License-Identifier: GPL-2.0-only /* * Copyright (C) 2023 Richtek Technology Corp. * * Authors: * ChiYuan Huang * Alice Chen */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include enum { MT6370_LED_ISNK1 = 0, MT6370_LED_ISNK2, MT6370_LED_ISNK3, MT6370_LED_ISNK4, MT6370_MAX_LEDS }; enum mt6370_led_mode { MT6370_LED_PWM_MODE = 0, MT6370_LED_BREATH_MODE, MT6370_LED_REG_MODE, MT6370_LED_MAX_MODE }; enum mt6370_led_field { F_RGB_EN = 0, F_CHGIND_EN, F_LED1_CURR, F_LED2_CURR, F_LED3_CURR, F_LED4_CURR, F_LED1_MODE, F_LED2_MODE, F_LED3_MODE, F_LED4_MODE, F_LED1_DUTY, F_LED2_DUTY, F_LED3_DUTY, F_LED4_DUTY, F_LED1_FREQ, F_LED2_FREQ, F_LED3_FREQ, F_LED4_FREQ, F_MAX_FIELDS }; enum mt6370_led_ranges { R_LED123_CURR = 0, R_LED4_CURR, R_LED_TRFON, R_LED_TOFF, R_MAX_RANGES }; enum mt6370_pattern { P_LED_TR1 = 0, P_LED_TR2, P_LED_TF1, P_LED_TF2, P_LED_TON, P_LED_TOFF, P_MAX_PATTERNS }; #define MT6370_REG_DEV_INFO 0x100 #define MT6370_REG_RGB1_DIM 0x182 #define MT6370_REG_RGB2_DIM 0x183 #define MT6370_REG_RGB3_DIM 0x184 #define MT6370_REG_RGB_EN 0x185 #define MT6370_REG_RGB1_ISNK 0x186 #define MT6370_REG_RGB2_ISNK 0x187 #define MT6370_REG_RGB3_ISNK 0x188 #define MT6370_REG_RGB1_TR 0x189 #define MT6370_REG_RGB_CHRIND_DIM 0x192 #define MT6370_REG_RGB_CHRIND_CTRL 0x193 #define MT6370_REG_RGB_CHRIND_TR 0x194 #define MT6372_REG_RGB_EN 0x182 #define MT6372_REG_RGB1_ISNK 0x183 #define MT6372_REG_RGB2_ISNK 0x184 #define MT6372_REG_RGB3_ISNK 0x185 #define MT6372_REG_RGB4_ISNK 0x186 #define MT6372_REG_RGB1_DIM 0x187 #define MT6372_REG_RGB2_DIM 0x188 #define MT6372_REG_RGB3_DIM 0x189 #define MT6372_REG_RGB4_DIM 0x18A #define MT6372_REG_RGB12_FREQ 0x18B #define MT6372_REG_RGB34_FREQ 0x18C #define MT6372_REG_RGB1_TR 0x18D #define MT6370_VENDOR_ID_MASK GENMASK(7, 4) #define MT6372_VENDOR_ID 0x9 #define MT6372C_VENDOR_ID 0xb #define MT6370_CHEN_BIT(id) BIT(MT6370_LED_ISNK4 - id) #define MT6370_VIRTUAL_MULTICOLOR 5 #define MC_CHANNEL_NUM 3 #define MT6370_PWM_DUTY (BIT(5) - 1) #define MT6372_PWM_DUTY (BIT(8) - 1) struct mt6370_led { /* * If the color of the LED in DT is set to * - 'LED_COLOR_ID_RGB' * - 'LED_COLOR_ID_MULTI' * The member 'index' of this struct will be set to * 'MT6370_VIRTUAL_MULTICOLOR'. * If so, this LED will choose 'struct led_classdev_mc mc' to use. * Instead, if the member 'index' of this struct is set to * 'MT6370_LED_ISNK1' ~ 'MT6370_LED_ISNK4', then this LED will choose * 'struct led_classdev isink' to use. */ union { struct led_classdev isink; struct led_classdev_mc mc; }; struct mt6370_priv *priv; enum led_default_state default_state; u32 index; }; struct mt6370_pdata { const unsigned int *tfreq; unsigned int tfreq_len; u16 reg_rgb1_tr; s16 reg_rgb_chrind_tr; u8 pwm_duty; }; struct mt6370_priv { /* Per LED access lock */ struct mutex lock; struct regmap *regmap; struct regmap_field *fields[F_MAX_FIELDS]; const struct reg_field *reg_fields; const struct linear_range *ranges; const struct mt6370_pdata *pdata; unsigned int leds_count; unsigned int leds_active; struct mt6370_led leds[] __counted_by(leds_count); }; static const struct reg_field common_reg_fields[F_MAX_FIELDS] = { [F_RGB_EN] = REG_FIELD(MT6370_REG_RGB_EN, 4, 7), [F_CHGIND_EN] = REG_FIELD(MT6370_REG_RGB_CHRIND_DIM, 7, 7), [F_LED1_CURR] = REG_FIELD(MT6370_REG_RGB1_ISNK, 0, 2), [F_LED2_CURR] = REG_FIELD(MT6370_REG_RGB2_ISNK, 0, 2), [F_LED3_CURR] = REG_FIELD(MT6370_REG_RGB3_ISNK, 0, 2), [F_LED4_CURR] = REG_FIELD(MT6370_REG_RGB_CHRIND_CTRL, 0, 1), [F_LED1_MODE] = REG_FIELD(MT6370_REG_RGB1_DIM, 5, 6), [F_LED2_MODE] = REG_FIELD(MT6370_REG_RGB2_DIM, 5, 6), [F_LED3_MODE] = REG_FIELD(MT6370_REG_RGB3_DIM, 5, 6), [F_LED4_MODE] = REG_FIELD(MT6370_REG_RGB_CHRIND_DIM, 5, 6), [F_LED1_DUTY] = REG_FIELD(MT6370_REG_RGB1_DIM, 0, 4), [F_LED2_DUTY] = REG_FIELD(MT6370_REG_RGB2_DIM, 0, 4), [F_LED3_DUTY] = REG_FIELD(MT6370_REG_RGB3_DIM, 0, 4), [F_LED4_DUTY] = REG_FIELD(MT6370_REG_RGB_CHRIND_DIM, 0, 4), [F_LED1_FREQ] = REG_FIELD(MT6370_REG_RGB1_ISNK, 3, 5), [F_LED2_FREQ] = REG_FIELD(MT6370_REG_RGB2_ISNK, 3, 5), [F_LED3_FREQ] = REG_FIELD(MT6370_REG_RGB3_ISNK, 3, 5), [F_LED4_FREQ] = REG_FIELD(MT6370_REG_RGB_CHRIND_CTRL, 2, 4), }; static const struct reg_field mt6372_reg_fields[F_MAX_FIELDS] = { [F_RGB_EN] = REG_FIELD(MT6372_REG_RGB_EN, 4, 7), [F_CHGIND_EN] = REG_FIELD(MT6372_REG_RGB_EN, 3, 3), [F_LED1_CURR] = REG_FIELD(MT6372_REG_RGB1_ISNK, 0, 3), [F_LED2_CURR] = REG_FIELD(MT6372_REG_RGB2_ISNK, 0, 3), [F_LED3_CURR] = REG_FIELD(MT6372_REG_RGB3_ISNK, 0, 3), [F_LED4_CURR] = REG_FIELD(MT6372_REG_RGB4_ISNK, 0, 3), [F_LED1_MODE] = REG_FIELD(MT6372_REG_RGB1_ISNK, 6, 7), [F_LED2_MODE] = REG_FIELD(MT6372_REG_RGB2_ISNK, 6, 7), [F_LED3_MODE] = REG_FIELD(MT6372_REG_RGB3_ISNK, 6, 7), [F_LED4_MODE] = REG_FIELD(MT6372_REG_RGB4_ISNK, 6, 7), [F_LED1_DUTY] = REG_FIELD(MT6372_REG_RGB1_DIM, 0, 7), [F_LED2_DUTY] = REG_FIELD(MT6372_REG_RGB2_DIM, 0, 7), [F_LED3_DUTY] = REG_FIELD(MT6372_REG_RGB3_DIM, 0, 7), [F_LED4_DUTY] = REG_FIELD(MT6372_REG_RGB4_DIM, 0, 7), [F_LED1_FREQ] = REG_FIELD(MT6372_REG_RGB12_FREQ, 5, 7), [F_LED2_FREQ] = REG_FIELD(MT6372_REG_RGB12_FREQ, 2, 4), [F_LED3_FREQ] = REG_FIELD(MT6372_REG_RGB34_FREQ, 5, 7), [F_LED4_FREQ] = REG_FIELD(MT6372_REG_RGB34_FREQ, 2, 4), }; /* Current unit: microamp, time unit: millisecond */ static const struct linear_range common_led_ranges[R_MAX_RANGES] = { [R_LED123_CURR] = { 4000, 1, 6, 4000 }, [R_LED4_CURR] = { 2000, 1, 3, 2000 }, [R_LED_TRFON] = { 125, 0, 15, 200 }, [R_LED_TOFF] = { 250, 0, 15, 400 }, }; static const struct linear_range mt6372_led_ranges[R_MAX_RANGES] = { [R_LED123_CURR] = { 2000, 1, 14, 2000 }, [R_LED4_CURR] = { 2000, 1, 14, 2000 }, [R_LED_TRFON] = { 125, 0, 15, 250 }, [R_LED_TOFF] = { 250, 0, 15, 500 }, }; static const unsigned int common_tfreqs[] = { 10000, 5000, 2000, 1000, 500, 200, 5, 1, }; static const unsigned int mt6372_tfreqs[] = { 8000, 4000, 2000, 1000, 500, 250, 8, 4, }; static const struct mt6370_pdata common_pdata = { .tfreq = common_tfreqs, .tfreq_len = ARRAY_SIZE(common_tfreqs), .pwm_duty = MT6370_PWM_DUTY, .reg_rgb1_tr = MT6370_REG_RGB1_TR, .reg_rgb_chrind_tr = MT6370_REG_RGB_CHRIND_TR, }; static const struct mt6370_pdata mt6372_pdata = { .tfreq = mt6372_tfreqs, .tfreq_len = ARRAY_SIZE(mt6372_tfreqs), .pwm_duty = MT6372_PWM_DUTY, .reg_rgb1_tr = MT6372_REG_RGB1_TR, .reg_rgb_chrind_tr = -1, }; static enum mt6370_led_field mt6370_get_led_current_field(unsigned int led_no) { switch (led_no) { case MT6370_LED_ISNK1: return F_LED1_CURR; case MT6370_LED_ISNK2: return F_LED2_CURR; case MT6370_LED_ISNK3: return F_LED3_CURR; default: return F_LED4_CURR; } } static int mt6370_set_led_brightness(struct mt6370_priv *priv, unsigned int led_no, unsigned int level) { enum mt6370_led_field sel_field; sel_field = mt6370_get_led_current_field(led_no); return regmap_field_write(priv->fields[sel_field], level); } static int mt6370_get_led_brightness(struct mt6370_priv *priv, unsigned int led_no, unsigned int *level) { enum mt6370_led_field sel_field; sel_field = mt6370_get_led_current_field(led_no); return regmap_field_read(priv->fields[sel_field], level); } static int mt6370_set_led_duty(struct mt6370_priv *priv, unsigned int led_no, unsigned int ton, unsigned int toff) { const struct mt6370_pdata *pdata = priv->pdata; enum mt6370_led_field sel_field; unsigned int divisor, ratio; divisor = pdata->pwm_duty; ratio = ton * divisor / (ton + toff); switch (led_no) { case MT6370_LED_ISNK1: sel_field = F_LED1_DUTY; break; case MT6370_LED_ISNK2: sel_field = F_LED2_DUTY; break; case MT6370_LED_ISNK3: sel_field = F_LED3_DUTY; break; default: sel_field = F_LED4_DUTY; break; } return regmap_field_write(priv->fields[sel_field], ratio); } static int mt6370_set_led_freq(struct mt6370_priv *priv, unsigned int led_no, unsigned int ton, unsigned int toff) { const struct mt6370_pdata *pdata = priv->pdata; enum mt6370_led_field sel_field; unsigned int tfreq_len = pdata->tfreq_len; unsigned int tsum, sel; tsum = ton + toff; if (tsum > pdata->tfreq[0] || tsum < pdata->tfreq[tfreq_len - 1]) return -EOPNOTSUPP; sel = find_closest_descending(tsum, pdata->tfreq, tfreq_len); switch (led_no) { case MT6370_LED_ISNK1: sel_field = F_LED1_FREQ; break; case MT6370_LED_ISNK2: sel_field = F_LED2_FREQ; break; case MT6370_LED_ISNK3: sel_field = F_LED3_FREQ; break; default: sel_field = F_LED4_FREQ; break; } return regmap_field_write(priv->fields[sel_field], sel); } static void mt6370_get_breath_reg_base(struct mt6370_priv *priv, unsigned int led_no, unsigned int *base) { const struct mt6370_pdata *pdata = priv->pdata; if (pdata->reg_rgb_chrind_tr < 0) { *base = pdata->reg_rgb1_tr + led_no * 3; return; } switch (led_no) { case MT6370_LED_ISNK1: case MT6370_LED_ISNK2: case MT6370_LED_ISNK3: *base = pdata->reg_rgb1_tr + led_no * 3; break; default: *base = pdata->reg_rgb_chrind_tr; break; } } static int mt6370_gen_breath_pattern(struct mt6370_priv *priv, struct led_pattern *pattern, u32 len, u8 *pattern_val, u32 val_len) { enum mt6370_led_ranges sel_range; struct led_pattern *curr; unsigned int sel; u32 val = 0; int i; if (len < P_MAX_PATTERNS && val_len < P_MAX_PATTERNS / 2) return -EINVAL; /* * Pattern list * tr1: byte 0, b'[7:4] * tr2: byte 0, b'[3:0] * tf1: byte 1, b'[7:4] * tf2: byte 1, b'[3:0] * ton: byte 2, b'[7:4] * toff: byte 2, b'[3:0] */ for (i = 0; i < P_MAX_PATTERNS; i++) { curr = pattern + i; sel_range = i == P_LED_TOFF ? R_LED_TOFF : R_LED_TRFON; linear_range_get_selector_within(priv->ranges + sel_range, curr->delta_t, &sel); if (i % 2) { val |= sel; } else { val <<= 8; val |= sel << 4; } } put_unaligned_be24(val, pattern_val); return 0; } static int mt6370_set_led_mode(struct mt6370_priv *priv, unsigned int led_no, enum mt6370_led_mode mode) { enum mt6370_led_field sel_field; switch (led_no) { case MT6370_LED_ISNK1: sel_field = F_LED1_MODE; break; case MT6370_LED_ISNK2: sel_field = F_LED2_MODE; break; case MT6370_LED_ISNK3: sel_field = F_LED3_MODE; break; default: sel_field = F_LED4_MODE; break; } return regmap_field_write(priv->fields[sel_field], mode); } static int mt6370_mc_brightness_set(struct led_classdev *lcdev, enum led_brightness level) { struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev); struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc); struct mt6370_priv *priv = led->priv; struct mc_subled *subled; unsigned int enable, disable; int i, ret; mutex_lock(&priv->lock); led_mc_calc_color_components(mccdev, level); ret = regmap_field_read(priv->fields[F_RGB_EN], &enable); if (ret) goto out_unlock; disable = enable; for (i = 0; i < mccdev->num_colors; i++) { u32 brightness; subled = mccdev->subled_info + i; brightness = min(subled->brightness, lcdev->max_brightness); disable &= ~MT6370_CHEN_BIT(subled->channel); if (level == 0) { enable &= ~MT6370_CHEN_BIT(subled->channel); ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_REG_MODE); if (ret) goto out_unlock; continue; } if (brightness == 0) { enable &= ~MT6370_CHEN_BIT(subled->channel); continue; } enable |= MT6370_CHEN_BIT(subled->channel); ret = mt6370_set_led_brightness(priv, subled->channel, brightness); if (ret) goto out_unlock; } ret = regmap_field_write(priv->fields[F_RGB_EN], disable); if (ret) goto out_unlock; ret = regmap_field_write(priv->fields[F_RGB_EN], enable); out_unlock: mutex_unlock(&priv->lock); return ret; } static int mt6370_mc_blink_set(struct led_classdev *lcdev, unsigned long *delay_on, unsigned long *delay_off) { struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev); struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc); struct mt6370_priv *priv = led->priv; struct mc_subled *subled; unsigned int enable, disable; int i, ret; mutex_lock(&priv->lock); if (!*delay_on && !*delay_off) *delay_on = *delay_off = 500; ret = regmap_field_read(priv->fields[F_RGB_EN], &enable); if (ret) goto out_unlock; disable = enable; for (i = 0; i < mccdev->num_colors; i++) { subled = mccdev->subled_info + i; disable &= ~MT6370_CHEN_BIT(subled->channel); ret = mt6370_set_led_duty(priv, subled->channel, *delay_on, *delay_off); if (ret) goto out_unlock; ret = mt6370_set_led_freq(priv, subled->channel, *delay_on, *delay_off); if (ret) goto out_unlock; ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_PWM_MODE); if (ret) goto out_unlock; } /* Toggle to make pattern timing the same */ ret = regmap_field_write(priv->fields[F_RGB_EN], disable); if (ret) goto out_unlock; ret = regmap_field_write(priv->fields[F_RGB_EN], enable); out_unlock: mutex_unlock(&priv->lock); return ret; } static int mt6370_mc_pattern_set(struct led_classdev *lcdev, struct led_pattern *pattern, u32 len, int repeat) { struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev); struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc); struct mt6370_priv *priv = led->priv; struct mc_subled *subled; unsigned int reg_base, enable, disable; u8 params[P_MAX_PATTERNS / 2]; int i, ret; mutex_lock(&priv->lock); ret = mt6370_gen_breath_pattern(priv, pattern, len, params, sizeof(params)); if (ret) goto out_unlock; ret = regmap_field_read(priv->fields[F_RGB_EN], &enable); if (ret) goto out_unlock; disable = enable; for (i = 0; i < mccdev->num_colors; i++) { subled = mccdev->subled_info + i; mt6370_get_breath_reg_base(priv, subled->channel, ®_base); disable &= ~MT6370_CHEN_BIT(subled->channel); ret = regmap_raw_write(priv->regmap, reg_base, params, sizeof(params)); if (ret) goto out_unlock; ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_BREATH_MODE); if (ret) goto out_unlock; } /* Toggle to make pattern timing be the same */ ret = regmap_field_write(priv->fields[F_RGB_EN], disable); if (ret) goto out_unlock; ret = regmap_field_write(priv->fields[F_RGB_EN], enable); out_unlock: mutex_unlock(&priv->lock); return ret; } static inline int mt6370_mc_pattern_clear(struct led_classdev *lcdev) { struct led_classdev_mc *mccdev = lcdev_to_mccdev(lcdev); struct mt6370_led *led = container_of(mccdev, struct mt6370_led, mc); struct mt6370_priv *priv = led->priv; struct mc_subled *subled; int i, ret = 0; mutex_lock(&led->priv->lock); for (i = 0; i < mccdev->num_colors; i++) { subled = mccdev->subled_info + i; ret = mt6370_set_led_mode(priv, subled->channel, MT6370_LED_REG_MODE); if (ret) break; } mutex_unlock(&led->priv->lock); return ret; } static int mt6370_isnk_brightness_set(struct led_classdev *lcdev, enum led_brightness level) { struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink); struct mt6370_priv *priv = led->priv; unsigned int enable; int ret; mutex_lock(&priv->lock); ret = regmap_field_read(priv->fields[F_RGB_EN], &enable); if (ret) goto out_unlock; if (level == 0) { enable &= ~MT6370_CHEN_BIT(led->index); ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_REG_MODE); if (ret) goto out_unlock; } else { enable |= MT6370_CHEN_BIT(led->index); ret = mt6370_set_led_brightness(priv, led->index, level); if (ret) goto out_unlock; } ret = regmap_field_write(priv->fields[F_RGB_EN], enable); out_unlock: mutex_unlock(&priv->lock); return ret; } static int mt6370_isnk_blink_set(struct led_classdev *lcdev, unsigned long *delay_on, unsigned long *delay_off) { struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink); struct mt6370_priv *priv = led->priv; int ret; mutex_lock(&priv->lock); if (!*delay_on && !*delay_off) *delay_on = *delay_off = 500; ret = mt6370_set_led_duty(priv, led->index, *delay_on, *delay_off); if (ret) goto out_unlock; ret = mt6370_set_led_freq(priv, led->index, *delay_on, *delay_off); if (ret) goto out_unlock; ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_PWM_MODE); out_unlock: mutex_unlock(&priv->lock); return ret; } static int mt6370_isnk_pattern_set(struct led_classdev *lcdev, struct led_pattern *pattern, u32 len, int repeat) { struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink); struct mt6370_priv *priv = led->priv; unsigned int reg_base; u8 params[P_MAX_PATTERNS / 2]; int ret; mutex_lock(&priv->lock); ret = mt6370_gen_breath_pattern(priv, pattern, len, params, sizeof(params)); if (ret) goto out_unlock; mt6370_get_breath_reg_base(priv, led->index, ®_base); ret = regmap_raw_write(priv->regmap, reg_base, params, sizeof(params)); if (ret) goto out_unlock; ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_BREATH_MODE); out_unlock: mutex_unlock(&priv->lock); return ret; } static inline int mt6370_isnk_pattern_clear(struct led_classdev *lcdev) { struct mt6370_led *led = container_of(lcdev, struct mt6370_led, isink); struct mt6370_priv *priv = led->priv; int ret; mutex_lock(&led->priv->lock); ret = mt6370_set_led_mode(priv, led->index, MT6370_LED_REG_MODE); mutex_unlock(&led->priv->lock); return ret; } static int mt6370_assign_multicolor_info(struct device *dev, struct mt6370_led *led, struct fwnode_handle *fwnode) { struct mt6370_priv *priv = led->priv; struct fwnode_handle *child; struct mc_subled *sub_led; u32 num_color = 0; int ret; sub_led = devm_kcalloc(dev, MC_CHANNEL_NUM, sizeof(*sub_led), GFP_KERNEL); if (!sub_led) return -ENOMEM; fwnode_for_each_child_node(fwnode, child) { u32 reg, color; ret = fwnode_property_read_u32(child, "reg", ®); if (ret || reg > MT6370_LED_ISNK3 || priv->leds_active & BIT(reg)) { fwnode_handle_put(child); return -EINVAL; } ret = fwnode_property_read_u32(child, "color", &color); if (ret) { fwnode_handle_put(child); return dev_err_probe(dev, ret, "LED %d, no color specified\n", led->index); } priv->leds_active |= BIT(reg); sub_led[num_color].color_index = color; sub_led[num_color].channel = reg; sub_led[num_color].intensity = 0; num_color++; } if (num_color < 2) return dev_err_probe(dev, -EINVAL, "Multicolor must include 2 or more LED channels\n"); led->mc.num_colors = num_color; led->mc.subled_info = sub_led; return 0; } static int mt6370_init_led_properties(struct device *dev, struct mt6370_led *led, struct led_init_data *init_data) { struct mt6370_priv *priv = led->priv; struct led_classdev *lcdev; enum mt6370_led_ranges sel_range; u32 max_uA, max_level; int ret; if (led->index == MT6370_VIRTUAL_MULTICOLOR) { ret = mt6370_assign_multicolor_info(dev, led, init_data->fwnode); if (ret) return ret; lcdev = &led->mc.led_cdev; lcdev->brightness_set_blocking = mt6370_mc_brightness_set; lcdev->blink_set = mt6370_mc_blink_set; lcdev->pattern_set = mt6370_mc_pattern_set; lcdev->pattern_clear = mt6370_mc_pattern_clear; } else { lcdev = &led->isink; lcdev->brightness_set_blocking = mt6370_isnk_brightness_set; lcdev->blink_set = mt6370_isnk_blink_set; lcdev->pattern_set = mt6370_isnk_pattern_set; lcdev->pattern_clear = mt6370_isnk_pattern_clear; } ret = fwnode_property_read_u32(init_data->fwnode, "led-max-microamp", &max_uA); if (ret) { dev_warn(dev, "Not specified led-max-microamp, config to the minimum\n"); max_uA = 0; } if (led->index == MT6370_LED_ISNK4) sel_range = R_LED4_CURR; else sel_range = R_LED123_CURR; linear_range_get_selector_within(priv->ranges + sel_range, max_uA, &max_level); lcdev->max_brightness = max_level; led->default_state = led_init_default_state_get(init_data->fwnode); return 0; } static int mt6370_isnk_init_default_state(struct mt6370_led *led) { struct mt6370_priv *priv = led->priv; unsigned int enable, level; int ret; ret = mt6370_get_led_brightness(priv, led->index, &level); if (ret) return ret; ret = regmap_field_read(priv->fields[F_RGB_EN], &enable); if (ret) return ret; if (!(enable & MT6370_CHEN_BIT(led->index))) level = 0; switch (led->default_state) { case LEDS_DEFSTATE_ON: led->isink.brightness = led->isink.max_brightness; break; case LEDS_DEFSTATE_KEEP: led->isink.brightness = min(level, led->isink.max_brightness); break; default: led->isink.brightness = 0; break; } return mt6370_isnk_brightness_set(&led->isink, led->isink.brightness); } static int mt6370_multicolor_led_register(struct device *dev, struct mt6370_led *led, struct led_init_data *init_data) { int ret; ret = mt6370_mc_brightness_set(&led->mc.led_cdev, 0); if (ret) return dev_err_probe(dev, ret, "Couldn't set multicolor brightness\n"); ret = devm_led_classdev_multicolor_register_ext(dev, &led->mc, init_data); if (ret) return dev_err_probe(dev, ret, "Couldn't register multicolor\n"); return 0; } static int mt6370_led_register(struct device *dev, struct mt6370_led *led, struct led_init_data *init_data) { struct mt6370_priv *priv = led->priv; int ret; if (led->index == MT6370_VIRTUAL_MULTICOLOR) return mt6370_multicolor_led_register(dev, led, init_data); /* If ISNK4 is declared, change its mode from HW auto to SW control */ if (led->index == MT6370_LED_ISNK4) { ret = regmap_field_write(priv->fields[F_CHGIND_EN], 1); if (ret) return dev_err_probe(dev, ret, "Failed to set CHRIND to SW\n"); } ret = mt6370_isnk_init_default_state(led); if (ret) return dev_err_probe(dev, ret, "Failed to init %d isnk state\n", led->index); ret = devm_led_classdev_register_ext(dev, &led->isink, init_data); if (ret) return dev_err_probe(dev, ret, "Couldn't register isink %d\n", led->index); return 0; } static int mt6370_check_vendor_info(struct mt6370_priv *priv) { unsigned int devinfo, vid; int ret; ret = regmap_read(priv->regmap, MT6370_REG_DEV_INFO, &devinfo); if (ret) return ret; vid = FIELD_GET(MT6370_VENDOR_ID_MASK, devinfo); if (vid == MT6372_VENDOR_ID || vid == MT6372C_VENDOR_ID) { priv->reg_fields = mt6372_reg_fields; priv->ranges = mt6372_led_ranges; priv->pdata = &mt6372_pdata; } else { /* Common for MT6370/71 */ priv->reg_fields = common_reg_fields; priv->ranges = common_led_ranges; priv->pdata = &common_pdata; } return 0; } static int mt6370_leds_probe(struct platform_device *pdev) { struct device *dev = &pdev->dev; struct mt6370_priv *priv; size_t count; unsigned int i = 0; int ret; count = device_get_child_node_count(dev); if (!count || count > MT6370_MAX_LEDS) return dev_err_probe(dev, -EINVAL, "No child node or node count over max LED number %zu\n", count); priv = devm_kzalloc(dev, struct_size(priv, leds, count), GFP_KERNEL); if (!priv) return -ENOMEM; priv->leds_count = count; mutex_init(&priv->lock); priv->regmap = dev_get_regmap(dev->parent, NULL); if (!priv->regmap) return dev_err_probe(dev, -ENODEV, "Failed to get parent regmap\n"); ret = mt6370_check_vendor_info(priv); if (ret) return dev_err_probe(dev, ret, "Failed to check vendor info\n"); ret = devm_regmap_field_bulk_alloc(dev, priv->regmap, priv->fields, priv->reg_fields, F_MAX_FIELDS); if (ret) return dev_err_probe(dev, ret, "Failed to allocate regmap field\n"); device_for_each_child_node_scoped(dev, child) { struct mt6370_led *led = priv->leds + i++; struct led_init_data init_data = { .fwnode = child }; u32 reg, color; ret = fwnode_property_read_u32(child, "reg", ®); if (ret) dev_err_probe(dev, ret, "Failed to parse reg property\n"); if (reg >= MT6370_MAX_LEDS) return dev_err_probe(dev, -EINVAL, "Error reg property number\n"); ret = fwnode_property_read_u32(child, "color", &color); if (ret) return dev_err_probe(dev, ret, "Failed to parse color property\n"); if (color == LED_COLOR_ID_RGB || color == LED_COLOR_ID_MULTI) reg = MT6370_VIRTUAL_MULTICOLOR; if (priv->leds_active & BIT(reg)) return dev_err_probe(dev, -EINVAL, "Duplicate reg property\n"); priv->leds_active |= BIT(reg); led->index = reg; led->priv = priv; ret = mt6370_init_led_properties(dev, led, &init_data); if (ret) return ret; ret = mt6370_led_register(dev, led, &init_data); if (ret) return ret; } return 0; } static const struct of_device_id mt6370_rgbled_device_table[] = { { .compatible = "mediatek,mt6370-indicator" }, {} }; MODULE_DEVICE_TABLE(of, mt6370_rgbled_device_table); static struct platform_driver mt6370_rgbled_driver = { .driver = { .name = "mt6370-indicator", .of_match_table = mt6370_rgbled_device_table, }, .probe = mt6370_leds_probe, }; module_platform_driver(mt6370_rgbled_driver); MODULE_AUTHOR("Alice Chen "); MODULE_AUTHOR("ChiYuan Huang "); MODULE_DESCRIPTION("MediaTek MT6370 RGB LED Driver"); MODULE_LICENSE("GPL");