Fix crash in double-push, and make it subject to chessFlag
[hachu.git] / hachu.c
diff --git a/hachu.c b/hachu.c
index 290fbf2..173aae9 100644 (file)
--- a/hachu.c
+++ b/hachu.c
@@ -11,7 +11,7 @@
 // promotions by pieces with Lion power stepping in & out the zone in same turn\r
 // promotion on capture\r
 \r
-#define VERSION "0.1beta"\r
+#define VERSION "0.4beta"\r
 \r
 #define PATH level==0 || path[0] == 0xc4028 &&  (level==1 /*|| path[1] == 0x75967 && (level == 2 || path[2] == 0x3400b && (level == 3))*/)\r
 //define PATH 0\r
@@ -40,7 +40,7 @@
 #define BSIZE BWMAX*BHMAX\r
 #define ZONE  zone\r
 \r
-#define ONE (currentVariant == V_SHO || currentVariant == V_CHESS)\r
+#define ONE (currentVariant == V_SHO || currentVariant == V_CHESS || currentVariant == V_SHATRANJ)\r
 \r
 #define BLACK      0\r
 #define WHITE      1\r
@@ -91,12 +91,14 @@ typedef struct {
   char fireMask;\r
 } UndoInfo;\r
 \r
-char *array, *reason;\r
-int bWidth, bHeight, bsize, zone, currentVariant;\r
-int stm, xstm, hashKeyH, hashKeyL, framePtr, msp, nonCapts, rootEval, retMSP, retFirst, retDep, pvPtr, level, cnt50, chuFlag=1, tenFlag, mobilityScore;\r
+char *array, fenArray[4000], *reason;\r
+int bWidth, bHeight, bsize, zone, currentVariant, chuFlag, tenFlag, chessFlag;\r
+int stm, xstm, hashKeyH, hashKeyL, framePtr, msp, nonCapts, rootEval, retMSP, retFirst, retDep, pvPtr, level, cnt50, mobilityScore;\r
 int nodes, startTime, tlim1, tlim2, repCnt, comp;\r
 Move retMove, moveStack[10000], path[100], repStack[300], pv[1000], repeatMove[300];\r
 \r
+      int maxDepth;                            // used by search\r
+\r
 #define X 36 /* slider              */\r
 #define R 37 /* jump capture        */\r
 #define N -1 /* Knight              */\r
@@ -324,6 +326,16 @@ PieceDesc chessPieces[] = {
   { NULL }  // sentinel\r
 };\r
 \r
+PieceDesc shatranjPieces[] = {\r
+  {"Q", "",  15, { 0,1,0,1,0,1,0,1 } },\r
+  {"R", "",  50, { X,0,X,0,X,0,X,0 } },\r
+  {"B", "",   9, { 0,J,0,J,0,J,0,J } },\r
+  {"N", "",  30, { N,N,N,N,N,N,N,N } },\r
+  {"K", "",  28, { 1,1,1,1,1,1,1,1 } },\r
+  {"P", "Q",  8, { M,C,0,0,0,0,0,C } },\r
+  { NULL }  // sentinel\r
+};\r
+\r
 char chuArray[] = "L:FLCSGK:DEGSC:FLL/:RV.B.:BT:KN:PH:BT.B.:RV/:SM:VMR:DH:DK:LN:FK:DK:DHR:VM:SM/PPPPPPPPPPPP/...:GB....:GB..."\r
                  "/............/............/"\r
                  "...:gb....:gb.../pppppppppppp/:sm:vmr:dh:dk:fk:ln:dk:dhr:vm:sm/:rv.b.:bt:ph:kn:bt.b.:rv/l:flcsg:dekgsc:fll";\r
