Inheritance diagram for SegmentFakeRatesModule:

Public Member Functions
def	__init__ (self, output_file_name, local_track_cands_store_array_name="LocalTrackCands", mc_track_cands_store_array_name="MCTrackCands", legendre_track_cand_store_array_name="LegendreTrackCands")

def	prepare (self)

def	peel (self, local_track_cand)

Public Attributes
	mc_track_cands_store_array_name
	cached name of the TrackCands StoreArray

	legendre_track_cand_store_array_name
	cached name of the Legendre TrackCands StoreArray

	mc_track_matcher_local
	function to match local tracks to MC tracks

	mc_track_matcher_legendre
	function to match Legendre tracks to MC tracks

	cdcHits
	cached CDCHits StoreArray

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

Harvesting module to check for basic matching information of the found segments.
If you want to have matching information, please use the MC matcher module to make all possible
combinations between legendre, local and MC track candidates.
Also, the Segments must be transformed into GF track cands.

Definition at line 141 of file harvester.py.

Constructor & Destructor Documentation

◆ init()

def __init__	(	self,
		output_file_name,
		local_track_cands_store_array_name = `"LocalTrackCands"`,
		mc_track_cands_store_array_name = `"MCTrackCands"`,
		legendre_track_cand_store_array_name = `"LegendreTrackCands"`
	)

Initialize the harvesting module with
Arguments
---------
  output_file_name Name of the root file
  local_track_cands_store_array_name Name of the StoreArray for local track cands
  mc_track_cands_store_array_name Name of the StoreArray for MC track cands
  legendre_track_cand_store_array_name Name of the StoreArray for legendre track cands

Reimplemented from HarvestingModule.

Definition at line 150 of file harvester.py.

            legendre_track_cand_store_array_name="LegendreTrackCands"):
        """
        Initialize the harvesting module with
        Arguments
        ---------
          output_file_name Name of the root file
          local_track_cands_store_array_name Name of the StoreArray for local track cands
          mc_track_cands_store_array_name Name of the StoreArray for MC track cands
          legendre_track_cand_store_array_name Name of the StoreArray for legendre track cands
        """
        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.legendre_track_cand_store_array_name = legendre_track_cand_store_array_name
 
        
        self.mc_track_matcher_local = Belle2.TrackMatchLookUp(self.mc_track_cands_store_array_name, self.foreach)
        
        self.mc_track_matcher_legendre = Belle2.TrackMatchLookUp(
            self.mc_track_cands_store_array_name,
            legendre_track_cand_store_array_name)
 

Member Function Documentation

◆ peel()

def peel	(	self,
		local_track_cand
	)

 Extract the information from the segments.

Reimplemented from HarvestingModule.

