import ROOT, argparse, operator, numpy, math, sys
sys.path.append('$PANDAPATH/macro/tpc/FOPI/mberger')
from functions import *

parser=argparse.ArgumentParser(description='Plot stuff along a track')
parser.add_argument('filename',help='filename of the reco file',type=str)
parser.add_argument('--rootfilename', help='name of the root file', type=str, default="not")
parser.add_argument('--cut', help='Real Data?', type=str, default="not")
#parser.add_argument('--doallmode',help='if you want to do more', type=str)
parser.add_argument("--nEvts",help='number of events to analyze',default=-1,type=int)
parser.add_argument('--zcut',help='zcut',type=int,default=-80)
parser.add_argument('--pdir',help='the picture print directory',type=str,default="not")
parser.add_argument('--step', help='step1 oder sqrt', type=str, default="not")
args=parser.parse_args()

ROOT.gROOT.ProcessLine(".x rootlogon.C") 
ROOT.gROOT.ProcessLine('gSystem->Load("libPhysics")')
ROOT.gROOT.ProcessLine('gStyle->SetPalette(1)')
ROOT.gROOT.ProcessLine('gROOT->SetStyle("Plain")')
ROOT.gROOT.ProcessLine('gSystem->Load("libGeom")')
ROOT.gROOT.LoadMacro("stlPYROOT.h+")

rfile=ROOT.TFile.Open(args.filename,"read")
if rfile.IsOpen()==ROOT.kTRUE:
    tree = rfile.Get("cbmsim")
if tree==None:
    print "No cbmsim found"
    if type(Rfile)==ROOT.TFile:
        Rfile.Close()
    exit()


#Histogramms
#{
Slices = [ -48.0, -36.0, -24.0, -12.0,  0.0, 20.0]
resbins = 40

htrkl1 = ROOT.TH1D("htrkl", "left - 1st sector", resbins,0,resbins)
htrkr1 = ROOT.TH1D("htrkr", "right - 1st sector", resbins,0,resbins)
hclsl1 = ROOT.TH1D("hclsl", "Clustercount - left - 1st sector", resbins,0,resbins)
hclsr1 = ROOT.TH1D("hclsr", "Clustercount - right - 1st sector", resbins,0,resbins)
htrk1 = ROOT.TH1D("htrkr", "1st sector", resbins,0,resbins)
hcls1 = ROOT.TH1D("hclsl", "Clustercount-1st sector", resbins,0,resbins)

htrkl2 = ROOT.TH1D("htrkl", "left - 2nd sector", resbins,0,resbins)
htrkr2 = ROOT.TH1D("htrkr", "right - 2nd sector", resbins,0,resbins)
hclsl2 = ROOT.TH1D("hclsl", "Clustercount - left - 2nd sector", resbins,0,resbins)
hclsr2 = ROOT.TH1D("hclsr", "Clustercount - right - 2nd sector", resbins,0,resbins)
htrk2 = ROOT.TH1D("htrkr", "2nd sector", resbins,0,resbins)
hcls2 = ROOT.TH1D("hclsl", "Clustercount-2nd sector", resbins,0,resbins)

htrkl3 = ROOT.TH1D("htrkl", "left - 3rd sector", resbins,0,resbins)
htrkr3 = ROOT.TH1D("htrkr", "right - 3rd sector", resbins,0,resbins)
hclsl3 = ROOT.TH1D("hclsl", "Clustercount - left - 3rd sector", resbins,0,resbins)
hclsr3 = ROOT.TH1D("hclsr", "Clustercount - right - 3rd sector", resbins,0,resbins)
htrk3 = ROOT.TH1D("htrkr", "3rd sector", resbins,0,resbins)
hcls3 = ROOT.TH1D("hclsl", "Clustercount-3rd sector", resbins,0,resbins)

htrkl4 = ROOT.TH1D("htrkl", "left - 4th sector", resbins,0,resbins)
htrkr4 = ROOT.TH1D("htrkr", "right - 4th sector", resbins,0,resbins)
hclsl4 = ROOT.TH1D("hclsl", "Clustercount - left - 4th sector", resbins,0,resbins)
hclsr4 = ROOT.TH1D("hclsr", "Clustercount - right - 4th sector", resbins,0,resbins)
htrk4 = ROOT.TH1D("htrkr", "4th sector", resbins,0,resbins)
hcls4 = ROOT.TH1D("hclsl", "Clustercount-4th sector", resbins,0,resbins)

