]> git.tdb.fi Git - ext/subsurface.git/blobdiff - profile.c
Print the end temperature of the dive
[ext/subsurface.git] / profile.c
index 9b91ca7b1a7ad608a0fa36f4de53c7c0234c8047..fdbbcdf165fe1746b3dbc85aeb7afe35a788ee09 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"
 
 int selected_dive = 0;
 
-/*
- * Cairo scaling really is horribly horribly mis-designed.
- *
- * Which is sad, because I really like Cairo otherwise. But
- * the fact that the line width is scaled with the same scale
- * as the coordinate system is a f*&%ing disaster. So we
- * can't use it, and instead have this butt-ugly wrapper thing..
- */
-struct graphics_context {
-       cairo_t *cr;
-       double maxx, maxy;
-       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;
+               struct plot_data *min[3];
+               struct plot_data *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
+#define SCALEX(gc,x)  (((x)-gc->leftx)/(gc->rightx-gc->leftx)*gc->maxx)
+#define SCALEY(gc,y)  (((y)-gc->topy)/(gc->bottomy-gc->topy)*gc->maxy)
+#define SCALE(gc,x,y) SCALEX(gc,x),SCALEY(gc,y)
 
 static void move_to(struct graphics_context *gc, double x, double y)
 {
@@ -36,6 +40,24 @@ static void line_to(struct graphics_context *gc, double x, double y)
        cairo_line_to(gc->cr, SCALE(gc, x, y));
 }
 
+static void set_source_rgba(struct graphics_context *gc, double r, double g, double b, double a)
+{
+       if (gc->printer) {
+               /* Black is white and white is black */
+               double sum = r+g+b;
+               if (sum > 2)
+                       r = g = b = 0;
+               else if (sum < 1)
+                       r = g = b = 1;
+       }
+       cairo_set_source_rgba(gc->cr, r, g, b, a);
+}
+
+static void set_source_rgb(struct graphics_context *gc, double r, double g, double b)
+{
+       set_source_rgba(gc, r, g, b, 1);
+}
+
 #define ROUND_UP(x,y) ((((x)+(y)-1)/(y))*(y))
 
 /*
@@ -63,7 +85,7 @@ typedef struct {
        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;
@@ -103,21 +125,20 @@ static void plot_text(struct graphics_context *gc, text_render_options_t *tro,
        cairo_rel_move_to(cr, dx, dy);
 
        cairo_text_path(cr, buffer);
-       cairo_set_source_rgb(cr, 0, 0, 0);
+       set_source_rgb(gc, 0, 0, 0);
        cairo_stroke(cr);
 
        move_to(gc, x, y);
        cairo_rel_move_to(cr, dx, dy);
 
-       cairo_set_source_rgb(cr, tro->r, tro->g, tro->b);
+       set_source_rgb(gc, tro->r, tro->g, tro->b);
        cairo_show_text(cr, buffer);
 }
 
-static void render_depth_sample(struct graphics_context *gc, struct sample *sample)
+static void render_depth_sample(struct graphics_context *gc, struct plot_data *entry, const text_render_options_t *tro)
 {
-       text_render_options_t tro = {14, 1.0, 0.2, 0.2, CENTER, TOP};
-       int sec = sample->time.seconds;
-       depth_t depth = sample->depth;
+       int sec = entry->sec;
+       depth_t depth = { entry->val };
        const char *fmt;
        double d;
 
@@ -131,124 +152,115 @@ static void render_depth_sample(struct graphics_context *gc, struct sample *samp
                fmt = "%.0f";
                break;
        }
-       plot_text(gc, &tro, sec, depth.mm, fmt, d);
+       plot_text(gc, tro, sec, depth.mm, fmt, d);
 }
 
-/*
- * Find the next minimum/maximum point.
- *
- * We exit early if we hit "enough" of a depth reversal,
- * which is roughly 10 feet.
- */
-static struct sample *next_minmax(struct sample *sample, struct sample *end, int minmax)
+static void plot_text_samples(struct graphics_context *gc, struct plot_info *pi)
 {
-       const int enough = 3000;
-       struct sample *result;
-       int depthlimit;
-
-       if (sample >= end)
-               return 0;
-
-       depthlimit = sample->depth.mm;
-       result = NULL;
-
-       for (;;) {
-               int time, depth;
+       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};
+       int i;
 
-               sample++;
-               if (sample >= end)
-                       return NULL;
-               time = sample->time.seconds;
-               depth = sample->depth.mm;
+       for (i = 0; i < pi->nr; i++) {
+               struct plot_data *entry = pi->entry + i;
 
-               if (minmax) {
-                       if (depth <= depthlimit) {
-                               if (depthlimit - depth > enough)
-                                       break;
-                               continue;
-                       }
-               } else {
-                       if (depth >= depthlimit) {
-                               if (depth - depthlimit > enough)
-                                       break;
-                               continue;
-                       }
-               }
+               if (entry->val < 2000)
+                       continue;
 
-               result = sample;
-               depthlimit = depth;
-       }
-       return result;
-}
+               if (entry == entry->max[2])
+                       render_depth_sample(gc, entry, &deep);
 
-static void plot_text_samples(struct graphics_context *gc, struct sample *a, struct sample *b)
-{
-       for (;;) {
-               if (b <= a)
-                       break;
-               a = next_minmax(a, b, 1);
-               if (!a)
-                       break;
-               render_depth_sample(gc, a);
-               a = next_minmax(a, b, 0);
-               if (!a)
-                       break;
+               if (entry == entry->min[2])
+                       render_depth_sample(gc, entry, &shallow);
        }
 }
 
-static void plot_depth_text(struct dive *dive, struct graphics_context *gc)
+static void plot_depth_text(struct dive *dive, struct graphics_context *gc, struct plot_info *pi)
 {
-       struct sample *sample, *end;
        int maxtime, maxdepth;
 
        /* Get plot scaling limits */
        maxtime = round_seconds_up(dive->duration.seconds);
        maxdepth = round_depth_up(dive->maxdepth);
 
-       gc->scalex = maxtime;
-       gc->scaley = maxdepth;
+       gc->leftx = 0; gc->rightx = maxtime;
+       gc->topy = 0; gc->bottomy = maxdepth;
+
+       plot_text_samples(gc, pi);
+}
+
+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);
+}
 
