SegmentFinderParameterExtractorModule Class Reference

Inheritance diagram for SegmentFinderParameterExtractorModule:

Public Member Functions
def	__init__ (self, local_track_cands_store_array_name, mc_track_cands_store_array_name, output_file_name)
def	peel (self, local_track_cand)

Public Attributes
	mc_track_cands_store_array_name
	cached name of the TrackCands StoreArray
	mc_track_matcher
	function to match tracks to MC tracks

Static Public Attributes
refiners	save_tree
	Refiners to be executed at the end of the harvesting / termination of the module Save a tree of all collected variables in a sub folder.

Detailed Description

Harvester module to extract momentum and position information from the found segments and tracks.
You need the apply MC track matcher module.

Definition at line 273 of file harvester.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			local_track_cands_store_array_name,
			mc_track_cands_store_array_name,
			output_file_name
	)

 Init the harvester with the local track candidates StoreArray name and the one for MC track cands
and the output file name for the result root file.

Reimplemented from HarvestingModule.

Definition at line 280 of file harvester.py.

    def __init__(self, local_track_cands_store_array_name, mc_track_cands_store_array_name, output_file_name):
        """ Init the harvester with the local track candidates StoreArray name and the one for MC track cands
        and the output file name for the result root file.
        """
        super().__init__(
            foreach=local_track_cands_store_array_name,
            output_file_name=output_file_name)
 
        
        self.mc_track_cands_store_array_name = mc_track_cands_store_array_name
 
        
        self.mc_track_matcher = Belle2.TrackMatchLookUp(self.mc_track_cands_store_array_name,
                                                        self.foreach)
 

Member Function Documentation

peel()

def peel	(		self,
			local_track_cand
	)

 Extract the information from the local track candidate. 

Reimplemented from HarvestingModule.

Definition at line 295 of file harvester.py.

    def peel(self, local_track_cand):
        """ Extract the information from the local track candidate. """
        mc_track_matcher = self.mc_track_matcher
 
        is_matched = mc_track_matcher.isAnyChargeMatchedPRRecoTrack(local_track_cand)
        is_background = mc_track_matcher.isBackgroundPRRecoTrack(local_track_cand)
        is_ghost = mc_track_matcher.isGhostPRRecoTrack(local_track_cand)
 
        related_mc_track_cand = mc_track_matcher.getRelatedMCRecoTrack(local_track_cand)
 
        track_momentum = np.NaN
        # track_position = np.NaN
        track_pt = np.NaN
        track_phi = np.NaN
 
        segment_momentum = np.NaN
        # segment_position = np.NaN
        segment_pt = np.NaN
        segment_phi = np.NaN
 
        if is_matched:
            track_momentum = related_mc_track_cand.getPosSeed()
            # track_position = related_mc_track_cand.getPosSeed()
            track_pt = track_momentum.Pt()
            track_phi = track_momentum.Phi()
            # trajectory_track = \
            #     Belle2.TrackFindingCDC.CDCTrajectory3D(Belle2.TrackFindingCDC.Vector3D(track_position),
            #                                            Belle2.TrackFindingCDC.Vector3D(track_momentum),
            #                                            related_mc_track_cand.getChargeSeed())
 
            segment_momentum = local_track_cand.getMomSeed()
            # segment_position = local_track_cand.getPosSeed()
            segment_pt = segment_momentum.Pt()
            segment_phi = segment_momentum.Phi()
            # trajectory_segment = \
            #     Belle2.TrackFindingCDC.CDCTrajectory3D(Belle2.TrackFindingCDC.Vector3D(segment_position),
            #                                            Belle2.TrackFindingCDC.Vector3D(segment_momentum),
            #                                            local_track_cand.getChargeSeed())
 
        return dict(is_matched=is_matched,
                    is_background=is_background,
                    is_ghost=is_ghost,
                    segment_pt=segment_pt,
                    track_pt=track_pt,
                    difference_pt=segment_pt - track_pt,
                    segment_phi=segment_phi,
                    track_phi=track_phi,
                    difference_phi=segment_phi - track_phi)
 

Member Data Documentation

mc_track_cands_store_array_name

mc_track_cands_store_array_name

cached name of the TrackCands StoreArray

Definition at line 289 of file harvester.py.

mc_track_matcher

mc_track_matcher

function to match tracks to MC tracks

Definition at line 292 of file harvester.py.

save_tree

refiners save_tree

static

Initial value:

=  refiners.save_tree(
        
 
 
    )

Refiners to be executed at the end of the harvesting / termination of the module Save a tree of all collected variables in a sub folder.

Definition at line 346 of file harvester.py.

---

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

• tracking/trackFindingCDC/scripts/trackfindingcdc/harvest/harvester.py