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

deck_analyzer: check for out of hand size (mostly 2p games)

Browse Source
This commit is contained in:
Maximilian Keßler 2023-03-16 09:34:03 +01:00
parent b819886ec5
commit 27055fd07f
Signed by: max
GPG Key ID: BCC5A619923C0BA5
1 changed files with 51 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
deck_analyzer.py
View File

@ -12,7 +12,26 @@ deck_str = "15xaivliynfkrhpdwtprfaskwvfhnpcmjdksmlabcquqoegxugub"
deck = decompress_deck(deck_str)
def analyze(deck: List[DeckCard], num_players):
class InfeasibilityType(Enum):
OutOfPace = 0 # idx denotes index of last card drawn before being forced to reduce pace, value denotes how bad pace is
OutOfHandSize = 1 # idx denotes index of last card drawn before being forced to discard a crit
class InfeasibilityReason():
def __init__(self, infeasibility_type, idx, value=None):
self.type = infeasibility_type
self.index = idx
self.value = value
def __repr__(self):
match self.type:
case InfeasibilityType.OutOfPace:
return "Deck runs out of pace ({}) after drawing card {}".format(self.value, self.index)
case InfeasibilityType.OutOfHandSize:
return "Deck runs out of hand size after drawing card {}".format(self.index)
def analyze(deck: List[DeckCard], num_players) -> InfeasibilityReason | None:
num_suits = max(map(lambda c: c.suitIndex, deck)) + 1
hand_size = STANDARD_HAND_SIZE[num_players]
@ -22,8 +41,10 @@ def analyze(deck: List[DeckCard], num_players):
stacks = [0] * num_suits
stored_cards = set()
stored_crits = set()
min_forced_pace = 100
for (i, card) in enumerate(deck):
worst_index = 0
for (i, card) in enumerate(deck[:-2]):
if card.rank == stacks[card.suitIndex] + 1:
# card is playable
stacks[card.suitIndex] += 1
@ -33,34 +54,54 @@ def analyze(deck: List[DeckCard], num_players):
if check_card in stored_cards:
stacks[card.suitIndex] += 1
stored_cards.remove(check_card)
if check_card in stored_crits:
stored_crits.remove(check_card)
else:
break
elif card.rank <= stacks[card.suitIndex]:
pass # card is trash
elif card.rank > stacks[card.suitIndex] + 1:
# need to store card
if card in stored_cards or card.rank == 5:
stored_crits.add(card)
stored_cards.add(card)
## check for out of handsize:
if len(stored_crits) == num_players * hand_size:
return InfeasibilityReason(InfeasibilityType.OutOfHandSize, i)
# the last - 1 is there because we have to discard 'next', causing a further draw
max_remaining_plays = (len(deck) - i - 1) + num_players - 1
needed_plays = 5 * num_suits - sum(stacks)
missing = max_remaining_plays - needed_plays
min_forced_pace = min(min_forced_pace, missing)
return min_forced_pace
if missing < min_forced_pace:
# print("update to {}: {}".format(i, missing))
min_forced_pace = missing
worst_index = i
if min_forced_pace < 0:
return InfeasibilityReason(InfeasibilityType.OutOfPace, worst_index, min_forced_pace)
else:
return None
def run_on_database():
cur = conn.cursor()
cur2 = conn.cursor()
cur.execute("SELECT seed, num_players, deck from seeds where variant_id = 0 order by num_players desc")
for (seed, num_players, deck) in cur:
p = analyze(decompress_deck(deck), num_players)
if p < 0:
print("seed {} runs out of pace".format(seed))
res = cur.fetchall()
for (seed, num_players, deck) in res:
deck = decompress_deck(deck)
a = analyze(deck, num_players)
if type(a) == InfeasibilityReason:
if a.type == InfeasibilityType.OutOfHandSize:
print("Seed {} infeasible: {}\n{}".format(seed, a, deck))
# if p < 0:
# print("seed {} ({} players) runs out of pace ({}) after drawing {}: {}:\n{}".format(seed, num_players, p, i, deck[i], deck))
# cur.execute("UPDATE seeds SET feasible = f WHERE seed = (%s)", seed)
if __name__ == "__main__":
print(deck)
a = analyze(deck, 2)
a = analyze(deck, 4)
print(a)
run_on_database()