]> git.tdb.fi Git - ext/subsurface.git/blobdiff - profile.c
Show the min/max data in funky purple shading
[ext/subsurface.git] / profile.c
index 29874d9c6861c66c57db9878c41a976dfc7d8a01..12e648be92abef21933c2ee6efd9e0e595c60ce7 100644 (file)
--- a/profile.c
+++ b/profile.c
@@ -1,6 +1,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <stdarg.h>
+#include <string.h>
 #include <time.h>
 
 #include "dive.h"
@@ -23,6 +24,21 @@ struct graphics_context {
        double scalex, scaley;
 };
 
+/* Plot info with smoothing and one-, two- and three-minute minimums and maximums */
+struct plot_info {
+       int nr;
+       int maxtime;
+       struct plot_data {
+               int sec;
+               int val;
+               int smoothed;
+               int min[3];
+               int max[3];
+               int avg[3];
+       } entry[];
+};
+#define plot_info_size(nr) (sizeof(struct plot_info) + (nr)*sizeof(struct plot_data))
+
 /* Scale to 0,0 -> maxx,maxy */
 #define SCALE(gc,x,y) (x)*gc->maxx/gc->scalex,(y)*gc->maxy/gc->scaley
 
@@ -57,11 +73,13 @@ static int round_depth_up(depth_t depth)
 }
 
 typedef struct {
+       int size;
        double r,g,b;
-       enum {CENTER,LEFT} allign;
+       enum {CENTER,LEFT} halign;
+       enum {MIDDLE,TOP,BOTTOM} valign;
 } text_render_options_t;
 
