// SPDX-License-Identifier: GPL-2.0-only
/*
 * axp288_fuel_gauge.c - Xpower AXP288 PMIC Fuel Gauge Driver
 *
 * Copyright (C) 2020-2021 Andrejus Basovas <xxx@yyy.tld>
 * Copyright (C) 2016-2021 Hans de Goede <hdegoede@redhat.com>
 * Copyright (C) 2014 Intel Corporation
 *
 * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 */

#include <linux/acpi.h>
#include <linux/dmi.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/device.h>
#include <linux/regmap.h>
#include <linux/jiffies.h>
#include <linux/interrupt.h>
#include <linux/mfd/axp20x.h>
#include <linux/platform_device.h>
#include <linux/power_supply.h>
#include <linux/iio/consumer.h>
#include <linux/unaligned.h>
#include <asm/iosf_mbi.h>

#define PS_STAT_VBUS_TRIGGER   (1 << 0)
#define PS_STAT_BAT_CHRG_DIR   (1 << 2)
#define PS_STAT_VBAT_ABOVE_VHOLD  (1 << 3)
#define PS_STAT_VBUS_VALID   (1 << 4)
#define PS_STAT_VBUS_PRESENT   (1 << 5)

#define CHRG_STAT_BAT_SAFE_MODE   (1 << 3)
#define CHRG_STAT_BAT_VALID   (1 << 4)
#define CHRG_STAT_BAT_PRESENT   (1 << 5)
#define CHRG_STAT_CHARGING   (1 << 6)
#define CHRG_STAT_PMIC_OTP   (1 << 7)

#define CHRG_CCCV_CC_MASK   0xf     /* 4 bits */
#define CHRG_CCCV_CC_BIT_POS   0
#define CHRG_CCCV_CC_OFFSET   200     /* 200mA */
#define CHRG_CCCV_CC_LSB_RES   200     /* 200mA */
#define CHRG_CCCV_ITERM_20P   (1 << 4)    /* 20% of CC */
#define CHRG_CCCV_CV_MASK   0x60        /* 2 bits */
#define CHRG_CCCV_CV_BIT_POS   5
#define CHRG_CCCV_CV_4100MV   0x0     /* 4.10V */
#define CHRG_CCCV_CV_4150MV   0x1     /* 4.15V */
#define CHRG_CCCV_CV_4200MV   0x2     /* 4.20V */
#define CHRG_CCCV_CV_4350MV   0x3     /* 4.35V */
#define CHRG_CCCV_CHG_EN   (1 << 7)

#define FG_CNTL_OCV_ADJ_STAT   (1 << 2)
#define FG_CNTL_OCV_ADJ_EN   (1 << 3)
#define FG_CNTL_CAP_ADJ_STAT   (1 << 4)
#define FG_CNTL_CAP_ADJ_EN   (1 << 5)
#define FG_CNTL_CC_EN    (1 << 6)
#define FG_CNTL_GAUGE_EN   (1 << 7)

#define FG_15BIT_WORD_VALID   (1 << 15)
#define FG_15BIT_VAL_MASK   0x7fff

#define FG_REP_CAP_VALID   (1 << 7)
#define FG_REP_CAP_VAL_MASK   0x7F

#define FG_DES_CAP1_VALID   (1 << 7)
#define FG_DES_CAP_RES_LSB   1456    /* 1.456mAhr */

#define FG_DES_CC_RES_LSB   1456    /* 1.456mAhr */

#define FG_OCV_CAP_VALID   (1 << 7)
#define FG_OCV_CAP_VAL_MASK   0x7F
#define FG_CC_CAP_VALID    (1 << 7)
#define FG_CC_CAP_VAL_MASK   0x7F

#define FG_LOW_CAP_THR1_MASK   0xf0    /* 5% tp 20% */
#define FG_LOW_CAP_THR1_VAL   0xa0    /* 15 perc */
#define FG_LOW_CAP_THR2_MASK   0x0f    /* 0% to 15% */
#define FG_LOW_CAP_WARN_THR   14  /* 14 perc */
#define FG_LOW_CAP_CRIT_THR   4   /* 4 perc */
#define FG_LOW_CAP_SHDN_THR   0   /* 0 perc */

#define DEV_NAME    "axp288_fuel_gauge"

