// SPDX-License-Identifier: GPL-2.0-only
/* ADC driver for AXP20X and AXP22X PMICs
 *
 * Copyright (c) 2016 Free Electrons NextThing Co.
 *      Quentin Schulz <quentin.schulz@free-electrons.com>
 */

#include <linux/unaligned.h>
#include <linux/bitfield.h>
#include <linux/completion.h>
#include <linux/interrupt.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/mod_devicetable.h>
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>
#include <linux/property.h>
#include <linux/regmap.h>
#include <linux/thermal.h>

#include <linux/iio/iio.h>
#include <linux/iio/driver.h>
#include <linux/iio/machine.h>
#include <linux/mfd/axp20x.h>

#define AXP192_ADC_EN1_MASK                     GENMASK(7, 0)
#define AXP192_ADC_EN2_MASK                     (GENMASK(3, 0) | BIT(7))

#define AXP20X_ADC_EN1_MASK                     GENMASK(7, 0)
#define AXP20X_ADC_EN2_MASK                     (GENMASK(3, 2) | BIT(7))

#define AXP22X_ADC_EN1_MASK                     (GENMASK(7, 5) | BIT(0))

#define AXP717_ADC_EN1_MASK                     GENMASK(7, 0)

#define AXP192_GPIO30_IN_RANGE_GPIO0            BIT(0)
#define AXP192_GPIO30_IN_RANGE_GPIO1            BIT(1)
#define AXP192_GPIO30_IN_RANGE_GPIO2            BIT(2)
#define AXP192_GPIO30_IN_RANGE_GPIO3            BIT(3)

#define AXP20X_GPIO10_IN_RANGE_GPIO0            BIT(0)
#define AXP20X_GPIO10_IN_RANGE_GPIO1            BIT(1)

#define AXP20X_ADC_RATE_MASK                    GENMASK(7, 6)
#define AXP20X_ADC_RATE_HZ(x)                   ((ilog2((x) / 25) << 6) & AXP20X_ADC_RATE_MASK)

#define AXP22X_ADC_RATE_HZ(x)                   ((ilog2((x) / 100) << 6) & AXP20X_ADC_RATE_MASK)

#define AXP717_ADC_DATA_TS                      0x00
#define AXP717_ADC_DATA_TEMP                    0x01
#define AXP717_ADC_DATA_VMID                    0x02
#define AXP717_ADC_DATA_BKUP_BATT               0x03

#define AXP717_ADC_DATA_MASK                    GENMASK(13, 0)

#define AXP813_V_I_ADC_RATE_MASK                GENMASK(5, 4)
#define AXP813_ADC_RATE_MASK                    (AXP20X_ADC_RATE_MASK | AXP813_V_I_ADC_RATE_MASK)
#define AXP813_TS_GPIO0_ADC_RATE_HZ(x)          AXP20X_ADC_RATE_HZ(x)
#define AXP813_V_I_ADC_RATE_HZ(x)               ((ilog2((x) / 100) << 4) & AXP813_V_I_ADC_RATE_MASK)
#define AXP813_ADC_RATE_HZ(x)                   (AXP20X_ADC_RATE_HZ(x) | AXP813_V_I_ADC_RATE_HZ(x))

#define AXP20X_ADC_CHANNEL(_channel, _name, _type, _reg)        \
        {                                                       \
                .type = _type,                                  \
                .indexed = 1,                                   \
                .channel = _channel,                            \
                .address = _reg,                                \
                .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |  \
                                      BIT(IIO_CHAN_INFO_SCALE), \
                .datasheet_name = _name,                        \
        }

#define AXP20X_ADC_CHANNEL_OFFSET(_channel, _name, _type, _reg) \
        {                                                       \
                .type = _type,                                  \
                .indexed = 1,                                   \
                .channel = _channel,                            \
                .address = _reg,                                \
                .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |  \
                                      BIT(IIO_CHAN_INFO_SCALE) |\
                                      BIT(IIO_CHAN_INFO_OFFSET),\
                .datasheet_name = _name,                        \
        }

struct axp_data;

struct axp20x_adc_iio {
        struct regmap           *regmap;
        const struct axp_data   *data;
};

enum axp192_adc_channel_v {
        AXP192_ACIN_V = 0,
        AXP192_VBUS_V,
        AXP192_TS_IN,
        AXP192_GPIO0_V,
        AXP192_GPIO1_V,
        AXP192_GPIO2_V,
        AXP192_GPIO3_V,
        AXP192_IPSOUT_V,
        AXP192_BATT_V,
};

enum axp192_adc_channel_i {
        AXP192_ACIN_I = 0,
        AXP192_VBUS_I,
        AXP192_BATT_CHRG_I,
        AXP192_BATT_DISCHRG_I,
};

enum axp20x_adc_channel_v {
        AXP20X_ACIN_V = 0,
        AXP20X_VBUS_V,
        AXP20X_TS_IN,
        AXP20X_GPIO0_V,
        AXP20X_GPIO1_V,
        AXP20X_IPSOUT_V,
        AXP20X_BATT_V,
};

enum axp20x_adc_channel_i {
        AXP20X_ACIN_I = 0,
        AXP20X_VBUS_I,
        AXP20X_BATT_CHRG_I,
        AXP20X_BATT_DISCHRG_I,
};

