1/*
2* Copyright (C) 2012 Invensense, Inc.
3*
4* This software is licensed under the terms of the GNU General Public
5* License version 2, as published by the Free Software Foundation, and
6* may be copied, distributed, and modified under those terms.
7*
8* This program is distributed in the hope that it will be useful,
9* but WITHOUT ANY WARRANTY; without even the implied warranty of
10* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11* GNU General Public License for more details.
12*/
13
14#include "inv_mpu_iio.h"
15
16static void inv_scan_query(struct iio_dev *indio_dev)
17{
18	struct inv_mpu6050_state  *st = iio_priv(indio_dev);
19
20	st->chip_config.gyro_fifo_enable =
21		test_bit(INV_MPU6050_SCAN_GYRO_X,
22			indio_dev->active_scan_mask) ||
23			test_bit(INV_MPU6050_SCAN_GYRO_Y,
24			indio_dev->active_scan_mask) ||
25			test_bit(INV_MPU6050_SCAN_GYRO_Z,
26			indio_dev->active_scan_mask);
27
28	st->chip_config.accl_fifo_enable =
29		test_bit(INV_MPU6050_SCAN_ACCL_X,
30			indio_dev->active_scan_mask) ||
31			test_bit(INV_MPU6050_SCAN_ACCL_Y,
32			indio_dev->active_scan_mask) ||
33			test_bit(INV_MPU6050_SCAN_ACCL_Z,
34			indio_dev->active_scan_mask);
35}
36
37/**
38 *  inv_mpu6050_set_enable() - enable chip functions.
39 *  @indio_dev:	Device driver instance.
40 *  @enable: enable/disable
41 */
42static int inv_mpu6050_set_enable(struct iio_dev *indio_dev, bool enable)
43{
44	struct inv_mpu6050_state *st = iio_priv(indio_dev);
45	int result;
46
47	if (enable) {
48		result = inv_mpu6050_set_power_itg(st, true);
49		if (result)
50			return result;
51		inv_scan_query(indio_dev);
52		if (st->chip_config.gyro_fifo_enable) {
53			result = inv_mpu6050_switch_engine(st, true,
54					INV_MPU6050_BIT_PWR_GYRO_STBY);
55			if (result)
56				return result;
57		}
58		if (st->chip_config.accl_fifo_enable) {
59			result = inv_mpu6050_switch_engine(st, true,
60					INV_MPU6050_BIT_PWR_ACCL_STBY);
61			if (result)
62				return result;
63		}
64		result = inv_reset_fifo(indio_dev);
65		if (result)
66			return result;
67	} else {
68		result = inv_mpu6050_write_reg(st, st->reg->fifo_en, 0);
69		if (result)
70			return result;
71
72		result = inv_mpu6050_write_reg(st, st->reg->int_enable, 0);
73		if (result)
74			return result;
75
76		result = inv_mpu6050_write_reg(st, st->reg->user_ctrl, 0);
77		if (result)
78			return result;
79
80		result = inv_mpu6050_switch_engine(st, false,
81					INV_MPU6050_BIT_PWR_GYRO_STBY);
82		if (result)
83			return result;
84
85		result = inv_mpu6050_switch_engine(st, false,
86					INV_MPU6050_BIT_PWR_ACCL_STBY);
87		if (result)
88			return result;
89		result = inv_mpu6050_set_power_itg(st, false);
90		if (result)
91			return result;
92	}
93	st->chip_config.enable = enable;
94
95	return 0;
96}
97
98/**
99 * inv_mpu_data_rdy_trigger_set_state() - set data ready interrupt state
100 * @trig: Trigger instance
101 * @state: Desired trigger state
102 */
103static int inv_mpu_data_rdy_trigger_set_state(struct iio_trigger *trig,
104						bool state)
105{
106	return inv_mpu6050_set_enable(iio_trigger_get_drvdata(trig), state);
107}
108
109static const struct iio_trigger_ops inv_mpu_trigger_ops = {
110	.owner = THIS_MODULE,
111	.set_trigger_state = &inv_mpu_data_rdy_trigger_set_state,
112};
113
114int inv_mpu6050_probe_trigger(struct iio_dev *indio_dev)
115{
116	int ret;
117	struct inv_mpu6050_state *st = iio_priv(indio_dev);
118
119	st->trig = iio_trigger_alloc("%s-dev%d",
120					indio_dev->name,
121					indio_dev->id);
122	if (st->trig == NULL) {
123		ret = -ENOMEM;
124		goto error_ret;
125	}
126	ret = request_irq(st->client->irq, &iio_trigger_generic_data_rdy_poll,
127				IRQF_TRIGGER_RISING,
128				"inv_mpu",
129				st->trig);
130	if (ret)
131		goto error_free_trig;
132	st->trig->dev.parent = &st->client->dev;
133	st->trig->ops = &inv_mpu_trigger_ops;
134	iio_trigger_set_drvdata(st->trig, indio_dev);
135	ret = iio_trigger_register(st->trig);
136	if (ret)
137		goto error_free_irq;
138	indio_dev->trig = iio_trigger_get(st->trig);
139
140	return 0;
141
142error_free_irq:
143	free_irq(st->client->irq, st->trig);
144error_free_trig:
145	iio_trigger_free(st->trig);
146error_ret:
147	return ret;
148}
149
150void inv_mpu6050_remove_trigger(struct inv_mpu6050_state *st)
151{
152	iio_trigger_unregister(st->trig);
153	free_irq(st->client->irq, st->trig);
154	iio_trigger_free(st->trig);
155}
156