]> sigrok.org Git - libsigrok.git/blob - src/input/trace32_ad.c
046cb53bef64cf2f258b71776118e623952b1ede
[libsigrok.git] / src / input / trace32_ad.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2015 Soeren Apel <soeren@apelpie.net>
5  * Copyright (C) 2015 Bert Vermeulen <bert@biot.com>
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation, either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19  */
20
21 /*
22  * Usage notes:
23  * This input module reads .ad files created using the
24  * following practice commands:
25  *
26  * I.SAVE <file> /NoCompress
27  * IPROBE.SAVE <file> /NoCompress
28  *
29  * It currently cannot make use of files that have been
30  * saved using /QuickCompress, /Compress or /ZIP.
31  * As a workaround you may load the file in PowerView
32  * using I.LOAD / IPROBE.LOAD and re-save using /NoCompress.
33  */
34
35 #include <config.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <sys/types.h>
39 #include <sys/stat.h>
40 #include <fcntl.h>
41 #include <unistd.h>
42 #include <sys/time.h>
43 #include <libsigrok/libsigrok.h>
44 #include "libsigrok-internal.h"
45
46 #define LOG_PREFIX "input/trace32_ad"
47
48 #define CHUNK_SIZE        (4 * 1024 * 1024)
49 #define MAX_POD_COUNT     12
50 #define HEADER_SIZE       80
51
52 #define TIMESTAMP_RESOLUTION ((double)0.000000000078125) /* 0.078125 ns */
53
54 /*
55  * The resolution equals a sampling freq of 12.8 GHz. That's a bit high
56  * for inter-record sample generation, so we scale it down to 200 MHz
57  * for now. That way, the scaling factor becomes 32.
58  */
59 #define DEFAULT_SAMPLERATE 200
60
61 enum {
62         AD_FORMAT_BINHDR = 1, /* Binary header, binary data, textual setup info */
63         AD_FORMAT_TXTHDR      /* Textual header, binary data */
64 };
65
66 enum {
67         AD_DEVICE_PI = 1, /* Data recorded by LA-7940 PowerIntegrator or */
68                           /* LA-394x PowerIntegrator II. */
69         AD_DEVICE_IPROBE  /* Data recorded by LA-769x PowerTrace II IProbe. */
70         /* Missing file format info for LA-793x ICD PowerProbe */
71         /* Missing file format info for LA-4530 uTrace analog probe */
72 };
73
74 enum {
75         AD_MODE_250MHZ = 0,
76         AD_MODE_500MHZ = 1
77 };
78
79 enum {
80         AD_COMPR_NONE  = 0, /* File created with /NOCOMPRESS */
81         AD_COMPR_QCOMP = 6, /* File created with /COMPRESS or /QUICKCOMPRESS */
82 };
83
84 struct context {
85         gboolean meta_sent;
86         gboolean header_read, records_read, trigger_sent;
87         char format, device, record_mode, compression;
88         char pod_status[MAX_POD_COUNT];
89         struct sr_channel *channels[MAX_POD_COUNT][17]; /* 16 + CLK */
90         uint64_t trigger_timestamp;
91         uint32_t record_size, record_count, cur_record;
92         int32_t last_record;
93         uint64_t samplerate;
94         double timestamp_scale;
95         GString *out_buf;
96 };
97
98 static int process_header(GString *buf, struct context *inc);
99 static void create_channels(struct sr_input *in);
100
101 static char get_pod_name_from_id(int id)
102 {
103         switch (id) {
104         case 0:  return 'A';
105         case 1:  return 'B';
106         case 2:  return 'C';
107         case 3:  return 'D';
108         case 4:  return 'E';
109         case 5:  return 'F';
110         case 6:  return 'J';
111         case 7:  return 'K';
112         case 8:  return 'L';
113         case 9:  return 'M';
114         case 10: return 'N';
115         case 11: return 'O';
116         default:
117                 sr_err("get_pod_name_from_id() called with invalid ID %d!", id);
118         }
119         return 'X';
120 }
121
122 static int init(struct sr_input *in, GHashTable *options)
123 {
124         struct context *inc;
125         int pod;
126         char id[17];
127
128         in->sdi = g_malloc0(sizeof(struct sr_dev_inst));
129         in->priv = g_malloc0(sizeof(struct context));
130
131         inc = in->priv;
132
133         /* Calculate the desired timestamp scaling factor. */
134         inc->samplerate = 1000000 *
135                 g_variant_get_uint32(g_hash_table_lookup(options, "samplerate"));
136
137         inc->timestamp_scale = ((1 / TIMESTAMP_RESOLUTION) / (double)inc->samplerate);
138
139         /* Enable the pods the user chose to see. */
140         for (pod = 0; pod < MAX_POD_COUNT; pod++) {
141                 g_snprintf(id, sizeof(id), "pod%c", get_pod_name_from_id(pod));
142                 if (g_variant_get_boolean(g_hash_table_lookup(options, id)))
143                         inc->pod_status[pod] = 1;
144         }
145
146         create_channels(in);
147         if (g_slist_length(in->sdi->channels) == 0) {
148                 sr_err("No pods were selected and thus no channels created, aborting.");
149                 g_free(in->priv);
150                 g_free(in->sdi);
151                 return SR_ERR;
152         }
153
154         inc->out_buf = g_string_sized_new(CHUNK_SIZE);
155
156         return SR_OK;
157 }
158
159 static int format_match(GHashTable *metadata, unsigned int *confidence)
160 {
161         GString *buf;
162         int rc;
163
164         buf = g_hash_table_lookup(metadata, GINT_TO_POINTER(SR_INPUT_META_HEADER));
165         rc = process_header(buf, NULL);
166
167         if (rc != SR_OK)
168                 return rc;
169         *confidence = 10;
170
171         return SR_OK;
172 }
173
174 static int process_header(GString *buf, struct context *inc)
175 {
176         char *format_name, *format_name_sig;
177         int i, record_size, device_id;
178
179         /*
180          * 00-31 (0x00-1F) file format name
181          * 32-39 (0x20-27) trigger timestamp       u64
182          * 40-47 (0x28-2F) unused
183          * 48    (0x30)    compression
184          * 49-53 (0x31-35) ??
185          *       50 (0x32) 0x00 (PI), 0x01 (iprobe)
186          * 54    (0x36)    0x08 (PI 250/500), 0x0A (iprobe 250)
187          * 55    (0x37)    0x00 (250), 0x01 (500)
188          * 56    (0x38)    record size             u8
189          * 57-59 (0x39-3B) const 0x00
190          * 60-63 (0x3C-3F) number of records       u32
191          * 64-67 (0x40-43) id of last record       s32
192          * 68-77 (0x44-4D) ??
193          *       71 (0x47) const 0x80=128
194          *       72 (0x48) const 0x01
195          * 78-79 (0x4E-4F) ??
196          */
197
198         /* Note: inc is off-limits until we check whether it's a valid pointer. */
199
200         format_name = g_strndup(buf->str, 32);
201
202         /* File format name ends on 0x20/0x1A, let's remove both. */
203         for (i = 1; i < 31; i++) {
204                 if (format_name[i] == 0x1A) {
205                         format_name[i - 1] = 0;
206                         format_name[i] = 0;
207                 }
208         }
209         g_strchomp(format_name); /* This is for additional padding spaces. */
210
211         format_name_sig = g_strndup(format_name, 5);
212
213         /* Desired file formats either start with digit+space or "trace32". */
214         if (g_strcmp0(format_name_sig, "trace32")) {
215                 if (inc)
216                         inc->format = AD_FORMAT_BINHDR;
217         } else if (g_ascii_isdigit(format_name[0]) && (format_name[1] == 0x20)) {
218                 if (inc)
219                         inc->format = AD_FORMAT_TXTHDR;
220                 g_free(format_name_sig);
221                 g_free(format_name);
222                 if (inc)
223                         sr_err("This format isn't implemented yet, aborting.");
224                 else
225                         sr_dbg("Not a supported trace32 input file.");
226                 return SR_ERR;
227         } else {
228                 g_free(format_name_sig);
229                 g_free(format_name);
230                 if (inc)
231                         sr_err("Don't know this file format, aborting.");
232                 else
233                         sr_dbg("Not a trace32 input file.");
234                 return SR_ERR;
235         }
236
237         sr_dbg("File says it's \"%s\"", format_name);
238
239         record_size = R8(buf->str + 56);
240         device_id = 0;
241
242         if (g_strcmp0(format_name, "trace32 power integrator data") == 0) {
243                 if (record_size == 28 || record_size == 45)
244                         device_id = AD_DEVICE_PI;
245         } else if (g_strcmp0(format_name, "trace32 iprobe data") == 0) {
246                 if (record_size == 11)
247                         device_id = AD_DEVICE_IPROBE;
248         }
249
250         if (!device_id) {
251                 g_free(format_name_sig);
252                 g_free(format_name);
253                 sr_err("Don't know how to handle this file with record size %d.",
254                         record_size);
255                 return SR_ERR;
256         }
257
258         g_free(format_name_sig);
259         g_free(format_name);
260
261         /* Stop processing the header if we just want to identify the file. */
262         if (!inc)
263                 return SR_OK;
264
265         inc->device       = device_id;
266         inc->trigger_timestamp = RL64(buf->str + 32);
267         inc->compression  = R8(buf->str + 48); /* Maps to the enum. */
268         inc->record_mode  = R8(buf->str + 55); /* Maps to the enum. */
269         inc->record_size  = record_size;
270         inc->record_count = RL32(buf->str + 60);
271         inc->last_record  = RL32S(buf->str + 64);
272
273         sr_dbg("Trigger occured at %lf s.",
274                 inc->trigger_timestamp * TIMESTAMP_RESOLUTION);
275         sr_dbg("File contains %d records: first one is %d, last one is %d.",
276                 inc->record_count, (inc->last_record - inc->record_count + 1),
277                 inc->last_record);
278
279         /* Check if we can work with this compression. */
280         if (inc->compression != AD_COMPR_NONE) {
281                 sr_err("File uses unsupported compression (0x%02X), can't continue.",
282                         inc->compression);
283                 return SR_ERR;
284         }
285
286         inc->header_read = TRUE;
287
288         return SR_OK;
289 }
290
291 static void create_channels(struct sr_input *in)
292 {
293         struct context *inc;
294         int pod, channel, chan_id;
295         char name[8];
296
297         inc = in->priv;
298         chan_id = 0;
299
300         for (pod = 0; pod < MAX_POD_COUNT; pod++) {
301                 if (!inc->pod_status[pod])
302                         continue;
303
304                 for (channel = 0; channel < 16; channel++) {
305                         snprintf(name, sizeof(name), "%c%d", get_pod_name_from_id(pod), channel);
306                         inc->channels[pod][channel] =
307                                 sr_channel_new(in->sdi, chan_id, SR_CHANNEL_LOGIC, TRUE, name);
308                         chan_id++;
309                 }
310
311                 snprintf(name, sizeof(name), "CLK%c", get_pod_name_from_id(pod));
312                 inc->channels[pod][16] =
313                         sr_channel_new(in->sdi, chan_id, SR_CHANNEL_LOGIC, TRUE, name);
314                 chan_id++;
315         }
316 }
317
318 static void send_metadata(struct sr_input *in)
319 {
320         struct sr_datafeed_packet packet;
321         struct sr_datafeed_meta meta;
322         struct sr_config *src;
323         struct context *inc;
324
325         inc = in->priv;
326
327         packet.type = SR_DF_META;
328         packet.payload = &meta;
329         src = sr_config_new(SR_CONF_SAMPLERATE, g_variant_new_uint64(inc->samplerate));
330         meta.config = g_slist_append(NULL, src);
331         sr_session_send(in->sdi, &packet);
332         g_slist_free(meta.config);
333         sr_config_free(src);
334
335         inc->meta_sent = TRUE;
336 }
337
338 static void flush_output_buffer(struct sr_input *in)
339 {
340         struct context *inc;
341         struct sr_datafeed_packet packet;
342         struct sr_datafeed_logic logic;
343
344         inc = in->priv;
345
346         if (inc->out_buf->len) {
347                 packet.type = SR_DF_LOGIC;
348                 packet.payload = &logic;
349                 logic.unitsize = (g_slist_length(in->sdi->channels) + 7) / 8;
350                 logic.data = inc->out_buf->str;
351                 logic.length = inc->out_buf->len;
352                 sr_session_send(in->sdi, &packet);
353
354                 g_string_truncate(inc->out_buf, 0);
355         }
356 }
357
358 static void process_record_pi(struct sr_input *in, gsize start)
359 {
360         struct sr_datafeed_packet packet;
361         struct context *inc;
362         uint64_t timestamp, next_timestamp;
363         uint32_t pod_data;
364         char single_payload[12 * 3];
365         GString *buf;
366         int i, pod_count, clk_offset, packet_count, pod;
367         int payload_bit, payload_len, value;
368
369         inc = in->priv;
370         buf = in->buf;
371
372         /*
373          * 00-07 timestamp
374          * 08-09 A15..0
375          * 10-11 B15..0
376          * 12-13 C15..0
377          * 14-15 D15..0
378          * 16-17 E15..0
379          * 18-19 F15..0
380          * 20-23 ??
381          * 24-25 J15..0                         Not present in 500MHz mode
382          * 26-27 K15..0                         Not present in 500MHz mode
383          * 28-29 L15..0                         Not present in 500MHz mode
384          * 30-31 M15..0                         Not present in 500MHz mode
385          * 32-33 N15..0                         Not present in 500MHz mode
386          * 34-35 O15..0                         Not present in 500MHz mode
387          * 36-39 ??                             Not present in 500MHz mode
388          * 40/24 CLKF..A (32=CLKF, .., 1=CLKA)
389          * 41    CLKO..J (32=CLKO, .., 1=CLKJ)  Not present in 500MHz mode
390          * 42/25    ??
391          * 43/26    ??
392          * 44/27    ??
393          */
394
395         timestamp = RL64(buf->str + start);
396
397         if (inc->record_mode == AD_MODE_500MHZ) {
398                 pod_count = 6;
399                 clk_offset = 24;
400         } else {
401                 pod_count = 12;
402                 clk_offset = 40;
403         }
404
405         payload_bit = 0;
406         payload_len = 0;
407         single_payload[0] = 0;
408
409         for (pod = 0; pod < pod_count; pod++) {
410                 if (!inc->pod_status[pod])
411                         continue;
412
413                 switch (pod) {
414                 case 0: /* A */
415                         pod_data = RL16(buf->str + start + 8);
416                         pod_data |= (RL16(buf->str + start + clk_offset) & 1) << 16;
417                         break;
418                 case 1: /* B */
419                         pod_data = RL16(buf->str + start + 10);
420                         pod_data |= (RL16(buf->str + start + clk_offset) & 2) << 15;
421                         break;
422                 case 2: /* C */
423                         pod_data = RL16(buf->str + start + 12);
424                         pod_data |= (RL16(buf->str + start + clk_offset) & 4) << 14;
425                         break;
426                 case 3: /* D */
427                         pod_data = RL16(buf->str + start + 14);
428                         pod_data |= (RL16(buf->str + start + clk_offset) & 8) << 13;
429                         break;
430                 case 4: /* E */
431                         pod_data = RL16(buf->str + start + 16);
432                         pod_data |= (RL16(buf->str + start + clk_offset) & 16) << 12;
433                         break;
434                 case 5: /* F */
435                         pod_data = RL16(buf->str + start + 18);
436                         pod_data |= (RL16(buf->str + start + clk_offset) & 32) << 11;
437                         break;
438                 case 6: /* J */
439                         pod_data = RL16(buf->str + start + 24);
440                         pod_data |= (RL16(buf->str + start + 41) & 1) << 16;
441                         break;
442                 case 7: /* K */
443                         pod_data = RL16(buf->str + start + 26);
444                         pod_data |= (RL16(buf->str + start + 41) & 2) << 15;
445                         break;
446                 case 8: /* L */
447                         pod_data = RL16(buf->str + start + 28);
448                         pod_data |= (RL16(buf->str + start + 41) & 4) << 14;
449                         break;
450                 case 9: /* M */
451                         pod_data = RL16(buf->str + start + 30);
452                         pod_data |= (RL16(buf->str + start + 41) & 8) << 13;
453                         break;
454                 case 10: /* N */
455                         pod_data = RL16(buf->str + start + 32);
456                         pod_data |= (RL16(buf->str + start + 41) & 16) << 12;
457                         break;
458                 case 11: /* O */
459                         pod_data = RL16(buf->str + start + 34);
460                         pod_data |= (RL16(buf->str + start + 41) & 32) << 11;
461                         break;
462                 default:
463                         sr_err("Don't know how to obtain data for pod %d.", pod);
464                 }
465
466                 for (i = 0; i < 17; i++) {
467                         value = (pod_data >> i) & 1;
468                         single_payload[payload_len] |= value << payload_bit;
469
470                         payload_bit++;
471                         if (payload_bit > 7) {
472                                 payload_bit = 0;
473                                 payload_len++;
474                                 single_payload[payload_len] = 0;
475                         }
476                 }
477         }
478
479         /* Make sure that payload_len accounts for any incomplete bytes used. */
480         if (payload_bit)
481                 payload_len++;
482
483         i = (g_slist_length(in->sdi->channels) + 7) / 8;
484         if (payload_len != i) {
485                 sr_err("Payload unit size is %d but should be %d!", payload_len, i);
486                 return;
487         }
488
489         if (timestamp == inc->trigger_timestamp && !inc->trigger_sent) {
490                 sr_dbg("Trigger @%lf s, record #%d.",
491                         timestamp * TIMESTAMP_RESOLUTION, inc->cur_record);
492
493                 packet.type = SR_DF_TRIGGER;
494                 packet.payload = NULL;
495                 sr_session_send(in->sdi, &packet);
496                 inc->trigger_sent = TRUE;
497         }
498
499         /* Is this the last record in the file? */
500         if (inc->cur_record == inc->record_count - 1) {
501                 /* It is, so send the last sample data only once. */
502                 g_string_append_len(inc->out_buf, single_payload, payload_len);
503         } else {
504                 /* It's not, so fill the time gap by sending lots of data. */
505                 next_timestamp = RL64(buf->str + start + inc->record_size);
506                 packet_count = (int)(next_timestamp - timestamp) / inc->timestamp_scale;
507
508                 /* Make sure we send at least one data set. */
509                 if (packet_count == 0)
510                         packet_count = 1;
511
512                 for (i = 0; i < packet_count; i++)
513                         g_string_append_len(inc->out_buf, single_payload, payload_len);
514         }
515
516         if (inc->out_buf->len >= CHUNK_SIZE)
517                 flush_output_buffer(in);
518 }
519
520 static void process_record_iprobe(struct sr_input *in, gsize start)
521 {
522         struct sr_datafeed_packet packet;
523         struct context *inc;
524         uint64_t timestamp, next_timestamp;
525         char single_payload[3];
526         int i, payload_len, packet_count;
527
528         inc = in->priv;
529
530         /*
531          * 00-07 timestamp
532          * 08-09 IP15..0
533          * 10    CLK
534          */
535
536         timestamp = RL64(in->buf->str + start);
537         single_payload[0] = R8(in->buf->str + start + 8);
538         single_payload[1] = R8(in->buf->str + start + 9);
539         single_payload[2] = R8(in->buf->str + start + 10) & 1;
540         payload_len = 3;
541
542         if (timestamp == inc->trigger_timestamp && !inc->trigger_sent) {
543                 sr_dbg("Trigger @%lf s, record #%d.",
544                         timestamp * TIMESTAMP_RESOLUTION, inc->cur_record);
545
546                 packet.type = SR_DF_TRIGGER;
547                 packet.payload = NULL;
548                 sr_session_send(in->sdi, &packet);
549                 inc->trigger_sent = TRUE;
550         }
551
552         /* Is this the last record in the file? */
553         if (inc->cur_record == inc->record_count - 1) {
554                 /* It is, so send the last sample data only once. */
555                 g_string_append_len(inc->out_buf, single_payload, payload_len);
556         } else {
557                 /* It's not, so fill the time gap by sending lots of data. */
558                 next_timestamp = RL64(in->buf->str + start + inc->record_size);
559                 packet_count = (int)(next_timestamp - timestamp) / inc->timestamp_scale;
560
561                 /* Make sure we send at least one data set. */
562                 if (packet_count == 0)
563                         packet_count = 1;
564
565                 for (i = 0; i < packet_count; i++)
566                         g_string_append_len(inc->out_buf, single_payload, payload_len);
567         }
568
569         if (inc->out_buf->len >= CHUNK_SIZE)
570                 flush_output_buffer(in);
571 }
572
573 static void process_practice_token(struct sr_input *in, char *cmd_token)
574 {
575         struct context *inc;
576         char **tokens;
577         char chan_suffix[2], chan_name[33];
578         char *s1, *s2;
579         int pod, ch;
580         struct sr_channel *channel;
581
582         inc = in->priv;
583
584         /*
585          * Commands of interest (I may also be IPROBE):
586          *
587          * I.TWIDTH
588          * I.TPREDELAY
589          * I.TDELAY
590          * I.TYSNC.SELECT I.A0 HIGH
591          * NAME.SET <port.chan> <name> <+/-> ...
592          */
593
594         if (!cmd_token)
595                 return;
596
597         if (cmd_token[0] == 0)
598                 return;
599
600         tokens = g_strsplit(cmd_token, " ", 0);
601
602         if (!tokens)
603                 return;
604
605         if (g_strcmp0(tokens[0], "NAME.SET") == 0) {
606                 /* Let the user know when the channel has been inverted. */
607                 /* This *should* be token #3 but there's an additonal space, making it #4. */
608                 chan_suffix[0] = 0;
609                 chan_suffix[1] = 0;
610                 if (tokens[4]) {
611                         if (tokens[4][0] == '-')
612                                 chan_suffix[0] = '-'; /* This is the way PowerView shows it. */
613                 }
614
615                 /*
616                  * Command is using structure "NAME.SET I.A00 I.XYZ" or
617                  * "NAME.SET IP.00 IP.XYZ", depending on the device used.
618                  * Let's get strings with the I./IP. from both tokens removed.
619                  */
620                 s1 = g_strstr_len(tokens[1], -1, ".") + 1;
621                 s2 = g_strstr_len(tokens[2], -1, ".") + 1;
622
623                 if (g_strcmp0(s1, "CLK") == 0) {
624                         /* CLK for iprobe */
625                         pod = 0;
626                         ch = 16;
627                 } else if ((strlen(s1) == 4) && g_ascii_isupper(s1[3])) {
628                         /* CLKA/B/J/K for PowerIntegrator */
629                         pod = s1[3] - (char)'A';
630                         ch = 16;
631                 } else if (g_ascii_isupper(s1[0])) {
632                         /* A00 for PowerIntegrator */
633                         pod = s1[0] - (char)'A';
634                         ch = atoi(s1 + 1);
635                 } else {
636                         /* 00 for iprobe */
637                         pod = 0;
638                         ch = atoi(s1);
639                 }
640
641                 channel = inc->channels[pod][ch];
642                 g_snprintf(chan_name, sizeof(chan_name), "%s%s", s2, chan_suffix);
643
644                 sr_dbg("Changing channel name for %s to %s.", s1, chan_name);
645                 sr_dev_channel_name_set(channel, chan_name);
646         }
647
648         g_strfreev(tokens);
649 }
650
651 static void process_practice(struct sr_input *in)
652 {
653         char delimiter[3];
654         char **tokens, *token;
655         int i;
656
657         /* Gather all input data until we see the end marker. */
658         if (in->buf->str[in->buf->len - 1] != 0x29)
659                 return;
660
661         delimiter[0] = 0x0A;
662         delimiter[1] = ' ';
663         delimiter[2] = 0;
664
665         tokens = g_strsplit(in->buf->str, delimiter, 0);
666
667         /* Special case: first token contains the start marker, too. Skip it. */
668         token = tokens[0];
669         for (i = 0; token[i]; i++) {
670                 if (token[i] == ' ')
671                         process_practice_token(in, token + i + 1);
672         }
673
674         for (i = 1; tokens[i]; i++)
675                 process_practice_token(in, tokens[i]);
676
677         g_strfreev(tokens);
678
679         g_string_erase(in->buf, 0, in->buf->len);
680 }
681
682 static int process_buffer(struct sr_input *in)
683 {
684         struct context *inc;
685         int i, chunk_size, res;
686
687         inc = in->priv;
688
689         if (!inc->header_read) {
690                 res = process_header(in->buf, inc);
691                 g_string_erase(in->buf, 0, HEADER_SIZE);
692                 if (res != SR_OK)
693                         return res;
694         }
695
696         if (!inc->meta_sent) {
697                 std_session_send_df_header(in->sdi);
698                 send_metadata(in);
699         }
700
701         if (!inc->records_read) {
702                 /* Cut off at a multiple of the record size. */
703                 chunk_size = ((in->buf->len) / inc->record_size) * inc->record_size;
704
705                 /* There needs to be at least one more record process_record() can peek into. */
706                 chunk_size -= inc->record_size;
707
708                 for (i = 0; (i < chunk_size) && (!inc->records_read); i += inc->record_size) {
709                         switch (inc->device) {
710                         case AD_DEVICE_PI:
711                                 process_record_pi(in, i);
712                                 break;
713                         case AD_DEVICE_IPROBE:
714                                 process_record_iprobe(in, i);
715                                 break;
716                         default:
717                                 sr_err("Trying to process records for unknown device!");
718                                 return SR_ERR;
719                         }
720
721                         inc->cur_record++;
722                         if (inc->cur_record == inc->record_count)
723                                 inc->records_read = TRUE;
724                 }
725
726                 g_string_erase(in->buf, 0, i);
727         }
728
729         if (inc->records_read) {
730                 /* Read practice commands that configure the setup. */
731                 process_practice(in);
732         }
733
734         return SR_OK;
735 }
736
737 static int receive(struct sr_input *in, GString *buf)
738 {
739         g_string_append_len(in->buf, buf->str, buf->len);
740
741         if (!in->sdi_ready) {
742                 /* sdi is ready, notify frontend. */
743                 in->sdi_ready = TRUE;
744                 return SR_OK;
745         }
746
747         return process_buffer(in);
748 }
749
750 static int end(struct sr_input *in)
751 {
752         struct context *inc;
753         int ret;
754
755         inc = in->priv;
756
757         if (in->sdi_ready)
758                 ret = process_buffer(in);
759         else
760                 ret = SR_OK;
761
762         flush_output_buffer(in);
763
764         if (inc->meta_sent)
765                 std_session_send_df_end(in->sdi);
766
767         return ret;
768 }
769
770 static int reset(struct sr_input *in)
771 {
772         struct context *inc = in->priv;
773
774         inc->meta_sent = FALSE;
775         inc->header_read = FALSE;
776         inc->records_read = FALSE;
777         inc->trigger_sent = FALSE;
778         inc->cur_record = 0;
779
780         g_string_truncate(in->buf, 0);
781
782         return SR_OK;
783 }
784
785 static struct sr_option options[] = {
786         { "podA", "Import pod A / iprobe",
787                 "Create channels and data for pod A / iprobe", NULL, NULL },
788
789         { "podB", "Import pod B", "Create channels and data for pod B", NULL, NULL },
790         { "podC", "Import pod C", "Create channels and data for pod C", NULL, NULL },
791         { "podD", "Import pod D", "Create channels and data for pod D", NULL, NULL },
792         { "podE", "Import pod E", "Create channels and data for pod E", NULL, NULL },
793         { "podF", "Import pod F", "Create channels and data for pod F", NULL, NULL },
794         { "podJ", "Import pod J", "Create channels and data for pod J", NULL, NULL },
795         { "podK", "Import pod K", "Create channels and data for pod K", NULL, NULL },
796         { "podL", "Import pod L", "Create channels and data for pod L", NULL, NULL },
797         { "podM", "Import pod M", "Create channels and data for pod M", NULL, NULL },
798         { "podN", "Import pod N", "Create channels and data for pod N", NULL, NULL },
799         { "podO", "Import pod O", "Create channels and data for pod O", NULL, NULL },
800
801         { "samplerate", "Reduced sample rate (MHz)", "Reduce the original sample rate of 12.8 GHz to the specified sample rate in MHz", NULL, NULL },
802
803         ALL_ZERO
804 };
805
806 static const struct sr_option *get_options(void)
807 {
808         if (!options[0].def) {
809                 options[0].def = g_variant_ref_sink(g_variant_new_boolean(TRUE));
810                 options[1].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
811                 options[2].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
812                 options[3].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
813                 options[4].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
814                 options[5].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
815                 options[6].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
816                 options[7].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
817                 options[8].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
818                 options[9].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
819                 options[10].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
820                 options[11].def = g_variant_ref_sink(g_variant_new_boolean(FALSE));
821                 options[12].def = g_variant_ref_sink(g_variant_new_uint32(DEFAULT_SAMPLERATE));
822         }
823
824         return options;
825 }
826
827 SR_PRIV struct sr_input_module input_trace32_ad = {
828         .id = "trace32_ad",
829         .name = "Trace32_ad",
830         .desc = "Lauterbach Trace32 logic analyzer data",
831         .exts = (const char*[]){"ad", NULL},
832         .options = get_options,
833         .metadata = { SR_INPUT_META_HEADER | SR_INPUT_META_REQUIRED },
834         .format_match = format_match,
835         .init = init,
836         .receive = receive,
837         .end = end,
838         .reset = reset,
839 };