diff options
| -rw-r--r-- | templatelayout.cpp | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/templatelayout.cpp b/templatelayout.cpp index 880b6d774..5f47b64a2 100644 --- a/templatelayout.cpp +++ b/templatelayout.cpp @@ -2,18 +2,13 @@ #include "templatelayout.h" #include "helpers.h" +#include "display.h" int getTotalWork() { - int dives = 0, i; - struct dive *dive; - for_each_dive (i, dive) { - //TODO check for exporting selected dives only - if (!dive->selected) - continue; - dives++; - } - return dives; + // return the correct number depending on all/selected dives + // but don't return 0 as we might divide by this number + return amount_selected ? amount_selected : 1; } TemplateLayout::TemplateLayout(print_options *PrintOptions) : |