Linux Audio

Check our new training course

Loading...
v6.2
  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3* Simple driver for Texas Instruments LM355x LED Flash driver chip
  4* Copyright (C) 2012 Texas Instruments
 
 
 
 
  5*/
  6
  7#include <linux/module.h>
  8#include <linux/delay.h>
  9#include <linux/i2c.h>
 
 10#include <linux/leds.h>
 11#include <linux/slab.h>
 12#include <linux/platform_device.h>
 13#include <linux/fs.h>
 14#include <linux/regmap.h>
 15#include <linux/platform_data/leds-lm355x.h>
 16
 17enum lm355x_type {
 18	CHIP_LM3554 = 0,
 19	CHIP_LM3556,
 20};
 21
 22enum lm355x_regs {
 23	REG_FLAG = 0,
 24	REG_TORCH_CFG,
 25	REG_TORCH_CTRL,
 26	REG_STROBE_CFG,
 27	REG_FLASH_CTRL,
 28	REG_INDI_CFG,
 29	REG_INDI_CTRL,
 30	REG_OPMODE,
 31	REG_MAX,
 32};
 33
 34/* operation mode */
 35enum lm355x_mode {
 36	MODE_SHDN = 0,
 37	MODE_INDIC,
 38	MODE_TORCH,
 39	MODE_FLASH
 40};
 41
 42/* register map info. */
 43struct lm355x_reg_data {
 44	u8 regno;
 45	u8 mask;
 46	u8 shift;
 47};
 48
 49struct lm355x_chip_data {
 50	struct device *dev;
 51	enum lm355x_type type;
 52
 53	struct led_classdev cdev_flash;
 54	struct led_classdev cdev_torch;
 55	struct led_classdev cdev_indicator;
 56
 57	struct lm355x_platform_data *pdata;
 58	struct regmap *regmap;
 59	struct mutex lock;
 60
 61	unsigned int last_flag;
 62	struct lm355x_reg_data *regs;
 63};
 64
 65/* specific indicator function for lm3556 */
 66enum lm3556_indic_pulse_time {
 67	PULSE_TIME_0_MS = 0,
 68	PULSE_TIME_32_MS,
 69	PULSE_TIME_64_MS,
 70	PULSE_TIME_92_MS,
 71	PULSE_TIME_128_MS,
 72	PULSE_TIME_160_MS,
 73	PULSE_TIME_196_MS,
 74	PULSE_TIME_224_MS,
 75	PULSE_TIME_256_MS,
 76	PULSE_TIME_288_MS,
 77	PULSE_TIME_320_MS,
 78	PULSE_TIME_352_MS,
 79	PULSE_TIME_384_MS,
 80	PULSE_TIME_416_MS,
 81	PULSE_TIME_448_MS,
 82	PULSE_TIME_480_MS,
 83};
 84
 85enum lm3556_indic_n_blank {
 86	INDIC_N_BLANK_0 = 0,
 87	INDIC_N_BLANK_1,
 88	INDIC_N_BLANK_2,
 89	INDIC_N_BLANK_3,
 90	INDIC_N_BLANK_4,
 91	INDIC_N_BLANK_5,
 92	INDIC_N_BLANK_6,
 93	INDIC_N_BLANK_7,
 94	INDIC_N_BLANK_8,
 95	INDIC_N_BLANK_9,
 96	INDIC_N_BLANK_10,
 97	INDIC_N_BLANK_11,
 98	INDIC_N_BLANK_12,
 99	INDIC_N_BLANK_13,
100	INDIC_N_BLANK_14,
101	INDIC_N_BLANK_15,
102};
103
104enum lm3556_indic_period {
105	INDIC_PERIOD_0 = 0,
106	INDIC_PERIOD_1,
107	INDIC_PERIOD_2,
108	INDIC_PERIOD_3,
109	INDIC_PERIOD_4,
110	INDIC_PERIOD_5,
111	INDIC_PERIOD_6,
112	INDIC_PERIOD_7,
113};
114
115#define INDIC_PATTERN_SIZE 4
116
117struct indicator {
118	u8 blinking;
119	u8 period_cnt;
120};
121
122/* indicator pattern data only for lm3556 */
123static struct indicator indicator_pattern[INDIC_PATTERN_SIZE] = {
124	[0] = {(INDIC_N_BLANK_1 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_1},
125	[1] = {(INDIC_N_BLANK_15 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_2},
126	[2] = {(INDIC_N_BLANK_10 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_4},
127	[3] = {(INDIC_N_BLANK_5 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_7},
128};
129
130static struct lm355x_reg_data lm3554_regs[REG_MAX] = {
131	[REG_FLAG] = {0xD0, 0xBF, 0},
132	[REG_TORCH_CFG] = {0xE0, 0x80, 7},
133	[REG_TORCH_CTRL] = {0xA0, 0x38, 3},
134	[REG_STROBE_CFG] = {0xE0, 0x04, 2},
135	[REG_FLASH_CTRL] = {0xB0, 0x78, 3},
136	[REG_INDI_CFG] = {0xE0, 0x08, 3},
137	[REG_INDI_CTRL] = {0xA0, 0xC0, 6},
138	[REG_OPMODE] = {0xA0, 0x03, 0},
139};
140
141static struct lm355x_reg_data lm3556_regs[REG_MAX] = {
142	[REG_FLAG] = {0x0B, 0xFF, 0},
143	[REG_TORCH_CFG] = {0x0A, 0x10, 4},
144	[REG_TORCH_CTRL] = {0x09, 0x70, 4},
145	[REG_STROBE_CFG] = {0x0A, 0x20, 5},
146	[REG_FLASH_CTRL] = {0x09, 0x0F, 0},
147	[REG_INDI_CFG] = {0xFF, 0xFF, 0},
148	[REG_INDI_CTRL] = {0x09, 0x70, 4},
149	[REG_OPMODE] = {0x0A, 0x03, 0},
150};
151
152static char lm355x_name[][I2C_NAME_SIZE] = {
153	[CHIP_LM3554] = LM3554_NAME,
154	[CHIP_LM3556] = LM3556_NAME,
155};
156
157/* chip initialize */
158static int lm355x_chip_init(struct lm355x_chip_data *chip)
159{
160	int ret;
161	unsigned int reg_val;
162	struct lm355x_platform_data *pdata = chip->pdata;
163
164	/* input and output pins configuration */
165	switch (chip->type) {
166	case CHIP_LM3554:
167		reg_val = (u32)pdata->pin_tx2 | (u32)pdata->ntc_pin;
168		ret = regmap_update_bits(chip->regmap, 0xE0, 0x28, reg_val);
169		if (ret < 0)
170			goto out;
171		reg_val = (u32)pdata->pass_mode;
172		ret = regmap_update_bits(chip->regmap, 0xA0, 0x04, reg_val);
173		if (ret < 0)
174			goto out;
175		break;
176
177	case CHIP_LM3556:
178		reg_val = (u32)pdata->pin_tx2 | (u32)pdata->ntc_pin |
179		          (u32)pdata->pass_mode;
180		ret = regmap_update_bits(chip->regmap, 0x0A, 0xC4, reg_val);
181		if (ret < 0)
182			goto out;
183		break;
184	default:
185		return -ENODATA;
186	}
187
188	return ret;
189out:
190	dev_err(chip->dev, "%s:i2c access fail to register\n", __func__);
191	return ret;
192}
193
194/* chip control */
195static int lm355x_control(struct lm355x_chip_data *chip,
196			   u8 brightness, enum lm355x_mode opmode)
197{
198	int ret;
199	unsigned int reg_val;
200	struct lm355x_platform_data *pdata = chip->pdata;
201	struct lm355x_reg_data *preg = chip->regs;
202
203	ret = regmap_read(chip->regmap, preg[REG_FLAG].regno, &chip->last_flag);
204	if (ret < 0)
205		goto out;
206	if (chip->last_flag & preg[REG_FLAG].mask)
207		dev_info(chip->dev, "%s Last FLAG is 0x%x\n",
208			 lm355x_name[chip->type],
209			 chip->last_flag & preg[REG_FLAG].mask);
210	/* brightness 0 means shutdown */
211	if (!brightness)
212		opmode = MODE_SHDN;
213
214	switch (opmode) {
215	case MODE_TORCH:
216		ret =
217		    regmap_update_bits(chip->regmap, preg[REG_TORCH_CTRL].regno,
218				       preg[REG_TORCH_CTRL].mask,
219				       (brightness - 1)
220				       << preg[REG_TORCH_CTRL].shift);
221		if (ret < 0)
222			goto out;
223
224		if (pdata->pin_tx1 != LM355x_PIN_TORCH_DISABLE) {
225			ret =
226			    regmap_update_bits(chip->regmap,
227					       preg[REG_TORCH_CFG].regno,
228					       preg[REG_TORCH_CFG].mask,
229					       0x01 <<
230					       preg[REG_TORCH_CFG].shift);
231			if (ret < 0)
232				goto out;
233			opmode = MODE_SHDN;
234			dev_info(chip->dev,
235				 "torch brt is set - ext. torch pin mode\n");
236		}
237		break;
238
239	case MODE_FLASH:
240
241		ret =
242		    regmap_update_bits(chip->regmap, preg[REG_FLASH_CTRL].regno,
243				       preg[REG_FLASH_CTRL].mask,
244				       (brightness - 1)
245				       << preg[REG_FLASH_CTRL].shift);
246		if (ret < 0)
247			goto out;
248
249		if (pdata->pin_strobe != LM355x_PIN_STROBE_DISABLE) {
250			if (chip->type == CHIP_LM3554)
251				reg_val = 0x00;
252			else
253				reg_val = 0x01;
254			ret =
255			    regmap_update_bits(chip->regmap,
256					       preg[REG_STROBE_CFG].regno,
257					       preg[REG_STROBE_CFG].mask,
258					       reg_val <<
259					       preg[REG_STROBE_CFG].shift);
260			if (ret < 0)
261				goto out;
262			opmode = MODE_SHDN;
263			dev_info(chip->dev,
264				 "flash brt is set - ext. strobe pin mode\n");
265		}
266		break;
267
268	case MODE_INDIC:
269		ret =
270		    regmap_update_bits(chip->regmap, preg[REG_INDI_CTRL].regno,
271				       preg[REG_INDI_CTRL].mask,
272				       (brightness - 1)
273				       << preg[REG_INDI_CTRL].shift);
274		if (ret < 0)
275			goto out;
276
277		if (pdata->pin_tx2 != LM355x_PIN_TX_DISABLE) {
278			ret =
279			    regmap_update_bits(chip->regmap,
280					       preg[REG_INDI_CFG].regno,
281					       preg[REG_INDI_CFG].mask,
282					       0x01 <<
283					       preg[REG_INDI_CFG].shift);
284			if (ret < 0)
285				goto out;
286			opmode = MODE_SHDN;
287		}
288		break;
289	case MODE_SHDN:
290		break;
291	default:
292		return -EINVAL;
293	}
294	/* operation mode control */
295	ret = regmap_update_bits(chip->regmap, preg[REG_OPMODE].regno,
296				 preg[REG_OPMODE].mask,
297				 opmode << preg[REG_OPMODE].shift);
298	if (ret < 0)
299		goto out;
300	return ret;
301out:
302	dev_err(chip->dev, "%s:i2c access fail to register\n", __func__);
303	return ret;
304}
305
306/* torch */
307
308static int lm355x_torch_brightness_set(struct led_classdev *cdev,
309					enum led_brightness brightness)
310{
311	struct lm355x_chip_data *chip =
312	    container_of(cdev, struct lm355x_chip_data, cdev_torch);
313	int ret;
314
315	mutex_lock(&chip->lock);
316	ret = lm355x_control(chip, brightness, MODE_TORCH);
317	mutex_unlock(&chip->lock);
318	return ret;
319}
320
321/* flash */
322
323static int lm355x_strobe_brightness_set(struct led_classdev *cdev,
324					 enum led_brightness brightness)
325{
326	struct lm355x_chip_data *chip =
327	    container_of(cdev, struct lm355x_chip_data, cdev_flash);
328	int ret;
329
330	mutex_lock(&chip->lock);
331	ret = lm355x_control(chip, brightness, MODE_FLASH);
332	mutex_unlock(&chip->lock);
333	return ret;
334}
335
336/* indicator */
337
338static int lm355x_indicator_brightness_set(struct led_classdev *cdev,
339					    enum led_brightness brightness)
340{
341	struct lm355x_chip_data *chip =
342	    container_of(cdev, struct lm355x_chip_data, cdev_indicator);
343	int ret;
344
345	mutex_lock(&chip->lock);
346	ret = lm355x_control(chip, brightness, MODE_INDIC);
347	mutex_unlock(&chip->lock);
348	return ret;
349}
350
351/* indicator pattern only for lm3556*/
352static ssize_t pattern_store(struct device *dev,
353			     struct device_attribute *attr,
354			     const char *buf, size_t size)
355{
356	ssize_t ret;
357	struct led_classdev *led_cdev = dev_get_drvdata(dev);
358	struct lm355x_chip_data *chip =
359	    container_of(led_cdev, struct lm355x_chip_data, cdev_indicator);
360	unsigned int state;
361
362	ret = kstrtouint(buf, 10, &state);
363	if (ret)
364		goto out;
365	if (state > INDIC_PATTERN_SIZE - 1)
366		state = INDIC_PATTERN_SIZE - 1;
367
368	ret = regmap_write(chip->regmap, 0x04,
369			   indicator_pattern[state].blinking);
370	if (ret < 0)
371		goto out;
372
373	ret = regmap_write(chip->regmap, 0x05,
374			   indicator_pattern[state].period_cnt);
375	if (ret < 0)
376		goto out;
377
378	return size;
379out:
380	dev_err(chip->dev, "%s:i2c access fail to register\n", __func__);
381	return ret;
382}
383
384static DEVICE_ATTR_WO(pattern);
385
386static struct attribute *lm355x_indicator_attrs[] = {
387	&dev_attr_pattern.attr,
388	NULL
389};
390ATTRIBUTE_GROUPS(lm355x_indicator);
391
392static const struct regmap_config lm355x_regmap = {
393	.reg_bits = 8,
394	.val_bits = 8,
395	.max_register = 0xFF,
396};
397
398/* module initialize */
399static int lm355x_probe(struct i2c_client *client,
400				  const struct i2c_device_id *id)
401{
402	struct lm355x_platform_data *pdata = dev_get_platdata(&client->dev);
403	struct lm355x_chip_data *chip;
404
405	int err;
406
407	if (!i2c_check_functionality(client->adapter, I2C_FUNC_I2C)) {
408		dev_err(&client->dev, "i2c functionality check fail.\n");
409		return -EOPNOTSUPP;
410	}
411
412	if (pdata == NULL) {
413		dev_err(&client->dev, "needs Platform Data.\n");
414		return -ENODATA;
415	}
416
417	chip = devm_kzalloc(&client->dev,
418			    sizeof(struct lm355x_chip_data), GFP_KERNEL);
419	if (!chip)
420		return -ENOMEM;
421
422	chip->dev = &client->dev;
423	chip->type = id->driver_data;
424	switch (id->driver_data) {
425	case CHIP_LM3554:
426		chip->regs = lm3554_regs;
427		break;
428	case CHIP_LM3556:
429		chip->regs = lm3556_regs;
430		break;
431	default:
432		return -ENOSYS;
433	}
434	chip->pdata = pdata;
435
436	chip->regmap = devm_regmap_init_i2c(client, &lm355x_regmap);
437	if (IS_ERR(chip->regmap)) {
438		err = PTR_ERR(chip->regmap);
439		dev_err(&client->dev,
440			"Failed to allocate register map: %d\n", err);
441		return err;
442	}
443
444	mutex_init(&chip->lock);
445	i2c_set_clientdata(client, chip);
446
447	err = lm355x_chip_init(chip);
448	if (err < 0)
449		goto err_out;
450
451	/* flash */
452	chip->cdev_flash.name = "flash";
453	chip->cdev_flash.max_brightness = 16;
454	chip->cdev_flash.brightness_set_blocking = lm355x_strobe_brightness_set;
455	chip->cdev_flash.default_trigger = "flash";
456	err = led_classdev_register(&client->dev, &chip->cdev_flash);
 
457	if (err < 0)
458		goto err_out;
459	/* torch */
460	chip->cdev_torch.name = "torch";
461	chip->cdev_torch.max_brightness = 8;
462	chip->cdev_torch.brightness_set_blocking = lm355x_torch_brightness_set;
463	chip->cdev_torch.default_trigger = "torch";
464	err = led_classdev_register(&client->dev, &chip->cdev_torch);
 
465	if (err < 0)
466		goto err_create_torch_file;
467	/* indicator */
468	chip->cdev_indicator.name = "indicator";
469	if (id->driver_data == CHIP_LM3554)
470		chip->cdev_indicator.max_brightness = 4;
471	else
472		chip->cdev_indicator.max_brightness = 8;
473	chip->cdev_indicator.brightness_set_blocking =
474					lm355x_indicator_brightness_set;
475	/* indicator pattern control only for LM3556 */
476	if (id->driver_data == CHIP_LM3556)
477		chip->cdev_indicator.groups = lm355x_indicator_groups;
478	err = led_classdev_register(&client->dev, &chip->cdev_indicator);
 
479	if (err < 0)
480		goto err_create_indicator_file;
481
482	dev_info(&client->dev, "%s is initialized\n",
483		 lm355x_name[id->driver_data]);
484	return 0;
485
486err_create_indicator_file:
487	led_classdev_unregister(&chip->cdev_torch);
488err_create_torch_file:
489	led_classdev_unregister(&chip->cdev_flash);
490err_out:
491	return err;
492}
493
494static void lm355x_remove(struct i2c_client *client)
495{
496	struct lm355x_chip_data *chip = i2c_get_clientdata(client);
497	struct lm355x_reg_data *preg = chip->regs;
498
499	regmap_write(chip->regmap, preg[REG_OPMODE].regno, 0);
500	led_classdev_unregister(&chip->cdev_indicator);
501	led_classdev_unregister(&chip->cdev_torch);
502	led_classdev_unregister(&chip->cdev_flash);
503	dev_info(&client->dev, "%s is removed\n", lm355x_name[chip->type]);
 
 
504}
505
506static const struct i2c_device_id lm355x_id[] = {
507	{LM3554_NAME, CHIP_LM3554},
508	{LM3556_NAME, CHIP_LM3556},
509	{}
510};
511
512MODULE_DEVICE_TABLE(i2c, lm355x_id);
513
514static struct i2c_driver lm355x_i2c_driver = {
515	.driver = {
516		   .name = LM355x_NAME,
517		   .pm = NULL,
518		   },
519	.probe = lm355x_probe,
520	.remove = lm355x_remove,
521	.id_table = lm355x_id,
522};
523
524module_i2c_driver(lm355x_i2c_driver);
525
526MODULE_DESCRIPTION("Texas Instruments Flash Lighting driver for LM355x");
527MODULE_AUTHOR("Daniel Jeong <daniel.jeong@ti.com>");
528MODULE_AUTHOR("G.Shark Jeong <gshark.jeong@gmail.com>");
529MODULE_LICENSE("GPL v2");
v4.10.11
 
  1/*
  2* Simple driver for Texas Instruments LM355x LED Flash driver chip
  3* Copyright (C) 2012 Texas Instruments
  4*
  5* This program is free software; you can redistribute it and/or modify
  6* it under the terms of the GNU General Public License version 2 as
  7* published by the Free Software Foundation.
  8*/
  9
 10#include <linux/module.h>
 11#include <linux/delay.h>
 12#include <linux/i2c.h>
 13#include <linux/gpio.h>
 14#include <linux/leds.h>
 15#include <linux/slab.h>
 16#include <linux/platform_device.h>
 17#include <linux/fs.h>
 18#include <linux/regmap.h>
 19#include <linux/platform_data/leds-lm355x.h>
 20
 21enum lm355x_type {
 22	CHIP_LM3554 = 0,
 23	CHIP_LM3556,
 24};
 25
 26enum lm355x_regs {
 27	REG_FLAG = 0,
 28	REG_TORCH_CFG,
 29	REG_TORCH_CTRL,
 30	REG_STROBE_CFG,
 31	REG_FLASH_CTRL,
 32	REG_INDI_CFG,
 33	REG_INDI_CTRL,
 34	REG_OPMODE,
 35	REG_MAX,
 36};
 37
 38/* operation mode */
 39enum lm355x_mode {
 40	MODE_SHDN = 0,
 41	MODE_INDIC,
 42	MODE_TORCH,
 43	MODE_FLASH
 44};
 45
 46/* register map info. */
 47struct lm355x_reg_data {
 48	u8 regno;
 49	u8 mask;
 50	u8 shift;
 51};
 52
 53struct lm355x_chip_data {
 54	struct device *dev;
 55	enum lm355x_type type;
 56
 57	struct led_classdev cdev_flash;
 58	struct led_classdev cdev_torch;
 59	struct led_classdev cdev_indicator;
 60
 61	struct lm355x_platform_data *pdata;
 62	struct regmap *regmap;
 63	struct mutex lock;
 64
 65	unsigned int last_flag;
 66	struct lm355x_reg_data *regs;
 67};
 68
 69/* specific indicator function for lm3556 */
 70enum lm3556_indic_pulse_time {
 71	PULSE_TIME_0_MS = 0,
 72	PULSE_TIME_32_MS,
 73	PULSE_TIME_64_MS,
 74	PULSE_TIME_92_MS,
 75	PULSE_TIME_128_MS,
 76	PULSE_TIME_160_MS,
 77	PULSE_TIME_196_MS,
 78	PULSE_TIME_224_MS,
 79	PULSE_TIME_256_MS,
 80	PULSE_TIME_288_MS,
 81	PULSE_TIME_320_MS,
 82	PULSE_TIME_352_MS,
 83	PULSE_TIME_384_MS,
 84	PULSE_TIME_416_MS,
 85	PULSE_TIME_448_MS,
 86	PULSE_TIME_480_MS,
 87};
 88
 89enum lm3556_indic_n_blank {
 90	INDIC_N_BLANK_0 = 0,
 91	INDIC_N_BLANK_1,
 92	INDIC_N_BLANK_2,
 93	INDIC_N_BLANK_3,
 94	INDIC_N_BLANK_4,
 95	INDIC_N_BLANK_5,
 96	INDIC_N_BLANK_6,
 97	INDIC_N_BLANK_7,
 98	INDIC_N_BLANK_8,
 99	INDIC_N_BLANK_9,
100	INDIC_N_BLANK_10,
101	INDIC_N_BLANK_11,
102	INDIC_N_BLANK_12,
103	INDIC_N_BLANK_13,
104	INDIC_N_BLANK_14,
105	INDIC_N_BLANK_15,
106};
107
108enum lm3556_indic_period {
109	INDIC_PERIOD_0 = 0,
110	INDIC_PERIOD_1,
111	INDIC_PERIOD_2,
112	INDIC_PERIOD_3,
113	INDIC_PERIOD_4,
114	INDIC_PERIOD_5,
115	INDIC_PERIOD_6,
116	INDIC_PERIOD_7,
117};
118
119#define INDIC_PATTERN_SIZE 4
120
121struct indicator {
122	u8 blinking;
123	u8 period_cnt;
124};
125
126/* indicator pattern data only for lm3556 */
127static struct indicator indicator_pattern[INDIC_PATTERN_SIZE] = {
128	[0] = {(INDIC_N_BLANK_1 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_1},
129	[1] = {(INDIC_N_BLANK_15 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_2},
130	[2] = {(INDIC_N_BLANK_10 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_4},
131	[3] = {(INDIC_N_BLANK_5 << 4) | PULSE_TIME_32_MS, INDIC_PERIOD_7},
132};
133
134static struct lm355x_reg_data lm3554_regs[REG_MAX] = {
135	[REG_FLAG] = {0xD0, 0xBF, 0},
136	[REG_TORCH_CFG] = {0xE0, 0x80, 7},
137	[REG_TORCH_CTRL] = {0xA0, 0x38, 3},
138	[REG_STROBE_CFG] = {0xE0, 0x04, 2},
139	[REG_FLASH_CTRL] = {0xB0, 0x78, 3},
140	[REG_INDI_CFG] = {0xE0, 0x08, 3},
141	[REG_INDI_CTRL] = {0xA0, 0xC0, 6},
142	[REG_OPMODE] = {0xA0, 0x03, 0},
143};
144
145static struct lm355x_reg_data lm3556_regs[REG_MAX] = {
146	[REG_FLAG] = {0x0B, 0xFF, 0},
147	[REG_TORCH_CFG] = {0x0A, 0x10, 4},
148	[REG_TORCH_CTRL] = {0x09, 0x70, 4},
149	[REG_STROBE_CFG] = {0x0A, 0x20, 5},
150	[REG_FLASH_CTRL] = {0x09, 0x0F, 0},
151	[REG_INDI_CFG] = {0xFF, 0xFF, 0},
152	[REG_INDI_CTRL] = {0x09, 0x70, 4},
153	[REG_OPMODE] = {0x0A, 0x03, 0},
154};
155
156static char lm355x_name[][I2C_NAME_SIZE] = {
157	[CHIP_LM3554] = LM3554_NAME,
158	[CHIP_LM3556] = LM3556_NAME,
159};
160
161/* chip initialize */
162static int lm355x_chip_init(struct lm355x_chip_data *chip)
163{
164	int ret;
165	unsigned int reg_val;
166	struct lm355x_platform_data *pdata = chip->pdata;
167
168	/* input and output pins configuration */
169	switch (chip->type) {
170	case CHIP_LM3554:
171		reg_val = pdata->pin_tx2 | pdata->ntc_pin;
172		ret = regmap_update_bits(chip->regmap, 0xE0, 0x28, reg_val);
173		if (ret < 0)
174			goto out;
175		reg_val = pdata->pass_mode;
176		ret = regmap_update_bits(chip->regmap, 0xA0, 0x04, reg_val);
177		if (ret < 0)
178			goto out;
179		break;
180
181	case CHIP_LM3556:
182		reg_val = pdata->pin_tx2 | pdata->ntc_pin | pdata->pass_mode;
 
183		ret = regmap_update_bits(chip->regmap, 0x0A, 0xC4, reg_val);
184		if (ret < 0)
185			goto out;
186		break;
187	default:
188		return -ENODATA;
189	}
190
191	return ret;
192out:
193	dev_err(chip->dev, "%s:i2c access fail to register\n", __func__);
194	return ret;
195}
196
197/* chip control */
198static int lm355x_control(struct lm355x_chip_data *chip,
199			   u8 brightness, enum lm355x_mode opmode)
200{
201	int ret;
202	unsigned int reg_val;
203	struct lm355x_platform_data *pdata = chip->pdata;
204	struct lm355x_reg_data *preg = chip->regs;
205
206	ret = regmap_read(chip->regmap, preg[REG_FLAG].regno, &chip->last_flag);
207	if (ret < 0)
208		goto out;
209	if (chip->last_flag & preg[REG_FLAG].mask)
210		dev_info(chip->dev, "%s Last FLAG is 0x%x\n",
211			 lm355x_name[chip->type],
212			 chip->last_flag & preg[REG_FLAG].mask);
213	/* brightness 0 means shutdown */
214	if (!brightness)
215		opmode = MODE_SHDN;
216
217	switch (opmode) {
218	case MODE_TORCH:
219		ret =
220		    regmap_update_bits(chip->regmap, preg[REG_TORCH_CTRL].regno,
221				       preg[REG_TORCH_CTRL].mask,
222				       (brightness - 1)
223				       << preg[REG_TORCH_CTRL].shift);
224		if (ret < 0)
225			goto out;
226
227		if (pdata->pin_tx1 != LM355x_PIN_TORCH_DISABLE) {
228			ret =
229			    regmap_update_bits(chip->regmap,
230					       preg[REG_TORCH_CFG].regno,
231					       preg[REG_TORCH_CFG].mask,
232					       0x01 <<
233					       preg[REG_TORCH_CFG].shift);
234			if (ret < 0)
235				goto out;
236			opmode = MODE_SHDN;
237			dev_info(chip->dev,
238				 "torch brt is set - ext. torch pin mode\n");
239		}
240		break;
241
242	case MODE_FLASH:
243
244		ret =
245		    regmap_update_bits(chip->regmap, preg[REG_FLASH_CTRL].regno,
246				       preg[REG_FLASH_CTRL].mask,
247				       (brightness - 1)
248				       << preg[REG_FLASH_CTRL].shift);
249		if (ret < 0)
250			goto out;
251
252		if (pdata->pin_strobe != LM355x_PIN_STROBE_DISABLE) {
253			if (chip->type == CHIP_LM3554)
254				reg_val = 0x00;
255			else
256				reg_val = 0x01;
257			ret =
258			    regmap_update_bits(chip->regmap,
259					       preg[REG_STROBE_CFG].regno,
260					       preg[REG_STROBE_CFG].mask,
261					       reg_val <<
262					       preg[REG_STROBE_CFG].shift);
263			if (ret < 0)
264				goto out;
265			opmode = MODE_SHDN;
266			dev_info(chip->dev,
267				 "flash brt is set - ext. strobe pin mode\n");
268		}
269		break;
270
271	case MODE_INDIC:
272		ret =
273		    regmap_update_bits(chip->regmap, preg[REG_INDI_CTRL].regno,
274				       preg[REG_INDI_CTRL].mask,
275				       (brightness - 1)
276				       << preg[REG_INDI_CTRL].shift);
277		if (ret < 0)
278			goto out;
279
280		if (pdata->pin_tx2 != LM355x_PIN_TX_DISABLE) {
281			ret =
282			    regmap_update_bits(chip->regmap,
283					       preg[REG_INDI_CFG].regno,
284					       preg[REG_INDI_CFG].mask,
285					       0x01 <<
286					       preg[REG_INDI_CFG].shift);
287			if (ret < 0)
288				goto out;
289			opmode = MODE_SHDN;
290		}
291		break;
292	case MODE_SHDN:
293		break;
294	default:
295		return -EINVAL;
296	}
297	/* operation mode control */
298	ret = regmap_update_bits(chip->regmap, preg[REG_OPMODE].regno,
299				 preg[REG_OPMODE].mask,
300				 opmode << preg[REG_OPMODE].shift);
301	if (ret < 0)
302		goto out;
303	return ret;
304out:
305	dev_err(chip->dev, "%s:i2c access fail to register\n", __func__);
306	return ret;
307}
308
309/* torch */
310
311static int lm355x_torch_brightness_set(struct led_classdev *cdev,
312					enum led_brightness brightness)
313{
314	struct lm355x_chip_data *chip =
315	    container_of(cdev, struct lm355x_chip_data, cdev_torch);
316	int ret;
317
318	mutex_lock(&chip->lock);
319	ret = lm355x_control(chip, brightness, MODE_TORCH);
320	mutex_unlock(&chip->lock);
321	return ret;
322}
323
324/* flash */
325
326static int lm355x_strobe_brightness_set(struct led_classdev *cdev,
327					 enum led_brightness brightness)
328{
329	struct lm355x_chip_data *chip =
330	    container_of(cdev, struct lm355x_chip_data, cdev_flash);
331	int ret;
332
333	mutex_lock(&chip->lock);
334	ret = lm355x_control(chip, brightness, MODE_FLASH);
335	mutex_unlock(&chip->lock);
336	return ret;
337}
338
339/* indicator */
340
341static int lm355x_indicator_brightness_set(struct led_classdev *cdev,
342					    enum led_brightness brightness)
343{
344	struct lm355x_chip_data *chip =
345	    container_of(cdev, struct lm355x_chip_data, cdev_indicator);
346	int ret;
347
348	mutex_lock(&chip->lock);
349	ret = lm355x_control(chip, brightness, MODE_INDIC);
350	mutex_unlock(&chip->lock);
351	return ret;
352}
353
354/* indicator pattern only for lm3556*/
355static ssize_t lm3556_indicator_pattern_store(struct device *dev,
356					      struct device_attribute *attr,
357					      const char *buf, size_t size)
358{
359	ssize_t ret;
360	struct led_classdev *led_cdev = dev_get_drvdata(dev);
361	struct lm355x_chip_data *chip =
362	    container_of(led_cdev, struct lm355x_chip_data, cdev_indicator);
363	unsigned int state;
364
365	ret = kstrtouint(buf, 10, &state);
366	if (ret)
367		goto out;
368	if (state > INDIC_PATTERN_SIZE - 1)
369		state = INDIC_PATTERN_SIZE - 1;
370
371	ret = regmap_write(chip->regmap, 0x04,
372			   indicator_pattern[state].blinking);
373	if (ret < 0)
374		goto out;
375
376	ret = regmap_write(chip->regmap, 0x05,
377			   indicator_pattern[state].period_cnt);
378	if (ret < 0)
379		goto out;
380
381	return size;
382out:
383	dev_err(chip->dev, "%s:i2c access fail to register\n", __func__);
384	return ret;
385}
386
387static DEVICE_ATTR(pattern, S_IWUSR, NULL, lm3556_indicator_pattern_store);
388
389static struct attribute *lm355x_indicator_attrs[] = {
390	&dev_attr_pattern.attr,
391	NULL
392};
393ATTRIBUTE_GROUPS(lm355x_indicator);
394
395static const struct regmap_config lm355x_regmap = {
396	.reg_bits = 8,
397	.val_bits = 8,
398	.max_register = 0xFF,
399};
400
401/* module initialize */
402static int lm355x_probe(struct i2c_client *client,
403				  const struct i2c_device_id *id)
404{
405	struct lm355x_platform_data *pdata = dev_get_platdata(&client->dev);
406	struct lm355x_chip_data *chip;
407
408	int err;
409
410	if (!i2c_check_functionality(client->adapter, I2C_FUNC_I2C)) {
411		dev_err(&client->dev, "i2c functionality check fail.\n");
412		return -EOPNOTSUPP;
413	}
414
415	if (pdata == NULL) {
416		dev_err(&client->dev, "needs Platform Data.\n");
417		return -ENODATA;
418	}
419
420	chip = devm_kzalloc(&client->dev,
421			    sizeof(struct lm355x_chip_data), GFP_KERNEL);
422	if (!chip)
423		return -ENOMEM;
424
425	chip->dev = &client->dev;
426	chip->type = id->driver_data;
427	switch (id->driver_data) {
428	case CHIP_LM3554:
429		chip->regs = lm3554_regs;
430		break;
431	case CHIP_LM3556:
432		chip->regs = lm3556_regs;
433		break;
434	default:
435		return -ENOSYS;
436	}
437	chip->pdata = pdata;
438
439	chip->regmap = devm_regmap_init_i2c(client, &lm355x_regmap);
440	if (IS_ERR(chip->regmap)) {
441		err = PTR_ERR(chip->regmap);
442		dev_err(&client->dev,
443			"Failed to allocate register map: %d\n", err);
444		return err;
445	}
446
447	mutex_init(&chip->lock);
448	i2c_set_clientdata(client, chip);
449
450	err = lm355x_chip_init(chip);
451	if (err < 0)
452		goto err_out;
453
454	/* flash */
455	chip->cdev_flash.name = "flash";
456	chip->cdev_flash.max_brightness = 16;
457	chip->cdev_flash.brightness_set_blocking = lm355x_strobe_brightness_set;
458	chip->cdev_flash.default_trigger = "flash";
459	err = led_classdev_register((struct device *)
460				    &client->dev, &chip->cdev_flash);
461	if (err < 0)
462		goto err_out;
463	/* torch */
464	chip->cdev_torch.name = "torch";
465	chip->cdev_torch.max_brightness = 8;
466	chip->cdev_torch.brightness_set_blocking = lm355x_torch_brightness_set;
467	chip->cdev_torch.default_trigger = "torch";
468	err = led_classdev_register((struct device *)
469				    &client->dev, &chip->cdev_torch);
470	if (err < 0)
471		goto err_create_torch_file;
472	/* indicator */
473	chip->cdev_indicator.name = "indicator";
474	if (id->driver_data == CHIP_LM3554)
475		chip->cdev_indicator.max_brightness = 4;
476	else
477		chip->cdev_indicator.max_brightness = 8;
478	chip->cdev_indicator.brightness_set_blocking =
479					lm355x_indicator_brightness_set;
480	/* indicator pattern control only for LM3556 */
481	if (id->driver_data == CHIP_LM3556)
482		chip->cdev_indicator.groups = lm355x_indicator_groups;
483	err = led_classdev_register((struct device *)
484				    &client->dev, &chip->cdev_indicator);
485	if (err < 0)
486		goto err_create_indicator_file;
487
488	dev_info(&client->dev, "%s is initialized\n",
489		 lm355x_name[id->driver_data]);
490	return 0;
491
492err_create_indicator_file:
493	led_classdev_unregister(&chip->cdev_torch);
494err_create_torch_file:
495	led_classdev_unregister(&chip->cdev_flash);
496err_out:
497	return err;
498}
499
500static int lm355x_remove(struct i2c_client *client)
501{
502	struct lm355x_chip_data *chip = i2c_get_clientdata(client);
503	struct lm355x_reg_data *preg = chip->regs;
504
505	regmap_write(chip->regmap, preg[REG_OPMODE].regno, 0);
506	led_classdev_unregister(&chip->cdev_indicator);
507	led_classdev_unregister(&chip->cdev_torch);
508	led_classdev_unregister(&chip->cdev_flash);
509	dev_info(&client->dev, "%s is removed\n", lm355x_name[chip->type]);
510
511	return 0;
512}
513
514static const struct i2c_device_id lm355x_id[] = {
515	{LM3554_NAME, CHIP_LM3554},
516	{LM3556_NAME, CHIP_LM3556},
517	{}
518};
519
520MODULE_DEVICE_TABLE(i2c, lm355x_id);
521
522static struct i2c_driver lm355x_i2c_driver = {
523	.driver = {
524		   .name = LM355x_NAME,
525		   .pm = NULL,
526		   },
527	.probe = lm355x_probe,
528	.remove = lm355x_remove,
529	.id_table = lm355x_id,
530};
531
532module_i2c_driver(lm355x_i2c_driver);
533
534MODULE_DESCRIPTION("Texas Instruments Flash Lighting driver for LM355x");
535MODULE_AUTHOR("Daniel Jeong <daniel.jeong@ti.com>");
536MODULE_AUTHOR("G.Shark Jeong <gshark.jeong@gmail.com>");
537MODULE_LICENSE("GPL v2");