1 // SPDX-License-Identifier: GPL-2.0-only
3 * PWM-based multi-color LED control
5 * Copyright 2022 Sven Schwermer <sven.schwermer@disruptive-technologies.com>
9 #include <linux/kernel.h>
10 #include <linux/led-class-multicolor.h>
11 #include <linux/leds.h>
12 #include <linux/mod_devicetable.h>
13 #include <linux/module.h>
14 #include <linux/mutex.h>
15 #include <linux/platform_device.h>
16 #include <linux/property.h>
17 #include <linux/pwm.h>
20 struct pwm_device *pwm;
21 struct pwm_state state;
25 struct led_classdev_mc mc_cdev;
27 struct pwm_led leds[];
30 static int led_pwm_mc_set(struct led_classdev *cdev,
31 enum led_brightness brightness)
33 struct led_classdev_mc *mc_cdev = lcdev_to_mccdev(cdev);
34 struct pwm_mc_led *priv = container_of(mc_cdev, struct pwm_mc_led, mc_cdev);
35 unsigned long long duty;
39 led_mc_calc_color_components(mc_cdev, brightness);
41 mutex_lock(&priv->lock);
43 for (i = 0; i < mc_cdev->num_colors; i++) {
44 duty = priv->leds[i].state.period;
45 duty *= mc_cdev->subled_info[i].brightness;
46 do_div(duty, cdev->max_brightness);
48 priv->leds[i].state.duty_cycle = duty;
49 priv->leds[i].state.enabled = duty > 0;
50 ret = pwm_apply_state(priv->leds[i].pwm,
51 &priv->leds[i].state);
56 mutex_unlock(&priv->lock);
61 static int iterate_subleds(struct device *dev, struct pwm_mc_led *priv,
62 struct fwnode_handle *mcnode)
64 struct mc_subled *subled = priv->mc_cdev.subled_info;
65 struct fwnode_handle *fwnode;
66 struct pwm_led *pwmled;
70 /* iterate over the nodes inside the multi-led node */
71 fwnode_for_each_child_node(mcnode, fwnode) {
72 pwmled = &priv->leds[priv->mc_cdev.num_colors];
73 pwmled->pwm = devm_fwnode_pwm_get(dev, fwnode, NULL);
74 if (IS_ERR(pwmled->pwm)) {
75 ret = PTR_ERR(pwmled->pwm);
76 dev_err(dev, "unable to request PWM: %d\n", ret);
79 pwm_init_state(pwmled->pwm, &pwmled->state);
81 ret = fwnode_property_read_u32(fwnode, "color", &color);
83 dev_err(dev, "cannot read color: %d\n", ret);
87 subled[priv->mc_cdev.num_colors].color_index = color;
88 priv->mc_cdev.num_colors++;
94 fwnode_handle_put(fwnode);
98 static int led_pwm_mc_probe(struct platform_device *pdev)
100 struct fwnode_handle *mcnode, *fwnode;
101 struct led_init_data init_data = {};
102 struct led_classdev *cdev;
103 struct mc_subled *subled;
104 struct pwm_mc_led *priv;
108 mcnode = device_get_named_child_node(&pdev->dev, "multi-led");
110 return dev_err_probe(&pdev->dev, -ENODEV,
111 "expected multi-led node\n");
113 /* count the nodes inside the multi-led node */
114 fwnode_for_each_child_node(mcnode, fwnode)
117 priv = devm_kzalloc(&pdev->dev, struct_size(priv, leds, count),
123 mutex_init(&priv->lock);
125 subled = devm_kcalloc(&pdev->dev, count, sizeof(*subled), GFP_KERNEL);
130 priv->mc_cdev.subled_info = subled;
132 /* init the multicolor's LED class device */
133 cdev = &priv->mc_cdev.led_cdev;
134 fwnode_property_read_u32(mcnode, "max-brightness",
135 &cdev->max_brightness);
136 cdev->flags = LED_CORE_SUSPENDRESUME;
137 cdev->brightness_set_blocking = led_pwm_mc_set;
139 ret = iterate_subleds(&pdev->dev, priv, mcnode);
143 init_data.fwnode = mcnode;
144 ret = devm_led_classdev_multicolor_register_ext(&pdev->dev,
149 "failed to register multicolor PWM led for %s: %d\n",
154 ret = led_pwm_mc_set(cdev, cdev->brightness);
156 return dev_err_probe(&pdev->dev, ret,
157 "failed to set led PWM value for %s: %d",
160 platform_set_drvdata(pdev, priv);
164 fwnode_handle_put(mcnode);
168 static const struct of_device_id of_pwm_leds_mc_match[] = {
169 { .compatible = "pwm-leds-multicolor", },
172 MODULE_DEVICE_TABLE(of, of_pwm_leds_mc_match);
174 static struct platform_driver led_pwm_mc_driver = {
175 .probe = led_pwm_mc_probe,
177 .name = "leds_pwm_multicolor",
178 .of_match_table = of_pwm_leds_mc_match,
181 module_platform_driver(led_pwm_mc_driver);
183 MODULE_AUTHOR("Sven Schwermer <sven.schwermer@disruptive-technologies.com>");
184 MODULE_DESCRIPTION("multi-color PWM LED driver");
185 MODULE_LICENSE("GPL v2");
186 MODULE_ALIAS("platform:leds-pwm-multicolor");