enum axp22x_adc_channel_v {
        AXP22X_TS_IN = 0,
        AXP22X_BATT_V,
};

enum axp22x_adc_channel_i {
        AXP22X_BATT_CHRG_I = 1,
        AXP22X_BATT_DISCHRG_I,
};

enum axp717_adc_channel_v {
        AXP717_BATT_V = 0,
        AXP717_TS_IN,
        AXP717_VBUS_V,
        AXP717_VSYS_V,
        AXP717_DIE_TEMP_V,
        AXP717_VMID_V = 6,
        AXP717_BKUP_BATT_V,
};

enum axp717_adc_channel_i {
        AXP717_BATT_CHRG_I = 5,
};

enum axp813_adc_channel_v {
        AXP813_TS_IN = 0,
        AXP813_GPIO0_V,
        AXP813_BATT_V,
};

static const struct iio_map axp20x_maps[] = {
        IIO_MAP("vbus_v", "axp20x-usb-power-supply", "vbus_v"),
        IIO_MAP("vbus_i", "axp20x-usb-power-supply", "vbus_i"),
        IIO_MAP("acin_v", "axp20x-ac-power-supply", "acin_v"),
        IIO_MAP("acin_i", "axp20x-ac-power-supply", "acin_i"),
        IIO_MAP("batt_v", "axp20x-battery-power-supply", "batt_v"),
        IIO_MAP("batt_chrg_i", "axp20x-battery-power-supply", "batt_chrg_i"),
        IIO_MAP("batt_dischrg_i", "axp20x-battery-power-supply", "batt_dischrg_i"),
        { }
};

static const struct iio_map axp22x_maps[] = {
        IIO_MAP("batt_v", "axp20x-battery-power-supply", "batt_v"),
        IIO_MAP("batt_chrg_i", "axp20x-battery-power-supply", "batt_chrg_i"),
        IIO_MAP("batt_dischrg_i", "axp20x-battery-power-supply", "batt_dischrg_i"),
        { }
};

static const struct iio_map axp717_maps[] = {
        {
                .consumer_dev_name = "axp20x-usb-power-supply",
                .consumer_channel = "vbus_v",
                .adc_channel_label = "vbus_v",
        }, {
                .consumer_dev_name = "axp20x-battery-power-supply",
                .consumer_channel = "batt_v",
                .adc_channel_label = "batt_v",
        }, {
                .consumer_dev_name = "axp20x-battery-power-supply",
                .consumer_channel = "batt_chrg_i",
                .adc_channel_label = "batt_chrg_i",
        },
        { }
};

/*
 * Channels are mapped by physical system. Their channels share the same index.
 * i.e. acin_i is in_current0_raw and acin_v is in_voltage0_raw.
 * The only exception is for the battery. batt_v will be in_voltage6_raw and
 * charge current in_current6_raw and discharge current will be in_current7_raw.
 */
static const struct iio_chan_spec axp192_adc_channels[] = {
        AXP20X_ADC_CHANNEL(AXP192_ACIN_V, "acin_v", IIO_VOLTAGE,
                           AXP20X_ACIN_V_ADC_H),
        AXP20X_ADC_CHANNEL(AXP192_ACIN_I, "acin_i", IIO_CURRENT,
                           AXP20X_ACIN_I_ADC_H),
        AXP20X_ADC_CHANNEL(AXP192_VBUS_V, "vbus_v", IIO_VOLTAGE,
                           AXP20X_VBUS_V_ADC_H),
        AXP20X_ADC_CHANNEL(AXP192_VBUS_I, "vbus_i", IIO_CURRENT,
                           AXP20X_VBUS_I_ADC_H),
        {
                .type = IIO_TEMP,
                .address = AXP20X_TEMP_ADC_H,
                .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
                                      BIT(IIO_CHAN_INFO_SCALE) |
                                      BIT(IIO_CHAN_INFO_OFFSET),
                .datasheet_name = "pmic_temp",
        },
        AXP20X_ADC_CHANNEL_OFFSET(AXP192_GPIO0_V, "gpio0_v", IIO_VOLTAGE,
                                  AXP20X_GPIO0_V_ADC_H),
        AXP20X_ADC_CHANNEL_OFFSET(AXP192_GPIO1_V, "gpio1_v", IIO_VOLTAGE,
                                  AXP20X_GPIO1_V_ADC_H),
        AXP20X_ADC_CHANNEL_OFFSET(AXP192_GPIO2_V, "gpio2_v", IIO_VOLTAGE,
                                  AXP192_GPIO2_V_ADC_H),
        AXP20X_ADC_CHANNEL_OFFSET(AXP192_GPIO3_V, "gpio3_v", IIO_VOLTAGE,
                                  AXP192_GPIO3_V_ADC_H),
        AXP20X_ADC_CHANNEL(AXP192_IPSOUT_V, "ipsout_v", IIO_VOLTAGE,
                           AXP20X_IPSOUT_V_HIGH_H),
        AXP20X_ADC_CHANNEL(AXP192_BATT_V, "batt_v", IIO_VOLTAGE,
                           AXP20X_BATT_V_H),
        AXP20X_ADC_CHANNEL(AXP192_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
                           AXP20X_BATT_CHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP192_BATT_DISCHRG_I, "batt_dischrg_i", IIO_CURRENT,
                           AXP20X_BATT_DISCHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP192_TS_IN, "ts_v", IIO_VOLTAGE,
                           AXP20X_TS_IN_H),
};

