ResultRecordingTask Class Reference

Inheritance diagram for ResultRecordingTask:

Collaboration diagram for ResultRecordingTask:

Public Member Functions
def	output (self)
def	requires (self)
def	create_result_recording_path (self, result_filter_records_name)
def	create_path (self)

Static Public Attributes
b2luigi	experiment_number = b2luigi.IntParameter()
	Experiment number of the conditions database, e.g.
b2luigi	n_events_training = b2luigi.IntParameter()
	Number of events to generate.
b2luigi	random_seed = b2luigi.Parameter()
	Random basf2 seed.
b2luigi	result_filter_records_name = b2luigi.Parameter()
	Name of the records file for training the final result filter.

Detailed Description

Task to record data for the final result filter. This only requires found and MC-matched SVD and CDC tracks that need to be
merged, all state filters are set to "all"

Definition at line 332 of file cdc_and_svd_ckf_merger_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 432 of file cdc_and_svd_ckf_merger_mva_training.py.

    def create_path(self):
        """
        Create basf2 path to process with event generation and simulation.
        """
        return self.create_result_recording_path(
            result_filter_records_name=self.get_output_file_name(self.result_filter_records_name),
        )
 
 

create_result_recording_path()

def create_result_recording_path	(		self,
			result_filter_records_name
	)

Create a path for the recording of the result filter. This file is then used to train the result filter.

:param result_filter_records_name: Name of the recording file.

Definition at line 367 of file cdc_and_svd_ckf_merger_mva_training.py.

    def create_result_recording_path(self, result_filter_records_name):
        """
        Create a path for the recording of the result filter. This file is then used to train the result filter.
 
        :param result_filter_records_name: Name of the recording 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"])
 
        # MCTrackFinding
        mc_reco_tracks = "MCRecoTracks"
        path.add_module('TrackFinderMCTruthRecoTracks',
                        RecoTracksStoreArrayName=mc_reco_tracks)
 
        # CDC track finding and MC matching
        cdc_reco_tracks = "CDCRecoTracks"
        add_cdc_track_finding(path, output_reco_tracks=cdc_reco_tracks)
        path.add_module("MCRecoTracksMatcher", UsePXDHits=False, UseSVDHits=False, UseCDCHits=True,
                        mcRecoTracksStoreArrayName=mc_reco_tracks,
                        prRecoTracksStoreArrayName=cdc_reco_tracks)
 
        path.add_module("DAFRecoFitter", recoTracksStoreArrayName=cdc_reco_tracks)
 
        # SVD track finding and MC matching
        svd_reco_tracks = "SVDRecoTracks"
        add_svd_standalone_tracking(path, reco_tracks=svd_reco_tracks)
        path.add_module("MCRecoTracksMatcher", UsePXDHits=False, UseSVDHits=True, UseCDCHits=False,
                        mcRecoTracksStoreArrayName=mc_reco_tracks,
                        prRecoTracksStoreArrayName=svd_reco_tracks)
 
        direction = "backward"
        path.add_module("CDCToSVDSeedCKF",
                        inputRecoTrackStoreArrayName=cdc_reco_tracks,
 
                        fromRelationStoreArrayName=cdc_reco_tracks,
                        toRelationStoreArrayName=svd_reco_tracks,
 
                        relatedRecoTrackStoreArrayName=svd_reco_tracks,
                        cdcTracksStoreArrayName=cdc_reco_tracks,
                        vxdTracksStoreArrayName=svd_reco_tracks,
 
                        relationCheckForDirection=direction,
                        reverseSeed=False,
                        firstHighFilterParameters={"direction": direction},
                        advanceHighFilterParameters={"direction": direction},
 
                        writeOutDirection=direction,
                        endEarly=False,
 
                        filter="recording_with_relations",
                        filterParameters={"rootFileName": result_filter_records_name})
 
        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 349 of file cdc_and_svd_ckf_merger_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.
        """
        yield self.add_to_output(self.result_filter_records_name)
 

requires()

def requires

(

self

)

This task requires that the training SplitMergeSimTask is finished.

Definition at line 356 of file cdc_and_svd_ckf_merger_mva_training.py.

    def requires(self):
        """
        This task requires that the training SplitMergeSimTask is finished.
        """
        yield SplitNMergeSimTask(
            bkgfiles_dir=MainTask.bkgfiles_by_exp[self.experiment_number],
            random_seed=self.random_seed,
            n_events=self.n_events_training,
            experiment_number=self.experiment_number,
        )
 

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 339 of file cdc_and_svd_ckf_merger_mva_training.py.

n_events_training

b2luigi n_events_training = b2luigi.IntParameter()

static

Number of events to generate.

Definition at line 341 of file cdc_and_svd_ckf_merger_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 344 of file cdc_and_svd_ckf_merger_mva_training.py.

result_filter_records_name

b2luigi result_filter_records_name = b2luigi.Parameter()

static

Name of the records file for training the final result filter.

Definition at line 347 of file cdc_and_svd_ckf_merger_mva_training.py.

---

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

• tracking/scripts/ckf_training/cdc_and_svd_merger/cdc_and_svd_ckf_merger_mva_training.py