/* 1.1mV per LSB expressed in uV */
#define VOLTAGE_FROM_ADC(a)   ((a * 11) / 10)
/* properties converted to uV, uA */
#define PROP_VOLT(a)    ((a) * 1000)
#define PROP_CURR(a)    ((a) * 1000)

#define AXP288_REG_UPDATE_INTERVAL  (60 * HZ)
#define AXP288_FG_INTR_NUM   6

#define AXP288_QUIRK_NO_BATTERY   BIT(0)

static bool no_current_sense_res;
module_param(no_current_sense_res, bool, 0444);
MODULE_PARM_DESC(no_current_sense_res, "No (or broken) current sense resistor");

enum {
 QWBTU_IRQ = 0,
 WBTU_IRQ,
 QWBTO_IRQ,
 WBTO_IRQ,
 WL2_IRQ,
 WL1_IRQ,
};

enum {
 BAT_CHRG_CURR,
 BAT_D_CURR,
 BAT_VOLT,
 IIO_CHANNEL_NUM
};

struct axp288_fg_info {
 struct device *dev;
 struct regmap *regmap;
 int irq[AXP288_FG_INTR_NUM];
 struct iio_channel *iio_channel[IIO_CHANNEL_NUM];
 struct power_supply *bat;
 struct mutex lock;
 int status;
 int max_volt;
 int pwr_op;
 int low_cap;
 struct dentry *debug_file;

 char valid;                 /* zero until following fields are valid */
 unsigned long last_updated; /* in jiffies */

 int pwr_stat;
 int fg_res;
 int bat_volt;
 int d_curr;
 int c_curr;
 int ocv;
 int fg_cc_mtr1;
 int fg_des_cap1;
};

static enum power_supply_property fuel_gauge_props[] = {
 POWER_SUPPLY_PROP_STATUS,
 POWER_SUPPLY_PROP_PRESENT,
 POWER_SUPPLY_PROP_HEALTH,
 POWER_SUPPLY_PROP_VOLTAGE_MAX_DESIGN,
 POWER_SUPPLY_PROP_VOLTAGE_NOW,
 POWER_SUPPLY_PROP_VOLTAGE_OCV,
 POWER_SUPPLY_PROP_CAPACITY,
 POWER_SUPPLY_PROP_CAPACITY_ALERT_MIN,
 POWER_SUPPLY_PROP_TECHNOLOGY,
 /* The 3 props below are not used when no_current_sense_res is set */
 POWER_SUPPLY_PROP_CHARGE_FULL,
 POWER_SUPPLY_PROP_CHARGE_NOW,
 POWER_SUPPLY_PROP_CURRENT_NOW,
};

static int fuel_gauge_reg_readb(struct axp288_fg_info *info, int reg)
{
 unsigned int val;
 int ret;

 ret = regmap_read(info->regmap, reg, &val);
 if (ret < 0) {
  dev_err(info->dev, "Error reading reg 0x%02x err: %d\n", reg, ret);
  return ret;
 }

 return val;
}

static int fuel_gauge_reg_writeb(struct axp288_fg_info *info, int reg, u8 val)
{
 int ret;

 ret = regmap_write(info->regmap, reg, (unsigned int)val);

 if (ret < 0)
  dev_err(info->dev, "Error writing reg 0x%02x err: %d\n", reg, ret);

 return ret;
}

static int fuel_gauge_read_15bit_word(struct axp288_fg_info *info, int reg)
{
 unsigned char buf[2];
 int ret;

 ret = regmap_bulk_read(info->regmap, reg, buf, 2);
 if (ret < 0) {
  dev_err(info->dev, "Error reading reg 0x%02x err: %d\n", reg, ret);
  return ret;
 }

 ret = get_unaligned_be16(buf);
 if (!(ret & FG_15BIT_WORD_VALID)) {
  dev_err(info->dev, "Error reg 0x%02x contents not valid\n", reg);
  return -ENXIO;
 }

 return ret & FG_15BIT_VAL_MASK;
}

static int fuel_gauge_read_12bit_word(struct axp288_fg_info *info, int reg)
{
 unsigned char buf[2];
 int ret;

 ret = regmap_bulk_read(info->regmap, reg, buf, 2);
 if (ret < 0) {
  dev_err(info->dev, "Error reading reg 0x%02x err: %d\n", reg, ret);
  return ret;
 }

 /* 12-bit data values have upper 8 bits in buf[0], lower 4 in buf[1] */
 return (buf[0] << 4) | ((buf[1] >> 4) & 0x0f);
}

