1 /*
2  * Driver for the Asahi Kasei EMD Corporation AK8974
3  * and Aichi Steel AMI305 magnetometer chips.
4  * Based on a patch from Samu Onkalo and the AK8975 IIO driver.
5  *
6  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
7  * Copyright (c) 2010 NVIDIA Corporation.
8  * Copyright (C) 2016 Linaro Ltd.
9  *
10  * Author: Samu Onkalo <samu.p.onkalo@nokia.com>
11  * Author: Linus Walleij <linus.walleij@linaro.org>
12  */
13 #include <linux/module.h>
14 #include <linux/kernel.h>
15 #include <linux/i2c.h>
16 #include <linux/interrupt.h>
17 #include <linux/irq.h> /* For irq_get_irq_data() */
18 #include <linux/completion.h>
19 #include <linux/err.h>
20 #include <linux/mutex.h>
21 #include <linux/delay.h>
22 #include <linux/bitops.h>
23 #include <linux/random.h>
24 #include <linux/regmap.h>
25 #include <linux/regulator/consumer.h>
26 #include <linux/pm_runtime.h>
27 
28 #include <linux/iio/iio.h>
29 #include <linux/iio/sysfs.h>
30 #include <linux/iio/buffer.h>
31 #include <linux/iio/trigger.h>
32 #include <linux/iio/trigger_consumer.h>
33 #include <linux/iio/triggered_buffer.h>
34 
35 /*
36  * 16-bit registers are little-endian. LSB is at the address defined below
37  * and MSB is at the next higher address.
38  */
39 
40 /* These registers are common for AK8974 and AMI30x */
41 #define AK8974_SELFTEST		0x0C
42 #define AK8974_SELFTEST_IDLE	0x55
43 #define AK8974_SELFTEST_OK	0xAA
44 
45 #define AK8974_INFO		0x0D
46 
47 #define AK8974_WHOAMI		0x0F
48 #define AK8974_WHOAMI_VALUE_AMI306 0x46
49 #define AK8974_WHOAMI_VALUE_AMI305 0x47
50 #define AK8974_WHOAMI_VALUE_AK8974 0x48
51 
52 #define AK8974_DATA_X		0x10
53 #define AK8974_DATA_Y		0x12
54 #define AK8974_DATA_Z		0x14
55 #define AK8974_INT_SRC		0x16
56 #define AK8974_STATUS		0x18
57 #define AK8974_INT_CLEAR	0x1A
58 #define AK8974_CTRL1		0x1B
59 #define AK8974_CTRL2		0x1C
60 #define AK8974_CTRL3		0x1D
61 #define AK8974_INT_CTRL		0x1E
62 #define AK8974_INT_THRES	0x26  /* Absolute any axis value threshold */
63 #define AK8974_PRESET		0x30
64 
65 /* AK8974-specific offsets */
66 #define AK8974_OFFSET_X		0x20
67 #define AK8974_OFFSET_Y		0x22
68 #define AK8974_OFFSET_Z		0x24
69 /* AMI305-specific offsets */
70 #define AMI305_OFFSET_X		0x6C
71 #define AMI305_OFFSET_Y		0x72
72 #define AMI305_OFFSET_Z		0x78
73 
74 /* Different temperature registers */
75 #define AK8974_TEMP		0x31
76 #define AMI305_TEMP		0x60
77 
78 /* AMI306-specific control register */
79 #define AMI306_CTRL4		0x5C
80 
81 /* AMI306 factory calibration data */
82 
83 /* fine axis sensitivity */
84 #define AMI306_FINEOUTPUT_X	0x90
85 #define AMI306_FINEOUTPUT_Y	0x92
86 #define AMI306_FINEOUTPUT_Z	0x94
87 
88 /* axis sensitivity */
89 #define AMI306_SENS_X		0x96
90 #define AMI306_SENS_Y		0x98
91 #define AMI306_SENS_Z		0x9A
92 
93 /* axis cross-interference */
94 #define AMI306_GAIN_PARA_XZ	0x9C
95 #define AMI306_GAIN_PARA_XY	0x9D
96 #define AMI306_GAIN_PARA_YZ	0x9E
97 #define AMI306_GAIN_PARA_YX	0x9F
98 #define AMI306_GAIN_PARA_ZY	0xA0
99 #define AMI306_GAIN_PARA_ZX	0xA1
100 
101 /* offset at ZERO magnetic field */
102 #define AMI306_OFFZERO_X	0xF8
103 #define AMI306_OFFZERO_Y	0xFA
104 #define AMI306_OFFZERO_Z	0xFC
105 
106 
107 #define AK8974_INT_X_HIGH	BIT(7) /* Axis over +threshold  */
108 #define AK8974_INT_Y_HIGH	BIT(6)
109 #define AK8974_INT_Z_HIGH	BIT(5)
110 #define AK8974_INT_X_LOW	BIT(4) /* Axis below -threshold	*/
111 #define AK8974_INT_Y_LOW	BIT(3)
112 #define AK8974_INT_Z_LOW	BIT(2)
113 #define AK8974_INT_RANGE	BIT(1) /* Range overflow (any axis) */
114 
115 #define AK8974_STATUS_DRDY	BIT(6) /* Data ready */
116 #define AK8974_STATUS_OVERRUN	BIT(5) /* Data overrun */
117 #define AK8974_STATUS_INT	BIT(4) /* Interrupt occurred */
118 
119 #define AK8974_CTRL1_POWER	BIT(7) /* 0 = standby; 1 = active */
120 #define AK8974_CTRL1_RATE	BIT(4) /* 0 = 10 Hz; 1 = 20 Hz	 */
121 #define AK8974_CTRL1_FORCE_EN	BIT(1) /* 0 = normal; 1 = force	 */
122 #define AK8974_CTRL1_MODE2	BIT(0) /* 0 */
123 
124 #define AK8974_CTRL2_INT_EN	BIT(4)  /* 1 = enable interrupts	      */
125 #define AK8974_CTRL2_DRDY_EN	BIT(3)  /* 1 = enable data ready signal */
126 #define AK8974_CTRL2_DRDY_POL	BIT(2)  /* 1 = data ready active high   */
127 #define AK8974_CTRL2_RESDEF	(AK8974_CTRL2_DRDY_POL)
128 
129 #define AK8974_CTRL3_RESET	BIT(7) /* Software reset		  */
130 #define AK8974_CTRL3_FORCE	BIT(6) /* Start forced measurement */
131 #define AK8974_CTRL3_SELFTEST	BIT(4) /* Set selftest register	  */
132 #define AK8974_CTRL3_RESDEF	0x00
133 
134 #define AK8974_INT_CTRL_XEN	BIT(7) /* Enable interrupt for this axis */
135 #define AK8974_INT_CTRL_YEN	BIT(6)
136 #define AK8974_INT_CTRL_ZEN	BIT(5)
137 #define AK8974_INT_CTRL_XYZEN	(BIT(7)|BIT(6)|BIT(5))
138 #define AK8974_INT_CTRL_POL	BIT(3) /* 0 = active low; 1 = active high */
139 #define AK8974_INT_CTRL_PULSE	BIT(1) /* 0 = latched; 1 = pulse (50 usec) */
140 #define AK8974_INT_CTRL_RESDEF	(AK8974_INT_CTRL_XYZEN | AK8974_INT_CTRL_POL)
141 
142 /* The AMI305 has elaborate FW version and serial number registers */
143 #define AMI305_VER		0xE8
144 #define AMI305_SN		0xEA
145 
146 #define AK8974_MAX_RANGE	2048
147 
148 #define AK8974_POWERON_DELAY	50
149 #define AK8974_ACTIVATE_DELAY	1
150 #define AK8974_SELFTEST_DELAY	1
151 /*
152  * Set the autosuspend to two orders of magnitude larger than the poweron
153  * delay to make sane reasonable power tradeoff savings (5 seconds in
154  * this case).
155  */
156 #define AK8974_AUTOSUSPEND_DELAY 5000
157 
158 #define AK8974_MEASTIME		3
159 
160 #define AK8974_PWR_ON		1
161 #define AK8974_PWR_OFF		0
162 
163 /**
164  * struct ak8974 - state container for the AK8974 driver
165  * @i2c: parent I2C client
166  * @orientation: mounting matrix, flipped axis etc
167  * @map: regmap to access the AK8974 registers over I2C
168  * @regs: the avdd and dvdd power regulators
169  * @name: the name of the part
170  * @variant: the whoami ID value (for selecting code paths)
171  * @lock: locks the magnetometer for exclusive use during a measurement
172  * @drdy_irq: uses the DRDY IRQ line
173  * @drdy_complete: completion for DRDY
174  * @drdy_active_low: the DRDY IRQ is active low
175  */
176 struct ak8974 {
177 	struct i2c_client *i2c;
178 	struct iio_mount_matrix orientation;
179 	struct regmap *map;
180 	struct regulator_bulk_data regs[2];
181 	const char *name;
182 	u8 variant;
183 	struct mutex lock;
184 	bool drdy_irq;
185 	struct completion drdy_complete;
186 	bool drdy_active_low;
187 };
188 
189 static const char ak8974_reg_avdd[] = "avdd";
190 static const char ak8974_reg_dvdd[] = "dvdd";
191 
192 static int ak8974_get_u16_val(struct ak8974 *ak8974, u8 reg, u16 *val)
193 {
194 	int ret;
195 	__le16 bulk;
196 
197 	ret = regmap_bulk_read(ak8974->map, reg, &bulk, 2);
198 	if (ret)
199 		return ret;
200 	*val = le16_to_cpu(bulk);
201 
202 	return 0;
203 }
204 
205 static int ak8974_set_u16_val(struct ak8974 *ak8974, u8 reg, u16 val)
206 {
207 	__le16 bulk = cpu_to_le16(val);
208 
209 	return regmap_bulk_write(ak8974->map, reg, &bulk, 2);
210 }
211 
212 static int ak8974_set_power(struct ak8974 *ak8974, bool mode)
213 {
214 	int ret;
215 	u8 val;
216 
217 	val = mode ? AK8974_CTRL1_POWER : 0;
218 	val |= AK8974_CTRL1_FORCE_EN;
219 	ret = regmap_write(ak8974->map, AK8974_CTRL1, val);
220 	if (ret < 0)
221 		return ret;
222 
223 	if (mode)
224 		msleep(AK8974_ACTIVATE_DELAY);
225 
226 	return 0;
227 }
228 
229 static int ak8974_reset(struct ak8974 *ak8974)
230 {
231 	int ret;
232 
233 	/* Power on to get register access. Sets CTRL1 reg to reset state */
234 	ret = ak8974_set_power(ak8974, AK8974_PWR_ON);
235 	if (ret)
236 		return ret;
237 	ret = regmap_write(ak8974->map, AK8974_CTRL2, AK8974_CTRL2_RESDEF);
238 	if (ret)
239 		return ret;
240 	ret = regmap_write(ak8974->map, AK8974_CTRL3, AK8974_CTRL3_RESDEF);
241 	if (ret)
242 		return ret;
243 	ret = regmap_write(ak8974->map, AK8974_INT_CTRL,
244 			   AK8974_INT_CTRL_RESDEF);
245 	if (ret)
246 		return ret;
247 
248 	/* After reset, power off is default state */
249 	return ak8974_set_power(ak8974, AK8974_PWR_OFF);
250 }
251 
252 static int ak8974_configure(struct ak8974 *ak8974)
253 {
254 	int ret;
255 
256 	ret = regmap_write(ak8974->map, AK8974_CTRL2, AK8974_CTRL2_DRDY_EN |
257 			   AK8974_CTRL2_INT_EN);
258 	if (ret)
259 		return ret;
260 	ret = regmap_write(ak8974->map, AK8974_CTRL3, 0);
261 	if (ret)
262 		return ret;
263 	if (ak8974->variant == AK8974_WHOAMI_VALUE_AMI306) {
264 		/* magic from datasheet: set high-speed measurement mode */
265 		ret = ak8974_set_u16_val(ak8974, AMI306_CTRL4, 0xA07E);
266 		if (ret)
267 			return ret;
268 	}
269 	ret = regmap_write(ak8974->map, AK8974_INT_CTRL, AK8974_INT_CTRL_POL);
270 	if (ret)
271 		return ret;
272 
273 	return regmap_write(ak8974->map, AK8974_PRESET, 0);
274 }
275 
276 static int ak8974_trigmeas(struct ak8974 *ak8974)
277 {
278 	unsigned int clear;
279 	u8 mask;
280 	u8 val;
281 	int ret;
282 
283 	/* Clear any previous measurement overflow status */
284 	ret = regmap_read(ak8974->map, AK8974_INT_CLEAR, &clear);
285 	if (ret)
286 		return ret;
287 
288 	/* If we have a DRDY IRQ line, use it */
289 	if (ak8974->drdy_irq) {
290 		mask = AK8974_CTRL2_INT_EN |
291 			AK8974_CTRL2_DRDY_EN |
292 			AK8974_CTRL2_DRDY_POL;
293 		val = AK8974_CTRL2_DRDY_EN;
294 
295 		if (!ak8974->drdy_active_low)
296 			val |= AK8974_CTRL2_DRDY_POL;
297 
298 		init_completion(&ak8974->drdy_complete);
299 		ret = regmap_update_bits(ak8974->map, AK8974_CTRL2,
300 					 mask, val);
301 		if (ret)
302 			return ret;
303 	}
304 
305 	/* Force a measurement */
306 	return regmap_update_bits(ak8974->map,
307 				  AK8974_CTRL3,
308 				  AK8974_CTRL3_FORCE,
309 				  AK8974_CTRL3_FORCE);
310 }
311 
312 static int ak8974_await_drdy(struct ak8974 *ak8974)
313 {
314 	int timeout = 2;
315 	unsigned int val;
316 	int ret;
317 
318 	if (ak8974->drdy_irq) {
319 		ret = wait_for_completion_timeout(&ak8974->drdy_complete,
320 					1 + msecs_to_jiffies(1000));
321 		if (!ret) {
322 			dev_err(&ak8974->i2c->dev,
323 				"timeout waiting for DRDY IRQ\n");
324 			return -ETIMEDOUT;
325 		}
326 		return 0;
327 	}
328 
329 	/* Default delay-based poll loop */
330 	do {
331 		msleep(AK8974_MEASTIME);
332 		ret = regmap_read(ak8974->map, AK8974_STATUS, &val);
333 		if (ret < 0)
334 			return ret;
335 		if (val & AK8974_STATUS_DRDY)
336 			return 0;
337 	} while (--timeout);
338 
339 	dev_err(&ak8974->i2c->dev, "timeout waiting for DRDY\n");
340 	return -ETIMEDOUT;
341 }
342 
343 static int ak8974_getresult(struct ak8974 *ak8974, __le16 *result)
344 {
345 	unsigned int src;
346 	int ret;
347 
348 	ret = ak8974_await_drdy(ak8974);
349 	if (ret)
350 		return ret;
351 	ret = regmap_read(ak8974->map, AK8974_INT_SRC, &src);
352 	if (ret < 0)
353 		return ret;
354 
355 	/* Out of range overflow! Strong magnet close? */
356 	if (src & AK8974_INT_RANGE) {
357 		dev_err(&ak8974->i2c->dev,
358 			"range overflow in sensor\n");
359 		return -ERANGE;
360 	}
361 
362 	ret = regmap_bulk_read(ak8974->map, AK8974_DATA_X, result, 6);
363 	if (ret)
364 		return ret;
365 
366 	return ret;
367 }
368 
369 static irqreturn_t ak8974_drdy_irq(int irq, void *d)
370 {
371 	struct ak8974 *ak8974 = d;
372 
373 	if (!ak8974->drdy_irq)
374 		return IRQ_NONE;
375 
376 	/* TODO: timestamp here to get good measurement stamps */
377 	return IRQ_WAKE_THREAD;
378 }
379 
380 static irqreturn_t ak8974_drdy_irq_thread(int irq, void *d)
381 {
382 	struct ak8974 *ak8974 = d;
383 	unsigned int val;
384 	int ret;
385 
386 	/* Check if this was a DRDY from us */
387 	ret = regmap_read(ak8974->map, AK8974_STATUS, &val);
388 	if (ret < 0) {
389 		dev_err(&ak8974->i2c->dev, "error reading DRDY status\n");
390 		return IRQ_HANDLED;
391 	}
392 	if (val & AK8974_STATUS_DRDY) {
393 		/* Yes this was our IRQ */
394 		complete(&ak8974->drdy_complete);
395 		return IRQ_HANDLED;
396 	}
397 
398 	/* We may be on a shared IRQ, let the next client check */
399 	return IRQ_NONE;
400 }
401 
402 static int ak8974_selftest(struct ak8974 *ak8974)
403 {
404 	struct device *dev = &ak8974->i2c->dev;
405 	unsigned int val;
406 	int ret;
407 
408 	ret = regmap_read(ak8974->map, AK8974_SELFTEST, &val);
409 	if (ret)
410 		return ret;
411 	if (val != AK8974_SELFTEST_IDLE) {
412 		dev_err(dev, "selftest not idle before test\n");
413 		return -EIO;
414 	}
415 
416 	/* Trigger self-test */
417 	ret = regmap_update_bits(ak8974->map,
418 			AK8974_CTRL3,
419 			AK8974_CTRL3_SELFTEST,
420 			AK8974_CTRL3_SELFTEST);
421 	if (ret) {
422 		dev_err(dev, "could not write CTRL3\n");
423 		return ret;
424 	}
425 
426 	msleep(AK8974_SELFTEST_DELAY);
427 
428 	ret = regmap_read(ak8974->map, AK8974_SELFTEST, &val);
429 	if (ret)
430 		return ret;
431 	if (val != AK8974_SELFTEST_OK) {
432 		dev_err(dev, "selftest result NOT OK (%02x)\n", val);
433 		return -EIO;
434 	}
435 
436 	ret = regmap_read(ak8974->map, AK8974_SELFTEST, &val);
437 	if (ret)
438 		return ret;
439 	if (val != AK8974_SELFTEST_IDLE) {
440 		dev_err(dev, "selftest not idle after test (%02x)\n", val);
441 		return -EIO;
442 	}
443 	dev_dbg(dev, "passed self-test\n");
444 
445 	return 0;
446 }
447 
448 static void ak8974_read_calib_data(struct ak8974 *ak8974, unsigned int reg,
449 				   __le16 *tab, size_t tab_size)
450 {
451 	int ret = regmap_bulk_read(ak8974->map, reg, tab, tab_size);
452 	if (ret) {
453 		memset(tab, 0xFF, tab_size);
454 		dev_warn(&ak8974->i2c->dev,
455 			 "can't read calibration data (regs %u..%zu): %d\n",
456 			 reg, reg + tab_size - 1, ret);
457 	} else {
458 		add_device_randomness(tab, tab_size);
459 	}
460 }
461 
462 static int ak8974_detect(struct ak8974 *ak8974)
463 {
464 	unsigned int whoami;
465 	const char *name;
466 	int ret;
467 	unsigned int fw;
468 	u16 sn;
469 
470 	ret = regmap_read(ak8974->map, AK8974_WHOAMI, &whoami);
471 	if (ret)
472 		return ret;
473 
474 	name = "ami305";
475 
476 	switch (whoami) {
477 	case AK8974_WHOAMI_VALUE_AMI306:
478 		name = "ami306";
479 		/* fall-through */
480 	case AK8974_WHOAMI_VALUE_AMI305:
481 		ret = regmap_read(ak8974->map, AMI305_VER, &fw);
482 		if (ret)
483 			return ret;
484 		fw &= 0x7f; /* only bits 0 thru 6 valid */
485 		ret = ak8974_get_u16_val(ak8974, AMI305_SN, &sn);
486 		if (ret)
487 			return ret;
488 		add_device_randomness(&sn, sizeof(sn));
489 		dev_info(&ak8974->i2c->dev,
490 			 "detected %s, FW ver %02x, S/N: %04x\n",
491 			 name, fw, sn);
492 		break;
493 	case AK8974_WHOAMI_VALUE_AK8974:
494 		name = "ak8974";
495 		dev_info(&ak8974->i2c->dev, "detected AK8974\n");
496 		break;
497 	default:
498 		dev_err(&ak8974->i2c->dev, "unsupported device (%02x) ",
499 			whoami);
500 		return -ENODEV;
501 	}
502 
503 	ak8974->name = name;
504 	ak8974->variant = whoami;
505 
506 	if (whoami == AK8974_WHOAMI_VALUE_AMI306) {
507 		__le16 fab_data1[9], fab_data2[3];
508 		int i;
509 
510 		ak8974_read_calib_data(ak8974, AMI306_FINEOUTPUT_X,
511 				       fab_data1, sizeof(fab_data1));
512 		ak8974_read_calib_data(ak8974, AMI306_OFFZERO_X,
513 				       fab_data2, sizeof(fab_data2));
514 
515 		for (i = 0; i < 3; ++i) {
516 			static const char axis[3] = "XYZ";
517 			static const char pgaxis[6] = "ZYZXYX";
518 			unsigned offz = le16_to_cpu(fab_data2[i]) & 0x7F;
519 			unsigned fine = le16_to_cpu(fab_data1[i]);
520 			unsigned sens = le16_to_cpu(fab_data1[i + 3]);
521 			unsigned pgain1 = le16_to_cpu(fab_data1[i + 6]);
522 			unsigned pgain2 = pgain1 >> 8;
523 
524 			pgain1 &= 0xFF;
525 
526 			dev_info(&ak8974->i2c->dev,
527 				 "factory calibration for axis %c: offz=%u sens=%u fine=%u pga%c=%u pga%c=%u\n",
528 				 axis[i], offz, sens, fine, pgaxis[i * 2],
529 				 pgain1, pgaxis[i * 2 + 1], pgain2);
530 		}
531 	}
532 
533 	return 0;
534 }
535 
536 static int ak8974_read_raw(struct iio_dev *indio_dev,
537 			   struct iio_chan_spec const *chan,
538 			   int *val, int *val2,
539 			   long mask)
540 {
541 	struct ak8974 *ak8974 = iio_priv(indio_dev);
542 	__le16 hw_values[3];
543 	int ret = -EINVAL;
544 
545 	pm_runtime_get_sync(&ak8974->i2c->dev);
546 	mutex_lock(&ak8974->lock);
547 
548 	switch (mask) {
549 	case IIO_CHAN_INFO_RAW:
550 		if (chan->address > 2) {
551 			dev_err(&ak8974->i2c->dev, "faulty channel address\n");
552 			ret = -EIO;
553 			goto out_unlock;
554 		}
555 		ret = ak8974_trigmeas(ak8974);
556 		if (ret)
557 			goto out_unlock;
558 		ret = ak8974_getresult(ak8974, hw_values);
559 		if (ret)
560 			goto out_unlock;
561 
562 		/*
563 		 * We read all axes and discard all but one, for optimized
564 		 * reading, use the triggered buffer.
565 		 */
566 		*val = le16_to_cpu(hw_values[chan->address]);
567 
568 		ret = IIO_VAL_INT;
569 	}
570 
571  out_unlock:
572 	mutex_unlock(&ak8974->lock);
573 	pm_runtime_mark_last_busy(&ak8974->i2c->dev);
574 	pm_runtime_put_autosuspend(&ak8974->i2c->dev);
575 
576 	return ret;
577 }
578 
579 static void ak8974_fill_buffer(struct iio_dev *indio_dev)
580 {
581 	struct ak8974 *ak8974 = iio_priv(indio_dev);
582 	int ret;
583 	__le16 hw_values[8]; /* Three axes + 64bit padding */
584 
585 	pm_runtime_get_sync(&ak8974->i2c->dev);
586 	mutex_lock(&ak8974->lock);
587 
588 	ret = ak8974_trigmeas(ak8974);
589 	if (ret) {
590 		dev_err(&ak8974->i2c->dev, "error triggering measure\n");
591 		goto out_unlock;
592 	}
593 	ret = ak8974_getresult(ak8974, hw_values);
594 	if (ret) {
595 		dev_err(&ak8974->i2c->dev, "error getting measures\n");
596 		goto out_unlock;
597 	}
598 
599 	iio_push_to_buffers_with_timestamp(indio_dev, hw_values,
600 					   iio_get_time_ns(indio_dev));
601 
602  out_unlock:
603 	mutex_unlock(&ak8974->lock);
604 	pm_runtime_mark_last_busy(&ak8974->i2c->dev);
605 	pm_runtime_put_autosuspend(&ak8974->i2c->dev);
606 }
607 
608 static irqreturn_t ak8974_handle_trigger(int irq, void *p)
609 {
610 	const struct iio_poll_func *pf = p;
611 	struct iio_dev *indio_dev = pf->indio_dev;
612 
613 	ak8974_fill_buffer(indio_dev);
614 	iio_trigger_notify_done(indio_dev->trig);
615 
616 	return IRQ_HANDLED;
617 }
618 
619 static const struct iio_mount_matrix *
620 ak8974_get_mount_matrix(const struct iio_dev *indio_dev,
621 			const struct iio_chan_spec *chan)
622 {
623 	struct ak8974 *ak8974 = iio_priv(indio_dev);
624 
625 	return &ak8974->orientation;
626 }
627 
628 static const struct iio_chan_spec_ext_info ak8974_ext_info[] = {
629 	IIO_MOUNT_MATRIX(IIO_SHARED_BY_DIR, ak8974_get_mount_matrix),
630 	{ },
631 };
632 
633 #define AK8974_AXIS_CHANNEL(axis, index)				\
634 	{								\
635 		.type = IIO_MAGN,					\
636 		.modified = 1,						\
637 		.channel2 = IIO_MOD_##axis,				\
638 		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),		\
639 		.ext_info = ak8974_ext_info,				\
640 		.address = index,					\
641 		.scan_index = index,					\
642 		.scan_type = {						\
643 			.sign = 's',					\
644 			.realbits = 16,					\
645 			.storagebits = 16,				\
646 			.endianness = IIO_LE				\
647 		},							\
648 	}
649 
650 static const struct iio_chan_spec ak8974_channels[] = {
651 	AK8974_AXIS_CHANNEL(X, 0),
652 	AK8974_AXIS_CHANNEL(Y, 1),
653 	AK8974_AXIS_CHANNEL(Z, 2),
654 	IIO_CHAN_SOFT_TIMESTAMP(3),
655 };
656 
657 static const unsigned long ak8974_scan_masks[] = { 0x7, 0 };
658 
659 static const struct iio_info ak8974_info = {
660 	.read_raw = &ak8974_read_raw,
661 	.driver_module = THIS_MODULE,
662 };
663 
664 static bool ak8974_writeable_reg(struct device *dev, unsigned int reg)
665 {
666 	struct i2c_client *i2c = to_i2c_client(dev);
667 	struct iio_dev *indio_dev = i2c_get_clientdata(i2c);
668 	struct ak8974 *ak8974 = iio_priv(indio_dev);
669 
670 	switch (reg) {
671 	case AK8974_CTRL1:
672 	case AK8974_CTRL2:
673 	case AK8974_CTRL3:
674 	case AK8974_INT_CTRL:
675 	case AK8974_INT_THRES:
676 	case AK8974_INT_THRES + 1:
677 	case AK8974_PRESET:
678 	case AK8974_PRESET + 1:
679 		return true;
680 	case AK8974_OFFSET_X:
681 	case AK8974_OFFSET_X + 1:
682 	case AK8974_OFFSET_Y:
683 	case AK8974_OFFSET_Y + 1:
684 	case AK8974_OFFSET_Z:
685 	case AK8974_OFFSET_Z + 1:
686 		if (ak8974->variant == AK8974_WHOAMI_VALUE_AK8974)
687 			return true;
688 		return false;
689 	case AMI305_OFFSET_X:
690 	case AMI305_OFFSET_X + 1:
691 	case AMI305_OFFSET_Y:
692 	case AMI305_OFFSET_Y + 1:
693 	case AMI305_OFFSET_Z:
694 	case AMI305_OFFSET_Z + 1:
695 		return ak8974->variant == AK8974_WHOAMI_VALUE_AMI305 ||
696 		       ak8974->variant == AK8974_WHOAMI_VALUE_AMI306;
697 	case AMI306_CTRL4:
698 	case AMI306_CTRL4 + 1:
699 		return ak8974->variant == AK8974_WHOAMI_VALUE_AMI306;
700 	default:
701 		return false;
702 	}
703 }
704 
705 static bool ak8974_precious_reg(struct device *dev, unsigned int reg)
706 {
707 	return reg == AK8974_INT_CLEAR;
708 }
709 
710 static const struct regmap_config ak8974_regmap_config = {
711 	.reg_bits = 8,
712 	.val_bits = 8,
713 	.max_register = 0xff,
714 	.writeable_reg = ak8974_writeable_reg,
715 	.precious_reg = ak8974_precious_reg,
716 };
717 
718 static int ak8974_probe(struct i2c_client *i2c,
719 			const struct i2c_device_id *id)
720 {
721 	struct iio_dev *indio_dev;
722 	struct ak8974 *ak8974;
723 	unsigned long irq_trig;
724 	int irq = i2c->irq;
725 	int ret;
726 
727 	/* Register with IIO */
728 	indio_dev = devm_iio_device_alloc(&i2c->dev, sizeof(*ak8974));
729 	if (indio_dev == NULL)
730 		return -ENOMEM;
731 
732 	ak8974 = iio_priv(indio_dev);
733 	i2c_set_clientdata(i2c, indio_dev);
734 	ak8974->i2c = i2c;
735 	mutex_init(&ak8974->lock);
736 
737 	ret = of_iio_read_mount_matrix(&i2c->dev,
738 				       "mount-matrix",
739 				       &ak8974->orientation);
740 	if (ret)
741 		return ret;
742 
743 	ak8974->regs[0].supply = ak8974_reg_avdd;
744 	ak8974->regs[1].supply = ak8974_reg_dvdd;
745 
746 	ret = devm_regulator_bulk_get(&i2c->dev,
747 				      ARRAY_SIZE(ak8974->regs),
748 				      ak8974->regs);
749 	if (ret < 0) {
750 		dev_err(&i2c->dev, "cannot get regulators\n");
751 		return ret;
752 	}
753 
754 	ret = regulator_bulk_enable(ARRAY_SIZE(ak8974->regs), ak8974->regs);
755 	if (ret < 0) {
756 		dev_err(&i2c->dev, "cannot enable regulators\n");
757 		return ret;
758 	}
759 
760 	/* Take runtime PM online */
761 	pm_runtime_get_noresume(&i2c->dev);
762 	pm_runtime_set_active(&i2c->dev);
763 	pm_runtime_enable(&i2c->dev);
764 
765 	ak8974->map = devm_regmap_init_i2c(i2c, &ak8974_regmap_config);
766 	if (IS_ERR(ak8974->map)) {
767 		dev_err(&i2c->dev, "failed to allocate register map\n");
768 		return PTR_ERR(ak8974->map);
769 	}
770 
771 	ret = ak8974_set_power(ak8974, AK8974_PWR_ON);
772 	if (ret) {
773 		dev_err(&i2c->dev, "could not power on\n");
774 		goto power_off;
775 	}
776 
777 	ret = ak8974_detect(ak8974);
778 	if (ret) {
779 		dev_err(&i2c->dev, "neither AK8974 nor AMI30x found\n");
780 		goto power_off;
781 	}
782 
783 	ret = ak8974_selftest(ak8974);
784 	if (ret)
785 		dev_err(&i2c->dev, "selftest failed (continuing anyway)\n");
786 
787 	ret = ak8974_reset(ak8974);
788 	if (ret) {
789 		dev_err(&i2c->dev, "AK8974 reset failed\n");
790 		goto power_off;
791 	}
792 
793 	pm_runtime_set_autosuspend_delay(&i2c->dev,
794 					 AK8974_AUTOSUSPEND_DELAY);
795 	pm_runtime_use_autosuspend(&i2c->dev);
796 	pm_runtime_put(&i2c->dev);
797 
798 	indio_dev->dev.parent = &i2c->dev;
799 	indio_dev->channels = ak8974_channels;
800 	indio_dev->num_channels = ARRAY_SIZE(ak8974_channels);
801 	indio_dev->info = &ak8974_info;
802 	indio_dev->available_scan_masks = ak8974_scan_masks;
803 	indio_dev->modes = INDIO_DIRECT_MODE;
804 	indio_dev->name = ak8974->name;
805 
806 	ret = iio_triggered_buffer_setup(indio_dev, NULL,
807 					 ak8974_handle_trigger,
808 					 NULL);
809 	if (ret) {
810 		dev_err(&i2c->dev, "triggered buffer setup failed\n");
811 		goto disable_pm;
812 	}
813 
814 	/* If we have a valid DRDY IRQ, make use of it */
815 	if (irq > 0) {
816 		irq_trig = irqd_get_trigger_type(irq_get_irq_data(irq));
817 		if (irq_trig == IRQF_TRIGGER_RISING) {
818 			dev_info(&i2c->dev, "enable rising edge DRDY IRQ\n");
819 		} else if (irq_trig == IRQF_TRIGGER_FALLING) {
820 			ak8974->drdy_active_low = true;
821 			dev_info(&i2c->dev, "enable falling edge DRDY IRQ\n");
822 		} else {
823 			irq_trig = IRQF_TRIGGER_RISING;
824 		}
825 		irq_trig |= IRQF_ONESHOT;
826 		irq_trig |= IRQF_SHARED;
827 
828 		ret = devm_request_threaded_irq(&i2c->dev,
829 						irq,
830 						ak8974_drdy_irq,
831 						ak8974_drdy_irq_thread,
832 						irq_trig,
833 						ak8974->name,
834 						ak8974);
835 		if (ret) {
836 			dev_err(&i2c->dev, "unable to request DRDY IRQ "
837 				"- proceeding without IRQ\n");
838 			goto no_irq;
839 		}
840 		ak8974->drdy_irq = true;
841 	}
842 
843 no_irq:
844 	ret = iio_device_register(indio_dev);
845 	if (ret) {
846 		dev_err(&i2c->dev, "device register failed\n");
847 		goto cleanup_buffer;
848 	}
849 
850 	return 0;
851 
852 cleanup_buffer:
853 	iio_triggered_buffer_cleanup(indio_dev);
854 disable_pm:
855 	pm_runtime_put_noidle(&i2c->dev);
856 	pm_runtime_disable(&i2c->dev);
857 	ak8974_set_power(ak8974, AK8974_PWR_OFF);
858 power_off:
859 	regulator_bulk_disable(ARRAY_SIZE(ak8974->regs), ak8974->regs);
860 
861 	return ret;
862 }
863 
864 static int ak8974_remove(struct i2c_client *i2c)
865 {
866 	struct iio_dev *indio_dev = i2c_get_clientdata(i2c);
867 	struct ak8974 *ak8974 = iio_priv(indio_dev);
868 
869 	iio_device_unregister(indio_dev);
870 	iio_triggered_buffer_cleanup(indio_dev);
871 	pm_runtime_get_sync(&i2c->dev);
872 	pm_runtime_put_noidle(&i2c->dev);
873 	pm_runtime_disable(&i2c->dev);
874 	ak8974_set_power(ak8974, AK8974_PWR_OFF);
875 	regulator_bulk_disable(ARRAY_SIZE(ak8974->regs), ak8974->regs);
876 
877 	return 0;
878 }
879 
880 static int __maybe_unused ak8974_runtime_suspend(struct device *dev)
881 {
882 	struct ak8974 *ak8974 =
883 		iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
884 
885 	ak8974_set_power(ak8974, AK8974_PWR_OFF);
886 	regulator_bulk_disable(ARRAY_SIZE(ak8974->regs), ak8974->regs);
887 
888 	return 0;
889 }
890 
891 static int __maybe_unused ak8974_runtime_resume(struct device *dev)
892 {
893 	struct ak8974 *ak8974 =
894 		iio_priv(i2c_get_clientdata(to_i2c_client(dev)));
895 	int ret;
896 
897 	ret = regulator_bulk_enable(ARRAY_SIZE(ak8974->regs), ak8974->regs);
898 	if (ret)
899 		return ret;
900 	msleep(AK8974_POWERON_DELAY);
901 	ret = ak8974_set_power(ak8974, AK8974_PWR_ON);
902 	if (ret)
903 		goto out_regulator_disable;
904 
905 	ret = ak8974_configure(ak8974);
906 	if (ret)
907 		goto out_disable_power;
908 
909 	return 0;
910 
911 out_disable_power:
912 	ak8974_set_power(ak8974, AK8974_PWR_OFF);
913 out_regulator_disable:
914 	regulator_bulk_disable(ARRAY_SIZE(ak8974->regs), ak8974->regs);
915 
916 	return ret;
917 }
918 
919 static const struct dev_pm_ops ak8974_dev_pm_ops = {
920 	SET_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
921 				pm_runtime_force_resume)
922 	SET_RUNTIME_PM_OPS(ak8974_runtime_suspend,
923 			   ak8974_runtime_resume, NULL)
924 };
925 
926 static const struct i2c_device_id ak8974_id[] = {
927 	{"ami305", 0 },
928 	{"ami306", 0 },
929 	{"ak8974", 0 },
930 	{}
931 };
932 MODULE_DEVICE_TABLE(i2c, ak8974_id);
933 
934 static const struct of_device_id ak8974_of_match[] = {
935 	{ .compatible = "asahi-kasei,ak8974", },
936 	{}
937 };
938 MODULE_DEVICE_TABLE(of, ak8974_of_match);
939 
940 static struct i2c_driver ak8974_driver = {
941 	.driver	 = {
942 		.name	= "ak8974",
943 		.pm = &ak8974_dev_pm_ops,
944 		.of_match_table = of_match_ptr(ak8974_of_match),
945 	},
946 	.probe	  = ak8974_probe,
947 	.remove	  = ak8974_remove,
948 	.id_table = ak8974_id,
949 };
950 module_i2c_driver(ak8974_driver);
951 
952 MODULE_DESCRIPTION("AK8974 and AMI30x 3-axis magnetometer driver");
953 MODULE_AUTHOR("Samu Onkalo");
954 MODULE_AUTHOR("Linus Walleij");
955 MODULE_LICENSE("GPL v2");
956