Geometry/test/single_neutrino_cfg.py

0001 # cmsRun single_neutrino_cfg.py nEvents=100000 etaMin=-6.0 etaMax=6.0 antiPart=0
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)