testHGCalCellHitSum_cfg.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms
import os, sys, imp, re
import FWCore.ParameterSet.VarParsing as VarParsing


options = VarParsing.VarParsing('standard')
options.register('geometry',
                 "D110",
                  VarParsing.VarParsing.multiplicity.singleton,
                  VarParsing.VarParsing.varType.string,
                  "geometry of operations: D105, D110, D114")

options.register('layers',
                 "1",
                  VarParsing.VarParsing.multiplicity.singleton,
                  VarParsing.VarParsing.varType.string,
                  "For single layer use 'layers=3' (default is 'layers=1'); or for multiple layers use 'layers=1,27,41,46'; or for all layers use 'layers=1-47'. Note that the size may increase by ~10 times in memory usage and ~50 times in file volume if 'all layers' option is applied.")

options.register('type',
                 "mu",
                 VarParsing.VarParsing.multiplicity.singleton,
                 VarParsing.VarParsing.varType.string,
                 "type of operations: mu, tt")


options.parseArguments()

import FWCore.ParameterSet.Config as cms
from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
process = cms.Process('HGCalCellHit',Phase2C17I13M9)

print(options)



loadFile = "Configuration.Geometry.GeometryExtendedRun4" + options.geometry + "Reco_cff"
inputFile = "file:step1" + options.geometry + options.type + ".root"
outputFile = "file:geantoutput" + options.geometry + options.type + ".root"

if (options.geometry == "D105"):
    geomFile = 'Validation/HGCalValidation/data/wafer_v16.csv'
elif (options.geometry == "D114"):
    geomFile = 'Validation/HGCalValidation/data/wafer_v17.csv'
else:
    geomFile = 'Validation/HGCalValidation/data/wafer_v17.csv'

print("Geometry file: ", loadFile)
print("Wafer file:    ", geomFile)
print("Input file:    ", inputFile)
print("Output file:   ", outputFile)

process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
process.load(loadFile)
process.load('FWCore.MessageService.MessageLogger_cfi')
process.MessageLogger.cerr.FwkReport.reportEvery = 100
if hasattr(process,'MessageLogger'):
    process.MessageLogger.ValidHGCal=dict()

process.load("IOMC.RandomEngine.IOMC_cff")
process.RandomNumberGeneratorService.generator.initialSeed = 456789

process.source = cms.Source("PoolSource",
    fileNames = cms.untracked.vstring(inputFile)
)

process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(-1))

process.load('Validation.HGCalValidation.hgcalCellHitSum_cff')

process.hgcalCellHitSumEE.geometryFileName = geomFile
process.hgcalCellHitSumHEF.geometryFileName = geomFile
process.hgcalCellHitSumHEB.geometryFileName = geomFile

process.TFileService = cms.Service("TFileService",
                                   fileName = cms.string(outputFile) )

process.p = cms.Path(process.hgcalCellHitSumEE*process.hgcalCellHitSumHEF*process.hgcalCellHitSumHEB)