static const struct iio_chan_spec axp20x_adc_channels[] = {
        AXP20X_ADC_CHANNEL(AXP20X_ACIN_V, "acin_v", IIO_VOLTAGE,
                           AXP20X_ACIN_V_ADC_H),
        AXP20X_ADC_CHANNEL(AXP20X_ACIN_I, "acin_i", IIO_CURRENT,
                           AXP20X_ACIN_I_ADC_H),
        AXP20X_ADC_CHANNEL(AXP20X_VBUS_V, "vbus_v", IIO_VOLTAGE,
                           AXP20X_VBUS_V_ADC_H),
        AXP20X_ADC_CHANNEL(AXP20X_VBUS_I, "vbus_i", IIO_CURRENT,
                           AXP20X_VBUS_I_ADC_H),
        {
                .type = IIO_TEMP,
                .address = AXP20X_TEMP_ADC_H,
                .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
                                      BIT(IIO_CHAN_INFO_SCALE) |
                                      BIT(IIO_CHAN_INFO_OFFSET),
                .datasheet_name = "pmic_temp",
        },
        AXP20X_ADC_CHANNEL_OFFSET(AXP20X_GPIO0_V, "gpio0_v", IIO_VOLTAGE,
                                  AXP20X_GPIO0_V_ADC_H),
        AXP20X_ADC_CHANNEL_OFFSET(AXP20X_GPIO1_V, "gpio1_v", IIO_VOLTAGE,
                                  AXP20X_GPIO1_V_ADC_H),
        AXP20X_ADC_CHANNEL(AXP20X_IPSOUT_V, "ipsout_v", IIO_VOLTAGE,
                           AXP20X_IPSOUT_V_HIGH_H),
        AXP20X_ADC_CHANNEL(AXP20X_BATT_V, "batt_v", IIO_VOLTAGE,
                           AXP20X_BATT_V_H),
        AXP20X_ADC_CHANNEL(AXP20X_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
                           AXP20X_BATT_CHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP20X_BATT_DISCHRG_I, "batt_dischrg_i", IIO_CURRENT,
                           AXP20X_BATT_DISCHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP20X_TS_IN, "ts_v", IIO_VOLTAGE,
                           AXP20X_TS_IN_H),
};

static const struct iio_chan_spec axp22x_adc_channels[] = {
        {
                .type = IIO_TEMP,
                .address = AXP22X_PMIC_TEMP_H,
                .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
                                      BIT(IIO_CHAN_INFO_SCALE) |
                                      BIT(IIO_CHAN_INFO_OFFSET),
                .datasheet_name = "pmic_temp",
        },
        AXP20X_ADC_CHANNEL(AXP22X_BATT_V, "batt_v", IIO_VOLTAGE,
                           AXP20X_BATT_V_H),
        AXP20X_ADC_CHANNEL(AXP22X_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
                           AXP20X_BATT_CHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP22X_BATT_DISCHRG_I, "batt_dischrg_i", IIO_CURRENT,
                           AXP20X_BATT_DISCHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP22X_TS_IN, "ts_v", IIO_VOLTAGE,
                           AXP22X_TS_ADC_H),
};

/*
 * Scale and offset is unknown for temp, ts, batt_chrg_i, vmid_v, and
 * bkup_batt_v channels. Leaving scale and offset undefined for now.
 */
static const struct iio_chan_spec axp717_adc_channels[] = {
        AXP20X_ADC_CHANNEL(AXP717_BATT_V, "batt_v", IIO_VOLTAGE,
                           AXP717_BATT_V_H),
        AXP20X_ADC_CHANNEL(AXP717_TS_IN, "ts_v", IIO_VOLTAGE,
                           AXP717_ADC_DATA_H),
        AXP20X_ADC_CHANNEL(AXP717_VBUS_V, "vbus_v", IIO_VOLTAGE,
                           AXP717_VBUS_V_H),
        AXP20X_ADC_CHANNEL(AXP717_VSYS_V, "vsys_v", IIO_VOLTAGE,
                           AXP717_VSYS_V_H),
        AXP20X_ADC_CHANNEL(AXP717_DIE_TEMP_V, "pmic_temp", IIO_TEMP,
                           AXP717_ADC_DATA_H),
        AXP20X_ADC_CHANNEL(AXP717_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
                           AXP717_BATT_CHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP717_VMID_V, "vmid_v", IIO_VOLTAGE,
                           AXP717_ADC_DATA_H),
        AXP20X_ADC_CHANNEL(AXP717_BKUP_BATT_V, "bkup_batt_v", IIO_VOLTAGE,
                           AXP717_ADC_DATA_H),
};

