// SPDX-License-Identifier: GPL-2.0+
  //
  // Regulator support for WM8400
  //
  // Copyright 2008 Wolfson Microelectronics PLC.
  //
  // Author: Mark Brown <broonie@opensource.wolfsonmicro.com>

  
  #include <linux/bug.h>
  #include <linux/err.h>
  #include <linux/kernel.h>

  #include <linux/module.h>

  #include <linux/regulator/driver.h>
  #include <linux/mfd/wm8400-private.h>

  static const struct linear_range wm8400_ldo_ranges[] = {

  	REGULATOR_LINEAR_RANGE(900000, 0, 14, 50000),
  	REGULATOR_LINEAR_RANGE(1700000, 15, 31, 100000),

  };

  static const struct regulator_ops wm8400_ldo_ops = {

  	.is_enabled = regulator_is_enabled_regmap,
  	.enable = regulator_enable_regmap,
  	.disable = regulator_disable_regmap,

  	.list_voltage = regulator_list_voltage_linear_range,

  	.get_voltage_sel = regulator_get_voltage_sel_regmap,
  	.set_voltage_sel = regulator_set_voltage_sel_regmap,

  	.map_voltage = regulator_map_voltage_linear_range,

  };

  static unsigned int wm8400_dcdc_get_mode(struct regulator_dev *dev)
  {

  	struct regmap *rmap = rdev_get_regmap(dev);

  	int offset = (rdev_get_id(dev) - WM8400_DCDC1) * 2;
  	u16 data[2];
  	int ret;

  	ret = regmap_bulk_read(rmap, WM8400_DCDC1_CONTROL_1 + offset, data, 2);

  	if (ret != 0)
  		return 0;
  
  	/* Datasheet: hibernate */
  	if (data[0] & WM8400_DC1_SLEEP)
  		return REGULATOR_MODE_STANDBY;
  
  	/* Datasheet: standby */
  	if (!(data[0] & WM8400_DC1_ACTIVE))
  		return REGULATOR_MODE_IDLE;
  
  	/* Datasheet: active with or without force PWM */
  	if (data[1] & WM8400_DC1_FRC_PWM)
  		return REGULATOR_MODE_FAST;
  	else
  		return REGULATOR_MODE_NORMAL;
  }
  
  static int wm8400_dcdc_set_mode(struct regulator_dev *dev, unsigned int mode)
  {

  	struct regmap *rmap = rdev_get_regmap(dev);

  	int offset = (rdev_get_id(dev) - WM8400_DCDC1) * 2;
  	int ret;
  
  	switch (mode) {
  	case REGULATOR_MODE_FAST:
  		/* Datasheet: active with force PWM */

  		ret = regmap_update_bits(rmap, WM8400_DCDC1_CONTROL_2 + offset,

  				      WM8400_DC1_FRC_PWM, WM8400_DC1_FRC_PWM);
  		if (ret != 0)
  			return ret;

  		return regmap_update_bits(rmap, WM8400_DCDC1_CONTROL_1 + offset,

  				       WM8400_DC1_ACTIVE | WM8400_DC1_SLEEP,
  				       WM8400_DC1_ACTIVE);
  
  	case REGULATOR_MODE_NORMAL:
  		/* Datasheet: active */

  		ret = regmap_update_bits(rmap, WM8400_DCDC1_CONTROL_2 + offset,

  				      WM8400_DC1_FRC_PWM, 0);
  		if (ret != 0)
  			return ret;

  		return regmap_update_bits(rmap, WM8400_DCDC1_CONTROL_1 + offset,

  				       WM8400_DC1_ACTIVE | WM8400_DC1_SLEEP,
  				       WM8400_DC1_ACTIVE);
  
  	case REGULATOR_MODE_IDLE:
  		/* Datasheet: standby */

  		return regmap_update_bits(rmap, WM8400_DCDC1_CONTROL_1 + offset,

  				       WM8400_DC1_ACTIVE | WM8400_DC1_SLEEP, 0);

  	default:
  		return -EINVAL;
  	}
  }
  
  static unsigned int wm8400_dcdc_get_optimum_mode(struct regulator_dev *dev,
  						 int input_uV, int output_uV,
  						 int load_uA)
  {
  	return REGULATOR_MODE_NORMAL;
  }

  static const struct regulator_ops wm8400_dcdc_ops = {

  	.is_enabled = regulator_is_enabled_regmap,
  	.enable = regulator_enable_regmap,
  	.disable = regulator_disable_regmap,
  	.list_voltage = regulator_list_voltage_linear,

  	.map_voltage = regulator_map_voltage_linear,

  	.get_voltage_sel = regulator_get_voltage_sel_regmap,
  	.set_voltage_sel = regulator_set_voltage_sel_regmap,

  	.get_mode = wm8400_dcdc_get_mode,
  	.set_mode = wm8400_dcdc_set_mode,
  	.get_optimum_mode = wm8400_dcdc_get_optimum_mode,
  };
  
  static struct regulator_desc regulators[] = {
  	{
  		.name = "LDO1",
  		.id = WM8400_LDO1,
  		.ops = &wm8400_ldo_ops,

  		.enable_reg = WM8400_LDO1_CONTROL,
  		.enable_mask = WM8400_LDO1_ENA,

  		.n_voltages = WM8400_LDO1_VSEL_MASK + 1,

  		.linear_ranges = wm8400_ldo_ranges,
  		.n_linear_ranges = ARRAY_SIZE(wm8400_ldo_ranges),

  		.vsel_reg = WM8400_LDO1_CONTROL,
  		.vsel_mask = WM8400_LDO1_VSEL_MASK,

  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  	},
  	{
  		.name = "LDO2",
  		.id = WM8400_LDO2,
  		.ops = &wm8400_ldo_ops,

  		.enable_reg = WM8400_LDO2_CONTROL,
  		.enable_mask = WM8400_LDO2_ENA,

  		.n_voltages = WM8400_LDO2_VSEL_MASK + 1,

  		.linear_ranges = wm8400_ldo_ranges,
  		.n_linear_ranges = ARRAY_SIZE(wm8400_ldo_ranges),

  		.type = REGULATOR_VOLTAGE,

  		.vsel_reg = WM8400_LDO2_CONTROL,
  		.vsel_mask = WM8400_LDO2_VSEL_MASK,

  		.owner = THIS_MODULE,
  	},
  	{
  		.name = "LDO3",
  		.id = WM8400_LDO3,
  		.ops = &wm8400_ldo_ops,

  		.enable_reg = WM8400_LDO3_CONTROL,
  		.enable_mask = WM8400_LDO3_ENA,

  		.n_voltages = WM8400_LDO3_VSEL_MASK + 1,

  		.linear_ranges = wm8400_ldo_ranges,
  		.n_linear_ranges = ARRAY_SIZE(wm8400_ldo_ranges),

  		.vsel_reg = WM8400_LDO3_CONTROL,
  		.vsel_mask = WM8400_LDO3_VSEL_MASK,

  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  	},
  	{
  		.name = "LDO4",
  		.id = WM8400_LDO4,
  		.ops = &wm8400_ldo_ops,

  		.enable_reg = WM8400_LDO4_CONTROL,
  		.enable_mask = WM8400_LDO4_ENA,

  		.n_voltages = WM8400_LDO4_VSEL_MASK + 1,

  		.linear_ranges = wm8400_ldo_ranges,
  		.n_linear_ranges = ARRAY_SIZE(wm8400_ldo_ranges),

  		.vsel_reg = WM8400_LDO4_CONTROL,
  		.vsel_mask = WM8400_LDO4_VSEL_MASK,

  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  	},
  	{
  		.name = "DCDC1",
  		.id = WM8400_DCDC1,
  		.ops = &wm8400_dcdc_ops,

  		.enable_reg = WM8400_DCDC1_CONTROL_1,
  		.enable_mask = WM8400_DC1_ENA_MASK,

  		.n_voltages = WM8400_DC1_VSEL_MASK + 1,

  		.vsel_reg = WM8400_DCDC1_CONTROL_1,
  		.vsel_mask = WM8400_DC1_VSEL_MASK,
  		.min_uV = 850000,
  		.uV_step = 25000,

  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  	},
  	{
  		.name = "DCDC2",
  		.id = WM8400_DCDC2,
  		.ops = &wm8400_dcdc_ops,

  		.enable_reg = WM8400_DCDC2_CONTROL_1,

  		.enable_mask = WM8400_DC2_ENA_MASK,

  		.n_voltages = WM8400_DC2_VSEL_MASK + 1,

  		.vsel_reg = WM8400_DCDC2_CONTROL_1,
  		.vsel_mask = WM8400_DC2_VSEL_MASK,
  		.min_uV = 850000,
  		.uV_step = 25000,

  		.type = REGULATOR_VOLTAGE,
  		.owner = THIS_MODULE,
  	},
  };

  static int wm8400_regulator_probe(struct platform_device *pdev)

  {

  	struct wm8400 *wm8400 = container_of(pdev, struct wm8400, regulators[pdev->id]);

  	struct regulator_config config = { };

  	struct regulator_dev *rdev;

  	config.dev = &pdev->dev;

  	config.init_data = dev_get_platdata(&pdev->dev);

  	config.driver_data = wm8400;

  	config.regmap = wm8400->regmap;

  	rdev = devm_regulator_register(&pdev->dev, &regulators[pdev->id],
  				       &config);

  	if (IS_ERR(rdev))
  		return PTR_ERR(rdev);

  	platform_set_drvdata(pdev, rdev);

  	return 0;
  }

  static struct platform_driver wm8400_regulator_driver = {
  	.driver = {
  		.name = "wm8400-regulator",
  	},
  	.probe = wm8400_regulator_probe,

  };
  
  /**
   * wm8400_register_regulator - enable software control of a WM8400 regulator
   *
   * This function enables software control of a WM8400 regulator via
   * the regulator API.  It is intended to be called from the
   * platform_init() callback of the WM8400 MFD driver.
   *

   * @dev:      The WM8400 device to operate on.
   * @reg:      The regulator to control.
   * @initdata: Regulator initdata for the regulator.

   */
  int wm8400_register_regulator(struct device *dev, int reg,
  			      struct regulator_init_data *initdata)
  {

  	struct wm8400 *wm8400 = dev_get_drvdata(dev);

  
  	if (wm8400->regulators[reg].name)
  		return -EBUSY;
  
  	initdata->driver_data = wm8400;
  
  	wm8400->regulators[reg].name = "wm8400-regulator";
  	wm8400->regulators[reg].id = reg;
  	wm8400->regulators[reg].dev.parent = dev;

  	wm8400->regulators[reg].dev.platform_data = initdata;
  
  	return platform_device_register(&wm8400->regulators[reg]);
  }
  EXPORT_SYMBOL_GPL(wm8400_register_regulator);
  
  static int __init wm8400_regulator_init(void)
  {
  	return platform_driver_register(&wm8400_regulator_driver);
  }

  subsys_initcall(wm8400_regulator_init);

  
  static void __exit wm8400_regulator_exit(void)
  {
  	platform_driver_unregister(&wm8400_regulator_driver);
  }
  module_exit(wm8400_regulator_exit);
  
  MODULE_AUTHOR("Mark Brown <broonie@opensource.wolfsonmicro.com>");
  MODULE_DESCRIPTION("WM8400 regulator driver");
  MODULE_LICENSE("GPL");
  MODULE_ALIAS("platform:wm8400-regulator");