SegmentPairCreationValidationModule Class Reference

Inheritance diagram for SegmentPairCreationValidationModule:

Public Member Functions
def	__init__ (self, output_file_name)
def	initialize (self)
def	prepare (self)
def	pick (self, segment_pair_relation)
def	peel (self, segment_pair_relation)
def	peel_target (self, segment_pair_relation)
def	peel_mc (self, segment_pair_relation)
def	peel_fit (self, segment_pair_relation)
def	peel_fitless (self, segment_pair_relation)
def	fit (self, segment_pair_relation)
def	select_fitless (self, fitless_crops)
def	select (self, crops)
def	print_signal_number (self, crops, tdirectory, **kwds)

Public Attributes
	mc_segment_lookup
	defer reference to CDCMCSegment2dLookUp singleton until after it is constructed
	mc_segment_pair_filter
	defer reference to MCSegmentPairFilter until after it is constructed
	segment_pair_fusion
	defer reference to CDCAxialStereoFusion until after it is constructed

Static Public Attributes
float	delta_phi_cut_value = 1.0
	default selection for the delta-phi of the segment pair
float	is_after_cut_value = 1.0
	default selection for the ordering of the segment pair
refiners	save_histograms = refiners.save_histograms(outlier_z_score=5.0, allow_discrete=True)
	Save histograms in a sub folder.
refiners	save_tree = refiners.save_tree()
	Save a tree of all collected variables in a sub folder.
refiners	save_fitless_selection_variables_histograms
	Save a tree of track-segment-without-fit variables in a sub folder.
refiners	save_view_is_after_cut_histograms
	Save a tree of mal-ordered track-segment-pair variables in a sub folder.
refiners	save_view_delta_phi_cut_histograms
	Save a tree of delta-phi-cut track-segment-pair variables in a sub folder.
refiners	save_selection_variables_after_fitless_selection_histograms
	Save a tree of track-segment-pair selection variables in a sub folder.
refiners	save_p_value_over_curvature_profile
	Save a tree of track-segment-pair p-value and curvature variables in a sub folder.

Detailed Description

Module to collect information about the generated segments and compose validation plots on terminate.

Definition at line 82 of file segmentPairCreationValidation.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			output_file_name
	)

Constructor

Definition at line 86 of file segmentPairCreationValidation.py.

    def __init__(self, output_file_name):
        """Constructor"""
        super().__init__(foreach="CDCSegmentPairVector",
                         output_file_name=output_file_name)
        
        self.mc_segment_lookup = None
        
        self.mc_segment_pair_filter = None
        
        self.segment_pair_fusion = None
 

Member Function Documentation

fit()

def fit	(		self,
			segment_pair_relation
	)

Fit the segment pair

Definition at line 249 of file segmentPairCreationValidation.py.

    def fit(self, segment_pair_relation):
        """Fit the segment pair"""
        self.segment_pair_fusion.reconstructFuseTrajectories(segment_pair_relation, True)
 

initialize()

def initialize

(

self

)

Receive signal at the start of event processing

Definition at line 97 of file segmentPairCreationValidation.py.

    def initialize(self):
        """Receive signal at the start of event processing"""
        super().initialize()
        self.mc_segment_lookup = Belle2.TrackFindingCDC.CDCMCSegment2DLookUp.getInstance()
        self.mc_segment_pair_filter = Belle2.TrackFindingCDC.MCSegmentPairFilter()
        self.segment_pair_fusion = Belle2.TrackFindingCDC.CDCAxialStereoFusion
 

peel()

def peel	(		self,
			segment_pair_relation
	)

Aggregate the track and MC information for track segment-pair analysis

Definition at line 119 of file segmentPairCreationValidation.py.

    def peel(self, segment_pair_relation):
        """Aggregate the track and MC information for track segment-pair analysis"""
        crops = self.peel_target(segment_pair_relation)
        crops.update(self.peel_mc(segment_pair_relation))
        crops.update(self.peel_fit(segment_pair_relation))
        return crops
 

peel_fit()

