def main(): global beamEnergy # Parse all command line arguments using the argparse module. parser = argparse.ArgumentParser( description='PyRoot analysis demostrating the us of a DST.') parser.add_argument("dst_file", help="ROOT DST file to process") parser.add_argument("-o", "--output", help="Name of output pdf file") parser.add_argument("-m", "--mc", help="is MonteCarlo") parser.add_argument("-p", "--pulser", help="is Pulser") parser.add_argument("-e", "--energy", help="beam energy") args = parser.parse_args() # If an output file name was not specified, set a default name and warn # the user if args.output: output_file = args.output else: output_file = "analysis_output.root" print "[ HPS ANALYSIS ]: An output file name was not specified. Setting the name to " print output_file print "[ HPS ANALYSIS ]: Output file is " + output_file isMC = False if args.mc: print "[ HPS ANALYSIS ]: Setting to run as MC" isMC = True isPulser = False if args.pulser: print "[ HPS ANALYSIS ]: Setting to run from a pulser file" isPulser = True if args.energy: print 'Setting beam energy to ' + args.energy beamEnergy = float(args.energy) myhist.setEnergyScales(beamEnergy) ################################# # Event Selection ################################ #clean up event first #### nominal selection nTrkMax = 3 nTrkMin = 3 nPosMax = 1 ###### two tracks (e+/e-) exactly # nTrkMax=2 # nTrkMin=2 # nPosMax=1 ###### more than 1 electron # nTrkMax=10 # nTrkMin=3 # nPosMax=1 ################### maxSharedHits = 2 #v0 cuts # v0Chi2=10 v0Chi2 = 99999999.0 #ESum -- full region v0PzMax = 1.2 * beamEnergy v0PzMin = 0.1 * beamEnergy #ESum -- Radiative region # v0PzMax=1.2 # v0PzMin=0.80 v0PyMax = 0.2 #absolute value v0PxMax = 0.2 #absolute value v0VzMax = 25.0 # mm from target # v0VyMax=1.0# mm from target v0VyMax = 6666.0 # mm from target v0VxMax = 2.0 # mm from target # track quality cuts trkChi2 = 10 beamCut = 0.8 * beamEnergy minPCut = 0.05 trkPyMax = 0.2 trkPxMax = 0.2 # slopeCut=0.03 slopeCut = 0.0 trkDeltaT = 4 #ns cluDeltaT = 2 #ns cluTrkDeltaT = 4 #ns ############## # ESum slices; upper limits nSlicesESum = 5 esumMin = 0.55 esumMax = 1.2 sliceSizeESum = 0.1 #100MeV starting at esumMin ############## trackKiller = False tkThreshold = 0.5 #GeV, below this start killing tracks tkThreshEff = 1.0 tkLowPoint = 0.20 tkLowPointEff = 0.40 # tkSlope=2.6 # tkIntercept=-0.04 #calculate tkSlope and Intercept tkSlope = (tkThreshEff - tkLowPointEff) / (tkThreshold - tkLowPoint) tkIntercept = tkThreshEff - tkSlope * tkThreshold ############## requireECalMatch = True requireECalFiducial = True requireECalSuperFiducial = False useGBL = True # Open the ROOT file # root_file = ROOT.TFile(str(args.dst_file)) # Get the TTree "HPS_EVENT" containing the HpsEvent branch and all # other colletions # tree = root_file.Get("HPS_Event") #use a TChain print "[ HPS ANALYSIS ]: Reading in root chain from " + args.dst_file tree = ROOT.TChain("HPS_Event") tree.Add(str(args.dst_file) + "*") # Create an HpsEvent object in order to read the TClonesArray # collections hps_event = HpsEvent() b_hps_event = tree.SetBranchAddress("Event", ROOT.AddressOf(hps_event)) #--- Analysis ---# #----------------# #counters nEvents = 0 nPassBasicCuts = 0 nPassV0Cuts = 0 nPassTrkCuts = 0 nPassNCand = 0 nPassECalMatch = 0 nFakeTri = 0 nTwoCand = 0 seedCnt = 0 # Loop over all events in the file for entry in xrange(0, tree.GetEntries()): # Print the event number every 500 events if (entry + 1) % 10000 == 0: print "Event " + str(entry + 1) tree.GetEntry(entry) if not hps_event.isPair1Trigger() and not isMC and not isPulser: continue nEvents += 1 addFakeEle = False # Loop over all tracks in the event npositrons = 0 n_tracks = 0 for track_n in xrange(0, hps_event.getNumberOfTracks()): track = hps_event.getGblTrack(track_n) if track is None: continue # print track.getParticle().getType() if trkMatchAndFiducial( track.getParticle(), requireECalSuperFiducial) and trkMomentum( track, minPCut, beamCut ): # count only matched tracks in defined fiducial region n_tracks += 1 if track.getCharge() > 0: npositrons += 1 myhist.rawposMom.Fill(pMag(track.getMomentum())) else: myhist.raweleMom.Fill(pMag(track.getMomentum())) # findWABPair(track.getParticle(),hps_event) # print "nTracks = "+str(n_tracks)+"; nPositrons = "+str(npositrons) # if n_tracks/2.0>nTrkMax : continue #do this very dumb thing (divide by 2 to un-double count GBL tracks) # if n_tracks/2.0<2: continue myhist.nTrk.Fill(n_tracks) myhist.nPos.Fill(npositrons) myhist.nEle.Fill(n_tracks - npositrons) myhist.nClust.Fill(hps_event.getNumberOfEcalClusters()) if n_tracks > nTrkMax: continue if n_tracks < nTrkMin: continue if npositrons < 1 or npositrons > nPosMax: continue nPassBasicCuts += 1 # print "passed basic cuts" candidateList = [] bestCandidate = -99 nCandidate = 0 # loop over all v0 candidates... for uc_index in xrange( 0, hps_event.getNumberOfParticles(HpsParticle.UC_V0_CANDIDATE)): particle = hps_event.getParticle(HpsParticle.UC_V0_CANDIDATE, uc_index) # print "Particle Type = "+ str(particle.getType()) if useGBL and particle.getType() < 32: continue if not useGBL and particle.getType() > 31: continue # print "found one..." vchi2 = particle.getVertexFitChi2() vposition = particle.getVertexPosition() vmomentum = particle.getMomentum() if vchi2 > v0Chi2: continue # use the measured sum of momentum # if vmomentum[2]>v0PzMax : continue # if vmomentum[2]<v0PzMin : continue #recon'ed vertex position cuts if abs(vposition[0]) > v0VxMax: continue if abs(vposition[1]) > v0VyMax: continue # if abs(vposition[2])>v0VzMax :continue # Only look at particles that have two daugther particles... daughter_particles = particle.getParticles() if daughter_particles.GetSize() != 2: continue # Only look at particles that are composed of e+e- pairs if daughter_particles.At(0).getCharge() * daughter_particles.At( 1).getCharge() > 0: continue # print "Passed daughter number cuts" electron = daughter_particles.At(0) positron = daughter_particles.At(1) if daughter_particles.At(0).getCharge() > 0: electron = daughter_particles.At(1) positron = daughter_particles.At(0) pEle = electron.getMomentum() pPos = positron.getMomentum() v0Sum = pMag(pSum(pEle, pPos)) #total momentum sum cuts if v0Sum > v0PzMax: continue if v0Sum < v0PzMin: continue nPassV0Cuts += 1 # print "Passed v0 cuts" ############# tracking cuts #momentum cuts...get rid of very soft or very hard tracks if pMag(pEle) > beamCut or pMag(pPos) > beamCut: continue if pMag(pEle) < minPCut or pMag(pPos) < minPCut: continue #top+bottom requirement if pEle[1] * pPos[1] > 0: continue # print 'looking at tracks now' # print len(electron.getTracks()) if len(electron.getTracks()) == 0 or len( positron.getTracks()) == 0: continue eleTrk = electron.getTracks().At(0) posTrk = positron.getTracks().At(0) if eleTrk is None or posTrk is None: continue # eleTrk.Print("v") #track timing if eleTrk.getTrackTime() - posTrk.getTrackTime() > trkDeltaT: continue #track slope (if any cut) if abs(eleTrk.getTanLambda()) < slopeCut or abs( posTrk.getTanLambda()) < slopeCut: continue # print 'satisfied timing cuts...' ############## nPassTrkCuts += 1 ############## # ECAL matching and timing cuts...also fiducial region cuts... if requireECalMatch: if positron.getClusters().GetEntries() == 0: continue if electron.getClusters().GetEntries() == 0: continue posCluster = positron.getClusters().First() eleCluster = electron.getClusters().First() if eleCluster.getClusterTime() - posCluster.getClusterTime( ) > cluDeltaT: continue if eleTrk.getTrackTime() - eleCluster.getClusterTime( ) + 43.5 > cluTrkDeltaT: continue if posTrk.getTrackTime() - posCluster.getClusterTime( ) + 43.5 > cluTrkDeltaT: continue if requireECalFiducial: #ANTI-fiducial cut # if myhist.inFiducialRegion(posCluster.getPosition()[0],posCluster.getPosition()[1]) : # continue # if myhist.inFiducialRegion(eleCluster.getPosition()[0],eleCluster.getPosition()[1]) : # continue #Fiducial cut if not myhist.inFiducialRegion( posCluster.getPosition()[0], posCluster.getPosition()[1]): continue if not myhist.inFiducialRegion( eleCluster.getPosition()[0], eleCluster.getPosition()[1]): continue if requireECalSuperFiducial: if not myhist.inSuperFiducialRegion( posCluster.getPosition()[0], posCluster.getPosition()[1]): continue if not myhist.inSuperFiducialRegion( eleCluster.getPosition()[0], eleCluster.getPosition()[1]): continue nPassECalMatch += 1 ############## #Passed the cuts..append the candidate index candidateList.append(particle) numCands = len(candidateList) myhist.nCand.Fill(numCands) if numCands != 2: continue #require 2 candidates...probably have to rethink this. print 'found 2 candidates' cand0 = candidateList[0] # take this as base candidate candpos = cand0.getParticles().At(0) candele = cand0.getParticles().At(1) if candpos.getCharge() < 0: candpos = cand0.getParticles().At(1) candele = cand0.getParticles().At(0) # print "swapping cand" cand1 = candidateList[1] # get the recoil electron from this recoil = cand1.getParticles().At(0) recpos = cand1.getParticles().At(1) if recoil.getCharge() > 0: # print "swapping recoil" recoil = cand1.getParticles().At(1) recpos = cand1.getParticles().At(0) # print 'Positron p' +str(pMag(candpos.getMomentum())) # print 'Electron p' +str(pMag(candele.getMomentum())) # print 'Recoil p' +str(pMag(recoil.getMomentum())) # print 'Recoil Positron p' +str(pMag(recpos.getMomentum())) # if candpos == recpos : # print "positrons are same" numSharedHits = myhist.getSharedHits(recoil.getTracks()[0], candele.getTracks()[0]) print "Number of overlapping hits for electrons is : " + str( numSharedHits) if numSharedHits <= maxSharedHits: nTwoCand += 1 myhist.fillCandidateHistograms(cand0, recoil) myhist.fillThreeTrackPlots(cand0, cand1) # if(nPassTrkCuts>0): myhist.saveHistograms(output_file) print "******************************************************************************************" print "Number of Events:\t\t", nEvents, "\t\t\t", float( nEvents) / nEvents, "\t\t\t", float(nEvents) / nEvents print "N(particle) Cuts:\t\t", nPassBasicCuts, "\t\t\t", float( nPassBasicCuts) / nEvents, "\t\t\t", float(nPassBasicCuts) / nEvents print "V0 Vertex Cuts:\t\t", nPassV0Cuts, "\t\t\t", float( nPassV0Cuts) / nPassBasicCuts, "\t\t\t", float(nPassV0Cuts) / nEvents print "Tracking Cuts:\t\t", nPassTrkCuts, "\t\t\t", float( nPassTrkCuts) / nPassV0Cuts, "\t\t\t", float(nPassTrkCuts) / nEvents print "ECal Match Cuts:\t\t", nPassECalMatch, "\t\t\t", float( nPassECalMatch) / nPassTrkCuts, "\t\t\t", float( nPassECalMatch) / nEvents print "Number of Fake Events Added: \t\t", nFakeTri, "\t\t\t", float( nFakeTri) / nPassECalMatch print "Number of events with 2 candidates (unshared) = ", nTwoCand
def main(): global beamEnergy # Parse all command line arguments using the argparse module. parser = argparse.ArgumentParser(description='PyRoot analysis demostrating the us of a DST.') parser.add_argument("dst_file", help="ROOT DST file to process") parser.add_argument("-o", "--output", help="Name of output pdf file") parser.add_argument("-m", "--mc", help="is MonteCarlo") parser.add_argument("-p", "--pulser", help="is Pulser") parser.add_argument("-e","--energy",help="beam energy") args = parser.parse_args() # If an output file name was not specified, set a default name and warn # the user if args.output: output_file = args.output else: output_file = "analysis_output.root" print "[ HPS ANALYSIS ]: An output file name was not specified. Setting the name to " print output_file print "[ HPS ANALYSIS ]: Output file is "+output_file isMC=False if args.mc: print "[ HPS ANALYSIS ]: Setting to run as MC" isMC=True isPulser=False if args.pulser: print "[ HPS ANALYSIS ]: Setting to run from a pulser file" isPulser=True if args.energy : print 'Setting beam energy to '+args.energy beamEnergy=float(args.energy) myhist.setEnergyScales(beamEnergy) ################################# # Event Selection ################################ #clean up event first #### nominal selection nTrkMax=3 nTrkMin=3 nPosMax=1 ###### two tracks (e+/e-) exactly # nTrkMax=2 # nTrkMin=2 # nPosMax=1 ###### more than 1 electron # nTrkMax=10 # nTrkMin=3 # nPosMax=1 ################### maxSharedHits=2 #v0 cuts # v0Chi2=10 v0Chi2=99999999.0 #ESum -- full region v0PzMax=1.2*beamEnergy v0PzMin=0.1*beamEnergy #ESum -- Radiative region # v0PzMax=1.2 # v0PzMin=0.80 v0PyMax=0.2 #absolute value v0PxMax=0.2 #absolute value v0VzMax=25.0# mm from target # v0VyMax=1.0# mm from target v0VyMax=6666.0# mm from target v0VxMax=2.0# mm from target # track quality cuts trkChi2=10 beamCut=0.8*beamEnergy minPCut=0.05 trkPyMax=0.2 trkPxMax=0.2 # slopeCut=0.03 slopeCut=0.0 trkDeltaT=4#ns cluDeltaT=2#ns cluTrkDeltaT=4#ns ############## # ESum slices; upper limits nSlicesESum=5 esumMin=0.55 esumMax=1.2 sliceSizeESum=0.1 #100MeV starting at esumMin ############## trackKiller=False tkThreshold=0.5 #GeV, below this start killing tracks tkThreshEff=1.0 tkLowPoint=0.20 tkLowPointEff=0.40 # tkSlope=2.6 # tkIntercept=-0.04 #calculate tkSlope and Intercept tkSlope=(tkThreshEff-tkLowPointEff)/(tkThreshold-tkLowPoint) tkIntercept=tkThreshEff-tkSlope*tkThreshold ############## requireECalMatch = True requireECalFiducial = True requireECalSuperFiducial = False useGBL=True # Open the ROOT file # root_file = ROOT.TFile(str(args.dst_file)) # Get the TTree "HPS_EVENT" containing the HpsEvent branch and all # other colletions # tree = root_file.Get("HPS_Event") #use a TChain print "[ HPS ANALYSIS ]: Reading in root chain from "+args.dst_file tree=ROOT.TChain("HPS_Event") tree.Add(str(args.dst_file)+"*") # Create an HpsEvent object in order to read the TClonesArray # collections hps_event = HpsEvent() b_hps_event = tree.SetBranchAddress("Event", ROOT.AddressOf(hps_event)) #--- Analysis ---# #----------------# #counters nEvents=0; nPassBasicCuts=0; nPassV0Cuts=0; nPassTrkCuts=0; nPassNCand=0 nPassECalMatch=0; nFakeTri=0 nTwoCand=0 seedCnt=0 # Loop over all events in the file for entry in xrange(0, tree.GetEntries()) : # Print the event number every 500 events if (entry+1)%10000 == 0 : print "Event " + str(entry+1) tree.GetEntry(entry) if not hps_event.isPair1Trigger() and not isMC and not isPulser: continue nEvents+=1 addFakeEle=False # Loop over all tracks in the event npositrons=0 n_tracks=0 for track_n in xrange(0, hps_event.getNumberOfTracks()) : track = hps_event.getGblTrack(track_n) if track is None : continue # print track.getParticle().getType() if trkMatchAndFiducial(track.getParticle(),requireECalSuperFiducial) and trkMomentum(track,minPCut,beamCut): # count only matched tracks in defined fiducial region n_tracks+=1 if track.getCharge()>0 : npositrons+=1 myhist.rawposMom.Fill(pMag(track.getMomentum())) else : myhist.raweleMom.Fill(pMag(track.getMomentum())) # findWABPair(track.getParticle(),hps_event) # print "nTracks = "+str(n_tracks)+"; nPositrons = "+str(npositrons) # if n_tracks/2.0>nTrkMax : continue #do this very dumb thing (divide by 2 to un-double count GBL tracks) # if n_tracks/2.0<2: continue myhist.nTrk.Fill(n_tracks); myhist.nPos.Fill(npositrons); myhist.nEle.Fill(n_tracks-npositrons); myhist.nClust.Fill(hps_event.getNumberOfEcalClusters()) if n_tracks>nTrkMax : continue if n_tracks<nTrkMin: continue if npositrons<1 or npositrons>nPosMax : continue nPassBasicCuts+=1 # print "passed basic cuts" candidateList=[] bestCandidate=-99 nCandidate=0 # loop over all v0 candidates... for uc_index in xrange(0, hps_event.getNumberOfParticles(HpsParticle.UC_V0_CANDIDATE)): particle = hps_event.getParticle(HpsParticle.UC_V0_CANDIDATE, uc_index) # print "Particle Type = "+ str(particle.getType()) if useGBL and particle.getType()<32 : continue if not useGBL and particle.getType()>31 : continue # print "found one..." vchi2=particle.getVertexFitChi2(); vposition=particle.getVertexPosition(); vmomentum=particle.getMomentum(); if vchi2>v0Chi2 : continue # use the measured sum of momentum # if vmomentum[2]>v0PzMax : continue # if vmomentum[2]<v0PzMin : continue #recon'ed vertex position cuts if abs(vposition[0])>v0VxMax : continue if abs(vposition[1])>v0VyMax :continue # if abs(vposition[2])>v0VzMax :continue # Only look at particles that have two daugther particles... daughter_particles = particle.getParticles() if daughter_particles.GetSize() != 2 : continue # Only look at particles that are composed of e+e- pairs if daughter_particles.At(0).getCharge()*daughter_particles.At(1).getCharge() > 0 : continue # print "Passed daughter number cuts" electron = daughter_particles.At(0) positron = daughter_particles.At(1) if daughter_particles.At(0).getCharge()>0: electron = daughter_particles.At(1) positron = daughter_particles.At(0) pEle=electron.getMomentum() pPos=positron.getMomentum() v0Sum=pMag(pSum(pEle,pPos)) #total momentum sum cuts if v0Sum>v0PzMax : continue if v0Sum<v0PzMin : continue nPassV0Cuts+=1 # print "Passed v0 cuts" ############# tracking cuts #momentum cuts...get rid of very soft or very hard tracks if pMag(pEle)>beamCut or pMag(pPos)>beamCut : continue if pMag(pEle)<minPCut or pMag(pPos)<minPCut : continue #top+bottom requirement if pEle[1]*pPos[1]>0 : continue # print 'looking at tracks now' # print len(electron.getTracks()) if len(electron.getTracks()) == 0 or len(positron.getTracks()) == 0: continue eleTrk=electron.getTracks().At(0) posTrk=positron.getTracks().At(0) if eleTrk is None or posTrk is None : continue # eleTrk.Print("v") #track timing if eleTrk.getTrackTime() - posTrk.getTrackTime()> trkDeltaT : continue #track slope (if any cut) if abs(eleTrk.getTanLambda())<slopeCut or abs(posTrk.getTanLambda())<slopeCut : continue # print 'satisfied timing cuts...' ############## nPassTrkCuts+=1 ############## # ECAL matching and timing cuts...also fiducial region cuts... if requireECalMatch: if positron.getClusters().GetEntries() == 0 : continue if electron.getClusters().GetEntries() == 0 : continue posCluster=positron.getClusters().First() eleCluster=electron.getClusters().First() if eleCluster.getClusterTime()- posCluster.getClusterTime() > cluDeltaT: continue if eleTrk.getTrackTime() - eleCluster.getClusterTime()+43.5 > cluTrkDeltaT : continue if posTrk.getTrackTime() - posCluster.getClusterTime()+43.5 > cluTrkDeltaT : continue if requireECalFiducial: #ANTI-fiducial cut # if myhist.inFiducialRegion(posCluster.getPosition()[0],posCluster.getPosition()[1]) : # continue # if myhist.inFiducialRegion(eleCluster.getPosition()[0],eleCluster.getPosition()[1]) : # continue #Fiducial cut if not myhist.inFiducialRegion(posCluster.getPosition()[0],posCluster.getPosition()[1]) : continue if not myhist.inFiducialRegion(eleCluster.getPosition()[0],eleCluster.getPosition()[1]) : continue if requireECalSuperFiducial : if not myhist.inSuperFiducialRegion(posCluster.getPosition()[0],posCluster.getPosition()[1]) : continue if not myhist.inSuperFiducialRegion(eleCluster.getPosition()[0],eleCluster.getPosition()[1]) : continue nPassECalMatch+=1 ############## #Passed the cuts..append the candidate index candidateList.append(particle) numCands=len(candidateList) myhist.nCand.Fill(numCands) if numCands != 2 : continue #require 2 candidates...probably have to rethink this. print 'found 2 candidates' cand0=candidateList[0] # take this as base candidate candpos=cand0.getParticles().At(0) candele=cand0.getParticles().At(1) if candpos.getCharge()<0 : candpos=cand0.getParticles().At(1) candele=cand0.getParticles().At(0) # print "swapping cand" cand1=candidateList[1] # get the recoil electron from this recoil= cand1.getParticles().At(0) recpos= cand1.getParticles().At(1) if recoil.getCharge()> 0 : # print "swapping recoil" recoil= cand1.getParticles().At(1) recpos= cand1.getParticles().At(0) # print 'Positron p' +str(pMag(candpos.getMomentum())) # print 'Electron p' +str(pMag(candele.getMomentum())) # print 'Recoil p' +str(pMag(recoil.getMomentum())) # print 'Recoil Positron p' +str(pMag(recpos.getMomentum())) # if candpos == recpos : # print "positrons are same" numSharedHits=myhist.getSharedHits(recoil.getTracks()[0],candele.getTracks()[0]) print "Number of overlapping hits for electrons is : "+str(numSharedHits) if numSharedHits <= maxSharedHits : nTwoCand+=1 myhist.fillCandidateHistograms(cand0,recoil) myhist.fillThreeTrackPlots(cand0,cand1) # if(nPassTrkCuts>0): myhist.saveHistograms(output_file) print "******************************************************************************************" print "Number of Events:\t\t",nEvents,"\t\t\t",float(nEvents)/nEvents,"\t\t\t",float(nEvents)/nEvents print "N(particle) Cuts:\t\t",nPassBasicCuts,"\t\t\t",float(nPassBasicCuts)/nEvents,"\t\t\t",float(nPassBasicCuts)/nEvents print "V0 Vertex Cuts:\t\t",nPassV0Cuts,"\t\t\t",float(nPassV0Cuts)/nPassBasicCuts,"\t\t\t",float(nPassV0Cuts)/nEvents print "Tracking Cuts:\t\t",nPassTrkCuts,"\t\t\t",float(nPassTrkCuts)/nPassV0Cuts,"\t\t\t",float(nPassTrkCuts)/nEvents print "ECal Match Cuts:\t\t",nPassECalMatch,"\t\t\t",float(nPassECalMatch)/nPassTrkCuts,"\t\t\t",float(nPassECalMatch)/nEvents print "Number of Fake Events Added: \t\t",nFakeTri,"\t\t\t",float(nFakeTri)/nPassECalMatch print "Number of events with 2 candidates (unshared) = ",nTwoCand
# Get an Ecal scoring plane hit from the event ecal_sp_hit = hps_event.getEcalScoringPlaneHit(ecal_sp_hit_n) # Get the associated particle particle = ecal_sp_hit.getParticle() # Get the track associated with the particle track = particle.getTracks().At(0) # Find the GBL track that corresponds to the Ecal scoring # plane hit for gbl_track_n in xrange(0, hps_event.getNumberOfGblTracks()): # Get a GBL track from the event gbl_track = hps_event.getGblTrack(gbl_track_n) # If the GBL seed track and the track associated with # the Ecal scoring plane hit do not match, continue # onto the next track if (gbl_track.getSeedD0() != track.getD0()): continue h_track_d0.Fill(gbl_track.getSeedD0()) h_gbl_track_d0.Fill(gbl_track.getD0()) gbl_track_parameters[0] = gbl_track.getD0() gbl_track_parameters[1] = gbl_track.getPhi() gbl_track_parameters[2] = gbl_track.getKappa() gbl_track_parameters[3] = gbl_track.getZ0() gbl_track_parameters[4] = math.tan(math.pi / 2 - gbl_track.getTheta())
# Get an Ecal scoring plane hit from the event ecal_sp_hit = hps_event.getEcalScoringPlaneHit(ecal_sp_hit_n) # Get the associated particle particle = ecal_sp_hit.getParticle() # Get the track associated with the particle track = particle.getTracks().At(0) # Find the GBL track that corresponds to the Ecal scoring # plane hit for gbl_track_n in xrange(0, hps_event.getNumberOfGblTracks()): # Get a GBL track from the event gbl_track = hps_event.getGblTrack(gbl_track_n) # If the GBL seed track and the track associated with # the Ecal scoring plane hit do not match, continue # onto the next track if(gbl_track.getSeedD0() != track.getD0()): continue h_track_d0.Fill(gbl_track.getSeedD0()) h_gbl_track_d0.Fill(gbl_track.getD0()) gbl_track_parameters[0] = gbl_track.getD0() gbl_track_parameters[1] = gbl_track.getPhi() gbl_track_parameters[2] = gbl_track.getKappa() gbl_track_parameters[3] = gbl_track.getZ0() gbl_track_parameters[4] = math.tan(math.pi/2 - gbl_track.getTheta())