LEDS: leds-pwm: Add init, notfiy and exit callbacks
[kernel.git] / drivers / leds / leds-pwm.c
1 /*
2  * linux/drivers/leds-pwm.c
3  *
4  * simple PWM based LED control
5  *
6  * Copyright 2009 Luotao Fu @ Pengutronix (l.fu@pengutronix.de)
7  *
8  * based on leds-gpio.c by Raphael Assenat <raph@8d.com>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License version 2 as
12  * published by the Free Software Foundation.
13  */
14
15 #include <linux/module.h>
16 #include <linux/kernel.h>
17 #include <linux/init.h>
18 #include <linux/platform_device.h>
19 #include <linux/fb.h>
20 #include <linux/leds.h>
21 #include <linux/err.h>
22 #include <linux/pwm.h>
23 #include <linux/leds_pwm.h>
24 #include <linux/slab.h>
25
26 struct led_pwm_data {
27         struct led_classdev     cdev;
28         struct pwm_device       *pwm;
29         struct led_pwm          *led;
30         struct device           *parent;
31 };
32
33 static void led_pwm_set(struct led_classdev *led_cdev,
34         enum led_brightness brightness)
35 {
36         struct led_pwm_data *led_dat =
37                 container_of(led_cdev, struct led_pwm_data, cdev);
38         struct device *parent = led_dat->parent;
39         struct led_pwm_platform_data *pdata = parent->platform_data;
40         unsigned int max = led_dat->cdev.max_brightness;
41         unsigned int period =  led_dat->led->pwm_period_ns;
42
43         if (pdata->notify)
44             brightness = pdata->notify(parent, led_dat->led, brightness);
45
46         if (brightness == 0) {
47                 pwm_config(led_dat->pwm, 0, period);
48                 pwm_disable(led_dat->pwm);
49         } else {
50                 pwm_config(led_dat->pwm, brightness * period / max, period);
51                 pwm_enable(led_dat->pwm);
52         }
53 }
54
55 static int led_pwm_probe(struct platform_device *pdev)
56 {
57         struct led_pwm_platform_data *pdata = pdev->dev.platform_data;
58         struct led_pwm *cur_led;
59         struct led_pwm_data *leds_data, *led_dat;
60         int i, ret = 0;
61
62         if (!pdata)
63                 return -EBUSY;
64
65         leds_data = kzalloc(sizeof(struct led_pwm_data) * pdata->num_leds,
66                                 GFP_KERNEL);
67         if (!leds_data)
68                 return -ENOMEM;
69
70         for (i = 0; i < pdata->num_leds; i++) {
71                 cur_led = &pdata->leds[i];
72                 led_dat = &leds_data[i];
73
74                 led_dat->pwm = pwm_request(cur_led->pwm_id,
75                                 cur_led->name);
76                 if (IS_ERR(led_dat->pwm)) {
77                         dev_err(&pdev->dev, "unable to request PWM %d\n",
78                                         cur_led->pwm_id);
79                         goto err;
80                 }
81
82                 led_dat->cdev.name = cur_led->name;
83                 led_dat->cdev.default_trigger = cur_led->default_trigger;
84                 led_dat->cdev.brightness_set = led_pwm_set;
85                 led_dat->cdev.brightness = LED_OFF;
86                 led_dat->cdev.max_brightness = cur_led->max_brightness;
87                 led_dat->cdev.flags |= LED_CORE_SUSPENDRESUME;
88
89                 led_dat->led = cur_led;
90                 led_dat->parent = &pdev->dev;
91
92                 ret = led_classdev_register(&pdev->dev, &led_dat->cdev);
93                 if (ret < 0) {
94                         pwm_free(led_dat->pwm);
95                         goto err;
96                 }
97
98                 if (pdata->init) {
99                         ret = pdata->init(&pdev->dev, cur_led);
100                         if (ret < 0) {
101                                 led_classdev_unregister(&led_dat->cdev);
102                                 pwm_free(led_dat->pwm);
103                                 goto err;
104                         }
105                 }
106         }
107
108         platform_set_drvdata(pdev, leds_data);
109
110         return 0;
111
112 err:
113         if (i > 0) {
114                 for (i = i - 1; i >= 0; i--) {
115                         if (pdata->exit)
116                                 pdata->exit(&pdev->dev, &pdata->leds[i]);
117                         led_classdev_unregister(&leds_data[i].cdev);
118                         pwm_free(leds_data[i].pwm);
119                 }
120         }
121
122         kfree(leds_data);
123
124         return ret;
125 }
126
127 static int __devexit led_pwm_remove(struct platform_device *pdev)
128 {
129         int i;
130         struct led_pwm_platform_data *pdata = pdev->dev.platform_data;
131         struct led_pwm_data *leds_data;
132
133         leds_data = platform_get_drvdata(pdev);
134
135         for (i = 0; i < pdata->num_leds; i++) {
136                 if (pdata->exit)
137                         pdata->exit(&pdev->dev, &pdata->leds[i]);
138                 led_classdev_unregister(&leds_data[i].cdev);
139                 pwm_free(leds_data[i].pwm);
140         }
141
142         kfree(leds_data);
143
144         return 0;
145 }
146
147 static struct platform_driver led_pwm_driver = {
148         .probe          = led_pwm_probe,
149         .remove         = __devexit_p(led_pwm_remove),
150         .driver         = {
151                 .name   = "leds_pwm",
152                 .owner  = THIS_MODULE,
153         },
154 };
155
156 static int __init led_pwm_init(void)
157 {
158         return platform_driver_register(&led_pwm_driver);
159 }
160
161 static void __exit led_pwm_exit(void)
162 {
163         platform_driver_unregister(&led_pwm_driver);
164 }
165
166 module_init(led_pwm_init);
167 module_exit(led_pwm_exit);
168
169 MODULE_AUTHOR("Luotao Fu <l.fu@pengutronix.de>");
170 MODULE_DESCRIPTION("PWM LED driver for PXA");
171 MODULE_LICENSE("GPL");
172 MODULE_ALIAS("platform:leds-pwm");