// SPDX-License-Identifier: GPL-2.0-only
/*
 * MFD core driver for the X-Powers' Power Management ICs
 *
 * AXP20x typically comprises an adaptive USB-Compatible PWM charger, BUCK DC-DC
 * converters, LDOs, multiple 12-bit ADCs of voltage, current and temperature
 * as well as configurable GPIOs.
 *
 * This file contains the interface independent core functions.
 *
 * Copyright (C) 2014 Carlo Caione
 *
 * Author: Carlo Caione <carlo@caione.org>
 */

#include <linux/acpi.h>
#include <linux/bitops.h>
#include <linux/delay.h>
#include <linux/err.h>
#include <linux/interrupt.h>
#include <linux/kernel.h>
#include <linux/mfd/axp20x.h>
#include <linux/mfd/core.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/property.h>
#include <linux/reboot.h>
#include <linux/regmap.h>
#include <linux/regulator/consumer.h>

#define AXP20X_OFF      BIT(7)

#define AXP806_REG_ADDR_EXT_ADDR_MASTER_MODE    0
#define AXP806_REG_ADDR_EXT_ADDR_SLAVE_MODE     BIT(4)

static const char * const axp20x_model_names[] = {
        [AXP152_ID] = "AXP152",
        [AXP192_ID] = "AXP192",
        [AXP202_ID] = "AXP202",
        [AXP209_ID] = "AXP209",
        [AXP221_ID] = "AXP221",
        [AXP223_ID] = "AXP223",
        [AXP288_ID] = "AXP288",
        [AXP313A_ID] = "AXP313a",
        [AXP323_ID] = "AXP323",
        [AXP717_ID] = "AXP717",
        [AXP803_ID] = "AXP803",
        [AXP806_ID] = "AXP806",
        [AXP809_ID] = "AXP809",
        [AXP813_ID] = "AXP813",
        [AXP15060_ID] = "AXP15060",
};

static const struct regmap_range axp152_writeable_ranges[] = {
        regmap_reg_range(AXP152_LDO3456_DC1234_CTRL, AXP152_IRQ3_STATE),
        regmap_reg_range(AXP152_DCDC_MODE, AXP152_PWM1_DUTY_CYCLE),
};

static const struct regmap_range axp152_volatile_ranges[] = {
        regmap_reg_range(AXP152_PWR_OP_MODE, AXP152_PWR_OP_MODE),
        regmap_reg_range(AXP152_IRQ1_EN, AXP152_IRQ3_STATE),
        regmap_reg_range(AXP152_GPIO_INPUT, AXP152_GPIO_INPUT),
};

static const struct regmap_access_table axp152_writeable_table = {
        .yes_ranges     = axp152_writeable_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp152_writeable_ranges),
};

static const struct regmap_access_table axp152_volatile_table = {
        .yes_ranges     = axp152_volatile_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp152_volatile_ranges),
};

static const struct regmap_range axp20x_writeable_ranges[] = {
        regmap_reg_range(AXP20X_DATACACHE(0), AXP20X_IRQ5_STATE),
        regmap_reg_range(AXP20X_CHRG_CTRL1, AXP20X_CHRG_CTRL2),
        regmap_reg_range(AXP20X_DCDC_MODE, AXP20X_FG_RES),
        regmap_reg_range(AXP20X_RDC_H, AXP20X_OCV(AXP20X_OCV_MAX)),
};

static const struct regmap_range axp20x_volatile_ranges[] = {
        regmap_reg_range(AXP20X_PWR_INPUT_STATUS, AXP20X_USB_OTG_STATUS),
        regmap_reg_range(AXP20X_CHRG_CTRL1, AXP20X_CHRG_CTRL2),
        regmap_reg_range(AXP20X_IRQ1_EN, AXP20X_IRQ5_STATE),
        regmap_reg_range(AXP20X_ACIN_V_ADC_H, AXP20X_IPSOUT_V_HIGH_L),
        regmap_reg_range(AXP20X_GPIO20_SS, AXP20X_GPIO3_CTRL),
        regmap_reg_range(AXP20X_FG_RES, AXP20X_RDC_L),
};

static const struct regmap_access_table axp20x_writeable_table = {
        .yes_ranges     = axp20x_writeable_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp20x_writeable_ranges),
};

static const struct regmap_access_table axp20x_volatile_table = {
        .yes_ranges     = axp20x_volatile_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp20x_volatile_ranges),
};

static const struct regmap_range axp192_writeable_ranges[] = {
        regmap_reg_range(AXP192_DATACACHE(0), AXP192_DATACACHE(5)),
        regmap_reg_range(AXP192_PWR_OUT_CTRL, AXP192_IRQ5_STATE),
        regmap_reg_range(AXP20X_DCDC_MODE, AXP192_N_RSTO_CTRL),
        regmap_reg_range(AXP20X_CC_CTRL, AXP20X_CC_CTRL),
};

static const struct regmap_range axp192_volatile_ranges[] = {
        regmap_reg_range(AXP20X_PWR_INPUT_STATUS, AXP192_USB_OTG_STATUS),
        regmap_reg_range(AXP192_IRQ1_STATE, AXP192_IRQ4_STATE),
        regmap_reg_range(AXP192_IRQ5_STATE, AXP192_IRQ5_STATE),
        regmap_reg_range(AXP20X_ACIN_V_ADC_H, AXP20X_IPSOUT_V_HIGH_L),
        regmap_reg_range(AXP20X_TIMER_CTRL, AXP20X_TIMER_CTRL),
        regmap_reg_range(AXP192_GPIO2_0_STATE, AXP192_GPIO2_0_STATE),
        regmap_reg_range(AXP192_GPIO4_3_STATE, AXP192_GPIO4_3_STATE),
        regmap_reg_range(AXP192_N_RSTO_CTRL, AXP192_N_RSTO_CTRL),
        regmap_reg_range(AXP20X_CHRG_CC_31_24, AXP20X_CC_CTRL),
};

static const struct regmap_access_table axp192_writeable_table = {
        .yes_ranges     = axp192_writeable_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp192_writeable_ranges),
};

static const struct regmap_access_table axp192_volatile_table = {
        .yes_ranges     = axp192_volatile_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp192_volatile_ranges),
};

/* AXP22x ranges are shared with the AXP809, as they cover the same range */
static const struct regmap_range axp22x_writeable_ranges[] = {
        regmap_reg_range(AXP20X_DATACACHE(0), AXP20X_IRQ5_STATE),
        regmap_reg_range(AXP20X_CHRG_CTRL1, AXP22X_CHRG_CTRL3),
        regmap_reg_range(AXP20X_DCDC_MODE, AXP22X_BATLOW_THRES1),
};

static const struct regmap_range axp22x_volatile_ranges[] = {
        regmap_reg_range(AXP20X_PWR_INPUT_STATUS, AXP20X_PWR_OP_MODE),
        regmap_reg_range(AXP20X_IRQ1_EN, AXP20X_IRQ5_STATE),
        regmap_reg_range(AXP22X_GPIO_STATE, AXP22X_GPIO_STATE),
        regmap_reg_range(AXP22X_PMIC_TEMP_H, AXP20X_IPSOUT_V_HIGH_L),
        regmap_reg_range(AXP20X_FG_RES, AXP20X_FG_RES),
};

static const struct regmap_access_table axp22x_writeable_table = {
        .yes_ranges     = axp22x_writeable_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp22x_writeable_ranges),
};

static const struct regmap_access_table axp22x_volatile_table = {
        .yes_ranges     = axp22x_volatile_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp22x_volatile_ranges),
};

/* AXP288 ranges are shared with the AXP803, as they cover the same range */
static const struct regmap_range axp288_writeable_ranges[] = {
        regmap_reg_range(AXP288_POWER_REASON, AXP288_POWER_REASON),
        regmap_reg_range(AXP20X_DATACACHE(0), AXP20X_IRQ6_STATE),
        regmap_reg_range(AXP20X_DCDC_MODE, AXP288_FG_TUNE5),
};

static const struct regmap_range axp288_volatile_ranges[] = {
        regmap_reg_range(AXP20X_PWR_INPUT_STATUS, AXP288_POWER_REASON),
        regmap_reg_range(AXP22X_PWR_OUT_CTRL1, AXP22X_ALDO3_V_OUT),
        regmap_reg_range(AXP288_BC_GLOBAL, AXP288_BC_GLOBAL),
        regmap_reg_range(AXP288_BC_DET_STAT, AXP20X_VBUS_IPSOUT_MGMT),
        regmap_reg_range(AXP20X_CHRG_BAK_CTRL, AXP20X_CHRG_BAK_CTRL),
        regmap_reg_range(AXP20X_IRQ1_EN, AXP20X_IPSOUT_V_HIGH_L),
        regmap_reg_range(AXP20X_TIMER_CTRL, AXP20X_TIMER_CTRL),
        regmap_reg_range(AXP20X_GPIO1_CTRL, AXP22X_GPIO_STATE),
        regmap_reg_range(AXP288_RT_BATT_V_H, AXP288_RT_BATT_V_L),
        regmap_reg_range(AXP20X_FG_RES, AXP288_FG_CC_CAP_REG),
};

static const struct regmap_access_table axp288_writeable_table = {
        .yes_ranges     = axp288_writeable_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp288_writeable_ranges),
};

static const struct regmap_access_table axp288_volatile_table = {
        .yes_ranges     = axp288_volatile_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp288_volatile_ranges),
};

