ncurses 6.1 - patch 20191102
[ncurses.git] / test / hanoi.c
index bd0fb10e1f8e9a3aaca0067f3aca82357921ae4e..f24ae54fa8e753cffbb978254061bae048f67af2 100644 (file)
@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright (c) 1998-2006,2008 Free Software Foundation, Inc.              *
+ * Copyright (c) 1998-2014,2017 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            *
  *
  *     Date: 05.Nov.90
  *
- * $Id: hanoi.c,v 1.27 2008/08/04 10:57:59 tom Exp $
+ * $Id: hanoi.c,v 1.39 2017/09/09 00:19:24 tom Exp $
  */
 
 #include <test.priv.h>
+#include <math.h>
 
 #define NPEGS                  3       /* This is not configurable !! */
 #define MINTILES               3
@@ -57,7 +58,7 @@
 #define MIDPEG                 39
 #define RIGHTPEG               59
 
-#define LENTOIND(x)            (((x)-1)/2)
+#define LENTOIND(x)            (((int)(x)-1)/2)
 #define OTHER(a,b)             (3-((a)+(b)))
 
 struct Peg {
@@ -67,8 +68,12 @@ struct Peg {
 
 static struct Peg Pegs[NPEGS];
 static int PegPos[] =
-{LEFTPEG, MIDPEG, RIGHTPEG};
-static int TileColour[] =
+{
+    LEFTPEG,
+    MIDPEG,
+    RIGHTPEG
+};
+static short TileColour[] =
 {
     COLOR_GREEN,               /* Length 3 */
     COLOR_MAGENTA,             /* Length 5 */
@@ -80,110 +85,10 @@ static int TileColour[] =
     COLOR_MAGENTA,             /* Length 17 */
     COLOR_RED,                 /* Length 19 */
 };
+static int NTiles = 0;
 static int NMoves = 0;
 static bool AutoFlag = FALSE;
 
-static void InitTiles(int NTiles);
-static void DisplayTiles(void);
-static void MakeMove(int From, int To);
-static void AutoMove(int From, int To, int Num);
-static void Usage(void);
-static int Solved(int NumTiles);
-static int GetMove(int *From, int *To);
-static int InvalidMove(int From, int To);
-
-int
-main(int argc, char **argv)
-{
-    int NTiles, FromCol, ToCol;
-
-    setlocale(LC_ALL, "");
-
-    switch (argc) {
-    case 1:
-       NTiles = DEFAULTTILES;
-       break;
-    case 2:
-       NTiles = atoi(argv[1]);
-       if (NTiles > MAXTILES || NTiles < MINTILES) {
-           fprintf(stderr, "Range %d to %d\n", MINTILES, MAXTILES);
-           ExitProgram(EXIT_FAILURE);
-       }
-       break;
-    case 3:
-       if (strcmp(argv[2], "a")) {
-           Usage();
-           ExitProgram(EXIT_FAILURE);
-       }
-       NTiles = atoi(argv[1]);
-       if (NTiles > MAXTILES || NTiles < MINTILES) {
-           fprintf(stderr, "Range %d to %d\n", MINTILES, MAXTILES);
-           ExitProgram(EXIT_FAILURE);
-       }
-       AutoFlag = TRUE;
-       break;
-    default:
-       Usage();
-       ExitProgram(EXIT_FAILURE);
-    }
-#ifdef TRACE
-    trace(TRACE_MAXIMUM);
-#endif
-    initscr();
-    if (has_colors()) {
-       int i;
-       int bg = COLOR_BLACK;
-       start_color();
-#if HAVE_USE_DEFAULT_COLORS
-       if (use_default_colors() == OK)
-           bg = -1;
-#endif
-       for (i = 0; i < 9; i++)
-           init_pair(i + 1, bg, TileColour[i]);
-    }
-    cbreak();
-    if (LINES < 24) {
-       endwin();
-       fprintf(stderr, "Min screen length 24 lines\n");
-       ExitProgram(EXIT_FAILURE);
-    }
-    if (AutoFlag) {
-       curs_set(0);
-       leaveok(stdscr, TRUE);  /* Attempt to remove cursor */
-    }
-    InitTiles(NTiles);
-    DisplayTiles();
-    if (AutoFlag) {
-       do {
-           noecho();
-           AutoMove(0, 2, NTiles);
-       } while (!Solved(NTiles));
-       sleep(2);
-    } else {
-       echo();
-       for (;;) {
-           if (GetMove(&FromCol, &ToCol))
-               break;
-           if (InvalidMove(FromCol, ToCol)) {
-               mvaddstr(STATUSLINE, 0, "Invalid Move !!");
-               refresh();
-               beep();
-               continue;
-           }
-           MakeMove(FromCol, ToCol);
-           if (Solved(NTiles)) {
-               mvprintw(STATUSLINE, 0,
-                        "Well Done !! You did it in %d moves", NMoves);
-               refresh();
-               sleep(5);
-               break;
-           }
-       }
-    }
-    endwin();
-    ExitProgram(EXIT_SUCCESS);
-}
-
 static int
 InvalidMove(int From, int To)
 {
@@ -207,12 +112,12 @@ InvalidMove(int From, int To)
 }
 
 static void
-InitTiles(int NTiles)
+InitTiles(void)
 {
     int Size, SlotNo;
 
     for (Size = NTiles * 2 + 1, SlotNo = 0; Size >= 3; Size -= 2)
-       Pegs[0].Length[SlotNo++] = Size;
+       Pegs[0].Length[SlotNo++] = (size_t) Size;
 
     Pegs[0].Count = NTiles;
     Pegs[1].Count = 0;
@@ -226,48 +131,48 @@ DisplayTiles(void)
     char TileBuf[BUFSIZ];
 
     erase();
-    mvaddstr(1, 24, "T O W E R S   O F   H A N O I");
-    mvaddstr(3, 34, "SJR 1990");
-    mvprintw(19, 5, "Moves : %d", NMoves);
-    attrset(A_REVERSE);
-    mvaddstr(BASELINE, 8,
+    MvAddStr(1, 24, "T O W E R S   O F   H A N O I");
+    MvAddStr(3, 34, "SJR 1990");
+    MvPrintw(19, 5, "Moves : %d of %.0f", NMoves, pow(2.0, (float) NTiles) - 1);
+    (void) attrset(A_REVERSE);
+    MvAddStr(BASELINE, 8,
             "                                                               ");
 
     for (Line = TOPLINE; Line < BASELINE; Line++) {
-       mvaddch(Line, LEFTPEG, ' ');
-       mvaddch(Line, MIDPEG, ' ');
-       mvaddch(Line, RIGHTPEG, ' ');
+       MvAddCh(Line, LEFTPEG, ' ');
+       MvAddCh(Line, MIDPEG, ' ');
+       MvAddCh(Line, RIGHTPEG, ' ');
     }
-    mvaddch(BASELINE, LEFTPEG, '1');
-    mvaddch(BASELINE, MIDPEG, '2');
-    mvaddch(BASELINE, RIGHTPEG, '3');
-    attrset(A_NORMAL);
+    MvAddCh(BASELINE, LEFTPEG, '1');
+    MvAddCh(BASELINE, MIDPEG, '2');
+    MvAddCh(BASELINE, RIGHTPEG, '3');
+    (void) attrset(A_NORMAL);
 
     /* Draw tiles */
     for (peg = 0; peg < NPEGS; peg++) {
        for (SlotNo = 0; SlotNo < Pegs[peg].Count; SlotNo++) {
-           unsigned len = Pegs[peg].Length[SlotNo];
-           if (len < sizeof(TileBuf) - 1 && len < (unsigned) PegPos[peg]) {
+           size_t len = Pegs[peg].Length[SlotNo];
+           if (len < sizeof(TileBuf) - 1 && len < (size_t) PegPos[peg]) {
                memset(TileBuf, ' ', len);
                TileBuf[len] = '\0';
                if (has_colors())
-                   attrset(COLOR_PAIR(LENTOIND(len)));
+                   (void) attrset(AttrArg(COLOR_PAIR(LENTOIND(len)), 0));
                else
-                   attrset(A_REVERSE);
-               mvaddstr(BASELINE - (SlotNo + 1),
-                        (int) (PegPos[peg] - len / 2),
+                   (void) attrset(A_REVERSE);
+               MvAddStr(BASELINE - (SlotNo + 1),
+                        (PegPos[peg] - (int) len / 2),
                         TileBuf);
            }
        }
     }
-    attrset(A_NORMAL);
+    (void) attrset(A_NORMAL);
     refresh();
 }
 
 static int
 GetMove(int *From, int *To)
 {
-    mvaddstr(STATUSLINE, 0, "Next move ('q' to quit) from ");
+    MvAddStr(STATUSLINE, 0, "Next move ('q' to quit) from ");
     clrtoeol();
     refresh();
     if ((*From = getch()) == 'q')
@@ -306,12 +211,12 @@ AutoMove(int From, int To, int Num)
     if (Num == 1) {
        MakeMove(From, To);
        napms(500);
-       return;
+    } else {
+       AutoMove(From, OTHER(From, To), Num - 1);
+       MakeMove(From, To);
+       napms(500);
+       AutoMove(OTHER(From, To), To, Num - 1);
     }
-    AutoMove(From, OTHER(From, To), Num - 1);
-    MakeMove(From, To);
-    napms(500);
-    AutoMove(OTHER(From, To), To, Num - 1);
 }
 
 static int
@@ -326,9 +231,129 @@ Solved(int NumTiles)
 }
 
 static void
-Usage(void)
+usage(void)
+{
+    static const char *msg[] =
+    {
+       "Usage: hanoi [options] [[<No Of Tiles>] [a]]"
+       ,""
+       ,"Options:"
+#if HAVE_USE_DEFAULT_COLORS
+       ," -d       invoke use_default_colors"
+#endif
+       ," -n NUM   set number of tiles (positional param is deprecated)"
+       ," -X       solve automatically (positional \"a\" is deprecated)"
+    };
+    size_t n;
+
+    for (n = 0; n < SIZEOF(msg); n++)
+       fprintf(stderr, "%s\n", msg[n]);
+
+    ExitProgram(EXIT_FAILURE);
+}
+
+int
+main(int argc, char **argv)
 {
-    fprintf(stderr, "Usage: hanoi [<No Of Tiles>] [a]\n");
-    fprintf(stderr,
-           "The 'a' option causes the tower to be solved automatically\n");
+    int ch, FromCol, ToCol;
+
+#if HAVE_USE_DEFAULT_COLORS
+    bool d_option = FALSE;
+#endif
+
+    NTiles = DEFAULTTILES;
+    while ((ch = getopt(argc, argv, "dn:X")) != -1) {
+       switch (ch) {
+#if HAVE_USE_DEFAULT_COLORS
+       case 'd':
+           d_option = TRUE;
+           break;
+#endif
+       case 'n':
+           NTiles = atoi(optarg);
+           break;
+       case 'X':
+           AutoFlag = TRUE;
+           break;
+       default:
+           usage();
+           /* NOTREACHED */
+       }
+    }
+    setlocale(LC_ALL, "");
+
+    switch (ch = (argc - optind)) {
+    case 2:
+       if (strcmp(argv[optind + 1], "a")) {
+           usage();
+       }
+       AutoFlag = TRUE;
+       /* FALLTHRU */
+    case 1:
+       NTiles = atoi(argv[optind]);
+       /* FALLTHRU */
+    case 0:
+       break;
+    default:
+       usage();
+    }
+
+    if (NTiles > MAXTILES || NTiles < MINTILES) {
+       fprintf(stderr, "Range %d to %d\n", MINTILES, MAXTILES);
+       usage();
+    }
+
+    initscr();
+    if (has_colors()) {
+       int i;
+       short bg = COLOR_BLACK;
+       start_color();
+#if HAVE_USE_DEFAULT_COLORS
+       if (d_option && (use_default_colors() == OK))
+           bg = -1;
+#endif
+       for (i = 0; i < 9; i++)
+           init_pair((short) (i + 1), bg, TileColour[i]);
+    }
+    cbreak();
+    if (LINES < 24) {
+       endwin();
+       fprintf(stderr, "Min screen length 24 lines\n");
+       ExitProgram(EXIT_FAILURE);
+    }
+    if (AutoFlag) {
+       curs_set(0);
+       leaveok(stdscr, TRUE);  /* Attempt to remove cursor */
+    }
+    InitTiles();
+    DisplayTiles();
+    if (AutoFlag) {
+       do {
+           noecho();
+           AutoMove(0, 2, NTiles);
+       } while (!Solved(NTiles));
+       sleep(2);
+    } else {
+       echo();
+       for (;;) {
+           if (GetMove(&FromCol, &ToCol))
+               break;
+           if (InvalidMove(FromCol, ToCol)) {
+               MvAddStr(STATUSLINE, 0, "Invalid Move !!");
+               refresh();
+               beep();
+               continue;
+           }
+           MakeMove(FromCol, ToCol);
+           if (Solved(NTiles)) {
+               MvPrintw(STATUSLINE, 0,
+                        "Well Done !! You did it in %d moves", NMoves);
+               refresh();
+               sleep(5);
+               break;
+           }
+       }
+    }
+    exit_curses();
+    ExitProgram(EXIT_SUCCESS);
 }