File indexing completed on 2024-04-06 12:32:19
0001
0002
0003 import FWCore.ParameterSet.Config as cms
0004 from FWCore.ParameterSet.VarParsing import VarParsing
0005
0006 process = cms.Process("TestProcess")
0007
0008 process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")
0009 process.load("GeneratorInterface.Core.generatorSmeared_cfi")
0010 process.load("Configuration.EventContent.EventContent_cff")
0011 from Configuration.StandardSequences.VtxSmeared import VtxSmeared
0012 process.load(VtxSmeared['NoSmear'])
0013
0014 options = VarParsing('analysis')
0015
0016 options.register('nEvents',
0017 100000,
0018 VarParsing.multiplicity.singleton,
0019 VarParsing.varType.int,
0020 "Maximum number of events"
0021 )
0022 options.register('etaMin',
0023 -6.0,
0024 VarParsing.multiplicity.singleton,
0025 VarParsing.varType.float,
0026 "Minimum Eta for the neutrinos"
0027 )
0028 options.register('etaMax',
0029 6.0,
0030 VarParsing.multiplicity.singleton,
0031 VarParsing.varType.float,
0032 "Maximum Eta for the neutrinos"
0033 )
0034 options.register('antiPart',
0035 0,
0036 VarParsing.multiplicity.singleton,
0037 VarParsing.varType.int,
0038 "Choice of AntiParticle"
0039 )
0040
0041 options.parseArguments()
0042 print(options)
0043
0044 if (options.antiPart == 0):
0045 antiPart = False
0046 else:
0047 antiPart = True
0048 print("antiPart: ", antiPart)
0049
0050 process.maxEvents = cms.untracked.PSet(
0051 input = cms.untracked.int32(options.nEvents)
0052 )
0053
0054 process.load("IOMC.RandomEngine.IOMC_cff")
0055 process.RandomNumberGeneratorService.generator.initialSeed = 456789
0056 process.RandomNumberGeneratorService.VtxSmeared.engineName = cms.untracked.string('HepJamesRandom')
0057 process.RandomNumberGeneratorService.VtxSmeared.initialSeed = cms.untracked.uint32(98765432)
0058
0059 process.source = cms.Source("EmptySource",
0060 firstRun = cms.untracked.uint32(1),
0061 firstEvent = cms.untracked.uint32(1)
0062 )
0063
0064 process.generator = cms.EDProducer("FlatRandomEGunProducer",
0065 PGunParameters = cms.PSet(
0066 PartID = cms.vint32(14),
0067 MinEta = cms.double(options.etaMin),
0068 MaxEta = cms.double(options.etaMax),
0069 MinPhi = cms.double(-3.14159265359),
0070 MaxPhi = cms.double(3.14159265359),
0071 MinE = cms.double(10.0),
0072 MaxE = cms.double(10.0)
0073 ),
0074 AddAntiParticle = cms.bool(antiPart),
0075 Verbosity = cms.untracked.int32(0)
0076 )
0077
0078 process.o1 = cms.OutputModule("PoolOutputModule",
0079 process.FEVTSIMEventContent,
0080 fileName = cms.untracked.string('single_neutrino_random.root')
0081 )
0082
0083 process.p1 = cms.Path(process.generator*process.VtxSmeared*process.generatorSmeared)
0084 process.outpath = cms.EndPath(process.o1)