HITSMerge_tf.py

Go to the documentation of this file.

#! /usr/bin/env python
 
# Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
 
"""
Merge multiple input HITS files into a single output HITS file.
Optionally run algorithms to tweak the output. Optionally merge Log
file tarballs at the same time.
"""
 
import os.path
import sys
import time
 
import logging
 
# Setup core logging here
from PyJobTransforms.trfLogger import msg
msg.info('logging set in %s' % sys.argv[0])
 
from PyJobTransforms.transform import transform
from PyJobTransforms.trfExe import athenaExecutor
from PyJobTransforms.trfArgs import addAthenaArguments, addDetectorArguments
from SimuJobTransforms.simTrfArgs import addHITSMergeArgs
from PyJobTransforms.trfDecorators import stdTrfExceptionHandler, sigUsrStackTrace
 
import PyJobTransforms.trfArgClasses as trfArgClasses
 
# Prodsys hack...
ListOfDefaultPositionalKeys=['--DBRelease', '--amiConfig', '--amiMetadataTag', '--asetup', '--athena', '--athenaopts', '--beamType', '--checkEventCount', '--command', '--conditionsTag', '--env', '--eventAcceptanceEfficiency', '--execOnly', '--geometryVersion', '--ignoreErrors', '--ignoreFiles', '--ignorePatterns', '--imf', '--inputHITSFile', '--inputLogsFile', '--maxEvents', '--outputHITS_MRGFile', '--postExec', '--postInclude', '--preExec', '--preInclude', '--reportName', '--reportType', '--runNumber', '--showGraph', '--showPath', '--showSteps', '--skipEvents', '--skipFileValidation', '--skipInputFileValidation', '--skipOutputFileValidation', '--tcmalloc']
 
@stdTrfExceptionHandler
@sigUsrStackTrace
def main():
 
    msg.info('This is %s' % sys.argv[0])
 
    trf = getTransform()
    trf.parseCmdLineArgs(sys.argv[1:])
    trf.execute()
    if 'outputFileValidation' in  trf._argdict and  trf._argdict['outputFileValidation'].value is False:
        msg.info('Skipping report generation')
    else:
        trf.generateReport()
 
    msg.info("%s stopped at %s, trf exit code %d" % (sys.argv[0], time.asctime(), trf.exitCode))
    sys.exit(trf.exitCode)
 
def getTransform():
    executorSet = set()
    from SimuJobTransforms.SimTransformUtils import addStandardHITSMergeSubstep, addHITSMergeArguments
    addStandardHITSMergeSubstep(executorSet)
    trf = transform(executor = executorSet, description = 'HITS Merging transform. Inputs must be HITS. Outputs must be HITS_MRG (i.e. HITS). ')
    addAthenaArguments(trf.parser)
    addDetectorArguments(trf.parser)
    addHITSMergeArguments(trf.parser)
    return trf
 

 

 
 
if __name__ == '__main__':
    main()