drivers/leds/leds-lp5521.c: adjust delays and add comments to them
[platform/kernel/linux-starfive.git] / drivers / leds / leds-lp5521.c
1 /*
2  * LP5521 LED chip driver.
3  *
4  * Copyright (C) 2010 Nokia Corporation
5  *
6  * Contact: Samu Onkalo <samu.p.onkalo@nokia.com>
7  *
8  * This program is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU General Public License
10  * version 2 as published by the Free Software Foundation.
11  *
12  * This program is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20  * 02110-1301 USA
21  */
22
23 #include <linux/module.h>
24 #include <linux/init.h>
25 #include <linux/i2c.h>
26 #include <linux/mutex.h>
27 #include <linux/gpio.h>
28 #include <linux/interrupt.h>
29 #include <linux/delay.h>
30 #include <linux/ctype.h>
31 #include <linux/spinlock.h>
32 #include <linux/wait.h>
33 #include <linux/leds.h>
34 #include <linux/leds-lp5521.h>
35 #include <linux/workqueue.h>
36 #include <linux/slab.h>
37
38 #define LP5521_PROGRAM_LENGTH           32      /* in bytes */
39
40 #define LP5521_MAX_LEDS                 3       /* Maximum number of LEDs */
41 #define LP5521_MAX_ENGINES              3       /* Maximum number of engines */
42
43 #define LP5521_ENG_MASK_BASE            0x30    /* 00110000 */
44 #define LP5521_ENG_STATUS_MASK          0x07    /* 00000111 */
45
46 #define LP5521_CMD_LOAD                 0x15    /* 00010101 */
47 #define LP5521_CMD_RUN                  0x2a    /* 00101010 */
48 #define LP5521_CMD_DIRECT               0x3f    /* 00111111 */
49 #define LP5521_CMD_DISABLED             0x00    /* 00000000 */
50
51 /* Registers */
52 #define LP5521_REG_ENABLE               0x00
53 #define LP5521_REG_OP_MODE              0x01
54 #define LP5521_REG_R_PWM                0x02
55 #define LP5521_REG_G_PWM                0x03
56 #define LP5521_REG_B_PWM                0x04
57 #define LP5521_REG_R_CURRENT            0x05
58 #define LP5521_REG_G_CURRENT            0x06
59 #define LP5521_REG_B_CURRENT            0x07
60 #define LP5521_REG_CONFIG               0x08
61 #define LP5521_REG_R_CHANNEL_PC         0x09
62 #define LP5521_REG_G_CHANNEL_PC         0x0A
63 #define LP5521_REG_B_CHANNEL_PC         0x0B
64 #define LP5521_REG_STATUS               0x0C
65 #define LP5521_REG_RESET                0x0D
66 #define LP5521_REG_GPO                  0x0E
67 #define LP5521_REG_R_PROG_MEM           0x10
68 #define LP5521_REG_G_PROG_MEM           0x30
69 #define LP5521_REG_B_PROG_MEM           0x50
70
71 #define LP5521_PROG_MEM_BASE            LP5521_REG_R_PROG_MEM
72 #define LP5521_PROG_MEM_SIZE            0x20
73
74 /* Base register to set LED current */
75 #define LP5521_REG_LED_CURRENT_BASE     LP5521_REG_R_CURRENT
76
77 /* Base register to set the brightness */
78 #define LP5521_REG_LED_PWM_BASE         LP5521_REG_R_PWM
79
80 /* Bits in ENABLE register */
81 #define LP5521_MASTER_ENABLE            0x40    /* Chip master enable */
82 #define LP5521_LOGARITHMIC_PWM          0x80    /* Logarithmic PWM adjustment */
83 #define LP5521_EXEC_RUN                 0x2A
84
85 /* Bits in CONFIG register */
86 #define LP5521_PWM_HF                   0x40    /* PWM: 0 = 256Hz, 1 = 558Hz */
87 #define LP5521_PWRSAVE_EN               0x20    /* 1 = Power save mode */
88 #define LP5521_CP_MODE_OFF              0       /* Charge pump (CP) off */
89 #define LP5521_CP_MODE_BYPASS           8       /* CP forced to bypass mode */
90 #define LP5521_CP_MODE_1X5              0x10    /* CP forced to 1.5x mode */
91 #define LP5521_CP_MODE_AUTO             0x18    /* Automatic mode selection */
92 #define LP5521_R_TO_BATT                4       /* R out: 0 = CP, 1 = Vbat */
93 #define LP5521_CLK_SRC_EXT              0       /* Ext-clk source (CLK_32K) */
94 #define LP5521_CLK_INT                  1       /* Internal clock */
95 #define LP5521_CLK_AUTO                 2       /* Automatic clock selection */
96
97 /* Status */
98 #define LP5521_EXT_CLK_USED             0x08
99
100 struct lp5521_engine {
101         const struct attribute_group *attributes;
102         int             id;
103         u8              mode;
104         u8              prog_page;
105         u8              engine_mask;
106 };
107
108 struct lp5521_led {
109         int                     id;
110         u8                      chan_nr;
111         u8                      led_current;
112         u8                      max_current;
113         struct led_classdev     cdev;
114         struct work_struct      brightness_work;
115         u8                      brightness;
116 };
117
118 struct lp5521_chip {
119         struct lp5521_platform_data *pdata;
120         struct mutex            lock; /* Serialize control */
121         struct i2c_client       *client;
122         struct lp5521_engine    engines[LP5521_MAX_ENGINES];
123         struct lp5521_led       leds[LP5521_MAX_LEDS];
124         u8                      num_channels;
125         u8                      num_leds;
126 };
127
128 static inline struct lp5521_led *cdev_to_led(struct led_classdev *cdev)
129 {
130         return container_of(cdev, struct lp5521_led, cdev);
131 }
132
133 static inline struct lp5521_chip *engine_to_lp5521(struct lp5521_engine *engine)
134 {
135         return container_of(engine, struct lp5521_chip,
136                             engines[engine->id - 1]);
137 }
138
139 static inline struct lp5521_chip *led_to_lp5521(struct lp5521_led *led)
140 {
141         return container_of(led, struct lp5521_chip,
142                             leds[led->id]);
143 }
144
145 static void lp5521_led_brightness_work(struct work_struct *work);
146
147 static inline int lp5521_write(struct i2c_client *client, u8 reg, u8 value)
148 {
149         return i2c_smbus_write_byte_data(client, reg, value);
150 }
151
152 static int lp5521_read(struct i2c_client *client, u8 reg, u8 *buf)
153 {
154         s32 ret;
155
156         ret = i2c_smbus_read_byte_data(client, reg);
157         if (ret < 0)
158                 return -EIO;
159
160         *buf = ret;
161         return 0;
162 }
163
164 static int lp5521_set_engine_mode(struct lp5521_engine *engine, u8 mode)
165 {
166         struct lp5521_chip *chip = engine_to_lp5521(engine);
167         struct i2c_client *client = chip->client;
168         int ret;
169         u8 engine_state;
170
171         /* Only transition between RUN and DIRECT mode are handled here */
172         if (mode == LP5521_CMD_LOAD)
173                 return 0;
174
175         if (mode == LP5521_CMD_DISABLED)
176                 mode = LP5521_CMD_DIRECT;
177
178         ret = lp5521_read(client, LP5521_REG_OP_MODE, &engine_state);
179
180         /* set mode only for this engine */
181         engine_state &= ~(engine->engine_mask);
182         mode &= engine->engine_mask;
183         engine_state |= mode;
184         ret |= lp5521_write(client, LP5521_REG_OP_MODE, engine_state);
185
186         return ret;
187 }
188
189 static int lp5521_load_program(struct lp5521_engine *eng, const u8 *pattern)
190 {
191         struct lp5521_chip *chip = engine_to_lp5521(eng);
192         struct i2c_client *client = chip->client;
193         int ret;
194         int addr;
195         u8 mode;
196
197         /* move current engine to direct mode and remember the state */
198         ret = lp5521_set_engine_mode(eng, LP5521_CMD_DIRECT);
199         /* Mode change requires min 500 us delay. 1 - 2 ms  with margin */
200         usleep_range(1000, 2000);
201         ret |= lp5521_read(client, LP5521_REG_OP_MODE, &mode);
202
203         /* For loading, all the engines to load mode */
204         lp5521_write(client, LP5521_REG_OP_MODE, LP5521_CMD_DIRECT);
205         /* Mode change requires min 500 us delay. 1 - 2 ms  with margin */
206         usleep_range(1000, 2000);
207         lp5521_write(client, LP5521_REG_OP_MODE, LP5521_CMD_LOAD);
208         /* Mode change requires min 500 us delay. 1 - 2 ms  with margin */
209         usleep_range(1000, 2000);
210
211         addr = LP5521_PROG_MEM_BASE + eng->prog_page * LP5521_PROG_MEM_SIZE;
212         i2c_smbus_write_i2c_block_data(client,
213                                 addr,
214                                 LP5521_PROG_MEM_SIZE,
215                                 pattern);
216
217         ret |= lp5521_write(client, LP5521_REG_OP_MODE, mode);
218         return ret;
219 }
220
221 static int lp5521_set_led_current(struct lp5521_chip *chip, int led, u8 curr)
222 {
223         return lp5521_write(chip->client,
224                     LP5521_REG_LED_CURRENT_BASE + chip->leds[led].chan_nr,
225                     curr);
226 }
227
228 static void lp5521_init_engine(struct lp5521_chip *chip,
229                         const struct attribute_group *attr_group)
230 {
231         int i;
232         for (i = 0; i < ARRAY_SIZE(chip->engines); i++) {
233                 chip->engines[i].id = i + 1;
234                 chip->engines[i].engine_mask = LP5521_ENG_MASK_BASE >> (i * 2);
235                 chip->engines[i].prog_page = i;
236                 chip->engines[i].attributes = &attr_group[i];
237         }
238 }
239
240 static int lp5521_configure(struct i2c_client *client,
241                         const struct attribute_group *attr_group)
242 {
243         struct lp5521_chip *chip = i2c_get_clientdata(client);
244         int ret;
245
246         lp5521_init_engine(chip, attr_group);
247
248         lp5521_write(client, LP5521_REG_RESET, 0xff);
249
250         usleep_range(10000, 20000); /*
251                                      * Exact value is not available. 10 - 20ms
252                                      * appears to be enough for reset.
253                                      */
254
255         /* Set all PWMs to direct control mode */
256         ret = lp5521_write(client, LP5521_REG_OP_MODE, 0x3F);
257
258         /* Enable auto-powersave, set charge pump to auto, red to battery */
259         ret |= lp5521_write(client, LP5521_REG_CONFIG,
260                 LP5521_PWRSAVE_EN | LP5521_CP_MODE_AUTO | LP5521_R_TO_BATT);
261
262         /* Initialize all channels PWM to zero -> leds off */
263         ret |= lp5521_write(client, LP5521_REG_R_PWM, 0);
264         ret |= lp5521_write(client, LP5521_REG_G_PWM, 0);
265         ret |= lp5521_write(client, LP5521_REG_B_PWM, 0);
266
267         /* Set engines are set to run state when OP_MODE enables engines */
268         ret |= lp5521_write(client, LP5521_REG_ENABLE,
269                         LP5521_MASTER_ENABLE | LP5521_LOGARITHMIC_PWM |
270                         LP5521_EXEC_RUN);
271         /* enable takes 500us. 1 - 2 ms leaves some margin */
272         usleep_range(1000, 2000);
273
274         return ret;
275 }
276
277 static int lp5521_run_selftest(struct lp5521_chip *chip, char *buf)
278 {
279         int ret;
280         u8 status;
281
282         ret = lp5521_read(chip->client, LP5521_REG_STATUS, &status);
283         if (ret < 0)
284                 return ret;
285
286         /* Check that ext clock is really in use if requested */
287         if (chip->pdata && chip->pdata->clock_mode == LP5521_CLOCK_EXT)
288                 if  ((status & LP5521_EXT_CLK_USED) == 0)
289                         return -EIO;
290         return 0;
291 }
292
293 static void lp5521_set_brightness(struct led_classdev *cdev,
294                              enum led_brightness brightness)
295 {
296         struct lp5521_led *led = cdev_to_led(cdev);
297         led->brightness = (u8)brightness;
298         schedule_work(&led->brightness_work);
299 }
300
301 static void lp5521_led_brightness_work(struct work_struct *work)
302 {
303         struct lp5521_led *led = container_of(work,
304                                               struct lp5521_led,
305                                               brightness_work);
306         struct lp5521_chip *chip = led_to_lp5521(led);
307         struct i2c_client *client = chip->client;
308
309         mutex_lock(&chip->lock);
310         lp5521_write(client, LP5521_REG_LED_PWM_BASE + led->chan_nr,
311                 led->brightness);
312         mutex_unlock(&chip->lock);
313 }
314
315 /* Detect the chip by setting its ENABLE register and reading it back. */
316 static int lp5521_detect(struct i2c_client *client)
317 {
318         int ret;
319         u8 buf;
320
321         ret = lp5521_write(client, LP5521_REG_ENABLE,
322                         LP5521_MASTER_ENABLE | LP5521_LOGARITHMIC_PWM);
323         if (ret)
324                 return ret;
325         /* enable takes 500us. 1 - 2 ms leaves some margin */
326         usleep_range(1000, 2000);
327         ret = lp5521_read(client, LP5521_REG_ENABLE, &buf);
328         if (ret)
329                 return ret;
330         if (buf != (LP5521_MASTER_ENABLE | LP5521_LOGARITHMIC_PWM))
331                 return -ENODEV;
332
333         return 0;
334 }
335
336 /* Set engine mode and create appropriate sysfs attributes, if required. */
337 static int lp5521_set_mode(struct lp5521_engine *engine, u8 mode)
338 {
339         struct lp5521_chip *chip = engine_to_lp5521(engine);
340         struct i2c_client *client = chip->client;
341         struct device *dev = &client->dev;
342         int ret = 0;
343
344         /* if in that mode already do nothing, except for run */
345         if (mode == engine->mode && mode != LP5521_CMD_RUN)
346                 return 0;
347
348         if (mode == LP5521_CMD_RUN) {
349                 ret = lp5521_set_engine_mode(engine, LP5521_CMD_RUN);
350         } else if (mode == LP5521_CMD_LOAD) {
351                 lp5521_set_engine_mode(engine, LP5521_CMD_DISABLED);
352                 lp5521_set_engine_mode(engine, LP5521_CMD_LOAD);
353
354                 ret = sysfs_create_group(&dev->kobj, engine->attributes);
355                 if (ret)
356                         return ret;
357         } else if (mode == LP5521_CMD_DISABLED) {
358                 lp5521_set_engine_mode(engine, LP5521_CMD_DISABLED);
359         }
360
361         /* remove load attribute from sysfs if not in load mode */
362         if (engine->mode == LP5521_CMD_LOAD && mode != LP5521_CMD_LOAD)
363                 sysfs_remove_group(&dev->kobj, engine->attributes);
364
365         engine->mode = mode;
366
367         return ret;
368 }
369
370 static int lp5521_do_store_load(struct lp5521_engine *engine,
371                                 const char *buf, size_t len)
372 {
373         struct lp5521_chip *chip = engine_to_lp5521(engine);
374         struct i2c_client *client = chip->client;
375         int  ret, nrchars, offset = 0, i = 0;
376         char c[3];
377         unsigned cmd;
378         u8 pattern[LP5521_PROGRAM_LENGTH] = {0};
379
380         while ((offset < len - 1) && (i < LP5521_PROGRAM_LENGTH)) {
381                 /* separate sscanfs because length is working only for %s */
382                 ret = sscanf(buf + offset, "%2s%n ", c, &nrchars);
383                 ret = sscanf(c, "%2x", &cmd);
384                 if (ret != 1)
385                         goto fail;
386                 pattern[i] = (u8)cmd;
387
388                 offset += nrchars;
389                 i++;
390         }
391
392         /* Each instruction is 16bit long. Check that length is even */
393         if (i % 2)
394                 goto fail;
395
396         mutex_lock(&chip->lock);
397         ret = lp5521_load_program(engine, pattern);
398         mutex_unlock(&chip->lock);
399
400         if (ret) {
401                 dev_err(&client->dev, "failed loading pattern\n");
402                 return ret;
403         }
404
405         return len;
406 fail:
407         dev_err(&client->dev, "wrong pattern format\n");
408         return -EINVAL;
409 }
410
411 static ssize_t store_engine_load(struct device *dev,
412                                      struct device_attribute *attr,
413                                      const char *buf, size_t len, int nr)
414 {
415         struct i2c_client *client = to_i2c_client(dev);
416         struct lp5521_chip *chip = i2c_get_clientdata(client);
417         return lp5521_do_store_load(&chip->engines[nr - 1], buf, len);
418 }
419
420 #define store_load(nr)                                                  \
421 static ssize_t store_engine##nr##_load(struct device *dev,              \
422                                      struct device_attribute *attr,     \
423                                      const char *buf, size_t len)       \
424 {                                                                       \
425         return store_engine_load(dev, attr, buf, len, nr);              \
426 }
427 store_load(1)
428 store_load(2)
429 store_load(3)
430
431 static ssize_t show_engine_mode(struct device *dev,
432                                 struct device_attribute *attr,
433                                 char *buf, int nr)
434 {
435         struct i2c_client *client = to_i2c_client(dev);
436         struct lp5521_chip *chip = i2c_get_clientdata(client);
437         switch (chip->engines[nr - 1].mode) {
438         case LP5521_CMD_RUN:
439                 return sprintf(buf, "run\n");
440         case LP5521_CMD_LOAD:
441                 return sprintf(buf, "load\n");
442         case LP5521_CMD_DISABLED:
443                 return sprintf(buf, "disabled\n");
444         default:
445                 return sprintf(buf, "disabled\n");
446         }
447 }
448
449 #define show_mode(nr)                                                   \
450 static ssize_t show_engine##nr##_mode(struct device *dev,               \
451                                     struct device_attribute *attr,      \
452                                     char *buf)                          \
453 {                                                                       \
454         return show_engine_mode(dev, attr, buf, nr);                    \
455 }
456 show_mode(1)
457 show_mode(2)
458 show_mode(3)
459
460 static ssize_t store_engine_mode(struct device *dev,
461                                  struct device_attribute *attr,
462                                  const char *buf, size_t len, int nr)
463 {
464         struct i2c_client *client = to_i2c_client(dev);
465         struct lp5521_chip *chip = i2c_get_clientdata(client);
466         struct lp5521_engine *engine = &chip->engines[nr - 1];
467         mutex_lock(&chip->lock);
468
469         if (!strncmp(buf, "run", 3))
470                 lp5521_set_mode(engine, LP5521_CMD_RUN);
471         else if (!strncmp(buf, "load", 4))
472                 lp5521_set_mode(engine, LP5521_CMD_LOAD);
473         else if (!strncmp(buf, "disabled", 8))
474                 lp5521_set_mode(engine, LP5521_CMD_DISABLED);
475
476         mutex_unlock(&chip->lock);
477         return len;
478 }
479
480 #define store_mode(nr)                                                  \
481 static ssize_t store_engine##nr##_mode(struct device *dev,              \
482                                      struct device_attribute *attr,     \
483                                      const char *buf, size_t len)       \
484 {                                                                       \
485         return store_engine_mode(dev, attr, buf, len, nr);              \
486 }
487 store_mode(1)
488 store_mode(2)
489 store_mode(3)
490
491 static ssize_t show_max_current(struct device *dev,
492                             struct device_attribute *attr,
493                             char *buf)
494 {
495         struct led_classdev *led_cdev = dev_get_drvdata(dev);
496         struct lp5521_led *led = cdev_to_led(led_cdev);
497
498         return sprintf(buf, "%d\n", led->max_current);
499 }
500
501 static ssize_t show_current(struct device *dev,
502                             struct device_attribute *attr,
503                             char *buf)
504 {
505         struct led_classdev *led_cdev = dev_get_drvdata(dev);
506         struct lp5521_led *led = cdev_to_led(led_cdev);
507
508         return sprintf(buf, "%d\n", led->led_current);
509 }
510
511 static ssize_t store_current(struct device *dev,
512                              struct device_attribute *attr,
513                              const char *buf, size_t len)
514 {
515         struct led_classdev *led_cdev = dev_get_drvdata(dev);
516         struct lp5521_led *led = cdev_to_led(led_cdev);
517         struct lp5521_chip *chip = led_to_lp5521(led);
518         ssize_t ret;
519         unsigned long curr;
520
521         if (strict_strtoul(buf, 0, &curr))
522                 return -EINVAL;
523
524         if (curr > led->max_current)
525                 return -EINVAL;
526
527         mutex_lock(&chip->lock);
528         ret = lp5521_set_led_current(chip, led->id, curr);
529         mutex_unlock(&chip->lock);
530
531         if (ret < 0)
532                 return ret;
533
534         led->led_current = (u8)curr;
535
536         return len;
537 }
538
539 static ssize_t lp5521_selftest(struct device *dev,
540                                struct device_attribute *attr,
541                                char *buf)
542 {
543         struct i2c_client *client = to_i2c_client(dev);
544         struct lp5521_chip *chip = i2c_get_clientdata(client);
545         int ret;
546
547         mutex_lock(&chip->lock);
548         ret = lp5521_run_selftest(chip, buf);
549         mutex_unlock(&chip->lock);
550         return sprintf(buf, "%s\n", ret ? "FAIL" : "OK");
551 }
552
553 /* led class device attributes */
554 static DEVICE_ATTR(led_current, S_IRUGO | S_IWUGO, show_current, store_current);
555 static DEVICE_ATTR(max_current, S_IRUGO , show_max_current, NULL);
556
557 static struct attribute *lp5521_led_attributes[] = {
558         &dev_attr_led_current.attr,
559         &dev_attr_max_current.attr,
560         NULL,
561 };
562
563 static struct attribute_group lp5521_led_attribute_group = {
564         .attrs = lp5521_led_attributes
565 };
566
567 /* device attributes */
568 static DEVICE_ATTR(engine1_mode, S_IRUGO | S_IWUGO,
569                    show_engine1_mode, store_engine1_mode);
570 static DEVICE_ATTR(engine2_mode, S_IRUGO | S_IWUGO,
571                    show_engine2_mode, store_engine2_mode);
572 static DEVICE_ATTR(engine3_mode, S_IRUGO | S_IWUGO,
573                    show_engine3_mode, store_engine3_mode);
574 static DEVICE_ATTR(engine1_load, S_IWUGO, NULL, store_engine1_load);
575 static DEVICE_ATTR(engine2_load, S_IWUGO, NULL, store_engine2_load);
576 static DEVICE_ATTR(engine3_load, S_IWUGO, NULL, store_engine3_load);
577 static DEVICE_ATTR(selftest, S_IRUGO, lp5521_selftest, NULL);
578
579 static struct attribute *lp5521_attributes[] = {
580         &dev_attr_engine1_mode.attr,
581         &dev_attr_engine2_mode.attr,
582         &dev_attr_engine3_mode.attr,
583         &dev_attr_selftest.attr,
584         NULL
585 };
586
587 static struct attribute *lp5521_engine1_attributes[] = {
588         &dev_attr_engine1_load.attr,
589         NULL
590 };
591
592 static struct attribute *lp5521_engine2_attributes[] = {
593         &dev_attr_engine2_load.attr,
594         NULL
595 };
596
597 static struct attribute *lp5521_engine3_attributes[] = {
598         &dev_attr_engine3_load.attr,
599         NULL
600 };
601
602 static const struct attribute_group lp5521_group = {
603         .attrs = lp5521_attributes,
604 };
605
606 static const struct attribute_group lp5521_engine_group[] = {
607         {.attrs = lp5521_engine1_attributes },
608         {.attrs = lp5521_engine2_attributes },
609         {.attrs = lp5521_engine3_attributes },
610 };
611
612 static int lp5521_register_sysfs(struct i2c_client *client)
613 {
614         struct device *dev = &client->dev;
615         return sysfs_create_group(&dev->kobj, &lp5521_group);
616 }
617
618 static void lp5521_unregister_sysfs(struct i2c_client *client)
619 {
620         struct lp5521_chip *chip = i2c_get_clientdata(client);
621         struct device *dev = &client->dev;
622         int i;
623
624         sysfs_remove_group(&dev->kobj, &lp5521_group);
625
626         for (i = 0; i <  ARRAY_SIZE(chip->engines); i++) {
627                 if (chip->engines[i].mode == LP5521_CMD_LOAD)
628                         sysfs_remove_group(&dev->kobj,
629                                         chip->engines[i].attributes);
630         }
631
632         for (i = 0; i < chip->num_leds; i++)
633                 sysfs_remove_group(&chip->leds[i].cdev.dev->kobj,
634                                 &lp5521_led_attribute_group);
635 }
636
637 static int __init lp5521_init_led(struct lp5521_led *led,
638                                 struct i2c_client *client,
639                                 int chan, struct lp5521_platform_data *pdata)
640 {
641         struct device *dev = &client->dev;
642         char name[32];
643         int res;
644
645         if (chan >= LP5521_MAX_LEDS)
646                 return -EINVAL;
647
648         if (pdata->led_config[chan].led_current == 0)
649                 return 0;
650
651         led->led_current = pdata->led_config[chan].led_current;
652         led->max_current = pdata->led_config[chan].max_current;
653         led->chan_nr = pdata->led_config[chan].chan_nr;
654
655         if (led->chan_nr >= LP5521_MAX_LEDS) {
656                 dev_err(dev, "Use channel numbers between 0 and %d\n",
657                         LP5521_MAX_LEDS - 1);
658                 return -EINVAL;
659         }
660
661         snprintf(name, sizeof(name), "%s:channel%d", client->name, chan);
662         led->cdev.brightness_set = lp5521_set_brightness;
663         led->cdev.name = name;
664         res = led_classdev_register(dev, &led->cdev);
665         if (res < 0) {
666                 dev_err(dev, "couldn't register led on channel %d\n", chan);
667                 return res;
668         }
669
670         res = sysfs_create_group(&led->cdev.dev->kobj,
671                         &lp5521_led_attribute_group);
672         if (res < 0) {
673                 dev_err(dev, "couldn't register current attribute\n");
674                 led_classdev_unregister(&led->cdev);
675                 return res;
676         }
677         return 0;
678 }
679
680 static int lp5521_probe(struct i2c_client *client,
681                         const struct i2c_device_id *id)
682 {
683         struct lp5521_chip              *chip;
684         struct lp5521_platform_data     *pdata;
685         int ret, i, led;
686
687         chip = kzalloc(sizeof(*chip), GFP_KERNEL);
688         if (!chip)
689                 return -ENOMEM;
690
691         i2c_set_clientdata(client, chip);
692         chip->client = client;
693
694         pdata = client->dev.platform_data;
695
696         if (!pdata) {
697                 dev_err(&client->dev, "no platform data\n");
698                 ret = -EINVAL;
699                 goto fail1;
700         }
701
702         mutex_init(&chip->lock);
703
704         chip->pdata   = pdata;
705
706         if (pdata->setup_resources) {
707                 ret = pdata->setup_resources();
708                 if (ret < 0)
709                         goto fail1;
710         }
711
712         if (pdata->enable) {
713                 pdata->enable(0);
714                 usleep_range(1000, 2000); /* Keep enable down at least 1ms */
715                 pdata->enable(1);
716                 usleep_range(1000, 2000); /* 500us abs min. */
717         }
718
719         ret = lp5521_detect(client);
720
721         if (ret) {
722                 dev_err(&client->dev, "Chip not found\n");
723                 goto fail2;
724         }
725
726         dev_info(&client->dev, "%s programmable led chip found\n", id->name);
727
728         ret = lp5521_configure(client, lp5521_engine_group);
729         if (ret < 0) {
730                 dev_err(&client->dev, "error configuring chip\n");
731                 goto fail2;
732         }
733
734         /* Initialize leds */
735         chip->num_channels = pdata->num_channels;
736         chip->num_leds = 0;
737         led = 0;
738         for (i = 0; i < pdata->num_channels; i++) {
739                 /* Do not initialize channels that are not connected */
740                 if (pdata->led_config[i].led_current == 0)
741                         continue;
742
743                 ret = lp5521_init_led(&chip->leds[led], client, i, pdata);
744                 if (ret) {
745                         dev_err(&client->dev, "error initializing leds\n");
746                         goto fail3;
747                 }
748                 chip->num_leds++;
749
750                 chip->leds[led].id = led;
751                 /* Set initial LED current */
752                 lp5521_set_led_current(chip, led,
753                                 chip->leds[led].led_current);
754
755                 INIT_WORK(&(chip->leds[led].brightness_work),
756                         lp5521_led_brightness_work);
757
758                 led++;
759         }
760
761         ret = lp5521_register_sysfs(client);
762         if (ret) {
763                 dev_err(&client->dev, "registering sysfs failed\n");
764                 goto fail3;
765         }
766         return ret;
767 fail3:
768         for (i = 0; i < chip->num_leds; i++) {
769                 led_classdev_unregister(&chip->leds[i].cdev);
770                 cancel_work_sync(&chip->leds[i].brightness_work);
771         }
772 fail2:
773         if (pdata->enable)
774                 pdata->enable(0);
775         if (pdata->release_resources)
776                 pdata->release_resources();
777 fail1:
778         kfree(chip);
779         return ret;
780 }
781
782 static int lp5521_remove(struct i2c_client *client)
783 {
784         struct lp5521_chip *chip = i2c_get_clientdata(client);
785         int i;
786
787         lp5521_unregister_sysfs(client);
788
789         for (i = 0; i < chip->num_leds; i++) {
790                 led_classdev_unregister(&chip->leds[i].cdev);
791                 cancel_work_sync(&chip->leds[i].brightness_work);
792         }
793
794         if (chip->pdata->enable)
795                 chip->pdata->enable(0);
796         if (chip->pdata->release_resources)
797                 chip->pdata->release_resources();
798         kfree(chip);
799         return 0;
800 }
801
802 static const struct i2c_device_id lp5521_id[] = {
803         { "lp5521", 0 }, /* Three channel chip */
804         { }
805 };
806 MODULE_DEVICE_TABLE(i2c, lp5521_id);
807
808 static struct i2c_driver lp5521_driver = {
809         .driver = {
810                 .name   = "lp5521",
811         },
812         .probe          = lp5521_probe,
813         .remove         = lp5521_remove,
814         .id_table       = lp5521_id,
815 };
816
817 static int __init lp5521_init(void)
818 {
819         int ret;
820
821         ret = i2c_add_driver(&lp5521_driver);
822
823         if (ret < 0)
824                 printk(KERN_ALERT "Adding lp5521 driver failed\n");
825
826         return ret;
827 }
828
829 static void __exit lp5521_exit(void)
830 {
831         i2c_del_driver(&lp5521_driver);
832 }
833
834 module_init(lp5521_init);
835 module_exit(lp5521_exit);
836
837 MODULE_AUTHOR("Mathias Nyman, Yuri Zaporozhets, Samu Onkalo");
838 MODULE_DESCRIPTION("LP5521 LED engine");
839 MODULE_LICENSE("GPL v2");