// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * lm3533-core.c -- LM3533 Core
 *
 * Copyright (C) 2011-2012 Texas Instruments
 *
 * Author: Johan Hovold <jhovold@gmail.com>
 */

#include <linux/bitfield.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/err.h>
#include <linux/gpio/consumer.h>
#include <linux/i2c.h>
#include <linux/mfd/core.h>
#include <linux/property.h>
#include <linux/regmap.h>
#include <linux/regulator/consumer.h>
#include <linux/seq_file.h>
#include <linux/slab.h>
#include <linux/uaccess.h>
#include <linux/units.h>

#include <linux/mfd/lm3533.h>


#define LM3533_BOOST_OVP_MASK		0x06
#define LM3533_BOOST_OVP_MIN		(16 * MICRO)
#define LM3533_BOOST_OVP_MAX		(40 * MICRO)

#define LM3533_BOOST_FREQ_MASK		0x01
#define LM3533_BOOST_FREQ_MIN		(500 * HZ_PER_KHZ)
#define LM3533_BOOST_FREQ_MAX		(1000 * HZ_PER_KHZ)

#define LM3533_BL_ID_MASK		1
#define LM3533_LED_ID_MASK		3
#define LM3533_BL_ID_MAX		1
#define LM3533_LED_ID_MAX		3

#define LM3533_HVLED_ID_MAX		2
#define LM3533_LVLED_ID_MAX		5
#define LM3533_CELLS_MAX		7

#define LM3533_REG_OUTPUT_CONF1		0x10
#define LM3533_REG_OUTPUT_CONF2		0x11
#define LM3533_REG_BOOST_PWM		0x2c

#define LM3533_REG_MAX			0xb2

/*
 * HVLED output config -- output hvled controlled by backlight bl
 */
static int lm3533_set_hvled_config(struct lm3533 *lm3533, u8 hvled, u8 bl)
{
	u8 val;
	u8 mask;
	int shift;
	int ret;

	if (hvled == 0 || hvled > LM3533_HVLED_ID_MAX)
		return -EINVAL;

	if (bl > LM3533_BL_ID_MAX)
		return -EINVAL;

	shift = hvled - 1;
	mask = LM3533_BL_ID_MASK << shift;
	val = bl << shift;

	ret = regmap_update_bits(lm3533->regmap, LM3533_REG_OUTPUT_CONF1,
				 mask, val);
	if (ret)
		dev_err(lm3533->dev, "failed to set hvled config\n");

	return ret;
}

/*
 * LVLED output config -- output lvled controlled by LED led
 */
static int lm3533_set_lvled_config(struct lm3533 *lm3533, u8 lvled, u8 led)
{
	u8 reg;
	u8 val;
	u8 mask;
	int shift;
	int ret;

	if (lvled == 0 || lvled > LM3533_LVLED_ID_MAX)
		return -EINVAL;

	if (led > LM3533_LED_ID_MAX)
		return -EINVAL;

	if (lvled < 4) {
		reg = LM3533_REG_OUTPUT_CONF1;
		shift = 2 * lvled;
	} else {
		reg = LM3533_REG_OUTPUT_CONF2;
		shift = 2 * (lvled - 4);
	}

	mask = LM3533_LED_ID_MASK << shift;
	val = led << shift;

	ret = regmap_update_bits(lm3533->regmap, reg, mask, val);
	if (ret)
		dev_err(lm3533->dev, "failed to set lvled config\n");

	return ret;
}

static int lm3533_enable(struct lm3533 *lm3533)
{
	int ret;

	ret = regulator_enable(lm3533->vin_supply);
	if (ret) {
		dev_err(lm3533->dev, "failed to enable vin power supply\n");
		return ret;
	}

	gpiod_set_value(lm3533->hwen, 1);

	return 0;
}

static void lm3533_disable(struct lm3533 *lm3533)
{
	gpiod_set_value(lm3533->hwen, 0);
	regulator_disable(lm3533->vin_supply);
}

enum lm3533_attribute_type {
	LM3533_ATTR_TYPE_BACKLIGHT,
	LM3533_ATTR_TYPE_LED,
};

struct lm3533_device_attribute {
	struct device_attribute dev_attr;
	enum lm3533_attribute_type type;
	union {
		struct {
			u8 id;
		} output;
	} u;
};

#define to_lm3533_dev_attr(_attr) \
	container_of(_attr, struct lm3533_device_attribute, dev_attr)

