rigol-ds1xx2: support digital channels.

This commit is contained in:
Martin Ling 2013-04-14 02:21:55 +01:00 committed by Bert Vermeulen
parent a789b61087
commit 6bb192bc05
3 changed files with 166 additions and 67 deletions

View File

@ -102,6 +102,22 @@ static const char *trigger_sources[] = {
"CH2", "CH2",
"EXT", "EXT",
"AC Line", "AC Line",
"D0",
"D1",
"D2",
"D3",
"D4",
"D5",
"D6",
"D7",
"D8",
"D9",
"D10",
"D11",
"D12",
"D13",
"D14",
"D15",
}; };
static const char *coupling[] = { static const char *coupling[] = {
@ -199,7 +215,9 @@ static GSList *hw_scan(GSList *options)
gchar **tokens; gchar **tokens;
const char *manufacturer, *model, *version; const char *manufacturer, *model, *version;
gboolean matched = FALSE; gboolean matched = FALSE;
gboolean has_digital = FALSE;
char buf[256]; char buf[256];
gchar *channel_name;
(void)options; (void)options;
@ -254,6 +272,7 @@ static GSList *hw_scan(GSList *options)
for (i = 0; i < ARRAY_SIZE(supported_models); i++) { for (i = 0; i < ARRAY_SIZE(supported_models); i++) {
if (!strcmp(model, supported_models[i])) { if (!strcmp(model, supported_models[i])) {
matched = 1; matched = 1;
has_digital = g_str_has_suffix(model, "D");
break; break;
} }
} }
@ -274,16 +293,29 @@ static GSList *hw_scan(GSList *options)
} }
devc->limit_frames = 0; devc->limit_frames = 0;
devc->device = device; devc->device = device;
devc->has_digital = has_digital;
sdi->priv = devc; sdi->priv = devc;
sdi->driver = di; sdi->driver = di;
for (i = 0; i < 2; i++) { for (i = 0; i < 2; i++) {
if (!(probe = sr_probe_new(0, SR_PROBE_ANALOG, TRUE, if (!(probe = sr_probe_new(i, SR_PROBE_ANALOG, TRUE,
i == 0 ? "CH1" : "CH2"))) i == 0 ? "CH1" : "CH2")))
return NULL; return NULL;
sdi->probes = g_slist_append(sdi->probes, probe); sdi->probes = g_slist_append(sdi->probes, probe);
} }
if (devc->has_digital) {
for (i = 0; i < 16; i++) {
if (!(channel_name = g_strdup_printf("D%d", i)))
return NULL;
probe = sr_probe_new(i, SR_PROBE_LOGIC, TRUE, channel_name);
g_free(channel_name);
if (!probe)
return NULL;
sdi->probes = g_slist_append(sdi->probes, probe);
}
}
drvc->instances = g_slist_append(drvc->instances, sdi); drvc->instances = g_slist_append(drvc->instances, sdi);
devices = g_slist_append(devices, sdi); devices = g_slist_append(devices, sdi);
} }
@ -469,8 +501,7 @@ static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
GVariant *tuple, *rational[2]; GVariant *tuple, *rational[2];
GVariantBuilder gvb; GVariantBuilder gvb;
unsigned int i; unsigned int i;
struct dev_context *devc = sdi->priv;
(void)sdi;
switch (key) { switch (key) {
case SR_CONF_DEVICE_OPTIONS: case SR_CONF_DEVICE_OPTIONS:
@ -502,7 +533,7 @@ static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
break; break;
case SR_CONF_TRIGGER_SOURCE: case SR_CONF_TRIGGER_SOURCE:
*data = g_variant_new_strv(trigger_sources, *data = g_variant_new_strv(trigger_sources,
ARRAY_SIZE(trigger_sources)); devc->has_digital ? ARRAY_SIZE(trigger_sources) : 4);
break; break;
default: default:
return SR_ERR_ARG; return SR_ERR_ARG;
@ -516,7 +547,6 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
struct dev_context *devc; struct dev_context *devc;
struct sr_probe *probe; struct sr_probe *probe;
GSList *l; GSList *l;
int probenum;
char cmd[256]; char cmd[256];
(void)cb_data; (void)cb_data;
@ -525,19 +555,32 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
for (l = sdi->probes; l; l = l->next) { for (l = sdi->probes; l; l = l->next) {
probe = l->data; probe = l->data;
probenum = probe->name[2] == '1' ? 0 : 1; sr_dbg("handling probe %s", probe->name);
if (probe->type == SR_PROBE_ANALOG) {
if (probe->enabled) if (probe->enabled)
devc->enabled_probes = g_slist_append(devc->enabled_probes, probe); devc->enabled_analog_probes = g_slist_append(
devc->enabled_analog_probes, probe);
if (probe->enabled != devc->channels[probenum]) { if (probe->enabled != devc->analog_channels[probe->index]) {
/* Enabled channel is currently disabled, or vice versa. */ /* Enabled channel is currently disabled, or vice versa. */
sprintf(cmd, ":CHAN%d:DISP %s", probenum + 1, sprintf(cmd, ":CHAN%d:DISP %s", probe->index + 1,
probe->enabled ? "ON" : "OFF");
if (rigol_ds1xx2_send(devc, cmd) != SR_OK)
return SR_ERR;
}
} else if (probe->type == SR_PROBE_LOGIC) {
if (probe->enabled)
devc->enabled_digital_probes = g_slist_append(
devc->enabled_digital_probes, probe);
if (probe->enabled != devc->digital_channels[probe->index]) {
/* Enabled channel is currently disabled, or vice versa. */
sprintf(cmd, ":DIG%d:TURN %s", probe->index,
probe->enabled ? "ON" : "OFF"); probe->enabled ? "ON" : "OFF");
if (rigol_ds1xx2_send(devc, cmd) != SR_OK) if (rigol_ds1xx2_send(devc, cmd) != SR_OK)
return SR_ERR; return SR_ERR;
} }
} }
if (!devc->enabled_probes) }
if (!devc->enabled_analog_probes && !devc->enabled_digital_probes)
return SR_ERR; return SR_ERR;
sr_source_add(devc->fd, G_IO_IN, 50, rigol_ds1xx2_receive, (void *)sdi); sr_source_add(devc->fd, G_IO_IN, 50, rigol_ds1xx2_receive, (void *)sdi);
@ -546,10 +589,16 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
std_session_send_df_header(cb_data, DRIVER_LOG_DOMAIN); std_session_send_df_header(cb_data, DRIVER_LOG_DOMAIN);
/* Fetch the first frame. */ /* Fetch the first frame. */
devc->channel_frame = devc->enabled_probes->data; if (devc->enabled_analog_probes) {
if (rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%c", devc->channel_frame = devc->enabled_analog_probes->data;
devc->channel_frame->name[2]) != SR_OK) if (rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%d",
devc->channel_frame->index + 1) != SR_OK)
return SR_ERR; return SR_ERR;
} else {
devc->channel_frame = devc->enabled_digital_probes->data;
if (rigol_ds1xx2_send(devc, ":WAV:DATA? DIG") != SR_OK)
return SR_ERR;
}
devc->num_frame_bytes = 0; devc->num_frame_bytes = 0;
@ -569,8 +618,10 @@ static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
return SR_ERR; return SR_ERR;
} }
g_slist_free(devc->enabled_probes); g_slist_free(devc->enabled_analog_probes);
devc->enabled_probes = NULL; g_slist_free(devc->enabled_digital_probes);
devc->enabled_analog_probes = NULL;
devc->enabled_digital_probes = NULL;
sr_source_remove(devc->fd); sr_source_remove(devc->fd);
return SR_OK; return SR_OK;

