43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
from typing import List, Dict, Tuple
|
|
|
|
from hanabi.hanab_game import Action, ActionType
|
|
from hanabi.live.hanab_live import HanabLiveInstance, parse_json_game, HanabLiveGameState
|
|
|
|
|
|
def analyze_game(instance: HanabLiveInstance, actions: List[Action]) -> Tuple[List[int], str]:
|
|
bdrs = []
|
|
termination = ''
|
|
game = HanabLiveGameState(instance)
|
|
for action in actions:
|
|
if action.type == ActionType.Discard or (action.type == ActionType.Play and not game.is_playable(instance.deck[action.target])):
|
|
discard = instance.deck[action.target]
|
|
if not game.is_trash(discard):
|
|
if game.is_critical(discard):
|
|
termination = 'Discard crit' if action.type == ActionType.Discard else 'Bomb crit'
|
|
break
|
|
if discard.rank != 1:
|
|
if discard in game.deck[game.progress:]:
|
|
bdrs.append(game.draw_pile_size)
|
|
else:
|
|
if game.deck[game.progress:].count(discard) == 2:
|
|
bdrs.append(game.draw_pile_size)
|
|
if action.type == ActionType.Play:
|
|
play = instance.deck[action.target]
|
|
if (not game.is_playable(play)) and game.is_critical(play):
|
|
termination = 'Bomb crit'
|
|
game.make_action(action)
|
|
if termination == '':
|
|
if game.strikes == 3:
|
|
termination = 'Strikeout'
|
|
elif actions[-1].type in [ActionType.EndGame, ActionType.VoteTerminate]:
|
|
termination = 'VTK'
|
|
elif game.score < 25:
|
|
termination = 'Lost Endgame'
|
|
return bdrs, termination
|
|
|
|
|
|
def describe_game(game_json: Dict) -> Tuple[List[int], str]:
|
|
instance, actions = parse_json_game(game_json)
|
|
bdrs, termination = analyze_game(instance, actions)
|
|
return bdrs, termination
|
|
|