pisqpipe.cpp

/** functions that communicate with manager through pipes */
/** don't modify this file */

#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <stdarg.h>
#include "pisqpipe.h"

int width, height; /* the board size */
int info_timeout_turn=30000; /* time for one turn in milliseconds */
int info_timeout_match=1000000000; /* total time for a game */
int info_time_left=1000000000; /* left time for a game */
int info_max_memory=0; /* maximum memory in bytes, zero if unlimited */
int info_game_type=1; /* 0:human opponent, 1:AI opponent, 2:tournament, 3:network tournament */
int info_exact5=0; /* 0:five or more stones win, 1:exactly five stones win */
int info_renju=0; /* 0:gomoku, 1:renju */
int info_continuous=0; /* 0:single game, 1:continuous */
int terminateAI; /* return from brain_turn when terminateAI>0 */
unsigned start_time; /* tick count at the beginning of turn */
char dataFolder[256]; /* folder for persistent files */

static char cmd[256];
static HANDLE event1, event2;


/** write a line to STDOUT */
int pipeOut(char *fmt, ...)
{
	int i;
	va_list va;
	va_start(va, fmt);
	i=vprintf(fmt, va);
	putchar('\n');
	fflush(stdout);
	va_end(va);
	return i;
}

/** read a line from STDIN */
static void get_line()
{
	int c, bytes;

	bytes=0;
	do{
		c=getchar();
		if(c==EOF) exit(0);
		if(bytes<sizeof(cmd)) cmd[bytes++]=(char)c;
	} while(c!='\n');
	cmd[bytes-1]=0;
	if(cmd[bytes-2]=='\r') cmd[bytes-2]=0;
}


/** parse coordinates x,y */
static int parse_coord(const char *param, int *x, int *y)
{
	if(sscanf(param, "%d,%d", x, y)!=2 ||
		*x<0 || *y<0 || *x>=width || *y>=height){
		return 0;
	}
	return 1;
}

/** parse coordinates x,y and player number z */
static void parse_3int_chk(const char *param, int *x, int *y, int *z)
{
	if(sscanf(param, "%d,%d,%d", x, y, z)!=3 || *x<0 || *y<0 ||
		*x>=width || *y>=height) *z=0;
}

/** return pointer to word after command if input starts with command, otherwise return NULL */
static const char *get_cmd_param(const char *command, const char *input)
{
	int n1, n2;
	n1=(int)strlen(command);
	n2=(int)strlen(input);
	if(n1>n2 || _strnicmp(command, input, n1)) return NULL; /* it is not command */
	input+=strlen(command);
	while(isspace(input[0])) input++;
	return input;
}

/** send suggest */
void suggest(int x, int y)
{
	pipeOut("SUGGEST %d,%d", x, y);
}

/** write move to the pipe and update internal data structures */
void do_mymove(int x, int y)
{
	brain_my(x, y);
	pipeOut("%d,%d", x, y);
}

/** main function for the working thread */
static DWORD WINAPI threadLoop(LPVOID)
{
	for(;;){
		WaitForSingleObject(event1, INFINITE);
		brain_turn();
		SetEvent(event2);
	}
}

/** start thinking */
static void turn()
{
	terminateAI=0;
	ResetEvent(event2);
	SetEvent(event1);
}

/** stop thinking */
static void stop()
{
	terminateAI=1;
	WaitForSingleObject(event2, INFINITE);
}

static void start()
{
	start_time=GetTickCount();
	stop();
	if(!width){
		width=height=20;
		brain_init();
	}
}