static int fuel_gauge_update_registers(struct axp288_fg_info *info)
{
 int ret;

 if (info->valid && time_before(jiffies, info->last_updated + AXP288_REG_UPDATE_INTERVAL))
  return 0;

 dev_dbg(info->dev, "Fuel Gauge updating register values...\n");

 ret = iosf_mbi_block_punit_i2c_access();
 if (ret < 0)
  return ret;

 ret = fuel_gauge_reg_readb(info, AXP20X_PWR_INPUT_STATUS);
 if (ret < 0)
  goto out;
 info->pwr_stat = ret;

 if (no_current_sense_res)
  ret = fuel_gauge_reg_readb(info, AXP288_FG_OCV_CAP_REG);
 else
  ret = fuel_gauge_reg_readb(info, AXP20X_FG_RES);
 if (ret < 0)
  goto out;
 info->fg_res = ret;

 ret = iio_read_channel_raw(info->iio_channel[BAT_VOLT], &info->bat_volt);
 if (ret < 0)
  goto out;

 ret = fuel_gauge_read_12bit_word(info, AXP288_FG_OCVH_REG);
 if (ret < 0)
  goto out;
 info->ocv = ret;

 if (no_current_sense_res)
  goto out_no_current_sense_res;

 if (info->pwr_stat & PS_STAT_BAT_CHRG_DIR) {
  info->d_curr = 0;
  ret = iio_read_channel_raw(info->iio_channel[BAT_CHRG_CURR], &info->c_curr);
  if (ret < 0)
   goto out;
 } else {
  info->c_curr = 0;
  ret = iio_read_channel_raw(info->iio_channel[BAT_D_CURR], &info->d_curr);
  if (ret < 0)
   goto out;
 }

 ret = fuel_gauge_read_15bit_word(info, AXP288_FG_CC_MTR1_REG);
 if (ret < 0)
  goto out;
 info->fg_cc_mtr1 = ret;

 ret = fuel_gauge_read_15bit_word(info, AXP288_FG_DES_CAP1_REG);
 if (ret < 0)
  goto out;
 info->fg_des_cap1 = ret;

out_no_current_sense_res:
 info->last_updated = jiffies;
 info->valid = 1;
 ret = 0;
out:
 iosf_mbi_unblock_punit_i2c_access();
 return ret;
}

static void fuel_gauge_get_status(struct axp288_fg_info *info)
{
 int pwr_stat = info->pwr_stat;
 int fg_res = info->fg_res;
 int curr = info->d_curr;

 /* Report full if Vbus is valid and the reported capacity is 100% */
 if (!(pwr_stat & PS_STAT_VBUS_VALID))
  goto not_full;

 if (!(fg_res & FG_REP_CAP_VALID))
  goto not_full;

 fg_res &= ~FG_REP_CAP_VALID;
 if (fg_res == 100) {
  info->status = POWER_SUPPLY_STATUS_FULL;
  return;
 }

 /*
  * Sometimes the charger turns itself off before fg-res reaches 100%.
  * When this happens the AXP288 reports a not-charging status and
  * 0 mA discharge current.
 */
 if (fg_res < 90 || (pwr_stat & PS_STAT_BAT_CHRG_DIR) || no_current_sense_res)
  goto not_full;

 if (curr == 0) {
  info->status = POWER_SUPPLY_STATUS_FULL;
  return;
 }

not_full:
 if (pwr_stat & PS_STAT_BAT_CHRG_DIR)
  info->status = POWER_SUPPLY_STATUS_CHARGING;
 else
  info->status = POWER_SUPPLY_STATUS_DISCHARGING;
}

static int fuel_gauge_battery_health(struct axp288_fg_info *info)
{
 int vocv = VOLTAGE_FROM_ADC(info->ocv);
 int health = POWER_SUPPLY_HEALTH_UNKNOWN;

 if (vocv > info->max_volt)
  health = POWER_SUPPLY_HEALTH_OVERVOLTAGE;
 else
  health = POWER_SUPPLY_HEALTH_GOOD;

 return health;
}

