[PATCH v2 2/2] iio: imu: inv_mpu6050: use set_power_itg function in i2c mux

[Jean-Baptiste Maneyrol <JManeyrol@xxxxxxxxxxxxxx>]

Set power function is rewritten manually inside i2c mux select.
Better use the already identical existing function.

Signed-off-by: Jean-Baptiste Maneyrol <jmaneyrol@xxxxxxxxxxxxxx>
---
 drivers/iio/imu/inv_mpu6050/inv_mpu_i2c.c | 33 ++++++++++++-------------------
 1 file changed, 13 insertions(+), 20 deletions(-)

diff --git a/drivers/iio/imu/inv_mpu6050/inv_mpu_i2c.c b/drivers/iio/imu/inv_mpu6050/inv_mpu_i2c.c
index fcd7a92..8a8a5bd 100644
--- a/drivers/iio/imu/inv_mpu6050/inv_mpu_i2c.c
+++ b/drivers/iio/imu/inv_mpu6050/inv_mpu_i2c.c
@@ -29,25 +29,19 @@ static int inv_mpu6050_select_bypass(struct i2c_mux_core *muxc, u32 chan_id)
 {
        struct iio_dev *indio_dev = i2c_mux_priv(muxc);
        struct inv_mpu6050_state *st = iio_priv(indio_dev);
-       int ret = 0;
+       int ret;

-       /* Use the same mutex which was used everywhere to protect power-op */
        mutex_lock(&st->lock);
-       if (!st->powerup_count) {
-               ret = regmap_write(st->map, st->reg->pwr_mgmt_1, 0);
-               if (ret)
-                       goto write_error;

-               usleep_range(INV_MPU6050_REG_UP_TIME_MIN,
-                            INV_MPU6050_REG_UP_TIME_MAX);
-       }
-       if (!ret) {
-               st->powerup_count++;
-               ret = regmap_write(st->map, st->reg->int_pin_cfg,
-                                  INV_MPU6050_INT_PIN_CFG |
-                                  INV_MPU6050_BIT_BYPASS_EN);
-       }
-write_error:
+       ret = inv_mpu6050_set_power_itg(st, true);
+       if (ret)
+               goto error_unlock;
+
+       ret = regmap_write(st->map, st->reg->int_pin_cfg,
+                          INV_MPU6050_INT_PIN_CFG |
+                          INV_MPU6050_BIT_BYPASS_EN);
+
+error_unlock:
        mutex_unlock(&st->lock);

        return ret;
@@ -59,12 +53,11 @@ static int inv_mpu6050_deselect_bypass(struct i2c_mux_core *muxc, u32 chan_id)
        struct inv_mpu6050_state *st = iio_priv(indio_dev);

        mutex_lock(&st->lock);
+
        /* It doesn't really mattter, if any of the calls fails */
        regmap_write(st->map, st->reg->int_pin_cfg, INV_MPU6050_INT_PIN_CFG);
-       st->powerup_count--;
-       if (!st->powerup_count)
-               regmap_write(st->map, st->reg->pwr_mgmt_1,
-                            INV_MPU6050_BIT_SLEEP);
+       inv_mpu6050_set_power_itg(st, false);
+
        mutex_unlock(&st->lock);

        return 0;
--
2.7.4
--
To unsubscribe from this list: send the line "unsubscribe linux-iio" in
the body of a message to majordomo@xxxxxxxxxxxxxxx
More majordomo info at  http://vger.kernel.org/majordomo-info.html