11 from ROOT
import Belle2
17 Prints PID log likelihoods + basic track info in a well formatted way.
18 User is prompted to continue or quit at each event
23 Print log likelihoods and wait for user respond.
27 exp = evtMetaData.obj().getExperiment()
28 run = evtMetaData.obj().
getRun()
29 evt = evtMetaData.obj().getEvent()
31 print(
'Experiment ' + str(exp) +
' Run ' + str(run) +
' Event ' + str(evt) +
':')
35 fitresult = track.getTrackFitResult(Belle2.Const.pion)
37 p_mag = fitresult.getMomentum().Mag()
38 pmom =
'{:5.3f}'.format(p_mag) +
' GeV/c'
41 mcpart = track.getRelated(
'MCParticles')
43 pdg = str(mcpart.getPDG())
46 index = track.getArrayIndex()
47 print(
'Track ' + str(index) +
': p = ' + pmom +
' MCtruth = ' + pdg)
48 likelihood = track.getRelated(
'PIDLikelihoods')
50 print(
'logLikelihoods:')
51 likelihood.printArray()
53 print(
'--> No relation to PIDLikelihood')
60 abc = input(
'Type <CR> to continue or Q to quit ')
61 evtMetaData.obj().setEndOfData()
66 set_log_level(LogLevel.ERROR)
72 roinput = register_module(
'RootInput')
73 main.add_module(roinput)