ui.py

import pygame
from pygame.locals import *
from socket import *
import sys
import thread
from Queue import *
import time
import threading
import pygbutton
import platform
from pgu import gui

#Socket to receive data
DEALER_SERVER = '192.168.117.112'
#DEALER_SERVER = '192.168.2.2'
R_PORT = 11716
R_HOST = ''
sock_recv_data = socket(AF_INET, SOCK_STREAM)
sock_recv_data.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
sock_recv_data.bind((R_HOST, R_PORT))

#Socket to send data
S_HOST = DEALER_SERVER
S_PORT = 11717
sock_send_data = socket(AF_INET, SOCK_STREAM)

#Message Queue
msg_q = Queue()

#Thread to handle received data
def handle_data():
    data = ''
    while True:
        if not msg_q.empty():
            q_data = msg_q.get()
            for data in q_data.split('$'):
                req_type = data.partition(':')[0].partition('=')[2]
                if req_type == 'check':
                    temp = data.partition(':')[2]
                    paramlist = temp.split(';')
                    for param in paramlist:
                        if param.partition('=')[0] == 'players_nick':
                            temp1 = param.partition('=')[2]
                            temp1 = temp1.strip('[').strip(']').split(',')
                            for i in range(len(temp1)):
                                poker_data.players_nick.append(temp1[i])
                    print 'Connection established'
                elif req_type == 'reset':
                    poker_data.hand_cards = ['images/front', 'images/front']
                    for i in range(5):
                        poker_data.com_cards[i] = 'images/front'
                elif req_type == 'init':
                    #print req_type
                    temp = data.partition(':')[2]
                    paramlist = temp.split(';')
                    for param in paramlist:
                        if param.partition('=')[0] == 'nop':
                            poker_data.NO_OF_PLAYERS = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'yourPos':
                            poker_data.mypos = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'sb':
                            poker_data.small_blind = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'players_money':
                            temp = param.partition('=')[2]
                            temp = temp.strip('[').strip(']').split(',')
                            for i in range(len(temp)):
                                poker_data.remaining_money[i] = int(temp[i])
                elif req_type == 'cards':
                    #print req_type
                    card1 = data.partition(':')[2].partition(',')[0]
                    card2 = data.partition(':')[2].partition(',')[2]
                    poker_data.hand_cards[0] = 'images/' + card1
                    poker_data.hand_cards[1] = 'images/' + card2
                elif req_type == 'data':
                    #print req_type
                    temp = data.partition(':')[2]
                    paramlist = temp.split(';')
                    for param in paramlist:
                        if param.partition('=')[0] == 'flop_cards':
                            temp = param.partition('=')[2]
                            temp = temp.strip('[').strip(']').split(',')
                            #print 'comm cards are ' + temp
                            for i in range(len(temp)):
                                poker_data.com_cards[i] = 'images/' + temp[i].split("'")[1]
                            #print 'after ' + poker_data.com_cards
                        elif param.partition('=')[0] == 'turn_cards':
                            temp = param.partition('=')[2]
                            poker_data.com_cards[3] = 'images/' + temp
                        elif param.partition('=')[0] == 'river_cards':
                            temp = param.partition('=')[2]
                            poker_data.com_cards[4] = 'images/' + temp
                        elif param.partition('=')[0] == 'live':
                            temp = param.partition('=')[2]
                            temp = temp.strip('[').strip(']').split(',')
                            for i in range(0, poker_data.NO_OF_PLAYERS):
                                poker_data.ingame[i] = int(temp[i])
                        elif param.partition('=')[0] == 'current_player':
                            poker_data.current_player = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'NO_OF_POTS':
                            poker_data.no_of_pots = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'last_raised_amt':
                            poker_data.last_raised_amt = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'players_money':
                            temp = param.partition('=')[2]
                            temp = temp.strip('[').strip(']').split(',')
                            for i in range(len(temp)):
                                poker_data.remaining_money[i] = int(temp[i])
                        elif param.partition('=')[0] == 'pot_investment':
                            temp = param.partition('=')[2]
                            temp = temp.strip('[').strip(']').split(',')
                            for i in range(len(temp)):
                                poker_data.money_in_pot[i] = int(temp[i])
                        elif param.partition('=')[0] == 'last_raised_by':
                            poker_data.last_raised_by = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'small_blind_amt':
                            poker_data.small_blind_amt = int(param.partition('=')[2])
                        elif param.partition('=')[0] == 'pot_money':
                            temp = param.partition('=')[2]
                            temp = temp.strip('[').strip(']').split(',')
                            for i in range(poker_data.no_of_pots):
                                poker_data.pot_money[i] = int(temp[i])
                        elif param.partition('=')[0] == 'showdown_list':
                            pass

                amt_min = max(poker_data.last_raised_amt + poker_data.last_raised_by, poker_data.small_blind_amt*2)
                amt_max = poker_data.remaining_money[poker_data.mypos]
                if amt_min > amt_max and amt_max > poker_data.last_raised_amt:
                    poker_data.bet_amount = amt_max + poker_data.money_in_pot[poker_data.mypos]
                else:
                    poker_data.bet_amount = amt_min

        else:
            time.sleep(1)

