X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=hardware%2Frigol-ds1xx2%2Fapi.c;h=34081f3db8cf4fe67de51f54d62b610cf56e1901;hb=29a27196a13de2ffd9b671185e4b464b9db9b549;hp=cdbdaad66ee9a01e48be11fb8404185f66705b7a;hpb=169dbe8577c2f7570f61799a064f95b39b3b6fe3;p=libsigrok.git

diff --git a/hardware/rigol-ds1xx2/api.c b/hardware/rigol-ds1xx2/api.c
index cdbdaad6..34081f3d 100644
--- a/hardware/rigol-ds1xx2/api.c
+++ b/hardware/rigol-ds1xx2/api.c
@@ -22,7 +22,6 @@
 #include <unistd.h>
 #include <stdlib.h>
 #include <string.h>
-#include <time.h>
 #include <glib.h>
 #include "libsigrok.h"
 #include "libsigrok-internal.h"
@@ -31,9 +30,12 @@
 #define NUM_TIMEBASE  12
 #define NUM_VDIV      8
 
+static const int32_t hwopts[] = {
+	SR_CONF_CONN,
+};
+
 static const int32_t hwcaps[] = {
 	SR_CONF_OSCILLOSCOPE,
-	SR_CONF_LIMIT_SAMPLES,
 	SR_CONF_TIMEBASE,
 	SR_CONF_TRIGGER_SOURCE,
 	SR_CONF_TRIGGER_SLOPE,
@@ -104,6 +106,22 @@ static const char *trigger_sources[] = {
 	"CH2",
 	"EXT",
 	"AC Line",
+	"D0",
+	"D1",
+	"D2",
+	"D3",
+	"D4",
+	"D5",
+	"D6",
+	"D7",
+	"D8",
+	"D9",
+	"D10",
+	"D11",
+	"D12",
+	"D13",
+	"D14",
+	"D15",
 };
 
 static const char *coupling[] = {
@@ -136,16 +154,16 @@ static int clear_instances(void)
 	for (l = drvc->instances; l; l = l->next) {
 		if (!(sdi = l->data))
 			continue;
-		if (!(devc = sdi->priv))
-			continue;
 
-		g_free(devc->device);
-		g_free(devc->coupling[0]);
-		g_free(devc->coupling[1]);
-		g_free(devc->trigger_source);
-		g_free(devc->trigger_slope);
-		close(devc->fd);
+		if (sdi->conn)
+			sr_serial_dev_inst_free(sdi->conn);
 
+		if ((devc = sdi->priv)) {
+			g_free(devc->coupling[0]);
+			g_free(devc->coupling[1]);
+			g_free(devc->trigger_source);
+			g_free(devc->trigger_slope);
+		}
 		sr_dev_inst_free(sdi);
 	}
 
@@ -157,143 +175,172 @@ static int clear_instances(void)
 
 static int set_cfg(const struct sr_dev_inst *sdi, const char *format, ...)
 {
-	struct dev_context *devc;
-	struct timespec delay;
 	va_list args;
 	char buf[256];
 
-	devc = sdi->priv;
-
 	va_start(args, format);
 	vsnprintf(buf, 255, format, args);
 	va_end(args);
-	if (rigol_ds1xx2_send(devc, buf) != SR_OK)
+	if (rigol_ds1xx2_send(sdi, buf) != SR_OK)
 		return SR_ERR;
 
 	/* When setting a bunch of parameters in a row, the DS1052E scrambles
 	 * some of them unless there is at least 100ms delay in between. */
-	delay.tv_sec = 0;
-	delay.tv_nsec = 100000000L;
-	sr_spew("delay %dms", delay.tv_nsec / 1000000);
-	nanosleep(&delay, NULL);
+	sr_spew("delay %dms", 100);
+	g_usleep(100000);
 
 	return SR_OK;
 }
 
 static int hw_init(struct sr_context *sr_ctx)
 {
-	return std_hw_init(sr_ctx, di, DRIVER_LOG_DOMAIN);
+	return std_hw_init(sr_ctx, di, LOG_PREFIX);
 }
 
-static GSList *hw_scan(GSList *options)
+static int probe_port(const char *port, GSList **devices)
 {
-	struct drv_context *drvc;
-	struct sr_dev_inst *sdi;
 	struct dev_context *devc;
+	struct sr_dev_inst *sdi;
+	struct sr_serial_dev_inst *serial;
 	struct sr_probe *probe;
-	GSList *devices;
-	GDir *dir;
-	const gchar *dev_name;
-	const gchar *dev_dir = "/dev/";
-	const gchar *prefix = "usbtmc";
-	gchar *device;
-	const gchar *idn_query = "*IDN?";
 	unsigned int i;
-	int len, num_tokens, fd;
-	const gchar *delimiter = ",";
-	gchar **tokens;
+	int len, num_tokens;
+	gboolean matched, has_digital;
 	const char *manufacturer, *model, *version;
-	gboolean matched = FALSE;
 	char buf[256];
+	gchar **tokens, *channel_name;
 
-	(void)options;
+	*devices = NULL;
+	if (!(serial = sr_serial_dev_inst_new(port, NULL)))
+		return SR_ERR_MALLOC;
 
-	drvc = di->priv;
-	drvc->instances = NULL;
+	if (serial_open(serial, SERIAL_RDWR) != SR_OK)
+		return SR_ERR;
+	len = serial_write(serial, "*IDN?", 5);
+	len = serial_read(serial, buf, sizeof(buf));
+	if (serial_close(serial) != SR_OK)
+		return SR_ERR;
 
-	devices = NULL;
+	sr_serial_dev_inst_free(serial);
 
-	dir = g_dir_open("/sys/class/usb/", 0, NULL);
+	if (len == 0)
+		return SR_ERR_NA;
 
-	if (dir == NULL)
-		return NULL;
+	buf[len] = 0;
+	tokens = g_strsplit(buf, ",", 0);
+	sr_dbg("response: %s [%s]", port, buf);
 
-	while ((dev_name = g_dir_read_name(dir)) != NULL) {
-		if (strncmp(dev_name, prefix, strlen(prefix))) 
-			continue;
+	for (num_tokens = 0; tokens[num_tokens] != NULL; num_tokens++);
 
-		device = g_strconcat(dev_dir, dev_name, NULL);
+	if (num_tokens < 4) {
+		g_strfreev(tokens);
+		return SR_ERR_NA;
+	}
 
-		fd = open(device, O_RDWR);
-		len = write(fd, idn_query, strlen(idn_query));
-		len = read(fd, buf, sizeof(buf));
-		close(fd);
-		if (len == 0) {
-			g_free(device);
-			return NULL;
+	manufacturer = tokens[0];
+	model = tokens[1];
+	version = tokens[3];
+
+	if (strcmp(manufacturer, "Rigol Technologies")) {
+		g_strfreev(tokens);
+		return SR_ERR_NA;
+	}
+
+	matched = has_digital = FALSE;
+	for (i = 0; i < ARRAY_SIZE(supported_models); i++) {
+		if (!strcmp(model, supported_models[i])) {
+			matched = TRUE;
+			has_digital = g_str_has_suffix(model, "D");
+			break;
 		}
+	}
 
-		buf[len] = 0;
-		tokens = g_strsplit(buf, delimiter, 0);
-		close(fd);
-		sr_dbg("response: %s %d [%s]", device, len, buf);
+	if (!matched || !(sdi = sr_dev_inst_new(0, SR_ST_ACTIVE,
+		manufacturer, model, version))) {
+		g_strfreev(tokens);
+		return SR_ERR_NA;
+	}
 
-		for (num_tokens = 0; tokens[num_tokens] != NULL; num_tokens++);
+	g_strfreev(tokens);
 
-		if (num_tokens < 4) {
-			g_strfreev(tokens);
-			g_free(device);
-			return NULL;
-		}
+	if (!(sdi->conn = sr_serial_dev_inst_new(port, NULL)))
+		return SR_ERR_MALLOC;
+	sdi->driver = di;
 
-		manufacturer = tokens[0];
-		model = tokens[1];
-		version = tokens[3];
+	if (!(devc = g_try_malloc0(sizeof(struct dev_context))))
+		return SR_ERR_MALLOC;
+	devc->limit_frames = 0;
+	devc->has_digital = has_digital;
 
-		if (strcmp(manufacturer, "Rigol Technologies")) {
-			g_strfreev(tokens);
-			g_free(device);
-			return NULL;
-		}
+	for (i = 0; i < 2; i++) {
+		if (!(probe = sr_probe_new(i, SR_PROBE_ANALOG, TRUE,
+				i == 0 ? "CH1" : "CH2")))
+			return SR_ERR_MALLOC;
+		sdi->probes = g_slist_append(sdi->probes, probe);
+	}
 
-		for (i = 0; i < ARRAY_SIZE(supported_models); i++) {
-			if (!strcmp(model, supported_models[i])) {
-				matched = 1;
-				break;
-			}
+	if (devc->has_digital) {
+		for (i = 0; i < 16; i++) {
+			if (!(channel_name = g_strdup_printf("D%d", i)))
+				return SR_ERR_MALLOC;
+			probe = sr_probe_new(i, SR_PROBE_LOGIC, TRUE, channel_name);
+			g_free(channel_name);
+			if (!probe)
+				return SR_ERR_MALLOC;
+			sdi->probes = g_slist_append(sdi->probes, probe);
 		}
+	}
+	sdi->priv = devc;
 
-		if (!matched || !(sdi = sr_dev_inst_new(0, SR_ST_ACTIVE,
-			manufacturer, model, version))) {
-			g_strfreev(tokens);
-			g_free(device);
-			return NULL;
-		}
+	*devices = g_slist_append(NULL, sdi);
 
-		g_strfreev(tokens);
+	return SR_OK;
+}
 
-		if (!(devc = g_try_malloc0(sizeof(struct dev_context)))) {
-			sr_err("Device context malloc failed.");
-			g_free(device);
-			return NULL;
+static GSList *hw_scan(GSList *options)
+{
+	struct drv_context *drvc;
+	struct sr_config *src;
+	GSList *l, *devices;
+	GDir *dir;
+	int ret;
+	const gchar *dev_name;
+	gchar *port = NULL;
+
+	drvc = di->priv;
+
+	for (l = options; l; l = l->next) {
+		src = l->data;
+		if (src->key == SR_CONF_CONN) {
+			port = (char *)g_variant_get_string(src->data, NULL);
+			break;
 		}
-		devc->limit_frames = 0;
-		devc->device = device;
-		sdi->priv = devc;
-		sdi->driver = di;
-
-		for (i = 0; i < 2; i++) {
-			if (!(probe = sr_probe_new(0, SR_PROBE_ANALOG, TRUE,
-			    i == 0 ? "CH1" : "CH2")))
+	}
+
+	devices = NULL;
+	if (port) {
+		if (probe_port(port, &devices) == SR_ERR_MALLOC)
+			return NULL;
+	} else {
+		if (!(dir = g_dir_open("/sys/class/usb/", 0, NULL)))
+			return NULL;
+		while ((dev_name = g_dir_read_name(dir))) {
+			if (strncmp(dev_name, "usbtmc", 6))
+				continue;
+			port = g_strconcat("/dev/", dev_name, NULL);
+			ret = probe_port(port, &devices);
+			g_free(port);
+			if (ret == SR_ERR_MALLOC) {
+				g_dir_close(dir);
 				return NULL;
-			sdi->probes = g_slist_append(sdi->probes, probe);
+			}
 		}
-
-		drvc->instances = g_slist_append(drvc->instances, sdi);
-		devices = g_slist_append(devices, sdi);
+		g_dir_close(dir);
 	}
 
-	g_dir_close(dir);
+	/* Tack a copy of the newly found devices onto the driver list. */
+	l = g_slist_copy(devices);
+	drvc->instances = g_slist_concat(drvc->instances, l);
 
 	return devices;
 }
@@ -305,29 +352,27 @@ static GSList *hw_dev_list(void)
 
 static int hw_dev_open(struct sr_dev_inst *sdi)
 {
-	struct dev_context *devc;
-	int fd;
-
-	devc = sdi->priv;
 
-	if ((fd = open(devc->device, O_RDWR)) == -1)
+	if (serial_open(sdi->conn, SERIAL_RDWR) != SR_OK)
 		return SR_ERR;
-	devc->fd = fd;
 
 	if (rigol_ds1xx2_get_dev_cfg(sdi) != SR_OK)
-		/* TODO: force configuration? */
 		return SR_ERR;
 
+	sdi->status = SR_ST_ACTIVE;
+
 	return SR_OK;
 }
 
 static int hw_dev_close(struct sr_dev_inst *sdi)
 {
-	struct dev_context *devc;
+	struct sr_serial_dev_inst *serial;
 
-	devc = sdi->priv;
-
-	close(devc->fd);
+	serial = sdi->conn;
+	if (serial && serial->fd != -1) {
+		serial_close(serial);
+		sdi->status = SR_ST_INACTIVE;
+	}
 
 	return SR_OK;
 }
@@ -352,7 +397,7 @@ static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi)
 		*data = g_variant_new_int32(NUM_VDIV);
 		break;
 	default:
-		return SR_ERR_ARG;
+		return SR_ERR_NA;
 	}
 
 	return SR_OK;
@@ -369,10 +414,8 @@ static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi)
 
 	devc = sdi->priv;
 
-	if (sdi->status != SR_ST_ACTIVE) {
-		sr_err("Device inactive, can't set config options.");
-		return SR_ERR;
-	}
+	if (sdi->status != SR_ST_ACTIVE)
+		return SR_ERR_DEV_CLOSED;
 
 	ret = SR_OK;
 	switch (id) {
@@ -461,8 +504,7 @@ static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi)
 			ret = SR_ERR_ARG;
 		break;
 	default:
-		sr_err("Unknown hardware capability: %d.", id);
-		ret = SR_ERR_ARG;
+		ret = SR_ERR_NA;
 		break;
 	}
 
@@ -474,10 +516,13 @@ static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
 	GVariant *tuple, *rational[2];
 	GVariantBuilder gvb;
 	unsigned int i;
-
-	(void)sdi;
+	struct dev_context *devc;
 
 	switch (key) {
+	case SR_CONF_SCAN_OPTIONS:
+		*data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
+				hwopts, ARRAY_SIZE(hwopts), sizeof(int32_t));
+		break;
 	case SR_CONF_DEVICE_OPTIONS:
 		*data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
 				hwcaps, ARRAY_SIZE(hwcaps), sizeof(int32_t));
@@ -506,11 +551,15 @@ static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
 		*data = g_variant_builder_end(&gvb);
 		break;
 	case SR_CONF_TRIGGER_SOURCE:
+		if (!sdi || !sdi->priv)
+			/* Can't know this until we have the exact model. */
+			return SR_ERR_ARG;
+		devc = sdi->priv;
 		*data = g_variant_new_strv(trigger_sources,
-				ARRAY_SIZE(trigger_sources));
+				devc->has_digital ? ARRAY_SIZE(trigger_sources) : 4);
 		break;
 	default:
-		return SR_ERR_ARG;
+		return SR_ERR_NA;
 	}
 
 	return SR_OK;
