/*
   * Copyright 2012 Texas Instruments
   *
   * Author: Milo(Woogyom) Kim <milo.kim@ti.com>
   *
   * This program is free software; you can redistribute it and/or modify
   * it under the terms of the GNU General Public License version 2 as
   * published by the Free Software Foundation.
   *
   */
  
  #include <linux/module.h>
  #include <linux/slab.h>
  #include <linux/i2c.h>
  #include <linux/regmap.h>
  #include <linux/err.h>
  #include <linux/gpio.h>

  #include <linux/delay.h>

  #include <linux/regulator/lp872x.h>
  #include <linux/regulator/driver.h>
  #include <linux/platform_device.h>

  #include <linux/of.h>
  #include <linux/of_gpio.h>
  #include <linux/regulator/of_regulator.h>

  
  /* Registers : LP8720/8725 shared */
  #define LP872X_GENERAL_CFG		0x00
  #define LP872X_LDO1_VOUT		0x01
  #define LP872X_LDO2_VOUT		0x02
  #define LP872X_LDO3_VOUT		0x03
  #define LP872X_LDO4_VOUT		0x04
  #define LP872X_LDO5_VOUT		0x05
  
  /* Registers : LP8720 */
  #define LP8720_BUCK_VOUT1		0x06
  #define LP8720_BUCK_VOUT2		0x07
  #define LP8720_ENABLE			0x08
  
  /* Registers : LP8725 */
  #define LP8725_LILO1_VOUT		0x06
  #define LP8725_LILO2_VOUT		0x07
  #define LP8725_BUCK1_VOUT1		0x08
  #define LP8725_BUCK1_VOUT2		0x09
  #define LP8725_BUCK2_VOUT1		0x0A
  #define LP8725_BUCK2_VOUT2		0x0B
  #define LP8725_BUCK_CTRL		0x0C
  #define LP8725_LDO_CTRL			0x0D
  
  /* Mask/shift : LP8720/LP8725 shared */
  #define LP872X_VOUT_M			0x1F
  #define LP872X_START_DELAY_M		0xE0
  #define LP872X_START_DELAY_S		5
  #define LP872X_EN_LDO1_M		BIT(0)
  #define LP872X_EN_LDO2_M		BIT(1)
  #define LP872X_EN_LDO3_M		BIT(2)
  #define LP872X_EN_LDO4_M		BIT(3)
  #define LP872X_EN_LDO5_M		BIT(4)
  
  /* Mask/shift : LP8720 */
  #define LP8720_TIMESTEP_S		0		/* Addr 00h */
  #define LP8720_TIMESTEP_M		BIT(0)
  #define LP8720_EXT_DVS_M		BIT(2)
  #define LP8720_BUCK_FPWM_S		5		/* Addr 07h */
  #define LP8720_BUCK_FPWM_M		BIT(5)
  #define LP8720_EN_BUCK_M		BIT(5)		/* Addr 08h */
  #define LP8720_DVS_SEL_M		BIT(7)
  
  /* Mask/shift : LP8725 */
  #define LP8725_TIMESTEP_M		0xC0		/* Addr 00h */
  #define LP8725_TIMESTEP_S		6
  #define LP8725_BUCK1_EN_M		BIT(0)
  #define LP8725_DVS1_M			BIT(2)
  #define LP8725_DVS2_M			BIT(3)
  #define LP8725_BUCK2_EN_M		BIT(4)
  #define LP8725_BUCK_CL_M		0xC0		/* Addr 09h, 0Bh */
  #define LP8725_BUCK_CL_S		6
  #define LP8725_BUCK1_FPWM_S		1		/* Addr 0Ch */
  #define LP8725_BUCK1_FPWM_M		BIT(1)
  #define LP8725_BUCK2_FPWM_S		5
  #define LP8725_BUCK2_FPWM_M		BIT(5)
  #define LP8725_EN_LILO1_M		BIT(5)		/* Addr 0Dh */
  #define LP8725_EN_LILO2_M		BIT(6)
  
  /* PWM mode */
  #define LP872X_FORCE_PWM		1
  #define LP872X_AUTO_PWM			0
  
  #define LP8720_NUM_REGULATORS		6
  #define LP8725_NUM_REGULATORS		9
  #define EXTERN_DVS_USED			0
  #define MAX_DELAY			6

  /* Default DVS Mode */
  #define LP8720_DEFAULT_DVS		0
  #define LP8725_DEFAULT_DVS		BIT(2)

  /* dump registers in regmap-debugfs */
  #define MAX_REGISTERS			0x0F
  
  enum lp872x_id {
  	LP8720,
  	LP8725,
  };
  
  struct lp872x {
  	struct regmap *regmap;
  	struct device *dev;
  	enum lp872x_id chipid;
  	struct lp872x_platform_data *pdata;

  	int num_regulators;
  	enum lp872x_dvs_state dvs_pin;

  };
  
  /* LP8720/LP8725 shared voltage table for LDOs */
  static const unsigned int lp872x_ldo_vtbl[] = {
  	1200000, 1250000, 1300000, 1350000, 1400000, 1450000, 1500000, 1550000,
  	1600000, 1650000, 1700000, 1750000, 1800000, 1850000, 1900000, 2000000,
  	2100000, 2200000, 2300000, 2400000, 2500000, 2600000, 2650000, 2700000,
  	2750000, 2800000, 2850000, 2900000, 2950000, 3000000, 3100000, 3300000,
  };
  
  /* LP8720 LDO4 voltage table */
  static const unsigned int lp8720_ldo4_vtbl[] = {
  	 800000,  850000,  900000, 1000000, 1100000, 1200000, 1250000, 1300000,
  	1350000, 1400000, 1450000, 1500000, 1550000, 1600000, 1650000, 1700000,
  	1750000, 1800000, 1850000, 1900000, 2000000, 2100000, 2200000, 2300000,
  	2400000, 2500000, 2600000, 2650000, 2700000, 2750000, 2800000, 2850000,
  };
  
  /* LP8725 LILO(Low Input Low Output) voltage table */
  static const unsigned int lp8725_lilo_vtbl[] = {
  	 800000,  850000,  900000,  950000, 1000000, 1050000, 1100000, 1150000,
  	1200000, 1250000, 1300000, 1350000, 1400000, 1500000, 1600000, 1700000,
  	1800000, 1900000, 2000000, 2100000, 2200000, 2300000, 2400000, 2500000,
  	2600000, 2700000, 2800000, 2850000, 2900000, 3000000, 3100000, 3300000,
  };
  
  /* LP8720 BUCK voltage table */
  #define EXT_R		0	/* external resistor divider */
  static const unsigned int lp8720_buck_vtbl[] = {
  	  EXT_R,  800000,  850000,  900000,  950000, 1000000, 1050000, 1100000,
  	1150000, 1200000, 1250000, 1300000, 1350000, 1400000, 1450000, 1500000,
  	1550000, 1600000, 1650000, 1700000, 1750000, 1800000, 1850000, 1900000,
  	1950000, 2000000, 2050000, 2100000, 2150000, 2200000, 2250000, 2300000,
  };
  
  /* LP8725 BUCK voltage table */
  static const unsigned int lp8725_buck_vtbl[] = {
  	 800000,  850000,  900000,  950000, 1000000, 1050000, 1100000, 1150000,
  	1200000, 1250000, 1300000, 1350000, 1400000, 1500000, 1600000, 1700000,
  	1750000, 1800000, 1850000, 1900000, 2000000, 2100000, 2200000, 2300000,
  	2400000, 2500000, 2600000, 2700000, 2800000, 2850000, 2900000, 3000000,
  };
  
  /* LP8725 BUCK current limit */
  static const unsigned int lp8725_buck_uA[] = {
  	460000, 780000, 1050000, 1370000,
  };
  
  static int lp872x_read_byte(struct lp872x *lp, u8 addr, u8 *data)
  {
  	int ret;
  	unsigned int val;
  
  	ret = regmap_read(lp->regmap, addr, &val);
  	if (ret < 0) {
  		dev_err(lp->dev, "failed to read 0x%.2x
  ", addr);
  		return ret;
  	}
  
  	*data = (u8)val;
  	return 0;
  }
  
  static inline int lp872x_write_byte(struct lp872x *lp, u8 addr, u8 data)
  {
  	return regmap_write(lp->regmap, addr, data);
  }
  
  static inline int lp872x_update_bits(struct lp872x *lp, u8 addr,
  				unsigned int mask, u8 data)
  {
  	return regmap_update_bits(lp->regmap, addr, mask, data);
  }

  static int lp872x_get_timestep_usec(struct lp872x *lp)
  {
  	enum lp872x_id chip = lp->chipid;
  	u8 val, mask, shift;
  	int *time_usec, size, ret;
  	int lp8720_time_usec[] = { 25, 50 };
  	int lp8725_time_usec[] = { 32, 64, 128, 256 };
  
  	switch (chip) {
  	case LP8720:
  		mask = LP8720_TIMESTEP_M;
  		shift = LP8720_TIMESTEP_S;
  		time_usec = &lp8720_time_usec[0];
  		size = ARRAY_SIZE(lp8720_time_usec);
  		break;
  	case LP8725:
  		mask = LP8725_TIMESTEP_M;
  		shift = LP8725_TIMESTEP_S;
  		time_usec = &lp8725_time_usec[0];
  		size = ARRAY_SIZE(lp8725_time_usec);
  		break;
  	default:
  		return -EINVAL;
  	}
  
  	ret = lp872x_read_byte(lp, LP872X_GENERAL_CFG, &val);
  	if (ret)

  		return ret;

  
  	val = (val & mask) >> shift;
  	if (val >= size)
  		return -EINVAL;
  
  	return *(time_usec + val);
  }
  
  static int lp872x_regulator_enable_time(struct regulator_dev *rdev)
  {
  	struct lp872x *lp = rdev_get_drvdata(rdev);

  	enum lp872x_regulator_id rid = rdev_get_id(rdev);

  	int time_step_us = lp872x_get_timestep_usec(lp);

  	int ret;

  	u8 addr, val;
  
  	if (time_step_us < 0)

  		return time_step_us;

  	switch (rid) {
  	case LP8720_ID_LDO1 ... LP8720_ID_BUCK:
  		addr = LP872X_LDO1_VOUT + rid;

  		break;

  	case LP8725_ID_LDO1 ... LP8725_ID_BUCK1:
  		addr = LP872X_LDO1_VOUT + rid - LP8725_ID_BASE;

  		break;
  	case LP8725_ID_BUCK2:
  		addr = LP8725_BUCK2_VOUT1;
  		break;
  	default:
  		return -EINVAL;
  	}
  
  	ret = lp872x_read_byte(lp, addr, &val);
  	if (ret)
  		return ret;
  
  	val = (val & LP872X_START_DELAY_M) >> LP872X_START_DELAY_S;
  
  	return val > MAX_DELAY ? 0 : val * time_step_us;
  }

  static void lp872x_set_dvs(struct lp872x *lp, enum lp872x_dvs_sel dvs_sel,
  			int gpio)

  {

  	enum lp872x_dvs_state state;
  
  	state = dvs_sel == SEL_V1 ? DVS_HIGH : DVS_LOW;
  	gpio_set_value(gpio, state);
  	lp->dvs_pin = state;
  }
  
  static u8 lp872x_select_buck_vout_addr(struct lp872x *lp,
  				enum lp872x_regulator_id buck)
  {
  	u8 val, addr;
  
  	if (lp872x_read_byte(lp, LP872X_GENERAL_CFG, &val))
  		return 0;
  
  	switch (buck) {
  	case LP8720_ID_BUCK:
  		if (val & LP8720_EXT_DVS_M) {
  			addr = (lp->dvs_pin == DVS_HIGH) ?
  				LP8720_BUCK_VOUT1 : LP8720_BUCK_VOUT2;
  		} else {
  			if (lp872x_read_byte(lp, LP8720_ENABLE, &val))
  				return 0;
  
  			addr = val & LP8720_DVS_SEL_M ?
  				LP8720_BUCK_VOUT1 : LP8720_BUCK_VOUT2;
  		}
  		break;
  	case LP8725_ID_BUCK1:
  		if (val & LP8725_DVS1_M)
  			addr = LP8725_BUCK1_VOUT1;
  		else
  			addr = (lp->dvs_pin == DVS_HIGH) ?
  				LP8725_BUCK1_VOUT1 : LP8725_BUCK1_VOUT2;
  		break;
  	case LP8725_ID_BUCK2:
  		addr =  val & LP8725_DVS2_M ?
  			LP8725_BUCK2_VOUT1 : LP8725_BUCK2_VOUT2;
  		break;
  	default:
  		return 0;
  	}
  
  	return addr;
  }
  
  static bool lp872x_is_valid_buck_addr(u8 addr)
  {
  	switch (addr) {
  	case LP8720_BUCK_VOUT1:
  	case LP8720_BUCK_VOUT2:
  	case LP8725_BUCK1_VOUT1:
  	case LP8725_BUCK1_VOUT2:
  	case LP8725_BUCK2_VOUT1:
  	case LP8725_BUCK2_VOUT2:
  		return true;
  	default:
  		return false;
  	}
  }
  
  static int lp872x_buck_set_voltage_sel(struct regulator_dev *rdev,
  					unsigned selector)
  {
  	struct lp872x *lp = rdev_get_drvdata(rdev);
  	enum lp872x_regulator_id buck = rdev_get_id(rdev);
  	u8 addr, mask = LP872X_VOUT_M;

  	struct lp872x_dvs *dvs = lp->pdata ? lp->pdata->dvs : NULL;

  
  	if (dvs && gpio_is_valid(dvs->gpio))

  		lp872x_set_dvs(lp, dvs->vsel, dvs->gpio);

  
  	addr = lp872x_select_buck_vout_addr(lp, buck);
  	if (!lp872x_is_valid_buck_addr(addr))
  		return -EINVAL;
  
  	return lp872x_update_bits(lp, addr, mask, selector);
  }
  
  static int lp872x_buck_get_voltage_sel(struct regulator_dev *rdev)
  {
  	struct lp872x *lp = rdev_get_drvdata(rdev);
  	enum lp872x_regulator_id buck = rdev_get_id(rdev);
  	u8 addr, val;
  	int ret;
  
  	addr = lp872x_select_buck_vout_addr(lp, buck);
  	if (!lp872x_is_valid_buck_addr(addr))
  		return -EINVAL;
  
  	ret = lp872x_read_byte(lp, addr, &val);
  	if (ret)
  		return ret;
  
  	return val & LP872X_VOUT_M;
  }
  
  static int lp8725_buck_set_current_limit(struct regulator_dev *rdev,
  					int min_uA, int max_uA)
  {
  	struct lp872x *lp = rdev_get_drvdata(rdev);
  	enum lp872x_regulator_id buck = rdev_get_id(rdev);

  	int i;
  	u8 addr;

  
  	switch (buck) {
  	case LP8725_ID_BUCK1:
  		addr = LP8725_BUCK1_VOUT2;
  		break;
  	case LP8725_ID_BUCK2:
  		addr = LP8725_BUCK2_VOUT2;
  		break;
  	default:
  		return -EINVAL;
  	}

  	for (i = ARRAY_SIZE(lp8725_buck_uA) - 1; i >= 0; i--) {

  		if (lp8725_buck_uA[i] >= min_uA &&
  			lp8725_buck_uA[i] <= max_uA)

  			return lp872x_update_bits(lp, addr,
  						  LP8725_BUCK_CL_M,
  						  i << LP8725_BUCK_CL_S);
  	}

  	return -EINVAL;

  }
  
  static int lp8725_buck_get_current_limit(struct regulator_dev *rdev)
  {
  	struct lp872x *lp = rdev_get_drvdata(rdev);
  	enum lp872x_regulator_id buck = rdev_get_id(rdev);
  	u8 addr, val;
  	int ret;
  
  	switch (buck) {
  	case LP8725_ID_BUCK1:
  		addr = LP8725_BUCK1_VOUT2;
  		break;
  	case LP8725_ID_BUCK2:
  		addr = LP8725_BUCK2_VOUT2;
  		break;
  	default:
  		return -EINVAL;
  	}
  
  	ret = lp872x_read_byte(lp, addr, &val);
  	if (ret)
  		return ret;
  
  	val = (val & LP8725_BUCK_CL_M) >> LP8725_BUCK_CL_S;
  
  	return (val < ARRAY_SIZE(lp8725_buck_uA)) ?
  			lp8725_buck_uA[val] : -EINVAL;
  }
  
  static int lp872x_buck_set_mode(struct regulator_dev *rdev, unsigned int mode)
  {
  	struct lp872x *lp = rdev_get_drvdata(rdev);
  	enum lp872x_regulator_id buck = rdev_get_id(rdev);
  	u8 addr, mask, shift, val;
  
  	switch (buck) {
  	case LP8720_ID_BUCK:
  		addr = LP8720_BUCK_VOUT2;
  		mask = LP8720_BUCK_FPWM_M;
  		shift = LP8720_BUCK_FPWM_S;
  		break;
  	case LP8725_ID_BUCK1:
  		addr = LP8725_BUCK_CTRL;
  		mask = LP8725_BUCK1_FPWM_M;
  		shift = LP8725_BUCK1_FPWM_S;
  		break;
  	case LP8725_ID_BUCK2:
  		addr = LP8725_BUCK_CTRL;
  		mask = LP8725_BUCK2_FPWM_M;
  		shift = LP8725_BUCK2_FPWM_S;
  		break;
  	default:
  		return -EINVAL;
  	}
  
  	if (mode == REGULATOR_MODE_FAST)
  		val = LP872X_FORCE_PWM << shift;
  	else if (mode == REGULATOR_MODE_NORMAL)
  		val = LP872X_AUTO_PWM << shift;
  	else
  		return -EINVAL;
  
  	return lp872x_update_bits(lp, addr, mask, val);
  }
  
  static unsigned int lp872x_buck_get_mode(struct regulator_dev *rdev)
  {
  	struct lp872x *lp = rdev_get_drvdata(rdev);
  	enum lp872x_regulator_id buck = rdev_get_id(rdev);
  	u8 addr, mask, val;
  	int ret;
  
  	switch (buck) {
  	case LP8720_ID_BUCK:
  		addr = LP8720_BUCK_VOUT2;
  		mask = LP8720_BUCK_FPWM_M;
  		break;
  	case LP8725_ID_BUCK1:
  		addr = LP8725_BUCK_CTRL;
  		mask = LP8725_BUCK1_FPWM_M;
  		break;
  	case LP8725_ID_BUCK2:
  		addr = LP8725_BUCK_CTRL;
  		mask = LP8725_BUCK2_FPWM_M;
  		break;
  	default:
  		return -EINVAL;
  	}
  
  	ret = lp872x_read_byte(lp, addr, &val);
  	if (ret)
  		return ret;
  
  	return val & mask ? REGULATOR_MODE_FAST : REGULATOR_MODE_NORMAL;
  }
  
  static struct regulator_ops lp872x_ldo_ops = {
  	.list_voltage = regulator_list_voltage_table,

  	.map_voltage = regulator_map_voltage_ascend,

  	.set_voltage_sel = regulator_set_voltage_sel_regmap,
  	.get_voltage_sel = regulator_get_voltage_sel_regmap,
  	.enable = regulator_enable_regmap,
  	.disable = regulator_disable_regmap,
  	.is_enabled = regulator_is_enabled_regmap,
  	.enable_time = lp872x_regulator_enable_time,
  };
  
  static struct regulator_ops lp8720_buck_ops = {
  	.list_voltage = regulator_list_voltage_table,

  	.map_voltage = regulator_map_voltage_ascend,

  	.set_voltage_sel = lp872x_buck_set_voltage_sel,
  	.get_voltage_sel = lp872x_buck_get_voltage_sel,
  	.enable = regulator_enable_regmap,
  	.disable = regulator_disable_regmap,
  	.is_enabled = regulator_is_enabled_regmap,
  	.enable_time = lp872x_regulator_enable_time,
  	.set_mode = lp872x_buck_set_mode,
  	.get_mode = lp872x_buck_get_mode,
  };
  
  static struct regulator_ops lp8725_buck_ops = {
  	.list_voltage = regulator_list_voltage_table,

  	.map_voltage = regulator_map_voltage_ascend,

  	.set_voltage_sel = lp872x_buck_set_voltage_sel,
  	.get_voltage_sel = lp872x_buck_get_voltage_sel,
  	.enable = regulator_enable_regmap,
  	.disable = regulator_disable_regmap,
  	.is_enabled = regulator_is_enabled_regmap,
  	.enable_time = lp872x_regulator_enable_time,
  	.set_mode = lp872x_buck_set_mode,
  	.get_mode = lp872x_buck_get_mode,
  	.set_current_limit = lp8725_buck_set_current_limit,
  	.get_current_limit = lp8725_buck_get_current_limit,
  };
  
  static struct regulator_desc lp8720_regulator_desc[] = {
  	{
  		.name = "ldo1",

  		.of_match = of_match_ptr("ldo1"),

  		.id = LP8720_ID_LDO1,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO1_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8720_ENABLE,
  		.enable_mask = LP872X_EN_LDO1_M,
  	},
  	{
  		.name = "ldo2",

  		.of_match = of_match_ptr("ldo2"),

  		.id = LP8720_ID_LDO2,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO2_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8720_ENABLE,
  		.enable_mask = LP872X_EN_LDO2_M,
  	},
  	{
  		.name = "ldo3",

  		.of_match = of_match_ptr("ldo3"),

  		.id = LP8720_ID_LDO3,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO3_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8720_ENABLE,
  		.enable_mask = LP872X_EN_LDO3_M,
  	},
  	{
  		.name = "ldo4",

  		.of_match = of_match_ptr("ldo4"),

  		.id = LP8720_ID_LDO4,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp8720_ldo4_vtbl),
  		.volt_table = lp8720_ldo4_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO4_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8720_ENABLE,
  		.enable_mask = LP872X_EN_LDO4_M,
  	},
  	{
  		.name = "ldo5",

  		.of_match = of_match_ptr("ldo5"),

  		.id = LP8720_ID_LDO5,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO5_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8720_ENABLE,
  		.enable_mask = LP872X_EN_LDO5_M,
  	},
  	{
  		.name = "buck",

  		.of_match = of_match_ptr("buck"),

  		.id = LP8720_ID_BUCK,
  		.ops = &lp8720_buck_ops,
  		.n_voltages = ARRAY_SIZE(lp8720_buck_vtbl),
  		.volt_table = lp8720_buck_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.enable_reg = LP8720_ENABLE,
  		.enable_mask = LP8720_EN_BUCK_M,
  	},
  };
  
  static struct regulator_desc lp8725_regulator_desc[] = {
  	{
  		.name = "ldo1",

  		.of_match = of_match_ptr("ldo1"),

  		.id = LP8725_ID_LDO1,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO1_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8725_LDO_CTRL,
  		.enable_mask = LP872X_EN_LDO1_M,
  	},
  	{
  		.name = "ldo2",

  		.of_match = of_match_ptr("ldo2"),

  		.id = LP8725_ID_LDO2,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO2_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8725_LDO_CTRL,
  		.enable_mask = LP872X_EN_LDO2_M,
  	},
  	{
  		.name = "ldo3",

  		.of_match = of_match_ptr("ldo3"),

  		.id = LP8725_ID_LDO3,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO3_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8725_LDO_CTRL,
  		.enable_mask = LP872X_EN_LDO3_M,
  	},
  	{
  		.name = "ldo4",

  		.of_match = of_match_ptr("ldo4"),

  		.id = LP8725_ID_LDO4,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO4_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8725_LDO_CTRL,
  		.enable_mask = LP872X_EN_LDO4_M,
  	},
  	{
  		.name = "ldo5",

  		.of_match = of_match_ptr("ldo5"),

  		.id = LP8725_ID_LDO5,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp872x_ldo_vtbl),
  		.volt_table = lp872x_ldo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP872X_LDO5_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8725_LDO_CTRL,
  		.enable_mask = LP872X_EN_LDO5_M,
  	},
  	{
  		.name = "lilo1",

  		.of_match = of_match_ptr("lilo1"),

  		.id = LP8725_ID_LILO1,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp8725_lilo_vtbl),
  		.volt_table = lp8725_lilo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP8725_LILO1_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8725_LDO_CTRL,
  		.enable_mask = LP8725_EN_LILO1_M,
  	},
  	{
  		.name = "lilo2",

  		.of_match = of_match_ptr("lilo2"),

  		.id = LP8725_ID_LILO2,
  		.ops = &lp872x_ldo_ops,
  		.n_voltages = ARRAY_SIZE(lp8725_lilo_vtbl),
  		.volt_table = lp8725_lilo_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.vsel_reg = LP8725_LILO2_VOUT,
  		.vsel_mask = LP872X_VOUT_M,
  		.enable_reg = LP8725_LDO_CTRL,
  		.enable_mask = LP8725_EN_LILO2_M,
  	},
  	{
  		.name = "buck1",

  		.of_match = of_match_ptr("buck1"),

  		.id = LP8725_ID_BUCK1,
  		.ops = &lp8725_buck_ops,
  		.n_voltages = ARRAY_SIZE(lp8725_buck_vtbl),
  		.volt_table = lp8725_buck_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.enable_reg = LP872X_GENERAL_CFG,
  		.enable_mask = LP8725_BUCK1_EN_M,
  	},
  	{
  		.name = "buck2",

  		.of_match = of_match_ptr("buck2"),

  		.id = LP8725_ID_BUCK2,
  		.ops = &lp8725_buck_ops,
  		.n_voltages = ARRAY_SIZE(lp8725_buck_vtbl),
  		.volt_table = lp8725_buck_vtbl,
  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  		.enable_reg = LP872X_GENERAL_CFG,
  		.enable_mask = LP8725_BUCK2_EN_M,
  	},
  };

  static int lp872x_init_dvs(struct lp872x *lp)
  {
  	int ret, gpio;

  	struct lp872x_dvs *dvs = lp->pdata ? lp->pdata->dvs : NULL;

  	enum lp872x_dvs_state pinstate;

  	u8 mask[] = { LP8720_EXT_DVS_M, LP8725_DVS1_M | LP8725_DVS2_M };
  	u8 default_dvs_mode[] = { LP8720_DEFAULT_DVS, LP8725_DEFAULT_DVS };
  
  	if (!dvs)
  		goto set_default_dvs_mode;

  	gpio = dvs->gpio;

  	if (!gpio_is_valid(gpio))

  		goto set_default_dvs_mode;

  
  	pinstate = dvs->init_state;
  	ret = devm_gpio_request_one(lp->dev, gpio, pinstate, "LP872X DVS");
  	if (ret) {
  		dev_err(lp->dev, "gpio request err: %d
  ", ret);
  		return ret;
  	}
  
  	lp->dvs_pin = pinstate;

  
  	return 0;

  
  set_default_dvs_mode:
  	return lp872x_update_bits(lp, LP872X_GENERAL_CFG, mask[lp->chipid],
  				default_dvs_mode[lp->chipid]);

  }

  static int lp872x_hw_enable(struct lp872x *lp)
  {
  	int ret, gpio;
  
  	if (!lp->pdata)
  		return -EINVAL;
  
  	gpio = lp->pdata->enable_gpio;
  	if (!gpio_is_valid(gpio))
  		return 0;
  
  	/* Always set enable GPIO high. */
  	ret = devm_gpio_request_one(lp->dev, gpio, GPIOF_OUT_INIT_HIGH, "LP872X EN");
  	if (ret) {
  		dev_err(lp->dev, "gpio request err: %d
  ", ret);
  		return ret;
  	}
  
  	/* Each chip has a different enable delay. */
  	if (lp->chipid == LP8720)
  		usleep_range(LP8720_ENABLE_DELAY, 1.5 * LP8720_ENABLE_DELAY);
  	else
  		usleep_range(LP8725_ENABLE_DELAY, 1.5 * LP8725_ENABLE_DELAY);
  
  	return 0;
  }

  static int lp872x_config(struct lp872x *lp)
  {
  	struct lp872x_platform_data *pdata = lp->pdata;
  	int ret;

  	if (!pdata || !pdata->update_config)
  		goto init_dvs;

  
  	ret = lp872x_write_byte(lp, LP872X_GENERAL_CFG, pdata->general_config);
  	if (ret)
  		return ret;

  init_dvs:

  	return lp872x_init_dvs(lp);
  }
  
  static struct regulator_init_data

  *lp872x_find_regulator_init_data(int id, struct lp872x *lp)

  {

  	struct lp872x_platform_data *pdata = lp->pdata;

  	int i;

  	if (!pdata)
  		return NULL;

  	for (i = 0; i < lp->num_regulators; i++) {

  		if (pdata->regulator_data[i].id == id)
  			return pdata->regulator_data[i].init_data;

  	}

  	return NULL;

  }
  
  static int lp872x_regulator_register(struct lp872x *lp)
  {
  	struct regulator_desc *desc;
  	struct regulator_config cfg = { };
  	struct regulator_dev *rdev;

  	int i;

  	for (i = 0; i < lp->num_regulators; i++) {

  		desc = (lp->chipid == LP8720) ? &lp8720_regulator_desc[i] :
  						&lp8725_regulator_desc[i];
  
  		cfg.dev = lp->dev;

  		cfg.init_data = lp872x_find_regulator_init_data(desc->id, lp);

  		cfg.driver_data = lp;
  		cfg.regmap = lp->regmap;

  		rdev = devm_regulator_register(lp->dev, desc, &cfg);

  		if (IS_ERR(rdev)) {
  			dev_err(lp->dev, "regulator register err");

  			return PTR_ERR(rdev);

  		}

  	}
  
  	return 0;

  }
  
  static const struct regmap_config lp872x_regmap_config = {
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = MAX_REGISTERS,
  };

  #ifdef CONFIG_OF
  
  #define LP872X_VALID_OPMODE	(REGULATOR_MODE_FAST | REGULATOR_MODE_NORMAL)
  
  static struct of_regulator_match lp8720_matches[] = {
  	{ .name = "ldo1", .driver_data = (void *)LP8720_ID_LDO1, },
  	{ .name = "ldo2", .driver_data = (void *)LP8720_ID_LDO2, },
  	{ .name = "ldo3", .driver_data = (void *)LP8720_ID_LDO3, },
  	{ .name = "ldo4", .driver_data = (void *)LP8720_ID_LDO4, },
  	{ .name = "ldo5", .driver_data = (void *)LP8720_ID_LDO5, },
  	{ .name = "buck", .driver_data = (void *)LP8720_ID_BUCK, },
  };
  
  static struct of_regulator_match lp8725_matches[] = {
  	{ .name = "ldo1", .driver_data = (void *)LP8725_ID_LDO1, },
  	{ .name = "ldo2", .driver_data = (void *)LP8725_ID_LDO2, },
  	{ .name = "ldo3", .driver_data = (void *)LP8725_ID_LDO3, },
  	{ .name = "ldo4", .driver_data = (void *)LP8725_ID_LDO4, },
  	{ .name = "ldo5", .driver_data = (void *)LP8725_ID_LDO5, },
  	{ .name = "lilo1", .driver_data = (void *)LP8725_ID_LILO1, },
  	{ .name = "lilo2", .driver_data = (void *)LP8725_ID_LILO2, },
  	{ .name = "buck1", .driver_data = (void *)LP8725_ID_BUCK1, },
  	{ .name = "buck2", .driver_data = (void *)LP8725_ID_BUCK2, },
  };
  
  static struct lp872x_platform_data
  *lp872x_populate_pdata_from_dt(struct device *dev, enum lp872x_id which)
  {
  	struct device_node *np = dev->of_node;
  	struct lp872x_platform_data *pdata;
  	struct of_regulator_match *match;

  	int num_matches;
  	int count;
  	int i;
  	u8 dvs_state;
  
  	pdata = devm_kzalloc(dev, sizeof(*pdata), GFP_KERNEL);
  	if (!pdata)

  		return ERR_PTR(-ENOMEM);

  
  	of_property_read_u8(np, "ti,general-config", &pdata->general_config);
  	if (of_find_property(np, "ti,update-config", NULL))
  		pdata->update_config = true;
  
  	pdata->dvs = devm_kzalloc(dev, sizeof(struct lp872x_dvs), GFP_KERNEL);
  	if (!pdata->dvs)

  		return ERR_PTR(-ENOMEM);

  
  	pdata->dvs->gpio = of_get_named_gpio(np, "ti,dvs-gpio", 0);
  	of_property_read_u8(np, "ti,dvs-vsel", (u8 *)&pdata->dvs->vsel);
  	of_property_read_u8(np, "ti,dvs-state", &dvs_state);
  	pdata->dvs->init_state = dvs_state ? DVS_HIGH : DVS_LOW;

  	pdata->enable_gpio = of_get_named_gpio(np, "enable-gpios", 0);

  	if (of_get_child_count(np) == 0)
  		goto out;
  
  	switch (which) {
  	case LP8720:
  		match = lp8720_matches;
  		num_matches = ARRAY_SIZE(lp8720_matches);
  		break;
  	case LP8725:
  		match = lp8725_matches;
  		num_matches = ARRAY_SIZE(lp8725_matches);
  		break;
  	default:
  		goto out;
  	}
  
  	count = of_regulator_match(dev, np, match, num_matches);
  	if (count <= 0)
  		goto out;
  
  	for (i = 0; i < num_matches; i++) {

  		pdata->regulator_data[i].id =
  				(enum lp872x_regulator_id)match[i].driver_data;

  		pdata->regulator_data[i].init_data = match[i].init_data;

  	}
  out:
  	return pdata;
  }
  #else
  static struct lp872x_platform_data
  *lp872x_populate_pdata_from_dt(struct device *dev, enum lp872x_id which)
  {
  	return NULL;
  }
  #endif

  static int lp872x_probe(struct i2c_client *cl, const struct i2c_device_id *id)
  {
  	struct lp872x *lp;

  	struct lp872x_platform_data *pdata;

  	int ret;

  	const int lp872x_num_regulators[] = {
  		[LP8720] = LP8720_NUM_REGULATORS,
  		[LP8725] = LP8725_NUM_REGULATORS,
  	};

  	if (cl->dev.of_node) {

  		pdata = lp872x_populate_pdata_from_dt(&cl->dev,

  					      (enum lp872x_id)id->driver_data);

  		if (IS_ERR(pdata))
  			return PTR_ERR(pdata);
  	} else {

  		pdata = dev_get_platdata(&cl->dev);

  	}

  	lp = devm_kzalloc(&cl->dev, sizeof(struct lp872x), GFP_KERNEL);
  	if (!lp)

  		return -ENOMEM;

  	lp->num_regulators = lp872x_num_regulators[id->driver_data];

  
  	lp->regmap = devm_regmap_init_i2c(cl, &lp872x_regmap_config);
  	if (IS_ERR(lp->regmap)) {
  		ret = PTR_ERR(lp->regmap);
  		dev_err(&cl->dev, "regmap init i2c err: %d
  ", ret);

  		return ret;

  	}
  
  	lp->dev = &cl->dev;

  	lp->pdata = pdata;

  	lp->chipid = id->driver_data;

  	i2c_set_clientdata(cl, lp);

  	ret = lp872x_hw_enable(lp);
  	if (ret)
  		return ret;

  	ret = lp872x_config(lp);
  	if (ret)

  		return ret;

  
  	return lp872x_regulator_register(lp);

  }

  static const struct of_device_id lp872x_dt_ids[] = {
  	{ .compatible = "ti,lp8720", },
  	{ .compatible = "ti,lp8725", },
  	{ }
  };
  MODULE_DEVICE_TABLE(of, lp872x_dt_ids);

  static const struct i2c_device_id lp872x_ids[] = {
  	{"lp8720", LP8720},
  	{"lp8725", LP8725},
  	{ }
  };
  MODULE_DEVICE_TABLE(i2c, lp872x_ids);
  
  static struct i2c_driver lp872x_driver = {
  	.driver = {
  		.name = "lp872x",

  		.of_match_table = of_match_ptr(lp872x_dt_ids),

  	},
  	.probe = lp872x_probe,

  	.id_table = lp872x_ids,
  };
  
  module_i2c_driver(lp872x_driver);
  
  MODULE_DESCRIPTION("TI/National Semiconductor LP872x PMU Regulator Driver");
  MODULE_AUTHOR("Milo Kim");
  MODULE_LICENSE("GPL");