Inheritance diagram for StateRecordingTask:

Collaboration diagram for StateRecordingTask:

Public Member Functions
def	output (self)

def	requires (self)

def	create_state_recording_path (self, layer, records1_fname, records2_fname, records3_fname)

def	create_path (self)

Static Public Attributes
b2luigi	experiment_number = b2luigi.IntParameter()
	Experiment number of the conditions database, e.g.

b2luigi	random_seed = b2luigi.Parameter()
	Random basf2 seed.

b2luigi	n_events = b2luigi.IntParameter()
	Number of events to generate for training.

b2luigi	layer = b2luigi.IntParameter()
	Layer on which to toggle for recording the information for training.

Detailed Description

Record the data for the three state filters for the CDCToSVDSpacePointCKF.

This task requires that the events used for training have been simulated before, which is done using the
``SplitMergeSimTask``.

Definition at line 345 of file combined_cdc_to_svd_ckf_mva_training.py.

Member Function Documentation

◆ create_path()

def create_path ( self )

Create basf2 path to process with event generation and simulation.

Definition at line 463 of file combined_cdc_to_svd_ckf_mva_training.py.

    def create_path(self):
        """
        Create basf2 path to process with event generation and simulation.
        """
        return self.create_state_recording_path(
            layer=self.layer,
            records1_fname=self.get_output_file_name("records1.root"),
            records2_fname=self.get_output_file_name("records2.root"),
            records3_fname=self.get_output_file_name("records3.root"),
        )
 
 

◆ create_state_recording_path()

def create_state_recording_path	(	self,
		layer,
		records1_fname,
		records2_fname,
		records3_fname
	)

Create a path for the recording. To record the data for the SVD state filters, CDC tracks are required, and these must
be truth matched before. The data have to recorded for each layer of the SVD, i.e. layers 3 to 6, but also an artificial
layer 7.

:param layer: The layer for which the data are recorded.
:param records1_fname: Name of the records1 file.
:param records2_fname: Name of the records2 file.
:param records3_fname: Name of the records3 file.

Definition at line 382 of file combined_cdc_to_svd_ckf_mva_training.py.

    def create_state_recording_path(self, layer, records1_fname, records2_fname, records3_fname):
        """
        Create a path for the recording. To record the data for the SVD state filters, CDC tracks are required, and these must
        be truth matched before. The data have to recorded for each layer of the SVD, i.e. layers 3 to 6, but also an artificial
        layer 7.
 
        :param layer: The layer for which the data are recorded.
        :param records1_fname: Name of the records1 file.
        :param records2_fname: Name of the records2 file.
        :param records3_fname: Name of the records3 file.
        """
        path = basf2.create_path()
 
        # get all the file names from the list of input files that are meant for training
        file_list = [fname for sublist in self.get_input_file_names().values()
                     for fname in sublist if "generated_mc_N" in fname and "training" in fname and fname.endswith(".root")]
        path.add_module("RootInput", inputFileNames=file_list)
 
        path.add_module("Gearbox")
        path.add_module("Geometry")
        path.add_module("SetupGenfitExtrapolation")
 
        add_hit_preparation_modules(path, components=["SVD"])
 
        add_track_finding(path, reco_tracks="CDCRecoTracks", components=["CDC"], prune_temporary_tracks=False)
 
        path.add_module('TrackFinderMCTruthRecoTracks',
                        RecoTracksStoreArrayName="MCRecoTracks",
                        WhichParticles=[],
                        UsePXDHits=True,
                        UseSVDHits=True,
                        UseCDCHits=True)
 
        path.add_module("MCRecoTracksMatcher", UsePXDHits=False, UseSVDHits=False, UseCDCHits=True,
                        mcRecoTracksStoreArrayName="MCRecoTracks",
                        prRecoTracksStoreArrayName="CDCRecoTracks")
        path.add_module("DAFRecoFitter", recoTracksStoreArrayName="CDCRecoTracks")
 
        path.add_module("CDCToSVDSpacePointCKF",
                        inputRecoTrackStoreArrayName="CDCRecoTracks",
                        outputRecoTrackStoreArrayName="VXDRecoTracks",
                        outputRelationRecoTrackStoreArrayName="CDCRecoTracks",
 
                        relationCheckForDirection="backward",
                        reverseSeed=False,
                        writeOutDirection="backward",
 
                        firstHighFilter="truth",
                        firstEqualFilter="recording",
                        firstEqualFilterParameters={"treeName": "records1", "rootFileName":
                                                    records1_fname, "returnWeight": 1.0},
                        firstLowFilter="none",
                        firstHighUseNStates=0,
                        firstToggleOnLayer=layer,
 
                        advanceHighFilter="advance",
 
                        secondHighFilter="truth",
                        secondEqualFilter="recording",
                        secondEqualFilterParameters={"treeName": "records2", "rootFileName":
                                                     records2_fname, "returnWeight": 1.0},
                        secondLowFilter="none",
                        secondHighUseNStates=0,
                        secondToggleOnLayer=layer,
 
                        updateHighFilter="fit",
 
                        thirdHighFilter="truth",
                        thirdEqualFilter="recording",
                        thirdEqualFilterParameters={"treeName": "records3", "rootFileName": records3_fname},
                        thirdLowFilter="none",
                        thirdHighUseNStates=0,
                        thirdToggleOnLayer=layer,
 
                        filter="none",
                        exportTracks=False,
 
                        enableOverlapResolving=False)
 
        return path
 

◆ output()

def output ( self )

Generate list of output files that the task should produce.
The task is considered finished if and only if the outputs all exist.

Definition at line 363 of file combined_cdc_to_svd_ckf_mva_training.py.

    def output(self):
        """
        Generate list of output files that the task should produce.
        The task is considered finished if and only if the outputs all exist.
        """
        for record_fname in ["records1.root", "records2.root", "records3.root"]:
            yield self.add_to_output(record_fname)
 

◆ requires()

def requires ( self )

This task only requires that the input files have been created.

Definition at line 371 of file combined_cdc_to_svd_ckf_mva_training.py.

    def requires(self):
        """
        This task only requires that the input files have been created.
        """
        yield SplitNMergeSimTask(
            bkgfiles_dir=MainTask.bkgfiles_by_exp[self.experiment_number],
            experiment_number=self.experiment_number,
            random_seed=self.random_seed,
            n_events=self.n_events,
        )
 

Member Data Documentation

◆ experiment_number

b2luigi experiment_number = b2luigi.IntParameter()

static

Experiment number of the conditions database, e.g.

defines simulation geometry

Definition at line 353 of file combined_cdc_to_svd_ckf_mva_training.py.

◆ layer

b2luigi layer = b2luigi.IntParameter()

static

Layer on which to toggle for recording the information for training.

Definition at line 361 of file combined_cdc_to_svd_ckf_mva_training.py.

◆ n_events

b2luigi n_events = b2luigi.IntParameter()

static

Number of events to generate for training.

Definition at line 358 of file combined_cdc_to_svd_ckf_mva_training.py.

◆ random_seed

b2luigi random_seed = b2luigi.Parameter()

static

Random basf2 seed.

It is further used to read of the production process to preserve clearness in the b2luigi output.

Definition at line 356 of file combined_cdc_to_svd_ckf_mva_training.py.

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

tracking/scripts/ckf_training/cdc_to_svd_spacepoint/combined_cdc_to_svd_ckf_mva_training.py

Public Member Functions

Static Public Attributes

Detailed Description

Member Function Documentation

◆ create_path()

◆ create_state_recording_path()

◆ output()

◆ requires()

Member Data Documentation

◆ experiment_number

◆ layer

◆ n_events

◆ random_seed