Color plots, integrate only flight portion of data.

Telemetry files have piles of pad data which shouldn't be integrated
into the velocity data as it tends to generate huge values from the
noise of the sensor.

Also make the data lines colored to keep them visually distinct from
the rest of the plot image.

Signed-off-by: Keith Packard <keithp@keithp.com>

2 files changed, 120 insertions, 20 deletions

diff --git a/ao-tools/ao-postflight/ao-postflight.1 b/ao-tools/ao-postflight/ao-postflight.1
index ac1c18a4..eca4bb34 100644
--- a/ao-tools/ao-postflight/ao-postflight.1
+++ b/ao-tools/ao-postflight/ao-postflight.1
@@ -23,15 +23,29 @@ ao-postflight \- Analyse a flight log (either telemetry or eeprom)
 .B "ao-postflight"
 [\-s <summary-file>]
 [\--summary=<summary-file>]
-[\-d <detail-file]
+[\-d <detail-file>]
 [\--detail=<detail-file>]
+[\-r <raw-file>]
+[\--raw=<raw-file>]
 [\-p <plot-file>]
 [\--plot=<plot-file>]
 {flight.eeprom|flight.telem}
 .SH DESCRIPTION
 .I ao-postflight
-reads the specified flight log and produces a summary of the flight on
-stdout or to the specified file along with an optional .svg format
-plot and detailed table of time/height/speed/accel.
+reads the specified flight log and produces several different kinds of
+output.
+.IP Summary
+By default, summary information is shown on stdout. With the --summary
+option, it can be redirected to a file.
+.IP Detail
+When requested with the --detail option, a filtered version of the
+flight position, speed and acceleration are written to the specified
+file.
+.IP Raw
+The --raw option writes the unfiltered, but converted acceleration
+and height data to the specified file.
+.IP Plot
+The --plot option writes plots of height, speed and acceleration to
+the specified file in .svg format
 .SH AUTHOR
 Keith Packard
diff --git a/ao-tools/ao-postflight/ao-postflight.c b/ao-tools/ao-postflight/ao-postflight.c
index 4ca39c24..ded2f3c2 100644
--- a/ao-tools/ao-postflight/ao-postflight.c
+++ b/ao-tools/ao-postflight/ao-postflight.c
@@ -39,9 +39,19 @@ static const char *state_names[] = {
 	"invalid"
 };
 
+static int plot_colors[3][3] = {
+	{ 0, 0x90, 0 },	/* height */
+	{ 0xa0, 0, 0 },	/* speed */
+	{ 0, 0, 0xc0 },	/* accel */
+};
+
+#define PLOT_HEIGHT	0
+#define PLOT_SPEED	1
+#define PLOT_ACCEL	2
+
 static void
 plot_perioddata(struct cc_perioddata *d, char *axis_label, char *plot_label,
-		double min_time, double max_time)
+		double min_time, double max_time, int plot_type)
 {
 	double	*times;
 	double	ymin, ymax;
@@ -60,11 +70,58 @@ plot_perioddata(struct cc_perioddata *d, char *axis_label, char *plot_label,
 	ymax_i = cc_perioddata_max(d, min_time, max_time);
 	ymin = d->data[ymin_i];
 	ymax = d->data[ymax_i];
+	plscol0(1, 0, 0, 0);
+	plscol0(2, plot_colors[plot_type][0],  plot_colors[plot_type][1],  plot_colors[plot_type][2]);
+	plcol0(1);
 	plenv(times[0], times[stop-start],
 	      ymin, ymax, 0, 2);
-	plcol0(1);
 	pllab("Time", axis_label, plot_label);
+	plcol0(2);
 	plline(stop - start + 1, times, d->data + start);
+	free(times);
+}
+
+static void
+plot_timedata(struct cc_timedata *d, char *axis_label, char *plot_label,
+		double min_time, double max_time)
+{
+	double	*times;
+	double	*values;
+	double	ymin, ymax;
+	int	ymin_i, ymax_i;
+	int	i;
+	int	start = -1, stop = -1;
+	double	start_time = 0, stop_time = 0;
+	int	num;
+
+	for (i = 0; i < d->num; i++) {
+		if (start < 0 && d->data[i].time >= min_time) {
+			start_time = d->data[i].time;
+			start = i;
+		}
+		if (d->data[i].time <= max_time) {
+			stop_time = d->data[i].time;
+			stop = i;
+		}
+	}
+
+	times = calloc(stop - start + 1, sizeof (double));
+	values = calloc(stop - start + 1, sizeof (double));
+
+	ymin_i = cc_timedata_min(d, min_time, max_time);
+	ymax_i = cc_timedata_max(d, min_time, max_time);
+	ymin = d->data[ymin_i].value;
+	ymax = d->data[ymax_i].value;
+	plcol0(1);
+	pllab("Time", axis_label, plot_label);
+	for (i = start; i <= stop; i++) {
+		times[i-start] = (d->data[i].time - start_time)/100.0;
+		values[i-start] = d->data[i].value;
+	}
+	plenv(times[0], times[stop-start], ymin, ymax, 0, 2);
+	plline(stop - start + 1, times, values);
+	free(times);
+	free(values);
 }
 
 static struct cc_perioddata *
@@ -102,7 +159,7 @@ merge_data(struct cc_perioddata *first, struct cc_perioddata *last, double split
 }
 
 static void
-analyse_flight(struct cc_flightraw *f, FILE *summary_file, FILE *detail_file, char *plot_name)
+analyse_flight(struct cc_flightraw *f, FILE *summary_file, FILE *detail_file, FILE *raw_file, char *plot_name)
 {
 	double	height;
 	double	accel;
@@ -241,6 +298,17 @@ analyse_flight(struct cc_flightraw *f, FILE *summary_file, FILE *detail_file, ch
 			       time, pos, speed, accel);
 		}
 	}