static const struct iio_chan_spec axp813_adc_channels[] = {
        {
                .type = IIO_TEMP,
                .address = AXP22X_PMIC_TEMP_H,
                .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
                                      BIT(IIO_CHAN_INFO_SCALE) |
                                      BIT(IIO_CHAN_INFO_OFFSET),
                .datasheet_name = "pmic_temp",
        },
        AXP20X_ADC_CHANNEL(AXP813_GPIO0_V, "gpio0_v", IIO_VOLTAGE,
                           AXP288_GP_ADC_H),
        AXP20X_ADC_CHANNEL(AXP813_BATT_V, "batt_v", IIO_VOLTAGE,
                           AXP20X_BATT_V_H),
        AXP20X_ADC_CHANNEL(AXP22X_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
                           AXP20X_BATT_CHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP22X_BATT_DISCHRG_I, "batt_dischrg_i", IIO_CURRENT,
                           AXP20X_BATT_DISCHRG_I_H),
        AXP20X_ADC_CHANNEL(AXP813_TS_IN, "ts_v", IIO_VOLTAGE,
                           AXP288_TS_ADC_H),
};

static int axp192_adc_raw(struct iio_dev *indio_dev,
                          struct iio_chan_spec const *chan, int *val)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        int ret, size;

        if (chan->type == IIO_CURRENT &&
            (chan->channel == AXP192_BATT_CHRG_I ||
             chan->channel == AXP192_BATT_DISCHRG_I))
                size = 13;
        else
                size = 12;

        ret = axp20x_read_variable_width(info->regmap, chan->address, size);
        if (ret < 0)
                return ret;

        *val = ret;
        return IIO_VAL_INT;
}

static int axp20x_adc_raw(struct iio_dev *indio_dev,
                          struct iio_chan_spec const *chan, int *val)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        int ret, size;

        /*
         * N.B.:  Unlike the Chinese datasheets tell, the charging current is
         * stored on 12 bits, not 13 bits. Only discharging current is on 13
         * bits.
         */
        if (chan->type == IIO_CURRENT && chan->channel == AXP20X_BATT_DISCHRG_I)
                size = 13;
        else
                size = 12;

        ret = axp20x_read_variable_width(info->regmap, chan->address, size);
        if (ret < 0)
                return ret;

        *val = ret;
        return IIO_VAL_INT;
}

static int axp22x_adc_raw(struct iio_dev *indio_dev,
                          struct iio_chan_spec const *chan, int *val)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        int ret;

        ret = axp20x_read_variable_width(info->regmap, chan->address, 12);
        if (ret < 0)
                return ret;

        *val = ret;
        return IIO_VAL_INT;
}

static int axp717_adc_raw(struct iio_dev *indio_dev,
                          struct iio_chan_spec const *chan, int *val)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        u8 bulk_reg[2];
        int ret;

        /*
         * A generic "ADC data" channel is used for TS, tdie, vmid,
         * and vbackup. This channel must both first be enabled and
         * also selected before it can be read.
         */
        switch (chan->channel) {
        case AXP717_TS_IN:
                regmap_write(info->regmap, AXP717_ADC_DATA_SEL,
                             AXP717_ADC_DATA_TS);
                break;
        case AXP717_DIE_TEMP_V:
                regmap_write(info->regmap, AXP717_ADC_DATA_SEL,
                             AXP717_ADC_DATA_TEMP);
                break;
        case AXP717_VMID_V:
                regmap_write(info->regmap, AXP717_ADC_DATA_SEL,
                             AXP717_ADC_DATA_VMID);
                break;
        case AXP717_BKUP_BATT_V:
                regmap_write(info->regmap, AXP717_ADC_DATA_SEL,
                             AXP717_ADC_DATA_BKUP_BATT);
                break;
        default:
                break;
        }

        /*
         * All channels are 14 bits, with the first 2 bits on the high
         * register reserved and the remaining bits as the ADC value.
         */
        ret = regmap_bulk_read(info->regmap, chan->address, bulk_reg, 2);
        if (ret < 0)
                return ret;

        *val = FIELD_GET(AXP717_ADC_DATA_MASK, get_unaligned_be16(bulk_reg));
        return IIO_VAL_INT;
}

static int axp813_adc_raw(struct iio_dev *indio_dev,
                          struct iio_chan_spec const *chan, int *val)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        int ret;

        ret = axp20x_read_variable_width(info->regmap, chan->address, 12);
        if (ret < 0)
                return ret;

        *val = ret;
        return IIO_VAL_INT;
}

static int axp192_adc_scale_voltage(int channel, int *val, int *val2)
{
        switch (channel) {
        case AXP192_ACIN_V:
        case AXP192_VBUS_V:
                *val = 1;
                *val2 = 700000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP192_GPIO0_V:
        case AXP192_GPIO1_V:
        case AXP192_GPIO2_V:
        case AXP192_GPIO3_V:
                *val = 0;
                *val2 = 500000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP192_BATT_V:
                *val = 1;
                *val2 = 100000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP192_IPSOUT_V:
                *val = 1;
                *val2 = 400000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP192_TS_IN:
                /* 0.8 mV per LSB */
                *val = 0;
                *val2 = 800000;
                return IIO_VAL_INT_PLUS_MICRO;

        default:
                return -EINVAL;
        }
}

static int axp20x_adc_scale_voltage(int channel, int *val, int *val2)
{
        switch (channel) {
        case AXP20X_ACIN_V:
        case AXP20X_VBUS_V:
                *val = 1;
                *val2 = 700000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP20X_GPIO0_V:
        case AXP20X_GPIO1_V:
                *val = 0;
                *val2 = 500000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP20X_BATT_V:
                *val = 1;
                *val2 = 100000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP20X_IPSOUT_V:
                *val = 1;
                *val2 = 400000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP20X_TS_IN:
                /* 0.8 mV per LSB */
                *val = 0;
                *val2 = 800000;
                return IIO_VAL_INT_PLUS_MICRO;

        default:
                return -EINVAL;
        }
}

