X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=output%2Foutput_gnuplot.c;h=ec50b9d2bfa8a6c3bc794986f5bb98351eb9134d;hb=20ebd1fe1d9314f421cfdfc925eba6eee498ce80;hp=94c566faf1bc0cbd71151f63e12217452ada02f3;hpb=f0411b1d170aa60c3aa79b12e88f53b225ea350d;p=libsigrok.git

diff --git a/output/output_gnuplot.c b/output/output_gnuplot.c
index 94c566fa..ec50b9d2 100644
--- a/output/output_gnuplot.c
+++ b/output/output_gnuplot.c
@@ -27,12 +27,14 @@
 struct context {
 	unsigned int num_enabled_probes;
 	unsigned int unitsize;
-	char *probelist[MAX_NUM_PROBES+1];
+	char *probelist[SR_MAX_NUM_PROBES + 1];
 	char *header;
 };
 
-#define MAX_HEADER_LEN   1024 + (MAX_NUM_PROBES * (MAX_PROBENAME_LEN + 10))
-const char *gnuplot_header = "\
+#define MAX_HEADER_LEN \
+	(1024 + (SR_MAX_NUM_PROBES * (SR_MAX_PROBENAME_LEN + 10)))
+
+static const char *gnuplot_header = "\
 # Sample data in space-separated columns format usable by gnuplot\n\
 #\n\
 # Generated by: %s on %s%s\
@@ -43,13 +45,13 @@ const char *gnuplot_header = "\
 ----------------------------------------\n\
 # 0\t\tSample counter (for internal gnuplot purposes)\n%s\n";
 
-const char *gnuplot_header_comment = "\
+static const char *gnuplot_header_comment = "\
 # Comment: Acquisition with %d/%d probes at %s\n";
 
