Belle II Software development
ElossHarvestingRun Class Reference
Inheritance diagram for ElossHarvestingRun:
HarvestingRun HarvestingRunMixin StandardEventGenerationRun BrowseTFileOnTerminateRunMixin PostProcessingRunMixin ReadOrGenerateEventsRun PostProcessingRunMixin EmptyRun MinimalRun EmptyRun EmptyRun

Public Member Functions

 output_file_name (self)
 
 harvesting_module (self, path=None)
 
 create_argument_parser (self, **kwds)
 
 create_path (self)
 
 pickle_crops (self, harvesting_module, crops, **kwds)
 
 unpickle_crops (self)
 
 postprocess (self)
 
 adjust_path (self, path)
 
 run (self, path)
 
 name (self)
 
 configure_and_execute_from_commandline (self)
 
 execute (self)
 
 configure_from_commandline (self)
 
 configure (self, arguments)
 

Public Attributes

 pickle_crops = self.output_file_name
 

Static Public Attributes

bool show_results = False
 By default, do not show the browsing results.
 
bool postprocess_only = False
 By default, browse the output TFile too.
 
str description = "Empty execution of basf2"
 Description of the run setup to be displayed on command line.
 
 generator_module = None
 By default, do not generate events.
 
str detector_setup = "Default"
 By default, use the default detector setup.
 
list bkg_files = []
 By default, no background overlay.
 
 components = None
 By default, do specific components.
 
bool disable_deltas = False
 By default, do not disable delta-ray generation.
 
 simulation_output = None
 By default, do no store the simulation output.
 
bool allow_input = True
 By default, this basf2 job can read events from an input ROOT TFile.
 
int n_events = 10000
 By default, process 10000 events.
 
 root_input_file = None
 By default, there is no input ROOT TFile.
 
 random_seed = None
 By default, the random-number seed is unassigned.
 
int n_processes = 0
 By default, no parallel processing.
 
int n_events_to_skip = 0
 By default, do not skip any events at the start of the input ROOT TFile.
 

Detailed Description

Harvester to generate, postprocess and inspect MC events for energy-loss evaluation

Definition at line 93 of file record.py.

Member Function Documentation

◆ adjust_path()

adjust_path ( self,
path )
inherited
Add the harvester to the basf2 path

Reimplemented from EmptyRun.

Definition at line 66 of file run.py.

66 def adjust_path(self, path):
67 """Add the harvester to the basf2 path"""
68 super().adjust_path(path)
69 harvesting_module = self.harvesting_module()
70 if self.output_file_name:
71 harvesting_module.output_file_name = self.output_file_name
72 harvesting_module.refiners.append(self.pickle_crops)
73 path.add_module(harvesting_module)
74 return path
75
76

◆ configure()

configure ( self,
arguments )
inherited
Save the command-line arguments as key-value pairs

Reimplemented in CDCDisplayRun, and ReadOrGenerateEventsRun.

Definition at line 73 of file minimal.py.

73 def configure(self, arguments):
74 """Save the command-line arguments as key-value pairs"""
75 # Simply translate the arguments that have
76 # the same name as valid instance arguments
77 for (key, value) in list(vars(arguments).items()):
78 if value is None:
79 continue
80 if hasattr(self, key):
81 get_logger().info("Setting %s to %s", key, value)
82 setattr(self, key, value)
83

◆ configure_and_execute_from_commandline()

configure_and_execute_from_commandline ( self)
inherited
Configure basf2 job script from command-line arguments then run it

Definition at line 45 of file minimal.py.

45 def configure_and_execute_from_commandline(self):
46 """Configure basf2 job script from command-line arguments then run it"""
47 self.configure_from_commandline()
48 self.execute()
49

◆ configure_from_commandline()

configure_from_commandline ( self)
inherited
Convert the command-line arguments to a basf2 job script

Definition at line 67 of file minimal.py.

67 def configure_from_commandline(self):
68 """Convert the command-line arguments to a basf2 job script"""
69 argument_parser = self.create_argument_parser()
70 arguments = argument_parser.parse_args()
71 self.configure(arguments)
72

◆ create_argument_parser()

create_argument_parser ( self,
** kwds )
Convert command-line arguments to basf2 argument list

Reimplemented from HarvestingRunMixin.

Definition at line 114 of file record.py.

114 def create_argument_parser(self, **kwds):
115 """Convert command-line arguments to basf2 argument list"""
116 argument_parser = super().create_argument_parser(**kwds)
117 return argument_parser
118

◆ create_path()

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 EmptyRun.

Definition at line 119 of file record.py.

119 def create_path(self):
120 """
121 Sets up a path that plays back pregenerated events or generates events
122 based on the properties in the base class.
123 """
124 path = super().create_path()
125
126 path.add_module("TFCDC_WireHitPreparer",
127 logLevel=8,
128 flightTimeEstimation="outwards",
129 UseNLoops=1)
130
131 path.add_module('TFCDC_AxialTrackCreatorMCTruth',
132 logLevel=8,
133 useOnlyBeforeTOP=True,
134 fit=True,
135 reconstructedDriftLength=True,
136 reconstructedPositions=True)
137
138 return path
139
140

◆ execute()