/** do command cmd */
static void do_command()
{
	const char *param;
	const char *info;
	char *t;
	int x, y, who, e;

	if((param=get_cmd_param("info", cmd))!=0) {
		if((info=get_cmd_param("max_memory", param))!=0) info_max_memory=atoi(info);
		if((info=get_cmd_param("timeout_match", param))!=0) info_timeout_match=atoi(info);
		if((info=get_cmd_param("timeout_turn", param))!=0) info_timeout_turn=atoi(info);
		if((info=get_cmd_param("time_left", param))!=0) info_time_left=atoi(info);
		if((info=get_cmd_param("game_type", param))!=0) info_game_type=atoi(info);
		if((info=get_cmd_param("rule", param))!=0){ e=atoi(info); info_exact5=e&1; info_continuous=(e>>1)&1; info_renju=(e>>2)&1; }
		if((info=get_cmd_param("folder", param))!=0) strncpy(dataFolder, info, sizeof(dataFolder)-1);
#ifdef DEBUG_EVAL
		if((info=get_cmd_param("evaluate", param))!=0){ if(parse_coord(info, &x, &y)) brain_eval(x, y); }
#endif
		/* unknown info is ignored */
	}
	else if((param=get_cmd_param("start", cmd))!=0) {
		if(sscanf(param, "%d", &width)!=1 || width<5){
			width=0;
			pipeOut("ERROR bad START parameter");
		}
		else{
			height=width;
			start();
			brain_init();
		}
	}
	else if((param=get_cmd_param("rectstart", cmd))!=0) {
		if(sscanf(param, "%d ,%d", &width, &height)!=2 || width<5 || height<5){
			width=height=0;
			pipeOut("ERROR bad RECTSTART parameters");
		}
		else{
			start();
			brain_init();
		}
	}
	else if((param=get_cmd_param("restart", cmd))!=0) {
		start();
		brain_restart();
	}
	else if((param=get_cmd_param("turn", cmd))!=0) {
		start();
		if(!parse_coord(param, &x, &y)){
			pipeOut("ERROR bad coordinates");
		}
		else{
			brain_opponents(x, y);
			turn();
		}
	}
	else if((param=get_cmd_param("play", cmd))!=0) {
		start();
		if(!parse_coord(param, &x, &y)){
			pipeOut("ERROR bad coordinates");
		}
		else{
			do_mymove(x, y);
		}
	}
	else if((param=get_cmd_param("begin", cmd))!=0) {
		start();
		turn();
	}
	else if((param=get_cmd_param("about", cmd))!=0) {
#ifdef ABOUT_FUNC
		brain_about();
#else
		pipeOut("%s", infotext);
#endif
	}
	else if((param=get_cmd_param("end", cmd))!=0) {
		stop();
		brain_end();
		exit(0);
	}
	else if((param=get_cmd_param("board", cmd))!=0) {
		start();
		for(;;){ /* fill the whole board */
			get_line();
			parse_3int_chk(cmd, &x, &y, &who);
			if(who==1) brain_my(x, y);
			else if(who==2) brain_opponents(x, y);
			else if(who==3) brain_block(x, y);
			else{
				if(_stricmp(cmd, "done")) pipeOut("ERROR x,y,who or DONE expected after BOARD");
				break;
			}
		}
		turn();
	}
	else if((param=get_cmd_param("takeback", cmd))!=0) {
		start();
		t="ERROR bad coordinates";
		if(parse_coord(param, &x, &y)){
			e= brain_takeback(x, y);
			if(e==0) t="OK";
			else if(e==1) t="UNKNOWN";
		}
		pipeOut(t);
	}
	else{
		pipeOut("UNKNOWN command");
	}
}


/** main function for AI console application  */
int main()
{
	DWORD mode;
	if(GetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), &mode))
		puts("MESSAGE Gomoku AI should not be started directly. Please install gomoku manager (http://sourceforge.net/projects/piskvork). Then enter path to this exe file in players settings.");

#ifdef DEBUG
	SetErrorMode(0);
#endif
	DWORD tid;
	event1=CreateEvent(0, FALSE, FALSE, 0);
	CreateThread(0, 0, threadLoop, 0, 0, &tid);
	event2=CreateEvent(0, TRUE, TRUE, 0);
	for(;;){
		get_line();
		do_command();
	}
}