File indexing completed on 2023-03-17 11:03:07
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 SwitchProducer in Task.')
0007 parser.add_argument("--disableTest2", help="Disable test2 SwitchProducer case", action="store_true")
0008 parser.add_argument("--input", help="Read input file from a previous step of this same configuration", action="store_true")
0009 parser.add_argument("--conditionalTask", help="Use ConditionalTask instead of Task", action="store_true")
0010
0011 argv = sys.argv[:]
0012 if '--' in argv:
0013 argv.remove("--")
0014 args, unknown = parser.parse_known_args(argv)
0015
0016 enableTest2 = not args.disableTest2
0017 class SwitchProducerTest(cms.SwitchProducer):
0018 def __init__(self, **kargs):
0019 super(SwitchProducerTest,self).__init__(
0020 dict(
0021 test1 = lambda accelerators: (True, -10),
0022 test2 = lambda accelerators: (enableTest2, -9)
0023 ), **kargs)
0024
0025 process = cms.Process("PROD2" if args.input else "PROD1")
0026
0027 if args.input:
0028
0029
0030 process.source = cms.Source("PoolSource",
0031 fileNames = cms.untracked.vstring("file:testSwitchProducer{}Task{}.root".format(
0032 "Conditional" if args.conditionalTask else "",
0033 1 if enableTest2 else 2,
0034 ))
0035 )
0036 process.maxEvents.input = -1
0037 else:
0038 process.source = cms.Source("EmptySource")
0039 process.maxEvents.input = 3
0040 if enableTest2:
0041 process.source.firstLuminosityBlock = cms.untracked.uint32(2)
0042
0043 process.out = cms.OutputModule("PoolOutputModule",
0044 fileName = cms.untracked.string('testSwitchProducer{}Task{}{}.root'.format(
0045 "Conditional" if args.conditionalTask else "",
0046 "Input" if args.input else "",
0047 1 if enableTest2 else 2,
0048 )),
0049 outputCommands = cms.untracked.vstring(
0050 'keep *_intProducer_*_*',
0051 'keep *_intProducerOther_*_*',
0052 'keep *_intProducerAlias_*_*',
0053 'keep *_intProducerAlias2_other_*',
0054 'keep *_intProducerDep1_*_*',
0055 'keep *_intProducerDep2_*_*',
0056 'keep *_intProducerDep3_*_*',
0057 )
0058 )
0059
0060 process.intProducer1 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(1))
0061 process.intProducer2 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(2))
0062 process.intProducer3 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(2), values = cms.VPSet(cms.PSet(instance=cms.string("foo"),value=cms.int32(2))))
0063 process.intProducer4 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(42), throw = cms.untracked.bool(True))
0064 process.intProducer5 = cms.EDProducer("ManyIntProducer", ivalue = cms.int32(3), values = cms.VPSet(cms.PSet(instance=cms.string("foo"),value=cms.int32(3))))
0065 if enableTest2:
0066 process.intProducer1.throw = cms.untracked.bool(True)
0067 else:
0068 process.intProducer2.throw = cms.untracked.bool(True)
0069 process.intProducer3.throw = cms.untracked.bool(True)
0070 process.intProducer5.throw = cms.untracked.bool(True)
0071
0072 process.intProducer = SwitchProducerTest(
0073 test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0074 test2 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer2"))
0075 )
0076
0077 process.intProducerOther = SwitchProducerTest(
0078 test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0079 test2 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer2"))
0080 )
0081
0082 process.intProducerAlias = SwitchProducerTest(
0083 test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0084 test2 = cms.EDAlias(intProducer3 = cms.VPSet(cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string(""), toProductInstance = cms.string("")),
0085 cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string("foo"), toProductInstance = cms.string("other"))))
0086 )
0087
0088 process.intProducerAlias2 = SwitchProducerTest(
0089 test1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer1")),
0090 test2 = cms.EDAlias(intProducer4 = cms.VPSet(cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string(""), toProductInstance = cms.string(""))),
0091 intProducer5 = cms.VPSet(cms.PSet(type = cms.string("edmtestIntProduct"), fromProductInstance = cms.string("foo"), toProductInstance = cms.string("other"))))
0092 )
0093
0094
0095 process.intProducerDep1 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer"))
0096 process.intProducerDep2 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer"))
0097 process.intProducerDep3 = cms.EDProducer("AddIntsProducer", labels = cms.VInputTag("intProducer"))
0098
0099 if args.conditionalTask:
0100 process.ct = cms.ConditionalTask(process.intProducer, process.intProducerOther, process.intProducerAlias, process.intProducerAlias2,
0101 process.intProducer1, process.intProducer2, process.intProducer3, process.intProducer4, process.intProducer5)
0102 process.p = cms.Path(process.intProducerDep1+process.intProducerDep2+process.intProducerDep3, process.ct)
0103 else:
0104 process.t = cms.Task(process.intProducer, process.intProducerOther, process.intProducerAlias, process.intProducerAlias2,
0105 process.intProducerDep1, process.intProducerDep2, process.intProducerDep3,
0106 process.intProducer1, process.intProducer2, process.intProducer3, process.intProducer4, process.intProducer5)
0107 process.p = cms.Path(process.t)
0108
0109 process.e = cms.EndPath(process.out)