Definition at line 187 of file harvester.py.

    def peel(self, local_track_cand):
        """ Extract the information from the segments. """
        mc_track_matcher_local = self.mc_track_matcher_local
        mc_track_matcher_legendre = self.mc_track_matcher_legendre
 
        cdc_hits = self.cdcHits
 
        is_background = mc_track_matcher_local.isBackgroundPRRecoTrack(local_track_cand)
        is_ghost = mc_track_matcher_local.isGhostPRRecoTrack(local_track_cand)
        is_matched = mc_track_matcher_local.isAnyChargeMatchedPRRecoTrack(local_track_cand)
        is_clone = mc_track_matcher_local.isAnyChargeClonePRRecoTrack(local_track_cand)
        is_clone_or_matched = is_matched or is_clone
        hit_purity = abs(mc_track_matcher_local.getRelatedPurity(local_track_cand))
 
        # Stereo Track?
        first_cdc_hit_id = local_track_cand.getHitIDs(Belle2.Const.CDC)[0]
        first_cdc_hit = cdc_hits[first_cdc_hit_id]
        is_stereo = first_cdc_hit.getISuperLayer() % 2 == 1
        superlayer_of_segment = first_cdc_hit.getISuperLayer()
 
        has_partner = np.NaN
        hit_purity_of_partner = np.NaN
        hit_efficiency_of_partner = np.NaN
        mc_track_pt = np.NaN
        mc_track_dist = np.NaN
        number_of_new_hits = local_track_cand.getNHits()
        number_of_hits = local_track_cand.getNHits()
        number_of_hits_in_same_superlayer = np.NaN
        partner_has_stereo_information = np.NaN
 
        if is_clone_or_matched:
            related_mc_track_cand = mc_track_matcher_local.getRelatedMCRecoTrack(local_track_cand)
            has_partner = (mc_track_matcher_legendre.isAnyChargeMatchedMCRecoTrack(related_mc_track_cand) or
                           mc_track_matcher_legendre.isAnyChargeMergedMCRecoTrack(related_mc_track_cand))
            mc_track_pt = related_mc_track_cand.getMomSeed().Pt()
            mc_track_dist = related_mc_track_cand.getPosSeed().Mag()
            if has_partner:
                partner_legendre_track_cand = mc_track_matcher_legendre.getRelatedPRRecoTrack(related_mc_track_cand)
                hit_purity_of_partner = abs(mc_track_matcher_legendre.getRelatedPurity(partner_legendre_track_cand))
                hit_efficiency_of_partner = abs(mc_track_matcher_legendre.getRelatedEfficiency(related_mc_track_cand))
 
                # Count number of new hits
                legendre_hits = set(list(partner_legendre_track_cand.getHitIDs()))
                local_hits = set(list(local_track_cand.getHitIDs()))
 
                local_hits_new = local_hits - legendre_hits
                number_of_new_hits = len(local_hits_new)
 
                # Count number of hits in this superlayer
                partner_has_stereo_information = 0
                number_of_hits_in_same_superlayer = 0
                for cdc_hit_ID in legendre_hits:
                    cdc_hit = cdc_hits[cdc_hit_ID]
                    if cdc_hit.getISuperLayer() == superlayer_of_segment:
                        number_of_hits_in_same_superlayer += 1
 
                    if cdc_hit.getISuperLayer() % 2 == 1:
                        partner_has_stereo_information = 1
 
        return dict(superlayer_of_segment=superlayer_of_segment,
                    mc_track_dist=mc_track_dist,
                    is_background=is_background,
                    is_ghost=is_ghost,
                    is_matched=is_matched,
                    is_clone=is_clone,
                    is_clone_or_matched=is_clone_or_matched,
                    is_stereo=is_stereo,
                    mc_track_pt=mc_track_pt,
                    hit_purity=hit_purity,
                    has_partner=has_partner,
                    hit_purity_of_partner=hit_purity_of_partner,
                    hit_efficiency_of_partner=hit_efficiency_of_partner,
                    number_of_new_hits=number_of_new_hits,
                    number_of_hits=number_of_hits,
                    number_of_hits_in_same_superlayer=number_of_hits_in_same_superlayer,
                    partner_has_stereo_information=partner_has_stereo_information)
 

◆ prepare()

def prepare ( self )

 Prepare the CDC lookup.

Reimplemented from HarvestingModule.

Definition at line 181 of file harvester.py.

    def prepare(self):
        """ Prepare the CDC lookup. """
        
        self.cdcHits = Belle2.PyStoreArray("CDCHits")
        return HarvestingModule.prepare(self)
 

Member Data Documentation

◆ cdcHits

cdcHits

cached CDCHits StoreArray

Definition at line 184 of file harvester.py.

◆ legendre_track_cand_store_array_name

legendre_track_cand_store_array_name

cached name of the Legendre TrackCands StoreArray

Definition at line 172 of file harvester.py.

◆ mc_track_cands_store_array_name

mc_track_cands_store_array_name

cached name of the TrackCands StoreArray

Definition at line 170 of file harvester.py.

◆ mc_track_matcher_legendre

mc_track_matcher_legendre

function to match Legendre tracks to MC tracks

Definition at line 177 of file harvester.py.

◆ mc_track_matcher_local

mc_track_matcher_local

function to match local tracks to MC tracks

Definition at line 175 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 266 of file harvester.py.

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

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

Public Member Functions

Public Attributes

Static Public Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

Member Function Documentation

◆ peel()

◆ prepare()

Member Data Documentation

◆ cdcHits

◆ legendre_track_cand_store_array_name

◆ mc_track_cands_store_array_name

◆ mc_track_matcher_legendre

◆ mc_track_matcher_local

◆ save_tree

◆ init()