HydjetInterface/test/testHydjet.py

0001 import FWCore.ParameterSet.Config as cms
0002 import FWCore.ParameterSet.VarParsing as VarParsing
0003 import os
0004
0005 hjenergy = os.getenv("HJENERGY", "0")
0006
0007 if hjenergy in "0":
0008     options = VarParsing.VarParsing("analysis")
0009     options.register("hjenergy", "999", VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.string, "Beam energy")
0010     options.parseArguments()
0011     hjenergy = options.hjenergy
0012
0013 if hjenergy in "999":
0014        raise RuntimeError("Stopping cmsRun testHydjet.py: this macro needs hjenergy=5362 command line parameter")
0015
0016 process = cms.Process("ANA")
0017
0018 process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")
0019 process.load("Configuration.StandardSequences.Services_cff")
0020 process.load("GeneratorInterface.HydjetInterface.hydjetDefault_cfi")
0021
0022 process.RandomNumberGeneratorService = cms.Service("RandomNumberGeneratorService",
0023     generator = cms.PSet(
0024         initialSeed = cms.untracked.uint32(123456789),
0025         engineName = cms.untracked.string('HepJamesRandom')
0026     )
0027 )
0028
0029 process.maxEvents = cms.untracked.PSet(
0030     input = cms.untracked.int32(-1)
0031     )
0032
0033 process.ana = cms.EDAnalyzer('HydjetAnalyzer',
0034
0035     doHistos        = cms.untracked.bool(True),
0036     userHistos      = cms.untracked.bool(False),
0037
0038     # Settings for USER histos
0039
0040     uStatus     = cms.untracked.int32(2),   # 1 - it's 1,2,3,4,5 of Pythia status; 2 - 11,12,13,14,15; 3 - All
0041     uPDG_1      = cms.untracked.int32(443),
0042     uPDG_2      = cms.untracked.int32(-443),
0043
0044     ### Eta cut for pT dep. dist.
0045     uPTetaCut   = cms.untracked.double(4.),
0046     dPTetaCut   = cms.untracked.double(2.5),
0047
0048     ### Vectors of bins borders (when 0 -  uniform bins would be used)
0049     PtBins      = cms.untracked.vdouble(0.,1.,2.,3.,4.,5.,6.,8.,12.,16.,20.),
0050     EtaBins         = cms.untracked.vdouble(0.),
0051     PhiBins         = cms.untracked.vdouble(0.),
0052     v2EtaBins       = cms.untracked.vdouble(0.),
0053     v2PtBins        = cms.untracked.vdouble(0.,1.,2.,3.,4.,6.,8.,12.,16.,20.),
0054
0055     ### Settings for uniform bins
0056     nintPt      = cms.untracked.int32(1000),
0057     nintEta     = cms.untracked.int32(100),
0058     nintPhi     = cms.untracked.int32(100),
0059     nintV2pt        = cms.untracked.int32(100),
0060     nintV2eta       = cms.untracked.int32(100),
0061
0062     minPt       = cms.untracked.double(0.),
0063     minEta      = cms.untracked.double(-10.),
0064     minPhi      = cms.untracked.double(-3.14159265358979),
0065     minV2pt     = cms.untracked.double(0.),
0066     minV2eta        = cms.untracked.double(-10.),
0067
0068     maxPt       = cms.untracked.double(100.),
0069     maxEta      = cms.untracked.double(10.),
0070     maxPhi      = cms.untracked.double(3.14159265358979),
0071     maxV2pt     = cms.untracked.double(10.),
0072     maxV2eta        = cms.untracked.double(10.),
0073
0074 )
0075
0076
0077 #process.generator.signalVtx = cms.untracked.vdouble(0.,0.,0.,0.) # Signal event vertex option, to set it by hand (instead of smearing)
0078
0079 process.TFileService = cms.Service('TFileService',
0080     fileName = cms.string('Hydjet.root')
0081 )
0082
0083 process.p = cms.Path(process.generator*process.ana)
0084
0085