ATLAS Offline Software
Functions | Variables
python.PATTransformUtils Namespace Reference

Functions

def addPhysValidationFiles (parser)
 
def addPhysValidationMergeFiles (parser)
 
def addNTUPMergeSubsteps (executorSet)
 
def addDAODArguments (parser, mergerTrf=True)
 Import list of known DAODs from the derivation framework and. More...
 
def addDAODMergerSubsteps (executorSet)
 
def knownDAODTypes ()
 

Variables

 msg = msg.getChild(__name__)
 

Function Documentation

◆ addDAODArguments()

def python.PATTransformUtils.addDAODArguments (   parser,
  mergerTrf = True 
)

Import list of known DAODs from the derivation framework and.

Definition at line 52 of file PATTransformUtils.py.

52 def addDAODArguments(parser, mergerTrf=True):
53  DAODTypes = knownDAODTypes()
54  if mergerTrf:
55  parser.defineArgGroup('Input DAOD', 'Input DAOD files to be merged')
56  parser.defineArgGroup('Output DAOD', 'Output merged DAOD files')
57  for DAOD in DAODTypes:
58  parser.add_argument("--input" + DAOD + "File", nargs="+",
59  type=trfArgClasses.argFactory(trfArgClasses.argPOOLFile, io="input", type="AOD", subtype=DAOD),
60  help="Input DAOD file of " + DAOD + " derivation", group="Input DAOD")
61  parser.add_argument("--output" + DAOD + "_MRGFile",
62  type=trfArgClasses.argFactory(trfArgClasses.argPOOLFile, io="output", type="AOD", subtype=DAOD),
63  help="Output merged DAOD file of " + DAOD + " derivation", group="Output DAOD")
64  else:
65  parser.defineArgGroup('Output DAOD', 'Output derivation DAOD files')
66  for DAOD in DAODTypes:
67  parser.add_argument("--output" + DAOD + "File",
68  type=trfArgClasses.argFactory(trfArgClasses.argPOOLFile, io="output", type="AOD", subtype=DAOD),
69  help="Output DAOD file of " + DAOD + " derivation", group="Output DAOD")
70 
71 

◆ addDAODMergerSubsteps()

def python.PATTransformUtils.addDAODMergerSubsteps (   executorSet)

Definition at line 72 of file PATTransformUtils.py.

72 def addDAODMergerSubsteps(executorSet):
73  DAODTypes = knownDAODTypes()
74  for DAOD in DAODTypes:
75  executorSet.add(POOLMergeExecutor(name = DAOD.removeprefix("DAOD_") + 'Merge', inData = [DAOD], outData = [DAOD+'_MRG']))
76 

◆ addNTUPMergeSubsteps()

def python.PATTransformUtils.addNTUPMergeSubsteps (   executorSet)

Definition at line 32 of file PATTransformUtils.py.

32 def addNTUPMergeSubsteps(executorSet):
33  # Ye olde NTUPs
34  try:
35  # 'Standard' D3PDs
36  inDataList, outDataList = listKnownD3PDs()
37  for (inData, outData) in iter(zip(inDataList, outDataList)):
38  executorSet.add(NTUPMergeExecutor(name='NTUPLEMerge'+inData.replace('_',''), exe='hadd', inData=[inData], outData=[outData], exeArgs=[]))
39  # Physics Validation NTUP
40  executorSet.add(NTUPMergeExecutor(name='NTUPLEMergePHYSVAL', exe='hadd', inData=['NTUP_PHYSVAL'], outData=['NTUP_PHYSVAL_MRG0'], exeArgs=[]))
41  #add post processing for some Pull histograms in ID track monitoring
42  executorSet.add(NTUPMergeExecutor(name='NTUPLEMergePHYSVALPostProc', exe='postProcessIDPVMHistos', inData=['NTUP_PHYSVAL_MRG0'], outData=['NTUP_PHYSVAL_MRG'], exeArgs=[]))
43  # Extra Tier-0 NTUPs
44  extraNTUPs = getExtraDPDList(NTUPOnly = True)
45  for ntup in extraNTUPs:
46  executorSet.add(NTUPMergeExecutor(name='NTUPLEMerge'+ntup.name.replace('_',''), exe='hadd', inData=[ntup.name], outData=[ntup.name+'_MRG'], exeArgs=[]))
47  except ImportError as e:
48  msg.warning("Failed to get D3PD lists - probably D3PDs are broken in this release: {0}".format(e))
49 
50 

