// SPDX-License-Identifier: GPL-2.0-only
/*
 * LP5812 LED driver
 *
 * Copyright (C) 2025 Texas Instruments
 *
 * Author: Jared Zhou <jared-zhou@ti.com>
 */

#include <linux/delay.h>
#include <linux/i2c.h>
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/led-class-multicolor.h>
#include <linux/leds.h>
#include <linux/module.h>
#include <linux/mutex.h>
#include <linux/sysfs.h>
#include <linux/types.h>

#include "leds-lp5812.h"

static const struct lp5812_mode_mapping chip_mode_map[] = {
	{"direct_mode", 0, 0, 0, 0, 0, 0},
	{"tcm:1:0", 1, 0, 0, 0, 0, 0},
	{"tcm:1:1", 1, 1, 0, 0, 0, 0},
	{"tcm:1:2", 1, 2, 0, 0, 0, 0},
	{"tcm:1:3", 1, 3, 0, 0, 0, 0},
	{"tcm:2:0:1", 2, 0, 1, 0, 0, 0},
	{"tcm:2:0:2", 2, 0, 2, 0, 0, 0},
	{"tcm:2:0:3", 2, 0, 3, 0, 0, 0},
	{"tcm:2:1:2", 2, 1, 2, 0, 0, 0},
	{"tcm:2:1:3", 2, 1, 3, 0, 0, 0},
	{"tcm:2:2:3", 2, 2, 3, 0, 0, 0},
	{"tcm:3:0:1:2", 3, 0, 1, 2, 0, 0},
	{"tcm:3:0:1:3", 3, 0, 1, 3, 0, 0},
	{"tcm:3:0:2:3", 3, 0, 2, 3, 0, 0},
	{"tcm:4:0:1:2:3", 4, 0, 1, 2, 3, 0},
	{"mix:1:0:1", 5, 1, 0, 0, 0, 0},
	{"mix:1:0:2", 5, 2, 0, 0, 0, 0},
	{"mix:1:0:3", 5, 3, 0, 0, 0, 0},
	{"mix:1:1:0", 5, 0, 0, 0, 0, 1},
	{"mix:1:1:2", 5, 2, 0, 0, 0, 1},
	{"mix:1:1:3", 5, 3, 0, 0, 0, 1},
	{"mix:1:2:0", 5, 0, 0, 0, 0, 2},
	{"mix:1:2:1", 5, 1, 0, 0, 0, 2},
	{"mix:1:2:3", 5, 3, 0, 0, 0, 2},
	{"mix:1:3:0", 5, 0, 0, 0, 0, 3},
	{"mix:1:3:1", 5, 1, 0, 0, 0, 3},
	{"mix:1:3:2", 5, 2, 0, 0, 0, 3},
	{"mix:2:0:1:2", 6, 1, 2, 0, 0, 0},
	{"mix:2:0:1:3", 6, 1, 3, 0, 0, 0},
	{"mix:2:0:2:3", 6, 2, 3, 0, 0, 0},
	{"mix:2:1:0:2", 6, 0, 2, 0, 0, 1},
	{"mix:2:1:0:3", 6, 0, 3, 0, 0, 1},
	{"mix:2:1:2:3", 6, 2, 3, 0, 0, 1},
	{"mix:2:2:0:1", 6, 0, 1, 0, 0, 2},
	{"mix:2:2:0:3", 6, 0, 3, 0, 0, 2},
	{"mix:2:2:1:3", 6, 1, 3, 0, 0, 2},
	{"mix:2:3:0:1", 6, 0, 1, 0, 0, 3},
	{"mix:2:3:0:2", 6, 0, 2, 0, 0, 3},
	{"mix:2:3:1:2", 6, 1, 2, 0, 0, 3},
	{"mix:3:0:1:2:3", 7, 1, 2, 3, 0, 0},
	{"mix:3:1:0:2:3", 7, 0, 2, 3, 0, 1},
	{"mix:3:2:0:1:3", 7, 0, 1, 3, 0, 2},
	{"mix:3:3:0:1:2", 7, 0, 1, 2, 0, 3}
};

