Add default configs and option to store individual bits for trigger s…

…tudies

PhysicsTools/Heppy/python/analyzers/core/TriggerBitAnalyzer.py

@@ -3,12 +3,15 @@
 from PhysicsTools.Heppy.analyzers.core.Analyzer import Analyzer
 from PhysicsTools.Heppy.analyzers.core.AutoHandle import AutoHandle
 from PhysicsTools.Heppy.analyzers.core.AutoFillTreeProducer  import NTupleVariable
+import PhysicsTools.HeppyCore.framework.config as cfg
 
 class TriggerBitAnalyzer( Analyzer ):
     def __init__(self, cfg_ana, cfg_comp, looperName ):
         super(TriggerBitAnalyzer,self).__init__(cfg_ana,cfg_comp,looperName)
         self.processName = getattr(self.cfg_ana,"processName","HLT")
         self.outprefix   = getattr(self.cfg_ana,"outprefix",  self.processName)
+        self.unrollbits = ( hasattr(self.cfg_ana,"unrollbits") and self.cfg_ana.unrollbits )
+
 
     def declareHandles(self):
         super(TriggerBitAnalyzer, self).declareHandles()
@@ -17,22 +20,72 @@ def declareHandles(self):
     def beginLoop(self, setup):
         super(TriggerBitAnalyzer,self).beginLoop(setup)
         self.triggerBitCheckers = []
+        if self.unrollbits :
+            self.allPaths = set()
+            self.triggerBitCheckersSingleBits = []
+
         for T, TL in self.cfg_ana.triggerBits.iteritems():
                 trigVec = ROOT.vector(ROOT.string)()
                 for TP in TL:
                     trigVec.push_back(TP)
-                outname="%s_%s"%(self.outprefix,T)
+                    if self.unrollbits :
+                        if TP not in self.allPaths :
+                            self.allPaths.update(TP)
+                            trigVecBit = ROOT.vector(ROOT.string)()
+                            trigVecBit.push_back(TP)
+                            outname="%s_BIT_%s"%(self.outprefix,TP)
+                            if not hasattr(setup ,"globalVariables") :
+                                setup.globalVariables = []
+                            setup.globalVariables.append( NTupleVariable(outname, eval("lambda ev: ev.%s" % outname), help="Trigger bit  %s"%TP) )
+                            self.triggerBitCheckersSingleBits.append( (TP, ROOT.heppy.TriggerBitChecker(trigVecBit)) )
+
+                outname="%s_%s"%(self.outprefix,T)  
                 if not hasattr(setup ,"globalVariables") :
                         setup.globalVariables = []
                 setup.globalVariables.append( NTupleVariable(outname, eval("lambda ev: ev.%s" % outname), help="OR of %s"%TL) )
                 self.triggerBitCheckers.append( (T, ROOT.heppy.TriggerBitChecker(trigVec)) )
+
 
     def process(self, event):
         self.readCollections( event.input )
         triggerResults = self.handles['TriggerResults'].product()
         for T,TC in self.triggerBitCheckers:
             outname="%s_%s"%(self.outprefix,T)
             setattr(event,outname, TC.check(event.input.object(), triggerResults))
+        if self.unrollbits :
+            for TP,TC in self.triggerBitCheckersSingleBits:
+               outname="%s_BIT_%s"%(self.outprefix,TP)
+               setattr(event,outname, TC.check(event.input.object(), triggerResults))
+
 
         return True
 
+
+setattr(TriggerBitAnalyzer,"defaultConfig",cfg.Analyzer(
+    TriggerBitAnalyzer, name="TriggerFlags",
+    processName = 'HLT',
+    triggerBits = {
+        # "<name>" : [ 'HLT_<Something>_v*', 'HLT_<SomethingElse>_v*' ] 
+}
+)
+)
+setattr(TriggerBitAnalyzer,"defaultEventFlagsConfig",cfg.Analyzer(
+    TriggerBitAnalyzer, name="EventFlags",
+    processName = 'PAT',
+    outprefix   = 'Flag',
+    triggerBits = {
+        "HBHENoiseFilter" : [ "Flag_HBHENoiseFilter" ],
+        "CSCTightHaloFilter" : [ "Flag_CSCTightHaloFilter" ],
+        "hcalLaserEventFilter" : [ "Flag_hcalLaserEventFilter" ],
+        "EcalDeadCellTriggerPrimitiveFilter" : [ "Flag_EcalDeadCellTriggerPrimitiveFilter" ],
+        "goodVertices" : [ "Flag_goodVertices" ],
+        "trackingFailureFilter" : [ "Flag_trackingFailureFilter" ],
+        "eeBadScFilter" : [ "Flag_eeBadScFilter" ],
+        "ecalLaserCorrFilter" : [ "Flag_ecalLaserCorrFilter" ],
+        "trkPOGFilters" : [ "Flag_trkPOGFilters" ],
+        "trkPOG_manystripclus53X" : [ "Flag_trkPOG_manystripclus53X" ],
+        "trkPOG_toomanystripclus53X" : [ "Flag_trkPOG_toomanystripclus53X" ],
+        "trkPOG_logErrorTooManyClusters" : [ "Flag_trkPOG_logErrorTooManyClusters" ],
+    }
+)
+)