AFPMenuSequence.py

Go to the documentation of this file.

#
#  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#
 
from AthenaConfiguration.AccumulatorCache import AccumulatorCache
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from TriggerMenuMT.HLT.Config.MenuComponents import InEventRecoCA, SelectionCA, MenuSequence
 
 
@AccumulatorCache
def AFPTrkRecoBaseSequenceCfg(flags):
    recoAcc = InEventRecoCA(name='AFPTrackingFS')
 
    if not flags.Input.isMC:
        acc = ComponentAccumulator()
 
        # Bytestream converter
        AFP_Raw = CompFactory.AFP_RawDataProvider("AFP_RawDataProvider")
        acc.addEventAlgo(AFP_Raw)
 
        # Digitalization
        AFP_R2D = CompFactory.AFP_Raw2Digi("AFP_Raw2Digi")
        acc.addEventAlgo(AFP_R2D)
 
        recoAcc.mergeReco(acc)
 
    # Cluster reconstruction
    from AFP_SiClusterTools.AFP_SiClusterTools import AFP_SiClusterTools_HLT
    AFP_SiCl = AFP_SiClusterTools_HLT(flags)
    recoAcc.mergeReco(AFP_SiCl)
 
    # Tracks reconstruction
    from AFP_LocReco.AFP_LocReco import AFP_LocReco_SiD_HLT
    AFP_SID = AFP_LocReco_SiD_HLT(flags)
    recoAcc.mergeReco(AFP_SID)
 
    return recoAcc
 
 
@AccumulatorCache
def AFPTrkSequenceGenCfg(flags):
    def trigStreamerAFPHypoTool(chainDict):
        return CompFactory.TrigStreamerHypoTool(chainDict['chainName'])
 
    recoAcc = AFPTrkRecoBaseSequenceCfg(flags)
 
    hypo = CompFactory.TrigStreamerHypoAlg('AFPPassThroughHypo')
 
    selAcc = SelectionCA('AFPPassThroughSequence')
    selAcc.mergeReco(recoAcc)
    selAcc.addHypoAlgo(hypo)
 
    return MenuSequence(flags, selAcc, HypoToolGen=trigStreamerAFPHypoTool)
 
 
