]> git.proxmox.com Git - mirror_ubuntu-bionic-kernel.git/blame - drivers/iio/light/rpr0521.c
Merge remote-tracking branches 'spi/topic/imx', 'spi/topic/mxs', 'spi/topic/orion...
[mirror_ubuntu-bionic-kernel.git] / drivers / iio / light / rpr0521.c
CommitLineData
efa86e9f
DB
1/*
2 * RPR-0521 ROHM 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 RPR-0521RS (7-bit I2C slave address 0x38).
11 *
e12ffd24 12 * TODO: illuminance channel
efa86e9f
DB
13 */
14
15#include <linux/module.h>
16#include <linux/init.h>
17#include <linux/i2c.h>
18#include <linux/regmap.h>
19#include <linux/delay.h>
20#include <linux/acpi.h>
21
22#include <linux/iio/iio.h>
e12ffd24
MK
23#include <linux/iio/buffer.h>
24#include <linux/iio/trigger.h>
25#include <linux/iio/trigger_consumer.h>
26#include <linux/iio/triggered_buffer.h>
efa86e9f
DB
27#include <linux/iio/sysfs.h>
28#include <linux/pm_runtime.h>
29
30#define RPR0521_REG_SYSTEM_CTRL 0x40
31#define RPR0521_REG_MODE_CTRL 0x41
32#define RPR0521_REG_ALS_CTRL 0x42
33#define RPR0521_REG_PXS_CTRL 0x43
34#define RPR0521_REG_PXS_DATA 0x44 /* 16-bit, little endian */
35#define RPR0521_REG_ALS_DATA0 0x46 /* 16-bit, little endian */
36#define RPR0521_REG_ALS_DATA1 0x48 /* 16-bit, little endian */
e12ffd24 37#define RPR0521_REG_INTERRUPT 0x4A
734c4423 38#define RPR0521_REG_PS_OFFSET_LSB 0x53
efa86e9f
DB
39#define RPR0521_REG_ID 0x92
40
41#define RPR0521_MODE_ALS_MASK BIT(7)
42#define RPR0521_MODE_PXS_MASK BIT(6)
43#define RPR0521_MODE_MEAS_TIME_MASK GENMASK(3, 0)
44#define RPR0521_ALS_DATA0_GAIN_MASK GENMASK(5, 4)
45#define RPR0521_ALS_DATA0_GAIN_SHIFT 4
46#define RPR0521_ALS_DATA1_GAIN_MASK GENMASK(3, 2)
47#define RPR0521_ALS_DATA1_GAIN_SHIFT 2
48#define RPR0521_PXS_GAIN_MASK GENMASK(5, 4)
49#define RPR0521_PXS_GAIN_SHIFT 4
e12ffd24
MK
50#define RPR0521_PXS_PERSISTENCE_MASK GENMASK(3, 0)
51#define RPR0521_INTERRUPT_INT_TRIG_PS_MASK BIT(0)
52#define RPR0521_INTERRUPT_INT_TRIG_ALS_MASK BIT(1)
53#define RPR0521_INTERRUPT_INT_REASSERT_MASK BIT(3)
54#define RPR0521_INTERRUPT_ALS_INT_STATUS_MASK BIT(6)
55#define RPR0521_INTERRUPT_PS_INT_STATUS_MASK BIT(7)
efa86e9f
DB
56
57#define RPR0521_MODE_ALS_ENABLE BIT(7)
58#define RPR0521_MODE_ALS_DISABLE 0x00
59#define RPR0521_MODE_PXS_ENABLE BIT(6)
60#define RPR0521_MODE_PXS_DISABLE 0x00
e12ffd24
MK
61#define RPR0521_PXS_PERSISTENCE_DRDY 0x00
62
63#define RPR0521_INTERRUPT_INT_TRIG_PS_ENABLE BIT(0)
64#define RPR0521_INTERRUPT_INT_TRIG_PS_DISABLE 0x00
65#define RPR0521_INTERRUPT_INT_TRIG_ALS_ENABLE BIT(1)
66#define RPR0521_INTERRUPT_INT_TRIG_ALS_DISABLE 0x00
67#define RPR0521_INTERRUPT_INT_REASSERT_ENABLE BIT(3)
68#define RPR0521_INTERRUPT_INT_REASSERT_DISABLE 0x00
efa86e9f
DB
69
70#define RPR0521_MANUFACT_ID 0xE0
71#define RPR0521_DEFAULT_MEAS_TIME 0x06 /* ALS - 100ms, PXS - 100ms */
72
73#define RPR0521_DRV_NAME "RPR0521"
e12ffd24 74#define RPR0521_IRQ_NAME "rpr0521_event"
efa86e9f
DB
75#define RPR0521_REGMAP_NAME "rpr0521_regmap"
76
77#define RPR0521_SLEEP_DELAY_MS 2000
78
79#define RPR0521_ALS_SCALE_AVAIL "0.007812 0.015625 0.5 1"
80#define RPR0521_PXS_SCALE_AVAIL "0.125 0.5 1"
81
82struct rpr0521_gain {
83 int scale;
84 int uscale;
85};
86
87static const struct rpr0521_gain rpr0521_als_gain[4] = {
88 {1, 0}, /* x1 */
89 {0, 500000}, /* x2 */
90 {0, 15625}, /* x64 */
91 {0, 7812}, /* x128 */
92};
93
94static const struct rpr0521_gain rpr0521_pxs_gain[3] = {
95 {1, 0}, /* x1 */
96 {0, 500000}, /* x2 */
97 {0, 125000}, /* x4 */
98};
99
100enum rpr0521_channel {
9ece370b 101 RPR0521_CHAN_PXS,
efa86e9f
DB
102 RPR0521_CHAN_ALS_DATA0,
103 RPR0521_CHAN_ALS_DATA1,
efa86e9f
DB
104};
105
106struct rpr0521_reg_desc {
107 u8 address;
108 u8 device_mask;
109};
110
111static const struct rpr0521_reg_desc rpr0521_data_reg[] = {
9ece370b
MK
112 [RPR0521_CHAN_PXS] = {
113 .address = RPR0521_REG_PXS_DATA,
114 .device_mask = RPR0521_MODE_PXS_MASK,
115 },
efa86e9f
DB
116 [RPR0521_CHAN_ALS_DATA0] = {
117 .address = RPR0521_REG_ALS_DATA0,
118 .device_mask = RPR0521_MODE_ALS_MASK,
119 },
120 [RPR0521_CHAN_ALS_DATA1] = {
121 .address = RPR0521_REG_ALS_DATA1,
122 .device_mask = RPR0521_MODE_ALS_MASK,
123 },
efa86e9f
DB
124};
125
126static const struct rpr0521_gain_info {
127 u8 reg;
128 u8 mask;
129 u8 shift;
130 const struct rpr0521_gain *gain;
131 int size;
132} rpr0521_gain[] = {
9ece370b
MK
133 [RPR0521_CHAN_PXS] = {
134 .reg = RPR0521_REG_PXS_CTRL,
135 .mask = RPR0521_PXS_GAIN_MASK,
136 .shift = RPR0521_PXS_GAIN_SHIFT,
137 .gain = rpr0521_pxs_gain,
138 .size = ARRAY_SIZE(rpr0521_pxs_gain),
139 },
efa86e9f
DB
140 [RPR0521_CHAN_ALS_DATA0] = {
141 .reg = RPR0521_REG_ALS_CTRL,
142 .mask = RPR0521_ALS_DATA0_GAIN_MASK,
143 .shift = RPR0521_ALS_DATA0_GAIN_SHIFT,
144 .gain = rpr0521_als_gain,
145 .size = ARRAY_SIZE(rpr0521_als_gain),
146 },
147 [RPR0521_CHAN_ALS_DATA1] = {
148 .reg = RPR0521_REG_ALS_CTRL,
149 .mask = RPR0521_ALS_DATA1_GAIN_MASK,
150 .shift = RPR0521_ALS_DATA1_GAIN_SHIFT,
151 .gain = rpr0521_als_gain,
152 .size = ARRAY_SIZE(rpr0521_als_gain),
153 },
efa86e9f
DB
154};
155
c91a88ef
MK
156struct rpr0521_samp_freq {
157 int als_hz;
158 int als_uhz;
159 int pxs_hz;
160 int pxs_uhz;
161};
162
163static const struct rpr0521_samp_freq rpr0521_samp_freq_i[13] = {
164/* {ALS, PXS}, W==currently writable option */
165 {0, 0, 0, 0}, /* W0000, 0=standby */
166 {0, 0, 100, 0}, /* 0001 */
167 {0, 0, 25, 0}, /* 0010 */
168 {0, 0, 10, 0}, /* 0011 */
169 {0, 0, 2, 500000}, /* 0100 */
170 {10, 0, 20, 0}, /* 0101 */
171 {10, 0, 10, 0}, /* W0110 */
172 {10, 0, 2, 500000}, /* 0111 */
173 {2, 500000, 20, 0}, /* 1000, measurement 100ms, sleep 300ms */
174 {2, 500000, 10, 0}, /* 1001, measurement 100ms, sleep 300ms */
175 {2, 500000, 0, 0}, /* 1010, high sensitivity mode */
176 {2, 500000, 2, 500000}, /* W1011, high sensitivity mode */
177 {20, 0, 20, 0} /* 1100, ALS_data x 0.5, see specification P.18 */
178};
179
efa86e9f
DB
180struct rpr0521_data {
181 struct i2c_client *client;
182
183 /* protect device params updates (e.g state, gain) */
184 struct mutex lock;
185
186 /* device active status */
187 bool als_dev_en;
188 bool pxs_dev_en;
189
e12ffd24
MK
190 struct iio_trigger *drdy_trigger0;
191 s64 irq_timestamp;
192
484c314b 193 /* optimize runtime pm ops - enable/disable device only if needed */
efa86e9f
DB
194 bool als_ps_need_en;
195 bool pxs_ps_need_en;
484c314b
MK
196 bool als_need_dis;
197 bool pxs_need_dis;
efa86e9f
DB
198
199 struct regmap *regmap;
200};
201
202static IIO_CONST_ATTR(in_intensity_scale_available, RPR0521_ALS_SCALE_AVAIL);
203static IIO_CONST_ATTR(in_proximity_scale_available, RPR0521_PXS_SCALE_AVAIL);
204
c91a88ef
MK
205/*
206 * Start with easy freq first, whole table of freq combinations is more
207 * complicated.
208 */
209static IIO_CONST_ATTR_SAMP_FREQ_AVAIL("2.5 10");
210
efa86e9f
DB
211static struct attribute *rpr0521_attributes[] = {
212 &iio_const_attr_in_intensity_scale_available.dev_attr.attr,
213 &iio_const_attr_in_proximity_scale_available.dev_attr.attr,
c91a88ef 214 &iio_const_attr_sampling_frequency_available.dev_attr.attr,
efa86e9f
DB
215 NULL,
216};
217
218static const struct attribute_group rpr0521_attribute_group = {
219 .attrs = rpr0521_attributes,
220};
221
e12ffd24
MK
222/* Order of the channel data in buffer */
223enum rpr0521_scan_index_order {
224 RPR0521_CHAN_INDEX_PXS,
225 RPR0521_CHAN_INDEX_BOTH,
226 RPR0521_CHAN_INDEX_IR,
227};
228
229static const unsigned long rpr0521_available_scan_masks[] = {
230 BIT(RPR0521_CHAN_INDEX_PXS) | BIT(RPR0521_CHAN_INDEX_BOTH) |
231 BIT(RPR0521_CHAN_INDEX_IR),
232 0
233};
234
efa86e9f 235static const struct iio_chan_spec rpr0521_channels[] = {
9ece370b
MK
236 {
237 .type = IIO_PROXIMITY,
238 .address = RPR0521_CHAN_PXS,
239 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
240 BIT(IIO_CHAN_INFO_OFFSET) |
241 BIT(IIO_CHAN_INFO_SCALE),
242 .info_mask_shared_by_all = BIT(IIO_CHAN_INFO_SAMP_FREQ),
e12ffd24
MK
243 .scan_index = RPR0521_CHAN_INDEX_PXS,
244 .scan_type = {
245 .sign = 'u',
246 .realbits = 16,
247 .storagebits = 16,
248 .endianness = IIO_LE,
249 },
9ece370b 250 },
efa86e9f
DB
251 {
252 .type = IIO_INTENSITY,
253 .modified = 1,
254 .address = RPR0521_CHAN_ALS_DATA0,
255 .channel2 = IIO_MOD_LIGHT_BOTH,
256 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
257 BIT(IIO_CHAN_INFO_SCALE),
c91a88ef 258 .info_mask_shared_by_all = BIT(IIO_CHAN_INFO_SAMP_FREQ),
e12ffd24
MK
259 .scan_index = RPR0521_CHAN_INDEX_BOTH,
260 .scan_type = {
261 .sign = 'u',
262 .realbits = 16,
263 .storagebits = 16,
264 .endianness = IIO_LE,
265 },
efa86e9f
DB
266 },
267 {
268 .type = IIO_INTENSITY,
269 .modified = 1,
270 .address = RPR0521_CHAN_ALS_DATA1,
271 .channel2 = IIO_MOD_LIGHT_IR,
272 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
273 BIT(IIO_CHAN_INFO_SCALE),
c91a88ef 274 .info_mask_shared_by_all = BIT(IIO_CHAN_INFO_SAMP_FREQ),
e12ffd24
MK
275 .scan_index = RPR0521_CHAN_INDEX_IR,
276 .scan_type = {
277 .sign = 'u',
278 .realbits = 16,
279 .storagebits = 16,
280 .endianness = IIO_LE,
281 },
efa86e9f 282 },
efa86e9f
DB
283};
284
285static int rpr0521_als_enable(struct rpr0521_data *data, u8 status)
286{
287 int ret;
288
289 ret = regmap_update_bits(data->regmap, RPR0521_REG_MODE_CTRL,
290 RPR0521_MODE_ALS_MASK,
291 status);
292 if (ret < 0)
293 return ret;
294
f87fa261
MK
295 if (status & RPR0521_MODE_ALS_MASK)
296 data->als_dev_en = true;
297 else
298 data->als_dev_en = false;
efa86e9f
DB
299
300 return 0;
301}
302
303static int rpr0521_pxs_enable(struct rpr0521_data *data, u8 status)
304{
305 int ret;
306
307 ret = regmap_update_bits(data->regmap, RPR0521_REG_MODE_CTRL,
308 RPR0521_MODE_PXS_MASK,
309 status);
310 if (ret < 0)
311 return ret;
312
f87fa261
MK
313 if (status & RPR0521_MODE_PXS_MASK)
314 data->pxs_dev_en = true;
315 else
316 data->pxs_dev_en = false;
efa86e9f
DB
317
318 return 0;
319}
320
321/**
322 * rpr0521_set_power_state - handles runtime PM state and sensors enabled status
323 *
324 * @data: rpr0521 device private data
325 * @on: state to be set for devices in @device_mask
326 * @device_mask: bitmask specifying for which device we need to update @on state
327 *
484c314b
MK
328 * Calls for this function must be balanced so that each ON should have matching
329 * OFF. Otherwise pm usage_count gets out of sync.
efa86e9f
DB
330 */
331static int rpr0521_set_power_state(struct rpr0521_data *data, bool on,
332 u8 device_mask)
333{
334#ifdef CONFIG_PM
335 int ret;
efa86e9f
DB
336
337 if (device_mask & RPR0521_MODE_ALS_MASK) {
484c314b
MK
338 data->als_ps_need_en = on;
339 data->als_need_dis = !on;
efa86e9f
DB
340 }
341
342 if (device_mask & RPR0521_MODE_PXS_MASK) {
484c314b
MK
343 data->pxs_ps_need_en = on;
344 data->pxs_need_dis = !on;
efa86e9f
DB
345 }
346
484c314b
MK
347 /*
348 * On: _resume() is called only when we are suspended
349 * Off: _suspend() is called after delay if _resume() is not
350 * called before that.
351 * Note: If either measurement is re-enabled before _suspend(),
352 * both stay enabled until _suspend().
353 */
efa86e9f
DB
354 if (on) {
355 ret = pm_runtime_get_sync(&data->client->dev);
356 } else {
357 pm_runtime_mark_last_busy(&data->client->dev);
358 ret = pm_runtime_put_autosuspend(&data->client->dev);
359 }
360 if (ret < 0) {
361 dev_err(&data->client->dev,
362 "Failed: rpr0521_set_power_state for %d, ret %d\n",
363 on, ret);
364 if (on)
365 pm_runtime_put_noidle(&data->client->dev);
366
367 return ret;
368 }
484c314b
MK
369
370 if (on) {
371 /* If _resume() was not called, enable measurement now. */
372 if (data->als_ps_need_en) {
373 ret = rpr0521_als_enable(data, RPR0521_MODE_ALS_ENABLE);
374 if (ret)
375 return ret;
376 data->als_ps_need_en = false;
377 }
378
379 if (data->pxs_ps_need_en) {
380 ret = rpr0521_pxs_enable(data, RPR0521_MODE_PXS_ENABLE);
381 if (ret)
382 return ret;
383 data->pxs_ps_need_en = false;
384 }
385 }
efa86e9f
DB
386#endif
387 return 0;
388}
389
e12ffd24
MK
390/* Interrupt register tells if this sensor caused the interrupt or not. */
391static inline bool rpr0521_is_triggered(struct rpr0521_data *data)
392{
393 int ret;
394 int reg;
395
396 ret = regmap_read(data->regmap, RPR0521_REG_INTERRUPT, &reg);
397 if (ret < 0)
398 return false; /* Reg read failed. */
399 if (reg &
400 (RPR0521_INTERRUPT_ALS_INT_STATUS_MASK |
401 RPR0521_INTERRUPT_PS_INT_STATUS_MASK))
402 return true;
403 else
404 return false; /* Int not from this sensor. */
405}
406
407/* IRQ to trigger handler */
408static irqreturn_t rpr0521_drdy_irq_handler(int irq, void *private)
409{
410 struct iio_dev *indio_dev = private;
411 struct rpr0521_data *data = iio_priv(indio_dev);
412
413 data->irq_timestamp = iio_get_time_ns(indio_dev);
414 /*
415 * We need to wake the thread to read the interrupt reg. It
416 * is not possible to do that here because regmap_read takes a
417 * mutex.
418 */
419
420 return IRQ_WAKE_THREAD;
421}
422
423static irqreturn_t rpr0521_drdy_irq_thread(int irq, void *private)
424{
425 struct iio_dev *indio_dev = private;
426 struct rpr0521_data *data = iio_priv(indio_dev);
427
428 if (rpr0521_is_triggered(data)) {
429 iio_trigger_poll_chained(data->drdy_trigger0);
430 return IRQ_HANDLED;
431 }
432
433 return IRQ_NONE;
434}
435
436static irqreturn_t rpr0521_trigger_consumer_store_time(int irq, void *p)
437{
438 struct iio_poll_func *pf = p;
439 struct iio_dev *indio_dev = pf->indio_dev;
440
441 /* Other trigger polls store time here. */
442 if (!iio_trigger_using_own(indio_dev))
443 pf->timestamp = iio_get_time_ns(indio_dev);
444
445 return IRQ_WAKE_THREAD;
446}
447
448static irqreturn_t rpr0521_trigger_consumer_handler(int irq, void *p)
449{
450 struct iio_poll_func *pf = p;
451 struct iio_dev *indio_dev = pf->indio_dev;
452 struct rpr0521_data *data = iio_priv(indio_dev);
453 int err;
454
455 u8 buffer[16]; /* 3 16-bit channels + padding + ts */
456
457 /* Use irq timestamp when reasonable. */
458 if (iio_trigger_using_own(indio_dev) && data->irq_timestamp) {
459 pf->timestamp = data->irq_timestamp;
460 data->irq_timestamp = 0;
461 }
462 /* Other chained trigger polls get timestamp only here. */
463 if (!pf->timestamp)
464 pf->timestamp = iio_get_time_ns(indio_dev);
465
466 err = regmap_bulk_read(data->regmap, RPR0521_REG_PXS_DATA,
467 &buffer,
468 (3 * 2) + 1); /* 3 * 16-bit + (discarded) int clear reg. */
469 if (!err)
470 iio_push_to_buffers_with_timestamp(indio_dev,
471 buffer, pf->timestamp);
472 else
473 dev_err(&data->client->dev,
474 "Trigger consumer can't read from sensor.\n");
475 pf->timestamp = 0;
476
477 iio_trigger_notify_done(indio_dev->trig);
478
479 return IRQ_HANDLED;
480}
481
482static int rpr0521_write_int_enable(struct rpr0521_data *data)
483{
484 int err;
485
486 /* Interrupt after each measurement */
487 err = regmap_update_bits(data->regmap, RPR0521_REG_PXS_CTRL,
488 RPR0521_PXS_PERSISTENCE_MASK,
489 RPR0521_PXS_PERSISTENCE_DRDY);
490 if (err) {
491 dev_err(&data->client->dev, "PS control reg write fail.\n");
492 return -EBUSY;
493 }
494
495 /* Ignore latch and mode because of drdy */
496 err = regmap_write(data->regmap, RPR0521_REG_INTERRUPT,
497 RPR0521_INTERRUPT_INT_REASSERT_DISABLE |
498 RPR0521_INTERRUPT_INT_TRIG_ALS_DISABLE |
499 RPR0521_INTERRUPT_INT_TRIG_PS_ENABLE
500 );
501 if (err) {
502 dev_err(&data->client->dev, "Interrupt setup write fail.\n");
503 return -EBUSY;
504 }
505
506 return 0;
507}
508
509static int rpr0521_write_int_disable(struct rpr0521_data *data)
510{
511 /* Don't care of clearing mode, assert and latch. */
512 return regmap_write(data->regmap, RPR0521_REG_INTERRUPT,
513 RPR0521_INTERRUPT_INT_TRIG_ALS_DISABLE |
514 RPR0521_INTERRUPT_INT_TRIG_PS_DISABLE
515 );
516}
517
518/*
519 * Trigger producer enable / disable. Note that there will be trigs only when
520 * measurement data is ready to be read.
521 */
522static int rpr0521_pxs_drdy_set_state(struct iio_trigger *trigger,
523 bool enable_drdy)
524{
525 struct iio_dev *indio_dev = iio_trigger_get_drvdata(trigger);
526 struct rpr0521_data *data = iio_priv(indio_dev);
527 int err;
528
529 if (enable_drdy)
530 err = rpr0521_write_int_enable(data);
531 else
532 err = rpr0521_write_int_disable(data);
533 if (err)
534 dev_err(&data->client->dev, "rpr0521_pxs_drdy_set_state failed\n");
535
536 return err;
537}
538
539static const struct iio_trigger_ops rpr0521_trigger_ops = {
540 .set_trigger_state = rpr0521_pxs_drdy_set_state,
541 .owner = THIS_MODULE,
542 };
543
544
545static int rpr0521_buffer_preenable(struct iio_dev *indio_dev)
546{
547 int err;
548 struct rpr0521_data *data = iio_priv(indio_dev);
549
550 mutex_lock(&data->lock);
551 err = rpr0521_set_power_state(data, true,
552 (RPR0521_MODE_PXS_MASK | RPR0521_MODE_ALS_MASK));
553 mutex_unlock(&data->lock);
554 if (err)
555 dev_err(&data->client->dev, "_buffer_preenable fail\n");
556
557 return err;
558}
559
560static int rpr0521_buffer_postdisable(struct iio_dev *indio_dev)
561{
562 int err;
563 struct rpr0521_data *data = iio_priv(indio_dev);
564
565 mutex_lock(&data->lock);
566 err = rpr0521_set_power_state(data, false,
567 (RPR0521_MODE_PXS_MASK | RPR0521_MODE_ALS_MASK));
568 mutex_unlock(&data->lock);
569 if (err)
570 dev_err(&data->client->dev, "_buffer_postdisable fail\n");
571
572 return err;
573}
574
575static const struct iio_buffer_setup_ops rpr0521_buffer_setup_ops = {
576 .preenable = rpr0521_buffer_preenable,
577 .postenable = iio_triggered_buffer_postenable,
578 .predisable = iio_triggered_buffer_predisable,
579 .postdisable = rpr0521_buffer_postdisable,
580};
581
efa86e9f
DB
582static int rpr0521_get_gain(struct rpr0521_data *data, int chan,
583 int *val, int *val2)
584{
585 int ret, reg, idx;
586
587 ret = regmap_read(data->regmap, rpr0521_gain[chan].reg, &reg);
588 if (ret < 0)
589 return ret;
590
591 idx = (rpr0521_gain[chan].mask & reg) >> rpr0521_gain[chan].shift;
592 *val = rpr0521_gain[chan].gain[idx].scale;
593 *val2 = rpr0521_gain[chan].gain[idx].uscale;
594
595 return 0;
596}
597
598static int rpr0521_set_gain(struct rpr0521_data *data, int chan,
599 int val, int val2)
600{
601 int i, idx = -EINVAL;
602
603 /* get gain index */
604 for (i = 0; i < rpr0521_gain[chan].size; i++)
605 if (val == rpr0521_gain[chan].gain[i].scale &&
606 val2 == rpr0521_gain[chan].gain[i].uscale) {
607 idx = i;
608 break;
609 }
610
611 if (idx < 0)
612 return idx;
613
614 return regmap_update_bits(data->regmap, rpr0521_gain[chan].reg,
615 rpr0521_gain[chan].mask,
616 idx << rpr0521_gain[chan].shift);
617}
618
c91a88ef
MK
619static int rpr0521_read_samp_freq(struct rpr0521_data *data,
620 enum iio_chan_type chan_type,
621 int *val, int *val2)
622{
623 int reg, ret;
624
625 ret = regmap_read(data->regmap, RPR0521_REG_MODE_CTRL, &reg);
626 if (ret < 0)
627 return ret;
628
629 reg &= RPR0521_MODE_MEAS_TIME_MASK;
630 if (reg >= ARRAY_SIZE(rpr0521_samp_freq_i))
631 return -EINVAL;
632
633 switch (chan_type) {
634 case IIO_INTENSITY:
635 *val = rpr0521_samp_freq_i[reg].als_hz;
636 *val2 = rpr0521_samp_freq_i[reg].als_uhz;
637 return 0;
638
639 case IIO_PROXIMITY:
640 *val = rpr0521_samp_freq_i[reg].pxs_hz;
641 *val2 = rpr0521_samp_freq_i[reg].pxs_uhz;
642 return 0;
643
644 default:
645 return -EINVAL;
646 }
647}
648
649static int rpr0521_write_samp_freq_common(struct rpr0521_data *data,
650 enum iio_chan_type chan_type,
651 int val, int val2)
652{
653 int i;
654
655 /*
656 * Ignore channel
657 * both pxs and als are setup only to same freq because of simplicity
658 */
659 switch (val) {
660 case 0:
661 i = 0;
662 break;
663
664 case 2:
665 if (val2 != 500000)
666 return -EINVAL;
667
668 i = 11;
669 break;
670
671 case 10:
672 i = 6;
673 break;
674
675 default:
676 return -EINVAL;
677 }
678
679 return regmap_update_bits(data->regmap,
680 RPR0521_REG_MODE_CTRL,
681 RPR0521_MODE_MEAS_TIME_MASK,
682 i);
683}
684
734c4423
MK
685static int rpr0521_read_ps_offset(struct rpr0521_data *data, int *offset)
686{
687 int ret;
688 __le16 buffer;
689
690 ret = regmap_bulk_read(data->regmap,
691 RPR0521_REG_PS_OFFSET_LSB, &buffer, sizeof(buffer));
692
693 if (ret < 0) {
694 dev_err(&data->client->dev, "Failed to read PS OFFSET register\n");
695 return ret;
696 }
697 *offset = le16_to_cpu(buffer);
698
699 return ret;
700}
701
702static int rpr0521_write_ps_offset(struct rpr0521_data *data, int offset)
703{
704 int ret;
705 __le16 buffer;
706
707 buffer = cpu_to_le16(offset & 0x3ff);
708 ret = regmap_raw_write(data->regmap,
709 RPR0521_REG_PS_OFFSET_LSB, &buffer, sizeof(buffer));
710
711 if (ret < 0) {
712 dev_err(&data->client->dev, "Failed to write PS OFFSET register\n");
713 return ret;
714 }
715
716 return ret;
717}
718
efa86e9f
DB
719static int rpr0521_read_raw(struct iio_dev *indio_dev,
720 struct iio_chan_spec const *chan, int *val,
721 int *val2, long mask)
722{
723 struct rpr0521_data *data = iio_priv(indio_dev);
724 int ret;
e12ffd24 725 int busy;
efa86e9f
DB
726 u8 device_mask;
727 __le16 raw_data;
728
729 switch (mask) {
730 case IIO_CHAN_INFO_RAW:
731 if (chan->type != IIO_INTENSITY && chan->type != IIO_PROXIMITY)
732 return -EINVAL;
733
e12ffd24
MK
734 busy = iio_device_claim_direct_mode(indio_dev);
735 if (busy)
736 return -EBUSY;
737
efa86e9f
DB
738 device_mask = rpr0521_data_reg[chan->address].device_mask;
739
740 mutex_lock(&data->lock);
741 ret = rpr0521_set_power_state(data, true, device_mask);
e12ffd24
MK
742 if (ret < 0)
743 goto rpr0521_read_raw_out;
efa86e9f
DB
744
745 ret = regmap_bulk_read(data->regmap,
746 rpr0521_data_reg[chan->address].address,
ae591650 747 &raw_data, sizeof(raw_data));
efa86e9f
DB
748 if (ret < 0) {
749 rpr0521_set_power_state(data, false, device_mask);
e12ffd24 750 goto rpr0521_read_raw_out;
efa86e9f
DB
751 }
752
753 ret = rpr0521_set_power_state(data, false, device_mask);
e12ffd24
MK
754
755rpr0521_read_raw_out:
efa86e9f 756 mutex_unlock(&data->lock);
e12ffd24 757 iio_device_release_direct_mode(indio_dev);
efa86e9f
DB
758 if (ret < 0)
759 return ret;
760
761 *val = le16_to_cpu(raw_data);
762
763 return IIO_VAL_INT;
624c389e 764
efa86e9f
DB
765 case IIO_CHAN_INFO_SCALE:
766 mutex_lock(&data->lock);
767 ret = rpr0521_get_gain(data, chan->address, val, val2);
768 mutex_unlock(&data->lock);
769 if (ret < 0)
770 return ret;
771
772 return IIO_VAL_INT_PLUS_MICRO;
624c389e 773
c91a88ef
MK
774 case IIO_CHAN_INFO_SAMP_FREQ:
775 mutex_lock(&data->lock);
776 ret = rpr0521_read_samp_freq(data, chan->type, val, val2);
777 mutex_unlock(&data->lock);
778 if (ret < 0)
779 return ret;
780
781 return IIO_VAL_INT_PLUS_MICRO;
782
734c4423
MK
783 case IIO_CHAN_INFO_OFFSET:
784 mutex_lock(&data->lock);
785 ret = rpr0521_read_ps_offset(data, val);
786 mutex_unlock(&data->lock);
787 if (ret < 0)
788 return ret;
789
790 return IIO_VAL_INT;
791
efa86e9f
DB
792 default:
793 return -EINVAL;
794 }
795}
796
797static int rpr0521_write_raw(struct iio_dev *indio_dev,
798 struct iio_chan_spec const *chan, int val,
799 int val2, long mask)
800{
801 struct rpr0521_data *data = iio_priv(indio_dev);
802 int ret;
803
804 switch (mask) {
805 case IIO_CHAN_INFO_SCALE:
806 mutex_lock(&data->lock);
807 ret = rpr0521_set_gain(data, chan->address, val, val2);
808 mutex_unlock(&data->lock);
809
810 return ret;
624c389e 811
c91a88ef
MK
812 case IIO_CHAN_INFO_SAMP_FREQ:
813 mutex_lock(&data->lock);
814 ret = rpr0521_write_samp_freq_common(data, chan->type,
815 val, val2);
816 mutex_unlock(&data->lock);
817
818 return ret;
819
734c4423
MK
820 case IIO_CHAN_INFO_OFFSET:
821 mutex_lock(&data->lock);
822 ret = rpr0521_write_ps_offset(data, val);
823 mutex_unlock(&data->lock);
824
825 return ret;
826
efa86e9f
DB
827 default:
828 return -EINVAL;
829 }
830}
831
832static const struct iio_info rpr0521_info = {
833 .driver_module = THIS_MODULE,
834 .read_raw = rpr0521_read_raw,
835 .write_raw = rpr0521_write_raw,
836 .attrs = &rpr0521_attribute_group,
837};
838
839static int rpr0521_init(struct rpr0521_data *data)
840{
841 int ret;
842 int id;
843
844 ret = regmap_read(data->regmap, RPR0521_REG_ID, &id);
845 if (ret < 0) {
846 dev_err(&data->client->dev, "Failed to read REG_ID register\n");
847 return ret;
848 }
849
850 if (id != RPR0521_MANUFACT_ID) {
851 dev_err(&data->client->dev, "Wrong id, got %x, expected %x\n",
852 id, RPR0521_MANUFACT_ID);
853 return -ENODEV;
854 }
855
856 /* set default measurement time - 100 ms for both ALS and PS */
857 ret = regmap_update_bits(data->regmap, RPR0521_REG_MODE_CTRL,
858 RPR0521_MODE_MEAS_TIME_MASK,
859 RPR0521_DEFAULT_MEAS_TIME);
860 if (ret) {
861 pr_err("regmap_update_bits returned %d\n", ret);
862 return ret;
863 }
864
484c314b 865#ifndef CONFIG_PM
efa86e9f
DB
866 ret = rpr0521_als_enable(data, RPR0521_MODE_ALS_ENABLE);
867 if (ret < 0)
868 return ret;
869 ret = rpr0521_pxs_enable(data, RPR0521_MODE_PXS_ENABLE);
870 if (ret < 0)
871 return ret;
484c314b 872#endif
efa86e9f 873
e12ffd24
MK
874 data->irq_timestamp = 0;
875
efa86e9f
DB
876 return 0;
877}
878
879static int rpr0521_poweroff(struct rpr0521_data *data)
880{
881 int ret;
e12ffd24 882 int tmp;
efa86e9f
DB
883
884 ret = regmap_update_bits(data->regmap, RPR0521_REG_MODE_CTRL,
885 RPR0521_MODE_ALS_MASK |
886 RPR0521_MODE_PXS_MASK,
887 RPR0521_MODE_ALS_DISABLE |
888 RPR0521_MODE_PXS_DISABLE);
889 if (ret < 0)
890 return ret;
891
892 data->als_dev_en = false;
893 data->pxs_dev_en = false;
894
e12ffd24
MK
895 /*
896 * Int pin keeps state after power off. Set pin to high impedance
897 * mode to prevent power drain.
898 */
899 ret = regmap_read(data->regmap, RPR0521_REG_INTERRUPT, &tmp);
900 if (ret) {
901 dev_err(&data->client->dev, "Failed to reset int pin.\n");
902 return ret;
903 }
904
efa86e9f
DB
905 return 0;
906}
907
908static bool rpr0521_is_volatile_reg(struct device *dev, unsigned int reg)
909{
910 switch (reg) {
911 case RPR0521_REG_MODE_CTRL:
912 case RPR0521_REG_ALS_CTRL:
913 case RPR0521_REG_PXS_CTRL:
914 return false;
915 default:
916 return true;
917 }
918}
919
920static const struct regmap_config rpr0521_regmap_config = {
921 .name = RPR0521_REGMAP_NAME,
922
923 .reg_bits = 8,
924 .val_bits = 8,
925
926 .max_register = RPR0521_REG_ID,
927 .cache_type = REGCACHE_RBTREE,
928 .volatile_reg = rpr0521_is_volatile_reg,
929};
930
931static int rpr0521_probe(struct i2c_client *client,
932 const struct i2c_device_id *id)
933{
934 struct rpr0521_data *data;
935 struct iio_dev *indio_dev;
936 struct regmap *regmap;
937 int ret;
938
939 indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
940 if (!indio_dev)
941 return -ENOMEM;
942
943 regmap = devm_regmap_init_i2c(client, &rpr0521_regmap_config);
944 if (IS_ERR(regmap)) {
945 dev_err(&client->dev, "regmap_init failed!\n");
946 return PTR_ERR(regmap);
947 }
948
949 data = iio_priv(indio_dev);
950 i2c_set_clientdata(client, indio_dev);
951 data->client = client;
952 data->regmap = regmap;
953
954 mutex_init(&data->lock);
955
956 indio_dev->dev.parent = &client->dev;
957 indio_dev->info = &rpr0521_info;
958 indio_dev->name = RPR0521_DRV_NAME;
959 indio_dev->channels = rpr0521_channels;
960 indio_dev->num_channels = ARRAY_SIZE(rpr0521_channels);
961 indio_dev->modes = INDIO_DIRECT_MODE;
962
963 ret = rpr0521_init(data);
964 if (ret < 0) {
965 dev_err(&client->dev, "rpr0521 chip init failed\n");
966 return ret;
967 }
efa86e9f
DB
968
969 ret = pm_runtime_set_active(&client->dev);
970 if (ret < 0)
12d74949 971 goto err_poweroff;
efa86e9f
DB
972
973 pm_runtime_enable(&client->dev);
974 pm_runtime_set_autosuspend_delay(&client->dev, RPR0521_SLEEP_DELAY_MS);
975 pm_runtime_use_autosuspend(&client->dev);
976
e12ffd24
MK
977 /*
978 * If sensor write/read is needed in _probe after _use_autosuspend,
979 * sensor needs to be _resumed first using rpr0521_set_power_state().
980 */
981
982 /* IRQ to trigger setup */
983 if (client->irq) {
984 /* Trigger0 producer setup */
985 data->drdy_trigger0 = devm_iio_trigger_alloc(
986 indio_dev->dev.parent,
987 "%s-dev%d", indio_dev->name, indio_dev->id);
988 if (!data->drdy_trigger0) {
989 ret = -ENOMEM;
990 goto err_pm_disable;
991 }
992 data->drdy_trigger0->dev.parent = indio_dev->dev.parent;
993 data->drdy_trigger0->ops = &rpr0521_trigger_ops;
994 indio_dev->available_scan_masks = rpr0521_available_scan_masks;
995 iio_trigger_set_drvdata(data->drdy_trigger0, indio_dev);
996
997 /* Ties irq to trigger producer handler. */
998 ret = devm_request_threaded_irq(&client->dev, client->irq,
999 rpr0521_drdy_irq_handler, rpr0521_drdy_irq_thread,
1000 IRQF_TRIGGER_FALLING | IRQF_ONESHOT,
1001 RPR0521_IRQ_NAME, indio_dev);
1002 if (ret < 0) {
1003 dev_err(&client->dev, "request irq %d for trigger0 failed\n",
1004 client->irq);
1005 goto err_pm_disable;
1006 }
1007
1008 ret = devm_iio_trigger_register(indio_dev->dev.parent,
1009 data->drdy_trigger0);
1010 if (ret) {
1011 dev_err(&client->dev, "iio trigger register failed\n");
1012 goto err_pm_disable;
1013 }
1014
1015 /*
1016 * Now whole pipe from physical interrupt (irq defined by
1017 * devicetree to device) to trigger0 output is set up.
1018 */
1019
1020 /* Trigger consumer setup */
1021 ret = devm_iio_triggered_buffer_setup(indio_dev->dev.parent,
1022 indio_dev,
1023 rpr0521_trigger_consumer_store_time,
1024 rpr0521_trigger_consumer_handler,
1025 &rpr0521_buffer_setup_ops);
1026 if (ret < 0) {
1027 dev_err(&client->dev, "iio triggered buffer setup failed\n");
1028 goto err_pm_disable;
1029 }
1030 }
1031
12d74949
MK
1032 ret = iio_device_register(indio_dev);
1033 if (ret)
1034 goto err_pm_disable;
1035
1036 return 0;
1037
1038err_pm_disable:
1039 pm_runtime_disable(&client->dev);
1040 pm_runtime_set_suspended(&client->dev);
1041 pm_runtime_put_noidle(&client->dev);
1042err_poweroff:
1043 rpr0521_poweroff(data);
1044
1045 return ret;
efa86e9f
DB
1046}
1047
1048static int rpr0521_remove(struct i2c_client *client)
1049{
1050 struct iio_dev *indio_dev = i2c_get_clientdata(client);
1051
7d0ead5c
AR
1052 iio_device_unregister(indio_dev);
1053
efa86e9f
DB
1054 pm_runtime_disable(&client->dev);
1055 pm_runtime_set_suspended(&client->dev);
1056 pm_runtime_put_noidle(&client->dev);
1057
efa86e9f
DB
1058 rpr0521_poweroff(iio_priv(indio_dev));
1059
1060 return 0;
1061}
1062
1063#ifdef CONFIG_PM
1064static int rpr0521_runtime_suspend(struct device *dev)
1065{
1066 struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
1067 struct rpr0521_data *data = iio_priv(indio_dev);
1068 int ret;
1069
efa86e9f 1070 mutex_lock(&data->lock);
484c314b
MK
1071 /* If measurements are enabled, enable them on resume */
1072 if (!data->als_need_dis)
1073 data->als_ps_need_en = data->als_dev_en;
1074 if (!data->pxs_need_dis)
1075 data->pxs_ps_need_en = data->pxs_dev_en;
1076
1077 /* disable channels and sets {als,pxs}_dev_en to false */
efa86e9f 1078 ret = rpr0521_poweroff(data);
484c314b 1079 regcache_mark_dirty(data->regmap);
efa86e9f
DB
1080 mutex_unlock(&data->lock);
1081
1082 return ret;
1083}
1084
1085static int rpr0521_runtime_resume(struct device *dev)
1086{
1087 struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
1088 struct rpr0521_data *data = iio_priv(indio_dev);
1089 int ret;
1090
484c314b 1091 regcache_sync(data->regmap);
efa86e9f
DB
1092 if (data->als_ps_need_en) {
1093 ret = rpr0521_als_enable(data, RPR0521_MODE_ALS_ENABLE);
1094 if (ret < 0)
1095 return ret;
1096 data->als_ps_need_en = false;
1097 }
1098
1099 if (data->pxs_ps_need_en) {
1100 ret = rpr0521_pxs_enable(data, RPR0521_MODE_PXS_ENABLE);
1101 if (ret < 0)
1102 return ret;
1103 data->pxs_ps_need_en = false;
1104 }
484c314b 1105 msleep(100); //wait for first measurement result
efa86e9f
DB
1106
1107 return 0;
1108}
1109#endif
1110
1111static const struct dev_pm_ops rpr0521_pm_ops = {
1112 SET_RUNTIME_PM_OPS(rpr0521_runtime_suspend,
1113 rpr0521_runtime_resume, NULL)
1114};
1115
1116static const struct acpi_device_id rpr0521_acpi_match[] = {
1117 {"RPR0521", 0},
1118 { }
1119};
1120MODULE_DEVICE_TABLE(acpi, rpr0521_acpi_match);
1121
1122static const struct i2c_device_id rpr0521_id[] = {
1123 {"rpr0521", 0},
1124 { }
1125};
1126
1127MODULE_DEVICE_TABLE(i2c, rpr0521_id);
1128
1129static struct i2c_driver rpr0521_driver = {
1130 .driver = {
1131 .name = RPR0521_DRV_NAME,
1132 .pm = &rpr0521_pm_ops,
1133 .acpi_match_table = ACPI_PTR(rpr0521_acpi_match),
1134 },
1135 .probe = rpr0521_probe,
1136 .remove = rpr0521_remove,
1137 .id_table = rpr0521_id,
1138};
1139
1140module_i2c_driver(rpr0521_driver);
1141
1142MODULE_AUTHOR("Daniel Baluta <daniel.baluta@intel.com>");
1143MODULE_DESCRIPTION("RPR0521 ROHM Ambient Light and Proximity Sensor driver");
1144MODULE_LICENSE("GPL v2");