htrkl5 = ROOT.TH1D("htrkl", "left - 5th sector", resbins,0,resbins)
htrkr5 = ROOT.TH1D("htrkr", "right - 5th sector", resbins,0,resbins)
hclsl5 = ROOT.TH1D("hclsl", "Clustercount - left - 5th sector", resbins,0,resbins)
hclsr5 = ROOT.TH1D("hclsr", "Clustercount - right - 5th sector", resbins,0,resbins)
htrk5 = ROOT.TH1D("htrkr", "5th sector", resbins,0,resbins)
hcls5 = ROOT.TH1D("hclsl", "Clustercount-5th sector", resbins,0,resbins)

htrkl6 = ROOT.TH1D("htrkl", "left - 6th sector", resbins,0,resbins)
htrkr6 = ROOT.TH1D("htrkr", "right - 6th sector", resbins,0,resbins)
hclsl6 = ROOT.TH1D("hclsl", "Clustercount - left - 6th sector", resbins,0,resbins)
hclsr6 = ROOT.TH1D("hclsr", "Clustercount - right - 6th sector", resbins,0,resbins)
htrk6 = ROOT.TH1D("htrkr", "6th sector", resbins,0,resbins)
hcls6 = ROOT.TH1D("hclsl", "Clustercount-6th sector", resbins,0,resbins)

bins = 60
clbins = 15
hclpos1 = ROOT.TH2D("hclpos1", "Clusterpos xy - right", 60,-15,15,60,-15,15)
hclpos2 = ROOT.TH2D("hclpos2", "Clusterpos xy - left", 60,-15,15,60,-15,15)
c1 = ROOT.TCanvas("c1", "Plot 1", 1000,700) 
c2 = ROOT.TCanvas("c2", "Plot 2", 1000,700)
c3 = ROOT.TCanvas("c3", "Plot 3", 1000,700)
c4 = ROOT.TCanvas("c4", "Plot 4", 700,1000)

c1.Divide(3,2)
c2.Divide(3,2)
c3.Divide(3,2)
c4.Divide(1,2)

#}

#cloose divide
#{
if args.step=="sqrt":
    div1=0.87
    div2=0.83
    div3=0.77
    div4=0.74
    div5=0.72
    div6=0.70

if args.step=="step1":
    div1=1.08
    div2=1.08
    div3=1.08
    div4=1.08
    div5=1.08
    div6=1.08
#}

#Loop over tree
#{
tree.SetBranchStatus("*", 0)
#tree.SetBranchStatus(args.branch+".*",1)
tree.SetBranchStatus("CutCosmics.*",1)

trackbranch=ROOT.TBranch()
evt=-1
done=0
olddone=0
if args.nEvts==-1:
    todoev=tree.GetEntriesFast()
else:
    todoev=args.nEvts
for e in tree:
    evt+=1
    if args.nEvts!=-1 and args.nEvts<evt:
        break
    
    eventcounter = 0
    trackbranch=e.CutCosmics
    for tr in trackbranch:

#Get-Parameter{
        size2D=tr.Get2DClSizes()
        size3D=tr.GetClSizes()
        length=tr.GetLength()
        cl_count=len(size3D)
        meanz=tr.GetMeanZ()
        cl_X=tr.GetHitPositionsX()
        cl_Y=tr.GetHitPositionsY()
       
        if length!=0:
            cpc=cl_count/length
        else:
            cpc=-1
        
        for cl in range(len(cl_X)):
            if cpc>div6:
                hclpos1.Fill(cl_X[cl],cl_Y[cl])
            if cpc<div6:
                hclpos2.Fill(cl_X[cl],cl_Y[cl])
        
        if meanz<-48.0:
            htrk1.Fill(length)
            hcls1.Fill(cl_count) 
            if cpc<div1:
                htrkl1.Fill(length)
                hclsl1.Fill(cl_count)
            else:
                htrkr1.Fill(length)
                hclsr1.Fill(cl_count)
                
        if -48<meanz<-36.0:
            htrk2.Fill(length)
            hcls2.Fill(cl_count) 
            if cpc<div2:
                htrkl2.Fill(length)
                hclsl2.Fill(cl_count)
            else:
                htrkr2.Fill(length)
                hclsr2.Fill(cl_count)
      
        if -36.0<meanz<-24.0:
            htrk3.Fill(length)
            hcls3.Fill(cl_count) 
            if cpc<div3:
                htrkl3.Fill(length)
                hclsl3.Fill(cl_count)
            else:
                htrkr3.Fill(length)
                hclsr3.Fill(cl_count)
           
        if -24.0<meanz<-12.0:
            htrk4.Fill(length)
            hcls4.Fill(cl_count) 
            if cpc<div4:
                htrkl4.Fill(length)
                hclsl4.Fill(cl_count)
            else:
                htrkr4.Fill(length)
                hclsr4.Fill(cl_count)

        if -12.0<meanz<0.0:
            htrk5.Fill(length)
            hcls5.Fill(cl_count) 
            if cpc<div5:
                htrkl5.Fill(length)
                hclsl5.Fill(cl_count)
            else:
                htrkr5.Fill(length)
                hclsr5.Fill(cl_count)
       
        if 0<meanz:
            htrk6.Fill(length)
            hcls6.Fill(cl_count) 
            if cpc<div6:
                htrkl6.Fill(length)
                hclsl6.Fill(cl_count)
            else:
                htrkr6.Fill(length)
                hclsr6.Fill(cl_count) 

