e2mu.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
 
from AthenaCommon import Logging
from ...decorators import timed
from ...utility import LHE
import shutil
 
 
logger = Logging.logging.getLogger("PowhegControl")
 
 
@timed("e2mu")
def e2mu(powheg_LHE_output):
    """! Post-process existing events from electrons to muons
 
    @param powheg_LHE_output  Name of LHE file produced by PowhegBox.
 
    @author Jan Kretzschmar <jan.kretzschmar@cern.ch>
    """
    logger.warning("Converting LHE events from electron to muon decays.")
 
    # Get opening and closing strings
    preamble = LHE.preamble(powheg_LHE_output)
    postamble = LHE.postamble(powheg_LHE_output)
    
    n_events = 0
    powheg_LHE_mu = "{}.mu".format(powheg_LHE_output)
    with open(powheg_LHE_mu, "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.e2mu(input_event)
            f_output.write(output_event)
            n_events += [0, 1][is_event_changed]
        f_output.write(postamble)
    logger.info("Changed e->mu in {} events!".format(n_events))
 
    # Make a backup of the original events
    shutil.move(powheg_LHE_output, "{}.e2mu_backup".format(powheg_LHE_output))
    shutil.move(powheg_LHE_mu, powheg_LHE_output)