/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Common library for ADIS16XXX devices
 *
 * Copyright 2012 Analog Devices Inc.
 *   Author: Lars-Peter Clausen <lars@metafoo.de>
 */

#ifndef __IIO_ADIS_H__
#define __IIO_ADIS_H__

#include <linux/cleanup.h>
#include <linux/spi/spi.h>
#include <linux/interrupt.h>
#include <linux/iio/iio.h>
#include <linux/iio/types.h>

#define ADIS_WRITE_REG(reg) ((0x80 | (reg)))
#define ADIS_READ_REG(reg) ((reg) & 0x7f)

#define ADIS_PAGE_SIZE 0x80
#define ADIS_REG_PAGE_ID 0x00

struct adis;
struct iio_dev_attr;

/**
 * struct adis_timeouts - ADIS chip variant timeouts
 * @reset_ms - Wait time after rst pin goes inactive
 * @sw_reset_ms - Wait time after sw reset command
 * @self_test_ms - Wait time after self test command
 */
struct adis_timeout {
        u16 reset_ms;
        u16 sw_reset_ms;
        u16 self_test_ms;
};

/**
 * struct adis_data - ADIS chip variant specific data
 * @read_delay: SPI delay for read operations in us
 * @write_delay: SPI delay for write operations in us
 * @cs_change_delay: SPI delay between CS changes in us
 * @glob_cmd_reg: Register address of the GLOB_CMD register
 * @msc_ctrl_reg: Register address of the MSC_CTRL register
 * @diag_stat_reg: Register address of the DIAG_STAT register
 * @diag_stat_size:     Length (in bytes) of the DIAG_STAT register. If 0 the
 *                      default length is 2 bytes long.
 * @prod_id_reg: Register address of the PROD_ID register
 * @prod_id: Product ID code that should be expected when reading @prod_id_reg
 * @self_test_mask: Bitmask of supported self-test operations
 * @self_test_reg: Register address to request self test command
 * @self_test_no_autoclear: True if device's self-test needs clear of ctrl reg
 * @status_error_msgs: Array of error messages
 * @status_error_mask: Bitmask of errors supported by the device
 * @timeouts: Chip specific delays
 * @enable_irq: Hook for ADIS devices that have a special IRQ enable/disable
 * @unmasked_drdy: True for devices that cannot mask/unmask the data ready pin
 * @has_paging: True if ADIS device has paged registers
 * @has_fifo: True if ADIS device has a hardware FIFO
 * @burst_reg_cmd:      Register command that triggers burst
 * @burst_len:          Burst size in the SPI RX buffer. If @burst_max_len is defined,
 *                      this should be the minimum size supported by the device.
 * @burst_max_len:      Holds the maximum burst size when the device supports
 *                      more than one burst mode with different sizes
 * @burst_max_speed_hz: Maximum spi speed that can be used in burst mode
 */
struct adis_data {
        unsigned int read_delay;
        unsigned int write_delay;
        unsigned int cs_change_delay;

        unsigned int glob_cmd_reg;
        unsigned int msc_ctrl_reg;
        unsigned int diag_stat_reg;
        unsigned int diag_stat_size;
        unsigned int prod_id_reg;

        unsigned int prod_id;

        unsigned int self_test_mask;
        unsigned int self_test_reg;
        bool self_test_no_autoclear;
        const struct adis_timeout *timeouts;

        const char * const *status_error_msgs;
        unsigned int status_error_mask;

        int (*enable_irq)(struct adis *adis, bool enable);
        bool unmasked_drdy;

        bool has_paging;
        bool has_fifo;

        unsigned int burst_reg_cmd;
        unsigned int burst_len;
        unsigned int burst_max_len;
        unsigned int burst_max_speed_hz;
};

/**
 * struct adis_ops: Custom ops for adis devices.
 * @write: Custom spi write implementation.
 * @read: Custom spi read implementation.
 * @reset: Custom sw reset implementation. The custom implementation does not
 *         need to sleep after the reset. It's done by the library already.
 */
struct adis_ops {
        int (*write)(struct adis *adis, unsigned int reg, unsigned int value,
                     unsigned int size);
        int (*read)(struct adis *adis, unsigned int reg, unsigned int *value,
                    unsigned int size);
        int (*reset)(struct adis *adis);
};