+	if (raw_file) {
+		fprintf(raw_file, "%9s %9s %9s\n",
+		       "time", "height", "accel");
+		for (i = 0; i < cooked->pres.num; i++) {
+			double time = cooked->pres.data[i].time;
+			double pres = cooked->pres.data[i].value;
+			double accel = cooked->accel.data[i].value;
+			fprintf(raw_file, "%9.2f %9.2f %9.2f %9.2f\n",
+				time, pres, accel);
+		}
+	}
 	if (cooked && plot_name) {
 		struct cc_perioddata	*speed;
 		plsdev("svgcairo");
@@ -252,12 +320,12 @@ analyse_flight(struct cc_flightraw *f, FILE *summary_file, FILE *detail_file, ch
 		plstar(2, 3);
 		speed = merge_data(&cooked->accel_speed, &cooked->pres_speed, apogee);
 
-		plot_perioddata(&cooked->pres_pos, "meters", "Height", -1e10, 1e10);
-		plot_perioddata(&cooked->pres_pos, "meters", "Height", boost_start, apogee);
-		plot_perioddata(speed, "meters/second", "Speed", -1e10, 1e10);
-		plot_perioddata(speed, "meters/second", "Speed", boost_start, apogee);
-		plot_perioddata(&cooked->accel_accel, "meters/second²", "Acceleration", -1e10, 1e10);
-		plot_perioddata(&cooked->accel_accel, "meters/second²", "Acceleration", boost_start, apogee);
+		plot_perioddata(&cooked->pres_pos, "meters", "Height", -1e10, 1e10, PLOT_HEIGHT);
+		plot_perioddata(&cooked->pres_pos, "meters", "Height to Apogee", boost_start, apogee, PLOT_HEIGHT);
+		plot_perioddata(speed, "meters/second", "Speed", -1e10, 1e10, PLOT_SPEED);
+		plot_perioddata(speed, "meters/second", "Speed to Apogee", boost_start, apogee, PLOT_SPEED);
+		plot_perioddata(&cooked->accel_accel, "meters/second²", "Acceleration", -1e10, 1e10, PLOT_ACCEL);
+		plot_perioddata(&cooked->accel_accel, "meters/second²", "Acceleration to Apogee", boost_start, apogee, PLOT_ACCEL);
 		free(speed->data);
 		free(speed);
 		plend();
@@ -270,12 +338,18 @@ static const struct option options[] = {
 	{ .name = "summary", .has_arg = 1, .val = 's' },
 	{ .name = "detail", .has_arg = 1, .val = 'd' },
 	{ .name = "plot", .has_arg = 1, .val = 'p' },
+	{ .name = "raw", .has_arg = 1, .val = 'r' },
 	{ 0, 0, 0, 0},
 };
 
 static void usage(char *program)
 {
-	fprintf(stderr, "usage: %s [--summary=<summary-file>] [-s <summary-file>] [--detail=<detail-file] [-d <detail-file>] [--plot=<plot-file> -p <plot-file>] {flight-log} ...\n", program);
+	fprintf(stderr, "usage: %s\n"
+		"\t[--summary=<summary-file>] [-s <summary-file>]\n"
+		"\t[--detail=<detail-file] [-d <detail-file>]\n"
+		"\t[--raw=<raw-file> -r <raw-file]\n"
+		"\t[--plot=<plot-file> -p <plot-file>]\n"
+		"\t{flight-log} ...\n", program);
 	exit(1);
 }
 
@@ -283,18 +357,21 @@ int
 main (int argc, char **argv)
 {
 	FILE			*file;
-	FILE			*summary_file;
-	FILE			*detail_file;
+	FILE			*summary_file = NULL;
+	FILE			*detail_file = NULL;
+	FILE			*raw_file = NULL;
 	int			i;
 	int			ret = 0;
 	struct cc_flightraw	*raw;
 	int			c;
 	int			serial;
 	char			*s;
-	char			*summary_name = NULL, *detail_name = NULL;
+	char			*summary_name = NULL;
+	char			*detail_name = NULL;
+	char			*raw_name = NULL;
 	char			*plot_name = NULL;
 
-	while ((c = getopt_long(argc, argv, "s:d:p:", options, NULL)) != -1) {
+	while ((c = getopt_long(argc, argv, "s:d:p:r:", options, NULL)) != -1) {
 		switch (c) {
 		case 's':
 			summary_name = optarg;
@@ -305,13 +382,15 @@ main (int argc, char **argv)
 		case 'p':
 			plot_name = optarg;
 			break;
+		case 'r':
+			raw_name = optarg;
+			break;
 		default:
 			usage(argv[0]);
 			break;
 		}
 	}
 	summary_file = stdout;
-	detail_file = NULL;
 	if (summary_name) {
 		summary_file = fopen(summary_name, "w");
 		if (!summary_file) {
@@ -330,6 +409,13 @@ main (int argc, char **argv)
 			}
 		}
 	}
+	if (raw_name) {
+		raw_file = fopen (raw_name, "w");
+		if (!raw_file) {
+			perror(raw_name);
+			exit(1);
+		}
+	}
 	for (i = optind; i < argc; i++) {
 		file = fopen(argv[i], "r");
 		if (!file) {
@@ -350,7 +436,7 @@ main (int argc, char **argv)
 		}
 		if (!raw->serial)
 			raw->serial = serial;
-		analyse_flight(raw, summary_file, detail_file, plot_name);
+		analyse_flight(raw, summary_file, detail_file, raw_file, plot_name);
 		cc_flightraw_free(raw);
 	}
 	return ret;