Spaces:
Sleeping
Sleeping
File size: 2,394 Bytes
466e3c0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | from __future__ import annotations
from typing import Any
from persistentpoker_bench.betting import LegalActionSet
from persistentpoker_bench.cards import cards_to_notation
from persistentpoker_bench.game_state import HandState
from persistentpoker_bench.pool import PersistentPool
def serialize_legal_actions(legal_actions: LegalActionSet) -> dict[str, Any]:
return {
"can_fold": legal_actions.can_fold,
"can_check": legal_actions.can_check,
"can_call": legal_actions.can_call,
"can_bet": legal_actions.can_bet,
"can_raise": legal_actions.can_raise,
"can_all_in": legal_actions.can_all_in,
"call_amount": legal_actions.call_amount,
"min_bet_to": legal_actions.min_bet_to,
"min_raise_to": legal_actions.min_raise_to,
"max_to": legal_actions.max_to,
}
def serialize_hand_state(
hand_state: HandState,
persistent_pool: PersistentPool,
*,
hand_id: str,
acting_player_index: int,
) -> dict[str, Any]:
return {
"hand_id": hand_id,
"street": hand_state.street.value,
"button_index": hand_state.button_index,
"actor_index": hand_state.actor_index,
"pot_total": hand_state.pot_total,
"current_bet": hand_state.current_bet,
"last_full_raise_size": hand_state.last_full_raise_size,
"community_cards": list(cards_to_notation(hand_state.community_cards)),
"persistent_pool": list(persistent_pool.notation_snapshot()),
"players": [
_serialize_player(hand_state, player_index, acting_player_index)
for player_index in range(len(hand_state.players))
],
}
def _serialize_player(
hand_state: HandState,
player_index: int,
acting_player_index: int,
) -> dict[str, Any]:
player = hand_state.players[player_index]
payload = {
"seat": player.seat,
"name": player.name,
"stack": player.stack,
"eliminated": player.eliminated,
"committed_street": player.committed_street,
"committed_total": player.committed_total,
"folded": player.folded,
"all_in": player.all_in,
"is_self": player_index == acting_player_index,
}
if player_index == acting_player_index and len(player.hole_cards) == 2:
payload["hole_cards"] = list(cards_to_notation(player.hole_cards))
return payload
|