@@ -337,6 +349,7 @@ char tenArray[] = "LN:FLICSGK:DEGSCI:FLNL/:RV.:CS:CS.:BT:KN:LN:FK:PH:BT.:CS:CS.:
                  ":ss:vsb:dh:dk:wb:fi:fe:lh:fi:wb:dk:dhb:vs:ss/:rv.:cs:cs.:bt:ph:fk:ln:kn:bt.:cs:cs.:rv/ln:flicsg:dekgsci:flnl";\r
 char shoArray[] = "LNSGKGSNL/.B..:DE..R./PPPPPPPPP/........./........./........./ppppppppp/.r..:de..b./lnsgkgsnl";\r
 char chessArray[] = "RNBQKBNR/PPPPPPPP/......../......../......../......../pppppppp/rnbqkbnr";\r
+char shatArray[]= "RNBKQBNR/PPPPPPPP/......../......../......../......../pppppppp/rnbkqbnr";\r
 \r
 typedef struct {\r
   int boardWidth, boardFiles, boardRanks, zoneDepth, varNr; // board sizes\r
@@ -344,7 +357,7 @@ typedef struct {
   char *array; // initial position\r
 } VariantDesc;\r
 \r
-typedef enum { V_CHESS, V_SHO, V_CHU, V_DAI, V_TENJIKU, V_DADA, V_MAKA, V_TAI, V_KYOKU } Variant;\r
+typedef enum { V_CHESS, V_SHO, V_CHU, V_DAI, V_TENJIKU, V_DADA, V_MAKA, V_TAI, V_KYOKU, V_SHATRANJ } Variant;\r
 \r
 VariantDesc variants[] = {\r
   { 16,  8,  8, 1, V_CHESS,  "normal", chessArray }, // FIDE\r
@@ -352,6 +365,8 @@ VariantDesc variants[] = {
   { 24, 12, 12, 4, V_CHU,     "chu",     chuArray }, // Chu\r
   { 30, 15, 15, 5, V_DAI,     "dai",     daiArray }, // Dai\r
   { 32, 16, 16, 5, V_TENJIKU, "tenjiku", tenArray }, // Tenjiku\r
+  { 16,  8,  8, 1, V_SHATRANJ,"shatranj",shatArray}, // Shatranj\r
+\r
 //  { 0, 0, 0, 0, 0 }, // sentinel\r
   { 34, 17, 17, 0, V_DADA,    "dada",    chuArray }, // Dai Dai\r
   { 38, 19, 19, 0, V_MAKA,    "maka",    chuArray }, // Maka Dai Dai\r
@@ -532,6 +547,8 @@ LookUp (char *name, int var)
       return ListLookUp(name, chuPieces);\r
     case V_CHESS: // FIDE\r
       return ListLookUp(name, chessPieces);\r
+    case V_SHATRANJ: // Shatranj\r
+      return ListLookUp(name, shatranjPieces);\r
   }\r
   return NULL;\r
 }\r
@@ -626,6 +643,7 @@ Compactify (int stm)
     }\r
   }\r
   for(i=stm+2; i<last[stm]; i+=2) { // second pass: promoted pieces\r
+\r
     if((k = p[i].promo) == -2 && p[i].pos == ABSENT) { // orphaned promoted piece not present\r
       SqueezeOut(i);\r
     }\r
@@ -670,11 +688,13 @@ SetUp(char *array, int var)
   char c, *q, name[3];\r
   PieceDesc *p1, *p2;\r
   last[WHITE] = 1; last[BLACK] = 0;\r
-  if(var == V_CHESS) // add dummy Crown Princes\r
+  if(var == V_CHESS || var == V_SHATRANJ) // add dummy Crown Princes\r
     p[AddPiece(WHITE, LookUp("CP", V_CHU))].pos =  p[AddPiece(BLACK, LookUp("CP", V_CHU))].pos = ABSENT;\r
   for(i=0; ; i++) {\r
 //printf("next rank: %s\n", array);\r
     for(j = BW*i; ; j++) {\r
+      int pflag=0;\r
+      if(*array == '+') pflag++, array++;\r
       c = name[0] = *array++;\r
       if(!c) goto eos;\r
       if(c == '.') continue;\r
@@ -687,9 +707,11 @@ SetUp(char *array, int var)
        if(name[1]) name[1] += 'A' - 'a';\r
       } else color = WHITE;\r
       p1 = LookUp(name, var);\r
+      if(!p1) printf("tellusererror Unknown piece '%s' in setup\n", name), exit(-1);\r
+      if(pflag && p1->promoted) p1 = LookUp(p1->promoted, var); // use promoted piece instead\r
       n = AddPiece(color, p1);\r
       p[n].pos = j;\r
-      if(p1->promoted[0]) {\r
+      if(p1->promoted[0] && !pflag) {\r
        p2 = LookUp(p1->promoted, var);\r
         m = AddPiece(color, p2);\r
        if(m <= n) n += 2;\r
@@ -738,6 +760,7 @@ Init (int var)
   bsize = bWidth*bHeight;\r
   chuFlag = (currentVariant == V_CHU);\r
   tenFlag = (currentVariant == V_TENJIKU);\r
+  chessFlag = (currentVariant == V_CHESS);\r
 \r
   for(i= -1; i<9; i++) { // board steps in linear coordinates\r
     kStep[i] = STEP(direction[i&7].x,   direction[i&7].y);       // King\r
@@ -923,8 +946,8 @@ GenNonCapts (int promoSuppress)
          }\r
        } else\r
        if(r == M) { // FIDE Pawn; check double-move\r
-         if(!NewNonCapture(x, x+v, pFlag) && promoBoard[x-v])\r
-           NewNonCapture(x, x+2*v+DEFER, pFlag); // use promoSuppress flag as e.p. flag\r
+         if(!NewNonCapture(x, x+v, pFlag) && chessFlag && promoBoard[x-v])\r
+           NewNonCapture(x, x+2*v, pFlag), moveStack[msp-1] |= DEFER; // use promoSuppress flag as e.p. flag\r
        }\r
        continue;\r
       }\r
@@ -1811,14 +1834,12 @@ pmoves(int start, int end)
     #define INVALID 0\r
 \r
     // some parameter of your engine\r
-    #define MAXMOVES 500  /* maximum game length  */\r
-    #define MAXPLY   60   /* maximum search depth */\r
+    #define MAXMOVES 2000 /* maximum game length  */\r
+    #define MAXPLY   20   /* maximum search depth */\r
 \r
     #define OFF 0\r
     #define ON  1\r
 \r
-#define DEFAULT_FEN ""\r
-\r
 typedef Move MOVE;\r
 \r
     int moveNr;              // part of game state; incremented by MakeMove\r
@@ -1863,13 +1884,56 @@ UnMake2 (MOVE move)
   sup2 = sup1; sup1 = sup0;\r
 }\r
 \r
+char fenNames[] = "RV....DKDEFL..DHGB......SMLNKN..FK....BT..VM..PH...."; // pairs of char\r
+char fenPromo[] = "WLDHSMSECPB R HFDE....WHFB..LNG ..DKVMFS..FO..FK...."; // pairs of char\r
+\r
+char *\r
+Convert (char *fen)\r
+{\r
+  char *p = fenArray, *q, *rows[36], tmp[4000];\r
+  int n=0;\r
+  printf("# convert FEN '%s'\n", fen);\r
+  q = strchr(fen, ' '); if(q) *q = 0; q = fen;\r
+  do { rows[n++] = q; q = strchr(q, '/'); if(!q) break; *q++ = 0; } while(1);\r
+  *tmp = 0;\r
+  while(--n >= 0) { strcat(tmp, rows[n]); if(n) strcat(tmp, "/"); }\r
+  fen = tmp;\r
+  printf("# flipped FEN '%s'\n", fen);\r
+  while(*fen) {\r
+    if(*fen == ' ') { *p = 0; break; }\r
+    if(n=atoi(fen)) fen++; // digits read\r
+    if(n > 9) fen++; // double digit\r
+    while(n-- > 0) *p++ = '.'; // expand to empty squares\r
+    if(isalpha(*fen)) {\r
+      char *table = fenNames;\r
+      n = *fen > 'Z' ? 'a' - 'A' : 0;\r
+      if((currentVariant == V_CHESS || currentVariant == V_SHATRANJ) && *fen - 'A' - n == 'N' // In Chess N is Knight, not Lion\r
+           || table[2* (*fen - 'A' - n)] == '.') *p++ = *fen; else {\r
+        *p++ = ':';\r
+        *p++ = table[2* (*fen - 'A' - n)] + n;\r
+        *p++ = table[2* (*fen - 'A' - n)+1] + n;\r
+      }\r
+    } else *p++ = *fen;\r
+    fen++;\r
+  }\r
+  printf("# converted FEN '%s'\n", fenArray);\r
+  return fenArray;\r
+}\r
+\r
 int\r
 Setup2 (char *fen)\r
 {\r
+  int stm = WHITE;\r
+  if(fen) {\r
+    char *q = strchr(fen, '\n');\r
+    if(q) *q = 0;\r
+    if(q = strchr(fen, ' ')) stm = (q[1] == 'b' ? BLACK : WHITE); // fen contains color field\r
+    if(strchr(fen, '.') || strchr(fen, ':')) array = fen; else array = Convert(fen);\r
+  }\r
   SetUp(array, currentVariant);\r
   sup0 = sup1 = sup2 = ABSENT;\r
-  rootEval = cnt50 = hashKeyH = hashKeyL = 0;\r
-  return WHITE;\r
+  rootEval = cnt50 = hashKeyH = hashKeyL = moveNr = 0;\r
+  return stm;\r
 }\r
 \r
 void\r
@@ -1943,6 +2007,7 @@ MapFromScratch(attacks);
     if((moveStack[i] & (PROMOTE | DEFER-1)) == ret) break;\r
     if((moveStack[i] & DEFER-1) == ret) deferred = i; // promoted version of entered non-promotion is legal\r
   }\r
