xref: /linux/drivers/iio/imu/inv_mpu6050/inv_mpu_trigger.c (revision 90d32e92011eaae8e70a9169b4e7acf4ca8f9d3a)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * Copyright (C) 2012 Invensense, Inc.
4 */
5 
6 #include <linux/pm_runtime.h>
7 
8 #include <linux/iio/common/inv_sensors_timestamp.h>
9 #include <linux/iio/events.h>
10 
11 #include "inv_mpu_iio.h"
12 
13 static unsigned int inv_scan_query_mpu6050(struct iio_dev *indio_dev)
14 {
15 	struct inv_mpu6050_state  *st = iio_priv(indio_dev);
16 	unsigned int mask;
17 
18 	/*
19 	 * If the MPU6050 is just used as a trigger, then the scan mask
20 	 * is not allocated so we simply enable the temperature channel
21 	 * as a dummy and bail out.
22 	 */
23 	if (!indio_dev->active_scan_mask) {
24 		st->chip_config.temp_fifo_enable = true;
25 		return INV_MPU6050_SENSOR_TEMP;
26 	}
27 
28 	st->chip_config.gyro_fifo_enable =
29 		test_bit(INV_MPU6050_SCAN_GYRO_X,
30 			 indio_dev->active_scan_mask) ||
31 		test_bit(INV_MPU6050_SCAN_GYRO_Y,
32 			 indio_dev->active_scan_mask) ||
33 		test_bit(INV_MPU6050_SCAN_GYRO_Z,
34 			 indio_dev->active_scan_mask);
35 
36 	st->chip_config.accl_fifo_enable =
37 		test_bit(INV_MPU6050_SCAN_ACCL_X,
38 			 indio_dev->active_scan_mask) ||
39 		test_bit(INV_MPU6050_SCAN_ACCL_Y,
40 			 indio_dev->active_scan_mask) ||
41 		test_bit(INV_MPU6050_SCAN_ACCL_Z,
42 			 indio_dev->active_scan_mask);
43 
44 	st->chip_config.temp_fifo_enable =
45 		test_bit(INV_MPU6050_SCAN_TEMP, indio_dev->active_scan_mask);
46 
47 	mask = 0;
48 	if (st->chip_config.gyro_fifo_enable)
49 		mask |= INV_MPU6050_SENSOR_GYRO;
50 	if (st->chip_config.accl_fifo_enable)
51 		mask |= INV_MPU6050_SENSOR_ACCL;
52 	if (st->chip_config.temp_fifo_enable)
53 		mask |= INV_MPU6050_SENSOR_TEMP;
54 
55 	return mask;
56 }
57 
58 static unsigned int inv_scan_query_mpu9x50(struct iio_dev *indio_dev)
59 {
60 	struct inv_mpu6050_state *st = iio_priv(indio_dev);
61 	unsigned int mask;
62 
63 	mask = inv_scan_query_mpu6050(indio_dev);
64 
65 	/* no magnetometer if i2c auxiliary bus is used */
66 	if (st->magn_disabled)
67 		return mask;
68 
69 	st->chip_config.magn_fifo_enable =
70 		test_bit(INV_MPU9X50_SCAN_MAGN_X,
71 			 indio_dev->active_scan_mask) ||
72 		test_bit(INV_MPU9X50_SCAN_MAGN_Y,
73 			 indio_dev->active_scan_mask) ||
74 		test_bit(INV_MPU9X50_SCAN_MAGN_Z,
75 			 indio_dev->active_scan_mask);
76 	if (st->chip_config.magn_fifo_enable)
77 		mask |= INV_MPU6050_SENSOR_MAGN;
78 
79 	return mask;
80 }
81 
82 static unsigned int inv_scan_query(struct iio_dev *indio_dev)
83 {
84 	struct inv_mpu6050_state *st = iio_priv(indio_dev);
85 
86 	switch (st->chip_type) {
87 	case INV_MPU9150:
88 	case INV_MPU9250:
89 	case INV_MPU9255:
90 		return inv_scan_query_mpu9x50(indio_dev);
91 	default:
92 		return inv_scan_query_mpu6050(indio_dev);
93 	}
94 }
95 
96 static unsigned int inv_compute_skip_samples(const struct inv_mpu6050_state *st)
97 {
98 	unsigned int skip_samples = 0;
99 
100 	/* mag first sample is always not ready, skip it */
101 	if (st->chip_config.magn_fifo_enable)
102 		skip_samples = 1;
103 
104 	return skip_samples;
105 }
106 
107 int inv_mpu6050_prepare_fifo(struct inv_mpu6050_state *st, bool enable)
108 {
109 	uint8_t d;
110 	int ret;
111 
112 	if (enable) {
113 		/* reset timestamping */
114 		inv_sensors_timestamp_reset(&st->timestamp);
115 		inv_sensors_timestamp_apply_odr(&st->timestamp, 0, 0, 0);
116 		/* reset FIFO */
117 		d = st->chip_config.user_ctrl | INV_MPU6050_BIT_FIFO_RST;
118 		ret = regmap_write(st->map, st->reg->user_ctrl, d);
119 		if (ret)
120 			return ret;
121 		/* enable sensor output to FIFO */
122 		d = 0;
123 		if (st->chip_config.gyro_fifo_enable)
124 			d |= INV_MPU6050_BITS_GYRO_OUT;
125 		if (st->chip_config.accl_fifo_enable)
126 			d |= INV_MPU6050_BIT_ACCEL_OUT;
127 		if (st->chip_config.temp_fifo_enable)
128 			d |= INV_MPU6050_BIT_TEMP_OUT;
129 		if (st->chip_config.magn_fifo_enable)
130 			d |= INV_MPU6050_BIT_SLAVE_0;
131 		ret = regmap_write(st->map, st->reg->fifo_en, d);
132 		if (ret)
133 			return ret;
134 		/* enable FIFO reading */
135 		d = st->chip_config.user_ctrl | INV_MPU6050_BIT_FIFO_EN;
136 		ret = regmap_write(st->map, st->reg->user_ctrl, d);
137 		if (ret)
138 			return ret;
139 		/* enable data interrupt */
140 		ret = regmap_update_bits(st->map, st->reg->int_enable,
141 				INV_MPU6050_BIT_DATA_RDY_EN, INV_MPU6050_BIT_DATA_RDY_EN);
142 	} else {
143 		/* disable data interrupt */
144 		ret = regmap_update_bits(st->map, st->reg->int_enable,
145 				INV_MPU6050_BIT_DATA_RDY_EN, 0);
146 		if (ret)
147 			return ret;
148 		ret = regmap_write(st->map, st->reg->fifo_en, 0);
149 		if (ret)
150 			return ret;
151 		/* restore user_ctrl for disabling FIFO reading */
152 		ret = regmap_write(st->map, st->reg->user_ctrl,
153 				   st->chip_config.user_ctrl);
154 	}
155 
156 	return ret;
157 }
158 
159 /**
160  *  inv_mpu6050_set_enable() - enable chip functions.
161  *  @indio_dev:	Device driver instance.
162  *  @enable: enable/disable
163  */
164 static int inv_mpu6050_set_enable(struct iio_dev *indio_dev, bool enable)
165 {
166 	struct inv_mpu6050_state *st = iio_priv(indio_dev);
167 	struct device *pdev = regmap_get_device(st->map);
168 	unsigned int scan;
169 	int result;
170 
171 	if (enable) {
172 		scan = inv_scan_query(indio_dev);
173 		result = pm_runtime_resume_and_get(pdev);
174 		if (result)
175 			return result;
176 		/*
177 		 * In case autosuspend didn't trigger, turn off first not
178 		 * required sensors excepted WoM
179 		 */
180 		result = inv_mpu6050_switch_engine(st, false, ~scan & ~INV_MPU6050_SENSOR_WOM);
181 		if (result)
182 			goto error_power_off;
183 		result = inv_mpu6050_switch_engine(st, true, scan);
184 		if (result)
185 			goto error_power_off;
186 		st->skip_samples = inv_compute_skip_samples(st);
187 		result = inv_mpu6050_prepare_fifo(st, true);
188 		if (result)
189 			goto error_power_off;
190 	} else {
191 		st->chip_config.gyro_fifo_enable = 0;
192 		st->chip_config.accl_fifo_enable = 0;
193 		st->chip_config.temp_fifo_enable = 0;
194 		st->chip_config.magn_fifo_enable = 0;
195 		result = inv_mpu6050_prepare_fifo(st, false);
196 		if (result)
197 			goto error_power_off;
198 		pm_runtime_mark_last_busy(pdev);
199 		pm_runtime_put_autosuspend(pdev);
200 	}
201 
202 	return 0;
203 
204 error_power_off:
205 	pm_runtime_put_autosuspend(pdev);
206 	return result;
207 }
208 
209 /**
210  * inv_mpu_data_rdy_trigger_set_state() - set data ready interrupt state
211  * @trig: Trigger instance
212  * @state: Desired trigger state
213  */
214 static int inv_mpu_data_rdy_trigger_set_state(struct iio_trigger *trig,
215 					      bool state)
216 {
217 	struct iio_dev *indio_dev = iio_trigger_get_drvdata(trig);
218 	struct inv_mpu6050_state *st = iio_priv(indio_dev);
219 	int result;
220 
221 	mutex_lock(&st->lock);
222 	result = inv_mpu6050_set_enable(indio_dev, state);
223 	mutex_unlock(&st->lock);
224 
225 	return result;
226 }
227 
228 static const struct iio_trigger_ops inv_mpu_trigger_ops = {
229 	.set_trigger_state = &inv_mpu_data_rdy_trigger_set_state,
230 };
231 
232 static irqreturn_t inv_mpu6050_interrupt_timestamp(int irq, void *p)
233 {
234 	struct iio_dev *indio_dev = p;
235 	struct inv_mpu6050_state *st = iio_priv(indio_dev);
236 
237 	st->it_timestamp = iio_get_time_ns(indio_dev);
238 
239 	return IRQ_WAKE_THREAD;
240 }
241 
242 static irqreturn_t inv_mpu6050_interrupt_handle(int irq, void *p)
243 {
244 	struct iio_dev *indio_dev = p;
245 	struct inv_mpu6050_state *st = iio_priv(indio_dev);
246 	unsigned int int_status, wom_bits;
247 	u64 ev_code;
248 	int result;
249 
250 	switch (st->chip_type) {
251 	case INV_MPU6050:
252 	case INV_MPU6500:
253 	case INV_MPU6515:
254 	case INV_MPU6880:
255 	case INV_MPU6000:
256 	case INV_MPU9150:
257 	case INV_MPU9250:
258 	case INV_MPU9255:
259 		wom_bits = INV_MPU6500_BIT_WOM_INT;
260 		break;
261 	default:
262 		wom_bits = INV_ICM20608_BIT_WOM_INT;
263 		break;
264 	}
265 
266 	scoped_guard(mutex, &st->lock) {
267 		/* ack interrupt and check status */
268 		result = regmap_read(st->map, st->reg->int_status, &int_status);
269 		if (result) {
270 			dev_err(regmap_get_device(st->map), "failed to ack interrupt\n");
271 			return IRQ_HANDLED;
272 		}
273 
274 		/* handle WoM event */
275 		if (st->chip_config.wom_en && (int_status & wom_bits)) {
276 			ev_code = IIO_MOD_EVENT_CODE(IIO_ACCEL, 0, IIO_MOD_X_OR_Y_OR_Z,
277 						     IIO_EV_TYPE_ROC, IIO_EV_DIR_RISING);
278 			iio_push_event(indio_dev, ev_code, st->it_timestamp);
279 		}
280 	}
281 
282 	/* handle raw data interrupt */
283 	if (int_status & INV_MPU6050_BIT_RAW_DATA_RDY_INT) {
284 		indio_dev->pollfunc->timestamp = st->it_timestamp;
285 		iio_trigger_poll_nested(st->trig);
286 	}
287 
288 	return IRQ_HANDLED;
289 }
290 
291 int inv_mpu6050_probe_trigger(struct iio_dev *indio_dev, int irq_type)
292 {
293 	int ret;
294 	struct inv_mpu6050_state *st = iio_priv(indio_dev);
295 
296 	st->trig = devm_iio_trigger_alloc(&indio_dev->dev,
297 					  "%s-dev%d",
298 					  indio_dev->name,
299 					  iio_device_id(indio_dev));
300 	if (!st->trig)
301 		return -ENOMEM;
302 
303 	ret = devm_request_threaded_irq(&indio_dev->dev, st->irq,
304 					&inv_mpu6050_interrupt_timestamp,
305 					&inv_mpu6050_interrupt_handle,
306 					irq_type, "inv_mpu", indio_dev);
307 	if (ret)
308 		return ret;
309 
310 	st->trig->dev.parent = regmap_get_device(st->map);
311 	st->trig->ops = &inv_mpu_trigger_ops;
312 	iio_trigger_set_drvdata(st->trig, indio_dev);
313 
314 	ret = devm_iio_trigger_register(&indio_dev->dev, st->trig);
315 	if (ret)
316 		return ret;
317 
318 	indio_dev->trig = iio_trigger_get(st->trig);
319 
320 	return 0;
321 }
322