X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=hardware%2Frigol-ds1xx2%2Fapi.c;h=40865950e9545f763b867684148aff055addadc0;hb=8f996b89481670219c7576e2c68b128a0a2ce026;hp=db8d88099cf296259c938a7ffc269a1530b5ce69;hpb=d2e0b1fa71b90faf1f7e77b72627868baf5e9135;p=libsigrok.git

diff --git a/hardware/rigol-ds1xx2/api.c b/hardware/rigol-ds1xx2/api.c
index db8d8809..40865950 100644
--- a/hardware/rigol-ds1xx2/api.c
+++ b/hardware/rigol-ds1xx2/api.c
@@ -30,6 +30,10 @@
 #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_TIMEBASE,
@@ -129,253 +133,249 @@ static const char *coupling[] = {
 static const char *supported_models[] = {
 	"DS1052E",
 	"DS1102E",
+	"DS1152E",
 	"DS1052D",
 	"DS1102D",
+	"DS1152D",
 };
 
 SR_PRIV struct sr_dev_driver rigol_ds1xx2_driver_info;
 static struct sr_dev_driver *di = &rigol_ds1xx2_driver_info;
 
-/* Properly close and free all devices. */
-static int clear_instances(void)
+static void clear_helper(void *priv)
 {
-	struct sr_dev_inst *sdi;
-	struct drv_context *drvc;
 	struct dev_context *devc;
-	GSList *l;
-
-	if (!(drvc = di->priv))
-		return SR_OK;
 
-	for (l = drvc->instances; l; l = l->next) {
-		if (!(sdi = l->data))
-			continue;
-		if (!(devc = sdi->priv))
-			continue;
+	devc = priv;
 
-		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);
-
-		sr_dev_inst_free(sdi);
-	}
-
-	g_slist_free(drvc->instances);
-	drvc->instances = NULL;
+	g_free(devc->coupling[0]);
+	g_free(devc->coupling[1]);
+	g_free(devc->trigger_source);
+	g_free(devc->trigger_slope);
+}
 
-	return SR_OK;
+static int dev_clear(void)
+{
+	return std_dev_clear(di, clear_helper);
 }
 
 static int set_cfg(const struct sr_dev_inst *sdi, const char *format, ...)
 {
-	struct dev_context *devc;
 	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. */
 	sr_spew("delay %dms", 100);
-	g_usleep(100);
+	g_usleep(100000);
 
 	return SR_OK;
 }
 
-static int hw_init(struct sr_context *sr_ctx)
+static int init(struct sr_context *sr_ctx)
 {
-	return std_hw_init(sr_ctx, di, DRIVER_LOG_DOMAIN);
+	return std_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;
-	gboolean has_digital = FALSE;
 	char buf[256];
-	gchar *channel_name;
+	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;
+	}
+
+	manufacturer = tokens[0];
+	model = tokens[1];
+	version = tokens[3];
 
-		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);
-			continue;
+	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);
-			continue;
-		}
+	if (!(sdi->conn = sr_serial_dev_inst_new(port, NULL)))
+		return SR_ERR_MALLOC;
+	sdi->driver = di;
+	sdi->inst_type = SR_INST_SERIAL;
 
-		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);
-			continue;
-		}
+	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;
-				has_digital = g_str_has_suffix(model, "D");
-				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);
-			continue;
-		}
+	*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);
-			goto hw_scan_abort;
-		}
+static GSList *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;
 
-		devc->limit_frames = 0;
-		devc->device = device;
-		devc->has_digital = has_digital;
-		sdi->priv = devc;
-		sdi->driver = di;
-		drvc->instances = g_slist_append(drvc->instances, sdi);
-		devices = g_slist_append(devices, sdi);
-
-		for (i = 0; i < 2; i++) {
-			if (!(probe = sr_probe_new(i, SR_PROBE_ANALOG, TRUE,
-				    i == 0 ? "CH1" : "CH2")))
-				goto hw_scan_abort;
-			sdi->probes = g_slist_append(sdi->probes, probe);
+	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;
 		}
+	}
 
-		if (devc->has_digital) {
-			for (i = 0; i < 16; i++) {
-				if (!(channel_name = g_strdup_printf("D%d", i)))
-					goto hw_scan_abort;
-				probe = sr_probe_new(i, SR_PROBE_LOGIC, TRUE, channel_name);
-				g_free(channel_name);
-				if (!probe)
-					goto hw_scan_abort;
-				sdi->probes = g_slist_append(sdi->probes, probe);
+	devices = NULL;
+	if (port) {
+		if (probe_port(port, &devices) == SR_ERR_MALLOC)
+			return NULL;
+	} else {
+		if (!(dir = g_dir_open("/sys/class/usbmisc/", 0, NULL)))
+			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;
 			}
 		}
+		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;
-
-hw_scan_abort:
-	g_dir_close(dir);
-	g_slist_free(devices);
-	clear_instances();
-	return NULL;
 }
 
