DQM/TrackingMonitorSource/python/TrackingDataMCValidation_Standalone_cff.py - master - githubmirror/cmssw

DQM/TrackingMonitorSource/python/TrackingDataMCValidation_Standalone_cff.py
316 строк12 KB
mmusich
make V0EventSelector produce skimmed V0 collection based on their mass
14 сен 2024, 15:07
14 сен 2024, 15:0743db8e1
import FWCore.ParameterSet.Config as cms
from DQM.TrackingMonitorSource.StandaloneTrackMonitor_cfi import *
from DQM.TrackingMonitorSource.ZEEDetails_cfi import *
from DQM.TrackingMonitorSource.V0Selections_cfi import *
from DQM.TrackingMonitor.V0Monitor_cfi import *

# Primary Vertex Selector
selectedPrimaryVertices = cms.EDFilter("VertexSelector",
                                       src = cms.InputTag('offlinePrimaryVertices'),
                                       # cut = cms.string("!isFake && ndof >= 4 && abs(z) < 24 && abs(position.Rho) < 2.0"),
                                       cut = cms.string(""),
                                       filter = cms.bool(True)
                                       )
# Track Selector
selectedTracks = cms.EDFilter("TrackSelector",
                              src = cms.InputTag('generalTracks'),
                              cut = cms.string("pt > 1.0"),
                              #cut = cms.string(""),
                              filter = cms.bool(True)
                              )

# Track Multiplicity Selector
selectedMultiplicityTracks = cms.EDFilter("TrackMultiplicityFilter",
                                          src = cms.InputTag('generalTracks'),
                                          #cut = cms.string("pt > 1.0"),
                                          nmin = cms.untracked.uint32(500),
                                          cut = cms.untracked.string(""),
                                          filter = cms.bool(True)
                                      )

# Track ALCARECO Selection for zerobias
selectedAlcaRecoZBTracks = cms.EDProducer("AlcaRecoTrackSelector",
                                        src = cms.InputTag('generalTracks'),
                                        #cut = cms.string("pt > 0.65 && abs(eta) < 3.5 && p > 1.5 && hitPattern.numberOfAllHits('TRACK_HITS') > 7"),
                                        #cut = cms.string(""),
                                        ptmin = cms.untracked.double(0.65),
                                        pmin = cms.untracked.double(1.5),
                                        etamin = cms.untracked.double(-3.5),
                                        etamax = cms.untracked.double(3.5),
                                        nhits = cms.untracked.uint32(7)
)
'''
# Track ALCARECO Selection for singlemuon
selectedAlcaRecoSMTracks = cms.EDFilter("TrackSelector",
                              src = cms.InputTag('selectedMultiplicityTracks'),
                              cut = cms.string("pt > 2.0 && abs(eta) < 3.5 && p > 1.5 && hitPattern.numberOfAllTrackerHits > 7"),
                              #cut = cms.string(""),
                              filter = cms.bool(True)
                              )
'''
# HLT path selector
hltPathFilter = cms.EDFilter("HLTPathSelector",
                             processName = cms.string("HLT"),
                             verbose = cms.untracked.bool(False),
                             hltPathsOfInterest = cms.vstring("HLT_ZeroBias_v"),
                             triggerResults = cms.untracked.InputTag("TriggerResults","","HLT"),
                             triggerEvent = cms.untracked.InputTag("hltTriggerSummaryAOD","","HLT")
                             )

# HLT path selector Muon
hltPathFilterMuon = cms.EDFilter("HLTPathSelector",
                             processName = cms.string("HLT"),
                             verbose = cms.untracked.bool(False),
                             hltPathsOfInterest = cms.vstring("HLT_IsoMu24_v"),
                             triggerResults = cms.untracked.InputTag("TriggerResults","","HLT"),
                             triggerEvent = cms.untracked.InputTag("hltTriggerSummaryAOD","","HLT")
                             )

# HLT path selector Electron
hltPathFilterElectron = cms.EDFilter("HLTPathSelector",
                             processName = cms.string("HLT"),
                             verbose = cms.untracked.bool(False),
                             hltPathsOfInterest = cms.vstring("HLT_Ele32_WPTight_Gsf_v"),
                             triggerResults = cms.untracked.InputTag("TriggerResults","","HLT"),
                             triggerEvent = cms.untracked.InputTag("hltTriggerSummaryAOD","","HLT")
                             )

