1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * ltr501.c - Support for Lite-On LTR501 ambient light and proximity sensor
4 *
5 * Copyright 2014 Peter Meerwald <pmeerw@pmeerw.net>
6 *
7 * 7-bit I2C slave address 0x23
8 *
9 * TODO: IR LED characteristics
10 */
11
12#include <linux/module.h>
13#include <linux/i2c.h>
14#include <linux/err.h>
15#include <linux/delay.h>
16#include <linux/regmap.h>
17#include <linux/acpi.h>
18
19#include <linux/iio/iio.h>
20#include <linux/iio/events.h>
21#include <linux/iio/sysfs.h>
22#include <linux/iio/trigger_consumer.h>
23#include <linux/iio/buffer.h>
24#include <linux/iio/triggered_buffer.h>
25
26#define LTR501_DRV_NAME "ltr501"
27
28#define LTR501_ALS_CONTR 0x80 /* ALS operation mode, SW reset */
29#define LTR501_PS_CONTR 0x81 /* PS operation mode */
30#define LTR501_PS_MEAS_RATE 0x84 /* measurement rate*/
31#define LTR501_ALS_MEAS_RATE 0x85 /* ALS integ time, measurement rate*/
32#define LTR501_PART_ID 0x86
33#define LTR501_MANUFAC_ID 0x87
34#define LTR501_ALS_DATA1 0x88 /* 16-bit, little endian */
35#define LTR501_ALS_DATA1_UPPER 0x89 /* upper 8 bits of LTR501_ALS_DATA1 */
36#define LTR501_ALS_DATA0 0x8a /* 16-bit, little endian */
37#define LTR501_ALS_DATA0_UPPER 0x8b /* upper 8 bits of LTR501_ALS_DATA0 */
38#define LTR501_ALS_PS_STATUS 0x8c
39#define LTR501_PS_DATA 0x8d /* 16-bit, little endian */
40#define LTR501_PS_DATA_UPPER 0x8e /* upper 8 bits of LTR501_PS_DATA */
41#define LTR501_INTR 0x8f /* output mode, polarity, mode */
42#define LTR501_PS_THRESH_UP 0x90 /* 11 bit, ps upper threshold */
43#define LTR501_PS_THRESH_LOW 0x92 /* 11 bit, ps lower threshold */
44#define LTR501_ALS_THRESH_UP 0x97 /* 16 bit, ALS upper threshold */
45#define LTR501_ALS_THRESH_LOW 0x99 /* 16 bit, ALS lower threshold */
46#define LTR501_INTR_PRST 0x9e /* ps thresh, als thresh */
47#define LTR501_MAX_REG 0x9f
48
49#define LTR501_ALS_CONTR_SW_RESET BIT(2)
50#define LTR501_CONTR_PS_GAIN_MASK (BIT(3) | BIT(2))
51#define LTR501_CONTR_PS_GAIN_SHIFT 2
52#define LTR501_CONTR_ALS_GAIN_MASK BIT(3)
53#define LTR501_CONTR_ACTIVE BIT(1)
54
55#define LTR501_STATUS_ALS_INTR BIT(3)
56#define LTR501_STATUS_ALS_RDY BIT(2)
57#define LTR501_STATUS_PS_INTR BIT(1)
58#define LTR501_STATUS_PS_RDY BIT(0)
59
60#define LTR501_PS_DATA_MASK 0x7ff
61#define LTR501_PS_THRESH_MASK 0x7ff
62#define LTR501_ALS_THRESH_MASK 0xffff
63
64#define LTR501_ALS_DEF_PERIOD 500000
65#define LTR501_PS_DEF_PERIOD 100000
66
67#define LTR501_REGMAP_NAME "ltr501_regmap"
68
69#define LTR501_LUX_CONV(vis_coeff, vis_data, ir_coeff, ir_data) \
70			((vis_coeff * vis_data) - (ir_coeff * ir_data))
71
72static const int int_time_mapping[] = {100000, 50000, 200000, 400000};
73
74static const struct reg_field reg_field_it =
75				REG_FIELD(LTR501_ALS_MEAS_RATE, 3, 4);
76static const struct reg_field reg_field_als_intr =
77				REG_FIELD(LTR501_INTR, 1, 1);
78static const struct reg_field reg_field_ps_intr =
79				REG_FIELD(LTR501_INTR, 0, 0);
80static const struct reg_field reg_field_als_rate =
81				REG_FIELD(LTR501_ALS_MEAS_RATE, 0, 2);
82static const struct reg_field reg_field_ps_rate =
83				REG_FIELD(LTR501_PS_MEAS_RATE, 0, 3);
84static const struct reg_field reg_field_als_prst =
85				REG_FIELD(LTR501_INTR_PRST, 0, 3);
86static const struct reg_field reg_field_ps_prst =
87				REG_FIELD(LTR501_INTR_PRST, 4, 7);
88
89struct ltr501_samp_table {
90	int freq_val;  /* repetition frequency in micro HZ*/
91	int time_val; /* repetition rate in micro seconds */
92};
93
94#define LTR501_RESERVED_GAIN -1
95
96enum {
97	ltr501 = 0,
98	ltr559,
99	ltr301,
100};
101
102struct ltr501_gain {
103	int scale;
104	int uscale;
105};
106
107static const struct ltr501_gain ltr501_als_gain_tbl[] = {
108	{1, 0},
109	{0, 5000},
110};
111
112static const struct ltr501_gain ltr559_als_gain_tbl[] = {
113	{1, 0},
114	{0, 500000},
115	{0, 250000},
116	{0, 125000},
117	{LTR501_RESERVED_GAIN, LTR501_RESERVED_GAIN},
118	{LTR501_RESERVED_GAIN, LTR501_RESERVED_GAIN},
119	{0, 20000},
120	{0, 10000},
121};
122
123static const struct ltr501_gain ltr501_ps_gain_tbl[] = {
124	{1, 0},
125	{0, 250000},
126	{0, 125000},
127	{0, 62500},
128};
129
130static const struct ltr501_gain ltr559_ps_gain_tbl[] = {
131	{0, 62500}, /* x16 gain */
132	{0, 31250}, /* x32 gain */
133	{0, 15625}, /* bits X1 are for x64 gain */
134	{0, 15624},
135};
136
137struct ltr501_chip_info {
138	u8 partid;
139	const struct ltr501_gain *als_gain;
140	int als_gain_tbl_size;
141	const struct ltr501_gain *ps_gain;
142	int ps_gain_tbl_size;
143	u8 als_mode_active;
144	u8 als_gain_mask;
145	u8 als_gain_shift;
146	struct iio_chan_spec const *channels;
147	const int no_channels;
148	const struct iio_info *info;
149	const struct iio_info *info_no_irq;
150};
151
152struct ltr501_data {
153	struct i2c_client *client;
154	struct mutex lock_als, lock_ps;
155	struct ltr501_chip_info *chip_info;
156	u8 als_contr, ps_contr;
157	int als_period, ps_period; /* period in micro seconds */
158	struct regmap *regmap;
159	struct regmap_field *reg_it;
160	struct regmap_field *reg_als_intr;
161	struct regmap_field *reg_ps_intr;
162	struct regmap_field *reg_als_rate;
163	struct regmap_field *reg_ps_rate;
164	struct regmap_field *reg_als_prst;
165	struct regmap_field *reg_ps_prst;
166};
167
168static const struct ltr501_samp_table ltr501_als_samp_table[] = {
169			{20000000, 50000}, {10000000, 100000},
170			{5000000, 200000}, {2000000, 500000},
171			{1000000, 1000000}, {500000, 2000000},
172			{500000, 2000000}, {500000, 2000000}
173};
174
175static const struct ltr501_samp_table ltr501_ps_samp_table[] = {
176			{20000000, 50000}, {14285714, 70000},
177			{10000000, 100000}, {5000000, 200000},
178			{2000000, 500000}, {1000000, 1000000},
179			{500000, 2000000}, {500000, 2000000},
180			{500000, 2000000}
181};
182
183static int ltr501_match_samp_freq(const struct ltr501_samp_table *tab,
184					   int len, int val, int val2)
185{
186	int i, freq;
187
188	freq = val * 1000000 + val2;
189
190	for (i = 0; i < len; i++) {
191		if (tab[i].freq_val == freq)
192			return i;
193	}
194
195	return -EINVAL;
196}
197
198static int ltr501_als_read_samp_freq(const struct ltr501_data *data,
199				     int *val, int *val2)
200{
201	int ret, i;
202
203	ret = regmap_field_read(data->reg_als_rate, &i);
204	if (ret < 0)
205		return ret;
206
207	if (i < 0 || i >= ARRAY_SIZE(ltr501_als_samp_table))
208		return -EINVAL;
209
210	*val = ltr501_als_samp_table[i].freq_val / 1000000;
211	*val2 = ltr501_als_samp_table[i].freq_val % 1000000;
212
213	return IIO_VAL_INT_PLUS_MICRO;
214}
215
216static int ltr501_ps_read_samp_freq(const struct ltr501_data *data,
217				    int *val, int *val2)
218{
219	int ret, i;
220
221	ret = regmap_field_read(data->reg_ps_rate, &i);
222	if (ret < 0)
223		return ret;
224
225	if (i < 0 || i >= ARRAY_SIZE(ltr501_ps_samp_table))
226		return -EINVAL;
227
228	*val = ltr501_ps_samp_table[i].freq_val / 1000000;
229	*val2 = ltr501_ps_samp_table[i].freq_val % 1000000;
230
231	return IIO_VAL_INT_PLUS_MICRO;
232}
233
234static int ltr501_als_write_samp_freq(struct ltr501_data *data,
235				      int val, int val2)
236{
237	int i, ret;
238
239	i = ltr501_match_samp_freq(ltr501_als_samp_table,
240				   ARRAY_SIZE(ltr501_als_samp_table),
241				   val, val2);
242
243	if (i < 0)
244		return i;
245
246	mutex_lock(&data->lock_als);
247	ret = regmap_field_write(data->reg_als_rate, i);
248	mutex_unlock(&data->lock_als);
249
250	return ret;
251}
252
253static int ltr501_ps_write_samp_freq(struct ltr501_data *data,
254				     int val, int val2)
255{
256	int i, ret;
257
258	i = ltr501_match_samp_freq(ltr501_ps_samp_table,
259				   ARRAY_SIZE(ltr501_ps_samp_table),
260				   val, val2);
261
262	if (i < 0)
263		return i;
264
265	mutex_lock(&data->lock_ps);
266	ret = regmap_field_write(data->reg_ps_rate, i);
267	mutex_unlock(&data->lock_ps);
268
269	return ret;
270}
271
272static int ltr501_als_read_samp_period(const struct ltr501_data *data, int *val)
273{
274	int ret, i;
275
276	ret = regmap_field_read(data->reg_als_rate, &i);
277	if (ret < 0)
278		return ret;
279
280	if (i < 0 || i >= ARRAY_SIZE(ltr501_als_samp_table))
281		return -EINVAL;
282
283	*val = ltr501_als_samp_table[i].time_val;
284
285	return IIO_VAL_INT;
286}
287
288static int ltr501_ps_read_samp_period(const struct ltr501_data *data, int *val)
289{
290	int ret, i;
291
292	ret = regmap_field_read(data->reg_ps_rate, &i);
293	if (ret < 0)
294		return ret;
295
296	if (i < 0 || i >= ARRAY_SIZE(ltr501_ps_samp_table))
297		return -EINVAL;
298
299	*val = ltr501_ps_samp_table[i].time_val;
300
301	return IIO_VAL_INT;
302}
303
304/* IR and visible spectrum coeff's are given in data sheet */
305static unsigned long ltr501_calculate_lux(u16 vis_data, u16 ir_data)
306{
307	unsigned long ratio, lux;
308
309	if (vis_data == 0)
310		return 0;
311
312	/* multiply numerator by 100 to avoid handling ratio < 1 */
313	ratio = DIV_ROUND_UP(ir_data * 100, ir_data + vis_data);
314
315	if (ratio < 45)
316		lux = LTR501_LUX_CONV(1774, vis_data, -1105, ir_data);
317	else if (ratio >= 45 && ratio < 64)
318		lux = LTR501_LUX_CONV(3772, vis_data, 1336, ir_data);
319	else if (ratio >= 64 && ratio < 85)
320		lux = LTR501_LUX_CONV(1690, vis_data, 169, ir_data);
321	else
322		lux = 0;
323
324	return lux / 1000;
325}
326
327static int ltr501_drdy(const struct ltr501_data *data, u8 drdy_mask)
328{
329	int tries = 100;
330	int ret, status;
331
332	while (tries--) {
333		ret = regmap_read(data->regmap, LTR501_ALS_PS_STATUS, &status);
334		if (ret < 0)
335			return ret;
336		if ((status & drdy_mask) == drdy_mask)
337			return 0;
338		msleep(25);
339	}
340
341	dev_err(&data->client->dev, "ltr501_drdy() failed, data not ready\n");
342	return -EIO;
343}
344
345static int ltr501_set_it_time(struct ltr501_data *data, int it)
346{
347	int ret, i, index = -1, status;
348
349	for (i = 0; i < ARRAY_SIZE(int_time_mapping); i++) {
350		if (int_time_mapping[i] == it) {
351			index = i;
352			break;
353		}
354	}
355	/* Make sure integ time index is valid */
356	if (index < 0)
357		return -EINVAL;
358
359	ret = regmap_read(data->regmap, LTR501_ALS_CONTR, &status);
360	if (ret < 0)
361		return ret;
362
363	if (status & LTR501_CONTR_ALS_GAIN_MASK) {
364		/*
365		 * 200 ms and 400 ms integ time can only be
366		 * used in dynamic range 1
367		 */
368		if (index > 1)
369			return -EINVAL;
370	} else
371		/* 50 ms integ time can only be used in dynamic range 2 */
372		if (index == 1)
373			return -EINVAL;
374
375	return regmap_field_write(data->reg_it, index);
376}
377
378/* read int time in micro seconds */
379static int ltr501_read_it_time(const struct ltr501_data *data,
380			       int *val, int *val2)
381{
382	int ret, index;
383
384	ret = regmap_field_read(data->reg_it, &index);
385	if (ret < 0)
386		return ret;
387
388	/* Make sure integ time index is valid */
389	if (index < 0 || index >= ARRAY_SIZE(int_time_mapping))
390		return -EINVAL;
391
392	*val2 = int_time_mapping[index];
393	*val = 0;
394
395	return IIO_VAL_INT_PLUS_MICRO;
396}
397
398static int ltr501_read_als(const struct ltr501_data *data, __le16 buf[2])
399{
400	int ret;
401
402	ret = ltr501_drdy(data, LTR501_STATUS_ALS_RDY);
403	if (ret < 0)
404		return ret;
405	/* always read both ALS channels in given order */
406	return regmap_bulk_read(data->regmap, LTR501_ALS_DATA1,
407				buf, 2 * sizeof(__le16));
408}
409
410static int ltr501_read_ps(const struct ltr501_data *data)
411{
412	__le16 status;
413	int ret;
414
415	ret = ltr501_drdy(data, LTR501_STATUS_PS_RDY);
416	if (ret < 0)
417		return ret;
418
419	ret = regmap_bulk_read(data->regmap, LTR501_PS_DATA,
420			       &status, sizeof(status));
421	if (ret < 0)
422		return ret;
423
424	return le16_to_cpu(status);
425}
426
427static int ltr501_read_intr_prst(const struct ltr501_data *data,
428				 enum iio_chan_type type,
429				 int *val2)
430{
431	int ret, samp_period, prst;
432
433	switch (type) {
434	case IIO_INTENSITY:
435		ret = regmap_field_read(data->reg_als_prst, &prst);
436		if (ret < 0)
437			return ret;
438
439		ret = ltr501_als_read_samp_period(data, &samp_period);
440
441		if (ret < 0)
442			return ret;
443		*val2 = samp_period * prst;
444		return IIO_VAL_INT_PLUS_MICRO;
445	case IIO_PROXIMITY:
446		ret = regmap_field_read(data->reg_ps_prst, &prst);
447		if (ret < 0)
448			return ret;
449
450		ret = ltr501_ps_read_samp_period(data, &samp_period);
451
452		if (ret < 0)
453			return ret;
454
455		*val2 = samp_period * prst;
456		return IIO_VAL_INT_PLUS_MICRO;
457	default:
458		return -EINVAL;
459	}
460
461	return -EINVAL;
462}
463
464static int ltr501_write_intr_prst(struct ltr501_data *data,
465				  enum iio_chan_type type,
466				  int val, int val2)
467{
468	int ret, samp_period, new_val;
469	unsigned long period;
470
471	if (val < 0 || val2 < 0)
472		return -EINVAL;
473
474	/* period in microseconds */
475	period = ((val * 1000000) + val2);
476
477	switch (type) {
478	case IIO_INTENSITY:
479		ret = ltr501_als_read_samp_period(data, &samp_period);
480		if (ret < 0)
481			return ret;
482
483		/* period should be atleast equal to sampling period */
484		if (period < samp_period)
485			return -EINVAL;
486
487		new_val = DIV_ROUND_UP(period, samp_period);
488		if (new_val < 0 || new_val > 0x0f)
489			return -EINVAL;
490
491		mutex_lock(&data->lock_als);
492		ret = regmap_field_write(data->reg_als_prst, new_val);
493		mutex_unlock(&data->lock_als);
494		if (ret >= 0)
495			data->als_period = period;
496
497		return ret;
498	case IIO_PROXIMITY:
499		ret = ltr501_ps_read_samp_period(data, &samp_period);
500		if (ret < 0)
501			return ret;
502
503		/* period should be atleast equal to rate */
504		if (period < samp_period)
505			return -EINVAL;
506
507		new_val = DIV_ROUND_UP(period, samp_period);
508		if (new_val < 0 || new_val > 0x0f)
509			return -EINVAL;
510
511		mutex_lock(&data->lock_ps);
512		ret = regmap_field_write(data->reg_ps_prst, new_val);
513		mutex_unlock(&data->lock_ps);
514		if (ret >= 0)
515			data->ps_period = period;
516
517		return ret;
518	default:
519		return -EINVAL;
520	}
521
522	return -EINVAL;
523}
524
525static const struct iio_event_spec ltr501_als_event_spec[] = {
526	{
527		.type = IIO_EV_TYPE_THRESH,
528		.dir = IIO_EV_DIR_RISING,
529		.mask_separate = BIT(IIO_EV_INFO_VALUE),
530	}, {
531		.type = IIO_EV_TYPE_THRESH,
532		.dir = IIO_EV_DIR_FALLING,
533		.mask_separate = BIT(IIO_EV_INFO_VALUE),
534	}, {
535		.type = IIO_EV_TYPE_THRESH,
536		.dir = IIO_EV_DIR_EITHER,
537		.mask_separate = BIT(IIO_EV_INFO_ENABLE) |
538				 BIT(IIO_EV_INFO_PERIOD),
539	},
540
541};
542
543static const struct iio_event_spec ltr501_pxs_event_spec[] = {
544	{
545		.type = IIO_EV_TYPE_THRESH,
546		.dir = IIO_EV_DIR_RISING,
547		.mask_separate = BIT(IIO_EV_INFO_VALUE),
548	}, {
549		.type = IIO_EV_TYPE_THRESH,
550		.dir = IIO_EV_DIR_FALLING,
551		.mask_separate = BIT(IIO_EV_INFO_VALUE),
552	}, {
553		.type = IIO_EV_TYPE_THRESH,
554		.dir = IIO_EV_DIR_EITHER,
555		.mask_separate = BIT(IIO_EV_INFO_ENABLE) |
556				 BIT(IIO_EV_INFO_PERIOD),
557	},
558};
559
560#define LTR501_INTENSITY_CHANNEL(_idx, _addr, _mod, _shared, \
561				 _evspec, _evsize) { \
562	.type = IIO_INTENSITY, \
563	.modified = 1, \
564	.address = (_addr), \
565	.channel2 = (_mod), \
566	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW), \
567	.info_mask_shared_by_type = (_shared), \
568	.scan_index = (_idx), \
569	.scan_type = { \
570		.sign = 'u', \
571		.realbits = 16, \
572		.storagebits = 16, \
573		.endianness = IIO_CPU, \
574	}, \
575	.event_spec = _evspec,\
576	.num_event_specs = _evsize,\
577}
578
579#define LTR501_LIGHT_CHANNEL() { \
580	.type = IIO_LIGHT, \
581	.info_mask_separate = BIT(IIO_CHAN_INFO_PROCESSED), \
582	.scan_index = -1, \
583}
584
585static const struct iio_chan_spec ltr501_channels[] = {
586	LTR501_LIGHT_CHANNEL(),
587	LTR501_INTENSITY_CHANNEL(0, LTR501_ALS_DATA0, IIO_MOD_LIGHT_BOTH, 0,
588				 ltr501_als_event_spec,
589				 ARRAY_SIZE(ltr501_als_event_spec)),
590	LTR501_INTENSITY_CHANNEL(1, LTR501_ALS_DATA1, IIO_MOD_LIGHT_IR,
591				 BIT(IIO_CHAN_INFO_SCALE) |
592				 BIT(IIO_CHAN_INFO_INT_TIME) |
593				 BIT(IIO_CHAN_INFO_SAMP_FREQ),
594				 NULL, 0),
595	{
596		.type = IIO_PROXIMITY,
597		.address = LTR501_PS_DATA,
598		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
599			BIT(IIO_CHAN_INFO_SCALE),
600		.scan_index = 2,
601		.scan_type = {
602			.sign = 'u',
603			.realbits = 11,
604			.storagebits = 16,
605			.endianness = IIO_CPU,
606		},
607		.event_spec = ltr501_pxs_event_spec,
608		.num_event_specs = ARRAY_SIZE(ltr501_pxs_event_spec),
609	},
610	IIO_CHAN_SOFT_TIMESTAMP(3),
611};
612
613static const struct iio_chan_spec ltr301_channels[] = {
614	LTR501_LIGHT_CHANNEL(),
615	LTR501_INTENSITY_CHANNEL(0, LTR501_ALS_DATA0, IIO_MOD_LIGHT_BOTH, 0,
616				 ltr501_als_event_spec,
617				 ARRAY_SIZE(ltr501_als_event_spec)),
618	LTR501_INTENSITY_CHANNEL(1, LTR501_ALS_DATA1, IIO_MOD_LIGHT_IR,
619				 BIT(IIO_CHAN_INFO_SCALE) |
620				 BIT(IIO_CHAN_INFO_INT_TIME) |
621				 BIT(IIO_CHAN_INFO_SAMP_FREQ),
622				 NULL, 0),
623	IIO_CHAN_SOFT_TIMESTAMP(2),
624};
625
626static int ltr501_read_raw(struct iio_dev *indio_dev,
627			   struct iio_chan_spec const *chan,
628			   int *val, int *val2, long mask)
629{
630	struct ltr501_data *data = iio_priv(indio_dev);
631	__le16 buf[2];
632	int ret, i;
633
634	switch (mask) {
635	case IIO_CHAN_INFO_PROCESSED:
636		switch (chan->type) {
637		case IIO_LIGHT:
638			ret = iio_device_claim_direct_mode(indio_dev);
639			if (ret)
640				return ret;
641
642			mutex_lock(&data->lock_als);
643			ret = ltr501_read_als(data, buf);
644			mutex_unlock(&data->lock_als);
645			iio_device_release_direct_mode(indio_dev);
646			if (ret < 0)
647				return ret;
648			*val = ltr501_calculate_lux(le16_to_cpu(buf[1]),
649						    le16_to_cpu(buf[0]));
650			return IIO_VAL_INT;
651		default:
652			return -EINVAL;
653		}
654	case IIO_CHAN_INFO_RAW:
655		ret = iio_device_claim_direct_mode(indio_dev);
656		if (ret)
657			return ret;
658
659		switch (chan->type) {
660		case IIO_INTENSITY:
661			mutex_lock(&data->lock_als);
662			ret = ltr501_read_als(data, buf);
663			mutex_unlock(&data->lock_als);
664			if (ret < 0)
665				break;
666			*val = le16_to_cpu(chan->address == LTR501_ALS_DATA1 ?
667					   buf[0] : buf[1]);
668			ret = IIO_VAL_INT;
669			break;
670		case IIO_PROXIMITY:
671			mutex_lock(&data->lock_ps);
672			ret = ltr501_read_ps(data);
673			mutex_unlock(&data->lock_ps);
674			if (ret < 0)
675				break;
676			*val = ret & LTR501_PS_DATA_MASK;
677			ret = IIO_VAL_INT;
678			break;
679		default:
680			ret = -EINVAL;
681			break;
682		}
683
684		iio_device_release_direct_mode(indio_dev);
685		return ret;
686
687	case IIO_CHAN_INFO_SCALE:
688		switch (chan->type) {
689		case IIO_INTENSITY:
690			i = (data->als_contr & data->chip_info->als_gain_mask)
691			     >> data->chip_info->als_gain_shift;
692			*val = data->chip_info->als_gain[i].scale;
693			*val2 = data->chip_info->als_gain[i].uscale;
694			return IIO_VAL_INT_PLUS_MICRO;
695		case IIO_PROXIMITY:
696			i = (data->ps_contr & LTR501_CONTR_PS_GAIN_MASK) >>
697				LTR501_CONTR_PS_GAIN_SHIFT;
698			*val = data->chip_info->ps_gain[i].scale;
699			*val2 = data->chip_info->ps_gain[i].uscale;
700			return IIO_VAL_INT_PLUS_MICRO;
701		default:
702			return -EINVAL;
703		}
704	case IIO_CHAN_INFO_INT_TIME:
705		switch (chan->type) {
706		case IIO_INTENSITY:
707			return ltr501_read_it_time(data, val, val2);
708		default:
709			return -EINVAL;
710		}
711	case IIO_CHAN_INFO_SAMP_FREQ:
712		switch (chan->type) {
713		case IIO_INTENSITY:
714			return ltr501_als_read_samp_freq(data, val, val2);
715		case IIO_PROXIMITY:
716			return ltr501_ps_read_samp_freq(data, val, val2);
717		default:
718			return -EINVAL;
719		}
720	}
721	return -EINVAL;
722}
723
724static int ltr501_get_gain_index(const struct ltr501_gain *gain, int size,
725				 int val, int val2)
726{
727	int i;
728
729	for (i = 0; i < size; i++)
730		if (val == gain[i].scale && val2 == gain[i].uscale)
731			return i;
732
733	return -1;
734}
735
736static int ltr501_write_raw(struct iio_dev *indio_dev,
737			    struct iio_chan_spec const *chan,
738			    int val, int val2, long mask)
739{
740	struct ltr501_data *data = iio_priv(indio_dev);
741	int i, ret, freq_val, freq_val2;
742	struct ltr501_chip_info *info = data->chip_info;
743
744	ret = iio_device_claim_direct_mode(indio_dev);
745	if (ret)
746		return ret;
747
748	switch (mask) {
749	case IIO_CHAN_INFO_SCALE:
750		switch (chan->type) {
751		case IIO_INTENSITY:
752			i = ltr501_get_gain_index(info->als_gain,
753						  info->als_gain_tbl_size,
754						  val, val2);
755			if (i < 0) {
756				ret = -EINVAL;
757				break;
758			}
759
760			data->als_contr &= ~info->als_gain_mask;
761			data->als_contr |= i << info->als_gain_shift;
762
763			ret = regmap_write(data->regmap, LTR501_ALS_CONTR,
764					   data->als_contr);
765			break;
766		case IIO_PROXIMITY:
767			i = ltr501_get_gain_index(info->ps_gain,
768						  info->ps_gain_tbl_size,
769						  val, val2);
770			if (i < 0) {
771				ret = -EINVAL;
772				break;
773			}
774			data->ps_contr &= ~LTR501_CONTR_PS_GAIN_MASK;
775			data->ps_contr |= i << LTR501_CONTR_PS_GAIN_SHIFT;
776
777			ret = regmap_write(data->regmap, LTR501_PS_CONTR,
778					   data->ps_contr);
779			break;
780		default:
781			ret = -EINVAL;
782			break;
783		}
784		break;
785
786	case IIO_CHAN_INFO_INT_TIME:
787		switch (chan->type) {
788		case IIO_INTENSITY:
789			if (val != 0) {
790				ret = -EINVAL;
791				break;
792			}
793			mutex_lock(&data->lock_als);
794			ret = ltr501_set_it_time(data, val2);
795			mutex_unlock(&data->lock_als);
796			break;
797		default:
798			ret = -EINVAL;
799			break;
800		}
801		break;
802
803	case IIO_CHAN_INFO_SAMP_FREQ:
804		switch (chan->type) {
805		case IIO_INTENSITY:
806			ret = ltr501_als_read_samp_freq(data, &freq_val,
807							&freq_val2);
808			if (ret < 0)
809				break;
810
811			ret = ltr501_als_write_samp_freq(data, val, val2);
812			if (ret < 0)
813				break;
814
815			/* update persistence count when changing frequency */
816			ret = ltr501_write_intr_prst(data, chan->type,
817						     0, data->als_period);
818
819			if (ret < 0)
820				ret = ltr501_als_write_samp_freq(data, freq_val,
821								 freq_val2);
822			break;
823		case IIO_PROXIMITY:
824			ret = ltr501_ps_read_samp_freq(data, &freq_val,
825						       &freq_val2);
826			if (ret < 0)
827				break;
828
829			ret = ltr501_ps_write_samp_freq(data, val, val2);
830			if (ret < 0)
831				break;
832
833			/* update persistence count when changing frequency */
834			ret = ltr501_write_intr_prst(data, chan->type,
835						     0, data->ps_period);
836
837			if (ret < 0)
838				ret = ltr501_ps_write_samp_freq(data, freq_val,
839								freq_val2);
840			break;
841		default:
842			ret = -EINVAL;
843			break;
844		}
845		break;
846
847	default:
848		ret = -EINVAL;
849		break;
850	}
851
852	iio_device_release_direct_mode(indio_dev);
853	return ret;
854}
855
856static int ltr501_read_thresh(const struct iio_dev *indio_dev,
857			      const struct iio_chan_spec *chan,
858			      enum iio_event_type type,
859			      enum iio_event_direction dir,
860			      enum iio_event_info info,
861			      int *val, int *val2)
862{
863	const struct ltr501_data *data = iio_priv(indio_dev);
864	int ret, thresh_data;
865
866	switch (chan->type) {
867	case IIO_INTENSITY:
868		switch (dir) {
869		case IIO_EV_DIR_RISING:
870			ret = regmap_bulk_read(data->regmap,
871					       LTR501_ALS_THRESH_UP,
872					       &thresh_data, 2);
873			if (ret < 0)
874				return ret;
875			*val = thresh_data & LTR501_ALS_THRESH_MASK;
876			return IIO_VAL_INT;
877		case IIO_EV_DIR_FALLING:
878			ret = regmap_bulk_read(data->regmap,
879					       LTR501_ALS_THRESH_LOW,
880					       &thresh_data, 2);
881			if (ret < 0)
882				return ret;
883			*val = thresh_data & LTR501_ALS_THRESH_MASK;
884			return IIO_VAL_INT;
885		default:
886			return -EINVAL;
887		}
888	case IIO_PROXIMITY:
889		switch (dir) {
890		case IIO_EV_DIR_RISING:
891			ret = regmap_bulk_read(data->regmap,
892					       LTR501_PS_THRESH_UP,
893					       &thresh_data, 2);
894			if (ret < 0)
895				return ret;
896			*val = thresh_data & LTR501_PS_THRESH_MASK;
897			return IIO_VAL_INT;
898		case IIO_EV_DIR_FALLING:
899			ret = regmap_bulk_read(data->regmap,
900					       LTR501_PS_THRESH_LOW,
901					       &thresh_data, 2);
902			if (ret < 0)
903				return ret;
904			*val = thresh_data & LTR501_PS_THRESH_MASK;
905			return IIO_VAL_INT;
906		default:
907			return -EINVAL;
908		}
909	default:
910		return -EINVAL;
911	}
912
913	return -EINVAL;
914}
915
916static int ltr501_write_thresh(struct iio_dev *indio_dev,
917			       const struct iio_chan_spec *chan,
918			       enum iio_event_type type,
919			       enum iio_event_direction dir,
920			       enum iio_event_info info,
921			       int val, int val2)
922{
923	struct ltr501_data *data = iio_priv(indio_dev);
924	int ret;
925
926	if (val < 0)
927		return -EINVAL;
928
929	switch (chan->type) {
930	case IIO_INTENSITY:
931		if (val > LTR501_ALS_THRESH_MASK)
932			return -EINVAL;
933		switch (dir) {
934		case IIO_EV_DIR_RISING:
935			mutex_lock(&data->lock_als);
936			ret = regmap_bulk_write(data->regmap,
937						LTR501_ALS_THRESH_UP,
938						&val, 2);
939			mutex_unlock(&data->lock_als);
940			return ret;
941		case IIO_EV_DIR_FALLING:
942			mutex_lock(&data->lock_als);
943			ret = regmap_bulk_write(data->regmap,
944						LTR501_ALS_THRESH_LOW,
945						&val, 2);
946			mutex_unlock(&data->lock_als);
947			return ret;
948		default:
949			return -EINVAL;
950		}
951	case IIO_PROXIMITY:
952		if (val > LTR501_PS_THRESH_MASK)
953			return -EINVAL;
954		switch (dir) {
955		case IIO_EV_DIR_RISING:
956			mutex_lock(&data->lock_ps);
957			ret = regmap_bulk_write(data->regmap,
958						LTR501_PS_THRESH_UP,
959						&val, 2);
960			mutex_unlock(&data->lock_ps);
961			return ret;
962		case IIO_EV_DIR_FALLING:
963			mutex_lock(&data->lock_ps);
964			ret = regmap_bulk_write(data->regmap,
965						LTR501_PS_THRESH_LOW,
966						&val, 2);
967			mutex_unlock(&data->lock_ps);
968			return ret;
969		default:
970			return -EINVAL;
971		}
972	default:
973		return -EINVAL;
974	}
975
976	return -EINVAL;
977}
978
979static int ltr501_read_event(struct iio_dev *indio_dev,
980			     const struct iio_chan_spec *chan,
981			     enum iio_event_type type,
982			     enum iio_event_direction dir,
983			     enum iio_event_info info,
984			     int *val, int *val2)
985{
986	int ret;
987
988	switch (info) {
989	case IIO_EV_INFO_VALUE:
990		return ltr501_read_thresh(indio_dev, chan, type, dir,
991					  info, val, val2);
992	case IIO_EV_INFO_PERIOD:
993		ret = ltr501_read_intr_prst(iio_priv(indio_dev),
994					    chan->type, val2);
995		*val = *val2 / 1000000;
996		*val2 = *val2 % 1000000;
997		return ret;
998	default:
999		return -EINVAL;
1000	}
1001
1002	return -EINVAL;
1003}
1004
1005static int ltr501_write_event(struct iio_dev *indio_dev,
1006			      const struct iio_chan_spec *chan,
1007			      enum iio_event_type type,
1008			      enum iio_event_direction dir,
1009			      enum iio_event_info info,
1010			      int val, int val2)
1011{
1012	switch (info) {
1013	case IIO_EV_INFO_VALUE:
1014		if (val2 != 0)
1015			return -EINVAL;
1016		return ltr501_write_thresh(indio_dev, chan, type, dir,
1017					   info, val, val2);
1018	case IIO_EV_INFO_PERIOD:
1019		return ltr501_write_intr_prst(iio_priv(indio_dev), chan->type,
1020					      val, val2);
1021	default:
1022		return -EINVAL;
1023	}
1024
1025	return -EINVAL;
1026}
1027
1028static int ltr501_read_event_config(struct iio_dev *indio_dev,
1029				    const struct iio_chan_spec *chan,
1030				    enum iio_event_type type,
1031				    enum iio_event_direction dir)
1032{
1033	struct ltr501_data *data = iio_priv(indio_dev);
1034	int ret, status;
1035
1036	switch (chan->type) {
1037	case IIO_INTENSITY:
1038		ret = regmap_field_read(data->reg_als_intr, &status);
1039		if (ret < 0)
1040			return ret;
1041		return status;
1042	case IIO_PROXIMITY:
1043		ret = regmap_field_read(data->reg_ps_intr, &status);
1044		if (ret < 0)
1045			return ret;
1046		return status;
1047	default:
1048		return -EINVAL;
1049	}
1050
1051	return -EINVAL;
1052}
1053
1054static int ltr501_write_event_config(struct iio_dev *indio_dev,
1055				     const struct iio_chan_spec *chan,
1056				     enum iio_event_type type,
1057				     enum iio_event_direction dir, int state)
1058{
1059	struct ltr501_data *data = iio_priv(indio_dev);
1060	int ret;
1061
1062	/* only 1 and 0 are valid inputs */
1063	if (state != 1  && state != 0)
1064		return -EINVAL;
1065
1066	switch (chan->type) {
1067	case IIO_INTENSITY:
1068		mutex_lock(&data->lock_als);
1069		ret = regmap_field_write(data->reg_als_intr, state);
1070		mutex_unlock(&data->lock_als);
1071		return ret;
1072	case IIO_PROXIMITY:
1073		mutex_lock(&data->lock_ps);
1074		ret = regmap_field_write(data->reg_ps_intr, state);
1075		mutex_unlock(&data->lock_ps);
1076		return ret;
1077	default:
1078		return -EINVAL;
1079	}
1080
1081	return -EINVAL;
1082}
1083
1084static ssize_t ltr501_show_proximity_scale_avail(struct device *dev,
1085						 struct device_attribute *attr,
1086						 char *buf)
1087{
1088	struct ltr501_data *data = iio_priv(dev_to_iio_dev(dev));
1089	struct ltr501_chip_info *info = data->chip_info;
1090	ssize_t len = 0;
1091	int i;
1092
1093	for (i = 0; i < info->ps_gain_tbl_size; i++) {
1094		if (info->ps_gain[i].scale == LTR501_RESERVED_GAIN)
1095			continue;
1096		len += scnprintf(buf + len, PAGE_SIZE - len, "%d.%06d ",
1097				 info->ps_gain[i].scale,
1098				 info->ps_gain[i].uscale);
1099	}
1100
1101	buf[len - 1] = '\n';
1102
1103	return len;
1104}
1105
1106static ssize_t ltr501_show_intensity_scale_avail(struct device *dev,
1107						 struct device_attribute *attr,
1108						 char *buf)
1109{
1110	struct ltr501_data *data = iio_priv(dev_to_iio_dev(dev));
1111	struct ltr501_chip_info *info = data->chip_info;
1112	ssize_t len = 0;
1113	int i;
1114
1115	for (i = 0; i < info->als_gain_tbl_size; i++) {
1116		if (info->als_gain[i].scale == LTR501_RESERVED_GAIN)
1117			continue;
1118		len += scnprintf(buf + len, PAGE_SIZE - len, "%d.%06d ",
1119				 info->als_gain[i].scale,
1120				 info->als_gain[i].uscale);
1121	}
1122
1123	buf[len - 1] = '\n';
1124
1125	return len;
1126}
1127
1128static IIO_CONST_ATTR_INT_TIME_AVAIL("0.05 0.1 0.2 0.4");
1129static IIO_CONST_ATTR_SAMP_FREQ_AVAIL("20 10 5 2 1 0.5");
1130
1131static IIO_DEVICE_ATTR(in_proximity_scale_available, S_IRUGO,
1132		       ltr501_show_proximity_scale_avail, NULL, 0);
1133static IIO_DEVICE_ATTR(in_intensity_scale_available, S_IRUGO,
1134		       ltr501_show_intensity_scale_avail, NULL, 0);
1135
1136static struct attribute *ltr501_attributes[] = {
1137	&iio_dev_attr_in_proximity_scale_available.dev_attr.attr,
1138	&iio_dev_attr_in_intensity_scale_available.dev_attr.attr,
1139	&iio_const_attr_integration_time_available.dev_attr.attr,
1140	&iio_const_attr_sampling_frequency_available.dev_attr.attr,
1141	NULL
1142};
1143
1144static struct attribute *ltr301_attributes[] = {
1145	&iio_dev_attr_in_intensity_scale_available.dev_attr.attr,
1146	&iio_const_attr_integration_time_available.dev_attr.attr,
1147	&iio_const_attr_sampling_frequency_available.dev_attr.attr,
1148	NULL
1149};
1150
1151static const struct attribute_group ltr501_attribute_group = {
1152	.attrs = ltr501_attributes,
1153};
1154
1155static const struct attribute_group ltr301_attribute_group = {
1156	.attrs = ltr301_attributes,
1157};
1158
1159static const struct iio_info ltr501_info_no_irq = {
1160	.read_raw = ltr501_read_raw,
1161	.write_raw = ltr501_write_raw,
1162	.attrs = &ltr501_attribute_group,
1163};
1164
1165static const struct iio_info ltr501_info = {
1166	.read_raw = ltr501_read_raw,
1167	.write_raw = ltr501_write_raw,
1168	.attrs = &ltr501_attribute_group,
1169	.read_event_value	= &ltr501_read_event,
1170	.write_event_value	= &ltr501_write_event,
1171	.read_event_config	= &ltr501_read_event_config,
1172	.write_event_config	= &ltr501_write_event_config,
1173};
1174
1175static const struct iio_info ltr301_info_no_irq = {
1176	.read_raw = ltr501_read_raw,
1177	.write_raw = ltr501_write_raw,
1178	.attrs = &ltr301_attribute_group,
1179};
1180
1181static const struct iio_info ltr301_info = {
1182	.read_raw = ltr501_read_raw,
1183	.write_raw = ltr501_write_raw,
1184	.attrs = &ltr301_attribute_group,
1185	.read_event_value	= &ltr501_read_event,
1186	.write_event_value	= &ltr501_write_event,
1187	.read_event_config	= &ltr501_read_event_config,
1188	.write_event_config	= &ltr501_write_event_config,
1189};
1190
1191static struct ltr501_chip_info ltr501_chip_info_tbl[] = {
1192	[ltr501] = {
1193		.partid = 0x08,
1194		.als_gain = ltr501_als_gain_tbl,
1195		.als_gain_tbl_size = ARRAY_SIZE(ltr501_als_gain_tbl),
1196		.ps_gain = ltr501_ps_gain_tbl,
1197		.ps_gain_tbl_size = ARRAY_SIZE(ltr501_ps_gain_tbl),
1198		.als_mode_active = BIT(0) | BIT(1),
1199		.als_gain_mask = BIT(3),
1200		.als_gain_shift = 3,
1201		.info = &ltr501_info,
1202		.info_no_irq = &ltr501_info_no_irq,
1203		.channels = ltr501_channels,
1204		.no_channels = ARRAY_SIZE(ltr501_channels),
1205	},
1206	[ltr559] = {
1207		.partid = 0x09,
1208		.als_gain = ltr559_als_gain_tbl,
1209		.als_gain_tbl_size = ARRAY_SIZE(ltr559_als_gain_tbl),
1210		.ps_gain = ltr559_ps_gain_tbl,
1211		.ps_gain_tbl_size = ARRAY_SIZE(ltr559_ps_gain_tbl),
1212		.als_mode_active = BIT(0),
1213		.als_gain_mask = BIT(2) | BIT(3) | BIT(4),
1214		.als_gain_shift = 2,
1215		.info = &ltr501_info,
1216		.info_no_irq = &ltr501_info_no_irq,
1217		.channels = ltr501_channels,
1218		.no_channels = ARRAY_SIZE(ltr501_channels),
1219	},
1220	[ltr301] = {
1221		.partid = 0x08,
1222		.als_gain = ltr501_als_gain_tbl,
1223		.als_gain_tbl_size = ARRAY_SIZE(ltr501_als_gain_tbl),
1224		.als_mode_active = BIT(0) | BIT(1),
1225		.als_gain_mask = BIT(3),
1226		.als_gain_shift = 3,
1227		.info = &ltr301_info,
1228		.info_no_irq = &ltr301_info_no_irq,
1229		.channels = ltr301_channels,
1230		.no_channels = ARRAY_SIZE(ltr301_channels),
1231	},
1232};
1233
1234static int ltr501_write_contr(struct ltr501_data *data, u8 als_val, u8 ps_val)
1235{
1236	int ret;
1237
1238	ret = regmap_write(data->regmap, LTR501_ALS_CONTR, als_val);
1239	if (ret < 0)
1240		return ret;
1241
1242	return regmap_write(data->regmap, LTR501_PS_CONTR, ps_val);
1243}
1244
1245static irqreturn_t ltr501_trigger_handler(int irq, void *p)
1246{
1247	struct iio_poll_func *pf = p;
1248	struct iio_dev *indio_dev = pf->indio_dev;
1249	struct ltr501_data *data = iio_priv(indio_dev);
1250	struct {
1251		u16 channels[3];
1252		s64 ts __aligned(8);
1253	} scan;
1254	__le16 als_buf[2];
1255	u8 mask = 0;
1256	int j = 0;
1257	int ret, psdata;
1258
1259	memset(&scan, 0, sizeof(scan));
1260
1261	/* figure out which data needs to be ready */
1262	if (test_bit(0, indio_dev->active_scan_mask) ||
1263	    test_bit(1, indio_dev->active_scan_mask))
1264		mask |= LTR501_STATUS_ALS_RDY;
1265	if (test_bit(2, indio_dev->active_scan_mask))
1266		mask |= LTR501_STATUS_PS_RDY;
1267
1268	ret = ltr501_drdy(data, mask);
1269	if (ret < 0)
1270		goto done;
1271
1272	if (mask & LTR501_STATUS_ALS_RDY) {
1273		ret = regmap_bulk_read(data->regmap, LTR501_ALS_DATA1,
1274				       als_buf, sizeof(als_buf));
1275		if (ret < 0)
1276			goto done;
1277		if (test_bit(0, indio_dev->active_scan_mask))
1278			scan.channels[j++] = le16_to_cpu(als_buf[1]);
1279		if (test_bit(1, indio_dev->active_scan_mask))
1280			scan.channels[j++] = le16_to_cpu(als_buf[0]);
1281	}
1282
1283	if (mask & LTR501_STATUS_PS_RDY) {
1284		ret = regmap_bulk_read(data->regmap, LTR501_PS_DATA,
1285				       &psdata, 2);
1286		if (ret < 0)
1287			goto done;
1288		scan.channels[j++] = psdata & LTR501_PS_DATA_MASK;
1289	}
1290
1291	iio_push_to_buffers_with_timestamp(indio_dev, &scan,
1292					   iio_get_time_ns(indio_dev));
1293
1294done:
1295	iio_trigger_notify_done(indio_dev->trig);
1296
1297	return IRQ_HANDLED;
1298}
1299
1300static irqreturn_t ltr501_interrupt_handler(int irq, void *private)
1301{
1302	struct iio_dev *indio_dev = private;
1303	struct ltr501_data *data = iio_priv(indio_dev);
1304	int ret, status;
1305
1306	ret = regmap_read(data->regmap, LTR501_ALS_PS_STATUS, &status);
1307	if (ret < 0) {
1308		dev_err(&data->client->dev,
1309			"irq read int reg failed\n");
1310		return IRQ_HANDLED;
1311	}
1312
1313	if (status & LTR501_STATUS_ALS_INTR)
1314		iio_push_event(indio_dev,
1315			       IIO_UNMOD_EVENT_CODE(IIO_INTENSITY, 0,
1316						    IIO_EV_TYPE_THRESH,
1317						    IIO_EV_DIR_EITHER),
1318			       iio_get_time_ns(indio_dev));
1319
1320	if (status & LTR501_STATUS_PS_INTR)
1321		iio_push_event(indio_dev,
1322			       IIO_UNMOD_EVENT_CODE(IIO_PROXIMITY, 0,
1323						    IIO_EV_TYPE_THRESH,
1324						    IIO_EV_DIR_EITHER),
1325			       iio_get_time_ns(indio_dev));
1326
1327	return IRQ_HANDLED;
1328}
1329
1330static int ltr501_init(struct ltr501_data *data)
1331{
1332	int ret, status;
1333
1334	ret = regmap_read(data->regmap, LTR501_ALS_CONTR, &status);
1335	if (ret < 0)
1336		return ret;
1337
1338	data->als_contr = status | data->chip_info->als_mode_active;
1339
1340	ret = regmap_read(data->regmap, LTR501_PS_CONTR, &status);
1341	if (ret < 0)
1342		return ret;
1343
1344	data->ps_contr = status | LTR501_CONTR_ACTIVE;
1345
1346	ret = ltr501_read_intr_prst(data, IIO_INTENSITY, &data->als_period);
1347	if (ret < 0)
1348		return ret;
1349
1350	ret = ltr501_read_intr_prst(data, IIO_PROXIMITY, &data->ps_period);
1351	if (ret < 0)
1352		return ret;
1353
1354	return ltr501_write_contr(data, data->als_contr, data->ps_contr);
1355}
1356
1357static bool ltr501_is_volatile_reg(struct device *dev, unsigned int reg)
1358{
1359	switch (reg) {
1360	case LTR501_ALS_DATA1:
1361	case LTR501_ALS_DATA1_UPPER:
1362	case LTR501_ALS_DATA0:
1363	case LTR501_ALS_DATA0_UPPER:
1364	case LTR501_ALS_PS_STATUS:
1365	case LTR501_PS_DATA:
1366	case LTR501_PS_DATA_UPPER:
1367		return true;
1368	default:
1369		return false;
1370	}
1371}
1372
1373static const struct regmap_config ltr501_regmap_config = {
1374	.name =  LTR501_REGMAP_NAME,
1375	.reg_bits = 8,
1376	.val_bits = 8,
1377	.max_register = LTR501_MAX_REG,
1378	.cache_type = REGCACHE_RBTREE,
1379	.volatile_reg = ltr501_is_volatile_reg,
1380};
1381
1382static int ltr501_powerdown(struct ltr501_data *data)
1383{
1384	return ltr501_write_contr(data, data->als_contr &
1385				  ~data->chip_info->als_mode_active,
1386				  data->ps_contr & ~LTR501_CONTR_ACTIVE);
1387}
1388
1389static const char *ltr501_match_acpi_device(struct device *dev, int *chip_idx)
1390{
1391	const struct acpi_device_id *id;
1392
1393	id = acpi_match_device(dev->driver->acpi_match_table, dev);
1394	if (!id)
1395		return NULL;
1396	*chip_idx = id->driver_data;
1397	return dev_name(dev);
1398}
1399
1400static int ltr501_probe(struct i2c_client *client,
1401			const struct i2c_device_id *id)
1402{
1403	struct ltr501_data *data;
1404	struct iio_dev *indio_dev;
1405	struct regmap *regmap;
1406	int ret, partid, chip_idx = 0;
1407	const char *name = NULL;
1408
1409	indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
1410	if (!indio_dev)
1411		return -ENOMEM;
1412
1413	regmap = devm_regmap_init_i2c(client, &ltr501_regmap_config);
1414	if (IS_ERR(regmap)) {
1415		dev_err(&client->dev, "Regmap initialization failed.\n");
1416		return PTR_ERR(regmap);
1417	}
1418
1419	data = iio_priv(indio_dev);
1420	i2c_set_clientdata(client, indio_dev);
1421	data->client = client;
1422	data->regmap = regmap;
1423	mutex_init(&data->lock_als);
1424	mutex_init(&data->lock_ps);
1425
1426	data->reg_it = devm_regmap_field_alloc(&client->dev, regmap,
1427					       reg_field_it);
1428	if (IS_ERR(data->reg_it)) {
1429		dev_err(&client->dev, "Integ time reg field init failed.\n");
1430		return PTR_ERR(data->reg_it);
1431	}
1432
1433	data->reg_als_intr = devm_regmap_field_alloc(&client->dev, regmap,
1434						     reg_field_als_intr);
1435	if (IS_ERR(data->reg_als_intr)) {
1436		dev_err(&client->dev, "ALS intr mode reg field init failed\n");
1437		return PTR_ERR(data->reg_als_intr);
1438	}
1439
1440	data->reg_ps_intr = devm_regmap_field_alloc(&client->dev, regmap,
1441						    reg_field_ps_intr);
1442	if (IS_ERR(data->reg_ps_intr)) {
1443		dev_err(&client->dev, "PS intr mode reg field init failed.\n");
1444		return PTR_ERR(data->reg_ps_intr);
1445	}
1446
1447	data->reg_als_rate = devm_regmap_field_alloc(&client->dev, regmap,
1448						     reg_field_als_rate);
1449	if (IS_ERR(data->reg_als_rate)) {
1450		dev_err(&client->dev, "ALS samp rate field init failed.\n");
1451		return PTR_ERR(data->reg_als_rate);
1452	}
1453
1454	data->reg_ps_rate = devm_regmap_field_alloc(&client->dev, regmap,
1455						    reg_field_ps_rate);
1456	if (IS_ERR(data->reg_ps_rate)) {
1457		dev_err(&client->dev, "PS samp rate field init failed.\n");
1458		return PTR_ERR(data->reg_ps_rate);
1459	}
1460
1461	data->reg_als_prst = devm_regmap_field_alloc(&client->dev, regmap,
1462						     reg_field_als_prst);
1463	if (IS_ERR(data->reg_als_prst)) {
1464		dev_err(&client->dev, "ALS prst reg field init failed\n");
1465		return PTR_ERR(data->reg_als_prst);
1466	}
1467
1468	data->reg_ps_prst = devm_regmap_field_alloc(&client->dev, regmap,
1469						    reg_field_ps_prst);
1470	if (IS_ERR(data->reg_ps_prst)) {
1471		dev_err(&client->dev, "PS prst reg field init failed.\n");
1472		return PTR_ERR(data->reg_ps_prst);
1473	}
1474
1475	ret = regmap_read(data->regmap, LTR501_PART_ID, &partid);
1476	if (ret < 0)
1477		return ret;
1478
1479	if (id) {
1480		name = id->name;
1481		chip_idx = id->driver_data;
1482	} else  if (ACPI_HANDLE(&client->dev)) {
1483		name = ltr501_match_acpi_device(&client->dev, &chip_idx);
1484	} else {
1485		return -ENODEV;
1486	}
1487
1488	data->chip_info = &ltr501_chip_info_tbl[chip_idx];
1489
1490	if ((partid >> 4) != data->chip_info->partid)
1491		return -ENODEV;
1492
1493	indio_dev->info = data->chip_info->info;
1494	indio_dev->channels = data->chip_info->channels;
1495	indio_dev->num_channels = data->chip_info->no_channels;
1496	indio_dev->name = name;
1497	indio_dev->modes = INDIO_DIRECT_MODE;
1498
1499	ret = ltr501_init(data);
1500	if (ret < 0)
1501		return ret;
1502
1503	if (client->irq > 0) {
1504		ret = devm_request_threaded_irq(&client->dev, client->irq,
1505						NULL, ltr501_interrupt_handler,
1506						IRQF_TRIGGER_FALLING |
1507						IRQF_ONESHOT,
1508						"ltr501_thresh_event",
1509						indio_dev);
1510		if (ret) {
1511			dev_err(&client->dev, "request irq (%d) failed\n",
1512				client->irq);
1513			return ret;
1514		}
1515	} else {
1516		indio_dev->info = data->chip_info->info_no_irq;
1517	}
1518
1519	ret = iio_triggered_buffer_setup(indio_dev, NULL,
1520					 ltr501_trigger_handler, NULL);
1521	if (ret)
1522		goto powerdown_on_error;
1523
1524	ret = iio_device_register(indio_dev);
1525	if (ret)
1526		goto error_unreg_buffer;
1527
1528	return 0;
1529
1530error_unreg_buffer:
1531	iio_triggered_buffer_cleanup(indio_dev);
1532powerdown_on_error:
1533	ltr501_powerdown(data);
1534	return ret;
1535}
1536
1537static int ltr501_remove(struct i2c_client *client)
1538{
1539	struct iio_dev *indio_dev = i2c_get_clientdata(client);
1540
1541	iio_device_unregister(indio_dev);
1542	iio_triggered_buffer_cleanup(indio_dev);
1543	ltr501_powerdown(iio_priv(indio_dev));
1544
1545	return 0;
1546}
1547
1548#ifdef CONFIG_PM_SLEEP
1549static int ltr501_suspend(struct device *dev)
1550{
1551	struct ltr501_data *data = iio_priv(i2c_get_clientdata(
1552					    to_i2c_client(dev)));
1553	return ltr501_powerdown(data);
1554}
1555
1556static int ltr501_resume(struct device *dev)
1557{
1558	struct ltr501_data *data = iio_priv(i2c_get_clientdata(
1559					    to_i2c_client(dev)));
1560
1561	return ltr501_write_contr(data, data->als_contr,
1562		data->ps_contr);
1563}
1564#endif
1565
1566static SIMPLE_DEV_PM_OPS(ltr501_pm_ops, ltr501_suspend, ltr501_resume);
1567
1568static const struct acpi_device_id ltr_acpi_match[] = {
1569	{"LTER0501", ltr501},
1570	{"LTER0559", ltr559},
1571	{"LTER0301", ltr301},
1572	{ },
1573};
1574MODULE_DEVICE_TABLE(acpi, ltr_acpi_match);
1575
1576static const struct i2c_device_id ltr501_id[] = {
1577	{ "ltr501", ltr501},
1578	{ "ltr559", ltr559},
1579	{ "ltr301", ltr301},
1580	{ }
1581};
1582MODULE_DEVICE_TABLE(i2c, ltr501_id);
1583
1584static struct i2c_driver ltr501_driver = {
1585	.driver = {
1586		.name   = LTR501_DRV_NAME,
1587		.pm	= &ltr501_pm_ops,
1588		.acpi_match_table = ACPI_PTR(ltr_acpi_match),
1589	},
1590	.probe  = ltr501_probe,
1591	.remove	= ltr501_remove,
1592	.id_table = ltr501_id,
1593};
1594
1595module_i2c_driver(ltr501_driver);
1596
1597MODULE_AUTHOR("Peter Meerwald <pmeerw@pmeerw.net>");
1598MODULE_DESCRIPTION("Lite-On LTR501 ambient light and proximity sensor driver");
1599MODULE_LICENSE("GPL");
1600