build/patch/kernel/archive/sm8550-6.12/0033_leds--Add-driver-for-HEROIC-HTR3212.patch

From 3a9cb9bab1475d1f907e0842f01b5d4ebaf860c6 Mon Sep 17 00:00:00 2001
From: Teguh Sobirin <teguh@sobir.in>
Date: Wed, 12 Feb 2025 15:29:16 +0800
Subject: [PATCH] leds: Add driver for HEROIC HTR3212

Signed-off-by: Teguh Sobirin <teguh@sobir.in>
---
 drivers/leds/Kconfig        |  10 ++
 drivers/leds/Makefile       |   1 +
 drivers/leds/leds-htr3212.c | 298 ++++++++++++++++++++++++++++++++++++
 3 files changed, 309 insertions(+)
 create mode 100644 drivers/leds/leds-htr3212.c

diff --git a/drivers/leds/Kconfig b/drivers/leds/Kconfig
index b784bb74a837..74dcaf546b4f 100644
--- a/drivers/leds/Kconfig
+++ b/drivers/leds/Kconfig
@@ -211,6 +211,16 @@ config LEDS_EXPRESSWIRE
 	bool
 	depends on GPIOLIB
 
+config LEDS_HTR3212
+	tristate "HEROIC HTR3212 LED driver"
+	depends on LEDS_CLASS && I2C
+	select REGMAP_I2C
+	help
+	  Driver for HEROIC HTR3212 LED controller.
+
+	  To compile this driver as a module, choose M here: the module
+	  will be called leds-htr3212.
+
 config LEDS_TURRIS_OMNIA
 	tristate "LED support for CZ.NIC's Turris Omnia"
 	depends on LEDS_CLASS_MULTICOLOR
diff --git a/drivers/leds/Makefile b/drivers/leds/Makefile
index 18afbb5a23ee..e2277fa7f00d 100644
--- a/drivers/leds/Makefile
+++ b/drivers/leds/Makefile
@@ -32,6 +32,7 @@ obj-$(CONFIG_LEDS_DA9052)		+= leds-da9052.o
 obj-$(CONFIG_LEDS_GPIO)			+= leds-gpio.o
 obj-$(CONFIG_LEDS_GPIO_REGISTER)	+= leds-gpio-register.o
 obj-$(CONFIG_LEDS_HP6XX)		+= leds-hp6xx.o
+obj-$(CONFIG_LEDS_HTR3212)		+= leds-htr3212.o
 obj-$(CONFIG_LEDS_INTEL_SS4200)		+= leds-ss4200.o
 obj-$(CONFIG_LEDS_IP30)			+= leds-ip30.o
 obj-$(CONFIG_LEDS_IPAQ_MICRO)		+= leds-ipaq-micro.o
