X-Git-Url: http://sigrok.org/gitweb/?a=blobdiff_plain;f=output%2Fcsv.c;h=0bec7ae281c4286825eb54f586828bdc2f683f10;hb=dba3e6826e3d37309ed49dd77cde346b369bc5a9;hp=cc22eb84bd9538052802a0f83c4d2f334935d00a;hpb=c09f0b578c0e9c03590cb814f66004bb3f6815ff;p=libsigrok.git

diff --git a/output/csv.c b/output/csv.c
index cc22eb84..0bec7ae2 100644
--- a/output/csv.c
+++ b/output/csv.c
@@ -1,5 +1,5 @@
 /*
- * This file is part of the sigrok project.
+ * This file is part of the libsigrok project.
  *
  * Copyright (C) 2011 Uwe Hermann <uwe@hermann-uwe.de>
  *
@@ -21,17 +21,18 @@
 #include <stdlib.h>
 #include <string.h>
 #include <glib.h>
-#include "config.h"
-#include "sigrok.h"
-#include "sigrok-internal.h"
+#include "config.h" /* Needed for PACKAGE_STRING and others. */
+#include "libsigrok.h"
+#include "libsigrok-internal.h"
+
+#define LOG_PREFIX "output/csv"
 
 struct context {
-	unsigned int num_enabled_probes;
-	unsigned int unitsize;
-	char *probelist[SR_MAX_NUM_PROBES + 1];
+	unsigned int num_enabled_channels;
 	uint64_t samplerate;
 	GString *header;
 	char separator;
+	int *channel_index;
 };
 
 /*
@@ -42,66 +43,49 @@ struct context {
  *  - Option to (not) print samplenumber / time as extra column.
  *  - Option to "compress" output (only print changed samples, VCD-like).
  *  - Option to print comma-separated bits, or whole bytes/words (for 8/16
- *    probe LAs) as ASCII/hex etc. etc.
+ *    channel LAs) as ASCII/hex etc. etc.
  *  - Trigger support.
  */
 
 static int init(struct sr_output *o)
 {
 	struct context *ctx;
-	struct sr_probe *probe;
+	struct sr_channel *ch;
 	GSList *l;
-	int num_probes;
-	uint64_t samplerate;
+	GVariant *gvar;
+	int num_channels, i, j;
 	time_t t;
-	unsigned int i;
-
-	if (!o) {
-		sr_err("csv out: %s: o was NULL", __func__);
-		return SR_ERR_ARG;
-	}
 
-	if (!o->dev) {
-		sr_err("csv out: %s: o->dev was NULL", __func__);
+	if (!o)
 		return SR_ERR_ARG;
-	}
 
-	if (!o->dev->driver) {
-		sr_err("csv out: %s: o->dev->driver was NULL", __func__);
+	if (!o->sdi)
 		return SR_ERR_ARG;
-	}
-
-	if (!(ctx = g_try_malloc0(sizeof(struct context)))) {
-		sr_err("csv out: %s: ctx malloc failed", __func__);
-		return SR_ERR_MALLOC;
-	}
 
+	ctx = g_malloc0(sizeof(struct context));
 	o->internal = ctx;
 
-	/* Get the number of probes, their names, and the unitsize. */
-	/* TODO: Error handling. */
-	for (l = o->dev->probes; l; l = l->next) {
-		probe = l->data;
-		if (!probe->enabled)
+	/* Get the number of channels, and the unitsize. */
+	for (l = o->sdi->channels; l; l = l->next) {
+		ch = l->data;
+		if (ch->type != SR_CHANNEL_LOGIC)
+			continue;
+		if (!ch->enabled)
 			continue;
-		ctx->probelist[ctx->num_enabled_probes++] = probe->name;
+		ctx->num_enabled_channels++;
 	}
-	ctx->probelist[ctx->num_enabled_probes] = 0;
-	ctx->unitsize = (ctx->num_enabled_probes + 7) / 8;
+	ctx->channel_index = g_malloc(sizeof(int) * ctx->num_enabled_channels);
 
-	num_probes = g_slist_length(o->dev->probes);
+	num_channels = g_slist_length(o->sdi->channels);
 
-	if (sr_dev_has_hwcap(o->dev, SR_HWCAP_SAMPLERATE)) {
-		samplerate = *((uint64_t *) o->dev->driver->dev_info_get(
-				o->dev->driver_index, SR_DI_CUR_SAMPLERATE));
-		/* TODO: Error checks. */
-	} else {
-		samplerate = 0; /* TODO: Error or set some value? */
-	}
-	ctx->samplerate = samplerate;
+	if (sr_config_get(o->sdi->driver, o->sdi, NULL, SR_CONF_SAMPLERATE,
+			&gvar) == SR_OK) {
+		ctx->samplerate = g_variant_get_uint64(gvar);
+		g_variant_unref(gvar);
+	} else
+		ctx->samplerate = 0;
 
 	ctx->separator = ',';
-
 	ctx->header = g_string_sized_new(512);
 
 	t = time(NULL);
@@ -113,114 +97,98 @@ static int init(struct sr_output *o)
 			       ctx->samplerate);
 
 	/* Columns / channels */
-	g_string_append_printf(ctx->header, "; Channels (%d/%d): ",
-			       ctx->num_enabled_probes, num_probes);
-	for (i = 0; i < ctx->num_enabled_probes; i++)
-		g_string_append_printf(ctx->header, "%s, ", ctx->probelist[i]);
+	g_string_append_printf(ctx->header, "; Channels (%d/%d):",
+			       ctx->num_enabled_channels, num_channels);
+	for (i = 0, j = 0, l = o->sdi->channels; l; l = l->next, i++) {
+		ch = l->data;
+		if (ch->type != SR_CHANNEL_LOGIC)
+			continue;
+		if (!ch->enabled)
+			continue;
+		g_string_append_printf(ctx->header, " %s,", ch->name);
+		/* Remember the enabled channel's index while we're at it. */
+		ctx->channel_index[j++] = i;
+	}
+	if (o->sdi->channels)
+		/* Drop last separator. */
+		g_string_truncate(ctx->header, ctx->header->len - 1);
 	g_string_append_printf(ctx->header, "\n");
 
-	return 0; /* TODO: SR_OK? */
+	return SR_OK;
 }
 
