1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * STMicroelectronics gyroscopes driver
4  *
5  * Copyright 2012-2013 STMicroelectronics Inc.
6  *
7  * Denis Ciocca <denis.ciocca@st.com>
8  */
9 
10 #include <linux/module.h>
11 #include <linux/kernel.h>
12 #include <linux/slab.h>
13 #include <linux/stat.h>
14 #include <linux/interrupt.h>
15 #include <linux/i2c.h>
16 #include <linux/delay.h>
17 #include <linux/iio/iio.h>
18 #include <linux/iio/buffer.h>
19 #include <linux/iio/trigger_consumer.h>
20 #include <linux/iio/triggered_buffer.h>
21 
22 #include <linux/iio/common/st_sensors.h>
23 #include "st_gyro.h"
24 
25 int st_gyro_trig_set_state(struct iio_trigger *trig, bool state)
26 {
27 	struct iio_dev *indio_dev = iio_trigger_get_drvdata(trig);
28 
29 	return st_sensors_set_dataready_irq(indio_dev, state);
30 }
31 
32 static int st_gyro_buffer_postenable(struct iio_dev *indio_dev)
33 {
34 	int err;
35 
36 	err = iio_triggered_buffer_postenable(indio_dev);
37 	if (err < 0)
38 		return err;
39 
40 	err = st_sensors_set_axis_enable(indio_dev,
41 					 (u8)indio_dev->active_scan_mask[0]);
42 	if (err < 0)
43 		goto st_gyro_buffer_predisable;
44 
45 	err = st_sensors_set_enable(indio_dev, true);
46 	if (err < 0)
47 		goto st_gyro_buffer_enable_all_axis;
48 
49 	return 0;
50 
51 st_gyro_buffer_enable_all_axis:
52 	st_sensors_set_axis_enable(indio_dev, ST_SENSORS_ENABLE_ALL_AXIS);
53 st_gyro_buffer_predisable:
54 	iio_triggered_buffer_predisable(indio_dev);
55 	return err;
56 }
57 
58 static int st_gyro_buffer_predisable(struct iio_dev *indio_dev)
59 {
60 	int err, err2;
61 
62 	err = st_sensors_set_enable(indio_dev, false);
63 	if (err < 0)
64 		goto st_gyro_buffer_predisable;
65 
66 	err = st_sensors_set_axis_enable(indio_dev, ST_SENSORS_ENABLE_ALL_AXIS);
67 
68 st_gyro_buffer_predisable:
69 	err2 = iio_triggered_buffer_predisable(indio_dev);
70 	if (!err)
71 		err = err2;
72 
73 	return err;
74 }
75 
76 static const struct iio_buffer_setup_ops st_gyro_buffer_setup_ops = {
77 	.postenable = &st_gyro_buffer_postenable,
78 	.predisable = &st_gyro_buffer_predisable,
79 };
80 
81 int st_gyro_allocate_ring(struct iio_dev *indio_dev)
82 {
83 	return iio_triggered_buffer_setup(indio_dev, NULL,
84 		&st_sensors_trigger_handler, &st_gyro_buffer_setup_ops);
85 }
86 
87 void st_gyro_deallocate_ring(struct iio_dev *indio_dev)
88 {
89 	iio_triggered_buffer_cleanup(indio_dev);
90 }
91 
92 MODULE_AUTHOR("Denis Ciocca <denis.ciocca@st.com>");
93 MODULE_DESCRIPTION("STMicroelectronics gyroscopes buffer");
94 MODULE_LICENSE("GPL v2");
95