// SPDX-License-Identifier: GPL-2.0-only

#include <linux/mfd/sm5703.h>
#include <linux/module.h>
#include <linux/mod_devicetable.h>
#include <linux/platform_device.h>
#include <linux/regmap.h>
#include <linux/regulator/driver.h>
#include <linux/regulator/of_regulator.h>

enum sm5703_regulators {
	SM5703_BUCK,
	SM5703_LDO1,
	SM5703_LDO2,
	SM5703_LDO3,
	SM5703_USBLDO1,
	SM5703_USBLDO2,
	SM5703_VBUS,
	SM5703_MAX_REGULATORS,
};

static const int sm5703_ldo_voltagemap[] = {
	1500000, 1800000, 2600000, 2800000, 3000000, 3300000,
};

static const int sm5703_buck_voltagemap[] = {
	1000000, 1000000, 1000000, 1000000,
	1000000, 1000000, 1000000, 1000000,
	1000000, 1000000, 1000000, 1100000,
	1200000, 1300000, 1400000, 1500000,
	1600000, 1700000, 1800000, 1900000,
	2000000, 2100000, 2200000, 2300000,
	2400000, 2500000, 2600000, 2700000,
	2800000, 2900000, 3000000, 3000000,
};

#define SM5703USBLDO(_name, _id)					\
	[SM5703_USBLDO ## _id] = {					\
		.name = _name,						\
		.of_match = _name,					\
		.regulators_node = "regulators",			\
		.type = REGULATOR_VOLTAGE,				\
		.id = SM5703_USBLDO ## _id,				\
		.ops = &sm5703_regulator_ops_fixed,			\
		.fixed_uV = SM5703_USBLDO_MICROVOLT,			\
		.enable_reg = SM5703_REG_USBLDO12,			\
		.enable_mask = SM5703_REG_EN_USBLDO ##_id,		\
		.owner			= THIS_MODULE,			\
	}

#define SM5703VBUS(_name)						\
	[SM5703_VBUS] = {						\
		.name = _name,						\
		.of_match = _name,					\
		.regulators_node = "regulators",			\
		.type = REGULATOR_VOLTAGE,				\
		.id = SM5703_VBUS,					\
		.ops = &sm5703_regulator_ops_fixed,			\
		.fixed_uV = SM5703_VBUS_MICROVOLT,			\
		.enable_reg = SM5703_REG_CNTL,				\
		.enable_mask = SM5703_OPERATION_MODE_MASK,		\
		.enable_val = SM5703_OPERATION_MODE_USB_OTG_MODE,	\
		.disable_val = SM5703_OPERATION_MODE_CHARGING_ON,	\
		.owner			= THIS_MODULE,			\
	}

#define SM5703BUCK(_name)						\
	[SM5703_BUCK] = {						\
		.name = _name,						\
		.of_match = _name,					\
		.regulators_node = "regulators",			\
		.type = REGULATOR_VOLTAGE,				\
		.id = SM5703_BUCK,					\
		.ops = &sm5703_regulator_ops,				\
		.n_voltages = ARRAY_SIZE(sm5703_buck_voltagemap),	\
		.volt_table = sm5703_buck_voltagemap,			\
		.vsel_reg = SM5703_REG_BUCK,				\
		.vsel_mask = SM5703_BUCK_VOLT_MASK,			\
		.enable_reg = SM5703_REG_BUCK,				\
		.enable_mask = SM5703_REG_EN_BUCK,			\
		.owner			= THIS_MODULE,			\
	}

#define SM5703LDO(_name, _id)						\
	[SM5703_LDO ## _id] = {						\
		.name = _name,						\
		.of_match = _name,					\
		.regulators_node = "regulators",			\
		.type = REGULATOR_VOLTAGE,				\
		.id = SM5703_LDO ## _id,				\
		.ops = &sm5703_regulator_ops,				\
		.n_voltages = ARRAY_SIZE(sm5703_ldo_voltagemap),	\
		.volt_table = sm5703_ldo_voltagemap,			\
		.vsel_reg = SM5703_REG_LDO ##_id,			\
		.vsel_mask = SM5703_LDO_VOLT_MASK,			\
		.enable_reg = SM5703_REG_LDO ##_id,			\
		.enable_mask = SM5703_LDO_EN,				\
		.owner			= THIS_MODULE,			\
	}

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

static const struct regulator_ops sm5703_regulator_ops_fixed = {
	.enable			= regulator_enable_regmap,
	.disable		= regulator_disable_regmap,
	.is_enabled		= regulator_is_enabled_regmap,
};

static struct regulator_desc sm5703_regulators_desc[SM5703_MAX_REGULATORS] = {
	SM5703BUCK("buck"),
	SM5703LDO("ldo1", 1),
	SM5703LDO("ldo2", 2),
	SM5703LDO("ldo3", 3),
	SM5703USBLDO("usbldo1", 1),
	SM5703USBLDO("usbldo2", 2),
	SM5703VBUS("vbus"),
};

static int sm5703_regulator_probe(struct platform_device *pdev)
{
	struct device *dev = &pdev->dev;
	struct regulator_config config = { NULL, };
	struct regulator_dev *rdev;
	struct sm5703_dev *sm5703 = dev_get_drvdata(pdev->dev.parent);
	int i;

	config.dev = dev->parent;
	config.regmap = sm5703->regmap;

	for (i = 0; i < SM5703_MAX_REGULATORS; i++) {
		rdev = devm_regulator_register(dev,
					       &sm5703_regulators_desc[i],
					       &config);
		if (IS_ERR(rdev))
			return dev_err_probe(dev, PTR_ERR(rdev),
					     "Failed to register a regulator\n");
	}

	return 0;
}

static const struct platform_device_id sm5703_regulator_id[] = {
	{ "sm5703-regulator", 0 },
	{}
};
MODULE_DEVICE_TABLE(platform, sm5703_regulator_id);

static struct platform_driver sm5703_regulator_driver = {
	.driver = {
		.name = "sm5703-regulator",
	},
	.probe	= sm5703_regulator_probe,
	.id_table	= sm5703_regulator_id,
};

module_platform_driver(sm5703_regulator_driver);

MODULE_DESCRIPTION("Silicon Mitus SM5703 LDO/Buck/USB regulator driver");
MODULE_AUTHOR("Markuss Broks <markuss.broks@gmail.com>");
MODULE_LICENSE("GPL");