Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
import FWCore.ParameterSet.Config as cms

##### event content for heavy-ion analysis objects
from Configuration.EventContent.EventContentHeavyIons_cff import *
from RecoHI.HiMuonAlgos.RecoHiMuon_EventContent_cff import *

#jets
jetContent = cms.PSet(
    outputCommands = cms.untracked.vstring(
    'keep double*_*CaloJets_*_*',
    'drop recoCaloJets_*_*_*',
    'keep recoGenJets_*_*_*',
    'keep patJets_*_*_*'
    )
  )

jetContentExtended = jetContent.clone()
jetContentExtended.outputCommands.extend(cms.untracked.vstring('keep *_towerMaker_*_*'))

#tracks
trkContent = cms.PSet(
    outputCommands = cms.untracked.vstring(
      'keep *_offlineBeamSpot_*_*',
      'keep recoTracks_hiSelectedTracks_*_*',
      'keep recoTracks_hiPixel3PrimTracks_*_*' # low-fake selection to lower pt?
      )
    )

lightTrackContent = cms.PSet(
    outputCommands = cms.untracked.vstring(
    'keep *_offlineBeamSpot_*_*',
    'keep *_allTracks_*_*' # Selected Charged Candidates
    )
    )

#muons
muonContent = cms.PSet(
    outputCommands = cms.untracked.vstring(
      'keep patMuons_*_*_*'
      )
    )

muonContentExtended = muonContent.clone()
muonContentExtended.outputCommands.extend(RecoMuonAOD.outputCommands)
muonContentExtended.outputCommands.extend(RecoHiMuonAOD.outputCommands)

#photons
photonContent = cms.PSet(
    outputCommands = cms.untracked.vstring(
      'keep patPhotons_selected*_*_*'
      )
    )

photonContentExtended = photonContent.clone()
photonContentExtended.outputCommands.extend(RecoEgammaAOD.outputCommands)

#correlations
corrContent = cms.PSet(
    outputCommands = cms.untracked.vstring( 
      'keep recoRecoChargedCandidates_allTracks_*_*',
      'keep recoRecoChargedCandidates_allPxltracks_*_*'
      )
    )

#common
hiCommon = cms.PSet(
    outputCommands = cms.untracked.vstring('drop *',
      'keep *_TriggerResults_*_HLT',
      'keep L1GlobalTriggerReadoutRecord_gtDigis_*_*',
      'keep recoVertexs_hiSelectedVertex_*_*',                              
      'keep *_heavyIon_*_*',
      'keep *_hiCentrality_*_*',
      'keep *_hiEvtPlane_*_*'
    )
)

#mc content
hiContentMC =  cms.PSet(
    outputCommands = cms.untracked.vstring(
      'keep *_generator_*_*',
      'keep *_hiSignal_*_*',
      'keep *_genParticles_*_*',
      'keep *_hiGenParticles_*_*'
    )
)

##### combinations for specific skims

# HI PAG skim
hiAnalysisSkimContent = hiCommon.clone()
hiAnalysisSkimContent.outputCommands.extend(jetContentExtended.outputCommands)
hiAnalysisSkimContent.outputCommands.extend(trkContent.outputCommands)
hiAnalysisSkimContent.outputCommands.extend(muonContent.outputCommands)
hiAnalysisSkimContent.outputCommands.extend(photonContent.outputCommands)
hiAnalysisSkimContent.outputCommands.extend(corrContent.outputCommands)

# [highpt] skim
jetTrkSkimContent = hiCommon.clone()
jetTrkSkimContent.outputCommands.extend(jetContentExtended.outputCommands)
jetTrkSkimContent.outputCommands.extend(lightTrackContent.outputCommands)

# [highpt] MC skim
jetTrkSkimContentMC = jetTrkSkimContent.clone()
jetTrkSkimContentMC.outputCommands.extend(hiContentMC.outputCommands)

# [dilepton] skim 0
muonSkimContent = hiCommon.clone()
muonSkimContent.outputCommands.extend(cms.untracked.vstring('keep patMuons_*_*_*'))
muonSkimContent.outputCommands.extend(RecoMuonRECO.outputCommands)
muonSkimContent.outputCommands.extend(RecoHiTrackerRECO.outputCommands)

# [dilepton] skim MC
muonSkimContentMC = muonSkimContent.clone()
muonSkimContentMC.outputCommands.extend(hiContentMC.outputCommands)

# [dilepton] skim 1
muonTrkSkimContent = hiCommon.clone()  # trigger, L!, vertex,centrality, etc
muonTrkSkimContent.outputCommands.extend(muonContentExtended.outputCommands) # muon AOD
muonTrkSkimContent.outputCommands.extend(trkContent.outputCommands)