/**
 * struct adis - ADIS device instance data
 * @spi: Reference to SPI device which owns this ADIS IIO device
 * @trig: IIO trigger object data
 * @data: ADIS chip variant specific data
 * @burst_extra_len: Burst extra length. Should only be used by devices that can
 *                   dynamically change their burst mode length.
 * @ops: ops struct for custom read and write functions
 * @state_lock: Lock used by the device to protect state
 * @msg: SPI message object
 * @xfer: SPI transfer objects to be used for a @msg
 * @current_page: Some ADIS devices have registers, this selects current page
 * @irq_flag: IRQ handling flags as passed to request_irq()
 * @buffer: Data buffer for information read from the device
 * @tx: DMA safe TX buffer for SPI transfers
 * @rx: DMA safe RX buffer for SPI transfers
 */
struct adis {
        struct spi_device       *spi;
        struct iio_trigger      *trig;

        const struct adis_data  *data;
        unsigned int            burst_extra_len;
        const struct adis_ops   *ops;
        /*
         * The state_lock is meant to be used during operations that require
         * a sequence of SPI R/W in order to protect the SPI transfer
         * information (fields 'xfer', 'msg' & 'current_page') between
         * potential concurrent accesses.
         * This lock is used by all "adis_{functions}" that have to read/write
         * registers. These functions also have unlocked variants
         * (see "__adis_{functions}"), which don't hold this lock.
         * This allows users of the ADIS library to group SPI R/W into
         * the drivers, but they also must manage this lock themselves.
         */
        struct mutex            state_lock;
        struct spi_message      msg;
        struct spi_transfer     *xfer;
        unsigned int            current_page;
        unsigned long           irq_flag;
        void                    *buffer;

        u8                      tx[10] __aligned(IIO_DMA_MINALIGN);
        u8                      rx[4];
};

int adis_init(struct adis *adis, struct iio_dev *indio_dev,
              struct spi_device *spi, const struct adis_data *data);
int __adis_reset(struct adis *adis);

/**
 * adis_reset() - Reset the device
 * @adis: The adis device
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_reset(struct adis *adis)
{
        guard(mutex)(&adis->state_lock);
        return __adis_reset(adis);
}

int __adis_write_reg(struct adis *adis, unsigned int reg,
                     unsigned int val, unsigned int size);
int __adis_read_reg(struct adis *adis, unsigned int reg,
                    unsigned int *val, unsigned int size);

/**
 * __adis_write_reg_8() - Write single byte to a register (unlocked)
 * @adis: The adis device
 * @reg: The address of the register to be written
 * @val: The value to write
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int __adis_write_reg_8(struct adis *adis, unsigned int reg,
                                     u8 val)
{
        return adis->ops->write(adis, reg, val, 1);
}

/**
 * __adis_write_reg_16() - Write 2 bytes to a pair of registers (unlocked)
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: Value to be written
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int __adis_write_reg_16(struct adis *adis, unsigned int reg,
                                      u16 val)
{
        return adis->ops->write(adis, reg, val, 2);
}

/**
 * __adis_write_reg_32() - write 4 bytes to four registers (unlocked)
 * @adis: The adis device
 * @reg: The address of the lower of the four register
 * @val: Value to be written
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int __adis_write_reg_32(struct adis *adis, unsigned int reg,
                                      u32 val)
{
        return adis->ops->write(adis, reg, val, 4);
}

/**
 * __adis_read_reg_16() - read 2 bytes from a 16-bit register (unlocked)
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: The value read back from the device
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int __adis_read_reg_16(struct adis *adis, unsigned int reg,
                                     u16 *val)
{
        unsigned int tmp;
        int ret;

        ret = adis->ops->read(adis, reg, &tmp, 2);
        if (ret == 0)
                *val = tmp;

        return ret;
}

/**
 * __adis_read_reg_32() - read 4 bytes from a 32-bit register (unlocked)
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: The value read back from the device
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int __adis_read_reg_32(struct adis *adis, unsigned int reg,
                                     u32 *val)
{
        unsigned int tmp;
        int ret;

        ret = adis->ops->read(adis, reg, &tmp, 4);
        if (ret == 0)
                *val = tmp;

        return ret;
}

/**
 * adis_write_reg() - write N bytes to register
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: The value to write to device (up to 4 bytes)
 * @size: The size of the @value (in bytes)
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_write_reg(struct adis *adis, unsigned int reg,
                                 unsigned int val, unsigned int size)
{
        guard(mutex)(&adis->state_lock);
        return adis->ops->write(adis, reg, val, size);
}

/**
 * adis_read_reg() - read N bytes from register
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: The value read back from the device
 * @size: The size of the @val buffer
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static int adis_read_reg(struct adis *adis, unsigned int reg,
                         unsigned int *val, unsigned int size)
{
        guard(mutex)(&adis->state_lock);
        return adis->ops->read(adis, reg, val, size);
}

/**
 * adis_write_reg_8() - Write single byte to a register
 * @adis: The adis device
 * @reg: The address of the register to be written
 * @val: The value to write
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_write_reg_8(struct adis *adis, unsigned int reg,
                                   u8 val)
{
        return adis_write_reg(adis, reg, val, 1);
}

/**
 * adis_write_reg_16() - Write 2 bytes to a pair of registers
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: Value to be written
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_write_reg_16(struct adis *adis, unsigned int reg,
                                    u16 val)
{
        return adis_write_reg(adis, reg, val, 2);
}

/**
 * adis_write_reg_32() - write 4 bytes to four registers
 * @adis: The adis device
 * @reg: The address of the lower of the four register
 * @val: Value to be written
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_write_reg_32(struct adis *adis, unsigned int reg,
                                    u32 val)
{
        return adis_write_reg(adis, reg, val, 4);
}

/**
 * adis_read_reg_16() - read 2 bytes from a 16-bit register
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: The value read back from the device
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_read_reg_16(struct adis *adis, unsigned int reg,
                                   u16 *val)
{
        unsigned int tmp;
        int ret;

        ret = adis_read_reg(adis, reg, &tmp, 2);
        if (ret == 0)
                *val = tmp;

        return ret;
}

/**
 * adis_read_reg_32() - read 4 bytes from a 32-bit register
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @val: The value read back from the device
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_read_reg_32(struct adis *adis, unsigned int reg,
                                   u32 *val)
{
        unsigned int tmp;
        int ret;

        ret = adis_read_reg(adis, reg, &tmp, 4);
        if (ret == 0)
                *val = tmp;

        return ret;
}

int __adis_update_bits_base(struct adis *adis, unsigned int reg, const u32 mask,
                            const u32 val, u8 size);
/**
 * adis_update_bits_base() - ADIS Update bits function - Locked version
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @mask: Bitmask to change
 * @val: Value to be written
 * @size: Size of the register to update
 *
 * Updates the desired bits of @reg in accordance with @mask and @val.
 *
 * Returns: %0 on success, a negative error code otherwise
 */
