xref: /netbsd-src/games/gomoku/main.c (revision d0fed6c87ddc40a8bffa6f99e7433ddfc864dd83)
1 /*	$NetBSD: main.c,v 1.3 1997/01/03 01:35:28 cgd Exp $	*/
2 
3 /*
4  * Copyright (c) 1994
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * This code is derived from software contributed to Berkeley by
8  * Ralph Campbell.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. All advertising materials mentioning features or use of this software
19  *    must display the following acknowledgement:
20  *	This product includes software developed by the University of
21  *	California, Berkeley and its contributors.
22  * 4. Neither the name of the University nor the names of its contributors
23  *    may be used to endorse or promote products derived from this software
24  *    without specific prior written permission.
25  *
26  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
27  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
30  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36  * SUCH DAMAGE.
37  */
38 
39 #ifndef lint
40 static char copyright[] =
41 "@(#) Copyright (c) 1994\n\
42 	The Regents of the University of California.  All rights reserved.\n";
43 #endif /* not lint */
44 
45 #ifndef lint
46 #if 0
47 static char sccsid[] = "@(#)main.c	8.4 (Berkeley) 5/4/95";
48 #else
49 static char rcsid[] = "$NetBSD: main.c,v 1.3 1997/01/03 01:35:28 cgd Exp $";
50 #endif
51 #endif /* not lint */
52 
53 #include <curses.h>
54 #include <err.h>
55 #include <signal.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <string.h>
59 #include <unistd.h>
60 
61 #include "gomoku.h"
62 
63 #define USER	0		/* get input from standard input */
64 #define PROGRAM	1		/* get input from program */
65 #define INPUTF	2		/* get input from a file */
66 
67 int	interactive = 1;	/* true if interactive */
68 int	debug;			/* true if debugging */
69 int	test;			/* both moves come from 1: input, 2: computer */
70 char	*prog;			/* name of program */
71 FILE	*debugfp;		/* file for debug output */
72 FILE	*inputfp;		/* file for debug input */
73 
74 char	pdir[4]		= "-\\|/";
75 char	fmtbuf[128];
76 
77 struct	spotstr	board[BAREA];		/* info for board */
78 struct	combostr frames[FAREA];		/* storage for all frames */
79 struct	combostr *sortframes[2];	/* sorted list of non-empty frames */
80 u_char	overlap[FAREA * FAREA];		/* true if frame [a][b] overlap */
81 short	intersect[FAREA * FAREA];	/* frame [a][b] intersection */
82 int	movelog[BSZ * BSZ];		/* log of all the moves */
83 int	movenum;			/* current move number */
84 char	*plyr[2];			/* who's who */
85 
86 extern void quit();
87 #ifdef DEBUG
88 extern void whatsup();
89 #endif
90 
91 main(argc, argv)
92 	int argc;
93 	char **argv;
94 {
95 	char buf[128];
96 	int color, curmove, i, ch;
97 	int input[2];
98 	static char *fmt[2] = {
99 		"%3d %-6s",
100 		"%3d        %-6s"
101 	};
102 
103 	prog = strrchr(argv[0], '/');
104 	if (prog)
105 		prog++;
106 	else
107 		prog = argv[0];
108 
109 	while ((ch = getopt(argc, argv, "bcdD:u")) != EOF) {
110 		switch (ch) {
111 		case 'b':	/* background */
112 			interactive = 0;
113 			break;
114 		case 'd':	/* debugging */
115 			debug++;
116 			break;
117 		case 'D':	/* log debug output to file */
118 			if ((debugfp = fopen(optarg, "w")) == NULL)
119 				err(1, "%s", optarg);
120 			break;
121 		case 'u':	/* testing: user verses user */
122 			test = 1;
123 			break;
124 		case 'c':	/* testing: computer verses computer */
125 			test = 2;
126 			break;
127 		}
128 	}
129 	argc -= optind;
130 	argv += optind;
131 	if (argc) {
132 		if ((inputfp = fopen(*argv, "r")) == NULL)
133 			err(1, "%s", *argv);
134 	}
135 
136 	if (!debug)
137 #ifdef SVR4
138 		srand(time(0));
139 #else
140 		srandom(time(0));
141 #endif
142 	if (interactive)
143 		cursinit();		/* initialize curses */
144 again:
145 	bdinit(board);			/* initialize board contents */
146 
147 	if (interactive) {
148 		plyr[BLACK] = plyr[WHITE] = "???";
149 		bdisp_init();		/* initialize display of board */
150 #ifdef DEBUG
151 		signal(SIGINT, whatsup);
152 #else
153 		signal(SIGINT, quit);
154 #endif
155 
156 		if (inputfp == NULL && test == 0) {
157 			for (;;) {
158 				ask("black or white? ");
159 				getline(buf, sizeof(buf));
160 				if (buf[0] == 'b' || buf[0] == 'B') {
161 					color = BLACK;
162 					break;
163 				}
164 				if (buf[0] == 'w' || buf[0] == 'W') {
165 					color = WHITE;
166 					break;
167 				}
168 				move(22, 0);
169 				printw("Black moves first. Please enter `black' or `white'\n");
170 			}
171 			move(22, 0);
172 			clrtoeol();
173 		}
174 	} else {
175 		setbuf(stdout, 0);
176 		getline(buf, sizeof(buf));
177 		if (strcmp(buf, "black") == 0)
178 			color = BLACK;
179 		else if (strcmp(buf, "white") == 0)
180 			color = WHITE;
181 		else {
182 			sprintf(fmtbuf,
183 			    "Huh?  Expected `black' or `white', got `%s'\n",
184 			    buf);
185 			panic(fmtbuf);
186 		}
187 	}
188 
189 	if (inputfp) {
190 		input[BLACK] = INPUTF;
191 		input[WHITE] = INPUTF;
192 	} else {
193 		switch (test) {
194 		case 0: /* user verses program */
195 			input[color] = USER;
196 			input[!color] = PROGRAM;
197 			break;
198 
199 		case 1: /* user verses user */
200 			input[BLACK] = USER;
201 			input[WHITE] = USER;
202 			break;
203 
204 		case 2: /* program verses program */
205 			input[BLACK] = PROGRAM;
206 			input[WHITE] = PROGRAM;
207 			break;
208 		}
209 	}
210 	if (interactive) {
211 		plyr[BLACK] = input[BLACK] == USER ? "you" : prog;
212 		plyr[WHITE] = input[WHITE] == USER ? "you" : prog;
213 		bdwho(1);
214 	}
215 
216 	for (color = BLACK; ; color = !color) {
217 	top:
218 		switch (input[color]) {
219 		case INPUTF: /* input comes from a file */
220 			curmove = readinput(inputfp);
221 			if (curmove != ILLEGAL)
222 				break;
223 			switch (test) {
224 			case 0: /* user verses program */
225 				input[color] = USER;
226 				input[!color] = PROGRAM;
227 				break;
228 
229 			case 1: /* user verses user */
230 				input[BLACK] = USER;
231 				input[WHITE] = USER;
232 				break;
233 
234 			case 2: /* program verses program */
235 				input[BLACK] = PROGRAM;
236 				input[WHITE] = PROGRAM;
237 				break;
238 			}
239 			plyr[BLACK] = input[BLACK] == USER ? "you" : prog;
240 			plyr[WHITE] = input[WHITE] == USER ? "you" : prog;
241 			bdwho(1);
242 			goto top;
243 
244 		case USER: /* input comes from standard input */
245 		getinput:
246 			if (interactive)
247 				ask("move? ");
248 			if (!getline(buf, sizeof(buf))) {
249 				curmove = RESIGN;
250 				break;
251 			}
252 			if (buf[0] == '\0')
253 				goto getinput;
254 			curmove = ctos(buf);
255 			if (interactive) {
256 				if (curmove == SAVE) {
257 					FILE *fp;
258 
259 					ask("save file name? ");
260 					(void)getline(buf, sizeof(buf));
261 					if ((fp = fopen(buf, "w")) == NULL) {
262 						log("cannot create save file");
263 						goto getinput;
264 					}
265 					for (i = 0; i < movenum - 1; i++)
266 						fprintf(fp, "%s\n",
267 							stoc(movelog[i]));
268 					fclose(fp);
269 					goto getinput;
270 				}
271 				if (curmove != RESIGN &&
272 				    board[curmove].s_occ != EMPTY) {
273 					log("Illegal move");
274 					goto getinput;
275 				}
276 			}
277 			break;
278 
279 		case PROGRAM: /* input comes from the program */
280 			curmove = pickmove(color);
281 			break;
282 		}
283 		if (interactive) {
284 			sprintf(fmtbuf, fmt[color], movenum, stoc(curmove));
285 			log(fmtbuf);
286 		}
287 		if ((i = makemove(color, curmove)) != MOVEOK)
288 			break;
289 		if (interactive)
290 			bdisp();
291 	}
292 	if (interactive) {
293 		move(22, 0);
294 		switch (i) {
295 		case WIN:
296 			if (input[color] == PROGRAM)
297 				addstr("Ha ha, I won");
298 			else
299 				addstr("Rats! you won");
300 			break;
301 		case TIE:
302 			addstr("Wow! its a tie");
303 			break;
304 		case ILLEGAL:
305 			addstr("Illegal move");
306 			break;
307 		}
308 		clrtoeol();
309 		bdisp();
310 		if (i != RESIGN) {
311 		replay:
312 			ask("replay? ");
313 			if (getline(buf, sizeof(buf)) &&
314 			    buf[0] == 'y' || buf[0] == 'Y')
315 				goto again;
316 			if (strcmp(buf, "save") == 0) {
317 				FILE *fp;
318 
319 				ask("save file name? ");
320 				(void)getline(buf, sizeof(buf));
321 				if ((fp = fopen(buf, "w")) == NULL) {
322 					log("cannot create save file");
323 					goto replay;
324 				}
325 				for (i = 0; i < movenum - 1; i++)
326 					fprintf(fp, "%s\n",
327 						stoc(movelog[i]));
328 				fclose(fp);
329 				goto replay;
330 			}
331 		}
332 	}
333 	quit();
334 }
335 
336 readinput(fp)
337 	FILE *fp;
338 {
339 	char *cp;
340 	int c;
341 
342 	cp = fmtbuf;
343 	while ((c = getc(fp)) != EOF && c != '\n')
344 		*cp++ = c;
345 	*cp = '\0';
346 	return (ctos(fmtbuf));
347 }
348 
349 #ifdef DEBUG
350 /*
351  * Handle strange situations.
352  */
353 void
354 whatsup(signum)
355 	int signum;
356 {
357 	int i, pnum, n, s1, s2, d1, d2;
358 	struct spotstr *sp;
359 	FILE *fp;
360 	char *str;
361 	struct elist *ep;
362 	struct combostr *cbp;
363 
364 	if (!interactive)
365 		quit();
366 top:
367 	ask("cmd? ");
368 	if (!getline(fmtbuf, sizeof(fmtbuf)))
369 		quit();
370 	switch (*fmtbuf) {
371 	case '\0':
372 		goto top;
373 	case 'q':		/* conservative quit */
374 		quit();
375 	case 'd':		/* set debug level */
376 		debug = fmtbuf[1] - '0';
377 		sprintf(fmtbuf, "Debug set to %d", debug);
378 		dlog(fmtbuf);
379 		sleep(1);
380 	case 'c':
381 		break;
382 	case 'b':		/* back up a move */
383 		if (movenum > 1) {
384 			movenum--;
385 			board[movelog[movenum - 1]].s_occ = EMPTY;
386 			bdisp();
387 		}
388 		goto top;
389 	case 's':		/* suggest a move */
390 		i = fmtbuf[1] == 'b' ? BLACK : WHITE;
391 		sprintf(fmtbuf, "suggest %c %s", i == BLACK ? 'B' : 'W',
392 			stoc(pickmove(i)));
393 		dlog(fmtbuf);
394 		goto top;
395 	case 'f':		/* go forward a move */
396 		board[movelog[movenum - 1]].s_occ = movenum & 1 ? BLACK : WHITE;
397 		movenum++;
398 		bdisp();
399 		goto top;
400 	case 'l':		/* print move history */
401 		if (fmtbuf[1] == '\0') {
402 			for (i = 0; i < movenum - 1; i++)
403 				dlog(stoc(movelog[i]));
404 			goto top;
405 		}
406 		if ((fp = fopen(fmtbuf + 1, "w")) == NULL)
407 			goto top;
408 		for (i = 0; i < movenum - 1; i++) {
409 			fprintf(fp, "%s", stoc(movelog[i]));
410 			if (++i < movenum - 1)
411 				fprintf(fp, " %s\n", stoc(movelog[i]));
412 			else
413 				fputc('\n', fp);
414 		}
415 		bdump(fp);
416 		fclose(fp);
417 		goto top;
418 	case 'o':
419 		n = 0;
420 		for (str = fmtbuf + 1; *str; str++)
421 			if (*str == ',') {
422 				for (d1 = 0; d1 < 4; d1++)
423 					if (str[-1] == pdir[d1])
424 						break;
425 				str[-1] = '\0';
426 				sp = &board[s1 = ctos(fmtbuf + 1)];
427 				n = (sp->s_frame[d1] - frames) * FAREA;
428 				*str++ = '\0';
429 				break;
430 			}
431 		sp = &board[s2 = ctos(str)];
432 		while (*str)
433 			str++;
434 		for (d2 = 0; d2 < 4; d2++)
435 			if (str[-1] == pdir[d2])
436 				break;
437 		n += sp->s_frame[d2] - frames;
438 		str = fmtbuf;
439 		sprintf(str, "overlap %s%c,", stoc(s1), pdir[d1]);
440 		str += strlen(str);
441 		sprintf(str, "%s%c = %x", stoc(s2), pdir[d2], overlap[n]);
442 		dlog(fmtbuf);
443 		goto top;
444 	case 'p':
445 		sp = &board[i = ctos(fmtbuf + 1)];
446 		sprintf(fmtbuf, "V %s %x/%d %d %x/%d %d %d %x", stoc(i),
447 			sp->s_combo[BLACK].s, sp->s_level[BLACK],
448 			sp->s_nforce[BLACK],
449 			sp->s_combo[WHITE].s, sp->s_level[WHITE],
450 			sp->s_nforce[WHITE], sp->s_wval, sp->s_flg);
451 		dlog(fmtbuf);
452 		sprintf(fmtbuf, "FB %s %x %x %x %x", stoc(i),
453 			sp->s_fval[BLACK][0].s, sp->s_fval[BLACK][1].s,
454 			sp->s_fval[BLACK][2].s, sp->s_fval[BLACK][3].s);
455 		dlog(fmtbuf);
456 		sprintf(fmtbuf, "FW %s %x %x %x %x", stoc(i),
457 			sp->s_fval[WHITE][0].s, sp->s_fval[WHITE][1].s,
458 			sp->s_fval[WHITE][2].s, sp->s_fval[WHITE][3].s);
459 		dlog(fmtbuf);
460 		goto top;
461 	case 'e':	/* e {b|w} [0-9] spot */
462 		str = fmtbuf + 1;
463 		if (*str >= '0' && *str <= '9')
464 			n = *str++ - '0';
465 		else
466 			n = 0;
467 		sp = &board[i = ctos(str)];
468 		for (ep = sp->s_empty; ep; ep = ep->e_next) {
469 			cbp = ep->e_combo;
470 			if (n) {
471 				if (cbp->c_nframes > n)
472 					continue;
473 				if (cbp->c_nframes != n)
474 					break;
475 			}
476 			printcombo(cbp, fmtbuf);
477 			dlog(fmtbuf);
478 		}
479 		goto top;
480 	default:
481 syntax:
482 		dlog("Options are:");
483 		dlog("q    - quit");
484 		dlog("c    - continue");
485 		dlog("d#   - set debug level to #");
486 		dlog("p#   - print values at #");
487 		goto top;
488 	}
489 }
490 #endif /* DEBUG */
491 
492 /*
493  * Display debug info.
494  */
495 dlog(str)
496 	char *str;
497 {
498 
499 	if (debugfp)
500 		fprintf(debugfp, "%s\n", str);
501 	if (interactive)
502 		dislog(str);
503 	else
504 		fprintf(stderr, "%s\n", str);
505 }
506 
507 log(str)
508 	char *str;
509 {
510 
511 	if (debugfp)
512 		fprintf(debugfp, "%s\n", str);
513 	if (interactive)
514 		dislog(str);
515 	else
516 		printf("%s\n", str);
517 }
518 
519 void
520 quit()
521 {
522 	if (interactive) {
523 		bdisp();		/* show final board */
524 		cursfini();
525 	}
526 	exit(0);
527 }
528 
529 /*
530  * Die gracefully.
531  */
532 panic(str)
533 	char *str;
534 {
535 	fprintf(stderr, "%s: %s\n", prog, str);
536 	fputs("resign\n", stdout);
537 	quit();
538 }
539