The read operation for the I2C function has many duplications that can
be generalized into a single function. This patch reworks the read
operation for I2C to centralizes all similar code in a single function.

It is possible to remove all the old interface to use the new one,
however, for keeping the things simple and working this patch maintain
legacy interface.

Signed-off-by: Rodrigo Siqueira <rodrigosiqueiram...@gmail.com>
---
Changes in v3:
 - Remove the use of defines that not improve the readability
 - Replace variable name "bytes" for "bits"
 - Update commit message

 drivers/staging/iio/meter/ade7854-i2c.c | 106 +++++++++++---------------------
 1 file changed, 37 insertions(+), 69 deletions(-)

diff --git a/drivers/staging/iio/meter/ade7854-i2c.c 
b/drivers/staging/iio/meter/ade7854-i2c.c
index 29e959fdb932..63793f9664c7 100644
--- a/drivers/staging/iio/meter/ade7854-i2c.c
+++ b/drivers/staging/iio/meter/ade7854-i2c.c
@@ -65,9 +65,10 @@ static int ade7854_i2c_write_reg(struct device *dev,
        return ret < 0 ? ret : 0;
 }
 
-static int ade7854_i2c_read_reg_8(struct device *dev,
-                                 u16 reg_address,
-                                 u8 *val)
+static int ade7854_i2c_read_reg(struct device *dev,
+                               u16 reg_address,
+                               u32 *val,
+                               int bits)
 {
        struct iio_dev *indio_dev = dev_to_iio_dev(dev);
        struct ade7854_state *st = iio_priv(indio_dev);
@@ -79,95 +80,62 @@ static int ade7854_i2c_read_reg_8(struct device *dev,
 
        ret = i2c_master_send(st->i2c, st->tx, 2);
        if (ret < 0)
-               goto out;
+               goto unlock;
 
-       ret = i2c_master_recv(st->i2c, st->rx, 1);
+       ret = i2c_master_recv(st->i2c, st->rx, bits);
        if (ret < 0)
-               goto out;
+               goto unlock;
+
+       switch (bits) {
+       case 8:
+               *val = st->rx[0];
+               break;
+       case 16:
+               *val = (st->rx[0] << 8) | st->rx[1];
+               break;
+       case 24:
+               *val = (st->rx[0] << 16) | (st->rx[1] << 8) | st->rx[2];
+               break;
+       case 32:
+               *val = (st->rx[0] << 24) | (st->rx[1] << 16) |
+                       (st->rx[2] << 8) | st->rx[3];
+               break;
+       default:
+               ret = -EINVAL;
+               goto unlock;
+       }
 
-       *val = st->rx[0];
-out:
+unlock:
        mutex_unlock(&st->buf_lock);
        return ret;
 }
 
+static int ade7854_i2c_read_reg_8(struct device *dev,
+                                 u16 reg_address,
+                                 u8 *val)
+{
+       return ade7854_i2c_read_reg(dev, reg_address, (u32 *)val, 8);
+}
+
 static int ade7854_i2c_read_reg_16(struct device *dev,
                                   u16 reg_address,
                                   u16 *val)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct ade7854_state *st = iio_priv(indio_dev);
-       int ret;
-
-       mutex_lock(&st->buf_lock);
-       st->tx[0] = (reg_address >> 8) & 0xFF;
-       st->tx[1] = reg_address & 0xFF;
-
-       ret = i2c_master_send(st->i2c, st->tx, 2);
-       if (ret < 0)
-               goto out;
-
-       ret = i2c_master_recv(st->i2c, st->rx, 2);
-       if (ret < 0)
-               goto out;
-
-       *val = (st->rx[0] << 8) | st->rx[1];
-out:
-       mutex_unlock(&st->buf_lock);
-       return ret;
+       return ade7854_i2c_read_reg(dev, reg_address, (u32 *)val, 16);
 }
 
 static int ade7854_i2c_read_reg_24(struct device *dev,
                                   u16 reg_address,
                                   u32 *val)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct ade7854_state *st = iio_priv(indio_dev);
-       int ret;
-
-       mutex_lock(&st->buf_lock);
-       st->tx[0] = (reg_address >> 8) & 0xFF;
-       st->tx[1] = reg_address & 0xFF;
-
-       ret = i2c_master_send(st->i2c, st->tx, 2);
-       if (ret < 0)
-               goto out;
-
-       ret = i2c_master_recv(st->i2c, st->rx, 3);
-       if (ret < 0)
-               goto out;
-
-       *val = (st->rx[0] << 16) | (st->rx[1] << 8) | st->rx[2];
-out:
-       mutex_unlock(&st->buf_lock);
-       return ret;
+       return ade7854_i2c_read_reg(dev, reg_address, (u32 *)val, 24);
 }
 
 static int ade7854_i2c_read_reg_32(struct device *dev,
                                   u16 reg_address,
                                   u32 *val)
 {
-       struct iio_dev *indio_dev = dev_to_iio_dev(dev);
-       struct ade7854_state *st = iio_priv(indio_dev);
-       int ret;
-
-       mutex_lock(&st->buf_lock);
-       st->tx[0] = (reg_address >> 8) & 0xFF;
-       st->tx[1] = reg_address & 0xFF;
-
-       ret = i2c_master_send(st->i2c, st->tx, 2);
-       if (ret < 0)
-               goto out;
-
-       ret = i2c_master_recv(st->i2c, st->rx, 4);
-       if (ret < 0)
-               goto out;
-
-       *val = (st->rx[0] << 24) | (st->rx[1] << 16) |
-               (st->rx[2] << 8) | st->rx[3];
-out:
-       mutex_unlock(&st->buf_lock);
-       return ret;
+       return ade7854_i2c_read_reg(dev, reg_address, (u32 *)val, 32);
 }
 
 static int ade7854_i2c_probe(struct i2c_client *client,
-- 
2.16.2

_______________________________________________
devel mailing list
de...@linuxdriverproject.org
http://driverdev.linuxdriverproject.org/mailman/listinfo/driverdev-devel

Reply via email to