htrkl1.GetXaxis().SetTitle("[cm]")
htrkl1.GetYaxis().SetTitle("count")

htrkl2.GetXaxis().SetTitle("[cm]")
htrkl2.GetYaxis().SetTitle("count")

htrkl3.GetXaxis().SetTitle("[cm]")
htrkl3.GetYaxis().SetTitle("count")

htrkl4.GetXaxis().SetTitle("[cm]")
htrkl4.GetYaxis().SetTitle("count")

htrkl5.GetXaxis().SetTitle("[cm]")
htrkl5.GetYaxis().SetTitle("count")

htrkl6.GetXaxis().SetTitle("[cm]")
htrkl6.GetYaxis().SetTitle("count")

htrk1.GetXaxis().SetTitle("[cm]")
htrk1.GetYaxis().SetTitle("count")

htrk2.GetXaxis().SetTitle("[cm]")
htrk2.GetYaxis().SetTitle("count")

htrk3.GetXaxis().SetTitle("[cm]")
htrk3.GetYaxis().SetTitle("count")

htrk4.GetXaxis().SetTitle("[cm]")
htrk4.GetYaxis().SetTitle("count")

htrk5.GetXaxis().SetTitle("[cm]")
htrk5.GetYaxis().SetTitle("count")

htrk6.GetXaxis().SetTitle("[cm]")
htrk6.GetYaxis().SetTitle("count")

htrkr1.GetXaxis().SetTitle("[cm]")
htrkr1.GetYaxis().SetTitle("count")

htrkr2.GetXaxis().SetTitle("[cm]")
htrkr2.GetYaxis().SetTitle("count")

htrkr3.GetXaxis().SetTitle("[cm]")
htrkr3.GetYaxis().SetTitle("count")

htrkr4.GetXaxis().SetTitle("[cm]")
htrkr4.GetYaxis().SetTitle("count")

htrkr5.GetXaxis().SetTitle("[cm]")
htrkr5.GetYaxis().SetTitle("count")

htrkr6.GetXaxis().SetTitle("[cm]")
htrkr6.GetYaxis().SetTitle("count")

hclpos1.GetXaxis().SetTitle("x-pos [cm]")
hclpos1.GetYaxis().SetTitle("y-pos [cm]")

hclpos2.GetXaxis().SetTitle("x-pos [cm]")
hclpos2.GetYaxis().SetTitle("y-pos [cm]")

color_cl=2
hclsr1.SetFillColor(color_cl)
hclsr2.SetFillColor(color_cl)
hclsr3.SetFillColor(color_cl)
hclsr4.SetFillColor(color_cl)
hclsr5.SetFillColor(color_cl)
hclsr6.SetFillColor(color_cl)
hclsr1.SetFillStyle(3003)
hclsr2.SetFillStyle(3003)
hclsr3.SetFillStyle(3003)
hclsr4.SetFillStyle(3003)
hclsr5.SetFillStyle(3003)
hclsr6.SetFillStyle(3003)

hcls1.SetFillColor(color_cl)
hcls2.SetFillColor(color_cl)
hcls3.SetFillColor(color_cl)
hcls4.SetFillColor(color_cl)
hcls5.SetFillColor(color_cl)
hcls6.SetFillColor(color_cl)
hcls1.SetFillStyle(3003)
hcls2.SetFillStyle(3003)
hcls3.SetFillStyle(3003)
hcls4.SetFillStyle(3003)
hcls5.SetFillStyle(3003)
hcls6.SetFillStyle(3003)