-       sample = dive->sample;
-       end = dive->sample + dive->samples;
+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;
 
-       plot_text_samples(gc, sample, end);
+       cairo_set_source_rgba(gc->cr, 1, 0.2, 1, a);
+       move_to(gc, entry->sec, entry->min[index]->val);
+       for (i = 1; i < pi->nr; i++) {
+               entry++;
+               line_to(gc, entry->sec, entry->min[index]->val);
+       }
+       for (i = 1; i < pi->nr; i++) {
+               line_to(gc, entry->sec, entry->max[index]->val);
+               entry--;
+       }
+       cairo_close_path(gc->cr);
+       cairo_fill(gc->cr);
 }
 
-static void plot_depth_profile(struct dive *dive, struct graphics_context *gc)
+static void plot_minmax_profile(struct graphics_context *gc, struct plot_info *pi)
 {
+       if (gc->printer)
+               return;
+       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 */
        maxtime = round_seconds_up(dive->duration.seconds);
        maxdepth = round_depth_up(dive->maxdepth);
 
        /* Time markers: every 5 min */
-       gc->scalex = maxtime;
-       gc->scaley = 1.0;
+       gc->leftx = 0; gc->rightx = maxtime;
+       gc->topy = 0; gc->bottomy = 1.0;
        for (i = 5*60; i < maxtime; i += 5*60) {
                move_to(gc, i, 0);
                line_to(gc, i, 1);
        }
 
        /* Depth markers: every 30 ft or 10 m*/
-       gc->scalex = 1.0;
-       gc->scaley = maxdepth;
+       gc->leftx = 0; gc->rightx = 1.0;
+       gc->topy = 0; gc->bottomy = maxdepth;
        switch (output_units.length) {
        case METERS: marker = 10000; break;
        case FEET: marker = 9144; break;        /* 30 ft */
        }
 
-       cairo_set_source_rgba(cr, 1, 1, 1, 0.5);
+       set_source_rgba(gc, 1, 1, 1, 0.5);
        for (i = marker; i < maxdepth; i += marker) {
                move_to(gc, 0, i);
                line_to(gc, 1, i);
@@ -256,32 +268,139 @@ static void plot_depth_profile(struct dive *dive, struct graphics_context *gc)
        cairo_stroke(cr);
 
        /* Show mean depth */
-       cairo_set_source_rgba(cr, 1, 0.2, 0.2, 0.40);
+       set_source_rgba(gc, 1, 0.2, 0.2, 0.40);
        move_to(gc, 0, dive->meandepth.mm);
        line_to(gc, 1, dive->meandepth.mm);
        cairo_stroke(cr);
 
-       gc->scalex = maxtime;
+       gc->leftx = 0; gc->rightx = maxtime;
 
-       sample = dive->sample;
-       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;
+       plot_smoothed_profile(gc, pi);
+       plot_minmax_profile(gc, pi);
+
+       entry = pi->entry;
+       set_source_rgba(gc, 1, 0.2, 0.2, 0.80);
+       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);
                }
        }
