]> git.tdb.fi Git - ext/subsurface.git/blobdiff - profile.c
Plot pressure data based on 'struct plot_info' rather than raw dive data
[ext/subsurface.git] / profile.c
index 796b318e71308a0166d28ac0190908b0398a0a2f..7b2f31fe5eb1974d8909bd53a8121a47bc44d68c 100644 (file)
--- a/profile.c
+++ b/profile.c
@@ -14,8 +14,13 @@ int selected_dive = 0;
 struct plot_info {
        int nr;
        int maxtime;
+       int meandepth, maxdepth;
+       int minpressure, maxpressure;
+       int mintemp, maxtemp;
        struct plot_data {
                int sec;
+               int pressure, temperature;
+               /* Depth info */
                int val;
                int smoothed;
                struct plot_data *min[3];
@@ -40,6 +45,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))
 
 /*
@@ -48,14 +71,15 @@ static void line_to(struct graphics_context *gc, double x, double y)
  * 30 minutes or 90 ft, just so that small dives show
  * up as such.
  */
-static int round_seconds_up(int seconds)
+static int get_maxtime(struct plot_info *pi)
 {
+       int seconds = pi->maxtime;
        return MAX(30*60, ROUND_UP(seconds, 60*10));
 }
 
-static int round_depth_up(depth_t depth)
+static int get_maxdepth(struct plot_info *pi)
 {
-       unsigned mm = depth.mm;
+       unsigned mm = pi->maxdepth;
        /* Minimum 30m */
        return MAX(30000, ROUND_UP(mm+3000, 10000));
 }
@@ -107,13 +131,13 @@ static void plot_text(struct graphics_context *gc, const text_render_options_t *
        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);
 }
 
@@ -157,13 +181,13 @@ static void plot_text_samples(struct graphics_context *gc, struct plot_info *pi)
        }
 }
 
