drivers/iio/light/ltr501.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * ltr501.c - Support for Lite-On LTR501 ambient light and proximity sensor
   4  *
   5  * Copyright 2014 Peter Meerwald <[email protected]>
   6  *
   7  * 7-bit I2C slave address 0x23
   8  *
   9  * TODO: IR LED characteristics
  10  */
  11
  12 #include <linux/module.h>
  13 #include <linux/i2c.h>
  14 #include <linux/err.h>
  15 #include <linux/delay.h>
  16 #include <linux/regmap.h>
  17 #include <linux/acpi.h>
  18 #include <linux/regulator/consumer.h>
  19
  20 #include <linux/iio/iio.h>
  21 #include <linux/iio/events.h>
  22 #include <linux/iio/sysfs.h>
  23 #include <linux/iio/trigger_consumer.h>
  24 #include <linux/iio/buffer.h>
  25 #include <linux/iio/triggered_buffer.h>
  26
  27 #define LTR501_DRV_NAME "ltr501"
  28
  29 #define LTR501_ALS_CONTR 0x80 /* ALS operation mode, SW reset */
  30 #define LTR501_PS_CONTR 0x81 /* PS operation mode */
  31 #define LTR501_PS_MEAS_RATE 0x84 /* measurement rate*/
  32 #define LTR501_ALS_MEAS_RATE 0x85 /* ALS integ time, measurement rate*/
  33 #define LTR501_PART_ID 0x86
  34 #define LTR501_MANUFAC_ID 0x87
  35 #define LTR501_ALS_DATA1 0x88 /* 16-bit, little endian */
  36 #define LTR501_ALS_DATA1_UPPER 0x89 /* upper 8 bits of LTR501_ALS_DATA1 */
  37 #define LTR501_ALS_DATA0 0x8a /* 16-bit, little endian */
  38 #define LTR501_ALS_DATA0_UPPER 0x8b /* upper 8 bits of LTR501_ALS_DATA0 */
  39 #define LTR501_ALS_PS_STATUS 0x8c
  40 #define LTR501_PS_DATA 0x8d /* 16-bit, little endian */
  41 #define LTR501_PS_DATA_UPPER 0x8e /* upper 8 bits of LTR501_PS_DATA */
  42 #define LTR501_INTR 0x8f /* output mode, polarity, mode */
  43 #define LTR501_PS_THRESH_UP 0x90 /* 11 bit, ps upper threshold */
  44 #define LTR501_PS_THRESH_LOW 0x92 /* 11 bit, ps lower threshold */
  45 #define LTR501_ALS_THRESH_UP 0x97 /* 16 bit, ALS upper threshold */
  46 #define LTR501_ALS_THRESH_LOW 0x99 /* 16 bit, ALS lower threshold */
  47 #define LTR501_INTR_PRST 0x9e /* ps thresh, als thresh */
  48 #define LTR501_MAX_REG 0x9f
  49
  50 #define LTR501_ALS_CONTR_SW_RESET BIT(2)
  51 #define LTR501_CONTR_PS_GAIN_MASK (BIT(3) | BIT(2))
  52 #define LTR501_CONTR_PS_GAIN_SHIFT 2
  53 #define LTR501_CONTR_ALS_GAIN_MASK BIT(3)
  54 #define LTR501_CONTR_ACTIVE BIT(1)
  55
  56 #define LTR501_STATUS_ALS_INTR BIT(3)
  57 #define LTR501_STATUS_ALS_RDY BIT(2)
  58 #define LTR501_STATUS_PS_INTR BIT(1)
  59 #define LTR501_STATUS_PS_RDY BIT(0)
  60
  61 #define LTR501_PS_DATA_MASK 0x7ff
  62 #define LTR501_PS_THRESH_MASK 0x7ff
  63 #define LTR501_ALS_THRESH_MASK 0xffff
  64
  65 #define LTR501_ALS_DEF_PERIOD 500000
  66 #define LTR501_PS_DEF_PERIOD 100000
  67
  68 #define LTR501_REGMAP_NAME "ltr501_regmap"
  69
  70 #define LTR501_LUX_CONV(vis_coeff, vis_data, ir_coeff, ir_data) \
  71                         ((vis_coeff * vis_data) - (ir_coeff * ir_data))
  72
  73 static const int int_time_mapping[] = {100000, 50000, 200000, 400000};
  74
  75 static const struct reg_field reg_field_it =
  76                                 REG_FIELD(LTR501_ALS_MEAS_RATE, 3, 4);
  77 static const struct reg_field reg_field_als_intr =
  78                                 REG_FIELD(LTR501_INTR, 1, 1);
  79 static const struct reg_field reg_field_ps_intr =
  80                                 REG_FIELD(LTR501_INTR, 0, 0);
  81 static const struct reg_field reg_field_als_rate =
  82                                 REG_FIELD(LTR501_ALS_MEAS_RATE, 0, 2);
  83 static const struct reg_field reg_field_ps_rate =
  84                                 REG_FIELD(LTR501_PS_MEAS_RATE, 0, 3);
  85 static const struct reg_field reg_field_als_prst =
  86                                 REG_FIELD(LTR501_INTR_PRST, 0, 3);
  87 static const struct reg_field reg_field_ps_prst =
  88                                 REG_FIELD(LTR501_INTR_PRST, 4, 7);
  89
  90 struct ltr501_samp_table {
  91         int freq_val;  /* repetition frequency in micro HZ*/
  92         int time_val; /* repetition rate in micro seconds */
  93 };
  94
  95 #define LTR501_RESERVED_GAIN -1
  96
  97 enum {
  98         ltr501 = 0,
  99         ltr559,
 100         ltr301,
 101 };
 102
 103 struct ltr501_gain {
 104         int scale;
 105         int uscale;
 106 };
 107
 108 static const struct ltr501_gain ltr501_als_gain_tbl[] = {
 109         {1, 0},
 110         {0, 5000},
 111 };
 112
 113 static const struct ltr501_gain ltr559_als_gain_tbl[] = {
 114         {1, 0},
 115         {0, 500000},
 116         {0, 250000},
 117         {0, 125000},
 118         {LTR501_RESERVED_GAIN, LTR501_RESERVED_GAIN},
 119         {LTR501_RESERVED_GAIN, LTR501_RESERVED_GAIN},
 120         {0, 20000},
 121         {0, 10000},
 122 };
 123
 124 static const struct ltr501_gain ltr501_ps_gain_tbl[] = {
 125         {1, 0},
 126         {0, 250000},
 127         {0, 125000},
 128         {0, 62500},
 129 };
 130
 131 static const struct ltr501_gain ltr559_ps_gain_tbl[] = {
 132         {0, 62500}, /* x16 gain */
 133         {0, 31250}, /* x32 gain */
 134         {0, 15625}, /* bits X1 are for x64 gain */
 135         {0, 15624},
 136 };
 137
 138 struct ltr501_chip_info {
 139         u8 partid;
 140         const struct ltr501_gain *als_gain;
 141         int als_gain_tbl_size;
 142         const struct ltr501_gain *ps_gain;
 143         int ps_gain_tbl_size;
 144         u8 als_mode_active;
 145         u8 als_gain_mask;
 146         u8 als_gain_shift;
 147         struct iio_chan_spec const *channels;
 148         const int no_channels;
 149         const struct iio_info *info;
 150         const struct iio_info *info_no_irq;
 151 };
 152
 153 struct ltr501_data {
 154         struct i2c_client *client;
 155         struct regulator_bulk_data regulators[2];
 156         struct mutex lock_als, lock_ps;
 157         const struct ltr501_chip_info *chip_info;
 158         u8 als_contr, ps_contr;
 159         int als_period, ps_period; /* period in micro seconds */
 160         struct regmap *regmap;
 161         struct regmap_field *reg_it;
 162         struct regmap_field *reg_als_intr;
 163         struct regmap_field *reg_ps_intr;
 164         struct regmap_field *reg_als_rate;
 165         struct regmap_field *reg_ps_rate;
 166         struct regmap_field *reg_als_prst;
 167         struct regmap_field *reg_ps_prst;
 168 };
 169
 170 static const struct ltr501_samp_table ltr501_als_samp_table[] = {
 171                         {20000000, 50000}, {10000000, 100000},
 172                         {5000000, 200000}, {2000000, 500000},
 173                         {1000000, 1000000}, {500000, 2000000},
 174                         {500000, 2000000}, {500000, 2000000}
 175 };
 176
 177 static const struct ltr501_samp_table ltr501_ps_samp_table[] = {
 178                         {20000000, 50000}, {14285714, 70000},
 179                         {10000000, 100000}, {5000000, 200000},
 180                         {2000000, 500000}, {1000000, 1000000},
 181                         {500000, 2000000}, {500000, 2000000},
 182                         {500000, 2000000}
 183 };
 184
 185 static int ltr501_match_samp_freq(const struct ltr501_samp_table *tab,
 186                                            int len, int val, int val2)
 187 {
 188         int i, freq;
 189
 190         freq = val * 1000000 + val2;
 191
 192         for (i = 0; i < len; i++) {
 193                 if (tab[i].freq_val == freq)
 194                         return i;
 195         }
 196
 197         return -EINVAL;
 198 }
 199
 200 static int ltr501_als_read_samp_freq(const struct ltr501_data *data,
 201                                      int *val, int *val2)
 202 {
 203         int ret, i;
 204
 205         ret = regmap_field_read(data->reg_als_rate, &i);
 206         if (ret < 0)
 207                 return ret;
 208
 209         if (i < 0 || i >= ARRAY_SIZE(ltr501_als_samp_table))
 210                 return -EINVAL;
 211
 212         *val = ltr501_als_samp_table[i].freq_val / 1000000;
 213         *val2 = ltr501_als_samp_table[i].freq_val % 1000000;
 214
 215         return IIO_VAL_INT_PLUS_MICRO;
 216 }
 217
 218 static int ltr501_ps_read_samp_freq(const struct ltr501_data *data,
 219                                     int *val, int *val2)
 220 {
 221         int ret, i;
 222
 223         ret = regmap_field_read(data->reg_ps_rate, &i);
 224         if (ret < 0)
 225                 return ret;
 226
 227         if (i < 0 || i >= ARRAY_SIZE(ltr501_ps_samp_table))
 228                 return -EINVAL;
 229
 230         *val = ltr501_ps_samp_table[i].freq_val / 1000000;
 231         *val2 = ltr501_ps_samp_table[i].freq_val % 1000000;
 232
 233         return IIO_VAL_INT_PLUS_MICRO;
 234 }
 235
 236 static int ltr501_als_write_samp_freq(struct ltr501_data *data,
 237                                       int val, int val2)
 238 {
 239         int i, ret;
 240
 241         i = ltr501_match_samp_freq(ltr501_als_samp_table,
 242                                    ARRAY_SIZE(ltr501_als_samp_table),
 243                                    val, val2);
 244
 245         if (i < 0)
 246                 return i;
 247
 248         mutex_lock(&data->lock_als);
 249         ret = regmap_field_write(data->reg_als_rate, i);
 250         mutex_unlock(&data->lock_als);
 251
 252         return ret;
 253 }
 254
 255 static int ltr501_ps_write_samp_freq(struct ltr501_data *data,
 256                                      int val, int val2)
 257 {
 258         int i, ret;
 259
 260         i = ltr501_match_samp_freq(ltr501_ps_samp_table,
 261                                    ARRAY_SIZE(ltr501_ps_samp_table),
 262                                    val, val2);
 263
 264         if (i < 0)
 265                 return i;
 266
 267         mutex_lock(&data->lock_ps);
 268         ret = regmap_field_write(data->reg_ps_rate, i);
 269         mutex_unlock(&data->lock_ps);
 270
 271         return ret;
 272 }
 273
 274 static int ltr501_als_read_samp_period(const struct ltr501_data *data, int *val)
 275 {
 276         int ret, i;
 277
 278         ret = regmap_field_read(data->reg_als_rate, &i);
 279         if (ret < 0)
 280                 return ret;
 281
 282         if (i < 0 || i >= ARRAY_SIZE(ltr501_als_samp_table))
 283                 return -EINVAL;
 284
 285         *val = ltr501_als_samp_table[i].time_val;
 286
 287         return IIO_VAL_INT;
 288 }
 289
 290 static int ltr501_ps_read_samp_period(const struct ltr501_data *data, int *val)
 291 {
 292         int ret, i;
 293
 294         ret = regmap_field_read(data->reg_ps_rate, &i);
 295         if (ret < 0)
 296                 return ret;
 297
 298         if (i < 0 || i >= ARRAY_SIZE(ltr501_ps_samp_table))
 299                 return -EINVAL;
 300
 301         *val = ltr501_ps_samp_table[i].time_val;
 302
 303         return IIO_VAL_INT;
 304 }
 305
 306 /* IR and visible spectrum coeff's are given in data sheet */
 307 static unsigned long ltr501_calculate_lux(u16 vis_data, u16 ir_data)
 308 {
 309         unsigned long ratio, lux;
 310
 311         if (vis_data == 0)
 312                 return 0;
 313
 314         /* multiply numerator by 100 to avoid handling ratio < 1 */
 315         ratio = DIV_ROUND_UP(ir_data * 100, ir_data + vis_data);
 316
 317         if (ratio < 45)
 318                 lux = LTR501_LUX_CONV(1774, vis_data, -1105, ir_data);
 319         else if (ratio >= 45 && ratio < 64)
 320                 lux = LTR501_LUX_CONV(3772, vis_data, 1336, ir_data);
 321         else if (ratio >= 64 && ratio < 85)
 322                 lux = LTR501_LUX_CONV(1690, vis_data, 169, ir_data);
 323         else
 324                 lux = 0;
 325
 326         return lux / 1000;
 327 }
 328
 329 static int ltr501_drdy(const struct ltr501_data *data, u8 drdy_mask)
 330 {
 331         int tries = 100;
 332         int ret, status;
 333
 334         while (tries--) {
 335                 ret = regmap_read(data->regmap, LTR501_ALS_PS_STATUS, &status);
 336                 if (ret < 0)
 337                         return ret;
 338                 if ((status & drdy_mask) == drdy_mask)
 339                         return 0;
 340                 msleep(25);
 341         }
 342
 343         dev_err(&data->client->dev, "ltr501_drdy() failed, data not ready\n");
 344         return -EIO;
 345 }
 346
 347 static int ltr501_set_it_time(struct ltr501_data *data, int it)
 348 {
 349         int ret, i, index = -1, status;
 350
 351         for (i = 0; i < ARRAY_SIZE(int_time_mapping); i++) {
 352                 if (int_time_mapping[i] == it) {
 353                         index = i;
 354                         break;
 355                 }
 356         }
 357         /* Make sure integ time index is valid */
 358         if (index < 0)
 359                 return -EINVAL;
 360
 361         ret = regmap_read(data->regmap, LTR501_ALS_CONTR, &status);
 362         if (ret < 0)
 363                 return ret;
 364
 365         if (status & LTR501_CONTR_ALS_GAIN_MASK) {
 366                 /*
 367                  * 200 ms and 400 ms integ time can only be
 368                  * used in dynamic range 1
 369                  */
 370                 if (index > 1)
 371                         return -EINVAL;
 372         } else
 373                 /* 50 ms integ time can only be used in dynamic range 2 */
 374                 if (index == 1)
 375                         return -EINVAL;
 376
 377         return regmap_field_write(data->reg_it, index);
 378 }
 379
 380 /* read int time in micro seconds */
 381 static int ltr501_read_it_time(const struct ltr501_data *data,
 382                                int *val, int *val2)
 383 {
 384         int ret, index;
 385
 386         ret = regmap_field_read(data->reg_it, &index);
 387         if (ret < 0)
 388                 return ret;
 389
 390         /* Make sure integ time index is valid */
 391         if (index < 0 || index >= ARRAY_SIZE(int_time_mapping))
 392                 return -EINVAL;
 393
 394         *val2 = int_time_mapping[index];
 395         *val = 0;
 396
 397         return IIO_VAL_INT_PLUS_MICRO;
 398 }
 399
 400 static int ltr501_read_als(const struct ltr501_data *data, __le16 buf[2])
 401 {
 402         int ret;
 403
 404         ret = ltr501_drdy(data, LTR501_STATUS_ALS_RDY);
 405         if (ret < 0)
 406                 return ret;
 407         /* always read both ALS channels in given order */
 408         return regmap_bulk_read(data->regmap, LTR501_ALS_DATA1,
 409                                 buf, 2 * sizeof(__le16));
 410 }
 411
 412 static int ltr501_read_ps(const struct ltr501_data *data)
 413 {
 414         __le16 status;
 415         int ret;
 416
 417         ret = ltr501_drdy(data, LTR501_STATUS_PS_RDY);
 418         if (ret < 0)
 419                 return ret;
 420
 421         ret = regmap_bulk_read(data->regmap, LTR501_PS_DATA,
 422                                &status, sizeof(status));
 423         if (ret < 0)
 424                 return ret;
 425
 426         return le16_to_cpu(status);
 427 }
 428
 429 static int ltr501_read_intr_prst(const struct ltr501_data *data,
 430                                  enum iio_chan_type type,
 431                                  int *val2)
 432 {
 433         int ret, samp_period, prst;
 434
 435         switch (type) {
 436         case IIO_INTENSITY:
 437                 ret = regmap_field_read(data->reg_als_prst, &prst);
 438                 if (ret < 0)
 439                         return ret;
 440
 441                 ret = ltr501_als_read_samp_period(data, &samp_period);
 442
 443                 if (ret < 0)
 444                         return ret;
 445                 *val2 = samp_period * prst;
 446                 return IIO_VAL_INT_PLUS_MICRO;
 447         case IIO_PROXIMITY:
 448                 ret = regmap_field_read(data->reg_ps_prst, &prst);
 449                 if (ret < 0)
 450                         return ret;
 451
 452                 ret = ltr501_ps_read_samp_period(data, &samp_period);
 453
 454                 if (ret < 0)
 455                         return ret;
 456
 457                 *val2 = samp_period * prst;
 458                 return IIO_VAL_INT_PLUS_MICRO;
 459         default:
 460                 return -EINVAL;
 461         }
 462
 463         return -EINVAL;
 464 }
 465
 466 static int ltr501_write_intr_prst(struct ltr501_data *data,
 467                                   enum iio_chan_type type,
 468                                   int val, int val2)
 469 {
 470         int ret, samp_period, new_val;
 471         unsigned long period;
 472
 473         if (val < 0 || val2 < 0)
 474                 return -EINVAL;
 475
 476         /* period in microseconds */
 477         period = ((val * 1000000) + val2);
 478
 479         switch (type) {
 480         case IIO_INTENSITY:
 481                 ret = ltr501_als_read_samp_period(data, &samp_period);
 482                 if (ret < 0)
 483                         return ret;
 484
 485                 /* period should be atleast equal to sampling period */
 486                 if (period < samp_period)
 487                         return -EINVAL;
 488
 489                 new_val = DIV_ROUND_UP(period, samp_period);
 490                 if (new_val < 0 || new_val > 0x0f)
 491                         return -EINVAL;
 492
 493                 mutex_lock(&data->lock_als);
 494                 ret = regmap_field_write(data->reg_als_prst, new_val);
 495                 mutex_unlock(&data->lock_als);
 496                 if (ret >= 0)
 497                         data->als_period = period;
 498
 499                 return ret;
 500         case IIO_PROXIMITY:
 501                 ret = ltr501_ps_read_samp_period(data, &samp_period);
 502                 if (ret < 0)
 503                         return ret;
 504
 505                 /* period should be atleast equal to rate */
 506                 if (period < samp_period)
 507                         return -EINVAL;
 508
 509                 new_val = DIV_ROUND_UP(period, samp_period);
 510                 if (new_val < 0 || new_val > 0x0f)
 511                         return -EINVAL;
 512
 513                 mutex_lock(&data->lock_ps);
 514                 ret = regmap_field_write(data->reg_ps_prst, new_val);
 515                 mutex_unlock(&data->lock_ps);
 516                 if (ret >= 0)
 517                         data->ps_period = period;
 518
 519                 return ret;
 520         default:
 521                 return -EINVAL;
 522         }
 523
 524         return -EINVAL;
 525 }
 526
 527 static const struct iio_event_spec ltr501_als_event_spec[] = {
 528         {
 529                 .type = IIO_EV_TYPE_THRESH,
 530                 .dir = IIO_EV_DIR_RISING,
 531                 .mask_separate = BIT(IIO_EV_INFO_VALUE),
 532         }, {
 533                 .type = IIO_EV_TYPE_THRESH,
 534                 .dir = IIO_EV_DIR_FALLING,
 535                 .mask_separate = BIT(IIO_EV_INFO_VALUE),
 536         }, {
 537                 .type = IIO_EV_TYPE_THRESH,
 538                 .dir = IIO_EV_DIR_EITHER,
 539                 .mask_separate = BIT(IIO_EV_INFO_ENABLE) |
 540                                  BIT(IIO_EV_INFO_PERIOD),
 541         },
 542
 543 };
 544
 545 static const struct iio_event_spec ltr501_pxs_event_spec[] = {
 546         {
 547                 .type = IIO_EV_TYPE_THRESH,
 548                 .dir = IIO_EV_DIR_RISING,
 549                 .mask_separate = BIT(IIO_EV_INFO_VALUE),
 550         }, {
 551                 .type = IIO_EV_TYPE_THRESH,
 552                 .dir = IIO_EV_DIR_FALLING,
 553                 .mask_separate = BIT(IIO_EV_INFO_VALUE),
 554         }, {
 555                 .type = IIO_EV_TYPE_THRESH,
 556                 .dir = IIO_EV_DIR_EITHER,
 557                 .mask_separate = BIT(IIO_EV_INFO_ENABLE) |
 558                                  BIT(IIO_EV_INFO_PERIOD),
 559         },
 560 };
 561
 562 #define LTR501_INTENSITY_CHANNEL(_idx, _addr, _mod, _shared, \
 563                                  _evspec, _evsize) { \
 564         .type = IIO_INTENSITY, \
 565         .modified = 1, \
 566         .address = (_addr), \
 567         .channel2 = (_mod), \
 568         .info_mask_separate = BIT(IIO_CHAN_INFO_RAW), \
 569         .info_mask_shared_by_type = (_shared), \
 570         .scan_index = (_idx), \
 571         .scan_type = { \
 572                 .sign = 'u', \
 573                 .realbits = 16, \
 574                 .storagebits = 16, \
 575                 .endianness = IIO_CPU, \
 576         }, \
 577         .event_spec = _evspec,\
 578         .num_event_specs = _evsize,\
 579 }
 580
 581 #define LTR501_LIGHT_CHANNEL() { \
 582         .type = IIO_LIGHT, \
 583         .info_mask_separate = BIT(IIO_CHAN_INFO_PROCESSED), \
 584         .scan_index = -1, \
 585 }
 586
 587 static const struct iio_chan_spec ltr501_channels[] = {
 588         LTR501_LIGHT_CHANNEL(),
 589         LTR501_INTENSITY_CHANNEL(0, LTR501_ALS_DATA0, IIO_MOD_LIGHT_BOTH, 0,
 590                                  ltr501_als_event_spec,
 591                                  ARRAY_SIZE(ltr501_als_event_spec)),
 592         LTR501_INTENSITY_CHANNEL(1, LTR501_ALS_DATA1, IIO_MOD_LIGHT_IR,
 593                                  BIT(IIO_CHAN_INFO_SCALE) |
 594                                  BIT(IIO_CHAN_INFO_INT_TIME) |
 595                                  BIT(IIO_CHAN_INFO_SAMP_FREQ),
 596                                  NULL, 0),
 597         {
 598                 .type = IIO_PROXIMITY,
 599                 .address = LTR501_PS_DATA,
 600                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
 601                         BIT(IIO_CHAN_INFO_SCALE),
 602                 .scan_index = 2,
 603                 .scan_type = {
 604                         .sign = 'u',
 605                         .realbits = 11,
 606                         .storagebits = 16,
 607                         .endianness = IIO_CPU,
 608                 },
 609                 .event_spec = ltr501_pxs_event_spec,
 610                 .num_event_specs = ARRAY_SIZE(ltr501_pxs_event_spec),
 611         },
 612         IIO_CHAN_SOFT_TIMESTAMP(3),
 613 };
 614
 615 static const struct iio_chan_spec ltr301_channels[] = {
 616         LTR501_LIGHT_CHANNEL(),
 617         LTR501_INTENSITY_CHANNEL(0, LTR501_ALS_DATA0, IIO_MOD_LIGHT_BOTH, 0,
 618                                  ltr501_als_event_spec,
 619                                  ARRAY_SIZE(ltr501_als_event_spec)),
 620         LTR501_INTENSITY_CHANNEL(1, LTR501_ALS_DATA1, IIO_MOD_LIGHT_IR,
 621                                  BIT(IIO_CHAN_INFO_SCALE) |
 622                                  BIT(IIO_CHAN_INFO_INT_TIME) |
 623                                  BIT(IIO_CHAN_INFO_SAMP_FREQ),
 624                                  NULL, 0),
 625         IIO_CHAN_SOFT_TIMESTAMP(2),
 626 };
 627
 628 static int ltr501_read_raw(struct iio_dev *indio_dev,
 629                            struct iio_chan_spec const *chan,
 630                            int *val, int *val2, long mask)
 631 {
 632         struct ltr501_data *data = iio_priv(indio_dev);
 633         __le16 buf[2];
 634         int ret, i;
 635
 636         switch (mask) {
 637         case IIO_CHAN_INFO_PROCESSED:
 638                 switch (chan->type) {
 639                 case IIO_LIGHT:
 640                         ret = iio_device_claim_direct_mode(indio_dev);
 641                         if (ret)
 642                                 return ret;
 643
 644                         mutex_lock(&data->lock_als);
 645                         ret = ltr501_read_als(data, buf);
 646                         mutex_unlock(&data->lock_als);
 647                         iio_device_release_direct_mode(indio_dev);
 648                         if (ret < 0)
 649                                 return ret;
 650                         *val = ltr501_calculate_lux(le16_to_cpu(buf[1]),
 651                                                     le16_to_cpu(buf[0]));
 652                         return IIO_VAL_INT;
 653                 default:
 654                         return -EINVAL;
 655                 }
 656         case IIO_CHAN_INFO_RAW:
 657                 ret = iio_device_claim_direct_mode(indio_dev);
 658                 if (ret)
 659                         return ret;
 660
 661                 switch (chan->type) {
 662                 case IIO_INTENSITY:
 663                         mutex_lock(&data->lock_als);
 664                         ret = ltr501_read_als(data, buf);
 665                         mutex_unlock(&data->lock_als);
 666                         if (ret < 0)
 667                                 break;
 668                         *val = le16_to_cpu(chan->address == LTR501_ALS_DATA1 ?
 669                                            buf[0] : buf[1]);
 670                         ret = IIO_VAL_INT;
 671                         break;
 672                 case IIO_PROXIMITY:
 673                         mutex_lock(&data->lock_ps);
 674                         ret = ltr501_read_ps(data);
 675                         mutex_unlock(&data->lock_ps);
 676                         if (ret < 0)
 677                                 break;
 678                         *val = ret & LTR501_PS_DATA_MASK;
 679                         ret = IIO_VAL_INT;
 680                         break;
 681                 default:
 682                         ret = -EINVAL;
 683                         break;
 684                 }
 685
 686                 iio_device_release_direct_mode(indio_dev);
 687                 return ret;
 688
 689         case IIO_CHAN_INFO_SCALE:
 690                 switch (chan->type) {
 691                 case IIO_INTENSITY:
 692                         i = (data->als_contr & data->chip_info->als_gain_mask)
 693                              >> data->chip_info->als_gain_shift;
 694                         *val = data->chip_info->als_gain[i].scale;
 695                         *val2 = data->chip_info->als_gain[i].uscale;
 696                         return IIO_VAL_INT_PLUS_MICRO;
 697                 case IIO_PROXIMITY:
 698                         i = (data->ps_contr & LTR501_CONTR_PS_GAIN_MASK) >>
 699                                 LTR501_CONTR_PS_GAIN_SHIFT;
 700                         *val = data->chip_info->ps_gain[i].scale;
 701                         *val2 = data->chip_info->ps_gain[i].uscale;
 702                         return IIO_VAL_INT_PLUS_MICRO;
 703                 default:
 704                         return -EINVAL;
 705                 }
 706         case IIO_CHAN_INFO_INT_TIME:
 707                 switch (chan->type) {
 708                 case IIO_INTENSITY:
 709                         return ltr501_read_it_time(data, val, val2);
 710                 default:
 711                         return -EINVAL;
 712                 }
 713         case IIO_CHAN_INFO_SAMP_FREQ:
 714                 switch (chan->type) {
 715                 case IIO_INTENSITY:
 716                         return ltr501_als_read_samp_freq(data, val, val2);
 717                 case IIO_PROXIMITY:
 718                         return ltr501_ps_read_samp_freq(data, val, val2);
 719                 default:
 720                         return -EINVAL;
 721                 }
 722         }
 723         return -EINVAL;
 724 }
 725
 726 static int ltr501_get_gain_index(const struct ltr501_gain *gain, int size,
 727                                  int val, int val2)
 728 {
 729         int i;
 730
 731         for (i = 0; i < size; i++)
 732                 if (val == gain[i].scale && val2 == gain[i].uscale)
 733                         return i;
 734
 735         return -1;
 736 }
 737
 738 static int ltr501_write_raw(struct iio_dev *indio_dev,
 739                             struct iio_chan_spec const *chan,
 740                             int val, int val2, long mask)
 741 {
 742         struct ltr501_data *data = iio_priv(indio_dev);
 743         int i, ret, freq_val, freq_val2;
 744         const struct ltr501_chip_info *info = data->chip_info;
 745
 746         ret = iio_device_claim_direct_mode(indio_dev);
 747         if (ret)
 748                 return ret;
 749
 750         switch (mask) {
 751         case IIO_CHAN_INFO_SCALE:
 752                 switch (chan->type) {
 753                 case IIO_INTENSITY:
 754                         i = ltr501_get_gain_index(info->als_gain,
 755                                                   info->als_gain_tbl_size,
 756                                                   val, val2);
 757                         if (i < 0) {
 758                                 ret = -EINVAL;
 759                                 break;
 760                         }
 761
 762                         data->als_contr &= ~info->als_gain_mask;
 763                         data->als_contr |= i << info->als_gain_shift;
 764
 765                         ret = regmap_write(data->regmap, LTR501_ALS_CONTR,
 766                                            data->als_contr);
 767                         break;
 768                 case IIO_PROXIMITY:
 769                         i = ltr501_get_gain_index(info->ps_gain,
 770                                                   info->ps_gain_tbl_size,
 771                                                   val, val2);
 772                         if (i < 0) {
 773                                 ret = -EINVAL;
 774                                 break;
 775                         }
 776                         data->ps_contr &= ~LTR501_CONTR_PS_GAIN_MASK;
 777                         data->ps_contr |= i << LTR501_CONTR_PS_GAIN_SHIFT;
 778
 779                         ret = regmap_write(data->regmap, LTR501_PS_CONTR,
 780                                            data->ps_contr);
 781                         break;
 782                 default:
 783                         ret = -EINVAL;
 784                         break;
 785                 }
 786                 break;
 787
 788         case IIO_CHAN_INFO_INT_TIME:
 789                 switch (chan->type) {
 790                 case IIO_INTENSITY:
 791                         if (val != 0) {
 792                                 ret = -EINVAL;
 793                                 break;
 794                         }
 795                         mutex_lock(&data->lock_als);
 796                         ret = ltr501_set_it_time(data, val2);
 797                         mutex_unlock(&data->lock_als);
 798                         break;
 799                 default:
 800                         ret = -EINVAL;
 801                         break;
 802                 }
 803                 break;
 804
 805         case IIO_CHAN_INFO_SAMP_FREQ:
 806                 switch (chan->type) {
 807                 case IIO_INTENSITY:
 808                         ret = ltr501_als_read_samp_freq(data, &freq_val,
 809                                                         &freq_val2);
 810                         if (ret < 0)
 811                                 break;
 812
 813                         ret = ltr501_als_write_samp_freq(data, val, val2);
 814                         if (ret < 0)
 815                                 break;
 816
 817                         /* update persistence count when changing frequency */
 818                         ret = ltr501_write_intr_prst(data, chan->type,
 819                                                      0, data->als_period);
 820
 821                         if (ret < 0)
 822                                 ret = ltr501_als_write_samp_freq(data, freq_val,
 823                                                                  freq_val2);
 824                         break;
 825                 case IIO_PROXIMITY:
 826                         ret = ltr501_ps_read_samp_freq(data, &freq_val,
 827                                                        &freq_val2);
 828                         if (ret < 0)
 829                                 break;
 830
 831                         ret = ltr501_ps_write_samp_freq(data, val, val2);
 832                         if (ret < 0)
 833                                 break;
 834
 835                         /* update persistence count when changing frequency */
 836                         ret = ltr501_write_intr_prst(data, chan->type,
 837                                                      0, data->ps_period);
 838
 839                         if (ret < 0)
 840                                 ret = ltr501_ps_write_samp_freq(data, freq_val,
 841                                                                 freq_val2);
 842                         break;
 843                 default:
 844                         ret = -EINVAL;
 845                         break;
 846                 }
 847                 break;
 848
 849         default:
 850                 ret = -EINVAL;
 851                 break;
 852         }
 853
 854         iio_device_release_direct_mode(indio_dev);
 855         return ret;
 856 }
 857
 858 static int ltr501_read_thresh(const struct iio_dev *indio_dev,
 859                               const struct iio_chan_spec *chan,
 860                               enum iio_event_type type,
 861                               enum iio_event_direction dir,
 862                               enum iio_event_info info,
 863                               int *val, int *val2)
 864 {
 865         const struct ltr501_data *data = iio_priv(indio_dev);
 866         int ret, thresh_data;
 867
 868         switch (chan->type) {
 869         case IIO_INTENSITY:
 870                 switch (dir) {
 871                 case IIO_EV_DIR_RISING:
 872                         ret = regmap_bulk_read(data->regmap,
 873                                                LTR501_ALS_THRESH_UP,
 874                                                &thresh_data, 2);
 875                         if (ret < 0)
 876                                 return ret;
 877                         *val = thresh_data & LTR501_ALS_THRESH_MASK;
 878                         return IIO_VAL_INT;
 879                 case IIO_EV_DIR_FALLING:
 880                         ret = regmap_bulk_read(data->regmap,
 881                                                LTR501_ALS_THRESH_LOW,
 882                                                &thresh_data, 2);
 883                         if (ret < 0)
 884                                 return ret;
 885                         *val = thresh_data & LTR501_ALS_THRESH_MASK;
 886                         return IIO_VAL_INT;
 887                 default:
 888                         return -EINVAL;
 889                 }
 890         case IIO_PROXIMITY:
 891                 switch (dir) {
 892                 case IIO_EV_DIR_RISING:
 893                         ret = regmap_bulk_read(data->regmap,
 894                                                LTR501_PS_THRESH_UP,
 895                                                &thresh_data, 2);
 896                         if (ret < 0)
 897                                 return ret;
 898                         *val = thresh_data & LTR501_PS_THRESH_MASK;
 899                         return IIO_VAL_INT;
 900                 case IIO_EV_DIR_FALLING:
 901                         ret = regmap_bulk_read(data->regmap,
 902                                                LTR501_PS_THRESH_LOW,
 903                                                &thresh_data, 2);
 904                         if (ret < 0)
 905                                 return ret;
 906                         *val = thresh_data & LTR501_PS_THRESH_MASK;
 907                         return IIO_VAL_INT;
 908                 default:
 909                         return -EINVAL;
 910                 }
 911         default:
 912                 return -EINVAL;
 913         }
 914
 915         return -EINVAL;
 916 }
 917
 918 static int ltr501_write_thresh(struct iio_dev *indio_dev,
 919                                const struct iio_chan_spec *chan,
 920                                enum iio_event_type type,
 921                                enum iio_event_direction dir,
 922                                enum iio_event_info info,
 923                                int val, int val2)
 924 {
 925         struct ltr501_data *data = iio_priv(indio_dev);
 926         int ret;
 927
 928         if (val < 0)
 929                 return -EINVAL;
 930
 931         switch (chan->type) {
 932         case IIO_INTENSITY:
 933                 if (val > LTR501_ALS_THRESH_MASK)
 934                         return -EINVAL;
 935                 switch (dir) {
 936                 case IIO_EV_DIR_RISING:
 937                         mutex_lock(&data->lock_als);
 938                         ret = regmap_bulk_write(data->regmap,
 939                                                 LTR501_ALS_THRESH_UP,
 940                                                 &val, 2);
 941                         mutex_unlock(&data->lock_als);
 942                         return ret;
 943                 case IIO_EV_DIR_FALLING:
 944                         mutex_lock(&data->lock_als);
 945                         ret = regmap_bulk_write(data->regmap,
 946                                                 LTR501_ALS_THRESH_LOW,
 947                                                 &val, 2);
 948                         mutex_unlock(&data->lock_als);
 949                         return ret;
 950                 default:
 951                         return -EINVAL;
 952                 }
 953         case IIO_PROXIMITY:
 954                 if (val > LTR501_PS_THRESH_MASK)
 955                         return -EINVAL;
 956                 switch (dir) {
 957                 case IIO_EV_DIR_RISING:
 958                         mutex_lock(&data->lock_ps);
 959                         ret = regmap_bulk_write(data->regmap,
 960                                                 LTR501_PS_THRESH_UP,
 961                                                 &val, 2);
 962                         mutex_unlock(&data->lock_ps);
 963                         return ret;
 964                 case IIO_EV_DIR_FALLING:
 965                         mutex_lock(&data->lock_ps);
 966                         ret = regmap_bulk_write(data->regmap,
 967                                                 LTR501_PS_THRESH_LOW,
 968                                                 &val, 2);
 969                         mutex_unlock(&data->lock_ps);
 970                         return ret;
 971                 default:
 972                         return -EINVAL;
 973                 }
 974         default:
 975                 return -EINVAL;
 976         }
 977
 978         return -EINVAL;
 979 }
 980
 981 static int ltr501_read_event(struct iio_dev *indio_dev,
 982                              const struct iio_chan_spec *chan,
 983                              enum iio_event_type type,
 984                              enum iio_event_direction dir,
 985                              enum iio_event_info info,
 986                              int *val, int *val2)
 987 {
 988         int ret;
 989
 990         switch (info) {
 991         case IIO_EV_INFO_VALUE:
 992                 return ltr501_read_thresh(indio_dev, chan, type, dir,
 993                                           info, val, val2);
 994         case IIO_EV_INFO_PERIOD:
 995                 ret = ltr501_read_intr_prst(iio_priv(indio_dev),
 996                                             chan->type, val2);
 997                 *val = *val2 / 1000000;
 998                 *val2 = *val2 % 1000000;
 999                 return ret;
1000         default:
1001                 return -EINVAL;
1002         }
1003
1004         return -EINVAL;
1005 }
1006
1007 static int ltr501_write_event(struct iio_dev *indio_dev,
1008                               const struct iio_chan_spec *chan,
1009                               enum iio_event_type type,
1010                               enum iio_event_direction dir,
1011                               enum iio_event_info info,
1012                               int val, int val2)
1013 {
1014         switch (info) {
1015         case IIO_EV_INFO_VALUE:
1016                 if (val2 != 0)
1017                         return -EINVAL;
1018                 return ltr501_write_thresh(indio_dev, chan, type, dir,
1019                                            info, val, val2);
1020         case IIO_EV_INFO_PERIOD:
1021                 return ltr501_write_intr_prst(iio_priv(indio_dev), chan->type,
1022                                               val, val2);
1023         default:
1024                 return -EINVAL;
1025         }
1026
1027         return -EINVAL;
1028 }
1029
1030 static int ltr501_read_event_config(struct iio_dev *indio_dev,
1031                                     const struct iio_chan_spec *chan,
1032                                     enum iio_event_type type,
1033                                     enum iio_event_direction dir)
1034 {
1035         struct ltr501_data *data = iio_priv(indio_dev);
1036         int ret, status;
1037
1038         switch (chan->type) {
1039         case IIO_INTENSITY:
1040                 ret = regmap_field_read(data->reg_als_intr, &status);
1041                 if (ret < 0)
1042                         return ret;
1043                 return status;
1044         case IIO_PROXIMITY:
1045                 ret = regmap_field_read(data->reg_ps_intr, &status);
1046                 if (ret < 0)
1047                         return ret;
1048                 return status;
1049         default:
1050                 return -EINVAL;
1051         }
1052
1053         return -EINVAL;
1054 }
1055
1056 static int ltr501_write_event_config(struct iio_dev *indio_dev,
1057                                      const struct iio_chan_spec *chan,
1058                                      enum iio_event_type type,
1059                                      enum iio_event_direction dir, int state)
1060 {
1061         struct ltr501_data *data = iio_priv(indio_dev);
1062         int ret;
1063
1064         /* only 1 and 0 are valid inputs */
1065         if (state != 1  && state != 0)
1066                 return -EINVAL;
1067
1068         switch (chan->type) {
1069         case IIO_INTENSITY:
1070                 mutex_lock(&data->lock_als);
1071                 ret = regmap_field_write(data->reg_als_intr, state);
1072                 mutex_unlock(&data->lock_als);
1073                 return ret;
1074         case IIO_PROXIMITY:
1075                 mutex_lock(&data->lock_ps);
1076                 ret = regmap_field_write(data->reg_ps_intr, state);
1077                 mutex_unlock(&data->lock_ps);
1078                 return ret;
1079         default:
1080                 return -EINVAL;
1081         }
1082
1083         return -EINVAL;
1084 }
1085
1086 static ssize_t ltr501_show_proximity_scale_avail(struct device *dev,
1087                                                  struct device_attribute *attr,
1088                                                  char *buf)
1089 {
1090         struct ltr501_data *data = iio_priv(dev_to_iio_dev(dev));
1091         const struct ltr501_chip_info *info = data->chip_info;
1092         ssize_t len = 0;
1093         int i;
1094
1095         for (i = 0; i < info->ps_gain_tbl_size; i++) {
1096                 if (info->ps_gain[i].scale == LTR501_RESERVED_GAIN)
1097                         continue;
1098                 len += scnprintf(buf + len, PAGE_SIZE - len, "%d.%06d ",
1099                                  info->ps_gain[i].scale,
1100                                  info->ps_gain[i].uscale);
1101         }
1102
1103         buf[len - 1] = '\n';
1104
1105         return len;
1106 }
1107
1108 static ssize_t ltr501_show_intensity_scale_avail(struct device *dev,
1109                                                  struct device_attribute *attr,
1110                                                  char *buf)
1111 {
1112         struct ltr501_data *data = iio_priv(dev_to_iio_dev(dev));
1113         const struct ltr501_chip_info *info = data->chip_info;
1114         ssize_t len = 0;
1115         int i;
1116
1117         for (i = 0; i < info->als_gain_tbl_size; i++) {
1118                 if (info->als_gain[i].scale == LTR501_RESERVED_GAIN)
1119                         continue;
1120                 len += scnprintf(buf + len, PAGE_SIZE - len, "%d.%06d ",
1121                                  info->als_gain[i].scale,
1122                                  info->als_gain[i].uscale);
1123         }
1124
1125         buf[len - 1] = '\n';
1126
1127         return len;
1128 }
1129
1130 static IIO_CONST_ATTR_INT_TIME_AVAIL("0.05 0.1 0.2 0.4");
1131 static IIO_CONST_ATTR_SAMP_FREQ_AVAIL("20 10 5 2 1 0.5");
1132
1133 static IIO_DEVICE_ATTR(in_proximity_scale_available, S_IRUGO,
1134                        ltr501_show_proximity_scale_avail, NULL, 0);
1135 static IIO_DEVICE_ATTR(in_intensity_scale_available, S_IRUGO,
1136                        ltr501_show_intensity_scale_avail, NULL, 0);
1137
1138 static struct attribute *ltr501_attributes[] = {
1139         &iio_dev_attr_in_proximity_scale_available.dev_attr.attr,
1140         &iio_dev_attr_in_intensity_scale_available.dev_attr.attr,
1141         &iio_const_attr_integration_time_available.dev_attr.attr,
1142         &iio_const_attr_sampling_frequency_available.dev_attr.attr,
1143         NULL
1144 };
1145
1146 static struct attribute *ltr301_attributes[] = {
1147         &iio_dev_attr_in_intensity_scale_available.dev_attr.attr,
1148         &iio_const_attr_integration_time_available.dev_attr.attr,
1149         &iio_const_attr_sampling_frequency_available.dev_attr.attr,
1150         NULL
1151 };
1152
1153 static const struct attribute_group ltr501_attribute_group = {
1154         .attrs = ltr501_attributes,
1155 };
1156
1157 static const struct attribute_group ltr301_attribute_group = {
1158         .attrs = ltr301_attributes,
1159 };
1160
1161 static const struct iio_info ltr501_info_no_irq = {
1162         .read_raw = ltr501_read_raw,
1163         .write_raw = ltr501_write_raw,
1164         .attrs = &ltr501_attribute_group,
1165 };
1166
1167 static const struct iio_info ltr501_info = {
1168         .read_raw = ltr501_read_raw,
1169         .write_raw = ltr501_write_raw,
1170         .attrs = &ltr501_attribute_group,
1171         .read_event_value       = &ltr501_read_event,
1172         .write_event_value      = &ltr501_write_event,
1173         .read_event_config      = &ltr501_read_event_config,
1174         .write_event_config     = &ltr501_write_event_config,
1175 };
1176
1177 static const struct iio_info ltr301_info_no_irq = {
1178         .read_raw = ltr501_read_raw,
1179         .write_raw = ltr501_write_raw,
1180         .attrs = &ltr301_attribute_group,
1181 };
1182
1183 static const struct iio_info ltr301_info = {
1184         .read_raw = ltr501_read_raw,
1185         .write_raw = ltr501_write_raw,
1186         .attrs = &ltr301_attribute_group,
1187         .read_event_value       = &ltr501_read_event,
1188         .write_event_value      = &ltr501_write_event,
1189         .read_event_config      = &ltr501_read_event_config,
1190         .write_event_config     = &ltr501_write_event_config,
1191 };
1192
1193 static const struct ltr501_chip_info ltr501_chip_info_tbl[] = {
1194         [ltr501] = {
1195                 .partid = 0x08,
1196                 .als_gain = ltr501_als_gain_tbl,
1197                 .als_gain_tbl_size = ARRAY_SIZE(ltr501_als_gain_tbl),
1198                 .ps_gain = ltr501_ps_gain_tbl,
1199                 .ps_gain_tbl_size = ARRAY_SIZE(ltr501_ps_gain_tbl),
1200                 .als_mode_active = BIT(0) | BIT(1),
1201                 .als_gain_mask = BIT(3),
1202                 .als_gain_shift = 3,
1203                 .info = &ltr501_info,
1204                 .info_no_irq = &ltr501_info_no_irq,
1205                 .channels = ltr501_channels,
1206                 .no_channels = ARRAY_SIZE(ltr501_channels),
1207         },
1208         [ltr559] = {
1209                 .partid = 0x09,
1210                 .als_gain = ltr559_als_gain_tbl,
1211                 .als_gain_tbl_size = ARRAY_SIZE(ltr559_als_gain_tbl),
1212                 .ps_gain = ltr559_ps_gain_tbl,
1213                 .ps_gain_tbl_size = ARRAY_SIZE(ltr559_ps_gain_tbl),
1214                 .als_mode_active = BIT(0),
1215                 .als_gain_mask = BIT(2) | BIT(3) | BIT(4),
1216                 .als_gain_shift = 2,
1217                 .info = &ltr501_info,
1218                 .info_no_irq = &ltr501_info_no_irq,
1219                 .channels = ltr501_channels,
1220                 .no_channels = ARRAY_SIZE(ltr501_channels),
1221         },
1222         [ltr301] = {
1223                 .partid = 0x08,
1224                 .als_gain = ltr501_als_gain_tbl,
1225                 .als_gain_tbl_size = ARRAY_SIZE(ltr501_als_gain_tbl),
1226                 .als_mode_active = BIT(0) | BIT(1),
1227                 .als_gain_mask = BIT(3),
1228                 .als_gain_shift = 3,
1229                 .info = &ltr301_info,
1230                 .info_no_irq = &ltr301_info_no_irq,
1231                 .channels = ltr301_channels,
1232                 .no_channels = ARRAY_SIZE(ltr301_channels),
1233         },
1234 };
1235
1236 static int ltr501_write_contr(struct ltr501_data *data, u8 als_val, u8 ps_val)
1237 {
1238         int ret;
1239
1240         ret = regmap_write(data->regmap, LTR501_ALS_CONTR, als_val);
1241         if (ret < 0)
1242                 return ret;
1243
1244         return regmap_write(data->regmap, LTR501_PS_CONTR, ps_val);
1245 }
1246
1247 static irqreturn_t ltr501_trigger_handler(int irq, void *p)
1248 {
1249         struct iio_poll_func *pf = p;
1250         struct iio_dev *indio_dev = pf->indio_dev;
1251         struct ltr501_data *data = iio_priv(indio_dev);
1252         struct {
1253                 u16 channels[3];
1254                 s64 ts __aligned(8);
1255         } scan;
1256         __le16 als_buf[2];
1257         u8 mask = 0;
1258         int j = 0;
1259         int ret, psdata;
1260
1261         memset(&scan, 0, sizeof(scan));
1262
1263         /* figure out which data needs to be ready */
1264         if (test_bit(0, indio_dev->active_scan_mask) ||
1265             test_bit(1, indio_dev->active_scan_mask))
1266                 mask |= LTR501_STATUS_ALS_RDY;
1267         if (test_bit(2, indio_dev->active_scan_mask))
1268                 mask |= LTR501_STATUS_PS_RDY;
1269
1270         ret = ltr501_drdy(data, mask);
1271         if (ret < 0)
1272                 goto done;
1273
1274         if (mask & LTR501_STATUS_ALS_RDY) {
1275                 ret = regmap_bulk_read(data->regmap, LTR501_ALS_DATA1,
1276                                        als_buf, sizeof(als_buf));
1277                 if (ret < 0)
1278                         goto done;
1279                 if (test_bit(0, indio_dev->active_scan_mask))
1280                         scan.channels[j++] = le16_to_cpu(als_buf[1]);
1281                 if (test_bit(1, indio_dev->active_scan_mask))
1282                         scan.channels[j++] = le16_to_cpu(als_buf[0]);
1283         }
1284
1285         if (mask & LTR501_STATUS_PS_RDY) {
1286                 ret = regmap_bulk_read(data->regmap, LTR501_PS_DATA,
1287                                        &psdata, 2);
1288                 if (ret < 0)
1289                         goto done;
1290                 scan.channels[j++] = psdata & LTR501_PS_DATA_MASK;
1291         }
1292
1293         iio_push_to_buffers_with_timestamp(indio_dev, &scan,
1294                                            iio_get_time_ns(indio_dev));
1295
1296 done:
1297         iio_trigger_notify_done(indio_dev->trig);
1298
1299         return IRQ_HANDLED;
1300 }
1301
1302 static irqreturn_t ltr501_interrupt_handler(int irq, void *private)
1303 {
1304         struct iio_dev *indio_dev = private;
1305         struct ltr501_data *data = iio_priv(indio_dev);
1306         int ret, status;
1307
1308         ret = regmap_read(data->regmap, LTR501_ALS_PS_STATUS, &status);
1309         if (ret < 0) {
1310                 dev_err(&data->client->dev,
1311                         "irq read int reg failed\n");
1312                 return IRQ_HANDLED;
1313         }
1314
1315         if (status & LTR501_STATUS_ALS_INTR)
1316                 iio_push_event(indio_dev,
1317                                IIO_UNMOD_EVENT_CODE(IIO_INTENSITY, 0,
1318                                                     IIO_EV_TYPE_THRESH,
1319                                                     IIO_EV_DIR_EITHER),
1320                                iio_get_time_ns(indio_dev));
1321
1322         if (status & LTR501_STATUS_PS_INTR)
1323                 iio_push_event(indio_dev,
1324                                IIO_UNMOD_EVENT_CODE(IIO_PROXIMITY, 0,
1325                                                     IIO_EV_TYPE_THRESH,
1326                                                     IIO_EV_DIR_EITHER),
1327                                iio_get_time_ns(indio_dev));
1328
1329         return IRQ_HANDLED;
1330 }
1331
1332 static int ltr501_init(struct ltr501_data *data)
1333 {
1334         int ret, status;
1335
1336         ret = regmap_read(data->regmap, LTR501_ALS_CONTR, &status);
1337         if (ret < 0)
1338                 return ret;
1339
1340         data->als_contr = status | data->chip_info->als_mode_active;
1341
1342         ret = regmap_read(data->regmap, LTR501_PS_CONTR, &status);
1343         if (ret < 0)
1344                 return ret;
1345
1346         data->ps_contr = status | LTR501_CONTR_ACTIVE;
1347
1348         ret = ltr501_read_intr_prst(data, IIO_INTENSITY, &data->als_period);
1349         if (ret < 0)
1350                 return ret;
1351
1352         ret = ltr501_read_intr_prst(data, IIO_PROXIMITY, &data->ps_period);
1353         if (ret < 0)
1354                 return ret;
1355
1356         return ltr501_write_contr(data, data->als_contr, data->ps_contr);
1357 }
1358
1359 static bool ltr501_is_volatile_reg(struct device *dev, unsigned int reg)
1360 {
1361         switch (reg) {
1362         case LTR501_ALS_DATA1:
1363         case LTR501_ALS_DATA1_UPPER:
1364         case LTR501_ALS_DATA0:
1365         case LTR501_ALS_DATA0_UPPER:
1366         case LTR501_ALS_PS_STATUS:
1367         case LTR501_PS_DATA:
1368         case LTR501_PS_DATA_UPPER:
1369                 return true;
1370         default:
1371                 return false;
1372         }
1373 }
1374
1375 static const struct regmap_config ltr501_regmap_config = {
1376         .name =  LTR501_REGMAP_NAME,
1377         .reg_bits = 8,
1378         .val_bits = 8,
1379         .max_register = LTR501_MAX_REG,
1380         .cache_type = REGCACHE_RBTREE,
1381         .volatile_reg = ltr501_is_volatile_reg,
1382 };
1383
1384 static void ltr501_disable_regulators(void *d)
1385 {
1386         struct ltr501_data *data = d;
1387
1388         regulator_bulk_disable(ARRAY_SIZE(data->regulators), data->regulators);
1389 }
1390
1391 static int ltr501_powerdown(struct ltr501_data *data)
1392 {
1393         return ltr501_write_contr(data, data->als_contr &
1394                                   ~data->chip_info->als_mode_active,
1395                                   data->ps_contr & ~LTR501_CONTR_ACTIVE);
1396 }
1397
1398 static const char *ltr501_match_acpi_device(struct device *dev, int *chip_idx)
1399 {
1400         const struct acpi_device_id *id;
1401
1402         id = acpi_match_device(dev->driver->acpi_match_table, dev);
1403         if (!id)
1404                 return NULL;
1405         *chip_idx = id->driver_data;
1406         return dev_name(dev);
1407 }
1408
1409 static int ltr501_probe(struct i2c_client *client,
1410                         const struct i2c_device_id *id)
1411 {
1412         struct ltr501_data *data;
1413         struct iio_dev *indio_dev;
1414         struct regmap *regmap;
1415         int ret, partid, chip_idx = 0;
1416         const char *name = NULL;
1417
1418         indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
1419         if (!indio_dev)
1420                 return -ENOMEM;
1421
1422         regmap = devm_regmap_init_i2c(client, &ltr501_regmap_config);
1423         if (IS_ERR(regmap)) {
1424                 dev_err(&client->dev, "Regmap initialization failed.\n");
1425                 return PTR_ERR(regmap);
1426         }
1427
1428         data = iio_priv(indio_dev);
1429         i2c_set_clientdata(client, indio_dev);
1430         data->client = client;
1431         data->regmap = regmap;
1432         mutex_init(&data->lock_als);
1433         mutex_init(&data->lock_ps);
1434
1435         data->regulators[0].supply = "vdd";
1436         data->regulators[1].supply = "vddio";
1437         ret = devm_regulator_bulk_get(&client->dev,
1438                                       ARRAY_SIZE(data->regulators),
1439                                       data->regulators);
1440         if (ret)
1441                 return dev_err_probe(&client->dev, ret,
1442                                      "Failed to get regulators\n");
1443
1444         ret = regulator_bulk_enable(ARRAY_SIZE(data->regulators),
1445                                     data->regulators);
1446         if (ret)
1447                 return ret;
1448
1449         ret = devm_add_action_or_reset(&client->dev,
1450                                        ltr501_disable_regulators, data);
1451         if (ret)
1452                 return ret;
1453
1454         data->reg_it = devm_regmap_field_alloc(&client->dev, regmap,
1455                                                reg_field_it);
1456         if (IS_ERR(data->reg_it)) {
1457                 dev_err(&client->dev, "Integ time reg field init failed.\n");
1458                 return PTR_ERR(data->reg_it);
1459         }
1460
1461         data->reg_als_intr = devm_regmap_field_alloc(&client->dev, regmap,
1462                                                      reg_field_als_intr);
1463         if (IS_ERR(data->reg_als_intr)) {
1464                 dev_err(&client->dev, "ALS intr mode reg field init failed\n");
1465                 return PTR_ERR(data->reg_als_intr);
1466         }
1467
1468         data->reg_ps_intr = devm_regmap_field_alloc(&client->dev, regmap,
1469                                                     reg_field_ps_intr);
1470         if (IS_ERR(data->reg_ps_intr)) {
1471                 dev_err(&client->dev, "PS intr mode reg field init failed.\n");
1472                 return PTR_ERR(data->reg_ps_intr);
1473         }
1474
1475         data->reg_als_rate = devm_regmap_field_alloc(&client->dev, regmap,
1476                                                      reg_field_als_rate);
1477         if (IS_ERR(data->reg_als_rate)) {
1478                 dev_err(&client->dev, "ALS samp rate field init failed.\n");
1479                 return PTR_ERR(data->reg_als_rate);
1480         }
1481
1482         data->reg_ps_rate = devm_regmap_field_alloc(&client->dev, regmap,
1483                                                     reg_field_ps_rate);
1484         if (IS_ERR(data->reg_ps_rate)) {
1485                 dev_err(&client->dev, "PS samp rate field init failed.\n");
1486                 return PTR_ERR(data->reg_ps_rate);
1487         }
1488
1489         data->reg_als_prst = devm_regmap_field_alloc(&client->dev, regmap,
1490                                                      reg_field_als_prst);
1491         if (IS_ERR(data->reg_als_prst)) {
1492                 dev_err(&client->dev, "ALS prst reg field init failed\n");
1493                 return PTR_ERR(data->reg_als_prst);
1494         }
1495
1496         data->reg_ps_prst = devm_regmap_field_alloc(&client->dev, regmap,
1497                                                     reg_field_ps_prst);
1498         if (IS_ERR(data->reg_ps_prst)) {
1499                 dev_err(&client->dev, "PS prst reg field init failed.\n");
1500                 return PTR_ERR(data->reg_ps_prst);
1501         }
1502
1503         ret = regmap_read(data->regmap, LTR501_PART_ID, &partid);
1504         if (ret < 0)
1505                 return ret;
1506
1507         if (id) {
1508                 name = id->name;
1509                 chip_idx = id->driver_data;
1510         } else  if (ACPI_HANDLE(&client->dev)) {
1511                 name = ltr501_match_acpi_device(&client->dev, &chip_idx);
1512         } else {
1513                 return -ENODEV;
1514         }
1515
1516         data->chip_info = &ltr501_chip_info_tbl[chip_idx];
1517
1518         if ((partid >> 4) != data->chip_info->partid)
1519                 return -ENODEV;
1520
1521         indio_dev->info = data->chip_info->info;
1522         indio_dev->channels = data->chip_info->channels;
1523         indio_dev->num_channels = data->chip_info->no_channels;
1524         indio_dev->name = name;
1525         indio_dev->modes = INDIO_DIRECT_MODE;
1526
1527         ret = ltr501_init(data);
1528         if (ret < 0)
1529                 return ret;
1530
1531         if (client->irq > 0) {
1532                 ret = devm_request_threaded_irq(&client->dev, client->irq,
1533                                                 NULL, ltr501_interrupt_handler,
1534                                                 IRQF_TRIGGER_FALLING |
1535                                                 IRQF_ONESHOT,
1536                                                 "ltr501_thresh_event",
1537                                                 indio_dev);
1538                 if (ret) {
1539                         dev_err(&client->dev, "request irq (%d) failed\n",
1540                                 client->irq);
1541                         return ret;
1542                 }
1543         } else {
1544                 indio_dev->info = data->chip_info->info_no_irq;
1545         }
1546
1547         ret = iio_triggered_buffer_setup(indio_dev, NULL,
1548                                          ltr501_trigger_handler, NULL);
1549         if (ret)
1550                 goto powerdown_on_error;
1551
1552         ret = iio_device_register(indio_dev);
1553         if (ret)
1554                 goto error_unreg_buffer;
1555
1556         return 0;
1557
1558 error_unreg_buffer:
1559         iio_triggered_buffer_cleanup(indio_dev);
1560 powerdown_on_error:
1561         ltr501_powerdown(data);
1562         return ret;
1563 }
1564
1565 static int ltr501_remove(struct i2c_client *client)
1566 {
1567         struct iio_dev *indio_dev = i2c_get_clientdata(client);
1568
1569         iio_device_unregister(indio_dev);
1570         iio_triggered_buffer_cleanup(indio_dev);
1571         ltr501_powerdown(iio_priv(indio_dev));
1572
1573         return 0;
1574 }
1575
1576 #ifdef CONFIG_PM_SLEEP
1577 static int ltr501_suspend(struct device *dev)
1578 {
1579         struct ltr501_data *data = iio_priv(i2c_get_clientdata(
1580                                             to_i2c_client(dev)));
1581         return ltr501_powerdown(data);
1582 }
1583
1584 static int ltr501_resume(struct device *dev)
1585 {
1586         struct ltr501_data *data = iio_priv(i2c_get_clientdata(
1587                                             to_i2c_client(dev)));
1588
1589         return ltr501_write_contr(data, data->als_contr,
1590                 data->ps_contr);
1591 }
1592 #endif
1593
1594 static SIMPLE_DEV_PM_OPS(ltr501_pm_ops, ltr501_suspend, ltr501_resume);
1595
1596 static const struct acpi_device_id ltr_acpi_match[] = {
1597         {"LTER0501", ltr501},
1598         {"LTER0559", ltr559},
1599         {"LTER0301", ltr301},
1600         { },
1601 };
1602 MODULE_DEVICE_TABLE(acpi, ltr_acpi_match);
1603
1604 static const struct i2c_device_id ltr501_id[] = {
1605         { "ltr501", ltr501},
1606         { "ltr559", ltr559},
1607         { "ltr301", ltr301},
1608         { }
1609 };
1610 MODULE_DEVICE_TABLE(i2c, ltr501_id);
1611
1612 static const struct of_device_id ltr501_of_match[] = {
1613         { .compatible = "liteon,ltr501", },
1614         { .compatible = "liteon,ltr559", },
1615         { .compatible = "liteon,ltr301", },
1616         {}
1617 };
1618 MODULE_DEVICE_TABLE(of, ltr501_of_match);
1619
1620 static struct i2c_driver ltr501_driver = {
1621         .driver = {
1622                 .name   = LTR501_DRV_NAME,
1623                 .of_match_table = ltr501_of_match,
1624                 .pm     = &ltr501_pm_ops,
1625                 .acpi_match_table = ACPI_PTR(ltr_acpi_match),
1626         },
1627         .probe  = ltr501_probe,
1628         .remove = ltr501_remove,
1629         .id_table = ltr501_id,
1630 };
1631
1632 module_i2c_driver(ltr501_driver);
1633
1634 MODULE_AUTHOR("Peter Meerwald <[email protected]>");
1635 MODULE_DESCRIPTION("Lite-On LTR501 ambient light and proximity sensor driver");
1636 MODULE_LICENSE("GPL");