-       gc->scaley = 1.0;
+       gc->topy = 0; gc->bottomy = 1.0;
        line_to(gc, MIN(sec,maxtime), 0);
        line_to(gc, begins, 0);
        cairo_close_path(cr);
-       cairo_set_source_rgba(cr, 1, 0.2, 0.2, 0.20);
+       set_source_rgba(gc, 1, 0.2, 0.2, 0.20);
        cairo_fill_preserve(cr);
-       cairo_set_source_rgba(cr, 1, 0.2, 0.2, 0.80);
+       set_source_rgba(gc, 1, 0.2, 0.2, 0.80);
+       cairo_stroke(cr);
+}
+
+static int setup_temperature_limits(struct dive *dive, struct graphics_context *gc)
+{
+       int i;
+       int maxtime, mintemp, maxtemp;
+
+       /* Get plot scaling limits */
+       maxtime = round_seconds_up(dive->duration.seconds);
+       mintemp = INT_MAX;
+       maxtemp = 0;
+       for (i = 0; i < dive->samples; i++) {
+               struct sample *sample = dive->sample+i;
+               int mkelvin = sample->temperature.mkelvin;
+               if (!mkelvin)
+                       continue;
+               if (mkelvin > maxtemp)
+                       maxtemp = mkelvin;
+               if (mkelvin < mintemp)
+                       mintemp = mkelvin;
+       }
+
+       gc->leftx = 0; gc->rightx = maxtime;
+       /* Show temperatures in roughly the lower third */
+       gc->topy = maxtemp + (maxtemp - mintemp)*2;
+       gc->bottomy = mintemp - (maxtemp - mintemp)/2;
+
+       return maxtemp > mintemp;
+}
+
+static void plot_single_temp_text(struct graphics_context *gc, int sec, temperature_t temperature)
+{
+       int deg;
+       const char *unit;
+       static const text_render_options_t tro = {12, 0.2, 0.2, 1.0, LEFT, TOP};
+
+       if (output_units.temperature == FAHRENHEIT) {
+               deg = to_F(temperature);
+               unit = "F";
+       } else {
+               deg = to_C(temperature);
+               unit = "C";
+       }
+       plot_text(gc, &tro, sec, temperature.mkelvin, "%d %s", deg, unit);
+}
+
+static void plot_temperature_text(struct dive *dive, struct graphics_context *gc)
+{
+       int i;
+       int last = 0;
+       temperature_t last_temperature, last_printed_temp;
+
+       if (!setup_temperature_limits(dive, gc))
+               return;
+
+       for (i = 0; i < dive->samples; i++) {
+               struct sample *sample = dive->sample+i;
+               if (sample->time.seconds > dive->duration.seconds)
+                       break; /* let's not plot surface temp events */
+               int mkelvin = sample->temperature.mkelvin;
+               int sec;
+               if (!mkelvin)
+                       continue;
+               last_temperature = sample->temperature;
+               sec = sample->time.seconds;
+               if (sec < last)
+                       continue;
+               last = sec + 300;
+               plot_single_temp_text(gc,sec,sample->temperature);
+               last_printed_temp = last_temperature ;
+       }
+       /* it would be nice to print the end temperature, if it's different */
+       if (last_temperature.mkelvin != last_printed_temp.mkelvin) {
+               plot_single_temp_text(gc,dive->duration.seconds,last_temperature);
+       }
+}
+
+static void plot_temperature_profile(struct dive *dive, struct graphics_context *gc)
+{
+       int i;
+       cairo_t *cr = gc->cr;
+       int last = 0;
+
+       if (!setup_temperature_limits(dive, gc))
+               return;
+
+       set_source_rgba(gc, 0.2, 0.2, 1.0, 0.8);
+       for (i = 0; i < dive->samples; i++) {
+               struct sample *sample = dive->sample+i;
+               if (sample->time.seconds > dive->duration.seconds)
+                       break; /* let's not plot surface temp events */
+               int mkelvin = sample->temperature.mkelvin;
+               if (!mkelvin) {
+                       if (!last)
+                               continue;
+                       mkelvin = last;
+               }
+               if (last)
+                       line_to(gc, sample->time.seconds, mkelvin);
+               else
+                       move_to(gc, sample->time.seconds, mkelvin);
+               last = mkelvin;
+       }
        cairo_stroke(cr);
 }
 
