]> sigrok.org Git - libsigrok.git/blame - hardware/rigol-ds/protocol.c
Rename rigol-ds1xx2 driver to rigol-ds.
[libsigrok.git] / hardware / rigol-ds / protocol.c
CommitLineData
f4816ac6
ML
1/*
2 * This file is part of the libsigrok project.
3 *
4 * Copyright (C) 2012 Martin Ling <martin-git@earth.li>
88e429c9 5 * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
f4816ac6
ML
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#include <stdlib.h>
e0b7d23c
ML
22#include <stdarg.h>
23#include <unistd.h>
24#include <errno.h>
a3df166f 25#include <string.h>
254dd102 26#include <math.h>
f4816ac6
ML
27#include <glib.h>
28#include "libsigrok.h"
29#include "libsigrok-internal.h"
30#include "protocol.h"
31
3086efdd 32SR_PRIV int rigol_ds_receive(int fd, int revents, void *cb_data)
f4816ac6 33{
e0b7d23c 34 struct sr_dev_inst *sdi;
9bd4c956 35 struct sr_serial_dev_inst *serial;
f4816ac6 36 struct dev_context *devc;
e0b7d23c
ML
37 struct sr_datafeed_packet packet;
38 struct sr_datafeed_analog analog;
6bb192bc
ML
39 struct sr_datafeed_logic logic;
40 unsigned char buf[DIGITAL_WAVEFORM_SIZE];
254dd102 41 double vdiv, offset;
6bb192bc
ML
42 float data[ANALOG_WAVEFORM_SIZE];
43 int len, i, waveform_size;
44 struct sr_probe *probe;
f4816ac6 45
decfe89d 46 (void)fd;
9bd4c956 47
f4816ac6
ML
48 if (!(sdi = cb_data))
49 return TRUE;
50
51 if (!(devc = sdi->priv))
52 return TRUE;
53
9bd4c956
ML
54 serial = sdi->conn;
55
f4816ac6 56 if (revents == G_IO_IN) {
6bb192bc
ML
57 probe = devc->channel_frame;
58 waveform_size = probe->type == SR_PROBE_ANALOG ?
59 ANALOG_WAVEFORM_SIZE : DIGITAL_WAVEFORM_SIZE;
9bd4c956 60 len = serial_read(serial, buf, waveform_size - devc->num_frame_bytes);
29d957ce 61 sr_dbg("Received %d bytes.", len);
e0b7d23c
ML
62 if (len == -1)
63 return TRUE;
75d8a4e5 64
ee7e9bee 65 if (devc->num_frame_bytes == 0) {
75d8a4e5
BV
66 /* Start of a new frame. */
67 packet.type = SR_DF_FRAME_BEGIN;
68 sr_session_send(sdi, &packet);
69 }
70
6bb192bc
ML
71 if (probe->type == SR_PROBE_ANALOG) {
72 for (i = 0; i < len; i++) {
73 vdiv = devc->vdiv[probe->index];
74 offset = devc->vert_offset[probe->index];
75 data[i] = vdiv / 25.6 * (128 - buf[i]) - offset;
76 }
77 analog.probes = g_slist_append(NULL, probe);
78 analog.num_samples = len;
79 analog.data = data;
80 analog.mq = SR_MQ_VOLTAGE;
81 analog.unit = SR_UNIT_VOLT;
82 analog.mqflags = 0;
83 packet.type = SR_DF_ANALOG;
84 packet.payload = &analog;
85 sr_session_send(cb_data, &packet);
86 g_slist_free(analog.probes);
87
88 if (len != ANALOG_WAVEFORM_SIZE)
89 /* Don't have the whole frame yet. */
90 return TRUE;
91 } else {
92 logic.length = len - 10;
93 logic.unitsize = 2;
94 logic.data = buf + 10;
95 packet.type = SR_DF_LOGIC;
96 packet.payload = &logic;
97 sr_session_send(cb_data, &packet);
98
99 if (len != DIGITAL_WAVEFORM_SIZE)
100 /* Don't have the whole frame yet. */
101 return TRUE;
ee7e9bee 102 }
75d8a4e5 103
254dd102
BV
104 /* End of the frame. */
105 packet.type = SR_DF_FRAME_END;
106 sr_session_send(sdi, &packet);
ee7e9bee 107 devc->num_frame_bytes = 0;
254dd102 108
6bb192bc
ML
109 if (devc->enabled_analog_probes
110 && devc->channel_frame == devc->enabled_analog_probes->data
111 && devc->enabled_analog_probes->next != NULL) {
112 /* We got the frame for the first analog channel, but
113 * there's a second analog channel. */
114 devc->channel_frame = devc->enabled_analog_probes->next->data;
3086efdd 115 rigol_ds_send(sdi, ":WAV:DATA? CHAN%c",
254dd102
BV
116 devc->channel_frame->name[2]);
117 } else {
6bb192bc
ML
118 /* Done with both analog channels in this frame. */
119 if (devc->enabled_digital_probes
120 && devc->channel_frame != devc->enabled_digital_probes->data) {
121 /* Now we need to get the digital data. */
122 devc->channel_frame = devc->enabled_digital_probes->data;
3086efdd 123 rigol_ds_send(sdi, ":WAV:DATA? DIG");
6bb192bc
ML
124 } else if (++devc->num_frames == devc->limit_frames) {
125 /* End of last frame. */
1130e422
DJ
126 packet.type = SR_DF_END;
127 sr_session_send(sdi, &packet);
75d8a4e5 128 sdi->driver->dev_acquisition_stop(sdi, cb_data);
254dd102 129 } else {
6bb192bc
ML
130 /* Get the next frame, starting with the first analog channel. */
131 if (devc->enabled_analog_probes) {
132 devc->channel_frame = devc->enabled_analog_probes->data;
3086efdd 133 rigol_ds_send(sdi, ":WAV:DATA? CHAN%c",
6bb192bc
ML
134 devc->channel_frame->name[2]);
135 } else {
136 devc->channel_frame = devc->enabled_digital_probes->data;
3086efdd 137 rigol_ds_send(sdi, ":WAV:DATA? DIG");
6bb192bc 138 }
254dd102 139 }
75d8a4e5 140 }
f4816ac6
ML
141 }
142
143 return TRUE;
144}
e0b7d23c 145
3086efdd 146SR_PRIV int rigol_ds_send(const struct sr_dev_inst *sdi, const char *format, ...)
e0b7d23c
ML
147{
148 va_list args;
149 char buf[256];
a3df166f 150 int len, out, ret;
29d957ce 151
e0b7d23c 152 va_start(args, format);
254dd102 153 len = vsnprintf(buf, 255, format, args);
e0b7d23c 154 va_end(args);
a3df166f
BV
155 strcat(buf, "\n");
156 len++;
9bd4c956 157 out = serial_write(sdi->conn, buf, len);
a3df166f
BV
158 buf[len - 1] = '\0';
159 if (out != len) {
160 sr_dbg("Only sent %d/%d bytes of '%s'.", out, len, buf);
161 ret = SR_ERR;
162 } else {
254dd102 163 sr_spew("Sent '%s'.", buf);
a3df166f
BV
164 ret = SR_OK;
165 }
29d957ce 166
a3df166f 167 return ret;
e0b7d23c 168}
254dd102
BV
169
170static int get_cfg(const struct sr_dev_inst *sdi, char *cmd, char *reply)
171{
254dd102
BV
172 int len;
173
3086efdd 174 if (rigol_ds_send(sdi, cmd) != SR_OK)
254dd102
BV
175 return SR_ERR;
176
9bd4c956 177 if ((len = serial_read(sdi->conn, reply, 255)) < 0)
254dd102
BV
178 return SR_ERR;
179 reply[len] = '\0';
180 sr_spew("Received '%s'.", reply);
181
182 return SR_OK;
183}
184
185static int get_cfg_float(const struct sr_dev_inst *sdi, char *cmd, float *f)
186{
187 char buf[256], *e;
188
189 if (get_cfg(sdi, cmd, buf) != SR_OK)
190 return SR_ERR;
191 *f = strtof(buf, &e);
169dbe85 192 if (e == buf || (fpclassify(*f) & (FP_ZERO | FP_NORMAL)) == 0) {
254dd102
BV
193 sr_dbg("failed to parse response to '%s': '%s'", cmd, buf);
194 return SR_ERR;
195 }
196
197 return SR_OK;
198}
199
200static int get_cfg_string(const struct sr_dev_inst *sdi, char *cmd, char **buf)
201{
202
203 if (!(*buf = g_try_malloc0(256)))
169dbe85 204 return SR_ERR_MALLOC;
254dd102
BV
205
206 if (get_cfg(sdi, cmd, *buf) != SR_OK)
207 return SR_ERR;
208
209 return SR_OK;
210}
211
3086efdd 212SR_PRIV int rigol_ds_get_dev_cfg(const struct sr_dev_inst *sdi)
254dd102
BV
213{
214 struct dev_context *devc;
6bb192bc
ML
215 char *t_s, *cmd;
216 int i, res;
254dd102
BV
217
218 devc = sdi->priv;
219
6bb192bc 220 /* Analog channel state. */
254dd102
BV
221 if (get_cfg_string(sdi, ":CHAN1:DISP?", &t_s) != SR_OK)
222 return SR_ERR;
6bb192bc 223 devc->analog_channels[0] = !strcmp(t_s, "ON") ? TRUE : FALSE;
254dd102
BV
224 g_free(t_s);
225 if (get_cfg_string(sdi, ":CHAN2:DISP?", &t_s) != SR_OK)
226 return SR_ERR;
6bb192bc 227 devc->analog_channels[1] = !strcmp(t_s, "ON") ? TRUE : FALSE;
254dd102 228 g_free(t_s);
6bb192bc
ML
229 sr_dbg("Current analog channel state CH1 %s CH2 %s",
230 devc->analog_channels[0] ? "on" : "off",
231 devc->analog_channels[1] ? "on" : "off");
232
233 /* Digital channel state. */
234 if (devc->has_digital) {
235 sr_dbg("Current digital channel state:");
236 for (i = 0; i < 16; i++) {
237 cmd = g_strdup_printf(":DIG%d:TURN?", i);
238 res = get_cfg_string(sdi, cmd, &t_s);
239 g_free(cmd);
240 if (res != SR_OK)
241 return SR_ERR;
242 devc->digital_channels[i] = !strcmp(t_s, "ON") ? TRUE : FALSE;
243 g_free(t_s);
244 sr_dbg("D%d: %s", i, devc->digital_channels[i] ? "on" : "off");
245 }
246 }
254dd102
BV
247
248 /* Timebase. */
249 if (get_cfg_float(sdi, ":TIM:SCAL?", &devc->timebase) != SR_OK)
250 return SR_ERR;
251 sr_dbg("Current timebase %f", devc->timebase);
252
253 /* Vertical gain. */
254 if (get_cfg_float(sdi, ":CHAN1:SCAL?", &devc->vdiv[0]) != SR_OK)
255 return SR_ERR;
256 if (get_cfg_float(sdi, ":CHAN2:SCAL?", &devc->vdiv[1]) != SR_OK)
257 return SR_ERR;
258 sr_dbg("Current vertical gain CH1 %f CH2 %f", devc->vdiv[0], devc->vdiv[1]);
259
260 /* Vertical offset. */
261 if (get_cfg_float(sdi, ":CHAN1:OFFS?", &devc->vert_offset[0]) != SR_OK)
262 return SR_ERR;
263 if (get_cfg_float(sdi, ":CHAN2:OFFS?", &devc->vert_offset[1]) != SR_OK)
264 return SR_ERR;
265 sr_dbg("Current vertical offset CH1 %f CH2 %f", devc->vert_offset[0],
266 devc->vert_offset[1]);
267
268 /* Coupling. */
269 if (get_cfg_string(sdi, ":CHAN1:COUP?", &devc->coupling[0]) != SR_OK)
270 return SR_ERR;
271 if (get_cfg_string(sdi, ":CHAN2:COUP?", &devc->coupling[1]) != SR_OK)
272 return SR_ERR;
273 sr_dbg("Current coupling CH1 %s CH2 %s", devc->coupling[0],
274 devc->coupling[1]);
275
276 /* Trigger source. */
277 if (get_cfg_string(sdi, ":TRIG:EDGE:SOUR?", &devc->trigger_source) != SR_OK)
278 return SR_ERR;
279 sr_dbg("Current trigger source %s", devc->trigger_source);
280
281 /* Horizontal trigger position. */
282 if (get_cfg_float(sdi, ":TIM:OFFS?", &devc->horiz_triggerpos) != SR_OK)
283 return SR_ERR;
284 sr_dbg("Current horizontal trigger position %f", devc->horiz_triggerpos);
285
286 /* Trigger slope. */
287 if (get_cfg_string(sdi, ":TRIG:EDGE:SLOP?", &devc->trigger_slope) != SR_OK)
288 return SR_ERR;
289 sr_dbg("Current trigger slope %s", devc->trigger_slope);
290
291 return SR_OK;
292}