#include <clib/alib_protos.h>
#include <stdio.h>
#include <time.h>
#include <string.h>
#include <ctype.h>
#include <time.h>
#include <errno.h>

#include "common.h"
#include "book.h"

void PGNSaveToFile(const char *file,const char *resultstr)
{
FILE *fp;
char s[100];
int len;
char *p;
short i;
time_t secs;
struct tm *timestruct;
if(!(fp=fopen(file,"a")))
	{
	sprintf(s,"Cannot write to file %s.",file);
	DoMethod(mui_app,MUIM_Chess_ShowThinking,s);
	return;
	}
fprintf(fp,"[Event \"\"]\n");
fprintf(fp,"[Site \"\"]\n");
secs=time(0);
timestruct=localtime((time_t *)&secs);
fprintf(fp,"[Date \"%4d.%02d.%02d\"]\n",timestruct->tm_year+1900,timestruct->tm_mon+1,timestruct->tm_mday);
fprintf(fp,"[Round \"\"]\n");
if(computerplays==white)  fprintf(fp,"[White \"GNUChess 5.09\"]\n");
else fprintf(fp,"[White \"%s\"]\n",name);
if(computerplays==black) fprintf(fp,"[Black \"GNUChess 5.09\"]\n");
else fprintf(fp,"[Black \"%s\"]\n",name);
fprintf(fp,"[WhiteELO \"%d\"]\n",computer==white?myrating:opprating);
fprintf(fp,"[BlackELO \"%d\"]\n",computer==white?opprating:myrating);
len=strcspn(resultstr," {");
fprintf(fp,"[Result \"%.*s\"]\n\n",len,resultstr);
s[0]=0;
for(i=0;i<=GameCnt;i+=2)
	{
	if(i==GameCnt) sprintf(s,"%s%d. %s ",s,i/2+1,Game[i].SANmv);
	else sprintf(s,"%s%d. %s %s ",s,i/2+1,Game[i].SANmv,Game[i+1].SANmv);
	if(strlen(s)>80)
		{
		p=s+79;
		while(*p--!=' ');
		*++p=0;
		fprintf(fp,"%s\n",s);
		strcpy(s,p+1);
		}
	}
fprintf(fp,"%s%s\n\n",s,resultstr);
fclose(fp);
}      

void PGNReadFromFile(const char *file)
{
FILE *fp;
char s[100],c,wmv[8],bmv[8],text[100];
int moveno;
leaf *p;
if(!(fp=fopen(file,"r")))
	{
	sprintf(text,"Cannot open file %s.",file);
	DoMethod(mui_app,MUIM_Chess_ShowThinking,text);
	return;
	}
DoMethod(mui_app,MUIM_Chess_ClearList);
do
	{
	if((c=fgetc(fp))=='[') fgets(s,100,fp);
	}
while(c=='[');
ungetc(c,fp);
InitVars();
while(!feof(fp))
	{
	int res;
	c=fgetc(fp);
	if(c=='*') break;
	ungetc(c,fp);
	res=fscanf(fp,"%d. %s %s ",&moveno,wmv,bmv);
	if(!(p=ValidateMove(wmv)))
		{
		sprintf(text,"Illegal move %d. %s",moveno,wmv);
		DoMethod(mui_app,MUIM_Chess_ShowThinking,text);
		DoMethod(mui_app,MUIM_Chess_ShowBoard);
		DoMethod(mui_app,MUIM_Chess_Side);
		break;
		}
	MakeMove(white,&p->move);
	strcpy(Game[GameCnt].SANmv,wmv);
	sprintf(text,"%d: %-5s",moveno,wmv);
	DoMethod(mui_app,MUIM_Chess_AddMove,white,text);
	if(res==2||bmv[0]=='*') break;
	if(!(p=ValidateMove(bmv)))
		{
		sprintf(text,"Illegal move %d. ... %s",moveno,bmv);
		DoMethod(mui_app,MUIM_Chess_ShowThinking,text);
		DoMethod(mui_app,MUIM_Chess_ShowBoard);
		DoMethod(mui_app,MUIM_Chess_Side);
		break;
		}
	MakeMove(black,&p->move);
	strcpy(Game[GameCnt].SANmv,bmv);
	sprintf(text,"%d: %-5s",moveno,bmv);
	DoMethod(mui_app,MUIM_Chess_AddMove,black,text);
	}
fclose(fp);
DoMethod(mui_app,MUIM_Chess_ShowBoard);
DoMethod(mui_app,MUIM_Chess_Side);
TTClear();
}

