recoTransforms.py

Go to the documentation of this file.

# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
 
 
import os
import subprocess
 
import logging
msg = logging.getLogger(__name__)
 
import PyJobTransforms.trfExceptions as trfExceptions
 
from PyJobTransforms.trfExitCodes import trfExit
from PyJobTransforms.trfExe import scriptExecutor
 
 
class skimRawExecutor(scriptExecutor):
 
    def preExecute(self, input = set(), output = set()):
        # First we need to strip the filter file down to events that are present 
        # in the RAW file we are going to skim. This is because the HI workflow
        # will provide millions of events in their filter file, more than acmd.py
        # can cope with.
        listEvtCommand = ['AtlListBSEvents', '-l']
        listEvtCommand.extend(self.conf.argdict['inputBSFile'].value)
        # For best lookup speed, we store the runnumber/eventnumber in a dictionary (set would also
        # be fast)
        rawEventList = {} 
        longlist = subprocess.check_output(listEvtCommand).decode('utf-8')
        try:
            for line in longlist.split("\n"):
                if line.startswith("Index="):
                    try:
                        splitStrings = line.split(" ")
                        runprefix, runstr = splitStrings[1].split("=")
                        evtprefix, evtstr = splitStrings[2].split("=")
                        # Check sanity
                        if runprefix != "Run" or evtprefix != "Event":
                            msg.warning("Failed to understand this line from AtlListBSEvents (1): %s", line)
                        else:
                            runnumber = int(runstr)  # noqa: F841
                            evtnumber = int(evtstr)  # noqa: F841
                            # We build up a string key as "RUN-EVENT", so that we can take advantage of
                            # the fast hash search against a dictionary 
                            rawEventList[runstr + "-" + evtstr] = True
                            msg.debug("Identified run %s, event %s in input RAW files", runstr, evtstr)
                    except ValueError:
                        msg.warning("Failed to understand this line from AtlListBSEvents (2): %s", line)
        except subprocess.CalledProcessError as e:
            errMsg = "Call to AtlListBSEvents failed: {0}".format(e)
            msg.error(errMsg)
            raise trfExceptions.TransformExecutionException(trfExit.nameToCode("TRF_EXEC_SETUP_FAIL"), errMsg)
        msg.info("Found %d events as skim candidates in RAW inputs", len(rawEventList))
        
        # Now open the list of filter events, and check through them
        slimmedFilterFile = "slimmedFilterFile.{0}".format(os.getpid())
        with open(slimmedFilterFile, "w") as slimFF, open(self.conf.argdict['filterFile'].value) as masterFF:
            count = 0
            for line in masterFF:
                try:
                    runstr, evtstr = str(line).split()
                    if runstr + "-" + evtstr in rawEventList:
                        msg.debug("Found run %s, event %s in master filter list", runstr, evtstr)
                        os.write(slimFF.fileno(), line.encode('utf-8'))
                        count += 1
                except ValueError as e:
                    msg.warning("Failed to understand this line from master filter file: %s %s", line, e)
            if count == 0:
                # If there are no matched events, create a bogus request for run and event 0 to keep
                # AtlCopyBSEvent.exe CLI
                msg.info("No events matched in this input file - empty RAW file output will be made")
                os.write(slimFF.fileno(), b"0 0\n")
        msg.info("Matched %d lines from the master filter file against input events; wrote these to %s", count, slimmedFilterFile)
        
        # Build up the right command line for AtlCopyBSEvent
 
        events = ''
        for line in open(slimmedFilterFile):
            events += '%s,' % line.split()[-1]
        events = events[:-1]
 
        self._cmd = ['AtlCopyBSEvent']
        
        self._cmd.extend(('-e', events))
        self._cmd.extend(('-o', self.conf.argdict['outputBS_SKIMFile'].value[0]))
        self._cmd.extend(self.conf.argdict['inputBSFile'].value)
 
        super(skimRawExecutor, self).preExecute()