trigbs_mixBSevents.py

Go to the documentation of this file.

#!/usr/bin/env python
 
# Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
 
from __future__ import print_function
import eformat
import libpyevent_storage as EventStorage
import sys
import getopt
 
eventsPerFile=500
verbose=False
 
def usage():
  print(" usage: %s [options] <outbasename> <file1> <file2> <file3>" % sys.argv[0])
  print("    Rewrites events with event ids numbered consequtively from 1")
  print("    This is mainly used for preparing events from playback partition")
  print()
  print("    Options:")
  print("       -n <number of events per file> : defaults to 500")
  print("       -v     : verbose output")
  sys.exit(1)
  
try:
  opts, args = getopt.getopt(sys.argv[1:], "vn:")
  if len(args)<2:
    usage()
except getopt:
  usage()
 
for opt,arg in opts:
  if opt == '-n':
    eventsPerFile=int(arg)
  if opt == '-v':
    verbose=True
 
basename=args[0]
files = args[1:]
 
ifile = 0
id = 1
outnum=1
dr=EventStorage.pickDataReader(files[0])
numFiles=0
output = eformat.ostream(core_name=basename,
                         run_number=dr.runNumber(),
                         trigger_type=dr.triggerType(),
                         detector_mask=dr.detectorMask(),
                         beam_type=dr.beamType(),
                         beam_energy=dr.beamEnergy())
output.max_events_per_file(eventsPerFile)
 
for input_file in files:
  ifile += 1
  print("Opening file %d: %s" % (ifile, input_file))
 
  input = eformat.istream([input_file])
 
  for event in input:
    if verbose:
      print("Reading event with Lvl1 Id = %ld and rewritting with %ld" % (event.lvl1_id(),id))
    new_event = eformat.write.FullEventFragment(event)
    new_event.lvl1_id(id)
    new_event.global_id(id)
    for rob in new_event.children():
      rob.rod_lvl1_id(id)
      rob.checksum_type(0)  #remove checksum
      rob.rod_lvl1_trigger_type(event.lvl1_trigger_type())
 
    output.write(new_event)
 
    id+=1
 
print('Wrote %d events to %d files' % ( id-1, 1+((id-1)//eventsPerFile) ))