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