+printf("# moveNr = %d in {%d,%d}\n", i, retFirst, retMSP);\r
   if(i>=retMSP) {  // no exact match\r
     if(deferred) { // but maybe non-sensical deferral\r
       int flags = p[board[f]].promoFlag;\r
@@ -2040,7 +2105,7 @@ SearchBestMove (int stm, int timeLeft, int mps, int timeControl, int inc, int ti
   nodes = 0;\r
 MapFromScratch(attacks);\r
   retMove = INVALID; repCnt = 0;\r
-  score = Search(-INF-1, INF+1, rootEval, 20, sup1, sup2);\r
+  score = Search(-INF-1, INF+1, rootEval, maxDepth, sup1, sup2);\r
   *move = retMove;\r
   *ponderMove = INVALID;\r
   return score;\r
@@ -2054,7 +2119,7 @@ PonderUntilInput (int stm)
     int TakeBack(int n)\r
     { // reset the game and then replay it to the desired point\r
       int last, stm;\r
-      stm = Setup2(NULL);\r
+      Init(currentVariant); stm = Setup2(NULL);\r
 printf("# setup done");fflush(stdout);\r
       last = moveNr - n; if(last < 0) last = 0;\r
       for(moveNr=0; moveNr<last; moveNr++) stm = MakeMove2(stm, gameMove[moveNr]),printf("make %2d: %x\n", moveNr, gameMove[moveNr]);\r
@@ -2075,7 +2140,6 @@ printf("# setup done");fflush(stdout);
       int engineSide=NONE;                     // side played by engine\r
       int timeLeft;                            // timeleft on engine's clock\r
       int mps, timeControl, inc, timePerMove;  // time-control parameters, to be used by Search\r
-      int maxDepth;                            // used by search\r
       MOVE move, ponderMove;\r
       int i, score;\r
       char inBuf[8000], command[80];\r
@@ -2088,6 +2152,7 @@ printf("# setup done");fflush(stdout);
 \r
         if(stm == engineSide) {         // if it is the engine's turn to move, set it thinking, and let it move\r
      \r
+pboard(board);\r
           score = SearchBestMove(stm, timeLeft, mps, timeControl, inc, timePerMove, &move, &ponderMove);\r
 \r
           if(move == INVALID) {         // game apparently ended\r
@@ -2134,8 +2199,6 @@ printf("# setup done");fflush(stdout);
         // wait for input, and read it until we have collected a complete line\r
         for(i = 0; (inBuf[i] = getchar()) != '\n'; i++);\r
         inBuf[i+1] = 0;\r
-pboard(board);\r
-pmoves(retFirst, retMSP);\r
 \r
         // extract the first word\r
         sscanf(inBuf, "%s", command);\r
@@ -2157,7 +2220,7 @@ printf("in: %s\n", command);
         }\r
         if(!strcmp(command, "protover")){\r
           printf("feature ping=1 setboard=1 colors=0 usermove=1 memory=1 debug=1 sigint=0 sigterm=0\n");\r
-          printf("feature variants=\"normal,chu,dai,tenjiku,12x12+0_fairy,9x9+0_shogi\"\n");\r
+          printf("feature variants=\"normal,shatranj,chu,dai,tenjiku,12x12+0_fairy,9x9+0_shogi\"\n");\r
           printf("feature myname=\"HaChu " VERSION "\" highlight=1\n");\r
           printf("feature option=\"Resign -check 0\"\n");           // example of an engine-defined option\r
           printf("feature option=\"Contempt -spin 0 -200 200\"\n"); // and another one\r
@@ -2173,7 +2236,6 @@ printf("in: %s\n", command);
           for(i=0; i<5; i++) {\r
             sscanf(inBuf+8, "%s", command);\r
             if(!strcmp(variants[i].name, command)) {\r
-printf("var %d\n",i);\r
               Init(i); stm = Setup2(NULL); break;\r
             }\r
          }\r
@@ -2185,10 +2247,10 @@ printf("var %d\n",i);
         if(!strcmp(command, "ping"))    { printf("pong%s", inBuf+4); continue; }\r
     //  if(!strcmp(command, ""))        { sscanf(inBuf, " %d", &); continue; }\r
         if(!strcmp(command, "new"))     {\r
-          engineSide = BLACK; Init(V_CHESS); stm = Setup2(DEFAULT_FEN); maxDepth = MAXPLY; randomize = OFF; comp = 0;\r
+          engineSide = BLACK; Init(V_CHESS); stm = Setup2(NULL); maxDepth = MAXPLY; randomize = OFF; comp = 0;\r
           continue;\r
         }\r
-        if(!strcmp(command, "setboard")){ engineSide = NONE;  stm = Setup2(inBuf+9); continue; }\r
+        if(!strcmp(command, "setboard")){ engineSide = NONE;  Init(currentVariant); stm = Setup2(inBuf+9); continue; }\r
         if(!strcmp(command, "easy"))    { ponder = OFF; continue; }\r
         if(!strcmp(command, "hard"))    { ponder = ON;  continue; }\r
         if(!strcmp(command, "undo"))    { stm = TakeBack(1); continue; }\r
@@ -2219,6 +2281,7 @@ printf("var %d\n",i);
         if(!strcmp(command, ""))  {  continue; }\r
         if(!strcmp(command, "usermove")){\r
           int move = ParseMove(inBuf+9);\r
+pboard(board);\r
           if(move == INVALID) {\r
             if(reason) printf("Illegal move {%s}\n", reason); else printf("%s\n", reason="Illegal move");\r
             if(comp) PrintResult(stm, -INF); // against computer: claim\r