# HLT path selector ttbar
hltPathFilterTtbar = cms.EDFilter("HLTPathSelector",
                             processName = cms.string("HLT"),
                             verbose = cms.untracked.bool(False),
                             hltPathsOfInterest = cms.vstring("HLT_Ele32_WPTight_Gsf_v","HLT_IsoMu24_v"),
                             triggerResults = cms.untracked.InputTag("TriggerResults","","HLT"),
                             triggerEvent = cms.untracked.InputTag("hltTriggerSummaryAOD","","HLT")
                             )

# Z->MuMu event selector
ztoMMEventSelector = cms.EDFilter("ZtoMMEventSelector")
muonTracks = cms.EDProducer("ZtoMMMuonTrackProducer")
# Z->ee event selector
ztoEEEventSelector = cms.EDFilter("ZtoEEEventSelector")
electronTracks = cms.EDProducer("ZtoEEElectronTrackProducer")
#ttbar event selector
ttbarEventSelector = cms.EDFilter("ttbarEventSelector")
ttbarTracks = cms.EDProducer("TtbarTrackProducer")

# Added modules for V0Monitoring (standard)
KshortMonitor = v0Monitor.clone()
KshortMonitor.FolderName = "StandaloneTrackMonitor/V0Monitoring/Ks"
KshortMonitor.v0         = "generalV0Candidates:Kshort"
KshortMonitor.histoPSet.massPSet = cms.PSet(nbins = cms.int32 (100),
                                            xmin  = cms.double(0.400),
                                            xmax  = cms.double(0.600))

LambdaMonitor = v0Monitor.clone()
LambdaMonitor.FolderName = "StandaloneTrackMonitor/V0Monitoring/Lambda"
LambdaMonitor.v0 = "generalV0Candidates:Lambda"
LambdaMonitor.histoPSet.massPSet = cms.PSet(nbins = cms.int32(100),
                                            xmin  = cms.double(1.050),
                                            xmax  = cms.double(1.250))

# Added modules for V0Monitoring (for restricted mass candidates)
SelectedKshortMonitor = KshortMonitor.clone(
    FolderName = "StandaloneTrackMonitor/V0Monitoring/SelectedKs",
    v0         = "KShortEventSelector"
)

SelectedLambdaMonitor = LambdaMonitor.clone(
    FolderName = "StandaloneTrackMonitor/V0Monitoring/SelectedLambda",
    v0         = "LambdaEventSelector"
)

##################
# For MinBias
##################
standaloneTrackMonitorMC = standaloneTrackMonitor.clone(
    puScaleFactorFile = "PileupScaleFactor_316060_wrt_nVertex_ZeroBias.root",
    doPUCorrection    = True,
    isMC              = True
    )
standaloneValidationMinbias = cms.Sequence(
    hltPathFilter
    * selectedPrimaryVertices 
#    * selectedMultiplicityTracks  # Use selectedMultiplicityTracks if needed nTracks > desired multiplicity
#    * selectedAlcaRecoZBTracks
    * selectedTracks
    * standaloneTrackMonitor
    * KshortMonitor
    * LambdaMonitor)

standaloneValidationMinbiasMC = cms.Sequence(
    hltPathFilter
    * selectedPrimaryVertices 
#    * selectedMultiplicityTracks  # Use selectedMultiplicityTracks if needed nTracks > desired multiplicity
#    * selectedAlcaRecoZBTracks
    * selectedTracks
    * standaloneTrackMonitorMC
    * KshortMonitor
    * LambdaMonitor)

##################
# For V0s in MinBias
##################
standaloneTrackMonitorK0 = standaloneTrackMonitor.clone(
    folderName = "K0Tracks",
    trackInputTag = 'KshortTracks',
    )

standaloneTrackMonitorK0MC = standaloneTrackMonitor.clone(
    folderName = "K0Tracks",
    trackInputTag = 'KshortTracks',
    puScaleFactorFile = "PileupScaleFactor_316082_wrt_nVertex_DYToLL.root",
    doPUCorrection    = True,
    isMC              = True
    )

standaloneTrackMonitorLambda = standaloneTrackMonitor.clone(
    folderName = "LambdaTracks",
    trackInputTag = 'LambdaTracks',
    )

standaloneTrackMonitorLambdaMC = standaloneTrackMonitor.clone(
    folderName = "LambdaTracks",
    trackInputTag = 'LambdaTracks',
    puScaleFactorFile = "PileupScaleFactor_316082_wrt_nVertex_DYToLL.root",
    doPUCorrection    = True,
    isMC              = True
    )

