baseballstats/graphgame.cgi at master · gregstoll/baseballstats · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
#!/usr/bin/python3

import cgi, os, sys, re, os.path, urllib.parse, shutil, time
import matplotlib
# Turn off DISPLAY
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import ticker
from getstats import getProbability
from getstats import getProbabilityOfString
from tempfile import mkstemp

form = cgi.FieldStorage()
print('Content-type: text/html\n\n')
print('<!DOCTYPE HTML>')
print('<html><head><title>Win probability graph</title></head><body>')
print('<p>These image files are cleaned out every few hours, so please download a copy of the image if you want to keep it. Or see <a href=\"#text\">below</a> to save a text representation that can be easily re-entered.</p>')
#for key in form:
    #print '<p>%s: %s</p>' % (key, form[key].value)
hasDirectText = False
for key in form:
    situationMatch = re.match(r'^situation(\d+)$', key)
    if (situationMatch):
        lastSituation = int(situationMatch.group(1))
    elif (key == 'directtext'):
        hasDirectText = True
        directSituationLines = form["directtext"].value.split("\n")
        directSituationLines = [line.strip() for line in directSituationLines]
        directSituationLines = [line for line in directSituationLines if line != '']
        directSituationLines = [line for line in directSituationLines if (not line.startswith('//'))]
        lastSituation = len(directSituationLines) - 1
title = form["title"].value
if ("doKey" in form):
    doKey = form["doKey"].value
else:
    doKey = False
if ("doRunsScored" in form):
    doRunsScored = form["doRunsScored"].value
else:
    doRunsScored = False
if (not doRunsScored):
    doKey = False
probs = []
xValues = []
inningStarts = []
# Stores (frame number, change in score)
runEntries = {}
runEntries['H'] = []
runEntries['V'] = []
lastScoreDiff = 0
lastHomeOrVisitor = 'H'
for i in range(0, lastSituation + 1):
    if (not hasDirectText):
        inningCombined = form["inning%d" % i].value
        outs = int(form["outs%d" % i].value)
        runners = int(form["runner%d" % i].value)
        scoreDiff = int(form["score%d" % i].value)
        if (len(inningStarts) == 0 or inningCombined != inningStarts[-1][0]):
            inningStarts.append((inningCombined, i))
        homeOrVisitor = inningCombined[0]
        inning = int(inningCombined[1:])
        prob = getProbability(homeOrVisitor, inning, outs, runners, scoreDiff)
    else:
        curLine = directSituationLines[i]
        curLine = curLine.strip()
        curLineComponents = curLine.split(',')
        homeOrVisitor = curLineComponents[0][1:2]
        inning = int(curLineComponents[1])
        inningCombined = homeOrVisitor + str(inning)
        runners = int(curLineComponents[3])
        outs = int(curLineComponents[2])
        scoreDiff = int(curLineComponents[4])
        if (len(inningStarts) == 0 or inningCombined != inningStarts[-1][0]):
            inningStarts.append((inningCombined, i))
        prob = getProbabilityOfString(curLine)
    # Keep a list of x-values, too
    if (prob != -1):
        # We want to show home probabilities
        if (homeOrVisitor == 'V'):
            prob = 1 - prob
        probs.append(prob)
        xValues.append(i)
    else:
        # Correct for end conditions
        if (inning >= 9 and homeOrVisitor == 'H' and scoreDiff > 0):
            prob = 1
            probs.append(prob)
            xValues.append(i)
        elif (inning > 9 and homeOrVisitor == 'V' and scoreDiff > 0):
            prob = 0
            probs.append(prob)
            xValues.append(i)
        else:
            # We should never see this value in the graph, but we need to keep
            # the probs list in sync.
            # TODO - put a marker on the graph?
            probs.append(-1)
    if (homeOrVisitor == lastHomeOrVisitor):
        if (scoreDiff != lastScoreDiff):
            runEntries[homeOrVisitor].append((i, scoreDiff - lastScoreDiff))
    else:
        if (scoreDiff != -1 * lastScoreDiff):
            runEntries[lastHomeOrVisitor].append((i, -1 * lastScoreDiff - scoreDiff))
    lastHomeOrVisitor = homeOrVisitor
    lastScoreDiff = scoreDiff
if (inningStarts[-1][1] == lastSituation):
    # This is just an entry to show the final outcome of the game, so don't
    # plot on x-axis.
    inningStarts = inningStarts[:-1]
#for key in os.environ:
    #print "<p>%s: %s</p>" % (key, os.environ[key])
