1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * apds9960.c - Support for Avago APDS9960 gesture/RGB/ALS/proximity sensor
4 *
5 * Copyright (C) 2015, 2018
6 * Author: Matt Ranostay <matt.ranostay@konsulko.com>
7 *
8 * TODO: gesture + proximity calib offsets
9 */
10
11#include <linux/acpi.h>
12#include <linux/module.h>
13#include <linux/init.h>
14#include <linux/interrupt.h>
15#include <linux/delay.h>
16#include <linux/mutex.h>
17#include <linux/err.h>
18#include <linux/irq.h>
19#include <linux/i2c.h>
20#include <linux/pm_runtime.h>
21#include <linux/regmap.h>
22#include <linux/iio/iio.h>
23#include <linux/iio/buffer.h>
24#include <linux/iio/events.h>
25#include <linux/iio/kfifo_buf.h>
26#include <linux/iio/sysfs.h>
27
28#define APDS9960_REGMAP_NAME	"apds9960_regmap"
29#define APDS9960_DRV_NAME	"apds9960"
30
31#define APDS9960_REG_RAM_START	0x00
32#define APDS9960_REG_RAM_END	0x7f
33
34#define APDS9960_REG_ENABLE	0x80
35#define APDS9960_REG_ATIME	0x81
36#define APDS9960_REG_WTIME	0x83
37
38#define APDS9960_REG_AILTL	0x84
39#define APDS9960_REG_AILTH	0x85
40#define APDS9960_REG_AIHTL	0x86
41#define APDS9960_REG_AIHTH	0x87
42
43#define APDS9960_REG_PILT	0x89
44#define APDS9960_REG_PIHT	0x8b
45#define APDS9960_REG_PERS	0x8c
46
47#define APDS9960_REG_CONFIG_1	0x8d
48#define APDS9960_REG_PPULSE	0x8e
49
50#define APDS9960_REG_CONTROL	0x8f
51#define APDS9960_REG_CONTROL_AGAIN_MASK		0x03
52#define APDS9960_REG_CONTROL_PGAIN_MASK		0x0c
53#define APDS9960_REG_CONTROL_AGAIN_MASK_SHIFT	0
54#define APDS9960_REG_CONTROL_PGAIN_MASK_SHIFT	2
55
56#define APDS9960_REG_CONFIG_2	0x90
57#define APDS9960_REG_ID		0x92
58
59#define APDS9960_REG_STATUS	0x93
60#define APDS9960_REG_STATUS_PS_INT	BIT(5)
61#define APDS9960_REG_STATUS_ALS_INT	BIT(4)
62#define APDS9960_REG_STATUS_GINT	BIT(2)
63
64#define APDS9960_REG_PDATA	0x9c
65#define APDS9960_REG_POFFSET_UR	0x9d
66#define APDS9960_REG_POFFSET_DL 0x9e
67#define APDS9960_REG_CONFIG_3	0x9f
68
69#define APDS9960_REG_GPENTH	0xa0
70#define APDS9960_REG_GEXTH	0xa1
71
72#define APDS9960_REG_GCONF_1	0xa2
73#define APDS9960_REG_GCONF_1_GFIFO_THRES_MASK		0xc0
74#define APDS9960_REG_GCONF_1_GFIFO_THRES_MASK_SHIFT	6
75
76#define APDS9960_REG_GCONF_2	0xa3
77#define APDS9960_REG_GCONF_2_GGAIN_MASK			0x60
78#define APDS9960_REG_GCONF_2_GGAIN_MASK_SHIFT		5
79
80#define APDS9960_REG_GOFFSET_U	0xa4
81#define APDS9960_REG_GOFFSET_D	0xa5
82#define APDS9960_REG_GPULSE	0xa6
83#define APDS9960_REG_GOFFSET_L	0xa7
84#define APDS9960_REG_GOFFSET_R	0xa9
85#define APDS9960_REG_GCONF_3	0xaa
86
87#define APDS9960_REG_GCONF_4	0xab
88#define APDS9960_REG_GFLVL	0xae
89#define APDS9960_REG_GSTATUS	0xaf
90
91#define APDS9960_REG_IFORCE	0xe4
92#define APDS9960_REG_PICLEAR	0xe5
93#define APDS9960_REG_CICLEAR	0xe6
94#define APDS9960_REG_AICLEAR	0xe7
95
96#define APDS9960_DEFAULT_PERS	0x33
97#define APDS9960_DEFAULT_GPENTH	0x50
98#define APDS9960_DEFAULT_GEXTH	0x40
99
100#define APDS9960_MAX_PXS_THRES_VAL	255
101#define APDS9960_MAX_ALS_THRES_VAL	0xffff
102#define APDS9960_MAX_INT_TIME_IN_US	1000000
103
104enum apds9960_als_channel_idx {
105	IDX_ALS_CLEAR, IDX_ALS_RED, IDX_ALS_GREEN, IDX_ALS_BLUE,
106};
107
108#define APDS9960_REG_ALS_BASE	0x94
109#define APDS9960_REG_ALS_CHANNEL(_colour) \
110	(APDS9960_REG_ALS_BASE + (IDX_ALS_##_colour * 2))
111
112enum apds9960_gesture_channel_idx {
113	IDX_DIR_UP, IDX_DIR_DOWN, IDX_DIR_LEFT, IDX_DIR_RIGHT,
114};
115
116#define APDS9960_REG_GFIFO_BASE	0xfc
117#define APDS9960_REG_GFIFO_DIR(_dir) \
118	(APDS9960_REG_GFIFO_BASE + IDX_DIR_##_dir)
119
120struct apds9960_data {
121	struct i2c_client *client;
122	struct iio_dev *indio_dev;
123	struct mutex lock;
124
125	/* regmap fields */
126	struct regmap *regmap;
127	struct regmap_field *reg_int_als;
128	struct regmap_field *reg_int_ges;
129	struct regmap_field *reg_int_pxs;
130
131	struct regmap_field *reg_enable_als;
132	struct regmap_field *reg_enable_ges;
133	struct regmap_field *reg_enable_pxs;
134
135	/* state */
136	int als_int;
137	int pxs_int;
138	int gesture_mode_running;
139
140	/* gain values */
141	int als_gain;
142	int pxs_gain;
143
144	/* integration time value in us */
145	int als_adc_int_us;
146
147	/* gesture buffer */
148	u8 buffer[4]; /* 4 8-bit channels */
149};
150
151static const struct reg_default apds9960_reg_defaults[] = {
152	/* Default ALS integration time = 2.48ms */
153	{ APDS9960_REG_ATIME, 0xff },
154};
155
156static const struct regmap_range apds9960_volatile_ranges[] = {
157	regmap_reg_range(APDS9960_REG_STATUS,
158				APDS9960_REG_PDATA),
159	regmap_reg_range(APDS9960_REG_GFLVL,
160				APDS9960_REG_GSTATUS),
161	regmap_reg_range(APDS9960_REG_GFIFO_DIR(UP),
162				APDS9960_REG_GFIFO_DIR(RIGHT)),
163	regmap_reg_range(APDS9960_REG_IFORCE,
164				APDS9960_REG_AICLEAR),
165};
166
167static const struct regmap_access_table apds9960_volatile_table = {
168	.yes_ranges	= apds9960_volatile_ranges,
169	.n_yes_ranges	= ARRAY_SIZE(apds9960_volatile_ranges),
170};
171
172static const struct regmap_range apds9960_precious_ranges[] = {
173	regmap_reg_range(APDS9960_REG_RAM_START, APDS9960_REG_RAM_END),
174};
175
176static const struct regmap_access_table apds9960_precious_table = {
177	.yes_ranges	= apds9960_precious_ranges,
178	.n_yes_ranges	= ARRAY_SIZE(apds9960_precious_ranges),
179};
180
181static const struct regmap_range apds9960_readable_ranges[] = {
182	regmap_reg_range(APDS9960_REG_ENABLE,
183				APDS9960_REG_GSTATUS),
184	regmap_reg_range(APDS9960_REG_GFIFO_DIR(UP),
185				APDS9960_REG_GFIFO_DIR(RIGHT)),
186};
187
188static const struct regmap_access_table apds9960_readable_table = {
189	.yes_ranges	= apds9960_readable_ranges,
190	.n_yes_ranges	= ARRAY_SIZE(apds9960_readable_ranges),
191};
192
193static const struct regmap_range apds9960_writeable_ranges[] = {
194	regmap_reg_range(APDS9960_REG_ENABLE, APDS9960_REG_CONFIG_2),
195	regmap_reg_range(APDS9960_REG_POFFSET_UR, APDS9960_REG_GCONF_4),
196	regmap_reg_range(APDS9960_REG_IFORCE, APDS9960_REG_AICLEAR),
197};
198
199static const struct regmap_access_table apds9960_writeable_table = {
200	.yes_ranges	= apds9960_writeable_ranges,
201	.n_yes_ranges	= ARRAY_SIZE(apds9960_writeable_ranges),
202};
203
204static const struct regmap_config apds9960_regmap_config = {
205	.name = APDS9960_REGMAP_NAME,
206	.reg_bits = 8,
207	.val_bits = 8,
208	.use_single_read = true,
209	.use_single_write = true,
210
211	.volatile_table = &apds9960_volatile_table,
212	.precious_table = &apds9960_precious_table,
213	.rd_table = &apds9960_readable_table,
214	.wr_table = &apds9960_writeable_table,
215
216	.reg_defaults = apds9960_reg_defaults,
217	.num_reg_defaults = ARRAY_SIZE(apds9960_reg_defaults),
218	.max_register = APDS9960_REG_GFIFO_DIR(RIGHT),
219	.cache_type = REGCACHE_RBTREE,
220};
221
222static const struct iio_event_spec apds9960_pxs_event_spec[] = {
223	{
224		.type = IIO_EV_TYPE_THRESH,
225		.dir = IIO_EV_DIR_RISING,
226		.mask_separate = BIT(IIO_EV_INFO_VALUE),
227	},
228	{
229		.type = IIO_EV_TYPE_THRESH,
230		.dir = IIO_EV_DIR_FALLING,
231		.mask_separate = BIT(IIO_EV_INFO_VALUE),
232	},
233	{
234		.type = IIO_EV_TYPE_THRESH,
235		.mask_separate = BIT(IIO_EV_INFO_ENABLE),
236	},
237};
238
239static const struct iio_event_spec apds9960_als_event_spec[] = {
240	{
241		.type = IIO_EV_TYPE_THRESH,
242		.dir = IIO_EV_DIR_RISING,
243		.mask_separate = BIT(IIO_EV_INFO_VALUE),
244	},
245	{
246		.type = IIO_EV_TYPE_THRESH,
247		.dir = IIO_EV_DIR_FALLING,
248		.mask_separate = BIT(IIO_EV_INFO_VALUE),
249	},
250	{
251		.type = IIO_EV_TYPE_THRESH,
252		.mask_separate = BIT(IIO_EV_INFO_ENABLE),
253	},
254};
255
256#define APDS9960_GESTURE_CHANNEL(_dir, _si) { \
257	.type = IIO_PROXIMITY, \
258	.channel = _si + 1, \
259	.scan_index = _si, \
260	.indexed = 1, \
261	.scan_type = { \
262		.sign = 'u', \
263		.realbits = 8, \
264		.storagebits = 8, \
265	}, \
266}
267
268#define APDS9960_INTENSITY_CHANNEL(_colour) { \
269	.type = IIO_INTENSITY, \
270	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW), \
271	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE) | \
272			BIT(IIO_CHAN_INFO_INT_TIME), \
273	.channel2 = IIO_MOD_LIGHT_##_colour, \
274	.address = APDS9960_REG_ALS_CHANNEL(_colour), \
275	.modified = 1, \
276	.scan_index = -1, \
277}
278
279static const unsigned long apds9960_scan_masks[] = {0xf, 0};
280
281static const struct iio_chan_spec apds9960_channels[] = {
282	{
283		.type = IIO_PROXIMITY,
284		.address = APDS9960_REG_PDATA,
285		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),
286		.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),
287		.channel = 0,
288		.indexed = 0,
289		.scan_index = -1,
290
291		.event_spec = apds9960_pxs_event_spec,
292		.num_event_specs = ARRAY_SIZE(apds9960_pxs_event_spec),
293	},
294	/* Gesture Sensor */
295	APDS9960_GESTURE_CHANNEL(UP, 0),
296	APDS9960_GESTURE_CHANNEL(DOWN, 1),
297	APDS9960_GESTURE_CHANNEL(LEFT, 2),
298	APDS9960_GESTURE_CHANNEL(RIGHT, 3),
299	/* ALS */
300	{
301		.type = IIO_INTENSITY,
302		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),
303		.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE) |
304			BIT(IIO_CHAN_INFO_INT_TIME),
305		.channel2 = IIO_MOD_LIGHT_CLEAR,
306		.address = APDS9960_REG_ALS_CHANNEL(CLEAR),
307		.modified = 1,
308		.scan_index = -1,
309
310		.event_spec = apds9960_als_event_spec,
311		.num_event_specs = ARRAY_SIZE(apds9960_als_event_spec),
312	},
313	/* RGB Sensor */
314	APDS9960_INTENSITY_CHANNEL(RED),
315	APDS9960_INTENSITY_CHANNEL(GREEN),
316	APDS9960_INTENSITY_CHANNEL(BLUE),
317};
318
319/* integration time in us */
320static const int apds9960_int_time[][2] = {
321	{ 28000, 246},
322	{100000, 219},
323	{200000, 182},
324	{700000,   0}
325};
326
327/* gain mapping */
328static const int apds9960_pxs_gain_map[] = {1, 2, 4, 8};
329static const int apds9960_als_gain_map[] = {1, 4, 16, 64};
330
331static IIO_CONST_ATTR(proximity_scale_available, "1 2 4 8");
332static IIO_CONST_ATTR(intensity_scale_available, "1 4 16 64");
333static IIO_CONST_ATTR_INT_TIME_AVAIL("0.028 0.1 0.2 0.7");
334
335static struct attribute *apds9960_attributes[] = {
336	&iio_const_attr_proximity_scale_available.dev_attr.attr,
337	&iio_const_attr_intensity_scale_available.dev_attr.attr,
338	&iio_const_attr_integration_time_available.dev_attr.attr,
339	NULL,
340};
341
342static const struct attribute_group apds9960_attribute_group = {
343	.attrs = apds9960_attributes,
344};
345
346static const struct reg_field apds9960_reg_field_int_als =
347				REG_FIELD(APDS9960_REG_ENABLE, 4, 4);
348
349static const struct reg_field apds9960_reg_field_int_ges =
350				REG_FIELD(APDS9960_REG_GCONF_4, 1, 1);
351
352static const struct reg_field apds9960_reg_field_int_pxs =
353				REG_FIELD(APDS9960_REG_ENABLE, 5, 5);
354
355static const struct reg_field apds9960_reg_field_enable_als =
356				REG_FIELD(APDS9960_REG_ENABLE, 1, 1);
357
358static const struct reg_field apds9960_reg_field_enable_ges =
359				REG_FIELD(APDS9960_REG_ENABLE, 6, 6);
360
361static const struct reg_field apds9960_reg_field_enable_pxs =
362				REG_FIELD(APDS9960_REG_ENABLE, 2, 2);
363
364static int apds9960_set_it_time(struct apds9960_data *data, int val2)
365{
366	int ret = -EINVAL;
367	int idx;
368
369	for (idx = 0; idx < ARRAY_SIZE(apds9960_int_time); idx++) {
370		if (apds9960_int_time[idx][0] == val2) {
371			mutex_lock(&data->lock);
372			ret = regmap_write(data->regmap, APDS9960_REG_ATIME,
373						 apds9960_int_time[idx][1]);
374			if (!ret)
375				data->als_adc_int_us = val2;
376			mutex_unlock(&data->lock);
377			break;
378		}
379	}
380
381	return ret;
382}
383
384static int apds9960_set_pxs_gain(struct apds9960_data *data, int val)
385{
386	int ret = -EINVAL;
387	int idx;
388
389	for (idx = 0; idx < ARRAY_SIZE(apds9960_pxs_gain_map); idx++) {
390		if (apds9960_pxs_gain_map[idx] == val) {
391			/* pxs + gesture gains are mirrored */
392			mutex_lock(&data->lock);
393			ret = regmap_update_bits(data->regmap,
394				APDS9960_REG_CONTROL,
395				APDS9960_REG_CONTROL_PGAIN_MASK,
396				idx << APDS9960_REG_CONTROL_PGAIN_MASK_SHIFT);
397			if (ret) {
398				mutex_unlock(&data->lock);
399				break;
400			}
401
402			ret = regmap_update_bits(data->regmap,
403				APDS9960_REG_GCONF_2,
404				APDS9960_REG_GCONF_2_GGAIN_MASK,
405				idx << APDS9960_REG_GCONF_2_GGAIN_MASK_SHIFT);
406			if (!ret)
407				data->pxs_gain = idx;
408			mutex_unlock(&data->lock);
409			break;
410		}
411	}
412
413	return ret;
414}
415
416static int apds9960_set_als_gain(struct apds9960_data *data, int val)
417{
418	int ret = -EINVAL;
419	int idx;
420
421	for (idx = 0; idx < ARRAY_SIZE(apds9960_als_gain_map); idx++) {
422		if (apds9960_als_gain_map[idx] == val) {
423			mutex_lock(&data->lock);
424			ret = regmap_update_bits(data->regmap,
425					APDS9960_REG_CONTROL,
426					APDS9960_REG_CONTROL_AGAIN_MASK, idx);
427			if (!ret)
428				data->als_gain = idx;
429			mutex_unlock(&data->lock);
430			break;
431		}
432	}
433
434	return ret;
435}
436
437#ifdef CONFIG_PM
438static int apds9960_set_power_state(struct apds9960_data *data, bool on)
439{
440	struct device *dev = &data->client->dev;
441	int ret = 0;
442
443	mutex_lock(&data->lock);
444
445	if (on) {
446		int suspended;
447
448		suspended = pm_runtime_suspended(dev);
449		ret = pm_runtime_get_sync(dev);
450
451		/* Allow one integration cycle before allowing a reading */
452		if (suspended)
453			usleep_range(data->als_adc_int_us,
454				     APDS9960_MAX_INT_TIME_IN_US);
455	} else {
456		pm_runtime_mark_last_busy(dev);
457		ret = pm_runtime_put_autosuspend(dev);
458	}
459
460	mutex_unlock(&data->lock);
461
462	return ret;
463}
464#else
465static int apds9960_set_power_state(struct apds9960_data *data, bool on)
466{
467	return 0;
468}
469#endif
470
471static int apds9960_read_raw(struct iio_dev *indio_dev,
472			     struct iio_chan_spec const *chan,
473			     int *val, int *val2, long mask)
474{
475	struct apds9960_data *data = iio_priv(indio_dev);
476	__le16 buf;
477	int ret = -EINVAL;
478
479	if (data->gesture_mode_running)
480		return -EBUSY;
481
482	switch (mask) {
483	case IIO_CHAN_INFO_RAW:
484		apds9960_set_power_state(data, true);
485		switch (chan->type) {
486		case IIO_PROXIMITY:
487			ret = regmap_read(data->regmap, chan->address, val);
488			if (!ret)
489				ret = IIO_VAL_INT;
490			break;
491		case IIO_INTENSITY:
492			ret = regmap_bulk_read(data->regmap, chan->address,
493					       &buf, 2);
494			if (!ret) {
495				ret = IIO_VAL_INT;
496				*val = le16_to_cpu(buf);
497			}
498			break;
499		default:
500			ret = -EINVAL;
501		}
502		apds9960_set_power_state(data, false);
503		break;
504	case IIO_CHAN_INFO_INT_TIME:
505		/* RGB + ALS sensors only have integration time */
506		mutex_lock(&data->lock);
507		switch (chan->type) {
508		case IIO_INTENSITY:
509			*val = 0;
510			*val2 = data->als_adc_int_us;
511			ret = IIO_VAL_INT_PLUS_MICRO;
512			break;
513		default:
514			ret = -EINVAL;
515		}
516		mutex_unlock(&data->lock);
517		break;
518	case IIO_CHAN_INFO_SCALE:
519		mutex_lock(&data->lock);
520		switch (chan->type) {
521		case IIO_PROXIMITY:
522			*val = apds9960_pxs_gain_map[data->pxs_gain];
523			ret = IIO_VAL_INT;
524			break;
525		case IIO_INTENSITY:
526			*val = apds9960_als_gain_map[data->als_gain];
527			ret = IIO_VAL_INT;
528			break;
529		default:
530			ret = -EINVAL;
531		}
532		mutex_unlock(&data->lock);
533		break;
534	}
535
536	return ret;
537};
538
539static int apds9960_write_raw(struct iio_dev *indio_dev,
540			     struct iio_chan_spec const *chan,
541			     int val, int val2, long mask)
542{
543	struct apds9960_data *data = iio_priv(indio_dev);
544
545	switch (mask) {
546	case IIO_CHAN_INFO_INT_TIME:
547		/* RGB + ALS sensors only have int time */
548		switch (chan->type) {
549		case IIO_INTENSITY:
550			if (val != 0)
551				return -EINVAL;
552			return apds9960_set_it_time(data, val2);
553		default:
554			return -EINVAL;
555		}
556	case IIO_CHAN_INFO_SCALE:
557		if (val2 != 0)
558			return -EINVAL;
559		switch (chan->type) {
560		case IIO_PROXIMITY:
561			return apds9960_set_pxs_gain(data, val);
562		case IIO_INTENSITY:
563			return apds9960_set_als_gain(data, val);
564		default:
565			return -EINVAL;
566		}
567	default:
568		return -EINVAL;
569	}
570
571	return 0;
572}
573
574static inline int apds9960_get_thres_reg(const struct iio_chan_spec *chan,
575					 enum iio_event_direction dir,
576					 u8 *reg)
577{
578	switch (dir) {
579	case IIO_EV_DIR_RISING:
580		switch (chan->type) {
581		case IIO_PROXIMITY:
582			*reg = APDS9960_REG_PIHT;
583			break;
584		case IIO_INTENSITY:
585			*reg = APDS9960_REG_AIHTL;
586			break;
587		default:
588			return -EINVAL;
589		}
590		break;
591	case IIO_EV_DIR_FALLING:
592		switch (chan->type) {
593		case IIO_PROXIMITY:
594			*reg = APDS9960_REG_PILT;
595			break;
596		case IIO_INTENSITY:
597			*reg = APDS9960_REG_AILTL;
598			break;
599		default:
600			return -EINVAL;
601		}
602		break;
603	default:
604		return -EINVAL;
605	}
606
607	return 0;
608}
609
610static int apds9960_read_event(struct iio_dev *indio_dev,
611			       const struct iio_chan_spec *chan,
612			       enum iio_event_type type,
613			       enum iio_event_direction dir,
614			       enum iio_event_info info,
615			       int *val, int *val2)
616{
617	u8 reg;
618	__le16 buf;
619	int ret = 0;
620	struct apds9960_data *data = iio_priv(indio_dev);
621
622	if (info != IIO_EV_INFO_VALUE)
623		return -EINVAL;
624
625	ret = apds9960_get_thres_reg(chan, dir, &reg);
626	if (ret < 0)
627		return ret;
628
629	if (chan->type == IIO_PROXIMITY) {
630		ret = regmap_read(data->regmap, reg, val);
631		if (ret < 0)
632			return ret;
633	} else if (chan->type == IIO_INTENSITY) {
634		ret = regmap_bulk_read(data->regmap, reg, &buf, 2);
635		if (ret < 0)
636			return ret;
637		*val = le16_to_cpu(buf);
638	} else
639		return -EINVAL;
640
641	*val2 = 0;
642
643	return IIO_VAL_INT;
644}
645
646static int apds9960_write_event(struct iio_dev *indio_dev,
647				const struct iio_chan_spec *chan,
648				enum iio_event_type type,
649				enum iio_event_direction dir,
650				enum iio_event_info info,
651				int val, int val2)
652{
653	u8 reg;
654	__le16 buf;
655	int ret = 0;
656	struct apds9960_data *data = iio_priv(indio_dev);
657
658	if (info != IIO_EV_INFO_VALUE)
659		return -EINVAL;
660
661	ret = apds9960_get_thres_reg(chan, dir, &reg);
662	if (ret < 0)
663		return ret;
664
665	if (chan->type == IIO_PROXIMITY) {
666		if (val < 0 || val > APDS9960_MAX_PXS_THRES_VAL)
667			return -EINVAL;
668		ret = regmap_write(data->regmap, reg, val);
669		if (ret < 0)
670			return ret;
671	} else if (chan->type == IIO_INTENSITY) {
672		if (val < 0 || val > APDS9960_MAX_ALS_THRES_VAL)
673			return -EINVAL;
674		buf = cpu_to_le16(val);
675		ret = regmap_bulk_write(data->regmap, reg, &buf, 2);
676		if (ret < 0)
677			return ret;
678	} else
679		return -EINVAL;
680
681	return 0;
682}
683
684static int apds9960_read_event_config(struct iio_dev *indio_dev,
685				      const struct iio_chan_spec *chan,
686				      enum iio_event_type type,
687				      enum iio_event_direction dir)
688{
689	struct apds9960_data *data = iio_priv(indio_dev);
690
691	switch (chan->type) {
692	case IIO_PROXIMITY:
693		return data->pxs_int;
694	case IIO_INTENSITY:
695		return data->als_int;
696	default:
697		return -EINVAL;
698	}
699
700	return 0;
701}
702
703static int apds9960_write_event_config(struct iio_dev *indio_dev,
704				       const struct iio_chan_spec *chan,
705				       enum iio_event_type type,
706				       enum iio_event_direction dir,
707				       int state)
708{
709	struct apds9960_data *data = iio_priv(indio_dev);
710	int ret;
711
712	state = !!state;
713
714	switch (chan->type) {
715	case IIO_PROXIMITY:
716		if (data->pxs_int == state)
717			return -EINVAL;
718
719		ret = regmap_field_write(data->reg_int_pxs, state);
720		if (ret)
721			return ret;
722		data->pxs_int = state;
723		apds9960_set_power_state(data, state);
724		break;
725	case IIO_INTENSITY:
726		if (data->als_int == state)
727			return -EINVAL;
728
729		ret = regmap_field_write(data->reg_int_als, state);
730		if (ret)
731			return ret;
732		data->als_int = state;
733		apds9960_set_power_state(data, state);
734		break;
735	default:
736		return -EINVAL;
737	}
738
739	return 0;
740}
741
742static const struct iio_info apds9960_info = {
743	.attrs = &apds9960_attribute_group,
744	.read_raw = apds9960_read_raw,
745	.write_raw = apds9960_write_raw,
746	.read_event_value = apds9960_read_event,
747	.write_event_value = apds9960_write_event,
748	.read_event_config = apds9960_read_event_config,
749	.write_event_config = apds9960_write_event_config,
750
751};
752
753static inline int apds9660_fifo_is_empty(struct apds9960_data *data)
754{
755	int cnt;
756	int ret;
757
758	ret = regmap_read(data->regmap, APDS9960_REG_GFLVL, &cnt);
759	if (ret)
760		return ret;
761
762	return cnt;
763}
764
765static void apds9960_read_gesture_fifo(struct apds9960_data *data)
766{
767	int ret, cnt = 0;
768
769	mutex_lock(&data->lock);
770	data->gesture_mode_running = 1;
771
772	while (cnt || (cnt = apds9660_fifo_is_empty(data) > 0)) {
773		ret = regmap_bulk_read(data->regmap, APDS9960_REG_GFIFO_BASE,
774				      &data->buffer, 4);
775
776		if (ret)
777			goto err_read;
778
779		iio_push_to_buffers(data->indio_dev, data->buffer);
780		cnt--;
781	}
782
783err_read:
784	data->gesture_mode_running = 0;
785	mutex_unlock(&data->lock);
786}
787
788static irqreturn_t apds9960_interrupt_handler(int irq, void *private)
789{
790	struct iio_dev *indio_dev = private;
791	struct apds9960_data *data = iio_priv(indio_dev);
792	int ret, status;
793
794	ret = regmap_read(data->regmap, APDS9960_REG_STATUS, &status);
795	if (ret < 0) {
796		dev_err(&data->client->dev, "irq status reg read failed\n");
797		return IRQ_HANDLED;
798	}
799
800	if ((status & APDS9960_REG_STATUS_ALS_INT) && data->als_int) {
801		iio_push_event(indio_dev,
802			       IIO_UNMOD_EVENT_CODE(IIO_INTENSITY, 0,
803						    IIO_EV_TYPE_THRESH,
804						    IIO_EV_DIR_EITHER),
805			       iio_get_time_ns(indio_dev));
806		regmap_write(data->regmap, APDS9960_REG_CICLEAR, 1);
807	}
808
809	if ((status & APDS9960_REG_STATUS_PS_INT) && data->pxs_int) {
810		iio_push_event(indio_dev,
811			       IIO_UNMOD_EVENT_CODE(IIO_PROXIMITY, 0,
812						    IIO_EV_TYPE_THRESH,
813						    IIO_EV_DIR_EITHER),
814			       iio_get_time_ns(indio_dev));
815		regmap_write(data->regmap, APDS9960_REG_PICLEAR, 1);
816	}
817
818	if (status & APDS9960_REG_STATUS_GINT)
819		apds9960_read_gesture_fifo(data);
820
821	return IRQ_HANDLED;
822}
823
824static int apds9960_set_powermode(struct apds9960_data *data, bool state)
825{
826	return regmap_update_bits(data->regmap, APDS9960_REG_ENABLE, 1, state);
827}
828
829static int apds9960_buffer_postenable(struct iio_dev *indio_dev)
830{
831	struct apds9960_data *data = iio_priv(indio_dev);
832	int ret;
833
834	ret = regmap_field_write(data->reg_int_ges, 1);
835	if (ret)
836		return ret;
837
838	ret = regmap_field_write(data->reg_enable_ges, 1);
839	if (ret)
840		return ret;
841
842	pm_runtime_get_sync(&data->client->dev);
843
844	return 0;
845}
846
847static int apds9960_buffer_predisable(struct iio_dev *indio_dev)
848{
849	struct apds9960_data *data = iio_priv(indio_dev);
850	int ret;
851
852	ret = regmap_field_write(data->reg_enable_ges, 0);
853	if (ret)
854		return ret;
855
856	ret = regmap_field_write(data->reg_int_ges, 0);
857	if (ret)
858		return ret;
859
860	pm_runtime_put_autosuspend(&data->client->dev);
861
862	return 0;
863}
864
865static const struct iio_buffer_setup_ops apds9960_buffer_setup_ops = {
866	.postenable = apds9960_buffer_postenable,
867	.predisable = apds9960_buffer_predisable,
868};
869
870static int apds9960_regfield_init(struct apds9960_data *data)
871{
872	struct device *dev = &data->client->dev;
873	struct regmap *regmap = data->regmap;
874
875	data->reg_int_als = devm_regmap_field_alloc(dev, regmap,
876						apds9960_reg_field_int_als);
877	if (IS_ERR(data->reg_int_als)) {
878		dev_err(dev, "INT ALS reg field init failed\n");
879		return PTR_ERR(data->reg_int_als);
880	}
881
882	data->reg_int_ges = devm_regmap_field_alloc(dev, regmap,
883						apds9960_reg_field_int_ges);
884	if (IS_ERR(data->reg_int_ges)) {
885		dev_err(dev, "INT gesture reg field init failed\n");
886		return PTR_ERR(data->reg_int_ges);
887	}
888
889	data->reg_int_pxs = devm_regmap_field_alloc(dev, regmap,
890						apds9960_reg_field_int_pxs);
891	if (IS_ERR(data->reg_int_pxs)) {
892		dev_err(dev, "INT pxs reg field init failed\n");
893		return PTR_ERR(data->reg_int_pxs);
894	}
895
896	data->reg_enable_als = devm_regmap_field_alloc(dev, regmap,
897						apds9960_reg_field_enable_als);
898	if (IS_ERR(data->reg_enable_als)) {
899		dev_err(dev, "Enable ALS reg field init failed\n");
900		return PTR_ERR(data->reg_enable_als);
901	}
902
903	data->reg_enable_ges = devm_regmap_field_alloc(dev, regmap,
904						apds9960_reg_field_enable_ges);
905	if (IS_ERR(data->reg_enable_ges)) {
906		dev_err(dev, "Enable gesture reg field init failed\n");
907		return PTR_ERR(data->reg_enable_ges);
908	}
909
910	data->reg_enable_pxs = devm_regmap_field_alloc(dev, regmap,
911						apds9960_reg_field_enable_pxs);
912	if (IS_ERR(data->reg_enable_pxs)) {
913		dev_err(dev, "Enable PXS reg field init failed\n");
914		return PTR_ERR(data->reg_enable_pxs);
915	}
916
917	return 0;
918}
919
920static int apds9960_chip_init(struct apds9960_data *data)
921{
922	int ret;
923
924	/* Default IT for ALS of 28 ms */
925	ret = apds9960_set_it_time(data, 28000);
926	if (ret)
927		return ret;
928
929	/* Ensure gesture interrupt is OFF */
930	ret = regmap_field_write(data->reg_int_ges, 0);
931	if (ret)
932		return ret;
933
934	/* Disable gesture sensor, since polling is useless from user-space */
935	ret = regmap_field_write(data->reg_enable_ges, 0);
936	if (ret)
937		return ret;
938
939	/* Ensure proximity interrupt is OFF */
940	ret = regmap_field_write(data->reg_int_pxs, 0);
941	if (ret)
942		return ret;
943
944	/* Enable proximity sensor for polling */
945	ret = regmap_field_write(data->reg_enable_pxs, 1);
946	if (ret)
947		return ret;
948
949	/* Ensure ALS interrupt is OFF */
950	ret = regmap_field_write(data->reg_int_als, 0);
951	if (ret)
952		return ret;
953
954	/* Enable ALS sensor for polling */
955	ret = regmap_field_write(data->reg_enable_als, 1);
956	if (ret)
957		return ret;
958	/*
959	 * When enabled trigger an interrupt after 3 readings
960	 * outside threshold for ALS + PXS
961	 */
962	ret = regmap_write(data->regmap, APDS9960_REG_PERS,
963			   APDS9960_DEFAULT_PERS);
964	if (ret)
965		return ret;
966
967	/*
968	 * Wait for 4 event outside gesture threshold to prevent interrupt
969	 * flooding.
970	 */
971	ret = regmap_update_bits(data->regmap, APDS9960_REG_GCONF_1,
972			APDS9960_REG_GCONF_1_GFIFO_THRES_MASK,
973			BIT(0) << APDS9960_REG_GCONF_1_GFIFO_THRES_MASK_SHIFT);
974	if (ret)
975		return ret;
976
977	/* Default ENTER and EXIT thresholds for the GESTURE engine. */
978	ret = regmap_write(data->regmap, APDS9960_REG_GPENTH,
979			   APDS9960_DEFAULT_GPENTH);
980	if (ret)
981		return ret;
982
983	ret = regmap_write(data->regmap, APDS9960_REG_GEXTH,
984			   APDS9960_DEFAULT_GEXTH);
985	if (ret)
986		return ret;
987
988	return apds9960_set_powermode(data, 1);
989}
990
991static int apds9960_probe(struct i2c_client *client)
992{
993	struct apds9960_data *data;
994	struct iio_dev *indio_dev;
995	int ret;
996
997	indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
998	if (!indio_dev)
999		return -ENOMEM;
1000
1001	indio_dev->info = &apds9960_info;
1002	indio_dev->name = APDS9960_DRV_NAME;
1003	indio_dev->channels = apds9960_channels;
1004	indio_dev->num_channels = ARRAY_SIZE(apds9960_channels);
1005	indio_dev->available_scan_masks = apds9960_scan_masks;
1006	indio_dev->modes = INDIO_DIRECT_MODE;
1007
1008	ret = devm_iio_kfifo_buffer_setup(&client->dev, indio_dev,
1009					  &apds9960_buffer_setup_ops);
1010	if (ret)
1011		return ret;
1012
1013	data = iio_priv(indio_dev);
1014	i2c_set_clientdata(client, indio_dev);
1015
1016	data->regmap = devm_regmap_init_i2c(client, &apds9960_regmap_config);
1017	if (IS_ERR(data->regmap)) {
1018		dev_err(&client->dev, "regmap initialization failed.\n");
1019		return PTR_ERR(data->regmap);
1020	}
1021
1022	data->client = client;
1023	data->indio_dev = indio_dev;
1024	mutex_init(&data->lock);
1025
1026	ret = pm_runtime_set_active(&client->dev);
1027	if (ret)
1028		goto error_power_down;
1029
1030	pm_runtime_enable(&client->dev);
1031	pm_runtime_set_autosuspend_delay(&client->dev, 5000);
1032	pm_runtime_use_autosuspend(&client->dev);
1033
1034	apds9960_set_power_state(data, true);
1035
1036	ret = apds9960_regfield_init(data);
1037	if (ret)
1038		goto error_power_down;
1039
1040	ret = apds9960_chip_init(data);
1041	if (ret)
1042		goto error_power_down;
1043
1044	if (client->irq <= 0) {
1045		dev_err(&client->dev, "no valid irq defined\n");
1046		ret = -EINVAL;
1047		goto error_power_down;
1048	}
1049	ret = devm_request_threaded_irq(&client->dev, client->irq,
1050					NULL, apds9960_interrupt_handler,
1051					IRQF_TRIGGER_FALLING | IRQF_ONESHOT,
1052					"apds9960_event",
1053					indio_dev);
1054	if (ret) {
1055		dev_err(&client->dev, "request irq (%d) failed\n", client->irq);
1056		goto error_power_down;
1057	}
1058
1059	ret = iio_device_register(indio_dev);
1060	if (ret)
1061		goto error_power_down;
1062
1063	apds9960_set_power_state(data, false);
1064
1065	return 0;
1066
1067error_power_down:
1068	apds9960_set_power_state(data, false);
1069
1070	return ret;
1071}
1072
1073static void apds9960_remove(struct i2c_client *client)
1074{
1075	struct iio_dev *indio_dev = i2c_get_clientdata(client);
1076	struct apds9960_data *data = iio_priv(indio_dev);
1077
1078	iio_device_unregister(indio_dev);
1079	pm_runtime_disable(&client->dev);
1080	pm_runtime_set_suspended(&client->dev);
1081	apds9960_set_powermode(data, 0);
1082}
1083
1084#ifdef CONFIG_PM
1085static int apds9960_runtime_suspend(struct device *dev)
1086{
1087	struct apds9960_data *data =
1088			iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
1089
1090	return apds9960_set_powermode(data, 0);
1091}
1092
1093static int apds9960_runtime_resume(struct device *dev)
1094{
1095	struct apds9960_data *data =
1096			iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
1097
1098	return apds9960_set_powermode(data, 1);
1099}
1100#endif
1101
1102static const struct dev_pm_ops apds9960_pm_ops = {
1103	SET_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
1104				pm_runtime_force_resume)
1105	SET_RUNTIME_PM_OPS(apds9960_runtime_suspend,
1106			   apds9960_runtime_resume, NULL)
1107};
1108
1109static const struct i2c_device_id apds9960_id[] = {
1110	{ "apds9960", 0 },
1111	{}
1112};
1113MODULE_DEVICE_TABLE(i2c, apds9960_id);
1114
1115static const struct acpi_device_id apds9960_acpi_match[] = {
1116	{ "MSHW0184" },
1117	{ }
1118};
1119MODULE_DEVICE_TABLE(acpi, apds9960_acpi_match);
1120
1121static const struct of_device_id apds9960_of_match[] = {
1122	{ .compatible = "avago,apds9960" },
1123	{ }
1124};
1125MODULE_DEVICE_TABLE(of, apds9960_of_match);
1126
1127static struct i2c_driver apds9960_driver = {
1128	.driver = {
1129		.name	= APDS9960_DRV_NAME,
1130		.of_match_table = apds9960_of_match,
1131		.pm	= &apds9960_pm_ops,
1132		.acpi_match_table = apds9960_acpi_match,
1133	},
1134	.probe		= apds9960_probe,
1135	.remove		= apds9960_remove,
1136	.id_table	= apds9960_id,
1137};
1138module_i2c_driver(apds9960_driver);
1139
1140MODULE_AUTHOR("Matt Ranostay <matt.ranostay@konsulko.com>");
1141MODULE_DESCRIPTION("APDS9960 Gesture/RGB/ALS/Proximity sensor");
1142MODULE_LICENSE("GPL");
1143