2 * This file is part of the libsigrok project.
4 * Copyright (C) 2012 Martin Ling <martin-git@earth.li>
5 * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
6 * Copyright (C) 2013 Mathias Grimmberger <mgri@zaphod.sax.de>
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.
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.
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/>.
27 #include "libsigrok.h"
28 #include "libsigrok-internal.h"
31 static const int32_t hwopts[] = {
36 static const int32_t hwcaps[] = {
39 SR_CONF_TRIGGER_SOURCE,
40 SR_CONF_TRIGGER_SLOPE,
41 SR_CONF_HORIZ_TRIGGERPOS,
47 static const int32_t analog_hwcaps[] = {
54 static const uint64_t timebases[][2] = {
93 /* { 1000, 1 }, Confuses other code? */
96 static const uint64_t vdivs[][2] = {
116 #define NUM_TIMEBASE ARRAY_SIZE(timebases)
117 #define NUM_VDIV ARRAY_SIZE(vdivs)
119 static const char *trigger_sources[] = {
144 static const char *coupling[] = {
150 /* Do not change the order of entries */
151 static const char *data_sources[] = {
158 * name, series, protocol flavor, min timebase, max timebase, min vdiv,
159 * digital channels, number of horizontal divs
162 #define RIGOL "Rigol Technologies"
163 #define AGILENT "Agilent Technologies"
165 static const struct rigol_ds_model supported_models[] = {
166 {RIGOL, "DS1052E", RIGOL_DS1000, PROTOCOL_LEGACY, {5, 1000000000}, {50, 1}, {2, 1000}, 2, false, 12},
167 {RIGOL, "DS1102E", RIGOL_DS1000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, false, 12},
168 {RIGOL, "DS1152E", RIGOL_DS1000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, false, 12},
169 {RIGOL, "DS1052D", RIGOL_DS1000, PROTOCOL_LEGACY, {5, 1000000000}, {50, 1}, {2, 1000}, 2, true, 12},
170 {RIGOL, "DS1102D", RIGOL_DS1000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, true, 12},
171 {RIGOL, "DS1152D", RIGOL_DS1000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, true, 12},
172 {RIGOL, "DS2072", RIGOL_DS2000, PROTOCOL_IEEE488_2, {5, 1000000000}, {500, 1}, {500, 1000000}, 2, false, 14},
173 {RIGOL, "DS2102", RIGOL_DS2000, PROTOCOL_IEEE488_2, {5, 1000000000}, {500, 1}, {500, 1000000}, 2, false, 14},
174 {RIGOL, "DS2202", RIGOL_DS2000, PROTOCOL_IEEE488_2, {2, 1000000000}, {500, 1}, {500, 1000000}, 2, false, 14},
175 {RIGOL, "DS2302", RIGOL_DS2000, PROTOCOL_IEEE488_2, {1, 1000000000}, {1000, 1}, {500, 1000000}, 2, false, 14},
176 {RIGOL, "DS2072A", RIGOL_DS2000, PROTOCOL_IEEE488_2, {5, 1000000000}, {1000, 1}, {500, 1000000}, 2, false, 14},
177 {RIGOL, "DS2102A", RIGOL_DS2000, PROTOCOL_IEEE488_2, {5, 1000000000}, {1000, 1}, {500, 1000000}, 2, false, 14},
178 {RIGOL, "DS2202A", RIGOL_DS2000, PROTOCOL_IEEE488_2, {2, 1000000000}, {1000, 1}, {500, 1000000}, 2, false, 14},
179 {RIGOL, "DS2302A", RIGOL_DS2000, PROTOCOL_IEEE488_2, {1, 1000000000}, {1000, 1}, {500, 1000000}, 2, false, 14},
180 {RIGOL, "VS5022", RIGOL_VS5000, PROTOCOL_LEGACY, {20, 1000000000}, {50, 1}, {2, 1000}, 2, false, 14},
181 {RIGOL, "VS5022D", RIGOL_VS5000, PROTOCOL_LEGACY, {20, 1000000000}, {50, 1}, {2, 1000}, 2, true, 14},
182 {RIGOL, "VS5042", RIGOL_VS5000, PROTOCOL_LEGACY, {10, 1000000000}, {50, 1}, {2, 1000}, 2, false, 14},
183 {RIGOL, "VS5042D", RIGOL_VS5000, PROTOCOL_LEGACY, {10, 1000000000}, {50, 1}, {2, 1000}, 2, true, 14},
184 {RIGOL, "VS5062", RIGOL_VS5000, PROTOCOL_LEGACY, {5, 1000000000}, {50, 1}, {2, 1000}, 2, false, 14},
185 {RIGOL, "VS5062D", RIGOL_VS5000, PROTOCOL_LEGACY, {5, 1000000000}, {50, 1}, {2, 1000}, 2, true, 14},
186 {RIGOL, "VS5102", RIGOL_VS5000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, false, 14},
187 {RIGOL, "VS5102D", RIGOL_VS5000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, true, 14},
188 {RIGOL, "VS5202", RIGOL_VS5000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, false, 14},
189 {RIGOL, "VS5202D", RIGOL_VS5000, PROTOCOL_LEGACY, {2, 1000000000}, {50, 1}, {2, 1000}, 2, true, 14},
190 {AGILENT, "DSO1002A", AGILENT_DSO1000, PROTOCOL_IEEE488_2, {5, 1000000000}, {50, 1}, {2, 1000}, 2, false, 12},
191 {AGILENT, "DSO1004A", AGILENT_DSO1000, PROTOCOL_IEEE488_2, {5, 1000000000}, {50, 1}, {2, 1000}, 4, false, 12},
192 {AGILENT, "DSO1012A", AGILENT_DSO1000, PROTOCOL_IEEE488_2, {2, 1000000000}, {50, 1}, {2, 1000}, 2, false, 12},
193 {AGILENT, "DSO1014A", AGILENT_DSO1000, PROTOCOL_IEEE488_2, {2, 1000000000}, {50, 1}, {2, 1000}, 4, false, 12},
194 {AGILENT, "DSO1022A", AGILENT_DSO1000, PROTOCOL_IEEE488_2, {2, 1000000000}, {50, 1}, {2, 1000}, 2, false, 12},
195 {AGILENT, "DSO1024A", AGILENT_DSO1000, PROTOCOL_IEEE488_2, {2, 1000000000}, {50, 1}, {2, 1000}, 4, false, 12},
198 SR_PRIV struct sr_dev_driver rigol_ds_driver_info;
199 static struct sr_dev_driver *di = &rigol_ds_driver_info;
201 static void clear_helper(void *priv)
203 struct dev_context *devc;
207 g_free(devc->buffer);
208 g_free(devc->coupling[0]);
209 g_free(devc->coupling[1]);
210 g_free(devc->trigger_source);
211 g_free(devc->trigger_slope);
212 g_slist_free(devc->analog_groups[0].probes);
213 g_slist_free(devc->analog_groups[1].probes);
214 g_slist_free(devc->digital_group.probes);
217 static int dev_clear(void)
219 return std_dev_clear(di, clear_helper);
222 static int set_cfg(const struct sr_dev_inst *sdi, const char *format, ...)
227 va_start(args, format);
228 ret = sr_scpi_send_variadic(sdi->conn, format, args);
234 /* When setting a bunch of parameters in a row, the DS1052E scrambles
235 * some of them unless there is at least 100ms delay in between. */
236 sr_spew("delay %dms", 100);
242 static int init(struct sr_context *sr_ctx)
244 return std_init(sr_ctx, di, LOG_PREFIX);
247 static int probe_port(const char *resource, const char *serialcomm, GSList **devices)
249 struct dev_context *devc;
250 struct sr_dev_inst *sdi;
251 struct sr_scpi_dev_inst *scpi;
252 struct sr_scpi_hw_info *hw_info;
253 struct sr_probe *probe;
255 const struct rigol_ds_model *model = NULL;
260 if (!(scpi = scpi_dev_inst_new(resource, serialcomm)))
263 if (sr_scpi_open(scpi) != SR_OK) {
264 sr_info("Couldn't open SCPI device.");
269 if (sr_scpi_get_hw_id(scpi, &hw_info) != SR_OK) {
270 sr_info("Couldn't get IDN response.");
276 for (i = 0; i < ARRAY_SIZE(supported_models); i++) {
277 if (!strcasecmp(hw_info->manufacturer, supported_models[i].vendor) &&
278 !strcmp(hw_info->model, supported_models[i].name)) {
279 model = &supported_models[i];
284 if (!model || !(sdi = sr_dev_inst_new(0, SR_ST_ACTIVE,
285 hw_info->manufacturer, hw_info->model,
286 hw_info->firmware_version))) {
287 sr_scpi_hw_info_free(hw_info);
293 sr_scpi_hw_info_free(hw_info);
299 sdi->inst_type = SR_INST_SCPI;
301 if (!(devc = g_try_malloc0(sizeof(struct dev_context))))
302 return SR_ERR_MALLOC;
304 devc->limit_frames = 0;
307 for (i = 0; i < model->analog_channels; i++) {
308 if (!(channel_name = g_strdup_printf("CH%d", i + 1)))
309 return SR_ERR_MALLOC;
310 probe = sr_probe_new(i, SR_PROBE_ANALOG, TRUE, channel_name);
311 sdi->probes = g_slist_append(sdi->probes, probe);
312 devc->analog_groups[i].name = channel_name;
313 devc->analog_groups[i].probes = g_slist_append(NULL, probe);
314 sdi->probe_groups = g_slist_append(sdi->probe_groups,
315 &devc->analog_groups[i]);
318 if (devc->model->has_digital) {
319 for (i = 0; i < 16; i++) {
320 if (!(channel_name = g_strdup_printf("D%d", i)))
321 return SR_ERR_MALLOC;
322 probe = sr_probe_new(i, SR_PROBE_LOGIC, TRUE, channel_name);
323 g_free(channel_name);
325 return SR_ERR_MALLOC;
326 sdi->probes = g_slist_append(sdi->probes, probe);
327 devc->digital_group.probes = g_slist_append(
328 devc->digital_group.probes, probe);
330 devc->digital_group.name = "LA";
331 sdi->probe_groups = g_slist_append(sdi->probe_groups,
332 &devc->digital_group);
335 for (i = 0; i < NUM_TIMEBASE; i++) {
336 if (!memcmp(&devc->model->min_timebase, &timebases[i], sizeof(uint64_t[2])))
337 devc->timebases = &timebases[i];
338 if (!memcmp(&devc->model->max_timebase, &timebases[i], sizeof(uint64_t[2])))
339 devc->num_timebases = &timebases[i] - devc->timebases + 1;
342 for (i = 0; i < NUM_VDIV; i++)
343 if (!memcmp(&devc->model->min_vdiv, &vdivs[i], sizeof(uint64_t[2])))
344 devc->vdivs = &vdivs[i];
346 if (!(devc->buffer = g_try_malloc(ACQ_BUFFER_SIZE)))
347 return SR_ERR_MALLOC;
348 if (!(devc->data = g_try_malloc(ACQ_BUFFER_SIZE * sizeof(float))))
349 return SR_ERR_MALLOC;
351 devc->data_source = DATA_SOURCE_LIVE;
355 *devices = g_slist_append(NULL, sdi);
360 static GSList *scan(GSList *options)
362 struct drv_context *drvc;
363 struct sr_config *src;
367 const gchar *dev_name;
369 gchar *serialcomm = NULL;
373 for (l = options; l; l = l->next) {
377 port = (char *)g_variant_get_string(src->data, NULL);
379 case SR_CONF_SERIALCOMM:
380 serialcomm = (char *)g_variant_get_string(src->data, NULL);
387 if (probe_port(port, serialcomm, &devices) == SR_ERR_MALLOC) {
394 if (!(dir = g_dir_open("/sys/class/usbmisc/", 0, NULL)))
395 if (!(dir = g_dir_open("/sys/class/usb/", 0, NULL)))
397 while ((dev_name = g_dir_read_name(dir))) {
398 if (strncmp(dev_name, "usbtmc", 6))
400 port = g_strconcat("/dev/", dev_name, NULL);
401 ret = probe_port(port, serialcomm, &devices);
405 if (ret == SR_ERR_MALLOC) {
413 /* Tack a copy of the newly found devices onto the driver list. */
414 l = g_slist_copy(devices);
415 drvc->instances = g_slist_concat(drvc->instances, l);
420 static GSList *dev_list(void)
422 return ((struct drv_context *)(di->priv))->instances;
425 static int dev_open(struct sr_dev_inst *sdi)
427 struct sr_scpi_dev_inst *scpi = sdi->conn;
429 if (sr_scpi_open(scpi) < 0)
432 if (rigol_ds_get_dev_cfg(sdi) != SR_OK)
435 sdi->status = SR_ST_ACTIVE;
440 static int dev_close(struct sr_dev_inst *sdi)
442 struct sr_scpi_dev_inst *scpi;
447 if (sr_scpi_close(scpi) < 0)
449 sdi->status = SR_ST_INACTIVE;
455 static int cleanup(void)
460 static int analog_frame_size(const struct sr_dev_inst *sdi)
462 struct dev_context *devc = sdi->priv;
463 struct sr_probe *probe;
464 int analog_probes = 0;
467 if (devc->model->protocol == PROTOCOL_LEGACY) {
468 if (devc->model->series == RIGOL_VS5000)
469 return VS5000_ANALOG_LIVE_WAVEFORM_SIZE;
471 return DS1000_ANALOG_LIVE_WAVEFORM_SIZE;
473 for (l = sdi->probes; l; l = l->next) {
475 if (probe->type == SR_PROBE_ANALOG && probe->enabled)
478 if (devc->data_source == DATA_SOURCE_MEMORY) {
479 if (analog_probes == 1)
480 return DS2000_ANALOG_MEM_WAVEFORM_SIZE_1C;
482 return DS2000_ANALOG_MEM_WAVEFORM_SIZE_2C;
484 if (devc->model->series == AGILENT_DSO1000)
485 return DSO1000_ANALOG_LIVE_WAVEFORM_SIZE;
487 return DS2000_ANALOG_LIVE_WAVEFORM_SIZE;
492 static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi,
493 const struct sr_probe_group *probe_group)
495 struct dev_context *devc;
498 if (!sdi || !(devc = sdi->priv))
501 /* If a probe group is specified, it must be a valid one. */
502 if (probe_group && !g_slist_find(sdi->probe_groups, probe_group)) {
503 sr_err("Invalid probe group specified.");
508 case SR_CONF_NUM_TIMEBASE:
509 *data = g_variant_new_int32(devc->model->num_horizontal_divs);
511 case SR_CONF_NUM_VDIV:
512 *data = g_variant_new_int32(8);
513 case SR_CONF_DATA_SOURCE:
514 if (devc->data_source == DATA_SOURCE_LIVE)
515 *data = g_variant_new_string("Live");
516 else if (devc->data_source == DATA_SOURCE_MEMORY)
517 *data = g_variant_new_string("Memory");
519 *data = g_variant_new_string("Segmented");
521 case SR_CONF_SAMPLERATE:
522 if (devc->data_source == DATA_SOURCE_LIVE) {
523 samplerate = analog_frame_size(sdi) /
524 (devc->timebase * devc->model->num_horizontal_divs);
525 *data = g_variant_new_uint64(samplerate);
537 static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi,
538 const struct sr_probe_group *probe_group)
540 struct dev_context *devc;
541 uint64_t tmp_u64, p, q;
547 if (!(devc = sdi->priv))
550 if (sdi->status != SR_ST_ACTIVE)
551 return SR_ERR_DEV_CLOSED;
553 /* If a probe group is specified, it must be a valid one. */
554 if (probe_group && !g_slist_find(sdi->probe_groups, probe_group)) {
555 sr_err("Invalid probe group specified.");
561 case SR_CONF_LIMIT_FRAMES:
562 devc->limit_frames = g_variant_get_uint64(data);
564 case SR_CONF_TRIGGER_SLOPE:
565 tmp_u64 = g_variant_get_uint64(data);
566 if (tmp_u64 != 0 && tmp_u64 != 1)
568 g_free(devc->trigger_slope);
569 devc->trigger_slope = g_strdup(tmp_u64 ? "POS" : "NEG");
570 ret = set_cfg(sdi, ":TRIG:EDGE:SLOP %s", devc->trigger_slope);
572 case SR_CONF_HORIZ_TRIGGERPOS:
573 t_dbl = g_variant_get_double(data);
574 if (t_dbl < 0.0 || t_dbl > 1.0)
576 devc->horiz_triggerpos = t_dbl;
577 /* We have the trigger offset as a percentage of the frame, but
578 * need to express this in seconds. */
579 t_dbl = -(devc->horiz_triggerpos - 0.5) * devc->timebase * devc->num_timebases;
580 ret = set_cfg(sdi, ":TIM:OFFS %.6f", t_dbl);
582 case SR_CONF_TIMEBASE:
583 g_variant_get(data, "(tt)", &p, &q);
584 for (i = 0; i < devc->num_timebases; i++) {
585 if (devc->timebases[i][0] == p && devc->timebases[i][1] == q) {
586 devc->timebase = (float)p / q;
587 ret = set_cfg(sdi, ":TIM:SCAL %.9f", devc->timebase);
591 if (i == devc->num_timebases)
594 case SR_CONF_TRIGGER_SOURCE:
595 tmp_str = g_variant_get_string(data, NULL);
596 for (i = 0; i < ARRAY_SIZE(trigger_sources); i++) {
597 if (!strcmp(trigger_sources[i], tmp_str)) {
598 g_free(devc->trigger_source);
599 devc->trigger_source = g_strdup(trigger_sources[i]);
600 if (!strcmp(devc->trigger_source, "AC Line"))
602 else if (!strcmp(devc->trigger_source, "CH1"))
604 else if (!strcmp(devc->trigger_source, "CH2"))
606 else if (!strcmp(devc->trigger_source, "CH3"))
608 else if (!strcmp(devc->trigger_source, "CH4"))
611 tmp_str = (char *)devc->trigger_source;
612 ret = set_cfg(sdi, ":TRIG:EDGE:SOUR %s", tmp_str);
616 if (i == ARRAY_SIZE(trigger_sources))
621 sr_err("No probe group specified.");
622 return SR_ERR_PROBE_GROUP;
624 g_variant_get(data, "(tt)", &p, &q);
625 for (i = 0; i < 2; i++) {
626 if (probe_group == &devc->analog_groups[i]) {
627 for (j = 0; j < ARRAY_SIZE(vdivs); j++) {
628 if (vdivs[j][0] != p || vdivs[j][1] != q)
630 devc->vdiv[i] = (float)p / q;
631 return set_cfg(sdi, ":CHAN%d:SCAL %.3f", i + 1,
638 case SR_CONF_COUPLING:
640 sr_err("No probe group specified.");
641 return SR_ERR_PROBE_GROUP;
643 tmp_str = g_variant_get_string(data, NULL);
644 for (i = 0; i < 2; i++) {
645 if (probe_group == &devc->analog_groups[i]) {
646 for (j = 0; j < ARRAY_SIZE(coupling); j++) {
647 if (!strcmp(tmp_str, coupling[j])) {
648 g_free(devc->coupling[i]);
649 devc->coupling[i] = g_strdup(coupling[j]);
650 return set_cfg(sdi, ":CHAN%d:COUP %s", i + 1,
658 case SR_CONF_DATA_SOURCE:
659 tmp_str = g_variant_get_string(data, NULL);
660 if (!strcmp(tmp_str, "Live"))
661 devc->data_source = DATA_SOURCE_LIVE;
662 else if (!strcmp(tmp_str, "Memory"))
663 devc->data_source = DATA_SOURCE_MEMORY;
664 else if (devc->model->protocol == PROTOCOL_IEEE488_2
665 && !strcmp(tmp_str, "Segmented"))
666 devc->data_source = DATA_SOURCE_SEGMENTED;
678 static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi,
679 const struct sr_probe_group *probe_group)
681 GVariant *tuple, *rational[2];
684 struct dev_context *devc = NULL;
689 if (key == SR_CONF_SCAN_OPTIONS) {
690 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
691 hwopts, ARRAY_SIZE(hwopts), sizeof(int32_t));
693 } else if (key == SR_CONF_DEVICE_OPTIONS && probe_group == NULL) {
694 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
695 hwcaps, ARRAY_SIZE(hwcaps), sizeof(int32_t));
699 /* Every other option requires a valid device instance. */
700 if (!sdi || !(devc = sdi->priv))
703 /* If a probe group is specified, it must be a valid one. */
705 if (probe_group != &devc->analog_groups[0]
706 && probe_group != &devc->analog_groups[1]) {
707 sr_err("Invalid probe group specified.");
713 case SR_CONF_DEVICE_OPTIONS:
715 sr_err("No probe group specified.");
716 return SR_ERR_PROBE_GROUP;
718 if (probe_group == &devc->digital_group) {
719 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
720 NULL, 0, sizeof(int32_t));
723 for (i = 0; i < 2; i++) {
724 if (probe_group == &devc->analog_groups[i]) {
725 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
726 analog_hwcaps, ARRAY_SIZE(analog_hwcaps), sizeof(int32_t));
733 case SR_CONF_COUPLING:
735 sr_err("No probe group specified.");
736 return SR_ERR_PROBE_GROUP;
738 *data = g_variant_new_strv(coupling, ARRAY_SIZE(coupling));
742 /* Can't know this until we have the exact model. */
745 sr_err("No probe group specified.");
746 return SR_ERR_PROBE_GROUP;
748 g_variant_builder_init(&gvb, G_VARIANT_TYPE_ARRAY);
749 for (i = 0; i < NUM_VDIV; i++) {
750 rational[0] = g_variant_new_uint64(devc->vdivs[i][0]);
751 rational[1] = g_variant_new_uint64(devc->vdivs[i][1]);
752 tuple = g_variant_new_tuple(rational, 2);
753 g_variant_builder_add_value(&gvb, tuple);
755 *data = g_variant_builder_end(&gvb);
757 case SR_CONF_TIMEBASE:
759 /* Can't know this until we have the exact model. */
761 if (devc->num_timebases <= 0)
763 g_variant_builder_init(&gvb, G_VARIANT_TYPE_ARRAY);
764 for (i = 0; i < devc->num_timebases; i++) {
765 rational[0] = g_variant_new_uint64(devc->timebases[i][0]);
766 rational[1] = g_variant_new_uint64(devc->timebases[i][1]);
767 tuple = g_variant_new_tuple(rational, 2);
768 g_variant_builder_add_value(&gvb, tuple);
770 *data = g_variant_builder_end(&gvb);
772 case SR_CONF_TRIGGER_SOURCE:
774 /* Can't know this until we have the exact model. */
776 *data = g_variant_new_strv(trigger_sources,
777 devc->model->has_digital ? ARRAY_SIZE(trigger_sources) : 4);
779 case SR_CONF_DATA_SOURCE:
781 /* Can't know this until we have the exact model. */
783 /* This needs tweaking by series/model! */
784 if (devc->model->series == RIGOL_DS2000)
785 *data = g_variant_new_strv(data_sources, ARRAY_SIZE(data_sources));
787 *data = g_variant_new_strv(data_sources, ARRAY_SIZE(data_sources) - 1);
796 static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
798 struct sr_scpi_dev_inst *scpi;
799 struct dev_context *devc;
800 struct sr_probe *probe;
803 if (sdi->status != SR_ST_ACTIVE)
804 return SR_ERR_DEV_CLOSED;
809 devc->num_frames = 0;
811 for (l = sdi->probes; l; l = l->next) {
813 sr_dbg("handling probe %s", probe->name);
814 if (probe->type == SR_PROBE_ANALOG) {
816 devc->enabled_analog_probes = g_slist_append(
817 devc->enabled_analog_probes, probe);
818 if (probe->enabled != devc->analog_channels[probe->index]) {
819 /* Enabled channel is currently disabled, or vice versa. */
820 if (set_cfg(sdi, ":CHAN%d:DISP %s", probe->index + 1,
821 probe->enabled ? "ON" : "OFF") != SR_OK)
823 devc->analog_channels[probe->index] = probe->enabled;
825 } else if (probe->type == SR_PROBE_LOGIC) {
826 if (probe->enabled) {
827 devc->enabled_digital_probes = g_slist_append(
828 devc->enabled_digital_probes, probe);
829 /* Turn on LA module if currently off. */
830 if (!devc->la_enabled) {
831 if (set_cfg(sdi, ":LA:DISP ON") != SR_OK)
833 devc->la_enabled = TRUE;
836 if (probe->enabled != devc->digital_channels[probe->index]) {
837 /* Enabled channel is currently disabled, or vice versa. */
838 if (set_cfg(sdi, ":DIG%d:TURN %s", probe->index,
839 probe->enabled ? "ON" : "OFF") != SR_OK)
841 devc->digital_channels[probe->index] = probe->enabled;
846 if (!devc->enabled_analog_probes && !devc->enabled_digital_probes)
849 /* Turn off LA module if on and no digital probes selected. */
850 if (devc->la_enabled && !devc->enabled_digital_probes)
851 if (set_cfg(sdi, ":LA:DISP OFF") != SR_OK)
854 if (devc->data_source == DATA_SOURCE_LIVE) {
855 if (set_cfg(sdi, ":RUN") != SR_OK)
857 } else if (devc->data_source == DATA_SOURCE_MEMORY) {
858 if (devc->model->series != RIGOL_DS2000) {
859 sr_err("Data source 'Memory' not supported for this device");
862 } else if (devc->data_source == DATA_SOURCE_SEGMENTED) {
863 sr_err("Data source 'Segmented' not yet supported");
867 sr_scpi_source_add(scpi, G_IO_IN, 50, rigol_ds_receive, (void *)sdi);
869 /* Send header packet to the session bus. */
870 std_session_send_df_header(cb_data, LOG_PREFIX);
872 if (devc->enabled_analog_probes)
873 devc->channel_entry = devc->enabled_analog_probes;
875 devc->channel_entry = devc->enabled_digital_probes;
877 devc->analog_frame_size = analog_frame_size(sdi);
879 if (devc->model->protocol == PROTOCOL_LEGACY) {
880 /* Fetch the first frame. */
881 if (rigol_ds_channel_start(sdi) != SR_OK)
884 if (devc->enabled_analog_probes) {
885 if (devc->data_source == DATA_SOURCE_MEMORY) {
886 /* Apparently for the DS2000 the memory
887 * depth can only be set in Running state -
888 * this matches the behaviour of the UI. */
889 if (set_cfg(sdi, ":RUN") != SR_OK)
891 if (set_cfg(sdi, "ACQ:MDEP %d", devc->analog_frame_size) != SR_OK)
893 if (set_cfg(sdi, ":STOP") != SR_OK)
896 if (rigol_ds_capture_start(sdi) != SR_OK)
904 static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
906 struct dev_context *devc;
907 struct sr_scpi_dev_inst *scpi;
908 struct sr_datafeed_packet packet;
914 if (sdi->status != SR_ST_ACTIVE) {
915 sr_err("Device inactive, can't stop acquisition.");
919 /* End of last frame. */
920 packet.type = SR_DF_END;
921 sr_session_send(sdi, &packet);
923 g_slist_free(devc->enabled_analog_probes);
924 g_slist_free(devc->enabled_digital_probes);
925 devc->enabled_analog_probes = NULL;
926 devc->enabled_digital_probes = NULL;
928 sr_scpi_source_remove(scpi);
933 SR_PRIV struct sr_dev_driver rigol_ds_driver_info = {
935 .longname = "Rigol DS",
940 .dev_list = dev_list,
941 .dev_clear = dev_clear,
942 .config_get = config_get,
943 .config_set = config_set,
944 .config_list = config_list,
945 .dev_open = dev_open,
946 .dev_close = dev_close,
947 .dev_acquisition_start = dev_acquisition_start,
948 .dev_acquisition_stop = dev_acquisition_stop,