AllocMonitor/test/moduleAlloc_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 import argparse
0004 import sys
0005
0006 parser = argparse.ArgumentParser(prog=sys.argv[0], description='Test ModuleAllocMonitor service.')
0007 parser.add_argument("--skipEvents", action="store_true", help="test skipping events")
0008 parser.add_argument("--edmodule", action="store_true", help="show only specific ed module")
0009 parser.add_argument("--esmodule", action="store_true", help="show only specific es module")
0010 args = parser.parse_args()
0011
0012
0013 process = cms.Process("TEST")
0014
0015 process.source = cms.Source("EmptySource")
0016
0017 process.maxEvents.input = 3
0018
0019 process.WhatsItESProducer = cms.ESProducer("WhatsItESProducer")
0020
0021 process.DoodadESSource = cms.ESSource("DoodadESSource")
0022
0023 process.Thing = cms.EDProducer("ThingProducer",
0024     offsetDelta = cms.int32(1)
0025 )
0026
0027 process.OtherThing = cms.EDProducer("OtherThingProducer")
0028
0029 process.thingProducer = cms.EDProducer("ThingProducer",
0030                                        offsetDelta = cms.int32(100),
0031                                        nThings = cms.int32(50)
0032 )
0033
0034 process.get = cms.EDAnalyzer("WhatsItAnalyzer")
0035 process.out = cms.OutputModule("AsciiOutputModule")
0036
0037
0038 process.ep = cms.EndPath(process.out+process.get, cms.Task(process.WhatsItESProducer, process.DoodadESSource, process.Thing, process.OtherThing, process.thingProducer))
0039
0040 #process.add_(cms.Service("Tracer"))
0041 process.add_(cms.Service("ModuleAllocMonitor", fileName = cms.untracked.string("moduleAlloc.log")))
0042
0043 if args.skipEvents:
0044     process.ModuleAllocMonitor.nEventsToSkip = cms.untracked.uint32(2)
0045
0046 if args.edmodule:
0047     process.ModuleAllocMonitor.moduleNames = cms.untracked.vstring(["thingProducer"])
0048
0049 if args.esmodule:
0050     process.ModuleAllocMonitor.moduleNames = cms.untracked.vstring(["WhatsItESProducer"])