Inheritance diagram for VXDQEDataCollectionTask:

Public Member Functions
	get_records_file_name (self, n_events=None, random_seed=None)
	Filename of the recorded/collected data for the final QE MVA training.

	get_input_files (self, n_events=None, random_seed=None)

	requires (self)

	output (self)

	create_path (self)

Static Public Attributes
	n_events = b2luigi.IntParameter()
	Number of events to generate.

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

	random_seed = b2luigi.Parameter()
	Random basf2 seed used by the GenerateSimTask.

str	queue = 'l'
	specify queue.

Detailed Description

Collect variables/features from VXDTF2 tracking and write them to a ROOT
file.

These variables are to be used as labelled training data for the MVA
classifier which is the VXD track quality estimator

Definition at line 660 of file combined_quality_estimator_teacher.py.

Member Function Documentation

◆ create_path()

create_path ( self )

Create basf2 path with VXDTF2 tracking and VXD QE data collection.

Definition at line 746 of file combined_quality_estimator_teacher.py.

    def create_path(self):
        """
        Create basf2 path with VXDTF2 tracking and VXD QE data collection.
        """
        path = basf2.create_path()
        inputFileNames = self.get_input_files()
        path.add_module(
            "RootInput",
            inputFileNames=inputFileNames,
        )
        path.add_module("Gearbox")
        tracking.add_geometry_modules(path)
        if 'DATA' in self.random_seed:
            from rawdata import add_unpackers
            add_unpackers(path, components=['SVD', 'PXD'])
        tracking.add_hit_preparation_modules(path)
        add_vxd_track_finding_vxdtf2(
            path, components=["SVD"], add_mva_quality_indicator=False
        )
        if 'DATA' in self.random_seed:
            path.add_module(
                "VXDQETrainingDataCollector",
                TrainingDataOutputName=self.get_output_file_name(self.get_records_file_name()),
                SpacePointTrackCandsStoreArrayName="SPTrackCands",
                EstimationMethod="tripletFit",
                UseTimingInfo=False,
                ClusterInformation="Average",
                MCStrictQualityEstimator=False,
                mva_target=False,
                MCInfo=False,
                )
        else:
            path.add_module(
                "TrackFinderMCTruthRecoTracks",
                RecoTracksStoreArrayName="MCRecoTracks",
                WhichParticles=[],
                UsePXDHits=False,
                UseSVDHits=True,
                UseCDCHits=False,
                )
            path.add_module(
                "VXDQETrainingDataCollector",
                TrainingDataOutputName=self.get_output_file_name(self.get_records_file_name()),
                SpacePointTrackCandsStoreArrayName="SPTrackCands",
                EstimationMethod="tripletFit",
                UseTimingInfo=False,
                ClusterInformation="Average",
                MCStrictQualityEstimator=True,
                mva_target=False,
                )
        return path
 
 

◆ get_input_files()

get_input_files	(	self,
		n_events = None,
		random_seed = None )

Get input file names depending on the use case: If they already exist, search in
the corresponding folders, for data check the specified list and if they are created
in the same run, check for the task that produced them.

Definition at line 699 of file combined_quality_estimator_teacher.py.

    def get_input_files(self, n_events=None, random_seed=None):
        """
        Get input file names depending on the use case: If they already exist, search in
        the corresponding folders, for data check the specified list and if they are created
        in the same run, check for the task that produced them.
        """
        if n_events is None:
            n_events = self.n_events
        if random_seed is None:
            random_seed = self.random_seed
        if "USESIM" in random_seed:
            if 'USESIMBB' in random_seed:
                random_seed = 'BBBAR_' + random_seed.split("_", 1)[1]
            elif 'USESIMEE' in random_seed:
                random_seed = 'BHABHA_' + random_seed.split("_", 1)[1]
            return ['datafiles/' + GenerateSimTask.output_file_name(GenerateSimTask,
                                                                    n_events=n_events, random_seed=random_seed)]
        elif "DATA" in random_seed:
            return MasterTask.datafiles
        else:
            return self.get_input_file_names(GenerateSimTask.output_file_name(
                GenerateSimTask, n_events=n_events, random_seed=random_seed))
 

◆ get_records_file_name()

get_records_file_name	(	self,
		n_events = None,
		random_seed = None )

Filename of the recorded/collected data for the final QE MVA training.

Create output file name depending on number of events and production
mode that is specified in the random_seed string.

Definition at line 679 of file combined_quality_estimator_teacher.py.

    def get_records_file_name(self, n_events=None, random_seed=None):
        """
        Create output file name depending on number of events and production
        mode that is specified in the random_seed string.
        """
        if n_events is None:
            n_events = self.n_events
        if random_seed is None:
            random_seed = self.random_seed
        if 'vxd' not in random_seed:
            random_seed += '_vxd'
        if 'DATA' in random_seed:
            return 'qe_records_DATA_vxd.root'
        else:
            if 'USESIMBB' in random_seed:
                random_seed = 'BBBAR_' + random_seed.split("_", 1)[1]
            elif 'USESIMEE' in random_seed:
                random_seed = 'BHABHA_' + random_seed.split("_", 1)[1]
            return 'qe_records_N' + str(n_events) + '_' + random_seed + '.root'
 

◆ output()

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 739 of file combined_quality_estimator_teacher.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.get_records_file_name())
 

◆ requires()

requires ( self )

Generate list of luigi Tasks that this Task depends on.

Definition at line 722 of file combined_quality_estimator_teacher.py.

    def requires(self):
        """
        Generate list of luigi Tasks that this Task depends on.
        """
        if "USESIM" in self.random_seed or "DATA" in self.random_seed:
            for filename in self.get_input_files():
                yield CheckExistingFile(
                    filename=filename,
                    )
        else:
            yield SplitNMergeSimTask(
                bkgfiles_dir=MasterTask.bkgfiles_by_exp[self.experiment_number],
                random_seed=self.random_seed,
                n_events=self.n_events,
                experiment_number=self.experiment_number,
                )
 

Member Data Documentation

◆ experiment_number

experiment_number = b2luigi.IntParameter()

static

Experiment number of the conditions database, e.g.

defines simulation geometry

Definition at line 671 of file combined_quality_estimator_teacher.py.

◆ n_events

n_events = b2luigi.IntParameter()

static

Number of events to generate.

Definition at line 669 of file combined_quality_estimator_teacher.py.

◆ queue

str queue = 'l'

static

specify queue.

E.g. choose between 'l' (long), 's' (short) or 'sx' (short, extra ram)

Definition at line 676 of file combined_quality_estimator_teacher.py.

◆ random_seed

random_seed = b2luigi.Parameter()

static

Random basf2 seed used by the GenerateSimTask.

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

Definition at line 674 of file combined_quality_estimator_teacher.py.

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

tracking/scripts/tracking/train/combined_quality_estimator_teacher.py

Public Member Functions

Static Public Attributes

Detailed Description

Member Function Documentation

◆ create_path()

◆ get_input_files()

◆ get_records_file_name()

◆ output()

◆ requires()

Member Data Documentation

◆ experiment_number

◆ n_events

◆ queue

◆ random_seed