// SPDX-License-Identifier: GPL-2.0-only
/*
 * TI LP8864/LP8866 4/6 Channel LED Driver
 *
 * Copyright (C) 2024 Siemens AG
 *
 * Based on LP8860 driver by Dan Murphy <dmurphy@ti.com>
 */

#include <linux/gpio/consumer.h>
#include <linux/i2c.h>
#include <linux/init.h>
#include <linux/leds.h>
#include <linux/module.h>
#include <linux/mutex.h>
#include <linux/of.h>
#include <linux/regmap.h>
#include <linux/regulator/consumer.h>
#include <linux/slab.h>

#define LP8864_BRT_CONTROL		0x00
#define LP8864_USER_CONFIG1		0x04
#define   LP8864_BRT_MODE_MASK		GENMASK(9, 8)
#define   LP8864_BRT_MODE_REG		BIT(9)		/* Brightness control by DISPLAY_BRT reg */
#define LP8864_SUPPLY_STATUS		0x0e
#define LP8864_BOOST_STATUS		0x10
#define LP8864_LED_STATUS		0x12
#define   LP8864_LED_STATUS_WR_MASK	GENMASK(14, 9)	/* Writeable bits in the LED_STATUS reg */

/* Textual meaning for status bits, starting from bit 1 */
static const char *const lp8864_supply_status_msg[] = {
	"Vin under-voltage fault",
	"Vin over-voltage fault",
	"Vdd under-voltage fault",
	"Vin over-current fault",
	"Missing charge pump fault",
	"Charge pump fault",
	"Missing boost sync fault",
	"CRC error fault ",
};

/* Textual meaning for status bits, starting from bit 1 */
static const char *const lp8864_boost_status_msg[] = {
	"Boost OVP low fault",
	"Boost OVP high fault",
	"Boost over-current fault",
	"Missing boost FSET resistor fault",
	"Missing MODE SEL resistor fault",
	"Missing LED resistor fault",
	"ISET resistor short to ground fault",
	"Thermal shutdown fault",
};

/* Textual meaning for every register bit */
static const char *const lp8864_led_status_msg[] = {
	"LED 1 fault",
	"LED 2 fault",
	"LED 3 fault",
	"LED 4 fault",
	"LED 5 fault",
	"LED 6 fault",
	"LED open fault",
	"LED internal short fault",
	"LED short to GND fault",
	NULL, NULL, NULL,
	"Invalid string configuration fault",
	NULL,
	"I2C time out fault",
};

/**
 * struct lp8864_led
 * @client: Pointer to the I2C client
 * @led_dev: led class device pointer
 * @regmap: Devices register map
 * @led_status_mask: Helps to report LED fault only once
 */
struct lp8864_led {
	struct i2c_client *client;
	struct led_classdev led_dev;
	struct regmap *regmap;
	u16 led_status_mask;
};

static int lp8864_fault_check(struct lp8864_led *led)
{
	int ret, i;
	unsigned int val;

	ret = regmap_read(led->regmap, LP8864_SUPPLY_STATUS, &val);
	if (ret)
		goto err;

	/* Odd bits are status bits, even bits are clear bits */
	for (i = 0; i < ARRAY_SIZE(lp8864_supply_status_msg); i++)
		if (val & BIT(i * 2 + 1))
			dev_warn(&led->client->dev, "%s\n", lp8864_supply_status_msg[i]);

	/*
	 * Clear bits have an index preceding the corresponding Status bits;
	 * both have to be written "1" simultaneously to clear the corresponding
	 * Status bit.
	 */
	if (val)
		ret = regmap_write(led->regmap, LP8864_SUPPLY_STATUS, val >> 1 | val);
	if (ret)
		goto err;

	ret = regmap_read(led->regmap, LP8864_BOOST_STATUS, &val);
	if (ret)
		goto err;

	/* Odd bits are status bits, even bits are clear bits */
	for (i = 0; i < ARRAY_SIZE(lp8864_boost_status_msg); i++)
		if (val & BIT(i * 2 + 1))
			dev_warn(&led->client->dev, "%s\n", lp8864_boost_status_msg[i]);

	if (val)
		ret = regmap_write(led->regmap, LP8864_BOOST_STATUS, val >> 1 | val);
	if (ret)
		goto err;

	ret = regmap_read(led->regmap, LP8864_LED_STATUS, &val);
	if (ret)
		goto err;

	/*
	 * Clear already reported faults that maintain their value until device
	 * power-down
	 */
	val &= ~led->led_status_mask;

	for (i = 0; i < ARRAY_SIZE(lp8864_led_status_msg); i++)
		if (lp8864_led_status_msg[i] && val & BIT(i))
			dev_warn(&led->client->dev, "%s\n", lp8864_led_status_msg[i]);

	/*
	 * Mark those which maintain their value until device power-down as
	 * "already reported"
	 */
	led->led_status_mask |= val & ~LP8864_LED_STATUS_WR_MASK;

	/*
	 * Only bits 14, 12, 10 have to be cleared here, but others are RO,
	 * we don't care what we write to them.
	 */
	if (val & LP8864_LED_STATUS_WR_MASK)
		ret = regmap_write(led->regmap, LP8864_LED_STATUS, val >> 1 | val);
	if (ret)
		goto err;

	return 0;

err:
	dev_err(&led->client->dev, "Failed to read/clear faults (%pe)\n", ERR_PTR(ret));

	return ret;
}

