def _make_thinning_tools(self, stream):
stream_name = self.stream_name
self.thinning_helper = ThinningHelper(stream_name+"ThinningHelper")
self.thinning_helper.AppendToStream(stream)
self.thinning_tools = []
tool = DerivationFramework__TrackParticleThinning(
stream_name + "TPThinningTool",
StreamName = stream_name,
SelectionString = "InDetTrackParticles.pt > {0}*GeV".format(
self.track_threshold) )
if self.track_sel is not None:
tool.SelectionString += " || InDetTrackParticles.{0.common_prefix}{0.track_sel}".format(self)
self.thinning_tools.append(tool)
self.thinning_tools += [
DerivationFramework__MuonTrackParticleThinning(
stream_name + "MuonTPThinningTool",
StreamName = stream_name,
MuonKey = "Muons",
InDetTrackParticlesKey = "InDetTrackParticles"),
DerivationFramework__EgammaTrackParticleThinning(
stream_name + "ElectronTPThinningTool",
StreamName = stream_name,
SGKey = "Electrons",
InDetTrackParticlesKey = "InDetTrackParticles"),
DerivationFramework__EgammaTrackParticleThinning(
stream_name + "PhotonTPThinningTool",
StreamName = stream_name,
SGKey = "Photons",
InDetTrackParticlesKey = "InDetTrackParticles"),
DerivationFramework__TauTrackParticleThinning(
stream_name + "TauTPThinningTool",
StreamName = stream_name,
TauKey = "TauJets",
InDetTrackParticlesKey = "InDetTrackParticles")
]
global ToolSvc
ToolSvc += self.thinning_tools
ElectronTriggers=electronTriggers,MuonTriggers=muonTriggers)
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_JETM2Stream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_JETM2Stream )
JETM2Stream = MSMgr.NewPoolRootStream( streamName, fileName )
JETM2Stream.AcceptAlgs(["JETM2Kernel"])
#=======================================
# ESTABLISH THE THINNING HELPER
#=======================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
JETM2ThinningHelper = ThinningHelper( "JETM2ThinningHelper" )
#JETM2ThinningHelper.TriggerChains = orstr.join(electronTriggers+muonTriggers)
JETM2ThinningHelper.AppendToStream( JETM2Stream )
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
# TrackParticles associated with Muons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
JETM2MuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(name = "JETM2MuonTPThinningTool",
StreamName = streamName,
MuonKey = "Muons",
InDetTrackParticlesKey = "InDetTrackParticles")
ToolSvc += JETM2MuonTPThinningTool
STDM5Stream.AcceptAlgs(["STDM5Kernel"])
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools=[]
#=====================
# TRIGGER NAV THINNING
#=====================
# Establish the thinning helper
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
STDM5ThinningHelper = ThinningHelper( "STDM5ThinningHelper" )
#trigger navigation content
STDM5ThinningHelper.TriggerChains = 'HLT_e.*|HLT_2e.*|HLT_mu.*|HLT_2mu.*'
STDM5ThinningHelper.AppendToStream( STDM5Stream )
#=====================
# TRACK THINNING
#=====================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
STDM5JetTPThinningTool = DerivationFramework__JetTrackParticleThinning( name = "STDM5JetTPThinningTool",
ThinningService = STDM5ThinningHelper.ThinningSvc(),
JetKey = "AntiKt4EMTopoJets",
from DerivationFrameworkJetEtMiss.ExtendedJetCommon import *
from DerivationFrameworkEGamma.EGammaCommon import *
from DerivationFrameworkMuons.MuonsCommon import *
from DerivationFrameworkTau.TauTruthCommon import *
from DerivationFrameworkInDet.InDetCommon import *
from DerivationFrameworkJetEtMiss.METCommon import *
### Set up stream
streamName = derivationFlags.WriteDAOD_SUSY2Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_SUSY2Stream)
SUSY2Stream = MSMgr.NewPoolRootStream(streamName, fileName)
SUSY2Stream.AcceptAlgs(["SUSY2KernelSkim"])
### Init
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
SUSY2ThinningHelper = ThinningHelper("SUSY2ThinningHelper")
thinningTools = []
AugmentationTools = []
# stream-specific sequence for on-the-fly jet building
SeqSUSY2 = CfgMgr.AthSequencer("SeqSUSY2")
DerivationFrameworkJob += SeqSUSY2
#====================================================================
# Trigger navigation thinning
#====================================================================
from DerivationFrameworkSUSY.SUSY2TriggerList import triggersNavThin
SUSY2ThinningHelper.TriggerChains = '|'.join(triggersNavThin)
SUSY2ThinningHelper.AppendToStream(SUSY2Stream)
"BTagging_AntiKt4EMTopo_201810SecVtx": "xAOD::VertexContainer",
"BTagging_AntiKt4EMTopo_201810SecVtxAux": "xAOD::VertexAuxContainer",
"BTagging_AntiKt2Track": "xAOD::BTaggingContainer",
"BTagging_AntiKt2TrackAux": "xAOD::BTaggingAuxContainer",
"BTagging_AntiKt2TrackJFVtx": "xAOD::BTagVertexContainer",
"BTagging_AntiKt2TrackJFVtxAux": "xAOD::BTagVertexAuxContainer",
"BTagging_AntiKt2TrackSecVtx": "xAOD::VertexContainer",
"BTagging_AntiKt2TrackSecVtxAux": "xAOD::VertexAuxContainer",
"BTagging_AntiKt10TruthWZ": "xAOD::BTaggingContainer",
"BTagging_AntiKt10TruthWZAux": "xAOD::BTaggingAuxContainer",
"BTagging_AntiKt10TruthWZJFVtx": "xAOD::BTagVertexContainer",
"BTagging_AntiKt10TruthWZJFVtxAux": "xAOD::BTagVertexAuxContainer",
"BTagging_AntiKt10TruthWZSecVtx": "xAOD::VertexContainer",
"BTagging_AntiKt10TruthWZSecVtxAux": "xAOD::VertexAuxContainer",
}
#----------------------------------------------------------------------
addJetOutputs(FTAG3SlimmingHelper, ["FTAG3"])
FTAG3SlimmingHelper.IncludeMuonTriggerContent = True
FTAG3SlimmingHelper.IncludeEGammaTriggerContent = False
FTAG3SlimmingHelper.IncludeJetTriggerContent = True
FTAG3SlimmingHelper.IncludeEtMissTriggerContent = False
FTAG3SlimmingHelper.IncludeBJetTriggerContent = True
#FTAG3 TrigNav Thinning
FTAG3ThinningHelper = ThinningHelper("FTAG3ThinningHelper")
FTAG3ThinningHelper.TriggerChains = 'HLT_mu*_j.*_dr05|HLT_mu*_j.*_bperf_split_dr05_dz02|HLT_mu*_j.*_boffperf_split_dr05_dz02|HLT_mu*_j.*_gsc.*_dr05_dz02'
FTAG3ThinningHelper.AppendToStream(FTAG3Stream)
FTAG3SlimmingHelper.AppendContentToStream(FTAG3Stream)
from DerivationFrameworkFlavourTag.HbbCommon import *
from TriggerMenu.api.TriggerAPI import TriggerAPI
from TriggerMenu.api.TriggerEnums import TriggerPeriod, TriggerType
from DerivationFrameworkTrigger.TriggerMatchingHelper import TriggerMatchingHelper
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_PHYSLITEStream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_PHYSLITEStream )
PHYSLITEStream = MSMgr.NewPoolRootStream( streamName, fileName )
PHYSLITEStream.AcceptAlgs(["PHYSLITEKernel"])
### Thinning and augmentation tools lists
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
PHYSLITEThinningHelper = ThinningHelper( "PHYSLITEThinningHelper" )
PHYSLITEThinningHelper.AppendToStream( PHYSLITEStream )
thinningTools = []
AugmentationTools = []
# Special sequence
SeqPHYSLITE = CfgMgr.AthSequencer("SeqPHYSLITE")
#====================================================================
# TRUTH CONTENT
#====================================================================
if DerivationFrameworkIsMonteCarlo:
from DerivationFrameworkMCTruth.MCTruthCommon import addStandardTruthContents,addPVCollection
addStandardTruthContents(SeqPHYSLITE)
addPVCollection(SeqPHYSLITE)
from DerivationFrameworkMCTruth.HFHadronsCommon import *
"HLT_e5_lhvloose", "HLT_e5_lhvloose_nod0", "HLT_e10_lhvloose_L1EM7",
"HLT_e12_lhvloose_L1EM10VH", "HLT_e12_lhvloose_nod0_L1EM10VH",
"HLT_e15_lhvloose_L1EM7", "HLT_e15_lhvloose_L1EM13VH",
"HLT_e15_lhvloose_nod0_L1EM13VH", "HLT_e17_lhvloose",
"HLT_e17_lhvloose_nod0", "HLT_e20_lhvloose_L1EM12", "HLT_e20_lhvloose",
"HLT_e20_lhvloose_nod0", "HLT_e24_lhvloose_nod0_L1EM18VH",
"HLT_e24_lhvloose_nod0_L1EM20VH"
]
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__TriggerSkimmingTool
HIGG3D3TrigSkimmingTool = DerivationFramework__TriggerSkimmingTool(
name="HIGG3D3TrigSkimmingTool", TriggerListOR=TriggerChains)
ToolSvc += HIGG3D3TrigSkimmingTool
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
HIGG3D3ThinningHelper = ThinningHelper("HIGG3D3ThinningHelper")
HIGG3D3ThinningHelper.TriggerChains = " | ".join(TriggerChains)
HIGG3D3ThinningHelper.AppendToStream(HIGG3D3Stream)
#================
# THINNING
#================
thinningTools = []
# MET/Jet tracks
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
HIGG3D3JetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
name="HIGG3D3JetTPThinningTool",
StreamName=streamName,
JetKey="AntiKt4EMTopoJets",
InDetTrackParticlesKey="InDetTrackParticles")
# Accept the most selective kernel (last one in sequence; later in derivation)
TOPQ3Stream.AcceptAlgs(["TOPQ3Kernel"])
#====================================================================
# PDF Weight Metadata
#====================================================================
if DFisMC:
from DerivationFrameworkCore.WeightMetadata import *
#====================================================================
# TRIGGER NAVIGATION THINNING
#====================================================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
import DerivationFrameworkTop.TOPQCommonThinning
TOPQ3ThinningHelper = ThinningHelper("TOPQ3ThinningHelper")
TOPQ3ThinningHelper.TriggerChains = DerivationFrameworkTop.TOPQCommonThinning.TOPQTriggerChains(
)
TOPQ3ThinningHelper.AppendToStream(TOPQ3Stream)
#====================================================================
# SKIMMING TOOLS
#====================================================================
import DerivationFrameworkTop.TOPQCommonSelection
skimmingTools_lep = DerivationFrameworkTop.TOPQCommonSelection.setup_lep(
'TOPQ3', ToolSvc)
skimmingTools_jet = DerivationFrameworkTop.TOPQCommonSelection.setup_jet(
'TOPQ3', ToolSvc)
#====================================================================
ToolSvc += DecoTool
#====================================================================
# SET UP STREAM
#====================================================================
MUON3Stream = MSMgr.NewPoolRootStream(
derivationFlags.WriteDAOD_MUON3Stream.StreamName,
buildFileName(derivationFlags.WriteDAOD_MUON3Stream))
MUON3Stream.AcceptAlgs(["MUON3Kernel"])
### trigger thinning
triggerList = ['HLT_mu14', 'HLT_2mu4', 'HLT_2mu6_bJpsimumu']
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
MUON3ThinningHelper = ThinningHelper("MUON3ThinningHelper")
MUON3ThinningHelper.TriggerChains = '|'.join(triggerList)
MUON3ThinningHelper.AppendToStream(MUON3Stream)
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_MUON3Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_MUON3Stream)
MUON3Stream = MSMgr.NewPoolRootStream(streamName, fileName)
MUON3Stream.AcceptAlgs(["MUON3Kernel"])
# Special lines for thinning
# Thinning service name must match the one passed to the thinning tools
augStream = MSMgr.GetStream(streamName)
evtStream = augStream.GetEventStream()
ToolSvc += JETM3SkimmingTool
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_JETM3Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_JETM3Stream)
JETM3Stream = MSMgr.NewPoolRootStream(streamName, fileName)
JETM3Stream.AcceptAlgs(["JETM3Kernel"])
#=======================================
# ESTABLISH THE THINNING HELPER
#=======================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
JETM3ThinningHelper = ThinningHelper("JETM3ThinningHelper")
JETM3ThinningHelper.TriggerChains = orstr.join(electronTriggers + muonTriggers)
JETM3ThinningHelper.AppendToStream(JETM3Stream)
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
# TrackParticles associated with Muons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
JETM3MuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name="JETM3MuonTPThinningTool",
ThinningService=JETM3ThinningHelper.ThinningSvc(),
MuonKey="Muons",
InDetTrackParticlesKey="InDetTrackParticles")
from DerivationFrameworkJetEtMiss.ExtendedJetCommon import *
from DerivationFrameworkEGamma.EGammaCommon import *
from DerivationFrameworkMuons.MuonsCommon import *
from DerivationFrameworkTau.TauTruthCommon import *
from DerivationFrameworkInDet.InDetCommon import *
from DerivationFrameworkJetEtMiss.METCommon import *
### Set up stream
streamName = derivationFlags.WriteDAOD_SUSY4Stream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_SUSY4Stream )
SUSY4Stream = MSMgr.NewPoolRootStream( streamName, fileName )
SUSY4Stream.AcceptAlgs(["SUSY4KernelSkim"])
### Init
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
SUSY4ThinningHelper = ThinningHelper( "SUSY4ThinningHelper" )
thinningTools = []
AugmentationTools = []
# stream-specific sequence for on-the-fly jet building
SeqSUSY4 = CfgMgr.AthSequencer("SeqSUSY4")
DerivationFrameworkJob += SeqSUSY4
#====================================================================
# Trigger navigation thinning
#====================================================================
# To be added.
SUSY4ThinningHelper.AppendToStream( SUSY4Stream )
from DerivationFrameworkFlavourTag.HbbCommon import *
from TriggerMenu.api.TriggerAPI import TriggerAPI
from TriggerMenu.api.TriggerEnums import TriggerPeriod, TriggerType
from DerivationFrameworkTrigger.TriggerMatchingHelper import TriggerMatchingHelper
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_PHYS2Stream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_PHYS2Stream )
PHYS2Stream = MSMgr.NewPoolRootStream( streamName, fileName )
PHYS2Stream.AcceptAlgs(["PHYS2Kernel"])
### Thinning and augmentation tools lists
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
PHYS2ThinningHelper = ThinningHelper( "PHYS2ThinningHelper" )
PHYS2ThinningHelper.AppendToStream( PHYS2Stream )
thinningTools = []
AugmentationTools = []
# Special sequence
SeqPHYS2 = CfgMgr.AthSequencer("SeqPHYS2")
DerivationFrameworkJob += SeqPHYS2
#====================================================================
# Truth collections
#====================================================================
if (DerivationFrameworkIsMonteCarlo):
from DerivationFrameworkMCTruth.MCTruthCommon import addStandardTruthContents,addMiniTruthCollectionLinks,addHFAndDownstreamParticles,addPVCollection
import DerivationFrameworkHiggs.TruthCategories
# Add charm quark collection
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_TEST10Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_TEST10Stream)
TEST10Stream = MSMgr.NewPoolRootStream(streamName, fileName)
TEST10Stream.AcceptAlgs(["TEST10Kernel"])
#====================================================================
# THINNING
#====================================================================
# Establish the thinning helper (which will set up the services behind the scenes)
# Plug in the required triggers
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
TEST10ThinningHelper = ThinningHelper("TEST10ThinningHelper")
TEST10ThinningHelper.TriggerChains = 'HLT_e.*|HLT_g.*|HLT_mu.*|HLT_tau.*'
TEST10ThinningHelper.AppendToStream(TEST10Stream)
#====================================================================
# CREATE THE DERIVATION KERNEL ALGORITHM AND PASS THE ABOVE TOOLS
#====================================================================
# The name of the kernel (TEST10Kernel in this case) must be unique to this derivation
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
DerivationFrameworkJob += CfgMgr.DerivationFramework__DerivationKernel(
"TEST10Kernel")
TEST10Stream.AcceptAlgs(["TEST10Kernel"])
#====================================================================
# CONTENT LIST
STDM3Stream = MSMgr.NewPoolRootStream(streamName, fileName)
STDM3Stream.AcceptAlgs(["STDM3Kernel"])
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
#=====================
# TRIGGER NAV THINNING
#=====================
# Establish the thinning helper
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
STDM3ThinningHelper = ThinningHelper("STDM3ThinningHelper")
#trigger navigation content
STDM3ThinningHelper.TriggerChains = 'HLT_e.*|HLT_2e.*|HLT_mu.*|HLT_2mu.*'
STDM3ThinningHelper.AppendToStream(STDM3Stream)
#=====================
# TRACK THINNING
#=====================
# Tracks associated with Jets
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
STDM3JetTPThinningTool = DerivationFramework__JetTrackParticleThinning(
name="STDM3JetTPThinningTool",
StreamName=streamName,
JetKey="AntiKt4EMTopoJets",
ToolSvc += JETM4SkimmingTool
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_JETM4Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_JETM4Stream)
JETM4Stream = MSMgr.NewPoolRootStream(streamName, fileName)
JETM4Stream.AcceptAlgs(["JETM4Kernel"])
#=======================================
# ESTABLISH THE THINNING HELPER
#=======================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
JETM4ThinningHelper = ThinningHelper("JETM4ThinningHelper")
JETM4ThinningHelper.TriggerChains = triggers
JETM4ThinningHelper.AppendToStream(JETM4Stream)
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
# TrackParticles associated with Muons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
JETM4MuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name="JETM4MuonTPThinningTool",
ThinningService=JETM4ThinningHelper.ThinningSvc(),
MuonKey="Muons",
InDetTrackParticlesKey="InDetTrackParticles")
ToolSvc += MUON5BkgElectronClassificationTool
augmentationTools.append(MUON5BkgElectronClassificationTool)
printfunc ("BkgElectronClassificationTool: ", MUON5BkgElectronClassificationTool)
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
# Establish the thinning helper (which will set up the services behind the scenes)
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
MUON5ThinningHelper = ThinningHelper("MUON5ThinningHelper")
# Trigger Thinning Tool
MUON5ThinningHelper.TriggerChains = 'HLT_e.*|HLT_mu.*|HLT_2e.*|HLT_2mu.*|HLT_tau.*'
MUON5ThinningHelper.AppendToStream(MUON5Stream)
#====================================================================
# TRACK THINNING
#====================================================================
do_track_thinning = False
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TrackParticleThinning
MUON5TrackThinningTool = DerivationFramework__TrackParticleThinning(name = "MUON5TrackThinningTool",
StreamName = streamName,
SelectionString = "abs(DFCommonInDetTrackZ0AtPV) < 10.0",
STDM7Stream = MSMgr.NewPoolRootStream(streamName, fileName)
STDM7Stream.AcceptAlgs(["STDM7Kernel"])
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
#=====================
# TRIGGER NAV THINNING
#=====================
# Establish the thinning helper
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
STDM7ThinningHelper = ThinningHelper("STDM7ThinningHelper")
#trigger navigation content
STDM7ThinningHelper.TriggerChains = 'HLT_e.*|HLT_2e.*|HLT_mu.*|HLT_2mu.*'
STDM7ThinningHelper.AppendToStream(STDM7Stream)
#=====================
# TRACK THINNING
#=====================
# removed for full tracking info...
# Truth leptons and their ancestors and descendants + final-state hadrons
truth_cond_boson = "((abs(TruthParticles.pdgId) == 23) || (abs(TruthParticles.pdgId) == 24))"
truth_cond_lepton = "((abs(TruthParticles.pdgId) >= 11) && (abs(TruthParticles.pdgId) <= 14) &&(TruthParticles.pt > 1*GeV) && (TruthParticles.status ==1) && (TruthParticles.barcode<200000))"
photonthinningexpr = "(TruthPhotons.classifierParticleOrigin != 42) && !(TruthPhotons.classifierParticleOrigin >= 23 && TruthPhotons.classifierParticleOrigin <= 35)"
from DerivationFrameworkJetEtMiss.JetCommon import *
from DerivationFrameworkJetEtMiss.ExtendedJetCommon import *
from DerivationFrameworkEGamma.EGammaCommon import *
from DerivationFrameworkMuons.MuonsCommon import *
from DerivationFrameworkInDet.InDetCommon import *
from DerivationFrameworkJetEtMiss.METCommon import *
### Set up stream
streamName = derivationFlags.WriteDAOD_SUSY8Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_SUSY8Stream)
SUSY8Stream = MSMgr.NewPoolRootStream(streamName, fileName)
SUSY8Stream.AcceptAlgs(["SUSY8KernelSkim"])
### Init
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
SUSY8ThinningHelper = ThinningHelper("SUSY8ThinningHelper")
thinningTools = []
AugmentationTools = []
# stream-specific sequence for on-the-fly jet building
SeqSUSY8 = CfgMgr.AthSequencer("SeqSUSY8")
DerivationFrameworkJob += SeqSUSY8
#====================================================================
# Trigger navigation thinning
#====================================================================
from DerivationFrameworkSUSY.SUSY8TriggerList import *
jetMETTriggers = SUSY8JetMETTriggers
muonTriggers = SUSY8MuonTriggers
dimuonTriggers = SUSY8DimuonTriggers
from DerivationFrameworkJetEtMiss.ExtendedJetCommon import *
from DerivationFrameworkEGamma.EGammaCommon import *
from DerivationFrameworkMuons.MuonsCommon import *
from DerivationFrameworkTau.TauTruthCommon import *
from DerivationFrameworkInDet.InDetCommon import *
from DerivationFrameworkJetEtMiss.METCommon import *
### Set up stream
streamName = derivationFlags.WriteDAOD_SUSY1Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_SUSY1Stream)
SUSY1Stream = MSMgr.NewPoolRootStream(streamName, fileName)
SUSY1Stream.AcceptAlgs(["SUSY1KernelSkim"])
### Init
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
SUSY1ThinningHelper = ThinningHelper("SUSY1ThinningHelper")
thinningTools = []
AugmentationTools = []
# stream-specific sequence for on-the-fly jet building
SeqSUSY1 = CfgMgr.AthSequencer("SeqSUSY1")
DerivationFrameworkJob += SeqSUSY1
#====================================================================
# Trigger navigation thinning
#====================================================================
triggerRegEx = [ # FIXME: disable unless tested
".?j.*", "xe.*", "e.*", "mu.*", "tau.*", "g.*"
]
SUSY1ThinningHelper.TriggerChains = 'HLT_(' + ' | '.join(triggerRegEx) + ')'
SUSY1ThinningHelper.AppendToStream(SUSY1Stream)
]
StaticContent += ["xAOD::VertexContainer#SoftBVrtClusterTool_Loose_Vertices"]
StaticContent += [
"xAOD::VertexAuxContainer#SoftBVrtClusterTool_Loose_VerticesAux." +
excludedVertexAuxData
]
StaticContent += [
"xAOD::VertexContainer#VrtSecInclusive_SoftBtagCandidateVertices"
]
StaticContent += [
"xAOD::VertexAuxContainer#VrtSecInclusive_SoftBtagCandidateVerticesAux."
]
FTAG4SlimmingHelper.StaticContent = StaticContent
#----------------------------------------------------------------------
addJetOutputs(FTAG4SlimmingHelper, ["FTAG4"])
FTAG4SlimmingHelper.IncludeMuonTriggerContent = True
FTAG4SlimmingHelper.IncludeEGammaTriggerContent = True
FTAG4SlimmingHelper.IncludeJetTriggerContent = True
FTAG4SlimmingHelper.IncludeEtMissTriggerContent = False
FTAG4SlimmingHelper.IncludeBJetTriggerContent = True
#FTAG4 TrigNav Thinning
FTAG4ThinningHelper = ThinningHelper("FTAG4ThinningHelper")
FTAG4ThinningHelper.TriggerChains = 'HLT_e[2-9][0-9]_.*|HLT_e[1-9][0-9][0-9]_.*|HLT_mu[2-9][0-9]_.*|HLT_mu[1-9][0-9][0-9]_.*|HLT_mu[2-9][0-9]'
FTAG4ThinningHelper.AppendToStream(FTAG4Stream)
FTAG4SlimmingHelper.AppendContentToStream(FTAG4Stream)
# Trigger matching decorations
#====================================================================
photonTriggers_matching = ['HLT_g60_loose', 'HLT_g140_loose', 'HLT_g160_loose']
from DerivationFrameworkCore.TriggerMatchingAugmentation import applyTriggerMatching
TrigMatchAug, NewTrigVars = applyTriggerMatching(ToolNamePrefix="JETM6",
ElectronTriggers=electronTriggers,
MuonTriggers=muonTriggers,
PhotonTriggers=photonTriggers_matching)
#====================================================================
# TRIGGER THINNING TOOL
#====================================================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
JETM6ThinningHelper = ThinningHelper( "JETM6ThinningHelper" )
JETM6ThinningHelper.TriggerChains = ''
JETM6ThinningHelper.TriggerChains += "|".join(electronTriggers)
JETM6ThinningHelper.TriggerChains += "|".join(muonTriggers)
JETM6ThinningHelper.TriggerChains += "|".join(photonTriggers)
JETM6ThinningHelper.TriggerChains += "|".join(jetTriggers)
JETM6ThinningHelper.AppendToStream( JETM6Stream )
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
#########################################
TOPQ2Stream = MSMgr.NewPoolRootStream(streamName, fileName)
# Accept the most selective kernel (last one in sequence; later in derivation)
TOPQ2Stream.AcceptAlgs(["TOPQ2Kernel"])
#====================================================================
# PDF Weight Metadata
#====================================================================
if DFisMC:
from DerivationFrameworkCore.WeightMetadata import *
#====================================================================
# TRIGGER NAVIGATION THINNING
#====================================================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
import DerivationFrameworkTop.TOPQCommonThinning
TOPQ2ThinningHelper = ThinningHelper("TOPQ2ThinningHelper")
TOPQ2ThinningHelper.TriggerChains = DerivationFrameworkTop.TOPQCommonThinning.TOPQTriggerChains(
)
TOPQ2ThinningHelper.AppendToStream(TOPQ2Stream)
#====================================================================
# SKIMMING TOOLS
#====================================================================
import DerivationFrameworkTop.TOPQCommonSelection
skimmingTools_lep = DerivationFrameworkTop.TOPQCommonSelection.setup_lep(
'TOPQ2', ToolSvc)
skimmingTools_jet = DerivationFrameworkTop.TOPQCommonSelection.setup_jet(
'TOPQ2', ToolSvc)
#====================================================================
# THINNING TOOLS
from DerivationFrameworkMuons.MuonsCommon import *
if DerivationFrameworkIsMonteCarlo:
from DerivationFrameworkMCTruth.MCTruthCommon import *
from DerivationFrameworkTau.TauTruthCommon import *
from DerivationFrameworkInDet.InDetCommon import *
from DerivationFrameworkJetEtMiss.METCommon import *
### Set up stream
streamName = derivationFlags.WriteDAOD_SUSY5Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_SUSY5Stream)
SUSY5Stream = MSMgr.NewPoolRootStream(streamName, fileName)
SUSY5Stream.AcceptAlgs(["SUSY5KernelSkim"])
### Init
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
SUSY5ThinningHelper = ThinningHelper("SUSY5ThinningHelper")
thinningTools = []
AugmentationTools = []
# stream-specific sequence for on-the-fly jet building
SeqSUSY5 = CfgMgr.AthSequencer("SeqSUSY5")
DerivationFrameworkJob += SeqSUSY5
#====================================================================
# Trigger navigation thinning
#====================================================================
triggerRegEx = ["e.*", "mu.*", "tau.*"]
SUSY5ThinningHelper.TriggerChains = 'HLT_(' + ' | '.join(triggerRegEx) + ')'
SUSY5ThinningHelper.AppendToStream(SUSY5Stream)
#====================================================================
#====================================================================
from DerivationFrameworkCalo.DerivationFrameworkCaloConf import DerivationFramework__MaxCellDecorator
EGAM9_MaxCellDecoratorTool = DerivationFramework__MaxCellDecorator(
name="EGAM9_MaxCellDecoratorTool",
SGKey_electrons="Electrons",
SGKey_photons="Photons",
)
ToolSvc += EGAM9_MaxCellDecoratorTool
#====================================================================
# SET UP THINNING
#====================================================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
EGAM9ThinningHelper = ThinningHelper("EGAM9ThinningHelper")
EGAM9ThinningHelper.TriggerChains = '(^(?!.*_[0-9]*(mu|j|xe|tau|ht|xs|te))(?!HLT_[eg].*_[0-9]*[eg][0-9].*)(?!HLT_eb.*)(?!.*larpeb.*)(?!HLT_.*_AFP_.*)(HLT_[eg].*))'
if globalflags.DataSource() != 'geant4':
ExtraContainersTrigger += ExtraContainersTriggerDataOnly
EGAM9ThinningHelper.AppendToStream(EGAM9Stream, ExtraContainersTrigger)
thinningTools = []
# Truth thinning
if globalflags.DataSource() == 'geant4':
truth_cond_WZH = "((abs(TruthParticles.pdgId) >= 23) && (abs(TruthParticles.pdgId) <= 25))" # W, Z and Higgs
truth_cond_lep = "((abs(TruthParticles.pdgId) >= 11) && (abs(TruthParticles.pdgId) <= 16))" # Leptons
truth_cond_top = "((abs(TruthParticles.pdgId) == 6))" # Top quark
truth_cond_gam = "((abs(TruthParticles.pdgId) == 22) && (TruthParticles.pt > 1*GeV))" # Photon
truth_cond_finalState = '(TruthParticles.status == 1 && TruthParticles.barcode < 200000)' # stable particles
truth_expression = '(' + truth_cond_WZH + ' || ' + truth_cond_lep + ' || ' + truth_cond_top + ' || ' + truth_cond_gam + ') || (' + truth_cond_finalState + ')'
applyJetCalibration_xAODColl(
"AntiKt4EMTopo") # adds this to DerivationFrameworkJob by default
updateJVT_xAODColl(
"AntiKt4EMTopo") # adds this to DerivationFrameworkJob by default
from DerivationFrameworkFlavourTag.FlavourTagCommon import applyBTagging_xAODColl
applyJetCalibration_xAODColl("AntiKt4EMTopo_BTagging201810")
updateJVT_xAODColl('AntiKt4EMTopo_BTagging201810')
applyBTagging_xAODColl('AntiKt4EMTopo_BTagging201810')
#=======================================
# ESTABLISH THE THINNING HELPER
#=======================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
JETM7ThinningHelper = ThinningHelper("JETM7ThinningHelper")
# JETM7ThinningHelper.TriggerChains = orstr.join(electronTriggers+muonTriggers)
JETM7ThinningHelper.AppendToStream(JETM7Stream)
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
# TrackParticles associated with Muons
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__MuonTrackParticleThinning
JETM7MuonTPThinningTool = DerivationFramework__MuonTrackParticleThinning(
name="JETM7MuonTPThinningTool",
StreamName=streamName,
MuonKey="Muons",
InDetTrackParticlesKey="InDetTrackParticles")
from DerivationFrameworkTau.TauTruthCommon import *
from DerivationFrameworkTau.TauCommon import *
# ==========================================================================================================================
# Set up stream
# ==========================================================================================================================
streamName = derivationFlags.WriteDAOD_TAUP1Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_TAUP1Stream)
TAUP1Stream = MSMgr.NewPoolRootStream(streamName, fileName)
TAUP1Stream.AcceptAlgs(["TAUP1Kernel"])
# ==========================================================================================================================
# Thinning tool
# ==========================================================================================================================
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
TAUP1ThinningHelper = ThinningHelper("TAUP1ThinningHelper")
TAUP1ThinningHelper.TriggerChains = "HLT_e24.*|HLT_e60.*"
TAUP1ThinningHelper.AppendToStream(TAUP1Stream)
# MET/Jet tracks
thinning_expression = "(InDetTrackParticles.pt > 0.5*GeV) && (InDetTrackParticles.numberOfPixelHits > 0) && (InDetTrackParticles.numberOfSCTHits > 5) && (abs(DFCommonInDetTrackZ0AtPV) < 1.5)"
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TrackParticleThinning
TAUP1MetTPThinningTool = DerivationFramework__TrackParticleThinning(
name="TAUP1MetTPThinningTool",
ThinningService=TAUP1ThinningHelper.ThinningSvc(),
SelectionString=thinning_expression,
InDetTrackParticlesKey="InDetTrackParticles",
ApplyAnd=True)
ToolSvc += TAUP1MetTPThinningTool
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__JetTrackParticleThinning
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_HIGG5D2Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_HIGG5D2Stream)
HIGG5D2Stream = MSMgr.NewPoolRootStream(streamName, fileName)
HIGG5D2Stream.AcceptAlgs(["HIGG5D2Kernel"])
#====================================================================
# THINNING TOOLS
#====================================================================
thinningTools = []
# Establish the thinning helper (which will set up the services behind the scenes)
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
HIGG5D2ThinningHelper = ThinningHelper("HIGG5D2ThinningHelper")
#trigger navigation content
HIGG5D2ThinningHelper.TriggerChains = 'HLT_e.*|HLT_mu.*|HLT_xe.*|HLT_j.*'
HIGG5D2ThinningHelper.AppendToStream(HIGG5D2Stream)
# MC truth thinning (not for data)
truth_cond_WZH = "((abs(TruthParticles.pdgId) >= 23) && (abs(TruthParticles.pdgId) <= 25))" # W, Z and Higgs
truth_cond_Lepton = "((abs(TruthParticles.pdgId) >= 11) && (abs(TruthParticles.pdgId) <= 16))" # Leptons
truth_cond_Quark = "((abs(TruthParticles.pdgId) == 6) || (abs(TruthParticles.pdgId) == 5))" # Top quark and Bottom quark
truth_cond_Photon = "((abs(TruthParticles.pdgId) == 22) && (TruthParticles.pt > 1*GeV))" # Photon
truth_expression = '(' + truth_cond_WZH + ' || ' + truth_cond_Lepton + ' || ' + truth_cond_Quark + ' || ' + truth_cond_Photon + ')'
from DerivationFrameworkMCTruth.DerivationFrameworkMCTruthConf import DerivationFramework__GenericTruthThinning
HIGG5D2MCThinningTool = DerivationFramework__GenericTruthThinning(
name="HIGG5D2MCThinningTool",
ThinningService=HIGG5D2ThinningHelper.ThinningSvc(),
ParticleSelectionString=truth_expression,
#====================================================================
# SET UP STREAM
#====================================================================
streamName = derivationFlags.WriteDAOD_EXOT10Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_EXOT10Stream)
EXOT10Stream = MSMgr.NewPoolRootStream(streamName, fileName)
EXOT10Stream.AcceptAlgs(["EXOT10Kernel"])
#====================================================================
# THINNING
#====================================================================
# Establish the thinning helper (which will set up the services behind the scenes)
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
EXOT10ThinningHelper = ThinningHelper("EXOT10ThinningHelper")
beamEnergy = jobproperties.Beam.energy()
if (beamEnergy < 4.1e+06):
EXOT10ThinningHelper.TriggerChains = 'EF_2g20_tight1 | EF_2g50_loose | HLT_g35_medium_g25_medium'
if (beamEnergy > 6.0e+06):
EXOT10ThinningHelper.TriggerChains = 'HLT_2g20_loose | HLT_2g50_loose | HLT_g35_medium_g25_medium'
EXOT10ThinningHelper.AppendToStream(EXOT10Stream)
#====================================================================
# THINNING TOOLS
#====================================================================
from DerivationFrameworkInDet.DerivationFrameworkInDetConf import DerivationFramework__TrackParticleThinning
EXOT10TPThinningTool = DerivationFramework__TrackParticleThinning(
name="EXOT10TPThinningTool",
StreamName=streamName,
print "Hello, my name is {} and I am running on {}".format(DAOD_StreamID, 'MC' if DFisMC else 'Data')
#==============
# SET UP STREAM
#==============
streamName = derivationFlags.WriteDAOD_HIGG4D3Stream.StreamName
fileName = buildFileName( derivationFlags.WriteDAOD_HIGG4D3Stream )
HIGG4D3Stream = MSMgr.NewPoolRootStream( streamName, fileName )
HIGG4D3Stream.AcceptAlgs([DAOD_StreamID+"Kernel"])
#============
# Setup tools
#============
# Establish the thinning helper
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
HIGG4D3ThinningHelper = ThinningHelper( DAOD_StreamID+"ThinningHelper" )
#trigger navigation thinning
import DerivationFrameworkHiggs.HIGG4DxThinning
HIGG4D3ThinningHelper.TriggerChains = DerivationFrameworkHiggs.HIGG4DxThinning.TriggerChains(DAOD_StreamID)
HIGG4D3ThinningHelper.AppendToStream( HIGG4D3Stream )
# thinning tools
thinningTools = DerivationFrameworkHiggs.HIGG4DxThinning.setup(DAOD_StreamID, HIGG4D3ThinningHelper.ThinningSvc(), ToolSvc)
# skimming tools
import DerivationFrameworkHiggs.HIGG4DxSkimming
skimmingTools = DerivationFrameworkHiggs.HIGG4DxSkimming.setup(DAOD_StreamID, ToolSvc)
#augmentation tools
from DerivationFrameworkHiggs.HIGG4DxAugmentation import *
from DerivationFrameworkJetEtMiss.ExtendedJetCommon import *
from DerivationFrameworkEGamma.EGammaCommon import *
from DerivationFrameworkMuons.MuonsCommon import *
from DerivationFrameworkTau.TauTruthCommon import *
from DerivationFrameworkInDet.InDetCommon import *
from DerivationFrameworkJetEtMiss.METCommon import *
from DerivationFrameworkTau.TauCommon import *
### Set up stream
streamName = derivationFlags.WriteDAOD_SUSY9Stream.StreamName
fileName = buildFileName(derivationFlags.WriteDAOD_SUSY9Stream)
SUSY9Stream = MSMgr.NewPoolRootStream(streamName, fileName)
SUSY9Stream.AcceptAlgs(["SUSY9KernelSkim"])
### Init
from DerivationFrameworkCore.ThinningHelper import ThinningHelper
SUSY9ThinningHelper = ThinningHelper("SUSY9ThinningHelper")
thinningTools = []
AugmentationTools = []
# stream-specific sequence for on-the-fly jet building
SeqSUSY9 = CfgMgr.AthSequencer("SeqSUSY9")
DerivationFrameworkJob += SeqSUSY9
#====================================================================
# Trigger navigation thinning
#====================================================================
# To be added.
SUSY9ThinningHelper.AppendToStream(SUSY9Stream)
#====================================================================
# THINNING TOOL
