ProvenanceAnalyzer.py

Go to the documentation of this file.

from __future__ import print_function
from builtins import range
import itertools
 
from PhysicsTools.Heppy.analyzers.core.Analyzer import Analyzer
from PhysicsTools.Heppy.analyzers.core.AutoHandle import AutoHandle
from PhysicsTools.HeppyCore.framework.event import Event
from PhysicsTools.HeppyCore.statistics.counter import Counter, Counters
 
from DataFormats.FWLite import Events, Handle,Lumis
 
class ProvenanceAnalyzer( Analyzer ):
    #---------------------------------------------
    # TO FINDS THE INITIAL EVENTS BEFORE THE SKIM
    #---------------------------------------------
    
    def __init__(self, cfg_ana, cfg_comp, looperName):
        super(ProvenanceAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName)
        self.lastId = (0,0)
        self.provenance = []
 
    def declareHandles(self):
        super(ProvenanceAnalyzer, self).declareHandles()
    
    def cmsswVNums(self, release):
        vpieces = release.split("_")[1:]
        vnums = [ int(v) for v in vpieces[:3] ]
        if len(vpieces) > 3:
            if "patch" in vpieces[3]:
                vnums.append(int(vpieces[3].replace("patch","")))
            elif "pre" in vpieces[3]:
                vnums.append(-100+int(vpieces[3].replace("pre","")))
        else:
            vnums.append(0)
        return tuple(vnums)
    def miniAODVersion(self, vnums):
        if vnums[:2] == (7,4):
            if vnums >= (7,4,14,0): return (2015,2.1)
            if vnums >= (7,4,12,0): return (2015,2.0)
            if vnums >= (7,4, 8,1): return (2015,1.1)
            return (2015,1.0)
        elif vnums[:2] == (7,2):
            return (2014,2.1)   
        elif vnums[:2] == (7,0):
            if vnums >= (7,0,9,1): return (2014,2.0)
            return (2014,1.0)
        else:
            return (-999,-999)
 
    def process(self, event):
        eid = (  event.input.eventAuxiliary().id().run(),
                 event.input.eventAuxiliary().id().luminosityBlock() )
        if eid != self.lastId:
            #import pdb; pdb.set_trace()
            history = event.input.object().processHistory()
            for i in reversed(list(range(history.size()))):
                conf = history.at(i)
                release = conf.releaseVersion().replace('"',"")
                vnums = self.cmsswVNums(release)
                if conf.processName() in  ("PAT", "RECO","REC2"):
                    print("processing step %d: process %s, release %r %s, miniAOD %s" % (i, conf.processName(), release, vnums, self.miniAODVersion(vnums)))
                    self.provenance = [
                        ('release', release),
                        ('releaseVersion', vnums),
                        ('miniAODVersion', self.miniAODVersion(vnums)),
                    ]
                    break
            self.lastId = eid
        for key,value in self.provenance:
            setattr(event, key, value)
        return True
 
import PhysicsTools.HeppyCore.framework.config as cfg
setattr(ProvenanceAnalyzer,"defaultConfig", cfg.Analyzer(
    class_object = ProvenanceAnalyzer
))