]> sigrok.org Git - libsigrok.git/blob - src/hardware/sysclk-lwla/api.c
drivers: Factor out std_*_idx*().
[libsigrok.git] / src / hardware / sysclk-lwla / api.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2014 Daniel Elstner <daniel.kitta@gmail.com>
5  *
6  * This program is free software: you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation, either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <config.h>
21 #include <glib.h>
22 #include <libusb.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <libsigrok/libsigrok.h>
26 #include <libsigrok-internal.h>
27 #include "protocol.h"
28
29 static const uint32_t scanopts[] = {
30         SR_CONF_CONN,
31 };
32
33 static const uint32_t drvopts[] = {
34         SR_CONF_LOGIC_ANALYZER,
35 };
36
37 static const int32_t trigger_matches[] = {
38         SR_TRIGGER_ZERO,
39         SR_TRIGGER_ONE,
40         SR_TRIGGER_RISING,
41         SR_TRIGGER_FALLING,
42 };
43
44 static const char *trigger_source_names[] = {
45         [TRIGGER_CHANNELS] = "CH",
46         [TRIGGER_EXT_TRG] = "TRG",
47 };
48
49 static const char *signal_edge_names[] = {
50         [EDGE_POSITIVE] = "r",
51         [EDGE_NEGATIVE] = "f",
52 };
53
54 static struct sr_dev_inst *dev_inst_new(const struct model_info *model)
55 {
56         struct sr_dev_inst *sdi;
57         struct dev_context *devc;
58         int i;
59         char name[8];
60
61         devc = g_malloc0(sizeof(struct dev_context));
62         devc->model = model;
63         devc->active_fpga_config = FPGA_NOCONF;
64         devc->cfg_rle = TRUE;
65         devc->samplerate = model->samplerates[0];
66         devc->channel_mask = (UINT64_C(1) << model->num_channels) - 1;
67
68         sdi = g_malloc0(sizeof(struct sr_dev_inst));
69         sdi->status = SR_ST_INACTIVE;
70         sdi->vendor = g_strdup("SysClk");
71         sdi->model = g_strdup(model->name);
72         sdi->priv = devc;
73
74         for (i = 0; i < model->num_channels; i++) {
75                 g_snprintf(name, sizeof(name), "CH%d", i + 1);
76                 sr_channel_new(sdi, i, SR_CHANNEL_LOGIC, TRUE, name);
77         }
78
79         return sdi;
80 }
81
82 /* Create a new device instance for a libusb device if it is a SysClk LWLA
83  * device and also matches the connection specification.
84  */
85 static struct sr_dev_inst *dev_inst_new_matching(GSList *conn_matches,
86                                                  libusb_device *dev)
87 {
88         GSList *node;
89         struct sr_usb_dev_inst *usb;
90         const struct model_info *model;
91         struct sr_dev_inst *sdi;
92         struct libusb_device_descriptor des;
93         int bus, address, ret;
94         unsigned int vid, pid;
95
96         bus = libusb_get_bus_number(dev);
97         address = libusb_get_device_address(dev);
98
99         for (node = conn_matches; node != NULL; node = node->next) {
100                 usb = node->data;
101                 if (usb && usb->bus == bus && usb->address == address)
102                         break; /* found */
103         }
104         if (conn_matches && !node)
105                 return NULL; /* no match */
106
107         ret = libusb_get_device_descriptor(dev, &des);
108         if (ret != 0) {
109                 sr_err("Failed to get USB device descriptor: %s.",
110                         libusb_error_name(ret));
111                 return NULL;
112         }
113         vid = des.idVendor;
114         pid = des.idProduct;
115
116         /* Create sigrok device instance. */
117         if (vid == USB_VID_SYSCLK && pid == USB_PID_LWLA1016) {
118                 model = &lwla1016_info;
119         } else if (vid == USB_VID_SYSCLK && pid == USB_PID_LWLA1034) {
120                 model = &lwla1034_info;
121         } else {
122                 if (conn_matches)
123                         sr_warn("USB device %d.%d (%04x:%04x) is not a"
124                                 " SysClk LWLA.", bus, address, vid, pid);
125                 return NULL;
126         }
127         sdi = dev_inst_new(model);
128
129         sdi->inst_type = SR_INST_USB;
130         sdi->conn = sr_usb_dev_inst_new(bus, address, NULL);
131
132         return sdi;
133 }
134
135 static GSList *scan(struct sr_dev_driver *di, GSList *options)
136 {
137         GSList *conn_devices, *devices, *node;
138         struct drv_context *drvc;
139         struct sr_dev_inst *sdi;
140         struct sr_config *src;
141         const char *conn;
142         libusb_device **devlist;
143         ssize_t num_devs, i;
144
145         drvc = di->context;
146         conn = NULL;
147         conn_devices = NULL;
148         devices = NULL;
149
150         for (node = options; node != NULL; node = node->next) {
151                 src = node->data;
152                 if (src->key == SR_CONF_CONN) {
153                         conn = g_variant_get_string(src->data, NULL);
154                         break;
155                 }
156         }
157         if (conn) {
158                 /* Find devices matching the connection specification. */
159                 conn_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn);
160         }
161
162         /* List all libusb devices. */
163         num_devs = libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
164         if (num_devs < 0) {
165                 sr_err("Failed to list USB devices: %s.",
166                         libusb_error_name(num_devs));
167                 g_slist_free_full(conn_devices,
168                         (GDestroyNotify)&sr_usb_dev_inst_free);
169                 return NULL;
170         }
171
172         /* Scan the USB device list for matching LWLA devices. */
173         for (i = 0; i < num_devs; i++) {
174                 sdi = dev_inst_new_matching(conn_devices, devlist[i]);
175                 if (!sdi)
176                         continue; /* no match */
177
178                 /* Register device instance with driver. */
179                 devices = g_slist_append(devices, sdi);
180         }
181
182         libusb_free_device_list(devlist, 1);
183         g_slist_free_full(conn_devices, (GDestroyNotify)&sr_usb_dev_inst_free);
184
185         return std_scan_complete(di, devices);
186 }
187
188 /* Drain any pending data from the USB transfer buffers on the device.
189  * This may be necessary e.g. after a crash or generally to clean up after
190  * an abnormal condition.
191  */
192 static int drain_usb(struct sr_usb_dev_inst *usb, unsigned int endpoint)
193 {
194         int drained, xfer_len, ret;
195         unsigned char buf[512];
196         const unsigned int drain_timeout_ms = 10;
197
198         drained = 0;
199         do {
200                 xfer_len = 0;
201                 ret = libusb_bulk_transfer(usb->devhdl, endpoint,
202                                            buf, sizeof(buf), &xfer_len,
203                                            drain_timeout_ms);
204                 drained += xfer_len;
205         } while (ret == LIBUSB_SUCCESS && xfer_len != 0);
206
207         if (ret != LIBUSB_SUCCESS && ret != LIBUSB_ERROR_TIMEOUT) {
208                 sr_err("Failed to drain USB endpoint %u: %s.",
209                        endpoint & (LIBUSB_ENDPOINT_IN - 1),
210                        libusb_error_name(ret));
211                 return SR_ERR;
212         }
213         if (drained > 0) {
214                 sr_warn("Drained %d bytes from USB endpoint %u.",
215                         drained, endpoint & (LIBUSB_ENDPOINT_IN - 1));
216         }
217
218         return SR_OK;
219 }
220
221 static int dev_open(struct sr_dev_inst *sdi)
222 {
223         struct drv_context *drvc;
224         struct dev_context *devc;
225         struct sr_usb_dev_inst *usb;
226         int i, ret;
227
228         drvc = sdi->driver->context;
229         devc = sdi->priv;
230         usb = sdi->conn;
231
232         /* Try the whole shebang three times, fingers crossed. */
233         for (i = 0; i < 3; i++) {
234                 ret = sr_usb_open(drvc->sr_ctx->libusb_ctx, usb);
235                 if (ret != SR_OK)
236                         return ret;
237
238                 ret = libusb_set_configuration(usb->devhdl, USB_CONFIG);
239                 if (ret != LIBUSB_SUCCESS) {
240                         sr_err("Failed to set USB configuration: %s.",
241                                 libusb_error_name(ret));
242                         sr_usb_close(usb);
243                         return SR_ERR;
244                 }
245
246                 ret = libusb_claim_interface(usb->devhdl, USB_INTERFACE);
247                 if (ret != LIBUSB_SUCCESS) {
248                         sr_err("Failed to claim interface: %s.",
249                                 libusb_error_name(ret));
250                         sr_usb_close(usb);
251                         return SR_ERR;
252                 }
253
254                 ret = drain_usb(usb, EP_REPLY);
255                 if (ret != SR_OK) {
256                         sr_usb_close(usb);
257                         return ret;
258                 }
259                 /* This delay appears to be necessary for reliable operation. */
260                 g_usleep(30 * 1000);
261
262                 devc->active_fpga_config = FPGA_NOCONF;
263                 devc->short_transfer_quirk = FALSE;
264                 devc->state = STATE_IDLE;
265
266                 ret = (*devc->model->apply_fpga_config)(sdi);
267
268                 if (ret == SR_OK)
269                         ret = (*devc->model->device_init_check)(sdi);
270                 if (ret == SR_OK)
271                         break;
272
273                 /* Rinse and repeat. */
274                 sr_usb_close(usb);
275         }
276
277         if (ret == SR_OK && devc->short_transfer_quirk)
278                 sr_warn("Short transfer quirk detected! "
279                         "Memory reads will be slow.");
280         return ret;
281 }
282
283 static int dev_close(struct sr_dev_inst *sdi)
284 {
285         struct dev_context *devc;
286         struct sr_usb_dev_inst *usb;
287         int ret;
288
289         devc = sdi->priv;
290         usb = sdi->conn;
291
292         if (devc->acquisition) {
293                 sr_err("Cannot close device during acquisition!");
294                 /* Request stop, leak handle, and prepare for the worst. */
295                 devc->cancel_requested = TRUE;
296                 return SR_ERR_BUG;
297         }
298
299         /* Download of the shutdown bitstream, if any. */
300         ret = (*devc->model->apply_fpga_config)(sdi);
301         if (ret != SR_OK)
302                 sr_warn("Unable to shut down device.");
303
304         libusb_release_interface(usb->devhdl, USB_INTERFACE);
305
306         sr_usb_close(usb);
307
308         return SR_OK;
309 }
310
311 /* Check whether the device options contain a specific key.
312  * Also match against get/set/list bits if specified.
313  */
314 static int has_devopt(const struct model_info *model, uint32_t key)
315 {
316         unsigned int i;
317
318         for (i = 0; i < model->num_devopts; i++) {
319                 if ((model->devopts[i] & (SR_CONF_MASK | key)) == key)
320                         return TRUE;
321         }
322
323         return FALSE;
324 }
325
326 static int config_get(uint32_t key, GVariant **data,
327         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
328 {
329         struct dev_context *devc;
330         unsigned int idx;
331
332         (void)cg;
333
334         if (!sdi)
335                 return SR_ERR_ARG;
336
337         devc = sdi->priv;
338
339         if (!has_devopt(devc->model, key | SR_CONF_GET))
340                 return SR_ERR_NA;
341
342         switch (key) {
343         case SR_CONF_SAMPLERATE:
344                 *data = g_variant_new_uint64(devc->samplerate);
345                 break;
346         case SR_CONF_LIMIT_MSEC:
347                 *data = g_variant_new_uint64(devc->limit_msec);
348                 break;
349         case SR_CONF_LIMIT_SAMPLES:
350                 *data = g_variant_new_uint64(devc->limit_samples);
351                 break;
352         case SR_CONF_RLE:
353                 *data = g_variant_new_boolean(devc->cfg_rle);
354                 break;
355         case SR_CONF_EXTERNAL_CLOCK:
356                 *data = g_variant_new_boolean(devc->cfg_clock_source
357                                                 == CLOCK_EXT_CLK);
358                 break;
359         case SR_CONF_CLOCK_EDGE:
360                 idx = devc->cfg_clock_edge;
361                 if (idx >= ARRAY_SIZE(signal_edge_names))
362                         return SR_ERR_BUG;
363                 *data = g_variant_new_string(signal_edge_names[idx]);
364                 break;
365         case SR_CONF_TRIGGER_SOURCE:
366                 idx = devc->cfg_trigger_source;
367                 if (idx >= ARRAY_SIZE(trigger_source_names))
368                         return SR_ERR_BUG;
369                 *data = g_variant_new_string(trigger_source_names[idx]);
370                 break;
371         case SR_CONF_TRIGGER_SLOPE:
372                 idx = devc->cfg_trigger_slope;
373                 if (idx >= ARRAY_SIZE(signal_edge_names))
374                         return SR_ERR_BUG;
375                 *data = g_variant_new_string(signal_edge_names[idx]);
376                 break;
377         default:
378                 /* Must not happen for a key listed in devopts. */
379                 return SR_ERR_BUG;
380         }
381
382         return SR_OK;
383 }
384
385 static int config_set(uint32_t key, GVariant *data,
386         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
387 {
388         uint64_t value;
389         struct dev_context *devc;
390         int idx;
391
392         (void)cg;
393
394         if (!sdi)
395                 return SR_ERR_ARG;
396
397         devc = sdi->priv;
398
399         if (!has_devopt(devc->model, key | SR_CONF_SET))
400                 return SR_ERR_NA;
401
402         switch (key) {
403         case SR_CONF_SAMPLERATE:
404                 value = g_variant_get_uint64(data);
405                 if (value < devc->model->samplerates[devc->model->num_samplerates - 1]
406                                 || value > devc->model->samplerates[0])
407                         return SR_ERR_SAMPLERATE;
408                 devc->samplerate = value;
409                 break;
410         case SR_CONF_LIMIT_MSEC:
411                 value = g_variant_get_uint64(data);
412                 if (value > MAX_LIMIT_MSEC)
413                         return SR_ERR_ARG;
414                 devc->limit_msec = value;
415                 break;
416         case SR_CONF_LIMIT_SAMPLES:
417                 value = g_variant_get_uint64(data);
418                 if (value > MAX_LIMIT_SAMPLES)
419                         return SR_ERR_ARG;
420                 devc->limit_samples = value;
421                 break;
422         case SR_CONF_RLE:
423                 devc->cfg_rle = g_variant_get_boolean(data);
424                 break;
425         case SR_CONF_EXTERNAL_CLOCK:
426                 devc->cfg_clock_source = (g_variant_get_boolean(data))
427                         ? CLOCK_EXT_CLK : CLOCK_INTERNAL;
428                 break;
429         case SR_CONF_CLOCK_EDGE:
430                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(signal_edge_names))) < 0)
431                         return SR_ERR_ARG;
432                 devc->cfg_clock_edge = idx;
433                 break;
434         case SR_CONF_TRIGGER_SOURCE:
435                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(trigger_source_names))) < 0)
436                         return SR_ERR_ARG;
437                 devc->cfg_trigger_source = idx;
438                 break;
439         case SR_CONF_TRIGGER_SLOPE:
440                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(signal_edge_names))) < 0)
441                         return SR_ERR_ARG;
442                 devc->cfg_trigger_slope = idx;
443                 break;
444         default:
445                 /* Must not happen for a key listed in devopts. */
446                 return SR_ERR_BUG;
447         }
448
449         return SR_OK;
450 }
451
452 static int config_channel_set(const struct sr_dev_inst *sdi,
453         struct sr_channel *ch, unsigned int changes)
454 {
455         uint64_t channel_bit;
456         struct dev_context *devc;
457
458         if (!sdi)
459                 return SR_ERR_ARG;
460
461         devc = sdi->priv;
462
463         if (ch->index < 0 || ch->index >= devc->model->num_channels) {
464                 sr_err("Channel index %d out of range.", ch->index);
465                 return SR_ERR_BUG;
466         }
467
468         if ((changes & SR_CHANNEL_SET_ENABLED) != 0) {
469                 channel_bit = UINT64_C(1) << ch->index;
470
471                 /* Enable or disable logic input for this channel. */
472                 if (ch->enabled)
473                         devc->channel_mask |= channel_bit;
474                 else
475                         devc->channel_mask &= ~channel_bit;
476         }
477
478         return SR_OK;
479 }
480
481 /* Derive trigger masks from the session's trigger configuration. */
482 static int prepare_trigger_masks(const struct sr_dev_inst *sdi)
483 {
484         uint64_t trigger_mask, trigger_values, trigger_edge_mask;
485         uint64_t level_bit, type_bit;
486         struct dev_context *devc;
487         struct sr_trigger *trigger;
488         struct sr_trigger_stage *stage;
489         struct sr_trigger_match *match;
490         const GSList *node;
491         int idx;
492         enum sr_trigger_matches trg;
493
494         devc = sdi->priv;
495
496         trigger = sr_session_trigger_get(sdi->session);
497         if (!trigger || !trigger->stages)
498                 return SR_OK;
499
500         if (trigger->stages->next) {
501                 sr_err("This device only supports 1 trigger stage.");
502                 return SR_ERR_ARG;
503         }
504         stage = trigger->stages->data;
505
506         trigger_mask = 0;
507         trigger_values = 0;
508         trigger_edge_mask = 0;
509
510         for (node = stage->matches; node; node = node->next) {
511                 match = node->data;
512
513                 if (!match->channel->enabled)
514                         continue; /* Ignore disabled channel. */
515
516                 idx = match->channel->index;
517                 trg = match->match;
518
519                 if (idx < 0 || idx >= devc->model->num_channels) {
520                         sr_err("Channel index %d out of range.", idx);
521                         return SR_ERR_BUG; /* Should not happen. */
522                 }
523                 if (trg != SR_TRIGGER_ZERO
524                                 && trg != SR_TRIGGER_ONE
525                                 && trg != SR_TRIGGER_RISING
526                                 && trg != SR_TRIGGER_FALLING) {
527                         sr_err("Unsupported trigger match for CH%d.", idx + 1);
528                         return SR_ERR_ARG;
529                 }
530                 level_bit = (trg == SR_TRIGGER_ONE
531                         || trg == SR_TRIGGER_RISING) ? 1 : 0;
532                 type_bit = (trg == SR_TRIGGER_RISING
533                         || trg == SR_TRIGGER_FALLING) ? 1 : 0;
534
535                 trigger_mask |= UINT64_C(1) << idx;
536                 trigger_values |= level_bit << idx;
537                 trigger_edge_mask |= type_bit << idx;
538         }
539         devc->trigger_mask = trigger_mask;
540         devc->trigger_values = trigger_values;
541         devc->trigger_edge_mask = trigger_edge_mask;
542
543         return SR_OK;
544 }
545
546 static int config_commit(const struct sr_dev_inst *sdi)
547 {
548         struct dev_context *devc;
549         int ret;
550
551         devc = sdi->priv;
552
553         if (devc->acquisition) {
554                 sr_err("Acquisition still in progress?");
555                 return SR_ERR;
556         }
557
558         ret = prepare_trigger_masks(sdi);
559         if (ret != SR_OK)
560                 return ret;
561
562         ret = (*devc->model->apply_fpga_config)(sdi);
563         if (ret != SR_OK) {
564                 sr_err("Failed to apply FPGA configuration.");
565                 return ret;
566         }
567
568         return SR_OK;
569 }
570
571 static int config_list(uint32_t key, GVariant **data,
572         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
573 {
574         struct dev_context *devc;
575
576         devc = (sdi) ? sdi->priv : NULL;
577
578         switch (key) {
579         case SR_CONF_SCAN_OPTIONS:
580         case SR_CONF_DEVICE_OPTIONS:
581                 return std_opts_config_list(key, data, sdi, cg,
582                         ARRAY_AND_SIZE(scanopts), ARRAY_AND_SIZE(drvopts),
583                         (devc) ? devc->model->devopts : NULL,
584                         (devc) ? devc->model->num_devopts : 0);
585         }
586
587         if (!has_devopt(devc->model, key | SR_CONF_LIST))
588                 return SR_ERR_NA;
589
590         switch (key) {
591         case SR_CONF_SAMPLERATE:
592                 *data = std_gvar_samplerates(devc->model->samplerates, devc->model->num_samplerates);
593                 break;
594         case SR_CONF_TRIGGER_MATCH:
595                 *data = std_gvar_array_i32(ARRAY_AND_SIZE(trigger_matches));
596                 break;
597         case SR_CONF_TRIGGER_SOURCE:
598                 *data = g_variant_new_strv(ARRAY_AND_SIZE(trigger_source_names));
599                 break;
600         case SR_CONF_TRIGGER_SLOPE:
601         case SR_CONF_CLOCK_EDGE:
602                 *data = g_variant_new_strv(ARRAY_AND_SIZE(signal_edge_names));
603                 break;
604         default:
605                 /* Must not happen for a key listed in devopts. */
606                 return SR_ERR_BUG;
607         }
608
609         return SR_OK;
610 }
611
612 /* Set up the device hardware to begin capturing samples as soon as the
613  * configured trigger conditions are met, or immediately if no triggers
614  * are configured.
615  */
616 static int dev_acquisition_start(const struct sr_dev_inst *sdi)
617 {
618         return lwla_start_acquisition(sdi);
619 }
620
621 static int dev_acquisition_stop(struct sr_dev_inst *sdi)
622 {
623         struct dev_context *devc;
624
625         devc = sdi->priv;
626
627         if (devc->state != STATE_IDLE && !devc->cancel_requested) {
628                 devc->cancel_requested = TRUE;
629                 sr_dbg("Requesting cancel.");
630         }
631
632         return SR_OK;
633 }
634
635 static struct sr_dev_driver sysclk_lwla_driver_info = {
636         .name = "sysclk-lwla",
637         .longname = "SysClk LWLA series",
638         .api_version = 1,
639         .init = std_init,
640         .cleanup = std_cleanup,
641         .scan = scan,
642         .dev_list = std_dev_list,
643         .dev_clear = std_dev_clear,
644         .config_get = config_get,
645         .config_set = config_set,
646         .config_channel_set = config_channel_set,
647         .config_commit = config_commit,
648         .config_list = config_list,
649         .dev_open = dev_open,
650         .dev_close = dev_close,
651         .dev_acquisition_start = dev_acquisition_start,
652         .dev_acquisition_stop = dev_acquisition_stop,
653         .context = NULL,
654 };
655 SR_REGISTER_DEV_DRIVER(sysclk_lwla_driver_info);