static const struct regmap_range axp806_writeable_ranges[] = {
        regmap_reg_range(AXP20X_DATACACHE(0), AXP20X_DATACACHE(3)),
        regmap_reg_range(AXP806_PWR_OUT_CTRL1, AXP806_CLDO3_V_CTRL),
        regmap_reg_range(AXP20X_IRQ1_EN, AXP20X_IRQ2_EN),
        regmap_reg_range(AXP20X_IRQ1_STATE, AXP20X_IRQ2_STATE),
        regmap_reg_range(AXP806_REG_ADDR_EXT, AXP806_REG_ADDR_EXT),
};

static const struct regmap_range axp313a_writeable_ranges[] = {
        regmap_reg_range(AXP313A_ON_INDICATE, AXP313A_IRQ_STATE),
};

static const struct regmap_range axp323_writeable_ranges[] = {
        regmap_reg_range(AXP313A_ON_INDICATE, AXP323_DCDC_MODE_CTRL2),
};

static const struct regmap_range axp313a_volatile_ranges[] = {
        regmap_reg_range(AXP313A_SHUTDOWN_CTRL, AXP313A_SHUTDOWN_CTRL),
        regmap_reg_range(AXP313A_IRQ_STATE, AXP313A_IRQ_STATE),
};

static const struct regmap_access_table axp313a_writeable_table = {
        .yes_ranges = axp313a_writeable_ranges,
        .n_yes_ranges = ARRAY_SIZE(axp313a_writeable_ranges),
};

static const struct regmap_access_table axp323_writeable_table = {
        .yes_ranges = axp323_writeable_ranges,
        .n_yes_ranges = ARRAY_SIZE(axp323_writeable_ranges),
};

static const struct regmap_access_table axp313a_volatile_table = {
        .yes_ranges = axp313a_volatile_ranges,
        .n_yes_ranges = ARRAY_SIZE(axp313a_volatile_ranges),
};

static const struct regmap_range axp717_writeable_ranges[] = {
        regmap_reg_range(AXP717_PMU_FAULT, AXP717_MODULE_EN_CONTROL_1),
        regmap_reg_range(AXP717_MIN_SYS_V_CONTROL, AXP717_BOOST_CONTROL),
        regmap_reg_range(AXP717_VSYS_V_POWEROFF, AXP717_VSYS_V_POWEROFF),
        regmap_reg_range(AXP717_IRQ0_EN, AXP717_IRQ4_EN),
        regmap_reg_range(AXP717_IRQ0_STATE, AXP717_IRQ4_STATE),
        regmap_reg_range(AXP717_TS_PIN_CFG, AXP717_TS_PIN_CFG),
        regmap_reg_range(AXP717_ICC_CHG_SET, AXP717_CV_CHG_SET),
        regmap_reg_range(AXP717_DCDC_OUTPUT_CONTROL, AXP717_CPUSLDO_CONTROL),
        regmap_reg_range(AXP717_ADC_CH_EN_CONTROL, AXP717_ADC_CH_EN_CONTROL),
        regmap_reg_range(AXP717_ADC_DATA_SEL, AXP717_ADC_DATA_SEL),
        regmap_reg_range(AXP717_TYPEC_CC_AA_EN, AXP717_TYPEC_CC_AA_EN),
        regmap_reg_range(AXP717_TYPEC_CC_MODE_CONTROL, AXP717_TYPEC_CC_MODE_CONTROL),
};

static const struct regmap_range axp717_volatile_ranges[] = {
        regmap_reg_range(AXP717_ON_INDICATE, AXP717_PMU_FAULT),
        regmap_reg_range(AXP717_IRQ0_STATE, AXP717_IRQ4_STATE),
        regmap_reg_range(AXP717_BATT_PERCENT_DATA, AXP717_BATT_PERCENT_DATA),
        regmap_reg_range(AXP717_BATT_V_H, AXP717_BATT_CHRG_I_L),
        regmap_reg_range(AXP717_ADC_DATA_H, AXP717_ADC_DATA_L),
        regmap_reg_range(AXP717_TYPEC_CC_STATUS, AXP717_TYPEC_CC_STATUS),
};

static const struct regmap_access_table axp717_writeable_table = {
        .yes_ranges = axp717_writeable_ranges,
        .n_yes_ranges = ARRAY_SIZE(axp717_writeable_ranges),
};

static const struct regmap_access_table axp717_volatile_table = {
        .yes_ranges = axp717_volatile_ranges,
        .n_yes_ranges = ARRAY_SIZE(axp717_volatile_ranges),
};

static const struct regmap_range axp806_volatile_ranges[] = {
        regmap_reg_range(AXP20X_IRQ1_STATE, AXP20X_IRQ2_STATE),
};

static const struct regmap_access_table axp806_writeable_table = {
        .yes_ranges     = axp806_writeable_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp806_writeable_ranges),
};

static const struct regmap_access_table axp806_volatile_table = {
        .yes_ranges     = axp806_volatile_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp806_volatile_ranges),
};

static const struct regmap_range axp15060_writeable_ranges[] = {
        regmap_reg_range(AXP15060_PWR_OUT_CTRL1, AXP15060_DCDC_MODE_CTRL2),
        regmap_reg_range(AXP15060_OUTPUT_MONITOR_DISCHARGE, AXP15060_CPUSLDO_V_CTRL),
        regmap_reg_range(AXP15060_PWR_WAKEUP_CTRL, AXP15060_PWR_DISABLE_DOWN_SEQ),
        regmap_reg_range(AXP15060_PEK_KEY, AXP15060_PEK_KEY),
        regmap_reg_range(AXP15060_IRQ1_EN, AXP15060_IRQ2_EN),
        regmap_reg_range(AXP15060_IRQ1_STATE, AXP15060_IRQ2_STATE),
};

static const struct regmap_range axp15060_volatile_ranges[] = {
        regmap_reg_range(AXP15060_STARTUP_SRC, AXP15060_STARTUP_SRC),
        regmap_reg_range(AXP15060_PWR_WAKEUP_CTRL, AXP15060_PWR_DISABLE_DOWN_SEQ),
        regmap_reg_range(AXP15060_IRQ1_STATE, AXP15060_IRQ2_STATE),
};

static const struct regmap_access_table axp15060_writeable_table = {
        .yes_ranges     = axp15060_writeable_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp15060_writeable_ranges),
};

static const struct regmap_access_table axp15060_volatile_table = {
        .yes_ranges     = axp15060_volatile_ranges,
        .n_yes_ranges   = ARRAY_SIZE(axp15060_volatile_ranges),
};

static const struct resource axp152_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP152_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP152_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct resource axp192_ac_power_supply_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP192_IRQ_ACIN_PLUGIN, "ACIN_PLUGIN"),
        DEFINE_RES_IRQ_NAMED(AXP192_IRQ_ACIN_REMOVAL, "ACIN_REMOVAL"),
        DEFINE_RES_IRQ_NAMED(AXP192_IRQ_ACIN_OVER_V, "ACIN_OVER_V"),
};

static const struct resource axp192_usb_power_supply_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP192_IRQ_VBUS_PLUGIN, "VBUS_PLUGIN"),
        DEFINE_RES_IRQ_NAMED(AXP192_IRQ_VBUS_REMOVAL, "VBUS_REMOVAL"),
        DEFINE_RES_IRQ_NAMED(AXP192_IRQ_VBUS_VALID, "VBUS_VALID"),
        DEFINE_RES_IRQ_NAMED(AXP192_IRQ_VBUS_NOT_VALID, "VBUS_NOT_VALID"),
};

static const struct resource axp20x_ac_power_supply_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_ACIN_PLUGIN, "ACIN_PLUGIN"),
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_ACIN_REMOVAL, "ACIN_REMOVAL"),
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_ACIN_OVER_V, "ACIN_OVER_V"),
};

static const struct resource axp20x_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct resource axp20x_usb_power_supply_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_VBUS_PLUGIN, "VBUS_PLUGIN"),
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_VBUS_REMOVAL, "VBUS_REMOVAL"),
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_VBUS_VALID, "VBUS_VALID"),
        DEFINE_RES_IRQ_NAMED(AXP20X_IRQ_VBUS_NOT_VALID, "VBUS_NOT_VALID"),
};

static const struct resource axp22x_usb_power_supply_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP22X_IRQ_VBUS_PLUGIN, "VBUS_PLUGIN"),
        DEFINE_RES_IRQ_NAMED(AXP22X_IRQ_VBUS_REMOVAL, "VBUS_REMOVAL"),
};

static const struct resource axp717_usb_power_supply_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP717_IRQ_VBUS_OVER_V, "VBUS_OVER_V"),
        DEFINE_RES_IRQ_NAMED(AXP717_IRQ_VBUS_PLUGIN, "VBUS_PLUGIN"),
        DEFINE_RES_IRQ_NAMED(AXP717_IRQ_VBUS_REMOVAL, "VBUS_REMOVAL"),
};

/* AXP803 and AXP813/AXP818 share the same interrupts */
static const struct resource axp803_usb_power_supply_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP803_IRQ_VBUS_PLUGIN, "VBUS_PLUGIN"),
        DEFINE_RES_IRQ_NAMED(AXP803_IRQ_VBUS_REMOVAL, "VBUS_REMOVAL"),
};

static const struct resource axp22x_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP22X_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP22X_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct resource axp288_power_button_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP288_IRQ_POKP, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP288_IRQ_POKN, "PEK_DBF"),
};

