xref: /linux/drivers/iio/accel/kxcjk-1013.c (revision 42874e4eb35bdfc54f8514685e50434098ba4f6c)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * KXCJK-1013 3-axis accelerometer driver
4  * Copyright (c) 2014, Intel Corporation.
5  */
6 
7 #include <linux/module.h>
8 #include <linux/i2c.h>
9 #include <linux/interrupt.h>
10 #include <linux/delay.h>
11 #include <linux/bitops.h>
12 #include <linux/slab.h>
13 #include <linux/string.h>
14 #include <linux/acpi.h>
15 #include <linux/pm.h>
16 #include <linux/pm_runtime.h>
17 #include <linux/regulator/consumer.h>
18 #include <linux/iio/iio.h>
19 #include <linux/iio/sysfs.h>
20 #include <linux/iio/buffer.h>
21 #include <linux/iio/trigger.h>
22 #include <linux/iio/events.h>
23 #include <linux/iio/trigger_consumer.h>
24 #include <linux/iio/triggered_buffer.h>
25 #include <linux/iio/accel/kxcjk_1013.h>
26 
27 #define KXCJK1013_DRV_NAME "kxcjk1013"
28 #define KXCJK1013_IRQ_NAME "kxcjk1013_event"
29 
30 #define KXTF9_REG_HP_XOUT_L		0x00
31 #define KXTF9_REG_HP_XOUT_H		0x01
32 #define KXTF9_REG_HP_YOUT_L		0x02
33 #define KXTF9_REG_HP_YOUT_H		0x03
34 #define KXTF9_REG_HP_ZOUT_L		0x04
35 #define KXTF9_REG_HP_ZOUT_H		0x05
36 
37 #define KXCJK1013_REG_XOUT_L		0x06
38 /*
39  * From low byte X axis register, all the other addresses of Y and Z can be
40  * obtained by just applying axis offset. The following axis defines are just
41  * provide clarity, but not used.
42  */
43 #define KXCJK1013_REG_XOUT_H		0x07
44 #define KXCJK1013_REG_YOUT_L		0x08
45 #define KXCJK1013_REG_YOUT_H		0x09
46 #define KXCJK1013_REG_ZOUT_L		0x0A
47 #define KXCJK1013_REG_ZOUT_H		0x0B
48 
49 #define KXCJK1013_REG_DCST_RESP		0x0C
50 #define KXCJK1013_REG_WHO_AM_I		0x0F
51 #define KXTF9_REG_TILT_POS_CUR		0x10
52 #define KXTF9_REG_TILT_POS_PREV		0x11
53 #define KXTF9_REG_INT_SRC1		0x15
54 #define KXTF9_REG_INT_SRC2		0x16
55 #define KXCJK1013_REG_INT_SRC1		0x16
56 #define KXCJK1013_REG_INT_SRC2		0x17
57 #define KXCJK1013_REG_STATUS_REG	0x18
58 #define KXCJK1013_REG_INT_REL		0x1A
59 #define KXCJK1013_REG_CTRL1		0x1B
60 #define KXTF9_REG_CTRL2			0x1C
61 #define KXTF9_REG_CTRL3			0x1D
62 #define KXCJK1013_REG_CTRL2		0x1D
63 #define KXCJK1013_REG_INT_CTRL1		0x1E
64 #define KXCJK1013_REG_INT_CTRL2		0x1F
65 #define KXTF9_REG_INT_CTRL3		0x20
66 #define KXCJK1013_REG_DATA_CTRL		0x21
67 #define KXTF9_REG_TILT_TIMER		0x28
68 #define KXCJK1013_REG_WAKE_TIMER	0x29
69 #define KXTF9_REG_TDT_TIMER		0x2B
70 #define KXTF9_REG_TDT_THRESH_H		0x2C
71 #define KXTF9_REG_TDT_THRESH_L		0x2D
72 #define KXTF9_REG_TDT_TAP_TIMER		0x2E
73 #define KXTF9_REG_TDT_TOTAL_TIMER	0x2F
74 #define KXTF9_REG_TDT_LATENCY_TIMER	0x30
75 #define KXTF9_REG_TDT_WINDOW_TIMER	0x31
76 #define KXCJK1013_REG_SELF_TEST		0x3A
77 #define KXTF9_REG_WAKE_THRESH		0x5A
78 #define KXTF9_REG_TILT_ANGLE		0x5C
79 #define KXTF9_REG_HYST_SET		0x5F
80 #define KXCJK1013_REG_WAKE_THRES	0x6A
81 
82 /* Everything up to 0x11 is equal to KXCJK1013/KXTF9 above */
83 #define KX023_REG_INS1			0x12
84 #define KX023_REG_INS2			0x13
85 #define KX023_REG_INS3			0x14
86 #define KX023_REG_STAT			0x15
87 #define KX023_REG_INT_REL		0x17
88 #define KX023_REG_CNTL1			0x18
89 #define KX023_REG_CNTL2			0x19
90 #define KX023_REG_CNTL3			0x1A
91 #define KX023_REG_ODCNTL		0x1B
92 #define KX023_REG_INC1			0x1C
93 #define KX023_REG_INC2			0x1D
94 #define KX023_REG_INC3			0x1E
95 #define KX023_REG_INC4			0x1F
96 #define KX023_REG_INC5			0x20
97 #define KX023_REG_INC6			0x21
98 #define KX023_REG_TILT_TIMER		0x22
99 #define KX023_REG_WUFC			0x23
100 #define KX023_REG_TDTRC			0x24
101 #define KX023_REG_TDTC			0x25
102 #define KX023_REG_TTH			0x26
103 #define KX023_REG_TTL			0x27
104 #define KX023_REG_FTD			0x28
105 #define KX023_REG_STD			0x29
106 #define KX023_REG_TLT			0x2A
107 #define KX023_REG_TWS			0x2B
108 #define KX023_REG_ATH			0x30
109 #define KX023_REG_TILT_ANGLE_LL		0x32
110 #define KX023_REG_TILT_ANGLE_HL		0x33
111 #define KX023_REG_HYST_SET		0x34
112 #define KX023_REG_LP_CNTL		0x35
113 #define KX023_REG_BUF_CNTL1		0x3A
114 #define KX023_REG_BUF_CNTL2		0x3B
115 #define KX023_REG_BUF_STATUS_1		0x3C
116 #define KX023_REG_BUF_STATUS_2		0x3D
117 #define KX023_REG_BUF_CLEAR		0x3E
118 #define KX023_REG_BUF_READ		0x3F
119 #define KX023_REG_SELF_TEST		0x60
120 
121 #define KXCJK1013_REG_CTRL1_BIT_PC1	BIT(7)
122 #define KXCJK1013_REG_CTRL1_BIT_RES	BIT(6)
123 #define KXCJK1013_REG_CTRL1_BIT_DRDY	BIT(5)
124 #define KXCJK1013_REG_CTRL1_BIT_GSEL1	BIT(4)
125 #define KXCJK1013_REG_CTRL1_BIT_GSEL0	BIT(3)
126 #define KXCJK1013_REG_CTRL1_BIT_WUFE	BIT(1)
127 
128 #define KXCJK1013_REG_INT_CTRL1_BIT_IEU	BIT(2)	/* KXTF9 */
129 #define KXCJK1013_REG_INT_CTRL1_BIT_IEL	BIT(3)
130 #define KXCJK1013_REG_INT_CTRL1_BIT_IEA	BIT(4)
131 #define KXCJK1013_REG_INT_CTRL1_BIT_IEN	BIT(5)
132 
133 #define KXTF9_REG_TILT_BIT_LEFT_EDGE	BIT(5)
134 #define KXTF9_REG_TILT_BIT_RIGHT_EDGE	BIT(4)
135 #define KXTF9_REG_TILT_BIT_LOWER_EDGE	BIT(3)
136 #define KXTF9_REG_TILT_BIT_UPPER_EDGE	BIT(2)
137 #define KXTF9_REG_TILT_BIT_FACE_DOWN	BIT(1)
138 #define KXTF9_REG_TILT_BIT_FACE_UP	BIT(0)
139 
140 #define KXCJK1013_DATA_MASK_12_BIT	0x0FFF
141 #define KXCJK1013_MAX_STARTUP_TIME_US	100000
142 
143 #define KXCJK1013_SLEEP_DELAY_MS	2000
144 
145 #define KXCJK1013_REG_INT_SRC1_BIT_TPS	BIT(0)	/* KXTF9 */
146 #define KXCJK1013_REG_INT_SRC1_BIT_WUFS	BIT(1)
147 #define KXCJK1013_REG_INT_SRC1_MASK_TDTS	(BIT(2) | BIT(3))	/* KXTF9 */
148 #define KXCJK1013_REG_INT_SRC1_TAP_NONE		0
149 #define KXCJK1013_REG_INT_SRC1_TAP_SINGLE		BIT(2)
150 #define KXCJK1013_REG_INT_SRC1_TAP_DOUBLE		BIT(3)
151 #define KXCJK1013_REG_INT_SRC1_BIT_DRDY	BIT(4)
152 
153 /* KXCJK: INT_SOURCE2: motion detect, KXTF9: INT_SRC_REG1: tap detect */
154 #define KXCJK1013_REG_INT_SRC2_BIT_ZP	BIT(0)
155 #define KXCJK1013_REG_INT_SRC2_BIT_ZN	BIT(1)
156 #define KXCJK1013_REG_INT_SRC2_BIT_YP	BIT(2)
157 #define KXCJK1013_REG_INT_SRC2_BIT_YN	BIT(3)
158 #define KXCJK1013_REG_INT_SRC2_BIT_XP	BIT(4)
159 #define KXCJK1013_REG_INT_SRC2_BIT_XN	BIT(5)
160 
161 /* KX023 interrupt routing to INT1. INT2 can be configured with INC6 */
162 #define KX023_REG_INC4_BFI1		BIT(6)
163 #define KX023_REG_INC4_WMI1		BIT(5)
164 #define KX023_REG_INC4_DRDY1		BIT(4)
165 #define KX023_REG_INC4_TDTI1		BIT(2)
166 #define KX023_REG_INC4_WUFI1		BIT(1)
167 #define KX023_REG_INC4_TPI1		BIT(0)
168 
169 #define KXCJK1013_DEFAULT_WAKE_THRES	1
170 
171 enum kx_chipset {
172 	KXCJK1013,
173 	KXCJ91008,
174 	KXTJ21009,
175 	KXTF9,
176 	KX0231025,
177 	KX_MAX_CHIPS /* this must be last */
178 };
179 
180 enum kx_acpi_type {
181 	ACPI_GENERIC,
182 	ACPI_SMO8500,
183 	ACPI_KIOX010A,
184 };
185 
186 struct kx_chipset_regs {
187 	u8 int_src1;
188 	u8 int_src2;
189 	u8 int_rel;
190 	u8 ctrl1;
191 	u8 wuf_ctrl;
192 	u8 int_ctrl1;
193 	u8 data_ctrl;
194 	u8 wake_timer;
195 	u8 wake_thres;
196 };
197 
198 static const struct kx_chipset_regs kxcjk1013_regs = {
199 	.int_src1	= KXCJK1013_REG_INT_SRC1,
200 	.int_src2	= KXCJK1013_REG_INT_SRC2,
201 	.int_rel	= KXCJK1013_REG_INT_REL,
202 	.ctrl1		= KXCJK1013_REG_CTRL1,
203 	.wuf_ctrl	= KXCJK1013_REG_CTRL2,
204 	.int_ctrl1	= KXCJK1013_REG_INT_CTRL1,
205 	.data_ctrl	= KXCJK1013_REG_DATA_CTRL,
206 	.wake_timer	= KXCJK1013_REG_WAKE_TIMER,
207 	.wake_thres	= KXCJK1013_REG_WAKE_THRES,
208 };
209 
210 static const struct kx_chipset_regs kxtf9_regs = {
211 	/* .int_src1 was moved to INT_SRC2 on KXTF9 */
212 	.int_src1	= KXTF9_REG_INT_SRC2,
213 	/* .int_src2 is not available */
214 	.int_rel	= KXCJK1013_REG_INT_REL,
215 	.ctrl1		= KXCJK1013_REG_CTRL1,
216 	.wuf_ctrl	= KXTF9_REG_CTRL3,
217 	.int_ctrl1	= KXCJK1013_REG_INT_CTRL1,
218 	.data_ctrl	= KXCJK1013_REG_DATA_CTRL,
219 	.wake_timer	= KXCJK1013_REG_WAKE_TIMER,
220 	.wake_thres	= KXTF9_REG_WAKE_THRESH,
221 };
222 
223 /* The registers have totally different names but the bits are compatible */
224 static const struct kx_chipset_regs kx0231025_regs = {
225 	.int_src1	= KX023_REG_INS2,
226 	.int_src2	= KX023_REG_INS3,
227 	.int_rel	= KX023_REG_INT_REL,
228 	.ctrl1		= KX023_REG_CNTL1,
229 	.wuf_ctrl	= KX023_REG_CNTL3,
230 	.int_ctrl1	= KX023_REG_INC1,
231 	.data_ctrl	= KX023_REG_ODCNTL,
232 	.wake_timer	= KX023_REG_WUFC,
233 	.wake_thres	= KX023_REG_ATH,
234 };
235 
236 enum kxcjk1013_axis {
237 	AXIS_X,
238 	AXIS_Y,
239 	AXIS_Z,
240 	AXIS_MAX
241 };
242 
243 struct kxcjk1013_data {
244 	struct i2c_client *client;
245 	struct iio_trigger *dready_trig;
246 	struct iio_trigger *motion_trig;
247 	struct iio_mount_matrix orientation;
248 	struct mutex mutex;
249 	/* Ensure timestamp naturally aligned */
250 	struct {
251 		s16 chans[AXIS_MAX];
252 		s64 timestamp __aligned(8);
253 	} scan;
254 	u8 odr_bits;
255 	u8 range;
256 	int wake_thres;
257 	int wake_dur;
258 	bool active_high_intr;
259 	bool dready_trigger_on;
260 	int ev_enable_state;
261 	bool motion_trigger_on;
262 	int64_t timestamp;
263 	enum kx_chipset chipset;
264 	enum kx_acpi_type acpi_type;
265 	const struct kx_chipset_regs *regs;
266 };
267 
268 enum kxcjk1013_mode {
269 	STANDBY,
270 	OPERATION,
271 };
272 
273 enum kxcjk1013_range {
274 	KXCJK1013_RANGE_2G,
275 	KXCJK1013_RANGE_4G,
276 	KXCJK1013_RANGE_8G,
277 };
278 
279 struct kx_odr_map {
280 	int val;
281 	int val2;
282 	int odr_bits;
283 	int wuf_bits;
284 };
285 
286 static const struct kx_odr_map samp_freq_table[] = {
287 	{ 0, 781000, 0x08, 0x00 },
288 	{ 1, 563000, 0x09, 0x01 },
289 	{ 3, 125000, 0x0A, 0x02 },
290 	{ 6, 250000, 0x0B, 0x03 },
291 	{ 12, 500000, 0x00, 0x04 },
292 	{ 25, 0, 0x01, 0x05 },
293 	{ 50, 0, 0x02, 0x06 },
294 	{ 100, 0, 0x03, 0x06 },
295 	{ 200, 0, 0x04, 0x06 },
296 	{ 400, 0, 0x05, 0x06 },
297 	{ 800, 0, 0x06, 0x06 },
298 	{ 1600, 0, 0x07, 0x06 },
299 };
300 
301 static const char *const kxcjk1013_samp_freq_avail =
302 	"0.781000 1.563000 3.125000 6.250000 12.500000 25 50 100 200 400 800 1600";
303 
304 static const struct kx_odr_map kxtf9_samp_freq_table[] = {
305 	{ 25, 0, 0x01, 0x00 },
306 	{ 50, 0, 0x02, 0x01 },
307 	{ 100, 0, 0x03, 0x01 },
308 	{ 200, 0, 0x04, 0x01 },
309 	{ 400, 0, 0x05, 0x01 },
310 	{ 800, 0, 0x06, 0x01 },
311 };
312 
313 static const char *const kxtf9_samp_freq_avail =
314 	"25 50 100 200 400 800";
315 
316 /* Refer to section 4 of the specification */
317 static __maybe_unused const struct {
318 	int odr_bits;
319 	int usec;
320 } odr_start_up_times[KX_MAX_CHIPS][12] = {
321 	/* KXCJK-1013 */
322 	{
323 		{0x08, 100000},
324 		{0x09, 100000},
325 		{0x0A, 100000},
326 		{0x0B, 100000},
327 		{0, 80000},
328 		{0x01, 41000},
329 		{0x02, 21000},
330 		{0x03, 11000},
331 		{0x04, 6400},
332 		{0x05, 3900},
333 		{0x06, 2700},
334 		{0x07, 2100},
335 	},
336 	/* KXCJ9-1008 */
337 	{
338 		{0x08, 100000},
339 		{0x09, 100000},
340 		{0x0A, 100000},
341 		{0x0B, 100000},
342 		{0, 80000},
343 		{0x01, 41000},
344 		{0x02, 21000},
345 		{0x03, 11000},
346 		{0x04, 6400},
347 		{0x05, 3900},
348 		{0x06, 2700},
349 		{0x07, 2100},
350 	},
351 	/* KXCTJ2-1009 */
352 	{
353 		{0x08, 1240000},
354 		{0x09, 621000},
355 		{0x0A, 309000},
356 		{0x0B, 151000},
357 		{0, 80000},
358 		{0x01, 41000},
359 		{0x02, 21000},
360 		{0x03, 11000},
361 		{0x04, 6000},
362 		{0x05, 4000},
363 		{0x06, 3000},
364 		{0x07, 2000},
365 	},
366 	/* KXTF9 */
367 	{
368 		{0x01, 81000},
369 		{0x02, 41000},
370 		{0x03, 21000},
371 		{0x04, 11000},
372 		{0x05, 5100},
373 		{0x06, 2700},
374 	},
375 	/* KX023-1025 */
376 	{
377 		/* First 4 are not in datasheet, taken from KXCTJ2-1009 */
378 		{0x08, 1240000},
379 		{0x09, 621000},
380 		{0x0A, 309000},
381 		{0x0B, 151000},
382 		{0, 81000},
383 		{0x01, 40000},
384 		{0x02, 22000},
385 		{0x03, 12000},
386 		{0x04, 7000},
387 		{0x05, 4400},
388 		{0x06, 3000},
389 		{0x07, 3000},
390 	},
391 };
392 
393 static const struct {
394 	u16 scale;
395 	u8 gsel_0;
396 	u8 gsel_1;
397 } KXCJK1013_scale_table[] = { {9582, 0, 0},
398 			      {19163, 1, 0},
399 			      {38326, 0, 1} };
400 
401 #ifdef CONFIG_ACPI
402 enum kiox010a_fn_index {
403 	KIOX010A_SET_LAPTOP_MODE = 1,
404 	KIOX010A_SET_TABLET_MODE = 2,
405 };
406 
407 static int kiox010a_dsm(struct device *dev, int fn_index)
408 {
409 	acpi_handle handle = ACPI_HANDLE(dev);
410 	guid_t kiox010a_dsm_guid;
411 	union acpi_object *obj;
412 
413 	if (!handle)
414 		return -ENODEV;
415 
416 	guid_parse("1f339696-d475-4e26-8cad-2e9f8e6d7a91", &kiox010a_dsm_guid);
417 
418 	obj = acpi_evaluate_dsm(handle, &kiox010a_dsm_guid, 1, fn_index, NULL);
419 	if (!obj)
420 		return -EIO;
421 
422 	ACPI_FREE(obj);
423 	return 0;
424 }
425 #endif
426 
427 static int kxcjk1013_set_mode(struct kxcjk1013_data *data,
428 			      enum kxcjk1013_mode mode)
429 {
430 	int ret;
431 
432 	ret = i2c_smbus_read_byte_data(data->client, data->regs->ctrl1);
433 	if (ret < 0) {
434 		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
435 		return ret;
436 	}
437 
438 	if (mode == STANDBY)
439 		ret &= ~KXCJK1013_REG_CTRL1_BIT_PC1;
440 	else
441 		ret |= KXCJK1013_REG_CTRL1_BIT_PC1;
442 
443 	ret = i2c_smbus_write_byte_data(data->client, data->regs->ctrl1, ret);
444 	if (ret < 0) {
445 		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
446 		return ret;
447 	}
448 
449 	return 0;
450 }
451 
452 static int kxcjk1013_get_mode(struct kxcjk1013_data *data,
453 			      enum kxcjk1013_mode *mode)
454 {
455 	int ret;
456 
457 	ret = i2c_smbus_read_byte_data(data->client, data->regs->ctrl1);
458 	if (ret < 0) {
459 		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
460 		return ret;
461 	}
462 
463 	if (ret & KXCJK1013_REG_CTRL1_BIT_PC1)
464 		*mode = OPERATION;
465 	else
466 		*mode = STANDBY;
467 
468 	return 0;
469 }
470 
471 static int kxcjk1013_set_range(struct kxcjk1013_data *data, int range_index)
472 {
473 	int ret;
474 
475 	ret = i2c_smbus_read_byte_data(data->client, data->regs->ctrl1);
476 	if (ret < 0) {
477 		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
478 		return ret;
479 	}
480 
481 	ret &= ~(KXCJK1013_REG_CTRL1_BIT_GSEL0 |
482 		 KXCJK1013_REG_CTRL1_BIT_GSEL1);
483 	ret |= (KXCJK1013_scale_table[range_index].gsel_0 << 3);
484 	ret |= (KXCJK1013_scale_table[range_index].gsel_1 << 4);
485 
486 	ret = i2c_smbus_write_byte_data(data->client, data->regs->ctrl1, ret);
487 	if (ret < 0) {
488 		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
489 		return ret;
490 	}
491 
492 	data->range = range_index;
493 
494 	return 0;
495 }
496 
497 static int kxcjk1013_chip_init(struct kxcjk1013_data *data)
498 {
499 	int ret;
500 
501 #ifdef CONFIG_ACPI
502 	if (data->acpi_type == ACPI_KIOX010A) {
503 		/* Make sure the kbd and touchpad on 2-in-1s using 2 KXCJ91008-s work */
504 		kiox010a_dsm(&data->client->dev, KIOX010A_SET_LAPTOP_MODE);
505 	}
506 #endif
507 
508 	ret = i2c_smbus_read_byte_data(data->client, KXCJK1013_REG_WHO_AM_I);
509 	if (ret < 0) {
510 		dev_err(&data->client->dev, "Error reading who_am_i\n");
511 		return ret;
512 	}
513 
514 	dev_dbg(&data->client->dev, "KXCJK1013 Chip Id %x\n", ret);
515 
516 	ret = kxcjk1013_set_mode(data, STANDBY);
517 	if (ret < 0)
518 		return ret;
519 
520 	ret = i2c_smbus_read_byte_data(data->client, data->regs->ctrl1);
521 	if (ret < 0) {
522 		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
523 		return ret;
524 	}
525 
526 	/* Set 12 bit mode */
527 	ret |= KXCJK1013_REG_CTRL1_BIT_RES;
528 
529 	ret = i2c_smbus_write_byte_data(data->client, data->regs->ctrl1, ret);
530 	if (ret < 0) {
531 		dev_err(&data->client->dev, "Error reading reg_ctrl\n");
532 		return ret;
533 	}
534 
535 	/* Setting range to 4G */
536 	ret = kxcjk1013_set_range(data, KXCJK1013_RANGE_4G);
537 	if (ret < 0)
538 		return ret;
539 
540 	ret = i2c_smbus_read_byte_data(data->client, data->regs->data_ctrl);
541 	if (ret < 0) {
542 		dev_err(&data->client->dev, "Error reading reg_data_ctrl\n");
543 		return ret;
544 	}
545 
546 	data->odr_bits = ret;
547 
548 	/* Set up INT polarity */
549 	ret = i2c_smbus_read_byte_data(data->client, data->regs->int_ctrl1);
550 	if (ret < 0) {
551 		dev_err(&data->client->dev, "Error reading reg_int_ctrl1\n");
552 		return ret;
553 	}
554 
555 	if (data->active_high_intr)
556 		ret |= KXCJK1013_REG_INT_CTRL1_BIT_IEA;
557 	else
558 		ret &= ~KXCJK1013_REG_INT_CTRL1_BIT_IEA;
559 
560 	ret = i2c_smbus_write_byte_data(data->client, data->regs->int_ctrl1, ret);
561 	if (ret < 0) {
562 		dev_err(&data->client->dev, "Error writing reg_int_ctrl1\n");
563 		return ret;
564 	}
565 
566 	/* On KX023, route all used interrupts to INT1 for now */
567 	if (data->chipset == KX0231025 && data->client->irq > 0) {
568 		ret = i2c_smbus_write_byte_data(data->client, KX023_REG_INC4,
569 						KX023_REG_INC4_DRDY1 |
570 						KX023_REG_INC4_WUFI1);
571 		if (ret < 0) {
572 			dev_err(&data->client->dev, "Error writing reg_inc4\n");
573 			return ret;
574 		}
575 	}
576 
577 	ret = kxcjk1013_set_mode(data, OPERATION);
578 	if (ret < 0)
579 		return ret;
580 
581 	data->wake_thres = KXCJK1013_DEFAULT_WAKE_THRES;
582 
583 	return 0;
584 }
585 
586 #ifdef CONFIG_PM
587 static int kxcjk1013_get_startup_times(struct kxcjk1013_data *data)
588 {
589 	int i;
590 	int idx = data->chipset;
591 
592 	for (i = 0; i < ARRAY_SIZE(odr_start_up_times[idx]); ++i) {
593 		if (odr_start_up_times[idx][i].odr_bits == data->odr_bits)
594 			return odr_start_up_times[idx][i].usec;
595 	}
596 
597 	return KXCJK1013_MAX_STARTUP_TIME_US;
598 }
599 #endif
600 
601 static int kxcjk1013_set_power_state(struct kxcjk1013_data *data, bool on)
602 {
603 #ifdef CONFIG_PM
604 	int ret;
605 
606 	if (on)
607 		ret = pm_runtime_resume_and_get(&data->client->dev);
608 	else {
609 		pm_runtime_mark_last_busy(&data->client->dev);
610 		ret = pm_runtime_put_autosuspend(&data->client->dev);
611 	}
612 	if (ret < 0) {
613 		dev_err(&data->client->dev,
614 			"Failed: %s for %d\n", __func__, on);
615 		return ret;
616 	}
617 #endif
618 
619 	return 0;
620 }
621 
622 static int kxcjk1013_chip_update_thresholds(struct kxcjk1013_data *data)
623 {
624 	int ret;
625 
626 	ret = i2c_smbus_write_byte_data(data->client, data->regs->wake_timer,
627 					data->wake_dur);
628 	if (ret < 0) {
629 		dev_err(&data->client->dev,
630 			"Error writing reg_wake_timer\n");
631 		return ret;
632 	}
633 
634 	ret = i2c_smbus_write_byte_data(data->client, data->regs->wake_thres,
635 					data->wake_thres);
636 	if (ret < 0) {
637 		dev_err(&data->client->dev, "Error writing reg_wake_thres\n");
638 		return ret;
639 	}
640 
641 	return 0;
642 }
643 
644 static int kxcjk1013_setup_any_motion_interrupt(struct kxcjk1013_data *data,
645 						bool status)
646 {
647 	int ret;
648 	enum kxcjk1013_mode store_mode;
649 
650 	ret = kxcjk1013_get_mode(data, &store_mode);
651 	if (ret < 0)
652 		return ret;
653 
654 	/* This is requirement by spec to change state to STANDBY */
655 	ret = kxcjk1013_set_mode(data, STANDBY);
656 	if (ret < 0)
657 		return ret;
658 
659 	ret = kxcjk1013_chip_update_thresholds(data);
660 	if (ret < 0)
661 		return ret;
662 
663 	ret = i2c_smbus_read_byte_data(data->client, data->regs->int_ctrl1);
664 	if (ret < 0) {
665 		dev_err(&data->client->dev, "Error reading reg_int_ctrl1\n");
666 		return ret;
667 	}
668 
669 	if (status)
670 		ret |= KXCJK1013_REG_INT_CTRL1_BIT_IEN;
671 	else
672 		ret &= ~KXCJK1013_REG_INT_CTRL1_BIT_IEN;
673 
674 	ret = i2c_smbus_write_byte_data(data->client, data->regs->int_ctrl1, ret);
675 	if (ret < 0) {
676 		dev_err(&data->client->dev, "Error writing reg_int_ctrl1\n");
677 		return ret;
678 	}
679 
680 	ret = i2c_smbus_read_byte_data(data->client, data->regs->ctrl1);
681 	if (ret < 0) {
682 		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
683 		return ret;
684 	}
685 
686 	if (status)
687 		ret |= KXCJK1013_REG_CTRL1_BIT_WUFE;
688 	else
689 		ret &= ~KXCJK1013_REG_CTRL1_BIT_WUFE;
690 
691 	ret = i2c_smbus_write_byte_data(data->client, data->regs->ctrl1, ret);
692 	if (ret < 0) {
693 		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
694 		return ret;
695 	}
696 
697 	if (store_mode == OPERATION) {
698 		ret = kxcjk1013_set_mode(data, OPERATION);
699 		if (ret < 0)
700 			return ret;
701 	}
702 
703 	return 0;
704 }
705 
706 static int kxcjk1013_setup_new_data_interrupt(struct kxcjk1013_data *data,
707 					      bool status)
708 {
709 	int ret;
710 	enum kxcjk1013_mode store_mode;
711 
712 	ret = kxcjk1013_get_mode(data, &store_mode);
713 	if (ret < 0)
714 		return ret;
715 
716 	/* This is requirement by spec to change state to STANDBY */
717 	ret = kxcjk1013_set_mode(data, STANDBY);
718 	if (ret < 0)
719 		return ret;
720 
721 	ret = i2c_smbus_read_byte_data(data->client, data->regs->int_ctrl1);
722 	if (ret < 0) {
723 		dev_err(&data->client->dev, "Error reading reg_int_ctrl1\n");
724 		return ret;
725 	}
726 
727 	if (status)
728 		ret |= KXCJK1013_REG_INT_CTRL1_BIT_IEN;
729 	else
730 		ret &= ~KXCJK1013_REG_INT_CTRL1_BIT_IEN;
731 
732 	ret = i2c_smbus_write_byte_data(data->client, data->regs->int_ctrl1, ret);
733 	if (ret < 0) {
734 		dev_err(&data->client->dev, "Error writing reg_int_ctrl1\n");
735 		return ret;
736 	}
737 
738 	ret = i2c_smbus_read_byte_data(data->client, data->regs->ctrl1);
739 	if (ret < 0) {
740 		dev_err(&data->client->dev, "Error reading reg_ctrl1\n");
741 		return ret;
742 	}
743 
744 	if (status)
745 		ret |= KXCJK1013_REG_CTRL1_BIT_DRDY;
746 	else
747 		ret &= ~KXCJK1013_REG_CTRL1_BIT_DRDY;
748 
749 	ret = i2c_smbus_write_byte_data(data->client, data->regs->ctrl1, ret);
750 	if (ret < 0) {
751 		dev_err(&data->client->dev, "Error writing reg_ctrl1\n");
752 		return ret;
753 	}
754 
755 	if (store_mode == OPERATION) {
756 		ret = kxcjk1013_set_mode(data, OPERATION);
757 		if (ret < 0)
758 			return ret;
759 	}
760 
761 	return 0;
762 }
763 
764 static const struct kx_odr_map *kxcjk1013_find_odr_value(
765 	const struct kx_odr_map *map, size_t map_size, int val, int val2)
766 {
767 	int i;
768 
769 	for (i = 0; i < map_size; ++i) {
770 		if (map[i].val == val && map[i].val2 == val2)
771 			return &map[i];
772 	}
773 
774 	return ERR_PTR(-EINVAL);
775 }
776 
777 static int kxcjk1013_convert_odr_value(const struct kx_odr_map *map,
778 				       size_t map_size, int odr_bits,
779 				       int *val, int *val2)
780 {
781 	int i;
782 
783 	for (i = 0; i < map_size; ++i) {
784 		if (map[i].odr_bits == odr_bits) {
785 			*val = map[i].val;
786 			*val2 = map[i].val2;
787 			return IIO_VAL_INT_PLUS_MICRO;
788 		}
789 	}
790 
791 	return -EINVAL;
792 }
793 
794 static int kxcjk1013_set_odr(struct kxcjk1013_data *data, int val, int val2)
795 {
796 	int ret;
797 	enum kxcjk1013_mode store_mode;
798 	const struct kx_odr_map *odr_setting;
799 
800 	ret = kxcjk1013_get_mode(data, &store_mode);
801 	if (ret < 0)
802 		return ret;
803 
804 	if (data->chipset == KXTF9)
805 		odr_setting = kxcjk1013_find_odr_value(kxtf9_samp_freq_table,
806 						       ARRAY_SIZE(kxtf9_samp_freq_table),
807 						       val, val2);
808 	else
809 		odr_setting = kxcjk1013_find_odr_value(samp_freq_table,
810 						       ARRAY_SIZE(samp_freq_table),
811 						       val, val2);
812 
813 	if (IS_ERR(odr_setting))
814 		return PTR_ERR(odr_setting);
815 
816 	/* To change ODR, the chip must be set to STANDBY as per spec */
817 	ret = kxcjk1013_set_mode(data, STANDBY);
818 	if (ret < 0)
819 		return ret;
820 
821 	ret = i2c_smbus_write_byte_data(data->client, data->regs->data_ctrl,
822 					odr_setting->odr_bits);
823 	if (ret < 0) {
824 		dev_err(&data->client->dev, "Error writing data_ctrl\n");
825 		return ret;
826 	}
827 
828 	data->odr_bits = odr_setting->odr_bits;
829 
830 	ret = i2c_smbus_write_byte_data(data->client, data->regs->wuf_ctrl,
831 					odr_setting->wuf_bits);
832 	if (ret < 0) {
833 		dev_err(&data->client->dev, "Error writing reg_ctrl2\n");
834 		return ret;
835 	}
836 
837 	if (store_mode == OPERATION) {
838 		ret = kxcjk1013_set_mode(data, OPERATION);
839 		if (ret < 0)
840 			return ret;
841 	}
842 
843 	return 0;
844 }
845 
846 static int kxcjk1013_get_odr(struct kxcjk1013_data *data, int *val, int *val2)
847 {
848 	if (data->chipset == KXTF9)
849 		return kxcjk1013_convert_odr_value(kxtf9_samp_freq_table,
850 						   ARRAY_SIZE(kxtf9_samp_freq_table),
851 						   data->odr_bits, val, val2);
852 	else
853 		return kxcjk1013_convert_odr_value(samp_freq_table,
854 						   ARRAY_SIZE(samp_freq_table),
855 						   data->odr_bits, val, val2);
856 }
857 
858 static int kxcjk1013_get_acc_reg(struct kxcjk1013_data *data, int axis)
859 {
860 	u8 reg = KXCJK1013_REG_XOUT_L + axis * 2;
861 	int ret;
862 
863 	ret = i2c_smbus_read_word_data(data->client, reg);
864 	if (ret < 0) {
865 		dev_err(&data->client->dev,
866 			"failed to read accel_%c registers\n", 'x' + axis);
867 		return ret;
868 	}
869 
870 	return ret;
871 }
872 
873 static int kxcjk1013_set_scale(struct kxcjk1013_data *data, int val)
874 {
875 	int ret, i;
876 	enum kxcjk1013_mode store_mode;
877 
878 	for (i = 0; i < ARRAY_SIZE(KXCJK1013_scale_table); ++i) {
879 		if (KXCJK1013_scale_table[i].scale == val) {
880 			ret = kxcjk1013_get_mode(data, &store_mode);
881 			if (ret < 0)
882 				return ret;
883 
884 			ret = kxcjk1013_set_mode(data, STANDBY);
885 			if (ret < 0)
886 				return ret;
887 
888 			ret = kxcjk1013_set_range(data, i);
889 			if (ret < 0)
890 				return ret;
891 
892 			if (store_mode == OPERATION) {
893 				ret = kxcjk1013_set_mode(data, OPERATION);
894 				if (ret)
895 					return ret;
896 			}
897 
898 			return 0;
899 		}
900 	}
901 
902 	return -EINVAL;
903 }
904 
905 static int kxcjk1013_read_raw(struct iio_dev *indio_dev,
906 			      struct iio_chan_spec const *chan, int *val,
907 			      int *val2, long mask)
908 {
909 	struct kxcjk1013_data *data = iio_priv(indio_dev);
910 	int ret;
911 
912 	switch (mask) {
913 	case IIO_CHAN_INFO_RAW:
914 		mutex_lock(&data->mutex);
915 		if (iio_buffer_enabled(indio_dev))
916 			ret = -EBUSY;
917 		else {
918 			ret = kxcjk1013_set_power_state(data, true);
919 			if (ret < 0) {
920 				mutex_unlock(&data->mutex);
921 				return ret;
922 			}
923 			ret = kxcjk1013_get_acc_reg(data, chan->scan_index);
924 			if (ret < 0) {
925 				kxcjk1013_set_power_state(data, false);
926 				mutex_unlock(&data->mutex);
927 				return ret;
928 			}
929 			*val = sign_extend32(ret >> chan->scan_type.shift,
930 					     chan->scan_type.realbits - 1);
931 			ret = kxcjk1013_set_power_state(data, false);
932 		}
933 		mutex_unlock(&data->mutex);
934 
935 		if (ret < 0)
936 			return ret;
937 
938 		return IIO_VAL_INT;
939 
940 	case IIO_CHAN_INFO_SCALE:
941 		*val = 0;
942 		*val2 = KXCJK1013_scale_table[data->range].scale;
943 		return IIO_VAL_INT_PLUS_MICRO;
944 
945 	case IIO_CHAN_INFO_SAMP_FREQ:
946 		mutex_lock(&data->mutex);
947 		ret = kxcjk1013_get_odr(data, val, val2);
948 		mutex_unlock(&data->mutex);
949 		return ret;
950 
951 	default:
952 		return -EINVAL;
953 	}
954 }
955 
956 static int kxcjk1013_write_raw(struct iio_dev *indio_dev,
957 			       struct iio_chan_spec const *chan, int val,
958 			       int val2, long mask)
959 {
960 	struct kxcjk1013_data *data = iio_priv(indio_dev);
961 	int ret;
962 
963 	switch (mask) {
964 	case IIO_CHAN_INFO_SAMP_FREQ:
965 		mutex_lock(&data->mutex);
966 		ret = kxcjk1013_set_odr(data, val, val2);
967 		mutex_unlock(&data->mutex);
968 		break;
969 	case IIO_CHAN_INFO_SCALE:
970 		if (val)
971 			return -EINVAL;
972 
973 		mutex_lock(&data->mutex);
974 		ret = kxcjk1013_set_scale(data, val2);
975 		mutex_unlock(&data->mutex);
976 		break;
977 	default:
978 		ret = -EINVAL;
979 	}
980 
981 	return ret;
982 }
983 
984 static int kxcjk1013_read_event(struct iio_dev *indio_dev,
985 				   const struct iio_chan_spec *chan,
986 				   enum iio_event_type type,
987 				   enum iio_event_direction dir,
988 				   enum iio_event_info info,
989 				   int *val, int *val2)
990 {
991 	struct kxcjk1013_data *data = iio_priv(indio_dev);
992 
993 	*val2 = 0;
994 	switch (info) {
995 	case IIO_EV_INFO_VALUE:
996 		*val = data->wake_thres;
997 		break;
998 	case IIO_EV_INFO_PERIOD:
999 		*val = data->wake_dur;
1000 		break;
1001 	default:
1002 		return -EINVAL;
1003 	}
1004 
1005 	return IIO_VAL_INT;
1006 }
1007 
1008 static int kxcjk1013_write_event(struct iio_dev *indio_dev,
1009 				    const struct iio_chan_spec *chan,
1010 				    enum iio_event_type type,
1011 				    enum iio_event_direction dir,
1012 				    enum iio_event_info info,
1013 				    int val, int val2)
1014 {
1015 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1016 
1017 	if (data->ev_enable_state)
1018 		return -EBUSY;
1019 
1020 	switch (info) {
1021 	case IIO_EV_INFO_VALUE:
1022 		data->wake_thres = val;
1023 		break;
1024 	case IIO_EV_INFO_PERIOD:
1025 		data->wake_dur = val;
1026 		break;
1027 	default:
1028 		return -EINVAL;
1029 	}
1030 
1031 	return 0;
1032 }
1033 
1034 static int kxcjk1013_read_event_config(struct iio_dev *indio_dev,
1035 					  const struct iio_chan_spec *chan,
1036 					  enum iio_event_type type,
1037 					  enum iio_event_direction dir)
1038 {
1039 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1040 
1041 	return data->ev_enable_state;
1042 }
1043 
1044 static int kxcjk1013_write_event_config(struct iio_dev *indio_dev,
1045 					   const struct iio_chan_spec *chan,
1046 					   enum iio_event_type type,
1047 					   enum iio_event_direction dir,
1048 					   int state)
1049 {
1050 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1051 	int ret;
1052 
1053 	if (state && data->ev_enable_state)
1054 		return 0;
1055 
1056 	mutex_lock(&data->mutex);
1057 
1058 	if (!state && data->motion_trigger_on) {
1059 		data->ev_enable_state = 0;
1060 		mutex_unlock(&data->mutex);
1061 		return 0;
1062 	}
1063 
1064 	/*
1065 	 * We will expect the enable and disable to do operation in
1066 	 * reverse order. This will happen here anyway as our
1067 	 * resume operation uses sync mode runtime pm calls, the
1068 	 * suspend operation will be delayed by autosuspend delay
1069 	 * So the disable operation will still happen in reverse of
1070 	 * enable operation. When runtime pm is disabled the mode
1071 	 * is always on so sequence doesn't matter
1072 	 */
1073 	ret = kxcjk1013_set_power_state(data, state);
1074 	if (ret < 0) {
1075 		mutex_unlock(&data->mutex);
1076 		return ret;
1077 	}
1078 
1079 	ret =  kxcjk1013_setup_any_motion_interrupt(data, state);
1080 	if (ret < 0) {
1081 		kxcjk1013_set_power_state(data, false);
1082 		data->ev_enable_state = 0;
1083 		mutex_unlock(&data->mutex);
1084 		return ret;
1085 	}
1086 
1087 	data->ev_enable_state = state;
1088 	mutex_unlock(&data->mutex);
1089 
1090 	return 0;
1091 }
1092 
1093 static int kxcjk1013_buffer_preenable(struct iio_dev *indio_dev)
1094 {
1095 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1096 
1097 	return kxcjk1013_set_power_state(data, true);
1098 }
1099 
1100 static int kxcjk1013_buffer_postdisable(struct iio_dev *indio_dev)
1101 {
1102 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1103 
1104 	return kxcjk1013_set_power_state(data, false);
1105 }
1106 
1107 static ssize_t kxcjk1013_get_samp_freq_avail(struct device *dev,
1108 					     struct device_attribute *attr,
1109 					     char *buf)
1110 {
1111 	struct iio_dev *indio_dev = dev_to_iio_dev(dev);
1112 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1113 	const char *str;
1114 
1115 	if (data->chipset == KXTF9)
1116 		str = kxtf9_samp_freq_avail;
1117 	else
1118 		str = kxcjk1013_samp_freq_avail;
1119 
1120 	return sprintf(buf, "%s\n", str);
1121 }
1122 
1123 static IIO_DEVICE_ATTR(in_accel_sampling_frequency_available, S_IRUGO,
1124 		       kxcjk1013_get_samp_freq_avail, NULL, 0);
1125 
1126 static IIO_CONST_ATTR(in_accel_scale_available, "0.009582 0.019163 0.038326");
1127 
1128 static struct attribute *kxcjk1013_attributes[] = {
1129 	&iio_dev_attr_in_accel_sampling_frequency_available.dev_attr.attr,
1130 	&iio_const_attr_in_accel_scale_available.dev_attr.attr,
1131 	NULL,
1132 };
1133 
1134 static const struct attribute_group kxcjk1013_attrs_group = {
1135 	.attrs = kxcjk1013_attributes,
1136 };
1137 
1138 static const struct iio_event_spec kxcjk1013_event = {
1139 		.type = IIO_EV_TYPE_THRESH,
1140 		.dir = IIO_EV_DIR_EITHER,
1141 		.mask_separate = BIT(IIO_EV_INFO_VALUE) |
1142 				 BIT(IIO_EV_INFO_ENABLE) |
1143 				 BIT(IIO_EV_INFO_PERIOD)
1144 };
1145 
1146 static const struct iio_mount_matrix *
1147 kxcjk1013_get_mount_matrix(const struct iio_dev *indio_dev,
1148 			   const struct iio_chan_spec *chan)
1149 {
1150 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1151 
1152 	return &data->orientation;
1153 }
1154 
1155 static const struct iio_chan_spec_ext_info kxcjk1013_ext_info[] = {
1156 	IIO_MOUNT_MATRIX(IIO_SHARED_BY_TYPE, kxcjk1013_get_mount_matrix),
1157 	{ }
1158 };
1159 
1160 #define KXCJK1013_CHANNEL(_axis) {					\
1161 	.type = IIO_ACCEL,						\
1162 	.modified = 1,							\
1163 	.channel2 = IIO_MOD_##_axis,					\
1164 	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),			\
1165 	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE) |		\
1166 				BIT(IIO_CHAN_INFO_SAMP_FREQ),		\
1167 	.scan_index = AXIS_##_axis,					\
1168 	.scan_type = {							\
1169 		.sign = 's',						\
1170 		.realbits = 12,						\
1171 		.storagebits = 16,					\
1172 		.shift = 4,						\
1173 		.endianness = IIO_LE,					\
1174 	},								\
1175 	.event_spec = &kxcjk1013_event,				\
1176 	.ext_info = kxcjk1013_ext_info,					\
1177 	.num_event_specs = 1						\
1178 }
1179 
1180 static const struct iio_chan_spec kxcjk1013_channels[] = {
1181 	KXCJK1013_CHANNEL(X),
1182 	KXCJK1013_CHANNEL(Y),
1183 	KXCJK1013_CHANNEL(Z),
1184 	IIO_CHAN_SOFT_TIMESTAMP(3),
1185 };
1186 
1187 static const struct iio_buffer_setup_ops kxcjk1013_buffer_setup_ops = {
1188 	.preenable		= kxcjk1013_buffer_preenable,
1189 	.postdisable		= kxcjk1013_buffer_postdisable,
1190 };
1191 
1192 static const struct iio_info kxcjk1013_info = {
1193 	.attrs			= &kxcjk1013_attrs_group,
1194 	.read_raw		= kxcjk1013_read_raw,
1195 	.write_raw		= kxcjk1013_write_raw,
1196 	.read_event_value	= kxcjk1013_read_event,
1197 	.write_event_value	= kxcjk1013_write_event,
1198 	.write_event_config	= kxcjk1013_write_event_config,
1199 	.read_event_config	= kxcjk1013_read_event_config,
1200 };
1201 
1202 static const unsigned long kxcjk1013_scan_masks[] = {0x7, 0};
1203 
1204 static irqreturn_t kxcjk1013_trigger_handler(int irq, void *p)
1205 {
1206 	struct iio_poll_func *pf = p;
1207 	struct iio_dev *indio_dev = pf->indio_dev;
1208 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1209 	int ret;
1210 
1211 	mutex_lock(&data->mutex);
1212 	ret = i2c_smbus_read_i2c_block_data_or_emulated(data->client,
1213 							KXCJK1013_REG_XOUT_L,
1214 							AXIS_MAX * 2,
1215 							(u8 *)data->scan.chans);
1216 	mutex_unlock(&data->mutex);
1217 	if (ret < 0)
1218 		goto err;
1219 
1220 	iio_push_to_buffers_with_timestamp(indio_dev, &data->scan,
1221 					   data->timestamp);
1222 err:
1223 	iio_trigger_notify_done(indio_dev->trig);
1224 
1225 	return IRQ_HANDLED;
1226 }
1227 
1228 static void kxcjk1013_trig_reen(struct iio_trigger *trig)
1229 {
1230 	struct iio_dev *indio_dev = iio_trigger_get_drvdata(trig);
1231 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1232 	int ret;
1233 
1234 	ret = i2c_smbus_read_byte_data(data->client, data->regs->int_rel);
1235 	if (ret < 0)
1236 		dev_err(&data->client->dev, "Error reading reg_int_rel\n");
1237 }
1238 
1239 static int kxcjk1013_data_rdy_trigger_set_state(struct iio_trigger *trig,
1240 						bool state)
1241 {
1242 	struct iio_dev *indio_dev = iio_trigger_get_drvdata(trig);
1243 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1244 	int ret;
1245 
1246 	mutex_lock(&data->mutex);
1247 
1248 	if (!state && data->ev_enable_state && data->motion_trigger_on) {
1249 		data->motion_trigger_on = false;
1250 		mutex_unlock(&data->mutex);
1251 		return 0;
1252 	}
1253 
1254 	ret = kxcjk1013_set_power_state(data, state);
1255 	if (ret < 0) {
1256 		mutex_unlock(&data->mutex);
1257 		return ret;
1258 	}
1259 	if (data->motion_trig == trig)
1260 		ret = kxcjk1013_setup_any_motion_interrupt(data, state);
1261 	else
1262 		ret = kxcjk1013_setup_new_data_interrupt(data, state);
1263 	if (ret < 0) {
1264 		kxcjk1013_set_power_state(data, false);
1265 		mutex_unlock(&data->mutex);
1266 		return ret;
1267 	}
1268 	if (data->motion_trig == trig)
1269 		data->motion_trigger_on = state;
1270 	else
1271 		data->dready_trigger_on = state;
1272 
1273 	mutex_unlock(&data->mutex);
1274 
1275 	return 0;
1276 }
1277 
1278 static const struct iio_trigger_ops kxcjk1013_trigger_ops = {
1279 	.set_trigger_state = kxcjk1013_data_rdy_trigger_set_state,
1280 	.reenable = kxcjk1013_trig_reen,
1281 };
1282 
1283 static void kxcjk1013_report_motion_event(struct iio_dev *indio_dev)
1284 {
1285 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1286 
1287 	int ret = i2c_smbus_read_byte_data(data->client, data->regs->int_src2);
1288 	if (ret < 0) {
1289 		dev_err(&data->client->dev, "Error reading reg_int_src2\n");
1290 		return;
1291 	}
1292 
1293 	if (ret & KXCJK1013_REG_INT_SRC2_BIT_XN)
1294 		iio_push_event(indio_dev,
1295 			       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1296 						  0,
1297 						  IIO_MOD_X,
1298 						  IIO_EV_TYPE_THRESH,
1299 						  IIO_EV_DIR_FALLING),
1300 			       data->timestamp);
1301 
1302 	if (ret & KXCJK1013_REG_INT_SRC2_BIT_XP)
1303 		iio_push_event(indio_dev,
1304 			       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1305 						  0,
1306 						  IIO_MOD_X,
1307 						  IIO_EV_TYPE_THRESH,
1308 						  IIO_EV_DIR_RISING),
1309 			       data->timestamp);
1310 
1311 	if (ret & KXCJK1013_REG_INT_SRC2_BIT_YN)
1312 		iio_push_event(indio_dev,
1313 			       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1314 						  0,
1315 						  IIO_MOD_Y,
1316 						  IIO_EV_TYPE_THRESH,
1317 						  IIO_EV_DIR_FALLING),
1318 			       data->timestamp);
1319 
1320 	if (ret & KXCJK1013_REG_INT_SRC2_BIT_YP)
1321 		iio_push_event(indio_dev,
1322 			       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1323 						  0,
1324 						  IIO_MOD_Y,
1325 						  IIO_EV_TYPE_THRESH,
1326 						  IIO_EV_DIR_RISING),
1327 			       data->timestamp);
1328 
1329 	if (ret & KXCJK1013_REG_INT_SRC2_BIT_ZN)
1330 		iio_push_event(indio_dev,
1331 			       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1332 						  0,
1333 						  IIO_MOD_Z,
1334 						  IIO_EV_TYPE_THRESH,
1335 						  IIO_EV_DIR_FALLING),
1336 			       data->timestamp);
1337 
1338 	if (ret & KXCJK1013_REG_INT_SRC2_BIT_ZP)
1339 		iio_push_event(indio_dev,
1340 			       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1341 						  0,
1342 						  IIO_MOD_Z,
1343 						  IIO_EV_TYPE_THRESH,
1344 						  IIO_EV_DIR_RISING),
1345 			       data->timestamp);
1346 }
1347 
1348 static irqreturn_t kxcjk1013_event_handler(int irq, void *private)
1349 {
1350 	struct iio_dev *indio_dev = private;
1351 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1352 	int ret;
1353 
1354 	ret = i2c_smbus_read_byte_data(data->client, data->regs->int_src1);
1355 	if (ret < 0) {
1356 		dev_err(&data->client->dev, "Error reading reg_int_src1\n");
1357 		goto ack_intr;
1358 	}
1359 
1360 	if (ret & KXCJK1013_REG_INT_SRC1_BIT_WUFS) {
1361 		if (data->chipset == KXTF9)
1362 			iio_push_event(indio_dev,
1363 				       IIO_MOD_EVENT_CODE(IIO_ACCEL,
1364 				       0,
1365 				       IIO_MOD_X_AND_Y_AND_Z,
1366 				       IIO_EV_TYPE_THRESH,
1367 				       IIO_EV_DIR_RISING),
1368 				       data->timestamp);
1369 		else
1370 			kxcjk1013_report_motion_event(indio_dev);
1371 	}
1372 
1373 ack_intr:
1374 	if (data->dready_trigger_on)
1375 		return IRQ_HANDLED;
1376 
1377 	ret = i2c_smbus_read_byte_data(data->client, data->regs->int_rel);
1378 	if (ret < 0)
1379 		dev_err(&data->client->dev, "Error reading reg_int_rel\n");
1380 
1381 	return IRQ_HANDLED;
1382 }
1383 
1384 static irqreturn_t kxcjk1013_data_rdy_trig_poll(int irq, void *private)
1385 {
1386 	struct iio_dev *indio_dev = private;
1387 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1388 
1389 	data->timestamp = iio_get_time_ns(indio_dev);
1390 
1391 	if (data->dready_trigger_on)
1392 		iio_trigger_poll(data->dready_trig);
1393 	else if (data->motion_trigger_on)
1394 		iio_trigger_poll(data->motion_trig);
1395 
1396 	if (data->ev_enable_state)
1397 		return IRQ_WAKE_THREAD;
1398 	else
1399 		return IRQ_HANDLED;
1400 }
1401 
1402 static const char *kxcjk1013_match_acpi_device(struct device *dev,
1403 					       enum kx_chipset *chipset,
1404 					       enum kx_acpi_type *acpi_type,
1405 					       const char **label)
1406 {
1407 	const struct acpi_device_id *id;
1408 
1409 	id = acpi_match_device(dev->driver->acpi_match_table, dev);
1410 	if (!id)
1411 		return NULL;
1412 
1413 	if (strcmp(id->id, "SMO8500") == 0) {
1414 		*acpi_type = ACPI_SMO8500;
1415 	} else if (strcmp(id->id, "KIOX010A") == 0) {
1416 		*acpi_type = ACPI_KIOX010A;
1417 		*label = "accel-display";
1418 	} else if (strcmp(id->id, "KIOX020A") == 0) {
1419 		*label = "accel-base";
1420 	}
1421 
1422 	*chipset = (enum kx_chipset)id->driver_data;
1423 
1424 	return dev_name(dev);
1425 }
1426 
1427 static int kxcjk1013_probe(struct i2c_client *client)
1428 {
1429 	const struct i2c_device_id *id = i2c_client_get_device_id(client);
1430 	static const char * const regulator_names[] = { "vdd", "vddio" };
1431 	struct kxcjk1013_data *data;
1432 	struct iio_dev *indio_dev;
1433 	struct kxcjk_1013_platform_data *pdata;
1434 	const char *name;
1435 	int ret;
1436 
1437 	indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
1438 	if (!indio_dev)
1439 		return -ENOMEM;
1440 
1441 	data = iio_priv(indio_dev);
1442 	i2c_set_clientdata(client, indio_dev);
1443 	data->client = client;
1444 
1445 	pdata = dev_get_platdata(&client->dev);
1446 	if (pdata) {
1447 		data->active_high_intr = pdata->active_high_intr;
1448 		data->orientation = pdata->orientation;
1449 	} else {
1450 		data->active_high_intr = true; /* default polarity */
1451 
1452 		ret = iio_read_mount_matrix(&client->dev, &data->orientation);
1453 		if (ret)
1454 			return ret;
1455 	}
1456 
1457 	ret = devm_regulator_bulk_get_enable(&client->dev,
1458 					     ARRAY_SIZE(regulator_names),
1459 					     regulator_names);
1460 	if (ret)
1461 		return dev_err_probe(&client->dev, ret, "Failed to get regulators\n");
1462 
1463 	/*
1464 	 * A typical delay of 10ms is required for powering up
1465 	 * according to the data sheets of supported chips.
1466 	 * Hence double that to play safe.
1467 	 */
1468 	msleep(20);
1469 
1470 	if (id) {
1471 		data->chipset = (enum kx_chipset)(id->driver_data);
1472 		name = id->name;
1473 	} else if (ACPI_HANDLE(&client->dev)) {
1474 		name = kxcjk1013_match_acpi_device(&client->dev,
1475 						   &data->chipset,
1476 						   &data->acpi_type,
1477 						   &indio_dev->label);
1478 	} else
1479 		return -ENODEV;
1480 
1481 	switch (data->chipset) {
1482 	case KXCJK1013:
1483 	case KXCJ91008:
1484 	case KXTJ21009:
1485 		data->regs = &kxcjk1013_regs;
1486 		break;
1487 	case KXTF9:
1488 		data->regs = &kxtf9_regs;
1489 		break;
1490 	case KX0231025:
1491 		data->regs = &kx0231025_regs;
1492 		break;
1493 	default:
1494 		return -EINVAL;
1495 	}
1496 
1497 	ret = kxcjk1013_chip_init(data);
1498 	if (ret < 0)
1499 		return ret;
1500 
1501 	mutex_init(&data->mutex);
1502 
1503 	indio_dev->channels = kxcjk1013_channels;
1504 	indio_dev->num_channels = ARRAY_SIZE(kxcjk1013_channels);
1505 	indio_dev->available_scan_masks = kxcjk1013_scan_masks;
1506 	indio_dev->name = name;
1507 	indio_dev->modes = INDIO_DIRECT_MODE;
1508 	indio_dev->info = &kxcjk1013_info;
1509 
1510 	if (client->irq > 0 && data->acpi_type != ACPI_SMO8500) {
1511 		ret = devm_request_threaded_irq(&client->dev, client->irq,
1512 						kxcjk1013_data_rdy_trig_poll,
1513 						kxcjk1013_event_handler,
1514 						IRQF_TRIGGER_RISING,
1515 						KXCJK1013_IRQ_NAME,
1516 						indio_dev);
1517 		if (ret)
1518 			goto err_poweroff;
1519 
1520 		data->dready_trig = devm_iio_trigger_alloc(&client->dev,
1521 							   "%s-dev%d",
1522 							   indio_dev->name,
1523 							   iio_device_id(indio_dev));
1524 		if (!data->dready_trig) {
1525 			ret = -ENOMEM;
1526 			goto err_poweroff;
1527 		}
1528 
1529 		data->motion_trig = devm_iio_trigger_alloc(&client->dev,
1530 							  "%s-any-motion-dev%d",
1531 							  indio_dev->name,
1532 							  iio_device_id(indio_dev));
1533 		if (!data->motion_trig) {
1534 			ret = -ENOMEM;
1535 			goto err_poweroff;
1536 		}
1537 
1538 		data->dready_trig->ops = &kxcjk1013_trigger_ops;
1539 		iio_trigger_set_drvdata(data->dready_trig, indio_dev);
1540 		ret = iio_trigger_register(data->dready_trig);
1541 		if (ret)
1542 			goto err_poweroff;
1543 
1544 		indio_dev->trig = iio_trigger_get(data->dready_trig);
1545 
1546 		data->motion_trig->ops = &kxcjk1013_trigger_ops;
1547 		iio_trigger_set_drvdata(data->motion_trig, indio_dev);
1548 		ret = iio_trigger_register(data->motion_trig);
1549 		if (ret) {
1550 			data->motion_trig = NULL;
1551 			goto err_trigger_unregister;
1552 		}
1553 	}
1554 
1555 	ret = iio_triggered_buffer_setup(indio_dev,
1556 					 &iio_pollfunc_store_time,
1557 					 kxcjk1013_trigger_handler,
1558 					 &kxcjk1013_buffer_setup_ops);
1559 	if (ret < 0) {
1560 		dev_err(&client->dev, "iio triggered buffer setup failed\n");
1561 		goto err_trigger_unregister;
1562 	}
1563 
1564 	ret = pm_runtime_set_active(&client->dev);
1565 	if (ret)
1566 		goto err_buffer_cleanup;
1567 
1568 	pm_runtime_enable(&client->dev);
1569 	pm_runtime_set_autosuspend_delay(&client->dev,
1570 					 KXCJK1013_SLEEP_DELAY_MS);
1571 	pm_runtime_use_autosuspend(&client->dev);
1572 
1573 	ret = iio_device_register(indio_dev);
1574 	if (ret < 0) {
1575 		dev_err(&client->dev, "unable to register iio device\n");
1576 		goto err_pm_cleanup;
1577 	}
1578 
1579 	return 0;
1580 
1581 err_pm_cleanup:
1582 	pm_runtime_dont_use_autosuspend(&client->dev);
1583 	pm_runtime_disable(&client->dev);
1584 err_buffer_cleanup:
1585 	iio_triggered_buffer_cleanup(indio_dev);
1586 err_trigger_unregister:
1587 	if (data->dready_trig)
1588 		iio_trigger_unregister(data->dready_trig);
1589 	if (data->motion_trig)
1590 		iio_trigger_unregister(data->motion_trig);
1591 err_poweroff:
1592 	kxcjk1013_set_mode(data, STANDBY);
1593 
1594 	return ret;
1595 }
1596 
1597 static void kxcjk1013_remove(struct i2c_client *client)
1598 {
1599 	struct iio_dev *indio_dev = i2c_get_clientdata(client);
1600 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1601 
1602 	iio_device_unregister(indio_dev);
1603 
1604 	pm_runtime_disable(&client->dev);
1605 	pm_runtime_set_suspended(&client->dev);
1606 
1607 	iio_triggered_buffer_cleanup(indio_dev);
1608 	if (data->dready_trig) {
1609 		iio_trigger_unregister(data->dready_trig);
1610 		iio_trigger_unregister(data->motion_trig);
1611 	}
1612 
1613 	mutex_lock(&data->mutex);
1614 	kxcjk1013_set_mode(data, STANDBY);
1615 	mutex_unlock(&data->mutex);
1616 }
1617 
1618 #ifdef CONFIG_PM_SLEEP
1619 static int kxcjk1013_suspend(struct device *dev)
1620 {
1621 	struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
1622 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1623 	int ret;
1624 
1625 	mutex_lock(&data->mutex);
1626 	ret = kxcjk1013_set_mode(data, STANDBY);
1627 	mutex_unlock(&data->mutex);
1628 
1629 	return ret;
1630 }
1631 
1632 static int kxcjk1013_resume(struct device *dev)
1633 {
1634 	struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
1635 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1636 	int ret = 0;
1637 
1638 	mutex_lock(&data->mutex);
1639 	ret = kxcjk1013_set_mode(data, OPERATION);
1640 	if (ret == 0)
1641 		ret = kxcjk1013_set_range(data, data->range);
1642 	mutex_unlock(&data->mutex);
1643 
1644 	return ret;
1645 }
1646 #endif
1647 
1648 #ifdef CONFIG_PM
1649 static int kxcjk1013_runtime_suspend(struct device *dev)
1650 {
1651 	struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
1652 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1653 	int ret;
1654 
1655 	ret = kxcjk1013_set_mode(data, STANDBY);
1656 	if (ret < 0) {
1657 		dev_err(&data->client->dev, "powering off device failed\n");
1658 		return -EAGAIN;
1659 	}
1660 	return 0;
1661 }
1662 
1663 static int kxcjk1013_runtime_resume(struct device *dev)
1664 {
1665 	struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
1666 	struct kxcjk1013_data *data = iio_priv(indio_dev);
1667 	int ret;
1668 	int sleep_val;
1669 
1670 	ret = kxcjk1013_set_mode(data, OPERATION);
1671 	if (ret < 0)
1672 		return ret;
1673 
1674 	sleep_val = kxcjk1013_get_startup_times(data);
1675 	if (sleep_val < 20000)
1676 		usleep_range(sleep_val, 20000);
1677 	else
1678 		msleep_interruptible(sleep_val/1000);
1679 
1680 	return 0;
1681 }
1682 #endif
1683 
1684 static const struct dev_pm_ops kxcjk1013_pm_ops = {
1685 	SET_SYSTEM_SLEEP_PM_OPS(kxcjk1013_suspend, kxcjk1013_resume)
1686 	SET_RUNTIME_PM_OPS(kxcjk1013_runtime_suspend,
1687 			   kxcjk1013_runtime_resume, NULL)
1688 };
1689 
1690 static const struct acpi_device_id kx_acpi_match[] = {
1691 	{"KXCJ1013", KXCJK1013},
1692 	{"KXCJ1008", KXCJ91008},
1693 	{"KXCJ9000", KXCJ91008},
1694 	{"KIOX0008", KXCJ91008},
1695 	{"KIOX0009", KXTJ21009},
1696 	{"KIOX000A", KXCJ91008},
1697 	{"KIOX010A", KXCJ91008}, /* KXCJ91008 in the display of a yoga 2-in-1 */
1698 	{"KIOX020A", KXCJ91008}, /* KXCJ91008 in the base of a yoga 2-in-1 */
1699 	{"KXTJ1009", KXTJ21009},
1700 	{"KXJ2109",  KXTJ21009},
1701 	{"SMO8500",  KXCJ91008},
1702 	{ },
1703 };
1704 MODULE_DEVICE_TABLE(acpi, kx_acpi_match);
1705 
1706 static const struct i2c_device_id kxcjk1013_id[] = {
1707 	{"kxcjk1013", KXCJK1013},
1708 	{"kxcj91008", KXCJ91008},
1709 	{"kxtj21009", KXTJ21009},
1710 	{"kxtf9",     KXTF9},
1711 	{"kx023-1025", KX0231025},
1712 	{"SMO8500",   KXCJ91008},
1713 	{}
1714 };
1715 
1716 MODULE_DEVICE_TABLE(i2c, kxcjk1013_id);
1717 
1718 static const struct of_device_id kxcjk1013_of_match[] = {
1719 	{ .compatible = "kionix,kxcjk1013", },
1720 	{ .compatible = "kionix,kxcj91008", },
1721 	{ .compatible = "kionix,kxtj21009", },
1722 	{ .compatible = "kionix,kxtf9", },
1723 	{ .compatible = "kionix,kx023-1025", },
1724 	{ }
1725 };
1726 MODULE_DEVICE_TABLE(of, kxcjk1013_of_match);
1727 
1728 static struct i2c_driver kxcjk1013_driver = {
1729 	.driver = {
1730 		.name	= KXCJK1013_DRV_NAME,
1731 		.acpi_match_table = ACPI_PTR(kx_acpi_match),
1732 		.of_match_table = kxcjk1013_of_match,
1733 		.pm	= &kxcjk1013_pm_ops,
1734 	},
1735 	.probe		= kxcjk1013_probe,
1736 	.remove		= kxcjk1013_remove,
1737 	.id_table	= kxcjk1013_id,
1738 };
1739 module_i2c_driver(kxcjk1013_driver);
1740 
1741 MODULE_AUTHOR("Srinivas Pandruvada <srinivas.pandruvada@linux.intel.com>");
1742 MODULE_LICENSE("GPL v2");
1743 MODULE_DESCRIPTION("KXCJK1013 accelerometer driver");
1744