| /* |
| * ADE7758 Poly Phase Multifunction Energy Metering IC driver |
| * |
| * Copyright 2010-2011 Analog Devices Inc. |
| * |
| * Licensed under the GPL-2. |
| */ |
| #include <linux/export.h> |
| #include <linux/interrupt.h> |
| #include <linux/kernel.h> |
| #include <linux/spi/spi.h> |
| #include <linux/slab.h> |
| #include <asm/unaligned.h> |
| |
| #include "../iio.h" |
| #include "../ring_sw.h" |
| #include "../trigger_consumer.h" |
| #include "ade7758.h" |
| |
| /** |
| * ade7758_spi_read_burst() - read data registers |
| * @dev: device associated with child of actual device (iio_dev or iio_trig) |
| **/ |
| static int ade7758_spi_read_burst(struct device *dev) |
| { |
| struct iio_dev *indio_dev = dev_get_drvdata(dev); |
| struct ade7758_state *st = iio_priv(indio_dev); |
| int ret; |
| |
| ret = spi_sync(st->us, &st->ring_msg); |
| if (ret) |
| dev_err(&st->us->dev, "problem when reading WFORM value\n"); |
| |
| return ret; |
| } |
| |
| static int ade7758_write_waveform_type(struct device *dev, unsigned type) |
| { |
| int ret; |
| u8 reg; |
| |
| ret = ade7758_spi_read_reg_8(dev, |
| ADE7758_WAVMODE, |
| ®); |
| if (ret) |
| goto out; |
| |
| reg &= ~0x1F; |
| reg |= type & 0x1F; |
| |
| ret = ade7758_spi_write_reg_8(dev, |
| ADE7758_WAVMODE, |
| reg); |
| out: |
| return ret; |
| } |
| |
| /* Whilst this makes a lot of calls to iio_sw_ring functions - it is to device |
| * specific to be rolled into the core. |
| */ |
| static irqreturn_t ade7758_trigger_handler(int irq, void *p) |
| { |
| struct iio_poll_func *pf = p; |
| struct iio_dev *indio_dev = pf->indio_dev; |
| struct iio_buffer *ring = indio_dev->buffer; |
| struct ade7758_state *st = iio_priv(indio_dev); |
| s64 dat64[2]; |
| u32 *dat32 = (u32 *)dat64; |
| |
| if (ring->scan_count) |
| if (ade7758_spi_read_burst(&indio_dev->dev) >= 0) |
| *dat32 = get_unaligned_be32(&st->rx_buf[5]) & 0xFFFFFF; |
| |
| /* Guaranteed to be aligned with 8 byte boundary */ |
| if (ring->scan_timestamp) |
| dat64[1] = pf->timestamp; |
| |
| ring->access->store_to(ring, (u8 *)dat64, pf->timestamp); |
| |
| iio_trigger_notify_done(indio_dev->trig); |
| |
| return IRQ_HANDLED; |
| } |
| |
| /** |
| * ade7758_ring_preenable() setup the parameters of the ring before enabling |
| * |
| * The complex nature of the setting of the nuber of bytes per datum is due |
| * to this driver currently ensuring that the timestamp is stored at an 8 |
| * byte boundary. |
| **/ |
| static int ade7758_ring_preenable(struct iio_dev *indio_dev) |
| { |
| struct ade7758_state *st = iio_priv(indio_dev); |
| struct iio_buffer *ring = indio_dev->buffer; |
| size_t d_size; |
| unsigned channel; |
| |
| if (!ring->scan_count) |
| return -EINVAL; |
| |
| channel = find_first_bit(ring->scan_mask, indio_dev->masklength); |
| |
| d_size = st->ade7758_ring_channels[channel].scan_type.storagebits / 8; |
| |
| if (ring->scan_timestamp) { |
| d_size += sizeof(s64); |
| |
| if (d_size % sizeof(s64)) |
| d_size += sizeof(s64) - (d_size % sizeof(s64)); |
| } |
| |
| if (indio_dev->buffer->access->set_bytes_per_datum) |
| indio_dev->buffer->access-> |
| set_bytes_per_datum(indio_dev->buffer, d_size); |
| |
| ade7758_write_waveform_type(&indio_dev->dev, |
| st->ade7758_ring_channels[channel].address); |
| |
| return 0; |
| } |
| |
| static const struct iio_buffer_setup_ops ade7758_ring_setup_ops = { |
| .preenable = &ade7758_ring_preenable, |
| .postenable = &iio_triggered_buffer_postenable, |
| .predisable = &iio_triggered_buffer_predisable, |
| }; |
| |
| void ade7758_unconfigure_ring(struct iio_dev *indio_dev) |
| { |
| iio_dealloc_pollfunc(indio_dev->pollfunc); |
| iio_sw_rb_free(indio_dev->buffer); |
| } |
| |
| int ade7758_configure_ring(struct iio_dev *indio_dev) |
| { |
| struct ade7758_state *st = iio_priv(indio_dev); |
| int ret = 0; |
| |
| indio_dev->buffer = iio_sw_rb_allocate(indio_dev); |
| if (!indio_dev->buffer) { |
| ret = -ENOMEM; |
| return ret; |
| } |
| |
| /* Effectively select the ring buffer implementation */ |
| indio_dev->buffer->access = &ring_sw_access_funcs; |
| indio_dev->buffer->setup_ops = &ade7758_ring_setup_ops; |
| indio_dev->buffer->owner = THIS_MODULE; |
| |
| indio_dev->pollfunc = iio_alloc_pollfunc(&iio_pollfunc_store_time, |
| &ade7758_trigger_handler, |
| 0, |
| indio_dev, |
| "ade7759_consumer%d", |
| indio_dev->id); |
| if (indio_dev->pollfunc == NULL) { |
| ret = -ENOMEM; |
| goto error_iio_sw_rb_free; |
| } |
| |
| indio_dev->modes |= INDIO_BUFFER_TRIGGERED; |
| |
| st->tx_buf[0] = ADE7758_READ_REG(ADE7758_RSTATUS); |
| st->tx_buf[1] = 0; |
| st->tx_buf[2] = 0; |
| st->tx_buf[3] = 0; |
| st->tx_buf[4] = ADE7758_READ_REG(ADE7758_WFORM); |
| st->tx_buf[5] = 0; |
| st->tx_buf[6] = 0; |
| st->tx_buf[7] = 0; |
| |
| /* build spi ring message */ |
| st->ring_xfer[0].tx_buf = &st->tx_buf[0]; |
| st->ring_xfer[0].len = 1; |
| st->ring_xfer[0].bits_per_word = 8; |
| st->ring_xfer[0].delay_usecs = 4; |
| st->ring_xfer[1].rx_buf = &st->rx_buf[1]; |
| st->ring_xfer[1].len = 3; |
| st->ring_xfer[1].bits_per_word = 8; |
| st->ring_xfer[1].cs_change = 1; |
| |
| st->ring_xfer[2].tx_buf = &st->tx_buf[4]; |
| st->ring_xfer[2].len = 1; |
| st->ring_xfer[2].bits_per_word = 8; |
| st->ring_xfer[2].delay_usecs = 1; |
| st->ring_xfer[3].rx_buf = &st->rx_buf[5]; |
| st->ring_xfer[3].len = 3; |
| st->ring_xfer[3].bits_per_word = 8; |
| |
| spi_message_init(&st->ring_msg); |
| spi_message_add_tail(&st->ring_xfer[0], &st->ring_msg); |
| spi_message_add_tail(&st->ring_xfer[1], &st->ring_msg); |
| spi_message_add_tail(&st->ring_xfer[2], &st->ring_msg); |
| spi_message_add_tail(&st->ring_xfer[3], &st->ring_msg); |
| |
| return 0; |
| |
| error_iio_sw_rb_free: |
| iio_sw_rb_free(indio_dev->buffer); |
| return ret; |
| } |
| |
| void ade7758_uninitialize_ring(struct iio_dev *indio_dev) |
| { |
| iio_buffer_unregister(indio_dev); |
| } |