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
 
"""
 
import sys
import time
 
from Configurables import ApplicationMgr, EventSelector, Gaudi__RootCnvSvc
 
from Gaudi.Configuration import INFO, EventDataSvc, EventPersistencySvc, MessageSvc
 
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(f"{s:>15s} : {(s1[s] - s0[s]):10.2F} sec")
        else:
            print(f"{s:>15s} : {(s1[s] - s0[s])/ 1.0e6:10.2F} MB")
 
 
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(f"Read {N} events")
printDelta(start, end)