static int lp5812_write(struct lp5812_chip *chip, u16 reg, u8 val)
{
	struct device *dev = &chip->client->dev;
	struct i2c_msg msg;
	u8 buf[LP5812_DATA_LENGTH];
	u8 reg_addr_bit8_9;
	int ret;

	/* Extract register address bits 9 and 8 for Address Byte 1 */
	reg_addr_bit8_9 = (reg >> LP5812_REG_ADDR_HIGH_SHIFT) & LP5812_REG_ADDR_BIT_8_9_MASK;

	/* Prepare payload: Address Byte 2 (bits [7:0]) and value to write */
	buf[LP5812_DATA_BYTE_0_IDX] = (u8)(reg & LP5812_REG_ADDR_LOW_MASK);
	buf[LP5812_DATA_BYTE_1_IDX] = val;

	/* Construct I2C message for a write operation */
	msg.addr = (chip->client->addr << LP5812_CHIP_ADDR_SHIFT) | reg_addr_bit8_9;
	msg.flags = 0;
	msg.len = sizeof(buf);
	msg.buf = buf;

	ret = i2c_transfer(chip->client->adapter, &msg, 1);
	if (ret == 1)
		return 0;

	dev_err(dev, "I2C write error, ret=%d\n", ret);
	return ret < 0 ? ret : -EIO;
}

static int lp5812_read(struct lp5812_chip *chip, u16 reg, u8 *val)
{
	struct device *dev = &chip->client->dev;
	struct i2c_msg msgs[LP5812_READ_MSG_LENGTH];
	u8 ret_val;
	u8 reg_addr_bit8_9;
	u8 converted_reg;
	int ret;

	/* Extract register address bits 9 and 8 for Address Byte 1 */
	reg_addr_bit8_9 = (reg >> LP5812_REG_ADDR_HIGH_SHIFT) & LP5812_REG_ADDR_BIT_8_9_MASK;

	/* Lower 8 bits go in Address Byte 2 */
	converted_reg = (u8)(reg & LP5812_REG_ADDR_LOW_MASK);

	/* Prepare I2C write message to set register address */
	msgs[LP5812_MSG_0_IDX].addr =
		(chip->client->addr << LP5812_CHIP_ADDR_SHIFT) | reg_addr_bit8_9;
	msgs[LP5812_MSG_0_IDX].flags = 0;
	msgs[LP5812_MSG_0_IDX].len = 1;
	msgs[LP5812_MSG_0_IDX].buf = &converted_reg;

	/* Prepare I2C read message to retrieve register value */
	msgs[LP5812_MSG_1_IDX].addr =
		(chip->client->addr << LP5812_CHIP_ADDR_SHIFT) | reg_addr_bit8_9;
	msgs[LP5812_MSG_1_IDX].flags = I2C_M_RD;
	msgs[LP5812_MSG_1_IDX].len = 1;
	msgs[LP5812_MSG_1_IDX].buf = &ret_val;

	ret = i2c_transfer(chip->client->adapter, msgs, LP5812_READ_MSG_LENGTH);
	if (ret == LP5812_READ_MSG_LENGTH) {
		*val = ret_val;
		return 0;
	}

	dev_err(dev, "I2C read error, ret=%d\n", ret);
	*val = 0;
	return ret < 0 ? ret : -EIO;
}

static int lp5812_read_tsd_config_status(struct lp5812_chip *chip, u8 *reg_val)
{
	return lp5812_read(chip, LP5812_TSD_CONFIG_STATUS, reg_val);
}

