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