def Config(self, stream, configKey):
### smart sliming
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
SlHelper = SlimmingHelper(configKey + "SlimmingHelper")
SlHelper.SmartCollections = self.smSlContainer[configKey]
useSmartSlimmingIfSupported(self.checkContainers.get(configKey, []),
self.allVarContainer[configKey],
self.Items[configKey],
SlHelper.NamesAndTypes)
SlHelper.AllVariables = self.allVarContainer[configKey]
# SlHelper.StaticContent = self.Items[configKey]
SlHelper.ExtraVariables = self.extraVariables.get(configKey, [])
SlHelper.IncludeMuonTriggerContent = self.UseTriggerContent.get(
configKey, True)
SlHelper.AppendContentToStream(stream)
### other items
for item in self.Items[configKey]:
stream.AddItem(item)
"Photons",
"Muons",
"TauJets",
"MET_Reference_AntiKt4EMTopo",
"AntiKt4LCTopoJets",
"AntiKt4EMTopoJets",
"BTagging_AntiKt4LCTopo",
"BTagging_AntiKt4EMTopo",
"InDetTrackParticles",
"PrimaryVertices" ]
STDM5SlimmingHelper.IncludeEGammaTriggerContent = True
STDM5SlimmingHelper.IncludeMuonTriggerContent = True
STDM5SlimmingHelper.ExtraVariables = ExtraContentAll
STDM5SlimmingHelper.AllVariables = ExtraContainersAll
if globalflags.DataSource()=='geant4':
STDM5SlimmingHelper.ExtraVariables += ExtraContentAllTruth
STDM5SlimmingHelper.AllVariables += ExtraContainersTruth
STDM5SlimmingHelper.AppendContentToStream(STDM5Stream)
STDM5Stream.AddItem("xAOD::EventShape#*")
STDM5Stream.AddItem("xAOD::EventShapeAuxInfo#*")
#STDM5Stream.AddItem("xAOD::TrigNavigation#*")
#STDM5Stream.AddItem("xAOD::TrigNavigationAuxInfo#*")
if globalflags.DataSource()=='geant4':
STDM5Stream.AddItem( "xAOD::TruthParticleContainer#TruthMuons" )
STDM5Stream.AddItem( "xAOD::TruthParticleAuxContainer#TruthMuonsAux." )
STDM5Stream.AddItem( "xAOD::TruthParticleContainer#TruthElectrons" )
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
#from DerivationFrameworkHiggs.HIGG5D2ContentList import HIGG5D2Content
from DerivationFrameworkHiggs.HIGG5D2ExtraContent import *
HIGG5D2SlimmingHelper = SlimmingHelper("HIGG5D2SlimmingHelper")
HIGG5D2SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo",
"MET_Reference_AntiKt4LCTopo", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets",
"BTagging_AntiKt4EMTopo", "BTagging_AntiKt4LCTopo", "InDetTrackParticles",
"PrimaryVertices"
]
HIGG5D2SlimmingHelper.ExtraVariables = ExtraContent
HIGG5D2SlimmingHelper.AllVariables = ExtraContainers
if globalflags.DataSource() == 'geant4':
HIGG5D2SlimmingHelper.ExtraVariables += ExtraContentTruth
HIGG5D2SlimmingHelper.AllVariables += ExtraContainersTruth
# Add the jet containers to the stream
addJetOutputs(HIGG5D2SlimmingHelper, ["HIGG5D2Jets"])
# Add the MET containers to the stream
addMETOutputs(HIGG5D2SlimmingHelper, ["AntiKt4LCTopo", "Track"])
HIGG5D2SlimmingHelper.IncludeMuonTriggerContent = True
HIGG5D2SlimmingHelper.IncludeEGammaTriggerContent = True
#HIGG5D2SlimmingHelper.IncludeBPhysTriggerContent = True
#HIGG5D2SlimmingHelper.IncludeJetTauEtMissTriggerContent = True
HIGG5D2SlimmingHelper.IncludeEtMissTriggerContent = True
#=======================================
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot7Seq
exot7Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT7Kernel_skim",
AugmentationTools=[EXOT7DeltaRTool],
SkimmingTools=[EXOT7StringSkimmingTool])
exot7Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT7Kernel", ThinningTools=thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT7ContentList import *
EXOT7SlimmingHelper = SlimmingHelper("EXOT7SlimmingHelper")
EXOT7SlimmingHelper.StaticContent = EXOT7Content
EXOT7SlimmingHelper.SmartCollections = EXOT7SmartCollections
EXOT7SlimmingHelper.ExtraVariables = EXOT7ExtraVariables
EXOT7SlimmingHelper.AllVariables = EXOT7AllVariables
#EXOT7SlimmingHelper.IncludeJetTauEtMissTriggerContent = True
EXOT7SlimmingHelper.IncludeJetTriggerContent = True
EXOT7SlimmingHelper.IncludeBJetTriggerContent = True
EXOT7SlimmingHelper.AppendContentToStream(EXOT7Stream)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
JETM7SlimmingHelper = SlimmingHelper("JETM7SlimmingHelper")
JETM7SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles",
"PrimaryVertices"
]
JETM7SlimmingHelper.AllVariables = [
"BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters",
"MuonTruthParticles", "egammaTruthParticles", "TruthParticles",
"TruthEvents", "TruthVertices", "MuonSegments"
]
JETM7SlimmingHelper.ExtraVariables = [
"Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus"
]
# Trigger content
JETM7SlimmingHelper.IncludeMuonTriggerContent = True
JETM7SlimmingHelper.IncludeEGammaTriggerContent = True
# Add the jet containers to the stream
addJetOutputs(JETM7SlimmingHelper, ["SmallR"])
# Add the MET containers to the stream
addMETOutputs(JETM7SlimmingHelper,
["Diagnostic", "AntiKt4LCTopo", "AntiKt4EMPFlow", "Track"])
JETM7SlimmingHelper.AppendContentToStream(JETM7Stream)
"AntiKtVR30Rmax4Rmin02TrackJets_BTagging201903",
"AntiKt4EMPFlowJets_BTagging201810",
"AntiKt4EMPFlowJets_BTagging201903",
"AntiKt4EMTopoJets_BTagging201810",
"BTagging_AntiKtVR30Rmax4Rmin02Track_201810",
"BTagging_AntiKtVR30Rmax4Rmin02Track_201903",
"BTagging_AntiKt4EMPFlow_201810",
"BTagging_AntiKt4EMPFlow_201903",
"BTagging_AntiKt4EMTopo_201810",
]
JETM6SlimmingHelper.AllVariables = [
"TruthEvents",
"Kt4EMTopoOriginEventShape","Kt4EMPFlowEventShape",
]
JETM6SlimmingHelper.ExtraVariables = ['CaloCalTopoClusters.calE.calEta.calM.calPhi.CENTER_MAG']
JETM6SlimmingHelper.ExtraVariables += ['Electrons.'+NewTrigVars["Electrons"],'Muons.'+NewTrigVars["Muons"],'Photons.'+NewTrigVars["Photons"]]
JETM6SlimmingHelper.ExtraVariables += [
'HLT_xAOD__JetContainer_a4tcemsubjesFS.ActiveArea.ActiveArea4vec_eta.ActiveArea4vec_m.ActiveArea4vec_phi.ActiveArea4vec_pt.AverageLArQF.BchCorrCell.CentroidR.DetectorEta.EMFrac.EnergyPerSampling.FracSamplingMax.FracSamplingMaxIndex.HECFrac.HECQuality.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_m.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_pt.JetEtaJESScaleMomentum_eta.JetEtaJESScaleMomentum_m.JetEtaJESScaleMomentum_phi.JetEtaJESScaleMomentum_pt.JetPileupScaleMomentum_eta.JetPileupScaleMomentum_m.JetPileupScaleMomentum_phi.JetPileupScaleMomentum_pt.LArQuality.N90Constituents.NegativeE.Timing.eta.kinematics.m.phi.pt',
'HLT_xAOD__JetContainer_a4tcemsubjesISFS.ActiveArea.ActiveArea4vec_eta.ActiveArea4vec_m.ActiveArea4vec_phi.ActiveArea4vec_pt.AverageLArQF.BchCorrCell.CentroidR.DetectorEta.EMFrac.EnergyPerSampling.FracSamplingMax.FracSamplingMaxIndex.HECFrac.HECQuality.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_m.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_pt.JetEtaJESScaleMomentum_eta.JetEtaJESScaleMomentum_m.JetEtaJESScaleMomentum_phi.JetEtaJESScaleMomentum_pt.JetPileupScaleMomentum_eta.JetPileupScaleMomentum_m.JetPileupScaleMomentum_phi.JetPileupScaleMomentum_pt.LArQuality.N90Constituents.NegativeE.Timing.eta.kinematics.m.phi.pt',
'HLT_xAOD__JetContainer_a10tclcwsubjesFS.ActiveArea.ActiveArea4vec_eta.ActiveArea4vec_m.ActiveArea4vec_phi.ActiveArea4vec_pt.AverageLArQF.BchCorrCell.CentroidR.DetectorEta.EMFrac.EnergyPerSampling.FracSamplingMax.FracSamplingMaxIndex.HECFrac.HECQuality.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_m.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_pt.JetEMScaleMomentum_eta.JetEMScaleMomentum_m.JetEMScaleMomentum_phi.JetEMScaleMomentum_pt.JetEtaJESScaleMomentum_eta.JetEtaJESScaleMomentum_m.JetEtaJESScaleMomentum_phi.JetEtaJESScaleMomentum_pt.JetPileupScaleMomentum_eta.JetPileupScaleMomentum_m.JetPileupScaleMomentum_phi.JetPileupScaleMomentum_pt.LArQuality.N90Constituents.NegativeE.Timing.eta.kinematics.m.phi.pt',
]
for truthc in [
"TruthTopQuark",
"TruthBosons",
"TruthHF"
]:
JETM6SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#"+truthc+"WithDecayParticles")
JETM6SlimmingHelper.StaticContent.append("xAOD::TruthParticleAuxContainer#"+truthc+"WithDecayParticlesAux.")
JETM6SlimmingHelper.StaticContent.append("xAOD::TruthVertexContainer#"+truthc+"WithDecayVertices")
PHYSLITESlimmingHelper.SmartCollections = [
"InDetTrackParticles",
"PrimaryVertices",
]
# Variables to save, per object type
PHYSLITESlimmingHelper.ExtraVariables = [
"AnalysisElectrons.trackParticleLinks.pt.eta.phi.m.charge.author.DFCommonElectronsLHVeryLoose.DFCommonElectronsLHLoose.DFCommonElectronsLHLooseBL.DFCommonElectronsLHMedium.DFCommonElectronsLHTight.DFCommonElectronsLHVeryLooseIsEMValue.DFCommonElectronsLHLooseIsEMValue.DFCommonElectronsLHLooseBLIsEMValue.DFCommonElectronsLHMediumIsEMValue.DFCommonElectronsLHTightIsEMValue.DFCommonElectronsECIDS.DFCommonElectronsECIDSResult.ptvarcone20.ptvarcone40.topoetcone20.topoetcone20ptCorrection.ptcone20_TightTTVA_pt500.ptcone20_TightTTVA_pt1000.ptvarcone20_TightTTVA_pt1000.ptvarcone30_TightTTVA_pt500.ptvarcone30_TightTTVA_pt1000.caloClusterLinks.ambiguityLink.truthParticleLink.truthOrigin.truthType.truthPdgId.firstEgMotherTruthType.firstEgMotherTruthOrigin.firstEgMotherTruthParticleLink.firstEgMotherPdgId.ambiguityType.OQ",
"AnalysisPhotons.pt.eta.phi.m.author.OQ.DFCommonPhotonsIsEMLoose.DFCommonPhotonsIsEMTight.DFCommonPhotonsIsEMTightIsEMValue.DFCommonPhotonsIsEMTightPtIncl.DFCommonPhotonsIsEMTightPtInclIsEMValue.DFCommonPhotonsCleaning.DFCommonPhotonsCleaningNoTime.ptcone20.topoetcone20.topoetcone40.topoetcone20ptCorrection.topoetcone40ptCorrection.caloClusterLinks.vertexLinks.ambiguityLink.truthParticleLink.truthOrigin.truthType",
"GSFTrackParticles.chiSquared.phi.d0.theta.qOverP.definingParametersCovMatrix.z0.vz.charge.vertexLink",
"CaloCalTopoClusters.rawE.rawEta.rawPhi.rawM.calE.calEta.calPhi.calM.e_sampl",
"egammaClusters.calE.calEta.calPhi.e_sampl.eta_sampl.ETACALOFRAME.PHICALOFRAME.ETA2CALOFRAME.PHI2CALOFRAME.constituentClusterLinks",
"AnalysisMuons.pt.eta.phi.truthType.truthOrigin.author.muonType.quality.inDetTrackParticleLink.muonSpectrometerTrackParticleLink.combinedTrackParticleLink.InnerDetectorPt.MuonSpectrometerPt.DFCommonGoodMuon.ptcone20.ptcone30.ptcone40.ptvarcone20.ptvarcone30.ptvarcone40.topoetcone20.topoetcone30.topoetcone40.truthParticleLink.charge.extrapolatedMuonSpectrometerTrackParticleLink.allAuthors.ptcone20_TightTTVA_pt1000.ptcone20_TightTTVA_pt500.ptvarcone30_TightTTVA_pt1000.ptvarcone30_TightTTVA_pt500.numberOfPrecisionLayers.combinedTrackOutBoundsPrecisionHits.numberOfPrecisionLayers.numberOfPrecisionHoleLayers.numberOfGoodPrecisionLayers.innerSmallHits.innerLargeHits.middleSmallHits.middleLargeHits.outerSmallHits.outerLargeHits.extendedSmallHits.extendedLargeHits.extendedSmallHoles.isSmallGoodSectors.cscUnspoiledEtaHits.EnergyLoss.energyLossType.momentumBalanceSignificance.scatteringCurvatureSignificance.scatteringNeighbourSignificance",
"CombinedMuonTrackParticles.qOverP.d0.z0.vz.phi.theta.truthOrigin.truthType.definingParametersCovMatrix.numberOfPixelDeadSensors.numberOfPixelHits.numberOfPixelHoles.numberOfSCTDeadSensors.numberOfSCTHits.numberOfSCTHoles.numberOfTRTHits.numberOfTRTOutliers.chiSquared.numberDoF",
"ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType.qOverP.theta.phi",
"MuonSpectrometerTrackParticles.phi.d0.z0.vz.definingParametersCovMatrix.vertexLink.theta.qOverP.truthParticleLink",
"AnalysisTauJets.pt.eta.phi.m.tauTrackLinks.jetLink.charge.isTauFlags.BDTJetScore.BDTEleScore.ptFinalCalib.etaFinalCalib.phiFinalCalib.mFinalCalib.electronLink.EleMatchLikelihoodScore.pt_combined.eta_combined.phi_combined.m_combined.BDTJetScoreSigTrans.BDTEleScoreSigTrans.PanTau_DecayMode.RNNJetScore.RNNJetScoreSigTrans.IsTruthMatched.truthOrigin.truthType.truthParticleLink.truthJetLink",
"AnalysisJets.pt.eta.phi.m.JetConstitScaleMomentum_pt.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_m.NumTrkPt500.SumPtTrkPt500.DetectorEta.Jvt.JVFCorr.JvtRpt.NumTrkPt1000.TrackWidthPt1000.GhostMuonSegmentCount.PartonTruthLabelID.ConeTruthLabelID.HadronConeExclExtendedTruthLabelID.HadronConeExclTruthLabelID.TrueFlavor.DFCommonJets_jetClean_LooseBad.DFCommonJets_jetClean_TightBad.Timing.btagging.btaggingLink.GhostTrack.DFCommonJets_fJvt.DFCommonJets_QGTagger_NTracks.DFCommonJets_QGTagger_TracksWidth.DFCommonJets_QGTagger_TracksC1.PSFrac",
"BTagging_AntiKt4EMPFlow_201903.DL1r_pu.DL1rmu_pu.DL1r_pb.DL1rmu_pb.DL1r_pc.DL1rmu_pc",
"TruthPrimaryVertices.t.x.y.z",
"MET_Core_AnalysisMET.name.mpx.mpy.sumet.source",
"METAssoc_AnalysisMET.",
]
if DerivationFrameworkIsMonteCarlo:
from DerivationFrameworkMCTruth.MCTruthCommon import addTruth3ContentToSlimmerTool
addTruth3ContentToSlimmerTool(PHYSLITESlimmingHelper)
# Extra trigger collections
trigmatching_helper_notau.add_to_slimming(PHYSLITESlimmingHelper)
outputGroup="EXOT12")
if globalflags.DataSource() == 'geant4':
replaceBuggyAntiKt4TruthWZJets(exot12Seq)
#exot12Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT12Kernel_jet",
# SkimmingTools = [EXOT12SkimmingTool_jet])
exot12Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT12Kernel", ThinningTools=thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT12ContentList import *
EXOT12SlimmingHelper = SlimmingHelper("EXOT12SlimmingHelper")
EXOT12SlimmingHelper.SmartCollections = EXOT12SmartContent
EXOT12SlimmingHelper.ExtraVariables = EXOT12ExtraVariablesContent
EXOT12SlimmingHelper.AllVariables = EXOT12AllVariablesContent
EXOT12SlimmingHelper.StaticContent = EXOT12Content
if globalflags.DataSource() == 'geant4':
EXOT12SlimmingHelper.ExtraVariables += EXOT12ExtraVariablesContentTruth
EXOT12SlimmingHelper.AllVariables += EXOT12AllVariablesContentTruth
EXOT12SlimmingHelper.StaticContent += EXOT12ContentTruth
EXOT12SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT12SlimmingHelper.IncludeMuonTriggerContent = True
EXOT12SlimmingHelper.AppendContentToStream(EXOT12Stream)
"MET_Reference_AntiKt4EMPFlow",
"AntiKt4EMTopoJets","AntiKt4LCTopoJets","AntiKt4EMPFlowJets",
"AntiKt4EMPFlowJets_BTagging201810",
"AntiKt4EMPFlowJets_BTagging201903",
"AntiKt4EMTopoJets_BTagging201810",
"BTagging_AntiKt4EMPFlow_201810",
"BTagging_AntiKt4EMPFlow_201903",
"BTagging_AntiKt4EMTopo_201810",
]
JETM2SlimmingHelper.AllVariables = ["MuonTruthParticles", "egammaTruthParticles",
"TruthParticles", "TruthEvents", "TruthVertices",
"MuonSegments",
"Kt4EMTopoOriginEventShape","Kt4LCTopoOriginEventShape","Kt4EMPFlowEventShape",
]
JETM2SlimmingHelper.ExtraVariables = ["Electrons."+NewTrigVars["Electrons"],
"Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus."+NewTrigVars["Muons"],
"TauJets.IsTruthMatched.truthParticleLink.truthJetLink"]
for truthc in [
"TruthMuons",
"TruthElectrons",
"TruthPhotons",
"TruthTaus",
# "TruthNeutrinos"
]:
JETM2SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#"+truthc)
JETM2SlimmingHelper.StaticContent.append("xAOD::TruthParticleAuxContainer#"+truthc+"Aux.")
# Trigger content
JETM2SlimmingHelper.IncludeMuonTriggerContent = True
JETM2SlimmingHelper.IncludeEGammaTriggerContent = True
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += CfgMgr.DerivationFramework__DerivationKernel(
"HIGG3D2Kernel", ThinningTools=thinningTools)
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_HIGG3D2Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_HIGG3D2Stream)
HIGG3D2Stream = MSMgr.NewPoolRootStream(streamName, fileName)
HIGG3D2Stream.AcceptAlgs(["HIGG3D2Kernel"])
augStream = MSMgr.GetStream(streamName)
evtStream = augStream.GetEventStream()
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
HIGG3D2SlimmingHelper = SlimmingHelper("HIGG3D2SlimmingHelper")
from DerivationFrameworkHiggs.HIGG3D2ExtraContent import *
HIGG3D2SlimmingHelper.ExtraVariables = HIGG3D2ExtraVariables
if globalflags.DataSource() == 'geant4':
HIGG3D2SlimmingHelper.AllVariables = HIGG3D2ExtraTruthContainers
HIGG3D2SlimmingHelper.AppendContentToStream(HIGG3D2Stream)
EXOT5SlimmingHelper = SlimmingHelper('EXOT5SlimmingHelper')
EXOT5SlimmingHelper.SmartCollections = [
'Electrons',
'Muons',
'Photons',
'TauJets',
'InDetTrackParticles',
'PrimaryVertices',
'MET_Reference_AntiKt4EMTopo',
'BTagging_AntiKt4EMTopo',
]
EXOT5SlimmingHelper.ExtraVariables = [
'Photons.author.Loose.Tight',
'Electrons.author.Medium.Tight.Loose.charge',
'Muons.ptcone20.ptcone30.ptcone40.etcone20.etcone30.etcone40',
'CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix', # SUSYTools
'ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix', # SUSYTools
'TauJets.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink',
]
EXOT5SlimmingHelper.AllVariables = [
'AntiKt4EMTopoJets',
'AntiKt4TruthJets',
'MET_Truth',
'MET_Track',
'MET_LocHadTopo',
'TruthEvents',
'TruthParticles',
'MuonSegments',
]
if DerivationFrameworkIsMonteCarlo:
EXOT5SlimmingHelper.StaticContent = [
svcMgr += createThinningSvc(svcName="STDM2ThinningSvc", outStreams=[evtStream])
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkSM.STDMExtraContent import *
STDM2SlimmingHelper = SlimmingHelper("STDM2SlimmingHelper")
STDM2SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets",
"BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "InDetTrackParticles",
"PrimaryVertices"
]
STDM2SlimmingHelper.IncludeEGammaTriggerContent = True
STDM2SlimmingHelper.ExtraVariables = ExtraContentElectrons + ExtraContentPhotons + ExtraContentBtag + ExtraContentVtx + ExtraContentTrk + ExtraContentJets + [
"Electrons.maxEcell_time.maxEcell_energy.maxEcell_gain.maxEcell_onlId.maxEcell_x.maxEcell_y.maxEcell_z"
] + [
"Photons.maxEcell_time.maxEcell_energy.maxEcell_gain.maxEcell_onlId.maxEcell_x.maxEcell_y.maxEcell_z"
]
STDM2SlimmingHelper.AllVariables = ExtraContainersJets #+ExtraContainers6Jets #do not exist for now
if globalflags.DataSource() == 'geant4':
STDM2SlimmingHelper.ExtraVariables += ExtraElectronsTruth + ExtraPhotonsTruth
STDM2SlimmingHelper.AllVariables += ExtraContainersTruth + ExtraContainersTruthPhotons #["AntiKt6TruthJets", "BTagging_AntiKt6TruthWZ","AntiKt6TruthWZJets"] do not exist for now
STDM2SlimmingHelper.AppendContentToStream(STDM2Stream)
STDM2Stream.AddItem("xAOD::EventShape#*")
STDM2Stream.AddItem("xAOD::EventShapeAuxInfo#*")
def setup(TOPQname, stream):
DFisMC = (globalflags.DataSource() == 'geant4')
TOPQSlimmingHelper = SlimmingHelper(TOPQname + "SlimmingHelper")
#=====================================================
# ADD BTagging_AntiKt4EMPFlow COLLECTION TO DICTIONARY
#=====================================================
TOPQSlimmingHelper.AppendToDictionary = {
"BTagging_AntiKt4EMPFlow": "xAOD::BTaggingContainer",
"BTagging_AntiKt4EMPFlowAux": "xAOD::BTaggingAuxContainer"
}
#================================
# SMART SLIMMING
#================================
TOPQSlimmingHelper.SmartCollections = []
TOPQSlimmingHelper.SmartCollections += TOPQSmartSlimmingCollections
print("TOPQSlimmingHelper.SmartCollections: ",
TOPQSlimmingHelper.SmartCollections)
#=================================================================
# EXTRA VARIABLES FROM DerivationFrameworkTop.TOPQCommonExtraContent
#=================================================================
TOPQSlimmingHelper.ExtraVariables = []
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMTopoJets
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMPFlowJets
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMPFlow
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMTopo
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotons
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectrons
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuons
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTaus
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTrackJets
TOPQSlimmingHelper.ExtraVariables += Config.GetExtraPromptVariablesForDxAOD(
)
# TOPQSlimmingHelper.ExtraVariables += ["CaloCalTopoClusters.calE.calEta.calPhi.calM.rawM.rawE.rawEta.rawPhi.e_sampl.eta_sampl.etaCalo.phiCalo"]
TOPQSlimmingHelper.ExtraVariables += [
"CaloCalTopoClusters.calPt.calEta.calPhi.calM.calE.CENTER_MAG"
]
if DFisMC:
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotonsTruth
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectronsTruth
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuonsTruth
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTausTruth
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsBTag_HLT_Container
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsJet_EF_Container
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsJet_Split_Container
print("TOPQSlimmingHelper.ExtraVariables: ",
TOPQSlimmingHelper.ExtraVariables)
#================================
# EXTRA COLLECTIONS - user added
#================================
TOPQSlimmingHelper.AllVariables = []
TOPQSlimmingHelper.AllVariables += TOPQExtraContainersStandard
TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTrigger
if DFisMC:
TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTruth
print("TOPQSlimmingHelper.AllVariables: ", TOPQSlimmingHelper.AllVariables)
#================================
# CREATED ON-THE-FLY COLLECTIONS
#================================
TOPQSlimmingHelper.StaticContent = []
TOPQSlimmingHelper.StaticContent += TOPQStaticContent
if DFisMC:
TOPQSlimmingHelper.StaticContent += TOPQStaticContentTruth
print("TOPQSlimmingHelper.StaticContent: ",
TOPQSlimmingHelper.StaticContent)
#================================
# TRIGGER CONTENT
#================================
TOPQSlimmingHelper.IncludeMuonTriggerContent = True
TOPQSlimmingHelper.IncludeEGammaTriggerContent = True
TOPQSlimmingHelper.IncludeJetTriggerContent = True
TOPQSlimmingHelper.IncludeTauTriggerContent = True
TOPQSlimmingHelper.IncludeEtMissTriggerContent = True
#TOPQSlimmingHelper.IncludeJetTauEtMissTriggerContent = True
#TOPQSlimmingHelper.IncludeBJetTriggerContent = True
#TOPQSlimmingHelper.IncludeBPhysTriggerContent = True
#TOPQSlimmingHelper.IncludeMinBiasTriggerContent = True
#====================================
# ADD AntiKT4EMPFlowJets MET OUTPUT
#====================================
# needs to be after where AllVaraibles[] and StaticContent[] are created
# adds METAssoc_XXXX, MET_Core_XXXX, and MET_Reference_XXXX
# uses DerivationFrameworkJetEtMiss.METCommon.py
addMETOutputs(TOPQSlimmingHelper, ["AntiKt4EMPFlow"])
#====================================
# ADD CUSTOM JET OUTPUT (new rel 21 method)
#====================================
addJetOutputs(
TOPQSlimmingHelper,
["SmallR", "LargeR", TOPQname],
[], # smart list
[
"AntiKt3PV0TrackJets",
"AntiKt4PV0TrackJets",
"AntiKt4LCTopoJets",
"AntiKt10LCTopoJets",
"AntiKt10TruthWZJets",
] # veto list
)
#================================
# THIS NEEDS TO BE THE LAST LINE
#================================
TOPQSlimmingHelper.AppendContentToStream(stream)
"AntiKt4TruthJets", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets"
]
#addDefaultTrimmedJets(jetm9Seq,"JETM9")
if jetFlags.useTruth:
replaceBuggyAntiKt4TruthWZJets(jetm9Seq, "JETM9")
for radius in [0.2, 0.3, 0.5, 0.6, 0.7, 0.8]:
if jetFlags.useTruth:
addRscanJets("AntiKt", radius, "Truth", jetm9Seq, "JETM9")
addRscanJets("AntiKt", radius, "TruthWZ", jetm9Seq, "JETM9")
addRscanJets("AntiKt", radius, "LCTopo", jetm9Seq, "JETM9")
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
JETM9SlimmingHelper = SlimmingHelper("JETM9SlimmingHelper")
JETM9SlimmingHelper.SmartCollections = [
"AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "PrimaryVertices"
]
JETM9SlimmingHelper.AllVariables = ["TruthEvents", "MuonSegments"]
JETM9SlimmingHelper.ExtraVariables = ["TruthVertices.z"]
# Trigger content
JETM9SlimmingHelper.IncludeJetTriggerContent = True
# Add the jet containers to the stream
addJetOutputs(JETM9SlimmingHelper, ["JETM9"])
JETM9SlimmingHelper.AppendContentToStream(JETM9Stream)
JETM9Stream.RemoveItem("xAOD::TrigNavigation#*")
JETM9Stream.RemoveItem("xAOD::TrigNavigationAuxInfo#*")
"EXOT3Kernel",
# ThinningTools = [EXOT3TPThinningTool,EXOT3MuonTPThinningTool,EXOT3ElectronTPThinningTool, EXOT3PhotonTPThinningTool]
ThinningTools=thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT3ContentList import *
EXOT3SlimmingHelper = SlimmingHelper("EXOT3SlimmingHelper")
# Containers to be smart slimmed, see https://svnweb.cern.ch/trac/atlasoff/browser/PhysicsAnalysis
# /DerivationFramework/DerivationFrameworkExamples/trunk/share/SlimmingExample.py#L38
EXOT3SlimmingHelper.SmartCollections = EXOT3SmartContent
EXOT3SlimmingHelper.ExtraVariables = EXOT3ExtraVariables
# Keep all variables for containers which we don't want to smart slim and were
# not created in the derivation
EXOT3SlimmingHelper.AllVariables = EXOT3AllVariablesContent
# Add jet collections created by derivation job
EXOT3SlimmingHelper.StaticContent = EXOT3StaticContent
# (Dont) Add jet triger content
#EXOT3SlimmingHelper.IncludeJetTauEtMissTriggerContent = True
EXOT3SlimmingHelper.IncludeJetTriggerContent = True
EXOT3SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT3SlimmingHelper.AppendContentToStream(EXOT3Stream)
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
HIGG3D1SlimmingHelper = SlimmingHelper("HIGG3D1SlimmingHelper")
from DerivationFrameworkHiggs.HIGG3D1ExtraContent import *
HIGG3D1SlimmingHelper.SmartCollections = [
"Electrons", "Muons", "Photons", "MET_Reference_AntiKt4LCTopo",
"MET_Reference_AntiKt4EMTopo", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets",
"BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo",
"BTagging_AntiKt4Track", "BTagging_AntiKt2Track", "InDetTrackParticles",
"PrimaryVertices"
]
HIGG3D1SlimmingHelper.ExtraVariables = HIGG3D1ExtraVariables
HIGG3D1SlimmingHelper.AllVariables = HIGG3D1ExtraContainers
if globalflags.DataSource() == 'geant4':
HIGG3D1SlimmingHelper.AllVariables += HIGG3D1ExtraTruthContainers
HIGG3D1SlimmingHelper.ExtraVariables += HIGG3D1ExtraTruthVariables
# Add Trigger content
HIGG3D1SlimmingHelper.IncludeMuonTriggerContent = True
HIGG3D1SlimmingHelper.IncludeEGammaTriggerContent = True
# Add MET to output stream
#addMETOutputs(HIGG3D1SlimmingHelper)
HIGG3D1SlimmingHelper.AppendContentToStream(HIGG3D1Stream)
#====================================================================
# CONTENT LIST
#====================================================================
# This might be the kind of set-up one would have for a muon based analysis
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
SUSY4SlimmingHelper = SlimmingHelper("SUSY4SlimmingHelper")
SUSY4SlimmingHelper.SmartCollections = ["Electrons", "Photons", "Muons", "TauJets", "AntiKt4EMTopoJets", "MET_Reference_AntiKt4EMTopo", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices", "AntiKt4EMPFlowJets"]
SUSY4SlimmingHelper.AllVariables = ["TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth", "InDetTrackParticles", "METAssoc_AntiKt4EMPFlow", "MET_Core_AntiKt4EMPFlow", "MET_Reference_AntiKt4EMPFlow"]
SUSY4SlimmingHelper.ExtraVariables = ["BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant",
"Electrons.Loose.Medium.Tight",
"Muons.ptcone30.ptcone20.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag",
"Photons.author.Loose.Tight",
"AntiKt4EMTopoJets.JvtJvfcorr.JvtRpt.Jvt.TruthLabelID.BchCorrCell.NumTrkPt1000.NumTrkPt500.SumPtTrkPt500.SumPtTrkPt1000.TrackWidthPt1000.TrackWidthPt500.DFCommonJets_Calib_pt.DFCommonJets_Calib_eta.DFCommonJets_Calib_phi",
"AntiKt4EMPFlowJets.JvtJvfcorr.JvtRpt.Jvt.TruthLabelID.BchCorrCell.NumTrkPt1000.NumTrkPt500.SumPtTrkPt500.SumPtTrkPt1000.TrackWidthPt1000.TrackWidthPt500",
"GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix",
"CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"TauJets.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink",
"MuonTruthParticles.barcode.decayVtxLink.e.m.pdgId.prodVtxLink.px.py.pz.recoMuonLink.status.truthOrigin.truthType",
"AntiKt4TruthJets.eta.m.phi.pt.TruthLabelID",
"InDetTrackParticles.truthOrigin.truthType.truthMatchProbability.TrkIso*"]
SUSY4SlimmingHelper.IncludeMuonTriggerContent = False
SUSY4SlimmingHelper.IncludeEGammaTriggerContent = False
SUSY4SlimmingHelper.IncludeBPhysTriggerContent = False
SUSY4SlimmingHelper.IncludeJetTriggerContent = False
SUSY4SlimmingHelper.IncludeTauTriggerContent = False
SUSY4SlimmingHelper.IncludeEtMissTriggerContent = False
SUSY4SlimmingHelper.IncludeBJetTriggerContent = False
SUSY5SlimmingHelper.SmartCollections = [
"PhotonCollection", "ElectronCollection", "Muons", "TauRecContainer",
"AntiKt4LCTopoJets", "MET_RefFinal", "BTagging_AntiKt4LCTopo",
"InDetTrackParticles", "PrimaryVertices"
]
SUSY5SlimmingHelper.AllVariables = [
"AntiKt4TruthJets", "BTagging_AntiKt4Truth", "MuonTruthParticle",
"TruthParticle", "TruthEvent", "TruthVertex", "MET_Truth",
"BTagging_AntiKt4LCTopo"
]
SUSY5SlimmingHelper.ExtraVariables = [
"BTagging_AntiKt4LCTopo.MV1_discriminant",
"PhotonCollection.author.Loose.Tight",
"ElectronCollection.author.Loose.Medium.Tight.truthOrigin.truthType.charge.ptcone20.etcone30",
"Muons.ptcone30.ptcone20.etcone30.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag",
"AntiKt4LCTopoJets.ActiveArea4vec_pt.ActiveArea4vec_eta.ActiveArea4vec_phi.ActiveArea4vec_m.EnergyPerSampling.NumTrkPt1000.TrackWidthPt1000.GhostMuonSegment.NumTrkPt500",
"GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix",
"CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"TauRecContainer.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType"
]
SUSY5SlimmingHelper.AppendContentToStream(SUSY5Stream)
# Add the MET containers to the stream
addMETOutputs(SUSY5Stream, ["SUSY"])
for item in OutputJets["LargeR"]:
if "Trim" not in item:
continue
SUSY5Stream.AddItem("xAOD::JetContainer_v1#" + item)
SUSY5Stream.AddItem("xAOD::JetAuxContainer_v1#" + item + "Aux.")
'SecVtx_ConvVtxContainer_Electrons' : 'xAOD::VertexContainer',
'SecVtx_ConvVtxContainer_ElectronsAux': 'xAOD::VertexAuxContainer',
'SecVtxContainer_Muons' : 'xAOD::VertexContainer',
'SecVtxContainer_MuonsAux' : 'xAOD::VertexAuxContainer',
'RefittedPriVtx' : 'xAOD::VertexContainer',
'RefittedPriVtxAux' : 'xAOD::VertexAuxContainer',
}
# Add extra variables to collections
MUON5SlimmingHelper.ExtraVariables = ["Muons.clusterLink.allAuthors.charge.extrapolatedMuonSpectrometerTrackParticleLink."
"scatteringCurvatureSignificance.scatteringNeighbourSignificance",
"Electrons.author.charge.topoetcone30.ptvarcone30",
"AntiKt4EMTopoJets.JetEMScaleMomentum_pt.JetEMScaleMomentum_eta.JetEMScaleMomentum_phi.JetEMScaleMomentum_m."
"ConeTruthLabelID.PartonTruthLabelID.SumPtTrkPt1000.Jvt.JvtJvfcorr.JvtRpt.TileStatus",
"GSFTrackParticles.z0.vz.definingParametersCovMatrix",
"CombinedMuonTrackParticles.z0.vz.definingParametersCovMatrix",
"ExtrapolatedMuonTrackParticles.z0.vz.definingParametersCovMatrix",
"PrimaryVertices.x.y.z.trackParticleLinks.vertexType.neutralParticleLinks",
"TauJets.pantau_CellBasedInput_isPanTauCandidate.pantau_CellBasedInput_DecayMode."
"ptPanTauCellBased.etaPanTauCellBased.phiPanTauCellBased.mPanTauCellBased."
"pantau_CellBasedInput_BDTValue_1p0n_vs_1p1n.pantau_CellBasedInput_BDTValue_1p1n_vs_1pXn."
"pantau_CellBasedInput_BDTValue_3p0n_vs_3pXn.ele_match_lhscore.ele_olr_pass."
"electronLink.ptDetectorAxis.etaDetectorAxis.phiDetectorAxis.mDetectorAxis",
"TauNeutralParticleFlowObjects.pt.eta.phi.m.e.rapidity.bdtPi0Score",
"TauChargedParticleFlowObjects.pt.eta.phi.m"]
MUON5SlimmingHelper.ExtraVariables += LepTagConfig.GetExtraPromptVariablesForDxAOD(onlyBDT=False)
MUON5SlimmingHelper.ExtraVariables += LepTagConfig.GetExtraImprovedPromptVariablesForDxAOD()
MUON5SlimmingHelper.ExtraVariables += ElectronsCPDetailedContent
MUON5SlimmingHelper.AllVariables = ["egammaClusters", "CaloCalTopoClusters", "MuonClusterCollection", "TopoClusterIsoCentralEventShape", "TopoClusterIsoForwardEventShape", "AntiKt4PV0TrackJets", "BTagging_AntiKt4Track", "InDetTrackParticles","GSFConversionVertices","GSFTrackParticles"]
if is_MC:
MUON5SlimmingHelper.AllVariables += ["TruthParticles", "TruthEvents", "TruthVertices", "AntiKt4TruthJets"]
SkimmingTools=skimmingTools,
ThinningTools=thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
HIGG3D3SlimmingHelper = SlimmingHelper("HIGG3D3SlimmingHelper")
from DerivationFrameworkHiggs.HIGG3D3ExtraContent import *
#HIGG3D3SlimmingHelper.UserContent = HIGG3D3Content
HIGG3D3SlimmingHelper.SmartCollections = [
"Electrons", "Muons", "AntiKt4EMTopoJets", "InDetTrackParticles",
"BTagging_AntiKt4EMTopo", "PrimaryVertices", "MET_Reference_AntiKt4EMTopo"
]
HIGG3D3SlimmingHelper.ExtraVariables = HIGG3D3ExtraVariables
HIGG3D3SlimmingHelper.AllVariables = HIGG3D3ExtraContainers
if globalflags.DataSource() == 'geant4':
HIGG3D3SlimmingHelper.AllVariables += HIGG3D3ExtraTruthContainers
HIGG3D3SlimmingHelper.ExtraVariables += HIGG3D3ExtraTruthVariables
# Add Trigger content
HIGG3D3SlimmingHelper.IncludeMuonTriggerContent = True
HIGG3D3SlimmingHelper.IncludeEGammaTriggerContent = True
# Add MET to output stream
#addMETOutputs(HIGG3D3SlimmingHelper)
HIGG3D3SlimmingHelper.AppendContentToStream(HIGG3D3Stream)
"SUSY8GenFilt", SimBarcodeOffset=DerivationFrameworkSimBarcodeOffset)
AugmentationTools.append(ToolSvc.SUSY8GenFilt)
#==============================================================================
# Augment after skim
#==============================================================================
SeqSUSY8 += CfgMgr.DerivationFramework__DerivationKernel(
"SUSY8KernelAug",
AugmentationTools=AugmentationTools,
ThinningTools=thinningTools,
)
#====================================================================
# CONTENT LIST
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkSUSY.SUSY8ContentList import *
SUSY8SlimmingHelper = SlimmingHelper("SUSY8SlimmingHelper")
SUSY8SlimmingHelper.SmartCollections = SUSY8SmartContent
SUSY8SlimmingHelper.ExtraVariables = SUSY8ExtraVariables
SUSY8SlimmingHelper.AllVariables = SUSY8AllVariablesContent
SUSY8SlimmingHelper.IncludeMuonTriggerContent = True
SUSY8SlimmingHelper.IncludeJetTriggerContent = True
SUSY8SlimmingHelper.IncludeEtMissTriggerContent = True
SUSY8SlimmingHelper.AppendContentToStream(SUSY8Stream)
SUSY5SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "Muons", "TauJets", "AntiKt4EMTopoJets",
"MET_Reference_AntiKt4EMTopo", "BTagging_AntiKt4EMTopo",
"InDetTrackParticles", "PrimaryVertices"
]
SUSY5SlimmingHelper.AllVariables = [
"TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth"
]
SUSY5SlimmingHelper.ExtraVariables = [
"BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant",
"Muons.ptcone30.ptcone20.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag",
"Photons.author.Loose.Tight",
"AntiKt4EMTopoJets.NumTrkPt1000.TrackWidthPt1000.NumTrkPt500.DFCommonJets_Calib_pt.DFCommonJets_Calib_eta.DFCommonJets_Calib_phi",
"GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix",
"CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"TauJets.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink",
"MuonTruthParticles.barcode.decayVtxLink.e.m.pdgId.prodVtxLink.px.py.pz.recoMuonLink.status.truthOrigin.truthType",
"AntiKt4TruthJets.eta.m.phi.pt.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T.TruthLabelID.ConeTruthLabelID.PartonTruthLabelID",
"Electrons.bkgTruthType.bkgTruthOrigin", "MET_Track",
"AntiKt3PV0TrackJets.eta.m.phi.pt.btagging.btaggingLink",
"BTagging_AntiKt3Track.MV2c20_discriminant"
]
SUSY5SlimmingHelper.IncludeMuonTriggerContent = True
SUSY5SlimmingHelper.IncludeEGammaTriggerContent = True
SUSY5SlimmingHelper.IncludeBPhysTriggerContent = False
SUSY5SlimmingHelper.IncludeJetTriggerContent = True
SUSY5SlimmingHelper.IncludeTauTriggerContent = True
SUSY5SlimmingHelper.IncludeEtMissTriggerContent = True
SUSY5SlimmingHelper.IncludeBJetTriggerContent = True
from DerivationFrameworkHiggs.HIGG2D2ExtraContent import *
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
HIGG2D2SlimmingHelper = SlimmingHelper("HIGG2D2SlimmingHelper")
HIGG2D2SlimmingHelper.SmartCollections = ["Electrons",
"Photons",
"Muons",
"TauJets",
"MET_Reference_AntiKt4EMTopo",
"MET_Reference_AntiKt4LCTopo",
"AntiKt4EMTopoJets",
"AntiKt4LCTopoJets",
"BTagging_AntiKt4EMTopo",
"BTagging_AntiKt4LCTopo",
"InDetTrackParticles",
"PrimaryVertices"]
HIGG2D2SlimmingHelper.ExtraVariables = HIGG2D2ExtraContent
HIGG2D2SlimmingHelper.AllVariables = HIGG2D2ExtraContainers
if globalflags.DataSource()=='geant4':
HIGG2D2SlimmingHelper.ExtraVariables += HIGG2D2ExtraContentTruth
HIGG2D2SlimmingHelper.AllVariables += HIGG2D2ExtraContainersTruth
# Add MET_RefFinalFix
addMETOutputs(HIGG2D2SlimmingHelper,["AntiKt4LCTopo","Track"])
HIGG2D2SlimmingHelper.IncludeMuonTriggerContent = True
HIGG2D2SlimmingHelper.IncludeEGammaTriggerContent = True
HIGG2D2SlimmingHelper.AppendContentToStream(HIGG2D2Stream)
"BTagging_AntiKt4EMPFlow_201903", "BTagging_AntiKt4EMTopo_201810"
]
JETM7SlimmingHelper.AllVariables = [ # "CaloCalTopoClusters",
"MuonTruthParticles",
"egammaTruthParticles",
"TruthParticles",
"TruthEvents",
"TruthVertices",
"MuonSegments",
"Kt4EMTopoOriginEventShape",
"Kt4LCTopoOriginEventShape",
"Kt4EMPFlowEventShape",
]
JETM7SlimmingHelper.ExtraVariables = [
"Electrons." + NewTrigVars["Electrons"],
"Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus."
+ NewTrigVars["Muons"]
]
for truthc in [
"TruthMuons",
"TruthElectrons",
"TruthPhotons",
"TruthTaus",
# "TruthNeutrinos"
]:
JETM7SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#" +
truthc)
JETM7SlimmingHelper.StaticContent.append(
"xAOD::TruthParticleAuxContainer#" + truthc + "Aux.")
# Trigger content
def setup(TOPQname, stream):
DFisMC = (globalflags.DataSource() == 'geant4')
TOPQSlimmingHelper = SlimmingHelper(TOPQname + "SlimmingHelper")
#=====================================================
# ADD BTagging_AntiKt4EMPFlow COLLECTION TO DICTIONARY
#=====================================================
TOPQSlimmingHelper.AppendToDictionary = {
"BTagging_AntiKt4EMPFlow": "xAOD::BTaggingContainer",
"BTagging_AntiKt4EMPFlowAux": "xAOD::BTaggingAuxContainer"
}
#================================
# SMART SLIMMING
#================================
TOPQSlimmingHelper.SmartCollections = []
TOPQSlimmingHelper.SmartCollections += TOPQSmartSlimmingCollections
print "TOPQSlimmingHelper.SmartCollections: ", TOPQSlimmingHelper.SmartCollections
#=================================================================
# EXTRA VARIABLES FROM DerivationFrameworkTop.TOPQCommonExtraContent
#=================================================================
TOPQSlimmingHelper.ExtraVariables = []
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMTopoJets
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMPFlowJets
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMPFlow
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMTopo
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotons
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectrons
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuons
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTaus
if DFisMC:
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotonsTruth
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectronsTruth
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuonsTruth
TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTausTruth
print "TOPQSlimmingHelper.ExtraVariables: ", TOPQSlimmingHelper.ExtraVariables
#================================
# EXTRA COLLECTIONS - user added
#================================
TOPQSlimmingHelper.AllVariables = []
TOPQSlimmingHelper.AllVariables += TOPQExtraContainersStandard
TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTrigger
if DFisMC:
TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTruth
print "TOPQSlimmingHelper.AllVariables: ", TOPQSlimmingHelper.AllVariables
#================================
# CREATED ON-THE-FLY COLLECTIONS
#================================
TOPQSlimmingHelper.StaticContent = []
TOPQSlimmingHelper.StaticContent += TOPQStaticContent
if DFisMC:
TOPQSlimmingHelper.StaticContent += TOPQStaticContentTruth
print "TOPQSlimmingHelper.StaticContent: ", TOPQSlimmingHelper.StaticContent
#================================
# TRIGGER CONTENT
#================================
TOPQSlimmingHelper.IncludeMuonTriggerContent = True
TOPQSlimmingHelper.IncludeEGammaTriggerContent = True
TOPQSlimmingHelper.IncludeJetTriggerContent = True
TOPQSlimmingHelper.IncludeTauTriggerContent = True
TOPQSlimmingHelper.IncludeEtMissTriggerContent = True
#TOPQSlimmingHelper.IncludeJetTauEtMissTriggerContent = True
#TOPQSlimmingHelper.IncludeBJetTriggerContent = True
#TOPQSlimmingHelper.IncludeBPhysTriggerContent = True
#TOPQSlimmingHelper.IncludeMinBiasTriggerContent = True
#====================================
# ADD AntiKT4EMPFlowJets MET OUTPUT
#====================================
# needs to be after where AllVaraibles[] and StaticContent[] are created
# adds METAssoc_XXXX, MET_Core_XXXX, and MET_Reference_XXXX
# uses DerivationFrameworkJetEtMiss.METCommon.py
addMETOutputs(TOPQSlimmingHelper, ["AntiKt4EMPFlow"])
#================================
# THIS NEEDS TO BE THE LAST LINE
#================================
TOPQSlimmingHelper.AppendContentToStream(stream)
SUSY2SlimmingHelper = SlimmingHelper("SUSY2SlimmingHelper")
SUSY2SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo",
"AntiKt4EMTopoJets", "BTagging_AntiKt4EMTopo", "InDetTrackParticles",
"PrimaryVertices"
]
SUSY2SlimmingHelper.AllVariables = [
"TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth"
]
SUSY2SlimmingHelper.ExtraVariables = [
"BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant",
"Muons.ptcone30.ptcone20.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag",
"AntiKt4EMTopoJets.NumTrkPt1000.TrackWidthPt1000.NumTrkPt500",
"GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"InDetTrackParticles.truthOrigin.truthType",
"CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType",
"TauJets.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink",
"MuonTruthParticles.barcode.decayVtxLink.e.m.pdgId.prodVtxLink.px.py.pz.recoMuonLink.status.truthOrigin.truthType",
"AntiKt4TruthJets.eta.m.phi.pt.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T.TruthLabelID.ConeTruthLabelID.PartonTruthLabelID",
"Electrons.bkgMotherPdgId.bkgTruthOrigin"
]
SUSY2SlimmingHelper.IncludeMuonTriggerContent = True
SUSY2SlimmingHelper.IncludeEGammaTriggerContent = True
#SUSY2SlimmingHelper.IncludeBPhysTriggerContent = True
#SUSY2SlimmingHelper.IncludeJetTauEtMissTriggerContent = True
if globalflags.DataSource() == 'geant4':
SUSY2SlimmingHelper.StaticContent += [
"xAOD::TruthParticleContainer#TruthTaus",
"xAOD::TruthParticleContainer#TruthTaus",
"xAOD::TruthParticleAuxContainer#TruthTausAux.",
"xAOD::TruthParticleContainer#TruthPhotons",
"xAOD::TruthParticleAuxContainer#TruthPhotonsAux.",
"xAOD::TruthParticleContainer#TruthNeutrinos",
"xAOD::TruthParticleAuxContainer#TruthNeutrinosAux."
]
TAUP1SlimmingHelper.IncludeMuonTriggerContent = True
TAUP1SlimmingHelper.IncludeTauTriggerContent = True
TAUP1SlimmingHelper.IncludeEGammaTriggerContent = True
TAUP1SlimmingHelper.IncludeEtMissTriggerContent = True
TAUP1SlimmingHelper.IncludeJetTriggerContent = True
TAUP1SlimmingHelper.IncludeBJetTriggerContent = True
TAUP1SlimmingHelper.ExtraVariables = ExtraContentTAUP1
TAUP1SlimmingHelper.AllVariables = ExtraContainersTAUP1
if globalflags.DataSource() == "geant4":
TAUP1SlimmingHelper.ExtraVariables += ExtraContentTruthTAUP1
TAUP1SlimmingHelper.AllVariables += ExtraContainersTruthTAUP1
TAUP1SlimmingHelper.AppendContentToStream(TAUP1Stream)
TAUP1Stream.AddItem("xAOD::EventShape#*")
TAUP1Stream.AddItem("xAOD::EventShapeAuxInfo#*")
# Add MET_RefFinalFix
# JRC TEMPORARILY COMMENTED
# addMETOutputs(TAUP1Stream)
exot6Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT6Kernel",
ThinningTools=[
EXOT6TPThinningTool, EXOT6MuonTPThinningTool,
EXOT6ElectronTPThinningTool, EXOT6PhotonTPThinningTool,
EXOT6PhotonCCThinningTool
])
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT6ContentList import EXOT6SmartContent, EXOT6AllVariablesContent, EXOT6ExtraVariables, EXOT6UnslimmedContent
EXOT6SlimmingHelper = SlimmingHelper("EXOT6SlimmingHelper")
# Containers to be smart slimmed, see https://svnweb.cern.ch/trac/atlasoff/browser/PhysicsAnalysis
# /DerivationFramework/DerivationFrameworkExamples/trunk/share/SlimmingExample.py#L38
EXOT6SlimmingHelper.SmartCollections = EXOT6SmartContent
# Keep all variables for containers which we don't want to smart slim and were
# not created in the derivation
EXOT6SlimmingHelper.AllVariables = EXOT6AllVariablesContent
# Adding extra missing variables from the smart slimming
EXOT6SlimmingHelper.ExtraVariables = EXOT6ExtraVariables
EXOT6SlimmingHelper.StaticContent = EXOT6UnslimmedContent
EXOT6SlimmingHelper.AppendContentToStream(EXOT6Stream)
#=======================================
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot10Seq
exot10Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT10Kernel_skim", SkimmingTools=[EXOT10SkimmingTool])
exot10Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT10Kernel",
AugmentationTools=[EXOT10_MaxCellDecoratorTool],
ThinningTools=[EXOT10PhotonCCThinningTool, EXOT10ElectronCCThinningTool])
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT10ContentList import *
EXOT10SlimmingHelper = SlimmingHelper("EXOT10SlimmingHelper")
EXOT10SlimmingHelper.StaticContent = EXOT10Content
EXOT10SlimmingHelper.SmartCollections = EXOT10SmartContent
EXOT10SlimmingHelper.AllVariables = EXOT10AllVariablesContent
EXOT10SlimmingHelper.UserContent = EXOT10Content
EXOT10SlimmingHelper.ExtraVariables = EXOT10ExtraVariables
EXOT10SlimmingHelper.IncludeEGammaTriggerContent = True
addMETOutputs(EXOT10SlimmingHelper)
EXOT10SlimmingHelper.AppendContentToStream(EXOT10Stream)
EXOT8SlimmingHelper.SmartCollections = [
"AntiKt4EMTopoJets",
"AntiKt4LCTopoJets",
"BTagging_AntiKt4LCTopo",
"BTagging_AntiKt4EMTopo",
"BTagging_AntiKt2Track",
"BTagging_AntiKt3Track",
"BTagging_AntiKt4Track",
"InDetTrackParticles",
"PrimaryVertices",
"Electrons",
"Muons",
"MET_Reference_AntiKt4LCTopo",
"MET_Reference_AntiKt4EMTopo",
]
EXOT8SlimmingHelper.ExtraVariables = ["Electrons.charge", "Muons.charge"]
EXOT8SlimmingHelper.AllVariables = [
"TruthParticles",
"TruthEvents",
"TruthVertices",
"AntiKt10LCTopoJets",
"AntiKt2PV0TrackJets",
"AntiKt3PV0TrackJets",
"AntiKt4PV0TrackJets",
#"BTagging_AntiKt4TruthWZ", #removed in new tags of MC15
"AntiKt4TruthJets",
"AntiKt10TruthJets",
"CombinedMuonTrackParticles",
"ExtrapolatedMuonTrackParticles",
]
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += CfgMgr.DerivationFramework__DerivationKernel("JETM2Kernel",
SkimmingTools = [JETM2SkimmingTool],
ThinningTools = thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
JETM2SlimmingHelper = SlimmingHelper("JETM2SlimmingHelper")
JETM2SlimmingHelper.SmartCollections = ["Electrons", "Photons", "Muons", "TauJets",
"InDetTrackParticles", "PrimaryVertices"]
JETM2SlimmingHelper.AllVariables = ["BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters",
"MuonTruthParticles", "egammaTruthParticles",
"TruthParticles", "TruthEvents", "TruthVertices",
"MuonSegments"
]
JETM2SlimmingHelper.ExtraVariables = ["Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus"]
# Trigger content
JETM2SlimmingHelper.IncludeMuonTriggerContent = True
JETM2SlimmingHelper.IncludeEGammaTriggerContent = True
# Add the jet containers to the stream
addJetOutputs(JETM2SlimmingHelper,["SmallR"])
# Add the MET containers to the stream
addMETOutputs(JETM2SlimmingHelper,["Diagnostic","AntiKt4LCTopo","AntiKt4EMPFlow","Track"])
JETM2SlimmingHelper.AppendContentToStream(JETM2Stream)
