postInclude.SortInput.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 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()) )
sortTag     = locals().get("AthenaInputSortTag",      "LumiBlockN")
sortOrd     = locals().get("AthenaInputSortOrder",    "Ascending")
 
from CollectionUtilities.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
sorter.execute(inputs, outputCollection=tmpCollFile, sortAttribute=sortTag, sortOrder=sortOrd)
 
# Reading Events through References require a populated FileCatalog
for inpfile in inputs:
    os.system('pool_insertFileToCatalog {}'.format(inpfile))
 
# Tell Athena to use the sorted collection instead of the original inputs
ServiceMgr.EventSelector.InputCollections = [tmpCollFile + ".root"]
ServiceMgr.EventSelector.CollectionType = "RootCollection"