static const struct resource axp288_fuel_gauge_resources[] = {
        DEFINE_RES_IRQ(AXP288_IRQ_QWBTU),
        DEFINE_RES_IRQ(AXP288_IRQ_WBTU),
        DEFINE_RES_IRQ(AXP288_IRQ_QWBTO),
        DEFINE_RES_IRQ(AXP288_IRQ_WBTO),
        DEFINE_RES_IRQ(AXP288_IRQ_WL2),
        DEFINE_RES_IRQ(AXP288_IRQ_WL1),
};

static const struct resource axp313a_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP313A_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP313A_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct resource axp717_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP717_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP717_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct resource axp803_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP803_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP803_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct resource axp806_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP806_IRQ_POK_RISE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP806_IRQ_POK_FALL, "PEK_DBF"),
};

static const struct resource axp809_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP809_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP809_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct resource axp15060_pek_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP15060_IRQ_PEK_RIS_EDGE, "PEK_DBR"),
        DEFINE_RES_IRQ_NAMED(AXP15060_IRQ_PEK_FAL_EDGE, "PEK_DBF"),
};

static const struct regmap_config axp152_regmap_config = {
        .reg_bits       = 8,
        .val_bits       = 8,
        .wr_table       = &axp152_writeable_table,
        .volatile_table = &axp152_volatile_table,
        .max_register   = AXP152_PWM1_DUTY_CYCLE,
        .cache_type     = REGCACHE_MAPLE,
};

static const struct regmap_config axp192_regmap_config = {
        .reg_bits       = 8,
        .val_bits       = 8,
        .wr_table       = &axp192_writeable_table,
        .volatile_table = &axp192_volatile_table,
        .max_register   = AXP20X_CC_CTRL,
        .cache_type     = REGCACHE_MAPLE,
};

static const struct regmap_config axp20x_regmap_config = {
        .reg_bits       = 8,
        .val_bits       = 8,
        .wr_table       = &axp20x_writeable_table,
        .volatile_table = &axp20x_volatile_table,
        .max_register   = AXP20X_OCV(AXP20X_OCV_MAX),
        .cache_type     = REGCACHE_MAPLE,
};

static const struct regmap_config axp22x_regmap_config = {
        .reg_bits       = 8,
        .val_bits       = 8,
        .wr_table       = &axp22x_writeable_table,
        .volatile_table = &axp22x_volatile_table,
        .max_register   = AXP22X_BATLOW_THRES1,
        .cache_type     = REGCACHE_MAPLE,
};

static const struct regmap_config axp288_regmap_config = {
        .reg_bits       = 8,
        .val_bits       = 8,
        .wr_table       = &axp288_writeable_table,
        .volatile_table = &axp288_volatile_table,
        .max_register   = AXP288_FG_TUNE5,
        .cache_type     = REGCACHE_MAPLE,
};

static const struct regmap_config axp313a_regmap_config = {
        .reg_bits = 8,
        .val_bits = 8,
        .wr_table = &axp313a_writeable_table,
        .volatile_table = &axp313a_volatile_table,
        .max_register = AXP313A_IRQ_STATE,
        .cache_type = REGCACHE_MAPLE,
};

static const struct regmap_config axp323_regmap_config = {
        .reg_bits = 8,
        .val_bits = 8,
        .wr_table = &axp323_writeable_table,
        .volatile_table = &axp313a_volatile_table,
        .max_register = AXP323_DCDC_MODE_CTRL2,
        .cache_type = REGCACHE_MAPLE,
};

static const struct regmap_config axp717_regmap_config = {
        .reg_bits = 8,
        .val_bits = 8,
        .wr_table = &axp717_writeable_table,
        .volatile_table = &axp717_volatile_table,
        .max_register = AXP717_TYPEC_CC_STATUS,
        .cache_type = REGCACHE_MAPLE,
};

static const struct regmap_config axp806_regmap_config = {
        .reg_bits       = 8,
        .val_bits       = 8,
        .wr_table       = &axp806_writeable_table,
        .volatile_table = &axp806_volatile_table,
        .max_register   = AXP806_REG_ADDR_EXT,
        .cache_type     = REGCACHE_MAPLE,
};

static const struct regmap_config axp15060_regmap_config = {
        .reg_bits       = 8,
        .val_bits       = 8,
        .wr_table       = &axp15060_writeable_table,
        .volatile_table = &axp15060_volatile_table,
        .max_register   = AXP15060_IRQ2_STATE,
        .cache_type     = REGCACHE_MAPLE,
};

#define INIT_REGMAP_IRQ(_variant, _irq, _off, _mask)                    \
        [_variant##_IRQ_##_irq] = { .reg_offset = (_off), .mask = BIT(_mask) }

static const struct regmap_irq axp152_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP152, LDO0IN_CONNECT,         0, 6),
        INIT_REGMAP_IRQ(AXP152, LDO0IN_REMOVAL,         0, 5),
        INIT_REGMAP_IRQ(AXP152, ALDO0IN_CONNECT,        0, 3),
        INIT_REGMAP_IRQ(AXP152, ALDO0IN_REMOVAL,        0, 2),
        INIT_REGMAP_IRQ(AXP152, DCDC1_V_LOW,            1, 5),
        INIT_REGMAP_IRQ(AXP152, DCDC2_V_LOW,            1, 4),
        INIT_REGMAP_IRQ(AXP152, DCDC3_V_LOW,            1, 3),
        INIT_REGMAP_IRQ(AXP152, DCDC4_V_LOW,            1, 2),
        INIT_REGMAP_IRQ(AXP152, PEK_SHORT,              1, 1),
        INIT_REGMAP_IRQ(AXP152, PEK_LONG,               1, 0),
        INIT_REGMAP_IRQ(AXP152, TIMER,                  2, 7),
        INIT_REGMAP_IRQ(AXP152, PEK_RIS_EDGE,           2, 6),
        INIT_REGMAP_IRQ(AXP152, PEK_FAL_EDGE,           2, 5),
        INIT_REGMAP_IRQ(AXP152, GPIO3_INPUT,            2, 3),
        INIT_REGMAP_IRQ(AXP152, GPIO2_INPUT,            2, 2),
        INIT_REGMAP_IRQ(AXP152, GPIO1_INPUT,            2, 1),
        INIT_REGMAP_IRQ(AXP152, GPIO0_INPUT,            2, 0),
};

static const struct regmap_irq axp192_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP192, ACIN_OVER_V,            0, 7),
        INIT_REGMAP_IRQ(AXP192, ACIN_PLUGIN,            0, 6),
        INIT_REGMAP_IRQ(AXP192, ACIN_REMOVAL,           0, 5),
        INIT_REGMAP_IRQ(AXP192, VBUS_OVER_V,            0, 4),
        INIT_REGMAP_IRQ(AXP192, VBUS_PLUGIN,            0, 3),
        INIT_REGMAP_IRQ(AXP192, VBUS_REMOVAL,           0, 2),
        INIT_REGMAP_IRQ(AXP192, VBUS_V_LOW,             0, 1),
        INIT_REGMAP_IRQ(AXP192, BATT_PLUGIN,            1, 7),
        INIT_REGMAP_IRQ(AXP192, BATT_REMOVAL,           1, 6),
        INIT_REGMAP_IRQ(AXP192, BATT_ENT_ACT_MODE,      1, 5),
        INIT_REGMAP_IRQ(AXP192, BATT_EXIT_ACT_MODE,     1, 4),
        INIT_REGMAP_IRQ(AXP192, CHARG,                  1, 3),
        INIT_REGMAP_IRQ(AXP192, CHARG_DONE,             1, 2),
        INIT_REGMAP_IRQ(AXP192, BATT_TEMP_HIGH,         1, 1),
        INIT_REGMAP_IRQ(AXP192, BATT_TEMP_LOW,          1, 0),
        INIT_REGMAP_IRQ(AXP192, DIE_TEMP_HIGH,          2, 7),
        INIT_REGMAP_IRQ(AXP192, CHARG_I_LOW,            2, 6),
        INIT_REGMAP_IRQ(AXP192, DCDC1_V_LONG,           2, 5),
        INIT_REGMAP_IRQ(AXP192, DCDC2_V_LONG,           2, 4),
        INIT_REGMAP_IRQ(AXP192, DCDC3_V_LONG,           2, 3),
        INIT_REGMAP_IRQ(AXP192, PEK_SHORT,              2, 1),
        INIT_REGMAP_IRQ(AXP192, PEK_LONG,               2, 0),
        INIT_REGMAP_IRQ(AXP192, N_OE_PWR_ON,            3, 7),
        INIT_REGMAP_IRQ(AXP192, N_OE_PWR_OFF,           3, 6),
        INIT_REGMAP_IRQ(AXP192, VBUS_VALID,             3, 5),
        INIT_REGMAP_IRQ(AXP192, VBUS_NOT_VALID,         3, 4),
        INIT_REGMAP_IRQ(AXP192, VBUS_SESS_VALID,        3, 3),
        INIT_REGMAP_IRQ(AXP192, VBUS_SESS_END,          3, 2),
        INIT_REGMAP_IRQ(AXP192, LOW_PWR_LVL,            3, 0),
        INIT_REGMAP_IRQ(AXP192, TIMER,                  4, 7),
        INIT_REGMAP_IRQ(AXP192, GPIO2_INPUT,            4, 2),
        INIT_REGMAP_IRQ(AXP192, GPIO1_INPUT,            4, 1),
        INIT_REGMAP_IRQ(AXP192, GPIO0_INPUT,            4, 0),
};

