[PATCH 2/4] iio: light: stk3310: Implement vdd supply and power it off during suspend

[Aren Moynihan]

From: Ondrej Jirman <m...@xff.cz>

VDD power input can be used to completely power off the chip during
system suspend. Do so if available.

Signed-off-by: Ondrej Jirman <m...@xff.cz>
Signed-off-by: Aren Moynihan <a...@peacevolution.org>
---
 drivers/iio/light/stk3310.c | 56 +++++++++++++++++++++++++++++++++++--
 1 file changed, 53 insertions(+), 3 deletions(-)

diff --git a/drivers/iio/light/stk3310.c b/drivers/iio/light/stk3310.c
index 7b71ad71d78d..bfa090538df7 100644
--- a/drivers/iio/light/stk3310.c
+++ b/drivers/iio/light/stk3310.c
@@ -16,6 +16,7 @@
 #include <linux/iio/events.h>
 #include <linux/iio/iio.h>
 #include <linux/iio/sysfs.h>
+#include <linux/regulator/consumer.h>
 
 #define STK3310_REG_STATE                      0x00
 #define STK3310_REG_PSCTRL                     0x01
@@ -117,6 +118,7 @@ struct stk3310_data {
        struct regmap_field *reg_int_ps;
        struct regmap_field *reg_flag_psint;
        struct regmap_field *reg_flag_nf;
+       struct regulator *vdd_reg;
 };
 
 static const struct iio_event_spec stk3310_events[] = {
@@ -607,6 +609,16 @@ static int stk3310_probe(struct i2c_client *client)
 
        mutex_init(&data->lock);
 
+       data->vdd_reg = devm_regulator_get_optional(&client->dev, "vdd");
+       if (IS_ERR(data->vdd_reg)) {
+               ret = PTR_ERR(data->vdd_reg);
+               if (ret == -ENODEV)
+                       data->vdd_reg = NULL;
+               else
+                       return dev_err_probe(&client->dev, ret,
+                                            "get regulator vdd failed\n");
+       }
+
        ret = stk3310_regmap_init(data);
        if (ret < 0)
                return ret;
@@ -617,9 +629,18 @@ static int stk3310_probe(struct i2c_client *client)
        indio_dev->channels = stk3310_channels;
        indio_dev->num_channels = ARRAY_SIZE(stk3310_channels);
 
+       if (data->vdd_reg) {
+               ret = regulator_enable(data->vdd_reg);
+               if (ret)
+                       return dev_err_probe(&client->dev, ret,
+                                            "regulator vdd enable failed\n");
+
+               usleep_range(1000, 2000);
+       }
+
        ret = stk3310_init(indio_dev);
        if (ret < 0)
-               return ret;
+               goto err_vdd_disable;
 
        if (client->irq > 0) {
                ret = devm_request_threaded_irq(&client->dev, client->irq,
@@ -645,32 +666,61 @@ static int stk3310_probe(struct i2c_client *client)
 
 err_standby:
        stk3310_set_state(data, STK3310_STATE_STANDBY);
+err_vdd_disable:
+       if (data->vdd_reg)
+               regulator_disable(data->vdd_reg);
        return ret;
 }
 
 static void stk3310_remove(struct i2c_client *client)
 {
        struct iio_dev *indio_dev = i2c_get_clientdata(client);
+       struct stk3310_data *data = iio_priv(indio_dev);
 
        iio_device_unregister(indio_dev);
        stk3310_set_state(iio_priv(indio_dev), STK3310_STATE_STANDBY);
+       if (data->vdd_reg)
+               regulator_disable(data->vdd_reg);
 }
 
 static int stk3310_suspend(struct device *dev)
 {
        struct stk3310_data *data;
+       int ret;
 
        data = iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
 
-       return stk3310_set_state(data, STK3310_STATE_STANDBY);
+       ret = stk3310_set_state(data, STK3310_STATE_STANDBY);
+       if (ret)
+               return ret;
+
+       if (data->vdd_reg) {
+               regcache_mark_dirty(data->regmap);
+               regulator_disable(data->vdd_reg);
+       }
+
+       return 0;
 }
 
 static int stk3310_resume(struct device *dev)
 {
-       u8 state = 0;
        struct stk3310_data *data;
+       u8 state = 0;
+       int ret;
 
        data = iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
+
+       if (data->vdd_reg) {
+               ret = regulator_enable(data->vdd_reg);
+               if (ret) {
+                       dev_err(dev, "Failed to re-enable regulator vdd\n");
+                       return ret;
+               }
+
+               usleep_range(1000, 2000);
+               regcache_sync(data->regmap);
+       }
+
        if (data->ps_enabled)
                state |= STK3310_STATE_EN_PS;
        if (data->als_enabled)
-- 
2.44.0