static inline int adis_update_bits_base(struct adis *adis, unsigned int reg,
                                        const u32 mask, const u32 val, u8 size)
{
        guard(mutex)(&adis->state_lock);
        return __adis_update_bits_base(adis, reg, mask, val, size);
}

/**
 * adis_update_bits() - Wrapper macro for adis_update_bits_base - Locked version
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @mask: Bitmask to change
 * @val: Value to be written
 *
 * This macro evaluates the sizeof of @val at compile time and calls
 * adis_update_bits_base() accordingly. Be aware that using MACROS/DEFINES for
 * @val can lead to undesired behavior if the register to update is 16bit.
 */
#define adis_update_bits(adis, reg, mask, val) ({                       \
        BUILD_BUG_ON(sizeof(val) != 2 && sizeof(val) != 4);             \
        adis_update_bits_base(adis, reg, mask, val, sizeof(val));       \
})

/**
 * adis_update_bits() - Wrapper macro for adis_update_bits_base
 * @adis: The adis device
 * @reg: The address of the lower of the two registers
 * @mask: Bitmask to change
 * @val: Value to be written
 *
 * This macro evaluates the sizeof of @val at compile time and calls
 * adis_update_bits_base() accordingly. Be aware that using MACROS/DEFINES for
 * @val can lead to undesired behavior if the register to update is 16bit.
 */
#define __adis_update_bits(adis, reg, mask, val) ({                     \
        BUILD_BUG_ON(sizeof(val) != 2 && sizeof(val) != 4);             \
        __adis_update_bits_base(adis, reg, mask, val, sizeof(val));     \
})

int __adis_check_status(struct adis *adis);
int __adis_initial_startup(struct adis *adis);
int __adis_enable_irq(struct adis *adis, bool enable);

static inline int adis_enable_irq(struct adis *adis, bool enable)
{
        guard(mutex)(&adis->state_lock);
        return __adis_enable_irq(adis, enable);
}

static inline int adis_check_status(struct adis *adis)
{
        guard(mutex)(&adis->state_lock);
        return __adis_check_status(adis);
}

#define adis_dev_auto_lock(adis)        guard(mutex)(&(adis)->state_lock)
#define adis_dev_auto_scoped_lock(adis) \
        scoped_guard(mutex, &(adis)->state_lock)

int adis_single_conversion(struct iio_dev *indio_dev,
                           const struct iio_chan_spec *chan,
                           unsigned int error_mask, int *val);