View File

@ -35,10 +35,12 @@ SR_PRIV int rigol_ds1xx2_receive(int fd, int revents, void *cb_data)
struct dev_context *devc; struct dev_context *devc;
struct sr_datafeed_packet packet; struct sr_datafeed_packet packet;
struct sr_datafeed_analog analog; struct sr_datafeed_analog analog;
unsigned char buf[WAVEFORM_SIZE]; struct sr_datafeed_logic logic;
unsigned char buf[DIGITAL_WAVEFORM_SIZE];
double vdiv, offset; double vdiv, offset;
float data[WAVEFORM_SIZE]; float data[ANALOG_WAVEFORM_SIZE];
int probenum, len, i; int len, i, waveform_size;
struct sr_probe *probe;
if (!(sdi = cb_data)) if (!(sdi = cb_data))
return TRUE; return TRUE;
@ -47,7 +49,10 @@ SR_PRIV int rigol_ds1xx2_receive(int fd, int revents, void *cb_data)
return TRUE; return TRUE;
if (revents == G_IO_IN) { if (revents == G_IO_IN) {
len = read(fd, buf, WAVEFORM_SIZE - devc->num_frame_bytes); probe = devc->channel_frame;
waveform_size = probe->type == SR_PROBE_ANALOG ?
ANALOG_WAVEFORM_SIZE : DIGITAL_WAVEFORM_SIZE;
len = read(fd, buf, waveform_size - devc->num_frame_bytes);
sr_dbg("Received %d bytes.", len); sr_dbg("Received %d bytes.", len);
if (len == -1) if (len == -1)
return TRUE; return TRUE;
@ -58,13 +63,13 @@ SR_PRIV int rigol_ds1xx2_receive(int fd, int revents, void *cb_data)
sr_session_send(sdi, &packet); sr_session_send(sdi, &packet);
} }
probenum = devc->channel_frame->name[2] == '1' ? 0 : 1; if (probe->type == SR_PROBE_ANALOG) {
for (i = 0; i < len; i++) { for (i = 0; i < len; i++) {
vdiv = devc->vdiv[probenum]; vdiv = devc->vdiv[probe->index];
offset = devc->vert_offset[probenum]; offset = devc->vert_offset[probe->index];
data[i] = vdiv / 25.6 * (128 - buf[i]) - offset; data[i] = vdiv / 25.6 * (128 - buf[i]) - offset;
} }
analog.probes = g_slist_append(NULL, devc->channel_frame); analog.probes = g_slist_append(NULL, probe);
analog.num_samples = len; analog.num_samples = len;
analog.data = data; analog.data = data;
analog.mq = SR_MQ_VOLTAGE; analog.mq = SR_MQ_VOLTAGE;
@ -75,9 +80,18 @@ SR_PRIV int rigol_ds1xx2_receive(int fd, int revents, void *cb_data)
sr_session_send(cb_data, &packet); sr_session_send(cb_data, &packet);
g_slist_free(analog.probes); g_slist_free(analog.probes);
devc->num_frame_bytes += len; if (len != ANALOG_WAVEFORM_SIZE)
/* Don't have the whole frame yet. */
return TRUE;
} else {
logic.length = len - 10;
logic.unitsize = 2;
logic.data = buf + 10;
packet.type = SR_DF_LOGIC;
packet.payload = &logic;
sr_session_send(cb_data, &packet);
if (devc->num_frame_bytes != WAVEFORM_SIZE) { if (len != DIGITAL_WAVEFORM_SIZE)
/* Don't have the whole frame yet. */ /* Don't have the whole frame yet. */
return TRUE; return TRUE;
} }
@ -87,22 +101,34 @@ SR_PRIV int rigol_ds1xx2_receive(int fd, int revents, void *cb_data)
sr_session_send(sdi, &packet); sr_session_send(sdi, &packet);
devc->num_frame_bytes = 0; devc->num_frame_bytes = 0;
if (devc->channel_frame == devc->enabled_probes->data if (devc->enabled_analog_probes
&& devc->enabled_probes->next != NULL) { && devc->channel_frame == devc->enabled_analog_probes->data
/* We got the frame for the first channel, but && devc->enabled_analog_probes->next != NULL) {
* there's a second channel. */ /* We got the frame for the first analog channel, but
devc->channel_frame = devc->enabled_probes->next->data; * there's a second analog channel. */
devc->channel_frame = devc->enabled_analog_probes->next->data;
rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%c", rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%c",
devc->channel_frame->name[2]); devc->channel_frame->name[2]);
} else { } else {
/* Done with both channels in this frame. */ /* Done with both analog channels in this frame. */
if (++devc->num_frames == devc->limit_frames) { if (devc->enabled_digital_probes
&& devc->channel_frame != devc->enabled_digital_probes->data) {
/* Now we need to get the digital data. */
devc->channel_frame = devc->enabled_digital_probes->data;
rigol_ds1xx2_send(devc, ":WAV:DATA? DIG");
} else if (++devc->num_frames == devc->limit_frames) {
/* End of last frame. */
sdi->driver->dev_acquisition_stop(sdi, cb_data); sdi->driver->dev_acquisition_stop(sdi, cb_data);
} else { } else {
/* Get the next frame, starting with the first channel. */ /* Get the next frame, starting with the first analog channel. */
devc->channel_frame = devc->enabled_probes->data; if (devc->enabled_analog_probes) {
devc->channel_frame = devc->enabled_analog_probes->data;
rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%c", rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%c",
devc->channel_frame->name[2]); devc->channel_frame->name[2]);
} else {
devc->channel_frame = devc->enabled_digital_probes->data;
rigol_ds1xx2_send(devc, ":WAV:DATA? DIG");
}
} }
} }
} }
@ -182,22 +208,38 @@ static int get_cfg_string(const struct sr_dev_inst *sdi, char *cmd, char **buf)
SR_PRIV int rigol_ds1xx2_get_dev_cfg(const struct sr_dev_inst *sdi) SR_PRIV int rigol_ds1xx2_get_dev_cfg(const struct sr_dev_inst *sdi)
{ {
struct dev_context *devc; struct dev_context *devc;
char *t_s; char *t_s, *cmd;
int i, res;
devc = sdi->priv; devc = sdi->priv;
/* Channel state. */ /* Analog channel state. */
if (get_cfg_string(sdi, ":CHAN1:DISP?", &t_s) != SR_OK) if (get_cfg_string(sdi, ":CHAN1:DISP?", &t_s) != SR_OK)
return SR_ERR; return SR_ERR;
devc->channels[0] = !strcmp(t_s, "ON") ? TRUE : FALSE; devc->analog_channels[0] = !strcmp(t_s, "ON") ? TRUE : FALSE;
g_free(t_s); g_free(t_s);
if (get_cfg_string(sdi, ":CHAN2:DISP?", &t_s) != SR_OK) if (get_cfg_string(sdi, ":CHAN2:DISP?", &t_s) != SR_OK)
return SR_ERR; return SR_ERR;
devc->channels[1] = !strcmp(t_s, "ON") ? TRUE : FALSE; devc->analog_channels[1] = !strcmp(t_s, "ON") ? TRUE : FALSE;
g_free(t_s); g_free(t_s);
sr_dbg("Current channel state CH1 %s CH2 %s", sr_dbg("Current analog channel state CH1 %s CH2 %s",
devc->channels[0] ? "on" : "off", devc->analog_channels[0] ? "on" : "off",
devc->channels[1] ? "on" : "off"); devc->analog_channels[1] ? "on" : "off");
/* Digital channel state. */
if (devc->has_digital) {
sr_dbg("Current digital channel state:");
for (i = 0; i < 16; i++) {
cmd = g_strdup_printf(":DIG%d:TURN?", i);
res = get_cfg_string(sdi, cmd, &t_s);
g_free(cmd);
if (res != SR_OK)
return SR_ERR;
devc->digital_channels[i] = !strcmp(t_s, "ON") ? TRUE : FALSE;
g_free(t_s);
sr_dbg("D%d: %s", i, devc->digital_channels[i] ? "on" : "off");
}
}
/* Timebase. */ /* Timebase. */
if (get_cfg_float(sdi, ":TIM:SCAL?", &devc->timebase) != SR_OK) if (get_cfg_float(sdi, ":TIM:SCAL?", &devc->timebase) != SR_OK)

View File

@ -34,17 +34,23 @@
#define sr_warn(s, args...) sr_warn(DRIVER_LOG_DOMAIN s, ## args) #define sr_warn(s, args...) sr_warn(DRIVER_LOG_DOMAIN s, ## args)
#define sr_err(s, args...) sr_err(DRIVER_LOG_DOMAIN s, ## args) #define sr_err(s, args...) sr_err(DRIVER_LOG_DOMAIN s, ## args)
#define WAVEFORM_SIZE 600 #define ANALOG_WAVEFORM_SIZE 600
#define DIGITAL_WAVEFORM_SIZE 1210
/** Private, per-device-instance driver context. */ /** Private, per-device-instance driver context. */
struct dev_context { struct dev_context {
/* Device features */
gboolean has_digital;
/* Acquisition settings */ /* Acquisition settings */
GSList *enabled_probes; GSList *enabled_analog_probes;
GSList *enabled_digital_probes;
uint64_t limit_frames; uint64_t limit_frames;
void *cb_data; void *cb_data;
/* Device settings */ /* Device settings */
gboolean channels[2]; gboolean analog_channels[2];
gboolean digital_channels[16];
float timebase; float timebase;
float vdiv[2]; float vdiv[2];
float vert_offset[2]; float vert_offset[2];