static int lp5812_update_regs_config(struct lp5812_chip *chip)
{
	u8 reg_val;
	int ret;

	ret = lp5812_write(chip, LP5812_CMD_UPDATE, LP5812_UPDATE_CMD_VAL);
	if (ret)
		return ret;

	ret = lp5812_read_tsd_config_status(chip, &reg_val);
	if (ret)
		return ret;

	return reg_val & LP5812_CFG_ERR_STATUS_MASK;
}

static ssize_t parse_drive_mode(struct lp5812_chip *chip, const char *str)
{
	int i;

	chip->drive_mode.bits.mix_sel_led_0 = false;
	chip->drive_mode.bits.mix_sel_led_1 = false;
	chip->drive_mode.bits.mix_sel_led_2 = false;
	chip->drive_mode.bits.mix_sel_led_3 = false;

	if (sysfs_streq(str, LP5812_MODE_DIRECT_NAME)) {
		chip->drive_mode.bits.led_mode = LP5812_MODE_DIRECT_VALUE;
		return 0;
	}

	for (i = 0; i < ARRAY_SIZE(chip_mode_map); i++) {
		if (!sysfs_streq(str, chip_mode_map[i].mode_name))
			continue;

		chip->drive_mode.bits.led_mode = chip_mode_map[i].mode;
		chip->scan_order.bits.order0 = chip_mode_map[i].scan_order_0;
		chip->scan_order.bits.order1 = chip_mode_map[i].scan_order_1;
		chip->scan_order.bits.order2 = chip_mode_map[i].scan_order_2;
		chip->scan_order.bits.order3 = chip_mode_map[i].scan_order_3;

		switch (chip_mode_map[i].selection_led) {
		case LP5812_MODE_MIX_SELECT_LED_0:
			chip->drive_mode.bits.mix_sel_led_0 = true;
			break;
		case LP5812_MODE_MIX_SELECT_LED_1:
			chip->drive_mode.bits.mix_sel_led_1 = true;
			break;
		case LP5812_MODE_MIX_SELECT_LED_2:
			chip->drive_mode.bits.mix_sel_led_2 = true;
			break;
		case LP5812_MODE_MIX_SELECT_LED_3:
			chip->drive_mode.bits.mix_sel_led_3 = true;
			break;
		default:
			return -EINVAL;
		}

		return 0;
	}

	return -EINVAL;
}

static int lp5812_set_drive_mode_scan_order(struct lp5812_chip *chip)
{
	u8 val;
	int ret;

	val = chip->drive_mode.val;
	ret = lp5812_write(chip, LP5812_DEV_CONFIG1, val);
	if (ret)
		return ret;

	val = chip->scan_order.val;
	ret = lp5812_write(chip, LP5812_DEV_CONFIG2, val);

	return ret;
}

static int lp5812_set_led_mode(struct lp5812_chip *chip, int led_number,
			       enum control_mode mode)
{
	u8 reg_val;
	u16 reg;
	int ret;

	/*
	 * Select device configuration register.
	 * Reg3 for LED_0–LED_3, LED_A0–A2, LED_B0
	 * Reg4 for LED_B1–B2, LED_C0–C2, LED_D0–D2
	 */
	if (led_number < LP5812_NUMBER_LED_IN_REG)
		reg = LP5812_DEV_CONFIG3;
	else
		reg = LP5812_DEV_CONFIG4;

	ret = lp5812_read(chip, reg, &reg_val);
	if (ret)
		return ret;

	if (mode == LP5812_MODE_MANUAL)
		reg_val &= ~(LP5812_ENABLE << (led_number % LP5812_NUMBER_LED_IN_REG));
	else
		reg_val |= (LP5812_ENABLE << (led_number % LP5812_NUMBER_LED_IN_REG));

	ret = lp5812_write(chip, reg, reg_val);
	if (ret)
		return ret;

	ret = lp5812_update_regs_config(chip);

	return ret;
}

