func InitMoveTable() {
ptr := 0
for _, pt := range ALL_PIECE_TYPES {
for sq := 0; sq < square.BOARD_SIZE; sq++ {
MoveTablePtrs[TMoveTableKey{square.TSquare(sq), piece.FromTypeAndColor(piece.TPieceType(pt), piece.BLACK)}] = ptr
MoveTablePtrs[TMoveTableKey{square.TSquare(sq), piece.FromTypeAndColor(piece.TPieceType(pt), piece.WHITE)}] = ptr
for df := -2; df <= 2; df++ {
for dr := -2; dr <= 2; dr++ {
vector_ok := false
dfabs := math.Abs(float64(df))
drabs := math.Abs(float64(dr))
prodabs := dfabs * drabs
sumabs := dfabs + drabs
if (pt & piece.IS_JUMPING) != 0 {
vector_ok = vector_ok || (prodabs == 2)
}
if (pt & piece.IS_STRAIGHT) != 0 {
vector_ok = vector_ok || (sumabs == 1)
}
if (pt & piece.IS_DIAGONAL) != 0 {
vector_ok = vector_ok || (prodabs == 1)
}
if vector_ok {
ok := true
f := int(square.FileOf(square.TSquare(sq)))
r := int(square.RankOf(square.TSquare(sq)))
vector_start := ptr
for ok {
f += df
r += dr
if square.FileRankOk(f, r) {
tsq := square.FromFileRank(square.TFile(f), square.TRank(r))
MoveTable[ptr].To = tsq
MoveTable[ptr].EndPiece = false
ptr++
} else {
ok = false
}
if (pt & piece.IS_SLIDING) == 0 {
ok = false
}
}
for vector_next_ptr := vector_start; vector_next_ptr < ptr; vector_next_ptr++ {
MoveTable[vector_next_ptr].NextVector = ptr
}
}
}
}
MoveTable[ptr].EndPiece = true
ptr++
}
}
}
func (b *TBoard) IsSquareColInCheck(sq square.TSquare, c piece.TColor) bool {
ksq := b.Pos.GetKingPos(c)
for _, pt := range ALL_PIECE_TYPES {
test_piece := piece.FromTypeAndColor(pt, piece.InvColorOf(c))
ptr := GetMoveTablePtr(ksq, test_piece)
for !MoveTable[ptr].EndPiece {
md := MoveTable[ptr]
p := b.PieceAtSq(md.To)
if p == test_piece {
return true
}
if piece.ColorOf(p) != piece.NO_COLOR {
ptr = md.NextVector
} else {
ptr++
}
}
}
return false
}
