L1TNtuples/analysis/example.py

0001 #!/usr/bin/env python
0002
0003 """
0004 Example L1TNtuple analysis program
0005 """
0006
0007
0008 import ROOT
0009
0010
0011 # apparently not needed...
0012 # ROOT.gSystem.Load("libL1TriggerL1TNtuples")
0013
0014 ROOT.PyConfig.IgnoreCommandLineOptions = True
0015 ROOT.gROOT.SetBatch(1)
0016 ROOT.TH1.SetDefaultSumw2(True)
0017 ROOT.gStyle.SetOptStat(0)
0018
0019
0020 def eventLoop(filename):
0021     f = ROOT.TFile(filename)
0022
0023     treeL1   = f.Get("l1NtupleProducer/L1Tree")
0024     treeTow  = f.Get("l1CaloTowerTreeProducer/L1CaloTowerTree")
0025 #    tree_l1ex = f.Get("l1ExtraTreeProducer/L1ExtraTree")
0026     treeL1up = f.Get("l1UpgradeTreeProducer/L1UpgradeTree")
0027
0028     treeJet  = f.Get("l1JetRecoTreeProducer/JetRecoTree")
0029 #    treeEG   = f.Get("l1EGRecoTreeProducer/EGRecoTree")
0030 #    treeTau  = f.Get("l1TauRecoTreeProducer/TauRecoTree")
0031 #    treeMu   = f.Get("l1MuonRecoTreeProducer/MuonRecoTree")
0032
0033     treeL1.AddFriend(treeTow)
0034 #    treeL1.AddFriend(treeL1ex)
0035     treeL1.AddFriend(treeL1up)
0036     treeL1.AddFriend(treeJet)
0037
0038
0039     for jentry, event in enumerate(tree):
0040         if jentry >= nevents:
0041             break
0042
0043         event      = treeL1.Event
0044         eventSim   = treeL1.Simulation
0045         eventRCT   = treeL1.RCT
0046         eventGCT   = treeL1.GCT
0047         eventDTTF  = treeL1.DTTF
0048         eventCSCTF = treeL1.CSCTF
0049         eventGMT   = treeL1.GMT
0050         eventGT    = treeL1.GT
0051
0052         eventCaloTP = treeTow.CaloTP
0053         eventTower  = treeTow.L1CaloTower
0054
0055         eventL1Up  = treeL1up.L1Upgrade
0056
0057         eventJet   = treeJet.Jet
0058
0059 #        eventEG    = treeEG.EG
0060
0061 #        eventTau   = treeTau.Tau
0062
0063 #        eventMuon  = treeMu.Mu
0064
0065
0066     # Print hists to file
0067     c = ROOT.TCanvas()
0068
0069
0070     h.Draw("")
0071     h.SetTitleOffset(0.55, 'Z')
0072     c.Print("plot.pdf")
0073
0074     f.Close()  # make sure I go last! otherwise your hists will be NoneType
0075
0076
0077 if __name__ == "__main__":
0078     towercorr()