import ROOT, argparse

parser=argparse.ArgumentParser(description="plot number of steps taken by devmap solver")
parser.add_argument("infile",help="the input devmap",type=str)

args=parser.parse_args()

steps=ROOT.TH1D("hsteps","steps",1000,0,0)
linecounter=-1
for line in open(args.infile,'r'):
    linecounter+=1
    words=line.split()
    if linecounter==0:
        rbins=int(words[0])
        zbins=int(words[1])
        rmin=float(words[2])
        rmax=float(words[3])
        zmin=float(words[4])
        zmax=float(words[5])
        steps2d=ROOT.TH2D("hsteps2d","steps2D",rbins,rmin,rmax,zbins,zmin,zmax)
        continue

    steps.Fill(int(words[3]))
    steps2d.SetBinContent(linecounter-1,int(words[3]))

c1=ROOT.TCanvas("c1","c1")
steps.Draw()
c2=ROOT.TCanvas("c2","c2")
steps2d.Draw("colz")

u=raw_input("done?")