#Thread to send data
def send_data(data):
    try:
        print 'sending.......'
        data = str(data)
        sock_send_data.send(data)
        print 'data sent ->' + data
    except:
        print 'Something wrong happened while sending data'



# Thread to receive data
def recv_data():
    print 'waiting for conn in recv_data'
    sock_recv_data.listen(1)
    conn_recv_data,addr = sock_recv_data.accept()

    while True:
        try:
            data = conn_recv_data.recv(4096)
            if data != '':
                msg_q.put(data)
                print data
        except:
            print 'connect error'
            sys.exit()


# Start recv_data thread
try:
    thread.start_new_thread(recv_data, ())
except:
    print'unable to start recv_data thread'

# Start handle_data thread
try:
    thread.start_new_thread(handle_data, ())
except:
    print'unable to start handle_data thread'

try:
    sock_send_data.connect((S_HOST, S_PORT))
    sock_send_data.send(raw_input('Enter Your Nick : '))
except:
    print 'connect error'


class Message:
    mypos = 0
    small_blind = 5
    small_blind_amt = 0
    no_of_pots = 0
    pot_money = [0] * 7
    NO_OF_PLAYERS = 0
    players_nick = []
    ingame = [0, 0, 0, 0, 0, 0, 0, 0]
    money_in_pot = [0, 0, 0, 0, 0, 0, 0, 0]
    remaining_money = [1500, 1500, 1500, 1500, 1500, 1500, 1500, 1500]
    last_raised_amt = 0
    last_raised_by = 0
    current_player = 0
    bet_amount = 0
    player_cards1 = ["images/front", "images/front", "images/front",
                     "images/front", "images/front", "images/front",
                     "images/front"]
    player_cards2 = ["images/front", "images/front", "images/front",
                     "images/front", "images/front", "images/front",
                     "images/front"]
    hand_cards = ["images/front", "images/front"]
    com_cards = ["images/front", "images/front", "images/front",
                 "images/front", "images/front"]

    def __init__(self):
        print 'object created'


#Creating message object
poker_data = Message()
pos = [[320, 300], [170, 300], [60, 165], [170, 25], [360, 25],
       [550, 25], [655, 165], [550, 300]]
pot_pos = [[396, 290], [206, 290], [170, 213], [206, 131], [396, 131],
           [586, 131], [630, 213], [586, 290]]
money_pos = [[396, 406], [206, 406], [100, 270], [206, 15], [396, 15],
             [586, 15], [700, 270], [586, 406]]
#SCREEN_X = 795
SCREEN_X = 950
SCREEN_Y = 511
bg = "images/back.png"