#define ADIS_VOLTAGE_CHAN(addr, si, chan, name, info_all, bits) { \
        .type = IIO_VOLTAGE, \
        .indexed = 1, \
        .channel = (chan), \
        .extend_name = name, \
        .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) | \
                BIT(IIO_CHAN_INFO_SCALE), \
        .info_mask_shared_by_all = info_all, \
        .address = (addr), \
        .scan_index = (si), \
        .scan_type = { \
                .sign = 'u', \
                .realbits = (bits), \
                .storagebits = 16, \
                .endianness = IIO_BE, \
        }, \
}

#define ADIS_SUPPLY_CHAN(addr, si, info_all, bits) \
        ADIS_VOLTAGE_CHAN(addr, si, 0, "supply", info_all, bits)

#define ADIS_AUX_ADC_CHAN(addr, si, info_all, bits) \
        ADIS_VOLTAGE_CHAN(addr, si, 1, NULL, info_all, bits)

#define ADIS_TEMP_CHAN(addr, si, info_all, bits) { \
        .type = IIO_TEMP, \
        .indexed = 1, \
        .channel = 0, \
        .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) | \
                BIT(IIO_CHAN_INFO_SCALE) | \
                BIT(IIO_CHAN_INFO_OFFSET), \
        .info_mask_shared_by_all = info_all, \
        .address = (addr), \
        .scan_index = (si), \
        .scan_type = { \
                .sign = 'u', \
                .realbits = (bits), \
                .storagebits = 16, \
                .endianness = IIO_BE, \
        }, \
}

#define ADIS_MOD_CHAN(_type, mod, addr, si, info_sep, info_all, bits) { \
        .type = (_type), \
        .modified = 1, \
        .channel2 = IIO_MOD_ ## mod, \
        .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) | \
                 (info_sep), \
        .info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE), \
        .info_mask_shared_by_all = info_all, \
        .address = (addr), \
        .scan_index = (si), \
        .scan_type = { \
                .sign = 's', \
                .realbits = (bits), \
                .storagebits = 16, \
                .endianness = IIO_BE, \
        }, \
}

#define ADIS_ACCEL_CHAN(mod, addr, si, info_sep, info_all, bits) \
        ADIS_MOD_CHAN(IIO_ACCEL, mod, addr, si, info_sep, info_all, bits)

#define ADIS_GYRO_CHAN(mod, addr, si, info_sep, info_all, bits)         \
        ADIS_MOD_CHAN(IIO_ANGL_VEL, mod, addr, si, info_sep, info_all, bits)

#define ADIS_INCLI_CHAN(mod, addr, si, info_sep, info_all, bits) \
        ADIS_MOD_CHAN(IIO_INCLI, mod, addr, si, info_sep, info_all, bits)

#define ADIS_ROT_CHAN(mod, addr, si, info_sep, info_all, bits) \
        ADIS_MOD_CHAN(IIO_ROT, mod, addr, si, info_sep, info_all, bits)

#define devm_adis_setup_buffer_and_trigger(adis, indio_dev, trigger_handler)    \
        devm_adis_setup_buffer_and_trigger_with_attrs((adis), (indio_dev),      \
                                                      (trigger_handler), NULL,  \
                                                      NULL)

#ifdef CONFIG_IIO_ADIS_LIB_BUFFER

int
devm_adis_setup_buffer_and_trigger_with_attrs(struct adis *adis,
                                              struct iio_dev *indio_dev,
                                              irq_handler_t trigger_handler,
                                              const struct iio_buffer_setup_ops *ops,
                                              const struct iio_dev_attr **buffer_attrs);

int devm_adis_probe_trigger(struct adis *adis, struct iio_dev *indio_dev);

int adis_update_scan_mode(struct iio_dev *indio_dev,
                          const unsigned long *scan_mask);

#else /* CONFIG_IIO_BUFFER */

static inline int
devm_adis_setup_buffer_and_trigger_with_attrs(struct adis *adis,
                                              struct iio_dev *indio_dev,
                                              irq_handler_t trigger_handler,
                                              const struct iio_buffer_setup_ops *ops,
                                              const struct iio_dev_attr **buffer_attrs)
{
        return 0;
}

static inline int devm_adis_probe_trigger(struct adis *adis,
                                          struct iio_dev *indio_dev)
{
        return 0;
}

#define adis_update_scan_mode NULL

#endif /* CONFIG_IIO_BUFFER */

#ifdef CONFIG_DEBUG_FS

int adis_debugfs_reg_access(struct iio_dev *indio_dev,
                            unsigned int reg, unsigned int writeval,
                            unsigned int *readval);

#else

#define adis_debugfs_reg_access NULL

#endif

#endif