Rework read SPI function to reduce the code duplication and centralizes
all the task in a single function.

Signed-off-by: Rodrigo Siqueira <rodrigosiqueiram...@gmail.com>
---
 drivers/staging/iio/meter/ade7854-spi.c | 132 ++++++++++----------------------
 1 file changed, 41 insertions(+), 91 deletions(-)

diff --git a/drivers/staging/iio/meter/ade7854-spi.c 
b/drivers/staging/iio/meter/ade7854-spi.c
index df3df85f9440..0feef24aa888 100644
--- a/drivers/staging/iio/meter/ade7854-spi.c
+++ b/drivers/staging/iio/meter/ade7854-spi.c
@@ -67,9 +67,10 @@ static int ade7854_spi_write_reg(struct device *dev,
        return ret;
 }
 
-static int ade7854_spi_read_reg_8(struct device *dev,
-                                 u16 reg_address,
-                                 u8 *val)
+static int ade7854_spi_read_reg(struct device *dev,
+                               u16 reg_address,
+                               u32 *val,
+                               enum data_size type)
 {
        struct iio_dev *indio_dev = dev_to_iio_dev(dev);
        struct ade7854_state *st = iio_priv(indio_dev);
@@ -82,7 +83,7 @@ static int ade7854_spi_read_reg_8(struct device *dev,
                }, {
                        .rx_buf = st->rx,
                        .bits_per_word = 8,
-                       .len = 1,
+                       .len = type,
                }
        };
 
@@ -94,51 +95,52 @@ static int ade7854_spi_read_reg_8(struct device *dev,
 
        ret = spi_sync_transfer(st->spi, xfers, ARRAY_SIZE(xfers));
        if (ret) {
-               dev_err(&st->spi->dev, "problem when reading 8 bit register 
0x%02X",
+               dev_err(&st->spi->dev, "problem when reading register 0x%02X",
                        reg_address);
-               goto error_ret;
+               goto error_spi_read_unlock;
+       }
+
+       switch (type) {
+       case DATA_SIZE_8_BITS:
+               *val = st->rx[0];
+               break;
+       case DATA_SIZE_16_BITS:
+               *val = be16_to_cpup((const __be16 *)st->rx);
+               break;
+       case DATA_SIZE_24_BITS:
+               *val = (st->rx[0] << 16) | (st->rx[1] << 8) | st->rx[2];
+               break;
+       case DATA_SIZE_32_BITS:
+               *val = be32_to_cpup((const __be32 *)st->rx);
+               break;
        }
-       *val = st->rx[0];
 
-error_ret:
+error_spi_read_unlock:
        mutex_unlock(&st->buf_lock);
        return ret;
 }
 
+static int ade7854_spi_read_reg_8(struct device *dev,
+                                 u16 reg_address,
+                                 u8 *val)
+{
+       int ret;
+
+       ret = ade7854_spi_read_reg(dev, reg_address, (u32 *)val,
+                                  DATA_SIZE_8_BITS);
+
+       return ret;
+}
+
 static int ade7854_spi_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;
-       struct spi_transfer xfers[] = {
-               {
-                       .tx_buf = st->tx,
-                       .bits_per_word = 8,
-                       .len = 3,
-               }, {
-                       .rx_buf = st->rx,
-                       .bits_per_word = 8,
-                       .len = 2,
-               }
-       };
 
-       mutex_lock(&st->buf_lock);
-       st->tx[0] = ADE7854_READ_REG;
-       st->tx[1] = (reg_address >> 8) & 0xFF;
-       st->tx[2] = reg_address & 0xFF;
+       ret = ade7854_spi_read_reg(dev, reg_address, (u32 *)val,
+                                  DATA_SIZE_16_BITS);
 
-       ret = spi_sync_transfer(st->spi, xfers, ARRAY_SIZE(xfers));
-       if (ret) {
-               dev_err(&st->spi->dev, "problem when reading 16 bit register 
0x%02X",
-                       reg_address);
-               goto error_ret;
-       }
-       *val = be16_to_cpup((const __be16 *)st->rx);
-
-error_ret:
-       mutex_unlock(&st->buf_lock);
        return ret;
 }
 
@@ -146,37 +148,11 @@ static int ade7854_spi_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;
-       struct spi_transfer xfers[] = {
-               {
-                       .tx_buf = st->tx,
-                       .bits_per_word = 8,
-                       .len = 3,
-               }, {
-                       .rx_buf = st->rx,
-                       .bits_per_word = 8,
-                       .len = 3,
-               }
-       };
 
-       mutex_lock(&st->buf_lock);
-
-       st->tx[0] = ADE7854_READ_REG;
-       st->tx[1] = (reg_address >> 8) & 0xFF;
-       st->tx[2] = reg_address & 0xFF;
+       ret = ade7854_spi_read_reg(dev, reg_address, (u32 *)val,
+                                  DATA_SIZE_24_BITS);
 
-       ret = spi_sync_transfer(st->spi, xfers, ARRAY_SIZE(xfers));
-       if (ret) {
-               dev_err(&st->spi->dev, "problem when reading 24 bit register 
0x%02X",
-                       reg_address);
-               goto error_ret;
-       }
-       *val = (st->rx[0] << 16) | (st->rx[1] << 8) | st->rx[2];
-
-error_ret:
-       mutex_unlock(&st->buf_lock);
        return ret;
 }
 
@@ -184,37 +160,11 @@ static int ade7854_spi_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;
-       struct spi_transfer xfers[] = {
-               {
-                       .tx_buf = st->tx,
-                       .bits_per_word = 8,
-                       .len = 3,
-               }, {
-                       .rx_buf = st->rx,
-                       .bits_per_word = 8,
-                       .len = 4,
-               }
-       };
 
-       mutex_lock(&st->buf_lock);
-
-       st->tx[0] = ADE7854_READ_REG;
-       st->tx[1] = (reg_address >> 8) & 0xFF;
-       st->tx[2] = reg_address & 0xFF;
+       ret = ade7854_spi_read_reg(dev, reg_address, (u32 *)val,
+                                  DATA_SIZE_32_BITS);
 
-       ret = spi_sync_transfer(st->spi, xfers, ARRAY_SIZE(xfers));
-       if (ret) {
-               dev_err(&st->spi->dev, "problem when reading 32 bit register 
0x%02X",
-                       reg_address);
-               goto error_ret;
-       }
-       *val = be32_to_cpup((const __be32 *)st->rx);
-
-error_ret:
-       mutex_unlock(&st->buf_lock);
        return ret;
 }
 
-- 
2.16.2

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

Reply via email to