Performance/python/MinimizeSourceAndOutputTime.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 def customise_min_source_output(process, nEventsToCache=10):
0004     from IOPool.Input.modules import RepeatingCachedRootSource
0005     from FWCore.Modules.modules import AsciiOutputModule
0006
0007     process.source = RepeatingCachedRootSource(fileName = process.source.fileNames[0],
0008                                                repeatNEvents = nEventsToCache)
0009
0010     for k,v in process.outputModules_().items():
0011         if v.type_() == 'PoolOutputModule':
0012             setattr(process,k, AsciiOutputModule(verbosity= 0, outputCommands = v.outputCommands))
0013             if hasattr(k,'SelectEvents'):
0014                 getattr(process,k).SelectEvents = k.SelectEvents
0015
0016     #decrease messages as events are processed quickly
0017     process.MessageLogger.cerr.FwkReport.reportEvery = 100
0018
0019     #avoid warning messages each event if running on empty events
0020     if hasattr(process, 'manystripclus53X'):
0021         process.manystripclus53X.multiplicityConfig = dict(firstMultiplicityConfig=dict(warnIfModuleMissing=cms.untracked.bool(False)),
0022                                                            secondMultiplicityConfig=dict(warnIfModuleMissing=cms.untracked.bool(False)))
0023     if hasattr(process, 'toomanystripclus53X'):
0024         process.toomanystripclus53X.multiplicityConfig = dict(firstMultiplicityConfig=dict(warnIfModuleMissing=cms.untracked.bool(False)),
0025                                                               secondMultiplicityConfig=dict(warnIfModuleMissing=cms.untracked.bool(False)))
0026
0027     return process