@@ -518,43 +567,66 @@ static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
 
 static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 {
+	struct sr_serial_dev_inst *serial;
 	struct dev_context *devc;
 	struct sr_probe *probe;
 	GSList *l;
-	int probenum;
 	char cmd[256];
 
-	(void)cb_data;
+	if (sdi->status != SR_ST_ACTIVE)
+		return SR_ERR_DEV_CLOSED;
 
+	serial = sdi->conn;
 	devc = sdi->priv;
 
 	for (l = sdi->probes; l; l = l->next) {
 		probe = l->data;
-		probenum = probe->name[2] == '1' ? 0 : 1;
-		if (probe->enabled)
-			devc->enabled_probes = g_slist_append(devc->enabled_probes, probe);
-
-		if (probe->enabled != devc->channels[probenum]) {
-			/* Enabled channel is currently disabled, or vice versa. */
-			sprintf(cmd, ":CHAN%d:DISP %s", probenum + 1,
-					probe->enabled ? "ON" : "OFF");
-			if (rigol_ds1xx2_send(devc, cmd) != SR_OK)
-				return SR_ERR;
+		sr_dbg("handling probe %s", probe->name);
+		if (probe->type == SR_PROBE_ANALOG) {
+			if (probe->enabled)
+				devc->enabled_analog_probes = g_slist_append(
+						devc->enabled_analog_probes, probe);
+			if (probe->enabled != devc->analog_channels[probe->index]) {
+				/* Enabled channel is currently disabled, or vice versa. */
+				sprintf(cmd, ":CHAN%d:DISP %s", probe->index + 1,
+						probe->enabled ? "ON" : "OFF");
+				if (rigol_ds1xx2_send(sdi, 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");
+				if (rigol_ds1xx2_send(sdi, cmd) != SR_OK)
+					return SR_ERR;
+			}
 		}
 	}
-	if (!devc->enabled_probes)
+	if (!devc->enabled_analog_probes && !devc->enabled_digital_probes)
 		return SR_ERR;
 
-	sr_source_add(devc->fd, G_IO_IN, 50, rigol_ds1xx2_receive, (void *)sdi);
+	sr_source_add(serial->fd, G_IO_IN, 50, rigol_ds1xx2_receive, (void *)sdi);
 
 	/* Send header packet to the session bus. */
-	std_session_send_df_header(cb_data, DRIVER_LOG_DOMAIN);
+	std_session_send_df_header(cb_data, LOG_PREFIX);
 
 	/* Fetch the first frame. */
-	devc->channel_frame = devc->enabled_probes->data;
-	if (rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%c",
-			devc->channel_frame->name[2]) != SR_OK)
-		return SR_ERR;
+	if (devc->enabled_analog_probes) {
+		devc->channel_frame = devc->enabled_analog_probes->data;
+		if (rigol_ds1xx2_send(sdi, ":WAV:DATA? CHAN%d",
+				devc->channel_frame->index + 1) != SR_OK)
+			return SR_ERR;
+	} else {
+		devc->channel_frame = devc->enabled_digital_probes->data;
+		if (rigol_ds1xx2_send(sdi, ":WAV:DATA? DIG") != SR_OK)
+			return SR_ERR;
+	}
+
+	devc->num_frame_bytes = 0;
 
 	return SR_OK;
 }
@@ -562,6 +634,7 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
 {
 	struct dev_context *devc;
+	struct sr_serial_dev_inst *serial;
 
 	(void)cb_data;
 
@@ -572,9 +645,12 @@ static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
 		return SR_ERR;
 	}
 
-	g_slist_free(devc->enabled_probes);
-	devc->enabled_probes = NULL;
-	sr_source_remove(devc->fd);
+	g_slist_free(devc->enabled_analog_probes);
+	g_slist_free(devc->enabled_digital_probes);
+	devc->enabled_analog_probes = NULL;
+	devc->enabled_digital_probes = NULL;
+	serial = sdi->conn;
+	sr_source_remove(serial->fd);
 
 	return SR_OK;
 }