static int lp5812_manual_dc_pwm_control(struct lp5812_chip *chip, int led_number,
					u8 val, enum dimming_type dimming_type)
{
	u16 led_base_reg;
	int ret;

	if (dimming_type == LP5812_DIMMING_ANALOG)
		led_base_reg = LP5812_MANUAL_DC_BASE;
	else
		led_base_reg = LP5812_MANUAL_PWM_BASE;

	ret = lp5812_write(chip, led_base_reg + led_number, val);

	return ret;
}

static int lp5812_multicolor_brightness(struct lp5812_led *led)
{
	struct lp5812_chip *chip = led->chip;
	int ret, i;

	guard(mutex)(&chip->lock);
	for (i = 0; i < led->mc_cdev.num_colors; i++) {
		ret = lp5812_manual_dc_pwm_control(chip, led->mc_cdev.subled_info[i].channel,
						   led->mc_cdev.subled_info[i].brightness,
						   LP5812_DIMMING_PWM);
		if (ret)
			return ret;
	}

	return 0;
}

static int lp5812_led_brightness(struct lp5812_led *led)
{
	struct lp5812_chip *chip = led->chip;
	struct lp5812_led_config *led_cfg;
	int ret;

	led_cfg = &chip->led_config[led->chan_nr];

	guard(mutex)(&chip->lock);
	ret = lp5812_manual_dc_pwm_control(chip, led_cfg->led_id[0],
					   led->brightness, LP5812_DIMMING_PWM);

	return ret;
}

static int lp5812_set_brightness(struct led_classdev *cdev,
				 enum led_brightness brightness)
{
	struct lp5812_led *led = container_of(cdev, struct lp5812_led, cdev);

	led->brightness = (u8)brightness;

	return lp5812_led_brightness(led);
}

static int lp5812_set_mc_brightness(struct led_classdev *cdev,
				    enum led_brightness brightness)
{
	struct led_classdev_mc *mc_dev = lcdev_to_mccdev(cdev);
	struct lp5812_led *led = container_of(mc_dev, struct lp5812_led, mc_cdev);

	led_mc_calc_color_components(&led->mc_cdev, brightness);

	return lp5812_multicolor_brightness(led);
}

static int lp5812_init_led(struct lp5812_led *led, struct lp5812_chip *chip, int chan)
{
	struct device *dev = &chip->client->dev;
	struct mc_subled *mc_led_info;
	struct led_classdev *led_cdev;
	int i, ret;

	if (chip->led_config[chan].name) {
		led->cdev.name = chip->led_config[chan].name;
	} else {
		led->cdev.name = devm_kasprintf(dev, GFP_KERNEL, "%s:channel%d",
						chip->label ? : chip->client->name, chan);
		if (!led->cdev.name)
			return -ENOMEM;
	}

	if (!chip->led_config[chan].is_sc_led) {
		mc_led_info = devm_kcalloc(dev, chip->led_config[chan].num_colors,
					   sizeof(*mc_led_info), GFP_KERNEL);
		if (!mc_led_info)
			return -ENOMEM;

		led_cdev = &led->mc_cdev.led_cdev;
		led_cdev->name = led->cdev.name;
		led_cdev->brightness_set_blocking = lp5812_set_mc_brightness;
		led->mc_cdev.num_colors = chip->led_config[chan].num_colors;

		for (i = 0; i < led->mc_cdev.num_colors; i++) {
			mc_led_info[i].color_index = chip->led_config[chan].color_id[i];
			mc_led_info[i].channel = chip->led_config[chan].led_id[i];
		}

		led->mc_cdev.subled_info = mc_led_info;
	} else {
		led->cdev.brightness_set_blocking = lp5812_set_brightness;
	}

	led->chan_nr = chan;

	if (chip->led_config[chan].is_sc_led) {
		ret = devm_led_classdev_register(dev, &led->cdev);
		if (ret == 0)
			led->cdev.dev->platform_data = led;
	} else {
		ret = devm_led_classdev_multicolor_register(dev, &led->mc_cdev);
		if (ret == 0)
			led->mc_cdev.led_cdev.dev->platform_data = led;
	}

	return ret;
}

