GNU Linux-libre 4.14.254-gnu1
[releases.git] / drivers / iio / proximity / srf08.c
1 /*
2  * srf08.c - Support for Devantech SRFxx ultrasonic ranger
3  *           with i2c interface
4  * actually supported are srf02, srf08, srf10
5  *
6  * Copyright (c) 2016, 2017 Andreas Klinger <ak@it-klinger.de>
7  *
8  * This file is subject to the terms and conditions of version 2 of
9  * the GNU General Public License. See the file COPYING in the main
10  * directory of this archive for more details.
11  *
12  * For details about the device see:
13  * http://www.robot-electronics.co.uk/htm/srf08tech.html
14  * http://www.robot-electronics.co.uk/htm/srf10tech.htm
15  * http://www.robot-electronics.co.uk/htm/srf02tech.htm
16  */
17
18 #include <linux/err.h>
19 #include <linux/i2c.h>
20 #include <linux/delay.h>
21 #include <linux/module.h>
22 #include <linux/bitops.h>
23 #include <linux/iio/iio.h>
24 #include <linux/iio/sysfs.h>
25 #include <linux/iio/buffer.h>
26 #include <linux/iio/trigger_consumer.h>
27 #include <linux/iio/triggered_buffer.h>
28
29 /* registers of SRF08 device */
30 #define SRF08_WRITE_COMMAND     0x00    /* Command Register */
31 #define SRF08_WRITE_MAX_GAIN    0x01    /* Max Gain Register: 0 .. 31 */
32 #define SRF08_WRITE_RANGE       0x02    /* Range Register: 0 .. 255 */
33 #define SRF08_READ_SW_REVISION  0x00    /* Software Revision */
34 #define SRF08_READ_LIGHT        0x01    /* Light Sensor during last echo */
35 #define SRF08_READ_ECHO_1_HIGH  0x02    /* Range of first echo received */
36 #define SRF08_READ_ECHO_1_LOW   0x03    /* Range of first echo received */
37
38 #define SRF08_CMD_RANGING_CM    0x51    /* Ranging Mode - Result in cm */
39
40 enum srf08_sensor_type {
41         SRF02,
42         SRF08,
43         SRF10,
44         SRF_MAX_TYPE
45 };
46
47 struct srf08_chip_info {
48         const int               *sensitivity_avail;
49         int                     num_sensitivity_avail;
50         int                     sensitivity_default;
51
52         /* default value of Range in mm */
53         int                     range_default;
54 };
55
56 struct srf08_data {
57         struct i2c_client       *client;
58
59         /*
60          * Gain in the datasheet is called sensitivity here to distinct it
61          * from the gain used with amplifiers of adc's
62          */
63         int                     sensitivity;
64
65         /* max. Range in mm */
66         int                     range_mm;
67         struct mutex            lock;
68
69         /* Ensure timestamp is naturally aligned */
70         struct {
71                 s16 chan;
72                 s64 timestamp __aligned(8);
73         } scan;
74
75         /* Sensor-Type */
76         enum srf08_sensor_type  sensor_type;
77
78         /* Chip-specific information */
79         const struct srf08_chip_info    *chip_info;
80 };
81
82 /*
83  * in the documentation one can read about the "Gain" of the device
84  * which is used here for amplifying the signal and filtering out unwanted
85  * ones.
86  * But with ADC's this term is already used differently and that's why it
87  * is called "Sensitivity" here.
88  */
89 static const struct srf08_chip_info srf02_chip_info = {
90         .sensitivity_avail      = NULL,
91         .num_sensitivity_avail  = 0,
92         .sensitivity_default    = 0,
93
94         .range_default          = 0,
95 };
96
97 static const int srf08_sensitivity_avail[] = {
98          94,  97, 100, 103, 107, 110, 114, 118,
99         123, 128, 133, 139, 145, 152, 159, 168,
100         177, 187, 199, 212, 227, 245, 265, 288,
101         317, 352, 395, 450, 524, 626, 777, 1025
102         };
103
104 static const struct srf08_chip_info srf08_chip_info = {
105         .sensitivity_avail      = srf08_sensitivity_avail,
106         .num_sensitivity_avail  = ARRAY_SIZE(srf08_sensitivity_avail),
107         .sensitivity_default    = 1025,
108
109         .range_default          = 6020,
110 };
111
112 static const int srf10_sensitivity_avail[] = {
113          40,  40,  50,  60,  70,  80, 100, 120,
114         140, 200, 250, 300, 350, 400, 500, 600,
115         700,
116         };
117
118 static const struct srf08_chip_info srf10_chip_info = {
119         .sensitivity_avail      = srf10_sensitivity_avail,
120         .num_sensitivity_avail  = ARRAY_SIZE(srf10_sensitivity_avail),
121         .sensitivity_default    = 700,
122
123         .range_default          = 6020,
124 };
125
126 static int srf08_read_ranging(struct srf08_data *data)
127 {
128         struct i2c_client *client = data->client;
129         int ret, i;
130         int waittime;
131
132         mutex_lock(&data->lock);
133
134         ret = i2c_smbus_write_byte_data(data->client,
135                         SRF08_WRITE_COMMAND, SRF08_CMD_RANGING_CM);
136         if (ret < 0) {
137                 dev_err(&client->dev, "write command - err: %d\n", ret);
138                 mutex_unlock(&data->lock);
139                 return ret;
140         }
141
142         /*
143          * we read here until a correct version number shows up as
144          * suggested by the documentation
145          *
146          * with an ultrasonic speed of 343 m/s and a roundtrip of it
147          * sleep the expected duration and try to read from the device
148          * if nothing useful is read try it in a shorter grid
149          *
150          * polling for not more than 20 ms should be enough
151          */
152         waittime = 1 + data->range_mm / 172;
153         msleep(waittime);
154         for (i = 0; i < 4; i++) {
155                 ret = i2c_smbus_read_byte_data(data->client,
156                                                 SRF08_READ_SW_REVISION);
157
158                 /* check if a valid version number is read */
159                 if (ret < 255 && ret > 0)
160                         break;
161                 msleep(5);
162         }
163
164         if (ret >= 255 || ret <= 0) {
165                 dev_err(&client->dev, "device not ready\n");
166                 mutex_unlock(&data->lock);
167                 return -EIO;
168         }
169
170         ret = i2c_smbus_read_word_swapped(data->client,
171                                                 SRF08_READ_ECHO_1_HIGH);
172         if (ret < 0) {
173                 dev_err(&client->dev, "cannot read distance: ret=%d\n", ret);
174                 mutex_unlock(&data->lock);
175                 return ret;
176         }
177
178         mutex_unlock(&data->lock);
179
180         return ret;
181 }
182
183 static irqreturn_t srf08_trigger_handler(int irq, void *p)
184 {
185         struct iio_poll_func *pf = p;
186         struct iio_dev *indio_dev = pf->indio_dev;
187         struct srf08_data *data = iio_priv(indio_dev);
188         s16 sensor_data;
189
190         sensor_data = srf08_read_ranging(data);
191         if (sensor_data < 0)
192                 goto err;
193
194         mutex_lock(&data->lock);
195
196         data->scan.chan = sensor_data;
197         iio_push_to_buffers_with_timestamp(indio_dev,
198                                            &data->scan, pf->timestamp);
199
200         mutex_unlock(&data->lock);
201 err:
202         iio_trigger_notify_done(indio_dev->trig);
203         return IRQ_HANDLED;
204 }
205
206 static int srf08_read_raw(struct iio_dev *indio_dev,
207                             struct iio_chan_spec const *channel, int *val,
208                             int *val2, long mask)
209 {
210         struct srf08_data *data = iio_priv(indio_dev);
211         int ret;
212
213         if (channel->type != IIO_DISTANCE)
214                 return -EINVAL;
215
216         switch (mask) {
217         case IIO_CHAN_INFO_RAW:
218                 ret = srf08_read_ranging(data);
219                 if (ret < 0)
220                         return ret;
221                 *val = ret;
222                 return IIO_VAL_INT;
223         case IIO_CHAN_INFO_SCALE:
224                 /* 1 LSB is 1 cm */
225                 *val = 0;
226                 *val2 = 10000;
227                 return IIO_VAL_INT_PLUS_MICRO;
228         default:
229                 return -EINVAL;
230         }
231 }
232
233 static ssize_t srf08_show_range_mm_available(struct device *dev,
234                                 struct device_attribute *attr, char *buf)
235 {
236         return sprintf(buf, "[0.043 0.043 11.008]\n");
237 }
238
239 static IIO_DEVICE_ATTR(sensor_max_range_available, S_IRUGO,
240                                 srf08_show_range_mm_available, NULL, 0);
241
242 static ssize_t srf08_show_range_mm(struct device *dev,
243                                 struct device_attribute *attr, char *buf)
244 {
245         struct iio_dev *indio_dev = dev_to_iio_dev(dev);
246         struct srf08_data *data = iio_priv(indio_dev);
247
248         return sprintf(buf, "%d.%03d\n", data->range_mm / 1000,
249                                                 data->range_mm % 1000);
250 }
251
252 /*
253  * set the range of the sensor to an even multiple of 43 mm
254  * which corresponds to 1 LSB in the register
255  *
256  * register value    corresponding range
257  *         0x00             43 mm
258  *         0x01             86 mm
259  *         0x02            129 mm
260  *         ...
261  *         0xFF          11008 mm
262  */
263 static ssize_t srf08_write_range_mm(struct srf08_data *data, unsigned int val)
264 {
265         int ret;
266         struct i2c_client *client = data->client;
267         unsigned int mod;
268         u8 regval;
269
270         ret = val / 43 - 1;
271         mod = val % 43;
272
273         if (mod || (ret < 0) || (ret > 255))
274                 return -EINVAL;
275
276         regval = ret;
277
278         mutex_lock(&data->lock);
279
280         ret = i2c_smbus_write_byte_data(client, SRF08_WRITE_RANGE, regval);
281         if (ret < 0) {
282                 dev_err(&client->dev, "write_range - err: %d\n", ret);
283                 mutex_unlock(&data->lock);
284                 return ret;
285         }
286
287         data->range_mm = val;
288
289         mutex_unlock(&data->lock);
290
291         return 0;
292 }
293
294 static ssize_t srf08_store_range_mm(struct device *dev,
295                                         struct device_attribute *attr,
296                                         const char *buf, size_t len)
297 {
298         struct iio_dev *indio_dev = dev_to_iio_dev(dev);
299         struct srf08_data *data = iio_priv(indio_dev);
300         int ret;
301         int integer, fract;
302
303         ret = iio_str_to_fixpoint(buf, 100, &integer, &fract);
304         if (ret)
305                 return ret;
306
307         ret = srf08_write_range_mm(data, integer * 1000 + fract);
308         if (ret < 0)
309                 return ret;
310
311         return len;
312 }
313
314 static IIO_DEVICE_ATTR(sensor_max_range, S_IRUGO | S_IWUSR,
315                         srf08_show_range_mm, srf08_store_range_mm, 0);
316
317 static ssize_t srf08_show_sensitivity_available(struct device *dev,
318                                 struct device_attribute *attr, char *buf)
319 {
320         int i, len = 0;
321         struct iio_dev *indio_dev = dev_to_iio_dev(dev);
322         struct srf08_data *data = iio_priv(indio_dev);
323
324         for (i = 0; i < data->chip_info->num_sensitivity_avail; i++)
325                 if (data->chip_info->sensitivity_avail[i])
326                         len += sprintf(buf + len, "%d ",
327                                 data->chip_info->sensitivity_avail[i]);
328
329         len += sprintf(buf + len, "\n");
330
331         return len;
332 }
333
334 static IIO_DEVICE_ATTR(sensor_sensitivity_available, S_IRUGO,
335                                 srf08_show_sensitivity_available, NULL, 0);
336
337 static ssize_t srf08_show_sensitivity(struct device *dev,
338                                 struct device_attribute *attr, char *buf)
339 {
340         struct iio_dev *indio_dev = dev_to_iio_dev(dev);
341         struct srf08_data *data = iio_priv(indio_dev);
342         int len;
343
344         len = sprintf(buf, "%d\n", data->sensitivity);
345
346         return len;
347 }
348
349 static ssize_t srf08_write_sensitivity(struct srf08_data *data,
350                                                         unsigned int val)
351 {
352         struct i2c_client *client = data->client;
353         int ret, i;
354         u8 regval;
355
356         if (!val)
357                 return -EINVAL;
358
359         for (i = 0; i < data->chip_info->num_sensitivity_avail; i++)
360                 if (val && (val == data->chip_info->sensitivity_avail[i])) {
361                         regval = i;
362                         break;
363                 }
364
365         if (i >= data->chip_info->num_sensitivity_avail)
366                 return -EINVAL;
367
368         mutex_lock(&data->lock);
369
370         ret = i2c_smbus_write_byte_data(client, SRF08_WRITE_MAX_GAIN, regval);
371         if (ret < 0) {
372                 dev_err(&client->dev, "write_sensitivity - err: %d\n", ret);
373                 mutex_unlock(&data->lock);
374                 return ret;
375         }
376
377         data->sensitivity = val;
378
379         mutex_unlock(&data->lock);
380
381         return 0;
382 }
383
384 static ssize_t srf08_store_sensitivity(struct device *dev,
385                                                 struct device_attribute *attr,
386                                                 const char *buf, size_t len)
387 {
388         struct iio_dev *indio_dev = dev_to_iio_dev(dev);
389         struct srf08_data *data = iio_priv(indio_dev);
390         int ret;
391         unsigned int val;
392
393         ret = kstrtouint(buf, 10, &val);
394         if (ret)
395                 return ret;
396
397         ret = srf08_write_sensitivity(data, val);
398         if (ret < 0)
399                 return ret;
400
401         return len;
402 }
403
404 static IIO_DEVICE_ATTR(sensor_sensitivity, S_IRUGO | S_IWUSR,
405                         srf08_show_sensitivity, srf08_store_sensitivity, 0);
406
407 static struct attribute *srf08_attributes[] = {
408         &iio_dev_attr_sensor_max_range.dev_attr.attr,
409         &iio_dev_attr_sensor_max_range_available.dev_attr.attr,
410         &iio_dev_attr_sensor_sensitivity.dev_attr.attr,
411         &iio_dev_attr_sensor_sensitivity_available.dev_attr.attr,
412         NULL,
413 };
414
415 static const struct attribute_group srf08_attribute_group = {
416         .attrs = srf08_attributes,
417 };
418
419 static const struct iio_chan_spec srf08_channels[] = {
420         {
421                 .type = IIO_DISTANCE,
422                 .info_mask_separate =
423                                 BIT(IIO_CHAN_INFO_RAW) |
424                                 BIT(IIO_CHAN_INFO_SCALE),
425                 .scan_index = 0,
426                 .scan_type = {
427                         .sign = 's',
428                         .realbits = 16,
429                         .storagebits = 16,
430                         .endianness = IIO_CPU,
431                 },
432         },
433         IIO_CHAN_SOFT_TIMESTAMP(1),
434 };
435
436 static const struct iio_info srf08_info = {
437         .read_raw = srf08_read_raw,
438         .attrs = &srf08_attribute_group,
439         .driver_module = THIS_MODULE,
440 };
441
442 /*
443  * srf02 don't have an adjustable range or sensitivity,
444  * so we don't need attributes at all
445  */
446 static const struct iio_info srf02_info = {
447         .read_raw = srf08_read_raw,
448         .driver_module = THIS_MODULE,
449 };
450
451 static int srf08_probe(struct i2c_client *client,
452                                          const struct i2c_device_id *id)
453 {
454         struct iio_dev *indio_dev;
455         struct srf08_data *data;
456         int ret;
457
458         if (!i2c_check_functionality(client->adapter,
459                                         I2C_FUNC_SMBUS_READ_BYTE_DATA |
460                                         I2C_FUNC_SMBUS_WRITE_BYTE_DATA |
461                                         I2C_FUNC_SMBUS_READ_WORD_DATA))
462                 return -ENODEV;
463
464         indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
465         if (!indio_dev)
466                 return -ENOMEM;
467
468         data = iio_priv(indio_dev);
469         i2c_set_clientdata(client, indio_dev);
470         data->client = client;
471         data->sensor_type = (enum srf08_sensor_type)id->driver_data;
472
473         switch (data->sensor_type) {
474         case SRF02:
475                 data->chip_info = &srf02_chip_info;
476                 indio_dev->info = &srf02_info;
477                 break;
478         case SRF08:
479                 data->chip_info = &srf08_chip_info;
480                 indio_dev->info = &srf08_info;
481                 break;
482         case SRF10:
483                 data->chip_info = &srf10_chip_info;
484                 indio_dev->info = &srf08_info;
485                 break;
486         default:
487                 return -EINVAL;
488         }
489
490         indio_dev->name = id->name;
491         indio_dev->dev.parent = &client->dev;
492         indio_dev->modes = INDIO_DIRECT_MODE;
493         indio_dev->channels = srf08_channels;
494         indio_dev->num_channels = ARRAY_SIZE(srf08_channels);
495
496         mutex_init(&data->lock);
497
498         ret = devm_iio_triggered_buffer_setup(&client->dev, indio_dev,
499                         iio_pollfunc_store_time, srf08_trigger_handler, NULL);
500         if (ret < 0) {
501                 dev_err(&client->dev, "setup of iio triggered buffer failed\n");
502                 return ret;
503         }
504
505         if (data->chip_info->range_default) {
506                 /*
507                  * set default range of device in mm here
508                  * these register values cannot be read from the hardware
509                  * therefore set driver specific default values
510                  *
511                  * srf02 don't have a default value so it'll be omitted
512                  */
513                 ret = srf08_write_range_mm(data,
514                                         data->chip_info->range_default);
515                 if (ret < 0)
516                         return ret;
517         }
518
519         if (data->chip_info->sensitivity_default) {
520                 /*
521                  * set default sensitivity of device here
522                  * these register values cannot be read from the hardware
523                  * therefore set driver specific default values
524                  *
525                  * srf02 don't have a default value so it'll be omitted
526                  */
527                 ret = srf08_write_sensitivity(data,
528                                 data->chip_info->sensitivity_default);
529                 if (ret < 0)
530                         return ret;
531         }
532
533         return devm_iio_device_register(&client->dev, indio_dev);
534 }
535
536 static const struct of_device_id of_srf08_match[] = {
537         { .compatible = "devantech,srf02", (void *)SRF02},
538         { .compatible = "devantech,srf08", (void *)SRF08},
539         { .compatible = "devantech,srf10", (void *)SRF10},
540         {},
541 };
542
543 MODULE_DEVICE_TABLE(of, of_srf08_match);
544
545 static const struct i2c_device_id srf08_id[] = {
546         { "srf02", SRF02 },
547         { "srf08", SRF08 },
548         { "srf10", SRF10 },
549         { }
550 };
551 MODULE_DEVICE_TABLE(i2c, srf08_id);
552
553 static struct i2c_driver srf08_driver = {
554         .driver = {
555                 .name   = "srf08",
556                 .of_match_table = of_srf08_match,
557         },
558         .probe = srf08_probe,
559         .id_table = srf08_id,
560 };
561 module_i2c_driver(srf08_driver);
562
563 MODULE_AUTHOR("Andreas Klinger <ak@it-klinger.de>");
564 MODULE_DESCRIPTION("Devantech SRF02/SRF08/SRF10 i2c ultrasonic ranger driver");
565 MODULE_LICENSE("GPL");