]> sigrok.org Git - libsigrok.git/blobdiff - src/hardware/gwinstek-gds-800/protocol.c
output/csv: use intermediate time_t var, silence compiler warning
[libsigrok.git] / src / hardware / gwinstek-gds-800 / protocol.c
index 8c5a10206230fbe1fca6e518e1eb6bd35d4d7d81..5feb25b140d716d336bb87e0d09fb721515a8674 100644 (file)
  */
 
 #include "protocol.h"
+#include <math.h>
+#include <string.h>
+
+#define ANALOG_CHANNELS 2
+#define VERTICAL_DIVISIONS 10
+
+static int read_data(struct sr_dev_inst *sdi,
+               struct sr_scpi_dev_inst *scpi, struct dev_context *devc,
+               int data_size)
+{
+       int len;
+
+       len = sr_scpi_read_data(scpi,
+               &devc->rcv_buffer[devc->cur_rcv_buffer_position],
+               data_size - devc->cur_rcv_buffer_position);
+       if (len < 0) {
+               sr_err("Read data error.");
+               sr_dev_acquisition_stop(sdi);
+               devc->cur_rcv_buffer_position = 0;
+               return SR_ERR;
+       }
+
+       devc->cur_rcv_buffer_position += len;
+
+       /* Handle the case where sr_scpi_read_data stopped at the newline. */
+       if (len < data_size && sr_scpi_read_complete(scpi)) {
+               devc->rcv_buffer[devc->cur_rcv_buffer_position] = '\n';
+               devc->cur_rcv_buffer_position++;
+       }
+
+       if (devc->cur_rcv_buffer_position < data_size)
+               return SR_ERR; /* Not finished yet. */
+       else if (devc->cur_rcv_buffer_position == data_size) {
+                devc->cur_rcv_buffer_position = 0;
+               return SR_OK;
+       } else {
+               sr_err("Too many bytes read.");
+               sr_dev_acquisition_stop(sdi);
+               devc->cur_rcv_buffer_position = 0;
+               return SR_ERR;
+       }
+}
 
 SR_PRIV int gwinstek_gds_800_receive_data(int fd, int revents, void *cb_data)
 {
-       const struct sr_dev_inst *sdi;
+       struct sr_dev_inst *sdi;
+       struct sr_scpi_dev_inst *scpi;
        struct dev_context *devc;
+       struct sr_datafeed_packet packet;
+       struct sr_datafeed_analog analog;
+       struct sr_analog_encoding encoding;
+       struct sr_analog_meaning meaning;
+       struct sr_analog_spec spec;
+       char command[32];
+       char *response;
+       float volts_per_division;
+       int num_samples, i;
+       float samples[MAX_SAMPLES];
+       uint32_t sample_rate;
+       char *end_ptr;
 
        (void)fd;
 
@@ -32,8 +87,197 @@ SR_PRIV int gwinstek_gds_800_receive_data(int fd, int revents, void *cb_data)
        if (!(devc = sdi->priv))
                return TRUE;
 
-       if (revents == G_IO_IN) {
-               /* TODO */
+       scpi = sdi->conn;
+
+       if (!(revents == G_IO_IN || revents == 0))
+               return TRUE;
+
+       switch (devc->state) {
+       case START_ACQUISITION:
+               if (sr_scpi_send(scpi, ":TRIG:MOD 3") != SR_OK) {
+                       sr_err("Failed to set trigger mode to SINGLE.");
+                       sr_dev_acquisition_stop(sdi);
+                       return TRUE;
+               }
+               if (sr_scpi_send(scpi, ":STOP") != SR_OK) {
+                       sr_err("Failed to put the trigger system into STOP state.");
+                       sr_dev_acquisition_stop(sdi);
+                       return TRUE;
+               }
+               if (sr_scpi_send(scpi, ":RUN") != SR_OK) {
+                       sr_err("Failed to put the trigger system into RUN state.");
+                       sr_dev_acquisition_stop(sdi);
+                       return TRUE;
+               }
+
+               devc->cur_acq_channel = 0;
+               devc->state = START_TRANSFER_OF_CHANNEL_DATA;
+               break;
+       case START_TRANSFER_OF_CHANNEL_DATA:
+               if (((struct sr_channel *)g_slist_nth_data(sdi->channels, devc->cur_acq_channel))->enabled) {
+                       if (sr_scpi_send(scpi, ":ACQ%d:MEM?", devc->cur_acq_channel+1) != SR_OK) {
+                               sr_err("Failed to acquire memory.");
+                               sr_dev_acquisition_stop(sdi);
+                               return TRUE;
+                       }
+                       if (sr_scpi_read_begin(scpi) != SR_OK) {
+                               sr_err("Could not begin reading SCPI response.");
+                               sr_dev_acquisition_stop(sdi);
+                               return TRUE;
+                       }
+                       devc->state = WAIT_FOR_TRANSFER_OF_BEGIN_TRANSMISSION_COMPLETE;
+                       devc->cur_rcv_buffer_position = 0;
+               } else {
+                       /* All channels acquired. */
+                       if (devc->cur_acq_channel == ANALOG_CHANNELS - 1) {
+                               sr_spew("All channels acquired.");
+
+                               if (devc->cur_acq_frame == devc->frame_limit - 1) {
+                                       /* All frames accquired. */
+                                       sr_spew("All frames acquired.");
+
+                                       sr_dev_acquisition_stop(sdi);
+                                       return TRUE;
+                               } else {
+                                       /* Start acquiring next frame. */
+                                       if (devc->df_started) {
+                                               std_session_send_df_frame_end(sdi);
+                                               std_session_send_df_frame_begin(sdi);
+                                       }
+
+                                       devc->cur_acq_frame++;
+                                       devc->state = START_ACQUISITION;
+                               }
+                       } else {
+                               /* Start acquiring next channel. */
+                               devc->cur_acq_channel++;
+                       }
+               }
+               break;
+       case WAIT_FOR_TRANSFER_OF_BEGIN_TRANSMISSION_COMPLETE:
+               if (read_data(sdi, scpi, devc, 1) < 0)
+                       break;
+               if (devc->rcv_buffer[0] == '#')
+                       devc->state = WAIT_FOR_TRANSFER_OF_DATA_SIZE_DIGIT_COMPLETE;
+               break;
+       case WAIT_FOR_TRANSFER_OF_DATA_SIZE_DIGIT_COMPLETE:
+               if (read_data(sdi, scpi, devc, 1) < 0)
+                       break;
+               if (devc->rcv_buffer[0] != '4' &&
+                       devc->rcv_buffer[0] != '5' &&
+                       devc->rcv_buffer[0] != '6') {
+                       sr_err("Data size digits is not 4, 5 or 6 but "
+                              "'%c'.", devc->rcv_buffer[0]);
+                       sr_dev_acquisition_stop(sdi);
+                       return TRUE;
+               } else {
+                       devc->data_size_digits = devc->rcv_buffer[0] - '0';
+                       devc->state = WAIT_FOR_TRANSFER_OF_DATA_SIZE_COMPLETE;
+               }
+               break;
+       case WAIT_FOR_TRANSFER_OF_DATA_SIZE_COMPLETE:
+               if (read_data(sdi, scpi, devc, devc->data_size_digits) < 0)
+                       break;
+               devc->rcv_buffer[devc->data_size_digits] = 0;
+               if (sr_atoi(devc->rcv_buffer, &devc->data_size) != SR_OK) {
+                       sr_err("Could not parse data size '%s'", devc->rcv_buffer);
+                       sr_dev_acquisition_stop(sdi);
+                       return TRUE;
+               } else
+                       devc->state = WAIT_FOR_TRANSFER_OF_SAMPLE_RATE_COMPLETE;
+               break;
+       case WAIT_FOR_TRANSFER_OF_SAMPLE_RATE_COMPLETE:
+               if (read_data(sdi, scpi, devc, sizeof(float)) < 0)
+                       break;
+               /*
+                * Contrary to the documentation, this field is
+                * transfered with most significant byte first!
+                */
+               sample_rate = RB32(devc->rcv_buffer);
+               memcpy(&devc->sample_rate, &sample_rate, sizeof(float));
+               devc->state = WAIT_FOR_TRANSFER_OF_CHANNEL_INDICATOR_COMPLETE;
+
+               if (!devc->df_started) {
+                       std_session_send_df_header(sdi);
+                       std_session_send_df_frame_begin(sdi);
+                       devc->df_started = TRUE;
+               }
+               break;
+       case WAIT_FOR_TRANSFER_OF_CHANNEL_INDICATOR_COMPLETE:
+               if (read_data(sdi, scpi, devc, 1) == SR_OK)
+                       devc->state = WAIT_FOR_TRANSFER_OF_RESERVED_DATA_COMPLETE;
+               break;
+       case WAIT_FOR_TRANSFER_OF_RESERVED_DATA_COMPLETE:
+               if (read_data(sdi, scpi, devc, 3) == SR_OK)
+                       devc->state = WAIT_FOR_TRANSFER_OF_CHANNEL_DATA_COMPLETE;
+               break;
+       case WAIT_FOR_TRANSFER_OF_CHANNEL_DATA_COMPLETE:
+               if (read_data(sdi, scpi, devc, devc->data_size - 8) < 0)
+                       break;
+
+               /* Fetch data needed for conversion from device. */
+               snprintf(command, sizeof(command), ":CHAN%d:SCAL?",
+                               devc->cur_acq_channel + 1);
+               if (sr_scpi_get_string(scpi, command, &response) != SR_OK) {
+                       sr_err("Failed to get volts per division.");
+                       sr_dev_acquisition_stop(sdi);
+                       return TRUE;
+               }
+               volts_per_division = g_ascii_strtod(response, &end_ptr);
+               if (!strcmp(end_ptr, "mV"))
+                       volts_per_division *= 1.e-3;
+               g_free(response);
+
+               num_samples = (devc->data_size - 8) / 2;
+               sr_spew("Received %d number of samples from channel "
+                       "%d.", num_samples, devc->cur_acq_channel + 1);
+
+               float vbit = volts_per_division * VERTICAL_DIVISIONS / 256.0;
+               float vbitlog = log10f(vbit);
+               int digits = -(int)vbitlog + (vbitlog < 0.0);
+
+               /* Convert data. */
+               for (i = 0; i < num_samples; i++)
+                       samples[i] = ((float) ((int16_t) (RB16(&devc->rcv_buffer[i*2])))) * vbit;
+
+               /* Fill frame. */
+               sr_analog_init(&analog, &encoding, &meaning, &spec, digits);
+               analog.meaning->channels = g_slist_append(NULL, g_slist_nth_data(sdi->channels, devc->cur_acq_channel));
+               analog.num_samples = num_samples;
+               analog.data = samples;
+               analog.meaning->mq = SR_MQ_VOLTAGE;
+               analog.meaning->unit = SR_UNIT_VOLT;
+               analog.meaning->mqflags = 0;
+               packet.type = SR_DF_ANALOG;
+               packet.payload = &analog;
+               sr_session_send(sdi, &packet);
+               g_slist_free(analog.meaning->channels);
+
+               /* All channels acquired. */
+               if (devc->cur_acq_channel == ANALOG_CHANNELS - 1) {
+                       sr_spew("All channels acquired.");
+
+                       if (devc->cur_acq_frame == devc->frame_limit - 1) {
+                               /* All frames acquired. */
+                               sr_spew("All frames acquired.");
+                               sr_dev_acquisition_stop(sdi);
+                               return TRUE;
+                       } else {
+                               /* Start acquiring next frame. */
+                               if (devc->df_started) {
+                                       std_session_send_df_frame_end(sdi);
+                                       std_session_send_df_frame_begin(sdi);
+                               }
+                               devc->cur_acq_frame++;
+                               devc->state = START_ACQUISITION;
+                       }
+               } else {
+                       /* Start acquiring next channel. */
+                       devc->state = START_TRANSFER_OF_CHANNEL_DATA;
+                       devc->cur_acq_channel++;
+                       return TRUE;
+               }
+               break;
        }
 
        return TRUE;