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)
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)
skimmingTools.append(EXOT1SkimmingTool)
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__FilterCombinationOR
EXOT1ORSkimmingTool = DerivationFramework__FilterCombinationOR(
name="EXOT1ORSkimmingTool", FilterList=skimmingTools)
ToolSvc += EXOT1ORSkimmingTool
printfunc(EXOT1SkimmingTool)
#=======================================
# CREATE THE DERIVATION KERNEL ALGORITHM
#=======================================
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot1Seq
exot1Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT1Kernel_skim", SkimmingTools=[EXOT1ORSkimmingTool])
exot1Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT1Kernel", ThinningTools=thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
# Currently using same list as EXOT1
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT1ContentList import *
EXOT1SlimmingHelper = SlimmingHelper("EXOT1SlimmingHelper")
EXOT1SlimmingHelper.StaticContent = EXOT1Content
EXOT1SlimmingHelper.SmartCollections = EXOT1SmartContent
EXOT1SlimmingHelper.AllVariables = EXOT1AllVariablesContent
EXOT1SlimmingHelper.AppendContentToStream(EXOT1Stream)
#=======================================
# 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)
name="JETM7Kernel",
SkimmingTools=[JETM7SkimmingTool],
ThinningTools=thinningTools)
#====================================================================
# 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"])
"AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets",
"AntiKt10UFOCSSKBottomUpSoftDropBeta100Zcut5Jets",
"AntiKt10UFOCSSKRecursiveSoftDropBeta100Zcut5NinfJets",
"AntiKtVR30Rmax4Rmin02TrackJets_BTagging201810",
"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"
]:
#====================================================================
# CONTENT LIST
#====================================================================
# About SmartCollections and AllVariables:
# If you want to use CP tools on a container, you MUST include that container as a SmartCollection.
# You are then free to include it as AllVariables in addition, to pick up variables not needed by the CP tools but needed by your analysis.
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
SUSY1SlimmingHelper = SlimmingHelper("SUSY1SlimmingHelper")
SUSY1SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "MET_Reference_AntiKt4EMTopo", "Muons", "TauJets",
"BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices",
"BTagging_AntiKt4Track"
]
SUSY1SlimmingHelper.AllVariables = [
"TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth",
"AntiKt4PV0TrackJets", "MET_Track"
]
SUSY1SlimmingHelper.ExtraVariables = [
"BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant",
"Muons.etcone30.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", # TODO: .DFCommonJets_Jvt",
"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.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T.TruthLabelID.ConeTruthLabelID.PartonTruthLabelID",
]
if globalflags.DataSource() == 'geant4':
SUSY1SlimmingHelper.StaticContent += [
EGAM2SlimmingHelper = SlimmingHelper("EGAM2SlimmingHelper")
from DerivationFrameworkEGamma.EGAM2ExtraContent import *
EGAM2SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo",
"AntiKt4EMTopoJets", "BTagging_AntiKt4EMTopo", "InDetTrackParticles",
"PrimaryVertices"
]
# muons, tau, MET, b-tagging could be switched off if not needed and use too much space
# Add egamma trigger objects
EGAM2SlimmingHelper.IncludeEGammaTriggerContent = True
# Extra variables
EGAM2SlimmingHelper.ExtraVariables = ExtraContentAll
EGAM2SlimmingHelper.AllVariables = ExtraContainersElectrons
EGAM2SlimmingHelper.AllVariables += ExtraContainersTrigger
if globalflags.DataSource() != 'geant4':
EGAM2SlimmingHelper.AllVariables += ExtraContainersTriggerDataOnly
if globalflags.DataSource() == 'geant4':
EGAM2SlimmingHelper.ExtraVariables += ExtraContentAllTruth
EGAM2SlimmingHelper.AllVariables += ExtraContainersTruth
for tool in EGAM2_ClusterEnergyPerLayerDecorators:
EGAM2SlimmingHelper.ExtraVariables.extend(
getClusterEnergyPerLayerDecorations(tool))
# This line must come after we have finished configuring EGAM2SlimmingHelper
EGAM2SlimmingHelper.AppendContentToStream(EGAM2Stream)
#if jetFlags.useTruth:
# includePreTools=True
# DerivationFrameworkJob += addTrimmedJets("AntiKt", 1.0, "Truth", rclus=0.2, ptfrac=0.05, includePreTools=True)
#if includePreTools==False:
# DerivationFrameworkJob += addTrimmedJets("AntiKt", 1.0, "LCTopo", rclus=0.2, ptfrac=0.05, includePreTools=True)
#else:
# DerivationFrameworkJob += addTrimmedJets("AntiKt", 1.0, "LCTopo", rclus=0.2, ptfrac=0.05)
#====================================================================
# 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
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
JETM3SlimmingHelper = SlimmingHelper("JETM3SlimmingHelper")
JETM3SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles",
"PrimaryVertices", "MET_Reference_AntiKt4EMTopo",
"MET_Reference_AntiKt4LCTopo", "MET_Reference_AntiKt4EMPFlow",
"AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "AntiKt4EMPFlowJets"
]
JETM3SlimmingHelper.AllVariables = [
"BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters",
"MuonTruthParticles", "egammaTruthParticles", "TruthParticles",
"TruthEvents", "TruthVertices", "MuonSegments",
"JetETMissChargedParticleFlowObjects",
"JetETMissNeutralParticleFlowObjects"
]
JETM3SlimmingHelper.ExtraVariables = [
"Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus"
]
for truthc in [
"TruthMuons",
"TruthElectrons",
"TruthPhotons",
"TruthTaus",
# "TruthNeutrinos"
]:
JETM3SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#" +
truthc)
#edmlist(set())
#====================================================================
# SET UP STREAM
#====================================================================
# SKIMMING
# The base name (DAOD_TRIG4 here) must match the string in
# DerivationFrameworkProdFlags (in DerivationFrameworkCore)
streamName = derivationFlags.WriteDAOD_TRIG4Stream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_TRIG4Stream )
TRIG4Stream = MSMgr.NewPoolRootStream( streamName, fileName )
# Only events that pass the filters listed below are written out.
# Name must match that of the kernel above
# AcceptAlgs = logical OR of filters
# RequireAlgs = logical AND of filters
TRIG4Stream.AcceptAlgs(["TRIG4Kernel"])
# Thinning
from DerivationFrameworkTrigger.TrigThinningHelper import ThinningHelper
TRIG4ThinningHelper = ThinningHelper("TRIG4ThinningHelper",edmlist,'slimming')
TRIG4ThinningHelper.AppendToStream(TRIG4Stream)
#====================================================================
# CONTENT LIST
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
TRIG4SlimmingHelper = SlimmingHelper("TRIG4SlimmingHelper")
##Smart Slimming
TRIG4SlimmingHelper.SmartCollections = []
TRIG4SlimmingHelper.AllVariables = edmlist
TRIG4SlimmingHelper.AppendContentToStream(TRIG4Stream)
"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)
# container variables. Thus BTagging_AntiKt4EMTopo is needed in SmartCollections as well as AllVariables
FTAG4SlimmingHelper.SmartCollections = ["Electrons","Muons",
"MET_Reference_AntiKt4EMTopo",
"AntiKt4EMTopoJets",
"BTagging_AntiKt4EMTopo"]
FTAG4SlimmingHelper.AllVariables = ["AntiKt3PV0TrackJets",
"AntiKt2PV0TrackJets",
"AntiKt4TruthJets",
"BTagging_AntiKt4EMTopo",
"BTagging_AntiKt2Track",
"BTagging_AntiKt3Track",
"BTagging_AntiKt4EMTopoJFVtx",
"BTagging_AntiKt2TrackJFVtx",
"BTagging_AntiKt3TrackJFVtx",
"BTagging_AntiKt4EMTopoSecVtx",
"BTagging_AntiKt2TrackSecVtx",
"BTagging_AntiKt3TrackSecVtx",
"TruthVertices",
"TruthParticles",
"TruthEvents",
"MET_Truth",
"MET_TruthRegions",
"InDetTrackParticles",
"PrimaryVertices"
]
from DerivationFrameworkCore.AntiKt4EMTopoJetsCPContent import AntiKt4EMTopoJetsCPContent
FTAG4SlimmingHelper.ExtraVariables.append(AntiKt4EMTopoJetsCPContent[1].replace("AntiKt4EMTopoJetsAux","AntiKt10LCTopoJets"))
FTAG4SlimmingHelper.IncludeMuonTriggerContent = True
#=======================================
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += CfgMgr.DerivationFramework__DerivationKernel( name = "JETM4Kernel",
SkimmingTools = [JETM4SkimmingTool],
ThinningTools = thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
JETM4SlimmingHelper = SlimmingHelper("JETM4SlimmingHelper")
JETM4SlimmingHelper.SmartCollections = ["Electrons", "Photons", "Muons", "TauJets",
"InDetTrackParticles", "PrimaryVertices"]
JETM4SlimmingHelper.AllVariables = ["BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters",
"MuonTruthParticles", "egammaTruthParticles",
"TruthParticles", "TruthEvents", "TruthVertices",
"MuonSegments"
]
#JETM4SlimmingHelper.ExtraVariables = []
# Trigger content
JETM4SlimmingHelper.IncludeEGammaTriggerContent = True
# Add the jet containers to the stream
addJetOutputs(JETM4SlimmingHelper,["SmallR"])
# Add the MET containers to the stream
addMETOutputs(JETM4SlimmingHelper,["Diagnostic","AntiKt4LCTopo","AntiKt4EMPFlow","Track"])
JETM4SlimmingHelper.AppendContentToStream(JETM4Stream)
"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" )
STDM5Stream.AddItem( "xAOD::TruthParticleAuxContainer#TruthElectronsAux." )
#====================================================================
streamName = derivationFlags.WriteDAOD_JETM9Stream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_JETM9Stream )
JETM9Stream = MSMgr.NewPoolRootStream( streamName, fileName )
JETM9Stream.AcceptAlgs(["JETM9Kernel"])
# for thinning
from AthenaServices.Configurables import ThinningSvc, createThinningSvc
augStream = MSMgr.GetStream( streamName )
evtStream = augStream.GetEventStream()
svcMgr += createThinningSvc( svcName="JETM9ThinningSvc", outStreams=[evtStream] )
#====================================================================
# 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", "TruthVertices",
"MuonSegments"
]
#JETM9SlimmingHelper.ExtraVariables = []
# 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#*")
"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"]
MUON5SlimmingHelper.ExtraVariables += ["Electrons.truthOrigin.truthType.truthParticleLink.bkgTruthType.bkgTruthOrigin."
"bkgTruthParticleLink.bkgMotherPdgId.deltaPhi1",
"MuonTruthParticles.truthOrigin.truthType.truthParticleLink",
"InDetTrackParticles.truthOrigin.truthType.truthParticleLink",
"CombinedMuonTrackParticles.truthOrigin.truthType.truthParticleLink",
"TauJets.IsTruthMatched.truthParticleLink.truthJetLink"]
MUON5SlimmingHelper.StaticContent = ["xAOD::TruthParticleContainer#TruthMuons",
"xAOD::TruthParticleAuxContainer#TruthMuonsAux.",
"xAOD::TruthParticleContainer#TruthElectrons",
"xAOD::TruthParticleAuxContainer#TruthElectronsAux.",
#====================================================================
# CONTENT LIST
#====================================================================
# This might be the kind of set-up one would have for a muon based analysis
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
SUSY5SlimmingHelper = SlimmingHelper("SUSY5SlimmingHelper")
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)
"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)
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)
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)
# container variables. Thus BTagging_AntiKt4EMTopo is needed in SmartCollections as well as AllVariables
FTAG3SlimmingHelper.SmartCollections = [
"Electrons",
"Muons",
"InDetTrackParticles",
"PrimaryVertices",
"AntiKt4EMTopoJets",
"AntiKt4EMPFlowJets",
"MET_Reference_AntiKt4EMTopo",
]
FTAG3SlimmingHelper.AllVariables = [
"AntiKt10TruthWZJets", "BTagging_AntiKt2Track",
"BTagging_AntiKt2TrackJFVtx", "BTagging_AntiKt10TruthWZ",
"BTagging_AntiKt10TruthWZJFVtx", "TruthEvents", "TruthVertices",
"TruthParticles", "TruthHFWithDecayParticles", "TruthHFWithDecayVertices",
"TruthMuons", "MuonTruthParticles", "MET_Truth", "MET_TruthRegions",
"HLT_xAOD__TrackParticleContainer_MuonEFInfo_CombTrackParticles",
"HLT_xAOD__TrackParticleContainer_MuonEFInfo_CombTrackParticles_FullScan",
"HLT_xAOD__MuonContainer_MuonEFInfo"
]
FTAG3SlimmingHelper.ExtraVariables += [
"BTagging_AntiKt2TrackSecVtx.-vxTrackAtVertex",
"BTagging_AntiKt10TruthSecVtx.-vxTrackAtVertex"
]
if BTaggingFlags.Do2019Retraining:
FTAG3SlimmingHelper.SmartCollections += \
["AntiKt4EMTopoJets_BTagging201810"]
FTAG3SlimmingHelper.AllVariables += \
# container variables. Thus BTagging_AntiKt4EMTopo is needed in SmartCollections as well as AllVariables
FTAG4SlimmingHelper.SmartCollections = [
"Electrons", "Muons", "InDetTrackParticles", "AntiKt4EMTopoJets",
"AntiKt4EMPFlowJets", "MET_Reference_AntiKt4EMPFlow",
"MET_Reference_AntiKt4EMTopo"
]
FTAG4SlimmingHelper.AllVariables = [
"BTagging_AntiKtVR30Rmax4Rmin02Track_201810",
"BTagging_AntiKtVR30Rmax4Rmin02Track_201810JFVtx",
"BTagging_AntiKtVR30Rmax4Rmin02Track_201903",
"BTagging_AntiKtVR30Rmax4Rmin02Track_201903JFVtx",
"BTagging_AntiKt10LCTopoTrimmedPtFrac5SmallR20ExCoM2Sub",
"BTagging_AntiKt10LCTopoTrimmedPtFrac5SmallR20ExCoM2SubJFVtx",
"BTagging_AntiKt2Track",
"BTagging_AntiKt2TrackJFVtx",
"TruthVertices",
"TruthEvents",
"TruthHFWithDecayParticles",
"TruthHFWithDecayVertices",
"MET_Truth",
"MET_TruthRegions",
"TruthParticles",
]
FTAG4SlimmingHelper.ExtraVariables += [
AntiKt4EMTopoJetsCPContent[1].replace("AntiKt4EMTopoJetsAux",
"AntiKt10LCTopoJets"),
"InDetTrackParticles.truthMatchProbability.x.y.z.vx.vy.vz",
"InDetTrackParticles.numberOfInnermostPixelLayerSplitHits.numberOfNextToInnermostPixelLayerSplitHits.numberOfNextToInnermostPixelLayerSharedHits",
"InDetTrackParticles.numberOfPixelSplitHits.numberOfInnermostPixelLayerSharedHits.numberOfContribPixelLayers.hitPattern.radiusOfFirstHit",
JETM7SlimmingHelper = SlimmingHelper("JETM7SlimmingHelper")
JETM7SlimmingHelper.SmartCollections = [
"Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles",
"PrimaryVertices", "MET_Reference_AntiKt4EMTopo",
"MET_Reference_AntiKt4LCTopo", "MET_Reference_AntiKt4EMPFlow",
"AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "AntiKt4EMPFlowJets",
"AntiKt4EMPFlowJets_BTagging201810", "AntiKt4EMPFlowJets_BTagging201903",
"AntiKt4EMTopoJets_BTagging201810", "BTagging_AntiKt4EMPFlow_201810",
"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",
"BTagging_AntiKt4EMTopo"]
FTAG5SlimmingHelper.AllVariables = ["AntiKt3PV0TrackJets",
"AntiKt2PV0TrackJets",
"AntiKt4PV0TrackJets",
"AntiKt4TruthJets",
"BTagging_AntiKt4EMTopo",
"BTagging_AntiKt2Track",
"BTagging_AntiKt3Track",
"BTagging_AntiKt4EMTopoJFVtx",
"BTagging_AntiKt2TrackJFVtx",
"BTagging_AntiKt3TrackJFVtx",
"BTagging_AntiKt4EMTopoSecVtx",
"BTagging_AntiKt2TrackSecVtx",
"BTagging_AntiKt3TrackSecVtx",
"TruthVertices",
"TruthParticles",
"TruthEvents",
"MET_Truth",
"MET_TruthRegions",
"InDetTrackParticles",
"PrimaryVertices",
"AntiKtVR50Rmax4Rmin0TrackJets",
"BTagging_AntiKtVR50Rmax4Rmin0Track",
"AntiKt10LCTopoJets",
"AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
"AntiKt10TrackJets",
"CaloCalTopoClusters",
]
from DerivationFrameworkJetEtMiss.AntiKt4EMTopoJetsCPContent import AntiKt4EMTopoJetsCPContent
ThinningTools=thinningTools,
)
#====================================================================
# CONTENT LIST
#====================================================================
# This might be the kind of set-up one would have for a muon based analysis
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
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
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)
# 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
HIGG5D2SlimmingHelper.IncludeJetTriggerContent = True
#=======================================
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += exot16Seq
exot16Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT16Kernel_skim", SkimmingTools=[EXOT16StringSkimmingTool])
exot16Seq += CfgMgr.DerivationFramework__DerivationKernel(
"EXOT16Kernel", ThinningTools=thinningTools)
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
# Currently using same list as EXOT16
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT16ContentList import *
EXOT16SlimmingHelper = SlimmingHelper("EXOT16SlimmingHelper")
EXOT16SlimmingHelper.AppendToDictionary = EXOT16Content
EXOT16SlimmingHelper.SmartCollections = EXOT16SmartCollections
EXOT16SlimmingHelper.AllVariables = EXOT16AllVariables
EXOT16SlimmingHelper.ExtraVariables = EXOT16ExtraVariables
addMETOutputs(EXOT16SlimmingHelper, ["MET_Track"], ["AntiKt4EMTopo"])
EXOT16SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT16SlimmingHelper.IncludeMuonTriggerContent = True
EXOT16SlimmingHelper.AppendContentToStream(EXOT16Stream)
if is_MC:
HIGG6D1SlimmingHelper.StaticContent = [
"xAOD::TruthParticleContainer#TruthMuons",
"xAOD::TruthParticleAuxContainer#TruthMuonsAux.",
"xAOD::TruthParticleContainer#TruthElectrons",
"xAOD::TruthParticleAuxContainer#TruthElectronsAux.",
"xAOD::TruthParticleContainer#TruthPhotons",
"xAOD::TruthParticleAuxContainer#TruthPhotonsAux.",
"xAOD::TruthParticleContainer#TruthNeutrinos",
"xAOD::TruthParticleAuxContainer#TruthNeutrinosAux.",
"xAOD::TruthParticleContainer#TruthTaus",
"xAOD::TruthParticleAuxContainer#TruthTausAux."
]
HIGG6D1SlimmingHelper.AllVariables = [
"TruthParticles", "TruthEvents", "MET_Truth", "METMap_Truth",
"TruthVertices", "AntiKt4TruthJets"
]
HIGG6D1SlimmingHelper.ExtraVariables += [
"AntiKt4LCTopoJets.PartonTruthLabelID.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T"
]
HIGG6D1SlimmingHelper.ExtraVariables += [
"AntiKt4EMTopoJets.PartonTruthLabelID.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T"
]
HIGG6D1SlimmingHelper.ExtraVariables += [
"Electrons.truthOrigin.truthType.truthParticleLink",
"MuonTruthParticles.truthOrigin.truthType"
]
HIGG6D1SlimmingHelper.ExtraVariables += [
"TauJets.IsTruthMatched.truthParticleLink"
]
"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)
"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",
]
EXOT8SlimmingHelper.StaticContent = [
"xAOD::JetContainer#AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets",
"xAOD::JetAuxContainer#AntiKt10LCTopoTrimmedPtFrac5SmallR20JetsAux.",
]
if globalflags.DataSource() == "geant4":
EXOT8SlimmingHelper.StaticContent += [
"xAOD::JetContainer#AntiKt10TruthTrimmedPtFrac5SmallR20Jets",
"xAOD::JetAuxContainer#AntiKt10TruthTrimmedPtFrac5SmallR20JetsAux.",
DerivationFrameworkJob += exot0Seq
exot0Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT0Kernel_skim", SkimmingTools = [EXOT0SkimmingTool])
exot0Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT0Kernel", ThinningTools = thinningTools)
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_EXOT0Stream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_EXOT0Stream )
EXOT0Stream = MSMgr.NewPoolRootStream( streamName, fileName )
EXOT0Stream.AcceptAlgs(["EXOT0Kernel"])
# Thinning
from AthenaServices.Configurables import ThinningSvc, createThinningSvc
augStream = MSMgr.GetStream( streamName )
evtStream = augStream.GetEventStream()
svcMgr += createThinningSvc( svcName="EXOT0ThinningSvc", outStreams=[evtStream] )
#====================================================================
# Add the containers to the output stream - slimming done here
#====================================================================
from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
from DerivationFrameworkExotics.EXOT0ContentList import *
EXOT0SlimmingHelper = SlimmingHelper("EXOT0SlimmingHelper")
EXOT0SlimmingHelper.StaticContent = EXOT0Content
EXOT0SlimmingHelper.AllVariables = EXOT0AllVariables
EXOT0SlimmingHelper.SmartCollections = EXOT0SmartCollections
EXOT0SlimmingHelper.IncludeEGammaTriggerContent = True
EXOT0SlimmingHelper.IncludeMuonTriggerContent = True
EXOT0SlimmingHelper.AppendContentToStream(EXOT0Stream)