static const struct regmap_irq axp20x_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP20X, ACIN_OVER_V,            0, 7),
        INIT_REGMAP_IRQ(AXP20X, ACIN_PLUGIN,            0, 6),
        INIT_REGMAP_IRQ(AXP20X, ACIN_REMOVAL,           0, 5),
        INIT_REGMAP_IRQ(AXP20X, VBUS_OVER_V,            0, 4),
        INIT_REGMAP_IRQ(AXP20X, VBUS_PLUGIN,            0, 3),
        INIT_REGMAP_IRQ(AXP20X, VBUS_REMOVAL,           0, 2),
        INIT_REGMAP_IRQ(AXP20X, VBUS_V_LOW,             0, 1),
        INIT_REGMAP_IRQ(AXP20X, BATT_PLUGIN,            1, 7),
        INIT_REGMAP_IRQ(AXP20X, BATT_REMOVAL,           1, 6),
        INIT_REGMAP_IRQ(AXP20X, BATT_ENT_ACT_MODE,      1, 5),
        INIT_REGMAP_IRQ(AXP20X, BATT_EXIT_ACT_MODE,     1, 4),
        INIT_REGMAP_IRQ(AXP20X, CHARG,                  1, 3),
        INIT_REGMAP_IRQ(AXP20X, CHARG_DONE,             1, 2),
        INIT_REGMAP_IRQ(AXP20X, BATT_TEMP_HIGH,         1, 1),
        INIT_REGMAP_IRQ(AXP20X, BATT_TEMP_LOW,          1, 0),
        INIT_REGMAP_IRQ(AXP20X, DIE_TEMP_HIGH,          2, 7),
        INIT_REGMAP_IRQ(AXP20X, CHARG_I_LOW,            2, 6),
        INIT_REGMAP_IRQ(AXP20X, DCDC1_V_LONG,           2, 5),
        INIT_REGMAP_IRQ(AXP20X, DCDC2_V_LONG,           2, 4),
        INIT_REGMAP_IRQ(AXP20X, DCDC3_V_LONG,           2, 3),
        INIT_REGMAP_IRQ(AXP20X, PEK_SHORT,              2, 1),
        INIT_REGMAP_IRQ(AXP20X, PEK_LONG,               2, 0),
        INIT_REGMAP_IRQ(AXP20X, N_OE_PWR_ON,            3, 7),
        INIT_REGMAP_IRQ(AXP20X, N_OE_PWR_OFF,           3, 6),
        INIT_REGMAP_IRQ(AXP20X, VBUS_VALID,             3, 5),
        INIT_REGMAP_IRQ(AXP20X, VBUS_NOT_VALID,         3, 4),
        INIT_REGMAP_IRQ(AXP20X, VBUS_SESS_VALID,        3, 3),
        INIT_REGMAP_IRQ(AXP20X, VBUS_SESS_END,          3, 2),
        INIT_REGMAP_IRQ(AXP20X, LOW_PWR_LVL1,           3, 1),
        INIT_REGMAP_IRQ(AXP20X, LOW_PWR_LVL2,           3, 0),
        INIT_REGMAP_IRQ(AXP20X, TIMER,                  4, 7),
        INIT_REGMAP_IRQ(AXP20X, PEK_RIS_EDGE,           4, 6),
        INIT_REGMAP_IRQ(AXP20X, PEK_FAL_EDGE,           4, 5),
        INIT_REGMAP_IRQ(AXP20X, GPIO3_INPUT,            4, 3),
        INIT_REGMAP_IRQ(AXP20X, GPIO2_INPUT,            4, 2),
        INIT_REGMAP_IRQ(AXP20X, GPIO1_INPUT,            4, 1),
        INIT_REGMAP_IRQ(AXP20X, GPIO0_INPUT,            4, 0),
};

static const struct regmap_irq axp22x_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP22X, ACIN_OVER_V,            0, 7),
        INIT_REGMAP_IRQ(AXP22X, ACIN_PLUGIN,            0, 6),
        INIT_REGMAP_IRQ(AXP22X, ACIN_REMOVAL,           0, 5),
        INIT_REGMAP_IRQ(AXP22X, VBUS_OVER_V,            0, 4),
        INIT_REGMAP_IRQ(AXP22X, VBUS_PLUGIN,            0, 3),
        INIT_REGMAP_IRQ(AXP22X, VBUS_REMOVAL,           0, 2),
        INIT_REGMAP_IRQ(AXP22X, VBUS_V_LOW,             0, 1),
        INIT_REGMAP_IRQ(AXP22X, BATT_PLUGIN,            1, 7),
        INIT_REGMAP_IRQ(AXP22X, BATT_REMOVAL,           1, 6),
        INIT_REGMAP_IRQ(AXP22X, BATT_ENT_ACT_MODE,      1, 5),
        INIT_REGMAP_IRQ(AXP22X, BATT_EXIT_ACT_MODE,     1, 4),
        INIT_REGMAP_IRQ(AXP22X, CHARG,                  1, 3),
        INIT_REGMAP_IRQ(AXP22X, CHARG_DONE,             1, 2),
        INIT_REGMAP_IRQ(AXP22X, BATT_TEMP_HIGH,         1, 1),
        INIT_REGMAP_IRQ(AXP22X, BATT_TEMP_LOW,          1, 0),
        INIT_REGMAP_IRQ(AXP22X, DIE_TEMP_HIGH,          2, 7),
        INIT_REGMAP_IRQ(AXP22X, PEK_SHORT,              2, 1),
        INIT_REGMAP_IRQ(AXP22X, PEK_LONG,               2, 0),
        INIT_REGMAP_IRQ(AXP22X, LOW_PWR_LVL1,           3, 1),
        INIT_REGMAP_IRQ(AXP22X, LOW_PWR_LVL2,           3, 0),
        INIT_REGMAP_IRQ(AXP22X, TIMER,                  4, 7),
        INIT_REGMAP_IRQ(AXP22X, PEK_RIS_EDGE,           4, 6),
        INIT_REGMAP_IRQ(AXP22X, PEK_FAL_EDGE,           4, 5),
        INIT_REGMAP_IRQ(AXP22X, GPIO1_INPUT,            4, 1),
        INIT_REGMAP_IRQ(AXP22X, GPIO0_INPUT,            4, 0),
};

/* some IRQs are compatible with axp20x models */
static const struct regmap_irq axp288_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP288, VBUS_FALL,              0, 2),
        INIT_REGMAP_IRQ(AXP288, VBUS_RISE,              0, 3),
        INIT_REGMAP_IRQ(AXP288, OV,                     0, 4),
        INIT_REGMAP_IRQ(AXP288, FALLING_ALT,            0, 5),
        INIT_REGMAP_IRQ(AXP288, RISING_ALT,             0, 6),
        INIT_REGMAP_IRQ(AXP288, OV_ALT,                 0, 7),

        INIT_REGMAP_IRQ(AXP288, DONE,                   1, 2),
        INIT_REGMAP_IRQ(AXP288, CHARGING,               1, 3),
        INIT_REGMAP_IRQ(AXP288, SAFE_QUIT,              1, 4),
        INIT_REGMAP_IRQ(AXP288, SAFE_ENTER,             1, 5),
        INIT_REGMAP_IRQ(AXP288, ABSENT,                 1, 6),
        INIT_REGMAP_IRQ(AXP288, APPEND,                 1, 7),

        INIT_REGMAP_IRQ(AXP288, QWBTU,                  2, 0),
        INIT_REGMAP_IRQ(AXP288, WBTU,                   2, 1),
        INIT_REGMAP_IRQ(AXP288, QWBTO,                  2, 2),
        INIT_REGMAP_IRQ(AXP288, WBTO,                   2, 3),
        INIT_REGMAP_IRQ(AXP288, QCBTU,                  2, 4),
        INIT_REGMAP_IRQ(AXP288, CBTU,                   2, 5),
        INIT_REGMAP_IRQ(AXP288, QCBTO,                  2, 6),
        INIT_REGMAP_IRQ(AXP288, CBTO,                   2, 7),

        INIT_REGMAP_IRQ(AXP288, WL2,                    3, 0),
        INIT_REGMAP_IRQ(AXP288, WL1,                    3, 1),
        INIT_REGMAP_IRQ(AXP288, GPADC,                  3, 2),
        INIT_REGMAP_IRQ(AXP288, OT,                     3, 7),

        INIT_REGMAP_IRQ(AXP288, GPIO0,                  4, 0),
        INIT_REGMAP_IRQ(AXP288, GPIO1,                  4, 1),
        INIT_REGMAP_IRQ(AXP288, POKO,                   4, 2),
        INIT_REGMAP_IRQ(AXP288, POKL,                   4, 3),
        INIT_REGMAP_IRQ(AXP288, POKS,                   4, 4),
        INIT_REGMAP_IRQ(AXP288, POKN,                   4, 5),
        INIT_REGMAP_IRQ(AXP288, POKP,                   4, 6),
        INIT_REGMAP_IRQ(AXP288, TIMER,                  4, 7),

        INIT_REGMAP_IRQ(AXP288, MV_CHNG,                5, 0),
        INIT_REGMAP_IRQ(AXP288, BC_USB_CHNG,            5, 1),
};

static const struct regmap_irq axp313a_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP313A, PEK_RIS_EDGE,          0, 7),
        INIT_REGMAP_IRQ(AXP313A, PEK_FAL_EDGE,          0, 6),
        INIT_REGMAP_IRQ(AXP313A, PEK_SHORT,             0, 5),
        INIT_REGMAP_IRQ(AXP313A, PEK_LONG,              0, 4),
        INIT_REGMAP_IRQ(AXP313A, DCDC3_V_LOW,           0, 3),
        INIT_REGMAP_IRQ(AXP313A, DCDC2_V_LOW,           0, 2),
        INIT_REGMAP_IRQ(AXP313A, DIE_TEMP_HIGH,         0, 0),
};

