1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (c) 2022 Qualcomm Innovation Center, Inc. All rights reserved.
4  */
5 
6 #include <linux/bitfield.h>
7 #include <linux/bits.h>
8 #include <linux/leds.h>
9 #include <linux/led-class-flash.h>
10 #include <linux/module.h>
11 #include <linux/platform_device.h>
12 #include <linux/property.h>
13 #include <linux/regmap.h>
14 #include <media/v4l2-flash-led-class.h>
15 
16 /* registers definitions */
17 #define FLASH_TYPE_REG			0x04
18 #define FLASH_TYPE_VAL			0x18
19 
20 #define FLASH_SUBTYPE_REG		0x05
21 #define FLASH_SUBTYPE_3CH_PM8150_VAL	0x04
22 #define FLASH_SUBTYPE_3CH_PMI8998_VAL	0x03
23 #define FLASH_SUBTYPE_4CH_VAL		0x07
24 
25 #define FLASH_STS_3CH_OTST1		BIT(0)
26 #define FLASH_STS_3CH_OTST2		BIT(1)
27 #define FLASH_STS_3CH_OTST3		BIT(2)
28 #define FLASH_STS_3CH_BOB_THM_OVERLOAD	BIT(3)
29 #define FLASH_STS_3CH_VPH_DROOP		BIT(4)
30 #define FLASH_STS_3CH_BOB_ILIM_S1	BIT(5)
31 #define FLASH_STS_3CH_BOB_ILIM_S2	BIT(6)
32 #define FLASH_STS_3CH_BCL_IBAT		BIT(7)
33 
34 #define FLASH_STS_4CH_VPH_LOW		BIT(0)
35 #define FLASH_STS_4CH_BCL_IBAT		BIT(1)
36 #define FLASH_STS_4CH_BOB_ILIM_S1	BIT(2)
37 #define FLASH_STS_4CH_BOB_ILIM_S2	BIT(3)
38 #define FLASH_STS_4CH_OTST2		BIT(4)
39 #define FLASH_STS_4CH_OTST1		BIT(5)
40 #define FLASH_STS_4CHG_BOB_THM_OVERLOAD	BIT(6)
41 
42 #define FLASH_TIMER_EN_BIT		BIT(7)
43 #define FLASH_TIMER_VAL_MASK		GENMASK(6, 0)
44 #define FLASH_TIMER_STEP_MS		10
45 
46 #define FLASH_STROBE_HW_SW_SEL_BIT	BIT(2)
47 #define SW_STROBE_VAL			0
48 #define HW_STROBE_VAL			1
49 #define FLASH_HW_STROBE_TRIGGER_SEL_BIT	BIT(1)
50 #define STROBE_LEVEL_TRIGGER_VAL	0
51 #define STROBE_EDGE_TRIGGER_VAL		1
52 #define FLASH_STROBE_POLARITY_BIT	BIT(0)
53 #define STROBE_ACTIVE_HIGH_VAL		1
54 
55 #define FLASH_IRES_MASK_4CH		BIT(0)
56 #define FLASH_IRES_MASK_3CH		GENMASK(1, 0)
57 #define FLASH_IRES_12P5MA_VAL		0
58 #define FLASH_IRES_5MA_VAL_4CH		1
59 #define FLASH_IRES_5MA_VAL_3CH		3
60 
61 /* constants */
62 #define FLASH_CURRENT_MAX_UA		1500000
63 #define TORCH_CURRENT_MAX_UA		500000
64 #define FLASH_TOTAL_CURRENT_MAX_UA	2000000
65 #define FLASH_CURRENT_DEFAULT_UA	1000000
66 #define TORCH_CURRENT_DEFAULT_UA	200000
67 
68 #define TORCH_IRES_UA			5000
69 #define FLASH_IRES_UA			12500
70 
71 #define FLASH_TIMEOUT_MAX_US		1280000
72 #define FLASH_TIMEOUT_STEP_US		10000
73 
74 #define UA_PER_MA			1000
75 
76 enum hw_type {
77 	QCOM_MVFLASH_3CH,
78 	QCOM_MVFLASH_4CH,
79 };
80 
81 enum led_mode {
82 	FLASH_MODE,
83 	TORCH_MODE,
84 };
85 
86 enum led_strobe {
87 	SW_STROBE,
88 	HW_STROBE,
89 };
90 
91 enum {
92 	REG_STATUS1,
93 	REG_STATUS2,
94 	REG_STATUS3,
95 	REG_CHAN_TIMER,
96 	REG_ITARGET,
97 	REG_MODULE_EN,
98 	REG_IRESOLUTION,
99 	REG_CHAN_STROBE,
100 	REG_CHAN_EN,
101 	REG_MAX_COUNT,
102 };
103 
104 static struct reg_field mvflash_3ch_regs[REG_MAX_COUNT] = {
105 	REG_FIELD(0x08, 0, 7),			/* status1	*/
106 	REG_FIELD(0x09, 0, 7),                  /* status2	*/
107 	REG_FIELD(0x0a, 0, 7),                  /* status3	*/
108 	REG_FIELD_ID(0x40, 0, 7, 3, 1),         /* chan_timer	*/
109 	REG_FIELD_ID(0x43, 0, 6, 3, 1),         /* itarget	*/
110 	REG_FIELD(0x46, 7, 7),                  /* module_en	*/
111 	REG_FIELD(0x47, 0, 5),                  /* iresolution	*/
112 	REG_FIELD_ID(0x49, 0, 2, 3, 1),         /* chan_strobe	*/
113 	REG_FIELD(0x4c, 0, 2),                  /* chan_en	*/
114 };
115 
116 static struct reg_field mvflash_4ch_regs[REG_MAX_COUNT] = {
117 	REG_FIELD(0x06, 0, 7),			/* status1	*/
118 	REG_FIELD(0x07, 0, 6),			/* status2	*/
119 	REG_FIELD(0x09, 0, 7),			/* status3	*/
120 	REG_FIELD_ID(0x3e, 0, 7, 4, 1),		/* chan_timer	*/
121 	REG_FIELD_ID(0x42, 0, 6, 4, 1),		/* itarget	*/
122 	REG_FIELD(0x46, 7, 7),			/* module_en	*/
123 	REG_FIELD(0x49, 0, 3),			/* iresolution	*/
124 	REG_FIELD_ID(0x4a, 0, 6, 4, 1),		/* chan_strobe	*/
125 	REG_FIELD(0x4e, 0, 3),			/* chan_en	*/
126 };
127 
128 struct qcom_flash_data {
129 	struct v4l2_flash	**v4l2_flash;
130 	struct regmap_field     *r_fields[REG_MAX_COUNT];
131 	struct mutex		lock;
132 	enum hw_type		hw_type;
133 	u8			leds_count;
134 	u8			max_channels;
135 	u8			chan_en_bits;
136 };
137 
138 struct qcom_flash_led {
139 	struct qcom_flash_data		*flash_data;
140 	struct led_classdev_flash	flash;
141 	u32				max_flash_current_ma;
142 	u32				max_torch_current_ma;
143 	u32				max_timeout_ms;
144 	u32				flash_current_ma;
145 	u32				flash_timeout_ms;
146 	u8				*chan_id;
147 	u8				chan_count;
148 	bool				enabled;
149 };
150 
151 static int set_flash_module_en(struct qcom_flash_led *led, bool en)
152 {
153 	struct qcom_flash_data *flash_data = led->flash_data;
154 	u8 led_mask = 0, enable;
155 	int i, rc;
156 
157 	for (i = 0; i < led->chan_count; i++)
158 		led_mask |= BIT(led->chan_id[i]);
159 
160 	mutex_lock(&flash_data->lock);
161 	if (en)
162 		flash_data->chan_en_bits |= led_mask;
163 	else
164 		flash_data->chan_en_bits &= ~led_mask;
165 
166 	enable = !!flash_data->chan_en_bits;
167 	rc = regmap_field_write(flash_data->r_fields[REG_MODULE_EN], enable);
168 	if (rc)
169 		dev_err(led->flash.led_cdev.dev, "write module_en failed, rc=%d\n", rc);
170 	mutex_unlock(&flash_data->lock);
171 
172 	return rc;
173 }
174 
175 static int set_flash_current(struct qcom_flash_led *led, u32 current_ma, enum led_mode mode)
176 {
177 	struct qcom_flash_data *flash_data = led->flash_data;
178 	u32 itarg_ua, ires_ua;
179 	u8 shift, ires_mask = 0, ires_val = 0, chan_id;
180 	int i, rc;
181 
182 	/*
183 	 * Split the current across the channels and set the
184 	 * IRESOLUTION and ITARGET registers accordingly.
185 	 */
186 	itarg_ua = (current_ma * UA_PER_MA) / led->chan_count + 1;
187 	ires_ua = (mode == FLASH_MODE) ? FLASH_IRES_UA : TORCH_IRES_UA;
188 
189 	for (i = 0; i < led->chan_count; i++) {
190 		u8 itarget = 0;
191 
192 		if (itarg_ua > ires_ua)
193 			itarget = itarg_ua / ires_ua - 1;
194 
195 		chan_id = led->chan_id[i];
196 
197 		rc = regmap_fields_write(flash_data->r_fields[REG_ITARGET], chan_id, itarget);
198 		if (rc)
199 			return rc;
200 
201 		if (flash_data->hw_type == QCOM_MVFLASH_3CH) {
202 			shift = chan_id * 2;
203 			ires_mask |= FLASH_IRES_MASK_3CH << shift;
204 			ires_val |= ((mode == FLASH_MODE) ?
205 				(FLASH_IRES_12P5MA_VAL << shift) :
206 				(FLASH_IRES_5MA_VAL_3CH << shift));
207 		} else if (flash_data->hw_type == QCOM_MVFLASH_4CH) {
208 			shift = chan_id;
209 			ires_mask |= FLASH_IRES_MASK_4CH << shift;
210 			ires_val |= ((mode == FLASH_MODE) ?
211 				(FLASH_IRES_12P5MA_VAL << shift) :
212 				(FLASH_IRES_5MA_VAL_4CH << shift));
213 		} else {
214 			dev_err(led->flash.led_cdev.dev,
215 					"HW type %d is not supported\n", flash_data->hw_type);
216 			return -EOPNOTSUPP;
217 		}
218 	}
219 
220 	return regmap_field_update_bits(flash_data->r_fields[REG_IRESOLUTION], ires_mask, ires_val);
221 }
222 
223 static int set_flash_timeout(struct qcom_flash_led *led, u32 timeout_ms)
224 {
225 	struct qcom_flash_data *flash_data = led->flash_data;
226 	u8 timer, chan_id;
227 	int rc, i;
228 
229 	/* set SAFETY_TIMER for all the channels connected to the same LED */
230 	timeout_ms = min_t(u32, timeout_ms, led->max_timeout_ms);
231 
232 	for (i = 0; i < led->chan_count; i++) {
233 		chan_id = led->chan_id[i];
234 
235 		timer = timeout_ms / FLASH_TIMER_STEP_MS;
236 		timer = clamp_t(u8, timer, 0, FLASH_TIMER_VAL_MASK);
237 
238 		if (timeout_ms)
239 			timer |= FLASH_TIMER_EN_BIT;
240 
241 		rc = regmap_fields_write(flash_data->r_fields[REG_CHAN_TIMER], chan_id, timer);
242 		if (rc)
243 			return rc;
244 	}
245 
246 	return 0;
247 }
248 
249 static int set_flash_strobe(struct qcom_flash_led *led, enum led_strobe strobe, bool state)
250 {
251 	struct qcom_flash_data *flash_data = led->flash_data;
252 	u8 strobe_sel, chan_en, chan_id, chan_mask = 0;
253 	int rc, i;
254 
255 	/* Set SW strobe config for all channels connected to the LED */
256 	for (i = 0; i < led->chan_count; i++) {
257 		chan_id = led->chan_id[i];
258 
259 		if (strobe == SW_STROBE)
260 			strobe_sel = FIELD_PREP(FLASH_STROBE_HW_SW_SEL_BIT, SW_STROBE_VAL);
261 		else
262 			strobe_sel = FIELD_PREP(FLASH_STROBE_HW_SW_SEL_BIT, HW_STROBE_VAL);
263 
264 		strobe_sel |=
265 			FIELD_PREP(FLASH_HW_STROBE_TRIGGER_SEL_BIT, STROBE_LEVEL_TRIGGER_VAL) |
266 			FIELD_PREP(FLASH_STROBE_POLARITY_BIT, STROBE_ACTIVE_HIGH_VAL);
267 
268 		rc = regmap_fields_write(
269 				flash_data->r_fields[REG_CHAN_STROBE], chan_id, strobe_sel);
270 		if (rc)
271 			return rc;
272 
273 		chan_mask |= BIT(chan_id);
274 	}
275 
276 	/* Enable/disable flash channels */
277 	chan_en = state ? chan_mask : 0;
278 	rc = regmap_field_update_bits(flash_data->r_fields[REG_CHAN_EN], chan_mask, chan_en);
279 	if (rc)
280 		return rc;
281 
282 	led->enabled = state;
283 	return 0;
284 }
285 
286 static inline struct qcom_flash_led *flcdev_to_qcom_fled(struct led_classdev_flash *flcdev)
287 {
288 	return container_of(flcdev, struct qcom_flash_led, flash);
289 }
290 
291 static int qcom_flash_brightness_set(struct led_classdev_flash *fled_cdev, u32 brightness)
292 {
293 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
294 
295 	led->flash_current_ma = min_t(u32, led->max_flash_current_ma, brightness / UA_PER_MA);
296 	return 0;
297 }
298 
299 static int qcom_flash_timeout_set(struct led_classdev_flash *fled_cdev, u32 timeout)
300 {
301 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
302 
303 	led->flash_timeout_ms = timeout / USEC_PER_MSEC;
304 	return 0;
305 }
306 
307 static int qcom_flash_strobe_set(struct led_classdev_flash *fled_cdev, bool state)
308 {
309 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
310 	int rc;
311 
312 	rc = set_flash_strobe(led, SW_STROBE, false);
313 	if (rc)
314 		return rc;
315 
316 	rc = set_flash_current(led, led->flash_current_ma, FLASH_MODE);
317 	if (rc)
318 		return rc;
319 
320 	rc = set_flash_timeout(led, led->flash_timeout_ms);
321 	if (rc)
322 		return rc;
323 
324 	rc = set_flash_module_en(led, state);
325 	if (rc)
326 		return rc;
327 
328 	return set_flash_strobe(led, SW_STROBE, state);
329 }
330 
331 static int qcom_flash_strobe_get(struct led_classdev_flash *fled_cdev, bool *state)
332 {
333 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
334 
335 	*state = led->enabled;
336 	return 0;
337 }
338 
339 static int qcom_flash_fault_get(struct led_classdev_flash *fled_cdev, u32 *fault)
340 {
341 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
342 	struct qcom_flash_data *flash_data = led->flash_data;
343 	u8 shift, chan_id, chan_mask = 0;
344 	u8 ot_mask = 0, oc_mask = 0, uv_mask = 0;
345 	u32 val, fault_sts = 0;
346 	int i, rc;
347 
348 	rc = regmap_field_read(flash_data->r_fields[REG_STATUS1], &val);
349 	if (rc)
350 		return rc;
351 
352 	for (i = 0; i < led->chan_count; i++) {
353 		chan_id = led->chan_id[i];
354 		shift = chan_id * 2;
355 
356 		if (val & BIT(shift))
357 			fault_sts |= LED_FAULT_SHORT_CIRCUIT;
358 
359 		chan_mask |= BIT(chan_id);
360 	}
361 
362 	rc = regmap_field_read(flash_data->r_fields[REG_STATUS2], &val);
363 	if (rc)
364 		return rc;
365 
366 	if (flash_data->hw_type == QCOM_MVFLASH_3CH) {
367 		ot_mask = FLASH_STS_3CH_OTST1 |
368 			  FLASH_STS_3CH_OTST2 |
369 			  FLASH_STS_3CH_OTST3 |
370 			  FLASH_STS_3CH_BOB_THM_OVERLOAD;
371 		oc_mask = FLASH_STS_3CH_BOB_ILIM_S1 |
372 			  FLASH_STS_3CH_BOB_ILIM_S2 |
373 			  FLASH_STS_3CH_BCL_IBAT;
374 		uv_mask = FLASH_STS_3CH_VPH_DROOP;
375 	} else if (flash_data->hw_type == QCOM_MVFLASH_4CH) {
376 		ot_mask = FLASH_STS_4CH_OTST2 |
377 			  FLASH_STS_4CH_OTST1 |
378 			  FLASH_STS_4CHG_BOB_THM_OVERLOAD;
379 		oc_mask = FLASH_STS_4CH_BCL_IBAT |
380 			  FLASH_STS_4CH_BOB_ILIM_S1 |
381 			  FLASH_STS_4CH_BOB_ILIM_S2;
382 		uv_mask = FLASH_STS_4CH_VPH_LOW;
383 	}
384 
385 	if (val & ot_mask)
386 		fault_sts |= LED_FAULT_OVER_TEMPERATURE;
387 
388 	if (val & oc_mask)
389 		fault_sts |= LED_FAULT_OVER_CURRENT;
390 
391 	if (val & uv_mask)
392 		fault_sts |= LED_FAULT_INPUT_VOLTAGE;
393 
394 	rc = regmap_field_read(flash_data->r_fields[REG_STATUS3], &val);
395 	if (rc)
396 		return rc;
397 
398 	if (flash_data->hw_type == QCOM_MVFLASH_3CH) {
399 		if (val & chan_mask)
400 			fault_sts |= LED_FAULT_TIMEOUT;
401 	} else if (flash_data->hw_type == QCOM_MVFLASH_4CH) {
402 		for (i = 0; i < led->chan_count; i++) {
403 			chan_id = led->chan_id[i];
404 			shift = chan_id * 2;
405 
406 			if (val & BIT(shift))
407 				fault_sts |= LED_FAULT_TIMEOUT;
408 		}
409 	}
410 
411 	*fault = fault_sts;
412 	return 0;
413 }
414 
415 static int qcom_flash_led_brightness_set(struct led_classdev *led_cdev,
416 					enum led_brightness brightness)
417 {
418 	struct led_classdev_flash *fled_cdev = lcdev_to_flcdev(led_cdev);
419 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
420 	u32 current_ma = brightness * led->max_torch_current_ma / LED_FULL;
421 	bool enable = !!brightness;
422 	int rc;
423 
424 	rc = set_flash_strobe(led, SW_STROBE, false);
425 	if (rc)
426 		return rc;
427 
428 	rc = set_flash_module_en(led, false);
429 	if (rc)
430 		return rc;
431 
432 	rc = set_flash_current(led, current_ma, TORCH_MODE);
433 	if (rc)
434 		return rc;
435 
436 	/* Disable flash timeout for torch LED */
437 	rc = set_flash_timeout(led, 0);
438 	if (rc)
439 		return rc;
440 
441 	rc = set_flash_module_en(led, enable);
442 	if (rc)
443 		return rc;
444 
445 	return set_flash_strobe(led, SW_STROBE, enable);
446 }
447 
448 static const struct led_flash_ops qcom_flash_ops = {
449 	.flash_brightness_set = qcom_flash_brightness_set,
450 	.strobe_set = qcom_flash_strobe_set,
451 	.strobe_get = qcom_flash_strobe_get,
452 	.timeout_set = qcom_flash_timeout_set,
453 	.fault_get = qcom_flash_fault_get,
454 };
455 
456 #if IS_ENABLED(CONFIG_V4L2_FLASH_LED_CLASS)
457 static int qcom_flash_external_strobe_set(struct v4l2_flash *v4l2_flash, bool enable)
458 {
459 	struct led_classdev_flash *fled_cdev = v4l2_flash->fled_cdev;
460 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
461 	int rc;
462 
463 	rc = set_flash_module_en(led, enable);
464 	if (rc)
465 		return rc;
466 
467 	if (enable)
468 		return set_flash_strobe(led, HW_STROBE, true);
469 	else
470 		return set_flash_strobe(led, SW_STROBE, false);
471 }
472 
473 static enum led_brightness
474 qcom_flash_intensity_to_led_brightness(struct v4l2_flash *v4l2_flash, s32 intensity)
475 {
476 	struct led_classdev_flash *fled_cdev = v4l2_flash->fled_cdev;
477 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
478 	u32 current_ma = intensity / UA_PER_MA;
479 
480 	current_ma = min_t(u32, current_ma, led->max_torch_current_ma);
481 	if (!current_ma)
482 		return LED_OFF;
483 
484 	return (current_ma * LED_FULL) / led->max_torch_current_ma;
485 }
486 
487 static s32 qcom_flash_brightness_to_led_intensity(struct v4l2_flash *v4l2_flash,
488 					enum led_brightness brightness)
489 {
490 	struct led_classdev_flash *fled_cdev = v4l2_flash->fled_cdev;
491 	struct qcom_flash_led *led = flcdev_to_qcom_fled(fled_cdev);
492 
493 	return (brightness * led->max_torch_current_ma * UA_PER_MA) / LED_FULL;
494 }
495 
496 static const struct v4l2_flash_ops qcom_v4l2_flash_ops = {
497 	.external_strobe_set = qcom_flash_external_strobe_set,
498 	.intensity_to_led_brightness = qcom_flash_intensity_to_led_brightness,
499 	.led_brightness_to_intensity = qcom_flash_brightness_to_led_intensity,
500 };
501 
502 static int
503 qcom_flash_v4l2_init(struct device *dev, struct qcom_flash_led *led, struct fwnode_handle *fwnode)
504 {
505 	struct qcom_flash_data *flash_data = led->flash_data;
506 	struct v4l2_flash_config v4l2_cfg = { 0 };
507 	struct led_flash_setting *intensity = &v4l2_cfg.intensity;
508 
509 	if (!(led->flash.led_cdev.flags & LED_DEV_CAP_FLASH))
510 		return 0;
511 
512 	intensity->min = intensity->step = TORCH_IRES_UA * led->chan_count;
513 	intensity->max = led->max_torch_current_ma * UA_PER_MA;
514 	intensity->val = min_t(u32, intensity->max, TORCH_CURRENT_DEFAULT_UA);
515 
516 	strscpy(v4l2_cfg.dev_name, led->flash.led_cdev.dev->kobj.name,
517 					sizeof(v4l2_cfg.dev_name));
518 
519 	v4l2_cfg.has_external_strobe = true;
520 	v4l2_cfg.flash_faults = LED_FAULT_INPUT_VOLTAGE |
521 				LED_FAULT_OVER_CURRENT |
522 				LED_FAULT_SHORT_CIRCUIT |
523 				LED_FAULT_OVER_TEMPERATURE |
524 				LED_FAULT_TIMEOUT;
525 
526 	flash_data->v4l2_flash[flash_data->leds_count] =
527 		v4l2_flash_init(dev, fwnode, &led->flash, &qcom_v4l2_flash_ops, &v4l2_cfg);
528 	return PTR_ERR_OR_ZERO(flash_data->v4l2_flash);
529 }
530 # else
531 static int
532 qcom_flash_v4l2_init(struct device *dev, struct qcom_flash_led *led, struct fwnode_handle *fwnode)
533 {
534 	return 0;
535 }
536 #endif
537 
538 static int qcom_flash_register_led_device(struct device *dev,
539 		struct fwnode_handle *node, struct qcom_flash_led *led)
540 {
541 	struct qcom_flash_data *flash_data = led->flash_data;
542 	struct led_init_data init_data;
543 	struct led_classdev_flash *flash = &led->flash;
544 	struct led_flash_setting *brightness, *timeout;
545 	u32 current_ua, timeout_us;
546 	u32 channels[4];
547 	int i, rc, count;
548 
549 	count = fwnode_property_count_u32(node, "led-sources");
550 	if (count <= 0) {
551 		dev_err(dev, "No led-sources specified\n");
552 		return -ENODEV;
553 	}
554 
555 	if (count > flash_data->max_channels) {
556 		dev_err(dev, "led-sources count %u exceeds maximum channel count %u\n",
557 				count, flash_data->max_channels);
558 		return -EINVAL;
559 	}
560 
561 	rc = fwnode_property_read_u32_array(node, "led-sources", channels, count);
562 	if (rc < 0) {
563 		dev_err(dev, "Failed to read led-sources property, rc=%d\n", rc);
564 		return rc;
565 	}
566 
567 	led->chan_count = count;
568 	led->chan_id = devm_kcalloc(dev, count, sizeof(u8), GFP_KERNEL);
569 	if (!led->chan_id)
570 		return -ENOMEM;
571 
572 	for (i = 0; i < count; i++) {
573 		if ((channels[i] == 0) || (channels[i] > flash_data->max_channels)) {
574 			dev_err(dev, "led-source out of HW support range [1-%u]\n",
575 					flash_data->max_channels);
576 			return -EINVAL;
577 		}
578 
579 		/* Make chan_id indexing from 0 */
580 		led->chan_id[i] = channels[i] - 1;
581 	}
582 
583 	rc = fwnode_property_read_u32(node, "led-max-microamp", &current_ua);
584 	if (rc < 0) {
585 		dev_err(dev, "Failed to read led-max-microamp property, rc=%d\n", rc);
586 		return rc;
587 	}
588 
589 	if (current_ua == 0) {
590 		dev_err(dev, "led-max-microamp shouldn't be 0\n");
591 		return -EINVAL;
592 	}
593 
594 	current_ua = min_t(u32, current_ua, TORCH_CURRENT_MAX_UA * led->chan_count);
595 	led->max_torch_current_ma = current_ua / UA_PER_MA;
596 
597 	if (fwnode_property_present(node, "flash-max-microamp")) {
598 		flash->led_cdev.flags |= LED_DEV_CAP_FLASH;
599 
600 		rc = fwnode_property_read_u32(node, "flash-max-microamp", &current_ua);
601 		if (rc < 0) {
602 			dev_err(dev, "Failed to read flash-max-microamp property, rc=%d\n",
603 					rc);
604 			return rc;
605 		}
606 
607 		current_ua = min_t(u32, current_ua, FLASH_CURRENT_MAX_UA * led->chan_count);
608 		current_ua = min_t(u32, current_ua, FLASH_TOTAL_CURRENT_MAX_UA);
609 
610 		/* Initialize flash class LED device brightness settings */
611 		brightness = &flash->brightness;
612 		brightness->min = brightness->step = FLASH_IRES_UA * led->chan_count;
613 		brightness->max = current_ua;
614 		brightness->val = min_t(u32, current_ua, FLASH_CURRENT_DEFAULT_UA);
615 
616 		led->max_flash_current_ma = current_ua / UA_PER_MA;
617 		led->flash_current_ma = brightness->val / UA_PER_MA;
618 
619 		rc = fwnode_property_read_u32(node, "flash-max-timeout-us", &timeout_us);
620 		if (rc < 0) {
621 			dev_err(dev, "Failed to read flash-max-timeout-us property, rc=%d\n",
622 					rc);
623 			return rc;
624 		}
625 
626 		timeout_us = min_t(u32, timeout_us, FLASH_TIMEOUT_MAX_US);
627 
628 		/* Initialize flash class LED device timeout settings */
629 		timeout = &flash->timeout;
630 		timeout->min = timeout->step = FLASH_TIMEOUT_STEP_US;
631 		timeout->val = timeout->max = timeout_us;
632 
633 		led->max_timeout_ms = led->flash_timeout_ms = timeout_us / USEC_PER_MSEC;
634 
635 		flash->ops = &qcom_flash_ops;
636 	}
637 
638 	flash->led_cdev.brightness_set_blocking = qcom_flash_led_brightness_set;
639 
640 	init_data.fwnode = node;
641 	init_data.devicename = NULL;
642 	init_data.default_label = NULL;
643 	init_data.devname_mandatory = false;
644 
645 	rc = devm_led_classdev_flash_register_ext(dev, flash, &init_data);
646 	if (rc < 0) {
647 		dev_err(dev, "Register flash LED classdev failed, rc=%d\n", rc);
648 		return rc;
649 	}
650 
651 	return qcom_flash_v4l2_init(dev, led, node);
652 }
653 
654 static int qcom_flash_led_probe(struct platform_device *pdev)
655 {
656 	struct qcom_flash_data *flash_data;
657 	struct qcom_flash_led *led;
658 	struct fwnode_handle *child;
659 	struct device *dev = &pdev->dev;
660 	struct regmap *regmap;
661 	struct reg_field *regs;
662 	int count, i, rc;
663 	u32 val, reg_base;
664 
665 	flash_data = devm_kzalloc(dev, sizeof(*flash_data), GFP_KERNEL);
666 	if (!flash_data)
667 		return -ENOMEM;
668 
669 	regmap = dev_get_regmap(dev->parent, NULL);
670 	if (!regmap) {
671 		dev_err(dev, "Failed to get parent regmap\n");
672 		return -EINVAL;
673 	}
674 
675 	rc = fwnode_property_read_u32(dev->fwnode, "reg", &reg_base);
676 	if (rc < 0) {
677 		dev_err(dev, "Failed to get register base address, rc=%d\n", rc);
678 		return rc;
679 	}
680 
681 	rc = regmap_read(regmap, reg_base + FLASH_TYPE_REG, &val);
682 	if (rc < 0) {
683 		dev_err(dev, "Read flash LED module type failed, rc=%d\n", rc);
684 		return rc;
685 	}
686 
687 	if (val != FLASH_TYPE_VAL) {
688 		dev_err(dev, "type %#x is not a flash LED module\n", val);
689 		return -ENODEV;
690 	}
691 
692 	rc = regmap_read(regmap, reg_base + FLASH_SUBTYPE_REG, &val);
693 	if (rc < 0) {
694 		dev_err(dev, "Read flash LED module subtype failed, rc=%d\n", rc);
695 		return rc;
696 	}
697 
698 	if (val == FLASH_SUBTYPE_3CH_PM8150_VAL || val == FLASH_SUBTYPE_3CH_PMI8998_VAL) {
699 		flash_data->hw_type = QCOM_MVFLASH_3CH;
700 		flash_data->max_channels = 3;
701 		regs = mvflash_3ch_regs;
702 	} else if (val == FLASH_SUBTYPE_4CH_VAL) {
703 		flash_data->hw_type = QCOM_MVFLASH_4CH;
704 		flash_data->max_channels = 4;
705 		regs = mvflash_4ch_regs;
706 	} else {
707 		dev_err(dev, "flash LED subtype %#x is not yet supported\n", val);
708 		return -ENODEV;
709 	}
710 
711 	for (i = 0; i < REG_MAX_COUNT; i++)
712 		regs[i].reg += reg_base;
713 
714 	rc = devm_regmap_field_bulk_alloc(dev, regmap, flash_data->r_fields, regs, REG_MAX_COUNT);
715 	if (rc < 0) {
716 		dev_err(dev, "Failed to allocate regmap field, rc=%d\n", rc);
717 		return rc;
718 	}
719 
720 	platform_set_drvdata(pdev, flash_data);
721 	mutex_init(&flash_data->lock);
722 
723 	count = device_get_child_node_count(dev);
724 	if (count == 0 || count > flash_data->max_channels) {
725 		dev_err(dev, "No child or child count exceeds %d\n", flash_data->max_channels);
726 		return -EINVAL;
727 	}
728 
729 	flash_data->v4l2_flash = devm_kcalloc(dev, count,
730 			sizeof(*flash_data->v4l2_flash), GFP_KERNEL);
731 	if (!flash_data->v4l2_flash)
732 		return -ENOMEM;
733 
734 	device_for_each_child_node(dev, child) {
735 		led = devm_kzalloc(dev, sizeof(*led), GFP_KERNEL);
736 		if (!led) {
737 			rc = -ENOMEM;
738 			goto release;
739 		}
740 
741 		led->flash_data = flash_data;
742 		rc = qcom_flash_register_led_device(dev, child, led);
743 		if (rc < 0)
744 			goto release;
745 
746 		flash_data->leds_count++;
747 	}
748 
749 	return 0;
750 
751 release:
752 	fwnode_handle_put(child);
753 	while (flash_data->v4l2_flash[flash_data->leds_count] && flash_data->leds_count)
754 		v4l2_flash_release(flash_data->v4l2_flash[flash_data->leds_count--]);
755 	return rc;
756 }
757 
758 static int qcom_flash_led_remove(struct platform_device *pdev)
759 {
760 	struct qcom_flash_data *flash_data = platform_get_drvdata(pdev);
761 
762 	while (flash_data->v4l2_flash[flash_data->leds_count] && flash_data->leds_count)
763 		v4l2_flash_release(flash_data->v4l2_flash[flash_data->leds_count--]);
764 
765 	mutex_destroy(&flash_data->lock);
766 	return 0;
767 }
768 
769 static const struct of_device_id qcom_flash_led_match_table[] = {
770 	{ .compatible = "qcom,spmi-flash-led" },
771 	{ }
772 };
773 
774 MODULE_DEVICE_TABLE(of, qcom_flash_led_match_table);
775 static struct platform_driver qcom_flash_led_driver = {
776 	.driver = {
777 		.name = "leds-qcom-flash",
778 		.of_match_table = qcom_flash_led_match_table,
779 	},
780 	.probe = qcom_flash_led_probe,
781 	.remove = qcom_flash_led_remove,
782 };
783 
784 module_platform_driver(qcom_flash_led_driver);
785 
786 MODULE_DESCRIPTION("QCOM Flash LED driver");
787 MODULE_LICENSE("GPL");
788