standaloneValidationK0s = cms.Sequence(
    hltPathFilter
    * selectedPrimaryVertices
    * KShortEventSelector
    * KshortTracks
    * standaloneTrackMonitorK0
    * SelectedKshortMonitor)

standaloneValidationK0sMC = cms.Sequence(
    hltPathFilter
    * selectedPrimaryVertices
    * KShortEventSelector
    * KshortTracks
    * standaloneTrackMonitorK0MC
    * SelectedKshortMonitor)

standaloneValidationLambdas = cms.Sequence(
    hltPathFilter
    * selectedPrimaryVertices
    * LambdaEventSelector
    * LambdaTracks
    * standaloneTrackMonitorLambda
    * SelectedLambdaMonitor)

standaloneValidationLambdasMC = cms.Sequence(
    hltPathFilter
    * selectedPrimaryVertices
    * LambdaEventSelector
    * LambdaTracks
    * standaloneTrackMonitorLambdaMC
    * SelectedLambdaMonitor)

##################
# For ZtoEE
##################
standaloneTrackMonitorElec = standaloneTrackMonitor.clone(
    folderName = "ElectronTracks",
    trackInputTag = 'electronTracks',
    )

standaloneTrackMonitorElecMC = standaloneTrackMonitor.clone(
    folderName = "ElectronTracks",
    trackInputTag = 'electronTracks',
    puScaleFactorFile = "PileupScaleFactor_316082_wrt_nVertex_DYToLL.root",
    doPUCorrection    = True,
    isMC              = True
    )

ZEEDetailsMC = ZEEDetails.clone(
    puScaleFactorFile = "PileupScaleFactor_316082_wrt_nVertex_DYToLL.root",
    doPUCorrection    = True,
    isMC              = True
    )

standaloneValidationElec = cms.Sequence(
    hltPathFilterElectron
    * selectedTracks
    * selectedPrimaryVertices
    * ztoEEEventSelector
    * electronTracks
    * standaloneTrackMonitorElec   
    * standaloneTrackMonitor
    * ZEEDetails)
standaloneValidationElecMC = cms.Sequence(
    hltPathFilterElectron
    * selectedTracks
    * selectedPrimaryVertices
    * ztoEEEventSelector
    * electronTracks
    * standaloneTrackMonitorElecMC   
    * standaloneTrackMonitorMC
    * ZEEDetailsMC)

##################
# For ZtoMM
##################
standaloneTrackMonitorMuon = standaloneTrackMonitor.clone(
    folderName = "MuonTracks",
    trackInputTag = 'muonTracks',
    )
standaloneTrackMonitorMuonMC = standaloneTrackMonitor.clone(
    folderName = "MuonTracks",
    trackInputTag = 'muonTracks',
    puScaleFactorFile = "PileupScaleFactor_316082_wrt_nVertex_DYToLL.root",
    doPUCorrection    = True,
    isMC              = True
    )

standaloneValidationMuon = cms.Sequence(
    hltPathFilterMuon
    * selectedTracks
    * selectedPrimaryVertices
    * ztoMMEventSelector
    * muonTracks
    * standaloneTrackMonitorMuon
    * standaloneTrackMonitor)

standaloneValidationMuonMC = cms.Sequence(
    hltPathFilterMuon
    * selectedTracks
    * selectedPrimaryVertices
    * ztoMMEventSelector
    * muonTracks
    * standaloneTrackMonitorMuonMC 
    * standaloneTrackMonitorMC)

##################
# For ttbar
##################
standaloneTrackMonitorTTbar = standaloneTrackMonitor.clone(
    folderName = "TTbarTracks",
    trackInputTag = 'ttbarTracks',
    )

standaloneTrackMonitorTTbarMC = standaloneTrackMonitor.clone(
    folderName = "TTbarTracks",
    trackInputTag = 'ttbarTracks',
    puScaleFactorFile = "PileupScaleFactor_316082_wrt_nVertex_DYToLL.root",
    doPUCorrection    = True,
    isMC              = True
    )

standaloneValidationTTbar = cms.Sequence(
    hltPathFilterTtbar
    * selectedPrimaryVertices
    * ttbarEventSelector
    * ttbarTracks
#    * selectedTracks
    * standaloneTrackMonitorTTbar)

standaloneValidationTTbarMC = cms.Sequence(
    hltPathFilterTtbar
    * selectedPrimaryVertices
    * ttbarEventSelector
    * ttbarTracks
#    * selectedTracks
    * standaloneTrackMonitorTTbarMC)