]> sigrok.org Git - libsigrok.git/blob - src/hardware/lecroy-xstream/protocol.c
lecroy-xstream: Fix compiler warnings
[libsigrok.git] / src / hardware / lecroy-xstream / protocol.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2017 Sven Schnelle <svens@stackframe.org>
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 <config.h>
21 #include <math.h>
22 #include <stdlib.h>
23 #include "scpi.h"
24 #include "protocol.h"
25
26 SR_PRIV void lecroy_queue_logic_data(struct dev_context *devc,
27                                   size_t group, GByteArray *pod_data);
28 SR_PRIV void lecroy_send_logic_packet(struct sr_dev_inst *sdi,
29                                    struct dev_context *devc);
30 SR_PRIV void lecroy_cleanup_logic_data(struct dev_context *devc);
31
32 struct lecroy_wavedesc_2_x {
33         uint16_t comm_type;
34         uint16_t comm_order; /* 1 - little endian */
35         uint32_t wave_descriptor_length;
36         uint32_t user_text_len;
37         uint32_t res_desc1;
38         uint32_t trigtime_array_length;
39         uint32_t ris_time1_array_length;
40         uint32_t res_array1;
41         uint32_t wave_array1_length;
42         uint32_t wave_array2_length;
43         uint32_t wave_array3_length;
44         uint32_t wave_array4_length;
45         char instrument_name[16];
46         uint32_t instrument_number;
47         char trace_label[16];
48         uint32_t reserved;
49         uint32_t wave_array_count;
50         uint32_t points_per_screen;
51         uint32_t first_valid_point;
52         uint32_t last_valid_point;
53         uint32_t first_point;
54         uint32_t sparsing_factor;
55         uint32_t segment_index;
56         uint32_t subarray_count;
57         uint32_t sweeps_per_acq;
58         uint16_t points_per_pair;
59         uint16_t pair_offset;
60         float vertical_gain;
61         float vertical_offset;
62         float max_value;
63         float min_value;
64         uint16_t nominal_bits;
65         uint16_t nom_subarray_count;
66         float horiz_interval;
67         double horiz_offset;
68         double pixel_offset;
69         char vertunit[48];
70         char horunit[48];
71         uint32_t reserved1;
72         double trigger_time;
73 } __attribute__((packed));
74
75 struct lecroy_wavedesc {
76         char descriptor_name[16];
77         char template_name[16];
78         union {
79                 struct lecroy_wavedesc_2_x version_2_x;
80         };
81 } __attribute__((packed));
82
83 static const uint32_t lecroy_devopts[] = {
84         SR_CONF_OSCILLOSCOPE,
85         SR_CONF_LIMIT_FRAMES | SR_CONF_GET | SR_CONF_SET,
86         SR_CONF_TRIGGER_SOURCE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
87         SR_CONF_TIMEBASE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
88         SR_CONF_NUM_HDIV | SR_CONF_GET,
89         SR_CONF_TRIGGER_SLOPE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
90         SR_CONF_HORIZ_TRIGGERPOS | SR_CONF_GET | SR_CONF_SET,
91         SR_CONF_SAMPLERATE | SR_CONF_GET,
92 };
93
94 static const uint32_t lecroy_analog_devopts[] = {
95         SR_CONF_NUM_VDIV | SR_CONF_GET,
96         SR_CONF_COUPLING | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
97         SR_CONF_VDIV | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
98 };
99
100 static const char *lecroy_coupling_options[] = {
101         "A1M", // AC with 1 MOhm termination
102         "D50",  // DC with 50 Ohm termination
103         "D1M", // DC with 1 MOhm termination
104         "GND",
105         "OVL",
106         NULL,
107 };
108
109 static const char *scope_trigger_slopes[] = {
110         "POS",
111         "NEG",
112         NULL,
113 };
114
115 static const char *lecroy_xstream_trigger_sources[] = {
116         "C1",
117         "C2",
118         "C3",
119         "C4",
120         "LINE",
121         "EXT",
122         NULL,
123 };
124
125 static const struct sr_rational lecroy_timebases[] = {
126         /* picoseconds */
127         { 20, 1000000000000 },
128         { 50, 1000000000000 },
129         { 100, 1000000000000 },
130         { 200, 1000000000000 },
131         { 500, 1000000000000 },
132         /* nanoseconds */
133         { 1, 1000000000 },
134         { 2, 1000000000 },
135         { 5, 1000000000 },
136         { 10, 1000000000 },
137         { 20, 1000000000 },
138         { 50, 1000000000 },
139         { 100, 1000000000 },
140         { 200, 1000000000 },
141         { 500, 1000000000 },
142         /* microseconds */
143         { 1, 1000000 },
144         { 2, 1000000 },
145         { 5, 1000000 },
146         { 10, 1000000 },
147         { 20, 1000000 },
148         { 50, 1000000 },
149         { 100, 1000000 },
150         { 200, 1000000 },
151         { 500, 1000000 },
152         /* milliseconds */
153         { 1, 1000 },
154         { 2, 1000 },
155         { 5, 1000 },
156         { 10, 1000 },
157         { 20, 1000 },
158         { 50, 1000 },
159         { 100, 1000 },
160         { 200, 1000 },
161         { 500, 1000 },
162         /* seconds */
163         { 1, 1 },
164         { 2, 1 },
165         { 5, 1 },
166         { 10, 1 },
167         { 20, 1 },
168         { 50, 1 },
169         { 100, 1 },
170         { 200, 1 },
171         { 500, 1 },
172         { 1000, 1 },
173 };
174
175 static const struct sr_rational lecroy_vdivs[] = {
176         /* millivolts */
177         { 1, 1000 },
178         { 2, 1000 },
179         { 5, 1000 },
180         { 10, 1000 },
181         { 20, 1000 },
182         { 50, 1000 },
183         { 100, 1000 },
184         { 200, 1000 },
185         { 500, 1000 },
186         /* volts */
187         { 1, 1 },
188         { 2, 1 },
189         { 5, 1 },
190         { 10, 1 },
191         { 20, 1 },
192         { 50, 1 },
193 };
194
195 static const char *scope_analog_channel_names[] = {
196         "CH1",
197         "CH2",
198         "CH3",
199         "CH4",
200 };
201
202 static const struct scope_config scope_models[] = {
203         {
204                 .name = { "WP7000", "WP7100", "WP7200", "WP7300" },
205
206                 .analog_channels = 4,
207                 .analog_names = &scope_analog_channel_names,
208
209                 .devopts = &lecroy_devopts,
210                 .num_devopts = ARRAY_SIZE(lecroy_devopts),
211
212                 .analog_devopts = &lecroy_analog_devopts,
213                 .num_analog_devopts = ARRAY_SIZE(lecroy_analog_devopts),
214
215                 .coupling_options = &lecroy_coupling_options,
216                 .trigger_sources = &lecroy_xstream_trigger_sources,
217                 .trigger_slopes = &scope_trigger_slopes,
218
219                 .timebases = lecroy_timebases,
220                 .num_timebases = ARRAY_SIZE(lecroy_timebases),
221
222                 .vdivs = lecroy_vdivs,
223                 .num_vdivs = ARRAY_SIZE(lecroy_vdivs),
224
225                 .num_xdivs = 10,
226                 .num_ydivs = 8,
227         },
228 };
229
230 static void scope_state_dump(const struct scope_config *config,
231                              struct scope_state *state)
232 {
233         unsigned int i;
234         char *tmp;
235
236         for (i = 0; i < config->analog_channels; i++) {
237                 tmp = sr_voltage_string(config->vdivs[state->analog_channels[i].vdiv].p,
238                                         config->vdivs[state->analog_channels[i].vdiv].q);
239                 sr_info("State of analog channel %d -> %s : %s (coupling) %s (vdiv) %2.2e (offset)",
240                         i + 1, state->analog_channels[i].state ? "On" : "Off",
241                         (*config->coupling_options)[state->analog_channels[i].coupling],
242                         tmp, state->analog_channels[i].vertical_offset);
243         }
244
245         tmp = sr_period_string_f(1.0/(((float)config->timebases[state->timebase].p) /
246                                       ((float)config->timebases[state->timebase].q)), 0);
247         sr_info("Current timebase: %s", tmp);
248         g_free(tmp);
249
250         tmp = sr_samplerate_string(state->sample_rate);
251         sr_info("Current samplerate: %s", tmp);
252         g_free(tmp);
253
254         sr_info("Current trigger: %s (source), %s (slope) %.2f (offset)",
255                 (*config->trigger_sources)[state->trigger_source],
256                 (*config->trigger_slopes)[state->trigger_slope],
257                 state->horiz_triggerpos);
258 }
259
260 static int scope_state_get_array_option(const char *resp,
261                                         const char *(*array)[],
262                                         int *result)
263 {
264         unsigned int i;
265
266         for (i = 0; (*array)[i]; i++) {
267                 if (!g_strcmp0(resp, (*array)[i])) {
268                         *result = i;
269                         return SR_OK;
270                 }
271         }
272
273         return SR_ERR;
274 }
275
276 /**
277  * This function takes a value of the form "2.000E-03" and returns the index
278  * of an array where a matching pair was found.
279  *
280  * @param value The string to be parsed.
281  * @param array The array of s/f pairs.
282  * @param array_len The number of pairs in the array.
283  * @param result The index at which a matching pair was found.
284  *
285  * @return SR_ERR on any parsing error, SR_OK otherwise.
286  */
287 static int array_float_get(gchar *value, const struct sr_rational *aval,
288                 int array_len, unsigned int *result)
289 {
290         struct sr_rational rval;
291
292         if (sr_parse_rational(value, &rval) != SR_OK)
293                 return SR_ERR;
294
295         for (int i = 0; i < array_len; i++) {
296                 if (sr_rational_eq(&rval, aval+i)) {
297                         *result = i;
298                         return SR_OK;
299                 }
300         }
301
302         return SR_ERR;
303 }
304
305 static int analog_channel_state_get(struct sr_scpi_dev_inst *scpi,
306                                     const struct scope_config *config,
307                                     struct scope_state *state)
308 {
309         unsigned int i, j;
310         char command[MAX_COMMAND_SIZE];
311         char *tmp_str;
312
313         for (i = 0; i < config->analog_channels; i++) {
314                 g_snprintf(command, sizeof(command), "C%d:TRACE?", i+1);
315
316                 if (sr_scpi_get_bool(scpi, command,
317                                         &state->analog_channels[i].state) != SR_OK)
318                         return SR_ERR;
319
320                 g_snprintf(command, sizeof(command), "C%d:VDIV?", i+1);
321
322                 if (sr_scpi_get_string(scpi, command, &tmp_str) != SR_OK)
323                         return SR_ERR;
324
325                 if (array_float_get(tmp_str, lecroy_vdivs, ARRAY_SIZE(lecroy_vdivs),
326                                     &j) != SR_OK) {
327                         g_free(tmp_str);
328                         sr_err("Could not determine array index for vertical div scale.");
329                         return SR_ERR;
330                 }
331
332                 g_free(tmp_str);
333                 state->analog_channels[i].vdiv = j;
334
335                 g_snprintf(command, sizeof(command), "C%d:OFFSET?", i+1);
336
337                 if (sr_scpi_get_float(scpi, command, &state->analog_channels[i].vertical_offset) != SR_OK)
338                         return SR_ERR;
339
340                 g_snprintf(command, sizeof(command), "C%d:COUPLING?", i+1);
341
342                 if (sr_scpi_get_string(scpi, command, &tmp_str) != SR_OK)
343                         return SR_ERR;
344
345
346                 if (scope_state_get_array_option(tmp_str, config->coupling_options,
347                                          &state->analog_channels[i].coupling) != SR_OK)
348                         return SR_ERR;
349
350                 g_free(tmp_str);
351         }
352
353         return SR_OK;
354 }
355
356 SR_PRIV int lecroy_xstream_update_sample_rate(const struct sr_dev_inst *sdi)
357 {
358         struct dev_context *devc;
359         struct scope_state *state;
360         const struct scope_config *config;
361         float memsize, timediv;
362
363         devc = sdi->priv;
364         state = devc->model_state;
365         config = devc->model_config;
366
367         if (sr_scpi_get_float(sdi->conn, "MEMORY_SIZE?", &memsize) != SR_OK)
368                         return SR_ERR;
369
370         if (sr_scpi_get_float(sdi->conn, "TIME_DIV?", &timediv) != SR_OK)
371                 return SR_ERR;
372
373         state->sample_rate = 1/((timediv * config->num_xdivs) / memsize);
374
375         return SR_OK;
376 }
377
378 SR_PRIV int lecroy_xstream_state_get(struct sr_dev_inst *sdi)
379 {
380         struct dev_context *devc;
381         struct scope_state *state
382 ;       const struct scope_config *config;
383         unsigned int i;
384         char *tmp_str, *tmp_str2, *tmpp, *p, *key;
385         char command[MAX_COMMAND_SIZE];
386         char *trig_source = NULL;
387
388         devc = sdi->priv;
389         config = devc->model_config;
390         state = devc->model_state;
391
392         sr_info("Fetching scope state");
393
394         if (analog_channel_state_get(sdi->conn, config, state) != SR_OK)
395                 return SR_ERR;
396
397         if (sr_scpi_get_string(sdi->conn, "TIME_DIV?", &tmp_str) != SR_OK)
398                 return SR_ERR;
399
400         if (array_float_get(tmp_str, lecroy_timebases, ARRAY_SIZE(lecroy_timebases),
401                             &i) != SR_OK) {
402                 g_free(tmp_str);
403                 sr_err("Could not determine array index for timbase scale.");
404                 return SR_ERR;
405         }
406         g_free(tmp_str);
407         state->timebase = i;
408
409         if (sr_scpi_get_string(sdi->conn, "TRIG_SELECT?", &tmp_str) != SR_OK)
410                 return SR_ERR;
411
412         key = tmpp = NULL;
413         tmp_str2 = tmp_str;
414         i = 0;
415         while((p = strtok_r(tmp_str2, ",", &tmpp))) {
416                 tmp_str2 = NULL;
417                 if (i == 0) {
418                         /* trigger type */
419                 } else if (i & 1) {
420                         key = p;
421                         /* key */
422                 } else if (!(i & 1)) {
423                         if (!strcmp(key, "SR"))
424                                 trig_source = p;
425                 }
426                 i++;
427         }
428
429         if (!trig_source || scope_state_get_array_option(trig_source, config->trigger_sources,
430                                                         &state->trigger_source) != SR_OK)
431                 return SR_ERR;
432
433
434         g_snprintf(command, sizeof(command), "%s:TRIG_SLOPE?", trig_source);
435         if (sr_scpi_get_string(sdi->conn, command, &tmp_str) != SR_OK)
436                 return SR_ERR;
437
438         if (scope_state_get_array_option(tmp_str,
439                 config->trigger_slopes, &state->trigger_slope) != SR_OK)
440                 return SR_ERR;
441
442         if (sr_scpi_get_float(sdi->conn, "TRIG_DELAY?", &state->horiz_triggerpos) != SR_OK)
443                 return SR_ERR;
444
445         if (lecroy_xstream_update_sample_rate(sdi) != SR_OK)
446                 return SR_ERR;
447
448         sr_info("Fetching finished.");
449
450         scope_state_dump(config, state);
451
452         return SR_OK;
453 }
454
455 static struct scope_state *scope_state_new(const struct scope_config *config)
456 {
457         struct scope_state *state;
458
459         state = g_malloc0(sizeof(struct scope_state));
460         state->analog_channels = g_malloc0_n(config->analog_channels,
461                         sizeof(struct analog_channel_state));
462         return state;
463 }
464
465 SR_PRIV void lecroy_xstream_state_free(struct scope_state *state)
466 {
467         g_free(state->analog_channels);
468         g_free(state);
469 }
470
471 SR_PRIV int lecroy_xstream_init_device(struct sr_dev_inst *sdi)
472 {
473         char command[MAX_COMMAND_SIZE];
474         int model_index;
475         unsigned int i, j;
476         struct sr_channel *ch;
477         struct dev_context *devc;
478         gboolean channel_enabled;
479
480         devc = sdi->priv;
481         model_index = -1;
482
483         /* Find the exact model. */
484         for (i = 0; i < ARRAY_SIZE(scope_models); i++) {
485                 for (j = 0; scope_models[i].name[j]; j++) {
486                         if (!strcmp(sdi->model, scope_models[i].name[j])) {
487                                 model_index = i;
488                                 break;
489                         }
490                 }
491                 if (model_index != -1)
492                         break;
493         }
494
495         if (model_index == -1) {
496                 sr_dbg("Unsupported LECROY device.");
497                 return SR_ERR_NA;
498         }
499
500         devc->analog_groups = g_malloc0(sizeof(struct sr_channel_group*) *
501                                         scope_models[model_index].analog_channels);
502
503         /* Add analog channels. */
504         for (i = 0; i < scope_models[model_index].analog_channels; i++) {
505
506                 g_snprintf(command, sizeof(command), "C%d:TRACE?", i+1);
507
508                 if (sr_scpi_get_bool(sdi->conn, command, &channel_enabled) != SR_OK)
509                         return SR_ERR;
510
511                 g_snprintf(command, sizeof(command), "C%d:VDIV?", i+1);
512
513                 ch = sr_channel_new(sdi, i, SR_CHANNEL_ANALOG, channel_enabled,
514                            (*scope_models[model_index].analog_names)[i]);
515
516                 devc->analog_groups[i] = g_malloc0(sizeof(struct sr_channel_group));
517
518                 devc->analog_groups[i]->name = g_strdup(
519                         (char *)(*scope_models[model_index].analog_names)[i]);
520                 devc->analog_groups[i]->channels = g_slist_append(NULL, ch);
521
522                 sdi->channel_groups = g_slist_append(sdi->channel_groups,
523                                                    devc->analog_groups[i]);
524         }
525
526         devc->model_config = &scope_models[model_index];
527         devc->frame_limit = 0;
528
529         if (!(devc->model_state = scope_state_new(devc->model_config)))
530                 return SR_ERR_MALLOC;
531
532         /* Set the desired response mode. */
533         sr_scpi_send(sdi->conn, "COMM_HEADER OFF,WORD,BIN");
534
535         return SR_OK;
536 }
537
538 static int lecroy_waveform_2_x_to_analog(GByteArray *data,
539                                          struct lecroy_wavedesc *desc,
540                                          struct sr_datafeed_analog *analog)
541 {
542         struct sr_analog_encoding *encoding = analog->encoding;
543         struct sr_analog_meaning *meaning = analog->meaning;
544         struct sr_analog_spec *spec = analog->spec;
545         float *data_float;
546         int16_t *waveform_data;
547         unsigned int i, num_samples;
548
549         data_float  = g_malloc(desc->version_2_x.wave_array_count * sizeof(float));
550         num_samples = desc->version_2_x.wave_array_count;
551
552         waveform_data = (int16_t *)(data->data +
553                                     + desc->version_2_x.wave_descriptor_length
554                                     + desc->version_2_x.user_text_len);
555
556         for(i = 0; i  < num_samples; i++)
557                 data_float[i] = (float)waveform_data[i]
558                         * desc->version_2_x.vertical_gain
559                         + desc->version_2_x.vertical_offset;
560
561
562         analog->data = data_float;
563         analog->num_samples = num_samples;
564
565         encoding->unitsize = sizeof(float);
566         encoding->is_signed = TRUE;
567         encoding->is_float = TRUE;
568         encoding->is_bigendian = FALSE;
569         encoding->scale.p = 1;
570         encoding->scale.q = 1;
571         encoding->offset.p = 0;
572         encoding->offset.q = 1;
573
574         encoding->digits = 6;
575         encoding->is_digits_decimal = FALSE;
576
577         if (strcmp(desc->version_2_x.vertunit, "A")) {
578                 meaning->mq = SR_MQ_CURRENT;
579                 meaning->unit = SR_UNIT_AMPERE;
580         } else {
581                 /* default to voltage */
582                 meaning->mq = SR_MQ_VOLTAGE;
583                 meaning->unit = SR_UNIT_VOLT;
584         }
585
586         meaning->mqflags = 0;
587         spec->spec_digits = 3;
588         return SR_OK;
589 }
590
591 static int lecroy_waveform_to_analog(GByteArray *data,
592                                      struct sr_datafeed_analog *analog)
593 {
594         struct lecroy_wavedesc *desc;
595
596         if (data->len < sizeof(struct lecroy_wavedesc))
597                 return SR_ERR;
598
599         desc = (struct lecroy_wavedesc *)data->data;
600
601         if (!strncmp(desc->template_name, "LECROY_2_2", 16) ||
602             !strncmp(desc->template_name, "LECROY_2_3", 16)) {
603                 return lecroy_waveform_2_x_to_analog(data, desc, analog);
604         }
605
606         sr_err("Waveformat template '%.16s' not supported\n", desc->template_name);
607         return SR_ERR;
608 }
609
610 SR_PRIV int lecroy_xstream_receive_data(int fd, int revents, void *cb_data)
611 {
612         struct sr_channel *ch;
613         struct sr_dev_inst *sdi;
614         struct dev_context *devc;
615         struct sr_datafeed_packet packet;
616         GByteArray *data;
617         struct sr_datafeed_analog analog;
618         struct sr_analog_encoding encoding;
619         struct sr_analog_meaning meaning;
620         struct sr_analog_spec spec;
621         char buf[8];
622         (void)fd;
623         (void)revents;
624
625         data = NULL;
626
627         if (!(sdi = cb_data))
628                 return TRUE;
629
630         if (!(devc = sdi->priv))
631                 return TRUE;
632
633         ch = devc->current_channel->data;
634
635         /*
636          * Send "frame begin" packet upon reception of data for the
637          * first enabled channel.
638          */
639         if (devc->current_channel == devc->enabled_channels) {
640                 packet.type = SR_DF_FRAME_BEGIN;
641                 sr_session_send(sdi, &packet);
642         }
643
644         if (ch->type != SR_CHANNEL_ANALOG)
645                 return SR_ERR;
646         /*
647          * Pass on the received data of the channel(s).
648          */
649         if (sr_scpi_read_data(sdi->conn, buf, 4) != 4) {
650                 sr_err("reading header failed\n");
651                 return TRUE;
652         }
653
654         if (sr_scpi_get_block(sdi->conn, NULL, &data) != SR_OK) {
655                 if (data)
656                         g_byte_array_free(data, TRUE);
657                 return TRUE;
658         }
659
660         analog.encoding = &encoding;
661         analog.meaning = &meaning;
662         analog.spec = &spec;
663
664         if (lecroy_waveform_to_analog(data, &analog) != SR_OK)
665                 return SR_ERR;
666
667         meaning.channels = g_slist_append(NULL, ch);
668         packet.payload = &analog;
669         packet.type = SR_DF_ANALOG;
670         sr_session_send(sdi, &packet);
671
672         g_byte_array_free(data, TRUE);
673         data = NULL;
674
675         g_slist_free(meaning.channels);
676         g_free(analog.data);
677
678
679         /*
680          * Advance to the next enabled channel. When data for all enabled
681          * channels was received, then flush potentially queued logic data,
682          * and send the "frame end" packet.
683          */
684         if (devc->current_channel->next) {
685                 devc->current_channel = devc->current_channel->next;
686                 lecroy_xstream_request_data(sdi);
687                 return TRUE;
688         }
689
690         packet.type = SR_DF_FRAME_END;
691         sr_session_send(sdi, &packet);
692
693         /*
694          * End of frame was reached. Stop acquisition after the specified
695          * number of frames, or continue reception by starting over at
696          * the first enabled channel.
697          */
698         if (++devc->num_frames == devc->frame_limit) {
699                 sdi->driver->dev_acquisition_stop(sdi);
700         } else {
701                 devc->current_channel = devc->enabled_channels;
702                 lecroy_xstream_request_data(sdi);
703         }
704
705         return TRUE;
706 }