RecoEgamma/test/ElectronMcSignalValidationMiniAOD_cfg.py

0001
0002 import sys
0003 import os
0004 import FWCore.ParameterSet.Config as cms
0005
0006 print('Number of arguments:', len(sys.argv), 'arguments.')
0007 print('Argument List:', str(sys.argv))
0008 # first arg : cmsRun
0009 # second arg : name of the _cfg file
0010 # third arg : sample name (ex. ZEE_14)
0011
0012 from electronValidationCheck_Env import env
0013
0014 cmsEnv = env() # be careful, cmsEnv != cmsenv. cmsEnv is local
0015
0016 cmsEnv.checkSample() # check the sample value
0017 cmsEnv.checkValues()
0018
0019 import DQMOffline.EGamma.electronDataDiscovery as dd
0020
0021 if cmsEnv.beginTag() == 'Run2_2017':
0022     from Configuration.Eras.Era_Run2_2017_cff import Run2_2017
0023     process = cms.Process("electronValidation",Run2_2017)
0024 elif cmsEnv.beginTag() == 'Run3':
0025     from Configuration.Eras.Era_Run3_cff import Run3
0026     process = cms.Process('electronValidation', Run3)
0027 else:
0028     from Configuration.Eras.Era_Phase2_cff import Phase2
0029     process = cms.Process('electronValidation',Phase2)
0030
0031 process.DQMStore = cms.Service("DQMStore")
0032 process.load("DQMServices.Components.DQMStoreStats_cfi")
0033 from DQMServices.Components.DQMStoreStats_cfi import *
0034
0035 dqmStoreStats.runOnEndJob = cms.untracked.bool(True)
0036
0037 print("reading files ...")
0038 max_number = -1 # number of events
0039 process.maxEvents = cms.untracked.PSet(input = cms.untracked.int32(max_number))
0040
0041 data = os.environ['data']
0042 flist = dd.getCMSdata(data)
0043 print(flist)
0044 process.source = cms.Source("PoolSource",
0045     #eventsToProcess = cms.untracked.VEventRange('1:38-1:40'),
0046     fileNames = cms.untracked.vstring(*flist))
0047
0048 #process.source = cms.Source ("PoolSource",
0049 #eventsToProcess = cms.untracked.VEventRange('1:2682-1:2682'),
0050 #eventsToProcess = cms.untracked.VEventRange('1:8259-1:8259'),
0051 #eventsToProcess = cms.untracked.VEventRange('1:10-1:10'),
0052 #fileNames = cms.untracked.vstring([
0053 #        'file:PAT_249120E2-D1EC-E611-83C2-0CC47A7C347A.root',
0054 #        'file:PAT_76F9AD07-D3EC-E611-AA87-0CC47A745250.root',
0055 #        'file:PAT_FA0E1D02-D5EC-E611-B8CA-0025905A6080.root',
0056 #        'file:PAT_EE728E01-D5EC-E611-9DC5-0025905A6126.root',
0057 #        '/store/relval/CMSSW_9_1_0_pre2/RelValSingleElectronPt10/MINIAODSIM/90X_upgrade2017_realistic_v20-v1/00000/66B5E60E-5619-E711-8E17-0CC47A4D7632.root',
0058 #        '/store/relval/CMSSW_9_1_0_pre2/RelValSingleElectronPt10/MINIAODSIM/90X_upgrade2017_realistic_v20-v1/00000/BC08FC0E-5619-E711-98E5-0CC47A4D760C.root',
0059 #    ]),
0060 #secondaryFileNames = cms.untracked.vstring() )
0061 #process.source.fileNames.extend(dd.search())
0062 print("reading files done")
0063
0064 process.load('Configuration.StandardSequences.Services_cff')
0065 process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
0066 process.load("FWCore.MessageService.MessageLogger_cfi")
0067 process.load('Configuration.EventContent.EventContent_cff')
0068 process.load('SimGeneral.MixingModule.mixNoPU_cfi')
0069 process.load('Configuration.StandardSequences.GeometryDB_cff')
0070 process.load('Configuration.StandardSequences.MagneticField_38T_cff')
0071 process.load('Configuration.StandardSequences.RawToDigi_cff')
0072 process.load('Configuration.StandardSequences.Reconstruction_cff')
0073 process.load('Configuration.StandardSequences.EndOfProcess_cff')
0074 process.load("Configuration.StandardSequences.EDMtoMEAtJobEnd_cff") # new
0075 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0076
0077 from Configuration.AlCa.autoCond import autoCond
0078 #process.GlobalTag.globaltag = os.environ['TEST_GLOBAL_TAG']#+'::All'
0079 #process.GlobalTag.globaltag = '122X_mcRun4_realistic_v1'
0080
0081 # FOR DATA REDONE FROM RAW, ONE MUST HIDE IsoFromDeps
0082 # CONFIGURATION
0083 process.load("Validation.RecoEgamma.electronIsoFromDeps_cff")
0084 process.load("Validation.RecoEgamma.ElectronIsolation_cfi")
0085 process.load("Validation.RecoEgamma.ElectronMcSignalValidatorMiniAOD_cfi")
0086
0087 print("miniAODElectronIsolation call")
0088 from Validation.RecoEgamma.electronValidationSequenceMiniAOD_cff import miniAODElectronIsolation # as _ElectronIsolationCITK
0089 process.miniAODElectronIsolation = miniAODElectronIsolation
0090 print("miniAODElectronIsolation clone done")
0091
0092 # load DQM
0093 process.load("DQMServices.Core.DQM_cfg")
0094 process.load("DQMServices.Components.DQMEnvironment_cfi")
0095
0096 #process.printContent = printContent
0097
0098 process.EDM = cms.OutputModule("PoolOutputModule",
0099 outputCommands = cms.untracked.vstring('drop *',"keep *_MEtoEDMConverter_*_*"),
0100 #fileName = cms.untracked.string(TEST_HISTOS_FILE)
0101 fileName = cms.untracked.string(os.environ['outputFile'])#.replace(".root", "_a.root"))
0102 )
0103
0104 process.electronMcSignalValidatorMiniAOD.InputFolderName = cms.string("EgammaV/ElectronMcSignalValidatorMiniAOD")
0105 process.electronMcSignalValidatorMiniAOD.OutputFolderName = cms.string("EgammaV/ElectronMcSignalValidatorMiniAOD")
0106
0107 #process.p = cms.Path(process.ElectronIsolation * process.electronMcSignalValidatorMiniAOD * process.MEtoEDMConverter ) # * process.dqmStoreStats
0108 #process.p = cms.Path(process.electronMcSignalValidatorMiniAOD * process.MEtoEDMConverter * process.dqmStoreStats)
0109 process.p = cms.Path( process.miniAODElectronIsolation * process.ElectronIsolation * process.electronMcSignalValidatorMiniAOD * process.MEtoEDMConverter ) #  process.printContent *
0110
0111 process.outpath = cms.EndPath(
0112 process.EDM,
0113 )