◆ addPhysValidationFiles()

def python.PATTransformUtils.addPhysValidationFiles (   parser)

Definition at line 15 of file PATTransformUtils.py.

15 def addPhysValidationFiles(parser):
16  # Better to somehow auto-import this from PhysicsAnalysis/PhysicsValidation/PhysValMonitoring
17  parser.defineArgGroup('Validation Files', 'Physics validation file options')
18  parser.add_argument('--outputNTUP_PHYSVALFile',
19  type=trfArgClasses.argFactory(trfArgClasses.argNTUPFile, io='output'),
20  help='Output physics validation file', group='Validation Files')
21 

◆ addPhysValidationMergeFiles()

def python.PATTransformUtils.addPhysValidationMergeFiles (   parser)

Definition at line 22 of file PATTransformUtils.py.

22 def addPhysValidationMergeFiles(parser):
23  # Better to somehow auto-import this from PhysicsAnalysis/PhysicsValidation/PhysValMonitoring
24  parser.defineArgGroup('Validation Files', 'Physics validation file options')
25  parser.add_argument('--inputNTUP_PHYSVALFile',
26  type=trfArgClasses.argFactory(trfArgClasses.argNTUPFile, io='input'),
27  help='Input physics validation file', group='Validation Files', nargs='+')
28  parser.add_argument('--outputNTUP_PHYSVAL_MRGFile',
29  type=trfArgClasses.argFactory(trfArgClasses.argNTUPFile, io='output'),
30  help='Output merged physics validation file', group='Validation Files')
31 

◆ knownDAODTypes()

def python.PATTransformUtils.knownDAODTypes ( )

Definition at line 77 of file PATTransformUtils.py.

77 def knownDAODTypes():
78  DAODTypes = []
79  try:
80  from DerivationFrameworkCore.DerivationFrameworkProdFlags import listAODtoDPD
81  DAODTypes = [ name.lstrip("Stream") for name in listAODtoDPD ]
82  except ImportError:
83  msg.warning("Could not import DAOD subtypes from DerivationFramework.DerivationFrameworkCore")
84  return DAODTypes

Variable Documentation

◆ msg

python.PATTransformUtils.msg = msg.getChild(__name__)

Definition at line 7 of file PATTransformUtils.py.

python.PATTransformUtils.addDAODArguments
def addDAODArguments(parser, mergerTrf=True)
Import list of known DAODs from the derivation framework and.
Definition: PATTransformUtils.py:52
vtune_athena.format
format
Definition: vtune_athena.py:14
python.trfArgs.listKnownD3PDs
def listKnownD3PDs()
This method adds the current valid list of D3PDs to two lists.
Definition: trfArgs.py:583
python.trfArgs.getExtraDPDList
def getExtraDPDList(NTUPOnly=False)
Definition: trfArgs.py:444
python.PATTransformUtils.addPhysValidationFiles
def addPhysValidationFiles(parser)
Definition: PATTransformUtils.py:15
python.PATTransformUtils.addNTUPMergeSubsteps
def addNTUPMergeSubsteps(executorSet)
Definition: PATTransformUtils.py:32
python.PATTransformUtils.addPhysValidationMergeFiles
def addPhysValidationMergeFiles(parser)
Definition: PATTransformUtils.py:22
python.PATTransformUtils.knownDAODTypes
def knownDAODTypes()
Definition: PATTransformUtils.py:77
python.PATTransformUtils.addDAODMergerSubsteps
def addDAODMergerSubsteps(executorSet)
Definition: PATTransformUtils.py:72