#Slider class
class bet_bar(gui.Table):
    def __init__(self,**params):
        self.value = range(100)
        gui.Table.__init__(self,**params)
        fg = (255,255,255)

        self.tr()
        self.td(gui.Label("Bet / Raise",color=fg),colspan=2)

        self.tr()
        self.td(gui.Label("Amount: ",color=fg),align=1)
        e = gui.HSlider(0,0,1000,size=10,width=500,height=16,name='amount')
        e.connect(gui.CHANGE,self.adjust,(0,e))
        self.td(e)
    def adjust(self,value):
        (num, slider) = value
        self.value[num] = slider.value
        amt_min = max(poker_data.last_raised_amt + poker_data.last_raised_by, poker_data.small_blind_amt*2)
        amt_max = poker_data.remaining_money[poker_data.mypos] + poker_data.money_in_pot[poker_data.mypos]
        if amt_min > amt_max and amt_max > poker_data.last_raised_amt:
            poker_data.bet_amount = amt_max
        else:
            poker_data.bet_amount = int(slider.value)/1000. * (amt_max-amt_min) + amt_min
            poker_data.bet_amount = int(poker_data.bet_amount)

app = gui.App()
betBar = bet_bar()

c = gui.Container(align=-1,valign=-1)
c.add(betBar,50,460)

pygame.init()
app.init(c)

screen = pygame.display.set_mode((SCREEN_X, SCREEN_Y), 0, 32)
screen.fill((28, 28, 30))
pygame.display.set_caption("Poker Project")
icon = pygame.image.load("images/icon.png").convert()
button_fold = pygbutton.PygButton((50, 420, 100, 30), 'FOLD')
button_call = pygbutton.PygButton((280, 420, 150, 30), 'CALL / CHECK')
button_raise = pygbutton.PygButton((550, 420, 150, 30), 'BET / RAISE')
button_turn = pygbutton.PygButton((50, 20, 100, 30), 'Your Turn')
current_player_image = pygame.image.load("images/current_player").convert()
small_blind_image = pygame.image.load("images/sb.png").convert()
#button = pygame.image.load("images/button").convert()
pygame.display.set_icon(icon)
back = pygame.image.load(bg).convert()
screen = pygame.display.get_surface()
clock = pygame.time.Clock()


