HitCleaner Class Reference

Inheritance diagram for HitCleaner:

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

Public Attributes
	number_of_tracks
	count the tracks
	number_of_deleted_hits
	count the deleted hits
	number_of_hits
	count the hits
	cdc_hit_look_up
	function to look up CDC MC hits
	mc_matcher_lookup
	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__()

def __init__

(

self

)

Constructor

Definition at line 76 of file modules.py.

    def __init__(self):
        """Constructor"""
        super().__init__()
 
        
        self.number_of_tracks = 0
        
        self.number_of_deleted_hits = 0
        
        self.number_of_hits = 0
 

Member Function Documentation

event()

def 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()

def 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()

def 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

function to look up CDC MC hits

Definition at line 90 of file modules.py.

mc_matcher_lookup

mc_matcher_lookup

function to match track candidatess to MC track candidates

Definition at line 92 of file modules.py.

number_of_deleted_hits

number_of_deleted_hits

count the deleted hits

Definition at line 83 of file modules.py.

number_of_hits

number_of_hits

count the hits

Definition at line 85 of file modules.py.

number_of_tracks

number_of_tracks

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