]> sigrok.org Git - libsigrok.git/blame_incremental - hardware/hantek-dso/api.c
Replace 'probe' with 'channel' in most places.
[libsigrok.git] / hardware / hantek-dso / api.c
... / ...
CommitLineData
1/*
2 * This file is part of the libsigrok project.
3 *
4 * Copyright (C) 2012 Bert Vermeulen <bert@biot.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 <stdio.h>
21#include <stdint.h>
22#include <stdlib.h>
23#include <sys/types.h>
24#include <sys/stat.h>
25#include <fcntl.h>
26#include <unistd.h>
27#include <string.h>
28#include <sys/time.h>
29#include <inttypes.h>
30#include <glib.h>
31#include <libusb.h>
32#include "libsigrok.h"
33#include "libsigrok-internal.h"
34#include "dso.h"
35
36/* Max time in ms before we want to check on USB events */
37/* TODO tune this properly */
38#define TICK 1
39
40#define NUM_TIMEBASE 10
41#define NUM_VDIV 8
42
43static const int32_t scanopts[] = {
44 SR_CONF_CONN,
45};
46
47static const int32_t devopts[] = {
48 SR_CONF_OSCILLOSCOPE,
49 SR_CONF_LIMIT_FRAMES,
50 SR_CONF_CONTINUOUS,
51 SR_CONF_TIMEBASE,
52 SR_CONF_BUFFERSIZE,
53 SR_CONF_TRIGGER_SOURCE,
54 SR_CONF_TRIGGER_SLOPE,
55 SR_CONF_HORIZ_TRIGGERPOS,
56 SR_CONF_FILTER,
57 SR_CONF_VDIV,
58 SR_CONF_COUPLING,
59 SR_CONF_NUM_TIMEBASE,
60 SR_CONF_NUM_VDIV,
61};
62
63static const char *channel_names[] = {
64 "CH1", "CH2",
65 NULL,
66};
67
68static const uint64_t buffersizes_32k[] = {
69 10240, 32768,
70};
71static const uint64_t buffersizes_512k[] = {
72 10240, 524288,
73};
74static const uint64_t buffersizes_14k[] = {
75 10240, 14336,
76};
77
78static const struct dso_profile dev_profiles[] = {
79 { 0x04b4, 0x2090, 0x04b5, 0x2090,
80 "Hantek", "DSO-2090",
81 buffersizes_32k,
82 FIRMWARE_DIR "/hantek-dso-2090.fw" },
83 { 0x04b4, 0x2150, 0x04b5, 0x2150,
84 "Hantek", "DSO-2150",
85 buffersizes_32k,
86 FIRMWARE_DIR "/hantek-dso-2150.fw" },
87 { 0x04b4, 0x2250, 0x04b5, 0x2250,
88 "Hantek", "DSO-2250",
89 buffersizes_512k,
90 FIRMWARE_DIR "/hantek-dso-2250.fw" },
91 { 0x04b4, 0x5200, 0x04b5, 0x5200,
92 "Hantek", "DSO-5200",
93 buffersizes_14k,
94 FIRMWARE_DIR "/hantek-dso-5200.fw" },
95 { 0x04b4, 0x520a, 0x04b5, 0x520a,
96 "Hantek", "DSO-5200A",
97 buffersizes_512k,
98 FIRMWARE_DIR "/hantek-dso-5200A.fw" },
99 { 0, 0, 0, 0, 0, 0, 0, 0 },
100};
101
102static const uint64_t timebases[][2] = {
103 /* microseconds */
104 { 10, 1000000 },
105 { 20, 1000000 },
106 { 40, 1000000 },
107 { 100, 1000000 },
108 { 200, 1000000 },
109 { 400, 1000000 },
110 /* milliseconds */
111 { 1, 1000 },
112 { 2, 1000 },
113 { 4, 1000 },
114 { 10, 1000 },
115 { 20, 1000 },
116 { 40, 1000 },
117 { 100, 1000 },
118 { 200, 1000 },
119 { 400, 1000 },
120};
121
122static const uint64_t vdivs[][2] = {
123 /* millivolts */
124 { 10, 1000 },
125 { 20, 1000 },
126 { 50, 1000 },
127 { 100, 1000 },
128 { 200, 1000 },
129 { 500, 1000 },
130 /* volts */
131 { 1, 1 },
132 { 2, 1 },
133 { 5, 1 },
134};
135
136static const char *trigger_sources[] = {
137 "CH1",
138 "CH2",
139 "EXT",
140 /* TODO: forced */
141};
142
143static const char *filter_targets[] = {
144 "CH1",
145 "CH2",
146 /* TODO: "TRIGGER", */
147};
148
149static const char *coupling[] = {
150 "AC",
151 "DC",
152 "GND",
153};
154
155SR_PRIV struct sr_dev_driver hantek_dso_driver_info;
156static struct sr_dev_driver *di = &hantek_dso_driver_info;
157
158static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data);
159
160static struct sr_dev_inst *dso_dev_new(int index, const struct dso_profile *prof)
161{
162 struct sr_dev_inst *sdi;
163 struct sr_channel *ch;
164 struct drv_context *drvc;
165 struct dev_context *devc;
166 int i;
167
168 sdi = sr_dev_inst_new(index, SR_ST_INITIALIZING,
169 prof->vendor, prof->model, NULL);
170 if (!sdi)
171 return NULL;
172 sdi->driver = di;
173
174 /*
175 * Add only the real channels -- EXT isn't a source of data, only
176 * a trigger source internal to the device.
177 */
178 for (i = 0; channel_names[i]; i++) {
179 if (!(ch = sr_probe_new(i, SR_PROBE_ANALOG, TRUE,
180 channel_names[i])))
181 return NULL;
182 sdi->channels = g_slist_append(sdi->channels, ch);
183 }
184
185 if (!(devc = g_try_malloc0(sizeof(struct dev_context)))) {
186 sr_err("Device context malloc failed.");
187 return NULL;
188 }
189
190 devc->profile = prof;
191 devc->dev_state = IDLE;
192 devc->timebase = DEFAULT_TIMEBASE;
193 devc->ch1_enabled = TRUE;
194 devc->ch2_enabled = TRUE;
195 devc->voltage_ch1 = DEFAULT_VOLTAGE;
196 devc->voltage_ch2 = DEFAULT_VOLTAGE;
197 devc->coupling_ch1 = DEFAULT_COUPLING;
198 devc->coupling_ch2 = DEFAULT_COUPLING;
199 devc->voffset_ch1 = DEFAULT_VERT_OFFSET;
200 devc->voffset_ch2 = DEFAULT_VERT_OFFSET;
201 devc->voffset_trigger = DEFAULT_VERT_TRIGGERPOS;
202 devc->framesize = DEFAULT_FRAMESIZE;
203 devc->triggerslope = SLOPE_POSITIVE;
204 devc->triggersource = g_strdup(DEFAULT_TRIGGER_SOURCE);
205 devc->triggerposition = DEFAULT_HORIZ_TRIGGERPOS;
206 sdi->priv = devc;
207 drvc = di->priv;
208 drvc->instances = g_slist_append(drvc->instances, sdi);
209
210 return sdi;
211}
212
213static int configure_channels(const struct sr_dev_inst *sdi)
214{
215 struct dev_context *devc;
216 struct sr_channel *ch;
217 const GSList *l;
218 int p;
219
220 devc = sdi->priv;
221
222 g_slist_free(devc->enabled_channels);
223 devc->ch1_enabled = devc->ch2_enabled = FALSE;
224 for (l = sdi->channels, p = 0; l; l = l->next, p++) {
225 ch = l->data;
226 if (p == 0)
227 devc->ch1_enabled = ch->enabled;
228 else
229 devc->ch2_enabled = ch->enabled;
230 if (ch->enabled)
231 devc->enabled_channels = g_slist_append(devc->enabled_channels, ch);
232 }
233
234 return SR_OK;
235}
236
237static void clear_dev_context(void *priv)
238{
239 struct dev_context *devc;
240
241 devc = priv;
242 g_free(devc->triggersource);
243 g_slist_free(devc->enabled_channels);
244
245}
246
247static int dev_clear(void)
248{
249 return std_dev_clear(di, clear_dev_context);
250}
251
252static int init(struct sr_context *sr_ctx)
253{
254 return std_init(sr_ctx, di, LOG_PREFIX);
255}
256
257static GSList *scan(GSList *options)
258{
259 struct drv_context *drvc;
260 struct dev_context *devc;
261 struct sr_dev_inst *sdi;
262 struct sr_usb_dev_inst *usb;
263 struct sr_config *src;
264 const struct dso_profile *prof;
265 GSList *l, *devices, *conn_devices;
266 struct libusb_device_descriptor des;
267 libusb_device **devlist;
268 int devcnt, ret, i, j;
269 const char *conn;
270
271 drvc = di->priv;
272
273 devcnt = 0;
274 devices = 0;
275
276 conn = NULL;
277 for (l = options; l; l = l->next) {
278 src = l->data;
279 if (src->key == SR_CONF_CONN) {
280 conn = g_variant_get_string(src->data, NULL);
281 break;
282 }
283 }
284 if (conn)
285 conn_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn);
286 else
287 conn_devices = NULL;
288
289 /* Find all Hantek DSO devices and upload firmware to all of them. */
290 libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
291 for (i = 0; devlist[i]; i++) {
292 if (conn) {
293 usb = NULL;
294 for (l = conn_devices; l; l = l->next) {
295 usb = l->data;
296 if (usb->bus == libusb_get_bus_number(devlist[i])
297 && usb->address == libusb_get_device_address(devlist[i]))
298 break;
299 }
300 if (!l)
301 /* This device matched none of the ones that
302 * matched the conn specification. */
303 continue;
304 }
305
306 if ((ret = libusb_get_device_descriptor(devlist[i], &des))) {
307 sr_err("Failed to get device descriptor: %s.",
308 libusb_error_name(ret));
309 continue;
310 }
311
312 prof = NULL;
313 for (j = 0; dev_profiles[j].orig_vid; j++) {
314 if (des.idVendor == dev_profiles[j].orig_vid
315 && des.idProduct == dev_profiles[j].orig_pid) {
316 /* Device matches the pre-firmware profile. */
317 prof = &dev_profiles[j];
318 sr_dbg("Found a %s %s.", prof->vendor, prof->model);
319 sdi = dso_dev_new(devcnt, prof);
320 devices = g_slist_append(devices, sdi);
321 devc = sdi->priv;
322 if (ezusb_upload_firmware(devlist[i], USB_CONFIGURATION,
323 prof->firmware) == SR_OK)
324 /* Remember when the firmware on this device was updated */
325 devc->fw_updated = g_get_monotonic_time();
326 else
327 sr_err("Firmware upload failed for "
328 "device %d.", devcnt);
329 /* Dummy USB address of 0xff will get overwritten later. */
330 sdi->conn = sr_usb_dev_inst_new(
331 libusb_get_bus_number(devlist[i]), 0xff, NULL);
332 devcnt++;
333 break;
334 } else if (des.idVendor == dev_profiles[j].fw_vid
335 && des.idProduct == dev_profiles[j].fw_pid) {
336 /* Device matches the post-firmware profile. */
337 prof = &dev_profiles[j];
338 sr_dbg("Found a %s %s.", prof->vendor, prof->model);
339 sdi = dso_dev_new(devcnt, prof);
340 sdi->status = SR_ST_INACTIVE;
341 devices = g_slist_append(devices, sdi);
342 devc = sdi->priv;
343 sdi->inst_type = SR_INST_USB;
344 sdi->conn = sr_usb_dev_inst_new(
345 libusb_get_bus_number(devlist[i]),
346 libusb_get_device_address(devlist[i]), NULL);
347 devcnt++;
348 break;
349 }
350 }
351 if (!prof)
352 /* not a supported VID/PID */
353 continue;
354 }
355 libusb_free_device_list(devlist, 1);
356
357 return devices;
358}
359
360static GSList *dev_list(void)
361{
362 return ((struct drv_context *)(di->priv))->instances;
363}
364
365static int dev_open(struct sr_dev_inst *sdi)
366{
367 struct dev_context *devc;
368 struct sr_usb_dev_inst *usb;
369 int64_t timediff_us, timediff_ms;
370 int err;
371
372 devc = sdi->priv;
373 usb = sdi->conn;
374
375 /*
376 * If the firmware was recently uploaded, wait up to MAX_RENUM_DELAY_MS
377 * for the FX2 to renumerate.
378 */
379 err = SR_ERR;
380 if (devc->fw_updated > 0) {
381 sr_info("Waiting for device to reset.");
382 /* Takes >= 300ms for the FX2 to be gone from the USB bus. */
383 g_usleep(300 * 1000);
384 timediff_ms = 0;
385 while (timediff_ms < MAX_RENUM_DELAY_MS) {
386 if ((err = dso_open(sdi)) == SR_OK)
387 break;
388 g_usleep(100 * 1000);
389 timediff_us = g_get_monotonic_time() - devc->fw_updated;
390 timediff_ms = timediff_us / 1000;
391 sr_spew("Waited %" PRIi64 " ms.", timediff_ms);
392 }
393 sr_info("Device came back after %d ms.", timediff_ms);
394 } else {
395 err = dso_open(sdi);
396 }
397
398 if (err != SR_OK) {
399 sr_err("Unable to open device.");
400 return SR_ERR;
401 }
402
403 err = libusb_claim_interface(usb->devhdl, USB_INTERFACE);
404 if (err != 0) {
405 sr_err("Unable to claim interface: %s.",
406 libusb_error_name(err));
407 return SR_ERR;
408 }
409
410 return SR_OK;
411}
412
413static int dev_close(struct sr_dev_inst *sdi)
414{
415 dso_close(sdi);
416
417 return SR_OK;
418}
419
420static int cleanup(void)
421{
422 return dev_clear();
423}
424
425static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi,
426 const struct sr_channel_group *cg)
427{
428 struct sr_usb_dev_inst *usb;
429 char str[128];
430
431 (void)cg;
432
433 switch (id) {
434 case SR_CONF_CONN:
435 if (!sdi || !sdi->conn)
436 return SR_ERR_ARG;
437 usb = sdi->conn;
438 if (usb->address == 255)
439 /* Device still needs to re-enumerate after firmware
440 * upload, so we don't know its (future) address. */
441 return SR_ERR;
442 snprintf(str, 128, "%d.%d", usb->bus, usb->address);
443 *data = g_variant_new_string(str);
444 break;
445 case SR_CONF_NUM_TIMEBASE:
446 *data = g_variant_new_int32(NUM_TIMEBASE);
447 break;
448 case SR_CONF_NUM_VDIV:
449 *data = g_variant_new_int32(NUM_VDIV);
450 break;
451 default:
452 return SR_ERR_NA;
453 }
454
455 return SR_OK;
456}
457
458static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi,
459 const struct sr_channel_group *cg)
460{
461 struct dev_context *devc;
462 double tmp_double;
463 uint64_t tmp_u64, p, q;
464 int tmp_int, ret;
465 unsigned int i;
466 const char *tmp_str;
467 char **targets;
468
469 (void)cg;
470
471 if (sdi->status != SR_ST_ACTIVE)
472 return SR_ERR_DEV_CLOSED;
473
474 ret = SR_OK;
475 devc = sdi->priv;
476 switch (id) {
477 case SR_CONF_LIMIT_FRAMES:
478 devc->limit_frames = g_variant_get_uint64(data);
479 break;
480 case SR_CONF_TRIGGER_SLOPE:
481 tmp_str = g_variant_get_string(data, NULL);
482 if (!tmp_str || !(tmp_str[0] == 'f' || tmp_str[0] == 'r'))
483 return SR_ERR_ARG;
484 devc->triggerslope = (tmp_str[0] == 'r')
485 ? SLOPE_POSITIVE : SLOPE_NEGATIVE;
486 break;
487 case SR_CONF_HORIZ_TRIGGERPOS:
488 tmp_double = g_variant_get_double(data);
489 if (tmp_double < 0.0 || tmp_double > 1.0) {
490 sr_err("Trigger position should be between 0.0 and 1.0.");
491 ret = SR_ERR_ARG;
492 } else
493 devc->triggerposition = tmp_double;
494 break;
495 case SR_CONF_BUFFERSIZE:
496 tmp_u64 = g_variant_get_uint64(data);
497 for (i = 0; i < 2; i++) {
498 if (devc->profile->buffersizes[i] == tmp_u64) {
499 devc->framesize = tmp_u64;
500 break;
501 }
502 }
503 if (i == 2)
504 ret = SR_ERR_ARG;
505 break;
506 case SR_CONF_TIMEBASE:
507 g_variant_get(data, "(tt)", &p, &q);
508 tmp_int = -1;
509 for (i = 0; i < ARRAY_SIZE(timebases); i++) {
510 if (timebases[i][0] == p && timebases[i][1] == q) {
511 tmp_int = i;
512 break;
513 }
514 }
515 if (tmp_int >= 0)
516 devc->timebase = tmp_int;
517 else
518 ret = SR_ERR_ARG;
519 break;
520 case SR_CONF_TRIGGER_SOURCE:
521 tmp_str = g_variant_get_string(data, NULL);
522 for (i = 0; trigger_sources[i]; i++) {
523 if (!strcmp(tmp_str, trigger_sources[i])) {
524 devc->triggersource = g_strdup(tmp_str);
525 break;
526 }
527 }
528 if (trigger_sources[i] == 0)
529 ret = SR_ERR_ARG;
530 break;
531 case SR_CONF_FILTER:
532 tmp_str = g_variant_get_string(data, NULL);
533 devc->filter_ch1 = devc->filter_ch2 = devc->filter_trigger = 0;
534 targets = g_strsplit(tmp_str, ",", 0);
535 for (i = 0; targets[i]; i++) {
536 if (targets[i] == '\0')
537 /* Empty filter string can be used to clear them all. */
538 ;
539 else if (!strcmp(targets[i], "CH1"))
540 devc->filter_ch1 = TRUE;
541 else if (!strcmp(targets[i], "CH2"))
542 devc->filter_ch2 = TRUE;
543 else if (!strcmp(targets[i], "TRIGGER"))
544 devc->filter_trigger = TRUE;
545 else {
546 sr_err("Invalid filter target %s.", targets[i]);
547 ret = SR_ERR_ARG;
548 }
549 }
550 g_strfreev(targets);
551 break;
552 case SR_CONF_VDIV:
553 /* TODO: Not supporting vdiv per channel yet. */
554 g_variant_get(data, "(tt)", &p, &q);
555 tmp_int = -1;
556 for (i = 0; i < ARRAY_SIZE(vdivs); i++) {
557 if (vdivs[i][0] == p && vdivs[i][1] == q) {
558 tmp_int = i;
559 break;
560 }
561 }
562 if (tmp_int >= 0) {
563 devc->voltage_ch1 = tmp_int;
564 devc->voltage_ch2 = tmp_int;
565 } else
566 ret = SR_ERR_ARG;
567 break;
568 case SR_CONF_COUPLING:
569 tmp_str = g_variant_get_string(data, NULL);
570 /* TODO: Not supporting coupling per channel yet. */
571 for (i = 0; coupling[i]; i++) {
572 if (!strcmp(tmp_str, coupling[i])) {
573 devc->coupling_ch1 = i;
574 devc->coupling_ch2 = i;
575 break;
576 }
577 }
578 if (coupling[i] == 0)
579 ret = SR_ERR_ARG;
580 break;
581 default:
582 ret = SR_ERR_NA;
583 break;
584 }
585
586 return ret;
587}
588
589static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi,
590 const struct sr_channel_group *cg)
591{
592 struct dev_context *devc;
593 GVariant *tuple, *rational[2];
594 GVariantBuilder gvb;
595 unsigned int i;
596
597 (void)cg;
598
599 switch (key) {
600 case SR_CONF_SCAN_OPTIONS:
601 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
602 scanopts, ARRAY_SIZE(scanopts), sizeof(int32_t));
603 break;
604 case SR_CONF_DEVICE_OPTIONS:
605 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
606 devopts, ARRAY_SIZE(devopts), sizeof(int32_t));
607 break;
608 case SR_CONF_BUFFERSIZE:
609 if (!sdi)
610 return SR_ERR_ARG;
611 devc = sdi->priv;
612 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT64,
613 devc->profile->buffersizes, 2, sizeof(uint64_t));
614 break;
615 case SR_CONF_COUPLING:
616 *data = g_variant_new_strv(coupling, ARRAY_SIZE(coupling));
617 break;
618 case SR_CONF_VDIV:
619 g_variant_builder_init(&gvb, G_VARIANT_TYPE_ARRAY);
620 for (i = 0; i < ARRAY_SIZE(vdivs); i++) {
621 rational[0] = g_variant_new_uint64(vdivs[i][0]);
622 rational[1] = g_variant_new_uint64(vdivs[i][1]);
623 tuple = g_variant_new_tuple(rational, 2);
624 g_variant_builder_add_value(&gvb, tuple);
625 }
626 *data = g_variant_builder_end(&gvb);
627 break;
628 case SR_CONF_FILTER:
629 *data = g_variant_new_strv(filter_targets,
630 ARRAY_SIZE(filter_targets));
631 break;
632 case SR_CONF_TIMEBASE:
633 g_variant_builder_init(&gvb, G_VARIANT_TYPE_ARRAY);
634 for (i = 0; i < ARRAY_SIZE(timebases); i++) {
635 rational[0] = g_variant_new_uint64(timebases[i][0]);
636 rational[1] = g_variant_new_uint64(timebases[i][1]);
637 tuple = g_variant_new_tuple(rational, 2);
638 g_variant_builder_add_value(&gvb, tuple);
639 }
640 *data = g_variant_builder_end(&gvb);
641 break;
642 case SR_CONF_TRIGGER_SOURCE:
643 *data = g_variant_new_strv(trigger_sources,
644 ARRAY_SIZE(trigger_sources));
645 break;
646 default:
647 return SR_ERR_NA;
648 }
649
650 return SR_OK;
651}
652
653static void send_chunk(struct sr_dev_inst *sdi, unsigned char *buf,
654 int num_samples)
655{
656 struct sr_datafeed_packet packet;
657 struct sr_datafeed_analog analog;
658 struct dev_context *devc;
659 float ch1, ch2, range;
660 int num_channels, data_offset, i;
661
662 devc = sdi->priv;
663 num_channels = (devc->ch1_enabled && devc->ch2_enabled) ? 2 : 1;
664 packet.type = SR_DF_ANALOG;
665 packet.payload = &analog;
666 /* TODO: support for 5xxx series 9-bit samples */
667 analog.channels = devc->enabled_channels;
668 analog.num_samples = num_samples;
669 analog.mq = SR_MQ_VOLTAGE;
670 analog.unit = SR_UNIT_VOLT;
671 /* TODO: Check malloc return value. */
672 analog.data = g_try_malloc(analog.num_samples * sizeof(float) * num_channels);
673 data_offset = 0;
674 for (i = 0; i < analog.num_samples; i++) {
675 /*
676 * The device always sends data for both channels. If a channel
677 * is disabled, it contains a copy of the enabled channel's
678 * data. However, we only send the requested channels to
679 * the bus.
680 *
681 * Voltage values are encoded as a value 0-255 (0-512 on the
682 * DSO-5200*), where the value is a point in the range
683 * represented by the vdiv setting. There are 8 vertical divs,
684 * so e.g. 500mV/div represents 4V peak-to-peak where 0 = -2V
685 * and 255 = +2V.
686 */
687 /* TODO: Support for DSO-5xxx series 9-bit samples. */
688 if (devc->ch1_enabled) {
689 range = ((float)vdivs[devc->voltage_ch1][0] / vdivs[devc->voltage_ch1][1]) * 8;
690 ch1 = range / 255 * *(buf + i * 2 + 1);
691 /* Value is centered around 0V. */
692 ch1 -= range / 2;
693 analog.data[data_offset++] = ch1;
694 }
695 if (devc->ch2_enabled) {
696 range = ((float)vdivs[devc->voltage_ch2][0] / vdivs[devc->voltage_ch2][1]) * 8;
697 ch2 = range / 255 * *(buf + i * 2);
698 ch2 -= range / 2;
699 analog.data[data_offset++] = ch2;
700 }
701 }
702 sr_session_send(devc->cb_data, &packet);
703}
704
705/*
706 * Called by libusb (as triggered by handle_event()) when a transfer comes in.
707 * Only channel data comes in asynchronously, and all transfers for this are
708 * queued up beforehand, so this just needs to chuck the incoming data onto
709 * the libsigrok session bus.
710 */
711static void receive_transfer(struct libusb_transfer *transfer)
712{
713 struct sr_datafeed_packet packet;
714 struct sr_dev_inst *sdi;
715 struct dev_context *devc;
716 int num_samples, pre;
717
718 sdi = transfer->user_data;
719 devc = sdi->priv;
720 sr_spew("receive_transfer(): status %d received %d bytes.",
721 transfer->status, transfer->actual_length);
722
723 if (transfer->actual_length == 0)
724 /* Nothing to send to the bus. */
725 return;
726
727 num_samples = transfer->actual_length / 2;
728
729 sr_spew("Got %d-%d/%d samples in frame.", devc->samp_received + 1,
730 devc->samp_received + num_samples, devc->framesize);
731
732 /*
733 * The device always sends a full frame, but the beginning of the frame
734 * doesn't represent the trigger point. The offset at which the trigger
735 * happened came in with the capture state, so we need to start sending
736 * from there up the session bus. The samples in the frame buffer
737 * before that trigger point came after the end of the device's frame
738 * buffer was reached, and it wrapped around to overwrite up until the
739 * trigger point.
740 */
741 if (devc->samp_received < devc->trigger_offset) {
742 /* Trigger point not yet reached. */
743 if (devc->samp_received + num_samples < devc->trigger_offset) {
744 /* The entire chunk is before the trigger point. */
745 memcpy(devc->framebuf + devc->samp_buffered * 2,
746 transfer->buffer, num_samples * 2);
747 devc->samp_buffered += num_samples;
748 } else {
749 /*
750 * This chunk hits or overruns the trigger point.
751 * Store the part before the trigger fired, and
752 * send the rest up to the session bus.
753 */
754 pre = devc->trigger_offset - devc->samp_received;
755 memcpy(devc->framebuf + devc->samp_buffered * 2,
756 transfer->buffer, pre * 2);
757 devc->samp_buffered += pre;
758
759 /* The rest of this chunk starts with the trigger point. */
760 sr_dbg("Reached trigger point, %d samples buffered.",
761 devc->samp_buffered);
762
763 /* Avoid the corner case where the chunk ended at
764 * exactly the trigger point. */
765 if (num_samples > pre)
766 send_chunk(sdi, transfer->buffer + pre * 2,
767 num_samples - pre);
768 }
769 } else {
770 /* Already past the trigger point, just send it all out. */
771 send_chunk(sdi, transfer->buffer,
772 num_samples);
773 }
774
775 devc->samp_received += num_samples;
776
777 /* Everything in this transfer was either copied to the buffer or
778 * sent to the session bus. */
779 g_free(transfer->buffer);
780 libusb_free_transfer(transfer);
781
782 if (devc->samp_received >= devc->framesize) {
783 /* That was the last chunk in this frame. Send the buffered
784 * pre-trigger samples out now, in one big chunk. */
785 sr_dbg("End of frame, sending %d pre-trigger buffered samples.",
786 devc->samp_buffered);
787 send_chunk(sdi, devc->framebuf, devc->samp_buffered);
788
789 /* Mark the end of this frame. */
790 packet.type = SR_DF_FRAME_END;
791 sr_session_send(devc->cb_data, &packet);
792
793 if (devc->limit_frames && ++devc->num_frames == devc->limit_frames) {
794 /* Terminate session */
795 devc->dev_state = STOPPING;
796 } else {
797 devc->dev_state = NEW_CAPTURE;
798 }
799 }
800}
801
802static int handle_event(int fd, int revents, void *cb_data)
803{
804 const struct sr_dev_inst *sdi;
805 struct sr_datafeed_packet packet;
806 struct timeval tv;
807 struct dev_context *devc;
808 struct drv_context *drvc = di->priv;
809 int num_channels;
810 uint32_t trigger_offset;
811 uint8_t capturestate;
812
813 (void)fd;
814 (void)revents;
815
816 sdi = cb_data;
817 devc = sdi->priv;
818 if (devc->dev_state == STOPPING) {
819 /* We've been told to wind up the acquisition. */
820 sr_dbg("Stopping acquisition.");
821 /*
822 * TODO: Doesn't really cancel pending transfers so they might
823 * come in after SR_DF_END is sent.
824 */
825 usb_source_remove(drvc->sr_ctx);
826
827 packet.type = SR_DF_END;
828 sr_session_send(sdi, &packet);
829
830 devc->dev_state = IDLE;
831
832 return TRUE;
833 }
834
835 /* Always handle pending libusb events. */
836 tv.tv_sec = tv.tv_usec = 0;
837 libusb_handle_events_timeout(drvc->sr_ctx->libusb_ctx, &tv);
838
839 /* TODO: ugh */
840 if (devc->dev_state == NEW_CAPTURE) {
841 if (dso_capture_start(sdi) != SR_OK)
842 return TRUE;
843 if (dso_enable_trigger(sdi) != SR_OK)
844 return TRUE;
845// if (dso_force_trigger(sdi) != SR_OK)
846// return TRUE;
847 sr_dbg("Successfully requested next chunk.");
848 devc->dev_state = CAPTURE;
849 return TRUE;
850 }
851 if (devc->dev_state != CAPTURE)
852 return TRUE;
853
854 if ((dso_get_capturestate(sdi, &capturestate, &trigger_offset)) != SR_OK)
855 return TRUE;
856
857 sr_dbg("Capturestate %d.", capturestate);
858 sr_dbg("Trigger offset 0x%.6x.", trigger_offset);
859 switch (capturestate) {
860 case CAPTURE_EMPTY:
861 if (++devc->capture_empty_count >= MAX_CAPTURE_EMPTY) {
862 devc->capture_empty_count = 0;
863 if (dso_capture_start(sdi) != SR_OK)
864 break;
865 if (dso_enable_trigger(sdi) != SR_OK)
866 break;
867// if (dso_force_trigger(sdi) != SR_OK)
868// break;
869 sr_dbg("Successfully requested next chunk.");
870 }
871 break;
872 case CAPTURE_FILLING:
873 /* No data yet. */
874 break;
875 case CAPTURE_READY_8BIT:
876 /* Remember where in the captured frame the trigger is. */
877 devc->trigger_offset = trigger_offset;
878
879 num_channels = (devc->ch1_enabled && devc->ch2_enabled) ? 2 : 1;
880 /* TODO: Check malloc return value. */
881 devc->framebuf = g_try_malloc(devc->framesize * num_channels * 2);
882 devc->samp_buffered = devc->samp_received = 0;
883
884 /* Tell the scope to send us the first frame. */
885 if (dso_get_channeldata(sdi, receive_transfer) != SR_OK)
886 break;
887
888 /*
889 * Don't hit the state machine again until we're done fetching
890 * the data we just told the scope to send.
891 */
892 devc->dev_state = FETCH_DATA;
893
894 /* Tell the frontend a new frame is on the way. */
895 packet.type = SR_DF_FRAME_BEGIN;
896 sr_session_send(sdi, &packet);
897 break;
898 case CAPTURE_READY_9BIT:
899 /* TODO */
900 sr_err("Not yet supported.");
901 break;
902 case CAPTURE_TIMEOUT:
903 /* Doesn't matter, we'll try again next time. */
904 break;
905 default:
906 sr_dbg("Unknown capture state: %d.", capturestate);
907 break;
908 }
909
910 return TRUE;
911}
912
913static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
914{
915 struct dev_context *devc;
916 struct drv_context *drvc = di->priv;
917
918 if (sdi->status != SR_ST_ACTIVE)
919 return SR_ERR_DEV_CLOSED;
920
921 devc = sdi->priv;
922 devc->cb_data = cb_data;
923
924 if (configure_channels(sdi) != SR_OK) {
925 sr_err("Failed to configure channels.");
926 return SR_ERR;
927 }
928
929 if (dso_init(sdi) != SR_OK)
930 return SR_ERR;
931
932 if (dso_capture_start(sdi) != SR_OK)
933 return SR_ERR;
934
935 devc->dev_state = CAPTURE;
936 usb_source_add(drvc->sr_ctx, TICK, handle_event, (void *)sdi);
937
938 /* Send header packet to the session bus. */
939 std_session_send_df_header(cb_data, LOG_PREFIX);
940
941 return SR_OK;
942}
943
944static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
945{
946 struct dev_context *devc;
947
948 (void)cb_data;
949
950 if (sdi->status != SR_ST_ACTIVE)
951 return SR_ERR;
952
953 devc = sdi->priv;
954 devc->dev_state = STOPPING;
955
956 return SR_OK;
957}
958
959SR_PRIV struct sr_dev_driver hantek_dso_driver_info = {
960 .name = "hantek-dso",
961 .longname = "Hantek DSO",
962 .api_version = 1,
963 .init = init,
964 .cleanup = cleanup,
965 .scan = scan,
966 .dev_list = dev_list,
967 .dev_clear = dev_clear,
968 .config_get = config_get,
969 .config_set = config_set,
970 .config_list = config_list,
971 .dev_open = dev_open,
972 .dev_close = dev_close,
973 .dev_acquisition_start = dev_acquisition_start,
974 .dev_acquisition_stop = dev_acquisition_stop,
975 .priv = NULL,
976};