static ssize_t show_output(struct device *dev,
				struct device_attribute *attr, char *buf)
{
	struct lm3533 *lm3533 = dev_get_drvdata(dev);
	struct lm3533_device_attribute *lattr = to_lm3533_dev_attr(attr);
	int id = lattr->u.output.id;
	u8 reg;
	u32 val;
	u8 mask;
	int shift;
	int ret;

	if (lattr->type == LM3533_ATTR_TYPE_BACKLIGHT) {
		reg = LM3533_REG_OUTPUT_CONF1;
		shift = id - 1;
		mask = LM3533_BL_ID_MASK << shift;
	} else {
		if (id < 4) {
			reg = LM3533_REG_OUTPUT_CONF1;
			shift = 2 * id;
		} else {
			reg = LM3533_REG_OUTPUT_CONF2;
			shift = 2 * (id - 4);
		}
		mask = LM3533_LED_ID_MASK << shift;
	}

	ret = regmap_read(lm3533->regmap, reg, &val);
	if (ret)
		return ret;

	val = (val & mask) >> shift;

	return sysfs_emit(buf, "%u\n", val);
}

static ssize_t store_output(struct device *dev,
					struct device_attribute *attr,
					const char *buf, size_t len)
{
	struct lm3533 *lm3533 = dev_get_drvdata(dev);
	struct lm3533_device_attribute *lattr = to_lm3533_dev_attr(attr);
	int id = lattr->u.output.id;
	u8 val;
	int ret;

	if (kstrtou8(buf, 0, &val))
		return -EINVAL;

	if (lattr->type == LM3533_ATTR_TYPE_BACKLIGHT)
		ret = lm3533_set_hvled_config(lm3533, id, val);
	else
		ret = lm3533_set_lvled_config(lm3533, id, val);

	if (ret)
		return ret;

	return len;
}

#define LM3533_OUTPUT_ATTR(_name, _mode, _show, _store, _type, _id) \
	struct lm3533_device_attribute lm3533_dev_attr_##_name = \
		{ .dev_attr	= __ATTR(_name, _mode, _show, _store), \
		  .type		= _type, \
		  .u.output	= { .id = _id }, }

#define LM3533_OUTPUT_ATTR_RW(_name, _type, _id) \
	LM3533_OUTPUT_ATTR(output_##_name, S_IRUGO | S_IWUSR, \
					show_output, store_output, _type, _id)