-static void plot_text(struct graphics_context *gc, text_render_options_t *tro,
+static void plot_text(struct graphics_context *gc, const text_render_options_t *tro,
                      double x, double y, const char *fmt, ...)
 {
        cairo_t *cr = gc->cr;
@@ -74,11 +92,28 @@ static void plot_text(struct graphics_context *gc, text_render_options_t *tro,
        vsnprintf(buffer, sizeof(buffer), fmt, args);
        va_end(args);
 
+       cairo_set_font_size(cr, tro->size);
        cairo_text_extents(cr, buffer, &extents);
        dx = 0;
-       if (tro->allign == CENTER)
+       switch (tro->halign) {
+       case CENTER:
                dx = -(extents.width/2 + extents.x_bearing);
-       dy = extents.height * 1.2;
+               break;
+       case LEFT:
+               dx = 0;
+               break;
+       }
+       switch (tro->valign) {
+       case TOP:
+               dy = extents.height * 1.2;
+               break;
+       case BOTTOM:
+               dy = -extents.height * 0.8;
+               break;
+       case MIDDLE:
+               dy = 0;
+               break;
+       }
 
        move_to(gc, x, y);
        cairo_rel_move_to(cr, dx, dy);
@@ -94,6 +129,26 @@ static void plot_text(struct graphics_context *gc, text_render_options_t *tro,
        cairo_show_text(cr, buffer);
 }
 
+static void render_depth_sample(struct graphics_context *gc, struct sample *sample, const text_render_options_t *tro)
+{
+       int sec = sample->time.seconds;
+       depth_t depth = sample->depth;
+       const char *fmt;
+       double d;
+
+       switch (output_units.length) {
+       case METERS:
+               d = depth.mm / 1000.0;
+               fmt = "%.1f";
+               break;
+       case FEET:
+               d = to_feet(depth);
+               fmt = "%.0f";
+               break;
+       }
+       plot_text(gc, tro, sec, depth.mm, fmt, d);
+}
+
 /*
  * Find the next minimum/maximum point.
  *
@@ -113,12 +168,11 @@ static struct sample *next_minmax(struct sample *sample, struct sample *end, int
        result = NULL;
 
        for (;;) {
-               int time, depth;
+               int depth;
 
                sample++;
                if (sample >= end)
                        return NULL;
-               time = sample->time.seconds;
                depth = sample->depth.mm;
 
                if (minmax) {
@@ -141,39 +195,24 @@ static struct sample *next_minmax(struct sample *sample, struct sample *end, int
        return result;
 }
 
-static void render_depth_sample(struct graphics_context *gc, struct sample *sample)
-{
-       text_render_options_t tro = {1.0, 0.2, 0.2, CENTER};
-       int sec = sample->time.seconds;
-       depth_t depth = sample->depth;
-       const char *fmt;
-       double d;
-
-       switch (output_units.length) {
-       case METERS:
-               d = depth.mm / 1000.0;
-               fmt = "%.1f";
-               break;
-       case FEET:
-               d = to_feet(depth);
-               fmt = "%.0f";
-               break;
-       }
-       plot_text(gc, &tro, sec, depth.mm, fmt, d);
-}
-
 static void plot_text_samples(struct graphics_context *gc, struct sample *a, struct sample *b)
 {
+       static const text_render_options_t deep = {14, 1.0, 0.2, 0.2, CENTER, TOP};
+       static const text_render_options_t shallow = {14, 1.0, 0.2, 0.2, CENTER, BOTTOM};
+
        for (;;) {
                if (b <= a)
                        break;
                a = next_minmax(a, b, 1);
                if (!a)
-                       return;
-               render_depth_sample(gc, a);
+                       break;
+               render_depth_sample(gc, a, &deep);
                a = next_minmax(a, b, 0);
                if (!a)
                        break;
+               if (a->depth.mm < 2500)
+                       continue;
+               render_depth_sample(gc, a, &shallow);
        }
 }
 
@@ -189,26 +228,61 @@ static void plot_depth_text(struct dive *dive, struct graphics_context *gc)
        gc->scalex = maxtime;
        gc->scaley = maxdepth;
 
-       cairo_set_font_size(gc->cr, 14);
-
        sample = dive->sample;
        end = dive->sample + dive->samples;
 
        plot_text_samples(gc, sample, end);
 }
 
-static void plot_depth_profile(struct dive *dive, struct graphics_context *gc)
+static void plot_smoothed_profile(struct graphics_context *gc, struct plot_info *pi)
+{
+       int i;
+       struct plot_data *entry = pi->entry;
+
+       cairo_set_source_rgba(gc->cr, 1, 0.2, 0.2, 0.20);
+       move_to(gc, entry->sec, entry->smoothed);
+       for (i = 1; i < pi->nr; i++) {
+               entry++;
+               line_to(gc, entry->sec, entry->smoothed);
+       }
+       cairo_stroke(gc->cr);
+}
+
+static void plot_minmax_profile_minute(struct graphics_context *gc, struct plot_info *pi,
+                               int index, double a)
+{
+       int i;
+       struct plot_data *entry = pi->entry;
+
+       cairo_set_source_rgba(gc->cr, 1, 0.2, 1, a);
+       move_to(gc, entry->sec, entry->min[index]);
+       for (i = 1; i < pi->nr; i++) {
+               entry++;
+               line_to(gc, entry->sec, entry->min[index]);
+       }
+       for (i = 1; i < pi->nr; i++) {
+               line_to(gc, entry->sec, entry->max[index]);
+               entry--;
+       }
+       cairo_close_path(gc->cr);
+       cairo_fill(gc->cr);
+}
+
+static void plot_minmax_profile(struct graphics_context *gc, struct plot_info *pi)
+{
+       plot_minmax_profile_minute(gc, pi, 2, 0.1);
+       plot_minmax_profile_minute(gc, pi, 1, 0.1);
+       plot_minmax_profile_minute(gc, pi, 0, 0.1);
+}
+
+static void plot_depth_profile(struct dive *dive, struct graphics_context *gc, struct plot_info *pi)
 {
+       int i;
        cairo_t *cr = gc->cr;
        int begins, sec, depth;
-       int i, samples;
-       struct sample *sample;
+       struct plot_data *entry;
        int maxtime, maxdepth, marker;
 
-       samples = dive->samples;
-       if (!samples)
-               return;
-
        cairo_set_line_width(gc->cr, 2);
 
        /* Get plot scaling limits */
@@ -246,15 +320,18 @@ static void plot_depth_profile(struct dive *dive, struct graphics_context *gc)
 
        gc->scalex = maxtime;
 
-       sample = dive->sample;
+       plot_smoothed_profile(gc, pi);
+       plot_minmax_profile(gc, pi);
+
+       entry = pi->entry;
        cairo_set_source_rgba(cr, 1, 0.2, 0.2, 0.80);
-       begins = sample->time.seconds;
-       move_to(gc, sample->time.seconds, sample->depth.mm);
-       for (i = 1; i < dive->samples; i++) {
-               sample++;
-               sec = sample->time.seconds;
+       begins = entry->sec;
+       move_to(gc, entry->sec, entry->val);
+       for (i = 1; i < pi->nr; i++) {
+               entry++;
+               sec = entry->sec;
                if (sec <= maxtime) {
-                       depth = sample->depth.mm;
+                       depth = entry->val;
                        line_to(gc, sec, depth);
                }
        }
@@ -363,7 +440,7 @@ static double calculate_airuse(struct dive *dive)
 
 static void plot_info(struct dive *dive, struct graphics_context *gc)
 {
-       text_render_options_t tro = {0.2, 1.0, 0.2, LEFT};
+       text_render_options_t tro = {10, 0.2, 1.0, 0.2, LEFT, TOP};
        const double liters_per_cuft = 28.317;
        const char *unit;
        double airuse;
@@ -394,8 +471,6 @@ static void plot_cylinder_pressure_text(struct dive *dive, struct graphics_conte
 {
        pressure_t startp, endp;
 
-       cairo_set_font_size(gc->cr, 10);
-
        if (get_cylinder_pressure_range(dive, gc, &startp, &endp)) {
                int start, end;
                const char *unit = "bar";
@@ -418,16 +493,114 @@ static void plot_cylinder_pressure_text(struct dive *dive, struct graphics_conte
                        break;
                }
 
-               text_render_options_t tro = {0.2, 1.0, 0.2, LEFT};
+               text_render_options_t tro = {10, 0.2, 1.0, 0.2, LEFT, TOP};
                plot_text(gc, &tro, 0, startp.mbar, "%d %s", start, unit);
                plot_text(gc, &tro, dive->duration.seconds, endp.mbar,
                          "%d %s", end, unit);
        }
 }
 
+static void analyze_plot_info_minmax_minute(struct plot_data *entry, struct plot_data *first, struct plot_data *last, int index)
+{
+       struct plot_data *p = entry;
+       int time = entry->sec;
+       int seconds = 60*(index+1);
+       int min, max, avg, nr;
+
+       /* Go back 'seconds' in time */
+       while (p > first) {
+               if (p[-1].sec < time - seconds)
+                       break;
+               p--;
+       }
+
+       /* Then go forward until we hit an entry past the time */
+       min = max = avg = p->val;
+       nr = 1;
+       while (++p < last) {
+               int val = p->val;
+               if (p->sec > time + seconds)
+                       break;
+               avg += val;
+               nr ++;
+               if (val < min)
+                       min = val;
+               if (val > max)
+                       max = val;
+       }
+       entry->min[index] = min;
+       entry->max[index] = max;
+       entry->avg[index] = (avg + nr/2) / nr;
+}
+
+static void analyze_plot_info_minmax(struct plot_data *entry, struct plot_data *first, struct plot_data *last)
+{
+       analyze_plot_info_minmax_minute(entry, first, last, 0);
+       analyze_plot_info_minmax_minute(entry, first, last, 1);
+       analyze_plot_info_minmax_minute(entry, first, last, 2);
+}
+
+static struct plot_info *analyze_plot_info(struct plot_info *pi)
+{
+       int i;
+       int nr = pi->nr;
+
+       /* Smoothing function: 5-point triangular smooth */
+       for (i = 2; i < nr-2; i++) {
+               struct plot_data *entry = pi->entry+i;
+               int val;
+
+               val = entry[-2].val + 2*entry[-1].val + 3*entry[0].val + 2*entry[1].val + entry[2].val;
+               entry->smoothed = (val+4) / 9;
+       }
+
+       /* One-, two- and three-minute minmax data */
+       for (i = 0; i < nr; i++) {
+               struct plot_data *entry = pi->entry +i;
+               analyze_plot_info_minmax(entry, pi->entry, pi->entry+nr);
+       }
+       
+       return pi;
+}
+
+/*
+ * Create a plot-info with smoothing and ranged min/max
+ *
+ * This also makes sure that we have extra empty events on both
+ * sides, so that you can do end-points without having to worry
+ * about it.
+ */
+static struct plot_info *depth_plot_info(struct dive *dive)
+{
+       int i, nr = dive->samples + 4, sec;
+       size_t alloc_size = plot_info_size(nr);
+       struct plot_info *pi;
+
+       pi = malloc(alloc_size);
+       if (!pi)
+               return pi;
+       memset(pi, 0, alloc_size);
+       pi->nr = nr;
+       sec = 0;
+       for (i = 0; i < dive->samples; i++) {
+               struct sample *sample = dive->sample+i;
+               struct plot_data *entry = pi->entry + i + 2;
+
+               sec = entry->sec = sample->time.seconds;
+               entry->val = sample->depth.mm;
+       }
+       /* Fill in the last two entries with empty values but valid times */
+       i = dive->samples + 2;
+       pi->entry[i].sec = sec + 20;
+       pi->entry[i+1].sec = sec + 40;
+
+       return analyze_plot_info(pi);
+}
+
 static void plot(struct graphics_context *gc, int w, int h, struct dive *dive)
 {
        double topx, topy;
+       struct plot_info *pi = depth_plot_info(dive);
 
        topx = w / 20.0;
        topy = h / 20.0;
@@ -447,7 +620,7 @@ static void plot(struct graphics_context *gc, int w, int h, struct dive *dive)
        plot_cylinder_pressure(dive, gc);
 
        /* Depth profile */
-       plot_depth_profile(dive, gc);
+       plot_depth_profile(dive, gc, pi);
 
        /* Text on top of all graphs.. */
        plot_depth_text(dive, gc);