#!/usr/bin/python

import mariosolver
import Gnuplot

mariosolver.setupSpaces()
g = Gnuplot.Gnuplot()
g('set data style lines')
g('set xrange[0:100]')
tempFile = open('temp', 'w')
for c in range(101):
    (n, d) = mariosolver.getProbabilityOfSuccess(94, c, (True, True), frozenset())
    (n1, d1) = mariosolver.getProbabilityOfSuccess(94, c, (True, False), frozenset())
    (n2, d2) = mariosolver.getProbabilityOfSuccess(94, c, (False, True), frozenset())
    (n3, d3) = mariosolver.getProbabilityOfSuccess(94, c, (False, False), frozenset())
    tempFile.write('%d %.10f %.10f %.10f %.10f\n' % (c, n/d, n1/d1, n2/d2, n3/d3))
tempFile.close()
g('set terminal png medium')
g('set output "plot.png"')
g('set ylabel "Probability of winning"')
g('set xlabel "Starting number of coins"')
g('set grid xtics ytics')
g('set key right bottom')
g('plot "temp" using 1:2 title "Normal", "temp" using 1:3 title "No picking up coins", "temp" using 1:4 title "No obstacles", "temp" using 1:5 title "No obstacles or picking up coins')