#define LM3533_OUTPUT_HVLED_ATTR_RW(_nr) \
	LM3533_OUTPUT_ATTR_RW(hvled##_nr, LM3533_ATTR_TYPE_BACKLIGHT, _nr)
#define LM3533_OUTPUT_LVLED_ATTR_RW(_nr) \
	LM3533_OUTPUT_ATTR_RW(lvled##_nr, LM3533_ATTR_TYPE_LED, _nr)
/*
 * Output config:
 *
 * output_hvled<nr>	0-1
 * output_lvled<nr>	0-3
 */
static LM3533_OUTPUT_HVLED_ATTR_RW(1);
static LM3533_OUTPUT_HVLED_ATTR_RW(2);
static LM3533_OUTPUT_LVLED_ATTR_RW(1);
static LM3533_OUTPUT_LVLED_ATTR_RW(2);
static LM3533_OUTPUT_LVLED_ATTR_RW(3);
static LM3533_OUTPUT_LVLED_ATTR_RW(4);
static LM3533_OUTPUT_LVLED_ATTR_RW(5);

static struct attribute *lm3533_attributes[] = {
	&lm3533_dev_attr_output_hvled1.dev_attr.attr,
	&lm3533_dev_attr_output_hvled2.dev_attr.attr,
	&lm3533_dev_attr_output_lvled1.dev_attr.attr,
	&lm3533_dev_attr_output_lvled2.dev_attr.attr,
	&lm3533_dev_attr_output_lvled3.dev_attr.attr,
	&lm3533_dev_attr_output_lvled4.dev_attr.attr,
	&lm3533_dev_attr_output_lvled5.dev_attr.attr,
	NULL,
};

#define to_dev_attr(_attr) \
	container_of(_attr, struct device_attribute, attr)

static umode_t lm3533_attr_is_visible(struct kobject *kobj,
					     struct attribute *attr, int n)
{
	struct device *dev = kobj_to_dev(kobj);
	struct lm3533 *lm3533 = dev_get_drvdata(dev);
	struct device_attribute *dattr = to_dev_attr(attr);
	struct lm3533_device_attribute *lattr = to_lm3533_dev_attr(dattr);
	enum lm3533_attribute_type type = lattr->type;
	umode_t mode = attr->mode;

	if (!lm3533->have_backlights && type == LM3533_ATTR_TYPE_BACKLIGHT)
		mode = 0;
	else if (!lm3533->have_leds && type == LM3533_ATTR_TYPE_LED)
		mode = 0;

	return mode;
};

static struct attribute_group lm3533_attribute_group = {
	.is_visible	= lm3533_attr_is_visible,
	.attrs		= lm3533_attributes
};

static const struct attribute_group *lm3533_attribute_groups[] = {
	&lm3533_attribute_group,
	NULL,
};

static int lm3533_device_init(struct lm3533 *lm3533)
{
	struct device *dev = lm3533->dev;
	struct mfd_cell *lm3533_devices;
	u32 reg, nchilds;
	u32 count = 0;
	int ret;

	nchilds = device_get_child_node_count(dev);
	if (!nchilds || nchilds > LM3533_CELLS_MAX)
		return dev_err_probe(dev, -ENODEV,
				     "num of child nodes is not supported\n");

	lm3533_devices = devm_kcalloc(dev, nchilds, sizeof(*lm3533_devices),
				      GFP_KERNEL);
	if (!lm3533_devices)
		return -ENOMEM;

	device_for_each_child_node_scoped(dev, child) {
		if (count >= nchilds)
			break;

		if (fwnode_device_is_compatible(child, "ti,lm3533-als")) {
			lm3533_devices[count].name = "lm3533-als";
			lm3533_devices[count].of_compatible = "ti,lm3533-als";
			lm3533_devices[count].id = PLATFORM_DEVID_NONE;

			lm3533->have_als = true;
			count++;
		} else if (fwnode_device_is_compatible(child, "ti,lm3533-backlight")) {
			ret = fwnode_property_read_u32(child, "reg", &reg);
			if (ret || reg >= LM3533_HVLED_ID_MAX) {
				dev_err(dev, "invalid backlight node %pfw\n", child);
				continue;
			}

			lm3533_devices[count].name = "lm3533-backlight";
			lm3533_devices[count].of_compatible = "ti,lm3533-backlight";
			lm3533_devices[count].id = reg;
			lm3533_devices[count].of_reg = reg;
			lm3533_devices[count].use_of_reg = true;

			lm3533->have_backlights = true;
			count++;
		} else if (fwnode_device_is_compatible(child, "ti,lm3533-leds")) {
			ret = fwnode_property_read_u32(child, "reg", &reg);
			if (ret || reg < LM3533_HVLED_ID_MAX ||
			    reg > LM3533_LVLED_ID_MAX) {
				dev_err(dev, "invalid LED node %pfw\n", child);
				continue;
			}

			lm3533_devices[count].name = "lm3533-leds";
			lm3533_devices[count].of_compatible = "ti,lm3533-leds";
			lm3533_devices[count].id = reg - LM3533_HVLED_ID_MAX;
			lm3533_devices[count].of_reg = reg;
			lm3533_devices[count].use_of_reg = true;

			lm3533->have_leds = true;
			count++;
		}
	}

	ret = lm3533_enable(lm3533);
	if (ret)
		return ret;

	ret = regmap_update_bits(lm3533->regmap, LM3533_REG_BOOST_PWM,
				 LM3533_BOOST_FREQ_MASK,
				 FIELD_PREP(LM3533_BOOST_FREQ_MASK, lm3533->boost_freq));
	if (ret) {
		dev_err(dev, "failed to set boost frequency\n");
		goto err_disable;
	}

	ret = regmap_update_bits(lm3533->regmap, LM3533_REG_BOOST_PWM,
				 LM3533_BOOST_OVP_MASK,
				 FIELD_PREP(LM3533_BOOST_OVP_MASK, lm3533->boost_ovp));
	if (ret) {
		dev_err(dev, "failed to set boost ovp\n");
		goto err_disable;
	}

	ret = mfd_add_devices(dev, 0, lm3533_devices, count, NULL, 0, NULL);
	if (ret) {
		dev_err(dev, "failed to add MFD devices: %d\n", ret);
		goto err_disable;
	}

	return 0;

err_disable:
	lm3533_disable(lm3533);

	return ret;
}

static void lm3533_device_exit(struct lm3533 *lm3533)
{
	dev_dbg(lm3533->dev, "%s\n", __func__);

	mfd_remove_devices(lm3533->dev);
	lm3533_disable(lm3533);
}

static bool lm3533_readable_register(struct device *dev, unsigned int reg)
{
	switch (reg) {
	case 0x10 ... 0x2c:
	case 0x30 ... 0x38:
	case 0x40 ... 0x45:
	case 0x50 ... 0x57:
	case 0x60 ... 0x6e:
	case 0x70 ... 0x75:
	case 0x80 ... 0x85:
	case 0x90 ... 0x95:
	case 0xa0 ... 0xa5:
	case 0xb0 ... 0xb2:
		return true;
	default:
		return false;
	}
}

static bool lm3533_volatile_register(struct device *dev, unsigned int reg)
{
	switch (reg) {
	case 0x34 ... 0x36:	/* zone */
	case 0x37 ... 0x38:	/* adc */
	case 0xb0 ... 0xb1:	/* fault */
		return true;
	default:
		return false;
	}
}

static bool lm3533_precious_register(struct device *dev, unsigned int reg)
{
	switch (reg) {
	case 0x34:		/* zone */
		return true;
	default:
		return false;
	}
}

static const struct regmap_config regmap_config = {
	.reg_bits	= 8,
	.val_bits	= 8,
	.max_register	= LM3533_REG_MAX,
	.readable_reg	= lm3533_readable_register,
	.volatile_reg	= lm3533_volatile_register,
	.precious_reg	= lm3533_precious_register,
};

static int lm3533_i2c_probe(struct i2c_client *i2c)
{
	struct lm3533 *lm3533;

	dev_dbg(&i2c->dev, "%s\n", __func__);

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

	i2c_set_clientdata(i2c, lm3533);

	lm3533->regmap = devm_regmap_init_i2c(i2c, &regmap_config);
	if (IS_ERR(lm3533->regmap))
		return PTR_ERR(lm3533->regmap);

	lm3533->dev = &i2c->dev;

	lm3533->hwen = devm_gpiod_get_optional(lm3533->dev, "enable",
					       GPIOD_OUT_LOW);
	if (IS_ERR(lm3533->hwen))
		return dev_err_probe(lm3533->dev, PTR_ERR(lm3533->hwen),
				     "failed to get HWEN GPIO\n");

	lm3533->vin_supply = devm_regulator_get(lm3533->dev, "vin");
	if (IS_ERR(lm3533->vin_supply))
		return dev_err_probe(lm3533->dev, PTR_ERR(lm3533->vin_supply),
				     "failed to get vin-supply\n");

	device_property_read_u32(lm3533->dev, "ti,boost-ovp-microvolt",
				 &lm3533->boost_ovp);

	lm3533->boost_ovp = clamp(lm3533->boost_ovp, LM3533_BOOST_OVP_MIN,
				  LM3533_BOOST_OVP_MAX);
	lm3533->boost_ovp = lm3533->boost_ovp / (8 * MICRO) - 2;

	device_property_read_u32(lm3533->dev, "ti,boost-freq-hz",
				 &lm3533->boost_freq);

	lm3533->boost_freq = clamp(lm3533->boost_freq, LM3533_BOOST_FREQ_MIN,
				   LM3533_BOOST_FREQ_MAX);
	lm3533->boost_freq = lm3533->boost_freq / (500 * KILO) - 1;

	/* LM3533 and child devices do not use DMA */
	i2c->dev.coherent_dma_mask = 0;
	i2c->dev.dma_mask = &i2c->dev.coherent_dma_mask;

	return lm3533_device_init(lm3533);
}

static void lm3533_i2c_remove(struct i2c_client *i2c)
{
	struct lm3533 *lm3533 = i2c_get_clientdata(i2c);

	dev_dbg(&i2c->dev, "%s\n", __func__);

	lm3533_device_exit(lm3533);
}

static const struct of_device_id lm3533_match_table[] = {
	{ .compatible = "ti,lm3533" },
	{ }
};
MODULE_DEVICE_TABLE(of, lm3533_match_table);

static const struct i2c_device_id lm3533_i2c_ids[] = {
	{ "lm3533" },
	{ }
};
MODULE_DEVICE_TABLE(i2c, lm3533_i2c_ids);

static struct i2c_driver lm3533_i2c_driver = {
	.driver = {
		   .name = "lm3533",
		   .dev_groups = lm3533_attribute_groups,
		   .of_match_table = lm3533_match_table,
	},
	.id_table	= lm3533_i2c_ids,
	.probe		= lm3533_i2c_probe,
	.remove		= lm3533_i2c_remove,
};

static int __init lm3533_i2c_init(void)
{
	return i2c_add_driver(&lm3533_i2c_driver);
}
subsys_initcall(lm3533_i2c_init);

static void __exit lm3533_i2c_exit(void)
{
	i2c_del_driver(&lm3533_i2c_driver);
}
module_exit(lm3533_i2c_exit);

MODULE_AUTHOR("Johan Hovold <jhovold@gmail.com>");
MODULE_DESCRIPTION("LM3533 Core");
MODULE_LICENSE("GPL");