static int lp8864_brightness_set(struct led_classdev *led_cdev,
				 enum led_brightness brt_val)
{
	struct lp8864_led *led = container_of(led_cdev, struct lp8864_led, led_dev);
	/* Scale 0..LED_FULL into 16-bit HW brightness */
	unsigned int val = brt_val * 0xffff / LED_FULL;
	int ret;

	ret = lp8864_fault_check(led);
	if (ret)
		return ret;

	ret = regmap_write(led->regmap, LP8864_BRT_CONTROL, val);
	if (ret)
		dev_err(&led->client->dev, "Failed to write brightness value\n");

	return ret;
}

static enum led_brightness lp8864_brightness_get(struct led_classdev *led_cdev)
{
	struct lp8864_led *led = container_of(led_cdev, struct lp8864_led, led_dev);
	unsigned int val;
	int ret;

	ret = regmap_read(led->regmap, LP8864_BRT_CONTROL, &val);
	if (ret) {
		dev_err(&led->client->dev, "Failed to read brightness value\n");
		return ret;
	}

	/* Scale 16-bit HW brightness into 0..LED_FULL */
	return val * LED_FULL / 0xffff;
}

static const struct regmap_config lp8864_regmap_config = {
	.reg_bits		= 8,
	.val_bits		= 16,
	.val_format_endian	= REGMAP_ENDIAN_LITTLE,
};

static void lp8864_disable_gpio(void *data)
{
	struct gpio_desc *gpio = data;

	gpiod_set_value(gpio, 0);
}

static int lp8864_probe(struct i2c_client *client)
{
	int ret;
	struct lp8864_led *led;
	struct device_node *np = dev_of_node(&client->dev);
	struct device_node *child_node;
	struct led_init_data init_data = {};
	struct gpio_desc *enable_gpio;

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

	child_node = of_get_next_available_child(np, NULL);
	if (!child_node) {
		dev_err(&client->dev, "No LED function defined\n");
		return -EINVAL;
	}

	ret = devm_regulator_get_enable_optional(&client->dev, "vled");
	if (ret && ret != -ENODEV)
		return dev_err_probe(&client->dev, ret, "Failed to enable vled regulator\n");

	enable_gpio = devm_gpiod_get_optional(&client->dev, "enable", GPIOD_OUT_HIGH);
	if (IS_ERR(enable_gpio))
		return dev_err_probe(&client->dev, PTR_ERR(enable_gpio),
				     "Failed to get enable GPIO\n");

	ret = devm_add_action_or_reset(&client->dev, lp8864_disable_gpio, enable_gpio);
	if (ret)
		return ret;

	led->client = client;
	led->led_dev.brightness_set_blocking = lp8864_brightness_set;
	led->led_dev.brightness_get = lp8864_brightness_get;

	led->regmap = devm_regmap_init_i2c(client, &lp8864_regmap_config);
	if (IS_ERR(led->regmap))
		return dev_err_probe(&client->dev, PTR_ERR(led->regmap),
				     "Failed to allocate regmap\n");

	/* Control brightness by DISPLAY_BRT register */
	ret = regmap_update_bits(led->regmap, LP8864_USER_CONFIG1, LP8864_BRT_MODE_MASK,
								   LP8864_BRT_MODE_REG);
	if (ret) {
		dev_err(&led->client->dev, "Failed to set brightness control mode\n");
		return ret;
	}

	ret = lp8864_fault_check(led);
	if (ret)
		return ret;

	init_data.fwnode = of_fwnode_handle(child_node);
	init_data.devicename = "lp8864";
	init_data.default_label = ":display_cluster";

	ret = devm_led_classdev_register_ext(&client->dev, &led->led_dev, &init_data);
	if (ret)
		dev_err(&client->dev, "Failed to register LED device (%pe)\n", ERR_PTR(ret));

	return ret;
}

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

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

static struct i2c_driver lp8864_driver = {
	.driver = {
		.name	= "lp8864",
		.of_match_table = of_lp8864_leds_match,
	},
	.probe		= lp8864_probe,
	.id_table	= lp8864_id,
};
module_i2c_driver(lp8864_driver);

MODULE_DESCRIPTION("Texas Instruments LP8864/LP8866 LED driver");
MODULE_AUTHOR("Alexander Sverdlin <alexander.sverdlin@siemens.com>");
MODULE_LICENSE("GPL");