-
Notifications
You must be signed in to change notification settings - Fork 4.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
improve ScoutingNano integration with autoNano with customisations an…
…d clean up code
- Loading branch information
Showing
3 changed files
with
742 additions
and
516 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
194 changes: 158 additions & 36 deletions
194
PhysicsTools/NanoAOD/python/custom_run3scouting_cff.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,62 +1,184 @@ | ||
| import FWCore.ParameterSet.Config as cms | ||
| from PhysicsTools.NanoAOD.run3scouting_cff import * | ||
| from PhysicsTools.NanoAOD.globals_cff import puTable | ||
| from PhysicsTools.NanoAOD.triggerObjects_cff import unpackedPatTrigger, triggerObjectTable, l1bits | ||
| from L1Trigger.Configuration.L1TRawToDigi_cff import * | ||
| from EventFilter.L1TRawToDigi.gtStage2Digis_cfi import gtStage2Digis | ||
| from PhysicsTools.PatAlgos.triggerLayer1.triggerProducer_cfi import patTrigger | ||
| from PhysicsTools.PatAlgos.slimming.selectedPatTrigger_cfi import selectedPatTrigger | ||
| from PhysicsTools.PatAlgos.slimming.slimmedPatTrigger_cfi import slimmedPatTrigger | ||
| from PhysicsTools.NanoAOD.triggerObjects_cff import l1bits | ||
| from PhysicsTools.NanoAOD.globals_cff import puTable | ||
|
|
||
| ############################ | ||
| ### Sub Task Definitions ### | ||
| ############################ | ||
|
|
||
| # common tasks | ||
| particleTask = cms.Task(scoutingPFCands) | ||
| ak4JetTableTask = cms.Task(ak4ScoutingJets,ak4ScoutingJetParticleNetJetTagInfos,ak4ScoutingJetParticleNetJetTags,ak4ScoutingJetTable) | ||
| ak8JetTableTask = cms.Task(ak8ScoutingJets,ak8ScoutingJetsSoftDrop,ak8ScoutingJetsSoftDropMass,ak8ScoutingJetEcfNbeta1,ak8ScoutingJetNjettiness,ak8ScoutingJetParticleNetJetTagInfos,ak8ScoutingJetParticleNetJetTags,ak8ScoutingJetParticleNetMassRegressionJetTags,ak8ScoutingJetTable) | ||
| # Task contains all dependent tasks | ||
| # ExtensionTask must be run on top of another Task | ||
|
|
||
| muonScoutingTableTask = cms.Task(muonScoutingTable) | ||
| displacedvertexScoutingTableTask = cms.Task(displacedvertexScoutingTable) | ||
| ############################# | ||
| # Scouting Original Objects # | ||
| ############################# | ||
|
|
||
| # from 2024, there are two scouting muon collections | ||
| # Scouting Muon | ||
| scoutingMuonTableTask = cms.Task(scoutingMuonTable) | ||
| scoutingMuonDisplacedVertexTableTask = cms.Task(scoutingMuonDisplacedVertexTable) | ||
|
|
||
| # from 2024, there are two muon collections | ||
| from Configuration.Eras.Modifier_run3_scouting_nanoAOD_post2023_cff import run3_scouting_nanoAOD_post2023 | ||
| run3_scouting_nanoAOD_post2023.toReplaceWith(muonScoutingTableTask, cms.Task(muonVtxScoutingTable, muonNoVtxScoutingTable))\ | ||
| .toReplaceWith(displacedvertexScoutingTableTask, cms.Task(displacedvertexVtxScoutingTable, displacedvertexNoVtxScoutingTable)) | ||
| run3_scouting_nanoAOD_post2023.toReplaceWith(scoutingMuonTableTask, cms.Task(scoutingMuonVtxTable, scoutingMuonNoVtxTable))\ | ||
| .toReplaceWith(scoutingMuonDisplacedVertexTableTask, cms.Task(scoutingMuonVtxDisplacedVertexTable, scoutingMuonNoVtxDisplacedVertexTable)) | ||
|
|
||
| # other collections are directly from original Run3Scouting objects, so unnessary to define tasks | ||
|
|
||
| ############################ | ||
| # Scouting Derived Objects # | ||
| ############################ | ||
|
|
||
| scoutingPFCandidateTask = cms.Task(scoutingPFCandidate, scoutingPFCandidateTable) | ||
| scoutingPFJetReclusterTask = cms.Task( | ||
| scoutingPFCandidate, # translate to reco::PFCandidate, used as input | ||
| scoutingPFJetRecluster, # jet clustering | ||
| scoutingPFJetReclusterParticleNetJetTagInfos, scoutingPFJetReclusterParticleNetJetTags, # jet tagging | ||
| scoutingPFJetReclusterTable | ||
| ) | ||
| scoutingPFJetReclusterMatchGenExtensionTask = cms.Task( | ||
| scoutingPFJetReclusterMatchGen, # gen jet matching | ||
| scoutingPFJetReclusterMatchGenExtensionTable | ||
| ) | ||
|
|
||
| scoutingFatPFJetReclusterTask = cms.Task( | ||
| scoutingPFCandidate, # translate to reco::PFCandidate, used as input | ||
| scoutingFatPFJetRecluster, # jet clustering | ||
| scoutingFatPFJetReclusterParticleNetJetTagInfos, scoutingFatPFJetReclusterParticleNetJetTags, # jet tagging | ||
| scoutingFatPFJetReclusterSoftDrop, scoutingFatPFJetReclusterSoftDropMass, # softdrop mass | ||
| scoutingFatPFJetReclusterParticleNetJetTagInfos, scoutingFatPFJetReclusterParticleNetMassRegressionJetTags, # regressed mass | ||
| scoutingFatPFJetReclusterEcfNbeta1, scoutingFatPFJetReclusterNjettiness, # substructure variables | ||
| scoutingFatPFJetReclusterTable | ||
| ) | ||
| scoutingFatPFJetReclusterMatchGenExtensionTask = cms.Task( | ||
| scoutingFatPFJetReclusterMatchGen, # gen jet matching | ||
| scoutingFatPFJetReclusterMatchGenExtensionTable | ||
| ) | ||
|
|
||
| ############################ | ||
| # Trigger Bits and Objects # | ||
| ############################ | ||
|
|
||
| ## L1 decisions | ||
| gtStage2DigisScouting = gtStage2Digis.clone(InputLabel="hltFEDSelectorL1") | ||
| l1bitsScouting = l1bits.clone(src="gtStage2DigisScouting") | ||
| l1bitsScouting = l1bits.clone(src="gtStage2DigisScouting") | ||
|
|
||
| ## L1 objects | ||
| from PhysicsTools.NanoAOD.l1trig_cff import * | ||
| l1MuScoutingTable = l1MuTable.clone(src=cms.InputTag("gtStage2DigisScouting","Muon")) | ||
| l1JetScoutingTable = l1JetTable.clone(src=cms.InputTag("gtStage2DigisScouting","Jet")) | ||
| l1EGScoutingTable = l1EGTable.clone(src=cms.InputTag("gtStage2DigisScouting","EGamma")) | ||
| l1TauScoutingTable = l1TauTable.clone(src=cms.InputTag("gtStage2DigisScouting","Tau")) | ||
| l1EtSumScoutingTable = l1EtSumTable.clone(src=cms.InputTag("gtStage2DigisScouting","EtSum")) | ||
| l1MuScoutingTable = l1MuTable.clone(src=cms.InputTag("gtStage2DigisScouting", "Muon")) | ||
| l1EGScoutingTable = l1EGTable.clone(src=cms.InputTag("gtStage2DigisScouting", "EGamma")) | ||
| l1TauScoutingTable = l1TauTable.clone(src=cms.InputTag("gtStage2DigisScouting", "Tau")) | ||
| l1JetScoutingTable = l1JetTable.clone(src=cms.InputTag("gtStage2DigisScouting", "Jet")) | ||
| l1EtSumScoutingTable = l1EtSumTable.clone(src=cms.InputTag("gtStage2DigisScouting", "EtSum")) | ||
|
|
||
| #reduce the variables to the core variables as only these are available in gtStage2Digis | ||
| l1EGScoutingTable.variables = cms.PSet(l1EGReducedVars) | ||
| # reduce the variables to the core variables as only these are available in gtStage2Digis | ||
| l1MuScoutingTable.variables = cms.PSet(l1MuonReducedVars) | ||
| l1JetScoutingTable.variables = cms.PSet(l1JetReducedVars) | ||
| l1EGScoutingTable.variables = cms.PSet(l1EGReducedVars) | ||
| l1TauScoutingTable.variables = cms.PSet(l1TauReducedVars) | ||
| l1JetScoutingTable.variables = cms.PSet(l1JetReducedVars) | ||
| l1EtSumScoutingTable.variables = cms.PSet(l1EtSumReducedVars) | ||
|
|
||
| triggerTask = cms.Task( | ||
| gtStage2DigisScouting, l1bitsScouting, | ||
| l1MuScoutingTable, l1EGScoutingTable, l1TauScoutingTable, l1JetScoutingTable, l1EtSumScoutingTable, | ||
| ) | ||
| triggerSequence = cms.Sequence(L1TRawToDigi+cms.Sequence(triggerTask)) | ||
| ############################## | ||
| ### Main Tasks Definitions ### | ||
| ############################## | ||
|
|
||
| # MC tasks | ||
| genJetTask = cms.Task(ak4ScoutingJetMatchGen,ak4ScoutingJetExtTable,ak8ScoutingJetMatchGen,ak8ScoutingJetExtTable) | ||
| puTask = cms.Task(puTable) | ||
| # default configuration for ScoutingNano common for both data and MC | ||
| def prepareScoutingNanoTaskCommon(): | ||
| # Scouting original objects | ||
| # all scouting objects are saved except PF Candidate and Track | ||
| scoutingNanoTaskCommon = cms.Task() | ||
| scoutingNanoTaskCommon.add(scoutingMuonTableTask, scoutingMuonDisplacedVertexTableTask) | ||
| scoutingNanoTaskCommon.add(scoutingElectronTable) | ||
| scoutingNanoTaskCommon.add(scoutingPhotonTable) | ||
| scoutingNanoTaskCommon.add(scoutingPrimaryVertexTable) | ||
| scoutingNanoTaskCommon.add(scoutingPFJetTable) | ||
| scoutingNanoTaskCommon.add(scoutingMETTable, scoutingRhoTable) | ||
|
|
||
| # Scouting derived objects | ||
| scoutingNanoTaskCommon.add(scoutingPFJetReclusterTask) | ||
| scoutingNanoTaskCommon.add(scoutingFatPFJetReclusterTask) | ||
|
|
||
| nanoTableTaskCommon = cms.Task(photonScoutingTable,muonScoutingTableTask,electronScoutingTable,primaryvertexScoutingTable,displacedvertexScoutingTableTask,jetScoutingTable,rhoScoutingTable,metScoutingTable,particleTask,ak4JetTableTask,ak8JetTableTask) | ||
| return scoutingNanoTaskCommon | ||
|
|
||
| nanoSequenceCommon = cms.Sequence(triggerSequence,nanoTableTaskCommon) | ||
| # tasks related to trigger bits and objects | ||
| def prepareScoutingTriggerTask(): | ||
| scoutingTriggerTask = cms.Task(gtStage2DigisScouting, l1bitsScouting) | ||
| scoutingTriggerTask.add(cms.Task(l1MuScoutingTable, l1EGScoutingTable, l1TauScoutingTable, l1JetScoutingTable, l1EtSumScoutingTable)) | ||
| return scoutingTriggerTask | ||
|
|
||
| nanoSequence = cms.Sequence(nanoSequenceCommon) | ||
| # additional tasks for running on MC | ||
| def prepareScoutingNanoTaskMC(): | ||
| scoutingNanoTaskMC = cms.Task() | ||
| scoutingNanoTaskMC.add(scoutingPFJetReclusterMatchGenExtensionTask) | ||
| scoutingNanoTaskMC.add(scoutingFatPFJetReclusterMatchGenExtensionTask) | ||
|
|
||
| nanoSequenceMC = cms.Sequence(nanoSequenceCommon + cms.Sequence(cms.Task(genJetTask,puTask))) | ||
| scoutingNanoTaskMC.add(puTable) | ||
| return scoutingNanoTaskMC | ||
|
|
||
| # Common tasks added to main scoutingNanoSequence | ||
| scoutingNanoTaskCommon = prepareScoutingNanoTaskCommon() | ||
| scoutingNanoSequence = cms.Sequence(scoutingNanoTaskCommon) | ||
|
|
||
| # Specific tasks which will be added to sequence during customization | ||
| scoutingTriggerTask = prepareScoutingTriggerTask() | ||
| scoutingTriggerSequence = cms.Sequence(L1TRawToDigi+cms.Sequence(scoutingTriggerTask)) | ||
| scoutingNanoTaskMC = prepareScoutingNanoTaskMC() | ||
|
|
||
| def customiseScoutingNano(process): | ||
| # if running with standard NanoAOD, triggerSequence is already added | ||
| # if running standalone, triggerSequence need to be added | ||
| if not ((hasattr(process, "nanoSequence") and process.schedule.contains(process.nanoSequence)) | ||
| or hasattr(process, "nanoSequenceMC") and process.schedule.contains(process.nanoSequenceMC)): | ||
| process.trigger_step = cms.Path(process.scoutingTriggerSequence) | ||
| process.schedule.extend([process.trigger_step]) | ||
|
|
||
| # specific tasks when running on MC | ||
| runOnMC = hasattr(process,"NANOEDMAODSIMoutput") or hasattr(process,"NANOAODSIMoutput") | ||
| if runOnMC: | ||
| process.scoutingNanoSequence.associate(scoutingNanoTaskMC) | ||
|
|
||
| return process | ||
|
|
||
| def customiseScoutingNanoFromMini(process): | ||
| # normal customise for ScoutingNano | ||
| process = customiseScoutingNano(process) | ||
|
|
||
| # remove L1TRawToDigi | ||
| process.scoutingTriggerSequence.remove(process.L1TRawToDigi) | ||
|
|
||
| # remove gtStage2Digis since they are already run for Mini | ||
| process.scoutingTriggerTask.remove(process.gtStage2DigisScouting) | ||
|
|
||
| # change src for l1 bits | ||
| process.l1bitsScouting.src = cms.InputTag("gtStage2Digis") | ||
|
|
||
| # change src for l1 objects | ||
| process.l1MuScoutingTable.src = cms.InputTag("gmtStage2Digis", "Muon") | ||
| process.l1EGScoutingTable.src = cms.InputTag("caloStage2Digis", "EGamma") | ||
| process.l1TauScoutingTable.src = cms.InputTag("caloStage2Digis", "Tau") | ||
| process.l1JetScoutingTable.src = cms.InputTag("caloStage2Digis", "Jet") | ||
| process.l1EtSumScoutingTable.src = cms.InputTag("caloStage2Digis", "EtSum") | ||
|
|
||
| return process | ||
|
|
||
| ##################### | ||
| ### Customisation ### | ||
| ##################### | ||
| # these function are designed to be used with --customise flag in cmsDriver.py | ||
| # e.g. --customise PhysicsTools/NanoAOD/python/custom_run3scouting_cff.addScoutingPFCandidate | ||
|
|
||
| def addScoutingTrack(process): | ||
| process.scoutingNanoSequence.associate(cms.Task(scoutingTrackTable)) | ||
| return process | ||
|
|
||
| def addScoutingParticle(process): | ||
| # original PF candidate without post-processing | ||
| process.scoutingNanoSequence.associate(cms.Task(scoutingParticleTable)) | ||
| return process | ||
|
|
||
| def nanoAOD_customizeCommon(process): | ||
| def addScoutingPFCandidate(process): | ||
| # PF candidate after translation to reco::PFCandidate | ||
| process.scoutingNanoSequence.associate(scoutingPFCandidateTask) | ||
| return process |
Oops, something went wrong.