static int fuel_gauge_get_property(struct power_supply *ps,
  enum power_supply_property prop,
  union power_supply_propval *val)
{
 struct axp288_fg_info *info = power_supply_get_drvdata(ps);
 int ret, value;

 mutex_lock(&info->lock);

 ret = fuel_gauge_update_registers(info);
 if (ret < 0)
  goto out;

 switch (prop) {
 case POWER_SUPPLY_PROP_STATUS:
  fuel_gauge_get_status(info);
  val->intval = info->status;
  break;
 case POWER_SUPPLY_PROP_HEALTH:
  val->intval = fuel_gauge_battery_health(info);
  break;
 case POWER_SUPPLY_PROP_VOLTAGE_NOW:
  value = VOLTAGE_FROM_ADC(info->bat_volt);
  val->intval = PROP_VOLT(value);
  break;
 case POWER_SUPPLY_PROP_VOLTAGE_OCV:
  value = VOLTAGE_FROM_ADC(info->ocv);
  val->intval = PROP_VOLT(value);
  break;
 case POWER_SUPPLY_PROP_CURRENT_NOW:
  if (info->d_curr > 0)
   value = -1 * info->d_curr;
  else
   value = info->c_curr;

  val->intval = PROP_CURR(value);
  break;
 case POWER_SUPPLY_PROP_PRESENT:
  if (info->pwr_op & CHRG_STAT_BAT_PRESENT)
   val->intval = 1;
  else
   val->intval = 0;
  break;
 case POWER_SUPPLY_PROP_CAPACITY:
  if (!(info->fg_res & FG_REP_CAP_VALID))
   dev_err(info->dev, "capacity measurement not valid\n");
  val->intval = (info->fg_res & FG_REP_CAP_VAL_MASK);
  break;
 case POWER_SUPPLY_PROP_CAPACITY_ALERT_MIN:
  val->intval = (info->low_cap & 0x0f);
  break;
 case POWER_SUPPLY_PROP_TECHNOLOGY:
  val->intval = POWER_SUPPLY_TECHNOLOGY_LION;
  break;
 case POWER_SUPPLY_PROP_CHARGE_NOW:
  val->intval = info->fg_cc_mtr1 * FG_DES_CAP_RES_LSB;
  break;
 case POWER_SUPPLY_PROP_CHARGE_FULL:
  val->intval = info->fg_des_cap1 * FG_DES_CAP_RES_LSB;
  break;
 case POWER_SUPPLY_PROP_VOLTAGE_MAX_DESIGN:
  val->intval = PROP_VOLT(info->max_volt);
  break;
 default:
  ret = -EINVAL;
 }

out:
 mutex_unlock(&info->lock);
 return ret;
}

static int fuel_gauge_set_property(struct power_supply *ps,
  enum power_supply_property prop,
  const union power_supply_propval *val)
{
 struct axp288_fg_info *info = power_supply_get_drvdata(ps);
 int new_low_cap, ret = 0;

 mutex_lock(&info->lock);
 switch (prop) {
 case POWER_SUPPLY_PROP_CAPACITY_ALERT_MIN:
  if ((val->intval < 0) || (val->intval > 15)) {
   ret = -EINVAL;
   break;
  }
  new_low_cap = info->low_cap;
  new_low_cap &= 0xf0;
  new_low_cap |= (val->intval & 0xf);
  ret = fuel_gauge_reg_writeb(info, AXP288_FG_LOW_CAP_REG, new_low_cap);
  if (ret == 0)
   info->low_cap = new_low_cap;
  break;
 default:
  ret = -EINVAL;
  break;
 }

 mutex_unlock(&info->lock);
 return ret;
}

static int fuel_gauge_property_is_writeable(struct power_supply *psy,
 enum power_supply_property psp)
{
 int ret;

 switch (psp) {
 case POWER_SUPPLY_PROP_CAPACITY_ALERT_MIN:
  ret = 1;
  break;
 default:
  ret = 0;
 }

 return ret;
}

