test/python/protoValid_cfg.py

0001 ###############################################################################
0002 # Way to use this:
0003 #   cmsRun protoValid_cfg.py geometry=D110 type=hgcalSimHitStudy defaultInput=1
0004 #
0005 #   Options for geometry D98, D99, D103, D104, D105, D106, D107, D108, D109
0006 #                        D110, D111, D112, D113, D114, D115
0007 #               type hgcalGeomCheck, hgcalSimHitStudy, hgcalDigiStudy,
0008 #                    hgcalRecHitStudy, hgcalSiliconValidation
0009 #               defaultInput 1, 0
0010 #
0011 ###############################################################################
0012 import FWCore.ParameterSet.Config as cms
0013 import os, sys, imp, re
0014 import FWCore.ParameterSet.VarParsing as VarParsing
0015
0016 ############################################################
0017 ### SETUP OPTIONS
0018 options = VarParsing.VarParsing('standard')
0019 options.register('geometry',
0020                  "D110",
0021                   VarParsing.VarParsing.multiplicity.singleton,
0022                   VarParsing.VarParsing.varType.string,
0023                   "geometry of operations: D98, D99, D103, D104, D105, D106, D107, D108, D109, D110, D111, D112, D113, D114, D115")
0024 options.register('type',
0025                  "hgcalGeomCheck",
0026                  VarParsing.VarParsing.multiplicity.singleton,
0027                  VarParsing.VarParsing.varType.string,
0028                  "type of operations: hgcalGeomCheck, hgcalSimHitStudy, hgcalDigiStudy, hgcalRecHitStudy, hgcalSiliconValidation")
0029 options.register('defaultInput',
0030                  1, # default Value = true
0031                  VarParsing.VarParsing.multiplicity.singleton,
0032                  VarParsing.VarParsing.varType.int,
0033                  "change files path in case of defaultInput=0 for using DIGI o/p")
0034
0035 ### get and parse the command line arguments
0036 options.parseArguments()
0037
0038 print(options)
0039
0040 ############################################################
0041 # Use the options
0042
0043 geomName = "Run4" + options.geometry
0044 print("Geometry Name:   ", geomName)
0045 import Configuration.Geometry.defaultPhase2ConditionsEra_cff as _settings
0046 GLOBAL_TAG, ERA = _settings.get_era_and_conditions(geomName)
0047 print("Global Tag Name: ", GLOBAL_TAG)
0048 print("Era Name:        ", ERA)
0049
0050 process = cms.Process('Valid',ERA)
0051
0052 geomFile = "Configuration.Geometry.GeometryExtendedRun4" + options.geometry + "Reco_cff"
0053 fileCheck = "testHGCalSimWatcher" + options.geometry + ".root"
0054 if (options.type == "hgcalSimHitStudy"):
0055     fileName = "hgcSimHit" + options.geometry + ".root"
0056 elif (options.type == "hgcalDigiStudy"):
0057     fileName = "hgcDigi" + options.geometry + ".root"
0058 elif (options.type == "hgcalRecHitStudy"):
0059     fileName = "hgcRecHit" + options.geometry + ".root"
0060 elif (options.type == "hgcalSiliconValidation"):
0061     if (options.defaultInput == 0):
0062         fileName = "hgcDigValid" + options.geometry + ".root"
0063     else:
0064         fileName = "hgcSilValid" + options.geometry + ".root"
0065 else:
0066     fileName = "hgcGeomCheck" + options.geometry + ".root"
0067
0068 print("Geometry file: ", geomFile)
0069 print("Output file:   ", fileName)
0070
0071 process.load(geomFile)
0072 process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")
0073 process.load("Configuration.StandardSequences.MagneticField_cff")
0074 process.load('Configuration.StandardSequences.Services_cff')
0075 process.load("Configuration.EventContent.EventContent_cff")
0076 process.load('FWCore.MessageService.MessageLogger_cfi')
0077 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0078 from Configuration.AlCa.GlobalTag import GlobalTag
0079 process.GlobalTag = GlobalTag(process.GlobalTag, GLOBAL_TAG, '')
0080
0081 process.MessageLogger.cerr.FwkReport.reportEvery = 5
0082 if hasattr(process,'MessageLogger'):
0083     process.MessageLogger.HGCalGeom=dict()
0084
0085 if (options.type == "hgcalSimHitStudy"):
0086     process.load('Validation.HGCalValidation.hgcSimHitStudy_cfi')
0087     process.source = cms.Source("PoolSource",
0088                                 fileNames = cms.untracked.vstring('file:step1.root') )
0089     process.analysis_step = cms.Path(process.hgcalSimHitStudy)
0090 elif (options.type == "hgcalDigiStudy"):
0091     process.load('Configuration.StandardSequences.RawToDigi_cff')
0092     process.load('Validation.HGCalValidation.hgcDigiStudy_cfi')
0093     process.source = cms.Source("PoolSource",
0094                                 fileNames = cms.untracked.vstring('file:step2.root') )
0095     process.analysis_step = cms.Path(process.RawToDigi+process.hgcalDigiStudyEE+process.hgcalDigiStudyHEF+process.hgcalDigiStudyHEB)
0096 elif (options.type == "hgcalRecHitStudy"):
0097     process.load('Validation.HGCalValidation.hgcalRecHitStudy_cff')
0098     process.source = cms.Source("PoolSource",
0099                                 fileNames = cms.untracked.vstring('file:step3.root') )
0100     process.analysis_step = cms.Path(process.hgcalRecHitStudyEE+process.hgcalRecHitStudyFH+process.hgcalRecHitStudyBH)
0101 elif (options.type == "hgcalSiliconValidation"):
0102     if (options.defaultInput == 0):
0103         fileIn = "file:step2.root"
0104     else:
0105         fileIn = "file:step1.root"
0106     process.load('Validation.HGCalValidation.hgcalSiliconValidation_cfi')
0107     process.source = cms.Source("PoolSource",
0108                                 fileNames = cms.untracked.vstring(fileIn) )
0109     process.analysis_step = cms.Path(process.hgcalSiliconAnalysisEE+process.hgcalSiliconAnalysisHEF)
0110 else:
0111     process.load('Validation.HGCalValidation.hgcGeomCheck_cff')
0112     process.source = cms.Source("PoolSource",
0113                                 fileNames = cms.untracked.vstring(fileCheck)
0114     )
0115     process.analysis_step = cms.Path(process.hgcGeomCheck)
0116
0117
0118 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(options.maxEvents) )
0119 process.TFileService = cms.Service("TFileService",
0120                                    fileName = cms.string(fileName),
0121                                    closeFileFast = cms.untracked.bool(True) )
0122 process.options = cms.untracked.PSet(
0123     wantSummary = cms.untracked.bool(True)
0124 )
0125
0126 # Schedule definition
0127 process.schedule = cms.Schedule(process.analysis_step)