OnlineImport2.py

#!/usr/bin/env python3
 
 
'''
DQM RootImport test
Test import for Online system root file
with histograms in subdirs
'''
import os
import basf2 as b2
from ROOT import Belle2, TFile, TH1F, gROOT
 
gROOT.SetBatch(True)
 
filein = "histino2.root"
fileout = 'histouto2.root'
statfile = "statso2.txt"
 
f = TFile(filein, "RECREATE")
 
f.mkdir("DQMInfo")
f.cd("DQMInfo")
h_expno = TH1F("expno", "1", 1, 0, 1)
h_runno = TH1F("runno", "1", 1, 0, 1)
h_rtype = TH1F("rtype", "null", 1, 0, 1)
f.cd("..")
 
f.mkdir("TEST")
f.cd("TEST")
h_test = TH1F("test", "", 1, 0, 1)
f.cd("..")
 
f.mkdir("DAQ")
f.cd("DAQ")
h_nevent = TH1F("Nevent", "", 1, 0, 1)
for n in range(0, 10):
    h_nevent.Fill(n)
f.cd("..")
 
f.Write()
f.Close()
 
Belle2.Environment.Instance().setNumberEventsOverride(1)
 
main = b2.create_path()
emptypath = b2.create_path()
 
dqminput = b2.register_module('DQMHistAnalysisInput2')
dqminput.param('HistMemoryPath', filein)
dqminput.param('RefreshInterval', 0)
dqminput.param('StatFileName', statfile)
dqminput.param("EnableRunInfo", True)
dqminput.if_false(emptypath)
main.add_module(dqminput)
 
main.add_module("DQMHistAutoCanvas")
 
dqmoutput = b2.register_module('DQMHistAnalysisOutputFile')
dqmoutput.param('OutputFolder', './')
dqmoutput.param('Filename', fileout)
main.add_module(dqmoutput)
 
# Process all events
b2.process(main)
 
expected = ["DQMInfo/c_info", "DAQ/c_Nevent", "DQMInfo/c_expno", "DQMInfo/c_runno", "DQMInfo/c_rtype", "TEST/c_test"]
b2.B2INFO("== resulting file content ==")
f = TFile(fileout, "READ")
for k in f.GetListOfKeys():
    o = k.ReadObj()
    b2.B2INFO(o.ClassName(), k)
    if o.GetName() == "DQMInfo/c_info":
        if "Exp 1, Run 1, RunType null" not in o.GetTitle():
            b2.B2ERROR(f"Run Info not found in {o.GetName()}: {o.GetTitle()}")
    if o.GetName() in expected:
        expected.remove(o.GetName())
b2.B2INFO("============================")
if len(expected) > 0:
    b2.B2ERROR("missing items in outfile: ", expected)
b2.B2INFO("== resulting stat content ==")
with open(statfile, 'r') as f:
    b2.B2INFO(f.read())
b2.B2INFO("=========================")
 
os.remove(statfile)
os.remove(filein)
os.remove(fileout)