class GameState:
def __init__(self):
self.state = None
self.me = Participant()
self.resetGame()
self.resetHand()
def resetGame(self):
self.matchID = None
# dummy entries, overridden on first newGame packet
self.leftOpp = Participant()
self.rightOpp = Participant()
self.numHands = None
self.stackSize = None
self.bigB = None
self.smallB = None
self.timebank = None
self.numArrivalsAtStreet = [0,0,0,0]
self.hand = Hand()
#self.trackedHands = [CHECK,BET,RAISE,CALL, POST]
def resetHand(self):
self.handID = None
self.me.newHand()
self.leftOpp.newHand()
self.rightOpp.newHand()
self.holeCard1 = None
self.holeCard2 = None
self.potSize = None
self.pot = 0
self.numBoardCards = None
self.boardCards = "__,__,__,__,__"#["__","__","__","__","__"]
self.numLastActions = None
self.lastActions = None
self.numLegalActions = None
self.legalActions = None
self.hand.clearHand()
self.lastBet = 0
self.street = PREFLOP
self.activePlayers = 3
self.lastActor = None
def parseInput(self, input):
numOptArgs = 0
packet = input.split(" ")
self.state = packet[0]
if self.state == NEWGAME:
self.resetGame()
self.matchID = int(packet[1])
self.leftOpp = Participant(packet[2])
self.rightOpp = Participant(packet[3])
self.numHands = int(packet[4])
self.stackSize = int(packet[5])
self.bigB = int(packet[6])
self.smallB = int(packet[7])
self.timebank = float(packet[8])
elif self.state == NEWHAND:
self.resetHand()
self.handID = int(packet[1])
self.me.position = int(packet[2])
self.rightOpp.position = (self.me.position - 1)%3
self.leftOpp.position = (self.me.position + 1)%3
self.holeCard1 = Card(packet[3])
self.holeCard2 = Card(packet[4])
self.me.holeCard1 = packet[3]
self.me.holeCard2 = packet[4]
self.me.bankroll = int(packet[5])
self.leftOpp.bankroll = int(packet[6])
self.rightOpp.bankroll = int(packet[7])
self.timeBank = float(packet[8])
self.numArrivalsAtStreet[self.street] += 1
elif self.state == GETACTION:
self.potSize = int(packet[1])
self.numBoardCards = int(packet[2])
if self.numBoardCards>0:
numOptArgs += 1
self.boardCards = packet[3] #Card(packet[3+i])
#parse action
self.numLastActions = int(packet[3+numOptArgs])
if self.numLastActions>0:
numOptArgs += 1
self.lastActions = packet[3+numOptArgs]
self.numLegalActions = int(packet[4+numOptArgs])
if self.numLegalActions > 0:
self.legalActions = packet[5+numOptArgs]
self.timebank = float(packet[-1])
self.parseBoardCards()
self.parseLastActions()
self.parseLegalActions()
print self.potSize, "=?", self.pot
elif self.state == HANDOVER:
self.me.bankroll = int(packet[1])
self.leftOpp.bankroll = int(packet[2])
self.rightOpp.bankroll = int(packet[3])
self.numLastActions = int(packet[4])
#parse actions
if self.numLastActions>0:
numOptArgs += 1
self.lastActions = packet[5]
self.numBoardCards = int(packet[5+numOptArgs])
if self.numBoardCards>0:
numOptArgs += 1
self.boardCards = packet[5+numOptArgs]
self.timebank = float(packet[-1])
self.parseBoardCards()
self.parseLastActions()
self.hand.splitActionsList()
#self.calculatePlayerStats()
self.leftOpp.archive.update(self)
self.rightOpp.archive.update(self)
print "my bankroll:", self.me.bankroll
print self.leftOpp.name, " bank:", self.leftOpp.bankroll
print self.rightOpp.name, " bank:", self.rightOpp.bankroll
def parseLastActions(self):
if self.lastActions:
self.lastActions = self.lastActions.split(",")
for i in range(self.numLastActions):
self.lastActions[i] = self.lastActions[i].split(":")
#add each action into structure, Hand
c1 = None
c2 = None
sla = self.lastActions[i][0]
actor = self.lastActions[i][1]
if actor == self.leftOpp.name:
player = self.leftOpp
elif actor == self.rightOpp.name:
player = self.rightOpp
else:
player = self.me
player.totalPot = self.pot + self.me.pip + self.leftOpp.pip + self.rightOpp.pip
potamt = 0
betamt = 0
amt = self.lastBet
if len(self.lastActions[i]) == 3:
amt = float(self.lastActions[i][2])
if sla == "RAISE":
betamt = amt/float(self.lastBet)
potamt = amt/float(player.totalPot)
self.lastBet = amt
player.stack -= amt - player.pip
if not player.aggFreqChanged:
player.numBets[self.street] += 1
player.aggFreqChanged = True
player.amountContributed[self.street] += amt - player.pip
player.amountBetRaise[self.street] += amt - player.pip
player.pip = amt
elif sla == "CALL":
betamt = 1.0
potamt = amt/float(player.totalPot)
player.stack -= self.lastBet - player.pip
player.amountContributed[self.street] += amt - player.pip
player.pip = self.lastBet
elif sla == "CHECK":
if self.street != PREFLOP:
amt = 0
elif sla == "BET":
betamt = amt/float(self.lastBet)
potamt = amt/float(player.totalPot)
self.lastBet = float(self.lastActions[i][2])
player.stack -= self.lastBet
player.pip = self.lastBet
if not player.aggFreqChanged:
player.numBets[self.street] += 1
player.aggFreqChanged = True
player.amountContributed[self.street] += amt
player.amountBetRaise[self.street] += amt
elif sla == "DEAL":
amt = 0
self.pot += self.me.pip + self.leftOpp.pip + self.rightOpp.pip
self.me.pip = 0
self.leftOpp.pip = 0
self.rightOpp.pip = 0
self.street += 1
self.numArrivalsAtStreet[self.street] += 1
for pl in [self.me, self.leftOpp, self.rightOpp]:
pl.aggFreqChanged = False
if pl.active==1 and pl.stack>0:
pl.numArrivalsAtStreet[self.street] += 1
print pl.name, "active:", pl.active, "stack size:", pl.stack, "arrivals:", pl.numArrivalsAtStreet
elif sla == "POST":
self.lastBet = float(self.lastActions[i][2])
player.stack -= self.lastBet
player.pip = self.lastBet
player.amountContributed[self.street] += self.lastBet
elif sla == "SHOWS":
c1 = self.lastActions[i][2]
c2 = self.lastActions[i][3]
amt = 0
elif sla == "FOLD":
player.active = 0
self.activePlayers -= 1
#elif sla == "REFUND":
#elif sla == "TIE":
#elif sla == "WIN":
if sla in ["POST","BET", "RAISE"]:
self.lastActor = player
print "lastactor:",self.lastActor.name
a = Action(ACTION_TYPES.index(sla), self.lastActions[i][1], self.street, c1,
c2, potamt, betamt, amt)
self.hand.actions.append(a)
player.lastActions.append(a)
# print "processed action: " + str(a)
# print "resulting in: stacks",self.me.stack, self.leftOpp.stack, self.rightOpp.stack, "and pips", self.me.pip, self.leftOpp.pip, self.rightOpp.pip
# print "lastActions", self.lastActions
def parseLegalActions(self):
if self.legalActions:
self.legalActions = self.legalActions.split(",")
for i in range(self.numLegalActions):
self.legalActions[i] = self.legalActions[i].split(":")
def parseBoardCards(self):
if(type(self.boardCards) == type("STRING")):
self.boardCards = self.boardCards.split(",")
for i in range(5-len(self.boardCards)):
self.boardCards += ["__"]
def calculatePlayerStats(self):
for p in [self.leftOpp, self.rightOpp]:
for s in [0,1,2,3]:
rounds = p.numArrivalsAtStreet[s]
if s==0:
rounds = self.handID
if rounds == 0:
p.aggFreq[s] = 0
p.avgChips[s] = 0
else:
p.avgChips[s] = float(p.amountContributed[s])/rounds
p.aggFreq[s] = float(p.numBets[s])/rounds
if p.numBets[s] >0:
p.avgRaiseAmt[s] = float(p.amountBetRaise[s])/p.numBets[s]
if self.handID !=0:
p.percentArrivals[s] = float(rounds)/self.handID
