postInclude.SortInput.py

Go to the documentation of this file.

# Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 
# Joboptions fragment that should be post-included after a typicaly configured Athena job
# It will read original input files, create a list of Events in memory, sort them and produce
# an intermediate Event Collection file that Athena will read instead of the original inputs
 
# Event information is read from EventInfoTags (stored by default in all Athena data files)
# The default sort key value (Lumi) can be oveerriden, as the sorting order
 
# The intermediate Collection file can be inspected using CollQuery cmdline utility
 
import os
 
from AthenaCommon.AppMgr import ServiceMgr
inputs = ServiceMgr.EventSelector.InputCollections
 
# set default sort parameters, read overrides from locals()
tmpCollFile = locals().get("AthenaInputSortCollName", "sortedEventRefs" + str(os.getpid()) + ".root")
sortTag     = locals().get("AthenaInputSortTag",      "LumiBlockN")
sortOrd     = locals().get("AthenaInputSortOrder",    "Ascending")
 
from CollectionSvc.SortedCollectionCreator import SortedCollectionCreator
sorter = SortedCollectionCreator(name="SortEvents")
# Sort Inputs based on one of the EventInfoTag attributes
# Store sorted event collection in a temporary file
# This should run as postInclude, so we assume EventSelector.InputCollections is set earlier
from PyUtils import PoolFile
sorter.execute(inputs, outputCollection=tmpCollFile, outputCollectionType=PoolFile.PoolOpts.CollectionType.RootTTreeCollection, sortAttribute=sortTag, sortOrder=sortOrd)
 
# Reading Events through References require a populated FileCatalog
for inpfile in inputs:
    os.system('pool_insertFileToCatalog {}'.format(inpfile))
 
ServiceMgr.PoolSvc.AttemptCatalogPatch = False
 
# Tell Athena to use the sorted collection instead of the original inputs
ServiceMgr.EventSelector.InputCollections = [tmpCollFile]
ServiceMgr.EventSelector.CollectionType = "RootCollection"