--- /dev/null
+#! /usr/bin/awk -f
+#
+# Plot Simply
+#
+# TODO: Incremental plotting (new canvas for each data line)
+# TODO: Support manual set of max y
+
+function orange(str) {return "\033[33m" str "\033[0m";}
+function orange_bright(str) {return "\033[1;33m" str "\033[0m";}
+function green(str) {return "\033[32m" str "\033[0m";}
+function grey_dark(str) {return "\033[30m" str "\033[0m";}
+function grey_light(str) {return "\033[37m" str "\033[0m";}
+
+function abs(n) {return n >= 0 ? n : -n}
+function round(n_float) {return int(n_float + 0.5)}
+function log_10(n) {return log(n) / log(10)}
+
+function width(n, w) {
+ #
+ # log(0) == -inf
+ #
+ w = 1 + round(n > 0 ? log_10(n) : (n == 0 ? 1 : log_10(-n) + 1)); # +1 for neg sign
+ #printf("n: %d, w: %d\n", n, w);
+ return w;
+}
+
+function bins_append_item_to_bin(bin, val, bin_i) {
+ bin_i = bin_item_count[bin]++;
+ bins[bin, bin_i] = val;
+}
+
+BEGIN {
+ # CLI options: w, h, p, a, b
+ limits["canvas_width"] = (w ? w : 70);
+ limits["canvas_height"] = (h ? h : 20);
+ char_point = orange_bright(p ? p : ".");
+ char_pad = grey_dark(a ? a : ""); # FIXME: non-blank pad is broken after adding the y labels
+ char_blank = grey_dark(b ? b : "|");
+ aggregation = g ? g : "mean";
+ data_field = df ? df : 1;
+
+ y_orig_min = 0;
+ y_orig_max = 0;
+}
+
+{
+ y_orig = $data_field;
+ if (y_orig > y_orig_max) y_orig_max = y_orig;
+ if (y_orig < y_orig_min) y_orig_min = y_orig;
+ data_0[NR] = y_orig;
+}
+
+function aggregate_mean() {
+}
+
+function aggregate(array) {
+ if (aggregation == "mean") {
+ return aggregate_mean(array);
+ } else {
+ printf("Unknown aggregation: %s\n", aggregation) > "/dev/stderr";
+ exit(2);
+ }
+}
+
+function data_scaled_x_to_width(\
+ data_in, data_out, limits,
+
+ x_orig,
+ datum,
+ x_bin,
+ bins,
+ bin_item_counts,
+ bin_sum,
+ bin_item_count,
+ i,
+ bin_item,
+ bin_mean \
+) {
+ # Find limit points
+ x_orig_min = 0;
+ x_orig_max = 0;
+ for (x_orig in data_in) {
+ if (x_orig > x_orig_max) x_orig_max = x_orig;
+ if (x_orig < x_orig_min) x_orig_min = x_orig;
+ }
+ #x_orig_range = x_orig_max - x_orig_min;
+ # Place items in bins
+ x_bin_min = 0;
+ x_bin_max = 0;
+ for (x_orig in data_in) {
+ datum = data_in[x_orig];
+ x_bin = round(x_orig * limits["canvas_width_x"] / length(data_in));
+ if (x_bin > x_bin_max) x_bin_max = x_bin;
+ if (x_bin < x_bin_min) x_bin_min = x_bin;
+ bin_item_count = ++bin_item_counts[x_bin];
+ bins[x_bin, bin_item_count] = datum;
+ #printf("x_orig: %f, x_bin: %f, bin_item_count: %f, datum: %f\n", x_orig, x_bin, bin_item_count, datum);
+ }
+ # Aggregate bins
+ for (x_bin in bin_item_counts) {
+ bin_sum = 0;
+ bin_item_count = bin_item_counts[x_bin];
+ for (i = 1; i <= bin_item_count; i++) {
+ bin_item = bins[x_bin, i];
+ bin_sum += bin_item;
+ #printf("x_bin: %f, bin_item_count: %f, bin_item: %f, bin_sum: %f\n", x_bin, bin_item_count, bin_item, bin_sum);
+ }
+ bin_mean = bin_sum / bin_item_count;
+ data_out[x_bin] = bin_mean;
+ #printf("x_bin: %f, bin_mean: %f, bin_sum: %f, bin_item_count: %f\n",
+ #x_bin, bin_mean, bin_sum, bin_item_count);
+ }
+ limits["x_min"] = x_bin_min;
+ limits["x_max"] = x_bin_max;
+}
+
+function data_scaled_y_to_height(\
+ data_in, data_out, limits,
+
+ data_scaled, x, y, y_orig, y_scaled \
+) {
+ # Offset orig
+ # TODO: Is there a better, closed-form way to get the offset?
+ # TODO: Is there better way to map canvas to value ranges altogether?
+ if (y_orig_min < 0) {
+ offset_orig = -1 * y_orig_min;
+ } else {
+ offset_orig = 0;
+ }
+ for (x in data_in) {
+ y = data_in[x];
+ #printf("x: %f, y: %f, y_min: %f, y_max: %f\n", x, y, y_min, y_max);
+ data_in_offsetted[x] = y + offset_orig;
+ }
+ y_orig_offseted_min = y_orig_min + offset_orig;
+ y_orig_offseted_max = y_orig_max + offset_orig;
+
+ # Scale to height
+ y_scaled_min = 0;
+ y_scaled_max = 0;
+ for (x in data_in_offsetted) {
+ y_orig_offsetted = data_in_offsetted[x];
+ y_scaled = \
+ y_orig_offseted_max > 0 \
+ ? round((y_orig_offsetted * limits["canvas_height"]) / y_orig_offseted_max) \
+ : 0;
+ #printf(\
+ #"x: %6.2f, y_orig_offsetted: %6.2f, y_orig_max: %6.2f, y_scaled: %6.2f\n",
+ #x, y_orig_offsetted, y_orig_max, y_scaled);
+ if (y_scaled > y_scaled_max) y_scaled_max = y_scaled;
+ if (y_scaled < y_scaled_min) y_scaled_min = y_scaled;
+ data_out[x] = y_scaled
+ }
+
+ # Save limits
+ limits["y_min"] = y_scaled_min;
+ limits["y_max"] = y_scaled_max;
+ range_orig = y_orig_max - y_orig_min;
+ offset_scaled = \
+ range_orig > 0 \
+ ? round(offset_orig * limits["canvas_height"] / range_orig) \
+ : 0;
+ limits["offset_scaled"] = offset_scaled;
+ #printf("offset_orig: %f, offset_scaled: %f\n", offset_orig, offset_scaled);
+}
+
+function canvas_init(canvas, width, height, row, col) {
+ for (row=0; row <= height; row++) {
+ for (col=0; col <= width; col++) {
+ canvas[row, col] = char_pad char_blank char_pad;
+ }
+ }
+}
+
+function canvas_overlay_highlight_ticks_x(canvas, limits, row, col) {
+ for (col=limits["canvas_width_y"] - 1; col <= limits["canvas_width"]; col++) {
+ offset = limits["offset_scaled"];
+ #printf("offset: %f\n", offset);
+ row = 0 + offset;
+ #row = 0;
+ canvas[row, col] = char_pad green("-") char_pad;
+ }
+}
+
+function canvas_overlay_highlight_zero_row(canvas, limits, row, col) {
+ #print "canvas_overlay_highlight_zero_row";
+ for (col=limits["canvas_width_y"] - 1; col <= limits["canvas_width"]; col++) {
+ offset = limits["offset_scaled"];
+ row = 0 + offset;
+ #printf("col: %6.2f, row: %6.2f, offset: %f\n", col, row, offset);
+ #row = 0;
+ canvas[row, col] = char_pad green("-") char_pad;
+ }
+}
+
+function canvas_overlay_highlight_zero_col(canvas, limits, row, col) {
+ #print "canvas_overlay_highlight_zero_col";
+ for (row=0; row <= limits["canvas_height"]; row++) {
+ col = limits["canvas_width_y"]; # was also -1. Why?
+ # TODO: Refactor color/character configs to ease composition
+ canvas[row, col] = green("|");
+ }
+ canvas[limits["canvas_height"], limits["canvas_width_y"]] = green("+");
+ canvas[0 , limits["canvas_width_y"]] = green("+");
+}
+function canvas_overlay_highlight_zero(canvas, limits) {
+ #print "canvas_overlay_highlight_zero";
+ canvas[0 + limits["offset_scaled"], 0 + limits["canvas_width_y"]] = green("+");
+}
+
+function canvas_overlay_data(canvas, data, limits, x_data, x_canvas, y, yi, yj) {
+ #print "canvas_overlay_data";
+ for (x_data in data) {
+ x_canvas = x_data + limits["canvas_width_y"] + 1;
+ y = data[x_data];
+ # TODO: Would be nice to scale width of all cells to the widest
+ #point = y;
+ #printf("canvas_width_y: %6.2f, x0: %6.2f, x1: %6.2f, x: %6.2f, y: %6.2f\n",
+ #limits["canvas_width_y"], x0, x1, x, y);
+ # TODO: This special case for 0 is kind of a kludge - can we do better?
+ canvas[y, x_canvas] = x_data == 0 ? char_point : char_pad char_point char_pad;
+
+ if (y > limits["offset_scaled"]) {
+ for (yi = y - 1; yi >= limits["offset_scaled"]; yi--) {
+ #printf("yi: %6.2f\n", yi);
+ canvas[yi, x_canvas] = x_data == 0 ? orange("|") : char_pad orange("|") char_pad;
+ }
+ } else if (y < limits["offset_scaled"]) {
+ for (yj = limits["offset_scaled"]; yj > y; yj--) {
+ #printf("yj: %6.2f\n", yj);
+ canvas[yj, x_canvas] = x_data == 0 ? orange("|") : char_pad orange("|") char_pad;
+ }
+ }
+ }
+}
+
+function canvas_overlay_y_lab(canvas, limits, y_lab_fmt, y_max_str, i) {
+ y_lab_fmt = "%" limits["canvas_width_y"] - 1 "d ";
+ y_max_str = sprintf(y_lab_fmt, y_orig_max);
+ y_min_str = sprintf(y_lab_fmt, y_orig_min);
+ #printf("y_width: %f, y_max_str: \"%s\", y_min_str: \"%s\"\n", limits["canvas_width_y"], y_max_str, y_min_str);
+ for (i=1; i<=length(y_max_str); i++) {
+ canvas[limits["canvas_height"], i - 1] = substr(y_max_str, i, 1);
+ }
+ canvas[0 + limits["offset_scaled"], 0 + limits["canvas_width_y"] - 1] = 0;
+ for (i=1; i<=length(y_min_str); i++) {
+ canvas[0, i - 1] = substr(y_min_str, i, 1);
+ }
+}
+
+function canvas_print(canvas, limits, row, col) {
+ for (row = limits["canvas_height"]; row >= 0; row--) {
+ for (col = 0; col <= limits["canvas_width"]; col++) {
+ printf("%s", canvas[row, col]);
+ }
+ printf("\n");
+ }
+}
+
+END {
+ # Find maximum y number width
+ y_orig_min_width = width(y_orig_min);
+ y_orig_max_width = width(y_orig_max);
+ if (y_orig_max_width >= y_orig_min_width) {
+ y_width = y_orig_max_width;
+ } else {
+ y_width = y_orig_min_width;
+ }
+ limits["canvas_width_y"] = y_width + 1;
+ limits["canvas_width_x"] = limits["canvas_width"] - limits["canvas_width_y"];
+
+ data_scaled_x_to_width(data_0, data_1, limits);
+ data_scaled_y_to_height(data_1, data_2, limits);
+
+ canvas_init(canvas, limits["canvas_width"], limits["canvas_height"]);
+ canvas_overlay_highlight_zero_row(canvas, limits);
+ canvas_overlay_highlight_zero_col(canvas, limits);
+ canvas_overlay_highlight_zero(canvas, limits);
+ canvas_overlay_y_lab(canvas, limits);
+ canvas_overlay_data(canvas, data_2, limits);
+ #for (l in limits) {
+ #printf("limits[%s] -> %s\n", l, limits[l]);
+ #}
+ canvas_print(canvas, limits);
+}
+
+# An even better way to think about scaling: ratios!!! Duh! :-D
+#
+# val_max / val_current = width / val_scaled
+#
+# val_max width
+# ----------- = ------------
+# val_current val_scaled
+#
+# val_max * val_scaled = val_current * width
+# val_scaled = (val_current * width) / val_max
+#
+#
+# num_data_points width
+# ---------------- = -----
+# x 1
+#
+# width * x = num_data_points
+# x = num_data_points / width
+#
+# But that is what I already tried, and it is awkward to scale up when
+# thinking thsese terms, so it is much better to first route each data
+# point to an appropriate bin and then aggregate each bin:
+# 1. Route: bins[scale(datum)]
+# 2. Aggregate: for bin in bins: for val in bin: aggregate(val)
home / code / khome.git / commitdiff
