14 #include "book_make.h"
\r
15 #include "book_merge.h"
\r
23 #include "mainloop.h"
\r
25 #include "move_gen.h"
\r
32 #include "xboard2uci.h"
\r
33 #include "uci2uci.h"
\r
39 static const char * const Version = "1.4.44b";
\r
40 static const char * const HelpMessage = "\
\r
42 * polyglot [configfile] [-noini] [-ec engine] [-ed enginedirectory] [-en enginename] [-log] [-lf logfile] [-hash value] [-bk book] [-pg <name>=<value>]* [-uci <name>=<value>]*\n\
\r
43 * polyglot make-book [-pgn inputfile] [-bin outputfile] [-max-ply ply] [-min-game games] [-min-score score] [-only-white] [-only-black] [-uniform]\n\
\r
44 * polyglot merge-book -in1 inputfile1 -in2 inputfile2 [-out outputfile]\n\
\r
45 * polyglot info-book [-bin inputfile] [-exact]\n\
\r
46 * polyglot dump-book [-bin inputfile] -color color [-out outputfile]\n\
\r
47 * polyglot [configfile] epd-test [engineoptions] [-epd inputfile] [-min-depth depth] [-max-depth depth] [-min-time time] [-max-time time] [-depth-delta delta]\n\
\r
48 * polyglot perft [-fen fen] [-max-depth depth]\
\r
51 static const int SearchDepth = 63;
\r
52 static const double SearchTime = 3600.0;
\r
53 static const int StringSize = 4096;
\r
61 static void init_book ();
\r
62 static void stop_search ();
\r
68 static void arg_shift_left(char **argv, int index){
\r
70 for(i=index; argv[i]!=NULL; i++){
\r
77 static void write_ini(const char *filename,
\r
78 option_list_t *pg_options,
\r
79 option_list_t *uci_options){
\r
81 char tmp[StringSize];
\r
83 f=fopen(filename,"w");
\r
85 my_fatal("write_ini(): Cannot open %s for writing.\n",filename);
\r
87 fprintf(f,"; You may edit this file to set options for the\n"
\r
88 "; UCI engine whose PolyGlot name is %s.\n"
\r
89 "; You may also safely delete this file\n"
\r
90 "; to restore the default options.\n",
\r
91 option_get_string(Option,"EngineName"));
\r
92 fprintf(f,"[PolyGlot]\n");
\r
93 option_start_iter(pg_options);
\r
94 while((opt=option_next(pg_options))){
\r
95 if(!my_string_equal(opt->value,opt->default_)&&
\r
97 (opt->mode & XBOARD)){
\r
98 snprintf(tmp,sizeof(tmp),"%s=%s\n",opt->name,opt->value);
\r
99 tmp[sizeof(tmp)-1]='\0';
\r
100 fprintf(f,"%s",tmp);
\r
103 fprintf(f,"[Engine]\n");
\r
104 option_start_iter(uci_options);
\r
105 while((opt=option_next(uci_options))){
\r
106 if(!my_string_equal(opt->value,opt->default_)&&
\r
108 snprintf(tmp,sizeof(tmp),"%s=%s\n",opt->name,opt->value);
\r
109 tmp[sizeof(tmp)-1]='\0';
\r
110 fprintf(f,"%s",tmp);
\r
118 static void write_ini_ex(const char *filename,
\r
120 ini_entry_t *entry;
\r
121 char tmp[StringSize];
\r
123 f=fopen(filename,"w");
\r
125 my_fatal("write_ini_ex(): Cannot open %s for writing.\n",filename);
\r
127 fprintf(f,"; You may edit this file to set options for the\n"
\r
128 "; UCI engine whose PolyGlot name is %s.\n"
\r
129 "; You may also safely delete this file\n"
\r
130 "; to restore the default options.\n",
\r
131 option_get_string(Option,"EngineName"));
\r
132 fprintf(f,"[PolyGlot]\n");
\r
133 ini_start_iter(ini);
\r
134 while((entry=ini_next(ini))){
\r
135 if(my_string_case_equal(entry->section,"polyglot")){
\r
136 snprintf(tmp,sizeof(tmp),"%s=%s\n",
\r
139 tmp[sizeof(tmp)-1]='\0';
\r
140 fprintf(f,"%s",tmp);
\r
143 fprintf(f,"[Engine]\n");
\r
144 ini_start_iter(ini);
\r
145 while((entry=ini_next(ini))){
\r
146 if(my_string_case_equal(entry->section,"engine")){
\r
147 snprintf(tmp,sizeof(tmp),"%s=%s\n",
\r
150 tmp[sizeof(tmp)-1]='\0';
\r
151 fprintf(f,"%s",tmp);
\r
160 int main(int argc, char * argv[]) {
\r
161 ini_t ini[1],ini_save[1];
\r
162 ini_entry_t *entry;
\r
165 bool NoIni, Persist;
\r
168 printf("PolyGlot %s by Fabien Letouzey.\n",Version);
\r
170 printf("PolyGlot %s by Fabien Letouzey (debug build).\n",Version);
\r
173 if(argc>=2 && ((my_string_case_equal(argv[1],"help")) || (my_string_case_equal(argv[1],"-help")) || (my_string_case_equal(argv[1],"--help")) || (my_string_case_equal(argv[1],"-h")) || my_string_case_equal(argv[1],"/?"))){
\r
174 printf("%s\n",HelpMessage);
\r
175 return EXIT_SUCCESS;
\r
194 ini_init(ini_save);
\r
198 if (argc >= 2 && my_string_equal(argv[1],"make-book")) {
\r
199 book_make(argc,argv);
\r
200 return EXIT_SUCCESS;
\r
203 if (argc >= 2 && my_string_equal(argv[1],"merge-book")) {
\r
204 book_merge(argc,argv);
\r
205 return EXIT_SUCCESS;
\r
208 if (argc >= 2 && my_string_equal(argv[1],"dump-book")) {
\r
209 book_dump(argc,argv);
\r
210 return EXIT_SUCCESS;
\r
213 if (argc >= 2 && my_string_equal(argv[1],"info-book")) {
\r
214 book_info(argc,argv);
\r
215 return EXIT_SUCCESS;
\r
220 if (argc >= 2 && my_string_equal(argv[1],"perft")) {
\r
221 do_perft(argc,argv);
\r
222 return EXIT_SUCCESS;
\r
225 // TODO: If logging is enabled on the command line turn it on NOW
\r
226 // and do not allow it to be overridden later.
\r
228 // What is the config file? This is very hacky right now.
\r
230 // Do we want a config file at all?
\r
234 while((arg=argv[arg_index++])){
\r
235 if(my_string_equal(arg,"-noini")){
\r
240 arg_shift_left(argv,arg_index-1);
\r
242 option_set(Option,"OptionFile","<empty>");
\r
245 // Ok see if first argument looks like config file
\r
247 if(argv[1] && !my_string_equal(argv[1],"epd-test") && !(argv[1][0]=='-')){
\r
248 // first argument must be config file
\r
250 option_set(Option,"OptionFile",argv[1]);
\r
254 arg_shift_left(argv,1);
\r
256 // Config file is the default.
\r
257 // This has already been set above or in "option_init_pg()"
\r
261 // if we use a config file: load it!
\r
263 if(!my_string_equal(option_get_string(Option,"OptionFile"),"<empty>")){
\r
264 if(ini_parse(ini,option_get_string(Option,"OptionFile"))){
\r
265 my_fatal("main(): Can't open file \"%s\": %s\n",
\r
266 option_get_string(Option,"OptionFile"),
\r
270 // remind the reader of what options are in effect
\r
272 my_log("POLYGLOG Options from ini file\n");
\r
275 // extract PG options
\r
277 ini_start_iter(ini);
\r
278 while((entry=ini_next(ini))){
\r
279 if(my_string_case_equal(entry->section,"polyglot")){
\r
280 option_set(Option,entry->name,entry->value);
\r
281 option_set_default(Option,entry->name,entry->value);
\r
285 // start logging if required
\r
287 if (option_get_bool(Option,"Log")) {
\r
288 my_log_open(option_get_string(Option,"LogFile"));
\r
291 // log welcome stuff
\r
294 my_log("PolyGlot %s by Fabien Letouzey\n",Version);
\r
296 my_log("PolyGlot %s by Fabien Letouzey (debug build)\n",Version);
\r
298 my_log("POLYGLOT *** START ***\n");
\r
299 my_log("POLYGLOT INI file \"%s\"\n",option_get_string(Option,"OptionFile"));
\r
301 // open book (presumably this should go else where)
\r
305 // scavenge command line for options necessary to start the engine
\r
308 while((arg=argv[arg_index])){
\r
309 if(my_string_equal(arg,"-ec") && argv[arg_index+1]){
\r
310 option_set(Option,"EngineCommand",argv[arg_index+1]);
\r
311 arg_shift_left(argv,arg_index);
\r
312 arg_shift_left(argv,arg_index);
\r
315 if(my_string_equal(arg,"-ed") && argv[arg_index+1]){
\r
316 option_set(Option,"EngineDir",argv[arg_index+1]);
\r
317 arg_shift_left(argv,arg_index);
\r
318 arg_shift_left(argv,arg_index);
\r
321 if(my_string_equal(arg,"-en") && argv[arg_index+1]){
\r
322 option_set(Option,"EngineName",argv[arg_index+1]);
\r
323 arg_shift_left(argv,arg_index);
\r
324 arg_shift_left(argv,arg_index);
\r
332 engine_open(Engine);
\r
333 if(!engine_active(Engine)){
\r
334 my_fatal("Could not start \"%s\"\n",option_get(Option,"EngineCommand"));
\r
337 // switch to UCI mode if necessary
\r
339 if (option_get_bool(Option,"UCI")) {
\r
340 my_log("POLYGLOT *** Switching to UCI mode ***\n");
\r
343 // initialize uci parsing and send uci command.
\r
344 // Parse options and wait for uciok
\r
346 uci_open(Uci,Engine);
\r
348 // get engine name from engine if not supplied in config file
\r
350 if (my_string_equal(option_get_string(Option,"EngineName"),"<empty>")) {
\r
351 option_set(Option,"EngineName",Uci->name);
\r
354 // what is the name of the persist file?
\r
356 if(my_string_equal(option_get_string(Option,"PersistFile"),"<empty>")){
\r
357 char tmp[StringSize];
\r
358 snprintf(tmp,sizeof(tmp),"PG_%s.ini",
\r
359 option_get_string(Option,"EngineName"));
\r
360 tmp[sizeof(tmp)-1]='\0';
\r
361 option_set(Option,"PersistFile",tmp);
\r
364 // Load the persist file
\r
366 my_log("POLYGLOT PersistFile=%s\n",option_get_string(Option,"PersistFile"));
\r
367 if(ini_parse(ini_save,option_get_string(Option,"PersistFile"))){
\r
368 my_log("POLYGLOT Unable to open PersistFile\n");
\r
371 // Do we want to use the persist file?
\r
373 entry=ini_find(ini_save,"polyglot","Persist");
\r
375 Persist=option_get_bool(Option,"Persist");
\r
377 Persist=(my_string_case_equal(entry->value,"false") ||
\r
378 my_string_equal(entry->value,"0"))?FALSE:TRUE;
\r
381 // if "Persist" now happens to be false, forget about the
\r
385 my_log("POLYGLOT Ignoring PersistFile");
\r
386 ini_clear(ini_save);
\r
389 option_set(Option,"Persist",Persist?"true":"false");
\r
391 // parse the command line and merge remaining options
\r
394 while((arg=argv[arg_index])){
\r
395 if(my_string_equal(arg,"-log")){
\r
396 ini_insert_ex(ini_save,"PolyGlot","Log","true");
\r
397 arg_shift_left(argv,arg_index);
\r
400 if(my_string_equal(arg,"-lf") && argv[arg_index+1]){
\r
401 ini_insert_ex(ini_save,"PolyGlot","LogFile",argv[arg_index+1]);
\r
402 arg_shift_left(argv,arg_index);
\r
403 arg_shift_left(argv,arg_index);
\r
406 if(my_string_equal(arg,"-hash") && argv[arg_index+1]){
\r
407 ini_insert_ex(ini_save,"Engine","Hash",argv[arg_index+1]);
\r
408 arg_shift_left(argv,arg_index);
\r
409 arg_shift_left(argv,arg_index);
\r
412 if(my_string_equal(arg,"-bk") && argv[arg_index+1]){
\r
413 ini_insert_ex(ini_save,"PolyGlot","Book","true");
\r
414 ini_insert_ex(ini_save,"PolyGlot","BookFile",argv[arg_index+1]);
\r
415 arg_shift_left(argv,arg_index);
\r
416 arg_shift_left(argv,arg_index);
\r
419 if((my_string_equal(arg,"-pg")||my_string_equal(arg,"-uci")) &&
\r
422 char section[StringSize];
\r
423 char name[StringSize];
\r
424 char value[StringSize];
\r
425 ret=ini_line_parse(argv[arg_index++],section,name,value);
\r
426 if(ret==NAME_VALUE){
\r
427 if(my_string_equal(arg,"-pg")){
\r
428 ini_insert_ex(ini_save,"PolyGlot",name,value);
\r
430 ini_insert_ex(ini_save,"Engine",name,value);
\r
433 arg_shift_left(argv,arg_index);
\r
434 arg_shift_left(argv,arg_index);
\r
440 // remind the reader once again about options
\r
442 my_log("POLYGLOG Options from PersistFile and command line\n");
\r
443 ini_disp(ini_save);
\r
445 // extract PG options; this time do not set the default
\r
446 // polyglot_set_option() performs the necessary actions such
\r
447 // as opening the log file/opening book etcetera.
\r
449 ini_start_iter(ini_save);
\r
450 while((entry=ini_next(ini_save))){
\r
451 if(my_string_case_equal(entry->section,"polyglot")){
\r
452 polyglot_set_option(entry->name,entry->value);
\r
456 // done initializing
\r
460 // collect engine options from config file(s) and send to engine
\r
462 ini_start_iter(ini);
\r
463 while((entry=ini_next(ini))){
\r
464 if(my_string_case_equal(entry->section,"engine")){
\r
465 // also updates value in Uci->option
\r
466 uci_send_option(Uci,entry->name,"%s",entry->value);
\r
467 // since this comes from the ini file, also update default
\r
468 option_set_default(Uci->option,entry->name,entry->value);
\r
469 // this is inherited, it probably does not work correctly
\r
470 if(my_string_case_equal(entry->name,"MultiPV") &&
\r
471 atoi(entry->value)>1){
\r
472 Uci->multipv_mode=TRUE;
\r
476 ini_start_iter(ini_save);
\r
477 while((entry=ini_next(ini_save))){
\r
478 if(my_string_case_equal(entry->section,"engine")){
\r
479 // also updates value in Uci->option
\r
480 uci_send_option(Uci,entry->name,"%s",entry->value);
\r
481 // this is inherited, it probably does not work correctly
\r
482 if(my_string_case_equal(entry->name,"MultiPV") &&
\r
483 atoi(entry->value)>1){
\r
484 Uci->multipv_mode=TRUE;
\r
492 if (argv[1] && my_string_equal(argv[1],"epd-test")){
\r
494 while((arg=argv[argc++]));
\r
495 epd_test(argc-1,argv);
\r
496 return EXIT_SUCCESS;
\r
499 // Anything that hasn't been parsed yet is a syntax error
\r
502 my_fatal("main(): Unknown option: %s\n",argv[1]);
\r
508 return EXIT_SUCCESS;
\r
511 // polyglot_set_option()
\r
513 void polyglot_set_option(const char *name, const char *value){ // this must be cleaned up!
\r
515 my_log("POLYGLOT Setting PolyGlot option %s=\"%s\"\n",name,value);
\r
516 if(my_string_case_equal(name,"Defaults")){
\r
517 option_start_iter(Uci->option);
\r
518 while((opt=option_next(Uci->option))){
\r
519 if(!IS_BUTTON(opt)){
\r
520 // also sets opt->value
\r
521 uci_send_option(Uci,opt->name,opt->default_);
\r
524 option_start_iter(Option);
\r
525 while((opt=option_next(Option))){
\r
526 if(!IS_BUTTON(opt)){
\r
527 polyglot_set_option(opt->name,opt->default_);
\r
530 xboard2uci_send_options();
\r
532 option_set(Option,name,value);
\r
533 if(option_get_bool(Option,"Book")&&(my_string_case_equal(name,"BookFile")||my_string_case_equal(name,"Book"))){
\r
534 my_log("POLYGLOT *** SETTING BOOK ***\n");
\r
535 my_log("POLYGLOT BOOK \"%s\"\n",option_get_string(Option,"BookFile"));
\r
538 book_open(option_get_string(Option,"BookFile"));
\r
539 if(!book_is_open()){
\r
540 my_log("POLYGLOT Unable to open book \"%s\"\n",option_get_string(Option,"BookFile"));
\r
542 }else if(option_get_bool(Option,"Log")&&(my_string_case_equal(name,"LogFile") ||my_string_case_equal(name,"Log"))){
\r
543 my_log("POLYGLOT *** SWITCHING LOGFILE ***\n");
\r
544 my_log("POLYGLOT LOGFILE \"%s\"\n",option_get_string(Option,"LogFile"));
\r
546 my_log_open(option_get_string(Option,"LogFile"));
\r
547 }else if(option_get_bool(Option,"UseNice") &&(my_string_case_equal(name,"NiceValue")||my_string_case_equal(name,"UseNice"))){
\r
548 my_log("POLYGLOT Adjust Engine Piority\n");
\r
549 engine_set_nice_value(Engine,atoi(option_get_string(Option,"NiceValue")));
\r
550 }else if(my_string_case_equal(name,"Book") && !option_get_bool(Option,"Book")){
\r
553 }else if(my_string_case_equal(name,"UseNice") && !option_get_bool(Option,"UseNice")){
\r
554 my_log("POLYGLOT Adjust Engine Piority\n");
\r
555 engine_set_nice_value(Engine,0);
\r
556 }else if(my_string_case_equal(name,"Log") && !option_get_bool(Option,"Log")){
\r
557 my_log("POLYGLOT QUIT LOGGING\n");
\r
565 static void init_book(){
\r
567 if (option_get_bool(Option,"Book")){
\r
568 my_log("POLYGLOT *** SETTING BOOK ***\n");
\r
569 my_log("POLYGLOT BOOK \"%s\"\n",option_get_string(Option,"BookFile"));
\r
570 book_open(option_get_string(Option,"BookFile"));
\r
571 if(!book_is_open()){
\r
572 my_log("POLYGLOT Unable to open book \"%s\"\n",
\r
573 option_get_string(Option,"BookFile"));
\r
587 my_log("POLYGLOT *** QUIT ***\n");
\r
592 engine_send(Engine,"quit");
\r
593 my_log("POLYGLOT Closing engine\n");
\r
594 engine_close(Engine);
\r
597 // printf("def=%s val=%s\n",option_get_default(Option,"Persist"),option_get_string(Option,"Persist"));
\r
598 if(option_get_bool(Option,"Persist")){
\r
599 write_ini(option_get_string(Option,"PersistFile"),
\r
600 Option,Uci->option);
\r
601 }else if(!my_string_case_equal(option_get_default(Option,"Persist"),
\r
602 option_get_string(Option,"Persist"))){
\r
604 ini_insert_ex(empty,"PolyGlot","Persist","false");
\r
605 write_ini_ex(option_get_string(Option,"PersistFile"),empty);
\r
607 write_ini_ex(option_get_string(Option,"PersistFile"),empty);
\r
609 my_log("POLYGLOT Calling exit\n");
\r
610 exit(EXIT_SUCCESS);
\r
615 static void stop_search() {
\r
617 if (Init && Uci->searching) {
\r
619 ASSERT(Uci->searching);
\r
620 ASSERT(Uci->pending_nb>=1);
\r
622 my_log("POLYGLOT STOP SEARCH\n");
\r
624 if (option_get_bool(Option,"SyncStop")) {
\r
625 uci_send_stop_sync(Uci);
\r
627 uci_send_stop(Uci);
\r