if __name__ == "__main__":
datfname = sys.argv[1]
jpgfname = sys.argv[2]
start = sys.argv[3]
end = sys.argv[4]
increment = sys.argv[5]
dt = datfname.split('.dat')[0].split('_')[2]
if float(increment) % 1 == 0:
start = int(start)
end = int(end)
increment = int(increment)
xtics = range(start, end+increment, increment)
elif float(increment) == 0.5:
start = int(float(start)*10.0)
end = int(float(end)*10.0)
increment = 5
strikes = range(start, end+increment, increment)
xticks = [x/10.0 for x in strikes]
yticks = [x/10.0 for x in range(0, 11)]
gpb = GNUPlotBase()
gpb.xmin = start*.9
gpb.xmax = end*1.1
gpb.ymin = 0
gpb.ymax = 1
commands = []
commands.append(gpb.set_output(fname=jpgfname))
commands.append(gpb.gen_header(ylabel='Probability', timestamp=dt))
commands.append(gpb.gen_ticks(xticks, yticks))
commands.append('plot "%s" with boxes' % datfname)
print '\n'.join(commands)
jpgfname = sys.argv[2]
start = sys.argv[3]
end = sys.argv[4]
increment = sys.argv[5]
dt = datfname.split('.dat')[0].split('_')[2]
if float(increment) % 1 == 0:
start = int(start)
end = int(end)
increment = int(increment)
xtics = range(start, end+increment, increment*2)
elif float(increment) == 0.5:
start = int(float(start)*10.0)
end = int(float(end)*10.0)
increment = 5
strikes = range(start, end+increment, increment*2)
xticks = [x/10.0 for x in strikes]
gpb = GNUPlotBase()
gpb.xmin = start*.9
gpb.xmax = end*1.1
gpb.ymin = 0
gpb.ymax = xticks[1] - xticks[0]
yticks = [float(gpb.ymax)/5.0*x for x in range(0, 6)]
commands = []
commands.append(gpb.set_output(fname=jpgfname))
commands.append(gpb.gpbase.gen_header(xlabel='Spot Price at Expiry',
ylabel='Butterfly Price',
timestamp=timestamp))
commands.append(gpb.gen_ticks(xticks, yticks))
commands.append('plot "%s" with boxes' % datfname)
print '\n'.join(commands)
