RecoTrack/test/analyseDuplicateFake.py

0001 #!/usr/bin/env python3
0002
0003 import ROOT
0004
0005 from Validation.RecoTrack.plotting.ntuple import *
0006
0007 # The purpose of this file is to demonstrate the printout features of the ntuple library
0008
0009 def main():
0010     ntuple = TrackingNtuple("trackingNtuple.root")
0011
0012     printTrack = TrackPrinter(trackingParticlePrinter=TrackingParticlePrinter())
0013     printTP = TrackingParticlePrinter(trackPrinter=TrackPrinter())
0014
0015     for event in ntuple:
0016         print("Event", event.eventIdStr())
0017         print("Fake tracks")
0018         for track in event.tracks():
0019             if track.nMatchedTrackingParticles() == 0:
0020                 printTrack(track)
0021                 print()
0022
0023         print("Duplicate tracks")
0024         for tp in event.trackingParticles():
0025             if tp.nMatchedTracks() >= 2:
0026                 printTP(tp)
0027                 print()
0028         print()
0029
0030         if event.entry() >= 1:
0031             break
0032
0033 if __name__ == "__main__":
0034     main()