static irqreturn_t fuel_gauge_thread_handler(int irq, void *dev)
{
 struct axp288_fg_info *info = dev;
 int i;

 for (i = 0; i < AXP288_FG_INTR_NUM; i++) {
  if (info->irq[i] == irq)
   break;
 }

 if (i >= AXP288_FG_INTR_NUM) {
  dev_warn(info->dev, "spurious interrupt!!\n");
  return IRQ_NONE;
 }

 switch (i) {
 case QWBTU_IRQ:
  dev_info(info->dev, "Quit Battery under temperature in work mode IRQ (QWBTU)\n");
  break;
 case WBTU_IRQ:
  dev_info(info->dev, "Battery under temperature in work mode IRQ (WBTU)\n");
  break;
 case QWBTO_IRQ:
  dev_info(info->dev, "Quit Battery over temperature in work mode IRQ (QWBTO)\n");
  break;
 case WBTO_IRQ:
  dev_info(info->dev, "Battery over temperature in work mode IRQ (WBTO)\n");
  break;
 case WL2_IRQ:
  dev_info(info->dev, "Low Batt Warning(2) INTR\n");
  break;
 case WL1_IRQ:
  dev_info(info->dev, "Low Batt Warning(1) INTR\n");
  break;
 default:
  dev_warn(info->dev, "Spurious Interrupt!!!\n");
 }

 mutex_lock(&info->lock);
 info->valid = 0; /* Force updating of the cached registers */
 mutex_unlock(&info->lock);

 power_supply_changed(info->bat);
 return IRQ_HANDLED;
}

static void fuel_gauge_external_power_changed(struct power_supply *psy)
{
 struct axp288_fg_info *info = power_supply_get_drvdata(psy);

 mutex_lock(&info->lock);
 info->valid = 0; /* Force updating of the cached registers */
 mutex_unlock(&info->lock);
 power_supply_changed(psy);
}

static struct power_supply_desc fuel_gauge_desc = {
 .name   = DEV_NAME,
 .type   = POWER_SUPPLY_TYPE_BATTERY,
 .properties  = fuel_gauge_props,
 .num_properties  = ARRAY_SIZE(fuel_gauge_props),
 .get_property  = fuel_gauge_get_property,
 .set_property  = fuel_gauge_set_property,
 .property_is_writeable = fuel_gauge_property_is_writeable,
 .external_power_changed = fuel_gauge_external_power_changed,
};

/*
 * Some devices have no battery (HDMI sticks) and the axp288 battery's
 * detection reports one despite it not being there.
 * Please keep this listed sorted alphabetically.
 */