def TrigAFPDijetComboHypoToolCfg(flags, chainDict):
    name = chainDict['chainName']
    tool = CompFactory.TrigAFPDijetComboHypoTool(name)
 
    tool.maxProtonDiff_x = 2.5 
    tool.maxProtonDiff_y = 100.0
    tool.maxProtonDist = 100.0
 
    from AthenaMonitoringKernel.GenericMonitoringTool import GenericMonitoringTool
    monTool = GenericMonitoringTool(flags, 'MonTool_' + name, HistPath='AFPComboHypo/' + tool.getName())
    monTool.defineHistogram('DijetMass', type='TH1F', path='EXPERT', title="Dijet mass", xbins=100, xmin=0, xmax=2000)
    monTool.defineHistogram('DijetRapidity', type='TH1F', path='EXPERT', title="Dijet rapidity", xbins=100, xmin=-5, xmax=5)
 
    monTool.defineHistogram('XiJet1', type='TH1F', path='EXPERT', title="Jet 1 xi", xbins=100, xmin=0, xmax=1)
    monTool.defineHistogram('XiJet2', type='TH1F', path='EXPERT', title="Jet 2 x1", xbins=100, xmin=0, xmax=1)
 
    monTool.defineHistogram('PredictProtonAEnergy', type='TH1F', path='EXPERT', title="Predicted proton energy A", xbins=100, xmin=0, xmax=10000)
    monTool.defineHistogram('PredictProtonCEnergy', type='TH1F', path='EXPERT', title="Predicted proton energy C", xbins=100, xmin=0, xmax=10000)
 
    monTool.defineHistogram('SideA_predictX', type='TH1F', path='EXPERT', title="Predicted X side A", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('SideA_predictY', type='TH1F', path='EXPERT', title="Predicted Y side A", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('SideC_predictX', type='TH1F', path='EXPERT', title="Predicted X side C", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('SideC_predictY', type='TH1F', path='EXPERT', title="Predicted Y side C", xbins=100, xmin=-100, xmax=100)
 
    monTool.defineHistogram('XDiff', type='TH1F', path='EXPERT', title="X difference", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('YDiff', type='TH1F', path='EXPERT', title="Y difference", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('distance', type='TH1F', path='EXPERT', title="distance", xbins=100, xmin=0, xmax=50)
 
    monTool.defineHistogram('SideA_trackX', type='TH1F', path='EXPERT', title="Track X side A", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('SideA_trackY', type='TH1F', path='EXPERT', title="Track Y side A", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('SideA_diffX', type='TH1F', path='EXPERT', title="Track X diff side A", xbins=100, xmin=-50, xmax=50)
    monTool.defineHistogram('SideA_diffY', type='TH1F', path='EXPERT', title="Track Y diff side A", xbins=100, xmin=-50, xmax=50)
 
    monTool.defineHistogram('SideC_trackX', type='TH1F', path='EXPERT', title="Track X side C", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('SideC_trackY', type='TH1F', path='EXPERT', title="Track Y side C", xbins=100, xmin=-100, xmax=100)
    monTool.defineHistogram('SideC_diffX', type='TH1F', path='EXPERT', title="Track X diff side C", xbins=100, xmin=-50, xmax=50)
    monTool.defineHistogram('SideC_diffY', type='TH1F', path='EXPERT', title="Track Y diff side C", xbins=100, xmin=-50, xmax=50)
 
    tool.MonTool = monTool
 
    return tool
 
 
@AccumulatorCache
def AFPGlobalRecoSequenceCfg(flags):
    recoAcc = InEventRecoCA(name='AFPGlobalFS')
 
    # ToF Tracks reconstruction
    from AFP_LocReco.AFP_LocReco import AFP_LocReco_TD_HLT
    AFP_TD = AFP_LocReco_TD_HLT(flags)
    recoAcc.mergeReco(AFP_TD)
 
    # Protons reconstruction
    from AFP_GlobReco.AFP_GlobReco import AFP_GlobReco_HLT
    AFP_Pr = AFP_GlobReco_HLT(flags)
    recoAcc.mergeReco(AFP_Pr)
 
    # Vertex reconstruction
    from AFP_VertexReco.AFP_VertexReco import AFP_VertexReco_HLT
    AFP_Vtx = AFP_VertexReco_HLT(flags)
    recoAcc.mergeReco(AFP_Vtx)
 
    return recoAcc
 
 
@AccumulatorCache
def AFPGlobalSequenceGenCfg(flags):
    def trigStreamerAFPToFHypoTool(chainDict):
        return CompFactory.TrigStreamerHypoTool(chainDict['chainName'])
    
    recoAcc = AFPGlobalRecoSequenceCfg(flags)
 
    hypo = CompFactory.TrigStreamerHypoAlg('AFPToFPassThroughHypo')
 
    selAcc = SelectionCA('AFPGlobalSequence')
    selAcc.mergeReco(recoAcc)
    selAcc.addHypoAlgo(hypo)
 
    return MenuSequence(flags, selAcc, HypoToolGen=trigStreamerAFPToFHypoTool)
 
def AFPToFDeltaZToolGen(chainDict):
    hypotool = CompFactory.TrigAFPToFHypoTool(chainDict['chainName'])
    if "afpdz5" in chainDict["chainName"]:
        hypotool.deltaZCut = 5.0
    elif "afpdz10" in chainDict["chainName"]:
        hypotool.deltaZCut = 10.0
    return hypotool
 
@AccumulatorCache
def AFPToFDeltaZSequenceGenCfg(flags):
 
    recoAcc = AFPGlobalRecoSequenceCfg(flags)
 
    hypo = CompFactory.TrigAFPToFHypoAlg('TrigAFPToFHypoAlg', AFPVertexContainer='HLT_AFPVertexContainer', VertexContainer='HLT_IDVertex_FS')
 
    # add monitoring
    from AthenaMonitoringKernel.GenericMonitoringTool import GenericMonitoringTool
    monTool = GenericMonitoringTool(flags, 'MonToolAFPToFHypo')
    monTool.defineHistogram('afp_vtx_count', path='EXPERT', type='TH1F', title='AFP Vertex Counts',
                     xbins=6, xmin=-0.5, xmax=5.5 )
    monTool.defineHistogram('id_vtx_count', path='EXPERT', type='TH1F', title='ID Vertex Counts',
                     xbins=6, xmin=-0.5, xmax=5.5 )
    monTool.defineHistogram('skipHypoTool', path='EXPERT', type='TH1F', title='Reason of skipping the trigger',
                     xbins=6, xmin=-0.5, xmax=5.5 )
    monTool.defineHistogram('afp_vtx_z', path='EXPERT', type='TH1F', title='AFP Vertex X',
                     xbins=401, xmin=-200, xmax=200 )
    monTool.defineHistogram('id_vtx_z', path='EXPERT', type='TH1F', title='ID Vertex Z',
                     xbins=401, xmin=-200, xmax=200 )
    monTool.defineHistogram('delta_z', path='EXPERT', type='TH1F', title='Delta Z',
                     xbins=201, xmin=-100, xmax=100 )
    hypo.MonTool = monTool
 
    selAcc = SelectionCA('AFPToFDeltaZSequence')
    selAcc.mergeReco(recoAcc)
    selAcc.addHypoAlgo(hypo)
 
    return MenuSequence(flags, selAcc, HypoToolGen=AFPToFDeltaZToolGen)
 
 
if __name__ == '__main__':
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.AOD_RUN3_MC
    flags.lock()
 
 
    afp_trk = AFPTrkSequenceGenCfg(flags)
    afp_trk.ca.printConfig(withDetails=True)
 
    afp_glob = AFPGlobalSequenceGenCfg(flags)
    afp_glob.ca.printConfig(withDetails=True)