1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * STMicroelectronics LSM9DS0 IMU driver
4  *
5  * Copyright (C) 2021, Intel Corporation
6  *
7  * Author: Andy Shevchenko <andriy.shevchenko@linux.intel.com>
8  */
9 
10 #include <linux/device.h>
11 #include <linux/err.h>
12 #include <linux/module.h>
13 #include <linux/regulator/consumer.h>
14 
15 #include <linux/iio/common/st_sensors.h>
16 #include <linux/iio/iio.h>
17 
18 #include "st_lsm9ds0.h"
19 
20 static int st_lsm9ds0_power_enable(struct device *dev, struct st_lsm9ds0 *lsm9ds0)
21 {
22 	int ret;
23 
24 	/* Regulators not mandatory, but if requested we should enable them. */
25 	lsm9ds0->vdd = devm_regulator_get(dev, "vdd");
26 	if (IS_ERR(lsm9ds0->vdd)) {
27 		dev_err(dev, "unable to get Vdd supply\n");
28 		return PTR_ERR(lsm9ds0->vdd);
29 	}
30 	ret = regulator_enable(lsm9ds0->vdd);
31 	if (ret) {
32 		dev_warn(dev, "Failed to enable specified Vdd supply\n");
33 		return ret;
34 	}
35 
36 	lsm9ds0->vdd_io = devm_regulator_get(dev, "vddio");
37 	if (IS_ERR(lsm9ds0->vdd_io)) {
38 		dev_err(dev, "unable to get Vdd_IO supply\n");
39 		regulator_disable(lsm9ds0->vdd);
40 		return PTR_ERR(lsm9ds0->vdd_io);
41 	}
42 	ret = regulator_enable(lsm9ds0->vdd_io);
43 	if (ret) {
44 		dev_warn(dev, "Failed to enable specified Vdd_IO supply\n");
45 		regulator_disable(lsm9ds0->vdd);
46 		return ret;
47 	}
48 
49 	return 0;
50 }
51 
52 static void st_lsm9ds0_power_disable(void *data)
53 {
54 	struct st_lsm9ds0 *lsm9ds0 = data;
55 
56 	regulator_disable(lsm9ds0->vdd_io);
57 	regulator_disable(lsm9ds0->vdd);
58 }
59 
60 static int devm_st_lsm9ds0_power_enable(struct st_lsm9ds0 *lsm9ds0)
61 {
62 	struct device *dev = lsm9ds0->dev;
63 	int ret;
64 
65 	ret = st_lsm9ds0_power_enable(dev, lsm9ds0);
66 	if (ret)
67 		return ret;
68 
69 	return devm_add_action_or_reset(dev, st_lsm9ds0_power_disable, lsm9ds0);
70 }
71 
72 static int st_lsm9ds0_probe_accel(struct st_lsm9ds0 *lsm9ds0, struct regmap *regmap)
73 {
74 	const struct st_sensor_settings *settings;
75 	struct device *dev = lsm9ds0->dev;
76 	struct st_sensor_data *data;
77 
78 	settings = st_accel_get_settings(lsm9ds0->name);
79 	if (!settings) {
80 		dev_err(dev, "device name %s not recognized.\n", lsm9ds0->name);
81 		return -ENODEV;
82 	}
83 
84 	lsm9ds0->accel = devm_iio_device_alloc(dev, sizeof(*data));
85 	if (!lsm9ds0->accel)
86 		return -ENOMEM;
87 
88 	lsm9ds0->accel->name = lsm9ds0->name;
89 
90 	data = iio_priv(lsm9ds0->accel);
91 	data->sensor_settings = (struct st_sensor_settings *)settings;
92 	data->dev = dev;
93 	data->irq = lsm9ds0->irq;
94 	data->regmap = regmap;
95 	data->vdd = lsm9ds0->vdd;
96 	data->vdd_io = lsm9ds0->vdd_io;
97 
98 	return st_accel_common_probe(lsm9ds0->accel);
99 }
100 
101 static int st_lsm9ds0_probe_magn(struct st_lsm9ds0 *lsm9ds0, struct regmap *regmap)
102 {
103 	const struct st_sensor_settings *settings;
104 	struct device *dev = lsm9ds0->dev;
105 	struct st_sensor_data *data;
106 
107 	settings = st_magn_get_settings(lsm9ds0->name);
108 	if (!settings) {
109 		dev_err(dev, "device name %s not recognized.\n", lsm9ds0->name);
110 		return -ENODEV;
111 	}
112 
113 	lsm9ds0->magn = devm_iio_device_alloc(dev, sizeof(*data));
114 	if (!lsm9ds0->magn)
115 		return -ENOMEM;
116 
117 	lsm9ds0->magn->name = lsm9ds0->name;
118 
119 	data = iio_priv(lsm9ds0->magn);
120 	data->sensor_settings = (struct st_sensor_settings *)settings;
121 	data->dev = dev;
122 	data->irq = lsm9ds0->irq;
123 	data->regmap = regmap;
124 	data->vdd = lsm9ds0->vdd;
125 	data->vdd_io = lsm9ds0->vdd_io;
126 
127 	return st_magn_common_probe(lsm9ds0->magn);
128 }
129 
130 int st_lsm9ds0_probe(struct st_lsm9ds0 *lsm9ds0, struct regmap *regmap)
131 {
132 	int ret;
133 
134 	ret = devm_st_lsm9ds0_power_enable(lsm9ds0);
135 	if (ret)
136 		return ret;
137 
138 	/* Setup accelerometer device */
139 	ret = st_lsm9ds0_probe_accel(lsm9ds0, regmap);
140 	if (ret)
141 		return ret;
142 
143 	/* Setup magnetometer device */
144 	ret = st_lsm9ds0_probe_magn(lsm9ds0, regmap);
145 	if (ret)
146 		st_accel_common_remove(lsm9ds0->accel);
147 
148 	return ret;
149 }
150 EXPORT_SYMBOL_GPL(st_lsm9ds0_probe);
151 
152 int st_lsm9ds0_remove(struct st_lsm9ds0 *lsm9ds0)
153 {
154 	st_magn_common_remove(lsm9ds0->magn);
155 	st_accel_common_remove(lsm9ds0->accel);
156 
157 	return 0;
158 }
159 EXPORT_SYMBOL_GPL(st_lsm9ds0_remove);
160 
161 MODULE_AUTHOR("Andy Shevchenko <andriy.shevchenko@linux.intel.com>");
162 MODULE_DESCRIPTION("STMicroelectronics LSM9DS0 IMU core driver");
163 MODULE_LICENSE("GPL v2");
164