hardware/asix-sigma/asix-sigma.c

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