]> ncurses.scripts.mit.edu Git - ncurses.git/blobdiff - test/hanoi.c
ncurses 5.7 - patch 20100116
[ncurses.git] / test / hanoi.c
index 3ee64b0a0d1453017e05f9271e55fe7c91e053e6..671c0cb47ed3ca2e76a11518e7cea05b4a1195f3 100644 (file)
@@ -1,3 +1,30 @@
+/****************************************************************************
+ * Copyright (c) 1998-2008,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            *
+ * "Software"), to deal in the Software without restriction, including      *
+ * without limitation the rights to use, copy, modify, merge, publish,      *
+ * distribute, distribute with modifications, sublicense, and/or sell       *
+ * copies of the Software, and to permit persons to whom the Software is    *
+ * furnished to do so, subject to the following conditions:                 *
+ *                                                                          *
+ * The above copyright notice and this permission notice shall be included  *
+ * in all copies or substantial portions of the Software.                   *
+ *                                                                          *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS  *
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF               *
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.   *
+ * IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,   *
+ * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR    *
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR    *
+ * THE USE OR OTHER DEALINGS IN THE SOFTWARE.                               *
+ *                                                                          *
+ * Except as contained in this notice, the name(s) of the above copyright   *
+ * holders shall not be used in advertising or otherwise to promote the     *
+ * sale, use or other dealings in this Software without prior written       *
+ * authorization.                                                           *
+ ****************************************************************************/
 /*
  *     Name: Towers of Hanoi.
  *
  *
  *     Date: 05.Nov.90
  *
- * $Id: hanoi.c,v 1.16 1999/10/23 15:01:01 tom Exp $
+ * $Id: hanoi.c,v 1.28 2009/08/29 19:02:25 tom Exp $
  */
 
 #include <test.priv.h>
 
-#include <string.h>
-
 #define NPEGS                  3       /* This is not configurable !! */
 #define MINTILES               3
 #define MAXTILES               9
-#define DEFAULTTILES   7
+#define DEFAULTTILES           7
 #define TOPLINE                        6
 #define BASELINE               16
 #define STATUSLINE             (LINES-3)
 #define OTHER(a,b)             (3-((a)+(b)))
 
 struct Peg {
-       size_t Length[MAXTILES];
-       int Count;
+    size_t Length[MAXTILES];
+    int Count;
 };
 
 static struct Peg Pegs[NPEGS];
-static int PegPos[] = { LEFTPEG, MIDPEG, RIGHTPEG };
-static int TileColour[] = {
-       COLOR_GREEN,    /* Length 3 */
-       COLOR_MAGENTA,  /* Length 5 */
-       COLOR_RED,      /* Length 7 */
-       COLOR_BLUE,     /* Length 9 */
-       COLOR_CYAN,     /* Length 11 */
-       COLOR_YELLOW,   /* Length 13 */
-       COLOR_GREEN,    /* Length 15 */
-       COLOR_MAGENTA,  /* Length 17 */
-       COLOR_RED,      /* Length 19 */
+static int PegPos[] =
+{LEFTPEG, MIDPEG, RIGHTPEG};
+static int TileColour[] =
+{
+    COLOR_GREEN,               /* Length 3 */
+    COLOR_MAGENTA,             /* Length 5 */
+    COLOR_RED,                 /* Length 7 */
+    COLOR_BLUE,                        /* Length 9 */
+    COLOR_CYAN,                        /* Length 11 */
+    COLOR_YELLOW,              /* Length 13 */
+    COLOR_GREEN,               /* Length 15 */
+    COLOR_MAGENTA,             /* Length 17 */
+    COLOR_RED,                 /* Length 19 */
 };
 static int NMoves = 0;
+static bool AutoFlag = FALSE;
 
 static void InitTiles(int NTiles);
 static void DisplayTiles(void);
@@ -67,232 +95,240 @@ static int InvalidMove(int From, int To);
 int
 main(int argc, char **argv)
 {
-int NTiles, FromCol, ToCol;
-unsigned char AutoFlag = 0;
+    int NTiles, FromCol, ToCol;
 
-       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);
-                       return EXIT_FAILURE;
-               }
-               break;
-       case 3:
-               if (strcmp(argv[2], "a")) {
-                       Usage();
-                       return EXIT_FAILURE;
-               }
-               NTiles = atoi(argv[1]);
-               if (NTiles > MAXTILES || NTiles < MINTILES) {
-                       fprintf(stderr, "Range %d to %d\n", MINTILES, MAXTILES);
-                       return EXIT_FAILURE;
-               }
-               AutoFlag = TRUE;
-               break;
-       default:
-               Usage();
-               return EXIT_FAILURE;
-       }
-#ifdef NCURSES_VERSION
-       trace(TRACE_MAXIMUM);
-#endif
-       initscr();
-       if (has_colors()) {
-               int i;
-               int bg = COLOR_BLACK;
-               start_color();
-#ifdef NCURSES_VERSION
-               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");
-               return EXIT_FAILURE;
+    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);
        }