static int axp22x_adc_scale_voltage(int channel, int *val, int *val2)
{
        switch (channel) {
        case AXP22X_BATT_V:
                /* 1.1 mV per LSB */
                *val = 1;
                *val2 = 100000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP22X_TS_IN:
                /* 0.8 mV per LSB */
                *val = 0;
                *val2 = 800000;
                return IIO_VAL_INT_PLUS_MICRO;

        default:
                return -EINVAL;
        }
}
static int axp813_adc_scale_voltage(int channel, int *val, int *val2)
{
        switch (channel) {
        case AXP813_GPIO0_V:
                *val = 0;
                *val2 = 800000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP813_BATT_V:
                *val = 1;
                *val2 = 100000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP813_TS_IN:
                /* 0.8 mV per LSB */
                *val = 0;
                *val2 = 800000;
                return IIO_VAL_INT_PLUS_MICRO;

        default:
                return -EINVAL;
        }
}

static int axp20x_adc_scale_current(int channel, int *val, int *val2)
{
        switch (channel) {
        case AXP20X_ACIN_I:
                *val = 0;
                *val2 = 625000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP20X_VBUS_I:
                *val = 0;
                *val2 = 375000;
                return IIO_VAL_INT_PLUS_MICRO;

        case AXP20X_BATT_DISCHRG_I:
        case AXP20X_BATT_CHRG_I:
                *val = 0;
                *val2 = 500000;
                return IIO_VAL_INT_PLUS_MICRO;

        default:
                return -EINVAL;
        }
}

static int axp192_adc_scale(struct iio_chan_spec const *chan, int *val,
                            int *val2)
{
        switch (chan->type) {
        case IIO_VOLTAGE:
                return axp192_adc_scale_voltage(chan->channel, val, val2);

        case IIO_CURRENT:
                /*
                 * AXP192 current channels are identical to the AXP20x,
                 * therefore we can re-use the scaling function.
                 */
                return axp20x_adc_scale_current(chan->channel, val, val2);

        case IIO_TEMP:
                *val = 100;
                return IIO_VAL_INT;

        default:
                return -EINVAL;
        }
}

static int axp20x_adc_scale(struct iio_chan_spec const *chan, int *val,
                            int *val2)
{
        switch (chan->type) {
        case IIO_VOLTAGE:
                return axp20x_adc_scale_voltage(chan->channel, val, val2);

        case IIO_CURRENT:
                return axp20x_adc_scale_current(chan->channel, val, val2);

        case IIO_TEMP:
                *val = 100;
                return IIO_VAL_INT;

        default:
                return -EINVAL;
        }
}

static int axp22x_adc_scale(struct iio_chan_spec const *chan, int *val,
                            int *val2)
{
        switch (chan->type) {
        case IIO_VOLTAGE:
                return axp22x_adc_scale_voltage(chan->channel, val, val2);

        case IIO_CURRENT:
                *val = 1;
                return IIO_VAL_INT;

        case IIO_TEMP:
                *val = 100;
                return IIO_VAL_INT;

        default:
                return -EINVAL;
        }
}

static int axp717_adc_scale(struct iio_chan_spec const *chan, int *val,
                            int *val2)
{
        switch (chan->type) {
        case IIO_VOLTAGE:
                *val = 1;
                return IIO_VAL_INT;

        case IIO_CURRENT:
                *val = 1;
                return IIO_VAL_INT;

        case IIO_TEMP:
                *val = 100;
                return IIO_VAL_INT;

        default:
                return -EINVAL;
        }
}

static int axp813_adc_scale(struct iio_chan_spec const *chan, int *val,
                            int *val2)
{
        switch (chan->type) {
        case IIO_VOLTAGE:
                return axp813_adc_scale_voltage(chan->channel, val, val2);

        case IIO_CURRENT:
                *val = 1;
                return IIO_VAL_INT;

        case IIO_TEMP:
                *val = 100;
                return IIO_VAL_INT;

        default:
                return -EINVAL;
        }
}

static int axp192_adc_offset_voltage(struct iio_dev *indio_dev, int channel,
                                     int *val)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        unsigned int regval;
        int ret;

        ret = regmap_read(info->regmap, AXP192_GPIO30_IN_RANGE, &regval);
        if (ret < 0)
                return ret;

        switch (channel) {
        case AXP192_GPIO0_V:
                regval = FIELD_GET(AXP192_GPIO30_IN_RANGE_GPIO0, regval);
                break;

        case AXP192_GPIO1_V:
                regval = FIELD_GET(AXP192_GPIO30_IN_RANGE_GPIO1, regval);
                break;

        case AXP192_GPIO2_V:
                regval = FIELD_GET(AXP192_GPIO30_IN_RANGE_GPIO2, regval);
                break;

        case AXP192_GPIO3_V:
                regval = FIELD_GET(AXP192_GPIO30_IN_RANGE_GPIO3, regval);
                break;

        default:
                return -EINVAL;
        }

        *val = regval ? 700000 : 0;
        return IIO_VAL_INT;
}