execute ( self)
inherited
Create the basf2 path then run the job

Reimplemented in ReadOrGenerateEventsRun.

Definition at line 50 of file minimal.py.

50 def execute(self):
51 """Create the basf2 path then run the job"""
52 # Create path and run #
53
54 path = self.create_path()
55 self.adjust_path(path)
56 self.run(path)
57

◆ harvesting_module()

harvesting_module ( self,
path = None )
Harvest and post-process the generated events

Reimplemented from HarvestingRunMixin.

Definition at line 107 of file record.py.

107 def harvesting_module(self, path=None):
108 """Harvest and post-process the generated events"""
109 harvesting_module = ElossHarvestingModule(self.output_file_name)
110 if path:
111 path.add_module(harvesting_module)
112 return harvesting_module
113

◆ name()

name ( self)
inherited
provide name of this object

Definition at line 41 of file minimal.py.

41 def name(self):
42 """provide name of this object"""
43 return self.__class__.__name__
44

◆ output_file_name()

output_file_name ( self)
Get the output ROOT filename

Reimplemented from HarvestingRunMixin.

Definition at line 103 of file record.py.

103 def output_file_name(self):
104 """Get the output ROOT filename"""
105 return 'eloss.root'
106

◆ pickle_crops()

pickle_crops ( self,
harvesting_module,
crops,
** kwds )
inherited
Save the raw crops as a pickle file

Definition at line 41 of file run.py.

41 def pickle_crops(self, harvesting_module, crops, **kwds):
42 """Save the raw crops as a pickle file"""
43 with open(self.output_file_name + ".pickle", "wb") as pickle_file:
44 pickle.dump(crops, pickle_file)
45

◆ postprocess()

postprocess ( self)
inherited
Post-process the crops

Reimplemented from BrowseTFileOnTerminateRunMixin.

Definition at line 51 of file run.py.

51 def postprocess(self):
52 """Post-process the crops"""
53 if self.postprocess_only:
54 harvesting_module = self.harvesting_module()
55 if self.output_file_name:
56 harvesting_module.output_file_name = self.output_file_name
57 try:
58 crops = self.unpickle_crops()
59 except FileNotFoundError:
60 print("Crops pickle file not found. Create it now.")
61 else:
62 harvesting_module.refine(crops)
63
64 super().postprocess()
65

◆ run()

run ( self,
path )
inherited
Post-process the basf2 job output

Reimplemented from EmptyRun.

Definition at line 45 of file mixins.py.

45 def run(self, path):
46 """Post-process the basf2 job output"""
47 if not self.postprocess_only:
48 super().run(path)
49
50 self.postprocess()
51

◆ unpickle_crops()

unpickle_crops ( self)
inherited
Load the raw crops from a pickle file

Definition at line 46 of file run.py.

46 def unpickle_crops(self):
47 """Load the raw crops from a pickle file"""
48 with open(self.output_file_name + ".pickle", "rb") as pickle_file:
49 return pickle.load(pickle_file)
50

Member Data Documentation

◆ allow_input

bool allow_input = True
staticinherited

By default, this basf2 job can read events from an input ROOT TFile.

Definition at line 109 of file minimal.py.

◆ bkg_files

list bkg_files = []
staticinherited

By default, no background overlay.

Definition at line 47 of file event_generation.py.

◆ components

components = None
staticinherited

By default, do specific components.

Definition at line 49 of file event_generation.py.

◆ description

str description = "Empty execution of basf2"
staticinherited

Description of the run setup to be displayed on command line.

Definition at line 33 of file minimal.py.

◆ detector_setup

str detector_setup = "Default"
staticinherited

By default, use the default detector setup.

Definition at line 45 of file event_generation.py.

◆ disable_deltas

bool disable_deltas = False
staticinherited

By default, do not disable delta-ray generation.

Definition at line 51 of file event_generation.py.

◆ generator_module

generator_module = None
staticinherited

By default, do not generate events.

Definition at line 43 of file event_generation.py.

◆ n_events

n_events = 10000
staticinherited

By default, process 10000 events.

Definition at line 111 of file minimal.py.

◆ n_events_to_skip

int n_events_to_skip = 0
staticinherited

By default, do not skip any events at the start of the input ROOT TFile.

Definition at line 119 of file minimal.py.

◆ n_processes

n_processes = 0
staticinherited

By default, no parallel processing.

Definition at line 117 of file minimal.py.

◆ pickle_crops

pickle_crops = self.output_file_name
inherited

Definition at line 72 of file run.py.

◆ postprocess_only

bool postprocess_only = False
staticinherited

By default, browse the output TFile too.

Post-process the basf2 job output

Definition at line 29 of file mixins.py.

◆ random_seed

random_seed = None
staticinherited

By default, the random-number seed is unassigned.

Definition at line 115 of file minimal.py.

◆ root_input_file

root_input_file = None
staticinherited

By default, there is no input ROOT TFile.

Definition at line 113 of file minimal.py.

◆ show_results

bool show_results = False
staticinherited

By default, do not show the browsing results.

Definition at line 62 of file mixins.py.

◆ simulation_output

simulation_output = None
staticinherited

By default, do no store the simulation output.

Definition at line 53 of file event_generation.py.


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