def peel_fit	(		self,
			segment_pair_relation
	)

Create a dictionary of track-segment-fit information

Definition at line 151 of file segmentPairCreationValidation.py.

    def peel_fit(self, segment_pair_relation):
        """Create a dictionary of track-segment-fit information"""
        fitless_crops = self.peel_fitless(segment_pair_relation)
 
        select_fitless = fitless_crops["select_fitless"]
        if select_fitless:
            # Now fit
            self.fit(segment_pair_relation)
            fit3d = segment_pair_relation.getTrajectory3D()
 
            i_curv = 0
            i_tan_lambda = 3
 
            chi2 = fit3d.getChi2()
            ndf = fit3d.getNDF()
 
            curvature_estimate = fit3d.getCurvatureXY()
            curvature_variance = fit3d.getLocalVariance(i_curv)
 
            tan_lambda_estimate = fit3d.getTanLambda()
            tan_lambda_variance = fit3d.getLocalVariance(i_tan_lambda)
 
            chi2 = chi2
            ndf = ndf
            p_value = prob(chi2, ndf)
            # select = True
 
        else:
            nan = float('nan')
            curvature_estimate = nan
            curvature_variance = nan
 
            tan_lambda_estimate = nan
            tan_lambda_variance = nan
 
            chi2 = nan
            ndf = nan
            p_value = nan
 
        crops = dict(
            curvature_estimate=curvature_estimate,
            curvature_variance=curvature_variance,
 
            tan_lambda_estimate=tan_lambda_estimate,
            tan_lambda_variance=tan_lambda_variance,
 
            chi2=chi2,
            ndf=ndf,
            p_value=p_value,
        )
 
        if select_fitless:
            crops["select"] = self.select(crops)
        else:
            crops["select"] = False
 
        crops.update(fitless_crops)
 
        return crops
 

peel_fitless()

def peel_fitless	(		self,
			segment_pair_relation
	)

Create a dictionary of track-segments-without-fit information

Definition at line 211 of file segmentPairCreationValidation.py.

    def peel_fitless(self, segment_pair_relation):
        """Create a dictionary of track-segments-without-fit information"""
        # Try to make some judgements without executing the common fit.
 
        start_segment = segment_pair_relation.getStartSegment()
        end_segment = segment_pair_relation.getEndSegment()
 
        start_fit2d = start_segment.getTrajectory2D()
        end_fit2d = end_segment.getTrajectory2D()
 
        start_superlayer_id = start_segment.getISuperLayer()
        end_superlayer_id = end_segment.getISuperLayer()
 
        sorted_superlayer_ids = sorted([start_superlayer_id, end_superlayer_id])
 
        superlayer_id_pair = 10.0 * sorted_superlayer_ids[1] + sorted_superlayer_ids[0]
 
        fitless_crops = dict(
            start_superlayer_id=start_superlayer_id,
            end_superlayer_id=end_superlayer_id,
            superlayer_id_pair=superlayer_id_pair,
 
            start_size=start_segment.size(),
            end_size=end_segment.size(),
 
            start_curvature_estimate=start_fit2d.getCurvature(),
            end_curvature_estimate=end_fit2d.getCurvature(),
 
            delta_phi=segment_pair_relation.computeDeltaPhiAtSuperLayerBound(),
            is_coaligned=segment_pair_relation.computeIsCoaligned(),
 
            start_is_before_end=segment_pair_relation.computeStartIsBeforeEnd(),
            end_is_after_start=segment_pair_relation.computeEndIsAfterStart(),
        )
 
        fitless_crops["select_fitless"] = self.select_fitless(fitless_crops)
        return fitless_crops
 

peel_mc()

def peel_mc	(		self,
			segment_pair_relation
	)

Create a dictionary of MC-truth (curvature,tanlambda) pairs