static int axp20x_adc_offset_voltage(struct iio_dev *indio_dev, int channel,
                                     int *val)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        unsigned int regval;
        int ret;

        ret = regmap_read(info->regmap, AXP20X_GPIO10_IN_RANGE, &regval);
        if (ret < 0)
                return ret;

        switch (channel) {
        case AXP20X_GPIO0_V:
                regval = FIELD_GET(AXP20X_GPIO10_IN_RANGE_GPIO0, regval);
                break;

        case AXP20X_GPIO1_V:
                regval = FIELD_GET(AXP20X_GPIO10_IN_RANGE_GPIO1, regval);
                break;

        default:
                return -EINVAL;
        }

        *val = regval ? 700000 : 0;
        return IIO_VAL_INT;
}

static int axp192_adc_offset(struct iio_dev *indio_dev,
                             struct iio_chan_spec const *chan, int *val)
{
        switch (chan->type) {
        case IIO_VOLTAGE:
                return axp192_adc_offset_voltage(indio_dev, chan->channel, val);

        case IIO_TEMP:
                *val = -1447;
                return IIO_VAL_INT;

        default:
                return -EINVAL;
        }
}

static int axp20x_adc_offset(struct iio_dev *indio_dev,
                             struct iio_chan_spec const *chan, int *val)
{
        switch (chan->type) {
        case IIO_VOLTAGE:
                return axp20x_adc_offset_voltage(indio_dev, chan->channel, val);

        case IIO_TEMP:
                *val = -1447;
                return IIO_VAL_INT;

        default:
                return -EINVAL;
        }
}

static int axp192_read_raw(struct iio_dev *indio_dev,
                           struct iio_chan_spec const *chan, int *val,
                           int *val2, long mask)
{
        switch (mask) {
        case IIO_CHAN_INFO_OFFSET:
                return axp192_adc_offset(indio_dev, chan, val);

        case IIO_CHAN_INFO_SCALE:
                return axp192_adc_scale(chan, val, val2);

        case IIO_CHAN_INFO_RAW:
                return axp192_adc_raw(indio_dev, chan, val);

        default:
                return -EINVAL;
        }
}

static int axp20x_read_raw(struct iio_dev *indio_dev,
                           struct iio_chan_spec const *chan, int *val,
                           int *val2, long mask)
{
        switch (mask) {
        case IIO_CHAN_INFO_OFFSET:
                return axp20x_adc_offset(indio_dev, chan, val);

        case IIO_CHAN_INFO_SCALE:
                return axp20x_adc_scale(chan, val, val2);

        case IIO_CHAN_INFO_RAW:
                return axp20x_adc_raw(indio_dev, chan, val);

        default:
                return -EINVAL;
        }
}

static int axp22x_read_raw(struct iio_dev *indio_dev,
                           struct iio_chan_spec const *chan, int *val,
                           int *val2, long mask)
{
        switch (mask) {
        case IIO_CHAN_INFO_OFFSET:
                /* For PMIC temp only */
                *val = -2677;
                return IIO_VAL_INT;

        case IIO_CHAN_INFO_SCALE:
                return axp22x_adc_scale(chan, val, val2);

        case IIO_CHAN_INFO_RAW:
                return axp22x_adc_raw(indio_dev, chan, val);

        default:
                return -EINVAL;
        }
}

static int axp717_read_raw(struct iio_dev *indio_dev,
                           struct iio_chan_spec const *chan, int *val,
                           int *val2, long mask)
{
        switch (mask) {
        case IIO_CHAN_INFO_SCALE:
                return axp717_adc_scale(chan, val, val2);

        case IIO_CHAN_INFO_RAW:
                return axp717_adc_raw(indio_dev, chan, val);

        default:
                return -EINVAL;
        }
}

static int axp813_read_raw(struct iio_dev *indio_dev,
                           struct iio_chan_spec const *chan, int *val,
                           int *val2, long mask)
{
        switch (mask) {
        case IIO_CHAN_INFO_OFFSET:
                *val = -2667;
                return IIO_VAL_INT;

        case IIO_CHAN_INFO_SCALE:
                return axp813_adc_scale(chan, val, val2);

        case IIO_CHAN_INFO_RAW:
                return axp813_adc_raw(indio_dev, chan, val);

        default:
                return -EINVAL;
        }
}

static int axp192_write_raw(struct iio_dev *indio_dev,
                            struct iio_chan_spec const *chan, int val, int val2,
                            long mask)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        unsigned int regmask, regval;

        /*
         * The AXP192 PMIC allows the user to choose between 0V and 0.7V offsets
         * for (independently) GPIO0-3 when in ADC mode.
         */
        if (mask != IIO_CHAN_INFO_OFFSET)
                return -EINVAL;

        if (val != 0 && val != 700000)
                return -EINVAL;

        switch (chan->channel) {
        case AXP192_GPIO0_V:
                regmask = AXP192_GPIO30_IN_RANGE_GPIO0;
                regval = FIELD_PREP(AXP192_GPIO30_IN_RANGE_GPIO0, !!val);
                break;

        case AXP192_GPIO1_V:
                regmask = AXP192_GPIO30_IN_RANGE_GPIO1;
                regval = FIELD_PREP(AXP192_GPIO30_IN_RANGE_GPIO1, !!val);
                break;

        case AXP192_GPIO2_V:
                regmask = AXP192_GPIO30_IN_RANGE_GPIO2;
                regval = FIELD_PREP(AXP192_GPIO30_IN_RANGE_GPIO2, !!val);
                break;

        case AXP192_GPIO3_V:
                regmask = AXP192_GPIO30_IN_RANGE_GPIO3;
                regval = FIELD_PREP(AXP192_GPIO30_IN_RANGE_GPIO3, !!val);
                break;

        default:
                return -EINVAL;
        }

        return regmap_update_bits(info->regmap, AXP192_GPIO30_IN_RANGE, regmask, regval);
}

