DRAW_EGZ.py

Go to the documentation of this file.

# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
#!/usr/bin/env python
# ====================================================================
# DRAW_EGZ.py
# This defines DRAW_EGZ, a skimmed DRAW format
# Z->ee, eey and mmy reduction for electron and photon ID and calibration
# ====================================================================
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaCommon.CFElements import seqAND
from AthenaCommon.Logging import logging
from PrimaryDPDMaker.DRAWCommonByteStream import DRAWCommonByteStreamCfg
 
 
def DRAW_EGZKernelCfg(flags, name='DRAW_EGZKernel', **kwargs):
    """Configure DRAW_EGZ kerne"""
 
    mlog = logging.getLogger(name)
    mlog.info('Start configuration')
 
    acc = ComponentAccumulator()
    acc.addSequence(seqAND('DRAW_EGZSequence'))
 
    # The selections
    DRAWEGZSel = {}
    DRAWEGZSel['Zee'] = [
        'Electrons.pt > 20*GeV && Electrons.LHMedium',
        'eeMass1',
        '(count(eeMass1 > 55*GeV) >= 1)']
    DRAWEGZSel['Zeey'] = [
        'Electrons.pt > 15*GeV && Electrons.LHMedium',
        'eeMass2',
        '(count(eeMass2 > 20*GeV && eeMass2 < 90*GeV) >= 1 && count(Photons.pt > 7*GeV && Photons.Tight) >= 1)']
    DRAWEGZSel['Zmmy'] = [
        'Muons.pt > 15*GeV',
        'mmMass',
        '(count(mmMass > 20*GeV && mmMass < 90*GeV) >= 1 && count(Photons.pt > 7*GeV && Photons.Tight) >= 1)']
    DRAWEGZSel['Zefe'] = [
        'Electrons.pt > 20*GeV && Electrons.LHMedium',
        'eeMass3',
        '(count(eeMass3 > 55*GeV) >=1'
        ' && count(Electrons.pt > 20*GeV && Electrons.LHMedium)'
        ' && count(ForwardElectrons.pt > 20*GeV && ForwardElectrons.Loose))',
        'ForwardElectrons.pt > 20*GeV && ForwardElectrons.Loose']
    # Augmentation tools for the di-lepton mass computations
    EventSels = []
    augmentationTools = []
    for key, sel in DRAWEGZSel.items():
        if key == 'Zefe':
            tool = CompFactory.DerivationFramework.EGInvariantMassTool(
                name=f'llmassToolFor{key}',
                Container1Name='Electrons',
                Container2Name='ForwardElectrons',
                Object1Requirements=sel[0],
                Object2Requirements=sel[3],
                Mass1Hypothesis=0.511,
                Mass2Hypothesis=0.511,
                CheckCharge=False,
                StoreGateEntryName=sel[1])
        else:
            tool = CompFactory.DerivationFramework.InvariantMassTool(
                name=f'llmassToolFor{key}',
                ContainerName='Electrons' if key.find('Zee') >= 0 else 'Muons',
                ObjectRequirements=sel[0],
                MassHypothesis=0.511 if key.find('Zee') >= 0 else 105.66,
                StoreGateEntryName=sel[1])
 
        augmentationTools.append(tool)
        acc.addPublicTool(tool)
        EventSels.append(sel[2])
    draw_egz = " || ".join(EventSels)
    mlog.info('DRAW_EGZ selection '+draw_egz)
 
    # The skimming tool
    from DerivationFrameworkTools.DerivationFrameworkToolsConfig import (
        xAODStringSkimmingToolCfg)
    skimmingTool = acc.getPrimaryAndMerge(xAODStringSkimmingToolCfg(
        flags, name='DRAW_EGZSkimmingTool', expression=draw_egz))
 
    # The main kernel algo
    DRAW_EGZKernel = CompFactory.DerivationFramework.DerivationKernel(
        name='DRAW_EGZKernel',
        doChronoStat=(flags.Concurrency.NumThreads <= 1),
        AugmentationTools=augmentationTools,
        SkimmingTools=[skimmingTool])
 
    acc.addEventAlgo(DRAW_EGZKernel, sequenceName='DRAW_EGZSequence')
    return acc
 
 
def DRAW_EGZCfg(flags):
    """Main config fragment for DRAW_EGZ"""
    acc = ComponentAccumulator()
 
    # Main algorithm (kernel)
    acc.merge(DRAW_EGZKernelCfg(flags, name='DRAW_EGZKernel'))
    acc.merge(DRAWCommonByteStreamCfg(flags,
                                      formatName='DRAW_EGZ',
                                      filename=flags.Output.DRAW_EGZFileName))
 
    return acc