static const struct regmap_irq axp717_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP717, SOC_DROP_LVL2,          0, 7),
        INIT_REGMAP_IRQ(AXP717, SOC_DROP_LVL1,          0, 6),
        INIT_REGMAP_IRQ(AXP717, GAUGE_NEW_SOC,          0, 4),
        INIT_REGMAP_IRQ(AXP717, BOOST_OVER_V,           0, 2),
        INIT_REGMAP_IRQ(AXP717, VBUS_OVER_V,            0, 1),
        INIT_REGMAP_IRQ(AXP717, VBUS_FAULT,             0, 0),
        INIT_REGMAP_IRQ(AXP717, VBUS_PLUGIN,            1, 7),
        INIT_REGMAP_IRQ(AXP717, VBUS_REMOVAL,           1, 6),
        INIT_REGMAP_IRQ(AXP717, BATT_PLUGIN,            1, 5),
        INIT_REGMAP_IRQ(AXP717, BATT_REMOVAL,           1, 4),
        INIT_REGMAP_IRQ(AXP717, PEK_SHORT,              1, 3),
        INIT_REGMAP_IRQ(AXP717, PEK_LONG,               1, 2),
        INIT_REGMAP_IRQ(AXP717, PEK_FAL_EDGE,           1, 1),
        INIT_REGMAP_IRQ(AXP717, PEK_RIS_EDGE,           1, 0),
        INIT_REGMAP_IRQ(AXP717, WDOG_EXPIRE,            2, 7),
        INIT_REGMAP_IRQ(AXP717, LDO_OVER_CURR,          2, 6),
        INIT_REGMAP_IRQ(AXP717, BATT_OVER_CURR,         2, 5),
        INIT_REGMAP_IRQ(AXP717, CHARG_DONE,             2, 4),
        INIT_REGMAP_IRQ(AXP717, CHARG,                  2, 3),
        INIT_REGMAP_IRQ(AXP717, DIE_TEMP_HIGH,          2, 2),
        INIT_REGMAP_IRQ(AXP717, CHARG_TIMER,            2, 1),
        INIT_REGMAP_IRQ(AXP717, BATT_OVER_V,            2, 0),
        INIT_REGMAP_IRQ(AXP717, BC_USB_DONE,            3, 7),
        INIT_REGMAP_IRQ(AXP717, BC_USB_CHNG,            3, 6),
        INIT_REGMAP_IRQ(AXP717, BATT_QUIT_TEMP_HIGH,    3, 4),
        INIT_REGMAP_IRQ(AXP717, BATT_CHG_TEMP_HIGH,     3, 3),
        INIT_REGMAP_IRQ(AXP717, BATT_CHG_TEMP_LOW,      3, 2),
        INIT_REGMAP_IRQ(AXP717, BATT_ACT_TEMP_HIGH,     3, 1),
        INIT_REGMAP_IRQ(AXP717, BATT_ACT_TEMP_LOW,      3, 0),
        INIT_REGMAP_IRQ(AXP717, TYPEC_REMOVE,           4, 6),
        INIT_REGMAP_IRQ(AXP717, TYPEC_PLUGIN,           4, 5),
};

static const struct regmap_irq axp803_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP803, ACIN_OVER_V,            0, 7),
        INIT_REGMAP_IRQ(AXP803, ACIN_PLUGIN,            0, 6),
        INIT_REGMAP_IRQ(AXP803, ACIN_REMOVAL,           0, 5),
        INIT_REGMAP_IRQ(AXP803, VBUS_OVER_V,            0, 4),
        INIT_REGMAP_IRQ(AXP803, VBUS_PLUGIN,            0, 3),
        INIT_REGMAP_IRQ(AXP803, VBUS_REMOVAL,           0, 2),
        INIT_REGMAP_IRQ(AXP803, BATT_PLUGIN,            1, 7),
        INIT_REGMAP_IRQ(AXP803, BATT_REMOVAL,           1, 6),
        INIT_REGMAP_IRQ(AXP803, BATT_ENT_ACT_MODE,      1, 5),
        INIT_REGMAP_IRQ(AXP803, BATT_EXIT_ACT_MODE,     1, 4),
        INIT_REGMAP_IRQ(AXP803, CHARG,                  1, 3),
        INIT_REGMAP_IRQ(AXP803, CHARG_DONE,             1, 2),
        INIT_REGMAP_IRQ(AXP803, BATT_CHG_TEMP_HIGH,     2, 7),
        INIT_REGMAP_IRQ(AXP803, BATT_CHG_TEMP_HIGH_END, 2, 6),
        INIT_REGMAP_IRQ(AXP803, BATT_CHG_TEMP_LOW,      2, 5),
        INIT_REGMAP_IRQ(AXP803, BATT_CHG_TEMP_LOW_END,  2, 4),
        INIT_REGMAP_IRQ(AXP803, BATT_ACT_TEMP_HIGH,     2, 3),
        INIT_REGMAP_IRQ(AXP803, BATT_ACT_TEMP_HIGH_END, 2, 2),
        INIT_REGMAP_IRQ(AXP803, BATT_ACT_TEMP_LOW,      2, 1),
        INIT_REGMAP_IRQ(AXP803, BATT_ACT_TEMP_LOW_END,  2, 0),
        INIT_REGMAP_IRQ(AXP803, DIE_TEMP_HIGH,          3, 7),
        INIT_REGMAP_IRQ(AXP803, GPADC,                  3, 2),
        INIT_REGMAP_IRQ(AXP803, LOW_PWR_LVL1,           3, 1),
        INIT_REGMAP_IRQ(AXP803, LOW_PWR_LVL2,           3, 0),
        INIT_REGMAP_IRQ(AXP803, TIMER,                  4, 7),
        INIT_REGMAP_IRQ(AXP803, PEK_RIS_EDGE,           4, 6),
        INIT_REGMAP_IRQ(AXP803, PEK_FAL_EDGE,           4, 5),
        INIT_REGMAP_IRQ(AXP803, PEK_SHORT,              4, 4),
        INIT_REGMAP_IRQ(AXP803, PEK_LONG,               4, 3),
        INIT_REGMAP_IRQ(AXP803, PEK_OVER_OFF,           4, 2),
        INIT_REGMAP_IRQ(AXP803, GPIO1_INPUT,            4, 1),
        INIT_REGMAP_IRQ(AXP803, GPIO0_INPUT,            4, 0),
        INIT_REGMAP_IRQ(AXP803, BC_USB_CHNG,            5, 1),
        INIT_REGMAP_IRQ(AXP803, MV_CHNG,                5, 0),
};

static const struct regmap_irq axp806_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP806, DIE_TEMP_HIGH_LV1,      0, 0),
        INIT_REGMAP_IRQ(AXP806, DIE_TEMP_HIGH_LV2,      0, 1),
        INIT_REGMAP_IRQ(AXP806, DCDCA_V_LOW,            0, 3),
        INIT_REGMAP_IRQ(AXP806, DCDCB_V_LOW,            0, 4),
        INIT_REGMAP_IRQ(AXP806, DCDCC_V_LOW,            0, 5),
        INIT_REGMAP_IRQ(AXP806, DCDCD_V_LOW,            0, 6),
        INIT_REGMAP_IRQ(AXP806, DCDCE_V_LOW,            0, 7),
        INIT_REGMAP_IRQ(AXP806, POK_LONG,               1, 0),
        INIT_REGMAP_IRQ(AXP806, POK_SHORT,              1, 1),
        INIT_REGMAP_IRQ(AXP806, WAKEUP,                 1, 4),
        INIT_REGMAP_IRQ(AXP806, POK_FALL,               1, 5),
        INIT_REGMAP_IRQ(AXP806, POK_RISE,               1, 6),
};

static const struct regmap_irq axp809_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP809, ACIN_OVER_V,            0, 7),
        INIT_REGMAP_IRQ(AXP809, ACIN_PLUGIN,            0, 6),
        INIT_REGMAP_IRQ(AXP809, ACIN_REMOVAL,           0, 5),
        INIT_REGMAP_IRQ(AXP809, VBUS_OVER_V,            0, 4),
        INIT_REGMAP_IRQ(AXP809, VBUS_PLUGIN,            0, 3),
        INIT_REGMAP_IRQ(AXP809, VBUS_REMOVAL,           0, 2),
        INIT_REGMAP_IRQ(AXP809, VBUS_V_LOW,             0, 1),
        INIT_REGMAP_IRQ(AXP809, BATT_PLUGIN,            1, 7),
        INIT_REGMAP_IRQ(AXP809, BATT_REMOVAL,           1, 6),
        INIT_REGMAP_IRQ(AXP809, BATT_ENT_ACT_MODE,      1, 5),
        INIT_REGMAP_IRQ(AXP809, BATT_EXIT_ACT_MODE,     1, 4),
        INIT_REGMAP_IRQ(AXP809, CHARG,                  1, 3),
        INIT_REGMAP_IRQ(AXP809, CHARG_DONE,             1, 2),
        INIT_REGMAP_IRQ(AXP809, BATT_CHG_TEMP_HIGH,     2, 7),
        INIT_REGMAP_IRQ(AXP809, BATT_CHG_TEMP_HIGH_END, 2, 6),
        INIT_REGMAP_IRQ(AXP809, BATT_CHG_TEMP_LOW,      2, 5),
        INIT_REGMAP_IRQ(AXP809, BATT_CHG_TEMP_LOW_END,  2, 4),
        INIT_REGMAP_IRQ(AXP809, BATT_ACT_TEMP_HIGH,     2, 3),
        INIT_REGMAP_IRQ(AXP809, BATT_ACT_TEMP_HIGH_END, 2, 2),
        INIT_REGMAP_IRQ(AXP809, BATT_ACT_TEMP_LOW,      2, 1),
        INIT_REGMAP_IRQ(AXP809, BATT_ACT_TEMP_LOW_END,  2, 0),
        INIT_REGMAP_IRQ(AXP809, DIE_TEMP_HIGH,          3, 7),
        INIT_REGMAP_IRQ(AXP809, LOW_PWR_LVL1,           3, 1),
        INIT_REGMAP_IRQ(AXP809, LOW_PWR_LVL2,           3, 0),
        INIT_REGMAP_IRQ(AXP809, TIMER,                  4, 7),
        INIT_REGMAP_IRQ(AXP809, PEK_RIS_EDGE,           4, 6),
        INIT_REGMAP_IRQ(AXP809, PEK_FAL_EDGE,           4, 5),
        INIT_REGMAP_IRQ(AXP809, PEK_SHORT,              4, 4),
        INIT_REGMAP_IRQ(AXP809, PEK_LONG,               4, 3),
        INIT_REGMAP_IRQ(AXP809, PEK_OVER_OFF,           4, 2),
        INIT_REGMAP_IRQ(AXP809, GPIO1_INPUT,            4, 1),
        INIT_REGMAP_IRQ(AXP809, GPIO0_INPUT,            4, 0),
};

