version 1.4.33b
[polyglot.git] / main.c
1 \r
2 // main.c\r
3 \r
4 // includes\r
5 \r
6 #include <errno.h>\r
7 #include <stdio.h>\r
8 #include <stdlib.h>\r
9 #include <string.h>\r
10 \r
11 #include "attack.h"\r
12 #include "board.h"\r
13 #include "book.h"\r
14 #include "book_make.h"\r
15 #include "book_merge.h"\r
16 #include "engine.h"\r
17 #include "epd.h"\r
18 #include "fen.h"\r
19 #include "gui.h"\r
20 #include "hash.h"\r
21 #include "list.h"\r
22 #include "main.h"\r
23 #include "mainloop.h"\r
24 #include "move.h"\r
25 #include "move_gen.h"\r
26 #include "option.h"\r
27 #include "piece.h"\r
28 #include "search.h"\r
29 #include "square.h"\r
30 #include "uci.h"\r
31 #include "util.h"\r
32 #include "xboard2uci.h"\r
33 #include "uci2uci.h"\r
34 \r
35 // constants\r
36 \r
37 \r
38 static const char * const Version = "1.4.33b";\r
39 static const char * const HelpMessage = "\\r
40 SYNTAX\n\\r
41 * polyglot [configfile]\n\\r
42 * polyglot -ec enginecommand\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 [-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
49 ";\r
50 \r
51 static const int SearchDepth = 63;\r
52 static const double SearchTime = 3600.0;\r
53 static const int StringSize = 4096;\r
54 \r
55 // variables\r
56 \r
57 static bool Init;\r
58 \r
59 // prototypes\r
60 \r
61 static void parse_option ();\r
62 static void init_book ();\r
63 static bool parse_line   (char line[], char * * name_ptr, char * * value_ptr);\r
64 static void stop_search  ();\r
65 \r
66 // functions\r
67 \r
68 // main()\r
69 \r
70 int main(int argc, char * argv[]) {\r
71 \r
72     if(!DEBUG){\r
73         printf("PolyGlot %s by Fabien Letouzey.\n",Version);\r
74     }else{\r
75         printf("PolyGlot %s by Fabien Letouzey (debug build).\n",Version);\r
76     }\r
77 \r
78     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
79         printf("%s\n",HelpMessage);\r
80         return EXIT_SUCCESS;\r
81     }\r
82 \r
83    // init\r
84 \r
85     Init = FALSE;\r
86 \r
87     util_init();\r
88     option_init();\r
89     \r
90     square_init();\r
91     piece_init();\r
92     attack_init();\r
93     \r
94     hash_init();\r
95     \r
96     my_random_init();\r
97 \r
98         // build book\r
99     \r
100     if (argc >= 2 && my_string_equal(argv[1],"make-book")) {\r
101         book_make(argc,argv);\r
102         return EXIT_SUCCESS;\r
103     }\r
104     \r
105     if (argc >= 2 && my_string_equal(argv[1],"merge-book")) {\r
106         book_merge(argc,argv);\r
107         return EXIT_SUCCESS;\r
108     }\r
109 \r
110        if (argc >= 2 && my_string_equal(argv[1],"merge-book")) {\r
111       book_merge(argc,argv);\r
112       return EXIT_SUCCESS;\r
113    }\r
114 \r
115    if (argc >= 2 && my_string_equal(argv[1],"dump-book")) {\r
116       book_dump(argc,argv);\r
117       return EXIT_SUCCESS;\r
118    }\r
119 \r
120    if (argc >= 2 && my_string_equal(argv[1],"info-book")) {\r
121       book_info(argc,argv);\r
122       return EXIT_SUCCESS;\r
123    }\r
124     \r
125     if (argc >= 2 && my_string_equal(argv[1],"perft")) {\r
126         do_perft(argc,argv);\r
127         return EXIT_SUCCESS;\r
128     }\r
129     \r
130     if (argc >= 3 && my_string_equal(argv[1],"-ec")) {\r
131         option_set("EngineCommand",argv[2]);\r
132         engine_open(Engine);\r
133         if(!engine_active(Engine)){\r
134             my_fatal("Could not start \"%s\"\n",option_get("EngineCommand"));\r
135         }\r
136         Init=TRUE;\r
137         gui_init(GUI);\r
138         uci_open(Uci,Engine);\r
139         if (my_string_equal(option_get_string("EngineName"),"<empty>")) {\r
140             option_set("EngineName",Uci->name);\r
141         }\r
142         mainloop();\r
143         return EXIT_SUCCESS; \r
144     }\r
145     \r
146         // read options\r
147     \r
148     if (argc == 2) option_set("OptionFile",argv[1]); // HACK for compatibility\r
149 \r
150     parse_option(); // HACK: also launches the engine\r
151     \r
152         // EPD test\r
153     \r
154     if (argc >= 2 && my_string_equal(argv[1],"epd-test")){\r
155         epd_test(argc,argv);\r
156         return EXIT_SUCCESS;\r
157     }else if(argc >= 3 && my_string_equal(argv[2],"epd-test")){\r
158         epd_test(argc-1,argv+1);\r
159         return EXIT_SUCCESS;\r
160     }\r
161     \r
162     if (argc >= 3) my_fatal("Too many arguments\n");\r
163 \r
164 \r
165     init_book();\r
166     gui_init(GUI);\r
167     mainloop();\r
168     return EXIT_SUCCESS; \r
169 }\r
170 \r
171 // polyglot_set_option\r
172 \r
173 void polyglot_set_option(char *name, char *value){ // this must be cleaned up!\r
174     option_set(name,value);\r
175     if(option_get_bool("Book")&&(my_string_case_equal(name,"BookFile")||my_string_case_equal(name,"Book"))){\r
176         my_log("POLYGLOT *** SETTING BOOK ***\n");\r
177         my_log("POLYGLOT BOOK \"%s\"\n",option_get_string("BookFile"));\r
178         book_close();\r
179         book_clear();\r
180         book_open(option_get_string("BookFile"));\r
181         if(!book_is_open()){\r
182             my_log("POLYGLOT Unable to open book \"%s\"\n",option_get_string("BookFile"));\r
183         }\r
184     }else if(option_get_bool("Log")&&(my_string_case_equal(name,"LogFile") ||my_string_case_equal(name,"Log"))){\r
185         my_log("POLYGLOT *** SETTING LOGFILE ***\n");\r
186         my_log("POLYGLOT LOGFILE \"%s\"\n",option_get_string("LogFile"));\r
187         my_log_close();\r
188         my_log_open(option_get_string("LogFile"));\r
189     }else if(option_get_bool("UseNice") &&(my_string_case_equal(name,"NiceValue")||my_string_case_equal(name,"UseNice"))){\r
190         my_log("POLYGLOT Adjust Engine Piority\n");\r
191         engine_set_nice_value(Engine,atoi(option_get_string("NiceValue")));\r
192     }else if(my_string_case_equal(name,"Book") && !option_get_bool("Book")){\r
193         book_close();\r
194         book_clear();\r
195     }else if(my_string_case_equal(name,"UseNice") && !option_get_bool("UseNice")){\r
196         my_log("POLYGLOT Adjust Engine Piority\n");\r
197         engine_set_nice_value(Engine,0);\r
198     }else if(my_string_case_equal(name,"Log") && !option_get_bool("Log")){\r
199         my_log("POLYGLOT QUIT LOGGING\n");\r
200         my_log_close();\r
201     }\r
202 }\r
203 \r
204 \r
205 // init_book()\r
206 \r
207 static void init_book(){\r
208     book_clear();\r
209     if (option_get_bool("Book")){\r
210         my_log("POLYGLOT *** SETTING BOOK ***\n");\r
211         my_log("POLYGLOT BOOK \"%s\"\n",option_get_string("BookFile"));\r
212         book_open(option_get_string("BookFile"));\r
213         if(!book_is_open()){\r
214             my_log("POLYGLOT Unable to open book \"%s\"\n",\r
215                    option_get_string("BookFile"));\r
216         }\r
217     }\r
218 }\r
219 \r
220 // parse_option()\r
221 \r
222 static void parse_option() {\r
223 \r
224     const char * file_name;\r
225     FILE * file;\r
226     char line[256];\r
227     char * name, * value;\r
228     file_name = option_get_string("OptionFile");\r
229     \r
230     file = fopen(file_name,"r");\r
231     if (file == NULL) {\r
232         my_fatal("Can't open file \"%s\": %s\n",file_name,strerror(errno));\r
233     }\r
234     \r
235         // PolyGlot options (assumed first)\r
236     \r
237    while (TRUE) {\r
238        \r
239        if (!my_file_read_line(file,line,256)) {\r
240            my_fatal("parse_option(): missing [Engine] section\n");\r
241        }\r
242        \r
243        if (my_string_case_equal(line,"[engine]")) break;\r
244        \r
245        if (parse_line(line,&name,&value)) {\r
246            option_set(name,value);\r
247            option_set_default(name,value);\r
248        }\r
249    }\r
250    \r
251    if (option_get_bool("Log")) {\r
252        my_log_open(option_get_string("LogFile"));\r
253    }\r
254    \r
255    if(!DEBUG){\r
256        my_log("PolyGlot %s by Fabien Letouzey\n",Version);\r
257    }else{\r
258        my_log("PolyGlot %s by Fabien Letouzey (debug build)\n",Version);\r
259    }\r
260 \r
261    my_log("POLYGLOT *** START ***\n");\r
262    my_log("POLYGLOT INI file \"%s\"\n",file_name);\r
263    engine_open(Engine);\r
264    if(!engine_active(Engine)){\r
265        my_fatal("Could not start \"%s\"\n",option_get("EngineCommand"));\r
266    }\r
267 \r
268    if (option_get_bool("UCI")) {\r
269        my_log("POLYGLOT *** Switching to UCI mode ***\n");\r
270    }\r
271    uci_open(Uci,Engine);\r
272    Init = TRUE;\r
273    while (my_file_read_line(file,line,256)) {\r
274        if (line[0] == '[') my_fatal("parse_option(): unknown section %s\n",line);\r
275        if (parse_line(line,&name,&value)) {\r
276            uci_send_option(Uci,name,"%s",value);\r
277                //to get a decent display in winboard_x we need to now if an engine really is doing multipv analysis\r
278                // "multipv 1" in the pv is meaningless,f.i. toga sends that all the time\r
279                //therefore check if MultiPV is set to a decent value in the polyglot ini file\r
280            if(my_string_case_equal(name,"MultiPV") && atoi(value)>1)  Uci->multipv_mode=TRUE;\r
281        }\r
282    }\r
283    if (my_string_equal(option_get_string("EngineName"),"<empty>")) {\r
284        option_set("EngineName",Uci->name);\r
285    }\r
286    \r
287    fclose(file);\r
288 }\r
289 \r
290 // parse_line()\r
291 \r
292 static bool parse_line(char line[], char * * name_ptr, char * * value_ptr) {\r
293     \r
294     char * ptr;\r
295     char * name, * value;\r
296     \r
297     ASSERT(line!=NULL);\r
298     ASSERT(name_ptr!=NULL);\r
299     ASSERT(value_ptr!=NULL);\r
300     \r
301         // remove comments\r
302     \r
303     ptr = strchr(line,';');\r
304     if (ptr != NULL) *ptr = '\0';\r
305     \r
306     ptr = strchr(line,'#');\r
307     if (ptr != NULL) *ptr = '\0';\r
308     \r
309         // split at '='\r
310     \r
311     ptr = strchr(line,'=');\r
312     if (ptr == NULL) return FALSE;\r
313     \r
314     name = line;\r
315     value = ptr+1;\r
316    \r
317         // cleanup name\r
318     \r
319     while (*name == ' ') name++; // remove leading spaces\r
320     \r
321     while (ptr > name && ptr[-1] == ' ') ptr--; // remove trailing spaces\r
322     *ptr = '\0';\r
323     \r
324     if (*name == '\0') return FALSE;\r
325     \r
326         // cleanup value\r
327     \r
328     ptr = &value[strlen(value)]; // pointer to string terminator\r
329     \r
330     while (*value == ' ') value++; // remove leading spaces\r
331     \r
332     while (ptr > value && ptr[-1] == ' ') ptr--; // remove trailing spaces\r
333     *ptr = '\0';\r
334     \r
335     if (*value == '\0') return FALSE;\r
336    \r
337         // end\r
338     \r
339     *name_ptr = name;\r
340     *value_ptr = value;\r
341     \r
342     return TRUE;\r
343 }\r
344 \r
345 // quit()\r
346 \r
347 void quit() {\r
348 \r
349     my_log("POLYGLOT *** QUIT ***\n");\r
350     \r
351     if (Init) {\r
352         \r
353         stop_search();\r
354         engine_send(Engine,"quit");\r
355         my_log("POLYGLOT Closing engine\n");\r
356         engine_close(Engine);\r
357         \r
358     }\r
359     my_log("POLYGLOT Calling exit\n");\r
360     exit(EXIT_SUCCESS);\r
361 }\r
362 \r
363 // stop_search()\r
364 \r
365 static void stop_search() {\r
366     \r
367     if (Init && Uci->searching) {\r
368         \r
369         ASSERT(Uci->searching);\r
370         ASSERT(Uci->pending_nb>=1);\r
371         \r
372         my_log("POLYGLOT STOP SEARCH\n");\r
373         \r
374         if (option_get_bool("SyncStop")) {\r
375             uci_send_stop_sync(Uci);\r
376         } else {\r
377             uci_send_stop(Uci);\r
378         }\r
379     }\r
380 }\r
381 \r
382 \r
383 // end of main.cpp\r
384 \r