Mercurial > hg > terminal-chess / file diff

     1.1 --- a/src/chess/rook.c	Mon Mar 31 11:41:08 2014 +0200
     1.2 +++ b/src/chess/rook.c	Mon Mar 31 14:00:58 2014 +0200
     1.3 @@ -31,16 +31,122 @@
     1.4  #include "rook.h"
     1.5  
     1.6  _Bool rook_chkrules(Move *move) {
     1.7 -    // TODO: implement
     1.8 +    return move->torow == move->fromrow || move->tofile == move->fromfile;
     1.9 +}
    1.10 +
    1.11 +_Bool rook_isblocked(Board board, Move *move) {
    1.12 +    
    1.13 +    if (move->torow == move->fromrow) {
    1.14 +        int d = move->tofile > move->fromfile ? 1 : -1;
    1.15 +        uint8_t f = move->fromfile;
    1.16 +        while (f != move->tofile-d) {
    1.17 +            f += d;
    1.18 +            if (board[move->fromrow][f]) {
    1.19 +                return 1;
    1.20 +            }
    1.21 +        }
    1.22 +    } else {
    1.23 +        int d = move->torow > move->fromrow ? 1 : -1;
    1.24 +        uint8_t r = move->fromrow;
    1.25 +        while (r != move->torow - d) {
    1.26 +            r += d;
    1.27 +            if (board[r][move->fromfile]) {
    1.28 +                return 1;
    1.29 +            }
    1.30 +        }
    1.31 +    }
    1.32 +    
    1.33      return 0;
    1.34  }
    1.35  
    1.36 -_Bool rook_isblocked(Board board, Move *move) {
    1.37 -    // TODO: implement
    1.38 -    return 1;
    1.39 +static int rook_getloc_fixedrow(Board board, Move *move) {
    1.40 +    uint8_t file = POS_UNSPECIFIED;
    1.41 +    for (uint8_t f = 0 ; f < 8 ; f++) {
    1.42 +        if (board[move->fromrow][f] == move->piece) {
    1.43 +            if (file == POS_UNSPECIFIED) {
    1.44 +                file = f;
    1.45 +            } else {
    1.46 +                return AMBIGUOUS_MOVE;
    1.47 +            }
    1.48 +        }
    1.49 +    }
    1.50 +    if (file == POS_UNSPECIFIED) {
    1.51 +        return INVALID_POSITION;
    1.52 +    } else {
    1.53 +        move->fromfile = file;
    1.54 +        return VALID_MOVE_SYNTAX;
    1.55 +    }
    1.56 +}
    1.57 +
    1.58 +static int rook_getloc_fixedfile(Board board, Move *move) {
    1.59 +    uint8_t row = POS_UNSPECIFIED;
    1.60 +    for (uint8_t r = 0 ; r < 8 ; r++) {
    1.61 +        if (board[r][move->fromfile] == move->piece) {
    1.62 +            if (row == POS_UNSPECIFIED) {
    1.63 +                row = r;
    1.64 +            } else {
    1.65 +                return AMBIGUOUS_MOVE;
    1.66 +            }
    1.67 +        }   
    1.68 +    }
    1.69 +    if (row == POS_UNSPECIFIED) {
    1.70 +        return INVALID_POSITION;
    1.71 +    } else {
    1.72 +        move->fromrow = row;
    1.73 +        return VALID_MOVE_SYNTAX;
    1.74 +    }
    1.75  }
    1.76  
    1.77  int rook_getlocation(Board board, Move *move) {
    1.78 -    // TODO: implement
    1.79 -    return INVALID_MOVE_SYNTAX;
    1.80 +    
    1.81 +    if (move->fromfile != POS_UNSPECIFIED) {
    1.82 +        if (move->fromfile == move->tofile) {
    1.83 +            return rook_getloc_fixedfile(board, move);
    1.84 +        } else {
    1.85 +            if (board[move->torow][move->fromfile] == move->piece) {
    1.86 +                move->fromrow = move->torow;
    1.87 +                return VALID_MOVE_SYNTAX;
    1.88 +            } else {
    1.89 +                return INVALID_POSITION;
    1.90 +            }
    1.91 +        }
    1.92 +    }
    1.93 +    
    1.94 +    if (move->fromrow != POS_UNSPECIFIED) {
    1.95 +        if (move->fromrow == move->torow) {
    1.96 +            return rook_getloc_fixedrow(board, move);
    1.97 +        } else {
    1.98 +            if (board[move->fromrow][move->tofile] == move->piece) {
    1.99 +                move->fromfile = move->tofile;
   1.100 +                return VALID_MOVE_SYNTAX;
   1.101 +            } else {
   1.102 +                return INVALID_POSITION;
   1.103 +            }
   1.104 +        }
   1.105 +    }
   1.106 +    
   1.107 +    Move chkrowmove = *move, chkfilemove = *move;
   1.108 +    
   1.109 +    chkrowmove.fromrow = move->torow;
   1.110 +    int chkrow = rook_getloc_fixedrow(board, &chkrowmove);
   1.111 +    
   1.112 +    chkfilemove.fromfile = move->tofile;
   1.113 +    int chkfile = rook_getloc_fixedfile(board, &chkfilemove);
   1.114 +    
   1.115 +    if ((chkrow == VALID_MOVE_SYNTAX && chkfile == VALID_MOVE_SYNTAX) ||
   1.116 +        chkrow == AMBIGUOUS_MOVE || chkfile == AMBIGUOUS_MOVE) {
   1.117 +        return AMBIGUOUS_MOVE;
   1.118 +    }
   1.119 +    
   1.120 +    if (chkrow == VALID_MOVE_SYNTAX) {
   1.121 +        *move = chkrowmove;
   1.122 +        return VALID_MOVE_SYNTAX;
   1.123 +    }
   1.124 +    
   1.125 +    if (chkfile == VALID_MOVE_SYNTAX) {
   1.126 +        *move = chkfilemove;
   1.127 +        return VALID_MOVE_SYNTAX;
   1.128 +    }
   1.129 +    
   1.130 +    return INVALID_POSITION;
   1.131  }