Definition at line 136 of file segmentPairCreationValidation.py.

    def peel_mc(self, segment_pair_relation):
        """Create a dictionary of MC-truth (curvature,tanlambda) pairs"""
        mc_segment_lookup = self.mc_segment_lookup
 
        end_segment = segment_pair_relation.getEndSegment()
 
        # Take the fit best at the middle of the segment pair
        # mc_particle = mc_segment_lookup.getMCParticle(end_segment)
        fit3d_truth = mc_segment_lookup.getTrajectory3D(end_segment)
 
        return dict(
            curvature_truth=fit3d_truth.getCurvatureXY(),
            tan_lambda_truth=fit3d_truth.getTanLambda(),
        )
 

peel_target()

def peel_target	(		self,
			segment_pair_relation
	)

Create a dictionary of MC-truth (weight,decision) pairs

Definition at line 126 of file segmentPairCreationValidation.py.

    def peel_target(self, segment_pair_relation):
        """Create a dictionary of MC-truth (weight,decision) pairs"""
        mc_weight = self.mc_segment_pair_filter(segment_pair_relation)
        mc_decision = np.isfinite(mc_weight)  # Filters for nan
 
        return dict(
            mc_weight=mc_weight,
            mc_decision=mc_decision,
        )
 

pick()

def pick	(		self,
			segment_pair_relation
	)

Select segment pairs with 4 or more hit in each segments and a matching primary MC particle

Definition at line 108 of file segmentPairCreationValidation.py.

    def pick(self, segment_pair_relation):
        """Select segment pairs with 4 or more hit in each segments and a matching primary MC particle"""
        mc_segment_lookup = self.mc_segment_lookup
        start_segment = segment_pair_relation.getStartSegment()
        end_segment = segment_pair_relation.getEndSegment()
        mc_particle = mc_segment_lookup.getMCParticle(start_segment)
        return (mc_particle and
                is_primary(mc_particle) and
                start_segment.size() > 3 and
                end_segment.size() > 3)
 

prepare()

def prepare

(

self

)

Initialize the MC-hit lookup method

Definition at line 104 of file segmentPairCreationValidation.py.

    def prepare(self):
        """Initialize the MC-hit lookup method"""
        Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
 

print_signal_number()

def print_signal_number	(		self,
			crops,
			tdirectory,
		**	kwds
	)

Print diagnostic information about the track-segment-pair selection

Definition at line 328 of file segmentPairCreationValidation.py.

    def print_signal_number(self, crops, tdirectory, **kwds):
        """Print diagnostic information about the track-segment-pair selection"""
        info = get_logger().info
 
        # start_superlayer_ids = crops["start_superlayer_id"]
        # end_superlayer_ids = crops["end_superlayer_id"]
 
        superlayer_id_pair = crops["superlayer_id_pair"]
        info("Number of pairs in superlayers %s : %s", np.unique(superlayer_id_pair), len(superlayer_id_pair))
 
        mc_decisions = crops["mc_decision"]
        n = len(mc_decisions)
        n_signal = np.sum(mc_decisions)
        n_background = n - n_signal
        info("#Signal : %s", n_signal)
        info("#Background : %s", n_background)
 
        fitless_selections = np.nonzero(crops["select_fitless"])
        info("#Signal after precut : %s", np.sum(mc_decisions[fitless_selections]))
        info("#Background after precut : %s", np.sum(1 - mc_decisions[fitless_selections]))
 
 

select()

def select	(		self,
			crops
	)

Select every track-segment-pair

Definition at line 266 of file segmentPairCreationValidation.py.

    def select(self, crops):
        """Select every track-segment-pair"""
        return True
 

select_fitless()

def select_fitless	(		self,
			fitless_crops
	)

Selection of track-segments-without-fit

Definition at line 258 of file segmentPairCreationValidation.py.

    def select_fitless(self, fitless_crops):
        """Selection of track-segments-without-fit"""
        delta_phi = fitless_crops["delta_phi"]
        start_is_before_end = fitless_crops["start_is_before_end"]
        end_is_after_start = fitless_crops["end_is_after_start"]
        is_after_select = (abs(start_is_before_end) < self.is_after_cut_value) & (abs(end_is_after_start) < self.is_after_cut_value)
        return (abs(delta_phi) < self.delta_phi_cut_value) & is_after_select
 

