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>
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.
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.
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/>.
28 #include "libsigrok.h"
29 #include "libsigrok-internal.h"
32 SR_PRIV int rigol_ds_receive(int fd, int revents, void *cb_data)
34 struct sr_dev_inst *sdi;
35 struct sr_serial_dev_inst *serial;
36 struct dev_context *devc;
37 struct sr_datafeed_packet packet;
38 struct sr_datafeed_analog analog;
39 struct sr_datafeed_logic logic;
40 unsigned char buf[DIGITAL_WAVEFORM_SIZE];
42 float data[ANALOG_WAVEFORM_SIZE];
43 int len, i, waveform_size;
44 struct sr_probe *probe;
51 if (!(devc = sdi->priv))
56 if (revents == G_IO_IN) {
57 probe = devc->channel_frame;
58 waveform_size = probe->type == SR_PROBE_ANALOG ?
59 ANALOG_WAVEFORM_SIZE : DIGITAL_WAVEFORM_SIZE;
60 len = serial_read(serial, buf, waveform_size - devc->num_frame_bytes);
61 sr_dbg("Received %d bytes.", len);
65 if (devc->num_frame_bytes == 0) {
66 /* Start of a new frame. */
67 packet.type = SR_DF_FRAME_BEGIN;
68 sr_session_send(sdi, &packet);
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;
77 analog.probes = g_slist_append(NULL, probe);
78 analog.num_samples = len;
80 analog.mq = SR_MQ_VOLTAGE;
81 analog.unit = SR_UNIT_VOLT;
83 packet.type = SR_DF_ANALOG;
84 packet.payload = &analog;
85 sr_session_send(cb_data, &packet);
86 g_slist_free(analog.probes);
88 if (len != ANALOG_WAVEFORM_SIZE)
89 /* Don't have the whole frame yet. */
92 logic.length = len - 10;
94 logic.data = buf + 10;
95 packet.type = SR_DF_LOGIC;
96 packet.payload = &logic;
97 sr_session_send(cb_data, &packet);
99 if (len != DIGITAL_WAVEFORM_SIZE)
100 /* Don't have the whole frame yet. */
104 /* End of the frame. */
105 packet.type = SR_DF_FRAME_END;
106 sr_session_send(sdi, &packet);
107 devc->num_frame_bytes = 0;
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;
115 rigol_ds_send(sdi, ":WAV:DATA? CHAN%c",
116 devc->channel_frame->name[2]);
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;
123 rigol_ds_send(sdi, ":WAV:DATA? DIG");
124 } else if (++devc->num_frames == devc->limit_frames) {
125 /* End of last frame. */
126 packet.type = SR_DF_END;
127 sr_session_send(sdi, &packet);
128 sdi->driver->dev_acquisition_stop(sdi, cb_data);
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;
133 rigol_ds_send(sdi, ":WAV:DATA? CHAN%c",
134 devc->channel_frame->name[2]);
136 devc->channel_frame = devc->enabled_digital_probes->data;
137 rigol_ds_send(sdi, ":WAV:DATA? DIG");
146 SR_PRIV int rigol_ds_send(const struct sr_dev_inst *sdi, const char *format, ...)
152 va_start(args, format);
153 len = vsnprintf(buf, 255, format, args);
157 out = serial_write(sdi->conn, buf, len);
160 sr_dbg("Only sent %d/%d bytes of '%s'.", out, len, buf);
163 sr_spew("Sent '%s'.", buf);
170 static int get_cfg(const struct sr_dev_inst *sdi, char *cmd, char *reply)
174 if (rigol_ds_send(sdi, cmd) != SR_OK)
177 if ((len = serial_read(sdi->conn, reply, 255)) < 0)
180 sr_spew("Received '%s'.", reply);
185 static int get_cfg_float(const struct sr_dev_inst *sdi, char *cmd, float *f)
189 if (get_cfg(sdi, cmd, buf) != SR_OK)
191 *f = strtof(buf, &e);
192 if (e == buf || (fpclassify(*f) & (FP_ZERO | FP_NORMAL)) == 0) {
193 sr_dbg("failed to parse response to '%s': '%s'", cmd, buf);
200 static int get_cfg_string(const struct sr_dev_inst *sdi, char *cmd, char **buf)
203 if (!(*buf = g_try_malloc0(256)))
204 return SR_ERR_MALLOC;
206 if (get_cfg(sdi, cmd, *buf) != SR_OK)
212 SR_PRIV int rigol_ds_get_dev_cfg(const struct sr_dev_inst *sdi)
214 struct dev_context *devc;
220 /* Analog channel state. */
221 if (get_cfg_string(sdi, ":CHAN1:DISP?", &t_s) != SR_OK)
223 devc->analog_channels[0] = !strcmp(t_s, "ON") ? TRUE : FALSE;
225 if (get_cfg_string(sdi, ":CHAN2:DISP?", &t_s) != SR_OK)
227 devc->analog_channels[1] = !strcmp(t_s, "ON") ? TRUE : FALSE;
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");
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);
242 devc->digital_channels[i] = !strcmp(t_s, "ON") ? TRUE : FALSE;
244 sr_dbg("D%d: %s", i, devc->digital_channels[i] ? "on" : "off");
249 if (get_cfg_float(sdi, ":TIM:SCAL?", &devc->timebase) != SR_OK)
251 sr_dbg("Current timebase %f", devc->timebase);
254 if (get_cfg_float(sdi, ":CHAN1:SCAL?", &devc->vdiv[0]) != SR_OK)
256 if (get_cfg_float(sdi, ":CHAN2:SCAL?", &devc->vdiv[1]) != SR_OK)
258 sr_dbg("Current vertical gain CH1 %f CH2 %f", devc->vdiv[0], devc->vdiv[1]);
260 /* Vertical offset. */
261 if (get_cfg_float(sdi, ":CHAN1:OFFS?", &devc->vert_offset[0]) != SR_OK)
263 if (get_cfg_float(sdi, ":CHAN2:OFFS?", &devc->vert_offset[1]) != SR_OK)
265 sr_dbg("Current vertical offset CH1 %f CH2 %f", devc->vert_offset[0],
266 devc->vert_offset[1]);
269 if (get_cfg_string(sdi, ":CHAN1:COUP?", &devc->coupling[0]) != SR_OK)
271 if (get_cfg_string(sdi, ":CHAN2:COUP?", &devc->coupling[1]) != SR_OK)
273 sr_dbg("Current coupling CH1 %s CH2 %s", devc->coupling[0],
276 /* Trigger source. */
277 if (get_cfg_string(sdi, ":TRIG:EDGE:SOUR?", &devc->trigger_source) != SR_OK)
279 sr_dbg("Current trigger source %s", devc->trigger_source);
281 /* Horizontal trigger position. */
282 if (get_cfg_float(sdi, ":TIM:OFFS?", &devc->horiz_triggerpos) != SR_OK)
284 sr_dbg("Current horizontal trigger position %f", devc->horiz_triggerpos);
287 if (get_cfg_string(sdi, ":TRIG:EDGE:SLOP?", &devc->trigger_slope) != SR_OK)
289 sr_dbg("Current trigger slope %s", devc->trigger_slope);