@@ -292,7 +411,7 @@ static int get_cylinder_pressure_range(struct dive *dive, struct graphics_contex
        int i;
        int min, max;
 
-       gc->scalex = round_seconds_up(dive->duration.seconds);
+       gc->leftx = 0; gc->rightx = round_seconds_up(dive->duration.seconds);
 
        max = 0;
        min = 5000000;
@@ -320,7 +439,7 @@ static int get_cylinder_pressure_range(struct dive *dive, struct graphics_contex
                endp->mbar = min;
        if (!max)
                return 0;
-       gc->scaley = max * 1.5;
+       gc->topy = 0; gc->bottomy = max * 1.5;
        return 1;
 }
 
@@ -382,7 +501,7 @@ static void plot_info(struct dive *dive, struct graphics_context *gc)
 {
        text_render_options_t tro = {10, 0.2, 1.0, 0.2, LEFT, TOP};
        const double liters_per_cuft = 28.317;
-       const char *unit;
+       const char *unit, *desc;
        double airuse;
 
        airuse = calculate_airuse(dive);
@@ -405,6 +524,15 @@ static void plot_info(struct dive *dive, struct graphics_context *gc)
                double sac = airuse / pressure * 60 / dive->duration.seconds;
                plot_text(gc, &tro, 0.8, 0.85, "SAC: %4.2f %s/min", sac, unit);
        }
+       desc = dive->cylinder[0].type.description;
+       if (desc || dive->cylinder[0].gasmix.o2.permille) {
+               int o2 = dive->cylinder[0].gasmix.o2.permille / 10;
+               if (!desc)
+                       desc = "";
+               if (!o2)
+                       o2 = 21;
+               plot_text(gc, &tro, 0.8, 0.9, "%s (%d%%)", desc, o2);
+       }
 }
 
 static void plot_cylinder_pressure_text(struct dive *dive, struct graphics_context *gc)
@@ -440,13 +568,116 @@ static void plot_cylinder_pressure_text(struct dive *dive, struct graphics_conte
        }
 }
 
-static void plot(struct graphics_context *gc, int w, int h, struct dive *dive)
+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 = 90*(index+1);
+       struct plot_data *min, *max;
+       int 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 = p;
+       avg = p->val;
+       nr = 1;
+       while (++p < last) {
+               int val = p->val;
+               if (p->sec > time + seconds)
+                       break;
+               avg += val;
+               nr ++;
+               if (val < min->val)
+                       min = p;
+               if (val > max->val)
+                       max = p;
+       }
+       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);
+}
+
+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;
        cairo_translate(gc->cr, topx, topy);
+       cairo_set_line_width(gc->cr, 2);
+       cairo_set_line_cap(gc->cr, CAIRO_LINE_CAP_ROUND);
+       cairo_set_line_join(gc->cr, CAIRO_LINE_JOIN_ROUND);
 
        /*
         * We can use "cairo_translate()" because that doesn't
@@ -458,22 +689,27 @@ static void plot(struct graphics_context *gc, int w, int h, struct dive *dive)
        gc->maxx = (w - 2*topx);
        gc->maxy = (h - 2*topy);
 
+       /* Temperature profile */
+       plot_temperature_profile(dive, gc);
+
        /* Cylinder pressure plot */
        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);
+       plot_temperature_text(dive, gc);
+       plot_depth_text(dive, gc, pi);
        plot_cylinder_pressure_text(dive, gc);
 
        /* And info box in the lower right corner.. */
-       gc->scalex = gc->scaley = 1.0;
+       gc->leftx = 0; gc->rightx = 1.0;
+       gc->topy = 0; gc->bottomy = 1.0;
        plot_info(dive, gc);
 
        /* Bounding box last */
-       cairo_set_source_rgb(gc->cr, 1, 1, 1);
+       set_source_rgb(gc, 1, 1, 1);
        move_to(gc, 0, 0);
        line_to(gc, 0, 1);
        line_to(gc, 1, 1);
@@ -486,14 +722,14 @@ static void plot(struct graphics_context *gc, int w, int h, struct dive *dive)
 static gboolean expose_event(GtkWidget *widget, GdkEventExpose *event, gpointer data)
 {
        struct dive *dive = current_dive;
-       struct graphics_context gc;
+       struct graphics_context gc = { .printer = 0 };
        int w,h;
 
        w = widget->allocation.width;
        h = widget->allocation.height;
 
        gc.cr = gdk_cairo_create(widget->window);
-       cairo_set_source_rgb(gc.cr, 0, 0, 0);
+       set_source_rgb(&gc, 0, 0, 0);
        cairo_paint(gc.cr);
 
        if (dive)
@@ -509,7 +745,7 @@ GtkWidget *dive_profile_widget(void)
        GtkWidget *da;
 
        da = gtk_drawing_area_new();
-       gtk_widget_set_size_request(da, 450, 350);
+       gtk_widget_set_size_request(da, 350, 250);
        g_signal_connect(da, "expose_event", G_CALLBACK(expose_event), NULL);
 
        return da;