xref: /openbmc/linux/drivers/iio/imu/kmx61.c (revision 7f904d7e1f3ec7c2de47c024a5a5c30988b54703)
1  // SPDX-License-Identifier: GPL-2.0-only
2  /*
3   * KMX61 - Kionix 6-axis Accelerometer/Magnetometer
4   *
5   * Copyright (c) 2014, Intel Corporation.
6   *
7   * IIO driver for KMX61 (7-bit I2C slave address 0x0E or 0x0F).
8   */
9  
10  #include <linux/module.h>
11  #include <linux/i2c.h>
12  #include <linux/acpi.h>
13  #include <linux/interrupt.h>
14  #include <linux/pm.h>
15  #include <linux/pm_runtime.h>
16  #include <linux/iio/iio.h>
17  #include <linux/iio/sysfs.h>
18  #include <linux/iio/events.h>
19  #include <linux/iio/trigger.h>
20  #include <linux/iio/buffer.h>
21  #include <linux/iio/triggered_buffer.h>
22  #include <linux/iio/trigger_consumer.h>
23  
24  #define KMX61_DRV_NAME "kmx61"
25  #define KMX61_IRQ_NAME "kmx61_event"
26  
27  #define KMX61_REG_WHO_AM_I	0x00
28  #define KMX61_REG_INS1		0x01
29  #define KMX61_REG_INS2		0x02
30  
31  /*
32   * three 16-bit accelerometer output registers for X/Y/Z axis
33   * we use only XOUT_L as a base register, all other addresses
34   * can be obtained by applying an offset and are provided here
35   * only for clarity.
36   */
37  #define KMX61_ACC_XOUT_L	0x0A
38  #define KMX61_ACC_XOUT_H	0x0B
39  #define KMX61_ACC_YOUT_L	0x0C
40  #define KMX61_ACC_YOUT_H	0x0D
41  #define KMX61_ACC_ZOUT_L	0x0E
42  #define KMX61_ACC_ZOUT_H	0x0F
43  
44  /*
45   * one 16-bit temperature output register
46   */
47  #define KMX61_TEMP_L		0x10
48  #define KMX61_TEMP_H		0x11
49  
50  /*
51   * three 16-bit magnetometer output registers for X/Y/Z axis
52   */
53  #define KMX61_MAG_XOUT_L	0x12
54  #define KMX61_MAG_XOUT_H	0x13
55  #define KMX61_MAG_YOUT_L	0x14
56  #define KMX61_MAG_YOUT_H	0x15
57  #define KMX61_MAG_ZOUT_L	0x16
58  #define KMX61_MAG_ZOUT_H	0x17
59  
60  #define KMX61_REG_INL		0x28
61  #define KMX61_REG_STBY		0x29
62  #define KMX61_REG_CTRL1		0x2A
63  #define KMX61_REG_CTRL2		0x2B
64  #define KMX61_REG_ODCNTL	0x2C
65  #define KMX61_REG_INC1		0x2D
66  
67  #define KMX61_REG_WUF_THRESH	0x3D
68  #define KMX61_REG_WUF_TIMER	0x3E
69  
70  #define KMX61_ACC_STBY_BIT	BIT(0)
71  #define KMX61_MAG_STBY_BIT	BIT(1)
72  #define KMX61_ACT_STBY_BIT	BIT(7)
73  
74  #define KMX61_ALL_STBY		(KMX61_ACC_STBY_BIT | KMX61_MAG_STBY_BIT)
75  
76  #define KMX61_REG_INS1_BIT_WUFS		BIT(1)
77  
78  #define KMX61_REG_INS2_BIT_ZP		BIT(0)
79  #define KMX61_REG_INS2_BIT_ZN		BIT(1)
80  #define KMX61_REG_INS2_BIT_YP		BIT(2)
81  #define KMX61_REG_INS2_BIT_YN		BIT(3)
82  #define KMX61_REG_INS2_BIT_XP		BIT(4)
83  #define KMX61_REG_INS2_BIT_XN		BIT(5)
84  
85  #define KMX61_REG_CTRL1_GSEL_MASK	0x03
86  
87  #define KMX61_REG_CTRL1_BIT_RES		BIT(4)
88  #define KMX61_REG_CTRL1_BIT_DRDYE	BIT(5)
89  #define KMX61_REG_CTRL1_BIT_WUFE	BIT(6)
90  #define KMX61_REG_CTRL1_BIT_BTSE	BIT(7)
91  
92  #define KMX61_REG_INC1_BIT_WUFS		BIT(0)
93  #define KMX61_REG_INC1_BIT_DRDYM	BIT(1)
94  #define KMX61_REG_INC1_BIT_DRDYA	BIT(2)
95  #define KMX61_REG_INC1_BIT_IEN		BIT(5)
96  
97  #define KMX61_ACC_ODR_SHIFT	0
98  #define KMX61_MAG_ODR_SHIFT	4
99  #define KMX61_ACC_ODR_MASK	0x0F
100  #define KMX61_MAG_ODR_MASK	0xF0
101  
102  #define KMX61_OWUF_MASK		0x7
103  
104  #define KMX61_DEFAULT_WAKE_THRESH	1
105  #define KMX61_DEFAULT_WAKE_DURATION	1
106  
107  #define KMX61_SLEEP_DELAY_MS	2000
108  
109  #define KMX61_CHIP_ID		0x12
110  
111  /* KMX61 devices */
112  #define KMX61_ACC	0x01
113  #define KMX61_MAG	0x02
114  
115  struct kmx61_data {
116  	struct i2c_client *client;
117  
118  	/* serialize access to non-atomic ops, e.g set_mode */
119  	struct mutex lock;
120  
121  	/* standby state */
122  	bool acc_stby;
123  	bool mag_stby;
124  
125  	/* power state */
126  	bool acc_ps;
127  	bool mag_ps;
128  
129  	/* config bits */
130  	u8 range;
131  	u8 odr_bits;
132  	u8 wake_thresh;
133  	u8 wake_duration;
134  
135  	/* accelerometer specific data */
136  	struct iio_dev *acc_indio_dev;
137  	struct iio_trigger *acc_dready_trig;
138  	struct iio_trigger *motion_trig;
139  	bool acc_dready_trig_on;
140  	bool motion_trig_on;
141  	bool ev_enable_state;
142  
143  	/* magnetometer specific data */
144  	struct iio_dev *mag_indio_dev;
145  	struct iio_trigger *mag_dready_trig;
146  	bool mag_dready_trig_on;
147  };
148  
149  enum kmx61_range {
150  	KMX61_RANGE_2G,
151  	KMX61_RANGE_4G,
152  	KMX61_RANGE_8G,
153  };
154  
155  enum kmx61_axis {
156  	KMX61_AXIS_X,
157  	KMX61_AXIS_Y,
158  	KMX61_AXIS_Z,
159  };
160  
161  static const u16 kmx61_uscale_table[] = {9582, 19163, 38326};
162  
163  static const struct {
164  	int val;
165  	int val2;
166  } kmx61_samp_freq_table[] = { {12, 500000},
167  			{25, 0},
168  			{50, 0},
169  			{100, 0},
170  			{200, 0},
171  			{400, 0},
172  			{800, 0},
173  			{1600, 0},
174  			{0, 781000},
175  			{1, 563000},
176  			{3, 125000},
177  			{6, 250000} };
178  
179  static const struct {
180  	int val;
181  	int val2;
182  	int odr_bits;
183  } kmx61_wake_up_odr_table[] = { {0, 781000, 0x00},
184  				 {1, 563000, 0x01},
185  				 {3, 125000, 0x02},
186  				 {6, 250000, 0x03},
187  				 {12, 500000, 0x04},
188  				 {25, 0, 0x05},
189  				 {50, 0, 0x06},
190  				 {100, 0, 0x06},
191  				 {200, 0, 0x06},
192  				 {400, 0, 0x06},
193  				 {800, 0, 0x06},
194  				 {1600, 0, 0x06} };
195  
196  static IIO_CONST_ATTR(accel_scale_available, "0.009582 0.019163 0.038326");
197  static IIO_CONST_ATTR(magn_scale_available, "0.001465");
198  static IIO_CONST_ATTR_SAMP_FREQ_AVAIL(
199  	"0.781000 1.563000 3.125000 6.250000 12.500000 25 50 100 200 400 800");
200  
201  static struct attribute *kmx61_acc_attributes[] = {
202  	&iio_const_attr_accel_scale_available.dev_attr.attr,
203  	&iio_const_attr_sampling_frequency_available.dev_attr.attr,
204  	NULL,
205  };
206  
207  static struct attribute *kmx61_mag_attributes[] = {
208  	&iio_const_attr_magn_scale_available.dev_attr.attr,
209  	&iio_const_attr_sampling_frequency_available.dev_attr.attr,
210  	NULL,
211  };
212  
213  static const struct attribute_group kmx61_acc_attribute_group = {
214  	.attrs = kmx61_acc_attributes,
215  };
216  
217  static const struct attribute_group kmx61_mag_attribute_group = {
218  	.attrs = kmx61_mag_attributes,
219  };
220  
221  static const struct iio_event_spec kmx61_event = {
222  	.type = IIO_EV_TYPE_THRESH,
223  	.dir = IIO_EV_DIR_EITHER,
224  	.mask_separate = BIT(IIO_EV_INFO_VALUE) |
225  			 BIT(IIO_EV_INFO_ENABLE) |
226  			 BIT(IIO_EV_INFO_PERIOD),
227  };
228  
229  #define KMX61_ACC_CHAN(_axis) { \
230  	.type = IIO_ACCEL, \
231  	.modified = 1, \
232  	.channel2 = IIO_MOD_ ## _axis, \
233  	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW), \
234  	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE) | \
235  				BIT(IIO_CHAN_INFO_SAMP_FREQ), \
236  	.address = KMX61_ACC, \
237  	.scan_index = KMX61_AXIS_ ## _axis, \
238  	.scan_type = { \
239  		.sign = 's', \
240  		.realbits = 12, \
241  		.storagebits = 16, \
242  		.shift = 4, \
243  		.endianness = IIO_LE, \
244  	}, \
245  	.event_spec = &kmx61_event, \
246  	.num_event_specs = 1 \
247  }
248  
249  #define KMX61_MAG_CHAN(_axis) { \
250  	.type = IIO_MAGN, \
251  	.modified = 1, \
252  	.channel2 = IIO_MOD_ ## _axis, \
253  	.address = KMX61_MAG, \
254  	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW), \
255  	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE) | \
256  				BIT(IIO_CHAN_INFO_SAMP_FREQ), \
257  	.scan_index = KMX61_AXIS_ ## _axis, \
258  	.scan_type = { \
259  		.sign = 's', \
260  		.realbits = 14, \
261  		.storagebits = 16, \
262  		.shift = 2, \
263  		.endianness = IIO_LE, \
264  	}, \
265  }
266  
267  static const struct iio_chan_spec kmx61_acc_channels[] = {
268  	KMX61_ACC_CHAN(X),
269  	KMX61_ACC_CHAN(Y),
270  	KMX61_ACC_CHAN(Z),
271  };
272  
273  static const struct iio_chan_spec kmx61_mag_channels[] = {
274  	KMX61_MAG_CHAN(X),
275  	KMX61_MAG_CHAN(Y),
276  	KMX61_MAG_CHAN(Z),
277  };
278  
279  static void kmx61_set_data(struct iio_dev *indio_dev, struct kmx61_data *data)
280  {
281  	struct kmx61_data **priv = iio_priv(indio_dev);
282  
283  	*priv = data;
284  }
285  
286  static struct kmx61_data *kmx61_get_data(struct iio_dev *indio_dev)
287  {
288  	return *(struct kmx61_data **)iio_priv(indio_dev);
289  }
290  
291  static int kmx61_convert_freq_to_bit(int val, int val2)
292  {
293  	int i;
294  
295  	for (i = 0; i < ARRAY_SIZE(kmx61_samp_freq_table); i++)
296  		if (val == kmx61_samp_freq_table[i].val &&
297  		    val2 == kmx61_samp_freq_table[i].val2)
298  			return i;
299  	return -EINVAL;
300  }
301  
302  static int kmx61_convert_wake_up_odr_to_bit(int val, int val2)
303  {
304  	int i;
305  
306  	for (i = 0; i < ARRAY_SIZE(kmx61_wake_up_odr_table); ++i)
307  		if (kmx61_wake_up_odr_table[i].val == val &&
308  			kmx61_wake_up_odr_table[i].val2 == val2)
309  				return kmx61_wake_up_odr_table[i].odr_bits;
310  	return -EINVAL;
311  }
312  
313  /**
314   * kmx61_set_mode() - set KMX61 device operating mode
315   * @data - kmx61 device private data pointer
316   * @mode - bitmask, indicating operating mode for @device
317   * @device - bitmask, indicating device for which @mode needs to be set
318   * @update - update stby bits stored in device's private  @data
319   *
320   * For each sensor (accelerometer/magnetometer) there are two operating modes
321   * STANDBY and OPERATION. Neither accel nor magn can be disabled independently
322   * if they are both enabled. Internal sensors state is saved in acc_stby and
323   * mag_stby members of driver's private @data.
324   */
325  static int kmx61_set_mode(struct kmx61_data *data, u8 mode, u8 device,
326  			  bool update)
327  {
328  	int ret;
329  	int acc_stby = -1, mag_stby = -1;
330  
331  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_STBY);
332  	if (ret < 0) {
333  		dev_err(&data->client->dev, "Error reading reg_stby\n");
334  		return ret;
335  	}
336  	if (device & KMX61_ACC) {
337  		if (mode & KMX61_ACC_STBY_BIT) {
338  			ret |= KMX61_ACC_STBY_BIT;
339  			acc_stby = 1;
340  		} else {
341  			ret &= ~KMX61_ACC_STBY_BIT;
342  			acc_stby = 0;
343  		}
344  	}
345  
346  	if (device & KMX61_MAG) {
347  		if (mode & KMX61_MAG_STBY_BIT) {
348  			ret |= KMX61_MAG_STBY_BIT;
349  			mag_stby = 1;
350  		} else {
351  			ret &= ~KMX61_MAG_STBY_BIT;
352  			mag_stby = 0;
353  		}
354  	}
355  
356  	if (mode & KMX61_ACT_STBY_BIT)
357  		ret |= KMX61_ACT_STBY_BIT;
358  
359  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_STBY, ret);
360  	if (ret < 0) {
361  		dev_err(&data->client->dev, "Error writing reg_stby\n");
362  		return ret;
363  	}
364  
365  	if (acc_stby != -1 && update)
366  		data->acc_stby = acc_stby;
367  	if (mag_stby != -1 && update)
368  		data->mag_stby = mag_stby;
369  
370  	return 0;
371  }
372  
373  static int kmx61_get_mode(struct kmx61_data *data, u8 *mode, u8 device)
374  {
375  	int ret;
376  
377  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_STBY);
378  	if (ret < 0) {
379  		dev_err(&data->client->dev, "Error reading reg_stby\n");
380  		return ret;
381  	}
382  	*mode = 0;
383  
384  	if (device & KMX61_ACC) {
385  		if (ret & KMX61_ACC_STBY_BIT)
386  			*mode |= KMX61_ACC_STBY_BIT;
387  		else
388  			*mode &= ~KMX61_ACC_STBY_BIT;
389  	}
390  
391  	if (device & KMX61_MAG) {
392  		if (ret & KMX61_MAG_STBY_BIT)
393  			*mode |= KMX61_MAG_STBY_BIT;
394  		else
395  			*mode &= ~KMX61_MAG_STBY_BIT;
396  	}
397  
398  	return 0;
399  }
400  
401  static int kmx61_set_wake_up_odr(struct kmx61_data *data, int val, int val2)
402  {
403  	int ret, odr_bits;
404  
405  	odr_bits = kmx61_convert_wake_up_odr_to_bit(val, val2);
406  	if (odr_bits < 0)
407  		return odr_bits;
408  
409  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_CTRL2,
410  					odr_bits);
411  	if (ret < 0)
412  		dev_err(&data->client->dev, "Error writing reg_ctrl2\n");
413  	return ret;
414  }
415  
416  static int kmx61_set_odr(struct kmx61_data *data, int val, int val2, u8 device)
417  {
418  	int ret;
419  	u8 mode;
420  	int lodr_bits, odr_bits;
421  
422  	ret = kmx61_get_mode(data, &mode, KMX61_ACC | KMX61_MAG);
423  	if (ret < 0)
424  		return ret;
425  
426  	lodr_bits = kmx61_convert_freq_to_bit(val, val2);
427  	if (lodr_bits < 0)
428  		return lodr_bits;
429  
430  	/* To change ODR, accel and magn must be in STDBY */
431  	ret = kmx61_set_mode(data, KMX61_ALL_STBY, KMX61_ACC | KMX61_MAG,
432  			     true);
433  	if (ret < 0)
434  		return ret;
435  
436  	odr_bits = 0;
437  	if (device & KMX61_ACC)
438  		odr_bits |= lodr_bits << KMX61_ACC_ODR_SHIFT;
439  	if (device & KMX61_MAG)
440  		odr_bits |= lodr_bits << KMX61_MAG_ODR_SHIFT;
441  
442  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_ODCNTL,
443  					odr_bits);
444  	if (ret < 0)
445  		return ret;
446  
447  	data->odr_bits = odr_bits;
448  
449  	if (device & KMX61_ACC) {
450  		ret = kmx61_set_wake_up_odr(data, val, val2);
451  		if (ret)
452  			return ret;
453  	}
454  
455  	return kmx61_set_mode(data, mode, KMX61_ACC | KMX61_MAG, true);
456  }
457  
458  static int kmx61_get_odr(struct kmx61_data *data, int *val, int *val2,
459  			 u8 device)
460  {
461  	u8 lodr_bits;
462  
463  	if (device & KMX61_ACC)
464  		lodr_bits = (data->odr_bits >> KMX61_ACC_ODR_SHIFT) &
465  			     KMX61_ACC_ODR_MASK;
466  	else if (device & KMX61_MAG)
467  		lodr_bits = (data->odr_bits >> KMX61_MAG_ODR_SHIFT) &
468  			     KMX61_MAG_ODR_MASK;
469  	else
470  		return -EINVAL;
471  
472  	if (lodr_bits >= ARRAY_SIZE(kmx61_samp_freq_table))
473  		return -EINVAL;
474  
475  	*val = kmx61_samp_freq_table[lodr_bits].val;
476  	*val2 = kmx61_samp_freq_table[lodr_bits].val2;
477  
478  	return 0;
479  }
480  
481  static int kmx61_set_range(struct kmx61_data *data, u8 range)
482  {
483  	int ret;
484  
485  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_CTRL1);
486  	if (ret < 0) {
487  		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
488  		return ret;
489  	}
490  
491  	ret &= ~KMX61_REG_CTRL1_GSEL_MASK;
492  	ret |= range & KMX61_REG_CTRL1_GSEL_MASK;
493  
494  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_CTRL1, ret);
495  	if (ret < 0) {
496  		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
497  		return ret;
498  	}
499  
500  	data->range = range;
501  
502  	return 0;
503  }
504  
505  static int kmx61_set_scale(struct kmx61_data *data, u16 uscale)
506  {
507  	int ret, i;
508  	u8  mode;
509  
510  	for (i = 0; i < ARRAY_SIZE(kmx61_uscale_table); i++) {
511  		if (kmx61_uscale_table[i] == uscale) {
512  			ret = kmx61_get_mode(data, &mode,
513  					     KMX61_ACC | KMX61_MAG);
514  			if (ret < 0)
515  				return ret;
516  
517  			ret = kmx61_set_mode(data, KMX61_ALL_STBY,
518  					     KMX61_ACC | KMX61_MAG, true);
519  			if (ret < 0)
520  				return ret;
521  
522  			ret = kmx61_set_range(data, i);
523  			if (ret < 0)
524  				return ret;
525  
526  			return  kmx61_set_mode(data, mode,
527  					       KMX61_ACC | KMX61_MAG, true);
528  		}
529  	}
530  	return -EINVAL;
531  }
532  
533  static int kmx61_chip_init(struct kmx61_data *data)
534  {
535  	int ret, val, val2;
536  
537  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_WHO_AM_I);
538  	if (ret < 0) {
539  		dev_err(&data->client->dev, "Error reading who_am_i\n");
540  		return ret;
541  	}
542  
543  	if (ret != KMX61_CHIP_ID) {
544  		dev_err(&data->client->dev,
545  			"Wrong chip id, got %x expected %x\n",
546  			 ret, KMX61_CHIP_ID);
547  		return -EINVAL;
548  	}
549  
550  	/* set accel 12bit, 4g range */
551  	ret = kmx61_set_range(data, KMX61_RANGE_4G);
552  	if (ret < 0)
553  		return ret;
554  
555  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_ODCNTL);
556  	if (ret < 0) {
557  		dev_err(&data->client->dev, "Error reading reg_odcntl\n");
558  		return ret;
559  	}
560  	data->odr_bits = ret;
561  
562  	/*
563  	 * set output data rate for wake up (motion detection) function
564  	 * to match data rate for accelerometer sampling
565  	 */
566  	ret = kmx61_get_odr(data, &val, &val2, KMX61_ACC);
567  	if (ret < 0)
568  		return ret;
569  
570  	ret = kmx61_set_wake_up_odr(data, val, val2);
571  	if (ret < 0)
572  		return ret;
573  
574  	/* set acc/magn to OPERATION mode */
575  	ret = kmx61_set_mode(data, 0, KMX61_ACC | KMX61_MAG, true);
576  	if (ret < 0)
577  		return ret;
578  
579  	data->wake_thresh = KMX61_DEFAULT_WAKE_THRESH;
580  	data->wake_duration = KMX61_DEFAULT_WAKE_DURATION;
581  
582  	return 0;
583  }
584  
585  static int kmx61_setup_new_data_interrupt(struct kmx61_data *data,
586  					  bool status, u8 device)
587  {
588  	u8 mode;
589  	int ret;
590  
591  	ret = kmx61_get_mode(data, &mode, KMX61_ACC | KMX61_MAG);
592  	if (ret < 0)
593  		return ret;
594  
595  	ret = kmx61_set_mode(data, KMX61_ALL_STBY, KMX61_ACC | KMX61_MAG, true);
596  	if (ret < 0)
597  		return ret;
598  
599  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_INC1);
600  	if (ret < 0) {
601  		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
602  		return ret;
603  	}
604  
605  	if (status) {
606  		ret |= KMX61_REG_INC1_BIT_IEN;
607  		if (device & KMX61_ACC)
608  			ret |= KMX61_REG_INC1_BIT_DRDYA;
609  		if (device & KMX61_MAG)
610  			ret |=  KMX61_REG_INC1_BIT_DRDYM;
611  	} else {
612  		ret &= ~KMX61_REG_INC1_BIT_IEN;
613  		if (device & KMX61_ACC)
614  			ret &= ~KMX61_REG_INC1_BIT_DRDYA;
615  		if (device & KMX61_MAG)
616  			ret &= ~KMX61_REG_INC1_BIT_DRDYM;
617  	}
618  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_INC1, ret);
619  	if (ret < 0) {
620  		dev_err(&data->client->dev, "Error writing reg_int_ctrl1\n");
621  		return ret;
622  	}
623  
624  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_CTRL1);
625  	if (ret < 0) {
626  		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
627  		return ret;
628  	}
629  
630  	if (status)
631  		ret |= KMX61_REG_CTRL1_BIT_DRDYE;
632  	else
633  		ret &= ~KMX61_REG_CTRL1_BIT_DRDYE;
634  
635  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_CTRL1, ret);
636  	if (ret < 0) {
637  		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
638  		return ret;
639  	}
640  
641  	return kmx61_set_mode(data, mode, KMX61_ACC | KMX61_MAG, true);
642  }
643  
644  static int kmx61_chip_update_thresholds(struct kmx61_data *data)
645  {
646  	int ret;
647  
648  	ret = i2c_smbus_write_byte_data(data->client,
649  					KMX61_REG_WUF_TIMER,
650  					data->wake_duration);
651  	if (ret < 0) {
652  		dev_err(&data->client->dev, "Errow writing reg_wuf_timer\n");
653  		return ret;
654  	}
655  
656  	ret = i2c_smbus_write_byte_data(data->client,
657  					KMX61_REG_WUF_THRESH,
658  					data->wake_thresh);
659  	if (ret < 0)
660  		dev_err(&data->client->dev, "Error writing reg_wuf_thresh\n");
661  
662  	return ret;
663  }
664  
665  static int kmx61_setup_any_motion_interrupt(struct kmx61_data *data,
666  					    bool status)
667  {
668  	u8 mode;
669  	int ret;
670  
671  	ret = kmx61_get_mode(data, &mode, KMX61_ACC | KMX61_MAG);
672  	if (ret < 0)
673  		return ret;
674  
675  	ret = kmx61_set_mode(data, KMX61_ALL_STBY, KMX61_ACC | KMX61_MAG, true);
676  	if (ret < 0)
677  		return ret;
678  
679  	ret = kmx61_chip_update_thresholds(data);
680  	if (ret < 0)
681  		return ret;
682  
683  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_INC1);
684  	if (ret < 0) {
685  		dev_err(&data->client->dev, "Error reading reg_inc1\n");
686  		return ret;
687  	}
688  	if (status)
689  		ret |= (KMX61_REG_INC1_BIT_IEN | KMX61_REG_INC1_BIT_WUFS);
690  	else
691  		ret &= ~(KMX61_REG_INC1_BIT_IEN | KMX61_REG_INC1_BIT_WUFS);
692  
693  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_INC1, ret);
694  	if (ret < 0) {
695  		dev_err(&data->client->dev, "Error writing reg_inc1\n");
696  		return ret;
697  	}
698  
699  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_CTRL1);
700  	if (ret < 0) {
701  		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
702  		return ret;
703  	}
704  
705  	if (status)
706  		ret |= KMX61_REG_CTRL1_BIT_WUFE | KMX61_REG_CTRL1_BIT_BTSE;
707  	else
708  		ret &= ~(KMX61_REG_CTRL1_BIT_WUFE | KMX61_REG_CTRL1_BIT_BTSE);
709  
710  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_CTRL1, ret);
711  	if (ret < 0) {
712  		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
713  		return ret;
714  	}
715  	mode |= KMX61_ACT_STBY_BIT;
716  	return kmx61_set_mode(data, mode, KMX61_ACC | KMX61_MAG, true);
717  }
718  
719  /**
720   * kmx61_set_power_state() - set power state for kmx61 @device
721   * @data - kmx61 device private pointer
722   * @on - power state to be set for @device
723   * @device - bitmask indicating device for which @on state needs to be set
724   *
725   * Notice that when ACC power state needs to be set to ON and MAG is in
726   * OPERATION then we know that kmx61_runtime_resume was already called
727   * so we must set ACC OPERATION mode here. The same happens when MAG power
728   * state needs to be set to ON and ACC is in OPERATION.
729   */
730  static int kmx61_set_power_state(struct kmx61_data *data, bool on, u8 device)
731  {
732  #ifdef CONFIG_PM
733  	int ret;
734  
735  	if (device & KMX61_ACC) {
736  		if (on && !data->acc_ps && !data->mag_stby) {
737  			ret = kmx61_set_mode(data, 0, KMX61_ACC, true);
738  			if (ret < 0)
739  				return ret;
740  		}
741  		data->acc_ps = on;
742  	}
743  	if (device & KMX61_MAG) {
744  		if (on && !data->mag_ps && !data->acc_stby) {
745  			ret = kmx61_set_mode(data, 0, KMX61_MAG, true);
746  			if (ret < 0)
747  				return ret;
748  		}
749  		data->mag_ps = on;
750  	}
751  
752  	if (on) {
753  		ret = pm_runtime_get_sync(&data->client->dev);
754  	} else {
755  		pm_runtime_mark_last_busy(&data->client->dev);
756  		ret = pm_runtime_put_autosuspend(&data->client->dev);
757  	}
758  	if (ret < 0) {
759  		dev_err(&data->client->dev,
760  			"Failed: kmx61_set_power_state for %d, ret %d\n",
761  			on, ret);
762  		if (on)
763  			pm_runtime_put_noidle(&data->client->dev);
764  
765  		return ret;
766  	}
767  #endif
768  	return 0;
769  }
770  
771  static int kmx61_read_measurement(struct kmx61_data *data, u8 base, u8 offset)
772  {
773  	int ret;
774  	u8 reg = base + offset * 2;
775  
776  	ret = i2c_smbus_read_word_data(data->client, reg);
777  	if (ret < 0)
778  		dev_err(&data->client->dev, "failed to read reg at %x\n", reg);
779  
780  	return ret;
781  }
782  
783  static int kmx61_read_raw(struct iio_dev *indio_dev,
784  			  struct iio_chan_spec const *chan, int *val,
785  			  int *val2, long mask)
786  {
787  	int ret;
788  	u8 base_reg;
789  	struct kmx61_data *data = kmx61_get_data(indio_dev);
790  
791  	switch (mask) {
792  	case IIO_CHAN_INFO_RAW:
793  		switch (chan->type) {
794  		case IIO_ACCEL:
795  			base_reg = KMX61_ACC_XOUT_L;
796  			break;
797  		case IIO_MAGN:
798  			base_reg = KMX61_MAG_XOUT_L;
799  			break;
800  		default:
801  			return -EINVAL;
802  		}
803  		mutex_lock(&data->lock);
804  
805  		ret = kmx61_set_power_state(data, true, chan->address);
806  		if (ret) {
807  			mutex_unlock(&data->lock);
808  			return ret;
809  		}
810  
811  		ret = kmx61_read_measurement(data, base_reg, chan->scan_index);
812  		if (ret < 0) {
813  			kmx61_set_power_state(data, false, chan->address);
814  			mutex_unlock(&data->lock);
815  			return ret;
816  		}
817  		*val = sign_extend32(ret >> chan->scan_type.shift,
818  				     chan->scan_type.realbits - 1);
819  		ret = kmx61_set_power_state(data, false, chan->address);
820  
821  		mutex_unlock(&data->lock);
822  		if (ret)
823  			return ret;
824  		return IIO_VAL_INT;
825  	case IIO_CHAN_INFO_SCALE:
826  		switch (chan->type) {
827  		case IIO_ACCEL:
828  			*val = 0;
829  			*val2 = kmx61_uscale_table[data->range];
830  			return IIO_VAL_INT_PLUS_MICRO;
831  		case IIO_MAGN:
832  			/* 14 bits res, 1465 microGauss per magn count */
833  			*val = 0;
834  			*val2 = 1465;
835  			return IIO_VAL_INT_PLUS_MICRO;
836  		default:
837  			return -EINVAL;
838  		}
839  	case IIO_CHAN_INFO_SAMP_FREQ:
840  		if (chan->type != IIO_ACCEL && chan->type != IIO_MAGN)
841  			return -EINVAL;
842  
843  		mutex_lock(&data->lock);
844  		ret = kmx61_get_odr(data, val, val2, chan->address);
845  		mutex_unlock(&data->lock);
846  		if (ret)
847  			return -EINVAL;
848  		return IIO_VAL_INT_PLUS_MICRO;
849  	}
850  	return -EINVAL;
851  }
852  
853  static int kmx61_write_raw(struct iio_dev *indio_dev,
854  			   struct iio_chan_spec const *chan, int val,
855  			   int val2, long mask)
856  {
857  	int ret;
858  	struct kmx61_data *data = kmx61_get_data(indio_dev);
859  
860  	switch (mask) {
861  	case IIO_CHAN_INFO_SAMP_FREQ:
862  		if (chan->type != IIO_ACCEL && chan->type != IIO_MAGN)
863  			return -EINVAL;
864  
865  		mutex_lock(&data->lock);
866  		ret = kmx61_set_odr(data, val, val2, chan->address);
867  		mutex_unlock(&data->lock);
868  		return ret;
869  	case IIO_CHAN_INFO_SCALE:
870  		switch (chan->type) {
871  		case IIO_ACCEL:
872  			if (val != 0)
873  				return -EINVAL;
874  			mutex_lock(&data->lock);
875  			ret = kmx61_set_scale(data, val2);
876  			mutex_unlock(&data->lock);
877  			return ret;
878  		default:
879  			return -EINVAL;
880  		}
881  	default:
882  		return -EINVAL;
883  	}
884  }
885  
886  static int kmx61_read_event(struct iio_dev *indio_dev,
887  			    const struct iio_chan_spec *chan,
888  			    enum iio_event_type type,
889  			    enum iio_event_direction dir,
890  			    enum iio_event_info info,
891  			    int *val, int *val2)
892  {
893  	struct kmx61_data *data = kmx61_get_data(indio_dev);
894  
895  	*val2 = 0;
896  	switch (info) {
897  	case IIO_EV_INFO_VALUE:
898  		*val = data->wake_thresh;
899  		return IIO_VAL_INT;
900  	case IIO_EV_INFO_PERIOD:
901  		*val = data->wake_duration;
902  		return IIO_VAL_INT;
903  	default:
904  		return -EINVAL;
905  	}
906  }
907  
908  static int kmx61_write_event(struct iio_dev *indio_dev,
909  			     const struct iio_chan_spec *chan,
910  			     enum iio_event_type type,
911  			     enum iio_event_direction dir,
912  			     enum iio_event_info info,
913  			     int val, int val2)
914  {
915  	struct kmx61_data *data = kmx61_get_data(indio_dev);
916  
917  	if (data->ev_enable_state)
918  		return -EBUSY;
919  
920  	switch (info) {
921  	case IIO_EV_INFO_VALUE:
922  		data->wake_thresh = val;
923  		return IIO_VAL_INT;
924  	case IIO_EV_INFO_PERIOD:
925  		data->wake_duration = val;
926  		return IIO_VAL_INT;
927  	default:
928  		return -EINVAL;
929  	}
930  }
931  
932  static int kmx61_read_event_config(struct iio_dev *indio_dev,
933  				   const struct iio_chan_spec *chan,
934  				   enum iio_event_type type,
935  				   enum iio_event_direction dir)
936  {
937  	struct kmx61_data *data = kmx61_get_data(indio_dev);
938  
939  	return data->ev_enable_state;
940  }
941  
942  static int kmx61_write_event_config(struct iio_dev *indio_dev,
943  				    const struct iio_chan_spec *chan,
944  				    enum iio_event_type type,
945  				    enum iio_event_direction dir,
946  				    int state)
947  {
948  	struct kmx61_data *data = kmx61_get_data(indio_dev);
949  	int ret = 0;
950  
951  	if (state && data->ev_enable_state)
952  		return 0;
953  
954  	mutex_lock(&data->lock);
955  
956  	if (!state && data->motion_trig_on) {
957  		data->ev_enable_state = false;
958  		goto err_unlock;
959  	}
960  
961  	ret = kmx61_set_power_state(data, state, KMX61_ACC);
962  	if (ret < 0)
963  		goto err_unlock;
964  
965  	ret = kmx61_setup_any_motion_interrupt(data, state);
966  	if (ret < 0) {
967  		kmx61_set_power_state(data, false, KMX61_ACC);
968  		goto err_unlock;
969  	}
970  
971  	data->ev_enable_state = state;
972  
973  err_unlock:
974  	mutex_unlock(&data->lock);
975  
976  	return ret;
977  }
978  
979  static int kmx61_acc_validate_trigger(struct iio_dev *indio_dev,
980  				      struct iio_trigger *trig)
981  {
982  	struct kmx61_data *data = kmx61_get_data(indio_dev);
983  
984  	if (data->acc_dready_trig != trig && data->motion_trig != trig)
985  		return -EINVAL;
986  
987  	return 0;
988  }
989  
990  static int kmx61_mag_validate_trigger(struct iio_dev *indio_dev,
991  				      struct iio_trigger *trig)
992  {
993  	struct kmx61_data *data = kmx61_get_data(indio_dev);
994  
995  	if (data->mag_dready_trig != trig)
996  		return -EINVAL;
997  
998  	return 0;
999  }
1000  
1001  static const struct iio_info kmx61_acc_info = {
1002  	.read_raw		= kmx61_read_raw,
1003  	.write_raw		= kmx61_write_raw,
1004  	.attrs			= &kmx61_acc_attribute_group,
1005  	.read_event_value	= kmx61_read_event,
1006  	.write_event_value	= kmx61_write_event,
1007  	.read_event_config	= kmx61_read_event_config,
1008  	.write_event_config	= kmx61_write_event_config,
1009  	.validate_trigger	= kmx61_acc_validate_trigger,
1010  };
1011  
1012  static const struct iio_info kmx61_mag_info = {
1013  	.read_raw		= kmx61_read_raw,
1014  	.write_raw		= kmx61_write_raw,
1015  	.attrs			= &kmx61_mag_attribute_group,
1016  	.validate_trigger	= kmx61_mag_validate_trigger,
1017  };
1018  
1019  
1020  static int kmx61_data_rdy_trigger_set_state(struct iio_trigger *trig,
1021  					    bool state)
1022  {
1023  	int ret = 0;
1024  	u8 device;
1025  
1026  	struct iio_dev *indio_dev = iio_trigger_get_drvdata(trig);
1027  	struct kmx61_data *data = kmx61_get_data(indio_dev);
1028  
1029  	mutex_lock(&data->lock);
1030  
1031  	if (!state && data->ev_enable_state && data->motion_trig_on) {
1032  		data->motion_trig_on = false;
1033  		goto err_unlock;
1034  	}
1035  
1036  	if (data->acc_dready_trig == trig || data->motion_trig == trig)
1037  		device = KMX61_ACC;
1038  	else
1039  		device = KMX61_MAG;
1040  
1041  	ret = kmx61_set_power_state(data, state, device);
1042  	if (ret < 0)
1043  		goto err_unlock;
1044  
1045  	if (data->acc_dready_trig == trig || data->mag_dready_trig == trig)
1046  		ret = kmx61_setup_new_data_interrupt(data, state, device);
1047  	else
1048  		ret = kmx61_setup_any_motion_interrupt(data, state);
1049  	if (ret < 0) {
1050  		kmx61_set_power_state(data, false, device);
1051  		goto err_unlock;
1052  	}
1053  
1054  	if (data->acc_dready_trig == trig)
1055  		data->acc_dready_trig_on = state;
1056  	else if (data->mag_dready_trig == trig)
1057  		data->mag_dready_trig_on = state;
1058  	else
1059  		data->motion_trig_on = state;
1060  err_unlock:
1061  	mutex_unlock(&data->lock);
1062  
1063  	return ret;
1064  }
1065  
1066  static int kmx61_trig_try_reenable(struct iio_trigger *trig)
1067  {
1068  	struct iio_dev *indio_dev = iio_trigger_get_drvdata(trig);
1069  	struct kmx61_data *data = kmx61_get_data(indio_dev);
1070  	int ret;
1071  
1072  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_INL);
1073  	if (ret < 0) {
1074  		dev_err(&data->client->dev, "Error reading reg_inl\n");
1075  		return ret;
1076  	}
1077  
1078  	return 0;
1079  }
1080  
1081  static const struct iio_trigger_ops kmx61_trigger_ops = {
1082  	.set_trigger_state = kmx61_data_rdy_trigger_set_state,
1083  	.try_reenable = kmx61_trig_try_reenable,
1084  };
1085  
1086  static irqreturn_t kmx61_event_handler(int irq, void *private)
1087  {
1088  	struct kmx61_data *data = private;
1089  	struct iio_dev *indio_dev = data->acc_indio_dev;
1090  	int ret;
1091  
1092  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_INS1);
1093  	if (ret < 0) {
1094  		dev_err(&data->client->dev, "Error reading reg_ins1\n");
1095  		goto ack_intr;
1096  	}
1097  
1098  	if (ret & KMX61_REG_INS1_BIT_WUFS) {
1099  		ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_INS2);
1100  		if (ret < 0) {
1101  			dev_err(&data->client->dev, "Error reading reg_ins2\n");
1102  			goto ack_intr;
1103  		}
1104  
1105  		if (ret & KMX61_REG_INS2_BIT_XN)
1106  			iio_push_event(indio_dev,
1107  				       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1108  				       0,
1109  				       IIO_MOD_X,
1110  				       IIO_EV_TYPE_THRESH,
1111  				       IIO_EV_DIR_FALLING),
1112  				       0);
1113  
1114  		if (ret & KMX61_REG_INS2_BIT_XP)
1115  			iio_push_event(indio_dev,
1116  				       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1117  				       0,
1118  				       IIO_MOD_X,
1119  				       IIO_EV_TYPE_THRESH,
1120  				       IIO_EV_DIR_RISING),
1121  				       0);
1122  
1123  		if (ret & KMX61_REG_INS2_BIT_YN)
1124  			iio_push_event(indio_dev,
1125  				       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1126  				       0,
1127  				       IIO_MOD_Y,
1128  				       IIO_EV_TYPE_THRESH,
1129  				       IIO_EV_DIR_FALLING),
1130  				       0);
1131  
1132  		if (ret & KMX61_REG_INS2_BIT_YP)
1133  			iio_push_event(indio_dev,
1134  				       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1135  				       0,
1136  				       IIO_MOD_Y,
1137  				       IIO_EV_TYPE_THRESH,
1138  				       IIO_EV_DIR_RISING),
1139  				       0);
1140  
1141  		if (ret & KMX61_REG_INS2_BIT_ZN)
1142  			iio_push_event(indio_dev,
1143  				       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1144  				       0,
1145  				       IIO_MOD_Z,
1146  				       IIO_EV_TYPE_THRESH,
1147  				       IIO_EV_DIR_FALLING),
1148  				       0);
1149  
1150  		if (ret & KMX61_REG_INS2_BIT_ZP)
1151  			iio_push_event(indio_dev,
1152  				       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1153  				       0,
1154  				       IIO_MOD_Z,
1155  				       IIO_EV_TYPE_THRESH,
1156  				       IIO_EV_DIR_RISING),
1157  				       0);
1158  	}
1159  
1160  ack_intr:
1161  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_CTRL1);
1162  	if (ret < 0)
1163  		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
1164  
1165  	ret |= KMX61_REG_CTRL1_BIT_RES;
1166  	ret = i2c_smbus_write_byte_data(data->client, KMX61_REG_CTRL1, ret);
1167  	if (ret < 0)
1168  		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
1169  
1170  	ret = i2c_smbus_read_byte_data(data->client, KMX61_REG_INL);
1171  	if (ret < 0)
1172  		dev_err(&data->client->dev, "Error reading reg_inl\n");
1173  
1174  	return IRQ_HANDLED;
1175  }
1176  
1177  static irqreturn_t kmx61_data_rdy_trig_poll(int irq, void *private)
1178  {
1179  	struct kmx61_data *data = private;
1180  
1181  	if (data->acc_dready_trig_on)
1182  		iio_trigger_poll(data->acc_dready_trig);
1183  	if (data->mag_dready_trig_on)
1184  		iio_trigger_poll(data->mag_dready_trig);
1185  
1186  	if (data->motion_trig_on)
1187  		iio_trigger_poll(data->motion_trig);
1188  
1189  	if (data->ev_enable_state)
1190  		return IRQ_WAKE_THREAD;
1191  	return IRQ_HANDLED;
1192  }
1193  
1194  static irqreturn_t kmx61_trigger_handler(int irq, void *p)
1195  {
1196  	struct iio_poll_func *pf = p;
1197  	struct iio_dev *indio_dev = pf->indio_dev;
1198  	struct kmx61_data *data = kmx61_get_data(indio_dev);
1199  	int bit, ret, i = 0;
1200  	u8 base;
1201  	s16 buffer[8];
1202  
1203  	if (indio_dev == data->acc_indio_dev)
1204  		base = KMX61_ACC_XOUT_L;
1205  	else
1206  		base = KMX61_MAG_XOUT_L;
1207  
1208  	mutex_lock(&data->lock);
1209  	for_each_set_bit(bit, indio_dev->active_scan_mask,
1210  			 indio_dev->masklength) {
1211  		ret = kmx61_read_measurement(data, base, bit);
1212  		if (ret < 0) {
1213  			mutex_unlock(&data->lock);
1214  			goto err;
1215  		}
1216  		buffer[i++] = ret;
1217  	}
1218  	mutex_unlock(&data->lock);
1219  
1220  	iio_push_to_buffers(indio_dev, buffer);
1221  err:
1222  	iio_trigger_notify_done(indio_dev->trig);
1223  
1224  	return IRQ_HANDLED;
1225  }
1226  
1227  static const char *kmx61_match_acpi_device(struct device *dev)
1228  {
1229  	const struct acpi_device_id *id;
1230  
1231  	id = acpi_match_device(dev->driver->acpi_match_table, dev);
1232  	if (!id)
1233  		return NULL;
1234  	return dev_name(dev);
1235  }
1236  
1237  static struct iio_dev *kmx61_indiodev_setup(struct kmx61_data *data,
1238  					    const struct iio_info *info,
1239  					    const struct iio_chan_spec *chan,
1240  					    int num_channels,
1241  					    const char *name)
1242  {
1243  	struct iio_dev *indio_dev;
1244  
1245  	indio_dev = devm_iio_device_alloc(&data->client->dev, sizeof(data));
1246  	if (!indio_dev)
1247  		return ERR_PTR(-ENOMEM);
1248  
1249  	kmx61_set_data(indio_dev, data);
1250  
1251  	indio_dev->dev.parent = &data->client->dev;
1252  	indio_dev->channels = chan;
1253  	indio_dev->num_channels = num_channels;
1254  	indio_dev->name = name;
1255  	indio_dev->modes = INDIO_DIRECT_MODE;
1256  	indio_dev->info = info;
1257  
1258  	return indio_dev;
1259  }
1260  
1261  static struct iio_trigger *kmx61_trigger_setup(struct kmx61_data *data,
1262  					       struct iio_dev *indio_dev,
1263  					       const char *tag)
1264  {
1265  	struct iio_trigger *trig;
1266  	int ret;
1267  
1268  	trig = devm_iio_trigger_alloc(&data->client->dev,
1269  				      "%s-%s-dev%d",
1270  				      indio_dev->name,
1271  				      tag,
1272  				      indio_dev->id);
1273  	if (!trig)
1274  		return ERR_PTR(-ENOMEM);
1275  
1276  	trig->dev.parent = &data->client->dev;
1277  	trig->ops = &kmx61_trigger_ops;
1278  	iio_trigger_set_drvdata(trig, indio_dev);
1279  
1280  	ret = iio_trigger_register(trig);
1281  	if (ret)
1282  		return ERR_PTR(ret);
1283  
1284  	return trig;
1285  }
1286  
1287  static int kmx61_probe(struct i2c_client *client,
1288  		       const struct i2c_device_id *id)
1289  {
1290  	int ret;
1291  	struct kmx61_data *data;
1292  	const char *name = NULL;
1293  
1294  	data = devm_kzalloc(&client->dev, sizeof(*data), GFP_KERNEL);
1295  	if (!data)
1296  		return -ENOMEM;
1297  
1298  	i2c_set_clientdata(client, data);
1299  	data->client = client;
1300  
1301  	mutex_init(&data->lock);
1302  
1303  	if (id)
1304  		name = id->name;
1305  	else if (ACPI_HANDLE(&client->dev))
1306  		name = kmx61_match_acpi_device(&client->dev);
1307  	else
1308  		return -ENODEV;
1309  
1310  	data->acc_indio_dev =
1311  		kmx61_indiodev_setup(data, &kmx61_acc_info,
1312  				     kmx61_acc_channels,
1313  				     ARRAY_SIZE(kmx61_acc_channels),
1314  				     name);
1315  	if (IS_ERR(data->acc_indio_dev))
1316  		return PTR_ERR(data->acc_indio_dev);
1317  
1318  	data->mag_indio_dev =
1319  		kmx61_indiodev_setup(data, &kmx61_mag_info,
1320  				     kmx61_mag_channels,
1321  				     ARRAY_SIZE(kmx61_mag_channels),
1322  				     name);
1323  	if (IS_ERR(data->mag_indio_dev))
1324  		return PTR_ERR(data->mag_indio_dev);
1325  
1326  	ret = kmx61_chip_init(data);
1327  	if (ret < 0)
1328  		return ret;
1329  
1330  	if (client->irq > 0) {
1331  		ret = devm_request_threaded_irq(&client->dev, client->irq,
1332  						kmx61_data_rdy_trig_poll,
1333  						kmx61_event_handler,
1334  						IRQF_TRIGGER_RISING,
1335  						KMX61_IRQ_NAME,
1336  						data);
1337  		if (ret)
1338  			goto err_chip_uninit;
1339  
1340  		data->acc_dready_trig =
1341  			kmx61_trigger_setup(data, data->acc_indio_dev,
1342  					    "dready");
1343  		if (IS_ERR(data->acc_dready_trig)) {
1344  			ret = PTR_ERR(data->acc_dready_trig);
1345  			goto err_chip_uninit;
1346  		}
1347  
1348  		data->mag_dready_trig =
1349  			kmx61_trigger_setup(data, data->mag_indio_dev,
1350  					    "dready");
1351  		if (IS_ERR(data->mag_dready_trig)) {
1352  			ret = PTR_ERR(data->mag_dready_trig);
1353  			goto err_trigger_unregister_acc_dready;
1354  		}
1355  
1356  		data->motion_trig =
1357  			kmx61_trigger_setup(data, data->acc_indio_dev,
1358  					    "any-motion");
1359  		if (IS_ERR(data->motion_trig)) {
1360  			ret = PTR_ERR(data->motion_trig);
1361  			goto err_trigger_unregister_mag_dready;
1362  		}
1363  
1364  		ret = iio_triggered_buffer_setup(data->acc_indio_dev,
1365  						 &iio_pollfunc_store_time,
1366  						 kmx61_trigger_handler,
1367  						 NULL);
1368  		if (ret < 0) {
1369  			dev_err(&data->client->dev,
1370  				"Failed to setup acc triggered buffer\n");
1371  			goto err_trigger_unregister_motion;
1372  		}
1373  
1374  		ret = iio_triggered_buffer_setup(data->mag_indio_dev,
1375  						 &iio_pollfunc_store_time,
1376  						 kmx61_trigger_handler,
1377  						 NULL);
1378  		if (ret < 0) {
1379  			dev_err(&data->client->dev,
1380  				"Failed to setup mag triggered buffer\n");
1381  			goto err_buffer_cleanup_acc;
1382  		}
1383  	}
1384  
1385  	ret = pm_runtime_set_active(&client->dev);
1386  	if (ret < 0)
1387  		goto err_buffer_cleanup_mag;
1388  
1389  	pm_runtime_enable(&client->dev);
1390  	pm_runtime_set_autosuspend_delay(&client->dev, KMX61_SLEEP_DELAY_MS);
1391  	pm_runtime_use_autosuspend(&client->dev);
1392  
1393  	ret = iio_device_register(data->acc_indio_dev);
1394  	if (ret < 0) {
1395  		dev_err(&client->dev, "Failed to register acc iio device\n");
1396  		goto err_buffer_cleanup_mag;
1397  	}
1398  
1399  	ret = iio_device_register(data->mag_indio_dev);
1400  	if (ret < 0) {
1401  		dev_err(&client->dev, "Failed to register mag iio device\n");
1402  		goto err_iio_unregister_acc;
1403  	}
1404  
1405  	return 0;
1406  
1407  err_iio_unregister_acc:
1408  	iio_device_unregister(data->acc_indio_dev);
1409  err_buffer_cleanup_mag:
1410  	if (client->irq > 0)
1411  		iio_triggered_buffer_cleanup(data->mag_indio_dev);
1412  err_buffer_cleanup_acc:
1413  	if (client->irq > 0)
1414  		iio_triggered_buffer_cleanup(data->acc_indio_dev);
1415  err_trigger_unregister_motion:
1416  	iio_trigger_unregister(data->motion_trig);
1417  err_trigger_unregister_mag_dready:
1418  	iio_trigger_unregister(data->mag_dready_trig);
1419  err_trigger_unregister_acc_dready:
1420  	iio_trigger_unregister(data->acc_dready_trig);
1421  err_chip_uninit:
1422  	kmx61_set_mode(data, KMX61_ALL_STBY, KMX61_ACC | KMX61_MAG, true);
1423  	return ret;
1424  }
1425  
1426  static int kmx61_remove(struct i2c_client *client)
1427  {
1428  	struct kmx61_data *data = i2c_get_clientdata(client);
1429  
1430  	iio_device_unregister(data->acc_indio_dev);
1431  	iio_device_unregister(data->mag_indio_dev);
1432  
1433  	pm_runtime_disable(&client->dev);
1434  	pm_runtime_set_suspended(&client->dev);
1435  	pm_runtime_put_noidle(&client->dev);
1436  
1437  	if (client->irq > 0) {
1438  		iio_triggered_buffer_cleanup(data->acc_indio_dev);
1439  		iio_triggered_buffer_cleanup(data->mag_indio_dev);
1440  		iio_trigger_unregister(data->acc_dready_trig);
1441  		iio_trigger_unregister(data->mag_dready_trig);
1442  		iio_trigger_unregister(data->motion_trig);
1443  	}
1444  
1445  	mutex_lock(&data->lock);
1446  	kmx61_set_mode(data, KMX61_ALL_STBY, KMX61_ACC | KMX61_MAG, true);
1447  	mutex_unlock(&data->lock);
1448  
1449  	return 0;
1450  }
1451  
1452  #ifdef CONFIG_PM_SLEEP
1453  static int kmx61_suspend(struct device *dev)
1454  {
1455  	int ret;
1456  	struct kmx61_data *data = i2c_get_clientdata(to_i2c_client(dev));
1457  
1458  	mutex_lock(&data->lock);
1459  	ret = kmx61_set_mode(data, KMX61_ALL_STBY, KMX61_ACC | KMX61_MAG,
1460  			     false);
1461  	mutex_unlock(&data->lock);
1462  
1463  	return ret;
1464  }
1465  
1466  static int kmx61_resume(struct device *dev)
1467  {
1468  	u8 stby = 0;
1469  	struct kmx61_data *data = i2c_get_clientdata(to_i2c_client(dev));
1470  
1471  	if (data->acc_stby)
1472  		stby |= KMX61_ACC_STBY_BIT;
1473  	if (data->mag_stby)
1474  		stby |= KMX61_MAG_STBY_BIT;
1475  
1476  	return kmx61_set_mode(data, stby, KMX61_ACC | KMX61_MAG, true);
1477  }
1478  #endif
1479  
1480  #ifdef CONFIG_PM
1481  static int kmx61_runtime_suspend(struct device *dev)
1482  {
1483  	struct kmx61_data *data = i2c_get_clientdata(to_i2c_client(dev));
1484  	int ret;
1485  
1486  	mutex_lock(&data->lock);
1487  	ret = kmx61_set_mode(data, KMX61_ALL_STBY, KMX61_ACC | KMX61_MAG, true);
1488  	mutex_unlock(&data->lock);
1489  
1490  	return ret;
1491  }
1492  
1493  static int kmx61_runtime_resume(struct device *dev)
1494  {
1495  	struct kmx61_data *data = i2c_get_clientdata(to_i2c_client(dev));
1496  	u8 stby = 0;
1497  
1498  	if (!data->acc_ps)
1499  		stby |= KMX61_ACC_STBY_BIT;
1500  	if (!data->mag_ps)
1501  		stby |= KMX61_MAG_STBY_BIT;
1502  
1503  	return kmx61_set_mode(data, stby, KMX61_ACC | KMX61_MAG, true);
1504  }
1505  #endif
1506  
1507  static const struct dev_pm_ops kmx61_pm_ops = {
1508  	SET_SYSTEM_SLEEP_PM_OPS(kmx61_suspend, kmx61_resume)
1509  	SET_RUNTIME_PM_OPS(kmx61_runtime_suspend, kmx61_runtime_resume, NULL)
1510  };
1511  
1512  static const struct acpi_device_id kmx61_acpi_match[] = {
1513  	{"KMX61021", 0},
1514  	{}
1515  };
1516  
1517  MODULE_DEVICE_TABLE(acpi, kmx61_acpi_match);
1518  
1519  static const struct i2c_device_id kmx61_id[] = {
1520  	{"kmx611021", 0},
1521  	{}
1522  };
1523  
1524  MODULE_DEVICE_TABLE(i2c, kmx61_id);
1525  
1526  static struct i2c_driver kmx61_driver = {
1527  	.driver = {
1528  		.name = KMX61_DRV_NAME,
1529  		.acpi_match_table = ACPI_PTR(kmx61_acpi_match),
1530  		.pm = &kmx61_pm_ops,
1531  	},
1532  	.probe		= kmx61_probe,
1533  	.remove		= kmx61_remove,
1534  	.id_table	= kmx61_id,
1535  };
1536  
1537  module_i2c_driver(kmx61_driver);
1538  
1539  MODULE_AUTHOR("Daniel Baluta <daniel.baluta@intel.com>");
1540  MODULE_DESCRIPTION("KMX61 accelerometer/magnetometer driver");
1541  MODULE_LICENSE("GPL v2");
1542