240 lines
8.8 KiB
Python
240 lines
8.8 KiB
Python
#!/usr/bin/env python
|
|
"""Mucked.py
|
|
|
|
Mucked cards display for FreePokerTools HUD.
|
|
"""
|
|
# Copyright 2008, Ray E. Barker
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
########################################################################
|
|
|
|
# to do
|
|
# problem with hand 30586
|
|
|
|
# Standard Library modules
|
|
import sys
|
|
import os
|
|
import string
|
|
import xml.dom.minidom
|
|
from xml.dom.minidom import Node
|
|
|
|
# pyGTK modules
|
|
import pygtk
|
|
import gtk
|
|
import gobject
|
|
|
|
# FreePokerTools modules
|
|
import Configuration
|
|
import Database
|
|
import Tables
|
|
import Hud
|
|
import Mucked
|
|
import HandHistory
|
|
|
|
class Mucked:
|
|
def __init__(self, parent, db_connection):
|
|
|
|
self.parent = parent #this is the parent of the mucked cards widget
|
|
self.db_connection = db_connection
|
|
|
|
self.vbox = gtk.VBox()
|
|
self.parent.add(self.vbox)
|
|
|
|
self.mucked_list = MuckedList (self.vbox, db_connection)
|
|
self.mucked_cards = MuckedCards(self.vbox, db_connection)
|
|
self.mucked_list.mucked_cards = self.mucked_cards
|
|
|
|
def update(self, new_hand_id):
|
|
self.mucked_list.update(new_hand_id)
|
|
|
|
class MuckedList:
|
|
def __init__(self, parent, db_connection):
|
|
|
|
self.parent = parent
|
|
self.db_connection = db_connection
|
|
|
|
# set up a scrolled window to hold the listbox
|
|
self.scrolled_window = gtk.ScrolledWindow()
|
|
self.scrolled_window.set_policy(gtk.POLICY_NEVER, gtk.POLICY_ALWAYS)
|
|
self.parent.add(self.scrolled_window)
|
|
|
|
# create a ListStore to use as the model
|
|
self.liststore = gtk.ListStore(str, str, str)
|
|
self.treeview = gtk.TreeView(self.liststore)
|
|
self.tvcolumn0 = gtk.TreeViewColumn('HandID')
|
|
self.tvcolumn1 = gtk.TreeViewColumn('Cards')
|
|
self.tvcolumn2 = gtk.TreeViewColumn('Net')
|
|
|
|
# add tvcolumn to treeview
|
|
self.treeview.append_column(self.tvcolumn0)
|
|
self.treeview.append_column(self.tvcolumn1)
|
|
self.treeview.append_column(self.tvcolumn2)
|
|
|
|
# create a CellRendererText to render the data
|
|
self.cell = gtk.CellRendererText()
|
|
|
|
# add the cell to the tvcolumn and allow it to expand
|
|
self.tvcolumn0.pack_start(self.cell, True)
|
|
self.tvcolumn1.pack_start(self.cell, True)
|
|
self.tvcolumn2.pack_start(self.cell, True)
|
|
self.tvcolumn0.add_attribute(self.cell, 'text', 0)
|
|
self.tvcolumn1.add_attribute(self.cell, 'text', 1)
|
|
self.tvcolumn2.add_attribute(self.cell, 'text', 2)
|
|
# resize the cols if nec
|
|
self.tvcolumn0.set_resizable(True)
|
|
self.treeview.connect("row-activated", self.activated_event)
|
|
|
|
self.scrolled_window.add_with_viewport(self.treeview)
|
|
|
|
def activated_event(self, path, column, data=None):
|
|
sel = self.treeview.get_selection()
|
|
(model, iter) = sel.get_selected()
|
|
self.mucked_cards.update(model.get_value(iter, 0))
|
|
|
|
def update(self, new_hand_id):
|
|
info_row = self.db_connection.get_hand_info(new_hand_id)
|
|
iter = self.liststore.append(info_row[0])
|
|
sel = self.treeview.get_selection()
|
|
sel.select_iter(iter)
|
|
|
|
vadj = self.scrolled_window.get_vadjustment()
|
|
vadj.set_value(vadj.upper)
|
|
self.mucked_cards.update(new_hand_id)
|
|
|
|
class MuckedCards:
|
|
def __init__(self, parent, db_connection):
|
|
|
|
self.parent = parent #this is the parent of the mucked cards widget
|
|
self.db_connection = db_connection
|
|
|
|
self.card_images = self.get_card_images()
|
|
self.seen_cards = {}
|
|
self.grid_contents = {}
|
|
self.eb = {}
|
|
|
|
self.rows = 8
|
|
self.cols = 7
|
|
self.grid = gtk.Table(self.rows, self.cols + 4, homogeneous = False)
|
|
|
|
for r in range(0, self.rows):
|
|
for c in range(0, self.cols):
|
|
self.seen_cards[(c, r)] = gtk.image_new_from_pixbuf(self.card_images[('B', 'S')])
|
|
self.eb[(c, r)]= gtk.EventBox()
|
|
|
|
# set up the contents for the cells
|
|
for r in range(0, self.rows):
|
|
self.grid_contents[( 0, r)] = gtk.Label("%d" % (r + 1))
|
|
self.grid_contents[( 1, r)] = gtk.Label("player %d" % (r + 1))
|
|
self.grid_contents[( 4, r)] = gtk.Label("-")
|
|
self.grid_contents[( 9, r)] = gtk.Label("-")
|
|
self.grid_contents[( 2, r)] = self.eb[( 0, r)]
|
|
self.grid_contents[( 3, r)] = self.eb[( 1, r)]
|
|
self.grid_contents[( 5, r)] = self.eb[( 2, r)]
|
|
self.grid_contents[( 6, r)] = self.eb[( 3, r)]
|
|
self.grid_contents[( 7, r)] = self.eb[( 4, r)]
|
|
self.grid_contents[( 8, r)] = self.eb[( 5, r)]
|
|
self.grid_contents[(10, r)] = self.eb[( 6, r)]
|
|
for c in range(0, self.cols):
|
|
self.eb[(c, r)].add(self.seen_cards[(c, r)])
|
|
|
|
# add the cell contents to the table
|
|
for c in range(0, self.cols + 4):
|
|
for r in range(0, self.rows):
|
|
self.grid.attach(self.grid_contents[(c, r)], c, c+1, r, r+1, xpadding = 1, ypadding = 1)
|
|
|
|
self.parent.add(self.grid)
|
|
|
|
def update(self, new_hand_id):
|
|
cards = self.db_connection.get_cards(new_hand_id)
|
|
self.clear()
|
|
|
|
for c in cards.keys():
|
|
self.grid_contents[(1, cards[c]['seat_number'] - 1)].set_text(cards[c]['screen_name'])
|
|
|
|
for i in ((0, 'hole_card_1'), (1, 'hole_card_2'), (2, 'hole_card_3'), (3, 'hole_card_4'),
|
|
(4, 'hole_card_5'), (5, 'hole_card_6'), (6, 'hole_card_7')):
|
|
if not cards[c][i[1]] == "":
|
|
self.seen_cards[(i[0], cards[c]['seat_number'] - 1)]. \
|
|
set_from_pixbuf(self.card_images[self.split_cards(cards[c][i[1]])])
|
|
|
|
xml_text = self.db_connection.get_xml(new_hand_id)
|
|
hh = HandHistory.HandHistory(xml_text, ('BETTING'))
|
|
|
|
# action in tool tips for 3rd street cards
|
|
tip = "%s" % hh.BETTING.rounds[0]
|
|
for c in (0, 1, 2):
|
|
for r in range(0, self.rows):
|
|
self.eb[(c, r)].set_tooltip_text(tip)
|
|
|
|
# action in tools tips for later streets
|
|
round_to_col = (0, 3, 4, 5, 6)
|
|
for round in range(1, len(hh.BETTING.rounds)):
|
|
tip = "%s" % hh.BETTING.rounds[round]
|
|
for r in range(0, self.rows):
|
|
self.eb[(round_to_col[round], r)].set_tooltip_text(tip)
|
|
|
|
def split_cards(self, card):
|
|
return (card[0], card[1].upper())
|
|
|
|
def clear(self):
|
|
for r in range(0, self.rows):
|
|
self.grid_contents[(1, r)].set_text(" ")
|
|
for c in range(0, 7):
|
|
self.seen_cards[(c, r)].set_from_pixbuf(self.card_images[('B', 'S')])
|
|
self.eb[(c, r)].set_tooltip_text('')
|
|
def get_card_images(self):
|
|
card_images = {}
|
|
suits = ('S', 'H', 'D', 'C')
|
|
ranks = ('A', 'K', 'Q', 'J', 'T', '9', '8', '7', '6', '5', '4', '3', '2', 'B')
|
|
pb = gtk.gdk.pixbuf_new_from_file("Cards01.png")
|
|
|
|
for j in range(0, 14):
|
|
for i in range(0, 4):
|
|
temp_pb = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, pb.get_has_alpha(), pb.get_bits_per_sample(), 30, 42)
|
|
pb.copy_area(30*j, 42*i, 30, 42, temp_pb, 0, 0)
|
|
card_images[(ranks[j], suits[i])] = temp_pb
|
|
return(card_images)
|
|
|
|
# cards are 30 wide x 42 high
|
|
|
|
if __name__== "__main__":
|
|
|
|
def destroy(*args): # call back for terminating the main eventloop
|
|
gtk.main_quit() # used only for testing
|
|
|
|
def process_new_hand(source, condition): #callback from stdin watch -- testing only
|
|
# there is a new hand_id to be processed
|
|
# just read it and pass it to update
|
|
new_hand_id = sys.stdin.readline()
|
|
new_hand_id = new_hand_id.rstrip() # remove trailing whitespace
|
|
m.update(new_hand_id)
|
|
return(True)
|
|
|
|
config = Configuration.Config()
|
|
db_connection = Database.Database(config, 'PTrackSv2', 'razz')
|
|
|
|
main_window = gtk.Window()
|
|
main_window.set_keep_above(True)
|
|
main_window.connect("destroy", destroy)
|
|
|
|
m = Mucked(main_window, db_connection)
|
|
main_window.show_all()
|
|
|
|
s_id = gobject.io_add_watch(sys.stdin, gobject.IO_IN, process_new_hand)
|
|
|
|
gtk.main()
|
|
|