gg4l_munue2all.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
from ... import Logging
from ...decorators import timed
from ...utility import LHE
import shutil
 

logger = Logging.logging.getLogger("PowhegControl")
 
 
@timed("gg4l_munue2all")
def gg4l_munue2all(powheg_LHE_output):
    """! Post-process existing events from muon pairs and electron neutrino
    pairs to arbitrary charged lepton pairs and neutrino pairs
 
    @param powheg_LHE_output  Name of LHE file produced by PowhegBox.
 
    @author Guglielmo Frattari <guglielmo.frattari@cern.ch>
    """
    logger.warning(
        "Converting LHE events from 2mu2nue to 2l2nu final states."
    )
 
    # Get opening and closing strings
    preamble = LHE.preamble(powheg_LHE_output)
    postamble = LHE.postamble(powheg_LHE_output)
    
    n_events = 0
    powheg_LHE_2mu2nue = "{}.mod".format(powheg_LHE_output)
    with open(powheg_LHE_2mu2nue, "w") as f_output:
        f_output.write("{}\n".format(preamble))
        for input_event in LHE.event_iterator(powheg_LHE_output):
            is_event_changed, output_event = LHE.gg4l_munue2all(input_event)
            f_output.write(output_event)
            n_events += [0, 1][is_event_changed]
        f_output.write(postamble)
    logger.info(
        "Changed 2mu2nue final state to 2l2nu in {} events!".format(n_events)
    )
 
    # Make a backup of the original events
    shutil.move(
        powheg_LHE_output,
        "{}.mod_backup".format(powheg_LHE_output)
    )
    shutil.move(powheg_LHE_2mu2nue, powheg_LHE_output)