trigbs_createHLTBSTestFile.py

Go to the documentation of this file.

#!/usr/bin/env python
 
# Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
"""
Prepare bytestream files for nightly and online tests
 
Renumbers all event ids, runnumbers and lumiblocks to create test files.
The MUCTPi fragment normally sent through the RoIB is also recreated, though
that should only be necessary for 2009 data
 
Note that a bunch of information is hardcoded and could change if
the data format of the CTP/MUCTPI/L1Calo ROIB fragments change.
"""
 
import os
import sys
 
import eformat
import libpyevent_storage as EventStorage
 
id=0
lumi=0
runno=0
runs=[]
def modify(event):
  """Fills in the MUCTPi ROB)."""
  global id
  global runs
  global runno
  global lumi
  id+=1
  
  map=[[eformat.helper.SubDetector.TDAQ_MUON_CTP_INTERFACE,0,1]]
  DaqRobs=[None]*len(map)
  
  for rob in event:
      cnt=0
      for match in map:
          if rob.source_id()==eformat.helper.SourceIdentifier(match[0],match[1]):
              DaqRobs[cnt]=rob
          cnt+=1
 
 
  #work around corrupted events:
  #new_event=eformat.write.FullEventFragment(event)
  new_event=eformat.write.FullEventFragment()
  new_event.copy_header(event)
  if not event.run_no() in runs:
    runs.append(event.run_no())
    lumi+=1
  new_event.run_no(runno) #fake it to avoid complaints and get latest conditions
  new_event.lumi_block(lumi)
  new_event.lvl1_id(id)
  new_event.global_id(id)
  for rob in event:
    keep=True
    for match in map:
      if rob.source_id()==eformat.helper.SourceIdentifier(match[0],match[2]):
        keep=False
        break
    if keep:
      newrob=eformat.write.ROBFragment(rob)
      newrob.rod_lvl1_id(id)
      newrob.rod_run_no(runno)
      newrob.checksum_type(0)
      newrob.rod_lvl1_trigger_type(event.lvl1_trigger_type())
      if rob.source_id()==eformat.helper.SourceIdentifier(eformat.helper.SubDetector.TDAQ_CTP,1):
        newrob.rod_detev_type(1<<16|lumi)
      new_event.append(newrob)
  for idx in range(len(map)):
      if DaqRobs[idx]:
          rob=eformat.write.ROBFragment(DaqRobs[idx])
          rob.source_id(eformat.helper.SourceIdentifier(map[idx][0],map[idx][2]))
          rob.rod_lvl1_id(id)
          rob.rod_run_no(runno)
          rob.checksum_type(0)
          rob.rod_lvl1_trigger_type(event.lvl1_trigger_type())
          if map[idx][0]==eformat.helper.SubDetector.TDAQ_CTP:
              version=rob.rod_minor_version()&0x1f
              if version:
                ctpSkip=(rob.rod_minor_version()>>8)&0x3f
              else:
                ctpSkip=(rob.rod_detev_type()>>16)&0x3f
              data=[DaqRobs[idx].rod_data()[0],DaqRobs[idx].rod_data()[1]]
              for ii in range(30):
                  data+=[DaqRobs[idx].rod_data()[2+ctpSkip*30+ii]]
              rob.rod_data(data)
          if map[idx][0]==eformat.helper.SubDetector.TDAQ_MUON_CTP_INTERFACE:
              data=[]
              muctpBC=(DaqRobs[idx].rod_data()[0]>>18)&7
              for word in DaqRobs[idx].rod_data()[1:]:
                  if (word>>14)&7==muctpBC and (word>>26)&1:
                      data+=[(word&0x3fff)|(((word>>17)&0x1ff)<<14)]
              if len(data)==0:
                  rob=eformat.write.ROBFragment()
                  rob.copy_header(DaqRobs[idx])
                  rob.source_id(eformat.helper.SourceIdentifier(map[idx][0],map[idx][2]))
              else:
                  rob.rod_data(data)
          new_event.append(rob)
      else:
          rob=eformat.write.ROBFragment()
          rob.source_id(eformat.helper.SourceIdentifier(map[idx][0],map[idx][2]))
          new_event.append(rob)
  return new_event.readonly()
 
if __name__ == "__main__":
  if len(sys.argv)!=4:
    print('usage: %s <infile> <outfile> <runno>' % sys.argv[0])
    sys.exit(1)
  global runno
  
  input_file = sys.argv[1]
  output_file = sys.argv[2]
  runno = int(sys.argv[3])
  
  input = eformat.istream([input_file])
  dr=EventStorage.pickDataReader(input_file)
  output = eformat.ostream(core_name="subset",
                           run_number=runno,
                           trigger_type=dr.triggerType(),
                           detector_mask=dr.detectorMask(),
                           beam_type=dr.beamType(),
                           beam_energy=dr.beamEnergy())
  for event in input:
    output.write(modify(event))
  tmp_file_name = output.last_filename()
  del output
  os.rename(tmp_file_name,output_file)