divelist.c

   1 /* divelist.c */
   2 /* this creates the UI for the dive list -
   3  * controlled through the following interfaces:
   4  *
   5  * void flush_divelist(struct dive *dive)
   6  * GtkWidget dive_list_create(void)
   7  * void dive_list_update_dives(void)
   8  * void update_dive_list_units(void)
   9  * void set_divelist_font(const char *font)
  10  * void mark_divelist_changed(int changed)
  11  * int unsaved_changes()
  12  */
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <string.h>
  16 #include <time.h>
  17 #include <math.h>
  18
  19 #include "divelist.h"
  20 #include "dive.h"
  21 #include "display.h"
  22 #include "display-gtk.h"
  23
  24 struct DiveList {
  25         GtkWidget    *tree_view;
  26         GtkWidget    *container_widget;
  27         GtkListStore *model;
  28         GtkTreeViewColumn *nr, *date, *depth, *duration, *location;
  29         GtkTreeViewColumn *temperature, *cylinder, *nitrox, *sac, *otu;
  30         int changed;
  31 };
  32
  33 static struct DiveList dive_list;
  34
  35 /*
  36  * The dive list has the dive data in both string format (for showing)
  37  * and in "raw" format (for sorting purposes)
  38  */
  39 enum {
  40         DIVE_INDEX = 0,
  41         DIVE_NR,                /* int: dive->nr */
  42         DIVE_DATE,              /* time_t: dive->when */
  43         DIVE_DEPTH,             /* int: dive->maxdepth in mm */
  44         DIVE_DURATION,          /* int: in seconds */
  45         DIVE_TEMPERATURE,       /* int: in mkelvin */
  46         DIVE_CYLINDER,
  47         DIVE_NITROX,            /* int: in permille */
  48         DIVE_SAC,               /* int: in ml/min */
  49         DIVE_OTU,               /* int: in OTUs */
  50         DIVE_LOCATION,          /* "2nd Cathedral, Lanai" */
  51         DIVELIST_COLUMNS
  52 };
  53
  54 static GList *selected_dives;
  55
  56 static void selection_cb(GtkTreeSelection *selection, GtkTreeModel *model)
  57 {
  58         GtkTreeIter iter;
  59         GValue value = {0, };
  60         GtkTreePath *path;
  61
  62         int nr_selected = gtk_tree_selection_count_selected_rows(selection);
  63
  64         if (selected_dives) {
  65                 g_list_foreach (selected_dives, (GFunc) gtk_tree_path_free, NULL);
  66                 g_list_free (selected_dives);
  67         }
  68         selected_dives = gtk_tree_selection_get_selected_rows(selection, NULL);
  69
  70         switch (nr_selected) {
  71         case 0: /* keep showing the last selected dive */
  72                 return;
  73         case 1:
  74                 /* just pick that dive as selected */
  75                 path = g_list_nth_data(selected_dives, 0);
  76                 if (gtk_tree_model_get_iter(model, &iter, path)) {
  77                         gtk_tree_model_get_value(model, &iter, DIVE_INDEX, &value);
  78                         selected_dive = g_value_get_int(&value);
  79                         repaint_dive();
  80                 }
  81                 return;
  82         default: /* multiple selections - what now? At this point I
  83                   * don't want to change the selected dive unless
  84                   * there is exactly one dive selected; not sure this
  85                   * is the most intuitive solution.
  86                   * I do however want to keep around which dives have
  87                   * been selected */
  88                 return;
  89         }
  90 }
  91
  92 static void date_data_func(GtkTreeViewColumn *col,
  93                            GtkCellRenderer *renderer,
  94                            GtkTreeModel *model,
  95                            GtkTreeIter *iter,
  96                            gpointer data)
  97 {
  98         int val;
  99         struct tm *tm;
 100         time_t when;
 101         char buffer[40];
 102
 103         gtk_tree_model_get(model, iter, DIVE_DATE, &val, -1);
 104
 105         /* 2038 problem */
 106         when = val;
 107
 108         tm = gmtime(&when);
 109         snprintf(buffer, sizeof(buffer),
 110                 "%s, %s %d, %d %02d:%02d",
 111                 weekday(tm->tm_wday),
 112                 monthname(tm->tm_mon),
 113                 tm->tm_mday, tm->tm_year + 1900,
 114                 tm->tm_hour, tm->tm_min);
 115         g_object_set(renderer, "text", buffer, NULL);
 116 }
 117
 118 static void depth_data_func(GtkTreeViewColumn *col,
 119                             GtkCellRenderer *renderer,
 120                             GtkTreeModel *model,
 121                             GtkTreeIter *iter,
 122                             gpointer data)
 123 {
 124         int depth, integer, frac, len;
 125         char buffer[40];
 126
 127         gtk_tree_model_get(model, iter, DIVE_DEPTH, &depth, -1);
 128
 129         switch (output_units.length) {
 130         case METERS:
 131                 /* To tenths of meters */
 132                 depth = (depth + 49) / 100;
 133                 integer = depth / 10;
 134                 frac = depth % 10;
 135                 if (integer < 20)
 136                         break;
 137                 frac = -1;
 138                 /* Rounding? */
 139                 break;
 140         case FEET:
 141                 integer = mm_to_feet(depth) + 0.5;
 142                 frac = -1;
 143                 break;
 144         default:
 145                 return;
 146         }
 147         len = snprintf(buffer, sizeof(buffer), "%d", integer);
 148         if (frac >= 0)
 149                 len += snprintf(buffer+len, sizeof(buffer)-len, ".%d", frac);
 150
 151         g_object_set(renderer, "text", buffer, NULL);
 152 }
 153
 154 static void duration_data_func(GtkTreeViewColumn *col,
 155                                GtkCellRenderer *renderer,
 156                                GtkTreeModel *model,
 157                                GtkTreeIter *iter,
 158                                gpointer data)
 159 {
 160         unsigned int sec;
 161         char buffer[16];
 162
 163         gtk_tree_model_get(model, iter, DIVE_DURATION, &sec, -1);
 164         snprintf(buffer, sizeof(buffer), "%d:%02d", sec / 60, sec % 60);
 165
 166         g_object_set(renderer, "text", buffer, NULL);
 167 }
 168
 169 static void temperature_data_func(GtkTreeViewColumn *col,
 170                                   GtkCellRenderer *renderer,
 171                                   GtkTreeModel *model,
 172                                   GtkTreeIter *iter,
 173                                   gpointer data)
 174 {
 175         int value;
 176         char buffer[80];
 177
 178         gtk_tree_model_get(model, iter, DIVE_TEMPERATURE, &value, -1);
 179
 180         *buffer = 0;
 181         if (value) {
 182                 double deg;
 183                 switch (output_units.temperature) {
 184                 case CELSIUS:
 185                         deg = mkelvin_to_C(value);
 186                         break;
 187                 case FAHRENHEIT:
 188                         deg = mkelvin_to_F(value);
 189                         break;
 190                 default:
 191                         return;
 192                 }
 193                 snprintf(buffer, sizeof(buffer), "%.1f", deg);
 194         }
 195
 196         g_object_set(renderer, "text", buffer, NULL);
 197 }
 198
 199 static void nitrox_data_func(GtkTreeViewColumn *col,
 200                              GtkCellRenderer *renderer,
 201                              GtkTreeModel *model,
 202                              GtkTreeIter *iter,
 203                              gpointer data)
 204 {
 205         int value;
 206         char buffer[80];
 207
 208         gtk_tree_model_get(model, iter, DIVE_NITROX, &value, -1);
 209
 210         if (value)
 211                 snprintf(buffer, sizeof(buffer), "%.1f", value/10.0);
 212         else
 213                 strcpy(buffer, "air");
 214
 215         g_object_set(renderer, "text", buffer, NULL);
 216 }
 217
 218 /* Render the SAC data (integer value of "ml / min") */
 219 static void sac_data_func(GtkTreeViewColumn *col,
 220                           GtkCellRenderer *renderer,
 221                           GtkTreeModel *model,
 222                           GtkTreeIter *iter,
 223                           gpointer data)
 224 {
 225         int value;
 226         const char *fmt;
 227         char buffer[16];
 228         double sac;
 229
 230         gtk_tree_model_get(model, iter, DIVE_SAC, &value, -1);
 231
 232         if (!value) {
 233                 g_object_set(renderer, "text", "", NULL);
 234                 return;
 235         }
 236
 237         sac = value / 1000.0;
 238         switch (output_units.volume) {
 239         case LITER:
 240                 fmt = "%4.1f";
 241                 break;
 242         case CUFT:
 243                 fmt = "%4.2f";
 244                 sac = ml_to_cuft(sac * 1000);
 245                 break;
 246         }
 247         snprintf(buffer, sizeof(buffer), fmt, sac);
 248
 249         g_object_set(renderer, "text", buffer, NULL);
 250 }
 251
 252 /* Render the OTU data (integer value of "OTU") */
 253 static void otu_data_func(GtkTreeViewColumn *col,
 254                           GtkCellRenderer *renderer,
 255                           GtkTreeModel *model,
 256                           GtkTreeIter *iter,
 257                           gpointer data)
 258 {
 259         int value;
 260         char buffer[16];
 261
 262         gtk_tree_model_get(model, iter, DIVE_OTU, &value, -1);
 263
 264         if (!value) {
 265                 g_object_set(renderer, "text", "", NULL);
 266                 return;
 267         }
 268
 269         snprintf(buffer, sizeof(buffer), "%d", value);
 270
 271         g_object_set(renderer, "text", buffer, NULL);
 272 }
 273
 274 /* calculate OTU for a dive */
 275 static int calculate_otu(struct dive *dive)
 276 {
 277         int i;
 278         double otu = 0.0;
 279
 280         for (i = 1; i < dive->samples; i++) {
 281                 int t;
 282                 double po2;
 283                 struct sample *sample = dive->sample + i;
 284                 struct sample *psample = sample - 1;
 285                 t = sample->time.seconds - psample->time.seconds;
 286                 po2 = dive->cylinder[sample->cylinderindex].gasmix.o2.permille / 1000.0 *
 287                         (sample->depth.mm + 10000) / 10000.0;
 288                 if (po2 >= 0.5)
 289                         otu += pow(po2 - 0.5, 0.83) * t / 30.0;
 290         }
 291         return otu + 0.5;
 292 }
 293 /*
 294  * Return air usage (in liters).
 295  */
 296 static double calculate_airuse(struct dive *dive)
 297 {
 298         double airuse = 0;
 299         int i;
 300
 301         for (i = 0; i < MAX_CYLINDERS; i++) {
 302                 pressure_t start, end;
 303                 cylinder_t *cyl = dive->cylinder + i;
 304                 int size = cyl->type.size.mliter;
 305                 double kilo_atm;
 306
 307                 if (!size)
 308                         continue;
 309
 310                 start = cyl->start.mbar ? cyl->start : cyl->sample_start;
 311                 end = cyl->end.mbar ? cyl->end : cyl->sample_end;
 312                 kilo_atm = (to_ATM(start) - to_ATM(end)) / 1000.0;
 313
 314                 /* Liters of air at 1 atm == milliliters at 1k atm*/
 315                 airuse += kilo_atm * size;
 316         }
 317         return airuse;
 318 }
 319
 320 static int calculate_sac(struct dive *dive)
 321 {
 322         double airuse, pressure, sac;
 323
 324         airuse = calculate_airuse(dive);
 325         if (!airuse)
 326                 return 0;
 327         if (!dive->duration.seconds)
 328                 return 0;
 329
 330         /* Mean pressure in atm: 1 atm per 10m */
 331         pressure = 1 + (dive->meandepth.mm / 10000.0);
 332         sac = airuse / pressure * 60 / dive->duration.seconds;
 333
 334         /* milliliters per minute.. */
 335         return sac * 1000;
 336 }
 337
 338 void update_cylinder_related_info(struct dive *dive)
 339 {
 340         if (dive != NULL) {
 341                 dive->sac = calculate_sac(dive);
 342                 dive->otu = calculate_otu(dive);
 343         }
 344 }
 345
 346 static void get_string(char **str, const char *s)
 347 {
 348         int len;
 349         char *n;
 350
 351         if (!s)
 352                 s = "";
 353         len = strlen(s);
 354         if (len > 60)
 355                 len = 60;
 356         n = malloc(len+1);
 357         memcpy(n, s, len);
 358         n[len] = 0;
 359         *str = n;
 360 }
 361
 362 static void get_location(struct dive *dive, char **str)
 363 {
 364         get_string(str, dive->location);
 365 }
 366
 367 static void get_cylinder(struct dive *dive, char **str)
 368 {
 369         get_string(str, dive->cylinder[0].type.description);
 370 }
 371
 372 static void fill_one_dive(struct dive *dive,
 373                           GtkTreeModel *model,
 374                           GtkTreeIter *iter)
 375 {
 376         char *location, *cylinder;
 377
 378         get_cylinder(dive, &cylinder);
 379         get_location(dive, &location);
 380
 381         /*
 382          * We only set the fields that changed: the strings.
 383          * The core data itself is unaffected by units
 384          */
 385         gtk_list_store_set(GTK_LIST_STORE(model), iter,
 386                 DIVE_NR, dive->number,
 387                 DIVE_LOCATION, location,
 388                 DIVE_CYLINDER, cylinder,
 389                 DIVE_SAC, dive->sac,
 390                 DIVE_OTU, dive->otu,
 391                 -1);
 392 }
 393
 394 static gboolean set_one_dive(GtkTreeModel *model,
 395                              GtkTreePath *path,
 396                              GtkTreeIter *iter,
 397                              gpointer data)
 398 {
 399         GValue value = {0, };
 400         struct dive *dive;
 401
 402         /* Get the dive number */
 403         gtk_tree_model_get_value(model, iter, DIVE_INDEX, &value);
 404         dive = get_dive(g_value_get_int(&value));
 405         if (!dive)
 406                 return TRUE;
 407         if (data && dive != data)
 408                 return FALSE;
 409
 410         fill_one_dive(dive, model, iter);
 411         return dive == data;
 412 }
 413
 414 void flush_divelist(struct dive *dive)
 415 {
 416         GtkTreeModel *model = GTK_TREE_MODEL(dive_list.model);
 417
 418         gtk_tree_model_foreach(model, set_one_dive, dive);
 419 }
 420
 421 void set_divelist_font(const char *font)
 422 {
 423         PangoFontDescription *font_desc = pango_font_description_from_string(font);
 424         gtk_widget_modify_font(dive_list.tree_view, font_desc);
 425         pango_font_description_free(font_desc);
 426 }
 427
 428 void update_dive_list_units(void)
 429 {
 430         const char *unit;
 431         GtkTreeModel *model = GTK_TREE_MODEL(dive_list.model);
 432
 433         (void) get_depth_units(0, NULL, &unit);
 434         gtk_tree_view_column_set_title(dive_list.depth, unit);
 435
 436         (void) get_temp_units(0, &unit);
 437         gtk_tree_view_column_set_title(dive_list.temperature, unit);
 438
 439         gtk_tree_model_foreach(model, set_one_dive, NULL);
 440 }
 441
 442 void update_dive_list_col_visibility(void)
 443 {
 444         gtk_tree_view_column_set_visible(dive_list.cylinder, visible_cols.cylinder);
 445         gtk_tree_view_column_set_visible(dive_list.temperature, visible_cols.temperature);
 446         gtk_tree_view_column_set_visible(dive_list.nitrox, visible_cols.nitrox);
 447         gtk_tree_view_column_set_visible(dive_list.sac, visible_cols.sac);
 448         gtk_tree_view_column_set_visible(dive_list.otu, visible_cols.otu);
 449         return;
 450 }
 451
 452 static void fill_dive_list(void)
 453 {
 454         int i;
 455         GtkTreeIter iter;
 456         GtkListStore *store;
 457
 458         store = GTK_LIST_STORE(dive_list.model);
 459
 460         i = dive_table.nr;
 461         while (--i >= 0) {
 462                 struct dive *dive = dive_table.dives[i];
 463
 464                 update_cylinder_related_info(dive);
 465                 gtk_list_store_append(store, &iter);
 466                 gtk_list_store_set(store, &iter,
 467                         DIVE_INDEX, i,
 468                         DIVE_NR, dive->number,
 469                         DIVE_DATE, dive->when,
 470                         DIVE_DEPTH, dive->maxdepth,
 471                         DIVE_DURATION, dive->duration.seconds,
 472                         DIVE_LOCATION, "location",
 473                         DIVE_TEMPERATURE, dive->watertemp.mkelvin,
 474                         DIVE_NITROX, dive->cylinder[0].gasmix.o2,
 475                         DIVE_SAC, 0,
 476                         -1);
 477         }
 478
 479         update_dive_list_units();
 480         if (gtk_tree_model_get_iter_first(GTK_TREE_MODEL(dive_list.model), &iter)) {
 481                 GtkTreeSelection *selection;
 482                 selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(dive_list.tree_view));
 483                 gtk_tree_selection_select_iter(selection, &iter);
 484         }
 485 }
 486
 487 void dive_list_update_dives(void)
 488 {
 489         gtk_list_store_clear(GTK_LIST_STORE(dive_list.model));
 490         fill_dive_list();
 491         repaint_dive();
 492 }
 493
 494 static GtkTreeViewColumn *divelist_column(struct DiveList *dl, int index, const char *title,
 495                                 data_func_t data_func, PangoAlignment align, gboolean visible)
 496 {
 497         return tree_view_column(dl->tree_view, index, title, data_func, align, visible);
 498 }
 499
 500 /*
 501  * This is some crazy crap. The only way to get default focus seems
 502  * to be to grab focus as the widget is being shown the first time.
 503  */
 504 static void realize_cb(GtkWidget *tree_view, gpointer userdata)
 505 {
 506         gtk_widget_grab_focus(tree_view);
 507 }
 508
 509 static void row_activated_cb(GtkTreeView *tree_view,
 510                         GtkTreePath *path,
 511                         GtkTreeViewColumn *column,
 512                         GtkTreeModel *model)
 513 {
 514         int index;
 515         GtkTreeIter iter;
 516
 517         if (!gtk_tree_model_get_iter(model, &iter, path))
 518                 return;
 519         gtk_tree_model_get(model, &iter, DIVE_INDEX, &index, -1);
 520         edit_dive_info(get_dive(index));
 521 }
 522
 523 GtkWidget *dive_list_create(void)
 524 {
 525         GtkTreeSelection  *selection;
 526
 527         dive_list.model = gtk_list_store_new(DIVELIST_COLUMNS,
 528                                 G_TYPE_INT,                     /* index */
 529                                 G_TYPE_INT,                     /* nr */
 530                                 G_TYPE_INT,                     /* Date */
 531                                 G_TYPE_INT,                     /* Depth */
 532                                 G_TYPE_INT,                     /* Duration */
 533                                 G_TYPE_INT,                     /* Temperature */
 534                                 G_TYPE_STRING,                  /* Cylinder */
 535                                 G_TYPE_INT,                     /* Nitrox */
 536                                 G_TYPE_INT,                     /* SAC */
 537                                 G_TYPE_INT,                     /* OTU */
 538                                 G_TYPE_STRING                   /* Location */
 539                                 );
 540         dive_list.tree_view = gtk_tree_view_new_with_model(GTK_TREE_MODEL(dive_list.model));
 541         set_divelist_font(divelist_font);
 542
 543         selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(dive_list.tree_view));
 544
 545         gtk_tree_selection_set_mode(GTK_TREE_SELECTION(selection), GTK_SELECTION_MULTIPLE);
 546         gtk_widget_set_size_request(dive_list.tree_view, 200, 200);
 547
 548         dive_list.nr = divelist_column(&dive_list, DIVE_NR, "#", NULL, PANGO_ALIGN_RIGHT, TRUE);
 549         gtk_tree_view_column_set_sort_column_id(dive_list.nr, -1);
 550         dive_list.date = divelist_column(&dive_list, DIVE_DATE, "Date", date_data_func, PANGO_ALIGN_LEFT, TRUE);
 551         dive_list.depth = divelist_column(&dive_list, DIVE_DEPTH, "ft", depth_data_func, PANGO_ALIGN_RIGHT, TRUE);
 552         dive_list.duration = divelist_column(&dive_list, DIVE_DURATION, "min", duration_data_func, PANGO_ALIGN_RIGHT, TRUE);
 553         dive_list.temperature = divelist_column(&dive_list, DIVE_TEMPERATURE, UTF8_DEGREE "F", temperature_data_func, PANGO_ALIGN_RIGHT, visible_cols.temperature);
 554         dive_list.cylinder = divelist_column(&dive_list, DIVE_CYLINDER, "Cyl", NULL, PANGO_ALIGN_CENTER, visible_cols.cylinder);
 555         dive_list.nitrox = divelist_column(&dive_list, DIVE_NITROX, "O" UTF8_SUBSCRIPT_2 "%", nitrox_data_func, PANGO_ALIGN_CENTER, visible_cols.nitrox);
 556         dive_list.sac = divelist_column(&dive_list, DIVE_SAC, "SAC", sac_data_func, PANGO_ALIGN_CENTER, visible_cols.sac);
 557         dive_list.otu = divelist_column(&dive_list, DIVE_OTU, "OTU", otu_data_func, PANGO_ALIGN_CENTER, visible_cols.otu);
 558         dive_list.location = divelist_column(&dive_list, DIVE_LOCATION, "Location", NULL, PANGO_ALIGN_LEFT, TRUE);
 559
 560         fill_dive_list();
 561
 562         g_object_set(G_OBJECT(dive_list.tree_view), "headers-visible", TRUE,
 563                                           "search-column", DIVE_LOCATION,
 564                                           "rules-hint", TRUE,
 565                                           NULL);
 566
 567         g_signal_connect_after(dive_list.tree_view, "realize", G_CALLBACK(realize_cb), NULL);
 568         g_signal_connect(dive_list.tree_view, "row-activated", G_CALLBACK(row_activated_cb), dive_list.model);
 569         g_signal_connect(selection, "changed", G_CALLBACK(selection_cb), dive_list.model);
 570
 571         dive_list.container_widget = gtk_scrolled_window_new(NULL, NULL);
 572         gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(dive_list.container_widget),
 573                                GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
 574         gtk_container_add(GTK_CONTAINER(dive_list.container_widget), dive_list.tree_view);
 575
 576         dive_list.changed = 0;
 577
 578         return dive_list.container_widget;
 579 }
 580
 581 void mark_divelist_changed(int changed)
 582 {
 583         dive_list.changed = changed;
 584 }
 585
 586 int unsaved_changes()
 587 {
 588         return dive_list.changed;
 589 }