ncurses 5.7 - patch 20100116

[ncurses.git] / test / savescreen.c

diff --git a/test/savescreen.c b/test/savescreen.c
old mode 100755 (executable)
new mode 100644 (file)
index 24e8a65..fe8d4e3
--- a/test/savescreen.c
+++ b/test/savescreen.c
@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright (c) 2007 Free Software Foundation, Inc.                        *
+ * Copyright (c) 2007,2009 Free Software Foundation, Inc.                   *
  *                                                                          *
  * Permission is hereby granted, free of charge, to any person obtaining a  *
  * copy of this software and associated documentation files (the            *
@@ -26,18 +26,17 @@
  * authorization.                                                           *
  ****************************************************************************/
 /*
- * $Id: savescreen.c,v 1.9 2007/07/14 23:16:55 tom Exp $
+ * $Id: savescreen.c,v 1.11 2009/10/10 19:38:21 tom Exp $
  *
  * Demonstrate save/restore functions from the curses library.
  * Thomas Dickey - 2007/7/14
  */
-/*
-scr_set                        -
-scr_init               -
-*/
 
 #include <test.priv.h>
 
+#include <sys/types.h>
+#include <sys/stat.h>
+
 #if TIME_WITH_SYS_TIME
 # include <sys/time.h>
 # include <time.h>
@@ -49,6 +48,15 @@ scr_init             -
 # endif
 #endif
 
+static bool use_init = FALSE;
+
+static int
+fexists(const char *name)
+{
+    struct stat sb;
+    return (stat(name, &sb) == 0 && (sb.st_mode & S_IFMT) == S_IFREG);
+}
+
 static void
 setup_next(void)
 {
@@ -66,6 +74,20 @@ cleanup(char *files[])
     }
 }
 
+static int
+load_screen(char *filename)
+{
+    int result;
+
+    if (use_init) {
+       if ((result = scr_init(filename)) != ERR)
+           result = scr_restore(filename);
+    } else {
+       result = scr_set(filename);
+    }
+    return result;
+}
+
 /*
  * scr_restore() or scr_set() operates on curscr.  If we read a character using
  * getch() that will refresh stdscr, wiping out the result.  To avoid that,
@@ -117,6 +139,7 @@ usage(void)
        "Usage: savescreen [-r] files",
        "",
        "Options:",
+       " -i  use scr_init/scr_restore rather than scr_set",
        " -r  replay the screen-dump files"
     };
     unsigned n;
@@ -136,8 +159,11 @@ main(int argc, char *argv[])
     bool done = FALSE;
     char **files;
 
-    while ((ch = getopt(argc, argv, "r")) != -1) {
+    while ((ch = getopt(argc, argv, "ir")) != -1) {
        switch (ch) {
+       case 'i':
+           use_init = TRUE;
+           break;
        case 'r':
            replaying = TRUE;
            break;
@@ -147,6 +173,14 @@ main(int argc, char *argv[])
        }
     }
 
+    files = argv + optind;
+    last = argc - optind - 1;
+
+    if (replaying) {
+       while (last >= 0 && !fexists(files[last]))
+           --last;
+    }
+
     initscr();
     cbreak();
     noecho();
@@ -160,8 +194,6 @@ main(int argc, char *argv[])
        }
     }
 
-    files = argv + optind;
-    last = argc - optind - 1;
     if (replaying) {
 
        /*
@@ -174,7 +206,7 @@ main(int argc, char *argv[])
        }
 
        which = last;
-       if (scr_set(files[which]) == ERR) {
+       if (load_screen(files[which]) == ERR) {
            endwin();
            printf("Cannot load screen-dump %s\n", files[which]);
            ExitProgram(EXIT_FAILURE);