VxdCdcMergerHarvester Class Reference

Inheritance diagram for VxdCdcMergerHarvester:

Public Member Functions
def	__init__ (self)
def	pick (self, mc_particle)
def	peel (self, mc_particle)

Public Attributes
	mc_track_matcher
	matcher used for the final MCTrack list
	mc_track_matcher_cdc
	matcher used for the MCTracks from the CDC
	mc_track_matcher_vxd
	matcher used for the MCTracks from the VXD

Static Public Attributes
refiners	save_tree = refiners.SaveTreeRefiner()
	Refiners to be executed at the end of the harvesting / termination of the module Store as a table in a ROOT file.

Detailed Description

Harvester module to extract useful information from VXD-CDC track merging.

Definition at line 68 of file vxdCdcMergerValidationGenerateSample.py.

Constructor & Destructor Documentation

__init__()

def __init__

(

self

)

Init harvester

Reimplemented from HarvestingModule.

Definition at line 73 of file vxdCdcMergerValidationGenerateSample.py.

    def __init__(self):
        """
        Init harvester
        """
        HarvestingModule.__init__(self, foreach="MCParticles", output_file_name="../VxdCdcValidationHarvested.root")
 
        
        self.mc_track_matcher = Belle2.TrackMatchLookUp("MCRecoTracks")
        
        self.mc_track_matcher_cdc = Belle2.TrackMatchLookUp("MCRecoTracks", "CDCRecoTracks")
        
        self.mc_track_matcher_vxd = Belle2.TrackMatchLookUp("MCRecoTracks", "SVDRecoTracks")
 

Member Function Documentation

peel()

def peel	(		self,
			mc_particle
	)

Extract the information.

Reimplemented from HarvestingModule.

Definition at line 96 of file vxdCdcMergerValidationGenerateSample.py.

    def peel(self, mc_particle):
        """
        Extract the information.
        """
 
        # mc is a genfit::TrackCand
 
        mc_track = mc_particle.getRelatedFrom("MCRecoTracks")
 
        mc_nhits = mc_track.getNumberOfTotalHits()
        mc_pt = mc_particle.getMomentum().Rho()
        mc_theta = mc_particle.getMomentum().Theta()
 
        # reco_tracks = Belle2.PyStoreArray('RecoTracks')
 
        this_best_track_cdc = self.mc_track_matcher_cdc.getRelatedPRRecoTrack(mc_track)
        this_best_track_vxd = self.mc_track_matcher_vxd.getRelatedPRRecoTrack(mc_track)
 
        # here we know, the above tracks should have been merged because they stem
        # from the same MC Track. Have they?
 
        this_best_track_merged = self.mc_track_matcher.getRelatedPRRecoTrack(mc_track)
 
        good_merge = False
        vxd_hits = 0
        cdc_hits = 0
 
        merged_hits = 0
 
        if this_best_track_merged:
            if this_best_track_cdc:
                cdc_hits = this_best_track_cdc.getNumberOfTotalHits()
            if this_best_track_vxd:
                vxd_hits = this_best_track_vxd.getNumberOfTotalHits()
 
            merged_hits = vxd_hits + cdc_hits
            good_merge = merged_hits == this_best_track_merged.getNumberOfTotalHits()
 
        return {
            "MC_PT": mc_pt,
            "MC_THETA": mc_theta,
            "MC_NHITS": mc_nhits,
            "PR_NHITS": merged_hits,
            "PR_CDC_NHITS": cdc_hits,
            "PR_VXD_NHITS": vxd_hits,
            "GOOD_MERGE": good_merge
        }
 

pick()

def pick	(		self,
			mc_particle
	)

Select the MCParticle if it is related to an MCRecoTrack

Reimplemented from HarvestingModule.

Definition at line 86 of file vxdCdcMergerValidationGenerateSample.py.

    def pick(self, mc_particle):
        """Select the MCParticle if it is related to an MCRecoTrack"""
        # mc_track = mc_particle.getRelated("MCRecoTracks")
        # return mc_track is not None
        mc_track = mc_particle.getRelatedFrom("MCRecoTracks")
        if mc_track:
            return True
        else:
            return False
 

Member Data Documentation

mc_track_matcher

mc_track_matcher

matcher used for the final MCTrack list

Definition at line 80 of file vxdCdcMergerValidationGenerateSample.py.

mc_track_matcher_cdc

mc_track_matcher_cdc

matcher used for the MCTracks from the CDC

Definition at line 82 of file vxdCdcMergerValidationGenerateSample.py.

mc_track_matcher_vxd

mc_track_matcher_vxd

matcher used for the MCTracks from the VXD

Definition at line 84 of file vxdCdcMergerValidationGenerateSample.py.

save_tree

refiners save_tree = refiners.SaveTreeRefiner()

static

Refiners to be executed at the end of the harvesting / termination of the module Store as a table in a ROOT file.

Definition at line 146 of file vxdCdcMergerValidationGenerateSample.py.

---

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

• tracking/validation/vxdCdcMergerValidationGenerateSample.py