drivers/iio/light/ltr501.c

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