-static int event(struct sr_output *o, int event_type, char **data_out,
-		 uint64_t *length_out)
+static int receive(struct sr_output *o, const struct sr_datafeed_packet *packet,
+		GString **out)
 {
+	const struct sr_datafeed_logic *logic;
 	struct context *ctx;
+	int idx;
+	uint64_t i, j;
+	gchar *p, c;
 
-	if (!o) {
-		sr_err("csv out: %s: o was NULL", __func__);
+	*out = NULL;
+	if (!o || !o->sdi)
 		return SR_ERR_ARG;
-	}
-
-	if (!(ctx = o->internal)) {
-		sr_err("csv out: %s: o->internal was NULL", __func__);
+	if (!(ctx = o->internal))
 		return SR_ERR_ARG;
-	}
 
-	if (!data_out) {
-		sr_err("csv out: %s: data_out was NULL", __func__);
-		return SR_ERR_ARG;
-	}
+	switch (packet->type) {
+	case SR_DF_LOGIC:
+		logic = packet->payload;
+		if (ctx->header) {
+			/*
+			 * First data packet: prime the output with the
+			 * previously prepared header.
+			 */
+			*out = ctx->header;
+			ctx->header = NULL;
+		} else {
+			*out = g_string_sized_new(512);
+		}
 
-	switch (event_type) {
-	case SR_DF_TRIGGER:
-		sr_dbg("csv out: %s: SR_DF_TRIGGER event", __func__);
-		/* TODO */
-		*data_out = NULL;
-		*length_out = 0;
-		break;
-	case SR_DF_END:
-		sr_dbg("csv out: %s: SR_DF_END event", __func__);
-		/* TODO */
-		*data_out = NULL;
-		*length_out = 0;
-		g_free(o->internal);
-		o->internal = NULL;
-		break;
-	default:
-		sr_err("csv out: %s: unsupported event type: %d", __func__,
-		       event_type);
-		*data_out = NULL;
-		*length_out = 0;
+		for (i = 0; i <= logic->length - logic->unitsize; i += logic->unitsize) {
+			for (j = 0; j < ctx->num_enabled_channels; j++) {
+				idx = ctx->channel_index[j];
+				p = logic->data + i + idx / 8;
+				c = *p & (1 << (idx % 8));
+				g_string_append_c(*out, c ? '1' : '0');
+				g_string_append_c(*out, ctx->separator);
+			}
+			if (j) {
+				/* Drop last separator. */
+				g_string_truncate(*out, (*out)->len - 1);
+			}
+			g_string_append_printf(*out, "\n");
+		}
 		break;
 	}
 
 	return SR_OK;
 }
 
-static int data(struct sr_output *o, const char *data_in, uint64_t length_in,
-		char **data_out, uint64_t *length_out)
+static int cleanup(struct sr_output *o)
 {
 	struct context *ctx;
-	GString *outstr;
-	uint64_t sample, i;
-	int j;
-
-	if (!o) {
-		sr_err("csv out: %s: o was NULL", __func__);
-		return SR_ERR_ARG;
-	}
 
-	if (!(ctx = o->internal)) {
-		sr_err("csv out: %s: o->internal was NULL", __func__);
+	if (!o || !o->sdi)
 		return SR_ERR_ARG;
-	}
-
-	if (!data_in) {
-		sr_err("csv out: %s: data_in was NULL", __func__);
-		return SR_ERR_ARG;
-	}
-
-	if (ctx->header) {
-		/* First data packet. */
-		outstr = ctx->header;
-		ctx->header = NULL;
-	} else {
-		outstr = g_string_sized_new(512);
-	}
 
-	for (i = 0; i <= length_in - ctx->unitsize; i += ctx->unitsize) {
-		memcpy(&sample, data_in + i, ctx->unitsize);
-		for (j = ctx->num_enabled_probes - 1; j >= 0; j--) {
-			g_string_append_printf(outstr, "%d%c",
-				(int)((sample & (1 << j)) >> j),
-				ctx->separator);
-		}
-		g_string_append_printf(outstr, "\n");
+	if (o->internal) {
+		ctx = o->internal;
+		if (ctx->header)
+			g_string_free(ctx->header, TRUE);
+		g_free(ctx->channel_index);
+		g_free(o->internal);
+		o->internal = NULL;
 	}
 
-	*data_out = outstr->str;
-	*length_out = outstr->len;
-	g_string_free(outstr, FALSE);
-
 	return SR_OK;
 }
 
 SR_PRIV struct sr_output_format output_csv = {
 	.id = "csv",
 	.description = "Comma-separated values (CSV)",
-	.df_type = SR_DF_LOGIC,
 	.init = init,
-	.data = data,
-	.event = event,
+	.receive = receive,
+	.cleanup = cleanup,
 };