GenerateSimTask Class Reference

Inheritance diagram for GenerateSimTask:

Collaboration diagram for GenerateSimTask:

Public Member Functions
def	output_file_name (self, n_events=None, random_seed=None)
	Name of the ROOT output file with generated and simulated events.
def	output (self)
def	create_path (self)

Static Public Attributes
b2luigi	experiment_number = b2luigi.IntParameter()
	Experiment number of the conditions database, e.g.
b2luigi	n_events = b2luigi.IntParameter()
	Number of events to generate.
b2luigi	random_seed = b2luigi.Parameter()
	Random basf2 seed.
b2luigi	bkgfiles_dir
	Directory with overlay background root files.
str	queue = 'l'
	specify queue.

Detailed Description

Generate simulated Monte Carlo with background overlay.

Make sure to use different ``random_seed`` parameters for the training data
format the classifier trainings and for the test data for the respective
evaluation/validation tasks.

Definition at line 166 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 213 of file cdc_and_svd_ckf_merger_mva_training.py.

    def create_path(self):
        """
        Create basf2 path to process with event generation and simulation.
        """
        basf2.set_random_seed(self.random_seed)
        path = basf2.create_path()
        path.add_module(
            "EventInfoSetter", evtNumList=[self.n_events], runList=[0], expList=[self.experiment_number]
        )
        path.add_module("EvtGenInput")
        bkg_files = ""
        # \cond suppress doxygen warning
        if self.experiment_number == 0:
            bkg_files = background.get_background_files()
        else:
            bkg_files = background.get_background_files(self.bkgfiles_dir)
        # \endcond
 
        simulation.add_simulation(path, bkgfiles=bkg_files, bkgOverlay=True, usePXDDataReduction=False)
 
        path.add_module(
            "RootOutput",
            outputFileName=self.get_output_file_name(self.output_file_name()),
        )
        return path
 
 
# I don't use the default MergeTask or similar because they only work if every input file is called the same.
# Additionally, I want to add more features like deleting the original input to save storage space.

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 206 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.output_file_name())
 

output_file_name()

def output_file_name	(		self,
			n_events = None,
			random_seed = None
	)

Name of the ROOT output file with generated and simulated events.

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

:param n_events: Number of events to simulate.
:param random_seed: Random seed to use for the simulation to create independent samples.

Definition at line 192 of file cdc_and_svd_ckf_merger_mva_training.py.

    def output_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.
 
        :param n_events: Number of events to simulate.
        :param random_seed: Random seed to use for the simulation to create independent samples.
        """
        if n_events is None:
            n_events = self.n_events
        if random_seed is None:
            random_seed = self.random_seed
        return "generated_mc_N" + str(n_events) + "_" + random_seed + ".root"
 

Member Data Documentation

bkgfiles_dir

b2luigi bkgfiles_dir

static

Initial value:

=  b2luigi.Parameter(
        
 
 
    )

Directory with overlay background root files.

Definition at line 183 of file cdc_and_svd_ckf_merger_mva_training.py.

experiment_number

b2luigi experiment_number = b2luigi.IntParameter()

static

Experiment number of the conditions database, e.g.

defines simulation geometry

Definition at line 176 of file cdc_and_svd_ckf_merger_mva_training.py.

n_events

b2luigi n_events = b2luigi.IntParameter()

static

Number of events to generate.

Definition at line 178 of file cdc_and_svd_ckf_merger_mva_training.py.

queue

str queue = 'l'

static

specify queue.

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

Definition at line 189 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 181 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