-static int init(struct output *o)
+static int init(struct sr_output *o)
 {
 	struct context *ctx;
-	struct probe *probe;
+	struct sr_probe *probe;
 	GSList *l;
 	uint64_t samplerate;
 	unsigned int i;
@@ -58,18 +60,38 @@ static int init(struct output *o)
 	char wbuf[1000], comment[128];
 	time_t t;
 
-	if (!(ctx = calloc(1, sizeof(struct context))))
-		return SIGROK_ERR_MALLOC;
+	if (!o) {
+		g_warning("gnuplot out: %s: o was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!o->device) {
+		g_warning("gnuplot out: %s: o->device was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!o->device->plugin) {
+		g_warning("gnuplot out: %s: o->device->plugin was NULL",
+			  __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!(ctx = calloc(1, sizeof(struct context)))) {
+		g_warning("gnuplot out: %s: ctx calloc failed", __func__);
+		return SR_ERR_MALLOC;
+	}
 
 	if (!(ctx->header = calloc(1, MAX_HEADER_LEN + 1))) {
+		g_warning("gnuplot out: %s: ctx->header calloc failed",
+			  __func__);
 		free(ctx);
-		return SIGROK_ERR_MALLOC;
+		return SR_ERR_MALLOC;
 	}
 
 	o->internal = ctx;
 	ctx->num_enabled_probes = 0;
 	for (l = o->device->probes; l; l = l->next) {
-		probe = l->data;
+		probe = l->data; /* TODO: Error checks. */
 		if (!probe->enabled)
 			continue;
 		ctx->probelist[ctx->num_enabled_probes++] = probe->name;
@@ -79,13 +101,15 @@ static int init(struct output *o)
 
 	num_probes = g_slist_length(o->device->probes);
 	comment[0] = '\0';
-	if (o->device->plugin) {
+	if (sr_device_has_hwcap(o->device, SR_HWCAP_SAMPLERATE)) {
 		samplerate = *((uint64_t *) o->device->plugin->get_device_info(
-				o->device->plugin_index, DI_CUR_SAMPLERATE));
-		if (!(frequency_s = sigrok_samplerate_string(samplerate))) {
+				o->device->plugin_index, SR_DI_CUR_SAMPLERATE));
+		if (!(frequency_s = sr_samplerate_string(samplerate))) {
+			g_warning("gnuplot out: %s: sr_samplerate_string "
+				  "failed", __func__);
 			free(ctx->header);
 			free(ctx);
-			return SIGROK_ERR;
+			return SR_ERR;
 		}
 		snprintf(comment, 127, gnuplot_header_comment,
 			ctx->num_enabled_probes, num_probes, frequency_s);
@@ -99,11 +123,13 @@ static int init(struct output *o)
 		sprintf(c, "# %d\t\t%s\n", i + 1, ctx->probelist[i]);
 	}
 
-	if (!(frequency_s = sigrok_period_string(samplerate))) {
+	if (!(frequency_s = sr_period_string(samplerate))) {
+		g_warning("gnuplot out: %s: sr_period_string failed", __func__);
 		free(ctx->header);
 		free(ctx);
-		return SIGROK_ERR;
+		return SR_ERR;
 	}
+
 	t = time(NULL);
 	b = snprintf(ctx->header, MAX_HEADER_LEN, gnuplot_header,
 		     PACKAGE_STRING, ctime(&t), comment, frequency_s,
@@ -111,36 +137,58 @@ static int init(struct output *o)
 	free(frequency_s);
 
 	if (b < 0) {
+		g_warning("gnuplot out: %s: sprintf failed", __func__);
 		free(ctx->header);
 		free(ctx);
-		return SIGROK_ERR;
+		return SR_ERR;
 	}
 
 	return 0;
 }
 
-static int event(struct output *o, int event_type, char **data_out,
+static int event(struct sr_output *o, int event_type, char **data_out,
 		 uint64_t *length_out)
 {
 	struct context *ctx;
 
+	if (!o) {
+		g_warning("gnuplot out: %s: o was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!data_out) {
+		g_warning("gnuplot out: %s: data_out was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!length_out) {
+		g_warning("gnuplot out: %s: length_out was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
 	ctx = o->internal;
+
 	switch (event_type) {
-	case DF_TRIGGER:
-		/* TODO */
+	case SR_DF_TRIGGER:
+		/* TODO: Can a trigger mark be in a gnuplot data file? */
 		break;
-	case DF_END:
-		*data_out = NULL;
-		*length_out = 0;
+	case SR_DF_END:
 		free(o->internal);
 		o->internal = NULL;
 		break;
+	default:
+		g_warning("gnuplot out: %s: unsupported event type: %d",
+		          __func__, event_type);
+		break;
 	}
 
-	return SIGROK_OK;
+	*data_out = NULL;
+	*length_out = 0;
+
+	return SR_OK;
 }
 
-static int data(struct output *o, char *data_in, uint64_t length_in,
+static int data(struct sr_output *o, const char *data_in, uint64_t length_in,
 		char **data_out, uint64_t *length_out)
 {
 	struct context *ctx;
@@ -149,14 +197,41 @@ static int data(struct output *o, char *data_in, uint64_t length_in,
 	static uint64_t samplecount = 0;
 	char *outbuf, *c;
 
+	if (!o) {
+		g_warning("gnuplot out: %s: o was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!o->internal) {
+		g_warning("gnuplot out: %s: o->internal was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!data_in) {
+		g_warning("gnuplot out: %s: data_in was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!data_out) {
+		g_warning("gnuplot out: %s: data_out was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
+	if (!length_out) {
+		g_warning("gnuplot out: %s: length_out was NULL", __func__);
+		return SR_ERR_ARG;
+	}
+
 	ctx = o->internal;
 	max_linelen = 16 + ctx->num_enabled_probes * 2;
 	outsize = length_in / ctx->unitsize * max_linelen;
 	if (ctx->header)
 		outsize += strlen(ctx->header);
 
-	if (!(outbuf = calloc(1, outsize)))
-		return SIGROK_ERR_MALLOC;
+	if (!(outbuf = calloc(1, outsize))) {
+		g_warning("gnuplot out: %s: outbuf calloc failed", __func__);
+		return SR_ERR_MALLOC;
+	}
 
 	outbuf[0] = '\0';
 	if (ctx->header) {
@@ -187,14 +262,161 @@ static int data(struct output *o, char *data_in, uint64_t length_in,
 	*data_out = outbuf;
 	*length_out = strlen(outbuf);
 
-	return SIGROK_OK;
+	return SR_OK;
+}
+
+struct sr_output_format output_gnuplot = {
+	.id = "gnuplot",
+	.description = "Gnuplot",
+	.df_type = SR_DF_LOGIC,
+	.init = init,
+	.data = data,
+	.event = event,
+};
+
+/* Temporarily disabled. */
+#if 0
+static int analog_init(struct sr_output *o)
+{
+	struct context *ctx;
+	struct sr_probe *probe;
+	GSList *l;
+	uint64_t samplerate;
+	unsigned int i;
+	int b, num_probes;
+	char *c, *frequency_s;
+	char wbuf[1000], comment[128];
+	time_t t;
+
+	if (!(ctx = calloc(1, sizeof(struct context))))
+		return SR_ERR_MALLOC;
+
+	if (!(ctx->header = calloc(1, MAX_HEADER_LEN + 1))) {
+		free(ctx);
+		return SR_ERR_MALLOC;
+	}
+
+	o->internal = ctx;
+	ctx->num_enabled_probes = 0;
+	for (l = o->device->probes; l; l = l->next) {
+		probe = l->data;
+		if (!probe->enabled)
+			continue;
+		ctx->probelist[ctx->num_enabled_probes++] = probe->name;
+	}
+	ctx->probelist[ctx->num_enabled_probes] = 0;
+//	ctx->unitsize = (ctx->num_enabled_probes + 7) / 8;
+	ctx->unitsize = sizeof(struct sr_analog_sample) +
+			(ctx->num_enabled_probes * sizeof(struct sr_analog_probe));
+
+	num_probes = g_slist_length(o->device->probes);
+	comment[0] = '\0';
+	if (o->device->plugin && sr_device_has_hwcap(o->device, SR_HWCAP_SAMPLERATE)) {
+		samplerate = *((uint64_t *) o->device->plugin->get_device_info(
+				o->device->plugin_index, SR_DI_CUR_SAMPLERATE));
+		if (!(frequency_s = sr_samplerate_string(samplerate))) {
+			free(ctx->header);
+			free(ctx);
+			return SR_ERR;
+		}
+		snprintf(comment, 127, gnuplot_header_comment,
+			ctx->num_enabled_probes, num_probes, frequency_s);
+		free(frequency_s);
+	}
+
+	/* Columns / channels */
+	wbuf[0] = '\0';
+	for (i = 0; i < ctx->num_enabled_probes; i++) {
+		c = (char *)&wbuf + strlen((char *)&wbuf);
+		sprintf(c, "# %d\t\t%s\n", i + 1, ctx->probelist[i]);
+	}
+
+	if (!(frequency_s = sr_period_string(samplerate))) {
+		free(ctx->header);
+		free(ctx);
+		return SR_ERR;
+	}
+	t = time(NULL);
+	b = snprintf(ctx->header, MAX_HEADER_LEN, gnuplot_header,
+		     PACKAGE_STRING, ctime(&t), comment, frequency_s,
+		     (char *)&wbuf);
+	free(frequency_s);
+
+	if (b < 0) {
+		free(ctx->header);
+		free(ctx);
+		return SR_ERR;
+	}
+
+	return 0;
+}
+
+static int analog_data(struct sr_output *o, char *data_in, uint64_t length_in,
+		char **data_out, uint64_t *length_out)
+{
+	struct context *ctx;
+	unsigned int max_linelen, outsize, p, /* curbit, */ i;
+//	uint64_t sample;
+	static uint64_t samplecount = 0;
+	char *outbuf, *c;
+	struct sr_analog_sample *sample;
+
+	ctx = o->internal;
+//	max_linelen = 16 + ctx->num_enabled_probes * 2;
+	max_linelen = 16 + ctx->num_enabled_probes * 30;
+	outsize = length_in / ctx->unitsize * max_linelen;
+	if (ctx->header)
+		outsize += strlen(ctx->header);
+
+	if (!(outbuf = calloc(1, outsize)))
+		return SR_ERR_MALLOC;
+
+	outbuf[0] = '\0';
+	if (ctx->header) {
+		/* The header is still here, this must be the first packet. */
+		strncpy(outbuf, ctx->header, outsize);
+		free(ctx->header);
+		ctx->header = NULL;
+	}
+
+	for (i = 0; i <= length_in - ctx->unitsize; i += ctx->unitsize) {
+//		memcpy(&sample, data_in + i, ctx->unitsize);
+		sample = (struct sr_analog_sample *) (data_in + i);
+
+		/* The first column is a counter (needed for gnuplot). */
+		c = outbuf + strlen(outbuf);
+		sprintf(c, "%" PRIu64 "\t", samplecount++);
+
+		/* The next columns are the values of all channels. */
+		for (p = 0; p < ctx->num_enabled_probes; p++) {
+//			curbit = (sample & ((uint64_t) (1 << p))) >> p;
+			c = outbuf + strlen(outbuf);
+//			sprintf(c, "%d ", curbit);
+			/*
+			 * FIXME: Should be doing proper raw->voltage conversion
+			 * here, casting to int16_t isn't it. Remember that if
+			 * res = 1 conversion isn't necessary.
+			 */
+			sprintf(c, "%f ", (double) ((int16_t) (sample->probes[p].val &
+					((1 << sample->probes[p].res) - 1))));
+		}
+
+		c = outbuf + strlen(outbuf);
+		sprintf(c, "\n");
+	}
+
+	*data_out = outbuf;
+	*length_out = strlen(outbuf);
+
+	return SR_OK;
 }
 
-struct output_format output_gnuplot = {
-	"gnuplot",
-	"Gnuplot",
-	DF_LOGIC,
-	init,
-	data,
-	event,
+struct sr_output_format output_analog_gnuplot = {
+	.id = "analog_gnuplot",
+	.description = "Gnuplot analog",
+	.df_type = SR_DF_ANALOG,
+	.init = analog_init,
+	.data = analog_data,
+	.event = event,
 };
+#endif