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

parser=argparse.ArgumentParser(description="plot Neff and other cluster error correlated stuff")
parser.add_argument('recofile',help='the reco file ',type=str)
parser.add_argument("--nEvts",help='number of events to analyze',default=-1,type=int)
parser.add_argument('--fC',help='the fC parameter',type=float,default=10)
parser.add_argument('--fG',help='the fG parameter',type=float,default=600)
parser.add_argument('--gain',help='the gain',type=float,default=3771)
parser.add_argument('--hlp',help='print help',action='store_true')
args=parser.parse_args()

ROOT.gROOT.ProcessLine(".x rootlogon.C") 
ROOT.gROOT.ProcessLine('gROOT->SetStyle("Plain")')
set_palette()

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


rfile=ROOT.TFile(args.recofile,'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("TpcCluster.*",1)
tree.SetBranchStatus("CutCosmics.*",1)

c1=ROOT.TCanvas("c1","c1",500,1000)
c1.Divide(1,3)
hNeff=ROOT.TH2D("hNeff","Cluster Neff Vs Z",75,-63,12,450,0,4500)
hNeffInv=ROOT.TH2D("hNeffInv","Cluster 1/Neff Vs Z",75,-63,12,500,0,0.01)
hNeffInvSqrt=ROOT.TH2D("hNeffInvSqrt","Cluster 1/Sqrt(Neff) Vs Z",75,-63,12,500,0,0.1)
evt=-1
for e in tree:
    evt+=1
    if args.nEvts!=-1 and args.nEvts<evt:
        break
    for tr in e.CutCosmics:
        posZ=tr.GetHitPositionsZ()
        amps=tr.GetAmps()
        
        for icl in range(len(posZ)):
            Neff=amps[icl]*(args.fG/args.gain)
            hNeff.Fill(posZ[icl],Neff)
            if Neff!=0:
                hNeffInv.Fill(posZ[icl],1/Neff)
                hNeffInvSqrt.Fill(posZ[icl],1/math.sqrt(Neff))

c1.cd(1)
hNeff.Draw("colz")
c1.cd(2)
hNeffInv.Draw("colz")
c1.cd(3)
hNeffInvSqrt.Draw("colz")
c1.Update()

u=' '
while u!='q':
    u=raw_input("done?")