static int lp5812_register_leds(struct lp5812_led *leds, struct lp5812_chip *chip)
{
	struct lp5812_led *led;
	int num_channels = chip->num_channels;
	u8 reg_val;
	u16 reg;
	int ret, i, j;

	for (i = 0; i < num_channels; i++) {
		led = &leds[i];
		ret = lp5812_init_led(led, chip, i);
		if (ret)
			goto err_init_led;

		led->chip = chip;

		for (j = 0; j < chip->led_config[i].num_colors; j++) {
			ret = lp5812_write(chip,
					   LP5812_AUTO_DC_BASE + chip->led_config[i].led_id[j],
					   chip->led_config[i].max_current[j]);
			if (ret)
				goto err_init_led;

			ret = lp5812_manual_dc_pwm_control(chip, chip->led_config[i].led_id[j],
							   chip->led_config[i].max_current[j],
							   LP5812_DIMMING_ANALOG);
			if (ret)
				goto err_init_led;

			ret = lp5812_set_led_mode(chip, chip->led_config[i].led_id[j],
						  LP5812_MODE_MANUAL);
			if (ret)
				goto err_init_led;

			reg = (chip->led_config[i].led_id[j] < LP5812_NUMBER_LED_IN_REG) ?
				LP5812_LED_EN_1 : LP5812_LED_EN_2;

			ret = lp5812_read(chip, reg, &reg_val);
			if (ret)
				goto err_init_led;

			reg_val |= (LP5812_ENABLE << (chip->led_config[i].led_id[j] %
				LP5812_NUMBER_LED_IN_REG));

			ret = lp5812_write(chip, reg, reg_val);
			if (ret)
				goto err_init_led;
		}
	}

	return 0;

err_init_led:
	return ret;
}

static int lp5812_init_device(struct lp5812_chip *chip)
{
	int ret;

	usleep_range(LP5812_WAIT_DEVICE_STABLE_MIN, LP5812_WAIT_DEVICE_STABLE_MAX);

	ret = lp5812_write(chip, LP5812_REG_ENABLE, LP5812_ENABLE);
	if (ret) {
		dev_err(&chip->client->dev, "failed to enable LP5812 device\n");
		return ret;
	}

	ret = lp5812_write(chip, LP5812_DEV_CONFIG12, LP5812_LSD_LOD_START_UP);
	if (ret) {
		dev_err(&chip->client->dev, "failed to configure device safety thresholds\n");
		return ret;
	}

	ret = parse_drive_mode(chip, chip->scan_mode);
	if (ret)
		return ret;

	ret = lp5812_set_drive_mode_scan_order(chip);
	if (ret)
		return ret;

	ret = lp5812_update_regs_config(chip);
	if (ret) {
		dev_err(&chip->client->dev, "failed to apply configuration updates\n");
		return ret;
	}

	return 0;
}

static void lp5812_deinit_device(struct lp5812_chip *chip)
{
	lp5812_write(chip, LP5812_LED_EN_1, LP5812_DISABLE);
	lp5812_write(chip, LP5812_LED_EN_2, LP5812_DISABLE);
	lp5812_write(chip, LP5812_REG_ENABLE, LP5812_DISABLE);
}

static int lp5812_parse_led_channel(struct device_node *np,
				    struct lp5812_led_config *cfg,
				    int color_number)
{
	int color_id, reg, ret;
	u32 max_cur;

	ret = of_property_read_u32(np, "reg", &reg);
	if (ret)
		return ret;

	cfg->led_id[color_number] = reg;

	ret = of_property_read_u32(np, "led-max-microamp", &max_cur);
	if (ret)
		max_cur = 0;
	/* Convert microamps to driver units */
	cfg->max_current[color_number] = max_cur / 100;

	ret = of_property_read_u32(np, "color", &color_id);
	if (ret)
		color_id = 0;
	cfg->color_id[color_number] = color_id;

	return 0;
}