while True:
    font = pygame.font.Font(None, 24)
    hc_images = [pygame.image.load(poker_data.hand_cards[0]).convert(),
                 pygame.image.load(poker_data.hand_cards[1]).convert()]
    cc_image = []
    for i in range(0, 5):
        cc_image.append(pygame.image.load(poker_data.com_cards[i])
                        .convert())
    card1_image = []
    card2_image = []
    for i in range(0, 7):
        card1_image.append(pygame.image.load(poker_data.player_cards1[i])
                           .convert())
        card2_image.append(pygame.image.load(poker_data.player_cards2[i])
                           .convert())
    clock.tick(10)
    for event in pygame.event.get():
        fold_events = button_fold.handleEvent(event)
        call_events = button_call.handleEvent(event)
        raise_events = button_raise.handleEvent(event)
        if event.type == QUIT:
            pygame.quit()
            sys.exit()
        elif 'click' in fold_events:
            print 'Fold button clicked'
            if poker_data.current_player == poker_data.mypos:
                poker_data.hand_cards[0] = 'images/front'
                poker_data.hand_cards[1] = 'images/front'
                thread.start_new_thread(send_data, ('req=fold:',))
                poker_data.current_player = -1
                break
        elif 'click' in call_events:
            print 'CALL button clicked'
            if poker_data.current_player == poker_data.mypos:
                print 'attempting to send data'
                thread.start_new_thread(send_data, ('req=call:amt='+str(min(poker_data.last_raised_amt, (poker_data.remaining_money[poker_data.mypos] + poker_data.money_in_pot[poker_data.mypos]))),))
                poker_data.current_player = -1
                break
        elif 'click' in raise_events:
            print 'RAISE button clicked'
            if poker_data.current_player == poker_data.mypos:
                print 'attempting to send data'
                slider_value = str(poker_data.bet_amount)
                thread.start_new_thread(send_data, ('req=raiseTO:amt='+str(poker_data.bet_amount),))
                poker_data.current_player = -1
                break
        elif event.type is KEYDOWN and event.key == K_ESCAPE:
            pass
        else:
            app.event(event)



    screen.blit(back, (0, 0))
    screen.blit(hc_images[0], (pos[0][0], pos[0][1]))
    screen.blit(hc_images[1], (pos[0][0] + 72, pos[0][1]))
    for i in range(0, 5):
        screen.blit(cc_image[i], (i * 72 + 225, 50 + 96 + 20))
    j = (poker_data.mypos + 1) % 8
    for i in range(1, 8):
        if(poker_data.ingame[j]):
            screen.blit(card1_image[i - 1], (pos[i][0], pos[i][1]))
            screen.blit(card2_image[i - 1], (pos[i][0] + 15, pos[i][1]))
        j += 1
        j %= 8
    j = 0 #(poker_data.mypos) % 8
    for k in range(0, 8):
        if(poker_data.ingame[j]):
            i = k
            rem_money_text = font.render("$" + str
                                         (poker_data.remaining_money[i]),
                                         1, (0, 255, 0))
            bet_money_text = font.render("$" + str(poker_data.money_in_pot[i]),
                                         1, (255, 0, 0))
            i = (k-poker_data.mypos+8)%8
            rem_money_textpos = rem_money_text.get_rect()
            bet_money_textpos = bet_money_text.get_rect()
            rem_money_textpos.centerx = money_pos[i][0] + 30
            bet_money_textpos.centerx = pot_pos[i][0]
            rem_money_textpos.centery = money_pos[i][1]
            bet_money_textpos.centery = pot_pos[i][1]
            screen.blit(rem_money_text, rem_money_textpos)
            screen.blit(bet_money_text, bet_money_textpos)
            nick_text = font.render(poker_data.players_nick[k] , 1, (255,255,255))
            nick_text_pos = nick_text.get_rect()
            nick_text_pos.centerx = money_pos[i][0] - 20
            nick_text_pos.centery = money_pos[i][1]
            screen.blit(nick_text, nick_text_pos)
        j += 1
        j %= 8
    '''
    screen.blit(button, (50, 450))
    fold = font.render("FOLD", 1, (255, 0, 0))
    screen.blit(fold, (75, 465))
    screen.blit(button, (280, 450))
    call = font.render("CALL/", 1, (255, 0, 0))
    screen.blit(call, (305, 455))
    check = font.render("CHECK", 1, (255, 0, 0))
    screen.blit(check, (305, 475))
    screen.blit(button, (510, 450))
    rise = font.render("RAISE", 1, (255, 0, 0))
    screen.blit(check, (535, 465))
    '''
    button_turn.visible = False
    button_call.visible = False
    button_raise.visible = False
    button_fold.visible = False
    if poker_data.mypos == poker_data.current_player:
        button_turn.visible = True
        button_call.visible = True
        if poker_data.last_raised_amt < poker_data.remaining_money[poker_data.mypos] + poker_data.money_in_pot[poker_data.mypos]:
            button_raise.visible = True
        button_fold.visible = True
    button_call.draw(screen)
    button_raise.draw(screen)
    button_fold.draw(screen)
    button_turn.draw(screen)
    sb = 8 + poker_data.small_blind - poker_data.mypos
    sb %= 8
    screen.blit(small_blind_image, (pos[sb][0]- 30, pos[sb][1]))
    cp = 8 + poker_data.current_player - poker_data.mypos
    cp %= 8
    screen.blit(current_player_image, (pos[cp][0]-16, pos[cp][1] + 20))
    if poker_data.mypos == poker_data.current_player:
        call_text = font.render('$ '+str(min(poker_data.last_raised_amt, (poker_data.remaining_money[poker_data.mypos] + poker_data.money_in_pot[poker_data.mypos]))), 1, (255, 255, 255))
        screen.blit(call_text, (450, 425))
        if poker_data.last_raised_amt < poker_data.remaining_money[poker_data.mypos] + poker_data.money_in_pot[poker_data.mypos]:
            app.paint()
            bet_text = font.render('$ '+str(poker_data.bet_amount), 1, (255, 255, 255))
            screen.blit(bet_text, (700, 480))
    for i in range(poker_data.no_of_pots):
        pot_text = font.render('Pot ' + str(i + 1) + ' $ '+str(poker_data.pot_money[i]), 1, (255, 255, 255))
        screen.blit(pot_text, (800, (i * 40) + 20))
    pygame.display.update()