test/private/testPATUserDataExternal.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 process = cms.Process("PAT")
0004
0005 # initialize MessageLogger and output report
0006 process.load("FWCore.MessageLogger.MessageLogger_cfi")
0007 process.MessageLogger.cerr.threshold = 'INFO'
0008 process.MessageLogger.cerr.INFO = cms.untracked.PSet(
0009     default          = cms.untracked.PSet( limit = cms.untracked.int32(0)  ),
0010     PATLayer0Summary = cms.untracked.PSet( limit = cms.untracked.int32(-1) )
0011 )
0012 process.options   = cms.untracked.PSet( wantSummary = cms.untracked.bool(True) )
0013
0014 # source
0015 process.source = cms.Source("PoolSource",
0016      fileNames = cms.untracked.vstring('file:/afs/cern.ch/cms/PRS/top/cmssw-data/relval200-for-pat-testing/FullSimTTBar-2_2_X_2008-11-03-STARTUP_V7-AODSIM.100.root')
0017 )
0018 process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(10) )
0019
0020 process.load("Configuration.StandardSequences.GeometryDB_cff")
0021 process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
0022 process.GlobalTag.globaltag = cms.string( autoCond[ 'phase1_2022_realistic' ] )
0023 process.load("Configuration.StandardSequences.MagneticField_cff")
0024
0025 # PAT Layer 0+1
0026 process.load("PhysicsTools.PatAlgos.patLayer0_cff")
0027 process.load("PhysicsTools.PatAlgos.patLayer1_cff")
0028
0029 process.answer = cms.EDProducer("PATUserDataTestModule", # each of this will produce all
0030     mode  = cms.string("external"),                      # but I don't care
0031     muons = cms.InputTag("allLayer0Muons"),
0032     label = cms.string("il"),                            # use instance label 'il' for the ints
0033 )
0034 process.pi = cms.EDProducer("PATUserDataTestModule",
0035     mode  = cms.string("external"),
0036     muons = cms.InputTag("allLayer0Muons"),
0037 )
0038 process.halfP4 = cms.EDProducer("PATUserDataTestModule",
0039     mode  = cms.string("external"),
0040     muons = cms.InputTag("allLayer0Muons"),
0041 )
0042 process.allLayer1Muons.userData.userInts.src    = cms.VInputTag(cms.InputTag("answer","il"))
0043 process.allLayer1Muons.userData.userFloats.src  = cms.VInputTag(cms.InputTag("pi"))
0044 process.allLayer1Muons.userData.userClasses.src = cms.VInputTag(cms.InputTag("halfP4"))
0045 process.testRead = cms.EDProducer("PATUserDataTestModule",
0046     mode  = cms.string("read"),
0047     muons = cms.InputTag("selectedLayer1Muons"),
0048 )
0049
0050
0051
0052 process.content = cms.EDAnalyzer("EventContentAnalyzer")
0053 process.p = cms.Path(
0054                 process.patLayer0
0055                 + process.answer * process.pi * process.halfP4
0056                 #+ process.content # uncomment to get a dump of the output after layer 0
0057                 + process.patLayer1
0058                 + process.testRead
0059             )
0060
0061 # Output module configuration
0062 process.out = cms.OutputModule("PoolOutputModule",
0063     fileName = cms.untracked.string('PATLayer1_Output.fromAOD_full.root'),
0064     # save only events passing the full path
0065     SelectEvents   = cms.untracked.PSet( SelectEvents = cms.vstring('p') ),
0066     outputCommands = cms.untracked.vstring('drop *')
0067 )
0068 process.outpath = cms.EndPath(process.out)
0069 # save PAT Layer 1 output
0070 process.load("PhysicsTools.PatAlgos.patLayer1_EventContent_cff")
0071 process.out.outputCommands.extend(process.patLayer1EventContent.outputCommands)
0072