hgcalTriggerChains.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms

class HGCalTriggerChains:
    def __init__(self):
        self.vfe = {}
        self.concentrator = {}
        self.truth_prod = {}
        self.backend1 = {}
        self.backend2 = {}
        self.selector = {}
        self.ntuple = {}
        self.chain = []
        self.truth_chain = []

    def register_vfe(self, name, generator):
        self.vfe[name] = generator

    def register_concentrator(self, name, generator):
        self.concentrator[name] = generator

    def register_truth_prod(self, name, generator): # for truth-matched trigger cells
        self.truth_prod[name] = generator

    def register_backend1(self, name, generator):
        self.backend1[name] = generator

    def register_backend2(self, name, generator):
        self.backend2[name] = generator

    def register_selector(self, name, generator):
        self.selector[name] = generator

    def register_ntuple(self, name, generator):
        self.ntuple[name] = generator

    def register_chain(self, vfe, concentrator, backend1, backend2, selector='', ntuple=''):
        if not vfe in self.vfe:
            raise KeyError('{} not registered as VFE producer'.format(vfe))
        if not concentrator in self.concentrator:
            raise KeyError('{} not registered as concentrator producer'.format(concentrator))
        if not backend1 in self.backend1:
            raise KeyError('{} not registered as backend1 producer'.format(backend1))
        if not backend2 in self.backend2:
            raise KeyError('{} not registered as backend2 producer'.format(backend2))
        if selector!='' and not selector in self.selector:
            raise KeyError('{} not registered as selector'.format(selector))
        if ntuple!='' and not ntuple in self.ntuple:
            raise KeyError('{} not registered as ntuplizer'.format(ntuple))
        self.chain.append( (vfe, concentrator, backend1, backend2, selector, ntuple) )

    def register_truth_chain(self, vfe, truth_prod, backend1='', backend2='', selector='', ntuple=''):
        if not vfe in self.vfe:
            raise KeyError('{} not registered as VFE producer'.format(vfe))
        if not truth_prod in self.truth_prod:
            raise KeyError('{} not registered as truth producer'.format(truth_prod))
        if backend1!='' and not backend1 in self.backend1:
            raise KeyError('{} not registered as backend1 producer'.format(backend1))
        if backend2!='' and not backend2 in self.backend2:
            raise KeyError('{} not registered as backend2 producer'.format(backend2))
        if selector!='' and not selector in self.selector:
            raise KeyError('{} not registered as selector'.format(selector))
        if ntuple!='' and not ntuple in self.ntuple:
            raise KeyError('{} not registered as ntuplizer'.format(ntuple))
        self.truth_chain.append( (vfe, truth_prod, backend1, backend2, selector, ntuple) )

    def create_sequences(self, process):
        if not hasattr(process, 'hgcalTriggerSelector'):
            process.load('L1Trigger.L1THGCalUtilities.HGC3DClusterGenMatchSelector_cff')
        tmp = cms.TaskPlaceholder("tmp")
        tmpseq = cms.SequencePlaceholder("tmp")
        vfe_task = cms.Task(tmp)
        concentrator_task = cms.Task(tmp)
        backend1_task = cms.Task(tmp)
        backend2_task = cms.Task(tmp)
        selector_sequence = cms.Sequence(tmpseq)
        ntuple_sequence = cms.Sequence(tmpseq)
        for vfe,concentrator,backend1,backend2,selector,ntuple in self.chain:
            concentrator_name = '{0}{1}'.format(vfe, concentrator)
            backend1_name = '{0}{1}{2}'.format(vfe, concentrator, backend1)
            backend2_name = '{0}{1}{2}{3}'.format(vfe, concentrator, backend1, backend2)
            selector_name = '{0}{1}{2}{3}{4}'.format(vfe, concentrator, backend1, backend2, selector)
            ntuple_name = '{0}{1}{2}{3}{4}{5}'.format(vfe, concentrator, backend1, backend2, selector, ntuple)
            if not hasattr(process, vfe):
                setattr(process, vfe, self.vfe[vfe](process))
                vfe_task.add(getattr(process, vfe))
            if not hasattr(process, concentrator_name):
                vfe_processor = getattr(process, vfe).ProcessorParameters.ProcessorName.value()
                setattr(process, concentrator_name, self.concentrator[concentrator](process, vfe+':'+vfe_processor))
                concentrator_task.add(getattr(process, concentrator_name))
            if not hasattr(process, backend1_name):
                concentrator_processor = getattr(process, concentrator_name).ProcessorParameters.ProcessorName.value()
                setattr(process, backend1_name, self.backend1[backend1](process, concentrator_name+':'+concentrator_processor))
                backend1_task.add(getattr(process, backend1_name))
            if not hasattr(process, backend2_name):
                backend1_processor = getattr(process, backend1_name).ProcessorParameters.ProcessorName.value()
                setattr(process, backend2_name, self.backend2[backend2](process, backend1_name+':'+backend1_processor))
                backend2_task.add(getattr(process, backend2_name))
            if selector!='' and not hasattr(process, selector_name):
                backend2_processor = getattr(process, backend2_name).ProcessorParameters.ProcessorName.value()
                setattr(process, selector_name, self.selector[selector](process, backend2_name+':'+backend2_processor))
                selector_sequence *= getattr(process, selector_name)

            if selector=='':
                ntuple_inputs = [
                        concentrator_name+':'+getattr(process, concentrator_name).ProcessorParameters.ProcessorName.value(),
                        backend1_name+':'+getattr(process, backend1_name).ProcessorParameters.ProcessorName.value(),
                        backend2_name+':'+getattr(process, backend2_name).ProcessorParameters.ProcessorName.value()
                        ]
            else:
                ntuple_inputs = [
                        concentrator_name+':'+getattr(process, concentrator_name).ProcessorParameters.ProcessorName.value(),
                        backend1_name+':'+getattr(process, backend1_name).ProcessorParameters.ProcessorName.value(),
                        selector_name]

            if ntuple!='' and not hasattr(process, ntuple_name):
                setattr(process, ntuple_name, self.ntuple[ntuple](process, ntuple_inputs))
                ntuple_sequence *= getattr(process, ntuple_name)
        vfe_task.remove(tmp)
        concentrator_task.remove(tmp)
        backend1_task.remove(tmp)
        backend2_task.remove(tmp)
        selector_sequence.remove(tmpseq)
        ntuple_sequence.remove(tmpseq)
        process.globalReplace('hgcalVFE', vfe_task)
        process.globalReplace('hgcalConcentrator', concentrator_task)
        if 'HGCalBackendStage1Processor' in backend1_processor:
            process.globalReplace('hgcalBackEndStage1', backend1_task)
            process.globalReplace('hgcalBackEndStage2', backend2_task)
        else:
            process.globalReplace('hgcalBackEndLayer1', backend1_task)
            process.globalReplace('hgcalBackEndLayer2', backend2_task)
        process.globalReplace('hgcalTriggerSelector', selector_sequence)
        process.globalReplace('hgcalTriggerNtuples', ntuple_sequence)
        return process

    def create_truth_sequences(self, process):
        if not hasattr(process, 'caloTruthCellsProducer'):
            from L1Trigger.L1THGCalUtilities.caloTruthCellsProducer_cfi import caloTruthCellsProducer
            process.load('L1Trigger.L1THGCalUtilities.caloTruthCells_cff')
            process.truthCellsProd = cms.Task(caloTruthCellsProducer)
            hgcalTriggerPrimitivesTask = cms.Task(process.hgcalVFE, process.truthCellsProd, process.hgcalBackEndLayer1, process.hgcalBackEndLayer2, process.hgcalTowerMap, process.hgcalTower)
            process.hgcalTriggerPrimitivesTruth = cms.Sequence(hgcalTriggerPrimitivesTask)
        if not hasattr(process, 'hgcalTriggerSelector'):
            process.load('L1Trigger.L1THGCalUtilities.HGC3DClusterGenMatchSelector_cff')
        tmp = cms.TaskPlaceholder("tmp")
        tmpseq = cms.SequencePlaceholder("tmp")
        vfe_task = cms.Task(tmp)
        truth_prod_task = cms.Task(tmp)
        backend1_task = cms.Task(tmp)
        backend2_task = cms.Task(tmp)
        selector_sequence = cms.Sequence(tmpseq)
        ntuple_sequence = cms.Sequence(tmpseq)
        for vfe,truth_prod,backend1,backend2,selector,ntuple in self.truth_chain:
            truth_prod_name = '{0}{1}'.format(vfe, truth_prod)
            backend1_name = truth_prod_name + '{0}'.format(backend1)
            backend2_name = backend1_name + '{0}'.format(backend2)
            selector_name = backend2_name + '{0}'.format(selector)
            ntuple_name = selector_name + '{0}'.format(ntuple)
            if selector=='':
                if backend2=='':
                    if backend1=='':
                        ntuple_inputs = [
                                truth_prod_name,
                                truth_prod_name,
                                truth_prod_name
                                ]
                    else:
                        ntuple_inputs = [
                                truth_prod_name,
                                backend1_name+':HGCalBackendLayer1Processor2DClustering',
                                truth_prod_name
                                ]
                else:
                    ntuple_inputs = [
                            truth_prod_name,
                            backend1_name+':HGCalBackendLayer1Processor2DClustering',
                            backend2_name+':HGCalBackendLayer2Processor3DClustering'
                            ]
            else:
                ntuple_inputs = [
                        truth_prod_name,
                        backend1_name+':HGCalBackendLayer1Processor2DClustering',
                        selector_name
                        ]

            if not hasattr(process, vfe):
                setattr(process, vfe, self.vfe[vfe](process))
                vfe_task.add(getattr(process, vfe))
            if not hasattr(process, truth_prod_name):
                setattr(process, truth_prod_name, self.truth_prod[truth_prod](process, vfe))
                truth_prod_task.add(getattr(process, truth_prod_name))
            if not hasattr(process, backend1_name):
                setattr(process, backend1_name, self.backend1[backend1](process, truth_prod_name))
                backend1_task.add(getattr(process, backend1_name))
            if backend2!='' and not hasattr(process, backend2_name):
                setattr(process, backend2_name, self.backend2[backend2](process, backend1_name))
                backend2_task.add(getattr(process, backend2_name))
            if selector!='' and not hasattr(process, selector_name):
                setattr(process, selector_name, self.selector[selector](process, backend2_name))
                selector_sequence *= getattr(process, selector_name)
            if ntuple!='' and not hasattr(process, ntuple_name):
                setattr(process, ntuple_name, self.ntuple[ntuple](process, ntuple_inputs))
                ntuple_sequence *= getattr(process, ntuple_name)
        vfe_task.remove(tmp)
        truth_prod_task.remove(tmp)
        backend1_task.remove(tmp)
        backend2_task.remove(tmp)
        selector_sequence.remove(tmpseq)
        ntuple_sequence.remove(tmpseq)
        process.globalReplace('hgcalVFE', vfe_task)
        process.globalReplace('truthCellsProd', truth_prod_task)
        process.globalReplace('hgcalBackEndLayer1', backend1_task)
        process.globalReplace('hgcalBackEndLayer2', backend2_task)
        process.globalReplace('hgcalTriggerSelector', selector_sequence)
        process.globalReplace('hgcalTriggerNtuples', ntuple_sequence)
        return process