static const struct regmap_irq axp15060_regmap_irqs[] = {
        INIT_REGMAP_IRQ(AXP15060, DIE_TEMP_HIGH_LV1,    0, 0),
        INIT_REGMAP_IRQ(AXP15060, DIE_TEMP_HIGH_LV2,    0, 1),
        INIT_REGMAP_IRQ(AXP15060, DCDC1_V_LOW,          0, 2),
        INIT_REGMAP_IRQ(AXP15060, DCDC2_V_LOW,          0, 3),
        INIT_REGMAP_IRQ(AXP15060, DCDC3_V_LOW,          0, 4),
        INIT_REGMAP_IRQ(AXP15060, DCDC4_V_LOW,          0, 5),
        INIT_REGMAP_IRQ(AXP15060, DCDC5_V_LOW,          0, 6),
        INIT_REGMAP_IRQ(AXP15060, DCDC6_V_LOW,          0, 7),
        INIT_REGMAP_IRQ(AXP15060, PEK_LONG,                     1, 0),
        INIT_REGMAP_IRQ(AXP15060, PEK_SHORT,                    1, 1),
        INIT_REGMAP_IRQ(AXP15060, GPIO1_INPUT,          1, 2),
        INIT_REGMAP_IRQ(AXP15060, PEK_FAL_EDGE,                 1, 3),
        INIT_REGMAP_IRQ(AXP15060, PEK_RIS_EDGE,                 1, 4),
        INIT_REGMAP_IRQ(AXP15060, GPIO2_INPUT,          1, 5),
};

static const struct regmap_irq_chip axp152_regmap_irq_chip = {
        .name                   = "axp152_irq_chip",
        .status_base            = AXP152_IRQ1_STATE,
        .ack_base               = AXP152_IRQ1_STATE,
        .unmask_base            = AXP152_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp152_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp152_regmap_irqs),
        .num_regs               = 3,
};

static unsigned int axp192_get_irq_reg(struct regmap_irq_chip_data *data,
                                       unsigned int base, int index)
{
        /* linear mapping for IRQ1 to IRQ4 */
        if (index < 4)
                return base + index;

        /* handle IRQ5 separately */
        if (base == AXP192_IRQ1_EN)
                return AXP192_IRQ5_EN;

        return AXP192_IRQ5_STATE;
}

static const struct regmap_irq_chip axp192_regmap_irq_chip = {
        .name                   = "axp192_irq_chip",
        .status_base            = AXP192_IRQ1_STATE,
        .ack_base               = AXP192_IRQ1_STATE,
        .unmask_base            = AXP192_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp192_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp192_regmap_irqs),
        .num_regs               = 5,
        .get_irq_reg            = axp192_get_irq_reg,
};

static const struct regmap_irq_chip axp20x_regmap_irq_chip = {
        .name                   = "axp20x_irq_chip",
        .status_base            = AXP20X_IRQ1_STATE,
        .ack_base               = AXP20X_IRQ1_STATE,
        .unmask_base            = AXP20X_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp20x_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp20x_regmap_irqs),
        .num_regs               = 5,

};

static const struct regmap_irq_chip axp22x_regmap_irq_chip = {
        .name                   = "axp22x_irq_chip",
        .status_base            = AXP20X_IRQ1_STATE,
        .ack_base               = AXP20X_IRQ1_STATE,
        .unmask_base            = AXP20X_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp22x_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp22x_regmap_irqs),
        .num_regs               = 5,
};

static const struct regmap_irq_chip axp288_regmap_irq_chip = {
        .name                   = "axp288_irq_chip",
        .status_base            = AXP20X_IRQ1_STATE,
        .ack_base               = AXP20X_IRQ1_STATE,
        .unmask_base            = AXP20X_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp288_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp288_regmap_irqs),
        .num_regs               = 6,

};

static const struct regmap_irq_chip axp313a_regmap_irq_chip = {
        .name                   = "axp313a_irq_chip",
        .status_base            = AXP313A_IRQ_STATE,
        .ack_base               = AXP313A_IRQ_STATE,
        .unmask_base            = AXP313A_IRQ_EN,
        .init_ack_masked        = true,
        .irqs                   = axp313a_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp313a_regmap_irqs),
        .num_regs               = 1,
};

static const struct regmap_irq_chip axp717_regmap_irq_chip = {
        .name                   = "axp717_irq_chip",
        .status_base            = AXP717_IRQ0_STATE,
        .ack_base               = AXP717_IRQ0_STATE,
        .unmask_base            = AXP717_IRQ0_EN,
        .init_ack_masked        = true,
        .irqs                   = axp717_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp717_regmap_irqs),
        .num_regs               = 5,
};

static const struct regmap_irq_chip axp803_regmap_irq_chip = {
        .name                   = "axp803",
        .status_base            = AXP20X_IRQ1_STATE,
        .ack_base               = AXP20X_IRQ1_STATE,
        .unmask_base            = AXP20X_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp803_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp803_regmap_irqs),
        .num_regs               = 6,
};

static const struct regmap_irq_chip axp806_regmap_irq_chip = {
        .name                   = "axp806",
        .status_base            = AXP20X_IRQ1_STATE,
        .ack_base               = AXP20X_IRQ1_STATE,
        .unmask_base            = AXP20X_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp806_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp806_regmap_irqs),
        .num_regs               = 2,
};

static const struct regmap_irq_chip axp809_regmap_irq_chip = {
        .name                   = "axp809",
        .status_base            = AXP20X_IRQ1_STATE,
        .ack_base               = AXP20X_IRQ1_STATE,
        .unmask_base            = AXP20X_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp809_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp809_regmap_irqs),
        .num_regs               = 5,
};

static const struct regmap_irq_chip axp15060_regmap_irq_chip = {
        .name                   = "axp15060",
        .status_base            = AXP15060_IRQ1_STATE,
        .ack_base               = AXP15060_IRQ1_STATE,
        .unmask_base            = AXP15060_IRQ1_EN,
        .init_ack_masked        = true,
        .irqs                   = axp15060_regmap_irqs,
        .num_irqs               = ARRAY_SIZE(axp15060_regmap_irqs),
        .num_regs               = 2,
};

static const struct mfd_cell axp192_cells[] = {
        {
                .name           = "axp192-adc",
                .of_compatible  = "x-powers,axp192-adc",
        }, {
                .name           = "axp20x-battery-power-supply",
                .of_compatible  = "x-powers,axp192-battery-power-supply",
        }, {
                .name           = "axp20x-ac-power-supply",
                .of_compatible  = "x-powers,axp202-ac-power-supply",
                .num_resources  = ARRAY_SIZE(axp192_ac_power_supply_resources),
                .resources      = axp192_ac_power_supply_resources,
        }, {
                .name           = "axp20x-usb-power-supply",
                .of_compatible  = "x-powers,axp192-usb-power-supply",
                .num_resources  = ARRAY_SIZE(axp192_usb_power_supply_resources),
                .resources      = axp192_usb_power_supply_resources,
        },
        {       .name           = "axp20x-regulator" },
};

static const struct mfd_cell axp20x_cells[] = {
        {
                .name           = "axp20x-gpio",
                .of_compatible  = "x-powers,axp209-gpio",
        }, {
                .name           = "axp20x-pek",
                .num_resources  = ARRAY_SIZE(axp20x_pek_resources),
                .resources      = axp20x_pek_resources,
        }, {
                .name           = "axp20x-regulator",
        }, {
                .name           = "axp20x-adc",
                .of_compatible  = "x-powers,axp209-adc",
        }, {
                .name           = "axp20x-battery-power-supply",
                .of_compatible  = "x-powers,axp209-battery-power-supply",
        }, {
                .name           = "axp20x-ac-power-supply",
                .of_compatible  = "x-powers,axp202-ac-power-supply",
                .num_resources  = ARRAY_SIZE(axp20x_ac_power_supply_resources),
                .resources      = axp20x_ac_power_supply_resources,
        }, {
                .name           = "axp20x-usb-power-supply",
                .of_compatible  = "x-powers,axp202-usb-power-supply",
                .num_resources  = ARRAY_SIZE(axp20x_usb_power_supply_resources),
                .resources      = axp20x_usb_power_supply_resources,
        },
};

