Linux Audio

Check our new training course

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