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