void BookPGNReadFromFile(const char *file)
{
FILE *fp;
char s[100],wmv[8],bmv[8];
int c;
unsigned int i;
char header[2000];
int moveno,result,ngames=0;
leaf *p;
time_t t1,t2;
double et;
short examinecolor,playerfound[2];

const char *const player[]={
"Alekhine",
"Adams",
"Anand",
"Anderssen",
"Andersson",
"Aronin",
"Averbakh",
"Balashov",
"Beliavsky",
"Benko",
"Bernstein",
"Bird",
"Bogoljubow",
"Bolbochan",
"Boleslavsky",
"Byrne",
"Botvinnik",
"Bronstein",
"Browne",
"Capablanca",
"Chigorin",
"Christiansen",
"De Firmian",
"Deep Blue",
"Deep Thought",
"Donner",
"Dreev",
"Duras",
"Euwe",
"Evans",
"Filip",
"Fine",
"Fischer",
"Flohr",
"Furman",
"Gelfand",
"Geller",
"Gereben",
"Glek",
"Gligoric",
"GNU",
"Golombek",
"Gruenfeld",
"Guimard",
"Hodgson",
"Ivanchuk",
"Ivkov",
"Janowsky",
"Kamsky",
"Kan",
"Karpov",
"Kasparov",
"Keres",
"Korchnoi",
"Kortschnoj",
"Kotov",
"Kramnik",
"Kupreich",
"Lasker",
"Lautier",
"Letelier",
"Lilienthal",
"Ljubojevic",
"Marshall",
"Maroczy",
"Mieses",
"Miles",
"Morphy",
"Mueller",
"Nimzowitsch",
"Nunn",
"Opocensky",
"Pachman",
"Petrosian",
"Piket",
"Pilnik",
"Pirc",
"Polgar",
"Portisch",
"Psakhis",
"Ragozin",
"Reshevsky",
"Reti",
"Romanish",
"Rubinstein",
"Saemisch",
"Seirawan",
"Shirov",
"Short",
"Silman",
"Smyslov",
"Sokolsky",
"Spassky",
"Sveshnikov",
"Stahlberg",
"Steinitz",
"Tal",
"Tarjan",
"Tartakower",
"Timman",
"Topalov",
"Torre",
"Vidmar"
};

et=0.0;
t1=time(0);
result=-1;
if(!(fp=fopen(file,"r")))
	{
	sprintf(s,"Cannot open file %s.",file);
	DoMethod(mui_app,MUIM_Chess_ShowThinking,s);
	return;
	}
if(BookBuilderOpen()!=BOOK_SUCCESS) return;
newpos=existpos=0;

nextgame:
header[0]=0;
InitVars();
NewPosition();
myrating=opprating=0;
playerfound[black]=playerfound[white]=0;
while((c=fgetc(fp))!=EOF)
	{
	if(c!=' '&&c!='\t'&&c!='\n'&&c!='\r')
		{
		ungetc(c,fp);
		break;
		}
	}
while((c=fgetc(fp))=='[')
	{
	ungetc(c,fp);
	fgets(s,100,fp);
	strcat(header,s);
	if(strncmp(s+1,"White ",6)==0)
		{
		examinecolor=white;
		ngames++;
		}
	else if(strncmp(s+1,"Black ",6)==0) examinecolor=black;
	else if(strncmp(s+1,"Result",6)==0)
		{
		if(strncmp(s+7," \"1-0",5)==0) result=R_WHITE_WINS;
		else if(strncmp(s+7," \"0-1",5)==0) result=R_BLACK_WINS;
		else if(strncmp(s+7," \"1/2-1/2",9)==0) result=R_DRAW;
		else result=R_NORESULT;
		continue;
		}
	else continue;
	for(i=0;i<(sizeof(player)/sizeof(*player));i++)
		{
		if(strstr(s+7,player[i])!=NULL) playerfound[examinecolor]=1;
		}
	}
ungetc(c,fp);
while(1)
	{
	if(fscanf(fp,"%d. %7s ",&moveno,wmv)<2) break;
	if(wmv[0]=='1'||wmv[0]=='['||wmv[0]=='*'||strcmp(wmv,"0-1")==0) break;
	p=ValidateMove(wmv);
	if(!p)
		{
		puts(header);
/*
		ShowBoard();
		printf("Illegal move %d. %s\n",moveno,wmv);
*/
		break;
		}
	MakeMove(white,&p->move);
	if(playerfound[white])
		{
		if(BookBuilder(result,white)!=BOOK_SUCCESS) break;
		}
	strcpy(Game[GameCnt].SANmv,wmv);
	if(fscanf(fp,"%7s ",bmv)<1) break;
	if(bmv[0]=='1'||bmv[0]=='['||bmv[0]=='*'||strcmp(bmv,"0-1")==0) break;
	p=ValidateMove(bmv);
	if(!p)
		{
/*
		puts(header);
		ShowBoard();
		printf("Illegal move %d. ... %s\n",moveno,bmv);
*/
		break;
		}
	MakeMove(black,&p->move);
	if(playerfound[black])
		{
		if(BookBuilder(result,black)!=BOOK_SUCCESS) break;
		}
	strcpy(Game[GameCnt].SANmv,bmv);
	}
while(!feof(fp))
	{
	fgets(s,100,fp);
	if(s[0]=='\n') break;
	}
if(!feof(fp))
	{
	if(ngames%10==0)
		{
		char t[20];
		sprintf(t,"%d Games",ngames);
		DoMethod(mui_app,MUIM_Chess_ShowThinking,t);
		}
	goto nextgame;
	}
fclose(fp);
if(BookBuilderClose()!=BOOK_SUCCESS) DoMethod(mui_app,MUIM_Chess_ShowThinking,"Error writing opening book.");
header[0]=0;
InitVars();
NewPosition();
myrating=opprating=0;
t2=time(0);
et+=difftime(t2,t1);
if(et<0.5) et=1.0;

sprintf(s,"Finished. %d positions added.",newpos);
DoMethod(mui_app,MUIM_Chess_ShowThinking,s);

/*
printf("Time = %.0f seconds\n",et);
printf("Games compiled: %d\n",ngames);
printf("Games per second: %f\n",ngames/et);
printf("Positions scanned: %d\n",newpos+existpos);
printf("Positions per second: %f\n",(newpos+existpos)/et);
printf("New & unique added: %d positions\n",newpos);
printf("Duplicates not added: %d positions\n",existpos);
*/
}