def push(self, move):
try:
self.move_stack.append(move)
self.color = not self.color
if self.variant[-5:] == "shogi":
# TODO: move this to pyffish.cpp
parts = sf.get_fen(self.variant, self.fen, [move],
self.chess960, self.sfen,
self.show_promoted).split()
color = "w" if parts[1] == "b" else "b"
placement, pockets = parts[0][:-1].split("[")
if pockets == "":
pockets = "-"
self.fen = "%s[%s] %s %s" % (placement, pockets, color,
len(self.move_stack))
else:
self.fen = sf.get_fen(self.variant, self.fen, [move],
self.chess960, self.sfen,
self.show_promoted)
except Exception:
self.move_stack.pop()
self.color = not self.color
log.error(
"ERROR: sf.get_fen() failed on %s %s %s" %
(self.initial_fen, ",".join(self.move_stack), self.chess960))
raise
def get_captured(self):
# TODO: this does not consider promoted pieces
previous_fen = sf.get_fen(self.variant, self.get_start_fen(), self.moves[:-1])
current_fen = sf.get_fen(self.variant, self.get_start_fen(), self.moves)
piece_filter = str.isupper if current_fen.split()[1] == 'w' else str.islower
previous_pieces = Counter(filter(piece_filter, previous_fen.split()[0]))
current_pieces = Counter(filter(piece_filter, current_fen.split()[0]))
return ''.join((previous_pieces - current_pieces).elements())
def get_fen(self):
if self.variant[-5:] == "shogi":
# TODO: move this to pyffish.cpp
parts = sf.get_fen(self.variant, self.initial_fen, self.move_stack).split()
color = "w" if parts[1] == "b" else "b"
placement, pockets = parts[0][:-1].split("[")
if pockets == "":
pockets = "-"
ply = parts[-1]
return "%s[%s] %s %s" % (placement, pockets, color, ply)
else:
try:
return sf.get_fen(self.variant, self.initial_fen, self.move_stack, self.chess960)
except Exception:
log.error("ERROR: sf.get_fen() failed on %s %s %s" % (self.initial_fen, ",".join(self.move_stack), self.chess960))
raise
def push(self, move):
try:
self.move_stack.append(move)
self.ply += 1
self.color = not self.color
self.fen = sf.get_fen(
self.variant,
self.fen,
[move],
self.chess960,
self.sfen,
self.show_promoted,
self.count_started,
)
except Exception:
self.move_stack.pop()
self.ply -= 1
self.color = not self.color
log.error(
"ERROR: sf.get_fen() failed on %s %s %s",
self.initial_fen,
",".join(self.move_stack),
self.chess960,
)
raise
def test_get_fen(self):
result = sf.get_fen("chess", CHESS, [])
self.assertEqual(result, CHESS)
result = sf.get_fen("capablanca", CAPA, [])
self.assertEqual(result, CAPA)
result = sf.get_fen("xiangqi", XIANGQI, [])
self.assertEqual(result, XIANGQI)
fen = "rnab1kbcnr/ppppPppppp/10/4q5/10/10/PPPPP1PPPP/RNABQKBCNR[p] b KQkq - 0 3"
result = sf.get_fen("capahouse", CAPA,
["f2f4", "e7e5", "f4e5", "e8e5", "P@e7"])
self.assertEqual(result, fen)
fen0 = "reb1k2r/ppppqppp/2nb1n2/4p3/4P3/N1P2N2/PB1PQPPP/RE2KBHR[h] b KQkqac - 2 6"
fen1 = "reb2rk1/ppppqppp/2nb1n2/4p3/4P3/N1P2N2/PB1PQPPP/RE2KBHR[h] w KQac - 3 7"
result = sf.get_fen("seirawan", fen0, ["e8g8"])
self.assertEqual(result, fen1)
result = sf.get_fen("chess", CHESS, [], True, False, False)
self.assertEqual(result, CHESS960)
# test O-O-O
fen = "rbkqnrbn/pppppppp/8/8/8/8/PPPPPPPP/RBKQNRBN w AFaf - 0 1"
moves = [
"d2d4", "f7f5", "e1f3", "h8g6", "h1g3", "c7c6", "c2c3", "e7e6",
"b1d3", "d7d5", "d1c2", "b8d6", "e2e3", "d8d7", "c1a1"
]
result = sf.get_fen("chess", fen, moves, True, False, False)
self.assertEqual(
result,
"r1k1nrb1/pp1q2pp/2pbp1n1/3p1p2/3P4/2PBPNN1/PPQ2PPP/2KR1RB1 b fa - 2 8",
CHESS960)
# SFEN
result = sf.get_fen("shogi", SHOGI, [], False, True)
self.assertEqual(result, SHOGI_SFEN)
# makruk FEN
fen = "rnsmksnr/8/1pM~1pppp/p7/8/PPPP1PPP/8/RNSKMSNR b - - 0 3"
result = sf.get_fen("makruk", MAKRUK,
["e3e4", "d6d5", "e4d5", "a6a5", "d5c6m"], False,
False, True)
self.assertEqual(result, fen)
result = sf.get_fen("makruk", fen, [], False, False, True)
self.assertEqual(result, fen)
def play(self):
self.initialize()
while not self.is_game_end() and not (self.partner and self.partner.is_game_end()):
idx = len(self.moves) % 2
engine = self.engines[idx]
engine.update_clocks(self.clock_times[idx], self.clock_times[idx - 1])
engine.usermove(self.moves[-1]) if self.moves else engine.go()
start_time = time.time()
with self.lock:
self.moves.append(engine.get_move())
end_time = time.time()
self.clock_times[idx] += self.time_control.increment - (end_time - start_time)
logging.debug('Position: {}, Move: {}'.format(sf.get_fen(self.variant, self.get_start_fen(), self.moves[:-1]),
self.moves[-1]))
if self.partner and self.is_legal():
captured = self.get_captured()
if captured:
self.partner.set_holdings(captured)
def test_get_san(self):
fen = "4k3/8/3R4/8/1R3R2/8/3R4/4K3 w - - 0 1"
result = sf.get_san("chess", fen, "b4d4")
self.assertEqual(result, "Rbd4")
result = sf.get_san("chess", fen, "f4d4")
self.assertEqual(result, "Rfd4")
result = sf.get_san("chess", fen, "d2d4")
self.assertEqual(result, "R2d4")
result = sf.get_san("chess", fen, "d6d4")
self.assertEqual(result, "R6d4")
fen = "4k3/8/3R4/3P4/1RP1PR2/8/3R4/4K3 w - - 0 1"
result = sf.get_san("chess", fen, "d2d4")
self.assertEqual(result, "Rd4")
fen = "1r2k3/P1P5/8/8/8/8/8/4K3 w - - 0 1"
result = sf.get_san("chess", fen, "c7b8q")
self.assertEqual(result, "cxb8=Q+")
fen = "1r2k3/P1P5/8/8/8/8/8/4K3 w - - 0 1"
result = sf.get_san("chess", fen, "c7b8q", False, sf.NOTATION_LAN)
self.assertEqual(result, "c7xb8=Q+")
result = sf.get_san("capablanca", CAPA, "e2e4")
self.assertEqual(result, "e4")
result = sf.get_san("capablanca", CAPA, "e2e4", False, sf.NOTATION_LAN)
self.assertEqual(result, "e2-e4")
result = sf.get_san("capablanca", CAPA, "h1i3")
self.assertEqual(result, "Ci3")
result = sf.get_san("sittuyin", SITTUYIN, "R@a1")
self.assertEqual(result, "R@a1")
fen = "3rr3/1kn3n1/1ss1p1pp/1pPpP3/6PP/p3KN2/2SSFN2/3R3R[] b - - 0 14"
result = sf.get_san("sittuyin", fen, "c6c5")
self.assertEqual(result, "Scxc5")
fen = "7R/1r6/3k1np1/3s2N1/3s3P/4n3/6p1/2R3K1[] w - - 2 55"
result = sf.get_san("sittuyin", fen, "h4h4f")
self.assertEqual(result, "h4=F")
fen = "k7/2K3P1/8/4P3/8/8/8/1R6[] w - - 0 1"
result = sf.get_san("sittuyin", fen, "e5f6f")
self.assertEqual(result, "e5f6=F")
result = sf.get_san("shogi", SHOGI, "i3i4")
self.assertEqual(result, "P-16")
result = sf.get_san("shogi", SHOGI, "i3i4", False, sf.NOTATION_SHOGI_HOSKING)
self.assertEqual(result, "P16")
result = sf.get_san("shogi", SHOGI, "f1e2", False, sf.NOTATION_SHOGI_HOSKING)
self.assertEqual(result, "G49-58")
result = sf.get_san("shogi", SHOGI, "f1e2", False, sf.NOTATION_SHOGI_HODGES)
self.assertEqual(result, "G4i-5h")
result = sf.get_san("shogi", SHOGI, "f1e2", False, sf.NOTATION_SHOGI_HODGES_NUMBER)
self.assertEqual(result, "G49-58")
fen = "lnsgkgsnl/1r5b1/pppppp1pp/6p2/9/2P6/PP1PPPPPP/1B5R1/LNSGKGSNL w -"
result = sf.get_san("shogi", fen, "b2h8", False, sf.NOTATION_SHOGI_HODGES)
self.assertEqual(result, "Bx2b=")
result = sf.get_san("shogi", fen, "b2h8+", False, sf.NOTATION_SHOGI_HODGES)
self.assertEqual(result, "Bx2b+")
fen = "lnsgkg1nl/1r5s1/pppppp1pp/6p2/9/2P6/PP1PPPPPP/7R1/LNSGKGSNL[Bb] w "
result = sf.get_san("shogi", fen, "B@g7", False, sf.NOTATION_SHOGI_HODGES)
self.assertEqual(result, "B*3c")
result = sf.get_san("shogi", fen, "B@g7", False, sf.NOTATION_SHOGI_HODGES_NUMBER)
self.assertEqual(result, "B*33")
fen = "lnsgkg1nl/1r4s+B1/pppppp1pp/6p2/9/2P6/PP1PPPPPP/7R1/LNSGKGSNL[B] w "
result = sf.get_san("shogi", fen, "h8g7", False, sf.NOTATION_SHOGI_HODGES)
self.assertEqual(result, "+B-3c")
fen = "lnk2gsnl/7b1/p1p+SGp1pp/6p2/1pP6/4P4/PP3PPPP/1S2G2R1/L2GK1bNL[PRppns] w "
result = sf.get_san("shogi", fen, "d7d8", False, sf.NOTATION_SHOGI_HODGES)
self.assertEqual(result, "+S-6b")
result = sf.get_san("xiangqi", XIANGQI, "h1g3")
self.assertEqual(result, "Hg3")
result = sf.get_san("xiangqi", XIANGQI, "h1g3", False, sf.NOTATION_XIANGQI_WXF)
self.assertEqual(result, "H2+3")
result = sf.get_san("xiangqi", XIANGQI, "c1e3")
self.assertEqual(result, "Ece3")
result = sf.get_san("xiangqi", XIANGQI, "c1e3", False, sf.NOTATION_XIANGQI_WXF)
self.assertEqual(result, "E7+5")
result = sf.get_san("xiangqi", XIANGQI, "h3h10")
self.assertEqual(result, "Cxh10")
result = sf.get_san("xiangqi", XIANGQI, "h3h10", False, sf.NOTATION_XIANGQI_WXF)
self.assertEqual(result, "C2+7")
result = sf.get_san("xiangqi", XIANGQI, "h3h5")
self.assertEqual(result, "Ch5")
# skip disambiguation for elephants and advisors, but not for pieces that require it
fen = "rnbakabnr/9/1c5c1/p1p1p1p1p/9/1NB6/P1P1P1P1P/1C1A3C1/9/RNBAK21R w - - 0 1"
result = sf.get_san("xiangqi", fen, "c5e3", False, sf.NOTATION_XIANGQI_WXF)
self.assertEqual(result, "E7-5")
result = sf.get_san("xiangqi", fen, "d1e2", False, sf.NOTATION_XIANGQI_WXF)
self.assertEqual(result, "A6+5")
result = sf.get_san("xiangqi", fen, "b5c7", False, sf.NOTATION_XIANGQI_WXF)
self.assertEqual(result, "H++7")
# Tandem pawns
fen = "rnbakabnr/9/1c5c1/p1p1P1p1p/4P4/9/P3P3P/1C5C1/9/RNBAKABNR w - - 0 1"
result = sf.get_san("xiangqi", fen, "e7d7", False, sf.NOTATION_XIANGQI_WXF)
self.assertEqual(result, "15=6")
fen = "1rb1ka2r/4a4/2ncb1nc1/p1p1p1p1p/9/2P6/P3PNP1P/2N1C2C1/9/R1BAKAB1R w - - 1 7"
result = sf.get_san("xiangqi", fen, "c3e2")
self.assertEqual(result, "Hce2")
result = sf.get_san("xiangqi", fen, "c3d5")
self.assertEqual(result, "Hd5")
result = sf.get_san("janggi", JANGGI, "b1c3", False, sf.NOTATION_JANGGI)
self.assertEqual(result, "H02-83")
fen = "1b1aa2b1/5k3/3ncn3/1pp1pp3/5r2p/9/P1PPB1PPB/2N1CCN1c/9/R2AKAR2 w - - 19 17"
result = sf.get_san("janggi", fen, "d1e2", False, sf.NOTATION_SAN)
self.assertEqual(result, "Ade2")
fen = "1Pbcka3/3nNn1c1/N2CaC3/1pB6/9/9/5P3/9/4K4/9 w - - 0 23"
result = sf.get_san("janggi", fen, "f8f10", False, sf.NOTATION_SAN)
self.assertEqual(result, "Cfxf10")
fen = "rnsm1s1r/4n1k1/1ppppppp/p7/2PPP3/PP3PPP/4N2R/RNSKMS2 b - - 1 5"
result = sf.get_san("makruk", fen, "f8f7")
self.assertEqual(result, "Sf7")
fen = "4k3/8/8/4S3/8/2S5/8/4K3 w - - 0 1"
result = sf.get_san("makruk", fen, "e5d4")
self.assertEqual(result, "Sed4")
result = sf.get_san("makruk", fen, "c3d4")
self.assertEqual(result, "Scd4")
fen = "4k3/8/8/3S4/8/3S4/8/4K3 w - - 0 1"
result = sf.get_san("makruk", fen, "d3d4")
self.assertEqual(result, "Sd4")
UCI_moves = ["e2e4", "e7e5", "g1f3", "b8c6h", "f1c4", "f8c5e"]
SAN_moves = ["e4", "e5", "Nf3", "Nc6/H", "Bc4", "Bc5/E"]
fen = SEIRAWAN
for i, move in enumerate(UCI_moves):
result = sf.get_san("seirawan", fen, move)
self.assertEqual(result, SAN_moves[i])
fen = sf.get_fen("seirawan", SEIRAWAN, UCI_moves[:i + 1])
result = sf.get_san("seirawan", fen, "e1g1")
self.assertEqual(result, "O-O")
result = sf.get_san("seirawan", fen, "e1g1h")
self.assertEqual(result, "O-O/He1")
result = sf.get_san("seirawan", fen, "e1g1e")
self.assertEqual(result, "O-O/Ee1")
result = sf.get_san("seirawan", fen, "h1e1h")
self.assertEqual(result, "O-O/Hh1")
result = sf.get_san("seirawan", fen, "h1e1e")
self.assertEqual(result, "O-O/Eh1")
def test_get_fen(self):
result = sf.get_fen("chess", CHESS, [])
self.assertEqual(result, CHESS)
# incomplete FENs
result = sf.get_fen("chess", "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR", [])
self.assertEqual(result, "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w - - 0 1")
result = sf.get_fen("chess", "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq -", [])
self.assertEqual(result, "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1")
result = sf.get_fen("chess", "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w 1 2", [])
self.assertEqual(result, "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w - - 1 2")
# alternative piece symbols
result = sf.get_fen("janggi", "rhea1aehr/4k4/1c5c1/p1p1p1p1p/9/9/P1P1P1P1P/1C5C1/4K4/RHEA1AEHR w - - 0 1", [])
self.assertEqual(result, JANGGI)
result = sf.get_fen("capablanca", CAPA, [])
self.assertEqual(result, CAPA)
result = sf.get_fen("xiangqi", XIANGQI, [])
self.assertEqual(result, XIANGQI)
fen = "rnab1kbcnr/ppppPppppp/10/4q5/10/10/PPPPP1PPPP/RNABQKBCNR[p] b KQkq - 0 3"
result = sf.get_fen("capahouse", CAPA, ["f2f4", "e7e5", "f4e5", "e8e5", "P@e7"])
self.assertEqual(result, fen)
fen0 = "reb1k2r/ppppqppp/2nb1n2/4p3/4P3/N1P2N2/PB1PQPPP/RE2KBHR[h] b KQkqac - 2 6"
fen1 = "reb2rk1/ppppqppp/2nb1n2/4p3/4P3/N1P2N2/PB1PQPPP/RE2KBHR[h] w KQac - 3 7"
result = sf.get_fen("seirawan", fen0, ["e8g8"])
self.assertEqual(result, fen1)
result = sf.get_fen("chess", CHESS, [], True, False, False)
self.assertEqual(result, CHESS960)
# test O-O-O
fen = "rbkqnrbn/pppppppp/8/8/8/8/PPPPPPPP/RBKQNRBN w AFaf - 0 1"
moves = ["d2d4", "f7f5", "e1f3", "h8g6", "h1g3", "c7c6", "c2c3", "e7e6", "b1d3", "d7d5", "d1c2", "b8d6", "e2e3", "d8d7", "c1a1"]
result = sf.get_fen("chess", fen, moves, True, False, False)
self.assertEqual(result, "r1k1nrb1/pp1q2pp/2pbp1n1/3p1p2/3P4/2PBPNN1/PPQ2PPP/2KR1RB1 b fa - 2 8", CHESS960)
# SFEN
result = sf.get_fen("shogi", SHOGI, [], False, True)
self.assertEqual(result, SHOGI_SFEN)
# makruk FEN
fen = "rnsmksnr/8/1pM~1pppp/p7/8/PPPP1PPP/8/RNSKMSNR b - - 0 3"
result = sf.get_fen("makruk", MAKRUK, ["e3e4", "d6d5", "e4d5", "a6a5", "d5c6m"], False, False, True)
self.assertEqual(result, fen)
result = sf.get_fen("makruk", fen, [], False, False, True)
self.assertEqual(result, fen)
# makruk piece honor counting
fen = "8/3k4/8/2K1S1P1/8/8/8/8 w - - 0 1"
moves = ["g5g6m"]
result = sf.get_fen("makruk", fen, moves, False, False, True)
self.assertEqual(result, "8/3k4/6M~1/2K1S3/8/8/8/8 b - 88 8 1")
fen = "8/2K3k1/5m2/4S1S1/8/8/8/8 w - 128 97 1"
moves = ["e5f6"]
result = sf.get_fen("makruk", fen, moves, False, False, True)
self.assertEqual(result, "8/2K3k1/5S2/6S1/8/8/8/8 b - 44 8 1")
# makruk board honor counting
fen = "3k4/2m5/8/4MP2/3KS3/8/8/8 w - - 0 1"
moves = ["f5f6m"]
result = sf.get_fen("makruk", fen, moves, False, False, True)
self.assertEqual(result, "3k4/2m5/5M~2/4M3/3KS3/8/8/8 b - 128 0 1")
fen = "3k4/2m5/5M~2/4M3/3KS3/8/8/8 w - 128 0 33"
moves = ["d4d5"]
result = sf.get_fen("makruk", fen, moves, False, False, True)
self.assertEqual(result, "3k4/2m5/5M~2/3KM3/4S3/8/8/8 b - 128 1 33")
fen = "3k4/2m5/5M~2/4M3/3KS3/8/8/8 w - 128 36 1"
moves = ["d4d5"]
result = sf.get_fen("makruk", fen, moves, False, False, True)
self.assertEqual(result, "3k4/2m5/5M~2/3KM3/4S3/8/8/8 b - 128 37 1")
fen = "3k4/2m5/5M~2/4M3/3KS3/8/8/8 w - 128 0 33"
moves = ["d4d5"]
result = sf.get_fen("makruk", fen, moves, False, False, True, -1)
self.assertEqual(result, "3k4/2m5/5M~2/3KM3/4S3/8/8/8 b - 128 0 33")
fen = "3k4/2m5/5M~2/4M3/3KS3/8/8/8 w - 128 7 33"
moves = ["d4d5"]
result = sf.get_fen("makruk", fen, moves, False, False, True, 58)
self.assertEqual(result, "3k4/2m5/5M~2/3KM3/4S3/8/8/8 b - 128 8 33")
def initial_sfen(self):
return sf.get_fen(self.variant, self.initial_fen, [], False, True)
def __str__(self):
return pyffish.get_fen("janggi", self.initial_fen, self.previous_moves)
def get_holdings(self):
fen = sf.get_fen(self.variant, self.get_start_fen(), self.moves)
return fen.split('[', 1)[1].split(']', 1)[0]
def addMove(self, move:str):
if move == "(none)":
print("Ingoring none move")
return
self.FEN.append(pyffish.get_fen(self.variant.name, self.FEN[-1], [move]))
self.moves.append(move)