diff --git a/drivers/leds/leds-htr3212.c b/drivers/leds/leds-htr3212.c
new file mode 100644
index 000000000000..e9367c4aad04
--- /dev/null
+++ b/drivers/leds/leds-htr3212.c
@@ -0,0 +1,298 @@
+// SPDX-License-Identifier: GPL-2.0-only
+/*
+ * Driver for HEROIC HTR3212 12-channel 8-bit PWM LED controller
+ *
+ * Copyright (c) 2024 Teguh Sobirin <teguh@sobir.in>
+ *
+ */
+
+#include <linux/delay.h>
+#include <linux/device.h>
+#include <linux/gpio/consumer.h>
+#include <linux/i2c.h>
+#include <linux/kernel.h>
+#include <linux/leds.h>
+#include <linux/module.h>
+#include <linux/of.h>
+
+#define HTR3212_CHANNELS 12
+#define HTR3212_ENABLE_BIT 1
+#define HTR3212_SHUTDOWN 0x00
+#define HTR3212_PWM_REGISTER_BASE 0x0d
+#define HTR3212_PWM_UPDATE 0x25
+#define HTR3212_LED_CONTROL_BASE 0x32
+#define HTR3212_GLOBAL_CONTROL 0x4a
+#define HTR3212_OUTPUT_FREQ 0x4b
+#define HTR3212_RESET 0x4f
+
+struct htr3212_priv;
+
+struct htr3212_led_data {
+	struct led_classdev cdev;
+	struct htr3212_priv *priv;
+	u8 channel;
+};
+
+struct htr3212_priv {
+	struct i2c_client *client;
+	unsigned int num_leds;
+	struct gpio_desc *sdb;
+	struct regulator *vdd;
+	struct htr3212_led_data leds[];
+};
+
+static int htr3212_write(struct htr3212_priv *priv, u8 reg, u8 val)
+{
+	int ret;
+
+	dev_dbg(&priv->client->dev, "writing register 0x%02X=0x%02X", reg, val);
+
+	ret =  i2c_smbus_write_byte_data(priv->client, reg, val);
+	if (ret) {
+		dev_err(&priv->client->dev,
+			"register write to 0x%02X failed (error %d)",
+			reg, ret);
+	}
+	return ret;
+}
+
+static int htr3212_brightness_set(struct led_classdev *led_cdev,
+				     enum led_brightness brightness)
+{
+	const struct htr3212_led_data *led_data =
+		container_of(led_cdev, struct htr3212_led_data, cdev);
+	u8 pwm_register_offset;
+	int ret;
+
+	dev_dbg(led_cdev->dev, "%s: %d\n", __func__, brightness);
+
+	pwm_register_offset = led_data->channel - 1;
+
+	ret = htr3212_write(led_data->priv,
+			       HTR3212_PWM_REGISTER_BASE + pwm_register_offset,
+			       brightness);
+	if (ret)
+		return ret;
+
+	return htr3212_write(led_data->priv, HTR3212_PWM_UPDATE, 0x00);
+}
+
+static int htr3212_reset_regs(struct htr3212_priv *priv)
+{
+	int ret;
+
+	ret = htr3212_write(priv, HTR3212_RESET, 0x00);
+	if (ret)
+		return ret;
+
+	return 0;
+}
+
+static int htr3212_init_regs(struct htr3212_priv *priv)
+{
+	int ret;
+
+	ret = htr3212_reset_regs(priv);
+	if (ret)
+		return ret;
+
+	u8 value = GENMASK(HTR3212_ENABLE_BIT, 0);
+	u8 num_regs = HTR3212_CHANNELS / HTR3212_ENABLE_BIT;
+
+	int i;
+
+	for (i = 0; i < num_regs; i++) {
+		ret = htr3212_write(priv,
+				HTR3212_LED_CONTROL_BASE + i, value);
+		if (ret)
+			return ret;
+	}
+
+	ret = htr3212_write(priv, HTR3212_SHUTDOWN, 0x01);
+	if (ret)
+		return ret;
+
+	ret = htr3212_write(priv, HTR3212_OUTPUT_FREQ, 0x01);
+	if (ret)
+		return ret;
+
+	ret = htr3212_write(priv, HTR3212_GLOBAL_CONTROL, 0x00);
+	if (ret)
+		return ret;
+
+	return 0;
+}
+
+static int htr3212_parse_child_dt(const struct device *dev,
+				     const struct device_node *child,
+				     struct htr3212_led_data *led_data)
+{
+	struct led_classdev *cdev = &led_data->cdev;
+	int ret = 0;
+	u32 reg;
+
+	ret = of_property_read_u32(child, "reg", &reg);
+	if (ret || reg < 1 || reg > HTR3212_CHANNELS) {
+		dev_err(dev,
+			"Child node %pOF does not have a valid reg property\n",
+			child);
+		return -EINVAL;
+	}
+	led_data->channel = reg;
+
+	cdev->brightness_set_blocking = htr3212_brightness_set;
+
+	if (!device_property_read_bool(dev, "always-on"))
+		cdev->flags |= LED_CORE_SUSPENDRESUME;
+
+	return 0;
+}
+
+static struct htr3212_led_data *htr3212_find_led_data(
+					struct htr3212_priv *priv,
+					u8 channel)
+{
+	size_t i;
+
+	for (i = 0; i < priv->num_leds; i++) {
+		if (priv->leds[i].channel == channel)
+			return &priv->leds[i];
+	}
+
+	return NULL;
+}
+
+static int htr3212_parse_dt(struct device *dev,
+			       struct htr3212_priv *priv)
+{
+	int ret = 0;
+
+	for_each_available_child_of_node_scoped(dev_of_node(dev), child) {
+		struct led_init_data init_data = {};
+		struct htr3212_led_data *led_data =
+			&priv->leds[priv->num_leds];
+		const struct htr3212_led_data *other_led_data;
+
+		led_data->priv = priv;
+
+		ret = htr3212_parse_child_dt(dev, child, led_data);
+		if (ret)
+			return ret;
+
+		other_led_data = htr3212_find_led_data(priv,
+							  led_data->channel);
+		if (other_led_data) {
+			dev_err(dev,
+				"Node %pOF 'reg' conflicts with another LED\n",
+				child);
+			return -EINVAL;
+		}
+
+		init_data.fwnode = of_fwnode_handle(child);
+
+		ret = devm_led_classdev_register_ext(dev, &led_data->cdev,
+						     &init_data);
+		if (ret) {
+			dev_err(dev, "Failed to register LED for %pOF: %d\n",
+				child, ret);
+			return ret;
+		}
+
+		priv->num_leds++;
+	}
+
+	return 0;
+}
+
+static const struct of_device_id of_htr3212_match[] = {
+	{ .compatible = "heroic,htr3212", },
+	{ /* sentinel */ }
+};
+
+MODULE_DEVICE_TABLE(of, of_htr3212_match);
+
+static int htr3212_probe(struct i2c_client *client)
+{
+	struct device *dev = &client->dev;
+	struct htr3212_priv *priv;
+	int count;
+	int ret = 0;
+
+	count = of_get_available_child_count(dev_of_node(dev));
+	if (!count)
+		return -EINVAL;
+
+	priv = devm_kzalloc(dev, struct_size(priv, leds, count),
+			    GFP_KERNEL);
+	if (!priv)
+		return -ENOMEM;
+
+	priv->sdb = devm_gpiod_get(dev, "sdb", GPIOD_OUT_HIGH);
+	if (PTR_ERR(priv->sdb) == -EPROBE_DEFER)
+		return -EPROBE_DEFER;
+
+	priv->vdd = devm_regulator_get(dev, "vdd");
+	if (IS_ERR(priv->vdd)) {
+		ret = PTR_ERR(priv->vdd);
+		return ret;
+	}
+
+	ret = regulator_enable(priv->vdd);
+	if (ret < 0) {
+		return ret;
+	}
+
+	gpiod_set_value_cansleep(priv->sdb, 1);
+	usleep_range(10000, 11000);
+
+	priv->client = client;
+	i2c_set_clientdata(client, priv);
+
+	ret = htr3212_init_regs(priv);
+	if (ret)
+		return ret;
+
+	ret = htr3212_parse_dt(dev, priv);
+	if (ret)
+		return ret;
+
+	return 0;
+}
+
+static void htr3212_remove(struct i2c_client *client)
+{
+	struct htr3212_priv *priv = i2c_get_clientdata(client);
+	int ret;
+
+	ret = htr3212_reset_regs(priv);
+	if (ret)
+		dev_err(&client->dev, "Failed to reset registers on removal (%pe)\n",
+			ERR_PTR(ret));
+
+	gpiod_set_value_cansleep(priv->sdb, 0);
+
+	regulator_disable(priv->vdd);
+}
+
+static const struct i2c_device_id htr3212_id[] = {
+	{ "htr3212" },
+	{},
+};
+
+MODULE_DEVICE_TABLE(i2c, htr3212_id);
+
+static struct i2c_driver htr3212_driver = {
+	.driver = {
+		.name	= "htr3212",
+		.of_match_table = of_htr3212_match,
+	},
+	.probe		= htr3212_probe,
+	.remove		= htr3212_remove,
+	.id_table	= htr3212_id,
+};
+
+module_i2c_driver(htr3212_driver);
+
+MODULE_AUTHOR("Teguh Sobirin <teguh@sobir.in");
+MODULE_DESCRIPTION("HEROIC HTR3212 LED Driver");
+MODULE_LICENSE("GPL v2");
-- 
2.34.1