HitCleaner Class Reference

Inheritance diagram for HitCleaner:

Public Member Functions
	__init__ (self)
	initialize (self)
	event (self)
	terminate (self)

Public Attributes
int	number_of_tracks = 0
	count the tracks
int	number_of_deleted_hits = 0
	count the deleted hits
int	number_of_hits = 0
	count the hits
	cdc_hit_look_up = Belle2.TrackFindingCDC.CDCMCHitLookUp()
	function to look up CDC MC hits
	mc_matcher_lookup = Belle2.TrackMatchLookUp("MCTrackCands", "TrackCands")
	function to match track candidatess to MC track candidates

Detailed Description

A small hit cleaner module to set the track information according to mc information.
This is surely not for later usage but for testing the genfitter module

Definition at line 71 of file modules.py.

Constructor & Destructor Documentation

__init__()

__init__

(

self

)

Constructor

Definition at line 76 of file modules.py.

    def __init__(self):
        """Constructor"""
        super().__init__()
 
        ## count the tracks
        self.number_of_tracks = 0
        ## count the deleted hits
        self.number_of_deleted_hits = 0
        ## count the hits
        self.number_of_hits = 0
 

Member Function Documentation

event()

event

(

self

)

Event method of the module

Definition at line 94 of file modules.py.

    def event(self):
        """Event method of the module"""
        tracks = Belle2.PyStoreArray("TrackCands")
        cdc_hits = Belle2.PyStoreArray("CDCHits")
        mc_particles = Belle2.PyStoreArray("MCParticles")
 
        cdc_hit_lookup = self.cdc_hit_look_up
        cdc_hit_lookup.fill()
 
        mc_matcher_lookup = self.mc_matcher_lookup
 
        self.number_of_tracks += tracks.getEntries()
 
        for track in tracks:
            # Store all Hit IDs and reset the track
            hitIDs = track.getHitIDs(Belle2.Const.CDC)
            # hits = [cdc_hits[i] for i in hitIDs]
            good_hits = []
 
            relation_track_particle = [0] * mc_particles.getEntries()
 
            # Now only add those hits which do belong to the track (with MC Info)
            for i in hitIDs:
                current_mc_track = cdc_hit_lookup.getMCTrackId(cdc_hits[i])
                if 0 <= current_mc_track:
                    relation_track_particle[current_mc_track] += 1
 
            should_belong_to_track = np.argmax(relation_track_particle)
 
            deleted_hits = sum(relation_track_particle) - relation_track_particle[should_belong_to_track]
 
            # plane_IDs_of_good_hits = []
            for i, hitID in enumerate(hitIDs):
                current_mc_track = cdc_hit_lookup.getMCTrackId(cdc_hits[hitID])
                if current_mc_track == should_belong_to_track:
                    good_hits.append(hitID)
 
            self.number_of_deleted_hits += deleted_hits
            self.number_of_hits += len(hitIDs)
 
            # Set the position and momentum
            mc_track = mc_matcher_lookup.getAnyChargeMatchedMCRecoTrack(track)
 
            if mc_track:
                mc_trajectory = Belle2.TrackFindingCDC.CDCTrajectory3D(Belle2.TrackFindingCDC.Vector3D(mc_track.getPosSeed()),
                                                                       Belle2.TrackFindingCDC.Vector3D(mc_track.getMomSeed()),
                                                                       mc_track.getChargeSeed())
                startingPosition = Belle2.TrackFindingCDC.Vector3D(track.getPosSeed().X(), track.getPosSeed().Y(), 0)
                sStartingPosition = mc_trajectory.calcArcLength2D(startingPosition)
                zStartingPosition = mc_trajectory.getTrajectorySZ().mapSToZ(sStartingPosition)
                mc_trajectory.setLocalOrigin(Belle2.TrackFindingCDC.Vector3D(startingPosition.xy(), zStartingPosition))
 
                # pos = ROOT.TVector3(
                #     mc_trajectory.getSupport().x(), mc_trajectory.getSupport().y(), mc_trajectory.getSupport().z())
                # mom = ROOT.TVector3(
                #     mc_trajectory.getMom3DAtSupport().x(),
                #     mc_trajectory.getMom3DAtSupport().y(),
                #     mc_trajectory.getMom3DAtSupport().z())
 
                # track.setPosMomSeedAndPdgCode(pos, mom , int(mc_track.getChargeSeed() * 211))
                track.setPdgCode(int(track.getChargeSeed() * 211))
 
            else:
                # track.reset()
                track.setPdgCode(int(track.getChargeSeed() * 211))
 

initialize()

initialize

(

self

)

Initialization signal at the start of the event processing

Definition at line 87 of file modules.py.

    def initialize(self):
        """Initialization signal at the start of the event processing"""
        
        self.cdc_hit_look_up = Belle2.TrackFindingCDC.CDCMCHitLookUp()
        
        self.mc_matcher_lookup = Belle2.TrackMatchLookUp("MCTrackCands", "TrackCands")
 

terminate()

terminate

(

self

)

Termination signal at the end of the event processing

Definition at line 160 of file modules.py.

    def terminate(self):
        """Termination signal at the end of the event processing"""
        print(f"Number of tracks in total: {int(self.number_of_tracks)}")
        print(f"Number of hits in total: {int(self.number_of_hits)}")
        print(f"Number of deleted hits: {int(self.number_of_deleted_hits)}")
 
        print(f"Number of deleted hits per track: {1.0 * self.number_of_deleted_hits / self.number_of_tracks:f}")
        print(f"Ratio of deleted hits: {100.0 * self.number_of_deleted_hits / self.number_of_hits:f}")

Member Data Documentation

cdc_hit_look_up

cdc_hit_look_up = Belle2.TrackFindingCDC.CDCMCHitLookUp()

function to look up CDC MC hits

Definition at line 90 of file modules.py.

mc_matcher_lookup

mc_matcher_lookup = Belle2.TrackMatchLookUp("MCTrackCands", "TrackCands")

function to match track candidatess to MC track candidates

Definition at line 92 of file modules.py.

number_of_deleted_hits

int number_of_deleted_hits = 0

count the deleted hits

Definition at line 83 of file modules.py.

number_of_hits

int number_of_hits = 0

count the hits

Definition at line 85 of file modules.py.

number_of_tracks

int number_of_tracks = 0

count the tracks

Definition at line 81 of file modules.py.

---

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

• tracking/trackFindingCDC/scripts/trackfindingcdc/modules.py