Member Data Documentation

delta_phi_cut_value

float delta_phi_cut_value = 1.0

static

default selection for the delta-phi of the segment pair

Definition at line 254 of file segmentPairCreationValidation.py.

is_after_cut_value

float is_after_cut_value = 1.0

static

default selection for the ordering of the segment pair

Definition at line 256 of file segmentPairCreationValidation.py.

mc_segment_lookup

mc_segment_lookup

defer reference to CDCMCSegment2dLookUp singleton until after it is constructed

Definition at line 91 of file segmentPairCreationValidation.py.

mc_segment_pair_filter

mc_segment_pair_filter

defer reference to MCSegmentPairFilter until after it is constructed

Definition at line 93 of file segmentPairCreationValidation.py.

save_fitless_selection_variables_histograms

refiners save_fitless_selection_variables_histograms

static

Initial value:

=  refiners.save_histograms(
        select=["mc_decision", "delta_phi", "start_is_before_end", "end_is_after_start", "is_coaligned"],
        outlier_z_score=5.0,
        allow_discrete=True,
        stackby="mc_decision",
        folder_name="fitless_selection_variables",
    )

Save a tree of track-segment-without-fit variables in a sub folder.

Definition at line 278 of file segmentPairCreationValidation.py.

save_histograms

refiners save_histograms = refiners.save_histograms(outlier_z_score=5.0, allow_discrete=True)

static

Save histograms in a sub folder.

Definition at line 272 of file segmentPairCreationValidation.py.

save_p_value_over_curvature_profile

refiners save_p_value_over_curvature_profile

static

Initial value:

=  refiners.save_profiles(
        select={"p_value": "p-value", "curvature_truth": "true curvature"},
        y="p-value",
        folder_name="selection_variables_after_fitless_selection",
        title=r"$p$-value versus true curvature after fitless selection",
        filter_on="select_fitless",
    )

Save a tree of track-segment-pair p-value and curvature variables in a sub folder.

Definition at line 317 of file segmentPairCreationValidation.py.

save_selection_variables_after_fitless_selection_histograms

refiners save_selection_variables_after_fitless_selection_histograms

static

Initial value:

=  refiners.save_histograms(
        select=["mc_decision", "chi2", "ndf", "p_value"],
        outlier_z_score=5.0,
        allow_discrete=True,
        stackby="mc_decision",
        folder_name="selection_variables_after_fitless_selection",
        filter_on="select_fitless",
    )

Save a tree of track-segment-pair selection variables in a sub folder.

Definition at line 306 of file segmentPairCreationValidation.py.

save_tree

refiners save_tree = refiners.save_tree()

static

Save a tree of all collected variables in a sub folder.

Definition at line 274 of file segmentPairCreationValidation.py.

save_view_delta_phi_cut_histograms

refiners save_view_delta_phi_cut_histograms

static

Initial value:

=  refiners.save_histograms(
        select=["mc_decision", "delta_phi"],
        lower_bound=-delta_phi_cut_value,
        upper_bound=delta_phi_cut_value,
        stackby="mc_decision",
        folder_name="view_fitless_cuts",
    )

Save a tree of delta-phi-cut track-segment-pair variables in a sub folder.

Definition at line 296 of file segmentPairCreationValidation.py.

save_view_is_after_cut_histograms

refiners save_view_is_after_cut_histograms

static

Initial value:

=  refiners.save_histograms(
        select=["mc_decision", "start_is_before_end", "end_is_after_start"],
        lower_bound=-is_after_cut_value,
        upper_bound=is_after_cut_value,
        stackby="mc_decision",
        folder_name="view_fitless_cuts",
    )

Save a tree of mal-ordered track-segment-pair variables in a sub folder.

Definition at line 287 of file segmentPairCreationValidation.py.

segment_pair_fusion

segment_pair_fusion

defer reference to CDCAxialStereoFusion until after it is constructed

Definition at line 95 of file segmentPairCreationValidation.py.

---

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

• tracking/trackFindingCDC/studies/segmentPair/segmentPairCreationValidation.py