SeedsAnalyser Class Reference

Inheritance diagram for SeedsAnalyser:

Public Member Functions
def	__init__ (self, output_file_name, track_cands, use_vxd_hits=True)
def	peel (self, legendre_track_cand)

Public Attributes
	use_vxd_hits
	cached flag to use VXD hits

Static Public Attributes
refiners	save_tree = refiners.SaveTreeRefiner()
	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

Extract the momentum seeds of the first and the last hit of the track candidates (when use_vxd_hits is true),
the helx and the mc momentum.

Definition at line 353 of file harvester.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			output_file_name,
			track_cands,
			use_vxd_hits = True
	)

Init with the root output file name and the name of the store array of the track cands to use.

Reimplemented from HarvestingModule.

Definition at line 360 of file harvester.py.

    def __init__(self, output_file_name, track_cands, use_vxd_hits=True):
        """
        Init with the root output file name and the name of the store array of the track cands to use.
        """
        HarvestingModule.__init__(self, foreach=track_cands, output_file_name=output_file_name)
 
        
        self.use_vxd_hits = use_vxd_hits
 

Member Function Documentation

peel()

def peel	(		self,
			legendre_track_cand
	)

Extract the information.

Reimplemented from HarvestingModule.

Definition at line 369 of file harvester.py.

    def peel(self, legendre_track_cand):
        """
        Extract the information.
        """
        if legendre_track_cand.getChargeSeed() > 0:
            helix = Belle2.Helix(legendre_track_cand.getPosSeed(), legendre_track_cand.getMomSeed(), 1, 1.5)
        if legendre_track_cand.getChargeSeed() < 0:
            helix = Belle2.Helix(legendre_track_cand.getPosSeed(), legendre_track_cand.getMomSeed(), -1, 1.5)
 
        matcher = Belle2.TrackMatchLookUp("MCTrackCands", self.foreach)
        mc_track_cand = matcher.getAnyChargeMatchedMCRecoTrack(legendre_track_cand)
 
        pxd_clusters = Belle2.PyStoreArray("PXDClusters")
        svd_clusters = Belle2.PyStoreArray("SVDClusters")
 
        if mc_track_cand:
            mc_x = mc_track_cand.getPosSeed().X()
            mc_y = mc_track_cand.getPosSeed().Y()
            mc_z = mc_track_cand.getPosSeed().Z()
            mc_mom_x = mc_track_cand.getMomSeed().X()
            mc_mom_y = mc_track_cand.getMomSeed().Y()
            mc_mom_z = mc_track_cand.getMomSeed().Z()
        else:
            mc_x = np.NaN
            mc_y = np.NaN
            mc_z = np.NaN
            mc_mom_x = np.NaN
            mc_mom_y = np.NaN
            mc_mom_z = np.NaN
 
        return_dict = dict(
            helix_z=helix.getZ0(),
            helix_x=helix.getPerigeeX(),
            helix_y=helix.getPerigeeY(),
            helix_mom_z=helix.getMomentumZ(1.5),
            helix_mom_x=helix.getMomentumX(1.5),
            helix_mom_y=helix.getMomentumY(1.5),
            mc_x=mc_x,
            mc_y=mc_y,
            mc_z=mc_z,
            mc_mom_x=mc_mom_x,
            mc_mom_y=mc_mom_y,
            mc_mom_z=mc_mom_z)
 
        if self.use_vxd_hits:
            first_hit = legendre_track_cand.getHit(0)
            if first_hit.getDetId() == Belle2.Const.PXD:
                first_cluster = pxd_clusters[first_hit.getHitId()]
                first_true_hit = first_cluster.getRelated("PXDTrueHits")
            elif first_hit.getDetId() == Belle2.Const.SVD:
                first_cluster = svd_clusters[first_hit.getHitId()]
                first_true_hit = first_cluster.getRelated("SVDTrueHits")
 
            vxdID = first_cluster.getSensorID()
            sensorInfoBase = Belle2.VXD.GeoCache.getInstance().getSensorInfo(vxdID)
            first_momentum = sensorInfoBase.vectorToGlobal(first_true_hit.getMomentum(), True)
 
            last_hit = legendre_track_cand.getHit(-1)
            if last_hit.getDetId() == Belle2.Const.PXD:
                last_cluster = pxd_clusters[last_hit.getHitId()]
                last_true_hit = last_cluster.getRelated("PXDTrueHits")
            elif last_hit.getDetId() == Belle2.Const.SVD:
                last_cluster = svd_clusters[last_hit.getHitId()]
                last_true_hit = last_cluster.getRelated("SVDTrueHits")
 
            vxdID = first_cluster.getSensorID()
            sensorInfoBase = Belle2.VXD.GeoCache.getInstance().getSensorInfo(vxdID)
            last_momentum = sensorInfoBase.vectorToGlobal(last_true_hit.getMomentum(), True)
 
            return_dict.update(dict(first_hit_mom_x=first_momentum.X(),
                                    first_hit_mom_y=first_momentum.Y(),
                                    first_hit_mom_z=first_momentum.Z(),
                                    last_hit_mom_x=last_momentum.X(),
                                    last_hit_mom_y=last_momentum.Y(),
                                    last_hit_mom_z=last_momentum.Z()))
 
        return return_dict
 

Member Data Documentation

save_tree

refiners save_tree = refiners.SaveTreeRefiner()

static

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 449 of file harvester.py.

use_vxd_hits

use_vxd_hits

cached flag to use VXD hits

Definition at line 367 of file harvester.py.

---

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

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