New Tau Embedding process modifiers (#6)

* first try distributing process Modifiers of Cleaning step

* disable finale state radiation for mu->mu embedding

* update center of mass energy for run3

* implement correct ModifierChain for tau embedding process modifiers

* revert some changes

* set all process modifier for tau embeddign cleaning step

* Introduce special cleaning and simulation process modifier

* add process modifiers for sim gen step

* Set process modifiers for reco sim step

* add process modifier for tau embedding merging step
Those "toReplaceWith" modifiers need to be applied as late as possible e.g. in StandardSequences.Reconstruction_Data_cff.
This is to avoid collisions with other modules that are cloned and modified versions from the original modules or "toModify" modifier calls that try to modify the original modules. They throw an exeption if they want to do this on the replaced merger modules.

* add autoNANO option for the tau embedding nanoaod files

* adjust test cmsDriver commands to work with the new process modifiers

---------

Co-authored-by: Christian Winter <[email protected]>

Author: winterchristian

Configuration/EventContent/python/EventContent_cff.py

@@ -1047,8 +1047,9 @@ def SwapKeepAndDrop(l):
 TauEmbeddingSimRecoEventContent = RAWRECOSIMHLTEventContent.clone()
 TauEmbeddingSimRecoEventContent.outputCommands.extend(TauEmbSimReco.outputCommands)
 
-TauEmbeddingMergeEventContent = MINIAODSIMEventContent.clone()
-TauEmbeddingMergeEventContent.outputCommands.extend(TauEmbMerge.outputCommands)
+# needs to have MINIAOD in its name to be recognized by cmsDriver as a MINIAOD output so that miniAOD_customizeOutput is applied
+TauEmbeddingMergeMINIAODEventContent = MINIAODSIMEventContent.clone()
+TauEmbeddingMergeMINIAODEventContent.outputCommands.extend(TauEmbMerge.outputCommands)
 
 # needs to have NANOAOD in its name to be recognized by cmsDriver as a NANOAOD output so that NanoAODOutputModule is used
 TauEmbeddingNANOAODEventContent = NANOAODSIMEventContent.clone()

Configuration/ProcessModifiers/python/tau_embedding_cleaning_cff.py

@@ -0,0 +1,4 @@
+import FWCore.ParameterSet.Config as cms
+
+# This modifier is to implement general changes in the RECO sequence needed for the tau embedding method. More info can be found in TauAnalysis/MCEmbeddingTools.
+tau_embedding_cleaning = cms.Modifier()

Configuration/ProcessModifiers/python/tau_embedding_merging_cff.py

@@ -0,0 +1,4 @@
+import FWCore.ParameterSet.Config as cms
+
+# This modifier is to implement general changes in the RECO sequence needed for the tau embedding method. More info can be found in TauAnalysis/MCEmbeddingTools.
+tau_embedding_merging = cms.Modifier()

Configuration/ProcessModifiers/python/tau_embedding_mu_to_e_cff.py

@@ -1,4 +1,4 @@
 import FWCore.ParameterSet.Config as cms
 
 # This modifier is to change tau embedding method to perform mu->e embedding. More info can be found in TauAnalysis/MCEmbeddingTools.
-tau_embedding_mu_to_e = cms.Modifier()
+tau_embedding_mu_to_e = cms.Modifier()

Configuration/ProcessModifiers/python/tau_embedding_mu_to_mu_cff.py

@@ -1,4 +1,4 @@
 import FWCore.ParameterSet.Config as cms
 
 # This modifier is to change tau embedding method to perform mu->mu embedding. More info can be found in TauAnalysis/MCEmbeddingTools.
-tau_embedding_mu_to_mu =  cms.Modifier()
+tau_embedding_mu_to_mu = cms.Modifier()

Configuration/ProcessModifiers/python/tau_embedding_mutauh_cff.py

@@ -1,4 +1,4 @@
 import FWCore.ParameterSet.Config as cms
 
 # This modifier sets the correct cuts for the taus decaying into one jet and one muon. More info can be found in TauAnalysis/MCEmbeddingTools.
-tau_embedding_mutauh = cms.Modifier()
+tau_embedding_mutauh = cms.Modifier()

Configuration/ProcessModifiers/python/tau_embedding_sim_cff.py

@@ -0,0 +1,4 @@
+import FWCore.ParameterSet.Config as cms
+
+# This modifier is to implement general changes in the RECO sequence needed for the tau embedding method. More info can be found in TauAnalysis/MCEmbeddingTools.
+tau_embedding_sim = cms.Modifier()

Configuration/StandardSequences/python/Reconstruction_cff.py

@@ -320,3 +320,12 @@
 #
 reconstruction_woCosmicMuonsTask = cms.Task(localrecoTask,globalrecoTask,highlevelrecoTask,logErrorHarvester)
 reconstruction_woCosmicMuons = cms.Sequence(reconstruction_woCosmicMuonsTask)
+
+##
+## Modify for the tau embedding methods reco sim step
+##
+from Configuration.ProcessModifiers.tau_embedding_sim_cff import tau_embedding_sim
+_tau_embedding_sim_globalreco_trackingTask = globalreco_trackingTask.copyAndExclude([offlineBeamSpotTask])
+tau_embedding_sim.toReplaceWith(globalreco_trackingTask, _tau_embedding_sim_globalreco_trackingTask)
+_tau_embedding_sim_reconstruction_pixelTrackingOnlyTask = reconstruction_pixelTrackingOnlyTask.copyAndExclude([offlineBeamSpotTask])
+tau_embedding_sim.toReplaceWith(reconstruction_pixelTrackingOnlyTask, _tau_embedding_sim_reconstruction_pixelTrackingOnlyTask)

IOMC/EventVertexGenerators/python/VtxSmearedRealistic_cfi.py

@@ -6,3 +6,10 @@
     src = "generator:unsmeared",
     readDB = True
 )
+
+##
+## Set the the vertex used for the simulation to the measured vertex for the tau embedding method
+##
+from Configuration.ProcessModifiers.tau_embedding_sim_cff import tau_embedding_sim
+from TauAnalysis.MCEmbeddingTools.Simulation_GEN_cfi import tau_embedding_vtx_corrected_to_input
+tau_embedding_sim.toReplaceWith(VtxSmeared, tau_embedding_vtx_corrected_to_input)

PhysicsTools/NanoAOD/python/autoNANO.py

@@ -110,4 +110,7 @@ def expandNanoMapping(seqList, mapping, key):
     # NANOGEN (from MiniAOD)
     'GENFromMini': {'sequence': 'PhysicsTools/NanoAOD/nanogen_cff.nanogenSequence',
                     'customize': 'PhysicsTools/NanoAOD/nanogen_cff.customizeNanoGENFromMini'},
+    # Tau embedding NanoAOD (includes Data as well as MC tables with an addition tau embedding table)
+    'TauEmbedding': {'sequence': '@PHYS+PhysicsTools/NanoAOD/nano_cff.nanoSequenceFS+TauAnalysis/MCEmbeddingTools/Nano_cff.embeddingTable_seq',
+                     'customize': '@PHYS'},
 }