static const struct mfd_cell axp221_cells[] = {
        {
                .name           = "axp20x-gpio",
                .of_compatible  = "x-powers,axp221-gpio",
        }, {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp22x_pek_resources),
                .resources      = axp22x_pek_resources,
        }, {
                .name           = "axp20x-regulator",
        }, {
                .name           = "axp22x-adc",
                .of_compatible  = "x-powers,axp221-adc",
        }, {
                .name           = "axp20x-ac-power-supply",
                .of_compatible  = "x-powers,axp221-ac-power-supply",
                .num_resources  = ARRAY_SIZE(axp20x_ac_power_supply_resources),
                .resources      = axp20x_ac_power_supply_resources,
        }, {
                .name           = "axp20x-battery-power-supply",
                .of_compatible  = "x-powers,axp221-battery-power-supply",
        }, {
                .name           = "axp20x-usb-power-supply",
                .of_compatible  = "x-powers,axp221-usb-power-supply",
                .num_resources  = ARRAY_SIZE(axp22x_usb_power_supply_resources),
                .resources      = axp22x_usb_power_supply_resources,
        },
};

static const struct mfd_cell axp223_cells[] = {
        {
                .name           = "axp20x-gpio",
                .of_compatible  = "x-powers,axp221-gpio",
        }, {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp22x_pek_resources),
                .resources      = axp22x_pek_resources,
        }, {
                .name           = "axp22x-adc",
                .of_compatible  = "x-powers,axp221-adc",
        }, {
                .name           = "axp20x-battery-power-supply",
                .of_compatible  = "x-powers,axp221-battery-power-supply",
        }, {
                .name           = "axp20x-regulator",
        }, {
                .name           = "axp20x-ac-power-supply",
                .of_compatible  = "x-powers,axp221-ac-power-supply",
                .num_resources  = ARRAY_SIZE(axp20x_ac_power_supply_resources),
                .resources      = axp20x_ac_power_supply_resources,
        }, {
                .name           = "axp20x-usb-power-supply",
                .of_compatible  = "x-powers,axp223-usb-power-supply",
                .num_resources  = ARRAY_SIZE(axp22x_usb_power_supply_resources),
                .resources      = axp22x_usb_power_supply_resources,
        },
};

static const struct mfd_cell axp152_cells[] = {
        {
                .name           = "axp20x-pek",
                .num_resources  = ARRAY_SIZE(axp152_pek_resources),
                .resources      = axp152_pek_resources,
        },
};

static struct mfd_cell axp313a_cells[] = {
        /* AXP323 is sometimes paired with AXP717 as sub-PMIC */
        MFD_CELL_BASIC("axp20x-regulator", NULL, NULL, 0, 1),
        MFD_CELL_RES("axp313a-pek", axp313a_pek_resources),
};

static struct mfd_cell axp717_cells[] = {
        MFD_CELL_NAME("axp20x-regulator"),
        MFD_CELL_RES("axp20x-pek", axp717_pek_resources),
        MFD_CELL_OF("axp717-adc",
                    NULL, NULL, 0, 0, "x-powers,axp717-adc"),
        MFD_CELL_OF("axp20x-usb-power-supply",
                    axp717_usb_power_supply_resources, NULL, 0, 0,
                    "x-powers,axp717-usb-power-supply"),
        MFD_CELL_OF("axp20x-battery-power-supply",
                    NULL, NULL, 0, 0, "x-powers,axp717-battery-power-supply"),
};

static const struct resource axp288_adc_resources[] = {
        DEFINE_RES_IRQ_NAMED(AXP288_IRQ_GPADC, "GPADC"),
};

static const struct resource axp288_extcon_resources[] = {
        DEFINE_RES_IRQ(AXP288_IRQ_VBUS_FALL),
        DEFINE_RES_IRQ(AXP288_IRQ_VBUS_RISE),
        DEFINE_RES_IRQ(AXP288_IRQ_MV_CHNG),
        DEFINE_RES_IRQ(AXP288_IRQ_BC_USB_CHNG),
};

static const struct resource axp288_charger_resources[] = {
        DEFINE_RES_IRQ(AXP288_IRQ_OV),
        DEFINE_RES_IRQ(AXP288_IRQ_DONE),
        DEFINE_RES_IRQ(AXP288_IRQ_CHARGING),
        DEFINE_RES_IRQ(AXP288_IRQ_SAFE_QUIT),
        DEFINE_RES_IRQ(AXP288_IRQ_SAFE_ENTER),
        DEFINE_RES_IRQ(AXP288_IRQ_QCBTU),
        DEFINE_RES_IRQ(AXP288_IRQ_CBTU),
        DEFINE_RES_IRQ(AXP288_IRQ_QCBTO),
        DEFINE_RES_IRQ(AXP288_IRQ_CBTO),
};

static const char * const axp288_fuel_gauge_suppliers[] = { "axp288_charger" };

static const struct property_entry axp288_fuel_gauge_properties[] = {
        PROPERTY_ENTRY_STRING_ARRAY("supplied-from", axp288_fuel_gauge_suppliers),
        { }
};

static const struct software_node axp288_fuel_gauge_sw_node = {
        .name = "axp288_fuel_gauge",
        .properties = axp288_fuel_gauge_properties,
};

static const struct mfd_cell axp288_cells[] = {
        {
                .name           = "axp288_adc",
                .num_resources  = ARRAY_SIZE(axp288_adc_resources),
                .resources      = axp288_adc_resources,
        }, {
                .name           = "axp288_extcon",
                .num_resources  = ARRAY_SIZE(axp288_extcon_resources),
                .resources      = axp288_extcon_resources,
        }, {
                .name           = "axp288_charger",
                .num_resources  = ARRAY_SIZE(axp288_charger_resources),
                .resources      = axp288_charger_resources,
        }, {
                .name           = "axp288_fuel_gauge",
                .num_resources  = ARRAY_SIZE(axp288_fuel_gauge_resources),
                .resources      = axp288_fuel_gauge_resources,
                .swnode         = &axp288_fuel_gauge_sw_node,
        }, {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp288_power_button_resources),
                .resources      = axp288_power_button_resources,
        }, {
                .name           = "axp288_pmic_acpi",
        },
};

static const struct mfd_cell axp803_cells[] = {
        {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp803_pek_resources),
                .resources      = axp803_pek_resources,
        }, {
                .name           = "axp20x-gpio",
                .of_compatible  = "x-powers,axp813-gpio",
        }, {
                .name           = "axp813-adc",
                .of_compatible  = "x-powers,axp813-adc",
        }, {
                .name           = "axp20x-battery-power-supply",
                .of_compatible  = "x-powers,axp813-battery-power-supply",
        }, {
                .name           = "axp20x-ac-power-supply",
                .of_compatible  = "x-powers,axp813-ac-power-supply",
                .num_resources  = ARRAY_SIZE(axp20x_ac_power_supply_resources),
                .resources      = axp20x_ac_power_supply_resources,
        }, {
                .name           = "axp20x-usb-power-supply",
                .num_resources  = ARRAY_SIZE(axp803_usb_power_supply_resources),
                .resources      = axp803_usb_power_supply_resources,
                .of_compatible  = "x-powers,axp813-usb-power-supply",
        },
        {       .name           = "axp20x-regulator" },
};

static const struct mfd_cell axp806_self_working_cells[] = {
        {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp806_pek_resources),
                .resources      = axp806_pek_resources,
        },
        {       .name           = "axp20x-regulator" },
};

static const struct mfd_cell axp806_cells[] = {
        {
                .id             = 2,
                .name           = "axp20x-regulator",
        },
};

static const struct mfd_cell axp809_cells[] = {
        {
                .name           = "axp20x-gpio",
                .of_compatible  = "x-powers,axp221-gpio",
        }, {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp809_pek_resources),
                .resources      = axp809_pek_resources,
        }, {
                .id             = 1,
                .name           = "axp20x-regulator",
        },
};

static const struct mfd_cell axp813_cells[] = {
        {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp803_pek_resources),
                .resources      = axp803_pek_resources,
        }, {
                .name           = "axp20x-regulator",
        }, {
                .name           = "axp20x-gpio",
                .of_compatible  = "x-powers,axp813-gpio",
        }, {
                .name           = "axp813-adc",
                .of_compatible  = "x-powers,axp813-adc",
        }, {
                .name           = "axp20x-battery-power-supply",
                .of_compatible  = "x-powers,axp813-battery-power-supply",
        }, {
                .name           = "axp20x-ac-power-supply",
                .of_compatible  = "x-powers,axp813-ac-power-supply",
                .num_resources  = ARRAY_SIZE(axp20x_ac_power_supply_resources),
                .resources      = axp20x_ac_power_supply_resources,
        }, {
                .name           = "axp20x-usb-power-supply",
                .num_resources  = ARRAY_SIZE(axp803_usb_power_supply_resources),
                .resources      = axp803_usb_power_supply_resources,
                .of_compatible  = "x-powers,axp813-usb-power-supply",
        },
};

static const struct mfd_cell axp15060_cells[] = {
        {
                .name           = "axp221-pek",
                .num_resources  = ARRAY_SIZE(axp15060_pek_resources),
                .resources      = axp15060_pek_resources,
        }, {
                .name           = "axp20x-regulator",
        },
};

/* For boards that don't have IRQ line connected to SOC. */
static const struct mfd_cell axp_regulator_only_cells[] = {
        /* PMIC without IRQ line may be secondary PMIC */
        MFD_CELL_BASIC("axp20x-regulator", NULL, NULL, 0, 1),
};

