331 lines
13 KiB
Python
331 lines
13 KiB
Python
#!/usr/bin/python
|
|
|
|
#Copyright 2008 Steffen Jobbagy-Felso
|
|
#This program is free software: you can redistribute it and/or modify
|
|
#it under the terms of the GNU Affero General Public License as published by
|
|
#the Free Software Foundation, version 3 of the License.
|
|
#
|
|
#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 Affero General Public License
|
|
#along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#In the "official" distribution you can find the license in
|
|
#agpl-3.0.txt in the docs folder of the package.
|
|
|
|
import threading
|
|
import pygtk
|
|
pygtk.require('2.0')
|
|
import gtk
|
|
import os
|
|
import sys
|
|
import traceback
|
|
from time import *
|
|
#import pokereval
|
|
|
|
try:
|
|
import matplotlib
|
|
matplotlib.use('GTK')
|
|
from matplotlib.figure import Figure
|
|
from matplotlib.backends.backend_gtk import FigureCanvasGTK as FigureCanvas
|
|
from matplotlib.backends.backend_gtkagg import NavigationToolbar2GTKAgg as NavigationToolbar
|
|
from matplotlib.font_manager import FontProperties
|
|
from numpy import arange, cumsum
|
|
from pylab import *
|
|
except ImportError, inst:
|
|
print """Failed to load libs for graphing, graphing will not function. Please in
|
|
stall numpy and matplotlib if you want to use graphs."""
|
|
print """This is of no consequence for other parts of the program, e.g. import
|
|
and HUD are NOT affected by this problem."""
|
|
print "ImportError: %s" % inst.args
|
|
|
|
import fpdb_import
|
|
import Database
|
|
import Filters
|
|
import Charset
|
|
|
|
class GuiGraphViewer (threading.Thread):
|
|
|
|
def __init__(self, querylist, config, debug=True):
|
|
"""Constructor for GraphViewer"""
|
|
self.sql = querylist
|
|
self.conf = config
|
|
self.debug = debug
|
|
#print "start of GraphViewer constructor"
|
|
self.db = Database.Database(self.conf, sql=self.sql)
|
|
|
|
|
|
filters_display = { "Heroes" : True,
|
|
"Sites" : True,
|
|
"Games" : True,
|
|
"Limits" : True,
|
|
"LimitSep" : True,
|
|
"LimitType" : True,
|
|
"Type" : False,
|
|
"UseType" : 'ring',
|
|
"Seats" : False,
|
|
"SeatSep" : False,
|
|
"Dates" : True,
|
|
"Groups" : False,
|
|
"Button1" : True,
|
|
"Button2" : True
|
|
}
|
|
|
|
self.filters = Filters.Filters(self.db, self.conf, self.sql, display = filters_display)
|
|
self.filters.registerButton1Name("Refresh _Graph")
|
|
self.filters.registerButton1Callback(self.generateGraph)
|
|
self.filters.registerButton2Name("_Export to File")
|
|
self.filters.registerButton2Callback(self.exportGraph)
|
|
|
|
self.mainHBox = gtk.HBox(False, 0)
|
|
self.mainHBox.show()
|
|
|
|
self.leftPanelBox = self.filters.get_vbox()
|
|
|
|
self.hpane = gtk.HPaned()
|
|
self.hpane.pack1(self.leftPanelBox)
|
|
self.mainHBox.add(self.hpane)
|
|
# hierarchy: self.mainHBox / self.hpane / self.graphBox / self.canvas / self.fig / self.ax
|
|
|
|
self.graphBox = gtk.VBox(False, 0)
|
|
self.graphBox.show()
|
|
self.hpane.pack2(self.graphBox)
|
|
self.hpane.show()
|
|
|
|
self.fig = None
|
|
#self.exportButton.set_sensitive(False)
|
|
self.canvas = None
|
|
|
|
|
|
self.db.rollback()
|
|
|
|
def get_vbox(self):
|
|
"""returns the vbox of this thread"""
|
|
return self.mainHBox
|
|
#end def get_vbox
|
|
|
|
def clearGraphData(self):
|
|
|
|
try:
|
|
try:
|
|
if self.canvas:
|
|
self.graphBox.remove(self.canvas)
|
|
except:
|
|
pass
|
|
|
|
if self.fig != None:
|
|
self.fig.clear()
|
|
self.fig = Figure(figsize=(5,4), dpi=100)
|
|
if self.canvas is not None:
|
|
self.canvas.destroy()
|
|
|
|
self.canvas = FigureCanvas(self.fig) # a gtk.DrawingArea
|
|
except:
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
|
print "***Error: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
|
raise
|
|
|
|
def generateGraph(self, widget, data):
|
|
try:
|
|
self.clearGraphData()
|
|
|
|
sitenos = []
|
|
playerids = []
|
|
|
|
sites = self.filters.getSites()
|
|
heroes = self.filters.getHeroes()
|
|
siteids = self.filters.getSiteIds()
|
|
limits = self.filters.getLimits()
|
|
games = self.filters.getGames()
|
|
|
|
for i in ('show', 'none'):
|
|
if i in limits:
|
|
limits.remove(i)
|
|
# Which sites are selected?
|
|
for site in sites:
|
|
if sites[site] == True:
|
|
sitenos.append(siteids[site])
|
|
c = self.db.get_cursor()
|
|
_hname = Charset.to_utf8(heroes[site])
|
|
# Nasty hack to deal with multiple sites + same player name -Eric
|
|
que = self.sql.query['getPlayerId'] + " AND siteId=%d" % siteids[site]
|
|
c.execute(que, (_hname,))
|
|
result = c.fetchall()
|
|
if len(result) == 1:
|
|
playerids.append( int(result[0][0]) )
|
|
|
|
if not sitenos:
|
|
#Should probably pop up here.
|
|
print "No sites selected - defaulting to PokerStars"
|
|
self.db.rollback()
|
|
return
|
|
|
|
if not playerids:
|
|
print "No player ids found"
|
|
self.db.rollback()
|
|
return
|
|
|
|
if not limits:
|
|
print "No limits found"
|
|
self.db.rollback()
|
|
return
|
|
|
|
#Set graph properties
|
|
self.ax = self.fig.add_subplot(111)
|
|
|
|
#Get graph data from DB
|
|
starttime = time()
|
|
(green, blue, red) = self.getRingProfitGraph(playerids, sitenos, limits, games)
|
|
print "Graph generated in: %s" %(time() - starttime)
|
|
|
|
self.ax.set_title("Profit graph for ring games")
|
|
|
|
#Set axis labels and grid overlay properites
|
|
self.ax.set_xlabel("Hands", fontsize = 12)
|
|
self.ax.set_ylabel("$", fontsize = 12)
|
|
self.ax.grid(color='g', linestyle=':', linewidth=0.2)
|
|
if green == None or green == []:
|
|
|
|
#TODO: Do something useful like alert user
|
|
print "No hands returned by graph query"
|
|
else:
|
|
#text = "Profit: $%.2f\nTotal Hands: %d" %(green[-1], len(green))
|
|
#self.ax.annotate(text,
|
|
# xy=(10, -10),
|
|
# xycoords='axes points',
|
|
# horizontalalignment='left', verticalalignment='top',
|
|
# fontsize=10)
|
|
|
|
#Draw plot
|
|
self.ax.plot(green, color='green', label='Hands: %d\nProfit: $%.2f' %(len(green), green[-1]))
|
|
self.ax.plot(blue, color='blue', label='Showdown: $%.2f' %(blue[-1]))
|
|
self.ax.plot(red, color='red', label='Non-showdown: $%.2f' %(red[-1]))
|
|
if sys.version[0:3] == '2.5':
|
|
self.ax.legend(loc='best', shadow=True, prop=FontProperties(size='smaller'))
|
|
else:
|
|
self.ax.legend(loc='best', fancybox=True, shadow=True, prop=FontProperties(size='smaller'))
|
|
|
|
|
|
self.graphBox.add(self.canvas)
|
|
self.canvas.show()
|
|
self.canvas.draw()
|
|
#self.exportButton.set_sensitive(True)
|
|
except:
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
|
print "***Error: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
|
|
|
#end of def showClicked
|
|
|
|
def getRingProfitGraph(self, names, sites, limits, games):
|
|
tmp = self.sql.query['getRingProfitAllHandsPlayerIdSite']
|
|
# print "DEBUG: getRingProfitGraph"
|
|
start_date, end_date = self.filters.getDates()
|
|
|
|
#Buggered if I can find a way to do this 'nicely' take a list of integers and longs
|
|
# and turn it into a tuple readale by sql.
|
|
# [5L] into (5) not (5,) and [5L, 2829L] into (5, 2829)
|
|
nametest = str(tuple(names))
|
|
sitetest = str(tuple(sites))
|
|
#nametest = nametest.replace("L", "")
|
|
|
|
q = []
|
|
for m in self.filters.display.items():
|
|
if m[0] == 'Games' and m[1]:
|
|
for n in games:
|
|
if games[n]:
|
|
q.append(n)
|
|
if len(q) > 0:
|
|
gametest = str(tuple(q))
|
|
gametest = gametest.replace("L", "")
|
|
gametest = gametest.replace(",)",")")
|
|
gametest = gametest.replace("u'","'")
|
|
gametest = "and gt.category in %s" % gametest
|
|
else:
|
|
gametest = "and gt.category IS NULL"
|
|
tmp = tmp.replace("<game_test>", gametest)
|
|
|
|
lims = [int(x) for x in limits if x.isdigit()]
|
|
potlims = [int(x[0:-2]) for x in limits if len(x) > 2 and x[-2:] == 'pl']
|
|
nolims = [int(x[0:-2]) for x in limits if len(x) > 2 and x[-2:] == 'nl']
|
|
limittest = "and ( (gt.limitType = 'fl' and gt.bigBlind in "
|
|
# and ( (limit and bb in()) or (nolimit and bb in ()) )
|
|
if lims:
|
|
blindtest = str(tuple(lims))
|
|
blindtest = blindtest.replace("L", "")
|
|
blindtest = blindtest.replace(",)",")")
|
|
limittest = limittest + blindtest + ' ) '
|
|
else:
|
|
limittest = limittest + '(-1) ) '
|
|
limittest = limittest + " or (gt.limitType = 'pl' and gt.bigBlind in "
|
|
if potlims:
|
|
blindtest = str(tuple(potlims))
|
|
blindtest = blindtest.replace("L", "")
|
|
blindtest = blindtest.replace(",)",")")
|
|
limittest = limittest + blindtest + ' ) '
|
|
else:
|
|
limittest = limittest + '(-1) ) '
|
|
limittest = limittest + " or (gt.limitType = 'nl' and gt.bigBlind in "
|
|
if nolims:
|
|
blindtest = str(tuple(nolims))
|
|
blindtest = blindtest.replace("L", "")
|
|
blindtest = blindtest.replace(",)",")")
|
|
limittest = limittest + blindtest + ' ) )'
|
|
else:
|
|
limittest = limittest + '(-1) ) )'
|
|
|
|
if type == 'ring':
|
|
limittest = limittest + " and gt.type = 'ring' "
|
|
elif type == 'tour':
|
|
limittest = limittest + " and gt.type = 'tour' "
|
|
|
|
#Must be a nicer way to deal with tuples of size 1 ie. (2,) - which makes sql barf
|
|
tmp = tmp.replace("<player_test>", nametest)
|
|
tmp = tmp.replace("<site_test>", sitetest)
|
|
tmp = tmp.replace("<startdate_test>", start_date)
|
|
tmp = tmp.replace("<enddate_test>", end_date)
|
|
tmp = tmp.replace("<limit_test>", limittest)
|
|
tmp = tmp.replace(",)", ")")
|
|
|
|
#print "DEBUG: sql query:"
|
|
#print tmp
|
|
self.db.cursor.execute(tmp)
|
|
#returns (HandId,Winnings,Costs,Profit)
|
|
winnings = self.db.cursor.fetchall()
|
|
self.db.rollback()
|
|
|
|
if winnings == ():
|
|
return None
|
|
|
|
green = map(lambda x:float(x[1]), winnings)
|
|
blue = map(lambda x: float(x[1]) if x[2] == True else 0.0, winnings)
|
|
red = map(lambda x: float(x[1]) if x[2] == False else 0.0, winnings)
|
|
greenline = cumsum(green)
|
|
blueline = cumsum(blue)
|
|
redline = cumsum(red)
|
|
return (greenline/100, blueline/100, redline/100)
|
|
#end of def getRingProfitGraph
|
|
|
|
def exportGraph (self, widget, data):
|
|
if self.fig is None:
|
|
return # Might want to disable export button until something has been generated.
|
|
dia_chooser = gtk.FileChooserDialog(title="Please choose the directory you wish to export to:",
|
|
action=gtk.FILE_CHOOSER_ACTION_OPEN,
|
|
buttons=(gtk.STOCK_CANCEL,gtk.RESPONSE_CANCEL,gtk.STOCK_OPEN,gtk.RESPONSE_OK))
|
|
#TODO: Suggest path and filename to start with
|
|
|
|
response = dia_chooser.run()
|
|
if response == gtk.RESPONSE_OK:
|
|
self.exportDir = dia_chooser.get_filename()
|
|
print "DEBUG: self.exportDir = %s" %(self.exportDir)
|
|
elif response == gtk.RESPONSE_CANCEL:
|
|
print 'Closed, no graph exported'
|
|
dia_chooser.destroy()
|
|
#TODO: Check to see if file exists
|
|
#NOTE: Dangerous - will happily overwrite any file we have write access too
|
|
#TODO: This asks for a directory but will take a filename and overwrite it.
|
|
self.fig.savefig(self.exportDir, format="png")
|
|
|
|
|