1// SPDX-License-Identifier: GPL-2.0-only
2/* Copyright (c) 2010, 2011, 2016 The Linux Foundation. All rights reserved.
3 */
4#include <linux/leds.h>
5#include <linux/module.h>
6#include <linux/of.h>
7#include <linux/platform_device.h>
8#include <linux/pm.h>
9#include <linux/regmap.h>
10
11#define PM8058_LED_TYPE_COMMON	0x00
12#define PM8058_LED_TYPE_KEYPAD	0x01
13#define PM8058_LED_TYPE_FLASH	0x02
14
15#define PM8058_LED_TYPE_COMMON_MASK	0xf8
16#define PM8058_LED_TYPE_KEYPAD_MASK	0xf0
17#define PM8058_LED_TYPE_COMMON_SHIFT	3
18#define PM8058_LED_TYPE_KEYPAD_SHIFT	4
19
20struct pm8058_led {
21	struct regmap *map;
22	u32 reg;
23	u32 ledtype;
24	struct led_classdev cdev;
25};
26
27static void pm8058_led_set(struct led_classdev *cled,
28	enum led_brightness value)
29{
30	struct pm8058_led *led;
31	int ret = 0;
32	unsigned int mask = 0;
33	unsigned int val = 0;
34
35	led = container_of(cled, struct pm8058_led, cdev);
36	switch (led->ledtype) {
37	case PM8058_LED_TYPE_COMMON:
38		mask = PM8058_LED_TYPE_COMMON_MASK;
39		val = value << PM8058_LED_TYPE_COMMON_SHIFT;
40		break;
41	case PM8058_LED_TYPE_KEYPAD:
42	case PM8058_LED_TYPE_FLASH:
43		mask = PM8058_LED_TYPE_KEYPAD_MASK;
44		val = value << PM8058_LED_TYPE_KEYPAD_SHIFT;
45		break;
46	default:
47		break;
48	}
49
50	ret = regmap_update_bits(led->map, led->reg, mask, val);
51	if (ret)
52		pr_err("Failed to set LED brightness\n");
53}
54
55static enum led_brightness pm8058_led_get(struct led_classdev *cled)
56{
57	struct pm8058_led *led;
58	int ret;
59	unsigned int val;
60
61	led = container_of(cled, struct pm8058_led, cdev);
62
63	ret = regmap_read(led->map, led->reg, &val);
64	if (ret) {
65		pr_err("Failed to get LED brightness\n");
66		return LED_OFF;
67	}
68
69	switch (led->ledtype) {
70	case PM8058_LED_TYPE_COMMON:
71		val &= PM8058_LED_TYPE_COMMON_MASK;
72		val >>= PM8058_LED_TYPE_COMMON_SHIFT;
73		break;
74	case PM8058_LED_TYPE_KEYPAD:
75	case PM8058_LED_TYPE_FLASH:
76		val &= PM8058_LED_TYPE_KEYPAD_MASK;
77		val >>= PM8058_LED_TYPE_KEYPAD_SHIFT;
78		break;
79	default:
80		val = LED_OFF;
81		break;
82	}
83
84	return val;
85}
86
87static int pm8058_led_probe(struct platform_device *pdev)
88{
89	struct led_init_data init_data = {};
90	struct device *dev = &pdev->dev;
91	struct pm8058_led *led;
92	struct device_node *np;
93	int ret;
94	struct regmap *map;
95	enum led_brightness maxbright;
96	enum led_default_state state;
97
98	led = devm_kzalloc(dev, sizeof(*led), GFP_KERNEL);
99	if (!led)
100		return -ENOMEM;
101
102	led->ledtype = (u32)(unsigned long)of_device_get_match_data(dev);
103
104	map = dev_get_regmap(dev->parent, NULL);
105	if (!map) {
106		dev_err(dev, "Parent regmap unavailable.\n");
107		return -ENXIO;
108	}
109	led->map = map;
110
111	np = dev_of_node(dev);
112
113	ret = of_property_read_u32(np, "reg", &led->reg);
114	if (ret) {
115		dev_err(dev, "no register offset specified\n");
116		return -EINVAL;
117	}
118
119	led->cdev.brightness_set = pm8058_led_set;
120	led->cdev.brightness_get = pm8058_led_get;
121	if (led->ledtype == PM8058_LED_TYPE_COMMON)
122		maxbright = 31; /* 5 bits */
123	else
124		maxbright = 15; /* 4 bits */
125	led->cdev.max_brightness = maxbright;
126
127	init_data.fwnode = of_fwnode_handle(np);
128
129	state = led_init_default_state_get(init_data.fwnode);
130	switch (state) {
131	case LEDS_DEFSTATE_ON:
132		led->cdev.brightness = maxbright;
133		pm8058_led_set(&led->cdev, maxbright);
134		break;
135	case LEDS_DEFSTATE_KEEP:
136		led->cdev.brightness = pm8058_led_get(&led->cdev);
137		break;
138	default:
139		led->cdev.brightness = LED_OFF;
140		pm8058_led_set(&led->cdev, LED_OFF);
141	}
142
143	if (led->ledtype == PM8058_LED_TYPE_KEYPAD ||
144	    led->ledtype == PM8058_LED_TYPE_FLASH)
145		led->cdev.flags	= LED_CORE_SUSPENDRESUME;
146
147	ret = devm_led_classdev_register_ext(dev, &led->cdev, &init_data);
148	if (ret)
149		dev_err(dev, "Failed to register LED for %pOF\n", np);
150
151	return ret;
152}
153
154static const struct of_device_id pm8058_leds_id_table[] = {
155	{
156		.compatible = "qcom,pm8058-led",
157		.data = (void *)PM8058_LED_TYPE_COMMON
158	},
159	{
160		.compatible = "qcom,pm8058-keypad-led",
161		.data = (void *)PM8058_LED_TYPE_KEYPAD
162	},
163	{
164		.compatible = "qcom,pm8058-flash-led",
165		.data = (void *)PM8058_LED_TYPE_FLASH
166	},
167	{ },
168};
169MODULE_DEVICE_TABLE(of, pm8058_leds_id_table);
170
171static struct platform_driver pm8058_led_driver = {
172	.probe		= pm8058_led_probe,
173	.driver		= {
174		.name	= "pm8058-leds",
175		.of_match_table = pm8058_leds_id_table,
176	},
177};
178module_platform_driver(pm8058_led_driver);
179
180MODULE_DESCRIPTION("PM8058 LEDs driver");
181MODULE_LICENSE("GPL v2");
182MODULE_ALIAS("platform:pm8058-leds");
183