FitValidationModule Class Reference

Inheritance diagram for FitValidationModule:

Collaboration diagram for FitValidationModule:

Public Member Functions
def	__init__ (self, mc_track_cands_store_array_name, legendre_track_cand_store_array_name, output_file_name)
def	prepare (self)
def	peel (self, legendre_track_cand)

Public Attributes
	mc_track_cands_store_array_name
	cached name of the MCTrackCands StoreArray
	legendre_track_cand_store_array_name
	cached name of the LegendreTrackCands StoreArray
	circle_fitter
	Use the Riemann fitter for circles.
	fast_fitter
	Use the standard track fitter for speed.
	cdcHits
	cached name of the CDCHits StoreArray

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

Validate the track fit

Definition at line 23 of file fitValidation.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			mc_track_cands_store_array_name,
			legendre_track_cand_store_array_name,
			output_file_name
	)

Constructor

Reimplemented from HarvestingModule.

Definition at line 26 of file fitValidation.py.

            output_file_name):
        """Constructor"""
        super().__init__(
            foreach=legendre_track_cand_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.circle_fitter = Belle2.TrackFindingCDC.CDCRiemannFitter.getFitter()
        
        self.fast_fitter = Belle2.TrackFindingCDC.TrackFitter()
 

Member Function Documentation

peel()

def peel	(		self,
			legendre_track_cand
	)

Aggregate the track and MC information for track-fit validation

Reimplemented from HarvestingModule.

Definition at line 52 of file fitValidation.py.

    def peel(self, legendre_track_cand):
        """Aggregate the track and MC information for track-fit validation"""
 
        cdc_hit_store_array = self.cdcHits
 
        # observations_variance = Belle2.TrackFindingCDC.CDCObservations2D()
        observations = Belle2.TrackFindingCDC.CDCObservations2D()
        hits = ROOT.std.vector("Belle2::TrackFindingCDC::TrackHit*")()
 
        cdc_hit_ids = legendre_track_cand.getHitIDs(Belle2.Const.CDC)
        for cdc_hit_id in cdc_hit_ids:
            cdc_hit = cdc_hit_store_array[cdc_hit_id]
 
            # We will only use the axial hits here as the FastFit can only process axial hits too
            if cdc_hit.getISuperLayer() % 2 == 0:
                cdc_wire_hit = Belle2.TrackFindingCDC.CDCWireHit(cdc_hit)
 
                wireRefPos2D = cdc_wire_hit.getRefPos2D()
                drift_length = cdc_wire_hit.getRefDriftLength()
                observations.append(wireRefPos2D.x(), wireRefPos2D.y(), 0, 1 / abs(drift_length))
                hits.push_back(Belle2.TrackFindingCDC.TrackHit(cdc_wire_hit))
 
        # Viktors method
        track_par = ROOT.std.pair("double, double")()
        ref_point = ROOT.std.pair("double, double")()
        # Careful: viktors result is a reduced chi2
        viktor_chi2 = self.fast_fitter.fitTrackCandidateFast(hits, track_par, ref_point, False) * (hits.size() - 4)
 
        # Riemann without drift variances
        trajectory2D = Belle2.TrackFindingCDC.CDCTrajectory2D()
        self.circle_fitter.update(trajectory2D, observations)
        riemann_chi2 = trajectory2D.getChi2()
 
        return_dict = dict(
            riemann_chi2=riemann_chi2,
            viktor_chi2=viktor_chi2,
        )
 
        return return_dict
 

prepare()

def prepare

(

self

)

 Initialize the harvester

Reimplemented from HarvestingModule.

Definition at line 46 of file fitValidation.py.

    def prepare(self):
        """ Initialize the harvester"""
        
        self.cdcHits = Belle2.PyStoreArray("CDCHits")
        return HarvestingModule.prepare(self)
 

Member Data Documentation

cdcHits

cdcHits

cached name of the CDCHits StoreArray

Definition at line 49 of file fitValidation.py.

circle_fitter

circle_fitter

Use the Riemann fitter for circles.

Definition at line 42 of file fitValidation.py.

fast_fitter

fast_fitter

Use the standard track fitter for speed.

Definition at line 44 of file fitValidation.py.

legendre_track_cand_store_array_name

legendre_track_cand_store_array_name

cached name of the LegendreTrackCands StoreArray

Definition at line 39 of file fitValidation.py.

mc_track_cands_store_array_name

mc_track_cands_store_array_name

cached name of the MCTrackCands StoreArray

Definition at line 37 of file fitValidation.py.

save_tree

refiners save_tree = refiners.save_tree(folder_name="tree")

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 94 of file fitValidation.py.

---

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

• tracking/trackFindingCDC/studies/legendreTrackFit/fitValidation.py