static const struct dmi_system_id axp288_quirks[] = {
 {
  /* ACEPC T8 Cherry Trail Z8350 mini PC */
  .matches = {
   DMI_EXACT_MATCH(DMI_BOARD_VENDOR, "To be filled by O.E.M."),
   DMI_EXACT_MATCH(DMI_BOARD_NAME, "Cherry Trail CR"),
   DMI_EXACT_MATCH(DMI_PRODUCT_SKU, "T8"),
   /* also match on somewhat unique bios-version */
   DMI_EXACT_MATCH(DMI_BIOS_VERSION, "1.000"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {
  /* ACEPC T11 Cherry Trail Z8350 mini PC */
  .matches = {
   DMI_EXACT_MATCH(DMI_BOARD_VENDOR, "To be filled by O.E.M."),
   DMI_EXACT_MATCH(DMI_BOARD_NAME, "Cherry Trail CR"),
   DMI_EXACT_MATCH(DMI_PRODUCT_SKU, "T11"),
   /* also match on somewhat unique bios-version */
   DMI_EXACT_MATCH(DMI_BIOS_VERSION, "1.000"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {
  /* Intel Bay Trail Compute Stick */
  .matches = {
   DMI_MATCH(DMI_SYS_VENDOR, "Intel"),
   /* Partial match for STCK1A32WFC STCK1A32FC, STCK1A8LFC variants */
   DMI_MATCH(DMI_PRODUCT_NAME, "STCK1A"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {
  /* Intel Cherry Trail Compute Stick */
  .matches = {
   DMI_MATCH(DMI_SYS_VENDOR, "Intel"),
   /* Partial match for STK1AW32SC and STK1A32SC variants */
   DMI_MATCH(DMI_PRODUCT_NAME, "STK1A"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {
  /* Meegopad T02 */
  .matches = {
   DMI_MATCH(DMI_PRODUCT_NAME, "MEEGOPAD T02"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 { /* Mele PCG03 Mini PC */
  .matches = {
   DMI_EXACT_MATCH(DMI_BOARD_VENDOR, "Mini PC"),
   DMI_EXACT_MATCH(DMI_BOARD_NAME, "Mini PC"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {
  /* Minix Neo Z83-4 mini PC */
  .matches = {
   DMI_MATCH(DMI_SYS_VENDOR, "MINIX"),
   DMI_MATCH(DMI_PRODUCT_NAME, "Z83-4"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {
  /*
   * One Mix 1, this uses the "T3 MRD" boardname used by
   * generic mini PCs, but it is a mini laptop so it does
   * actually have a battery!
 */
  .matches = {
   DMI_MATCH(DMI_BOARD_NAME, "T3 MRD"),
   DMI_MATCH(DMI_BIOS_DATE, "06/14/2018"),
  },
  .driver_data = NULL,
 },
 {
  /* Radxa ROCK Pi X Single Board Computer */
  .matches = {
   DMI_MATCH(DMI_BOARD_NAME, "ROCK Pi X"),
   DMI_MATCH(DMI_BOARD_VENDOR, "Radxa"),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {
  /*
   * Various Ace PC/Meegopad/MinisForum/Wintel Mini-PCs/HDMI-sticks
   * This entry must be last because it is generic, this allows
   * adding more specifuc quirks overriding this generic entry.
 */
  .matches = {
   DMI_MATCH(DMI_BOARD_NAME, "T3 MRD"),
   DMI_MATCH(DMI_CHASSIS_TYPE, "3"),
   DMI_MATCH(DMI_BIOS_VENDOR, "American Megatrends Inc."),
  },
  .driver_data = (void *)AXP288_QUIRK_NO_BATTERY,
 },
 {}
};

static int axp288_fuel_gauge_read_initial_regs(struct axp288_fg_info *info)
{
 unsigned int val;
 int ret;

 /*
  * On some devices the fuelgauge and charger parts of the axp288 are
  * not used, check that the fuelgauge is enabled (CC_CTRL != 0).
 */
 ret = regmap_read(info->regmap, AXP20X_CC_CTRL, &val);
 if (ret < 0)
  return ret;
 if (val == 0)
  return -ENODEV;

 ret = fuel_gauge_reg_readb(info, AXP288_FG_DES_CAP1_REG);
 if (ret < 0)
  return ret;

 if (!(ret & FG_DES_CAP1_VALID)) {
  dev_err(info->dev, "axp288 not configured by firmware\n");
  return -ENODEV;
 }

 ret = fuel_gauge_reg_readb(info, AXP20X_CHRG_CTRL1);
 if (ret < 0)
  return ret;
 switch ((ret & CHRG_CCCV_CV_MASK) >> CHRG_CCCV_CV_BIT_POS) {
 case CHRG_CCCV_CV_4100MV:
  info->max_volt = 4100;
  break;
 case CHRG_CCCV_CV_4150MV:
  info->max_volt = 4150;
  break;
 case CHRG_CCCV_CV_4200MV:
  info->max_volt = 4200;
  break;
 case CHRG_CCCV_CV_4350MV:
  info->max_volt = 4350;
  break;
 }

 ret = fuel_gauge_reg_readb(info, AXP20X_PWR_OP_MODE);
 if (ret < 0)
  return ret;
 info->pwr_op = ret;

 ret = fuel_gauge_reg_readb(info, AXP288_FG_LOW_CAP_REG);
 if (ret < 0)
  return ret;
 info->low_cap = ret;

 return 0;
}

static void axp288_fuel_gauge_release_iio_chans(void *data)
{
 struct axp288_fg_info *info = data;
 int i;

 for (i = 0; i < IIO_CHANNEL_NUM; i++)
  if (!IS_ERR_OR_NULL(info->iio_channel[i]))
   iio_channel_release(info->iio_channel[i]);
}

static int axp288_fuel_gauge_probe(struct platform_device *pdev)
{
 struct axp288_fg_info *info;
 struct axp20x_dev *axp20x = dev_get_drvdata(pdev->dev.parent);
 struct power_supply_config psy_cfg = {};
 static const char * const iio_chan_name[] = {
  [BAT_CHRG_CURR] = "axp288-chrg-curr",
  [BAT_D_CURR] = "axp288-chrg-d-curr",
  [BAT_VOLT] = "axp288-batt-volt",
 };
 const struct dmi_system_id *dmi_id;
 struct device *dev = &pdev->dev;
 unsigned long quirks = 0;
 int i, pirq, ret;

 /*
  * Normally the native AXP288 fg/charger drivers are preferred but
  * on some devices the ACPI drivers should be used instead.
 */
 if (!acpi_quirk_skip_acpi_ac_and_battery())
  return -ENODEV;

 dmi_id = dmi_first_match(axp288_quirks);
 if (dmi_id)
  quirks = (unsigned long)dmi_id->driver_data;

 if (quirks & AXP288_QUIRK_NO_BATTERY)
  return -ENODEV;

 info = devm_kzalloc(dev, sizeof(*info), GFP_KERNEL);
 if (!info)
  return -ENOMEM;

 info->dev = dev;
 info->regmap = axp20x->regmap;
 info->status = POWER_SUPPLY_STATUS_UNKNOWN;
 info->valid = 0;

 platform_set_drvdata(pdev, info);

 mutex_init(&info->lock);

 for (i = 0; i < AXP288_FG_INTR_NUM; i++) {
  pirq = platform_get_irq(pdev, i);
  if (pirq < 0)
   continue;
  ret = regmap_irq_get_virq(axp20x->regmap_irqc, pirq);
  if (ret < 0)
   return dev_err_probe(dev, ret, "getting vIRQ %d\n", pirq);

  info->irq[i] = ret;
 }

 for (i = 0; i < IIO_CHANNEL_NUM; i++) {
  /*
   * Note cannot use devm_iio_channel_get because x86 systems
   * lack the device<->channel maps which iio_channel_get will
   * try to use when passed a non NULL device pointer.
 */
  info->iio_channel[i] =
   iio_channel_get(NULL, iio_chan_name[i]);
  if (IS_ERR(info->iio_channel[i])) {
   ret = PTR_ERR(info->iio_channel[i]);
   dev_dbg(dev, "error getting iiochan %s: %d\n", iio_chan_name[i], ret);
   /* Wait for axp288_adc to load */
   if (ret == -ENODEV)
    ret = -EPROBE_DEFER;

   axp288_fuel_gauge_release_iio_chans(info);
   return ret;
  }
 }

 ret = devm_add_action_or_reset(dev, axp288_fuel_gauge_release_iio_chans, info);
 if (ret)
  return ret;

 ret = iosf_mbi_block_punit_i2c_access();
 if (ret < 0)
  return ret;

 ret = axp288_fuel_gauge_read_initial_regs(info);
 iosf_mbi_unblock_punit_i2c_access();
 if (ret < 0)
  return ret;

 psy_cfg.drv_data = info;
 if (no_current_sense_res)
  fuel_gauge_desc.num_properties = ARRAY_SIZE(fuel_gauge_props) - 3;
 info->bat = devm_power_supply_register(dev, &fuel_gauge_desc, &psy_cfg);
 if (IS_ERR(info->bat)) {
  ret = PTR_ERR(info->bat);
  dev_err(dev, "failed to register battery: %d\n", ret);
  return ret;
 }

 for (i = 0; i < AXP288_FG_INTR_NUM; i++) {
  ret = devm_request_threaded_irq(dev, info->irq[i], NULL,
      fuel_gauge_thread_handler,
      IRQF_ONESHOT, DEV_NAME, info);
  if (ret)
   return dev_err_probe(dev, ret, "requesting IRQ %d\n", info->irq[i]);
 }

 return 0;
}

static const struct platform_device_id axp288_fg_id_table[] = {
 { .name = DEV_NAME },
 {},
};
MODULE_DEVICE_TABLE(platform, axp288_fg_id_table);

static struct platform_driver axp288_fuel_gauge_driver = {
 .probe = axp288_fuel_gauge_probe,
 .id_table = axp288_fg_id_table,
 .driver = {
  .name = DEV_NAME,
 },
};

module_platform_driver(axp288_fuel_gauge_driver);

MODULE_AUTHOR("Ramakrishna Pallala <ramakrishna.pallala@intel.com>");
MODULE_AUTHOR("Todd Brandt <todd.e.brandt@linux.intel.com>");
MODULE_DESCRIPTION("Xpower AXP288 Fuel Gauge Driver");
MODULE_LICENSE("GPL");