ReconstructionPositionHarvester Class Reference

Inheritance diagram for ReconstructionPositionHarvester:

Public Member Functions
def	__init__ (self, output_file_name, tracks_store_vector_name="CDCTrackVector")
def	peel (self, track_cand)

Public Attributes
	mc_hit_lookup
	function to look up CDC MC hits

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 check for the reconstructed positions 

Definition at line 17 of file harvester.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			output_file_name,
			tracks_store_vector_name = "CDCTrackVector"
	)

 Initialize with the output file name of the root file and the store obj with the
CDCTrack vector to use. MC track cands are needed. 

Reimplemented from HarvestingModule.

Definition at line 21 of file harvester.py.

    def __init__(self, output_file_name, tracks_store_vector_name="CDCTrackVector"):
        """ Initialize with the output file name of the root file and the store obj with the
        CDCTrack vector to use. MC track cands are needed. """
        super().__init__(
            foreach=tracks_store_vector_name,
            output_file_name=output_file_name)
 
        
        self.mc_hit_lookup = Belle2.TrackFindingCDC.CDCMCHitLookUp()
 

Member Function Documentation

peel()

def peel	(		self,
			track_cand
	)

 Extract the information 

Reimplemented from HarvestingModule.

Definition at line 31 of file harvester.py.

    def peel(self, track_cand):
        """ Extract the information """
        mc_hit_lookup = self.mc_hit_lookup
        mc_hit_lookup.fill()
 
        sum_of_difference_norms_stereo = 0
        num_norms_stereo = 0
        sum_of_difference_norms_axial = 0
        num_norms_axial = 0
 
        number_of_wrong_rl_infos = 0
 
        for reco_hit3D in list(track_cand.items()):
            underlaying_cdc_hit = reco_hit3D.getWireHit().getHit()
            hit_difference = mc_hit_lookup.getRecoPos3D(underlaying_cdc_hit) - reco_hit3D.getRecoPos3D()
            sum_of_difference_norms_axial += hit_difference.xy().norm()
            num_norms_axial += 1
 
            if reco_hit3D.getStereoType() != 0:  # AXIAL
                sum_of_difference_norms_stereo += abs(hit_difference.z())
                num_norms_stereo += 1
 
                correct_rl_info = mc_hit_lookup.getRLInfo(underlaying_cdc_hit)
 
                if correct_rl_info != reco_hit3D.getRLInfo():
                    number_of_wrong_rl_infos += 1
 
        return dict(sum_of_difference_norms_axial=sum_of_difference_norms_axial,
                    num_norms_axial=num_norms_axial,
                    mean_of_difference_norms_axial=np.true_divide(sum_of_difference_norms_axial, num_norms_axial),
                    sum_of_difference_norms_stereo=sum_of_difference_norms_stereo,
                    num_norms_stereo=num_norms_stereo,
                    mean_of_difference_norms_stereo=np.true_divide(sum_of_difference_norms_stereo, num_norms_stereo),
                    number_of_wrong_rl_infos=number_of_wrong_rl_infos,
                    mean_wrong_rl_infos=np.true_divide(number_of_wrong_rl_infos, num_norms_stereo))
 

Member Data Documentation

mc_hit_lookup

mc_hit_lookup

function to look up CDC MC hits

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

---

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

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