import ROOT, argparse,sys
sys.path.append('$PANDAPATH/macro/tpc/FOPI/mberger')
from functions import sort_inner, sort_inner2, Drawh, Divideh, fillcutred, plotfullevent
from  cosmic_histos import *

parser=argparse.ArgumentParser(description='analyze cosmics from ana file')
parser.add_argument("--branch",help="which branch to use",default="TrackFitStat_0")
parser.add_argument("filename",help="the file to analyze")
parser.add_argument("--slicef",help='a root file with the slices histo',default="")
parser.add_argument("--nEvts",help='number of events to analyze',default=-1,type=int)
parser.add_argument("--outfname",help="name of the output file",default="./someAnaFile.root",type=str)
parser.add_argument("--drawSingle",help="draw a single histo or dict",default="",type=str)

parser.add_argument("--hlp",help="display help",action='store_true')
args=parser.parse_args()

if args.hlp:
    parser.print_help()
    exit(0)

ROOT.gROOT.ProcessLine(".x rootlogon.C") 
ROOT.gROOT.ProcessLine('gStyle->SetPalette(1)') 
ROOT.gROOT.ProcessLine('gROOT->SetStyle("Plain")')

histsdict=cosmic_histos()
histsopt=cosmic_histopt()
canvasopt=cosmic_canvasopt()
axlabel=cosmic_axlabel()

zSlices = []
if args.slicef!="":
    print "slicename", args.slicef
    infiles=ROOT.TFile(args.slicef)
    zSlices = []
    histsdict['hslices'] = infiles.Get("hslices")
    for k in range(1,7):
        print histsdict['hslices'].GetBinContent(k)
        zSlices.append(histsdict['hslices']).GetBinContent(k)
else:
    zSlices = (  -48.0, -36.0, -24.0, -12.0,  0.0, 20.0 ) # cosmic ne
    #zSlices =  (   20,  30,  40,  49,  57, 70 ) # beam
    for i in range(len(zSlices)):
        histsdict['hslices'].SetBinContent(i+1,zSlices[i])

hslices=cosmic_hslices(zSlices)

