barak/tim-the-enchanter
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

pep8 and drop pymc/numpy requirement

Browse source
This commit is contained in:
Barak Michener 2013-07-30 14:27:32 -04:00
parent 9eb3625393
commit d4a6aa3925
2 changed files with 347 additions and 336 deletions
Download patch file Download diff file Expand all files Collapse all files

2
requirements.txt
View file

@ -1,6 +1,4 @@
argparse==1.2.1
clint==0.3.1
numpy==1.7.1
progressbar==2.3
pymc==2.2
wsgiref==0.1.2

119
tim.py
View file

@ -1,43 +1,54 @@
import pymc as mc
import functools
import itertools
from collections import defaultdict as dd
import pprint
import progressbar
import readline
import random
import sys
from clint.textui import colored, puts
def ResistanceGame(n_players):
full_set = [("G1", True), ("G2", True), ("G3", True), ("E1", False), ("E2", False),
("G4", True), ("E3", False), ("G5", True), ("G6", True), ("E4", False)]
full_set = [("G1", True), ("G2", True), ("G3", True),
("E1", False), ("E2", False), ("G4", True),
("E3", False), ("G5", True), ("G6", True), ("E4", False)]
return full_set[:n_players]
class Bernoulli(object):
def __init__(self, percentage):
self.percentage = percentage
def rand(self):
return random.random() < self.percentage
def random(self):
return self.rand()
class DeceptionGame(object):
def __init__(self, player_array):
self.player_array = player_array
self.all_permutations = list(itertools.permutations(player_array))
self.n_players = len(player_array)
self.n_good = len([x for x in player_array if x[1] == True])
self.n_good = len([x for x in player_array if x[1] is True])
self.trace = None
self.observations = []
self.seen = []
self.tid = 0
self.lady_will_duck = mc.Bernoulli("lady_will_duck", 0.5)
self.lady_will_duck = Bernoulli(0.5)
self.mission_ducks_on_round = [None] * 5
self.mission_ducks_on_round[0] = mc.Bernoulli("missionduck_on_0", 0.5)
self.mission_ducks_on_round[1] = mc.Bernoulli("missionduck_on_1", 0.5)
self.mission_ducks_on_round[2] = mc.Bernoulli("missionduck_on_2", 0.5)
self.mission_ducks_on_round[3] = mc.Bernoulli("missionduck_on_3", 0.0)
self.mission_ducks_on_round[4] = mc.Bernoulli("missionduck_on_4", 0.0)
self.mission_ducks_on_round[0] = Bernoulli(0.5)
self.mission_ducks_on_round[1] = Bernoulli(0.5)
self.mission_ducks_on_round[2] = Bernoulli(0.5)
self.mission_ducks_on_round[3] = Bernoulli(0.0)
self.mission_ducks_on_round[4] = Bernoulli(0.0)
self.ignorance_on_round = [None] * 5
self.ignorance_on_round[0] = mc.Bernoulli("ignorance_on_0", 0.9)
self.ignorance_on_round[1] = mc.Bernoulli("ignorance_on_1", 0.7)
self.ignorance_on_round[2] = mc.Bernoulli("ignorance_on_2", 0.5)
self.ignorance_on_round[3] = mc.Bernoulli("ignorance_on_3", 0.3)
self.ignorance_on_round[4] = mc.Bernoulli("ignorance_on_4", 0.3)
self.ignorance_on_round[0] = Bernoulli(0.9)
self.ignorance_on_round[1] = Bernoulli(0.7)
self.ignorance_on_round[2] = Bernoulli(0.5)
self.ignorance_on_round[3] = Bernoulli(0.3)
self.ignorance_on_round[4] = Bernoulli(0.3)
def player_is_good(self, deal, player):
return deal[player][1]
@ -47,6 +58,7 @@ class DeceptionGame(object):
def add_known_alliance(self, player_id, is_good):
transaction = []
def obs(deal):
if self.player_is_good(deal, player_id) == is_good:
return True
@ -54,11 +66,13 @@ class DeceptionGame(object):
return None
transaction.append(obs)
self.observations.append(transaction)
self.seen.append(["Known role ", player_id, "Good" if is_good else "Evil"])
self.seen.append(
["Known role ", player_id, "Good" if is_good else "Evil"])
self.tid += 1
def add_known_role(self, player_id, role_str):
transaction = []
def obs(deal):
if self.player_is_role(deal, player_id) == role_str:
return True
@ -71,6 +85,7 @@ class DeceptionGame(object):
def player_sees_player_and_claims(self, p1, p2, claim):
transaction = []
def obs(deal):
if self.player_is_good(deal, p1):
if self.player_is_good(deal, p2) == claim:
@ -90,14 +105,17 @@ class DeceptionGame(object):
return True
transaction.append(obs)
self.observations.append(transaction)
self.seen.append(["Lady:", p1, "says", p2 , "is", "Good" if claim else "Evil"])
self.seen.append(
["Lady:", p1, "says", p2, "is", "Good" if claim else "Evil"])
self.tid += 1
def do_mission(self, team, fails, must_fail, r):
transaction = []
rnd = r - 1
def obs(deal):
n_actually_good_people = sum([int(self.player_is_good(deal, x)) for x in team])
n_actually_good_people = sum(
[int(self.player_is_good(deal, x)) for x in team])
n_spies = len(team) - n_actually_good_people
if n_spies == 0:
if fails != 0:
@ -119,14 +137,17 @@ class DeceptionGame(object):
transaction.append(obs)
self.observations.append(transaction)
self.seen.append(["Mission:"] + team + ["with %d fails on round %d" % (fails, r)])
self.seen.append(
["Mission:"] + team + ["with %d fails on round %d" % (fails, r)])
self.tid += 1
def do_vote(self, team, votes, r):
transaction = []
rnd = r - 1
def obs(deal):
n_actually_good_people = sum([int(self.player_is_good(deal, x)) for x in team])
n_actually_good_people = sum(
[int(self.player_is_good(deal, x)) for x in team])
n_spies = len(team) - n_actually_good_people
could_happen = True
for player, vote in enumerate(votes):
@ -167,12 +188,15 @@ class DeceptionGame(object):
self.seen.append(["Vote:"] + team + ["voted %s round %d" % (votes, r)])
self.tid += 1
def eval(self, length=10):
random.seed()
deck = self.all_permutations[:]
new_deck = []
trace = []
progress = progressbar.ProgressBar(widgets=["Simulating games: ", progressbar.Bar(marker="*"), " ", progressbar.ETA()])
progress = progressbar.ProgressBar(
widgets=["Simulating games: ",
progressbar.Bar(marker="*"),
" ", progressbar.ETA()])
for i in progress(range(length)):
for deal in deck:
f_list = []
@ -214,7 +238,10 @@ class DeceptionGame(object):
out[i]["role"] = dd(float)
out[i]["side"] = dd(float)
progress = progressbar.ProgressBar(widgets=["Reticulating splines: ", progressbar.Bar(marker="*"), " ", progressbar.ETA()])
progress = progressbar.ProgressBar(
widgets=["Reticulating splines: ",
progressbar.Bar(marker="*"),
" ", progressbar.ETA()])
size = len(self.trace) * 1.0
for deal in progress(self.trace):
for i, card in enumerate(deal):
@ -234,7 +261,8 @@ class DeceptionGame(object):
return dict(out)
def __str__(self):
return "%d Player Game (%d constraints)" % (self.n_players, len(self.seen))
return "%d Player Game (%d constraints)" % (self.n_players,
len(self.seen))
def disbelieve(self, i):
self.observations = self.observations[:i] + self.observations[i + 1:]
@ -245,12 +273,19 @@ class DeceptionGame(object):
pp = pprint.PrettyPrinter(indent=4)
pp.pprint(self.report())
def repl_report(report, namemap, ngood):
sort_order = sorted([(report[i]["side"].get(True, 0.0), i) for i in range(len(report))], reverse=True)
sort_order = sorted(
[(report[i]["side"].get(True, 0.0), i)
for i in range(len(report))],
reverse=True)
still_good = 0
for goodness, i in sort_order:
row = "%s: %2f%% Good %2f%% Evil" % (namemap.get(i, "") + "(%s)" % str(i), goodness * 100, (1.0 - goodness) * 100)
row = "%s: %2f%% Good %2f%% Evil" % (
namemap.get(i, "") + " (%s)" % str(i),
goodness * 100,
(1.0 - goodness) * 100)
if still_good < ngood:
puts(colored.cyan(row))
else:
@ -271,7 +306,8 @@ def main():
command = command_list[0]
if command == "quit" or command == "q" or command == "exit":
sys.exit(0)
if (command != "newgame" and command != "testgame") and game is None:
if (command != "newgame" and command != "testgame") \
and game is None:
puts(colored.red("Need to create a game"))
continue
elif command == "newgame":
@ -293,7 +329,8 @@ def main():
continue
elif command == "ls":
for i, statement in enumerate(game.seen):
name = " ".join([namemap.get(x, str(x)) for x in statement])
name = " ".join(
[namemap.get(x, str(x)) for x in statement])
print "%d: %s" % (i, name)
continue
elif command == "vote":
@ -354,29 +391,5 @@ def main():
continue
if __name__ == '__main__':
main()
base_game = DeceptionGame(ResistanceGame(5))
#base_game.add_known_role(0, "G1")
#base_game.add_known_alliance(1, False)
#base_game.player_sees_player_and_claims(0, 1, True)
#base_game.player_sees_player_and_claims(1, 2, False)
#base_game.player_sees_player_and_claims(2, 3, False)
base_game.do_vote([1,2], [0,1,1,0,1], 1)
base_game.do_mission([1,2], 0, False, 1)
base_game.do_vote([0, 1, 2], [1,1,1,0,1], 2)
base_game.do_mission([0, 1, 2], 1, False, 2)
base_game.do_vote([3, 4], [0,0,1,1,1], 3)
base_game.do_mission([3, 4], 0, False, 3)
base_game.do_vote([3, 4], [0,0,1,1,1], 4)
base_game.do_mission([0, 3, 4], 1, False, 4)
#base_game.do_vote([1, 3, 4], [0,1,1,0,1], 5)
#base_game.do_mission([1, 3, 4], 1, True, 5)
base_game.eval(100)
base_game.print_report()