printPIDLikelihoods Class Reference

Inheritance diagram for printPIDLikelihoods:

Collaboration diagram for printPIDLikelihoods:

Public Member Functions
def	event (self)

Detailed Description

Prints PID log likelihoods + basic track info in a well formatted way.
User is prompted to continue or quit at each event

Definition at line 22 of file printPIDLikelihoods.py.

Member Function Documentation

event()

def event

(

self

)

Print log likelihoods and wait for user respond.

Definition at line 29 of file printPIDLikelihoods.py.

    def event(self):
        '''
        Print log likelihoods and wait for user respond.
        '''
 
        evtMetaData = Belle2.PyStoreObj('EventMetaData')
        exp = evtMetaData.obj().getExperiment()
        run = evtMetaData.obj().getRun()
        evt = evtMetaData.obj().getEvent()
        print()
        print('Experiment ' + str(exp) + ' Run ' + str(run) + ' Event ' + str(evt) + ':')
        print()
        tracks = Belle2.PyStoreArray('Tracks')
        for track in tracks:
            fitresult = track.getTrackFitResult(Belle2.Const.pion)
            if fitresult:
                p_mag = fitresult.getMomentum().Mag()
                pmom = '{:5.3f}'.format(p_mag) + ' GeV/c'
            else:
                pmom = '?'
            mcpart = track.getRelated('MCParticles')
            if mcpart:
                pdg = str(mcpart.getPDG())
            else:
                pdg = '?'
            index = track.getArrayIndex()
            print('Track ' + str(index) + ': p = ' + pmom + '  MCtruth = ' + pdg)
            likelihood = track.getRelated('PIDLikelihoods')
            try:
                print('logLikelihoods:')
                likelihood.printArray()
            except BaseException:
                print('--> No relation to PIDLikelihood')
            print()
 
        # wait for user respond
        response = input("Type Enter to continue or Q to quit.\n").lower().strip()
        if response == "q":
            evtMetaData.obj().setEndOfData()
 
 
b2.set_log_level(b2.LogLevel.ERROR)
 
# Create path

---

The documentation for this class was generated from the following file:

• mdst/examples/printPIDLikelihoods.py