SegmentPairFitValidationRun Class Reference

Inheritance diagram for SegmentPairFitValidationRun:

Public Member Functions
def	output_file_name (self)
def	harvesting_module (self, path=None)
def	create_argument_parser (self, **kwds)
def	get_fit_method (self)
def	create_path (self)

Public Attributes
	monte_carlo
	Degree of refinement of the segment generation.

Static Public Attributes
int	n_events = 10000
	number of events to generate
str	generator_module = "simple_gun"
	use the low-momentum particle gun event generator
str	monte_carlo = "no"
	do not generate new events
str	segment_orientation = "outwards"
	ordering of the segments in the pair
str	fit_method_name = "fuse-sz"
	use the Kalmanesk fuse of the two trajectory fits

Detailed Description

Harvester to generate, postprocess and inspect MC events for track-segment-pair fit validation

Definition at line 33 of file record.py.

Member Function Documentation

create_argument_parser()

def create_argument_parser	(		self,
		**	kwds
	)

Convert command-line arguments to basf2 argument list

Reimplemented from HarvestingRunMixin.

Definition at line 68 of file record.py.

    def create_argument_parser(self, **kwds):
        """Convert command-line arguments to basf2 argument list"""
        argument_parser = super().create_argument_parser(**kwds)
 
        argument_parser.add_argument(
            '-m',
            '--monte-carlo',
            choices=["no", "medium", "full"],
            default=self.monte_carlo,
            dest='monte_carlo',
            help='Amount of monte carlo information to be used in the segment generation.',
        )
 
        argument_parser.add_argument(
            "--fit",
            choices=["zreco", "fuse-pre", "fuse-sz", "fuse-sz-re"],
            default=self.fit_method_name,
            dest="fit_method_name",
            help=("Choose which fit positional information of the segment should be used. \n"
                  "* 'zreco' means the z coordinate is reconstructed and a linear sz fit is made. "
                  "No covariance between the circle and the linear sz part can be made.\n"
                  "* 'fuse-sz' means the Kalmanesk fuse of the two trajectory fits.\n"
                  "* 'fuse-sz-re' means the Kalmanesk fuse of the two trajectory fits but reestimate the drift length."
                  )
        )
 
        return argument_parser
 

create_path()

def create_path

(

self

)

Sets up a path that plays back pregenerated events or generates events
based on the properties in the base class.

Reimplemented from ReadOrGenerateEventsRun.

Definition at line 161 of file record.py.

    def create_path(self):
        """
        Sets up a path that plays back pregenerated events or generates events
        based on the properties in the base class.
        """
        path = super().create_path()
 
        path.add_module("TFCDC_WireHitPreparer",
                        flightTimeEstimation="outwards",
                        UseNLoops=0.5
                        )
 
        path.add_module("TFCDC_ClusterPreparer")
 
        
        if self.monte_carlo == "no":
            # MC free - default
            path.add_module("TFCDC_SegmentFinderFacetAutomaton",
                            SegmentOrientation="outwards"
                            )
 
            path.add_module("TFCDC_SegmentFitter",
                            inputSegments="CDCSegment2DVector",
                            updateDriftLength=True,
                            useAlphaInDriftLength=True,
                            )
 
        elif self.monte_carlo == "medium":
            # Medium MC - proper generation logic, but true facets and facet relations
            path.add_module("TFCDC_SegmentFinderFacetAutomaton",
                            FacetFilter="truth",
                            FacetRelationFilter="truth",
                            SegmentOrientation="outwards"
                            )
 
            path.add_module("TFCDC_SegmentFitter",
                            inputSegments="CDCSegment2DVector",
                            updateDriftLength=True,
                            useAlphaInDriftLength=True,
                            )
 
        elif self.monte_carlo == "full":
            # Only true monte carlo segments
            # make the positions realistic but keep the true drift length
            path.add_module("TFCDC_SegmentCreatorMCTruth",
                            reconstructedDriftLength=False,
                            reconstructedPositions=True,
                            # segments="MCSegments"
                            )
 
            path.add_module("TFCDC_SegmentFitter",
                            inputSegments="CDCSegment2DVector",
                            updateDriftLength=False,
                            # useAlphaInDriftLength=True,
                            )
 
        else:
            raise ValueError("Invalid degree of Monte Carlo information")
 
        path.add_module("TFCDC_SegmentOrienter",
                        SegmentOrientation="outwards",
                        # SegmentOrientation="none",
                        inputSegments="CDCSegment2DVector",
                        segments="CDCSegment2DVectorOriented"
                        )
 
        path.add_module("TFCDC_TrackFinderSegmentPairAutomaton",
                        inputSegments="CDCSegment2DVectorOriented",
                        WriteSegmentPairs=True,
                        SegmentPairFilter="truth",
                        SegmentPairFilterParameters={"allowReverse": True},
                        SegmentPairRelationFilter="none"
                        )
 
        path.add_module(AxialStereoPairFitterModule(fit_method=self.get_fit_method()))
        return path
 
 

