| 1 | from BaseControlPlots import BaseControlPlots
|
|---|
| 2 |
|
|---|
| 3 | # Requirements:
|
|---|
| 4 | # event.jets
|
|---|
| 5 | # event.bjets
|
|---|
| 6 | # event.MEt
|
|---|
| 7 |
|
|---|
| 8 | class JetControlPlots(BaseControlPlots):
|
|---|
| 9 | """A class to create control plots for jetmet"""
|
|---|
| 10 |
|
|---|
| 11 | def __init__(self, dir=None, dataset=None, mode="plots"):
|
|---|
| 12 | # create output file if needed. If no file is given, it means it is delegated
|
|---|
| 13 | BaseControlPlots.__init__(self, dir=dir, purpose="jetmet", dataset=dataset, mode=mode)
|
|---|
| 14 |
|
|---|
| 15 | def beginJob(self):
|
|---|
| 16 | # declare histograms
|
|---|
| 17 | self.add("JetPt","Jet Pt",100,0,200)
|
|---|
| 18 | self.add("JetEta","Jet Eta",100,-5,5)
|
|---|
| 19 | self.add("JetPhi","Jet Phi",64,-3.2,3.2)
|
|---|
| 20 | self.add("BjetPt","B Jet Pt",100,0,200)
|
|---|
| 21 | self.add("BjetEta","B Jet Eta",100,-5,5)
|
|---|
| 22 | self.add("BjetPhi","B Jet Phi",64,-3.2,3.2)
|
|---|
| 23 | self.add("MET","MET",100,0,200)
|
|---|
| 24 | self.add("METphi","MET phi",64,-3.2,3.2)
|
|---|
| 25 | self.add("Njets","Jet multiplicity",10,0,10)
|
|---|
| 26 | self.add("Nbjets","B Jet multiplicity",10,0,10)
|
|---|
| 27 |
|
|---|
| 28 | def process(self, event):
|
|---|
| 29 | #get information
|
|---|
| 30 | result = { }
|
|---|
| 31 | result["JetPt"] = [ ]
|
|---|
| 32 | result["JetEta"] = [ ]
|
|---|
| 33 | result["JetPhi"] = [ ]
|
|---|
| 34 | result["BjetPt"] = [ ]
|
|---|
| 35 | result["BjetEta"] = [ ]
|
|---|
| 36 | result["BjetPhi"] = [ ]
|
|---|
| 37 | nb=0
|
|---|
| 38 | for jet in event.selectedJets:
|
|---|
| 39 | result["JetPt"].append(jet.PT)
|
|---|
| 40 | result["JetEta"].append(jet.Eta)
|
|---|
| 41 | result["JetPhi"].append(jet.Phi)
|
|---|
| 42 | if jet.BTag:
|
|---|
| 43 | nb+=1
|
|---|
| 44 | result["BjetPt"].append(jet.PT)
|
|---|
| 45 | result["BjetEta"].append(jet.Eta)
|
|---|
| 46 | result["BjetPhi"].append(jet.Phi)
|
|---|
| 47 | result["Njets"] = len(event.selectedJets)
|
|---|
| 48 | result["Nbjets"] = nb
|
|---|
| 49 | result["MET"] = event.MEt[0].MET
|
|---|
| 50 | result["METphi"] = event.MEt[0].Phi
|
|---|
| 51 | return result
|
|---|
| 52 |
|
|---|
| 53 | if __name__=="__main__":
|
|---|
| 54 | import sys
|
|---|
| 55 | from DelphesAnalysis.BaseControlPlots import runTest
|
|---|
| 56 | runTest(sys.argv[1], JetControlPlots())
|
|---|
| 57 |
|
|---|
