proximity: vl53l0x: Handle the VDD regulator

Handle the regulator supplying the VDD pin of VL53L0X.

Signed-off-by: Markuss Broks <markuss.broks@gmail.com>
Link: https://lore.kernel.org/r/20220523175344.5845-4-markuss.broks@gmail.com
Signed-off-by: Jonathan Cameron <Jonathan.Cameron@huawei.com>

diff --git a/drivers/iio/proximity/vl53l0x-i2c.c b/drivers/iio/proximity/vl53l0x-i2c.c
index 7272732..db2bdba 100644 (file)
--- a/drivers/iio/proximity/vl53l0x-i2c.c
+++ b/drivers/iio/proximity/vl53l0x-i2c.c
@@ -43,6 +43,7 @@
 struct vl53l0x_data {
        struct i2c_client *client;
        struct completion completion;
+       struct regulator *vdd_supply;
 };
 
 static irqreturn_t vl53l0x_handle_irq(int irq, void *priv)
@@ -191,10 +192,31 @@ static const struct iio_info vl53l0x_info = {
        .read_raw = vl53l0x_read_raw,
 };
 
+static void vl53l0x_power_off(void *_data)
+{
+       struct vl53l0x_data *data = _data;
+
+       regulator_disable(data->vdd_supply);
+}
+
+static int vl53l0x_power_on(struct vl53l0x_data *data)
+{
+       int ret;
+
+       ret = regulator_enable(data->vdd_supply);
+       if (ret)
+               return ret;
+
+       usleep_range(3200, 5000);
+
+       return 0;
+}
+
 static int vl53l0x_probe(struct i2c_client *client)
 {
        struct vl53l0x_data *data;
        struct iio_dev *indio_dev;
+       int error;
 
        indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
        if (!indio_dev)
@@ -209,6 +231,21 @@ static int vl53l0x_probe(struct i2c_client *client)
                                     I2C_FUNC_SMBUS_BYTE_DATA))
                return -EOPNOTSUPP;
 
+       data->vdd_supply = devm_regulator_get_optional(&client->dev, "vdd");
+       if (IS_ERR(data->vdd_supply))
+               return dev_err_probe(&client->dev, PTR_ERR(data->vdd_supply),
+                                    "Unable to get VDD regulator\n");
+
+       error = vl53l0x_power_on(data);
+       if (error)
+               return dev_err_probe(&client->dev, error,
+                                    "Failed to power on the chip\n");
+
+       error = devm_add_action_or_reset(&client->dev, vl53l0x_power_off, data);
+       if (error)
+               return dev_err_probe(&client->dev, error,
+                                    "Failed to install poweroff action\n");
+
        indio_dev->name = "vl53l0x";
        indio_dev->info = &vl53l0x_info;
        indio_dev->channels = vl53l0x_channels;