Analyzer.py

Go to the documentation of this file.

from __future__ import print_function
import os 
import logging 

from PhysicsTools.HeppyCore.framework.analyzer import Analyzer as CoreAnalyzer
import six

class Analyzer(CoreAnalyzer):
    '''Base Analyzer class. Used in Looper.'''
    
    def declareHandles(self):
        self.handles = {}
        self.mchandles = {}

    def beginLoop(self, setup):
        '''Automatically called by Looper, for all analyzers.'''
        super(Analyzer, self).beginLoop(setup)
        self.declareHandles()
        

    def process(self, event ):
        '''Automatically called by Looper, for all analyzers.
        each analyzer in the sequence will be passed the same event instance.
        each analyzer can access, modify, and store event information, of any type.'''
        print(self.cfg_ana.name)
        self.readCollections( event.input )

    def readCollections(self, iEvent ):
        '''You must call this function at the beginning of the process
        function of your child analyzer.'''
        # if not self.beginLoopCalled:
        #    # necessary in case the user calls process to go straight to a given event, before looping
        #    self.beginLoop(setup)
        for str,handle in six.iteritems(self.handles):
            handle.Load( iEvent )
        if self.cfg_comp.isMC:
            for str,handle in six.iteritems(self.mchandles):
                handle.Load( iEvent )