import glob,sys,subprocess,time,os,ROOT,commands,random
import array
pandapath=os.environ.get('PANDAPATH')
print pandapath
sys.path.append(pandapath+'/macro/tpc/FOPI/python/argparse-1.2.1')
import argparse

parser=argparse.ArgumentParser(description='make tree for marian parametrization')
parser.add_argument('recofile',help='the reco file',type=str)
parser.add_argument('outfile',help='the output file',type=str)
parser.add_argument('--events',help='number of events',type=int,default=-1)
args=parser.parse_args()
ROOT.gROOT.ProcessLine(".x rootlogon.C") 
infile=ROOT.TFile(args.recofile,'read')
tree=infile.Get('cbmsim')

if tree==None:
    print 'no tree!'
    exit()
    
tuplevars=''
tuplevars+='X:Y:Z'
tuplevars+=':theta:phi'
tuplevars+=':ttheta:tphi'
tuplevars+=':A:N'
tuplevars+=':rmsU:rmsV'
tuplevars+=':rmsA1P:rmsA2P'
tuplevars+=':rmsA0:rmsA1:rmsA2'
tuplevars+=':rmsX:rmsY:rmsZ'
tuplevars+=':sigA0:sigA1:sigA2'
tuplevars+=':resX:resY:resZ'
tuplevars+=':resU:resV'
tuplevars+=':resA0:resA1:resA2'

outfile=ROOT.TFile(args.outfile,'recreate')
outtuple=ROOT.TNtuple("ClusterColl",'',tuplevars)

ev=-1
for e in tree:
    ev+=1
    if args.events!=-1 and args.events<ev:
        break
    
    if ev%1000==0:
        print ev
    
    for tr in e.CutCosmics:
        
        for icl in range(tr.nhits()):
                pos=tr.GetHitPosition(icl)
                theta=tr.GetMidPlaneW(icl).Theta()
                phi=tr.GetMidPlaneW(icl).Phi()
                ttheta=tr.GetTheta()
                tphi=tr.GetPhi()
                amp=tr.GetAmp(icl)
                ndigis=tr.GetClusterSize(icl)
                rmsU=tr.GetRMS_UV(icl).X()
                rmsV=tr.GetRMS_UV(icl).Y()
                rmsA1P=tr.GetRMS_A_UV(icl).X()
                rmsA2P=tr.GetRMS_A_UV(icl).Y()
                rmsA0=tr.GetRMS_A(icl).X()
                rmsA1=tr.GetRMS_A(icl).Y()
                rmsA2=tr.GetRMS_A(icl).Z()
                rmsXYZ=tr.GetRMS(icl)
                sig=tr.GetSigA(icl)
                resXYZ=tr.GetRes(icl)
                resU=tr.GetResU(icl)
                resV=tr.GetResV(icl)
                resA=tr.GetResA(icl)
                
                tuparray=[pos.X(),pos.Y(),pos.Z(),
                          theta,phi,
                          ttheta,tphi,
                          amp,ndigis,
                          rmsU,rmsV,
                          rmsA1P,rmsA2P,
                          rmsA0,rmsA1,rmsA2,
                          rmsXYZ.X(),rmsXYZ.Y(),rmsXYZ.Z(),
                          sig.X(),sig.Y(),sig.Z(),
                          resXYZ.X(),resXYZ.Y(),resXYZ.Z(),
                          resU,resV,
                          resA.X(),resA.Y(),resA.Z()]
                
                outtuple.Fill(array.array("f",tuparray))
                

outtuple.Write()
outfile.Close()