Hanabi/Py-Hanabi
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add selection of default hanab.live variant id to allow compressing of regular GameStates

Browse source
This commit is contained in:
Maximilian Keßler 2023-05-19 12:43:49 +02:00
parent 16cee69c82
commit 2af847d5b8
Signed by: max
GPG key ID: BCC5A619923C0BA5
2 changed files with 24 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
compress.py
View file

@ -5,10 +5,11 @@ import more_itertools
from enum import Enum from enum import Enum
from termcolor import colored from termcolor import colored
from typing import List, Optional from typing import List, Optional, Union
from variants import variant_id, variant_name from variants import variant_id, variant_name
from hanabi import DeckCard, ActionType, Action, GameState, HanabiInstance from hanabi import DeckCard, ActionType, Action, GameState, HanabiInstance
from hanab_live import HanabLiveGameState, HanabLiveInstance
# use same BASE62 as on hanab.live to encode decks # use same BASE62 as on hanab.live to encode decks
@ -164,14 +165,18 @@ def decompress_deck(deck_str: str) -> List[DeckCard]:
# which can be used in json replay links # which can be used in json replay links
# The GameState object has to be standard / fitting hanab.live variants, # The GameState object has to be standard / fitting hanab.live variants,
# otherwise compression is not possible # otherwise compression is not possible
def compress_game_state(state: GameState) -> str: def compress_game_state(state: Union[GameState, HanabLiveGameState]) -> str:
# if not state.instance.is_standard(): var_id = -1
# raise ValueError("Cannot compress non-standard hanabi instance") if isinstance(state, GameState):
var_id = HanabLiveInstance.select_standard_variant_id(state.instance)
else:
assert isinstance(state, HanabLiveGameState)
var_id = state.instance.variant_id
out = "{}{},{},{}".format( out = "{}{},{},{}".format(
state.instance.num_players, state.instance.num_players,
compress_deck(state.instance.deck), compress_deck(state.instance.deck),
compress_actions(state.actions), compress_actions(state.actions),
state.instance.variant_id # Note that a sane default is chosen if construction did not provide one var_id
) )
with_dashes = ''.join(more_itertools.intersperse("-", out, 20)) with_dashes = ''.join(more_itertools.intersperse("-", out, 20))
return with_dashes return with_dashes

15
hanab_live.py
View file

@ -1,4 +1,4 @@
from typing import List from typing import List, Optional
import hanabi import hanabi
import constants import constants
@ -26,6 +26,19 @@ class HanabLiveInstance(hanabi.HanabiInstance):
self.variant_id = variant_id self.variant_id = variant_id
self.variant = Variant.from_db(self.variant_id) self.variant = Variant.from_db(self.variant_id)
@staticmethod
def select_standard_variant_id(instance: hanabi.HanabiInstance):
err_msg = "Hanabi instance not supported by hanab.live, cannot convert to HanabLiveInstance: "
assert 3 <= instance.num_suits <= 6, \
err_msg + "Illegal number of suits ({}) found, must be in range [3,6]".format(instance.num_suits)
assert 0 <= instance.num_dark_suits <= 2, \
err_msg + "Illegal number of dark suits ({}) found, must be in range [0,2]".format(instance.num_dark_suits)
assert 4 <= instance.num_suits - instance.num_dark_suits, \
err_msg + "Illegal ratio of dark suits to suits, can have at most {} dark suits with {} total suits".format(
max(instance.num_suits - 4, 0), instance.num_suits
)
return constants.VARIANT_IDS_STANDARD_DISTRIBUTIONS[instance.num_suits][instance.num_dark_suits]
class HanabLiveGameState(hanabi.GameState): class HanabLiveGameState(hanabi.GameState):
def __init__(self, instance: HanabLiveInstance): def __init__(self, instance: HanabLiveInstance):