static int lp5812_parse_led(struct device_node *np,
			    struct lp5812_led_config *cfg,
			    int led_index)
{
	int num_colors, ret;

	of_property_read_string(np, "label", &cfg[led_index].name);

	ret = of_property_read_u32(np, "reg", &cfg[led_index].chan_nr);
	if (ret)
		return ret;

	num_colors = 0;
	for_each_available_child_of_node_scoped(np, child) {
		ret = lp5812_parse_led_channel(child, &cfg[led_index], num_colors);
		if (ret)
			return ret;

		num_colors++;
	}

	if (num_colors == 0) {
		ret = lp5812_parse_led_channel(np, &cfg[led_index], 0);
		if (ret)
			return ret;

		num_colors = 1;
		cfg[led_index].is_sc_led = true;
	} else {
		cfg[led_index].is_sc_led = false;
	}

	cfg[led_index].num_colors = num_colors;

	return 0;
}

static int lp5812_of_probe(struct device *dev,
			   struct device_node *np,
			   struct lp5812_chip *chip)
{
	struct lp5812_led_config *cfg;
	int num_channels, i = 0, ret;

	num_channels = of_get_available_child_count(np);
	if (num_channels == 0) {
		dev_err(dev, "no LED channels\n");
		return -EINVAL;
	}

	cfg = devm_kcalloc(dev, num_channels, sizeof(*cfg), GFP_KERNEL);
	if (!cfg)
		return -ENOMEM;

	chip->led_config = &cfg[0];
	chip->num_channels = num_channels;

	for_each_available_child_of_node_scoped(np, child) {
		ret = lp5812_parse_led(child, cfg, i);
		if (ret)
			return -EINVAL;
		i++;
	}

	ret = of_property_read_string(np, "ti,scan-mode", &chip->scan_mode);
	if (ret)
		chip->scan_mode = LP5812_MODE_DIRECT_NAME;

	of_property_read_string(np, "label", &chip->label);

	return 0;
}

static int lp5812_probe(struct i2c_client *client)
{
	struct lp5812_chip *chip;
	struct device_node *np = dev_of_node(&client->dev);
	struct lp5812_led *leds;
	int ret;

	if (!np)
		return -EINVAL;

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

	ret = lp5812_of_probe(&client->dev, np, chip);
	if (ret)
		return ret;

	leds = devm_kcalloc(&client->dev, chip->num_channels, sizeof(*leds), GFP_KERNEL);
	if (!leds)
		return -ENOMEM;

	chip->client = client;
	mutex_init(&chip->lock);
	i2c_set_clientdata(client, chip);

	ret = lp5812_init_device(chip);
	if (ret)
		return ret;

	ret = lp5812_register_leds(leds, chip);
	if (ret)
		goto err_out;

	return 0;

err_out:
	lp5812_deinit_device(chip);
	return ret;
}

static void lp5812_remove(struct i2c_client *client)
{
	struct lp5812_chip *chip = i2c_get_clientdata(client);

	lp5812_deinit_device(chip);
}

static const struct of_device_id of_lp5812_match[] = {
	{ .compatible = "ti,lp5812" },
	{ /* sentinel */ }
};
MODULE_DEVICE_TABLE(of, of_lp5812_match);

static struct i2c_driver lp5812_driver = {
	.driver = {
		.name   = "lp5812",
		.of_match_table = of_lp5812_match,
	},
	.probe		= lp5812_probe,
	.remove		= lp5812_remove,
};
module_i2c_driver(lp5812_driver);

MODULE_DESCRIPTION("Texas Instruments LP5812 LED Driver");
MODULE_AUTHOR("Jared Zhou <jared-zhou@ti.com>");
MODULE_LICENSE("GPL");