static int axp20x_power_off(struct sys_off_data *data)
{
        struct axp20x_dev *axp20x = data->cb_data;
        unsigned int shutdown_reg;

        switch (axp20x->variant) {
        case AXP323_ID:
        case AXP313A_ID:
                shutdown_reg = AXP313A_SHUTDOWN_CTRL;
                break;
        default:
                shutdown_reg = AXP20X_OFF_CTRL;
                break;
        }

        regmap_write(axp20x->regmap, shutdown_reg, AXP20X_OFF);

        /* Give capacitors etc. time to drain to avoid kernel panic msg. */
        mdelay(500);

        return NOTIFY_DONE;
}

int axp20x_match_device(struct axp20x_dev *axp20x)
{
        struct device *dev = axp20x->dev;
        const struct mfd_cell *cells_no_irq = NULL;
        int nr_cells_no_irq = 0;

        axp20x->variant = (long)device_get_match_data(dev);
        switch (axp20x->variant) {
        case AXP152_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp152_cells);
                axp20x->cells = axp152_cells;
                axp20x->regmap_cfg = &axp152_regmap_config;
                axp20x->regmap_irq_chip = &axp152_regmap_irq_chip;
                break;
        case AXP192_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp192_cells);
                axp20x->cells = axp192_cells;
                axp20x->regmap_cfg = &axp192_regmap_config;
                axp20x->regmap_irq_chip = &axp192_regmap_irq_chip;
                break;
        case AXP202_ID:
        case AXP209_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp20x_cells);
                axp20x->cells = axp20x_cells;
                axp20x->regmap_cfg = &axp20x_regmap_config;
                axp20x->regmap_irq_chip = &axp20x_regmap_irq_chip;
                break;
        case AXP221_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp221_cells);
                axp20x->cells = axp221_cells;
                axp20x->regmap_cfg = &axp22x_regmap_config;
                axp20x->regmap_irq_chip = &axp22x_regmap_irq_chip;
                break;
        case AXP223_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp223_cells);
                axp20x->cells = axp223_cells;
                axp20x->regmap_cfg = &axp22x_regmap_config;
                axp20x->regmap_irq_chip = &axp22x_regmap_irq_chip;
                break;
        case AXP288_ID:
                axp20x->cells = axp288_cells;
                axp20x->nr_cells = ARRAY_SIZE(axp288_cells);
                axp20x->regmap_cfg = &axp288_regmap_config;
                axp20x->regmap_irq_chip = &axp288_regmap_irq_chip;
                axp20x->irq_flags = IRQF_TRIGGER_LOW;
                break;
        case AXP313A_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp313a_cells);
                axp20x->cells = axp313a_cells;
                axp20x->regmap_cfg = &axp313a_regmap_config;
                axp20x->regmap_irq_chip = &axp313a_regmap_irq_chip;
                break;
        case AXP323_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp313a_cells);
                axp20x->cells = axp313a_cells;
                axp20x->regmap_cfg = &axp323_regmap_config;
                axp20x->regmap_irq_chip = &axp313a_regmap_irq_chip;
                break;
        case AXP717_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp717_cells);
                axp20x->cells = axp717_cells;
                axp20x->regmap_cfg = &axp717_regmap_config;
                axp20x->regmap_irq_chip = &axp717_regmap_irq_chip;
                break;
        case AXP803_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp803_cells);
                axp20x->cells = axp803_cells;
                axp20x->regmap_cfg = &axp288_regmap_config;
                axp20x->regmap_irq_chip = &axp803_regmap_irq_chip;
                break;
        case AXP806_ID:
                /*
                 * Don't register the power key part if in slave mode or
                 * if there is no interrupt line.
                 */
                if (of_property_read_bool(axp20x->dev->of_node,
                                          "x-powers,self-working-mode")) {
                        axp20x->nr_cells = ARRAY_SIZE(axp806_self_working_cells);
                        axp20x->cells = axp806_self_working_cells;
                } else {
                        axp20x->nr_cells = ARRAY_SIZE(axp806_cells);
                        axp20x->cells = axp806_cells;
                }
                nr_cells_no_irq = ARRAY_SIZE(axp806_cells);
                cells_no_irq = axp806_cells;
                axp20x->regmap_cfg = &axp806_regmap_config;
                axp20x->regmap_irq_chip = &axp806_regmap_irq_chip;
                break;
        case AXP809_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp809_cells);
                axp20x->cells = axp809_cells;
                axp20x->regmap_cfg = &axp22x_regmap_config;
                axp20x->regmap_irq_chip = &axp809_regmap_irq_chip;
                break;
        case AXP813_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp813_cells);
                axp20x->cells = axp813_cells;
                axp20x->regmap_cfg = &axp288_regmap_config;
                /*
                 * The IRQ table given in the datasheet is incorrect.
                 * In IRQ enable/status registers 1, there are separate
                 * IRQs for ACIN and VBUS, instead of bits [7:5] being
                 * the same as bits [4:2]. So it shares the same IRQs
                 * as the AXP803, rather than the AXP288.
                 */
                axp20x->regmap_irq_chip = &axp803_regmap_irq_chip;
                break;
        case AXP15060_ID:
                axp20x->nr_cells = ARRAY_SIZE(axp15060_cells);
                axp20x->cells = axp15060_cells;
                axp20x->regmap_cfg = &axp15060_regmap_config;
                axp20x->regmap_irq_chip = &axp15060_regmap_irq_chip;
                break;
        default:
                dev_err(dev, "unsupported AXP20X ID %u\n", axp20x->variant);
                return -EINVAL;
        }

        /*
         * Use an alternative cell array when no interrupt line is connected,
         * since IRQs are required by some drivers.
         * The default is the safe "regulator-only", as this works fine without
         * an interrupt specified.
         */
        if (axp20x->irq <= 0) {
                if (cells_no_irq) {
                        axp20x->nr_cells = nr_cells_no_irq;
                        axp20x->cells = cells_no_irq;
                } else {
                        axp20x->nr_cells = ARRAY_SIZE(axp_regulator_only_cells);
                        axp20x->cells = axp_regulator_only_cells;
                }
        }

        dev_info(dev, "AXP20x variant %s found\n",
                 axp20x_model_names[axp20x->variant]);

        return 0;
}
EXPORT_SYMBOL(axp20x_match_device);

int axp20x_device_probe(struct axp20x_dev *axp20x)
{
        int ret;

        /*
         * The AXP806 supports either master/standalone or slave mode.
         * Slave mode allows sharing the serial bus, even with multiple
         * AXP806 which all have the same hardware address.
         *
         * This is done with extra "serial interface address extension",
         * or AXP806_BUS_ADDR_EXT, and "register address extension", or
         * AXP806_REG_ADDR_EXT, registers. The former is read-only, with
         * 1 bit customizable at the factory, and 1 bit depending on the
         * state of an external pin. The latter is writable. The device
         * will only respond to operations to its other registers when
         * the these device addressing bits (in the upper 4 bits of the
         * registers) match.
         *
         * By default we support an AXP806 chained to an AXP809 in slave
         * mode. Boards which use an AXP806 in master mode can set the
         * property "x-powers,master-mode" to override the default.
         */
        if (axp20x->variant == AXP806_ID) {
                if (of_property_read_bool(axp20x->dev->of_node,
                                          "x-powers,master-mode") ||
                    of_property_read_bool(axp20x->dev->of_node,
                                          "x-powers,self-working-mode"))
                        regmap_write(axp20x->regmap, AXP806_REG_ADDR_EXT,
                                     AXP806_REG_ADDR_EXT_ADDR_MASTER_MODE);
                else
                        regmap_write(axp20x->regmap, AXP806_REG_ADDR_EXT,
                                     AXP806_REG_ADDR_EXT_ADDR_SLAVE_MODE);
        }

        /* Only if there is an interrupt line connected towards the CPU. */
        if (axp20x->irq > 0) {
                ret = regmap_add_irq_chip(axp20x->regmap, axp20x->irq,
                                IRQF_ONESHOT | IRQF_SHARED | axp20x->irq_flags,
                                -1, axp20x->regmap_irq_chip,
                                &axp20x->regmap_irqc);
                if (ret) {
                        dev_err(axp20x->dev, "failed to add irq chip: %d\n",
                                ret);
                        return ret;
                }
        }

        ret = mfd_add_devices(axp20x->dev, PLATFORM_DEVID_NONE, axp20x->cells,
                              axp20x->nr_cells, NULL, 0, NULL);

        if (ret) {
                dev_err(axp20x->dev, "failed to add MFD devices: %d\n", ret);
                regmap_del_irq_chip(axp20x->irq, axp20x->regmap_irqc);
                return ret;
        }

        if (axp20x->variant != AXP288_ID)
                devm_register_power_off_handler(axp20x->dev, axp20x_power_off, axp20x);

        dev_info(axp20x->dev, "AXP20X driver loaded\n");

        return 0;
}
EXPORT_SYMBOL(axp20x_device_probe);

void axp20x_device_remove(struct axp20x_dev *axp20x)
{
        mfd_remove_devices(axp20x->dev);
        regmap_del_irq_chip(axp20x->irq, axp20x->regmap_irqc);
}
EXPORT_SYMBOL(axp20x_device_remove);

MODULE_DESCRIPTION("PMIC MFD core driver for AXP20X");
MODULE_AUTHOR("Carlo Caione <carlo@caione.org>");
MODULE_LICENSE("GPL");