leds-lp55xx: use lp55xx common led registration function

 LED class devices are registered in lp5521_register_leds() and
 lp5523_register_leds().
 Two separate functions are merged into consolidated lp55xx function,
 lp55xx_register_leds().

 Error handling fix:
 Unregistering LEDS are handled in lp55xx_register_leds() when LED registration
 failure occurs. So each driver error handler is changed to 'err_register_leds'

 Chip dependency: 'brightness_work_fn' and 'set_led_current'
 To make the structure abstract, both functions are configured in each driver.
 Those functions should be done by each driver because register control is
 chip-dependant work.

 lp55xx_init_led: skeleton
 Will be filled in next patch

Signed-off-by: Milo(Woogyom) Kim <milo.kim@ti.com>
Signed-off-by: Bryan Wu <cooloney@gmail.com>

4 files changed, 72 insertions, 82 deletions

diff --git a/drivers/leds/leds-lp5521.c b/drivers/leds/leds-lp5521.c
index dd4526e168fa..dc58f4106d09 100644
--- a/drivers/leds/leds-lp5521.c
+++ b/drivers/leds/leds-lp5521.c
@@ -738,44 +738,6 @@ static int lp5521_init_led(struct lp5521_led *led,
         return 0;
 }
 
-static int lp5521_register_leds(struct lp5521_chip *chip)
-{
-        struct lp5521_platform_data *pdata = chip->pdata;
-        struct i2c_client *client = chip->client;
-        int i;
-        int led;
-        int ret;
-
-        /* Initialize leds */
-        chip->num_channels = pdata->num_channels;
-        chip->num_leds = 0;
-        led = 0;
-        for (i = 0; i < pdata->num_channels; i++) {
-                /* Do not initialize channels that are not connected */
-                if (pdata->led_config[i].led_current == 0)
-                        continue;
-
-                ret = lp5521_init_led(&chip->leds[led], client, i, pdata);
-                if (ret) {
-                        dev_err(&client->dev, "error initializing leds\n");
-                        return ret;
-                }
-                chip->num_leds++;
-
-                chip->leds[led].id = led;
-                /* Set initial LED current */
-                lp5521_set_led_current(chip, led,
-                                chip->leds[led].led_current);
-
-                INIT_WORK(&(chip->leds[led].brightness_work),
-                        lp5521_led_brightness_work);
-
-                led++;
-        }
-
-        return 0;
-}
-
 static void lp5521_unregister_leds(struct lp5521_chip *chip)
 {
         int i;
@@ -836,9 +798,9 @@ static int lp5521_probe(struct i2c_client *client,
 
         dev_info(&client->dev, "%s programmable led chip found\n", id->name);
 
-        ret = lp5521_register_leds(old_chip);
+        ret = lp55xx_register_leds(led, chip);
         if (ret)
-                goto fail2;
+                goto err_register_leds;
 
         ret = lp5521_register_sysfs(client);
         if (ret) {
@@ -848,6 +810,7 @@ static int lp5521_probe(struct i2c_client *client,
         return ret;
 fail2:
         lp5521_unregister_leds(old_chip);
+err_register_leds:
         lp55xx_deinit_device(chip);
 err_init:
         return ret;
diff --git a/drivers/leds/leds-lp5523.c b/drivers/leds/leds-lp5523.c
index 3f506e3d4986..41ac502ff82f 100644
--- a/drivers/leds/leds-lp5523.c
+++ b/drivers/leds/leds-lp5523.c
@@ -822,46 +822,6 @@ static int lp5523_init_led(struct lp5523_led *led, struct device *dev,
         return 0;
 }
 
-static int lp5523_register_leds(struct lp5523_chip *chip, const char *name)
-{
-        struct lp5523_platform_data *pdata = chip->pdata;
-        struct i2c_client *client = chip->client;
-        int i;
-        int led;
-        int ret;
-
-        /* Initialize leds */
-        chip->num_channels = pdata->num_channels;
-        chip->num_leds = 0;
-        led = 0;
-        for (i = 0; i < pdata->num_channels; i++) {
-                /* Do not initialize channels that are not connected */
-                if (pdata->led_config[i].led_current == 0)
-                        continue;
-
-                INIT_WORK(&chip->leds[led].brightness_work,
-                        lp5523_led_brightness_work);
-
-                ret = lp5523_init_led(&chip->leds[led], &client->dev, i, pdata,
-                                name);
-                if (ret) {
-                        dev_err(&client->dev, "error initializing leds\n");
-                        return ret;
-                }
-                chip->num_leds++;
-
-                chip->leds[led].id = led;
-                /* Set LED current */
-                lp5523_write(client,
-                          LP5523_REG_LED_CURRENT_BASE + chip->leds[led].chan_nr,
-                          chip->leds[led].led_current);
-
-                led++;
-        }
-
-        return 0;
-}
-
 static void lp5523_unregister_leds(struct lp5523_chip *chip)
 {
         int i;
@@ -922,9 +882,9 @@ static int lp5523_probe(struct i2c_client *client,
 
         dev_info(&client->dev, "%s Programmable led chip found\n", id->name);
 
-        ret = lp5523_register_leds(old_chip, id->name);
+        ret = lp55xx_register_leds(led, chip);
         if (ret)
-                goto fail2;
+                goto err_register_leds;
 
         ret = lp5523_register_sysfs(client);
         if (ret) {
@@ -934,6 +894,7 @@ static int lp5523_probe(struct i2c_client *client,
         return ret;
 fail2:
         lp5523_unregister_leds(old_chip);
+err_register_leds:
         lp55xx_deinit_device(chip);
 err_init:
         return ret;
diff --git a/drivers/leds/leds-lp55xx-common.c b/drivers/leds/leds-lp55xx-common.c
index bcabf2cb949a..1fab1d1c4502 100644
--- a/drivers/leds/leds-lp55xx-common.c
+++ b/drivers/leds/leds-lp55xx-common.c
@@ -63,6 +63,12 @@ static int lp55xx_post_init_device(struct lp55xx_chip *chip)
         return cfg->post_init_device(chip);
 }
 
+static int lp55xx_init_led(struct lp55xx_led *led,
+                        struct lp55xx_chip *chip, int chan)
+{
+        return 0;
+}
+
 int lp55xx_write(struct lp55xx_chip *chip, u8 reg, u8 val)
 {
         return i2c_smbus_write_byte_data(chip->cl, reg, val);
@@ -170,6 +176,55 @@ void lp55xx_deinit_device(struct lp55xx_chip *chip)
 }
 EXPORT_SYMBOL_GPL(lp55xx_deinit_device);
 
+int lp55xx_register_leds(struct lp55xx_led *led, struct lp55xx_chip *chip)
+{
+        struct lp55xx_platform_data *pdata = chip->pdata;
+        struct lp55xx_device_config *cfg = chip->cfg;
+        int num_channels = pdata->num_channels;
+        struct lp55xx_led *each;
+        u8 led_current;
+        int ret;
+        int i;
+
+        if (!cfg->brightness_work_fn) {
+                dev_err(&chip->cl->dev, "empty brightness configuration\n");
+                return -EINVAL;
+        }
+
+        for (i = 0; i < num_channels; i++) {
+
+                /* do not initialize channels that are not connected */
+                if (pdata->led_config[i].led_current == 0)
+                        continue;
+
+                led_current = pdata->led_config[i].led_current;
+                each = led + i;
+                ret = lp55xx_init_led(each, chip, i);
+                if (ret)
+                        goto err_init_led;
+
+                INIT_WORK(&each->brightness_work, cfg->brightness_work_fn);
+
+                chip->num_leds++;
+                each->chip = chip;
+
+                /* setting led current at each channel */
+                if (cfg->set_led_current)
+                        cfg->set_led_current(each, led_current);
+        }
+
+        return 0;
+
+err_init_led:
+        for (i = 0; i < chip->num_leds; i++) {
+                each = led + i;
+                led_classdev_unregister(&each->cdev);
+                flush_work(&each->brightness_work);
+        }
+        return ret;
+}
+EXPORT_SYMBOL_GPL(lp55xx_register_leds);
+
 MODULE_AUTHOR("Milo Kim <milo.kim@ti.com>");
 MODULE_DESCRIPTION("LP55xx Common Driver");
 MODULE_LICENSE("GPL");
diff --git a/drivers/leds/leds-lp55xx-common.h b/drivers/leds/leds-lp55xx-common.h
index 908b00a56b7e..219780a2d4eb 100644
--- a/drivers/leds/leds-lp55xx-common.h
+++ b/drivers/leds/leds-lp55xx-common.h
@@ -33,6 +33,8 @@ struct lp55xx_reg {
  * @reset              : Chip specific reset command
  * @enable             : Chip specific enable command
  * @post_init_device   : Chip specific initialization code
+ * @brightness_work_fn : Brightness work function
+ * @set_led_current    : LED current set function
  */
 struct lp55xx_device_config {
         const struct lp55xx_reg reset;
@@ -40,6 +42,12 @@ struct lp55xx_device_config {
 
         /* define if the device has specific initialization process */
         int (*post_init_device) (struct lp55xx_chip *chip);
+
+        /* access brightness register */
+        void (*brightness_work_fn)(struct work_struct *work);
+
+        /* current setting function */
+        void (*set_led_current) (struct lp55xx_led *led, u8 led_current);
 };
 
 /*
@@ -88,4 +96,7 @@ extern int lp55xx_update_bits(struct lp55xx_chip *chip, u8 reg,
 extern int lp55xx_init_device(struct lp55xx_chip *chip);
 extern void lp55xx_deinit_device(struct lp55xx_chip *chip);
 
+/* common LED class device functions */
+extern int lp55xx_register_leds(struct lp55xx_led *led,
+                                struct lp55xx_chip *chip);
 #endif /* _LEDS_LP55XX_COMMON_H */