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