Linux Audio

Check our new training course

Loading...
v4.6
 
  1/**
  2 * Sensortek STK3310/STK3311 Ambient Light and Proximity Sensor
  3 *
  4 * Copyright (c) 2015, Intel Corporation.
  5 *
  6 * This file is subject to the terms and conditions of version 2 of
  7 * the GNU General Public License. See the file COPYING in the main
  8 * directory of this archive for more details.
  9 *
 10 * IIO driver for STK3310/STK3311. 7-bit I2C address: 0x48.
 11 */
 12
 13#include <linux/acpi.h>
 14#include <linux/i2c.h>
 15#include <linux/interrupt.h>
 16#include <linux/kernel.h>
 17#include <linux/module.h>
 18#include <linux/regmap.h>
 19#include <linux/gpio/consumer.h>
 20#include <linux/iio/events.h>
 21#include <linux/iio/iio.h>
 22#include <linux/iio/sysfs.h>
 23
 24#define STK3310_REG_STATE			0x00
 25#define STK3310_REG_PSCTRL			0x01
 26#define STK3310_REG_ALSCTRL			0x02
 27#define STK3310_REG_INT				0x04
 28#define STK3310_REG_THDH_PS			0x06
 29#define STK3310_REG_THDL_PS			0x08
 30#define STK3310_REG_FLAG			0x10
 31#define STK3310_REG_PS_DATA_MSB			0x11
 32#define STK3310_REG_PS_DATA_LSB			0x12
 33#define STK3310_REG_ALS_DATA_MSB		0x13
 34#define STK3310_REG_ALS_DATA_LSB		0x14
 35#define STK3310_REG_ID				0x3E
 36#define STK3310_MAX_REG				0x80
 37
 38#define STK3310_STATE_EN_PS			BIT(0)
 39#define STK3310_STATE_EN_ALS			BIT(1)
 40#define STK3310_STATE_STANDBY			0x00
 41
 42#define STK3310_CHIP_ID_VAL			0x13
 43#define STK3311_CHIP_ID_VAL			0x1D
 
 44#define STK3310_PSINT_EN			0x01
 45#define STK3310_PS_MAX_VAL			0xFFFF
 46
 47#define STK3310_DRIVER_NAME			"stk3310"
 48#define STK3310_REGMAP_NAME			"stk3310_regmap"
 49#define STK3310_EVENT				"stk3310_event"
 50
 51#define STK3310_SCALE_AVAILABLE			"6.4 1.6 0.4 0.1"
 52
 53#define STK3310_IT_AVAILABLE \
 54	"0.000185 0.000370 0.000741 0.001480 0.002960 0.005920 0.011840 " \
 55	"0.023680 0.047360 0.094720 0.189440 0.378880 0.757760 1.515520 " \
 56	"3.031040 6.062080"
 57
 58#define STK3310_REGFIELD(name)						    \
 59	do {								    \
 60		data->reg_##name =					    \
 61			devm_regmap_field_alloc(&client->dev, regmap,	    \
 62				stk3310_reg_field_##name);		    \
 63		if (IS_ERR(data->reg_##name)) {				    \
 64			dev_err(&client->dev, "reg field alloc failed.\n"); \
 65			return PTR_ERR(data->reg_##name);		    \
 66		}							    \
 67	} while (0)
 68
 69static const struct reg_field stk3310_reg_field_state =
 70				REG_FIELD(STK3310_REG_STATE, 0, 2);
 71static const struct reg_field stk3310_reg_field_als_gain =
 72				REG_FIELD(STK3310_REG_ALSCTRL, 4, 5);
 73static const struct reg_field stk3310_reg_field_ps_gain =
 74				REG_FIELD(STK3310_REG_PSCTRL, 4, 5);
 75static const struct reg_field stk3310_reg_field_als_it =
 76				REG_FIELD(STK3310_REG_ALSCTRL, 0, 3);
 77static const struct reg_field stk3310_reg_field_ps_it =
 78				REG_FIELD(STK3310_REG_PSCTRL, 0, 3);
 79static const struct reg_field stk3310_reg_field_int_ps =
 80				REG_FIELD(STK3310_REG_INT, 0, 2);
 81static const struct reg_field stk3310_reg_field_flag_psint =
 82				REG_FIELD(STK3310_REG_FLAG, 4, 4);
 83static const struct reg_field stk3310_reg_field_flag_nf =
 84				REG_FIELD(STK3310_REG_FLAG, 0, 0);
 85
 86/* Estimate maximum proximity values with regard to measurement scale. */
 87static const int stk3310_ps_max[4] = {
 88	STK3310_PS_MAX_VAL / 640,
 89	STK3310_PS_MAX_VAL / 160,
 90	STK3310_PS_MAX_VAL /  40,
 91	STK3310_PS_MAX_VAL /  10
 92};
 93
 94static const int stk3310_scale_table[][2] = {
 95	{6, 400000}, {1, 600000}, {0, 400000}, {0, 100000}
 96};
 97
 98/* Integration time in seconds, microseconds */
 99static const int stk3310_it_table[][2] = {
100	{0, 185},	{0, 370},	{0, 741},	{0, 1480},
101	{0, 2960},	{0, 5920},	{0, 11840},	{0, 23680},
102	{0, 47360},	{0, 94720},	{0, 189440},	{0, 378880},
103	{0, 757760},	{1, 515520},	{3, 31040},	{6, 62080},
104};
105
106struct stk3310_data {
107	struct i2c_client *client;
108	struct mutex lock;
109	bool als_enabled;
110	bool ps_enabled;
111	u64 timestamp;
112	struct regmap *regmap;
113	struct regmap_field *reg_state;
114	struct regmap_field *reg_als_gain;
115	struct regmap_field *reg_ps_gain;
116	struct regmap_field *reg_als_it;
117	struct regmap_field *reg_ps_it;
118	struct regmap_field *reg_int_ps;
119	struct regmap_field *reg_flag_psint;
120	struct regmap_field *reg_flag_nf;
121};
122
123static const struct iio_event_spec stk3310_events[] = {
124	/* Proximity event */
125	{
126		.type = IIO_EV_TYPE_THRESH,
127		.dir = IIO_EV_DIR_RISING,
128		.mask_separate = BIT(IIO_EV_INFO_VALUE) |
129				 BIT(IIO_EV_INFO_ENABLE),
130	},
131	/* Out-of-proximity event */
132	{
133		.type = IIO_EV_TYPE_THRESH,
134		.dir = IIO_EV_DIR_FALLING,
135		.mask_separate = BIT(IIO_EV_INFO_VALUE) |
136				 BIT(IIO_EV_INFO_ENABLE),
137	},
138};
139
140static const struct iio_chan_spec stk3310_channels[] = {
141	{
142		.type = IIO_LIGHT,
143		.info_mask_separate =
144			BIT(IIO_CHAN_INFO_RAW) |
145			BIT(IIO_CHAN_INFO_SCALE) |
146			BIT(IIO_CHAN_INFO_INT_TIME),
147	},
148	{
149		.type = IIO_PROXIMITY,
150		.info_mask_separate =
151			BIT(IIO_CHAN_INFO_RAW) |
152			BIT(IIO_CHAN_INFO_SCALE) |
153			BIT(IIO_CHAN_INFO_INT_TIME),
154		.event_spec = stk3310_events,
155		.num_event_specs = ARRAY_SIZE(stk3310_events),
156	}
157};
158
159static IIO_CONST_ATTR(in_illuminance_scale_available, STK3310_SCALE_AVAILABLE);
160
161static IIO_CONST_ATTR(in_proximity_scale_available, STK3310_SCALE_AVAILABLE);
162
163static IIO_CONST_ATTR(in_illuminance_integration_time_available,
164		      STK3310_IT_AVAILABLE);
165
166static IIO_CONST_ATTR(in_proximity_integration_time_available,
167		      STK3310_IT_AVAILABLE);
168
169static struct attribute *stk3310_attributes[] = {
170	&iio_const_attr_in_illuminance_scale_available.dev_attr.attr,
171	&iio_const_attr_in_proximity_scale_available.dev_attr.attr,
172	&iio_const_attr_in_illuminance_integration_time_available.dev_attr.attr,
173	&iio_const_attr_in_proximity_integration_time_available.dev_attr.attr,
174	NULL,
175};
176
177static const struct attribute_group stk3310_attribute_group = {
178	.attrs = stk3310_attributes
179};
180
181static int stk3310_get_index(const int table[][2], int table_size,
182			     int val, int val2)
183{
184	int i;
185
186	for (i = 0; i < table_size; i++) {
187		if (val == table[i][0] && val2 == table[i][1])
188			return i;
189	}
190
191	return -EINVAL;
192}
193
194static int stk3310_read_event(struct iio_dev *indio_dev,
195			      const struct iio_chan_spec *chan,
196			      enum iio_event_type type,
197			      enum iio_event_direction dir,
198			      enum iio_event_info info,
199			      int *val, int *val2)
200{
201	u8 reg;
202	__be16 buf;
203	int ret;
204	struct stk3310_data *data = iio_priv(indio_dev);
205
206	if (info != IIO_EV_INFO_VALUE)
207		return -EINVAL;
208
209	/* Only proximity interrupts are implemented at the moment. */
210	if (dir == IIO_EV_DIR_RISING)
211		reg = STK3310_REG_THDH_PS;
212	else if (dir == IIO_EV_DIR_FALLING)
213		reg = STK3310_REG_THDL_PS;
214	else
215		return -EINVAL;
216
217	mutex_lock(&data->lock);
218	ret = regmap_bulk_read(data->regmap, reg, &buf, 2);
219	mutex_unlock(&data->lock);
220	if (ret < 0) {
221		dev_err(&data->client->dev, "register read failed\n");
222		return ret;
223	}
224	*val = be16_to_cpu(buf);
225
226	return IIO_VAL_INT;
227}
228
229static int stk3310_write_event(struct iio_dev *indio_dev,
230			       const struct iio_chan_spec *chan,
231			       enum iio_event_type type,
232			       enum iio_event_direction dir,
233			       enum iio_event_info info,
234			       int val, int val2)
235{
236	u8 reg;
237	__be16 buf;
238	int ret;
239	unsigned int index;
240	struct stk3310_data *data = iio_priv(indio_dev);
241	struct i2c_client *client = data->client;
242
243	ret = regmap_field_read(data->reg_ps_gain, &index);
244	if (ret < 0)
245		return ret;
246
247	if (val < 0 || val > stk3310_ps_max[index])
248		return -EINVAL;
249
250	if (dir == IIO_EV_DIR_RISING)
251		reg = STK3310_REG_THDH_PS;
252	else if (dir == IIO_EV_DIR_FALLING)
253		reg = STK3310_REG_THDL_PS;
254	else
255		return -EINVAL;
256
257	buf = cpu_to_be16(val);
258	ret = regmap_bulk_write(data->regmap, reg, &buf, 2);
259	if (ret < 0)
260		dev_err(&client->dev, "failed to set PS threshold!\n");
261
262	return ret;
263}
264
265static int stk3310_read_event_config(struct iio_dev *indio_dev,
266				     const struct iio_chan_spec *chan,
267				     enum iio_event_type type,
268				     enum iio_event_direction dir)
269{
270	unsigned int event_val;
271	int ret;
272	struct stk3310_data *data = iio_priv(indio_dev);
273
274	ret = regmap_field_read(data->reg_int_ps, &event_val);
275	if (ret < 0)
276		return ret;
277
278	return event_val;
279}
280
281static int stk3310_write_event_config(struct iio_dev *indio_dev,
282				      const struct iio_chan_spec *chan,
283				      enum iio_event_type type,
284				      enum iio_event_direction dir,
285				      int state)
286{
287	int ret;
288	struct stk3310_data *data = iio_priv(indio_dev);
289	struct i2c_client *client = data->client;
290
291	if (state < 0 || state > 7)
292		return -EINVAL;
293
294	/* Set INT_PS value */
295	mutex_lock(&data->lock);
296	ret = regmap_field_write(data->reg_int_ps, state);
297	if (ret < 0)
298		dev_err(&client->dev, "failed to set interrupt mode\n");
299	mutex_unlock(&data->lock);
300
301	return ret;
302}
303
304static int stk3310_read_raw(struct iio_dev *indio_dev,
305			    struct iio_chan_spec const *chan,
306			    int *val, int *val2, long mask)
307{
308	u8 reg;
309	__be16 buf;
310	int ret;
311	unsigned int index;
312	struct stk3310_data *data = iio_priv(indio_dev);
313	struct i2c_client *client = data->client;
314
315	if (chan->type != IIO_LIGHT && chan->type != IIO_PROXIMITY)
316		return -EINVAL;
317
318	switch (mask) {
319	case IIO_CHAN_INFO_RAW:
320		if (chan->type == IIO_LIGHT)
321			reg = STK3310_REG_ALS_DATA_MSB;
322		else
323			reg = STK3310_REG_PS_DATA_MSB;
324
325		mutex_lock(&data->lock);
326		ret = regmap_bulk_read(data->regmap, reg, &buf, 2);
327		if (ret < 0) {
328			dev_err(&client->dev, "register read failed\n");
329			mutex_unlock(&data->lock);
330			return ret;
331		}
332		*val = be16_to_cpu(buf);
333		mutex_unlock(&data->lock);
334		return IIO_VAL_INT;
335	case IIO_CHAN_INFO_INT_TIME:
336		if (chan->type == IIO_LIGHT)
337			ret = regmap_field_read(data->reg_als_it, &index);
338		else
339			ret = regmap_field_read(data->reg_ps_it, &index);
340		if (ret < 0)
341			return ret;
342
343		*val = stk3310_it_table[index][0];
344		*val2 = stk3310_it_table[index][1];
345		return IIO_VAL_INT_PLUS_MICRO;
346	case IIO_CHAN_INFO_SCALE:
347		if (chan->type == IIO_LIGHT)
348			ret = regmap_field_read(data->reg_als_gain, &index);
349		else
350			ret = regmap_field_read(data->reg_ps_gain, &index);
351		if (ret < 0)
352			return ret;
353
354		*val = stk3310_scale_table[index][0];
355		*val2 = stk3310_scale_table[index][1];
356		return IIO_VAL_INT_PLUS_MICRO;
357	}
358
359	return -EINVAL;
360}
361
362static int stk3310_write_raw(struct iio_dev *indio_dev,
363			     struct iio_chan_spec const *chan,
364			     int val, int val2, long mask)
365{
366	int ret;
367	int index;
368	struct stk3310_data *data = iio_priv(indio_dev);
369
370	if (chan->type != IIO_LIGHT && chan->type != IIO_PROXIMITY)
371		return -EINVAL;
372
373	switch (mask) {
374	case IIO_CHAN_INFO_INT_TIME:
375		index = stk3310_get_index(stk3310_it_table,
376					  ARRAY_SIZE(stk3310_it_table),
377					  val, val2);
378		if (index < 0)
379			return -EINVAL;
380		mutex_lock(&data->lock);
381		if (chan->type == IIO_LIGHT)
382			ret = regmap_field_write(data->reg_als_it, index);
383		else
384			ret = regmap_field_write(data->reg_ps_it, index);
385		if (ret < 0)
386			dev_err(&data->client->dev,
387				"sensor configuration failed\n");
388		mutex_unlock(&data->lock);
389		return ret;
390
391	case IIO_CHAN_INFO_SCALE:
392		index = stk3310_get_index(stk3310_scale_table,
393					  ARRAY_SIZE(stk3310_scale_table),
394					  val, val2);
395		if (index < 0)
396			return -EINVAL;
397		mutex_lock(&data->lock);
398		if (chan->type == IIO_LIGHT)
399			ret = regmap_field_write(data->reg_als_gain, index);
400		else
401			ret = regmap_field_write(data->reg_ps_gain, index);
402		if (ret < 0)
403			dev_err(&data->client->dev,
404				"sensor configuration failed\n");
405		mutex_unlock(&data->lock);
406		return ret;
407	}
408
409	return -EINVAL;
410}
411
412static const struct iio_info stk3310_info = {
413	.driver_module		= THIS_MODULE,
414	.read_raw		= stk3310_read_raw,
415	.write_raw		= stk3310_write_raw,
416	.attrs			= &stk3310_attribute_group,
417	.read_event_value	= stk3310_read_event,
418	.write_event_value	= stk3310_write_event,
419	.read_event_config	= stk3310_read_event_config,
420	.write_event_config	= stk3310_write_event_config,
421};
422
423static int stk3310_set_state(struct stk3310_data *data, u8 state)
424{
425	int ret;
426	struct i2c_client *client = data->client;
427
428	/* 3-bit state; 0b100 is not supported. */
429	if (state > 7 || state == 4)
430		return -EINVAL;
431
432	mutex_lock(&data->lock);
433	ret = regmap_field_write(data->reg_state, state);
434	if (ret < 0) {
435		dev_err(&client->dev, "failed to change sensor state\n");
436	} else if (state != STK3310_STATE_STANDBY) {
437		/* Don't reset the 'enabled' flags if we're going in standby */
438		data->ps_enabled  = !!(state & STK3310_STATE_EN_PS);
439		data->als_enabled = !!(state & STK3310_STATE_EN_ALS);
440	}
441	mutex_unlock(&data->lock);
442
443	return ret;
444}
445
446static int stk3310_init(struct iio_dev *indio_dev)
447{
448	int ret;
449	int chipid;
450	u8 state;
451	struct stk3310_data *data = iio_priv(indio_dev);
452	struct i2c_client *client = data->client;
453
454	ret = regmap_read(data->regmap, STK3310_REG_ID, &chipid);
455	if (ret < 0)
456		return ret;
457
458	if (chipid != STK3310_CHIP_ID_VAL &&
459	    chipid != STK3311_CHIP_ID_VAL) {
 
460		dev_err(&client->dev, "invalid chip id: 0x%x\n", chipid);
461		return -ENODEV;
462	}
463
464	state = STK3310_STATE_EN_ALS | STK3310_STATE_EN_PS;
465	ret = stk3310_set_state(data, state);
466	if (ret < 0) {
467		dev_err(&client->dev, "failed to enable sensor");
468		return ret;
469	}
470
471	/* Enable PS interrupts */
472	ret = regmap_field_write(data->reg_int_ps, STK3310_PSINT_EN);
473	if (ret < 0)
474		dev_err(&client->dev, "failed to enable interrupts!\n");
475
476	return ret;
477}
478
479static bool stk3310_is_volatile_reg(struct device *dev, unsigned int reg)
480{
481	switch (reg) {
482	case STK3310_REG_ALS_DATA_MSB:
483	case STK3310_REG_ALS_DATA_LSB:
484	case STK3310_REG_PS_DATA_LSB:
485	case STK3310_REG_PS_DATA_MSB:
486	case STK3310_REG_FLAG:
487		return true;
488	default:
489		return false;
490	}
491}
492
493static struct regmap_config stk3310_regmap_config = {
494	.name = STK3310_REGMAP_NAME,
495	.reg_bits = 8,
496	.val_bits = 8,
497	.max_register = STK3310_MAX_REG,
498	.cache_type = REGCACHE_RBTREE,
499	.volatile_reg = stk3310_is_volatile_reg,
500};
501
502static int stk3310_regmap_init(struct stk3310_data *data)
503{
504	struct regmap *regmap;
505	struct i2c_client *client;
506
507	client = data->client;
508	regmap = devm_regmap_init_i2c(client, &stk3310_regmap_config);
509	if (IS_ERR(regmap)) {
510		dev_err(&client->dev, "regmap initialization failed.\n");
511		return PTR_ERR(regmap);
512	}
513	data->regmap = regmap;
514
515	STK3310_REGFIELD(state);
516	STK3310_REGFIELD(als_gain);
517	STK3310_REGFIELD(ps_gain);
518	STK3310_REGFIELD(als_it);
519	STK3310_REGFIELD(ps_it);
520	STK3310_REGFIELD(int_ps);
521	STK3310_REGFIELD(flag_psint);
522	STK3310_REGFIELD(flag_nf);
523
524	return 0;
525}
526
527static irqreturn_t stk3310_irq_handler(int irq, void *private)
528{
529	struct iio_dev *indio_dev = private;
530	struct stk3310_data *data = iio_priv(indio_dev);
531
532	data->timestamp = iio_get_time_ns();
533
534	return IRQ_WAKE_THREAD;
535}
536
537static irqreturn_t stk3310_irq_event_handler(int irq, void *private)
538{
539	int ret;
540	unsigned int dir;
541	u64 event;
542
543	struct iio_dev *indio_dev = private;
544	struct stk3310_data *data = iio_priv(indio_dev);
545
546	/* Read FLAG_NF to figure out what threshold has been met. */
547	mutex_lock(&data->lock);
548	ret = regmap_field_read(data->reg_flag_nf, &dir);
549	if (ret < 0) {
550		dev_err(&data->client->dev, "register read failed\n");
551		mutex_unlock(&data->lock);
552		return ret;
553	}
554	event = IIO_UNMOD_EVENT_CODE(IIO_PROXIMITY, 1,
555				     IIO_EV_TYPE_THRESH,
556				     (dir ? IIO_EV_DIR_FALLING :
557					    IIO_EV_DIR_RISING));
558	iio_push_event(indio_dev, event, data->timestamp);
559
560	/* Reset the interrupt flag */
561	ret = regmap_field_write(data->reg_flag_psint, 0);
562	if (ret < 0)
563		dev_err(&data->client->dev, "failed to reset interrupts\n");
564	mutex_unlock(&data->lock);
565
566	return IRQ_HANDLED;
567}
568
569static int stk3310_probe(struct i2c_client *client,
570			 const struct i2c_device_id *id)
571{
572	int ret;
573	struct iio_dev *indio_dev;
574	struct stk3310_data *data;
575
576	indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
577	if (!indio_dev) {
578		dev_err(&client->dev, "iio allocation failed!\n");
579		return -ENOMEM;
580	}
581
582	data = iio_priv(indio_dev);
583	data->client = client;
584	i2c_set_clientdata(client, indio_dev);
585	mutex_init(&data->lock);
586
587	ret = stk3310_regmap_init(data);
588	if (ret < 0)
589		return ret;
590
591	indio_dev->dev.parent = &client->dev;
592	indio_dev->info = &stk3310_info;
593	indio_dev->name = STK3310_DRIVER_NAME;
594	indio_dev->modes = INDIO_DIRECT_MODE;
595	indio_dev->channels = stk3310_channels;
596	indio_dev->num_channels = ARRAY_SIZE(stk3310_channels);
597
598	ret = stk3310_init(indio_dev);
599	if (ret < 0)
600		return ret;
601
602	if (client->irq > 0) {
603		ret = devm_request_threaded_irq(&client->dev, client->irq,
604						stk3310_irq_handler,
605						stk3310_irq_event_handler,
606						IRQF_TRIGGER_FALLING |
607						IRQF_ONESHOT,
608						STK3310_EVENT, indio_dev);
609		if (ret < 0) {
610			dev_err(&client->dev, "request irq %d failed\n",
611				client->irq);
612			goto err_standby;
613		}
614	}
615
616	ret = iio_device_register(indio_dev);
617	if (ret < 0) {
618		dev_err(&client->dev, "device_register failed\n");
619		goto err_standby;
620	}
621
622	return 0;
623
624err_standby:
625	stk3310_set_state(data, STK3310_STATE_STANDBY);
626	return ret;
627}
628
629static int stk3310_remove(struct i2c_client *client)
630{
631	struct iio_dev *indio_dev = i2c_get_clientdata(client);
632
633	iio_device_unregister(indio_dev);
634	return stk3310_set_state(iio_priv(indio_dev), STK3310_STATE_STANDBY);
635}
636
637#ifdef CONFIG_PM_SLEEP
638static int stk3310_suspend(struct device *dev)
639{
640	struct stk3310_data *data;
641
642	data = iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
643
644	return stk3310_set_state(data, STK3310_STATE_STANDBY);
645}
646
647static int stk3310_resume(struct device *dev)
648{
649	u8 state = 0;
650	struct stk3310_data *data;
651
652	data = iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
653	if (data->ps_enabled)
654		state |= STK3310_STATE_EN_PS;
655	if (data->als_enabled)
656		state |= STK3310_STATE_EN_ALS;
657
658	return stk3310_set_state(data, state);
659}
660
661static SIMPLE_DEV_PM_OPS(stk3310_pm_ops, stk3310_suspend, stk3310_resume);
662
663#define STK3310_PM_OPS (&stk3310_pm_ops)
664#else
665#define STK3310_PM_OPS NULL
666#endif
667
668static const struct i2c_device_id stk3310_i2c_id[] = {
669	{"STK3310", 0},
670	{"STK3311", 0},
 
671	{}
672};
673MODULE_DEVICE_TABLE(i2c, stk3310_i2c_id);
674
675static const struct acpi_device_id stk3310_acpi_id[] = {
676	{"STK3310", 0},
677	{"STK3311", 0},
 
678	{}
679};
680
681MODULE_DEVICE_TABLE(acpi, stk3310_acpi_id);
682
 
 
 
 
 
 
 
 
683static struct i2c_driver stk3310_driver = {
684	.driver = {
685		.name = "stk3310",
 
686		.pm = STK3310_PM_OPS,
687		.acpi_match_table = ACPI_PTR(stk3310_acpi_id),
688	},
689	.probe =            stk3310_probe,
690	.remove =           stk3310_remove,
691	.id_table =         stk3310_i2c_id,
692};
693
694module_i2c_driver(stk3310_driver);
695
696MODULE_AUTHOR("Tiberiu Breana <tiberiu.a.breana@intel.com>");
697MODULE_DESCRIPTION("STK3310 Ambient Light and Proximity Sensor driver");
698MODULE_LICENSE("GPL v2");
v5.4
  1// SPDX-License-Identifier: GPL-2.0-only
  2/**
  3 * Sensortek STK3310/STK3311 Ambient Light and Proximity Sensor
  4 *
  5 * Copyright (c) 2015, Intel Corporation.
  6 *
 
 
 
 
  7 * IIO driver for STK3310/STK3311. 7-bit I2C address: 0x48.
  8 */
  9
 10#include <linux/acpi.h>
 11#include <linux/i2c.h>
 12#include <linux/interrupt.h>
 13#include <linux/kernel.h>
 14#include <linux/module.h>
 15#include <linux/regmap.h>
 
 16#include <linux/iio/events.h>
 17#include <linux/iio/iio.h>
 18#include <linux/iio/sysfs.h>
 19
 20#define STK3310_REG_STATE			0x00
 21#define STK3310_REG_PSCTRL			0x01
 22#define STK3310_REG_ALSCTRL			0x02
 23#define STK3310_REG_INT				0x04
 24#define STK3310_REG_THDH_PS			0x06
 25#define STK3310_REG_THDL_PS			0x08
 26#define STK3310_REG_FLAG			0x10
 27#define STK3310_REG_PS_DATA_MSB			0x11
 28#define STK3310_REG_PS_DATA_LSB			0x12
 29#define STK3310_REG_ALS_DATA_MSB		0x13
 30#define STK3310_REG_ALS_DATA_LSB		0x14
 31#define STK3310_REG_ID				0x3E
 32#define STK3310_MAX_REG				0x80
 33
 34#define STK3310_STATE_EN_PS			BIT(0)
 35#define STK3310_STATE_EN_ALS			BIT(1)
 36#define STK3310_STATE_STANDBY			0x00
 37
 38#define STK3310_CHIP_ID_VAL			0x13
 39#define STK3311_CHIP_ID_VAL			0x1D
 40#define STK3335_CHIP_ID_VAL			0x51
 41#define STK3310_PSINT_EN			0x01
 42#define STK3310_PS_MAX_VAL			0xFFFF
 43
 44#define STK3310_DRIVER_NAME			"stk3310"
 45#define STK3310_REGMAP_NAME			"stk3310_regmap"
 46#define STK3310_EVENT				"stk3310_event"
 47
 48#define STK3310_SCALE_AVAILABLE			"6.4 1.6 0.4 0.1"
 49
 50#define STK3310_IT_AVAILABLE \
 51	"0.000185 0.000370 0.000741 0.001480 0.002960 0.005920 0.011840 " \
 52	"0.023680 0.047360 0.094720 0.189440 0.378880 0.757760 1.515520 " \
 53	"3.031040 6.062080"
 54
 55#define STK3310_REGFIELD(name)						    \
 56	do {								    \
 57		data->reg_##name =					    \
 58			devm_regmap_field_alloc(&client->dev, regmap,	    \
 59				stk3310_reg_field_##name);		    \
 60		if (IS_ERR(data->reg_##name)) {				    \
 61			dev_err(&client->dev, "reg field alloc failed.\n"); \
 62			return PTR_ERR(data->reg_##name);		    \
 63		}							    \
 64	} while (0)
 65
 66static const struct reg_field stk3310_reg_field_state =
 67				REG_FIELD(STK3310_REG_STATE, 0, 2);
 68static const struct reg_field stk3310_reg_field_als_gain =
 69				REG_FIELD(STK3310_REG_ALSCTRL, 4, 5);
 70static const struct reg_field stk3310_reg_field_ps_gain =
 71				REG_FIELD(STK3310_REG_PSCTRL, 4, 5);
 72static const struct reg_field stk3310_reg_field_als_it =
 73				REG_FIELD(STK3310_REG_ALSCTRL, 0, 3);
 74static const struct reg_field stk3310_reg_field_ps_it =
 75				REG_FIELD(STK3310_REG_PSCTRL, 0, 3);
 76static const struct reg_field stk3310_reg_field_int_ps =
 77				REG_FIELD(STK3310_REG_INT, 0, 2);
 78static const struct reg_field stk3310_reg_field_flag_psint =
 79				REG_FIELD(STK3310_REG_FLAG, 4, 4);
 80static const struct reg_field stk3310_reg_field_flag_nf =
 81				REG_FIELD(STK3310_REG_FLAG, 0, 0);
 82
 83/* Estimate maximum proximity values with regard to measurement scale. */
 84static const int stk3310_ps_max[4] = {
 85	STK3310_PS_MAX_VAL / 640,
 86	STK3310_PS_MAX_VAL / 160,
 87	STK3310_PS_MAX_VAL /  40,
 88	STK3310_PS_MAX_VAL /  10
 89};
 90
 91static const int stk3310_scale_table[][2] = {
 92	{6, 400000}, {1, 600000}, {0, 400000}, {0, 100000}
 93};
 94
 95/* Integration time in seconds, microseconds */
 96static const int stk3310_it_table[][2] = {
 97	{0, 185},	{0, 370},	{0, 741},	{0, 1480},
 98	{0, 2960},	{0, 5920},	{0, 11840},	{0, 23680},
 99	{0, 47360},	{0, 94720},	{0, 189440},	{0, 378880},
100	{0, 757760},	{1, 515520},	{3, 31040},	{6, 62080},
101};
102
103struct stk3310_data {
104	struct i2c_client *client;
105	struct mutex lock;
106	bool als_enabled;
107	bool ps_enabled;
108	u64 timestamp;
109	struct regmap *regmap;
110	struct regmap_field *reg_state;
111	struct regmap_field *reg_als_gain;
112	struct regmap_field *reg_ps_gain;
113	struct regmap_field *reg_als_it;
114	struct regmap_field *reg_ps_it;
115	struct regmap_field *reg_int_ps;
116	struct regmap_field *reg_flag_psint;
117	struct regmap_field *reg_flag_nf;
118};
119
120static const struct iio_event_spec stk3310_events[] = {
121	/* Proximity event */
122	{
123		.type = IIO_EV_TYPE_THRESH,
124		.dir = IIO_EV_DIR_RISING,
125		.mask_separate = BIT(IIO_EV_INFO_VALUE) |
126				 BIT(IIO_EV_INFO_ENABLE),
127	},
128	/* Out-of-proximity event */
129	{
130		.type = IIO_EV_TYPE_THRESH,
131		.dir = IIO_EV_DIR_FALLING,
132		.mask_separate = BIT(IIO_EV_INFO_VALUE) |
133				 BIT(IIO_EV_INFO_ENABLE),
134	},
135};
136
137static const struct iio_chan_spec stk3310_channels[] = {
138	{
139		.type = IIO_LIGHT,
140		.info_mask_separate =
141			BIT(IIO_CHAN_INFO_RAW) |
142			BIT(IIO_CHAN_INFO_SCALE) |
143			BIT(IIO_CHAN_INFO_INT_TIME),
144	},
145	{
146		.type = IIO_PROXIMITY,
147		.info_mask_separate =
148			BIT(IIO_CHAN_INFO_RAW) |
149			BIT(IIO_CHAN_INFO_SCALE) |
150			BIT(IIO_CHAN_INFO_INT_TIME),
151		.event_spec = stk3310_events,
152		.num_event_specs = ARRAY_SIZE(stk3310_events),
153	}
154};
155
156static IIO_CONST_ATTR(in_illuminance_scale_available, STK3310_SCALE_AVAILABLE);
157
158static IIO_CONST_ATTR(in_proximity_scale_available, STK3310_SCALE_AVAILABLE);
159
160static IIO_CONST_ATTR(in_illuminance_integration_time_available,
161		      STK3310_IT_AVAILABLE);
162
163static IIO_CONST_ATTR(in_proximity_integration_time_available,
164		      STK3310_IT_AVAILABLE);
165
166static struct attribute *stk3310_attributes[] = {
167	&iio_const_attr_in_illuminance_scale_available.dev_attr.attr,
168	&iio_const_attr_in_proximity_scale_available.dev_attr.attr,
169	&iio_const_attr_in_illuminance_integration_time_available.dev_attr.attr,
170	&iio_const_attr_in_proximity_integration_time_available.dev_attr.attr,
171	NULL,
172};
173
174static const struct attribute_group stk3310_attribute_group = {
175	.attrs = stk3310_attributes
176};
177
178static int stk3310_get_index(const int table[][2], int table_size,
179			     int val, int val2)
180{
181	int i;
182
183	for (i = 0; i < table_size; i++) {
184		if (val == table[i][0] && val2 == table[i][1])
185			return i;
186	}
187
188	return -EINVAL;
189}
190
191static int stk3310_read_event(struct iio_dev *indio_dev,
192			      const struct iio_chan_spec *chan,
193			      enum iio_event_type type,
194			      enum iio_event_direction dir,
195			      enum iio_event_info info,
196			      int *val, int *val2)
197{
198	u8 reg;
199	__be16 buf;
200	int ret;
201	struct stk3310_data *data = iio_priv(indio_dev);
202
203	if (info != IIO_EV_INFO_VALUE)
204		return -EINVAL;
205
206	/* Only proximity interrupts are implemented at the moment. */
207	if (dir == IIO_EV_DIR_RISING)
208		reg = STK3310_REG_THDH_PS;
209	else if (dir == IIO_EV_DIR_FALLING)
210		reg = STK3310_REG_THDL_PS;
211	else
212		return -EINVAL;
213
214	mutex_lock(&data->lock);
215	ret = regmap_bulk_read(data->regmap, reg, &buf, 2);
216	mutex_unlock(&data->lock);
217	if (ret < 0) {
218		dev_err(&data->client->dev, "register read failed\n");
219		return ret;
220	}
221	*val = be16_to_cpu(buf);
222
223	return IIO_VAL_INT;
224}
225
226static int stk3310_write_event(struct iio_dev *indio_dev,
227			       const struct iio_chan_spec *chan,
228			       enum iio_event_type type,
229			       enum iio_event_direction dir,
230			       enum iio_event_info info,
231			       int val, int val2)
232{
233	u8 reg;
234	__be16 buf;
235	int ret;
236	unsigned int index;
237	struct stk3310_data *data = iio_priv(indio_dev);
238	struct i2c_client *client = data->client;
239
240	ret = regmap_field_read(data->reg_ps_gain, &index);
241	if (ret < 0)
242		return ret;
243
244	if (val < 0 || val > stk3310_ps_max[index])
245		return -EINVAL;
246
247	if (dir == IIO_EV_DIR_RISING)
248		reg = STK3310_REG_THDH_PS;
249	else if (dir == IIO_EV_DIR_FALLING)
250		reg = STK3310_REG_THDL_PS;
251	else
252		return -EINVAL;
253
254	buf = cpu_to_be16(val);
255	ret = regmap_bulk_write(data->regmap, reg, &buf, 2);
256	if (ret < 0)
257		dev_err(&client->dev, "failed to set PS threshold!\n");
258
259	return ret;
260}
261
262static int stk3310_read_event_config(struct iio_dev *indio_dev,
263				     const struct iio_chan_spec *chan,
264				     enum iio_event_type type,
265				     enum iio_event_direction dir)
266{
267	unsigned int event_val;
268	int ret;
269	struct stk3310_data *data = iio_priv(indio_dev);
270
271	ret = regmap_field_read(data->reg_int_ps, &event_val);
272	if (ret < 0)
273		return ret;
274
275	return event_val;
276}
277
278static int stk3310_write_event_config(struct iio_dev *indio_dev,
279				      const struct iio_chan_spec *chan,
280				      enum iio_event_type type,
281				      enum iio_event_direction dir,
282				      int state)
283{
284	int ret;
285	struct stk3310_data *data = iio_priv(indio_dev);
286	struct i2c_client *client = data->client;
287
288	if (state < 0 || state > 7)
289		return -EINVAL;
290
291	/* Set INT_PS value */
292	mutex_lock(&data->lock);
293	ret = regmap_field_write(data->reg_int_ps, state);
294	if (ret < 0)
295		dev_err(&client->dev, "failed to set interrupt mode\n");
296	mutex_unlock(&data->lock);
297
298	return ret;
299}
300
301static int stk3310_read_raw(struct iio_dev *indio_dev,
302			    struct iio_chan_spec const *chan,
303			    int *val, int *val2, long mask)
304{
305	u8 reg;
306	__be16 buf;
307	int ret;
308	unsigned int index;
309	struct stk3310_data *data = iio_priv(indio_dev);
310	struct i2c_client *client = data->client;
311
312	if (chan->type != IIO_LIGHT && chan->type != IIO_PROXIMITY)
313		return -EINVAL;
314
315	switch (mask) {
316	case IIO_CHAN_INFO_RAW:
317		if (chan->type == IIO_LIGHT)
318			reg = STK3310_REG_ALS_DATA_MSB;
319		else
320			reg = STK3310_REG_PS_DATA_MSB;
321
322		mutex_lock(&data->lock);
323		ret = regmap_bulk_read(data->regmap, reg, &buf, 2);
324		if (ret < 0) {
325			dev_err(&client->dev, "register read failed\n");
326			mutex_unlock(&data->lock);
327			return ret;
328		}
329		*val = be16_to_cpu(buf);
330		mutex_unlock(&data->lock);
331		return IIO_VAL_INT;
332	case IIO_CHAN_INFO_INT_TIME:
333		if (chan->type == IIO_LIGHT)
334			ret = regmap_field_read(data->reg_als_it, &index);
335		else
336			ret = regmap_field_read(data->reg_ps_it, &index);
337		if (ret < 0)
338			return ret;
339
340		*val = stk3310_it_table[index][0];
341		*val2 = stk3310_it_table[index][1];
342		return IIO_VAL_INT_PLUS_MICRO;
343	case IIO_CHAN_INFO_SCALE:
344		if (chan->type == IIO_LIGHT)
345			ret = regmap_field_read(data->reg_als_gain, &index);
346		else
347			ret = regmap_field_read(data->reg_ps_gain, &index);
348		if (ret < 0)
349			return ret;
350
351		*val = stk3310_scale_table[index][0];
352		*val2 = stk3310_scale_table[index][1];
353		return IIO_VAL_INT_PLUS_MICRO;
354	}
355
356	return -EINVAL;
357}
358
359static int stk3310_write_raw(struct iio_dev *indio_dev,
360			     struct iio_chan_spec const *chan,
361			     int val, int val2, long mask)
362{
363	int ret;
364	int index;
365	struct stk3310_data *data = iio_priv(indio_dev);
366
367	if (chan->type != IIO_LIGHT && chan->type != IIO_PROXIMITY)
368		return -EINVAL;
369
370	switch (mask) {
371	case IIO_CHAN_INFO_INT_TIME:
372		index = stk3310_get_index(stk3310_it_table,
373					  ARRAY_SIZE(stk3310_it_table),
374					  val, val2);
375		if (index < 0)
376			return -EINVAL;
377		mutex_lock(&data->lock);
378		if (chan->type == IIO_LIGHT)
379			ret = regmap_field_write(data->reg_als_it, index);
380		else
381			ret = regmap_field_write(data->reg_ps_it, index);
382		if (ret < 0)
383			dev_err(&data->client->dev,
384				"sensor configuration failed\n");
385		mutex_unlock(&data->lock);
386		return ret;
387
388	case IIO_CHAN_INFO_SCALE:
389		index = stk3310_get_index(stk3310_scale_table,
390					  ARRAY_SIZE(stk3310_scale_table),
391					  val, val2);
392		if (index < 0)
393			return -EINVAL;
394		mutex_lock(&data->lock);
395		if (chan->type == IIO_LIGHT)
396			ret = regmap_field_write(data->reg_als_gain, index);
397		else
398			ret = regmap_field_write(data->reg_ps_gain, index);
399		if (ret < 0)
400			dev_err(&data->client->dev,
401				"sensor configuration failed\n");
402		mutex_unlock(&data->lock);
403		return ret;
404	}
405
406	return -EINVAL;
407}
408
409static const struct iio_info stk3310_info = {
 
410	.read_raw		= stk3310_read_raw,
411	.write_raw		= stk3310_write_raw,
412	.attrs			= &stk3310_attribute_group,
413	.read_event_value	= stk3310_read_event,
414	.write_event_value	= stk3310_write_event,
415	.read_event_config	= stk3310_read_event_config,
416	.write_event_config	= stk3310_write_event_config,
417};
418
419static int stk3310_set_state(struct stk3310_data *data, u8 state)
420{
421	int ret;
422	struct i2c_client *client = data->client;
423
424	/* 3-bit state; 0b100 is not supported. */
425	if (state > 7 || state == 4)
426		return -EINVAL;
427
428	mutex_lock(&data->lock);
429	ret = regmap_field_write(data->reg_state, state);
430	if (ret < 0) {
431		dev_err(&client->dev, "failed to change sensor state\n");
432	} else if (state != STK3310_STATE_STANDBY) {
433		/* Don't reset the 'enabled' flags if we're going in standby */
434		data->ps_enabled  = !!(state & STK3310_STATE_EN_PS);
435		data->als_enabled = !!(state & STK3310_STATE_EN_ALS);
436	}
437	mutex_unlock(&data->lock);
438
439	return ret;
440}
441
442static int stk3310_init(struct iio_dev *indio_dev)
443{
444	int ret;
445	int chipid;
446	u8 state;
447	struct stk3310_data *data = iio_priv(indio_dev);
448	struct i2c_client *client = data->client;
449
450	ret = regmap_read(data->regmap, STK3310_REG_ID, &chipid);
451	if (ret < 0)
452		return ret;
453
454	if (chipid != STK3310_CHIP_ID_VAL &&
455	    chipid != STK3311_CHIP_ID_VAL &&
456	    chipid != STK3335_CHIP_ID_VAL) {
457		dev_err(&client->dev, "invalid chip id: 0x%x\n", chipid);
458		return -ENODEV;
459	}
460
461	state = STK3310_STATE_EN_ALS | STK3310_STATE_EN_PS;
462	ret = stk3310_set_state(data, state);
463	if (ret < 0) {
464		dev_err(&client->dev, "failed to enable sensor");
465		return ret;
466	}
467
468	/* Enable PS interrupts */
469	ret = regmap_field_write(data->reg_int_ps, STK3310_PSINT_EN);
470	if (ret < 0)
471		dev_err(&client->dev, "failed to enable interrupts!\n");
472
473	return ret;
474}
475
476static bool stk3310_is_volatile_reg(struct device *dev, unsigned int reg)
477{
478	switch (reg) {
479	case STK3310_REG_ALS_DATA_MSB:
480	case STK3310_REG_ALS_DATA_LSB:
481	case STK3310_REG_PS_DATA_LSB:
482	case STK3310_REG_PS_DATA_MSB:
483	case STK3310_REG_FLAG:
484		return true;
485	default:
486		return false;
487	}
488}
489
490static struct regmap_config stk3310_regmap_config = {
491	.name = STK3310_REGMAP_NAME,
492	.reg_bits = 8,
493	.val_bits = 8,
494	.max_register = STK3310_MAX_REG,
495	.cache_type = REGCACHE_RBTREE,
496	.volatile_reg = stk3310_is_volatile_reg,
497};
498
499static int stk3310_regmap_init(struct stk3310_data *data)
500{
501	struct regmap *regmap;
502	struct i2c_client *client;
503
504	client = data->client;
505	regmap = devm_regmap_init_i2c(client, &stk3310_regmap_config);
506	if (IS_ERR(regmap)) {
507		dev_err(&client->dev, "regmap initialization failed.\n");
508		return PTR_ERR(regmap);
509	}
510	data->regmap = regmap;
511
512	STK3310_REGFIELD(state);
513	STK3310_REGFIELD(als_gain);
514	STK3310_REGFIELD(ps_gain);
515	STK3310_REGFIELD(als_it);
516	STK3310_REGFIELD(ps_it);
517	STK3310_REGFIELD(int_ps);
518	STK3310_REGFIELD(flag_psint);
519	STK3310_REGFIELD(flag_nf);
520
521	return 0;
522}
523
524static irqreturn_t stk3310_irq_handler(int irq, void *private)
525{
526	struct iio_dev *indio_dev = private;
527	struct stk3310_data *data = iio_priv(indio_dev);
528
529	data->timestamp = iio_get_time_ns(indio_dev);
530
531	return IRQ_WAKE_THREAD;
532}
533
534static irqreturn_t stk3310_irq_event_handler(int irq, void *private)
535{
536	int ret;
537	unsigned int dir;
538	u64 event;
539
540	struct iio_dev *indio_dev = private;
541	struct stk3310_data *data = iio_priv(indio_dev);
542
543	/* Read FLAG_NF to figure out what threshold has been met. */
544	mutex_lock(&data->lock);
545	ret = regmap_field_read(data->reg_flag_nf, &dir);
546	if (ret < 0) {
547		dev_err(&data->client->dev, "register read failed\n");
548		mutex_unlock(&data->lock);
549		return ret;
550	}
551	event = IIO_UNMOD_EVENT_CODE(IIO_PROXIMITY, 1,
552				     IIO_EV_TYPE_THRESH,
553				     (dir ? IIO_EV_DIR_FALLING :
554					    IIO_EV_DIR_RISING));
555	iio_push_event(indio_dev, event, data->timestamp);
556
557	/* Reset the interrupt flag */
558	ret = regmap_field_write(data->reg_flag_psint, 0);
559	if (ret < 0)
560		dev_err(&data->client->dev, "failed to reset interrupts\n");
561	mutex_unlock(&data->lock);
562
563	return IRQ_HANDLED;
564}
565
566static int stk3310_probe(struct i2c_client *client,
567			 const struct i2c_device_id *id)
568{
569	int ret;
570	struct iio_dev *indio_dev;
571	struct stk3310_data *data;
572
573	indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
574	if (!indio_dev) {
575		dev_err(&client->dev, "iio allocation failed!\n");
576		return -ENOMEM;
577	}
578
579	data = iio_priv(indio_dev);
580	data->client = client;
581	i2c_set_clientdata(client, indio_dev);
582	mutex_init(&data->lock);
583
584	ret = stk3310_regmap_init(data);
585	if (ret < 0)
586		return ret;
587
588	indio_dev->dev.parent = &client->dev;
589	indio_dev->info = &stk3310_info;
590	indio_dev->name = STK3310_DRIVER_NAME;
591	indio_dev->modes = INDIO_DIRECT_MODE;
592	indio_dev->channels = stk3310_channels;
593	indio_dev->num_channels = ARRAY_SIZE(stk3310_channels);
594
595	ret = stk3310_init(indio_dev);
596	if (ret < 0)
597		return ret;
598
599	if (client->irq > 0) {
600		ret = devm_request_threaded_irq(&client->dev, client->irq,
601						stk3310_irq_handler,
602						stk3310_irq_event_handler,
603						IRQF_TRIGGER_FALLING |
604						IRQF_ONESHOT,
605						STK3310_EVENT, indio_dev);
606		if (ret < 0) {
607			dev_err(&client->dev, "request irq %d failed\n",
608				client->irq);
609			goto err_standby;
610		}
611	}
612
613	ret = iio_device_register(indio_dev);
614	if (ret < 0) {
615		dev_err(&client->dev, "device_register failed\n");
616		goto err_standby;
617	}
618
619	return 0;
620
621err_standby:
622	stk3310_set_state(data, STK3310_STATE_STANDBY);
623	return ret;
624}
625
626static int stk3310_remove(struct i2c_client *client)
627{
628	struct iio_dev *indio_dev = i2c_get_clientdata(client);
629
630	iio_device_unregister(indio_dev);
631	return stk3310_set_state(iio_priv(indio_dev), STK3310_STATE_STANDBY);
632}
633
634#ifdef CONFIG_PM_SLEEP
635static int stk3310_suspend(struct device *dev)
636{
637	struct stk3310_data *data;
638
639	data = iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
640
641	return stk3310_set_state(data, STK3310_STATE_STANDBY);
642}
643
644static int stk3310_resume(struct device *dev)
645{
646	u8 state = 0;
647	struct stk3310_data *data;
648
649	data = iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
650	if (data->ps_enabled)
651		state |= STK3310_STATE_EN_PS;
652	if (data->als_enabled)
653		state |= STK3310_STATE_EN_ALS;
654
655	return stk3310_set_state(data, state);
656}
657
658static SIMPLE_DEV_PM_OPS(stk3310_pm_ops, stk3310_suspend, stk3310_resume);
659
660#define STK3310_PM_OPS (&stk3310_pm_ops)
661#else
662#define STK3310_PM_OPS NULL
663#endif
664
665static const struct i2c_device_id stk3310_i2c_id[] = {
666	{"STK3310", 0},
667	{"STK3311", 0},
668	{"STK3335", 0},
669	{}
670};
671MODULE_DEVICE_TABLE(i2c, stk3310_i2c_id);
672
673static const struct acpi_device_id stk3310_acpi_id[] = {
674	{"STK3310", 0},
675	{"STK3311", 0},
676	{"STK3335", 0},
677	{}
678};
679
680MODULE_DEVICE_TABLE(acpi, stk3310_acpi_id);
681
682static const struct of_device_id stk3310_of_match[] = {
683	{ .compatible = "sensortek,stk3310", },
684	{ .compatible = "sensortek,stk3311", },
685	{ .compatible = "sensortek,stk3335", },
686	{}
687};
688MODULE_DEVICE_TABLE(of, stk3310_of_match);
689
690static struct i2c_driver stk3310_driver = {
691	.driver = {
692		.name = "stk3310",
693		.of_match_table = stk3310_of_match,
694		.pm = STK3310_PM_OPS,
695		.acpi_match_table = ACPI_PTR(stk3310_acpi_id),
696	},
697	.probe =            stk3310_probe,
698	.remove =           stk3310_remove,
699	.id_table =         stk3310_i2c_id,
700};
701
702module_i2c_driver(stk3310_driver);
703
704MODULE_AUTHOR("Tiberiu Breana <tiberiu.a.breana@intel.com>");
705MODULE_DESCRIPTION("STK3310 Ambient Light and Proximity Sensor driver");
706MODULE_LICENSE("GPL v2");