-static GSList *hw_dev_list(void)
+static GSList *dev_list(void)
 {
 	return ((struct drv_context *)(di->priv))->instances;
 }
 
-static int hw_dev_open(struct sr_dev_inst *sdi)
+static int 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)
+static int 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;
 }
 
-static int hw_cleanup(void)
+static int cleanup(void)
 {
-	clear_instances();
-
-	return SR_OK;
+	return dev_clear();
 }
 
-static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi)
+static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi,
+		const struct sr_probe_group *probe_group)
 {
 
 	(void)sdi;
+	(void)probe_group;
 
 	switch (id) {
 	case SR_CONF_NUM_TIMEBASE:
@@ -391,7 +391,8 @@ static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi)
 	return SR_OK;
 }
 
-static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi)
+static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi,
+		const struct sr_probe_group *probe_group)
 {
 	struct dev_context *devc;
 	uint64_t tmp_u64, p, q;
@@ -400,12 +401,12 @@ static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi)
 	int ret;
 	const char *tmp_str;
 
+	(void)probe_group;
+
 	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) {
@@ -501,14 +502,21 @@ static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi)
 	return ret;
 }
 
-static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
+static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi,
+		const struct sr_probe_group *probe_group)
 {
 	GVariant *tuple, *rational[2];
 	GVariantBuilder gvb;
 	unsigned int i;
 	struct dev_context *devc;
 
+	(void)probe_group;
+
 	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));
@@ -553,13 +561,16 @@ 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;
 	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) {
@@ -573,7 +584,7 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 				/* 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(devc, cmd) != SR_OK)
+				if (rigol_ds1xx2_send(sdi, cmd) != SR_OK)
 					return SR_ERR;
 			}
 		} else if (probe->type == SR_PROBE_LOGIC) {
@@ -584,7 +595,7 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 				/* Enabled channel is currently disabled, or vice versa. */
 				sprintf(cmd, ":DIG%d:TURN %s", probe->index,
 						probe->enabled ? "ON" : "OFF");
-				if (rigol_ds1xx2_send(devc, cmd) != SR_OK)
+				if (rigol_ds1xx2_send(sdi, cmd) != SR_OK)
 					return SR_ERR;
 			}
 		}
@@ -592,20 +603,20 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 	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. */
 	if (devc->enabled_analog_probes) {
 		devc->channel_frame = devc->enabled_analog_probes->data;
-		if (rigol_ds1xx2_send(devc, ":WAV:DATA? CHAN%d",
+		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(devc, ":WAV:DATA? DIG") != SR_OK)
+		if (rigol_ds1xx2_send(sdi, ":WAV:DATA? DIG") != SR_OK)
 			return SR_ERR;
 	}
 
@@ -617,6 +628,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;
 
@@ -631,7 +643,8 @@ static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
 	g_slist_free(devc->enabled_digital_probes);
 	devc->enabled_analog_probes = NULL;
 	devc->enabled_digital_probes = NULL;
-	sr_source_remove(devc->fd);
+	serial = sdi->conn;
+	sr_source_remove(serial->fd);
 
 	return SR_OK;
 }
@@ -640,16 +653,16 @@ SR_PRIV struct sr_dev_driver rigol_ds1xx2_driver_info = {
 	.name = "rigol-ds1xx2",
 	.longname = "Rigol DS1xx2",
 	.api_version = 1,
-	.init = hw_init,
-	.cleanup = hw_cleanup,
-	.scan = hw_scan,
-	.dev_list = hw_dev_list,
-	.dev_clear = clear_instances,
+	.init = init,
+	.cleanup = cleanup,
+	.scan = scan,
+	.dev_list = dev_list,
+	.dev_clear = dev_clear,
 	.config_get = config_get,
 	.config_set = config_set,
 	.config_list = config_list,
-	.dev_open = hw_dev_open,
-	.dev_close = hw_dev_close,
+	.dev_open = dev_open,
+	.dev_close = dev_close,
 	.dev_acquisition_start = dev_acquisition_start,
 	.dev_acquisition_stop = dev_acquisition_stop,
 	.priv = NULL,