RecoUtils/test/pf_pu_assomap_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 process = cms.Process("OWNPARTICLES")
0004
0005 process.load("FWCore.MessageService.MessageLogger_cfi")
0006
0007 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(100) )
0008
0009 process.source = cms.Source("PoolSource",
0010     fileNames = cms.untracked.vstring('file:/user/geisler/RelValZmumuJets_Pt_20_300_PU_START53_V6-v1_GEN-SIM-RECO.root'),
0011     secondaryFileNames =  cms.untracked.vstring('file:/user/geisler/RelValZmumuJets_Pt_20_300_PU_START53_V6-v1_GEN-SIM-DIGI-RAW-HLTDEBUG.root'),
0012 )
0013
0014 #process.source = cms.Source("PoolSource",
0015     #fileNames = cms.untracked.vstring('file:/user/geisler/QCD_Pt-15to3000_Tune2C_Flat_8TeV_pythia8_AODSIM.root'),
0016 #)
0017
0018 ### conditions
0019 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0020 process.GlobalTag.globaltag = 'START53_V11::All'
0021
0022 ### standard includes
0023 process.load('Configuration.StandardSequences.Services_cff')
0024 process.load('Configuration.Geometry.GeometryPilot2_cff')
0025 process.load("Configuration.StandardSequences.RawToDigi_cff")
0026 process.load("Configuration.EventContent.EventContent_cff")
0027 process.load("Configuration.StandardSequences.Reconstruction_cff")
0028 process.load("Configuration.StandardSequences.MagneticField_cff")
0029
0030 process.selectedPrimaryVertexQuality = cms.EDFilter("VertexSelector",
0031     src = cms.InputTag('offlinePrimaryVertices'),
0032     cut = cms.string("isValid & ndof >= 4 & chi2 > 0 & tracksSize > 0 & abs(z) < 24 & abs(position.Rho) < 2."),
0033     filter = cms.bool(True),
0034 )
0035
0036 ### AssociationMap-specific includes
0037 from CommonTools.RecoUtils.pf_pu_assomap_cfi import AssociationMaps
0038
0039 process.assMap = AssociationMaps.clone(
0040           VertexCollection = cms.InputTag('selectedPrimaryVertexQuality'),
0041 )
0042
0043 ### FirstVertexTracks-specific includes
0044 from CommonTools.RecoUtils.pf_pu_firstvertextracks_cfi import FirstVertexTracks
0045
0046 process.firstVertexTracks = FirstVertexTracks.clone(
0047           AssociationMap = cms.InputTag('assMap'),
0048           VertexCollection = cms.InputTag('selectedPrimaryVertexQuality'),
0049 )
0050
0051
0052
0053 process.p = cms.Path(
0054     process.selectedPrimaryVertexQuality *
0055     process.assMap *
0056     process.firstVertexTracks
0057 )
0058
0059 process.output = cms.OutputModule("PoolOutputModule",
0060     fileName = cms.untracked.string("myOutput.root"),
0061     outputCommands = cms.untracked.vstring('drop *',
0062           'keep *_*_*_RECO'),
0063 )
0064
0065 process.out_step = cms.EndPath(process.output)
0066