drivers/iio/light/rpr0521.c

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