FitHarvester Class Reference

Inheritance diagram for FitHarvester:

Public Member Functions
def	__init__ (self, output_file_name, queue)
def	pick (self, track_fit_result)
def	peel (self, track_fit_result)

Public Attributes
	data_store
	access the DataStore singletion

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

Collect dE/dX and the correct p of each VXD hit associated with the fitted tracks. 

Definition at line 338 of file harvester.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			output_file_name,
			queue
	)

Constructor

Definition at line 341 of file harvester.py.

    def __init__(self, output_file_name, queue):
        """Constructor"""
        QueueHarvester.__init__(self, queue, foreach="TrackFitResults", output_file_name=output_file_name)
        
        self.data_store = Belle2.DataStore.Instance()
 

Member Function Documentation

peel()

def peel	(		self,
			track_fit_result
	)

Aggregate the track-fit information associated with a TrackFitResult

Definition at line 358 of file harvester.py.

    def peel(self, track_fit_result):
        """Aggregate the track-fit information associated with a TrackFitResult"""
        mc_track_cand = track_fit_result.getRelationsFrom("TrackCands")[0]
        mc_particle = self.data_store.getRelated(mc_track_cand, "MCParticles")
 
        fit_momentum = track_fit_result.getMomentum()
        true_momentum = mc_particle.getMomentum()
 
        related_reco_track = track_fit_result.getRelated("GF2Tracks")
        cardinal_rep = related_reco_track.getCardinalRep()
        kalman_fit_state = related_reco_track.getKalmanFitStatus()
 
        number_of_measurements_in_total = 0
        number_of_measurements_with_smaller_weight = 0
 
        number_of_momentum_measurements_in_total = 0
        number_of_momentum_measurements_with_smaller_weight = 0
 
        for track_point_ID in range(related_reco_track.getNumPointsWithMeasurement()):
            track_point = related_reco_track.getPointWithMeasurement(track_point_ID)
 
            is_momentum_measurement = track_point.getRawMeasurement().__class__.__name__ == "genfit::PlanarMomentumMeasurement"
 
            if is_momentum_measurement:
                number_of_momentum_measurements_in_total += 1
 
            if track_point.hasFitterInfo(cardinal_rep):
                fitter_info = track_point.getFitterInfo(cardinal_rep)
                num_measurements = fitter_info.getNumMeasurements()
 
                for measurement_id in range(num_measurements):
                    number_of_measurements_in_total += 1
                    weight = fitter_info.getMeasurementOnPlane(measurement_id).getWeight()
                    if weight != 1:
                        number_of_measurements_with_smaller_weight += 1
 
                        if is_momentum_measurement:
                            number_of_momentum_measurements_with_smaller_weight += 1
 
        return dict(fit_momentum_x=fit_momentum.X(),
                    fit_momentum_y=fit_momentum.Y(),
                    fit_momentum_z=fit_momentum.Z(),
                    p_value=kalman_fit_state.getForwardPVal(),
                    backward_p_value=kalman_fit_state.getBackwardPVal(),
                    true_momentum_x=true_momentum.X(),
                    true_momentum_y=true_momentum.Y(),
                    true_momentum_z=true_momentum.Z(),
                    number_of_measurements_in_total=number_of_measurements_in_total,
                    number_of_measurements_with_smaller_weight=number_of_measurements_with_smaller_weight,
                    number_of_momentum_measurements_in_total=number_of_momentum_measurements_in_total,
                    number_of_momentum_measurements_with_smaller_weight=number_of_momentum_measurements_with_smaller_weight)
 

pick()

def pick	(		self,
			track_fit_result
	)

Select a TrackFitResult if it is associated with exactly one MCParticle

Definition at line 347 of file harvester.py.

    def pick(self, track_fit_result):
        """Select a TrackFitResult if it is associated with exactly one MCParticle"""
        mc_track_cands = track_fit_result.getRelationsFrom("TrackCands")
        if len(mc_track_cands) != 1:
            return False
 
        mc_track_cand = mc_track_cands[0]
        mc_particles = self.data_store.getRelationsFromObj(mc_track_cand, "MCParticles")
 
        return len(mc_particles) == 1
 

Member Data Documentation

data_store

data_store

access the DataStore singletion

Definition at line 345 of file harvester.py.

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

---

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

• tracking/studies/low_momentum_vxd_estimation/harvester.py