get_fit_method()

def get_fit_method

(

self

)

Determine which track-segment-pair fitter to use

Definition at line 96 of file record.py.

    def get_fit_method(self):
        """Determine which track-segment-pair fitter to use"""
        fit_method_name = self.fit_method_name
 
        if fit_method_name == 'zreco':
            sz_fitter = Belle2.TrackFindingCDC.CDCSZFitter.getFitter()
 
            def z_reconstruction_fit(pair):
                return sz_fitter.update(pair)
            return z_reconstruction_fit
 
        elif fit_method_name.startswith('fuse-pre'):
            CDCAxialStereoFusion = Belle2.TrackFindingCDC.CDCAxialStereoFusion
            CDCSegmentPair = Belle2.TrackFindingCDC.CDCSegmentPair
            fusionFit = CDCAxialStereoFusion()
 
            def sz_segment_pair_preliminary_fit(pair):
                fusionFit.fusePreliminary(pair)
            return sz_segment_pair_preliminary_fit
 
        elif fit_method_name.startswith('fuse-sz'):
            CDCAxialStereoFusion = Belle2.TrackFindingCDC.CDCAxialStereoFusion
            CDCSegmentPair = Belle2.TrackFindingCDC.CDCSegmentPair
            reestimateDriftLength = fit_method_name.endswith("re")
            fusionFit = CDCAxialStereoFusion(reestimateDriftLength)
 
            def sz_segment_pair_fusion_fit(pair):
                fusionFit.reconstructFuseTrajectories(pair)
                return
 
                trajectory3D = pair.getTrajectory3D()
                revFromSegment = pair.getFromSegment().reversed()
                revToSegment = pair.getToSegment().reversed()
                revPair = CDCSegmentPair(revToSegment, revFromSegment)
 
                CDCAxialStereoFusion.reconstructFuseTrajectories(revPair)
                revTrajectory3D = revPair.getTrajectory3D().reversed()
 
                # print("One origin x", trajectory3D.getLocalOrigin().x())
                # print("One origin y", trajectory3D.getLocalOrigin().y())
                # print("One origin z", trajectory3D.getLocalOrigin().z())
 
                # print("Rev origin x", revTrajectory3D.getLocalOrigin().x())
                # print("Rev origin y", revTrajectory3D.getLocalOrigin().y())
                # print("Rev origin z", revTrajectory3D.getLocalOrigin().z())
 
                print("One parameters", [trajectory3D.getLocalHelix().parameters()[i] for i in range(5)])
                print("Rev parameters", [revTrajectory3D.getLocalHelix().parameters()[i] for i in range(5)])
 
                print("One covariance")
                for j in range(5):
                    print([trajectory3D.getLocalHelix().helixCovariance()(i, j) for i in range(5)])
 
                print("Rev covariance")
                for j in range(5):
                    print([revTrajectory3D.getLocalHelix().helixCovariance()(i, j) for i in range(5)])
 
                # return revTrajectory3D
                # return trajectory3D
 
            return sz_segment_pair_fusion_fit
 
        else:
            raise ValueError(f"Unexpected fit_positions {fit_method_name}")
 

harvesting_module()

def harvesting_module	(		self,
			path = None
	)

Harvest and post-process the MC events

Reimplemented from HarvestingRunMixin.

Definition at line 61 of file record.py.

    def harvesting_module(self, path=None):
        """Harvest and post-process the MC events"""
        harvesting_module = SegmentPairFitValidationModule(self.output_file_name)
        if path:
            path.add_module(harvesting_module)
        return harvesting_module
 

output_file_name()

def output_file_name

(

self

)

Get the output ROOT filename

Reimplemented from HarvestingRunMixin.

Definition at line 49 of file record.py.

    def output_file_name(self):
        """Get the output ROOT filename"""
        file_name = self.fit_method_name
        file_name += "-mc-" + self.monte_carlo
 
        if self.root_input_file:
            file_name += "-" + os.path.split(self.root_input_file)[1]
        else:
            file_name += ".root"
 
        return file_name
 

Member Data Documentation

fit_method_name

str fit_method_name = "fuse-sz"

static

use the Kalmanesk fuse of the two trajectory fits

Definition at line 46 of file record.py.

generator_module

str generator_module = "simple_gun"

static

use the low-momentum particle gun event generator

Definition at line 38 of file record.py.

monte_carlo [1/2]

str monte_carlo = "no"

static

do not generate new events

Definition at line 41 of file record.py.

monte_carlo [2/2]

monte_carlo

Degree of refinement of the segment generation.

Definition at line 176 of file record.py.

n_events

int n_events = 10000

static

number of events to generate

Definition at line 36 of file record.py.

segment_orientation

str segment_orientation = "outwards"

static

ordering of the segments in the pair

Definition at line 43 of file record.py.

---

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

• tracking/trackFindingCDC/studies/segmentPairFit/record.py