b8893e6243beba33acf724d4ba33f78b3d3ec91d
[ncurses.git] / test / bs.c
1 /****************************************************************************
2  * Copyright (c) 1998-2017,2018 Free Software Foundation, Inc.              *
3  *                                                                          *
4  * Permission is hereby granted, free of charge, to any person obtaining a  *
5  * copy of this software and associated documentation files (the            *
6  * "Software"), to deal in the Software without restriction, including      *
7  * without limitation the rights to use, copy, modify, merge, publish,      *
8  * distribute, distribute with modifications, sublicense, and/or sell       *
9  * copies of the Software, and to permit persons to whom the Software is    *
10  * furnished to do so, subject to the following conditions:                 *
11  *                                                                          *
12  * The above copyright notice and this permission notice shall be included  *
13  * in all copies or substantial portions of the Software.                   *
14  *                                                                          *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS  *
16  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF               *
17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.   *
18  * IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,   *
19  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR    *
20  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR    *
21  * THE USE OR OTHER DEALINGS IN THE SOFTWARE.                               *
22  *                                                                          *
23  * Except as contained in this notice, the name(s) of the above copyright   *
24  * holders shall not be used in advertising or otherwise to promote the     *
25  * sale, use or other dealings in this Software without prior written       *
26  * authorization.                                                           *
27  ****************************************************************************/
28 /* 
29  * bs.c - original author: Bruce Holloway
30  *              salvo option by: Chuck A DeGaul
31  * with improved user interface, autoconfiguration and code cleanup
32  *              by Eric S. Raymond <esr@snark.thyrsus.com>
33  * v1.2 with color support and minor portability fixes, November 1990
34  * v2.0 featuring strict ANSI/POSIX conformance, November 1993.
35  * v2.1 with ncurses mouse support, September 1995
36  *
37  * $Id: bs.c,v 1.73 2018/05/12 15:07:51 tom Exp $
38  */
39
40 #include <test.priv.h>
41
42 #include <time.h>
43
44 #ifndef SIGIOT
45 #define SIGIOT SIGABRT
46 #endif
47
48 static int getcoord(int);
49
50 /*
51  * Constants for tuning the random-fire algorithm. It prefers moves that
52  * diagonal-stripe the board with a stripe separation of srchstep. If
53  * no such preferred moves are found, srchstep is decremented.
54  */
55 #define BEGINSTEP       3       /* initial value of srchstep */
56
57 /* miscellaneous constants */
58 #define SHIPTYPES       5
59 #define OTHER           (1-turn)
60 #define PLAYER          0
61 #define COMPUTER        1
62 #define MARK_HIT        'H'
63 #define MARK_MISS       'o'
64 #define CTRLC           '\003'  /* used as terminate command */
65 #define FF              '\014'  /* used as redraw command */
66
67 #define is_QUIT(c) ((c) == CTRLC || (c) == QUIT)
68
69 /* coordinate handling */
70 #define BWIDTH          10
71 #define BDEPTH          10
72
73 /* display symbols */
74 #define SHOWHIT         '*'
75 #define SHOWSPLASH      ' '
76 #define IS_SHIP(c)      (isupper(UChar(c)) ? TRUE : FALSE)
77
78 /* how to position us on player board */
79 #define PYBASE  3
80 #define PXBASE  3
81 #define PY(y)   (PYBASE + (y))
82 #define PX(x)   (PXBASE + (x)*3)
83 #define pgoto(y, x)     (void)move(PY(y), PX(x))
84
85 /* how to position us on cpu board */
86 #define CYBASE  3
87 #define CXBASE  48
88 #define CY(y)   (CYBASE + (y))
89 #define CX(x)   (CXBASE + (x)*3)
90 #define CYINV(y)        ((y) - CYBASE)
91 #define CXINV(x)        (((x) - CXBASE) / 3)
92 #define cgoto(y, x)     (void)move(CY(y), CX(x))
93
94 #define ONBOARD(x, y)   (x >= 0 && x < BWIDTH && y >= 0 && y < BDEPTH)
95
96 /* other board locations */
97 #define COLWIDTH        80
98 #define PROMPTLINE      21      /* prompt line */
99 #define SYBASE          CYBASE + BDEPTH + 3     /* move key diagram */
100 #define SXBASE          63
101 #define MYBASE          SYBASE - 1      /* diagram caption */
102 #define MXBASE          64
103 #define HYBASE          SYBASE - 1      /* help area */
104 #define HXBASE          0
105
106 /* this will need to be changed if BWIDTH changes */
107 static char numbers[] = "   0  1  2  3  4  5  6  7  8  9";
108
109 static char carrier[] = "Aircraft Carrier";
110 static char battle[] = "Battleship";
111 static char sub[] = "Submarine";
112 static char destroy[] = "Destroyer";
113 static char ptboat[] = "PT Boat";
114
115 static char *your_name;
116 static char dftname[] = "stranger";
117
118 /* direction constants */
119 typedef enum {
120     dir_E = 0
121     ,dir_SE
122     ,dir_S
123     ,dir_SW
124     ,dir_W
125     ,dir_NW
126     ,dir_N
127     ,dir_NE
128     ,dir_MAX
129 } DIRECTIONS;
130 static int xincr[dir_MAX + 2] =
131 {1, 1, 0, -1, -1, -1, 0, 1};
132 static int yincr[dir_MAX + 2] =
133 {0, 1, 1, 1, 0, -1, -1, -1};
134
135 /* current ship position and direction */
136 static int curx = (BWIDTH / 2);
137 static int cury = (BDEPTH / 2);
138
139 typedef struct {
140     char *name;                 /* name of the ship type */
141     int hits;                   /* how many times has this ship been hit? */
142     char symbol;                /* symbol for game purposes */
143     int length;                 /* length of ship */
144     int x, y;                   /* coordinates of ship start point */
145     int dir;                    /* direction of `bow' */
146     bool placed;                /* has it been placed on the board? */
147 } ship_t;
148
149 static bool checkplace(int b, ship_t * ss, int vis);
150
151 #define SHIPIT(name, symbol, length) { name, 0, symbol, length, 0,0, 0, FALSE }
152
153 /* "ply=player", "cpu=computer" */
154 static ship_t plyship[SHIPTYPES] =
155 {
156     SHIPIT(carrier, 'A', 5),
157     SHIPIT(battle, 'B', 4),
158     SHIPIT(destroy, 'D', 3),
159     SHIPIT(sub, 'S', 3),
160     SHIPIT(ptboat, 'P', 2),
161 };
162
163 static ship_t cpuship[SHIPTYPES] =
164 {
165     SHIPIT(carrier, 'A', 5),
166     SHIPIT(battle, 'B', 4),
167     SHIPIT(destroy, 'D', 3),
168     SHIPIT(sub, 'S', 3),
169     SHIPIT(ptboat, 'P', 2),
170 };
171
172 /* "Hits" board, and main board. */
173 static char hits[2][BWIDTH][BDEPTH];
174 static char board[2][BWIDTH][BDEPTH];
175
176 static int turn;                /* 0=player, 1=computer */
177 static int plywon = 0, cpuwon = 0;      /* How many games has each won? */
178
179 static int salvo, blitz, closepack;
180
181 #define PR      (void)addstr
182
183 static void uninitgame(int sig) GCC_NORETURN;
184
185 static void
186 uninitgame(int sig GCC_UNUSED)
187 /* end the game, either normally or due to signal */
188 {
189     clear();
190     (void) refresh();
191     (void) reset_shell_mode();
192     (void) echo();
193     (void) endwin();
194     free(your_name);
195     ExitProgram(sig ? EXIT_FAILURE : EXIT_SUCCESS);
196 }
197
198 static void
199 announceopts(void)
200 /* announce which game options are enabled */
201 {
202     if (salvo || blitz || closepack) {
203         (void) printw("Playing optional game (");
204         if (salvo)
205             (void) printw("salvo, ");
206         else
207             (void) printw("nosalvo, ");
208         if (blitz)
209             (void) printw("blitz ");
210         else
211             (void) printw("noblitz, ");
212         if (closepack)
213             (void) printw("closepack)");
214         else
215             (void) printw("noclosepack)");
216     } else
217         (void) printw(
218                          "Playing standard game (noblitz, nosalvo, noclosepack)");
219 }
220
221 static void
222 intro(void)
223 {
224     const char *tmpname;
225
226     srand((unsigned) (time(0L) + getpid()));    /* Kick the random number generator */
227
228     InitAndCatch(initscr(), uninitgame);
229
230     if ((tmpname = getlogin()) != 0 &&
231         (your_name = strdup(tmpname)) != 0) {
232         your_name[0] = (char) toupper(UChar(your_name[0]));
233     } else {
234         your_name = strdup(dftname);
235     }
236
237     keypad(stdscr, TRUE);
238     (void) def_prog_mode();
239     (void) nonl();
240     (void) cbreak();
241     (void) noecho();
242
243 #ifdef PENGUIN
244     (void) clear();
245     MvAddStr(4, 29, "Welcome to Battleship!");
246     (void) move(8, 0);
247     PR("                                                  \\\n");
248     PR("                           \\                     \\ \\\n");
249     PR("                          \\ \\                   \\ \\ \\_____________\n");
250     PR("                         \\ \\ \\_____________      \\ \\/            |\n");
251     PR("                          \\ \\/             \\      \\/             |\n");
252     PR("                           \\/               \\_____/              |__\n");
253     PR("           ________________/                                       |\n");
254     PR("           \\  S.S. Penguin                                         |\n");
255     PR("            \\                                                     /\n");
256     PR("             \\___________________________________________________/\n");
257
258     MvAddStr(22, 27, "Hit any key to continue...");
259     (void) refresh();
260     (void) getch();
261 #endif /* PENGUIN */
262
263 #ifdef A_COLOR
264     start_color();
265
266     init_pair(COLOR_BLACK, COLOR_BLACK, COLOR_BLACK);
267     init_pair(COLOR_GREEN, COLOR_GREEN, COLOR_BLACK);
268     init_pair(COLOR_RED, COLOR_RED, COLOR_BLACK);
269     init_pair(COLOR_CYAN, COLOR_CYAN, COLOR_BLACK);
270     init_pair(COLOR_WHITE, COLOR_WHITE, COLOR_BLACK);
271     init_pair(COLOR_MAGENTA, COLOR_MAGENTA, COLOR_BLACK);
272     init_pair(COLOR_BLUE, COLOR_BLUE, COLOR_BLACK);
273     init_pair(COLOR_YELLOW, COLOR_YELLOW, COLOR_BLACK);
274 #endif /* A_COLOR */
275
276 #ifdef NCURSES_MOUSE_VERSION
277     (void) mousemask(BUTTON1_CLICKED, (mmask_t *) NULL);
278 #endif /* NCURSES_MOUSE_VERSION */
279 }
280
281 /* VARARGS1 */
282 static void
283 prompt(int n, NCURSES_CONST char *f, const char *s)
284 /* print a message at the prompt line */
285 {
286     (void) move(PROMPTLINE + n, 0);
287     (void) clrtoeol();
288     (void) printw(f, s);
289     (void) refresh();
290 }
291
292 static void
293 error(NCURSES_CONST char *s)
294 {
295     (void) move(PROMPTLINE + 2, 0);
296     (void) clrtoeol();
297     if (s) {
298         (void) addstr(s);
299         (void) beep();
300     }
301 }
302
303 static void
304 placeship(int b, ship_t * ss, int vis)
305 {
306     int l;
307
308     for (l = 0; l < ss->length; ++l) {
309         int newx = ss->x + l * xincr[ss->dir];
310         int newy = ss->y + l * yincr[ss->dir];
311
312         board[b][newx][newy] = ss->symbol;
313         if (vis) {
314             pgoto(newy, newx);
315             AddCh(ss->symbol);
316         }
317     }
318     ss->hits = 0;
319 }
320
321 static int
322 rnd(int n)
323 {
324     return (((rand() & 0x7FFF) % n));
325 }
326
327 static void
328 randomplace(int b, ship_t * ss)
329 /* generate a valid random ship placement into px,py */
330 {
331
332     do {
333         ss->dir = rnd(2) ? dir_E : dir_S;
334         ss->x = rnd(BWIDTH - (ss->dir == dir_E ? ss->length : 0));
335         ss->y = rnd(BDEPTH - (ss->dir == dir_S ? ss->length : 0));
336     } while
337         (!checkplace(b, ss, FALSE));
338 }
339
340 static void
341 initgame(void)
342 {
343     int i, j, unplaced;
344     ship_t *ss;
345
346     (void) clear();
347     MvAddStr(0, 35, "BATTLESHIPS");
348     (void) move(PROMPTLINE + 2, 0);
349     announceopts();
350
351     memset(board, 0, sizeof(char) * BWIDTH * BDEPTH * 2);
352     memset(hits, 0, sizeof(char) * BWIDTH * BDEPTH * 2);
353     for (i = 0; i < SHIPTYPES; i++) {
354         ss = cpuship + i;
355
356         ss->x =
357             ss->y =
358             ss->dir =
359             ss->hits = 0;
360         ss->placed = FALSE;
361
362         ss = plyship + i;
363
364         ss->x =
365             ss->y =
366             ss->dir =
367             ss->hits = 0;
368         ss->placed = FALSE;
369     }
370
371     /* draw empty boards */
372     MvAddStr(PYBASE - 2, PXBASE + 5, "Main Board");
373     MvAddStr(PYBASE - 1, PXBASE - 3, numbers);
374     for (i = 0; i < BDEPTH; ++i) {
375         MvAddCh(PYBASE + i, PXBASE - 3, (chtype) (i + 'A'));
376 #ifdef A_COLOR
377         if (has_colors())
378             attron(COLOR_PAIR(COLOR_BLUE));
379 #endif /* A_COLOR */
380         AddCh(' ');
381         for (j = 0; j < BWIDTH; j++)
382             (void) addstr(" . ");
383 #ifdef A_COLOR
384         (void) attrset(0);
385 #endif /* A_COLOR */
386         AddCh(' ');
387         AddCh(i + 'A');
388     }
389     MvAddStr(PYBASE + BDEPTH, PXBASE - 3, numbers);
390     MvAddStr(CYBASE - 2, CXBASE + 7, "Hit/Miss Board");
391     MvAddStr(CYBASE - 1, CXBASE - 3, numbers);
392     for (i = 0; i < BDEPTH; ++i) {
393         MvAddCh(CYBASE + i, CXBASE - 3, (chtype) (i + 'A'));
394 #ifdef A_COLOR
395         if (has_colors())
396             attron(COLOR_PAIR(COLOR_BLUE));
397 #endif /* A_COLOR */
398         AddCh(' ');
399         for (j = 0; j < BWIDTH; j++)
400             (void) addstr(" . ");
401 #ifdef A_COLOR
402         (void) attrset(0);
403 #endif /* A_COLOR */
404         AddCh(' ');
405         AddCh(i + 'A');
406     }
407
408     MvAddStr(CYBASE + BDEPTH, CXBASE - 3, numbers);
409
410     MvPrintw(HYBASE, HXBASE,
411              "To position your ships: move the cursor to a spot, then");
412     MvPrintw(HYBASE + 1, HXBASE,
413              "type the first letter of a ship type to select it, then");
414     MvPrintw(HYBASE + 2, HXBASE,
415              "type a direction ([hjkl] or [4862]), indicating how the");
416     MvPrintw(HYBASE + 3, HXBASE,
417              "ship should be pointed. You may also type a ship letter");
418     MvPrintw(HYBASE + 4, HXBASE,
419              "followed by `r' to position it randomly, or type `R' to");
420     MvPrintw(HYBASE + 5, HXBASE,
421              "place all remaining ships randomly.");
422
423     MvAddStr(MYBASE, MXBASE, "Aiming keys:");
424     MvAddStr(SYBASE, SXBASE, "y k u    7 8 9");
425     MvAddStr(SYBASE + 1, SXBASE, " \\|/      \\|/ ");
426     MvAddStr(SYBASE + 2, SXBASE, "h-+-l    4-+-6");
427     MvAddStr(SYBASE + 3, SXBASE, " /|\\      /|\\ ");
428     MvAddStr(SYBASE + 4, SXBASE, "b j n    1 2 3");
429
430     /* have the computer place ships */
431     for (ss = cpuship; ss < cpuship + SHIPTYPES; ss++) {
432         randomplace(COMPUTER, ss);
433         placeship(COMPUTER, ss, FALSE);
434     }
435
436     do {
437         char c, docked[SHIPTYPES + 2], *cp = docked;
438
439         ss = (ship_t *) NULL;
440
441         /* figure which ships still wait to be placed */
442         *cp++ = 'R';
443         for (i = 0; i < SHIPTYPES; i++)
444             if (!plyship[i].placed)
445                 *cp++ = plyship[i].symbol;
446         *cp = '\0';
447
448         /* get a command letter */
449         prompt(1, "Type one of [%s] to pick a ship.", docked + 1);
450         do {
451             c = (char) getcoord(PLAYER);
452         } while
453             (!(strchr) (docked, c));
454
455         if (c == 'R')
456             (void) ungetch('R');
457         else {
458             /* map that into the corresponding symbol */
459             for (ss = plyship; ss < plyship + SHIPTYPES; ss++)
460                 if (ss->symbol == c)
461                     break;
462
463             prompt(1, "Type one of [hjklrR] to place your %s.", ss->name);
464             pgoto(cury, curx);
465         }
466
467         do {
468             c = (char) getch();
469         } while
470             (!(strchr("hjkl8462rR", c) || c == FF || is_QUIT(c)));
471
472         if (is_QUIT(c)) {
473             uninitgame(0);
474         } else if (c == FF) {
475             (void) clearok(stdscr, TRUE);
476             (void) refresh();
477         } else if (ss == 0) {
478             beep();             /* simple to verify, unlikely to happen */
479         } else if (c == 'r') {
480             prompt(1, "Random-placing your %s", ss->name);
481             randomplace(PLAYER, ss);
482             placeship(PLAYER, ss, TRUE);
483             error((char *) NULL);
484             ss->placed = TRUE;
485         } else if (c == 'R') {
486             prompt(1, "Placing the rest of your fleet at random...", "");
487             for (ss = plyship; ss < plyship + SHIPTYPES; ss++)
488                 if (!ss->placed) {
489                     randomplace(PLAYER, ss);
490                     placeship(PLAYER, ss, TRUE);
491                     ss->placed = TRUE;
492                 }
493             error((char *) NULL);
494         } else if (strchr("hjkl8462", c)) {
495             ss->x = curx;
496             ss->y = cury;
497
498             switch (c) {
499             case 'k':
500             case '8':
501                 ss->dir = dir_N;
502                 break;
503             case 'j':
504             case '2':
505                 ss->dir = dir_S;
506                 break;
507             case 'h':
508             case '4':
509                 ss->dir = dir_W;
510                 break;
511             case 'l':
512             case '6':
513                 ss->dir = dir_E;
514                 break;
515             }
516
517             if (checkplace(PLAYER, ss, TRUE)) {
518                 placeship(PLAYER, ss, TRUE);
519                 error((char *) NULL);
520                 ss->placed = TRUE;
521             }
522         }
523
524         for (unplaced = i = 0; i < SHIPTYPES; i++)
525             unplaced += !plyship[i].placed;
526     } while
527         (unplaced);
528
529     turn = rnd(2);
530
531     MvPrintw(HYBASE, HXBASE,
532              "To fire, move the cursor to your chosen aiming point   ");
533     MvPrintw(HYBASE + 1, HXBASE,
534              "and strike any key other than a motion key.            ");
535     MvPrintw(HYBASE + 2, HXBASE,
536              "                                                       ");
537     MvPrintw(HYBASE + 3, HXBASE,
538              "                                                       ");
539     MvPrintw(HYBASE + 4, HXBASE,
540              "                                                       ");
541     MvPrintw(HYBASE + 5, HXBASE,
542              "                                                       ");
543
544     (void) prompt(0, "Press any key to start...", "");
545     (void) getch();
546 }
547
548 static int
549 getcoord(int atcpu)
550 {
551     int ny, nx, c;
552
553     if (atcpu)
554         cgoto(cury, curx);
555     else
556         pgoto(cury, curx);
557     (void) refresh();
558     for (;;) {
559         if (atcpu) {
560             MvPrintw(CYBASE + BDEPTH + 1, CXBASE + 11, "(%d, %c)",
561                      curx, 'A' + cury);
562             cgoto(cury, curx);
563         } else {
564             MvPrintw(PYBASE + BDEPTH + 1, PXBASE + 11, "(%d, %c)",
565                      curx, 'A' + cury);
566             pgoto(cury, curx);
567         }
568
569         switch (c = getch()) {
570         case 'k':
571         case '8':
572         case KEY_UP:
573             ny = cury + BDEPTH - 1;
574             nx = curx;
575             break;
576         case 'j':
577         case '2':
578         case KEY_DOWN:
579             ny = cury + 1;
580             nx = curx;
581             break;
582         case 'h':
583         case '4':
584         case KEY_LEFT:
585             ny = cury;
586             nx = curx + BWIDTH - 1;
587             break;
588         case 'l':
589         case '6':
590         case KEY_RIGHT:
591             ny = cury;
592             nx = curx + 1;
593             break;
594         case 'y':
595         case '7':
596         case KEY_A1:
597             ny = cury + BDEPTH - 1;
598             nx = curx + BWIDTH - 1;
599             break;
600         case 'b':
601         case '1':
602         case KEY_C1:
603             ny = cury + 1;
604             nx = curx + BWIDTH - 1;
605             break;
606         case 'u':
607         case '9':
608         case KEY_A3:
609             ny = cury + BDEPTH - 1;
610             nx = curx + 1;
611             break;
612         case 'n':
613         case '3':
614         case KEY_C3:
615             ny = cury + 1;
616             nx = curx + 1;
617             break;
618         case FF:
619             nx = curx;
620             ny = cury;
621             (void) clearok(stdscr, TRUE);
622             (void) refresh();
623             break;
624 #ifdef NCURSES_MOUSE_VERSION
625         case KEY_MOUSE:
626             {
627                 MEVENT myevent;
628
629                 getmouse(&myevent);
630                 if (atcpu
631                     && myevent.y >= CY(0) && myevent.y <= CY(BDEPTH)
632                     && myevent.x >= CX(0) && myevent.x <= CX(BDEPTH)) {
633                     curx = CXINV(myevent.x);
634                     cury = CYINV(myevent.y);
635                     return (' ');
636                 } else {
637                     beep();
638                     continue;
639                 }
640             }
641             /* no fall through */
642 #endif /* NCURSES_MOUSE_VERSION */
643
644         default:
645             if (atcpu)
646                 MvAddStr(CYBASE + BDEPTH + 1, CXBASE + 11, "      ");
647             else
648                 MvAddStr(PYBASE + BDEPTH + 1, PXBASE + 11, "      ");
649             return (c);
650         }
651
652         curx = nx % BWIDTH;
653         cury = ny % BDEPTH;
654     }
655 }
656
657 static bool
658 collidecheck(int b, int y, int x)
659 /* is this location on the selected zboard adjacent to a ship? */
660 {
661     bool collide;
662
663     /* anything on the square */
664     if ((collide = IS_SHIP(board[b][x][y])) != FALSE)
665         return (collide);
666
667     /* anything on the neighbors */
668     if (!closepack) {
669         int i;
670
671         for (i = 0; i < dir_MAX; i++) {
672             int xend, yend;
673
674             yend = y + yincr[i];
675             xend = x + xincr[i];
676             if (ONBOARD(xend, yend)
677                 && IS_SHIP(board[b][xend][yend])) {
678                 collide = TRUE;
679                 break;
680             }
681         }
682     }
683     return (collide);
684 }
685
686 static bool
687 checkplace(int b, ship_t * ss, int vis)
688 {
689     int l, xend, yend;
690
691     /* first, check for board edges */
692     xend = ss->x + (ss->length - 1) * xincr[ss->dir];
693     yend = ss->y + (ss->length - 1) * yincr[ss->dir];
694     if (!ONBOARD(xend, yend)) {
695         if (vis)
696             switch (rnd(3)) {
697             case 0:
698                 error("Ship is hanging from the edge of the world");
699                 break;
700             case 1:
701                 error("Try fitting it on the board");
702                 break;
703             case 2:
704                 error("Figure I won't find it if you put it there?");
705                 break;
706             }
707         return (FALSE);
708     }
709
710     for (l = 0; l < ss->length; ++l) {
711         if (collidecheck(b, ss->y + l * yincr[ss->dir], ss->x + l * xincr[ss->dir])) {
712             if (vis)
713                 switch (rnd(3)) {
714                 case 0:
715                     error("There's already a ship there");
716                     break;
717                 case 1:
718                     error("Collision alert!  Aaaaaagh!");
719                     break;
720                 case 2:
721                     error("Er, Admiral, what about the other ship?");
722                     break;
723                 }
724             return (FALSE);
725         }
726     }
727     return (TRUE);
728 }
729
730 static int
731 awinna(void)
732 {
733     int i, j;
734     ship_t *ss;
735
736     for (i = 0; i < 2; ++i) {
737         ss = (i) ? cpuship : plyship;
738         for (j = 0; j < SHIPTYPES; ++j, ++ss)
739             if (ss->length > ss->hits)
740                 break;
741         if (j == SHIPTYPES)
742             return (OTHER);
743     }
744     return (-1);
745 }
746
747 static ship_t *
748 hitship(int x, int y)
749 /* register a hit on the targeted ship */
750 {
751     ship_t *sb, *ss;
752     char sym;
753     int oldx, oldy;
754
755     getyx(stdscr, oldy, oldx);
756     sb = (turn) ? plyship : cpuship;
757     if ((sym = board[OTHER][x][y]) == 0)
758         return ((ship_t *) NULL);
759     for (ss = sb; ss < sb + SHIPTYPES; ++ss)
760         if (ss->symbol == sym) {
761             if (++ss->hits < ss->length)        /* still afloat? */
762                 return ((ship_t *) NULL);
763             else {              /* sunk! */
764                 int i, j;
765
766                 if (!closepack)
767                     for (j = -1; j <= 1; j++) {
768                         int bx = ss->x + j * xincr[(ss->dir + 2) % dir_MAX];
769                         int by = ss->y + j * yincr[(ss->dir + 2) % dir_MAX];
770
771                         for (i = -1; i <= ss->length; ++i) {
772                             int x1, y1;
773
774                             x1 = bx + i * xincr[ss->dir];
775                             y1 = by + i * yincr[ss->dir];
776                             if (ONBOARD(x1, y1)) {
777                                 hits[turn][x1][y1] = MARK_MISS;
778                                 if (turn % 2 == PLAYER) {
779                                     cgoto(y1, x1);
780 #ifdef A_COLOR
781                                     if (has_colors())
782                                         attron(COLOR_PAIR(COLOR_GREEN));
783 #endif /* A_COLOR */
784                                     AddCh(MARK_MISS);
785 #ifdef A_COLOR
786                                     (void) attrset(0);
787 #endif /* A_COLOR */
788                                 } else {
789                                     pgoto(y1, x1);
790                                     AddCh(SHOWSPLASH);
791                                 }
792                             }
793                         }
794                     }
795
796                 for (i = 0; i < ss->length; ++i) {
797                     int x1 = ss->x + i * xincr[ss->dir];
798                     int y1 = ss->y + i * yincr[ss->dir];
799
800                     hits[turn][x1][y1] = ss->symbol;
801                     if (turn % 2 == PLAYER) {
802                         cgoto(y1, x1);
803                         AddCh(ss->symbol);
804                     } else {
805                         pgoto(y1, x1);
806 #ifdef A_COLOR
807                         if (has_colors())
808                             attron(COLOR_PAIR(COLOR_RED));
809 #endif /* A_COLOR */
810                         AddCh(SHOWHIT);
811 #ifdef A_COLOR
812                         (void) attrset(0);
813 #endif /* A_COLOR */
814                     }
815                 }
816
817                 (void) move(oldy, oldx);
818                 return (ss);
819             }
820         }
821     (void) move(oldy, oldx);
822     return ((ship_t *) NULL);
823 }
824
825 static bool
826 plyturn(void)
827 {
828     ship_t *ss;
829     bool hit;
830     NCURSES_CONST char *m = NULL;
831
832     prompt(1, "Where do you want to shoot? ", "");
833     for (;;) {
834         (void) getcoord(COMPUTER);
835         if (hits[PLAYER][curx][cury]) {
836             prompt(1, "You shelled this spot already! Try again.", "");
837             beep();
838         } else
839             break;
840     }
841     hit = IS_SHIP(board[COMPUTER][curx][cury]);
842     hits[PLAYER][curx][cury] = (char) (hit ? MARK_HIT : MARK_MISS);
843     cgoto(cury, curx);
844 #ifdef A_COLOR
845     if (has_colors()) {
846         if (hit)
847             attron(COLOR_PAIR(COLOR_RED));
848         else
849             attron(COLOR_PAIR(COLOR_GREEN));
850     }
851 #endif /* A_COLOR */
852     AddCh(hits[PLAYER][curx][cury]);
853 #ifdef A_COLOR
854     (void) attrset(0);
855 #endif /* A_COLOR */
856
857     prompt(1, "You %s.", hit ? "scored a hit" : "missed");
858     if (hit && (ss = hitship(curx, cury))) {
859         switch (rnd(5)) {
860         case 0:
861             m = " You sank my %s!";
862             break;
863         case 1:
864             m = " I have this sinking feeling about my %s....";
865             break;
866         case 2:
867             m = " My %s has gone to Davy Jones's locker!";
868             break;
869         case 3:
870             m = " Glub, glub -- my %s is headed for the bottom!";
871             break;
872         case 4:
873             m = " You'll pick up survivors from my %s, I hope...!";
874             break;
875         }
876         if (m != 0) {
877             (void) printw(m, ss->name);
878         }
879         (void) beep();
880     }
881     return (hit);
882 }
883
884 static int
885 sgetc(const char *s)
886 {
887     const char *s1;
888     int ch;
889
890     (void) refresh();
891     for (;;) {
892         ch = getch();
893         if (islower(ch))
894             ch = toupper(ch);
895         if (is_QUIT(ch))
896             uninitgame(0);
897         for (s1 = s; *s1 && ch != *s1; ++s1)
898             continue;
899         if (*s1) {
900             AddCh(ch);
901             (void) refresh();
902             return (ch);
903         }
904     }
905 }
906
907 static void
908 randomfire(int *px, int *py)
909 /* random-fire routine -- implements simple diagonal-striping strategy */
910 {
911     static int turncount = 0;
912     static int srchstep = BEGINSTEP;
913     static int huntoffs;        /* Offset on search strategy */
914     int ypossible[BWIDTH * BDEPTH], xpossible[BWIDTH * BDEPTH], nposs;
915     int ypreferred[BWIDTH * BDEPTH], xpreferred[BWIDTH * BDEPTH], npref;
916     int x, y, i;
917
918     if (turncount++ == 0)
919         huntoffs = rnd(srchstep);
920
921     /* first, list all possible moves */
922     nposs = npref = 0;
923     for (x = 0; x < BWIDTH; x++)
924         for (y = 0; y < BDEPTH; y++)
925             if (!hits[COMPUTER][x][y]) {
926                 xpossible[nposs] = x;
927                 ypossible[nposs] = y;
928                 nposs++;
929                 if (((x + huntoffs) % srchstep) != (y % srchstep)) {
930                     xpreferred[npref] = x;
931                     ypreferred[npref] = y;
932                     npref++;
933                 }
934             }
935
936     if (npref) {
937         i = rnd(npref);
938
939         *px = xpreferred[i];
940         *py = ypreferred[i];
941     } else if (nposs) {
942         i = rnd(nposs);
943
944         *px = xpossible[i];
945         *py = ypossible[i];
946
947         if (srchstep > 1)
948             --srchstep;
949     } else {
950         error("No moves possible?? Help!");
951         ExitProgram(EXIT_FAILURE);
952         /*NOTREACHED */
953     }
954 }
955
956 #define S_MISS  0
957 #define S_HIT   1
958 #define S_SUNK  -1
959
960 static int
961 cpufire(int x, int y)
962 /* fire away at given location */
963 {
964     bool hit, sunk;
965     ship_t *ss = NULL;
966
967     hit = (bool) board[PLAYER][x][y];
968     hits[COMPUTER][x][y] = (hit ? MARK_HIT : MARK_MISS);
969     MvPrintw(PROMPTLINE, 0,
970              "I shoot at %c%d. I %s!", y + 'A', x, hit ? "hit" :
971              "miss");
972     if ((sunk = (hit && (ss = hitship(x, y)))) != 0)
973         (void) printw(" I've sunk your %s", ss->name);
974     (void) clrtoeol();
975
976     pgoto(y, x);
977 #ifdef A_COLOR
978     if (has_colors()) {
979         if (hit)
980             attron(COLOR_PAIR(COLOR_RED));
981         else
982             attron(COLOR_PAIR(COLOR_GREEN));
983     }
984 #endif /* A_COLOR */
985     AddCh((hit ? SHOWHIT : SHOWSPLASH));
986 #ifdef A_COLOR
987     (void) attrset(0);
988 #endif /* A_COLOR */
989
990     return hit ? (sunk ? S_SUNK : S_HIT) : S_MISS;
991 }
992
993 /*
994  * This code implements a fairly irregular FSM, so please forgive the rampant
995  * unstructuredness below. The five labels are states which need to be held
996  * between computer turns.
997  *
998  * The FSM is not externally reset to RANDOM_FIRE if the player wins. Instead,
999  * the other states check for "impossible" conditions which signify a new
1000  * game, then if found transition to RANDOM_FIRE.
1001  */
1002 static bool
1003 cputurn(void)
1004 {
1005 #define POSSIBLE(x, y)  (ONBOARD(x, y) && !hits[COMPUTER][x][y])
1006 #define RANDOM_FIRE     0
1007 #define RANDOM_HIT      1
1008 #define HUNT_DIRECT     2
1009 #define FIRST_PASS      3
1010 #define REVERSE_JUMP    4
1011 #define SECOND_PASS     5
1012     static int next = RANDOM_FIRE;
1013     static bool used[5];
1014     static ship_t ts;
1015     int navail, x, y, d, n;
1016     int hit = S_MISS;
1017
1018     switch (next) {
1019     case RANDOM_FIRE:           /* last shot was random and missed */
1020       refire:
1021         randomfire(&x, &y);
1022         if (!(hit = cpufire(x, y)))
1023             next = RANDOM_FIRE;
1024         else {
1025             ts.x = x;
1026             ts.y = y;
1027             ts.hits = 1;
1028             next = (hit == S_SUNK) ? RANDOM_FIRE : RANDOM_HIT;
1029         }
1030         break;
1031
1032     case RANDOM_HIT:            /* last shot was random and hit */
1033         used[dir_E / 2] =
1034             used[dir_S / 2] =
1035             used[dir_W / 2] =
1036             used[dir_N / 2] = FALSE;
1037         /* FALLTHROUGH */
1038
1039     case HUNT_DIRECT:           /* last shot hit, we're looking for ship's long axis */
1040         for (d = navail = 0; d < (dir_MAX) / 2; d++) {
1041             x = ts.x + xincr[d * 2];
1042             y = ts.y + yincr[d * 2];
1043             if (!used[d] && POSSIBLE(x, y))
1044                 navail++;
1045             else
1046                 used[d] = TRUE;
1047         }
1048         if (navail == 0)        /* no valid places for shots adjacent... */
1049             goto refire;        /* ...so we must random-fire */
1050         else {
1051             n = rnd(navail) + 1;
1052             for (d = 0; d < (dir_MAX) / 2 && used[d]; d++) ;
1053             /* used[d] is first that == 0 */
1054             for (; n > 1; n--)
1055                 while (d < (dir_MAX) / 2 && used[++d]) ;
1056             /* used[d] is next that == 0 */
1057
1058             assert(d < (dir_MAX) / 2);
1059             assert(used[d] == FALSE);
1060
1061             used[d] = TRUE;
1062             x = ts.x + xincr[d * 2];
1063             y = ts.y + yincr[d * 2];
1064
1065             assert(POSSIBLE(x, y));
1066
1067             if (!(hit = cpufire(x, y)))
1068                 next = HUNT_DIRECT;
1069             else {
1070                 ts.x = x;
1071                 ts.y = y;
1072                 ts.dir = d * 2;
1073                 ts.hits++;
1074                 next = (hit == S_SUNK) ? RANDOM_FIRE : FIRST_PASS;
1075             }
1076         }
1077         break;
1078
1079     case FIRST_PASS:            /* we have a start and a direction now */
1080         x = ts.x + xincr[ts.dir];
1081         y = ts.y + yincr[ts.dir];
1082         if (POSSIBLE(x, y) && (hit = cpufire(x, y))) {
1083             ts.x = x;
1084             ts.y = y;
1085             ts.hits++;
1086             next = (hit == S_SUNK) ? RANDOM_FIRE : FIRST_PASS;
1087         } else
1088             next = REVERSE_JUMP;
1089         break;
1090
1091     case REVERSE_JUMP:          /* nail down the ship's other end */
1092         d = (ts.dir + (dir_MAX) / 2) % dir_MAX;
1093         x = ts.x + ts.hits * xincr[d];
1094         y = ts.y + ts.hits * yincr[d];
1095         if (POSSIBLE(x, y) && (hit = cpufire(x, y))) {
1096             ts.x = x;
1097             ts.y = y;
1098             ts.dir = d;
1099             ts.hits++;
1100             next = (hit == S_SUNK) ? RANDOM_FIRE : SECOND_PASS;
1101         } else
1102             next = RANDOM_FIRE;
1103         break;
1104
1105     case SECOND_PASS:           /* continue shooting after reversing */
1106         x = ts.x + xincr[ts.dir];
1107         y = ts.y + yincr[ts.dir];
1108         if (POSSIBLE(x, y) && (hit = cpufire(x, y))) {
1109             ts.x = x;
1110             ts.y = y;
1111             ts.hits++;
1112             next = (hit == S_SUNK) ? RANDOM_FIRE : SECOND_PASS;
1113             break;
1114         } else
1115             next = RANDOM_FIRE;
1116         break;
1117     }
1118
1119     /* pause between shots in salvo */
1120     if (salvo) {
1121         (void) refresh();
1122         (void) sleep(1);
1123     }
1124 #ifdef DEBUG
1125     MvPrintw(PROMPTLINE + 2, 0,
1126              "New state %d, x=%d, y=%d, d=%d",
1127              next, x, y, d);
1128 #endif /* DEBUG */
1129     return ((hit) ? TRUE : FALSE);
1130 }
1131
1132 static int
1133 playagain(void)
1134 {
1135     int j;
1136     ship_t *ss;
1137
1138     for (ss = cpuship; ss < cpuship + SHIPTYPES; ss++)
1139         for (j = 0; j < ss->length; j++) {
1140             cgoto(ss->y + j * yincr[ss->dir], ss->x + j * xincr[ss->dir]);
1141             AddCh(ss->symbol);
1142         }
1143
1144     if (awinna())
1145         ++cpuwon;
1146     else
1147         ++plywon;
1148     j = 18 + (int) strlen(your_name);
1149     if (plywon >= 10)
1150         ++j;
1151     if (cpuwon >= 10)
1152         ++j;
1153     MvPrintw(1, (COLWIDTH - j) / 2,
1154              "%s: %d     Computer: %d", your_name, plywon, cpuwon);
1155
1156     prompt(2, (awinna())? "Want to be humiliated again, %s [yn]? "
1157            : "Going to give me a chance for revenge, %s [yn]? ", your_name);
1158     return (sgetc("YN") == 'Y');
1159 }
1160
1161 static void
1162 do_options(int c, char *op[])
1163 {
1164     register int i;
1165
1166     if (c > 1) {
1167         for (i = 1; i < c; i++) {
1168             switch (op[i][0]) {
1169             default:
1170             case '?':
1171                 (void) fprintf(stderr, "Usage: bs [-s | -b] [-c]\n");
1172                 (void) fprintf(stderr, "\tWhere the options are:\n");
1173                 (void) fprintf(stderr, "\t-s : play a salvo game\n");
1174                 (void) fprintf(stderr, "\t-b : play a blitz game\n");
1175                 (void) fprintf(stderr, "\t-c : ships may be adjacent\n");
1176                 ExitProgram(EXIT_FAILURE);
1177                 break;
1178             case '-':
1179                 switch (op[i][1]) {
1180                 case 'b':
1181                     blitz = 1;
1182                     if (salvo == 1) {
1183                         (void) fprintf(stderr,
1184                                        "Bad Arg: -b and -s are mutually exclusive\n");
1185                         ExitProgram(EXIT_FAILURE);
1186                     }
1187                     break;
1188                 case 's':
1189                     salvo = 1;
1190                     if (blitz == 1) {
1191                         (void) fprintf(stderr,
1192                                        "Bad Arg: -s and -b are mutually exclusive\n");
1193                         ExitProgram(EXIT_FAILURE);
1194                     }
1195                     break;
1196                 case 'c':
1197                     closepack = 1;
1198                     break;
1199                 default:
1200                     (void) fprintf(stderr,
1201                                    "Bad arg: type \"%s ?\" for usage message\n",
1202                                    op[0]);
1203                     ExitProgram(EXIT_FAILURE);
1204                 }
1205             }
1206         }
1207     }
1208 }
1209
1210 static int
1211 scount(int who)
1212 {
1213     register int i, shots;
1214     register ship_t *sp;
1215
1216     if (who)
1217         sp = cpuship;           /* count cpu shots */
1218     else
1219         sp = plyship;           /* count player shots */
1220
1221     for (i = 0, shots = 0; i < SHIPTYPES; i++, sp++) {
1222         if (sp->hits >= sp->length)
1223             continue;           /* dead ship */
1224         else
1225             shots++;
1226     }
1227     return (shots);
1228 }
1229
1230 int
1231 main(int argc, char *argv[])
1232 {
1233     setlocale(LC_ALL, "");
1234
1235     do_options(argc, argv);
1236
1237     intro();
1238     do {
1239         initgame();
1240         while (awinna() == -1) {
1241             if (!blitz) {
1242                 if (!salvo) {
1243                     if (turn)
1244                         (void) cputurn();
1245                     else
1246                         (void) plyturn();
1247                 } else {
1248                     register int i;
1249
1250                     i = scount(turn);
1251                     while (i--) {
1252                         if (turn) {
1253                             if (cputurn() && awinna() != -1)
1254                                 i = 0;
1255                         } else {
1256                             if (plyturn() && awinna() != -1)
1257                                 i = 0;
1258                         }
1259                     }
1260                 }
1261             } else
1262                 while ((turn ? cputurn() : plyturn()) && awinna() == -1)
1263                     continue;
1264             turn = OTHER;
1265         }
1266     } while
1267         (playagain());
1268     uninitgame(0);
1269     /*NOTREACHED */
1270 }
1271
1272 /* bs.c ends here */