Do cylinder pressure plot first, then depth, then text notes

[ext/subsurface.git] / profile.c

diff --git a/profile.c b/profile.c
index 070e6fa7054445e26031335c4c1b5e1ac35ccbff..f10e856c29fb19e48cb1c757e31f290bf131fe17 100644 (file)
--- a/profile.c
+++ b/profile.c
@@ -1,9 +1,11 @@
 #include <stdio.h>
 #include <stdlib.h>
+#include <stdarg.h>
 #include <time.h>
 
 #include "dive.h"
 #include "display.h"
+#include "divelist.h"
 
 int selected_dive = 0;
 
@@ -25,10 +27,117 @@ static int round_feet_up(int feet)
        return MAX(90, ROUND_UP(feet+5, 15));
 }
 
+static void plot_text(cairo_t *cr, double x, double y, const char *fmt, ...)
+{
+       cairo_text_extents_t extents;
+       char buffer[80];
+       va_list args;
+
+       va_start(args, fmt);
+       vsnprintf(buffer, sizeof(buffer), fmt, args);
+       va_end(args);
+
+       cairo_text_extents(cr, buffer, &extents);
+
+       x -= extents.width/2 + extents.x_bearing;
+       y += extents.height * 1.2;
+
+       cairo_move_to(cr, x, y);
+       cairo_text_path(cr, buffer);
+       cairo_set_source_rgb(cr, 0, 0, 0);
+       cairo_stroke(cr);
+
+       cairo_move_to(cr, x, y);
+       cairo_set_source_rgb(cr, 1, 0, 0);
+       cairo_show_text(cr, buffer);
+}
+
+/*
+ * Find the next maximum point in a 10-minute window.
+ *
+ * We exit early if we hit "enough" of a depth reversal,
+ * which is roughly 10 feet.
+ */
+static int next_minmax(struct dive *dive, int index, int minmax)
+{
+       const int enough = 3000;
+       int timelimit, depthlimit, result;
+       struct sample *sample = dive->sample + index;
+
+       if (index >= dive->samples)
+               return 0;
+
+       timelimit = 24*60*60;
+       depthlimit = sample->depth.mm;
+       result = 0;
+
+       for (;;) {
+               int time, depth;
+
+               index++;
+               sample++;
+               if (index >= dive->samples)
+                       break;
+               time = sample->time.seconds;
+               depth = sample->depth.mm;
+               if (time > timelimit)
+                       break;
+
+               if (minmax) {
+                       if (depth <= depthlimit) {
+                               if (depthlimit - depth > enough)
+                                       break;
+                               continue;
+                       }
+               } else {
+                       if (depth >= depthlimit) {
+                               if (depth - depthlimit > enough)
+                                       break;
+                               continue;
+                       }
+               }
+
+               result = index;
+               depthlimit = depth;
+               /* Look up to ten minutes into the future */
+               timelimit = time + 600;
+       }
+       return result;
+}
+
 /* Scale to 0,0 -> maxx,maxy */
 #define SCALE(x,y) (x)*maxx/scalex+topx,(y)*maxy/scaley+topy
 
-static void plot_profile(struct dive *dive, cairo_t *cr,
+static void plot_depth_text(struct dive *dive, cairo_t *cr,
+       double topx, double topy, double maxx, double maxy)
+{
+       double scalex, scaley;
+       int maxtime, maxdepth;
+       int i;
+
+       /* Get plot scaling limits */
+       maxtime = round_seconds_up(dive->duration.seconds);
+       maxdepth = round_feet_up(to_feet(dive->maxdepth));
+
+       scalex = maxtime;
+       scaley = maxdepth;
+
+       cairo_set_font_size(cr, 14);
+       cairo_set_source_rgb(cr, 1, 0.2, 0.2);
+       i = 0;
+       while ((i = next_minmax(dive, i, 1)) != 0) {
+               struct sample *sample = dive->sample+i;
+               int sec = sample->time.seconds;
+               int depth = to_feet(sample->depth);
+
+               plot_text(cr, SCALE(sec, depth), "%d ft", depth);
+               i = next_minmax(dive, i, 0);
+               if (!i)
+                       break;
+       }
+}
+
+static void plot_depth_profile(struct dive *dive, cairo_t *cr,
        double topx, double topy, double maxx, double maxy)
 {
        double scalex, scaley;
@@ -106,6 +215,9 @@ static int get_cylinder_pressure_range(struct dive *dive, double *scalex, double
                struct sample *sample = dive->sample + i;
                double bar;
 
+               /* FIXME! We only track cylinder 0 right now */
+               if (sample->cylinderindex)
+                       continue;
                if (!sample->cylinderpressure.mbar)
                        continue;
                bar = sample->cylinderpressure.mbar;
@@ -123,7 +235,7 @@ static int get_cylinder_pressure_range(struct dive *dive, double *scalex, double
 static void plot_cylinder_pressure(struct dive *dive, cairo_t *cr,
        double topx, double topy, double maxx, double maxy)
 {
-       int i;
+       int i, sec = -1;
        double scalex, scaley;
 
        if (!get_cylinder_pressure_range(dive, &scalex, &scaley))
@@ -131,18 +243,23 @@ static void plot_cylinder_pressure(struct dive *dive, cairo_t *cr,
 
        cairo_set_source_rgba(cr, 0.2, 1.0, 0.2, 0.80);
 
-       cairo_move_to(cr, SCALE(0, dive->beginning_pressure.mbar));
+       cairo_move_to(cr, SCALE(0, dive->cylinder[0].start.mbar));
        for (i = 1; i < dive->samples; i++) {
-               int sec, mbar;
+               int mbar;
                struct sample *sample = dive->sample + i;
 
-               sec = sample->time.seconds;
                mbar = sample->cylinderpressure.mbar;
                if (!mbar)
                        continue;
+               sec = sample->time.seconds;
                cairo_line_to(cr, SCALE(sec, mbar));
        }
-       cairo_line_to(cr, SCALE(dive->duration.seconds, dive->end_pressure.mbar));
+       /*
+        * We may have "surface time" events, in which case we don't go
+        * back to dive duration
+        */
+       if (sec < dive->duration.seconds)
+               cairo_line_to(cr, SCALE(dive->duration.seconds, dive->cylinder[0].end.mbar));
        cairo_stroke(cr);
 }
 
@@ -156,11 +273,14 @@ static void plot(cairo_t *cr, int w, int h, struct dive *dive)
        maxx = (w - 2*topx);
        maxy = (h - 2*topy);
 
+       /* Cylinder pressure plot */
+       plot_cylinder_pressure(dive, cr, topx, topy, maxx, maxy);
+
        /* Depth profile */
-       plot_profile(dive, cr, topx, topy, maxx, maxy);
+       plot_depth_profile(dive, cr, topx, topy, maxx, maxy);
 
-       /* Cylinder pressure plot? */
-       plot_cylinder_pressure(dive, cr, topx, topy, maxx, maxy);
+       /* Text on top of all graphs.. */
+       plot_depth_text(dive, cr, topx, topy, maxx, maxy);
 
        /* Bounding box last */
        scalex = scaley = 1.0;