IOTest.py

Go to the documentation of this file.

"""

   Simple test application to read LHCb dst files
   For reading other root files: change the data access
   if '/Event/Rec/Header' is not present.

   Requires: 'SetupProject LHCb'

   Script inherited from T.Ruf
   M.Frank  CERN/LHCb

"""
from __future__ import print_function
import os
import sys
import time
from Configurables import ApplicationMgr, EventSelector, Gaudi__RootCnvSvc
from Gaudi.Configuration import *
from GaudiKernel import *

input_file = 'castor://castorlhcb.cern.ch//castor/cern.ch/grid/lhcb/LHCb/Collision11/SEMILEPTONIC.DST/00012569/0000/00012569_00000004_1.semileptonic.dst?svcClass=lhcbdisk'
input_file = 'root://castorlhcb.cern.ch//castor/cern.ch/grid//lhcb/data/2010/BHADRON.DST/00008399/0000/00008399_00001052_1.bhadron.dst?svcClass=lhcbdisk'

# -----------------------------------------------------------------------------


def storeExplorer(load=1, freq=0.0001, name='StoreExplorerAlg'):
    from Configurables import StoreExplorerAlg
    alg = StoreExplorerAlg(name)
    alg.Load = load
    alg.PrintFreq = freq
    return alg


# -----------------------------------------------------------------------------

if len(sys.argv) > 1:
    input_file = sys.argv[1]

appConf = ApplicationMgr(OutputLevel=INFO)
appConf.HistogramPersistency = "NONE"
appConf.ExtSvc.append('Gaudi::IODataManager/IODataManager')
appConf.ExtSvc.append('Gaudi::RootCnvSvc/RootCnvSvc')
appConf.TopAlg.append(storeExplorer(freq=1.0))

EventDataSvc().RootCLID = 1
EventDataSvc().EnableFaultHandler = True
root = Gaudi__RootCnvSvc('RootCnvSvc')
root.CacheBranches = []
root.VetoBranches = ['*']

#root.OutputLevel  = 2
# Enable specialized branch caching:
root.CacheBranches = [
    '_Event.*', '_Event_R.*', '_Event_Rec.*', '_Event_Rec_R.*',
    '_Event_Rec_Header.*'
    '_Event_Rec_Header_R.*'
]
root.VetoBranches = [
    #    '_Event_pRec_*',
    #    '_Event_Semileptonic_*'
]

EventPersistencySvc().CnvServices.append(root)
EventSelector().Input = [
    "DATA='PFN:" + input_file + "' SVC='Gaudi::RootEvtSelector'"
]
EventSelector().PrintFreq = 1000
MessageSvc().OutputLevel = 3


def update():
    statistic = {}
    test = open('/proc/self/io')
    io = test.readlines()
    test.close()
    for l in io:
        temp = l.split(':')
        statistic[temp[0]] = int(temp[1])
    statistic['time'] = time.time()
    return statistic


def printDelta(s0, s1):
    for s in s1:
        if s == 'time':
            print('%15s : %10.2F sec' % (s, (s1[s] - s0[s])))
        else:
            print('%15s : %10.2F MB' % (s, (s1[s] - s0[s]) / 1.E6))


import GaudiPython
appMgr = GaudiPython.AppMgr()
sel = appMgr.evtsel()
evt = appMgr.evtsvc()
print(sel.Input)

start = update()
# test 1
N = 0
while 1 > 0:
    rc = appMgr.run(1)
    if N == 10:
        start = update()
    # if not evt['/Event/DAQ/RawEvent']:
    #    print 'Failed to access /Event/DAQ/RawEvent'
    if not evt['/Event/Rec/Header']:
        print('Failed to access /Event/Rec/Header')
        break
    N += 1
    if N > 100000:
        break
end = update()
print('Read %d events' % N)
printDelta(start, end)