]> sigrok.org Git - libsigrok.git/blob - hardware/openbench-logic-sniffer/ols.c
OLS: fix multi-channel capture
[libsigrok.git] / hardware / openbench-logic-sniffer / ols.c
1 /*
2  * This file is part of the sigrok project.
3  *
4  * Copyright (C) 2010 Bert Vermeulen <bert@biot.com>
5  *
6  * This program is free software: you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation, either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <stdio.h>
21 #include <stdint.h>
22 #include <stdlib.h>
23 #include <sys/types.h>
24 #include <sys/stat.h>
25 #include <fcntl.h>
26 #include <unistd.h>
27 #ifndef _WIN32
28 #include <termios.h>
29 #endif
30 #include <string.h>
31 #include <sys/time.h>
32 #include <inttypes.h>
33 #ifdef _WIN32
34 /* TODO */
35 #else
36 #include <arpa/inet.h>
37 #endif
38 #include <glib.h>
39 #include <sigrok.h>
40
41 #define NUM_PROBES                      32
42 #define NUM_TRIGGER_STAGES              4
43 #define TRIGGER_TYPES                   "01"
44 #define SERIAL_SPEED                    B115200
45 /* TODO: SERIAL_ bits, parity, stop bit */
46 #define CLOCK_RATE                      100000000
47
48 /* Command opcodes */
49 #define CMD_RESET                       0x00
50 #define CMD_ID                          0x02
51 #define CMD_SET_FLAGS                   0x82
52 #define CMD_SET_DIVIDER                 0x80
53 #define CMD_RUN                         0x01
54 #define CMD_CAPTURE_SIZE                0x81
55 #define CMD_SET_TRIGGER_MASK_0          0xc0
56 #define CMD_SET_TRIGGER_MASK_1          0xc4
57 #define CMD_SET_TRIGGER_MASK_2          0xc8
58 #define CMD_SET_TRIGGER_MASK_3          0xcc
59 #define CMD_SET_TRIGGER_VALUE_0         0xc1
60 #define CMD_SET_TRIGGER_VALUE_1         0xc5
61 #define CMD_SET_TRIGGER_VALUE_2         0xc9
62 #define CMD_SET_TRIGGER_VALUE_3         0xcd
63 #define CMD_SET_TRIGGER_CONFIG_0        0xc2
64 #define CMD_SET_TRIGGER_CONFIG_1        0xc6
65 #define CMD_SET_TRIGGER_CONFIG_2        0xca
66 #define CMD_SET_TRIGGER_CONFIG_3        0xce
67
68 /* Bitmasks for CMD_FLAGS */
69 #define FLAG_DEMUX                      0x01
70 #define FLAG_FILTER                     0x02
71 #define FLAG_CHANNELGROUP_1             0x04
72 #define FLAG_CHANNELGROUP_2             0x08
73 #define FLAG_CHANNELGROUP_3             0x10
74 #define FLAG_CHANNELGROUP_4             0x20
75 #define FLAG_CLOCK_EXTERNAL             0x40
76 #define FLAG_CLOCK_INVERTED             0x80
77 #define FLAG_RLE                        0x0100
78
79 static int capabilities[] = {
80         HWCAP_LOGIC_ANALYZER,
81         HWCAP_SAMPLERATE,
82         HWCAP_CAPTURE_RATIO,
83         HWCAP_LIMIT_SAMPLES,
84         0,
85 };
86
87 static struct samplerates samplerates = {
88         10,
89         MHZ(200),
90         1,
91         0,
92 };
93
94 /* List of struct serial_device_instance */
95 static GSList *device_instances = NULL;
96
97 /* Current state of the flag register */
98 static uint32_t flag_reg = 0;
99
100 static uint64_t cur_samplerate = 0;
101 static uint64_t limit_samples = 0;
102 /*
103  * Pre/post trigger capture ratio, in percentage.
104  * 0 means no pre-trigger data.
105  */
106 static int capture_ratio = 0;
107 static int trigger_at = -1;
108 static uint32_t probe_mask = 0xffffffff;
109 static uint32_t trigger_mask[4] = { 0, 0, 0, 0 };
110 static uint32_t trigger_value[4] = { 0, 0, 0, 0 };
111 static int num_stages = 0;
112
113 static int send_shortcommand(int fd, uint8_t command)
114 {
115         char buf[1];
116
117         g_debug("ols: sending cmd 0x%.2x", command);
118         buf[0] = command;
119         if (write(fd, buf, 1) != 1)
120                 return SIGROK_ERR;
121
122         return SIGROK_OK;
123 }
124
125 static int send_longcommand(int fd, uint8_t command, uint32_t data)
126 {
127         char buf[5];
128
129         g_debug("ols: sending cmd 0x%.2x data 0x%.8x", command, data);
130         buf[0] = command;
131         buf[1] = (data & 0xff000000) >> 24;
132         buf[2] = (data & 0xff0000) >> 16;
133         buf[3] = (data & 0xff00) >> 8;
134         buf[4] = data & 0xff;
135         if (write(fd, buf, 5) != 5)
136                 return SIGROK_ERR;
137
138         return SIGROK_OK;
139 }
140
141 static int configure_probes(GSList *probes)
142 {
143         struct probe *probe;
144         GSList *l;
145         int probe_bit, stage, i;
146         char *tc;
147
148         probe_mask = 0;
149         for (i = 0; i < NUM_TRIGGER_STAGES; i++) {
150                 trigger_mask[i] = 0;
151                 trigger_value[i] = 0;
152         }
153
154         num_stages = 0;
155         for (l = probes; l; l = l->next) {
156                 probe = (struct probe *)l->data;
157                 if (!probe->enabled)
158                         continue;
159
160                 /*
161                  * Set up the probe mask for later configuration into the
162                  * flag register.
163                  */
164                 probe_bit = 1 << (probe->index - 1);
165                 probe_mask |= probe_bit;
166
167                 if (!probe->trigger)
168                         continue;
169
170                 /* Configure trigger mask and value. */
171                 stage = 0;
172                 for (tc = probe->trigger; tc && *tc; tc++) {
173                         trigger_mask[stage] |= probe_bit;
174                         if (*tc == '1')
175                                 trigger_value[stage] |= probe_bit;
176                         stage++;
177                         if (stage > 3)
178                                 /*
179                                  * TODO: Only supporting parallel mode, with
180                                  * up to 4 stages.
181                                  */
182                                 return SIGROK_ERR;
183                 }
184                 if (stage > num_stages)
185                         num_stages = stage;
186         }
187
188         return SIGROK_OK;
189 }
190
191 static uint32_t reverse16(uint32_t in)
192 {
193         uint32_t out;
194
195         out = (in & 0xff) << 8;
196         out |= (in & 0xff00) >> 8;
197         out |= (in & 0xff0000) << 8;
198         out |= (in & 0xff000000) >> 8;
199
200         return out;
201 }
202
203 static uint32_t reverse32(uint32_t in)
204 {
205         uint32_t out;
206
207         out = (in & 0xff) << 24;
208         out |= (in & 0xff00) << 8;
209         out |= (in & 0xff0000) >> 8;
210         out |= (in & 0xff000000) >> 24;
211
212         return out;
213 }
214
215 static int hw_init(char *deviceinfo)
216 {
217         struct sigrok_device_instance *sdi;
218         GSList *ports, *l;
219         GPollFD *fds;
220         int devcnt, final_devcnt, num_ports, fd, ret, i;
221         char buf[8], **device_names, **serial_params;
222
223         if (deviceinfo)
224                 ports = g_slist_append(NULL, strdup(deviceinfo));
225         else
226                 /* No specific device given, so scan all serial ports. */
227                 ports = list_serial_ports();
228
229         num_ports = g_slist_length(ports);
230         fds = calloc(1, num_ports * sizeof(GPollFD));
231         device_names = malloc(num_ports * sizeof(char *));
232         serial_params = malloc(num_ports * sizeof(char *));
233         devcnt = 0;
234         for (l = ports; l; l = l->next) {
235                 /* The discovery procedure is like this: first send the Reset
236                  * command (0x00) 5 times, since the device could be anywhere
237                  * in a 5-byte command. Then send the ID command (0x02).
238                  * If the device responds with 4 bytes ("OLS1" or "SLA1"), we
239                  * have a match.
240                  *
241                  * Since it may take the device a while to respond at 115Kb/s,
242                  * we do all the sending first, then wait for all of them to
243                  * respond with g_poll().
244                  */
245                 g_message("probing %s...", (char *)l->data);
246 #ifdef _WIN32
247                 // FIXME
248                 // hdl = serial_open(l->data, 0);
249 #else
250                 fd = serial_open(l->data, O_RDWR | O_NONBLOCK);
251 #endif
252                 if (fd != -1) {
253                         serial_params[devcnt] = serial_backup_params(fd);
254                         serial_set_params(fd, 115200, 8, 0, 1, 2);
255                         ret = SIGROK_OK;
256                         for (i = 0; i < 5; i++) {
257                                 if ((ret = send_shortcommand(fd,
258                                         CMD_RESET)) != SIGROK_OK) {
259                                         /* Serial port is not writable. */
260                                         break;
261                                 }
262                         }
263                         if (ret != SIGROK_OK) {
264                                 serial_restore_params(fd,
265                                         serial_params[devcnt]);
266                                 serial_close(fd);
267                                 continue;
268                         }
269                         send_shortcommand(fd, CMD_ID);
270                         fds[devcnt].fd = fd;
271                         fds[devcnt].events = G_IO_IN;
272                         device_names[devcnt] = strdup(l->data);
273                         devcnt++;
274                 }
275                 free(l->data);
276         }
277
278         /* 2ms isn't enough for reliable transfer with pl2303, let's try 10 */
279         usleep(10000);
280
281         final_devcnt = 0;
282         g_poll(fds, devcnt, 1);
283         for (i = 0; i < devcnt; i++) {
284                 if (fds[i].revents == G_IO_IN) {
285                         if (read(fds[i].fd, buf, 4) == 4) {
286                                 if (!strncmp(buf, "1SLO", 4)
287                                     || !strncmp(buf, "1ALS", 4)) {
288                                         if (!strncmp(buf, "1SLO", 4))
289                                                 sdi = sigrok_device_instance_new
290                                                     (final_devcnt, ST_INACTIVE,
291                                                      "Openbench",
292                                                      "Logic Sniffer", "v1.0");
293                                         else
294                                                 sdi = sigrok_device_instance_new
295                                                     (final_devcnt, ST_INACTIVE,
296                                                      "Sump", "Logic Analyzer",
297                                                      "v1.0");
298                                         sdi->serial = serial_device_instance_new
299                                             (device_names[i], -1);
300                                         device_instances =
301                                             g_slist_append(device_instances, sdi);
302                                         final_devcnt++;
303                                         serial_close(fds[i].fd);
304                                         fds[i].fd = 0;
305                                 }
306                         }
307                         free(device_names[i]);
308                 }
309
310                 if (fds[i].fd != 0) {
311                         serial_restore_params(fds[i].fd, serial_params[i]);
312                         serial_close(fds[i].fd);
313                 }
314                 free(serial_params[i]);
315         }
316
317         free(fds);
318         free(device_names);
319         free(serial_params);
320         g_slist_free(ports);
321
322         cur_samplerate = samplerates.low;
323
324         return final_devcnt;
325 }
326
327 static int hw_opendev(int device_index)
328 {
329         struct sigrok_device_instance *sdi;
330
331         if (!(sdi = get_sigrok_device_instance(device_instances, device_index)))
332                 return SIGROK_ERR;
333
334         sdi->serial->fd = serial_open(sdi->serial->port, O_RDWR);
335         if (sdi->serial->fd == -1)
336                 return SIGROK_ERR;
337
338         sdi->status = ST_ACTIVE;
339
340         return SIGROK_OK;
341 }
342
343 static void hw_closedev(int device_index)
344 {
345         struct sigrok_device_instance *sdi;
346
347         if (!(sdi = get_sigrok_device_instance(device_instances, device_index)))
348                 return;
349
350         if (sdi->serial->fd != -1) {
351                 serial_close(sdi->serial->fd);
352                 sdi->serial->fd = -1;
353                 sdi->status = ST_INACTIVE;
354         }
355 }
356
357 static void hw_cleanup(void)
358 {
359         GSList *l;
360         struct sigrok_device_instance *sdi;
361
362         /* Properly close all devices. */
363         for (l = device_instances; l; l = l->next) {
364                 sdi = l->data;
365                 if (sdi->serial->fd != -1)
366                         serial_close(sdi->serial->fd);
367                 sigrok_device_instance_free(sdi);
368         }
369         g_slist_free(device_instances);
370         device_instances = NULL;
371 }
372
373 static void *hw_get_device_info(int device_index, int device_info_id)
374 {
375         struct sigrok_device_instance *sdi;
376         void *info;
377
378         if (!(sdi = get_sigrok_device_instance(device_instances, device_index)))
379                 return NULL;
380
381         info = NULL;
382         switch (device_info_id) {
383         case DI_INSTANCE:
384                 info = sdi;
385                 break;
386         case DI_NUM_PROBES:
387                 info = GINT_TO_POINTER(NUM_PROBES);
388                 break;
389         case DI_SAMPLERATES:
390                 info = &samplerates;
391                 break;
392         case DI_TRIGGER_TYPES:
393                 info = (char *)TRIGGER_TYPES;
394                 break;
395         case DI_CUR_SAMPLERATE:
396                 info = &cur_samplerate;
397                 break;
398         }
399
400         return info;
401 }
402
403 static int hw_get_status(int device_index)
404 {
405         struct sigrok_device_instance *sdi;
406
407         if (!(sdi = get_sigrok_device_instance(device_instances, device_index)))
408                 return ST_NOT_FOUND;
409
410         return sdi->status;
411 }
412
413 static int *hw_get_capabilities(void)
414 {
415         return capabilities;
416 }
417
418 static int set_configuration_samplerate(struct sigrok_device_instance *sdi,
419                                         uint64_t samplerate)
420 {
421         uint32_t divider;
422
423         if (samplerate < samplerates.low || samplerate > samplerates.high)
424                 return SIGROK_ERR_SAMPLERATE;
425
426         if (samplerate > CLOCK_RATE) {
427                 flag_reg |= FLAG_DEMUX;
428                 divider = (CLOCK_RATE * 2 / samplerate) - 1;
429         } else {
430                 flag_reg &= ~FLAG_DEMUX;
431                 divider = (CLOCK_RATE / samplerate) - 1;
432         }
433
434         g_message("setting samplerate to %" PRIu64 " Hz (divider %u, demux %s)",
435                         samplerate, divider, flag_reg & FLAG_DEMUX ? "on" : "off");
436
437         if (send_longcommand(sdi->serial->fd, CMD_SET_DIVIDER, reverse32(divider)) != SIGROK_OK)
438                 return SIGROK_ERR;
439         cur_samplerate = samplerate;
440
441         return SIGROK_OK;
442 }
443
444 static int hw_set_configuration(int device_index, int capability, void *value)
445 {
446         struct sigrok_device_instance *sdi;
447         int ret;
448         uint64_t *tmp_u64;
449
450         if (!(sdi = get_sigrok_device_instance(device_instances, device_index)))
451                 return SIGROK_ERR;
452
453         if (sdi->status != ST_ACTIVE)
454                 return SIGROK_ERR;
455
456         switch (capability) {
457         case HWCAP_SAMPLERATE:
458                 tmp_u64 = value;
459                 ret = set_configuration_samplerate(sdi, *tmp_u64);
460                 break;
461         case HWCAP_PROBECONFIG:
462                 ret = configure_probes((GSList *) value);
463                 break;
464         case HWCAP_LIMIT_SAMPLES:
465                 tmp_u64 = value;
466                 limit_samples = *tmp_u64;
467                 ret = SIGROK_OK;
468                 break;
469         case HWCAP_CAPTURE_RATIO:
470                 tmp_u64 = value;
471                 capture_ratio = *tmp_u64;
472                 if (capture_ratio < 0 || capture_ratio > 100) {
473                         capture_ratio = 0;
474                         ret = SIGROK_ERR;
475                 } else
476                         ret = SIGROK_OK;
477                 break;
478         default:
479                 ret = SIGROK_ERR;
480         }
481
482         return ret;
483 }
484
485 static int receive_data(int fd, int revents, void *user_data)
486 {
487         static unsigned int num_transfers = 0;
488         static int num_bytes = 0;
489         static char last_sample[4] = { 0xff, 0xff, 0xff, 0xff };
490         static unsigned char sample[4] = { 0, 0, 0, 0 };
491         static unsigned char tmp_sample[4];
492         static unsigned char *raw_sample_buf = NULL;
493         int count, buflen, num_channels, offset, i, j;
494         struct datafeed_packet packet;
495         unsigned char byte, *buffer;
496
497         if (num_transfers++ == 0) {
498                 /*
499                  * First time round, means the device started sending data,
500                  * and will not stop until done. If it stops sending for
501                  * longer than it takes to send a byte, that means it's
502                  * finished. We'll double that to 30ms to be sure...
503                  */
504                 source_remove(fd);
505                 source_add(fd, G_IO_IN, 30, receive_data, user_data);
506                 raw_sample_buf = malloc(limit_samples * 4);
507                 /* fill with 1010... for debugging */
508                 memset(raw_sample_buf, 0x82, limit_samples * 4);
509         }
510
511         num_channels = 0;
512         for (i = 0x20; i > 0x02; i /= 2) {
513                 if ((flag_reg & i) == 0)
514                         num_channels++;
515         }
516
517         if (revents == G_IO_IN
518             && num_transfers / num_channels <= limit_samples) {
519                 if (read(fd, &byte, 1) != 1)
520                         return FALSE;
521
522                 sample[num_bytes++] = byte;
523                 g_debug("received byte 0x%.2x", byte);
524                 if (num_bytes == num_channels) {
525                         g_debug("received sample 0x%.*x", num_bytes * 2, (int) *sample);
526                         /* Got a full sample. */
527                         if (flag_reg & FLAG_RLE) {
528                                 /*
529                                  * In RLE mode -1 should never come in as a
530                                  * sample, because bit 31 is the "count" flag.
531                                  * TODO: Endianness may be wrong here, could be
532                                  * sample[3].
533                                  */
534                                 if (sample[0] & 0x80
535                                     && !(last_sample[0] & 0x80)) {
536                                         count = (int)(*sample) & 0x7fffffff;
537                                         buffer = g_malloc(count);
538                                         buflen = 0;
539                                         for (i = 0; i < count; i++) {
540                                                 memcpy(buffer + buflen, last_sample, 4);
541                                                 buflen += 4;
542                                         }
543                                 } else {
544                                         /*
545                                          * Just a single sample, next sample
546                                          * will probably be a count referring
547                                          * to this -- but this one is still a
548                                          * part of the stream.
549                                          */
550                                         buffer = sample;
551                                         buflen = 4;
552                                 }
553                         } else {
554                                 /* No compression. */
555                                 buffer = sample;
556                                 buflen = 4;
557                         }
558
559                         if (num_channels < 4) {
560                                 /*
561                                  * Some channel groups may have been turned
562                                  * off, to speed up transfer between the
563                                  * hardware and the PC. Expand that here before
564                                  * submitting it over the session bus --
565                                  * whatever is listening on the bus will be
566                                  * expecting a full 32-bit sample, based on
567                                  * the number of probes.
568                                  */
569                                 j = 0;
570                                 memset(tmp_sample, 0, 4);
571                                 for (i = 0; i < 4; i++) {
572                                         if (((flag_reg >> 2) & (1 << i)) == 0) {
573                                                 /*
574                                                  * This channel group was
575                                                  * enabled, copy from received
576                                                  * sample.
577                                                  */
578                                                 tmp_sample[i] = sample[j++];
579                                         }
580                                 }
581                                 memcpy(sample, tmp_sample, 4);
582                                 g_debug("full sample 0x%.8x", (int) *sample);
583                         }
584
585                         /* the OLS sends its sample buffer backwards.
586                          * store it in reverse order here, so we can dump
587                          * this on the session bus later.
588                          */
589                         offset = (limit_samples - num_transfers / num_channels) * 4;
590                         memcpy(raw_sample_buf + offset, sample, 4);
591
592                         if (buffer == sample)
593                                 memcpy(last_sample, buffer, num_channels);
594                         else
595                                 g_free(buffer);
596
597                         memset(sample, 0, 4);
598                         num_bytes = 0;
599                 }
600         } else {
601                 /*
602                  * This is the main loop telling us a timeout was reached, or
603                  * we've acquired all the samples we asked for -- we're done.
604                  * Send the (properly-ordered) buffer to the frontend.
605                  */
606                 if (trigger_at != -1) {
607                         /* a trigger was set up, so we need to tell the frontend
608                          * about it.
609                          */
610                         if (trigger_at > 0) {
611                                 /* there are pre-trigger samples, send those first */
612                                 packet.type = DF_LOGIC32;
613                                 packet.length = trigger_at * 4;
614                                 packet.payload = raw_sample_buf;
615                                 session_bus(user_data, &packet);
616                         }
617
618                         packet.type = DF_TRIGGER;
619                         packet.length = 0;
620                         session_bus(user_data, &packet);
621
622                         packet.type = DF_LOGIC32;
623                         packet.length = (limit_samples * 4) - (trigger_at * 4);
624                         packet.payload = raw_sample_buf + trigger_at * 4;
625                         session_bus(user_data, &packet);
626                 } else {
627                         packet.type = DF_LOGIC32;
628                         packet.length = limit_samples * 4;
629                         packet.payload = raw_sample_buf;
630                         session_bus(user_data, &packet);
631                 }
632                 free(raw_sample_buf);
633
634 #ifndef _WIN32
635                 /* TODO: Move to serial.c? */
636                 tcflush(fd, TCIOFLUSH);
637 #endif
638                 serial_close(fd);
639                 packet.type = DF_END;
640                 packet.length = 0;
641                 session_bus(user_data, &packet);
642         }
643
644         return TRUE;
645 }
646
647 static int hw_start_acquisition(int device_index, gpointer session_device_id)
648 {
649         int i;
650         struct datafeed_packet *packet;
651         struct datafeed_header *header;
652         struct sigrok_device_instance *sdi;
653         uint32_t trigger_config[4];
654         uint32_t data;
655         uint16_t readcount, delaycount;
656         uint8_t changrp_mask;
657
658         if (!(sdi = get_sigrok_device_instance(device_instances, device_index)))
659                 return SIGROK_ERR;
660
661         if (sdi->status != ST_ACTIVE)
662                 return SIGROK_ERR;
663
664         readcount = limit_samples / 4;
665
666         memset(trigger_config, 0, 16);
667         trigger_config[num_stages-1] |= 0x08;
668         if (trigger_mask[0]) {
669                 delaycount = readcount * (1 - capture_ratio / 100.0);
670                 trigger_at = (readcount - delaycount) * 4 - num_stages;
671                 g_message("ta %d", trigger_at);
672
673                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_0,
674                         reverse32(trigger_mask[0])) != SIGROK_OK)
675                         return SIGROK_ERR;
676                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_0,
677                         reverse32(trigger_value[0])) != SIGROK_OK)
678                         return SIGROK_ERR;
679                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_0,
680                         trigger_config[0]) != SIGROK_OK)
681                         return SIGROK_ERR;
682
683                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_1,
684                         reverse32(trigger_mask[1])) != SIGROK_OK)
685                         return SIGROK_ERR;
686                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_1,
687                         reverse32(trigger_value[1])) != SIGROK_OK)
688                         return SIGROK_ERR;
689                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_1,
690                         trigger_config[1]) != SIGROK_OK)
691                         return SIGROK_ERR;
692
693                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_2,
694                         reverse32(trigger_mask[2])) != SIGROK_OK)
695                         return SIGROK_ERR;
696                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_2,
697                         reverse32(trigger_value[2])) != SIGROK_OK)
698                         return SIGROK_ERR;
699                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_2,
700                         trigger_config[2]) != SIGROK_OK)
701                         return SIGROK_ERR;
702
703                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_3,
704                         reverse32(trigger_mask[3])) != SIGROK_OK)
705                         return SIGROK_ERR;
706                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_3,
707                         reverse32(trigger_value[3])) != SIGROK_OK)
708                         return SIGROK_ERR;
709                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_3,
710                         trigger_config[3]) != SIGROK_OK)
711                         return SIGROK_ERR;
712         } else {
713                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_0,
714                      trigger_mask[0]) != SIGROK_OK)
715                         return SIGROK_ERR;
716                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_0,
717                      trigger_value[0]) != SIGROK_OK)
718                         return SIGROK_ERR;
719                 if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_0,
720                      0x00000008) != SIGROK_OK)
721                         return SIGROK_ERR;
722                 delaycount = readcount;
723         }
724
725         set_configuration_samplerate(sdi, cur_samplerate);
726
727         /* Send sample limit and pre/post-trigger capture ratio. */
728         data = ((readcount - 1) & 0xffff) << 16;
729         data |= (delaycount - 1) & 0xffff;
730         if (send_longcommand(sdi->serial->fd, CMD_CAPTURE_SIZE, reverse16(data)) != SIGROK_OK)
731                 return SIGROK_ERR;
732
733         /*
734          * Enable/disable channel groups in the flag register according to the
735          * probe mask.
736          */
737         changrp_mask = 0;
738         for (i = 0; i < 4; i++) {
739                 if (probe_mask & (0xff << (i * 8)))
740                         changrp_mask |= (1 << i);
741         }
742         g_message("changrp_mask 0x%.2x", changrp_mask);
743
744         /* The flag register wants them here, and 1 means "disable channel". */
745         flag_reg |= ~(changrp_mask << 2) & 0x3c;
746         g_message("flag_reg 0x%.2x", flag_reg & 0x3c);
747
748         flag_reg |= FLAG_FILTER;
749         data = flag_reg << 24;
750         if (send_longcommand(sdi->serial->fd, CMD_SET_FLAGS, data) != SIGROK_OK)
751                 return SIGROK_ERR;
752
753         /* Start acquisition on the device. */
754         if (send_shortcommand(sdi->serial->fd, CMD_RUN) != SIGROK_OK)
755                 return SIGROK_ERR;
756
757         source_add(sdi->serial->fd, G_IO_IN, -1, receive_data,
758                    session_device_id);
759
760         /* Send header packet to the session bus. */
761         packet = g_malloc(sizeof(struct datafeed_packet));
762         header = g_malloc(sizeof(struct datafeed_header));
763         if (!packet || !header)
764                 return SIGROK_ERR;
765         packet->type = DF_HEADER;
766         packet->length = sizeof(struct datafeed_header);
767         packet->payload = (unsigned char *)header;
768         header->feed_version = 1;
769         gettimeofday(&header->starttime, NULL);
770         header->samplerate = cur_samplerate;
771         header->protocol_id = PROTO_RAW;
772         header->num_probes = NUM_PROBES;
773         session_bus(session_device_id, packet);
774         g_free(header);
775         g_free(packet);
776
777         return SIGROK_OK;
778 }
779
780 static void hw_stop_acquisition(int device_index, gpointer session_device_id)
781 {
782         struct datafeed_packet packet;
783
784         /* QUICK HACK */
785         device_index = device_index;
786
787         packet.type = DF_END;
788         packet.length = 0;
789         session_bus(session_device_id, &packet);
790 }
791
792 struct device_plugin ols_plugin_info = {
793         "sump",
794         1,
795         hw_init,
796         hw_cleanup,
797         hw_opendev,
798         hw_closedev,
799         hw_get_device_info,
800         hw_get_status,
801         hw_get_capabilities,
802         hw_set_configuration,
803         hw_start_acquisition,
804         hw_stop_acquisition,
805 };