static int axp20x_write_raw(struct iio_dev *indio_dev,
                            struct iio_chan_spec const *chan, int val, int val2,
                            long mask)
{
        struct axp20x_adc_iio *info = iio_priv(indio_dev);
        unsigned int regmask, regval;

        /*
         * The AXP20X PMIC allows the user to choose between 0V and 0.7V offsets
         * for (independently) GPIO0 and GPIO1 when in ADC mode.
         */
        if (mask != IIO_CHAN_INFO_OFFSET)
                return -EINVAL;

        if (val != 0 && val != 700000)
                return -EINVAL;

        switch (chan->channel) {
        case AXP20X_GPIO0_V:
                regmask = AXP20X_GPIO10_IN_RANGE_GPIO0;
                regval = FIELD_PREP(AXP20X_GPIO10_IN_RANGE_GPIO0, !!val);
                break;

        case AXP20X_GPIO1_V:
                regmask = AXP20X_GPIO10_IN_RANGE_GPIO1;
                regval = FIELD_PREP(AXP20X_GPIO10_IN_RANGE_GPIO1, !!val);
                break;

        default:
                return -EINVAL;
        }

        return regmap_update_bits(info->regmap, AXP20X_GPIO10_IN_RANGE, regmask, regval);
}

static const struct iio_info axp192_adc_iio_info = {
        .read_raw = axp192_read_raw,
        .write_raw = axp192_write_raw,
};

static const struct iio_info axp20x_adc_iio_info = {
        .read_raw = axp20x_read_raw,
        .write_raw = axp20x_write_raw,
};

static const struct iio_info axp22x_adc_iio_info = {
        .read_raw = axp22x_read_raw,
};

static const struct iio_info axp717_adc_iio_info = {
        .read_raw = axp717_read_raw,
};

static const struct iio_info axp813_adc_iio_info = {
        .read_raw = axp813_read_raw,
};

static int axp20x_adc_rate(struct axp20x_adc_iio *info, int rate)
{
        return regmap_update_bits(info->regmap, AXP20X_ADC_RATE,
                                  AXP20X_ADC_RATE_MASK,
                                  AXP20X_ADC_RATE_HZ(rate));
}

static int axp22x_adc_rate(struct axp20x_adc_iio *info, int rate)
{
        return regmap_update_bits(info->regmap, AXP20X_ADC_RATE,
                                  AXP20X_ADC_RATE_MASK,
                                  AXP22X_ADC_RATE_HZ(rate));
}

static int axp813_adc_rate(struct axp20x_adc_iio *info, int rate)
{
        return regmap_update_bits(info->regmap, AXP813_ADC_RATE,
                                 AXP813_ADC_RATE_MASK,
                                 AXP813_ADC_RATE_HZ(rate));
}

struct axp_data {
        const struct iio_info           *iio_info;
        int                             num_channels;
        struct iio_chan_spec const      *channels;
        unsigned long                   adc_en1;
        unsigned long                   adc_en1_mask;
        unsigned long                   adc_en2;
        unsigned long                   adc_en2_mask;
        int                             (*adc_rate)(struct axp20x_adc_iio *info,
                                                    int rate);
        const struct iio_map            *maps;
};

static const struct axp_data axp192_data = {
        .iio_info = &axp192_adc_iio_info,
        .num_channels = ARRAY_SIZE(axp192_adc_channels),
        .channels = axp192_adc_channels,
        .adc_en1_mask = AXP192_ADC_EN1_MASK,
        .adc_en2_mask = AXP192_ADC_EN2_MASK,
        .adc_rate = axp20x_adc_rate,
        .maps = axp20x_maps,
};

static const struct axp_data axp20x_data = {
        .iio_info = &axp20x_adc_iio_info,
        .num_channels = ARRAY_SIZE(axp20x_adc_channels),
        .channels = axp20x_adc_channels,
        .adc_en1 = AXP20X_ADC_EN1,
        .adc_en1_mask = AXP20X_ADC_EN1_MASK,
        .adc_en2 = AXP20X_ADC_EN2,
        .adc_en2_mask = AXP20X_ADC_EN2_MASK,
        .adc_rate = axp20x_adc_rate,
        .maps = axp20x_maps,
};

static const struct axp_data axp22x_data = {
        .iio_info = &axp22x_adc_iio_info,
        .num_channels = ARRAY_SIZE(axp22x_adc_channels),
        .channels = axp22x_adc_channels,
        .adc_en1 = AXP20X_ADC_EN1,
        .adc_en1_mask = AXP22X_ADC_EN1_MASK,
        .adc_rate = axp22x_adc_rate,
        .maps = axp22x_maps,
};