-static void plot_depth_text(struct dive *dive, struct graphics_context *gc, struct plot_info *pi)
+static void plot_depth_text(struct graphics_context *gc, struct plot_info *pi)
 {
        int maxtime, maxdepth;
 
        /* Get plot scaling limits */
-       maxtime = round_seconds_up(dive->duration.seconds);
-       maxdepth = round_depth_up(dive->maxdepth);
+       maxtime = get_maxtime(pi);
+       maxdepth = get_maxdepth(pi);
 
        gc->leftx = 0; gc->rightx = maxtime;
        gc->topy = 0; gc->bottomy = maxdepth;
@@ -207,12 +231,14 @@ static void plot_minmax_profile_minute(struct graphics_context *gc, struct plot_
 
 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)
+static void plot_depth_profile(struct graphics_context *gc, struct plot_info *pi)
 {
        int i;
        cairo_t *cr = gc->cr;
@@ -221,8 +247,8 @@ static void plot_depth_profile(struct dive *dive, struct graphics_context *gc, s
        int maxtime, maxdepth, marker;
 
        /* Get plot scaling limits */
-       maxtime = round_seconds_up(dive->duration.seconds);
-       maxdepth = round_depth_up(dive->maxdepth);
+       maxtime = get_maxtime(pi);
+       maxdepth = get_maxdepth(pi);
 
        /* Time markers: every 5 min */
        gc->leftx = 0; gc->rightx = maxtime;
@@ -240,7 +266,7 @@ static void plot_depth_profile(struct dive *dive, struct graphics_context *gc, s
        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);
@@ -248,9 +274,9 @@ static void plot_depth_profile(struct dive *dive, struct graphics_context *gc, s
        cairo_stroke(cr);
 
        /* Show mean depth */
-       cairo_set_source_rgba(cr, 1, 0.2, 0.2, 0.40);
-       move_to(gc, 0, dive->meandepth.mm);
-       line_to(gc, 1, dive->meandepth.mm);
+       set_source_rgba(gc, 1, 0.2, 0.2, 0.40);
+       move_to(gc, 0, pi->meandepth);
+       line_to(gc, 1, pi->meandepth);
        cairo_stroke(cr);
 
        gc->leftx = 0; gc->rightx = maxtime;
@@ -259,7 +285,7 @@ static void plot_depth_profile(struct dive *dive, struct graphics_context *gc, s
        plot_minmax_profile(gc, pi);
 
        entry = pi->entry;
-       cairo_set_source_rgba(cr, 1, 0.2, 0.2, 0.80);
+       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++) {
@@ -274,31 +300,20 @@ static void plot_depth_profile(struct dive *dive, struct graphics_context *gc, s
        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)
+static int setup_temperature_limits(struct graphics_context *gc, struct plot_info *pi)
 {
-       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;
-       }
+       maxtime = get_maxtime(pi);
+       mintemp = pi->mintemp;
+       maxtemp = pi->maxtemp;
 
        gc->leftx = 0; gc->rightx = maxtime;
        /* Show temperatures in roughly the lower third */
@@ -308,50 +323,67 @@ static int setup_temperature_limits(struct dive *dive, struct graphics_context *
        return maxtemp > mintemp;
 }
 
-static void plot_temperature_text(struct dive *dive, struct graphics_context *gc)
+static void plot_single_temp_text(struct graphics_context *gc, int sec, int mkelvin)
 {
-       int i;
+       int deg;
+       const char *unit;
        static const text_render_options_t tro = {12, 0.2, 0.2, 1.0, LEFT, TOP};
+       temperature_t temperature = { mkelvin };
+
+       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 graphics_context *gc, struct plot_info *pi)
+{
+       int i;
        int last = 0;
+       int last_temperature = 0, last_printed_temp = 0;
 
-       if (!setup_temperature_limits(dive, gc))
+       if (!setup_temperature_limits(gc, pi))
                return;
 
-       for (i = 0; i < dive->samples; i++) {
-               const char *unit;
-               struct sample *sample = dive->sample+i;
-               int mkelvin = sample->temperature.mkelvin;
-               int sec, deg;
+       for (i = 0; i < pi->nr; i++) {
+               int sec;
+               struct plot_data *entry = pi->entry+i;
+               int mkelvin = entry->temperature;
+
                if (!mkelvin)
                        continue;
-               sec = sample->time.seconds;
-               if (sec < last)
+               last_temperature = mkelvin;
+               sec = entry->sec;
+               if (sec < last + 300)
                        continue;
-               last = sec + 300;
-               if (output_units.temperature == FAHRENHEIT) {
-                       deg = to_F(sample->temperature);
-                       unit = "F";
-               } else {
-                       deg = to_C(sample->temperature);
-                       unit = "C";
-               }
-               plot_text(gc, &tro, sec, mkelvin, "%d %s", deg, unit);
+               last = sec;
+               plot_single_temp_text(gc,sec,mkelvin);
+               last_printed_temp = mkelvin;
        }
+       /* it would be nice to print the end temperature, if it's different */
+       if (last_temperature != last_printed_temp)
+               plot_single_temp_text(gc, last, last_temperature);
 }
 
-static void plot_temperature_profile(struct dive *dive, struct graphics_context *gc)
+static void plot_temperature_profile(struct dive *dive, struct graphics_context *gc,
+       struct plot_info *pi)
 {
        int i;
        cairo_t *cr = gc->cr;
        int last = 0;
 
-       if (!setup_temperature_limits(dive, gc))
+       if (!setup_temperature_limits(gc, pi))
                return;
 
-       cairo_set_source_rgba(cr, 0.2, 0.2, 1.0, 0.8);
+       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)
@@ -368,49 +400,20 @@ static void plot_temperature_profile(struct dive *dive, struct graphics_context
 }
 
 /* gets both the actual start and end pressure as well as the scaling factors */
-static int get_cylinder_pressure_range(struct dive *dive, struct graphics_context *gc,
-       pressure_t *startp, pressure_t *endp)
+static int get_cylinder_pressure_range(struct graphics_context *gc, struct plot_info *pi)
 {
-       int i;
-       int min, max;
-
-       gc->leftx = 0; gc->rightx = round_seconds_up(dive->duration.seconds);
-
-       max = 0;
-       min = 5000000;
-       if (startp)
-               startp->mbar = endp->mbar = 0;
-
-       for (i = 0; i < dive->samples; i++) {
-               int mbar;
-               struct sample *sample = dive->sample + i;
+       gc->leftx = 0;
+       gc->rightx = get_maxtime(pi);
 
-               /* FIXME! We only track cylinder 0 right now */
-               if (sample->cylinderindex)
-                       continue;
-               mbar = sample->cylinderpressure.mbar;
-               if (!mbar)
-                       continue;
-               if (mbar < min)
-                       min = mbar;
-               if (mbar > max)
-                       max = mbar;
-       }
-       if (startp)
-               startp->mbar = max;
-       if (endp)
-               endp->mbar = min;
-       if (!max)
-               return 0;
-       gc->topy = 0; gc->bottomy = max * 1.5;
-       return 1;
+       gc->topy = 0; gc->bottomy = pi->maxpressure * 1.5;
+       return pi->maxpressure != 0;
 }
 
-static void plot_cylinder_pressure(struct dive *dive, struct graphics_context *gc)
+static void plot_cylinder_pressure(struct dive *dive, struct graphics_context *gc, struct plot_info *pi)
 {
        int i, sec = -1;
 
-       if (!get_cylinder_pressure_range(dive, gc, NULL, NULL))
+       if (!get_cylinder_pressure_range(gc, pi))
                return;
 
        cairo_set_source_rgba(gc->cr, 0.2, 1.0, 0.2, 0.80);
@@ -498,35 +501,39 @@ static void plot_info(struct dive *dive, struct graphics_context *gc)
        }
 }
 
-static void plot_cylinder_pressure_text(struct dive *dive, struct graphics_context *gc)
+static int mbar_to_PSI(int mbar)
 {
-       pressure_t startp, endp;
+       pressure_t p = {mbar};
+       return to_PSI(p);
+}
 
-       if (get_cylinder_pressure_range(dive, gc, &startp, &endp)) {
+static void plot_cylinder_pressure_text(struct graphics_context *gc, struct plot_info *pi)
+{
+       if (get_cylinder_pressure_range(gc, pi)) {
                int start, end;
                const char *unit = "bar";
 
                switch (output_units.pressure) {
                case PASCAL:
-                       start = startp.mbar * 100;
-                       end = startp.mbar * 100;
+                       start = pi->maxpressure * 100;
+                       end = pi->minpressure * 100;
                        unit = "pascal";
                        break;
                case BAR:
-                       start = (startp.mbar + 500) / 1000;
-                       end = (endp.mbar + 500) / 1000;
+                       start = (pi->maxpressure + 500) / 1000;
+                       end = (pi->minpressure + 500) / 1000;
                        unit = "bar";
                        break;
                case PSI:
-                       start = to_PSI(startp);
-                       end = to_PSI(endp);
+                       start = mbar_to_PSI(pi->maxpressure);
+                       end = mbar_to_PSI(pi->minpressure);
                        unit = "psi";
                        break;
                }
 
                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,
+               plot_text(gc, &tro, 0, pi->maxpressure, "%d %s", start, unit);
+               plot_text(gc, &tro, pi->maxtime, pi->minpressure,
                          "%d %s", end, unit);
        }
 }
@@ -578,6 +585,27 @@ static struct plot_info *analyze_plot_info(struct plot_info *pi)
        int i;
        int nr = pi->nr;
 
+       /* Do pressure min/max based on the non-surface data */
+       for (i = 0; i < nr; i++) {
+               struct plot_data *entry = pi->entry+i;
+               int pressure = entry->pressure;
+               int temperature = entry->temperature;
+
+               if (pressure) {
+                       if (!pi->minpressure || pressure < pi->minpressure)
+                               pi->minpressure = pressure;
+                       if (pressure > pi->maxpressure)
+                               pi->maxpressure = pressure;
+               }
+
+               if (temperature) {
+                       if (!pi->mintemp || temperature < pi->mintemp)
+                               pi->mintemp = temperature;
+                       if (temperature > pi->maxtemp)
+                               pi->maxtemp = temperature;
+               }
+       }
+
        /* Smoothing function: 5-point triangular smooth */
        for (i = 2; i < nr-2; i++) {
                struct plot_data *entry = pi->entry+i;
@@ -603,8 +631,9 @@ static struct plot_info *analyze_plot_info(struct plot_info *pi)
  * sides, so that you can do end-points without having to worry
  * about it.
  */
-static struct plot_info *depth_plot_info(struct dive *dive)
+static struct plot_info *create_plot_info(struct dive *dive)
 {
+       int lastdepth, lastindex;
        int i, nr = dive->samples + 4, sec;
        size_t alloc_size = plot_info_size(nr);
        struct plot_info *pi;
@@ -615,25 +644,47 @@ static struct plot_info *depth_plot_info(struct dive *dive)
        memset(pi, 0, alloc_size);
        pi->nr = nr;
        sec = 0;
+       lastindex = 0;
+       lastdepth = -1;
        for (i = 0; i < dive->samples; i++) {
+               int depth;
                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;
+               depth = entry->val = sample->depth.mm;
+               entry->pressure = sample->cylinderpressure.mbar;
+               entry->temperature = sample->temperature.mkelvin;
+
+               if (depth || lastdepth)
+                       lastindex = i+2;
+
+               lastdepth = depth;
+               if (depth > pi->maxdepth)
+                       pi->maxdepth = depth;
        }
+       if (lastdepth)
+               lastindex = i + 2;
        /* 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;
 
+       pi->nr = lastindex+1;
+       pi->maxtime = pi->entry[lastindex].sec;
+
+       pi->minpressure = dive->cylinder[0].end.mbar;
+       pi->maxpressure = dive->cylinder[0].start.mbar;
+
+       pi->meandepth = dive->meandepth.mm;
+
        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);
+       struct plot_info *pi = create_plot_info(dive);
 
        topx = w / 20.0;
        topy = h / 20.0;
@@ -653,18 +704,18 @@ void plot(struct graphics_context *gc, int w, int h, struct dive *dive)
        gc->maxy = (h - 2*topy);
 
        /* Temperature profile */
-       plot_temperature_profile(dive, gc);
+       plot_temperature_profile(dive, gc, pi);
 
        /* Cylinder pressure plot */
-       plot_cylinder_pressure(dive, gc);
+       plot_cylinder_pressure(dive, gc, pi);
 
        /* Depth profile */
-       plot_depth_profile(dive, gc, pi);
+       plot_depth_profile(gc, pi);
 
        /* Text on top of all graphs.. */
-       plot_temperature_text(dive, gc);
-       plot_depth_text(dive, gc, pi);
-       plot_cylinder_pressure_text(dive, gc);
+       plot_temperature_text(gc, pi);
+       plot_depth_text(gc, pi);
+       plot_cylinder_pressure_text(gc, pi);
 
        /* And info box in the lower right corner.. */
        gc->leftx = 0; gc->rightx = 1.0;
@@ -672,7 +723,7 @@ void plot(struct graphics_context *gc, int w, int h, struct dive *dive)
        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);
@@ -685,14 +736,14 @@ 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)