class JetIDAnalyzer(Analyzer): '''A simple jet analyzer.''' def declareHandles(self): super(JetIDAnalyzer, self).declareHandles() self.handles['jets'] = AutoHandle(*self.cfg_ana.jetCollection) if self.cfg_ana.useGenLeptons: self.mchandles['leptons2'] = AutoHandle( 'genLeptonsStatus2', 'std::vector<reco::GenParticle>') else: self.mchandles['genParticles'] = AutoHandle( 'prunedGen', 'std::vector<reco::GenParticle>') ### self.mchandles['genJets'] = AutoHandle ( ### *self.cfg_ana.genJetsCollection ### ) self.handles['vertices'] = AutoHandle('offlinePrimaryVertices', 'std::vector<reco::Vertex>') # .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... def beginLoop(self): super(JetIDAnalyzer, self).beginLoop() self.file = TFile('/'.join([self.looperName, 'testJets.root']), 'recreate') if self.cfg_ana.applyPFLooseId: from ROOT import PFJetIDSelectionFunctor self.isPFLooseFunc = PFJetIDSelectionFunctor( 0, PFJetIDSelectionFunctor.LOOSE) ## Workaround: for some reason PyROOT does not bind PFJetIDSelectionFunctor(Jet) nor PFJetIDSelectionFunctor.getBitsTemplates from ROOT import pat self.isPFLooseFunc.bits = pat.strbitset() for i in "CHF", "NHF", "CEF", "NEF", "NCH", "nConstituents": self.isPFLooseFunc.bits.push_back(i) ## /Workaround self.isPFLoose = lambda x: self.isPFLooseFunc( x, self.isPFLooseFunc.bits) else: self.isPFLoose = lambda x: True # general histograms self.jetHistos = JetHistograms('Jets') self.cleanJetHistos = JetHistograms('CleanJets') self.matchedCleanJetHistos = JetHistograms('MatchedCleanJets') self.unmatchedCleanJetHistos = JetHistograms('UnmatchedCleanJets') ## Pileup JetID if self.cfg_ana.jetIdMva: self.puidalgo = PileupJetIdNtupleAlgo(*self.cfg_ana.jetIdMva) self.runMva = True else: self.puidalgo = PileupJetIdNtupleAlgo() self.runMva = False self.vtxBins = (0, 10, 15, 20, 30) ## (0,2,4,6,10,15,20,30,35) self.vtxBinLabels = mkBinLabels(self.vtxBins) self.ptBins = (20, 30, 50) ## (20,30,40,50,100) self.etaBins = (0, 1.4, 2.5, 3.0) self.puEtaLables = ["_barrel", "_endtk", "_endNOtk", "_fwd"] reweight_f = TF1("f", "pol2(0)+expo(3)") reweight_f.SetParameters(0.1955298, -0.003830591, 1.944794e-05, 4.649755, -0.1722024) self.reweight = ("pt", reweight_f) ## PileUp histograms if self.cfg_ana.doJetIdHisto: # histograms for pileup jet identification variables self.gluCleanHistosId = PileupJetHistograms( "GluonMatchedCleanHistosId", self.vtxBins, self.ptBins, self.etaBins, etalabels=self.puEtaLables, reweight=self.reweight) self.quarkCleanHistosId = PileupJetHistograms( "QuarkMatchedCleanHistosId", self.vtxBins, self.ptBins, self.etaBins, etalabels=self.puEtaLables, reweight=self.reweight) self.reweiMatchedCleanHistosId = PileupJetHistograms( "ReweiMatchedCleanHistosId", self.vtxBins, self.ptBins, self.etaBins, etalabels=self.puEtaLables, reweight=self.reweight) self.unmatchedCleanHistosId = PileupJetHistograms( "UnmatchedCleanHistosId", self.vtxBins, self.ptBins, self.etaBins, etalabels=self.puEtaLables) ### self.jetKin = [] ### for vtx in self.vtxBinLabels: ### self.jetKin.append([]) ### for eta in self.puEtaLables: ### self.jetKin[-1].append( JetHistograms('Jets%s%s' % (vtx,eta) ) ) if self.cfg_ana.dumpTree: self.tree = TTree("tree", "tree") self.tree.SetDirectory(self.file) self.puidalgo.bookBranches(self.tree) gROOT.ProcessLine( """struct MyTreeVars{ Int_t nvtx; Bool_t isMatched; Int_t jetFlavour; Float_t partonPt; Float_t genPt; Float_t genDr; }""" ) from ROOT import MyTreeVars self.treevars = MyTreeVars() for i, j in ("nvtx", "I"), ("isMatched", "O"), ("jetFlavour", "I"), ("partonPt", "F"), ("genPt", "F"), ("genDr", "F"): self.tree.Branch(i, AddressOf(self.treevars, i), "%s/%s" % (i, j)) self.h_nvtx = TH1F("h_nvtx", "", 50, 0, 50) self.h_genjetspt = TH1F("h_genjetspt", "", 500, 0, 500) # .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... def process(self, iEvent, event): #read all the handles defined beforehand self.readCollections(iEvent) jetEtaCut = 4.5 # get the vertexes event.vertices = self.handles['vertices'].product() self.h_nvtx.Fill(len(event.vertices)) event.vertexBin = int(len(event.vertices)) # get the jets in the jets variable jets = self.handles['jets'].product() # filter jets with some selections event.jets = [ jet for jet in jets if (abs(jet.eta()) < jetEtaCut and jet.pt() > self.cfg_ana.ptCut and self.isPFLoose(jet)) ] self.jetHistos.fillEvent(event.jets) try: # get status 2 leptons if 'leptons2' in self.mchandles: event.genLeptons = map(GenParticle, self.mchandles['leptons2'].product()) else: event.genLeptons = [ lep for lep in self.mchandles['genParticles'].product() if lep.status() == 3 and (abs(lep.pdgId()) == 11 or abs( lep.pdgId()) == 13 or abs(lep.pdgId()) == 15) ] except: event.genLeptons = None if event.genLeptons: #FIXME why are there cases in which there's 4 or 6 leptons? if len(event.genLeptons) != 2: return # in case I want to filter out taus # 11, 13, 15 : e, u, T # remove leptons from jets if closer than 0.2 event.cleanJets = cleanObjectCollection(event.jets, event.genLeptons, 0.2) else: event.cleanJets = event.jets ### event.matchingCleanJets = matchObjectCollection2 (event.cleanJets, event.selGenJets, 0.25) # assign to each jet its gen match (easy life :)) event.selGenJets = [] for jet in event.cleanJets: try: jet.gen = jet.genJet() jet.gen.pt() event.selGenJets.append(jet.gen) except: jet.gen = None for jet in event.selGenJets: self.h_genjetspt.Fill(jet.pt()) self.jetHistos.fillStats(len(event.selGenJets), len(event.jets)) ### try: ### # get genJets ### event.genJets = map (GenJet, self.mchandles['genJets'].product ()) ### event.selGenJets = event.genJets ### for jet in event.selGenJets : ### self.h_genjetspt.Fill (jet.pt ()) ### ### # first stats plots ### self.jetHistos.fillStats (len (event.selGenJets), len (event.jets)) ### except: ### pass ## run the PileupJetIdAlgo and attach the id to the jet ijet = 0 for jet in event.cleanJets: try: jet.puIdentifier = self.puidalgo.computeIdVariables( jet.sourcePtr().get(), 0., event.vertices[0], self.runMva) except: jet.puIdentifier = self.puidalgo.computeIdVariables( jet, 0., event.vertices[0], self.runMva) if self.cfg_ana.dumpTree: ## self.puidalgo.fillJet(jet.puIdentifier,ijet,0) self.puidalgo.setIJetIEvent(ijet, 0) self.treevars.nvtx = len(event.vertices) self.treevars.isMatched = False self.treevars.jetFlavour = -999 self.treevars.genPt = 0. self.treevars.partonPt = 0. self.treevars.genDr = -100. if jet.gen: self.treevars.isMatched = True self.treevars.jetFlavour = jet.partonFlavour() self.treevars.genPt = jet.gen.pt() self.treevars.genDr = deltaR(jet.gen.eta(), jet.gen.phi(), jet.eta(), jet.phi()) try: self.treevars.partonPt = jet.genParton.pt() except: pass self.tree.Fill() ijet += 1 self.cleanJetHistos.fillEvent(event.cleanJets) event.matchedCleanJets = [ jet for jet in event.cleanJets if jet.gen != None ] event.unmatchedCleanJets = [ jet for jet in event.cleanJets if jet.gen == None ] self.matchedCleanJetHistos.fillEvent(event.matchedCleanJets) self.unmatchedCleanJetHistos.fillEvent(event.unmatchedCleanJets) event.cleanGluJets = [] event.cleanQuarkJets = [] vtxbin = findBin(self.vtxBinLabels, len(event.vertices)) for jet in event.matchedCleanJets: flav = abs(jet.partonFlavour()) if flav == 21: event.cleanGluJets.append(jet) elif flav > 0 and flav <= 3: event.cleanQuarkJets.append(jet) jet.weight = self.reweight[1](getattr(jet, self.reweight[0])()) etabin = findBin(self.etaBins, jet.eta()) ### self.jetKin[vtxbin][etabin].fillJet(jet) if self.cfg_ana.doJetIdHisto: self.gluCleanHistosId.fillEvent(event.cleanGluJets, event.vertices) self.quarkCleanHistosId.fillEvent(event.cleanQuarkJets, event.vertices) ### self.matchedCleanHistosId.fillEvent(event.matchedCleanJets,event.vertices) self.reweiMatchedCleanHistosId.fillEvent(event.matchedCleanJets, event.vertices) self.unmatchedCleanHistosId.fillEvent(event.unmatchedCleanJets, event.vertices) # .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... def write(self): from ROOT import gROOT gROOT.SetBatch(True) self.jetHistos.Write(self.file) self.cleanJetHistos.Write(self.file) self.matchedCleanJetHistos.Write(self.file) self.unmatchedCleanJetHistos.Write(self.file) if self.cfg_ana.doJetIdHisto: self.gluCleanHistosId.summary() self.gluCleanHistosId.Write(self.file) self.quarkCleanHistosId.summary() self.quarkCleanHistosId.Write(self.file) self.reweiMatchedCleanHistosId.summary() self.reweiMatchedCleanHistosId.Write(self.file) ### self.matchedCleanHistosId.Write(self.file) self.unmatchedCleanHistosId.Write(self.file) ### for j in self.jetKin: ### for h in j: ### h.Write(self.file) self.file.cd() self.h_nvtx.Write() self.h_genjetspt.Write() if self.cfg_ana.dumpTree: self.tree.Write() self.file.Close()
class SimpleJetAnalyzer (Analyzer) : '''A simple jet analyzer for Pietro.''' ### def __init__(self,cfg_ana, cfg_comp, looperName): ### loadLibs() ### super (SimpleJetAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName) def declareHandles (self) : super (SimpleJetAnalyzer, self).declareHandles () self.handles['jets'] = AutoHandle ( *self.cfg_ana.jetCollection ) if self.cfg_ana.useGenLeptons: self.mchandles['genParticlesPruned'] = AutoHandle ( *self.cfg_ana.GenParticlesCollection ) else: self.mchandles['genParticles'] = AutoHandle ( 'prunedGen', 'std::vector<reco::GenParticle>' ) self.mchandles['genJets'] = AutoHandle ( *self.cfg_ana.genJetsCollection ) self.handles['vertices'] = AutoHandle ( *self.cfg_ana.VtxCollection ) # .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... def beginLoop (self) : super (SimpleJetAnalyzer,self).beginLoop () self.file = TFile ('/'.join ([self.looperName, 'testJets.root']), 'recreate') if self.cfg_ana.applyPFLooseId: from ROOT import PFJetIDSelectionFunctor self.isPFLooseFunc = PFJetIDSelectionFunctor(0,PFJetIDSelectionFunctor.LOOSE) ## Workaround: for some reason PyROOT does not bind nor PFJetIDSelectionFunctor(Jet)PFJetIDSelectionFunctor.getBitsTemplates from ROOT import pat self.isPFLooseFunc.bits = pat.strbitset() for i in "CHF","NHF","CEF","NEF","NCH","nConstituents": self.isPFLooseFunc.bits.push_back(i) ## /Workaround self.isPFLoose = lambda x : self.isPFLooseFunc(x,self.isPFLooseFunc.bits) else: self.isPFLoose = lambda x : True # general histograms self.jetHistos = JetHistograms ('Jets') self.cleanJetHistos = JetHistograms ('CleanJets') self.matchedCleanJetHistos = JetHistograms ('MatchedCleanJets') self.matchedCleanJetHistos_barrel = JetHistograms ('MatchedCleanJets_barrel') self.matchedCleanJetHistos_endtk = JetHistograms ('MatchedCleanJets_endtk') self.matchedCleanJetHistos_endNOtk = JetHistograms ('MatchedCleanJets_endNOtk') self.matchedCleanJetHistos_fwd = JetHistograms ('MatchedCleanJets_fwd') self.LPtmatchedCleanJetHistos = JetHistograms ('LPtMatchedCleanJets') self.HPtmatchedCleanJetHistos = JetHistograms ('HPtMatchedCleanJets') self.unmatchedCleanJetHistos = JetHistograms ('UnmatchedCleanJets') self.LPtUnmatchedCleanJetHistos = JetHistograms ('LPtUnmatchedCleanJets') self.HPtUnmatchedCleanJetHistos = JetHistograms ('HPtUnmatchedCleanJets') # histograms of the components fraction self.matchedCleanJetHistosComponents = FractionJetHistograms ('MatchedCleanJetsCompontents') self.unmatchedCleanJetHistosComponents = FractionJetHistograms ('UnmatchedCleanJetsCompontents') # histograms for the resolution of matched jets self.matchedCleanJetHistosResolution = ResolutionJetHistograms ('MatchedCleanJetsResolution', 50, 1) self.matchedCleanJetHistosResolution_barrel = ResolutionJetHistograms ('MatchedCleanJetsResolution_barrel', 50, 1) self.matchedCleanJetHistosResolution_endtk = ResolutionJetHistograms ('MatchedCleanJetsResolution_endtk', 50, 1) self.matchedCleanJetHistosResolution_endNOtk = ResolutionJetHistograms ('MatchedCleanJetsResolution_endNOtk', 50, 1) self.matchedCleanJetHistosResolution_fwd = ResolutionJetHistograms ('MatchedCleanJetsResolution_fwd', 50, 1) print 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA' self.doJetIdHisto = False if self.cfg_ana.doJetIdHisto: print 'doing jet ID' # histograms for pileup jet identification variables self.vtxBins = (0,5,10,15,20,30) ## (0,2,4,6,10,15,20,30,35) self.ptBins = (20,30,50) ## (20,30,40,50,100) self.etaBins = (0,1.4,2.5,3.0) self.puEtaLables = ["_barrel","_endtk","_endNOtk","_fwd"] reweight_f = TF1("f","pol2(0)+expo(3)") reweight_f.SetParameters(0.1955298,-0.003830591,1.944794e-05,4.649755,-0.1722024) self.reweight = ("pt", reweight_f) self.doJetIdHisto = True self.gluCleanHistosId = PileupJetHistograms("GluonMatchedCleanHistosId",self.vtxBins,self.ptBins,self.etaBins,etalabels=self.puEtaLables,reweight=self.reweight, jetIdMva=self.cfg_ana.jetIdMva) self.quarkCleanHistosId = PileupJetHistograms("QuarkMatchedCleanHistosId",self.vtxBins,self.ptBins,self.etaBins,etalabels=self.puEtaLables,reweight=self.reweight, jetIdMva=self.cfg_ana.jetIdMva) self.reweiMatchedCleanHistosId = PileupJetHistograms("ReweiMatchedCleanHistosId",self.vtxBins,self.ptBins,self.etaBins,etalabels=self.puEtaLables,reweight=self.reweight, jetIdMva=self.cfg_ana.jetIdMva) self.unmatchedCleanHistosId = PileupJetHistograms("UnmatchedCleanHistosId",self.vtxBins,self.ptBins,self.etaBins,etalabels=self.puEtaLables, jetIdMva=self.cfg_ana.jetIdMva) self.h_nvtx = TH1F ("h_nvtx", "" ,50, 0, 50) self.h_genjetspt = TH1F ("h_genjetspt", "" ,500, 0, 500) self.h_secondClosestVsPtratio = TH2F ("h_secondClosestVsPtratio", "" ,100, 0, 2, 100, 0, 6) self.h_avedistanceVSNvtx = TH2F ("h_avedistanceVSNvtx", "" ,50, 0, 50, 100, 0, 6) self.h_PTRatioVSgenEta = TH2F ("h_PTRatioVSgenEta", "" ,150, -5, 5, 100, 0, 2) self.h_PTRatioVSgenPt = TH2F ("h_PTRatioVSgenPt", "" ,200, 0, 100, 100, 0, 2) self.h_matchDR = TH1F ("h_matchDR", "" ,60, 0, 0.30) self.h_relPtVSmatchDR = TH2F ("h_relPtVSmatchDR", "" ,60, 0, 0.30, 100, 0, 2) self.h_relPtVSchFrac = TH2F ("h_relPtVSchFrac", "" ,100, 0, 1, 100, 0, 2) # .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... def process (self, iEvent, event) : #read all the handles defined beforehand self.readCollections (iEvent) jetEtaCut = 4.5 # get the vertexes event.vertices = self.handles['vertices'].product () self.h_nvtx.Fill (len (event.vertices)) event.vertexBin = int (len (event.vertices)) # get the jets in the jets variable jets = self.handles['jets'].product () # filter jets with some selections event.jets = [ jet for jet in jets if ( abs(jet.eta()) < jetEtaCut and jet.pt()>self.cfg_ana.ptCut and self.isPFLoose(jet) ) ] self.jetHistos.fillEvent (event.jets) # get status 2 leptons if 'genParticlesPruned' in self.mchandles: event.genLeptons = [ lep for lep in self.mchandles['genParticlesPruned'].product() if lep.status() == 3 and (abs(lep.pdgId()) == 11 or abs(lep.pdgId()) == 13 or abs(lep.pdgId()) == 15) ] else: event.genLeptons = [ lep for lep in self.mchandles['genParticles'].product() if lep.status() == 3 and (abs(lep.pdgId()) == 11 or abs(lep.pdgId()) == 13 or abs(lep.pdgId()) == 15) ] # @ Pasquale: why level 3 and not level 2? # event.selGenLeptons = [GenParticle (lep) for lep in event.genLeptons if (lep.pt ()>self.cfg_ana.ptCut and abs (lep.eta ()) < jetEtaCut)] # get genJets event.genJets = map (GenJet, self.mchandles['genJets'].product ()) # filter genjets as for reco jets event.myGenJets = [GenJet (jet) for jet in event.genJets if (jet.pt ()>self.cfg_ana.genPtCut)] event.selGenJets = cleanObjectCollection (event.myGenJets, event.genLeptons, 0.2) # event.selGenJets = event.genJets for jet in event.selGenJets : self.h_genjetspt.Fill (jet.pt ()) event.noNegJets = [ jet for jet in event.jets if (jet.jecFactor(0) > 0) ] # event.noNegJets = [ jet for jet in event.jets] # first stats plots # print 'genLeptons : ' + repr (len (event.genLeptons)) + ' | genJets : ' + repr (len (event.genJets)) + ' | recoJets : ' + repr (len (event.jets)) self.jetHistos.fillStats (len (event.selGenJets), len (event.noNegJets)) #FIXME why are there cases in which there's 4 or 6 leptons? if len (event.genLeptons) > 2 : return # in case I want to filter out taus # 11, 13, 15 : e, u, T # event.genOneLepton = [GenParticle (part) for part in event.genLeptons if abs (part.pdgId ()) == 15] # remove leptons from jets if closer than 0.2 event.cleanJets = cleanObjectCollection (event.noNegJets, event.genLeptons, 0.2) # event.cleanJets = event.noNegJets self.cleanJetHistos.fillEvent (event.cleanJets) # print len (jets),len (event.jets), len (event.noNegJets), len (event.cleanJets), len (event.genLeptons),"-->",(len (event.noNegJets) - len (event.cleanJets) - len (event.genLeptons)) event.matchingCleanJets = matchObjectCollection2 (event.cleanJets, event.selGenJets, 0.25) # assign to each jet its gen match (easy life :)) for jet in event.cleanJets : jet.gen = event.matchingCleanJets[ jet ] # FIXME next step might be to put this in the matching and remove the boolean flags event.matchedCleanJets = [jet for jet in event.matchingCleanJets if jet.gen != None] event.cleanGluJets = [] event.cleanQuarkJets = [] for jet in event.matchedCleanJets: flav = abs(jet.partonFlavour()) if flav == 21: event.cleanGluJets.append(jet) elif flav > 0 and flav <= 3: event.cleanQuarkJets.append(jet) event.LPtmatchedCleanJets = [jet for jet in event.matchingCleanJets if jet.gen != None and jet.pt () <= 30] event.HPtmatchedCleanJets = [jet for jet in event.matchingCleanJets if jet.gen != None and jet.pt () > 30] self.matchedCleanJetHistos.fillEvent (event.matchedCleanJets) self.LPtmatchedCleanJetHistos.fillEvent (event.LPtmatchedCleanJets) self.HPtmatchedCleanJetHistos.fillEvent (event.HPtmatchedCleanJets) event.unmatchedCleanJets = [jet for jet in event.matchingCleanJets if jet.gen == None] event.LPtunmatchedCleanJets = [jet for jet in event.matchingCleanJets if jet.gen == None and jet.pt () <= 30] event.HPtunmatchedCleanJets = [jet for jet in event.matchingCleanJets if jet.gen == None and jet.pt () > 30] self.unmatchedCleanJetHistos.fillEvent (event.unmatchedCleanJets) self.LPtUnmatchedCleanJetHistos.fillEvent (event.LPtunmatchedCleanJets) self.HPtUnmatchedCleanJetHistos.fillEvent (event.HPtunmatchedCleanJets) self.matchedCleanJetHistosComponents.fillEvent (event.matchedCleanJets) self.unmatchedCleanJetHistosComponents.fillEvent (event.unmatchedCleanJets) self.matchedCleanJetHistosResolution.fillEvent (event.matchedCleanJets, len (event.vertices)) for jet in event.matchedCleanJets : if abs (jet.gen.eta ()) < 1.4 : self.matchedCleanJetHistosResolution_barrel.fillJet (jet, len (event.vertices)) self.matchedCleanJetHistos_barrel.fillJet (jet) elif 1.6 < abs (jet.gen.eta ()) < 2.5 : self.matchedCleanJetHistosResolution_endtk.fillJet (jet, len (event.vertices)) self.matchedCleanJetHistos_endtk.fillJet (jet) elif 2.6 < abs (jet.gen.eta ()) < 2.9 : self.matchedCleanJetHistosResolution_endNOtk.fillJet (jet, len (event.vertices)) self.matchedCleanJetHistos_endNOtk.fillJet (jet) elif 3.1 < abs (jet.gen.eta ()) : self.matchedCleanJetHistosResolution_fwd.fillJet (jet, len (event.vertices)) self.matchedCleanJetHistos_fwd.fillJet (jet) ##PG debugging for tails #for jet in event.matchedCleanJets : #deltaRR = deltaR( jet.eta (), jet.phi (), jet.gen.eta (), jet.gen.phi ()) #self.h_matchDR.Fill (deltaRR) #self.h_relPtVSmatchDR.Fill (deltaRR, jet.pt () / jet.gen.pt ()) #if abs (jet.gen.eta ()) > 2.5 and abs (jet.gen.eta ()) < 3 : #self.h_relPtVSchFrac.Fill (jet.chargedHadronEnergyFraction (), jet.pt () / jet.gen.pt ()) #if jet.gen.pt () > 20 and jet.gen.pt () < 40 : #self.h_PTRatioVSgenEta.Fill (jet.gen.eta (), jet.pt () / jet.gen.pt ()) #if abs (jet.gen.eta ()) > 1.6 : #self.h_PTRatioVSgenPt.Fill (jet.gen.pt (), jet.pt () / jet.gen.pt ()) #minDelta = 10 #secondClosest = jet #for recojet in event.cleanJets : #if recojet == jet : continue #dr2 = deltaR2( jet.gen.eta (), jet.gen.phi (), recojet.eta (), recojet.phi ()) #if dr2 < minDelta : #minDelta = dr2 #secondClosest = recojet ##if len(event.vertices) < 10 or abs (jet.gen.eta ()) < 1.6: continue #self.h_secondClosestVsPtratio.Fill (jet.pt () / jet.gen.pt (), math.sqrt (minDelta)) #if jet.pt () / jet.gen.pt () < 0.2 and jet.gen.pt () > 20 and abs (jet.gen.eta ()) < 3 and abs (jet.gen.eta ()) > 2.5 : #print '============',len(event.genLeptons) #print jet.pt (), jet.eta (), jet.phi (), jet.jecFactor (0) #print jet.gen.pt (), jet.gen.eta (), jet.gen.phi () #print '------------ leptons:' #for lept in event.genLeptons : #print lept.pt (), lept.eta (), lept.phi () #print '------------' #for recojet in event.cleanJets : #print "RECO",recojet.pt (), recojet.eta (), recojet.phi (), recojet.jecFactor (0) #for genjet in event.selGenJets : #print "GEN ",genjet.pt (), genjet.eta (), genjet.phi () #aveDeltaR = 0 #num = 0 #for recojet1 in event.cleanJets : #minDelta = 10 #closest = recojet1 #for recojet2 in event.cleanJets : #if recojet1 == recojet2 : continue #dr2 = deltaR2( recojet1.eta (), recojet1.phi (), recojet2.eta (), recojet2.phi ()) #if dr2 < minDelta : #minDelta = dr2 #closest = recojet2 #if minDelta == 10 : continue #aveDeltaR = aveDeltaR + math.sqrt (minDelta) #num = num + 1 #if num > 0 : #aveDeltaR = aveDeltaR / num #self.h_avedistanceVSNvtx.Fill (len(event.vertices), aveDeltaR) if self.doJetIdHisto: self.gluCleanHistosId.fillEvent(event.cleanGluJets,event.vertices) self.quarkCleanHistosId.fillEvent(event.cleanQuarkJets,event.vertices) ### self.matchedCleanHistosId.fillEvent(event.matchedCleanJets,event.vertices) self.reweiMatchedCleanHistosId.fillEvent(event.matchedCleanJets,event.vertices) self.unmatchedCleanHistosId.fillEvent(event.unmatchedCleanJets,event.vertices) # .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... .... def write (self): from ROOT import gROOT gROOT.SetBatch(True) self.jetHistos.Write (self.file) self.cleanJetHistos.Write (self.file) self.matchedCleanJetHistos.Write (self.file) self.matchedCleanJetHistos_barrel.Write (self.file) self.matchedCleanJetHistos_endtk.Write (self.file) self.matchedCleanJetHistos_endNOtk.Write (self.file) self.matchedCleanJetHistos_fwd.Write (self.file) self.LPtmatchedCleanJetHistos.Write (self.file) self.HPtmatchedCleanJetHistos.Write (self.file) self.LPtUnmatchedCleanJetHistos.Write (self.file) self.HPtUnmatchedCleanJetHistos.Write (self.file) self.unmatchedCleanJetHistos.Write (self.file) self.matchedCleanJetHistosComponents.summary () self.matchedCleanJetHistosComponents.Write (self.file) self.unmatchedCleanJetHistosComponents.summary () self.unmatchedCleanJetHistosComponents.Write (self.file) self.matchedCleanJetHistosResolution.summary () self.matchedCleanJetHistosResolution.Write (self.file) self.matchedCleanJetHistosResolution_barrel.summary () self.matchedCleanJetHistosResolution_barrel.Write (self.file) self.matchedCleanJetHistosResolution_endtk.summary () self.matchedCleanJetHistosResolution_endtk.Write (self.file) self.matchedCleanJetHistosResolution_endNOtk.summary () self.matchedCleanJetHistosResolution_endNOtk.Write (self.file) self.matchedCleanJetHistosResolution_fwd.summary () self.matchedCleanJetHistosResolution_fwd.Write (self.file) if self.doJetIdHisto: self.gluCleanHistosId.summary() self.gluCleanHistosId.Write(self.file) self.quarkCleanHistosId.summary() self.quarkCleanHistosId.Write(self.file) self.reweiMatchedCleanHistosId.summary() self.reweiMatchedCleanHistosId.Write(self.file) ### self.matchedCleanHistosId.Write(self.file) self.unmatchedCleanHistosId.Write(self.file) self.file.cd () self.h_nvtx.Write () self.h_genjetspt.Write () self.h_secondClosestVsPtratio.Write () self.h_avedistanceVSNvtx.Write () self.h_PTRatioVSgenEta.Write () self.h_PTRatioVSgenPt.Write () self.h_matchDR.Write () self.h_relPtVSmatchDR.Write () self.h_relPtVSchFrac.Write () self.file.Close()