static const struct axp_data axp717_data = {
        .iio_info = &axp717_adc_iio_info,
        .num_channels = ARRAY_SIZE(axp717_adc_channels),
        .channels = axp717_adc_channels,
        .adc_en1 = AXP717_ADC_CH_EN_CONTROL,
        .adc_en1_mask = AXP717_ADC_EN1_MASK,
        .maps = axp717_maps,
};

static const struct axp_data axp813_data = {
        .iio_info = &axp813_adc_iio_info,
        .num_channels = ARRAY_SIZE(axp813_adc_channels),
        .channels = axp813_adc_channels,
        .adc_en1 = AXP20X_ADC_EN1,
        .adc_en1_mask = AXP22X_ADC_EN1_MASK,
        .adc_rate = axp813_adc_rate,
        .maps = axp22x_maps,
};

static const struct of_device_id axp20x_adc_of_match[] = {
        { .compatible = "x-powers,axp192-adc", .data = (void *)&axp192_data, },
        { .compatible = "x-powers,axp209-adc", .data = (void *)&axp20x_data, },
        { .compatible = "x-powers,axp221-adc", .data = (void *)&axp22x_data, },
        { .compatible = "x-powers,axp717-adc", .data = (void *)&axp717_data, },
        { .compatible = "x-powers,axp813-adc", .data = (void *)&axp813_data, },
        { }
};
MODULE_DEVICE_TABLE(of, axp20x_adc_of_match);

static const struct platform_device_id axp20x_adc_id_match[] = {
        { .name = "axp192-adc", .driver_data = (kernel_ulong_t)&axp192_data, },
        { .name = "axp20x-adc", .driver_data = (kernel_ulong_t)&axp20x_data, },
        { .name = "axp22x-adc", .driver_data = (kernel_ulong_t)&axp22x_data, },
        { .name = "axp717-adc", .driver_data = (kernel_ulong_t)&axp717_data, },
        { .name = "axp813-adc", .driver_data = (kernel_ulong_t)&axp813_data, },
        { }
};
MODULE_DEVICE_TABLE(platform, axp20x_adc_id_match);

static int axp20x_probe(struct platform_device *pdev)
{
        struct axp20x_adc_iio *info;
        struct iio_dev *indio_dev;
        struct axp20x_dev *axp20x_dev;
        int ret;

        axp20x_dev = dev_get_drvdata(pdev->dev.parent);

        indio_dev = devm_iio_device_alloc(&pdev->dev, sizeof(*info));
        if (!indio_dev)
                return -ENOMEM;

        info = iio_priv(indio_dev);
        platform_set_drvdata(pdev, indio_dev);

        info->regmap = axp20x_dev->regmap;
        indio_dev->modes = INDIO_DIRECT_MODE;

        if (!dev_fwnode(&pdev->dev)) {
                const struct platform_device_id *id;

                id = platform_get_device_id(pdev);
                info->data = (const struct axp_data *)id->driver_data;
        } else {
                struct device *dev = &pdev->dev;

                info->data = device_get_match_data(dev);
        }

        indio_dev->name = platform_get_device_id(pdev)->name;
        indio_dev->info = info->data->iio_info;
        indio_dev->num_channels = info->data->num_channels;
        indio_dev->channels = info->data->channels;

        /* Enable the ADCs on IP */
        regmap_write(info->regmap, info->data->adc_en1,
                     info->data->adc_en1_mask);

        if (info->data->adc_en2_mask)
                regmap_set_bits(info->regmap, info->data->adc_en2,
                                info->data->adc_en2_mask);

        /* Configure ADCs rate */
        if (info->data->adc_rate)
                info->data->adc_rate(info, 100);

        ret = iio_map_array_register(indio_dev, info->data->maps);
        if (ret < 0) {
                dev_err(&pdev->dev, "failed to register IIO maps: %d\n", ret);
                goto fail_map;
        }

        ret = iio_device_register(indio_dev);
        if (ret < 0) {
                dev_err(&pdev->dev, "could not register the device\n");
                goto fail_register;
        }

        return 0;

fail_register:
        iio_map_array_unregister(indio_dev);

fail_map:
        regmap_write(info->regmap, info->data->adc_en1, 0);

        if (info->data->adc_en2_mask)
                regmap_write(info->regmap, info->data->adc_en2, 0);

        return ret;
}

static void axp20x_remove(struct platform_device *pdev)
{
        struct iio_dev *indio_dev = platform_get_drvdata(pdev);
        struct axp20x_adc_iio *info = iio_priv(indio_dev);

        iio_device_unregister(indio_dev);
        iio_map_array_unregister(indio_dev);

        regmap_write(info->regmap, info->data->adc_en1, 0);

        if (info->data->adc_en2_mask)
                regmap_write(info->regmap, info->data->adc_en2, 0);
}

static struct platform_driver axp20x_adc_driver = {
        .driver = {
                .name = "axp20x-adc",
                .of_match_table = axp20x_adc_of_match,
        },
        .id_table = axp20x_adc_id_match,
        .probe = axp20x_probe,
        .remove = axp20x_remove,
};

module_platform_driver(axp20x_adc_driver);

MODULE_DESCRIPTION("ADC driver for AXP20X and AXP22X PMICs");
MODULE_AUTHOR("Quentin Schulz <quentin.schulz@free-electrons.com>");
MODULE_LICENSE("GPL");