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

greedy_solver: remove global variables. allow exporting games with no actions

Browse source
This commit is contained in:
Maximilian Keßler 2023-03-17 11:55:46 +01:00
parent 0851f83e12
commit 93eca1f0b1
Signed by: max
GPG key ID: BCC5A619923C0BA5
1 changed files with 20 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
greedy_solver.py
View file

@ -129,6 +129,13 @@ class GameState():
self.__make_turn() self.__make_turn()
def to_json(self): def to_json(self):
# ensure we have at least one action
if len(self.actions) == 0:
self.actions.append(Action(
ActionType.EndGame,
target: 0
)
)
return { return {
"deck": self.deck, "deck": self.deck,
"players": self.players, "players": self.players,
@ -295,33 +302,33 @@ wins = open("won_seeds.txt", "a")
losses = open("lost_seeds.txt", "a") losses = open("lost_seeds.txt", "a")
crits = open("crits_lost.txt", "a") crits = open("crits_lost.txt", "a")
lost = 0
won = 0
crits_lost = 0
def run_deck(seed, num_players, deck_str): def run_deck(seed, num_players, deck_str):
global lost
global won
global crits_lost
deck = decompress_deck(deck_str) deck = decompress_deck(deck_str)
gs = GameState(num_players, deck) gs = GameState(num_players, deck)
strat = GreedyStrategy(gs) strat = GreedyStrategy(gs)
while not gs.is_over(): while not gs.is_over():
strat.make_move() strat.make_move()
if not gs.score == 25: if not gs.score == 25:
losses.write("Seed {:10} {}:\n{}\n".format(seed, str(deck), link(gs.to_json()))) losses.write("{}-player seed {:10} {}:\n{}\n".format(num_players, seed, str(deck), link(gs.to_json())))
lost += 1 return False
else: return True
won += 1
def run_samples(num_players, sample_size): def run_samples(num_players, sample_size):
won = 0
lost = 0
cur = conn.cursor() cur = conn.cursor()
cur.execute("SELECT seed, num_players, deck FROM seeds WHERE variant_id = 0 AND num_players = (%s) order by seed desc limit (%s)", (num_players, sample_size)) cur.execute("SELECT seed, num_players, deck FROM seeds WHERE variant_id = 0 AND num_players = (%s) order by seed desc limit (%s)", (num_players, sample_size))
for r in cur: for r in cur:
run_deck(*r) succ = run_deck(*r)
print("won: {:4}, lost: {:4}, crits lost: {:3}".format(won, lost, crits_lost), end = "\r") if succ:
won += 1
else:
lost += 1
print("won: {:4}, lost: {:4}".format(won, lost), end = "\r")
print() print()
print("Total wins: {}%".format(round(100 * won / (lost + won + crits_lost), 2))) print("Total wins: {}%".format(round(100 * won / (lost + won), 2)))
if __name__ == "__main__": if __name__ == "__main__":
for p in range(2, 6): for p in range(2, 6):