hclsl1.SetFillColor(color_cl)
hclsl2.SetFillColor(color_cl)
hclsl3.SetFillColor(color_cl)
hclsl4.SetFillColor(color_cl)
hclsl5.SetFillColor(color_cl)
hclsl6.SetFillColor(color_cl)
hclsl1.SetFillStyle(3003)
hclsl2.SetFillStyle(3003)
hclsl3.SetFillStyle(3003)
hclsl4.SetFillStyle(3003)
hclsl5.SetFillStyle(3003)
hclsl6.SetFillStyle(3003)

color=16
htrkl1.SetFillColor(color)
htrkl2.SetFillColor(color)
htrkl3.SetFillColor(color)
htrkl4.SetFillColor(color)
htrkl5.SetFillColor(color)
htrkl6.SetFillColor(color)

htrkr1.SetFillColor(color)
htrkr2.SetFillColor(color)
htrkr3.SetFillColor(color)
htrkr4.SetFillColor(color)
htrkr5.SetFillColor(color)
htrkr6.SetFillColor(color)

htrk1.SetFillColor(color)
htrk2.SetFillColor(color)
htrk3.SetFillColor(color)
htrk4.SetFillColor(color)
htrk5.SetFillColor(color)
htrk6.SetFillColor(color)

leg =  ROOT.TLegend(0.6,0.7,0.90,0.9)
leg.AddEntry(htrkr6,"Tracklength","f") 
leg.AddEntry(hclsr6,"Clustercount","f") 




c1.cd(1)
ROOT.gStyle.SetOptStat(0)
htrkl1.Draw()
hclsl1.Draw("SAME")
leg.Draw()
c1.cd(2)
htrkl2.Draw()
hclsl2.Draw("SAME")
leg.Draw()
c1.cd(3)
htrkl3.Draw()
hclsl3.Draw("SAME")
leg.Draw()
c1.cd(4)
htrkl4.Draw()
hclsl4.Draw("SAME")
leg.Draw()
c1.cd(5)
htrkl5.Draw()
hclsl5.Draw("SAME")
leg.Draw()
c1.cd(6)
htrkl6.Draw()
hclsl6.Draw("SAME")
leg.Draw()
c1.Update()

c2.cd(1)
ROOT.gStyle.SetOptStat(0)
htrkr1.Draw()
hclsr1.Draw("SAME")
leg.Draw()
c2.cd(2)
htrkr2.Draw()
hclsr2.Draw("SAME")
leg.Draw()
c2.cd(3)
htrkr3.Draw()
hclsr3.Draw("SAME")
leg.Draw()
c2.cd(4)
htrkr4.Draw()
hclsr4.Draw("SAME")
leg.Draw()
c2.cd(5)
htrkr5.Draw()
hclsr5.Draw("SAME")
leg.Draw()
c2.cd(6)
htrkr6.Draw()
hclsr6.Draw("SAME")
leg.Draw()
c2.Update()

c3.cd(1)
ROOT.gStyle.SetOptStat(0)
htrk1.Draw()
hcls1.Draw("SAME")
leg.Draw()
c3.cd(2)
htrk2.Draw()
hcls2.Draw("SAME")
leg.Draw()
c3.cd(3)
htrk3.Draw()
hcls3.Draw("SAME")
leg.Draw()
c3.cd(4)
htrk4.Draw()
hcls4.Draw("SAME")
leg.Draw()
c3.cd(5)
htrk5.Draw()
hcls5.Draw("SAME")
leg.Draw()
c3.cd(6)
htrk6.Draw()
hcls6.Draw("SAME")
leg.Draw()
c3.Update()

c4.cd(1)
hclpos1.Draw("colz")
c4.cd(2)
hclpos2.Draw("colz")
c4.Update()
  
#}


if args.pdir!="not":
    c1.SaveAs(args.pdir+"Trk_Cl left.eps")
    c2.SaveAs(args.pdir+"Trk_cl right.eps")
    c3.SaveAs(args.pdir+"Trk_cl both.eps")
    c4.SaveAs(args.pdir+"Cut - Clpos")
 


u = raw_input("Exit?")
if u == 'q':
    exit()