RFile=ROOT.TFile()
Rfile = ROOT.TFile.Open(args.filename,"read")
tree = Rfile.Get("cbmsim")
if tree == None :
    print "No cbmsim in file", file
    if type(Rfile)==ROOT.TFile:
        Rfile.Close()
        exit()

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{{
        phi=tr.GetPhi()

        resX=tr.GetResX()
        resY=tr.GetResY()
        posX=tr.GetHitPositionsX()
        posY=tr.GetHitPositionsY()
        posZ=tr.GetHitPositionsZ()
        size2D=tr.Get2DClSizes()
        size3D=tr.GetClSizes()
        length=tr.GetLength()
        cl_count=len(size3D)
        distance=tr.GetClusterDists()
        
    
        chi2=tr.getChi2()
        NDF=tr.getNDF() 
        meanz=tr.GetMeanZ()
    
        resMcX=tr.GetResMcX()  
        
        gesdigi=0
        for clsi in size3D:
            gesdigi+=clsi
#}}
#Fill Histogramms{{        
        for i in range(len(size3D)):
            histsdict['hdbc'].Fill(posZ[i],distance[i])
            histsdict['hcs2D'].Fill(posZ[i],size2D[i])
            histsdict['hcs3D'].Fill(posZ[i],size3D[i])
            diff = abs(size3D[i]-size2D[i])
            histsdict['hdsoc'].Fill(posZ[i],diff)
            if posZ[i]<-48.0:
                histsdict['hdpc1'].Fill(size3D[i])
            if -48<posZ[i]<-36.0:
                histsdict['hdpc2'].Fill(size3D[i])
            if -36.0<posZ[i]<-24.0:
                histsdict['hdpc3'].Fill(size3D[i]) 
            if -24.0<posZ[i]<-12.0:
                histsdict['hdpc4'].Fill(size3D[i])
            if -12.0<posZ[i]<0.0:
                histsdict['hdpc5'].Fill(size3D[i])
            if 0<posZ[i]:
                histsdict['hdpc6'].Fill(size3D[i])
        
            for i in range(len(resMcX)):
                if abs(phi)>70 and abs(phi)<110:
                    histsdict['hresZ'].Fill(posZ[i],resX[i])


        for icl in range(len(resX)):
            resX[icl]*=  10000
            resY[icl]*=  10000
            resMcX[icl]*=10000

            if abs(phi)>70 and abs(phi)<110:
                histsdict['hxy res'].Fill(posX[icl],posY[icl],resX[icl])
                histsdict['hxy'].Fill(posX[icl],posY[icl])
                histsdict['hyz res'].Fill(posZ[icl],posY[icl],resX[icl])
                histsdict['hyz'].Fill(posZ[icl],posY[icl])
                for j in range(len(zSlices)):
                    if posZ[icl]<zSlices[j]:
                        hslices['hxy slice'][j].Fill(posX[icl],posY[icl])
                        hslices['hxy res slice'][j].Fill(posX[icl],posY[icl],resX[icl])
                        hslices['residualsX'][j].Fill(resX[icl])
                        hslices['McresidualsX'][j].Fill(resMcX[icl])
                        break
            elif(abs(phi)<20 or abs(phi)>160):
                histsdict['hxy resY'].Fill(posX[icl],posY[icl],resY[icl])
                histsdict['hxy Y'].Fill(posX[icl],posY[icl])
                histsdict['hxz resY'].Fill(posZ[icl],posY[icl],resY[icl])
                histsdict['hxz'].Fill(posZ[icl],posY[icl])
                for j in range(len(zSlices)):
                    if posZ[icl]<zSlices[j]:
                        hslices['hxy Y slice'][j].Fill(posX[icl],posY[icl])
                        hslices['hxy resY slice'][j].Fill(posX[icl],posY[icl],resY[icl])
                        hslices['residualsY'][j].Fill(resY[icl])
                        break

            histsdict['hclusterSize2DVsZ'].Fill(posZ[icl],size2D[icl])
            histsdict['hclusterSize3DVsZ'].Fill(posZ[icl],size3D[icl])

            for j in range(len(zSlices)):
                if posZ[icl]<zSlices[j]:
                    hslices['clustersize'][j].Fill(size3D[icl])
                    hslices['clustersize2D'][j].Fill(size2D[icl])
#}}
#calculation{{       
        eventcounter+=1

        if NDF!=0:
            chi2_over_ndf = chi2/NDF
        else:
            chi2_over_ndf = -10

        if length!=0:
            cpc=cl_count/length
            dpc=gesdigi/length
        else:
            cpc=-1
            dpc=-1
        
       # print "NDF:", NDF
        #print "chi2:", chi2
        
#print "chi2/1000000:", chi2/1000000
 #       print "ndf/1000000:", NDF/1000000

#}}
#Fill Hisogramms{{

        histsdict['trackchi2']['chindf'].Fill(chi2_over_ndf)
        histsdict['trackchi2']['chindf over z'].Fill(meanz, chi2_over_ndf)
        histsdict['trackchi2']['chi over z'].Fill(meanz, chi2/1000)
        histsdict['trackchi2']['ndf over z'].Fill(meanz, NDF)
        histsdict['hcpc'].Fill(cpc)
        histsdict['hcpt'].Fill(cl_count)
        histsdict['hcvsac'].Fill(cl_count, cpc)
        
       
        if meanz<-48.0:
            histsdict['trackchi2']['chindf - 1st sector'].Fill(chi2_over_ndf)
            histsdict['hcpc1'].Fill(cpc)
            histsdict['hdpcm1'].Fill(dpc)
            histsdict['hlvsc1']. Fill(cl_count, length)
            histsdict['hcvsac1'].Fill(cl_count, cpc)
        if -48<meanz<-36.0:
            histsdict['trackchi2']['chindf - 2st sector'].Fill(chi2_over_ndf)
            histsdict['hcpc2'].Fill(cpc)
            histsdict['hdpcm2'].Fill(dpc)
            histsdict['hlvsc2']. Fill(cl_count, length)
            histsdict['hcvsac2'].Fill(cl_count, cpc)
        if -36.0<meanz<-24.0:
            histsdict['trackchi2']['chindf - 3st sector'].Fill(chi2_over_ndf)
            histsdict['hcpc3'].Fill(cpc)
            histsdict['hdpcm3'].Fill(dpc)
            histsdict['hlvsc3']. Fill(cl_count, length)
            histsdict['hcvsac3'].Fill(cl_count, cpc)
        if -24.0<meanz<-12.0:
            histsdict['trackchi2']['chindf - 4st sector'].Fill(chi2_over_ndf)
            histsdict['hcpc4'].Fill(cpc)
            histsdict['hdpcm4'].Fill(dpc)
            histsdict['hlvsc4']. Fill(cl_count, length)
            histsdict['hcvsac4'].Fill(cl_count, cpc)
        if -12.0<meanz<0.0:
            histsdict['trackchi2']['chindf - 5st sector'].Fill(chi2_over_ndf)
            histsdict['hcpc5'].Fill(cpc)
            histsdict['hdpcm5'].Fill(dpc)
            histsdict['hlvsc5']. Fill(cl_count, length)
            histsdict['hcvsac5'].Fill(cl_count, cpc)
        if 0<meanz:
            histsdict['trackchi2']['chindf - 6st sector'].Fill(chi2_over_ndf)
            histsdict['hcpc6'].Fill(cpc)
            histsdict['hdpcm6'].Fill(dpc)
            histsdict['hlvsc6']. Fill(cl_count, length)
            histsdict['hcvsac6'].Fill(cl_count, cpc)

        histsdict['hldt'].Fill(length)
        histsdict['hnof'].Fill(eventcounter)
        histsdict['hdpcz'].Fill(meanz, dpc)
        histsdict['hlvsc']. Fill(cl_count, length)

   
        
        
    #}}
            
    
    done=int(float(evt)/float(todoev)*100)
    if done!=olddone:
        print done,"% done, Good event:",evt#,"(",totgoodev,")"
    olddone=done

histsdict['hxy res norm'] =Divideh(histsdict['hxy res'] ,histsdict['hxy']  ,"hxyresnorm" ,"xy X-residual normalized")
histsdict['hyz res norm'] =Divideh(histsdict['hyz res'] ,histsdict['hyz']  ,"hyzresnorm" ,"yz X-residual normalized")
histsdict['hxy resY norm']=Divideh(histsdict['hxy resY'],histsdict['hxy Y'],"hxyresYnorm","xy Y-residual normalized")

hslices['hxy res norm slice']={}
hslices['hxy resY norm slice']={}
for i in range(len(zSlices)):
    hslices['hxy res norm slice'][i]=Divideh(hslices['hxy res slice'][i],hslices['hxy slice'][i],"hxyresnorm"+str(zSlices[i]),"xy X-residual normalized "+str(zSlices[i]))
    hslices['hxy resY norm slice'][i]=Divideh(hslices['hxy resY slice'][i],hslices['hxy Y slice'][i],"hxyresYnorm"+str(zSlices[i]),"xy Y-residual normalized")

outfile = ROOT.TFile(args.outfname,"recreate")
outfile.cd()

histcounter = 0
for hists in histsdict:
    histcounter+=1
    hist=histsdict[hists]
    if type(hist)==dict:
        for h in hist:
            if hist[h].GetEntries()>0:
                hist[h].Write()
    else:
        if hist.GetEntries()>0:
            hist.Write()

for d in hslices:
    sdict=hslices[d]
    for i in range(len(zSlices)):
        h=sdict[i]
        if h.GetEntries()>0:
            h.Write()

print 'Histos are written into file:', args.outfname
print "Anzahl der Histogramme:", " ", histcounter