-       if(AutoFlag) {
-               curs_set(0);
-               leaveok(stdscr, TRUE);  /* Attempt to remove cursor */
+       break;
+    case 3:
+       if (strcmp(argv[2], "a")) {
+           Usage();
+           ExitProgram(EXIT_FAILURE);
        }
-       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;
-                       }
-               }
+       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();
-       return EXIT_SUCCESS;
+       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)
 {
-       if(From >= NPEGS)
-               return TRUE;
-       if(From < 0)
-               return TRUE;
-       if(To >= NPEGS)
-               return TRUE;
-       if(To < 0)
-               return TRUE;
-       if(From == To)
-               return TRUE;
-       if(!Pegs[From].Count)
-               return TRUE;
-       if(Pegs[To].Count &&
-                       Pegs[From].Length[Pegs[From].Count-1] >
-                       Pegs[To].Length[Pegs[To].Count-1])
-               return TRUE;
-       return FALSE;
+    if (From >= NPEGS)
+       return TRUE;
+    if (From < 0)
+       return TRUE;
+    if (To >= NPEGS)
+       return TRUE;
+    if (To < 0)
+       return TRUE;
+    if (From == To)
+       return TRUE;
+    if (!Pegs[From].Count)
+       return TRUE;
+    if (Pegs[To].Count &&
+       Pegs[From].Length[Pegs[From].Count - 1] >
+       Pegs[To].Length[Pegs[To].Count - 1])
+       return TRUE;
+    return FALSE;
 }
 
 static void
 InitTiles(int NTiles)
 {
-       int Size, SlotNo;
+    int Size, SlotNo;
 
-       for(Size=NTiles*2+1, SlotNo=0; Size>=3; Size-=2)
-               Pegs[0].Length[SlotNo++] = Size;
+    for (Size = NTiles * 2 + 1, SlotNo = 0; Size >= 3; Size -= 2)
+       Pegs[0].Length[SlotNo++] = Size;
 
-       Pegs[0].Count = NTiles;
-       Pegs[1].Count = 0;
-       Pegs[2].Count = 0;
+    Pegs[0].Count = NTiles;
+    Pegs[1].Count = 0;
+    Pegs[2].Count = 0;
 }
 
 static void
 DisplayTiles(void)
 {
-       int Line, peg, SlotNo;
-       char TileBuf[BUFSIZ];
+    int Line, peg, SlotNo;
+    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, "                                                               ");
+    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);
+    (void) attrset(A_REVERSE);
+    mvaddstr(BASELINE, 8,
+            "                                                               ");
 
