Reload file before printing final histogram

[mnemon] / mnemon.c

diff --git a/mnemon.c b/mnemon.c
index 4c5acd26287ce381e13a25efa86c81eb800536da..db8f62fe87c917de251bcb4537c65eed7f65c15c 100644 (file)
--- a/mnemon.c
+++ b/mnemon.c
@@ -1151,7 +1151,8 @@ mnemon_do_challenges (mnemon_t *mnemon)
 
        while (1) {
            if (category->time_limit > 0.0) {
-               readline ("The next one is timed. Press enter when ready:");
+               response = readline ("The next one is timed. Press enter when ready:");
+               free (response);
            }
                
            printf ("%s\n", item->challenge);
@@ -1166,10 +1167,12 @@ mnemon_do_challenges (mnemon_t *mnemon)
                return;
            }
 
-           if (response[0] == '/')
+           if (response[0] == '/') {
                mnemon_handle_command (mnemon, response + 1);
-           else
+               free (response);
+           } else {
                break;
+           }
        }
 
        mnemon_handle_response (mnemon, bin, item_index,
@@ -1177,17 +1180,17 @@ mnemon_do_challenges (mnemon_t *mnemon)
                                (end.tv_sec + end.tv_usec / 1e6) -
                                (start.tv_sec + start.tv_usec / 1e6),
                                category->time_limit);
+       free (response);
     } while (mnemon->to_introduce ||
             mnemon->unlearned ||
             mnemon->to_master > 0);
-
-    printf ("Great job.\n");
 }
 
 int
 main (int argc, char *argv[])
 {
     mnemon_t mnemon;
+    char *response;
 
     srand (time (NULL));
 
@@ -1201,5 +1204,15 @@ main (int argc, char *argv[])
 
     mnemon_fini (&mnemon);
 
+    mnemon_init (&mnemon);
+    mnemon_load (&mnemon);
+
+    printf ("Great job.\nHere are your current results:\n");
+    mnemon_print_histogram (&mnemon, NULL, 0);
+    response = readline ("Press enter to quit.\n");
+    free (response);
+
+    mnemon_fini (&mnemon);
+
     return 0;
 }