generateCFChains.py

Go to the documentation of this file.

#
#  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#
 
 
 
import functools
 
def generateCFChains(flags):
    from TriggerMenuMT.HLT.Menu.SignatureDicts import ChainStore
    from TriggerMenuMT.CFtest.TestUtils import makeChain, makeChainStep
    from TriggerMenuMT.HLT.Config.GenerateMenuMT import GenerateMenuMT
    menu = GenerateMenuMT()
    menu.chainsInMenu = ChainStore()
    
    if 'Egamma' in flags.Trigger.enabledSignatures:
        from TriggerMenuMT.HLT.CommonSequences.CaloSequences import fastCaloSequenceGenCfg
        from TriggerMenuMT.HLT.Electron.FastElectronMenuSequences import fastElectronSequenceGenCfg
        from TriggerMenuMT.HLT.Electron.PrecisionCaloMenuSequences import precisionCaloSequenceGenCfg
 
        fastCaloSeq = functools.partial(fastCaloSequenceGenCfg, flags, name='Electron' )
        electronSeq = functools.partial(fastElectronSequenceGenCfg, flags )
        precisionCaloSeq = functools.partial(precisionCaloSequenceGenCfg, flags )
        
        FastCaloStep      = makeChainStep("ElectronFastCaloStep", [fastCaloSeq])
        FastElectronStep  = makeChainStep("ElectronFastTrackStep", [electronSeq])
        PrecisionCaloStep = makeChainStep("ElectronPrecisionCaloStep", [precisionCaloSeq])
        
        electronChains  = [
            makeChain(flags, name='HLT_e3_etcut1step_L1EM3',  L1Thresholds=["EM3"],  ChainSteps=[FastCaloStep]  ),
            makeChain(flags, name='HLT_e3_etcut_L1EM3',       L1Thresholds=["EM3"],  ChainSteps=[FastCaloStep, FastElectronStep, PrecisionCaloStep]  ),
            makeChain(flags, name='HLT_e5_etcut_L1EM3',       L1Thresholds=["EM3"],  ChainSteps=[FastCaloStep, FastElectronStep, PrecisionCaloStep]  ),
            makeChain(flags, name='HLT_e7_etcut_L1EM3',       L1Thresholds=["EM3"],  ChainSteps=[FastCaloStep, FastElectronStep, PrecisionCaloStep]  )
            ]
        menu.chainsInMenu['Egamma'] += electronChains
 
        from TriggerMenuMT.HLT.Photon.FastPhotonMenuSequences import fastPhotonSequenceGenCfg
        from TriggerMenuMT.HLT.Photon.PrecisionCaloMenuSequences import precisionCaloSequenceGenCfg
 
        fastCaloSeq            = functools.partial(fastCaloSequenceGenCfg, flags, name='Photon' )
        fastPhotonSeq          = functools.partial(fastPhotonSequenceGenCfg, flags )
        precisionCaloPhotonSeq = functools.partial(precisionCaloSequenceGenCfg, flags )
        
        FastCaloStep            = makeChainStep("PhotonFastCaloStep", [fastCaloSeq])
        fastPhotonStep          = makeChainStep("PhotonStep2", [fastPhotonSeq])
        precisionCaloPhotonStep = makeChainStep("precisionCaloPhotonStep", [precisionCaloPhotonSeq])
        
        photonChains = [
            makeChain(flags, name='HLT_g5_etcut_L1EM3',    L1Thresholds=["EM3"],  ChainSteps=[ FastCaloStep,  fastPhotonStep, precisionCaloPhotonStep]  )
            ]
        menu.chainsInMenu['Egamma'] += photonChains
 
    
    if 'Muon' in flags.Trigger.enabledSignatures:
        from TriggerMenuMT.HLT.Muon.MuonMenuSequences import (
            muFastSequenceGenCfg, muCombSequenceGenCfg, 
            muEFSASequenceGenCfg, muEFCBSequenceGenCfg, muEFSAFSSequenceGenCfg, muEFCBFSSequenceGenCfg
        )
 
        MuonChains  = []
        # step1
        mufastS= functools.partial(muFastSequenceGenCfg,flags)
        step1mufast=makeChainStep("Step1_muFast", [ mufastS ])
        # step2
        mucombS = functools.partial(muCombSequenceGenCfg,flags)
        step2muComb=makeChainStep("Step2_muComb", [ mucombS ])
        # step3
        muEFSAS = functools.partial(muEFSASequenceGenCfg,flags)
        step3muEFSA=makeChainStep("Step3_muEFSA", [ muEFSAS ])
        #/step3muIso =makeChainStep("Step3_muIso",  [ muIsoSequence() ])
        # step4
        muEFCBS = functools.partial(muEFCBSequenceGenCfg,flags)
        step4muEFCB = makeChainStep("Step4_muEFCB", [ muEFCBS ])
        emptyStep   = makeChainStep("Step2_empty", multiplicity=[])
 
        
        MuonChains += [ makeChain(flags, name='HLT_mu6fast_L1MU5VF',     L1Thresholds=["MU5VF"], ChainSteps=[ step1mufast ])]
        MuonChains += [ makeChain(flags, name='HLT_mu6Comb_L1MU5VF',     L1Thresholds=["MU5VF"], ChainSteps=[ step1mufast, step2muComb ])]
        MuonChains += [ makeChain(flags, name='HLT_mu6_L1MU5VF',         L1Thresholds=["MU5VF"], ChainSteps=[ step1mufast, step2muComb, step3muEFSA, step4muEFCB ])]
        MuonChains += [ makeChain(flags, name='HLT_mu6msonly_L1MU5VF',   L1Thresholds=["MU5VF"], ChainSteps=[ step1mufast, emptyStep,   step3muEFSA ])] # removed due to muEFSA isuue(?)
 
        # multi muon trigger
        # 2muons symmetric
        step1_2mufast_sym= makeChainStep("Step1_2muFast_sym", [ mufastS], multiplicity=[2])
        step2_2muComb_sym= makeChainStep("Step2_2muComb_sym", [ mucombS], multiplicity=[2])
    
        MuonChains += [ makeChain(flags, name='HLT_2mu6Comb_L12MU5VF',  L1Thresholds=["MU5VF"], ChainSteps=[ step1_2mufast_sym, step2_2muComb_sym ])]
 
        # 2muons asymmetric (this will change): 2 sequences, 2 seeds
        step1_2mufast_asym= makeChainStep("Step1_2muFast_asym", [ mufastS, mufastS], multiplicity=[1,1])
        step2_2muComb_asym= makeChainStep("Step1_2muComb_asym", [ mucombS, mucombS], multiplicity=[1,1])
    
        MuonChains += [ makeChain(flags, name='HLT_mu6_mu4_L12MU3V',
                                L1Thresholds=["MU3V", "MU3V"],
                                ChainSteps=[ step1_2mufast_asym, step2_2muComb_asym ])]        
        
        
        #FS Muon trigger
        # Full scan MS tracking step
        muEFSAFSS = functools.partial(muEFSAFSSequenceGenCfg,flags)
        muEFCBFSS = functools.partial(muEFCBFSSequenceGenCfg,flags)
        stepFSmuEFSA=makeChainStep("Step_FSmuEFSA", [muEFSAFSS])
        stepFSmuEFCB=makeChainStep("Step_FSmuEFCB", [muEFCBFSS])
        MuonChains += [ makeChain(flags, name='HLT_mu6noL1_L1MU5VF', L1Thresholds=["FSNOSEED"],  ChainSteps=[stepFSmuEFSA, stepFSmuEFCB])]
 
        menu.chainsInMenu['Muon'] += MuonChains
 
        
    
 
    from TriggerMenuMT.HLT.Jet.JetRecoCommon import jetRecoDictFromString
    def jetCaloHypoMenuSequenceFromString(jet_def_str):
        jetRecoDict = jetRecoDictFromString(jet_def_str)
        from TriggerMenuMT.HLT.Jet.JetRecoSequencesConfig import JetRecoDataDeps
        jetDefDict = JetRecoDataDeps(flags, **jetRecoDict)
        from TriggerMenuMT.HLT.Jet.JetMenuSequencesConfig import jetCaloHypoMenuSequenceGenCfg
        return functools.partial(jetCaloHypoMenuSequenceGenCfg, flags, isPerf=False, **jetDefDict)
 
    def jetCaloPreselMenuSequenceFromString(jet_def_str):
        jetRecoDict = jetRecoDictFromString(jet_def_str)
        from TriggerMenuMT.HLT.Jet.JetRecoSequencesConfig import JetRecoDataDeps
        jetDefDict = JetRecoDataDeps(flags, **jetRecoDict)
        from TriggerMenuMT.HLT.Jet.JetMenuSequencesConfig import jetCaloPreselMenuSequenceGenCfg
        return functools.partial(jetCaloPreselMenuSequenceGenCfg, flags, **jetDefDict)
 
    def jetTrackingHypoMenuSequenceFromString(jet_def_str):
        jetRecoDict = jetRecoDictFromString(jet_def_str)
        from TriggerMenuMT.HLT.Jet.JetRecoSequencesConfig import JetRecoDataDeps
        jetDefDict = JetRecoDataDeps(flags, **jetRecoDict)
        from TriggerMenuMT.HLT.Jet.JetMenuSequencesConfig import jetFSTrackingHypoMenuSequenceGenCfg
        return functools.partial(jetFSTrackingHypoMenuSequenceGenCfg, flags, isPerf=False, **jetDefDict)
 
    if 'Jet' in flags.Trigger.enabledSignatures:
 
        # small-R jets
        jetSeq_a4_tc_em = jetCaloHypoMenuSequenceFromString("a4_tc_em_subjesIS")
        step_a4_tc_em = makeChainStep("Step_jet_a4_tc_em", [jetSeq_a4_tc_em])
        
        # large-R jets
        jetSeq_a10_tc_lcw_subjes = jetCaloHypoMenuSequenceFromString("a10_tc_lcw_subjes")
        step_a10_tc_lcw_subjes = makeChainStep("Step_jet_a10_subjes_tc_lcw", [jetSeq_a10_tc_lcw_subjes])
        
        jetSeq_a10r = jetCaloHypoMenuSequenceFromString("a10r_tc_em_subjesIS")
        step_a10r = makeChainStep("Step_jet_a10r", [jetSeq_a10r])
 
        jetSeq_a10t = jetCaloHypoMenuSequenceFromString("a10t_tc_lcw_jes")
        step_a10t = makeChainStep("Step_jet_a10t", [jetSeq_a10t])
        
        # Jet chains with tracking
        jetSeq_a4_tc_em_presel = jetCaloPreselMenuSequenceFromString("a4_tc_em_subjesIS")
        step_a4_tc_em_presel = makeChainStep("Step_jet_a4_tc_em_presel", [jetSeq_a4_tc_em_presel])
        jetSeq_a4_pf_em_ftf = jetTrackingHypoMenuSequenceFromString("a4_tc_em_subresjesgscIS_ftf")
        step_a4_pf_em_ftf = makeChainStep("Step_jet_a4_pf_em_ftf", [jetSeq_a4_pf_em_ftf])
 
        menu.chainsInMenu['Jet'] = [
            makeChain(flags, name='HLT_j45_L1J20',  L1Thresholds=["FSNOSEED"], ChainSteps=[step_a4_tc_em]  ),
            makeChain(flags, name='HLT_j85_L1J20',  L1Thresholds=["FSNOSEED"], ChainSteps=[step_a4_tc_em]  ),
            makeChain(flags, name='HLT_j420_L1J20', L1Thresholds=["FSNOSEED"], ChainSteps=[step_a4_tc_em]  ),
            makeChain(flags, name='HLT_j260f_L1J20',  L1Thresholds=["FSNOSEED"], ChainSteps=[step_a4_tc_em]  ),
            makeChain(flags, name='HLT_j460_a10_lcw_subjes_L1J20', L1Thresholds=["FSNOSEED"], ChainSteps=[step_a10_tc_lcw_subjes]  ),
            makeChain(flags, name='HLT_j460_a10r_L1J20', L1Thresholds=["FSNOSEED"], ChainSteps=[step_a10r]  ),
            makeChain(flags, name='HLT_j460_a10t_L1J20', L1Thresholds=["FSNOSEED"], ChainSteps=[step_a10t]  ),
            makeChain(flags, name='HLT_3j200_L1J20', L1Thresholds=["FSNOSEED"], ChainSteps=[step_a4_tc_em]  ),
            makeChain(flags, name='HLT_5j70c_L1J20', L1Thresholds=["FSNOSEED"], ChainSteps=[step_a4_tc_em]  ), # 5j70_0eta240_L14J15 (J20 until multi-object L1 seeds supported)
            makeChain(flags, name='HLT_j45_pf_subresjesgscIS_ftf_preselj20_L1J20',  L1Thresholds=["FSNOSEED"], ChainSteps=[step_a4_tc_em_presel,step_a4_pf_em_ftf]  ),
            ]
 
 
    
    if 'Bjet' in flags.Trigger.enabledSignatures:
        from TriggerMenuMT.HLT.Bjet.BjetMenuSequences import getBJetSequenceGenCfg
 
        jetSeq_a4_tc_em_presel = jetCaloPreselMenuSequenceFromString("a4_tc_em_subjesIS")
        jetSeq_a4_tc_em_gsc_ftf = jetTrackingHypoMenuSequenceFromString("a4_tc_em_subjesgscIS_ftf")
        jc_name = "HLT_AntiKt4EMPFlowJets_subresjesgscIS_ftf"
 
        bjet_sec= functools.partial(getBJetSequenceGenCfg, flags, jc_name)
        
        step1 = makeChainStep("Step_jet_a4_tc_em_presel", [jetSeq_a4_tc_em_presel])
        step2 = makeChainStep("Step_jet_a4_tc_em_gsc_ftf", [jetSeq_a4_tc_em_gsc_ftf])
        step3 = makeChainStep("Step3_bjet", [bjet_sec])
        
        menu.chainsInMenu['Bjet']  = [
            makeChain(flags, name='HLT_j45_boffperf_ftf_subjesgscIS_preselj20_L1J20', L1Thresholds=["FSNOSEED"], ChainSteps=[step1,step2,step3] ),
            makeChain(flags, name='HLT_j45_bdl1d70_ftf_subjesgscIS_preselj20_L1J20',  L1Thresholds=["FSNOSEED"], ChainSteps=[step1,step2,step3] ),
            ]
 
   
    
    if 'Tau' in flags.Trigger.enabledSignatures and False:  # not working at the moment
        from TriggerMenuMT.HLT.Tau.TauMenuSequences import getTauSequence
 
        step1=makeChainStep("Step1_tau", [getTauSequence('calo')])
        step1MVA=makeChainStep("Step1MVA_tau", [getTauSequence('calo_mva')])
 
        #This runs the tau-preselection(TP) step
        step2TP=makeChainStep("Step2TP_tau", [getTauSequence('track_core')])
 
        #This runs the EFTauMV hypo on top of fast tracks 
        step2PT=makeChainStep("Step2PT_tau", [getTauSequence('precision')])    
  
        menu.chainsInMenu['Tau'] = [
            makeChain(flags, name='HLT_tau0_perf_ptonly_L1TAU8',              L1Thresholds=["TAU8"], ChainSteps=[step1, step2] ),
            makeChain(flags, name='HLT_tau25_medium1_tracktwo_L1TAU12IM',      L1Thresholds=["TAU12IM"],  ChainSteps=[step1, step2TP] ),
            makeChain(flags, name='HLT_tau35_mediumRNN_tracktwoMVA_L1TAU12IM', L1Thresholds=["TAU20IM"], ChainSteps=[step1MVA, step2PT])
        ]
 
    
    if 'Bphysics' in flags.Trigger.enabledSignatures:
        from TriggerMenuMT.HLT.Muon.MuonMenuSequences import muFastSequenceGenCfg, muCombSequenceGenCfg, muEFSASequenceGenCfg, muEFCBSequenceGenCfg
        from TrigBphysHypo.TrigMultiTrkComboHypoConfig import StreamerDimuL2ComboHypoCfg, DimuEFComboHypoCfg
        
        muFast = functools.partial(muFastSequenceGenCfg, flags)
        step1_dimufast=makeChainStep("Step1_dimuFast", [muFast], multiplicity=[2])
        mucombS = functools.partial(muCombSequenceGenCfg, flags)
        step2_dimuComb=makeChainStep("Step2_dimuComb", [mucombS], multiplicity=[2], comboHypoCfg=functools.partial(StreamerDimuL2ComboHypoCfg,flags))
        muEFSAS = functools.partial(muEFSASequenceGenCfg, flags)
        muEFCBS = functools.partial(muEFCBSequenceGenCfg, flags)
 
        step3_dimuEFSA=makeChainStep("Step3_dimuEFSA", [muEFSAS], multiplicity=[2])
        step4_dimuEFCB=makeChainStep("Step4_dimuEFCB", [muEFCBS], multiplicity=[2], comboHypoCfg=functools.partial(DimuEFComboHypoCfg,flags))
        steps = [step1_dimufast, step2_dimuComb, step3_dimuEFSA, step4_dimuEFCB]
 
        menu.chainsInMenu['Bphysics'] = [
            makeChain(flags, name='HLT_2mu4_bBmumu_L12MU3V', L1Thresholds=["MU3V"], ChainSteps=steps),
            makeChain(flags, name='HLT_2mu4_bDimu_L12MU3V', L1Thresholds=["MU3V"], ChainSteps=steps),
            makeChain(flags, name='HLT_2mu4_bJpsimumu_L12MU3V', L1Thresholds=["MU3V"], ChainSteps=steps),
            makeChain(flags, name='HLT_2mu6_bJpsimumu_L12MU5VF', L1Thresholds=["MU5VF"], ChainSteps=steps),
            makeChain(flags, name='HLT_2mu4_bUpsimumu_L12MU3V', L1Thresholds=["MU3V"], ChainSteps=steps)
            ]
 
    
    doCombinedSlice = True
    if doCombinedSlice:
        from TriggerMenuMT.HLT.CommonSequences.CaloSequences import fastCaloSequenceGenCfg
        fastCaloSeq = functools.partial(fastCaloSequenceGenCfg,flags, name='Electron')
        
        from TriggerMenuMT.HLT.Muon.MuonMenuSequences import muFastSequenceGenCfg
        muFast = functools.partial(muFastSequenceGenCfg,flags)
 
        comboStep_et_mufast = makeChainStep("Step1_et_mufast", [fastCaloSeq, muFast], multiplicity=[1,1])
 
        menu.chainsInMenu['Combined'] = [
            makeChain(flags, name='HLT_e3_etcut_mu6_L12eEM10L_MU8F', L1Thresholds=["eEM10L", "MU8F"],  ChainSteps=[comboStep_et_mufast ])
        ]
 
    return menu