Reference Manual

Classes
class	TestHasInputTagModuleLabel

Functions
def	_hasInputTagModuleLabel

def	customiseEarlyDelete

Function Documentation

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

private

Definition at line 12 of file earlyDeleteSettings_cff.py.

Referenced by customiseEarlyDelete().

 
 def _hasInputTagModuleLabel(process, pset, psetModLabel, moduleLabels, result):
     for name in pset.parameterNames_():
         value = getattr(pset,name)
         if isinstance(value, cms.PSet):
             _hasInputTagModuleLabel(process, value, psetModLabel, moduleLabels, result)
         elif isinstance(value, cms.VPSet):
             for ps in value:
                 _hasInputTagModuleLabel(process, ps, psetModLabel, 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_":
             try:
                 ps = getattr(process, value.value())
             except AttributeError:
                 raise RuntimeError("Module %s has a 'PSet(refToPSet_ = cms.string(\"%s\"))', but the referenced-to PSet does not exist in the Process." % (psetModLabel, value.value()))
             _hasInputTagModuleLabel(process, ps, psetModLabel, moduleLabels, result)
 

earlyDeleteSettings_cff._hasInputTagModuleLabel

def _hasInputTagModuleLabel

Definition: earlyDeleteSettings_cff.py:12

def earlyDeleteSettings_cff.customiseEarlyDelete ( process )

Definition at line 42 of file earlyDeleteSettings_cff.py.

References _hasInputTagModuleLabel(), sistrip::SpyUtilities.range(), and makeHLTPrescaleTable.values.

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 = customiseEarlyDeleteForMkFit(process, products)
     products = customiseEarlyDeleteForCKF(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.values():
         for branch in branches:
             branchSet.add(branch)
     branchList = sorted(branchSet)
     process.options.canDeleteEarly.extend(branchList)
 
     # LogErrorHarvester should not wait for deleted items
     for prod in process.producers_().values():
         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 branchList])
             prod.excludeModules = t
 
     # Find the consumers
     producers=[]
     branchesList=[]
     for producer, branches in products.items():
         producers.append(producer)
         branchesList.append(branches)
 
     for moduleType in [process.producers_(), process.filters_(), process.analyzers_()]:
         for name, module in moduleType.items():
             result=[]
             for producer in producers:
                 result.append(False)
 
             _hasInputTagModuleLabel(process, module, name, producers, result)
             for i in range(len(result)):
                 if result[i]:
                     #if it exists it might be optional or empty, both evaluate to False
                     if hasattr(module, "mightGet") and module.mightGet:
                         module.mightGet.extend(branchesList[i])
                     else:
                         module.mightGet = cms.untracked.vstring(branchesList[i])
     return process
 