iio: initial pass at getting channel information from scan_elements
[contrib/mraa.git] / src / iio / iio.c
1 /*
2  * Author: Brendan Le Foll <brendan.le.foll@intel.com>
3  * Copyright (c) 2015 Intel Corporation.
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining
6  * a copy of this software and associated documentation files (the
7  * "Software"), to deal in the Software without restriction, including
8  * without limitation the rights to use, copy, modify, merge, publish,
9  * distribute, sublicense, and/or sell copies of the Software, and to
10  * permit persons to whom the Software is furnished to do so, subject to
11  * the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be
14  * included in all copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
20  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
21  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
22  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
23  */
24
25 #include "iio.h"
26 #include "mraa_internal.h"
27 #include "dirent.h"
28 #include <string.h>
29 #include <poll.h>
30
31 #define MAX_SIZE 128
32 #define IIO_DEVICE "iio:device"
33 #define IIO_SCAN_ELEM "scan_elements"
34 #define IIO_SLASH_DEV "/dev/"IIO_DEVICE
35 #define IIO_SYSFS_DEVICE "/sys/bus/iio/devices/"IIO_DEVICE
36
37 mraa_iio_context
38 mraa_iio_init(int device)
39 {
40     if (device > plat->iio_device_count) {
41         return NULL;
42     }
43
44     mraa_iio_get_channel_data(&plat->iio_devices[device]);
45     return &plat->iio_devices[device];
46 }
47
48 int
49 mraa_iio_read_size(mraa_iio_context dev)
50 {
51     return dev->datasize;
52 }
53
54 mraa_iio_channel*
55 mraa_iio_get_channels(mraa_iio_context dev)
56 {
57     return dev->channels;
58 }
59
60 int
61 mraa_iio_get_channel_count(mraa_iio_context dev)
62 {
63     return dev->chan_num;
64 }
65
66 mraa_result_t
67 mraa_iio_get_channel_data(mraa_iio_context dev)
68 {
69     const struct dirent *ent;
70     DIR* dir;
71     int chan_num = 0;
72     char buf[MAX_SIZE];
73     char readbuf[32];
74     int fd;
75     int ret = 0;
76     int padint = 0;
77     int curr_bytes = 0;
78     char shortbuf, signchar;
79     memset(buf, 0, MAX_SIZE);
80     snprintf(buf, MAX_SIZE, IIO_SYSFS_DEVICE "%d/" IIO_SCAN_ELEM, dev->num);
81     dir = opendir(buf);
82     if (dir != NULL) {
83         while ((ent = readdir(dir)) != NULL) {
84             if (strcmp(ent->d_name + strlen(ent->d_name) - strlen("_en"), "_en") == 0) {
85                     chan_num++;
86             }
87         }
88     }
89     dev->chan_num = chan_num;
90     mraa_iio_channel* chan;
91     dev->channels = calloc(chan_num, sizeof(mraa_iio_channel));
92     seekdir(dir, 0);
93     while ((ent = readdir(dir)) != NULL) {
94         if (strcmp(ent->d_name + strlen(ent->d_name) - strlen("_index"), "_index") == 0) {
95             snprintf(buf, MAX_SIZE, IIO_SYSFS_DEVICE "%d/" IIO_SCAN_ELEM "/%s", dev->num, ent->d_name);
96             fd = open(buf, O_RDONLY);
97             if (fd > 0) {
98                 if (read(fd, readbuf, 2 * sizeof(char)) != 2) {
99                     break;
100                 }
101                 chan_num = ((int) strtol(readbuf, NULL, 10));
102                 chan = &dev->channels[chan_num];
103                 chan->index = chan_num;
104                 close(fd);
105
106                 buf[(strlen(buf)-5)] = '\0';
107                 char* str = strdup(buf);
108                 snprintf(buf, MAX_SIZE, "%stype", str);
109                 fd = open(buf, O_RDONLY);
110                 if (fd > 0) {
111                     read(fd, readbuf, 31 * sizeof(char));
112                     ret = sscanf(readbuf, "%ce:%c%u/%u>>%u", &shortbuf,
113                                     &signchar, &chan->bits_used,
114                                     &padint, &chan->shift);
115                     chan->bytes = padint / 8;
116                     if (curr_bytes % chan->bytes == 0) {
117                         chan->location = curr_bytes;
118                     } else {
119                         chan->location = curr_bytes - curr_bytes%chan->bytes + chan->bytes;
120                     }
121                     curr_bytes = chan->location + chan->bytes;
122                     if (ret < 0) {
123                         // probably should be 5?
124                         return MRAA_IO_SETUP_FAILURE;
125                     }
126                     chan->signedd = (signchar == 's');
127                     chan->lendian = (shortbuf == 'l');
128                     if (chan->bits_used == 64) {
129                         chan->mask = ~0;
130                     } else {
131                         chan->mask = (1 << chan->bits_used) - 1;
132                     }
133
134                     close(fd);
135                 }
136                 // todo - read scale & _en attributes
137             }
138         }
139     }
140     dev->datasize = curr_bytes;
141
142     return MRAA_SUCCESS;
143 }
144
145 const char*
146 mraa_iio_get_device_name(mraa_iio_context dev)
147 {
148     return dev->name;
149 }
150
151 mraa_result_t
152 mraa_iio_read(mraa_iio_context dev, const char* attr_chan, float* data)
153 {
154     char buf[64];
155     snprintf(buf, 64, IIO_SYSFS_DEVICE "%d/%s", dev->num, attr_chan);
156     int fd = open(buf, O_RDONLY);
157     if (fd != -1) {
158         int len = read(fd, &buf, 64);
159         *data = strtol(buf, NULL, 10);
160         return MRAA_SUCCESS;
161     }
162     return MRAA_ERROR_UNSPECIFIED;
163 }
164
165 mraa_result_t
166 mraa_iio_write(mraa_iio_context dev, const char* attr_chan)
167 {
168     return MRAA_ERROR_FEATURE_NOT_IMPLEMENTED;
169 }
170
171 static mraa_result_t
172 mraa_iio_wait_event(int fd, char* data)
173 {
174     struct pollfd pfd;
175
176     if (fd < 0) {
177         return MRAA_ERROR_INVALID_RESOURCE;
178     }
179
180     pfd.fd = fd;
181     pfd.events = POLLIN;
182
183     // Wait for it forever or until pthread_cancel
184     // poll is a cancelable point like sleep()
185     int x = poll(&pfd, 1, -1);
186
187     memset(data, 0, 100);
188     read(fd, data, 100);
189
190     return MRAA_SUCCESS;
191 }
192
193 static void*
194 mraa_iio_trigger_handler(void* arg)
195 {
196     mraa_iio_context dev = (mraa_iio_context) arg;
197     char data[MAX_SIZE*100];
198
199     for (;;) {
200         if (mraa_iio_wait_event(dev->fp, &data[0]) == MRAA_SUCCESS) {
201             pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, NULL);
202             dev->isr(&data);
203             pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
204         } else {
205             // we must have got an error code so die nicely
206             pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, NULL);
207             return NULL;
208         }
209     }
210 }
211
212 mraa_result_t
213 mraa_iio_trigger_buffer(mraa_iio_context dev, void (*fptr)(char* data), void* args)
214 {
215     char bu[MAX_SIZE];
216     if (dev->thread_id != 0) {
217         return MRAA_ERROR_NO_RESOURCES;
218     }
219
220     sprintf(bu, IIO_SLASH_DEV "%d", dev->num);
221     dev->fp = open(bu, O_RDONLY | O_NONBLOCK);
222     if (dev->fp == -1) {
223         return MRAA_ERROR_INVALID_RESOURCE;
224     }
225
226     dev->isr = fptr;
227     pthread_create(&dev->thread_id, NULL, mraa_iio_trigger_handler, (void*) dev);
228
229     return MRAA_SUCCESS;
230 }
231
232 #if 0
233 // does stop make any sense on iio devices?
234 mraa_result_t
235 mraa_iio_stop(mraa_iio_context dev)
236 {
237     return MRAA_SUCCESS;
238 }
239 #endif
240