earlyDeleteSettings_cff Namespace Reference

Classes
class	TestHasInputTagModuleLabel

Functions
def	_hasInputTagModuleLabel (process, pset, moduleLabels, result)
def	customiseEarlyDelete (process)

Function Documentation

def earlyDeleteSettings_cff._hasInputTagModuleLabel (   process,   pset,   moduleLabels,   result  )

private

Definition at line 10 of file earlyDeleteSettings_cff.py.

Referenced by customiseEarlyDelete().

def _hasInputTagModuleLabel(process, pset, moduleLabels,result):
    for name in pset.parameterNames_():
        value = getattr(pset,name)
        if isinstance(value, cms.PSet):
            _hasInputTagModuleLabel(process, value, moduleLabels,result)
        elif isinstance(value, cms.VPSet):
            for ps in value:
                _hasInputTagModuleLabel(process, ps, moduleLabels,result)
        elif isinstance(value, cms.VInputTag):
            for t in value:
                t2 = t
                if not isinstance(t, cms.InputTag):
                    t2 = cms.InputTag(t2)
                for i,moduleLabel in enumerate(moduleLabels):
                    if result[i]: continue #no need
                    if t2.getModuleLabel() == moduleLabel:
                        result[i]=True
        elif isinstance(value, cms.InputTag):
            for i,moduleLabel in enumerate(moduleLabels):
                if result[i]: continue #no need
                if value.getModuleLabel() == moduleLabel:
                    result[i]=True
        elif isinstance(value, cms.string) and name == "refToPSet_":
            _hasInputTagModuleLabel(process, getattr(process, value.value()), moduleLabels,result)

def earlyDeleteSettings_cff.customiseEarlyDelete

(

process

)

Definition at line 36 of file earlyDeleteSettings_cff.py.

References _hasInputTagModuleLabel(), and list().

Referenced by ConfigBuilder.ConfigBuilder.prepare().

def customiseEarlyDelete(process):
    # Mapping label -> [branches]
    # for the producers whose products are to be deleted early
    products = collections.defaultdict(list)

    products = customiseEarlyDeleteForSeeding(process, products)

    products = customiseEarlyDeleteForCandIsoDeposits(process, products)

    # Set process.options.canDeleteEarly
    if not hasattr(process.options, "canDeleteEarly"):
        process.options.canDeleteEarly = cms.untracked.vstring()

    branchSet = set()
    for branches in products.itervalues():
        for branch in branches:
            branchSet.add(branch)
    process.options.canDeleteEarly.extend(list(branchSet))

    # LogErrorHarvester should not wait for deleted items
    for prod in process.producers_().itervalues():
        if prod.type_() == "LogErrorHarvester":
            if not hasattr(prod,'excludeModules'):
                prod.excludeModules = cms.untracked.vstring()
            t = prod.excludeModules.value()
            t.extend([b.split('_')[1] for b in branchSet])
            prod.excludeModules = t

    # Find the consumers
    producers=[]
    branchesList=[]
    for producer, branches in products.iteritems():
        producers.append(producer)
        branchesList.append(branches)

    for moduleType in [process.producers_(), process.filters_(), process.analyzers_()]:
        for name, module in moduleType.iteritems():
            result=[]
            for producer in producers:
                result.append(False)

            _hasInputTagModuleLabel(process, module, producers,result)
            for i in range(len(result)):
                if result[i]:
                    if hasattr(module, "mightGet"):
                        module.mightGet.extend(branchesList[i])
                    else:
                        module.mightGet = cms.untracked.vstring(branchesList[i])
    return process