scenario_htdipz.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
 
from TrigHLTJetHypo.RepeatedConditionParams import RepeatedConditionParams
from TrigHLTJetHypo.HelperConfigToolParams import HelperConfigToolParams
from TrigHLTJetHypo.ConditionDefaults import defaults
from TrigHLTJetHypo.make_treevec import make_treevec
 
import re
 
pattern_full = r'^HTZ(?P<htlo>\d+)XX(?P<N>\d+)c(?P<ptlo>\d*)' #
rgx_pattern = re.compile(pattern_full)
 
def get_kin_args_from_matchdict(groupdict):
    """ Get kinematic cuts on jets for HT+DIPZ MLPL hypo """
    
    if not groupdict['ptlo']:  # then default filtering for pt
        groupdict['ptlo'] = '20'
        groupdict['pthi'] = ''
    
    condargs = []
    vals = defaults('pt',
                    groupdict['ptlo'],
                    '')  # will be assigned default value 
    condargs.append(('pt', vals))
        
    vals = defaults('eta',
                    '', # will be assigned default value 
                    '240')  
    condargs.append(('eta', vals))
 
    return condargs
 
def get_mult_args_from_matchdict(groupdict):
    """Get jet multiplicity for HT+DIPZ MLPL hypo """
     
    if not groupdict['N']:
        raise ValueError('DIPZ scenario requires a pre-defined jet multiplicity.')
 
    condargs = []
 
    vals = defaults('dipz_njet', lo = groupdict['N'], hi = groupdict['N'])
    condargs.append(('dipz_njet', vals))
    
    return condargs
 
def get_htdipz_from_matchdict(groupdict, njets):
    """Get DIPz WP, capacity (njets) and decorator names"""
    
    condargs = []
    vals = defaults('ht',
                    groupdict['htlo'])
 
    vals['decName_z']='dipz20231122_z'
    vals['decName_sigma']='dipz20231122_negLogSigma2'
    vals['capacity']=str(njets)
    
    condargs.append(('htdipz', vals))
 
    return condargs
 
 
def scenario_htdipz(scenario, chainPartInd):
    """calculate the parameters needed to generate a hypo helper config AlgTool
    starting from a the hypoScenario which appears in the chainname for
    a DIPZ MLPL + HT condition. """
 
    assert scenario.startswith('HTZ'),\
        'routing error, module %s: bad scenario %s' % (__name__, scenario)
 
    m = rgx_pattern.match(scenario)
 
    assert m is not None, \
        'scenario_htdipz.py - regex part %s does not match scenario %s' % (        
            pattern_full, scenario)
 
    groupdict = m.groupdict()
 
    # list for the repeatedCondition parameters, FilterParams and their indices
    # needed for the HelperConfigToolParams
    repcondargs = []
    filterparams = []
    filterparam_inds = []
 
    # treeVec is [0, 0, 1] handle non-root nodes here
 
    
    condargs_kin = get_kin_args_from_matchdict(groupdict)
    chooseN = float(get_mult_args_from_matchdict(groupdict)[0][1]['min']) 
    condargsHTZ = get_htdipz_from_matchdict(groupdict, chooseN)
 
    
    repcondargs.append(RepeatedConditionParams(tree_id=2,
                                               tree_pid=1,
                                               multiplicity=1,
                                               chainPartInd=chainPartInd,
                                               condargs=condargsHTZ))
    filterparam_inds.append(-1) # no filter
 
    # N jet: single jet condition with multiplicity N
    repcondargs.append(RepeatedConditionParams(tree_id=1,
                                           tree_pid=0,
                                           multiplicity=int(chooseN),
                                           chainPartInd=chainPartInd,
                                           condargs=condargs_kin))    
    filterparam_inds.append(-1)
 
    # treevec[i] gives the tree_id of the parent of the
    # node with tree_id = i
    treevec = make_treevec(repcondargs)
    assert treevec == [0, 0, 1]
 
    assert len(repcondargs) == len(filterparam_inds)
 
 
    helper_params = HelperConfigToolParams(treevec=treevec,
                                           repcondargs=repcondargs,
                                           filterparams=filterparams,
                                           filterparam_inds=filterparam_inds)
    return [helper_params]  # a list with one entry per FastReduction tree