(tempPngFile, tempPngFileName) = mkstemp(suffix=".png")
os.close(tempPngFile)

axes = plt.axes()
plt.ylim(0,1)
plt.xlim(0, max(xValues))
plt.ylabel("Win Probability")
plt.title(title)
plt.plot(xValues, probs, 'g')

xTicks = []
for entry in inningStarts:
    combinedInning = entry[0]
    index = entry[1]
    xTicks.append((index, combinedInning))
plt.yticks(np.linspace(0, 1, num=5, endpoint=True))
def y_axis_formatter(x, p):
    return "%d%%" % (int(x * 100),)
axes.yaxis.set_major_formatter(ticker.FuncFormatter(y_axis_formatter))
plt.xticks(rotation=90)
plt.xticks([x[0] for x in xTicks])
axes.set_xticklabels([x[1] for x in xTicks])
plt.grid(axis='both', linestyle='-', color='0.5')

# Find the points where runs were scored.
if (doRunsScored):
    pointsToPlot = {}
    for homeOrVisitor in list(runEntries.keys()):
        pointsToPlot[homeOrVisitor] = []
        for runTuple in runEntries[homeOrVisitor]:
            index = runTuple[0]
            yVal = probs[index]
            pointsToPlot[homeOrVisitor].append((index, yVal, runTuple[1]))
#if (len(xTicksString) > 2):
#    g('set xtics %s' % xTicksString)
#    g('set grid ytics xtics')
#else:
#    g('set grid ytics')
#g.plot(list(zip(xValues, probs)))
# Plot the points where runs were scored
# Indicate how many runs scored by the number of boxes around it (the boxes
# get bigger)
def plotRunsScored(pointsToPlot, homeOrVisitorName, color, doKey):
    haveTitled = False
    for pointToPlot in pointsToPlot:
        for pointSize in range(1, pointToPlot[2] + 1):
            label = None
            if (doKey and not haveTitled and pointSize == 1):
                label = 'Runs for %s' % (homeOrVisitorName,)
                haveTitled = True
            pointPlot = plt.scatter(pointToPlot[0], pointToPlot[1], 25*pointSize*pointSize, edgecolor=color, facecolor=(1, 1, 1, 0), marker='s', label=label)

if (doRunsScored):
    if ('H' in pointsToPlot and len(pointsToPlot['H']) > 0):
        plotRunsScored(pointsToPlot['H'], 'Home', 'b', doKey)
    if ('V' in pointsToPlot and len(pointsToPlot['V']) > 0):
        plotRunsScored(pointsToPlot['V'], 'Visitor', 'r', doKey)

if doKey:
    plt.legend(loc='best', scatterpoints=1)
pngOptions = ""
# TODO - make the size an option.
if (xValues[-1] >= 100):
    # More than 100 entries means we're kind of long, so make the graph bigger.
    if (xValues[-1] >= 150):
        pngOptions = "size %d, %d" % (1024, 768)
    else:
        pngOptions = "size %d, %d" % (800, 600)
plt.savefig(tempPngFileName)
time.sleep(0.5)
pictureName = os.path.join(os.getcwd() + '/images', os.path.basename(tempPngFileName))
#print "<p>%s to %s</p>" % (newTempName, pictureName)
shutil.copyfile(tempPngFileName, pictureName)
urlName = urllib.parse.urljoin(os.environ['SCRIPT_URI'], 'images/' + os.path.basename(pictureName))
print('<img src="%s" alt="Win probability graph">' % urlName)
if (doRunsScored):
    print('<p>Note that the number of boxes around a point indicate the number of runs scored on that play.</p>')
print('<p id=\"text\">Here is a text representation of the game that you can use to input again:</p>')
print('<pre>')
# print out textual representation
for i in range(0, lastSituation + 1):
    if (not hasDirectText):
        inningCombined = form["inning%d" % i].value
        outs = int(form["outs%d" % i].value)
        runners = int(form["runner%d" % i].value)
        scoreDiff = int(form["score%d" % i].value)
        homeOrVisitor = inningCombined[0]
        inning = int(inningCombined[1:])
        print('"%s",%d,%d,%d,%d' % (homeOrVisitor, inning, outs, runners, scoreDiff))
    else:
        print(directSituationLines[i])
print('</pre>')
# TODO - output .csv representation of probs or something?
print('<script src="//www.google-analytics.com/urchin.js" type="text/javascript"></script>')
print('<script type="text/javascript">')
print('_uacct = "UA-362292-1";')
print('urchinTracker();')
print('</script>')
print('</body></html>')