// SPDX-License-Identifier: GPL-2.0-only
/*
 * Device driver for Hi6421 PMIC
 *
 * Copyright (c) <2011-2014> HiSilicon Technologies Co., Ltd.
 *              http://www.hisilicon.com
 * Copyright (c) <2013-2017> Linaro Ltd.
 *              https://www.linaro.org
 *
 * Author: Guodong Xu <guodong.xu@linaro.org>
 */

#include <linux/device.h>
#include <linux/err.h>
#include <linux/mfd/core.h>
#include <linux/mfd/hi6421-pmic.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/platform_device.h>
#include <linux/property.h>
#include <linux/regmap.h>

static const struct mfd_cell hi6421_devs[] = {
 { .name = "hi6421-regulator", },
};

static const struct mfd_cell hi6421v530_devs[] = {
 { .name = "hi6421v530-regulator", },
};

static const struct regmap_config hi6421_regmap_config = {
 .reg_bits = 32,
 .reg_stride = 4,
 .val_bits = 8,
 .max_register = HI6421_REG_TO_BUS_ADDR(HI6421_REG_MAX),
};

static const struct of_device_id of_hi6421_pmic_match[] = {
 {
  .compatible = "hisilicon,hi6421-pmic",
  .data = (void *)HI6421
 },
 {
  .compatible = "hisilicon,hi6421v530-pmic",
  .data = (void *)HI6421_V530
 },
 { },
};
MODULE_DEVICE_TABLE(of, of_hi6421_pmic_match);

static int hi6421_pmic_probe(struct platform_device *pdev)
{
 struct hi6421_pmic *pmic;
 const struct mfd_cell *subdevs;
 enum hi6421_type type;
 void __iomem *base;
 int n_subdevs, ret;

 type = (uintptr_t)device_get_match_data(&pdev->dev);

 pmic = devm_kzalloc(&pdev->dev, sizeof(*pmic), GFP_KERNEL);
 if (!pmic)
  return -ENOMEM;

 base = devm_platform_get_and_ioremap_resource(pdev, 0, NULL);
 if (IS_ERR(base))
  return PTR_ERR(base);

 pmic->regmap = devm_regmap_init_mmio_clk(&pdev->dev, NULL, base,
       &hi6421_regmap_config);
 if (IS_ERR(pmic->regmap)) {
  dev_err(&pdev->dev, "Failed to initialise Regmap: %ld\n",
      PTR_ERR(pmic->regmap));
  return PTR_ERR(pmic->regmap);
 }

 platform_set_drvdata(pdev, pmic);

 switch (type) {
 case HI6421:
  /* set over-current protection debounce 8ms */
  regmap_update_bits(pmic->regmap, HI6421_OCP_DEB_CTRL_REG,
    (HI6421_OCP_DEB_SEL_MASK
     | HI6421_OCP_EN_DEBOUNCE_MASK
     | HI6421_OCP_AUTO_STOP_MASK),
    (HI6421_OCP_DEB_SEL_8MS
     | HI6421_OCP_EN_DEBOUNCE_ENABLE));

  subdevs = hi6421_devs;
  n_subdevs = ARRAY_SIZE(hi6421_devs);
  break;
 case HI6421_V530:
  subdevs = hi6421v530_devs;
  n_subdevs = ARRAY_SIZE(hi6421v530_devs);
  break;
 default:
  dev_err(&pdev->dev, "Unknown device type %d\n",
      (unsigned int)type);
  return -EINVAL;
 }

 ret = devm_mfd_add_devices(&pdev->dev, PLATFORM_DEVID_NONE,
       subdevs, n_subdevs, NULL, 0, NULL);
 if (ret) {
  dev_err(&pdev->dev, "Failed to add child devices: %d\n", ret);
  return ret;
 }

 return 0;
}

static struct platform_driver hi6421_pmic_driver = {
 .driver = {
  .name = "hi6421_pmic",
  .of_match_table = of_hi6421_pmic_match,
 },
 .probe = hi6421_pmic_probe,
};
module_platform_driver(hi6421_pmic_driver);

MODULE_AUTHOR("Guodong Xu <guodong.xu@linaro.org>");
MODULE_DESCRIPTION("Hi6421 PMIC driver");
MODULE_LICENSE("GPL v2");