LArMerge_trf Namespace Reference

Functions
def	getFileMap (fname, dsname, nevts=0, guid='')
def	larMerge (dataMap)

Variables
	picklefile = sys.argv[1][len('--argdict='):]
	main() More...
	f = open(picklefile, 'rb')
	dataMap = pickle.load(f)
	inputFiles = sys.argv[1][13:].split(',')
	outputFile = sys.argv[2][13:]

Function Documentation

getFileMap()

def LArMerge_trf.getFileMap	(		fname,
			dsname,
			nevts = 0,
			guid = ''
	)

Definition at line 30 of file LArMerge_trf.py.

def getFileMap(fname, dsname, nevts=0, guid='') :
  if os.path.isfile(fname) :
    dataMap = { 'lfn': fname,
                'dataset' : dsname,
                'events' : nevts
          }
    if guid : 
      dataMap['GUID'] = guid      
  else : 
    dataMap = {}
  return dataMap
 

larMerge()

def LArMerge_trf.larMerge

(

dataMap

)

Definition at line 45 of file LArMerge_trf.py.

def larMerge(dataMap) :
 
  print ("\n##################################################################")
  print (  "##                 ATLAS Tier0 LAr CAF file Merging             ##")
  print (  "##################################################################\n")
 
  print ("\nFull Tier-0 run options:\n")
  pprint.pprint(dataMap)
  
  inputList = []
  inputDSName = None
 
  for val in dataMap['inputLArFiles'] :
    inputList.append(val.split('#')[1]) 
    if not inputDSName :
      inputDSName = val.split('#')[0]
    
  print ("\ninputLArFiles list:\n")
  pprint.pprint(inputList)
 
  # output file
  outputDSName = dataMap['outputLArFile'].split('#')[0]
  outputFile = dataMap['outputLArFile'].split('#')[1]
 
  print ('\nOutput file name:', outputFile)
 
  retcode = 0
  acronym = 'OK'
 
  outFiles = []
  inFiles = []
  nEvents=0
 
  cmdline=["LArSamplesMerge.exe"]
  for fileName in inputList:
    if os.access(fileName,os.R_OK):
      cmdline.append(fileName)
    else:
      print ('ERROR : could not open file', fileName)
      retcode = 74001
      acronym = 'TRF_LAR_FILE_INPUT_ERROR'
      break
    pass
  
  if (retcode==0):
    writepath=os.path.split(outputFile)[0]
    if writepath=="":
      writepath="./"
    if (os.access(writepath,os.W_OK)):
      cmdline.append(outputFile)
    else:
      print ("ERROR, not allowed to write to oututfile", outputFile)
      retcode = 74002
      acronym = 'TRF_LAR_MERGE_ERROR'
      pass
    pass
  
  if (retcode==0):
 
    logfile=open("log.LArMerge","w")
 
    try:
      subprocess.check_call(cmdline,stdout=logfile,stderr=subprocess.STDOUT)
    except Exception as e:
      print ("ERROR exectution of subprocess failed")
      print (e)
      acronym = 'TRF_LAR_MERGE_ERROR'
      pass
 
    logfile.close()
    logfile=open("log.LArMerge","r")
 
    for line in logfile:
      print (line, end='')
      if line.startswith("Open file "):
        linetok=line.split(" ")
        inFiles.append(getFileMap(linetok[2], None, nevts=int(linetok[4])))
        pass
      if line.startswith("Wrote output file "):
        linetok=line.split(" ")
        nEvents=int(linetok[5])
        outFiles=[getFileMap(linetok[3], outputDSName, nevts=nEvents),]
        pass
      
    logfile.close()
 
    
 
  # assemble job report map, pickle it
  outMap = { 'prodsys': { 'trfCode': retcode,
                          'trfAcronym': acronym,  
                          'jobOutputs': outFiles,
                          'jobInputs': inFiles,
                          'nevents': nEvents,              
                        }
           }              
  f = open('jobReport.gpickle', 'wb')
  pickle.dump(outMap, f)
  f.close()
 
  print ("\n##################################################################")
  print (  "## End of job.")
  print (  "##################################################################\n")
 
  print (outMap)
 

Variable Documentation

dataMap

dictionary LArMerge_trf.dataMap = pickle.load(f)

Definition at line 162 of file LArMerge_trf.py.

f

LArMerge_trf.f = open(picklefile, 'rb')

Definition at line 161 of file LArMerge_trf.py.

inputFiles

LArMerge_trf.inputFiles = sys.argv[1][13:].split(',')

Definition at line 166 of file LArMerge_trf.py.

outputFile

LArMerge_trf.outputFile = sys.argv[2][13:]

Definition at line 167 of file LArMerge_trf.py.

picklefile

LArMerge_trf.picklefile = sys.argv[1][len('--argdict='):]

main()

Definition at line 158 of file LArMerge_trf.py.