-       for(Line=TOPLINE; Line<BASELINE; Line++) {
-               mvaddch(Line, LEFTPEG, ' ');
-               mvaddch(Line, MIDPEG, ' ');
-               mvaddch(Line, RIGHTPEG, ' ');
-       }
-       mvaddch(BASELINE, LEFTPEG, '1');
-       mvaddch(BASELINE, MIDPEG, '2');
-       mvaddch(BASELINE, RIGHTPEG, '3');
-       attrset(A_NORMAL);
+    for (Line = TOPLINE; Line < BASELINE; Line++) {
+       mvaddch(Line, LEFTPEG, ' ');
+       mvaddch(Line, MIDPEG, ' ');
+       mvaddch(Line, RIGHTPEG, ' ');
+    }
+    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++) {
-                       memset(TileBuf, ' ', Pegs[peg].Length[SlotNo]);
-                       TileBuf[Pegs[peg].Length[SlotNo]] = '\0';
-                       if (has_colors())
-                               attrset(COLOR_PAIR(LENTOIND(Pegs[peg].Length[SlotNo])));
-                       else
-                               attrset(A_REVERSE);
-                       mvaddstr(BASELINE-(SlotNo+1),
-                               (int)(PegPos[peg] - Pegs[peg].Length[SlotNo]/2),
-                               TileBuf);
-               }
+    /* 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]) {
+               memset(TileBuf, ' ', len);
+               TileBuf[len] = '\0';
+               if (has_colors())
+                   (void) attrset(COLOR_PAIR(LENTOIND(len)));
+               else
+                   (void) attrset(A_REVERSE);
+               mvaddstr(BASELINE - (SlotNo + 1),
+                        (int) (PegPos[peg] - len / 2),
+                        TileBuf);
+           }
        }
-       attrset(A_NORMAL);
-       refresh();
+    }
+    (void) attrset(A_NORMAL);
+    refresh();
 }
 
 static int
 GetMove(int *From, int *To)
 {
-       mvaddstr(STATUSLINE, 0, "Next move ('q' to quit) from ");
-       clrtoeol();
-       refresh();
-       if((*From = getch()) == 'q')
-               return TRUE;
-       *From -= ('0'+1);
-       addstr(" to ");
-       clrtoeol();
-       refresh();
+    mvaddstr(STATUSLINE, 0, "Next move ('q' to quit) from ");
+    clrtoeol();
+    refresh();
+    if ((*From = getch()) == 'q')
+       return TRUE;
+    *From -= ('0' + 1);
+    addstr(" to ");
+    clrtoeol();
+    refresh();
 
-       if((*To = getch()) == 'q')
-               return TRUE;
-       *To -= ('0'+1);
-       refresh();
+    if ((*To = getch()) == 'q')
+       return TRUE;
+    *To -= ('0' + 1);
+    refresh();
+    if (!AutoFlag)
        napms(500);
 
-       move(STATUSLINE, 0);
-       clrtoeol();
-       refresh();
-       return FALSE;
+    move(STATUSLINE, 0);
+    clrtoeol();
+    refresh();
+    return FALSE;
 }
 
 static void
 MakeMove(int From, int To)
 {
-       Pegs[From].Count--;
-       Pegs[To].Length[Pegs[To].Count] = Pegs[From].Length[Pegs[From].Count];
-       Pegs[To].Count++;
-       NMoves++;
-       DisplayTiles();
+    Pegs[From].Count--;
+    Pegs[To].Length[Pegs[To].Count] = Pegs[From].Length[Pegs[From].Count];
+    Pegs[To].Count++;
+    NMoves++;
+    DisplayTiles();
 }
 
 static void
 AutoMove(int From, int To, int Num)
 {
-       if(Num == 1) {
-               MakeMove(From, To);
-               napms(500);
-               return;
-       }
-       AutoMove(From, OTHER(From, To), Num-1);
+    if (Num == 1) {
        MakeMove(From, To);
        napms(500);
-       AutoMove(OTHER(From, To), To, Num-1);
+       return;
+    }
+    AutoMove(From, OTHER(From, To), Num - 1);
+    MakeMove(From, To);
+    napms(500);
+    AutoMove(OTHER(From, To), To, Num - 1);
 }
 
 static int
 Solved(int NumTiles)
 {
-       int i;
+    int i;
 
-       for(i = 1; i < NPEGS; i++)
-               if (Pegs[i].Count == NumTiles)
-                       return TRUE;
-       return FALSE;
+    for (i = 1; i < NPEGS; i++)
+       if (Pegs[i].Count == NumTiles)
+           return TRUE;
+    return FALSE;
 }
 
 static void
-Usage()
+Usage(void)
 {
-       fprintf(stderr, "Usage: hanoi [<No Of Tiles>] [a]\n");
-       fprintf(stderr, "The 'a' option causes the tower to be solved automatically\n");
+    fprintf(stderr, "Usage: hanoi [<No Of Tiles>] [a]\n");
+    fprintf(stderr,
+           "The 'a' option causes the tower to be solved automatically\n");
 }