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