rigol-ds: Experimental support for V5 frame reading

[libsigrok.git] / src / hardware / rigol-ds / api.c

diff --git a/src/hardware/rigol-ds/api.c b/src/hardware/rigol-ds/api.c
index af15587db2e36fef02c2a6c1963e01ad56e81542..01914ff85c61d3f4e7c3abc628aed86ca66c923e 100644 (file)
--- a/src/hardware/rigol-ds/api.c
+++ b/src/hardware/rigol-ds/api.c
@@ -500,6 +500,7 @@ static int analog_frame_size(const struct sr_dev_inst *sdi)
        case DATA_SOURCE_LIVE:
                return devc->model->series->live_samples;
        case DATA_SOURCE_MEMORY:
+       case DATA_SOURCE_SEGMENTED:
                return devc->model->series->buffer_samples / analog_channels;
        default:
                return 0;
@@ -514,6 +515,7 @@ static int digital_frame_size(const struct sr_dev_inst *sdi)
        case DATA_SOURCE_LIVE:
                return devc->model->series->live_samples * 2;
        case DATA_SOURCE_MEMORY:
+       case DATA_SOURCE_SEGMENTED:
                return devc->model->series->buffer_samples * 2;
        default:
                return 0;
@@ -879,6 +881,7 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi)
        devc = sdi->priv;
 
        devc->num_frames = 0;
+       devc->num_frames_segmented = 0;
 
        some_digital = FALSE;
        for (l = sdi->channels; l; l = l->next) {
@@ -943,8 +946,28 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi)
 
        /* Set memory mode. */
        if (devc->data_source == DATA_SOURCE_SEGMENTED) {
-               sr_err("Data source 'Segmented' not yet supported");
-               return SR_ERR;
+               switch (devc->model->series->protocol) {
+               case PROTOCOL_V4:
+               {
+                       int frames = 0;
+                       sr_scpi_get_int(sdi->conn, "FUNC:WREP:FEND?", &frames);
+                       if (frames <= 0) {
+                               sr_err("No segmented data available");
+                               return SR_ERR;
+                       }
+                       devc->num_frames_segmented = frames;
+                       break;
+               }
+               case PROTOCOL_V5:
+                       /* The frame limit has to be read on the fly, just set up
+                        * reading of the first frame */
+                       if (rigol_ds_config_set(sdi, "REC:CURR 1") != SR_OK)
+                               return SR_ERR;
+                       break;
+               default:
+                       sr_err("Data source 'Segmented' not yet supported");
+                       return SR_ERR;
+               }
        }
 
        devc->analog_frame_size = analog_frame_size(sdi);