]> sigrok.org Git - libsigrok.git/blob - hardware/asix-sigma/asix-sigma.c
088095acb96d9177d2d24ff0d3996ed6c93df382
[libsigrok.git] / hardware / asix-sigma / asix-sigma.c
1 /*
2  * This file is part of the sigrok project.
3  *
4  * Copyright (C) 2010-2012 Håvard Espeland <gus@ping.uio.no>,
5  * Copyright (C) 2010 Martin Stensgård <mastensg@ping.uio.no>
6  * Copyright (C) 2010 Carl Henrik Lunde <chlunde@ping.uio.no>
7  *
8  * This program is free software: you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation, either version 3 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
20  */
21
22 /*
23  * ASIX SIGMA/SIGMA2 logic analyzer driver
24  */
25
26 #include <glib.h>
27 #include <glib/gstdio.h>
28 #include <ftdi.h>
29 #include <string.h>
30 #include "libsigrok.h"
31 #include "libsigrok-internal.h"
32 #include "asix-sigma.h"
33
34 #define USB_VENDOR                      0xa600
35 #define USB_PRODUCT                     0xa000
36 #define USB_DESCRIPTION                 "ASIX SIGMA"
37 #define USB_VENDOR_NAME                 "ASIX"
38 #define USB_MODEL_NAME                  "SIGMA"
39 #define USB_MODEL_VERSION               ""
40 #define TRIGGER_TYPES                   "rf10"
41 #define NUM_PROBES                      16
42
43 SR_PRIV struct sr_dev_driver asix_sigma_driver_info;
44 static struct sr_dev_driver *adi = &asix_sigma_driver_info;
45
46 static const uint64_t supported_samplerates[] = {
47         SR_KHZ(200),
48         SR_KHZ(250),
49         SR_KHZ(500),
50         SR_MHZ(1),
51         SR_MHZ(5),
52         SR_MHZ(10),
53         SR_MHZ(25),
54         SR_MHZ(50),
55         SR_MHZ(100),
56         SR_MHZ(200),
57         0,
58 };
59
60 /*
61  * Probe numbers seem to go from 1-16, according to this image:
62  * http://tools.asix.net/img/sigma_sigmacab_pins_720.jpg
63  * (the cable has two additional GND pins, and a TI and TO pin)
64  */
65 static const char *probe_names[NUM_PROBES + 1] = {
66         "1",
67         "2",
68         "3",
69         "4",
70         "5",
71         "6",
72         "7",
73         "8",
74         "9",
75         "10",
76         "11",
77         "12",
78         "13",
79         "14",
80         "15",
81         "16",
82         NULL,
83 };
84
85 static const struct sr_samplerates samplerates = {
86         0,
87         0,
88         0,
89         supported_samplerates,
90 };
91
92 static const int hwcaps[] = {
93         SR_HWCAP_LOGIC_ANALYZER,
94         SR_HWCAP_SAMPLERATE,
95         SR_HWCAP_CAPTURE_RATIO,
96         SR_HWCAP_PROBECONFIG,
97
98         SR_HWCAP_LIMIT_MSEC,
99         0,
100 };
101
102 /* Force the FPGA to reboot. */
103 static uint8_t suicide[] = {
104         0x84, 0x84, 0x88, 0x84, 0x88, 0x84, 0x88, 0x84,
105 };
106
107 /* Prepare to upload firmware (FPGA specific). */
108 static uint8_t init[] = {
109         0x03, 0x03, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01,
110 };
111
112 /* Initialize the logic analyzer mode. */
113 static uint8_t logic_mode_start[] = {
114         0x00, 0x40, 0x0f, 0x25, 0x35, 0x40,
115         0x2a, 0x3a, 0x40, 0x03, 0x20, 0x38,
116 };
117
118 static const char *firmware_files[] = {
119         "asix-sigma-50.fw",     /* 50 MHz, supports 8 bit fractions */
120         "asix-sigma-100.fw",    /* 100 MHz */
121         "asix-sigma-200.fw",    /* 200 MHz */
122         "asix-sigma-50sync.fw", /* Synchronous clock from pin */
123         "asix-sigma-phasor.fw", /* Frequency counter */
124 };
125
126 static int hw_dev_acquisition_stop(int dev_index, void *cb_data);
127
128 static int sigma_read(void *buf, size_t size, struct context *ctx)
129 {
130         int ret;
131
132         ret = ftdi_read_data(&ctx->ftdic, (unsigned char *)buf, size);
133         if (ret < 0) {
134                 sr_err("sigma: ftdi_read_data failed: %s",
135                        ftdi_get_error_string(&ctx->ftdic));
136         }
137
138         return ret;
139 }
140
141 static int sigma_write(void *buf, size_t size, struct context *ctx)
142 {
143         int ret;
144
145         ret = ftdi_write_data(&ctx->ftdic, (unsigned char *)buf, size);
146         if (ret < 0) {
147                 sr_err("sigma: ftdi_write_data failed: %s",
148                        ftdi_get_error_string(&ctx->ftdic));
149         } else if ((size_t) ret != size) {
150                 sr_err("sigma: ftdi_write_data did not complete write.");
151         }
152
153         return ret;
154 }
155
156 static int sigma_write_register(uint8_t reg, uint8_t *data, size_t len,
157                                 struct context *ctx)
158 {
159         size_t i;
160         uint8_t buf[len + 2];
161         int idx = 0;
162
163         buf[idx++] = REG_ADDR_LOW | (reg & 0xf);
164         buf[idx++] = REG_ADDR_HIGH | (reg >> 4);
165
166         for (i = 0; i < len; ++i) {
167                 buf[idx++] = REG_DATA_LOW | (data[i] & 0xf);
168                 buf[idx++] = REG_DATA_HIGH_WRITE | (data[i] >> 4);
169         }
170
171         return sigma_write(buf, idx, ctx);
172 }
173
174 static int sigma_set_register(uint8_t reg, uint8_t value, struct context *ctx)
175 {
176         return sigma_write_register(reg, &value, 1, ctx);
177 }
178
179 static int sigma_read_register(uint8_t reg, uint8_t *data, size_t len,
180                                struct context *ctx)
181 {
182         uint8_t buf[3];
183
184         buf[0] = REG_ADDR_LOW | (reg & 0xf);
185         buf[1] = REG_ADDR_HIGH | (reg >> 4);
186         buf[2] = REG_READ_ADDR;
187
188         sigma_write(buf, sizeof(buf), ctx);
189
190         return sigma_read(data, len, ctx);
191 }
192
193 static uint8_t sigma_get_register(uint8_t reg, struct context *ctx)
194 {
195         uint8_t value;
196
197         if (1 != sigma_read_register(reg, &value, 1, ctx)) {
198                 sr_err("sigma: sigma_get_register: 1 byte expected");
199                 return 0;
200         }
201
202         return value;
203 }
204
205 static int sigma_read_pos(uint32_t *stoppos, uint32_t *triggerpos,
206                           struct context *ctx)
207 {
208         uint8_t buf[] = {
209                 REG_ADDR_LOW | READ_TRIGGER_POS_LOW,
210
211                 REG_READ_ADDR | NEXT_REG,
212                 REG_READ_ADDR | NEXT_REG,
213                 REG_READ_ADDR | NEXT_REG,
214                 REG_READ_ADDR | NEXT_REG,
215                 REG_READ_ADDR | NEXT_REG,
216                 REG_READ_ADDR | NEXT_REG,
217         };
218         uint8_t result[6];
219
220         sigma_write(buf, sizeof(buf), ctx);
221
222         sigma_read(result, sizeof(result), ctx);
223
224         *triggerpos = result[0] | (result[1] << 8) | (result[2] << 16);
225         *stoppos = result[3] | (result[4] << 8) | (result[5] << 16);
226
227         /* Not really sure why this must be done, but according to spec. */
228         if ((--*stoppos & 0x1ff) == 0x1ff)
229                 stoppos -= 64;
230
231         if ((*--triggerpos & 0x1ff) == 0x1ff)
232                 triggerpos -= 64;
233
234         return 1;
235 }
236
237 static int sigma_read_dram(uint16_t startchunk, size_t numchunks,
238                            uint8_t *data, struct context *ctx)
239 {
240         size_t i;
241         uint8_t buf[4096];
242         int idx = 0;
243
244         /* Send the startchunk. Index start with 1. */
245         buf[0] = startchunk >> 8;
246         buf[1] = startchunk & 0xff;
247         sigma_write_register(WRITE_MEMROW, buf, 2, ctx);
248
249         /* Read the DRAM. */
250         buf[idx++] = REG_DRAM_BLOCK;
251         buf[idx++] = REG_DRAM_WAIT_ACK;
252
253         for (i = 0; i < numchunks; ++i) {
254                 /* Alternate bit to copy from DRAM to cache. */
255                 if (i != (numchunks - 1))
256                         buf[idx++] = REG_DRAM_BLOCK | (((i + 1) % 2) << 4);
257
258                 buf[idx++] = REG_DRAM_BLOCK_DATA | ((i % 2) << 4);
259
260                 if (i != (numchunks - 1))
261                         buf[idx++] = REG_DRAM_WAIT_ACK;
262         }
263
264         sigma_write(buf, idx, ctx);
265
266         return sigma_read(data, numchunks * CHUNK_SIZE, ctx);
267 }
268
269 /* Upload trigger look-up tables to Sigma. */
270 static int sigma_write_trigger_lut(struct triggerlut *lut, struct context *ctx)
271 {
272         int i;
273         uint8_t tmp[2];
274         uint16_t bit;
275
276         /* Transpose the table and send to Sigma. */
277         for (i = 0; i < 16; ++i) {
278                 bit = 1 << i;
279
280                 tmp[0] = tmp[1] = 0;
281
282                 if (lut->m2d[0] & bit)
283                         tmp[0] |= 0x01;
284                 if (lut->m2d[1] & bit)
285                         tmp[0] |= 0x02;
286                 if (lut->m2d[2] & bit)
287                         tmp[0] |= 0x04;
288                 if (lut->m2d[3] & bit)
289                         tmp[0] |= 0x08;
290
291                 if (lut->m3 & bit)
292                         tmp[0] |= 0x10;
293                 if (lut->m3s & bit)
294                         tmp[0] |= 0x20;
295                 if (lut->m4 & bit)
296                         tmp[0] |= 0x40;
297
298                 if (lut->m0d[0] & bit)
299                         tmp[1] |= 0x01;
300                 if (lut->m0d[1] & bit)
301                         tmp[1] |= 0x02;
302                 if (lut->m0d[2] & bit)
303                         tmp[1] |= 0x04;
304                 if (lut->m0d[3] & bit)
305                         tmp[1] |= 0x08;
306
307                 if (lut->m1d[0] & bit)
308                         tmp[1] |= 0x10;
309                 if (lut->m1d[1] & bit)
310                         tmp[1] |= 0x20;
311                 if (lut->m1d[2] & bit)
312                         tmp[1] |= 0x40;
313                 if (lut->m1d[3] & bit)
314                         tmp[1] |= 0x80;
315
316                 sigma_write_register(WRITE_TRIGGER_SELECT0, tmp, sizeof(tmp),
317                                      ctx);
318                 sigma_set_register(WRITE_TRIGGER_SELECT1, 0x30 | i, ctx);
319         }
320
321         /* Send the parameters */
322         sigma_write_register(WRITE_TRIGGER_SELECT0, (uint8_t *) &lut->params,
323                              sizeof(lut->params), ctx);
324
325         return SR_OK;
326 }
327
328 /* Generate the bitbang stream for programming the FPGA. */
329 static int bin2bitbang(const char *filename,
330                        unsigned char **buf, size_t *buf_size)
331 {
332         FILE *f;
333         unsigned long file_size;
334         unsigned long offset = 0;
335         unsigned char *p;
336         uint8_t *firmware;
337         unsigned long fwsize = 0;
338         const int buffer_size = 65536;
339         size_t i;
340         int c, bit, v;
341         uint32_t imm = 0x3f6df2ab;
342
343         f = g_fopen(filename, "rb");
344         if (!f) {
345                 sr_err("sigma: g_fopen(\"%s\", \"rb\")", filename);
346                 return SR_ERR;
347         }
348
349         if (-1 == fseek(f, 0, SEEK_END)) {
350                 sr_err("sigma: fseek on %s failed", filename);
351                 fclose(f);
352                 return SR_ERR;
353         }
354
355         file_size = ftell(f);
356
357         fseek(f, 0, SEEK_SET);
358
359         if (!(firmware = g_try_malloc(buffer_size))) {
360                 sr_err("sigma: %s: firmware malloc failed", __func__);
361                 fclose(f);
362                 return SR_ERR_MALLOC;
363         }
364
365         while ((c = getc(f)) != EOF) {
366                 imm = (imm + 0xa853753) % 177 + (imm * 0x8034052);
367                 firmware[fwsize++] = c ^ imm;
368         }
369         fclose(f);
370
371         if(fwsize != file_size) {
372             sr_err("sigma: %s: Error reading firmware", filename);
373             fclose(f);
374             g_free(firmware);
375             return SR_ERR;
376         }
377
378         *buf_size = fwsize * 2 * 8;
379
380         *buf = p = (unsigned char *)g_try_malloc(*buf_size);
381         if (!p) {
382                 sr_err("sigma: %s: buf/p malloc failed", __func__);
383                 g_free(firmware);
384                 return SR_ERR_MALLOC;
385         }
386
387         for (i = 0; i < fwsize; ++i) {
388                 for (bit = 7; bit >= 0; --bit) {
389                         v = firmware[i] & 1 << bit ? 0x40 : 0x00;
390                         p[offset++] = v | 0x01;
391                         p[offset++] = v;
392                 }
393         }
394
395         g_free(firmware);
396
397         if (offset != *buf_size) {
398                 g_free(*buf);
399                 sr_err("sigma: Error reading firmware %s "
400                        "offset=%ld, file_size=%ld, buf_size=%zd.",
401                        filename, offset, file_size, *buf_size);
402
403                 return SR_ERR;
404         }
405
406         return SR_OK;
407 }
408
409 static void clear_instances(void)
410 {
411         GSList *l;
412         struct sr_dev_inst *sdi;
413         struct context *ctx;
414
415         /* Properly close all devices. */
416         for (l = adi->instances; l; l = l->next) {
417                 if (!(sdi = l->data)) {
418                         /* Log error, but continue cleaning up the rest. */
419                         sr_err("sigma: %s: sdi was NULL, continuing", __func__);
420                         continue;
421                 }
422                 if (sdi->priv) {
423                         ctx = sdi->priv;
424                         ftdi_free(&ctx->ftdic);
425                         g_free(ctx);
426                 }
427                 sr_dev_inst_free(sdi);
428         }
429         g_slist_free(adi->instances);
430         adi->instances = NULL;
431
432 }
433
434 static int hw_init(void)
435 {
436
437         /* Nothing to do. */
438
439         return SR_OK;
440 }
441
442 static GSList *hw_scan(GSList *options)
443 {
444         struct sr_dev_inst *sdi;
445         struct context *ctx;
446         GSList *devices;
447         struct ftdi_device_list *devlist;
448         char serial_txt[10];
449         uint32_t serial;
450         int ret;
451
452         (void)options;
453         devices = NULL;
454         clear_instances();
455
456         if (!(ctx = g_try_malloc(sizeof(struct context)))) {
457                 sr_err("sigma: %s: ctx malloc failed", __func__);
458                 return NULL;
459         }
460
461         ftdi_init(&ctx->ftdic);
462
463         /* Look for SIGMAs. */
464
465         if ((ret = ftdi_usb_find_all(&ctx->ftdic, &devlist,
466             USB_VENDOR, USB_PRODUCT)) <= 0) {
467                 if (ret < 0)
468                         sr_err("ftdi_usb_find_all(): %d", ret);
469                 goto free;
470         }
471
472         /* Make sure it's a version 1 or 2 SIGMA. */
473         ftdi_usb_get_strings(&ctx->ftdic, devlist->dev, NULL, 0, NULL, 0,
474                              serial_txt, sizeof(serial_txt));
475         sscanf(serial_txt, "%x", &serial);
476
477         if (serial < 0xa6010000 || serial > 0xa602ffff) {
478                 sr_err("sigma: Only SIGMA and SIGMA2 are supported "
479                        "in this version of sigrok.");
480                 goto free;
481         }
482
483         sr_info("Found ASIX SIGMA - Serial: %s", serial_txt);
484
485         ctx->cur_samplerate = 0;
486         ctx->period_ps = 0;
487         ctx->limit_msec = 0;
488         ctx->cur_firmware = -1;
489         ctx->num_probes = 0;
490         ctx->samples_per_event = 0;
491         ctx->capture_ratio = 50;
492         ctx->use_triggers = 0;
493
494         /* Register SIGMA device. */
495         if (!(sdi = sr_dev_inst_new(0, SR_ST_INITIALIZING, USB_VENDOR_NAME,
496                                     USB_MODEL_NAME, USB_MODEL_VERSION))) {
497                 sr_err("sigma: %s: sdi was NULL", __func__);
498                 goto free;
499         }
500         sdi->driver = adi;
501         devices = g_slist_append(devices, sdi);
502         adi->instances = g_slist_append(adi->instances, sdi);
503         sdi->priv = ctx;
504
505         /* We will open the device again when we need it. */
506         ftdi_list_free(&devlist);
507
508         return devices;
509
510 free:
511         ftdi_deinit(&ctx->ftdic);
512         g_free(ctx);
513         return NULL;
514 }
515
516 static int upload_firmware(int firmware_idx, struct context *ctx)
517 {
518         int ret;
519         unsigned char *buf;
520         unsigned char pins;
521         size_t buf_size;
522         unsigned char result[32];
523         char firmware_path[128];
524
525         /* Make sure it's an ASIX SIGMA. */
526         if ((ret = ftdi_usb_open_desc(&ctx->ftdic,
527                 USB_VENDOR, USB_PRODUCT, USB_DESCRIPTION, NULL)) < 0) {
528                 sr_err("sigma: ftdi_usb_open failed: %s",
529                        ftdi_get_error_string(&ctx->ftdic));
530                 return 0;
531         }
532
533         if ((ret = ftdi_set_bitmode(&ctx->ftdic, 0xdf, BITMODE_BITBANG)) < 0) {
534                 sr_err("sigma: ftdi_set_bitmode failed: %s",
535                        ftdi_get_error_string(&ctx->ftdic));
536                 return 0;
537         }
538
539         /* Four times the speed of sigmalogan - Works well. */
540         if ((ret = ftdi_set_baudrate(&ctx->ftdic, 750000)) < 0) {
541                 sr_err("sigma: ftdi_set_baudrate failed: %s",
542                        ftdi_get_error_string(&ctx->ftdic));
543                 return 0;
544         }
545
546         /* Force the FPGA to reboot. */
547         sigma_write(suicide, sizeof(suicide), ctx);
548         sigma_write(suicide, sizeof(suicide), ctx);
549         sigma_write(suicide, sizeof(suicide), ctx);
550         sigma_write(suicide, sizeof(suicide), ctx);
551
552         /* Prepare to upload firmware (FPGA specific). */
553         sigma_write(init, sizeof(init), ctx);
554
555         ftdi_usb_purge_buffers(&ctx->ftdic);
556
557         /* Wait until the FPGA asserts INIT_B. */
558         while (1) {
559                 ret = sigma_read(result, 1, ctx);
560                 if (result[0] & 0x20)
561                         break;
562         }
563
564         /* Prepare firmware. */
565         snprintf(firmware_path, sizeof(firmware_path), "%s/%s", FIRMWARE_DIR,
566                  firmware_files[firmware_idx]);
567
568         if ((ret = bin2bitbang(firmware_path, &buf, &buf_size)) != SR_OK) {
569                 sr_err("sigma: An error occured while reading the firmware: %s",
570                        firmware_path);
571                 return ret;
572         }
573
574         /* Upload firmare. */
575         sr_info("sigma: Uploading firmware %s", firmware_files[firmware_idx]);
576         sigma_write(buf, buf_size, ctx);
577
578         g_free(buf);
579
580         if ((ret = ftdi_set_bitmode(&ctx->ftdic, 0x00, BITMODE_RESET)) < 0) {
581                 sr_err("sigma: ftdi_set_bitmode failed: %s",
582                        ftdi_get_error_string(&ctx->ftdic));
583                 return SR_ERR;
584         }
585
586         ftdi_usb_purge_buffers(&ctx->ftdic);
587
588         /* Discard garbage. */
589         while (1 == sigma_read(&pins, 1, ctx))
590                 ;
591
592         /* Initialize the logic analyzer mode. */
593         sigma_write(logic_mode_start, sizeof(logic_mode_start), ctx);
594
595         /* Expect a 3 byte reply. */
596         ret = sigma_read(result, 3, ctx);
597         if (ret != 3 ||
598             result[0] != 0xa6 || result[1] != 0x55 || result[2] != 0xaa) {
599                 sr_err("sigma: Configuration failed. Invalid reply received.");
600                 return SR_ERR;
601         }
602
603         ctx->cur_firmware = firmware_idx;
604
605         sr_info("sigma: Firmware uploaded");
606
607         return SR_OK;
608 }
609
610 static int hw_dev_open(struct sr_dev_inst *sdi)
611 {
612         struct context *ctx;
613         int ret;
614
615         ctx = sdi->priv;
616
617         /* Make sure it's an ASIX SIGMA. */
618         if ((ret = ftdi_usb_open_desc(&ctx->ftdic,
619                 USB_VENDOR, USB_PRODUCT, USB_DESCRIPTION, NULL)) < 0) {
620
621                 sr_err("sigma: ftdi_usb_open failed: %s",
622                        ftdi_get_error_string(&ctx->ftdic));
623
624                 return 0;
625         }
626
627         sdi->status = SR_ST_ACTIVE;
628
629         return SR_OK;
630 }
631
632 static int set_samplerate(const struct sr_dev_inst *sdi, uint64_t samplerate)
633 {
634         int i, ret;
635         struct context *ctx = sdi->priv;
636
637         for (i = 0; supported_samplerates[i]; i++) {
638                 if (supported_samplerates[i] == samplerate)
639                         break;
640         }
641         if (supported_samplerates[i] == 0)
642                 return SR_ERR_SAMPLERATE;
643
644         if (samplerate <= SR_MHZ(50)) {
645                 ret = upload_firmware(0, ctx);
646                 ctx->num_probes = 16;
647         }
648         if (samplerate == SR_MHZ(100)) {
649                 ret = upload_firmware(1, ctx);
650                 ctx->num_probes = 8;
651         }
652         else if (samplerate == SR_MHZ(200)) {
653                 ret = upload_firmware(2, ctx);
654                 ctx->num_probes = 4;
655         }
656
657         ctx->cur_samplerate = samplerate;
658         ctx->period_ps = 1000000000000 / samplerate;
659         ctx->samples_per_event = 16 / ctx->num_probes;
660         ctx->state.state = SIGMA_IDLE;
661
662         return ret;
663 }
664
665 /*
666  * In 100 and 200 MHz mode, only a single pin rising/falling can be
667  * set as trigger. In other modes, two rising/falling triggers can be set,
668  * in addition to value/mask trigger for any number of probes.
669  *
670  * The Sigma supports complex triggers using boolean expressions, but this
671  * has not been implemented yet.
672  */
673 static int configure_probes(const struct sr_dev_inst *sdi, const GSList *probes)
674 {
675         struct context *ctx = sdi->priv;
676         const struct sr_probe *probe;
677         const GSList *l;
678         int trigger_set = 0;
679         int probebit;
680
681         memset(&ctx->trigger, 0, sizeof(struct sigma_trigger));
682
683         for (l = probes; l; l = l->next) {
684                 probe = (struct sr_probe *)l->data;
685                 probebit = 1 << (probe->index - 1);
686
687                 if (!probe->enabled || !probe->trigger)
688                         continue;
689
690                 if (ctx->cur_samplerate >= SR_MHZ(100)) {
691                         /* Fast trigger support. */
692                         if (trigger_set) {
693                                 sr_err("sigma: ASIX SIGMA only supports a single "
694                                        "pin trigger in 100 and 200MHz mode.");
695                                 return SR_ERR;
696                         }
697                         if (probe->trigger[0] == 'f')
698                                 ctx->trigger.fallingmask |= probebit;
699                         else if (probe->trigger[0] == 'r')
700                                 ctx->trigger.risingmask |= probebit;
701                         else {
702                                 sr_err("sigma: ASIX SIGMA only supports "
703                                        "rising/falling trigger in 100 "
704                                        "and 200MHz mode.");
705                                 return SR_ERR;
706                         }
707
708                         ++trigger_set;
709                 } else {
710                         /* Simple trigger support (event). */
711                         if (probe->trigger[0] == '1') {
712                                 ctx->trigger.simplevalue |= probebit;
713                                 ctx->trigger.simplemask |= probebit;
714                         }
715                         else if (probe->trigger[0] == '0') {
716                                 ctx->trigger.simplevalue &= ~probebit;
717                                 ctx->trigger.simplemask |= probebit;
718                         }
719                         else if (probe->trigger[0] == 'f') {
720                                 ctx->trigger.fallingmask |= probebit;
721                                 ++trigger_set;
722                         }
723                         else if (probe->trigger[0] == 'r') {
724                                 ctx->trigger.risingmask |= probebit;
725                                 ++trigger_set;
726                         }
727
728                         /*
729                          * Actually, Sigma supports 2 rising/falling triggers,
730                          * but they are ORed and the current trigger syntax
731                          * does not permit ORed triggers.
732                          */
733                         if (trigger_set > 1) {
734                                 sr_err("sigma: ASIX SIGMA only supports 1 "
735                                        "rising/falling triggers.");
736                                 return SR_ERR;
737                         }
738                 }
739
740                 if (trigger_set)
741                         ctx->use_triggers = 1;
742         }
743
744         return SR_OK;
745 }
746
747 static int hw_dev_close(struct sr_dev_inst *sdi)
748 {
749         struct context *ctx;
750
751         if (!(ctx = sdi->priv)) {
752                 sr_err("sigma: %s: sdi->priv was NULL", __func__);
753                 return SR_ERR_BUG;
754         }
755
756         /* TODO */
757         if (sdi->status == SR_ST_ACTIVE)
758                 ftdi_usb_close(&ctx->ftdic);
759
760         sdi->status = SR_ST_INACTIVE;
761
762         return SR_OK;
763 }
764
765 static int hw_cleanup(void)
766 {
767
768         clear_instances();
769
770         return SR_OK;
771 }
772
773 static int hw_info_get(int info_id, const void **data,
774        const struct sr_dev_inst *sdi)
775 {
776         struct context *ctx;
777
778         switch (info_id) {
779         case SR_DI_INST:
780                 *data = sdi;
781                 break;
782         case SR_DI_HWCAPS:
783                 *data = hwcaps;
784                 break;
785         case SR_DI_NUM_PROBES:
786                 *data = GINT_TO_POINTER(NUM_PROBES);
787                 break;
788         case SR_DI_PROBE_NAMES:
789                 *data = probe_names;
790                 break;
791         case SR_DI_SAMPLERATES:
792                 *data = &samplerates;
793                 break;
794         case SR_DI_TRIGGER_TYPES:
795                 *data = (char *)TRIGGER_TYPES;
796                 break;
797         case SR_DI_CUR_SAMPLERATE:
798                 if (sdi) {
799                         ctx = sdi->priv;
800                         *data = &ctx->cur_samplerate;
801                 } else
802                         return SR_ERR;
803                 break;
804         default:
805                 return SR_ERR_ARG;
806         }
807
808         return SR_OK;
809 }
810
811 static int hw_dev_status_get(int dev_index)
812 {
813         struct sr_dev_inst *sdi;
814
815         sdi = sr_dev_inst_get(adi->instances, dev_index);
816         if (sdi)
817                 return sdi->status;
818         else
819                 return SR_ST_NOT_FOUND;
820 }
821
822 static int hw_dev_config_set(const struct sr_dev_inst *sdi, int hwcap,
823                 const void *value)
824 {
825         struct context *ctx;
826         int ret;
827
828         ctx = sdi->priv;
829
830         if (hwcap == SR_HWCAP_SAMPLERATE) {
831                 ret = set_samplerate(sdi, *(const uint64_t *)value);
832         } else if (hwcap == SR_HWCAP_PROBECONFIG) {
833                 ret = configure_probes(sdi, value);
834         } else if (hwcap == SR_HWCAP_LIMIT_MSEC) {
835                 ctx->limit_msec = *(const uint64_t *)value;
836                 if (ctx->limit_msec > 0)
837                         ret = SR_OK;
838                 else
839                         ret = SR_ERR;
840         } else if (hwcap == SR_HWCAP_CAPTURE_RATIO) {
841                 ctx->capture_ratio = *(const uint64_t *)value;
842                 if (ctx->capture_ratio < 0 || ctx->capture_ratio > 100)
843                         ret = SR_ERR;
844                 else
845                         ret = SR_OK;
846         } else {
847                 ret = SR_ERR;
848         }
849
850         return ret;
851 }
852
853 /* Software trigger to determine exact trigger position. */
854 static int get_trigger_offset(uint16_t *samples, uint16_t last_sample,
855                               struct sigma_trigger *t)
856 {
857         int i;
858
859         for (i = 0; i < 8; ++i) {
860                 if (i > 0)
861                         last_sample = samples[i-1];
862
863                 /* Simple triggers. */
864                 if ((samples[i] & t->simplemask) != t->simplevalue)
865                         continue;
866
867                 /* Rising edge. */
868                 if ((last_sample & t->risingmask) != 0 || (samples[i] &
869                     t->risingmask) != t->risingmask)
870                         continue;
871
872                 /* Falling edge. */
873                 if ((last_sample & t->fallingmask) != t->fallingmask ||
874                     (samples[i] & t->fallingmask) != 0)
875                         continue;
876
877                 break;
878         }
879
880         /* If we did not match, return original trigger pos. */
881         return i & 0x7;
882 }
883
884 /*
885  * Decode chunk of 1024 bytes, 64 clusters, 7 events per cluster.
886  * Each event is 20ns apart, and can contain multiple samples.
887  *
888  * For 200 MHz, events contain 4 samples for each channel, spread 5 ns apart.
889  * For 100 MHz, events contain 2 samples for each channel, spread 10 ns apart.
890  * For 50 MHz and below, events contain one sample for each channel,
891  * spread 20 ns apart.
892  */
893 static int decode_chunk_ts(uint8_t *buf, uint16_t *lastts,
894                            uint16_t *lastsample, int triggerpos,
895                            uint16_t limit_chunk, void *cb_data)
896 {
897         struct sr_dev_inst *sdi = cb_data;
898         struct context *ctx = sdi->priv;
899         uint16_t tsdiff, ts;
900         uint16_t samples[65536 * ctx->samples_per_event];
901         struct sr_datafeed_packet packet;
902         struct sr_datafeed_logic logic;
903         int i, j, k, l, numpad, tosend;
904         size_t n = 0, sent = 0;
905         int clustersize = EVENTS_PER_CLUSTER * ctx->samples_per_event;
906         uint16_t *event;
907         uint16_t cur_sample;
908         int triggerts = -1;
909
910         /* Check if trigger is in this chunk. */
911         if (triggerpos != -1) {
912                 if (ctx->cur_samplerate <= SR_MHZ(50))
913                         triggerpos -= EVENTS_PER_CLUSTER - 1;
914
915                 if (triggerpos < 0)
916                         triggerpos = 0;
917
918                 /* Find in which cluster the trigger occured. */
919                 triggerts = triggerpos / 7;
920         }
921
922         /* For each ts. */
923         for (i = 0; i < 64; ++i) {
924                 ts = *(uint16_t *) &buf[i * 16];
925                 tsdiff = ts - *lastts;
926                 *lastts = ts;
927
928                 /* Decode partial chunk. */
929                 if (limit_chunk && ts > limit_chunk)
930                         return SR_OK;
931
932                 /* Pad last sample up to current point. */
933                 numpad = tsdiff * ctx->samples_per_event - clustersize;
934                 if (numpad > 0) {
935                         for (j = 0; j < numpad; ++j)
936                                 samples[j] = *lastsample;
937
938                         n = numpad;
939                 }
940
941                 /* Send samples between previous and this timestamp to sigrok. */
942                 sent = 0;
943                 while (sent < n) {
944                         tosend = MIN(2048, n - sent);
945
946                         packet.type = SR_DF_LOGIC;
947                         packet.payload = &logic;
948                         logic.length = tosend * sizeof(uint16_t);
949                         logic.unitsize = 2;
950                         logic.data = samples + sent;
951                         sr_session_send(ctx->session_dev_id, &packet);
952
953                         sent += tosend;
954                 }
955                 n = 0;
956
957                 event = (uint16_t *) &buf[i * 16 + 2];
958                 cur_sample = 0;
959
960                 /* For each event in cluster. */
961                 for (j = 0; j < 7; ++j) {
962
963                         /* For each sample in event. */
964                         for (k = 0; k < ctx->samples_per_event; ++k) {
965                                 cur_sample = 0;
966
967                                 /* For each probe. */
968                                 for (l = 0; l < ctx->num_probes; ++l)
969                                         cur_sample |= (!!(event[j] & (1 << (l *
970                                            ctx->samples_per_event + k)))) << l;
971
972                                 samples[n++] = cur_sample;
973                         }
974                 }
975
976                 /* Send data up to trigger point (if triggered). */
977                 sent = 0;
978                 if (i == triggerts) {
979                         /*
980                          * Trigger is not always accurate to sample because of
981                          * pipeline delay. However, it always triggers before
982                          * the actual event. We therefore look at the next
983                          * samples to pinpoint the exact position of the trigger.
984                          */
985                         tosend = get_trigger_offset(samples, *lastsample,
986                                                     &ctx->trigger);
987
988                         if (tosend > 0) {
989                                 packet.type = SR_DF_LOGIC;
990                                 packet.payload = &logic;
991                                 logic.length = tosend * sizeof(uint16_t);
992                                 logic.unitsize = 2;
993                                 logic.data = samples;
994                                 sr_session_send(ctx->session_dev_id, &packet);
995
996                                 sent += tosend;
997                         }
998
999                         /* Only send trigger if explicitly enabled. */
1000                         if (ctx->use_triggers) {
1001                                 packet.type = SR_DF_TRIGGER;
1002                                 sr_session_send(ctx->session_dev_id, &packet);
1003                         }
1004                 }
1005
1006                 /* Send rest of the chunk to sigrok. */
1007                 tosend = n - sent;
1008
1009                 if (tosend > 0) {
1010                         packet.type = SR_DF_LOGIC;
1011                         packet.payload = &logic;
1012                         logic.length = tosend * sizeof(uint16_t);
1013                         logic.unitsize = 2;
1014                         logic.data = samples + sent;
1015                         sr_session_send(ctx->session_dev_id, &packet);
1016                 }
1017
1018                 *lastsample = samples[n - 1];
1019         }
1020
1021         return SR_OK;
1022 }
1023
1024 static int receive_data(int fd, int revents, void *cb_data)
1025 {
1026         struct sr_dev_inst *sdi = cb_data;
1027         struct context *ctx = sdi->priv;
1028         struct sr_datafeed_packet packet;
1029         const int chunks_per_read = 32;
1030         unsigned char buf[chunks_per_read * CHUNK_SIZE];
1031         int bufsz, numchunks, i, newchunks;
1032         uint64_t running_msec;
1033         struct timeval tv;
1034
1035         /* Avoid compiler warnings. */
1036         (void)fd;
1037         (void)revents;
1038
1039         /* Get the current position. */
1040         sigma_read_pos(&ctx->state.stoppos, &ctx->state.triggerpos, ctx);
1041
1042         numchunks = (ctx->state.stoppos + 511) / 512;
1043
1044         if (ctx->state.state == SIGMA_IDLE)
1045                 return TRUE;
1046
1047         if (ctx->state.state == SIGMA_CAPTURE) {
1048                 /* Check if the timer has expired, or memory is full. */
1049                 gettimeofday(&tv, 0);
1050                 running_msec = (tv.tv_sec - ctx->start_tv.tv_sec) * 1000 +
1051                         (tv.tv_usec - ctx->start_tv.tv_usec) / 1000;
1052
1053                 if (running_msec < ctx->limit_msec && numchunks < 32767)
1054                         return TRUE; /* While capturing... */
1055                 else
1056                         hw_dev_acquisition_stop(sdi->index, sdi);
1057
1058         } else if (ctx->state.state == SIGMA_DOWNLOAD) {
1059                 if (ctx->state.chunks_downloaded >= numchunks) {
1060                         /* End of samples. */
1061                         packet.type = SR_DF_END;
1062                         sr_session_send(ctx->session_dev_id, &packet);
1063
1064                         ctx->state.state = SIGMA_IDLE;
1065
1066                         return TRUE;
1067                 }
1068
1069                 newchunks = MIN(chunks_per_read,
1070                                 numchunks - ctx->state.chunks_downloaded);
1071
1072                 sr_info("sigma: Downloading sample data: %.0f %%",
1073                         100.0 * ctx->state.chunks_downloaded / numchunks);
1074
1075                 bufsz = sigma_read_dram(ctx->state.chunks_downloaded,
1076                                         newchunks, buf, ctx);
1077                 /* TODO: Check bufsz. For now, just avoid compiler warnings. */
1078                 (void)bufsz;
1079
1080                 /* Find first ts. */
1081                 if (ctx->state.chunks_downloaded == 0) {
1082                         ctx->state.lastts = *(uint16_t *) buf - 1;
1083                         ctx->state.lastsample = 0;
1084                 }
1085
1086                 /* Decode chunks and send them to sigrok. */
1087                 for (i = 0; i < newchunks; ++i) {
1088                         int limit_chunk = 0;
1089
1090                         /* The last chunk may potentially be only in part. */
1091                         if (ctx->state.chunks_downloaded == numchunks - 1) {
1092                                 /* Find the last valid timestamp */
1093                                 limit_chunk = ctx->state.stoppos % 512 + ctx->state.lastts;
1094                         }
1095
1096                         if (ctx->state.chunks_downloaded + i == ctx->state.triggerchunk)
1097                                 decode_chunk_ts(buf + (i * CHUNK_SIZE),
1098                                                 &ctx->state.lastts,
1099                                                 &ctx->state.lastsample,
1100                                                 ctx->state.triggerpos & 0x1ff,
1101                                                 limit_chunk, sdi);
1102                         else
1103                                 decode_chunk_ts(buf + (i * CHUNK_SIZE),
1104                                                 &ctx->state.lastts,
1105                                                 &ctx->state.lastsample,
1106                                                 -1, limit_chunk, sdi);
1107
1108                         ++ctx->state.chunks_downloaded;
1109                 }
1110         }
1111
1112         return TRUE;
1113 }
1114
1115 /* Build a LUT entry used by the trigger functions. */
1116 static void build_lut_entry(uint16_t value, uint16_t mask, uint16_t *entry)
1117 {
1118         int i, j, k, bit;
1119
1120         /* For each quad probe. */
1121         for (i = 0; i < 4; ++i) {
1122                 entry[i] = 0xffff;
1123
1124                 /* For each bit in LUT. */
1125                 for (j = 0; j < 16; ++j)
1126
1127                         /* For each probe in quad. */
1128                         for (k = 0; k < 4; ++k) {
1129                                 bit = 1 << (i * 4 + k);
1130
1131                                 /* Set bit in entry */
1132                                 if ((mask & bit) &&
1133                                     ((!(value & bit)) !=
1134                                     (!(j & (1 << k)))))
1135                                         entry[i] &= ~(1 << j);
1136                         }
1137         }
1138 }
1139
1140 /* Add a logical function to LUT mask. */
1141 static void add_trigger_function(enum triggerop oper, enum triggerfunc func,
1142                                  int index, int neg, uint16_t *mask)
1143 {
1144         int i, j;
1145         int x[2][2], tmp, a, b, aset, bset, rset;
1146
1147         memset(x, 0, 4 * sizeof(int));
1148
1149         /* Trigger detect condition. */
1150         switch (oper) {
1151         case OP_LEVEL:
1152                 x[0][1] = 1;
1153                 x[1][1] = 1;
1154                 break;
1155         case OP_NOT:
1156                 x[0][0] = 1;
1157                 x[1][0] = 1;
1158                 break;
1159         case OP_RISE:
1160                 x[0][1] = 1;
1161                 break;
1162         case OP_FALL:
1163                 x[1][0] = 1;
1164                 break;
1165         case OP_RISEFALL:
1166                 x[0][1] = 1;
1167                 x[1][0] = 1;
1168                 break;
1169         case OP_NOTRISE:
1170                 x[1][1] = 1;
1171                 x[0][0] = 1;
1172                 x[1][0] = 1;
1173                 break;
1174         case OP_NOTFALL:
1175                 x[1][1] = 1;
1176                 x[0][0] = 1;
1177                 x[0][1] = 1;
1178                 break;
1179         case OP_NOTRISEFALL:
1180                 x[1][1] = 1;
1181                 x[0][0] = 1;
1182                 break;
1183         }
1184
1185         /* Transpose if neg is set. */
1186         if (neg) {
1187                 for (i = 0; i < 2; ++i) {
1188                         for (j = 0; j < 2; ++j) {
1189                                 tmp = x[i][j];
1190                                 x[i][j] = x[1-i][1-j];
1191                                 x[1-i][1-j] = tmp;
1192                         }
1193                 }
1194         }
1195
1196         /* Update mask with function. */
1197         for (i = 0; i < 16; ++i) {
1198                 a = (i >> (2 * index + 0)) & 1;
1199                 b = (i >> (2 * index + 1)) & 1;
1200
1201                 aset = (*mask >> i) & 1;
1202                 bset = x[b][a];
1203
1204                 if (func == FUNC_AND || func == FUNC_NAND)
1205                         rset = aset & bset;
1206                 else if (func == FUNC_OR || func == FUNC_NOR)
1207                         rset = aset | bset;
1208                 else if (func == FUNC_XOR || func == FUNC_NXOR)
1209                         rset = aset ^ bset;
1210
1211                 if (func == FUNC_NAND || func == FUNC_NOR || func == FUNC_NXOR)
1212                         rset = !rset;
1213
1214                 *mask &= ~(1 << i);
1215
1216                 if (rset)
1217                         *mask |= 1 << i;
1218         }
1219 }
1220
1221 /*
1222  * Build trigger LUTs used by 50 MHz and lower sample rates for supporting
1223  * simple pin change and state triggers. Only two transitions (rise/fall) can be
1224  * set at any time, but a full mask and value can be set (0/1).
1225  */
1226 static int build_basic_trigger(struct triggerlut *lut, struct context *ctx)
1227 {
1228         int i,j;
1229         uint16_t masks[2] = { 0, 0 };
1230
1231         memset(lut, 0, sizeof(struct triggerlut));
1232
1233         /* Contant for simple triggers. */
1234         lut->m4 = 0xa000;
1235
1236         /* Value/mask trigger support. */
1237         build_lut_entry(ctx->trigger.simplevalue, ctx->trigger.simplemask,
1238                         lut->m2d);
1239
1240         /* Rise/fall trigger support. */
1241         for (i = 0, j = 0; i < 16; ++i) {
1242                 if (ctx->trigger.risingmask & (1 << i) ||
1243                     ctx->trigger.fallingmask & (1 << i))
1244                         masks[j++] = 1 << i;
1245         }
1246
1247         build_lut_entry(masks[0], masks[0], lut->m0d);
1248         build_lut_entry(masks[1], masks[1], lut->m1d);
1249
1250         /* Add glue logic */
1251         if (masks[0] || masks[1]) {
1252                 /* Transition trigger. */
1253                 if (masks[0] & ctx->trigger.risingmask)
1254                         add_trigger_function(OP_RISE, FUNC_OR, 0, 0, &lut->m3);
1255                 if (masks[0] & ctx->trigger.fallingmask)
1256                         add_trigger_function(OP_FALL, FUNC_OR, 0, 0, &lut->m3);
1257                 if (masks[1] & ctx->trigger.risingmask)
1258                         add_trigger_function(OP_RISE, FUNC_OR, 1, 0, &lut->m3);
1259                 if (masks[1] & ctx->trigger.fallingmask)
1260                         add_trigger_function(OP_FALL, FUNC_OR, 1, 0, &lut->m3);
1261         } else {
1262                 /* Only value/mask trigger. */
1263                 lut->m3 = 0xffff;
1264         }
1265
1266         /* Triggertype: event. */
1267         lut->params.selres = 3;
1268
1269         return SR_OK;
1270 }
1271
1272 static int hw_dev_acquisition_start(int dev_index, void *cb_data)
1273 {
1274         struct sr_dev_inst *sdi;
1275         struct context *ctx;
1276         struct sr_datafeed_packet *packet;
1277         struct sr_datafeed_header *header;
1278         struct sr_datafeed_meta_logic meta;
1279         struct clockselect_50 clockselect;
1280         int frac, triggerpin, ret;
1281         uint8_t triggerselect;
1282         struct triggerinout triggerinout_conf;
1283         struct triggerlut lut;
1284
1285         if (!(sdi = sr_dev_inst_get(adi->instances, dev_index)))
1286                 return SR_ERR;
1287
1288         ctx = sdi->priv;
1289
1290         /* If the samplerate has not been set, default to 200 kHz. */
1291         if (ctx->cur_firmware == -1) {
1292                 if ((ret = set_samplerate(sdi, SR_KHZ(200))) != SR_OK)
1293                         return ret;
1294         }
1295
1296         /* Enter trigger programming mode. */
1297         sigma_set_register(WRITE_TRIGGER_SELECT1, 0x20, ctx);
1298
1299         /* 100 and 200 MHz mode. */
1300         if (ctx->cur_samplerate >= SR_MHZ(100)) {
1301                 sigma_set_register(WRITE_TRIGGER_SELECT1, 0x81, ctx);
1302
1303                 /* Find which pin to trigger on from mask. */
1304                 for (triggerpin = 0; triggerpin < 8; ++triggerpin)
1305                         if ((ctx->trigger.risingmask | ctx->trigger.fallingmask) &
1306                             (1 << triggerpin))
1307                                 break;
1308
1309                 /* Set trigger pin and light LED on trigger. */
1310                 triggerselect = (1 << LEDSEL1) | (triggerpin & 0x7);
1311
1312                 /* Default rising edge. */
1313                 if (ctx->trigger.fallingmask)
1314                         triggerselect |= 1 << 3;
1315
1316         /* All other modes. */
1317         } else if (ctx->cur_samplerate <= SR_MHZ(50)) {
1318                 build_basic_trigger(&lut, ctx);
1319
1320                 sigma_write_trigger_lut(&lut, ctx);
1321
1322                 triggerselect = (1 << LEDSEL1) | (1 << LEDSEL0);
1323         }
1324
1325         /* Setup trigger in and out pins to default values. */
1326         memset(&triggerinout_conf, 0, sizeof(struct triggerinout));
1327         triggerinout_conf.trgout_bytrigger = 1;
1328         triggerinout_conf.trgout_enable = 1;
1329
1330         sigma_write_register(WRITE_TRIGGER_OPTION,
1331                              (uint8_t *) &triggerinout_conf,
1332                              sizeof(struct triggerinout), ctx);
1333
1334         /* Go back to normal mode. */
1335         sigma_set_register(WRITE_TRIGGER_SELECT1, triggerselect, ctx);
1336
1337         /* Set clock select register. */
1338         if (ctx->cur_samplerate == SR_MHZ(200))
1339                 /* Enable 4 probes. */
1340                 sigma_set_register(WRITE_CLOCK_SELECT, 0xf0, ctx);
1341         else if (ctx->cur_samplerate == SR_MHZ(100))
1342                 /* Enable 8 probes. */
1343                 sigma_set_register(WRITE_CLOCK_SELECT, 0x00, ctx);
1344         else {
1345                 /*
1346                  * 50 MHz mode (or fraction thereof). Any fraction down to
1347                  * 50 MHz / 256 can be used, but is not supported by sigrok API.
1348                  */
1349                 frac = SR_MHZ(50) / ctx->cur_samplerate - 1;
1350
1351                 clockselect.async = 0;
1352                 clockselect.fraction = frac;
1353                 clockselect.disabled_probes = 0;
1354
1355                 sigma_write_register(WRITE_CLOCK_SELECT,
1356                                      (uint8_t *) &clockselect,
1357                                      sizeof(clockselect), ctx);
1358         }
1359
1360         /* Setup maximum post trigger time. */
1361         sigma_set_register(WRITE_POST_TRIGGER,
1362                            (ctx->capture_ratio * 255) / 100, ctx);
1363
1364         /* Start acqusition. */
1365         gettimeofday(&ctx->start_tv, 0);
1366         sigma_set_register(WRITE_MODE, 0x0d, ctx);
1367
1368         ctx->session_dev_id = cb_data;
1369
1370         if (!(packet = g_try_malloc(sizeof(struct sr_datafeed_packet)))) {
1371                 sr_err("sigma: %s: packet malloc failed.", __func__);
1372                 return SR_ERR_MALLOC;
1373         }
1374
1375         if (!(header = g_try_malloc(sizeof(struct sr_datafeed_header)))) {
1376                 sr_err("sigma: %s: header malloc failed.", __func__);
1377                 return SR_ERR_MALLOC;
1378         }
1379
1380         /* Send header packet to the session bus. */
1381         packet->type = SR_DF_HEADER;
1382         packet->payload = header;
1383         header->feed_version = 1;
1384         gettimeofday(&header->starttime, NULL);
1385         sr_session_send(ctx->session_dev_id, packet);
1386
1387         /* Send metadata about the SR_DF_LOGIC packets to come. */
1388         packet->type = SR_DF_META_LOGIC;
1389         packet->payload = &meta;
1390         meta.samplerate = ctx->cur_samplerate;
1391         meta.num_probes = ctx->num_probes;
1392         sr_session_send(ctx->session_dev_id, packet);
1393
1394         /* Add capture source. */
1395         sr_source_add(0, G_IO_IN, 10, receive_data, sdi);
1396
1397         g_free(header);
1398         g_free(packet);
1399
1400         ctx->state.state = SIGMA_CAPTURE;
1401
1402         return SR_OK;
1403 }
1404
1405 static int hw_dev_acquisition_stop(int dev_index, void *cb_data)
1406 {
1407         struct sr_dev_inst *sdi;
1408         struct context *ctx;
1409         uint8_t modestatus;
1410
1411         /* Avoid compiler warnings. */
1412         (void)cb_data;
1413
1414         if (!(sdi = sr_dev_inst_get(adi->instances, dev_index))) {
1415                 sr_err("sigma: %s: sdi was NULL", __func__);
1416                 return SR_ERR_BUG;
1417         }
1418
1419         if (!(ctx = sdi->priv)) {
1420                 sr_err("sigma: %s: sdi->priv was NULL", __func__);
1421                 return SR_ERR_BUG;
1422         }
1423
1424         /* Stop acquisition. */
1425         sigma_set_register(WRITE_MODE, 0x11, ctx);
1426
1427         /* Set SDRAM Read Enable. */
1428         sigma_set_register(WRITE_MODE, 0x02, ctx);
1429
1430         /* Get the current position. */
1431         sigma_read_pos(&ctx->state.stoppos, &ctx->state.triggerpos, ctx);
1432
1433         /* Check if trigger has fired. */
1434         modestatus = sigma_get_register(READ_MODE, ctx);
1435         if (modestatus & 0x20)
1436                 ctx->state.triggerchunk = ctx->state.triggerpos / 512;
1437         else
1438                 ctx->state.triggerchunk = -1;
1439
1440         ctx->state.chunks_downloaded = 0;
1441
1442         ctx->state.state = SIGMA_DOWNLOAD;
1443
1444         return SR_OK;
1445 }
1446
1447 SR_PRIV struct sr_dev_driver asix_sigma_driver_info = {
1448         .name = "asix-sigma",
1449         .longname = "ASIX SIGMA/SIGMA2",
1450         .api_version = 1,
1451         .init = hw_init,
1452         .cleanup = hw_cleanup,
1453         .scan = hw_scan,
1454         .dev_open = hw_dev_open,
1455         .dev_close = hw_dev_close,
1456         .info_get = hw_info_get,
1457         .dev_status_get = hw_dev_status_get,
1458         .dev_config_set = hw_dev_config_set,
1459         .dev_acquisition_start = hw_dev_acquisition_start,
1460         .dev_acquisition_stop = hw_dev_acquisition_stop,
1461         .instances = NULL,
1462 };