Belle II Software  release-05-02-19
ElossHarvestingModule Class Reference
Inheritance diagram for ElossHarvestingModule:
Collaboration diagram for ElossHarvestingModule:

Public Member Functions

def __init__ (self, output_file_name)
 
def initialize (self)
 
def prepare (self)
 
def pick (self, track)
 
def peel (self, track)
 

Public Attributes

 mc_track_lookup
 by default, there is no method to find matching MC tracks More...
 
 track_fitter
 Use the CDCReimannFitter with a constrained origin for track fitting.
 
 mc_hit_lookup
 Method to find matching MC hits.
 
 eloss_estimator
 Method to estimate dE/dx in the CDC.
 
 bfield
 Method to interrogate the magnetic field values.
 

Static Public Attributes

 save_tree = refiners.save_tree()
 Refiners to be executed at the end of the harvesting / termination of the module Save a tree of all collected variables in a sub folder.
 
 save_histograms
 Save histograms in a sub folder. More...
 
 save_histograms_stackby_charge
 Save the histograms, stacked by charge, in a sub folder. More...
 
 save_scatter
 Save the scatterplots in a sub folder. More...
 
 save_profiles
 Save the profile histograms to the output ROOT file. More...
 
 save_bz_profiles
 Save the magnetic-field profile histogram in a sub folder. More...
 
 save_cid_histogram
 Save the eloss-displacement histograms in a sub folder. More...
 
 save_cid_profiles
 Save the eloss-displacement profile histograms in a sub folder. More...
 
 save_cid_scatters
 Save the eloss-displacement scatterplots in a sub folder. More...
 
 save_energy_cid_histogram
 Save the eloss histograms in a sub folder. More...
 
 save_energy_cid_profiles
 Save the eloss profile histograms in a sub folder. More...
 
 save_energy_cid_scatters
 Save the eloss profile scatterplots in a sub folder. More...
 

Detailed Description

Module to collect information about the generated segments and
compose validation plots on terminate.

Definition at line 146 of file record.py.

Constructor & Destructor Documentation

◆ __init__()

def __init__ (   self,
  output_file_name 
)
Constructor

Definition at line 151 of file record.py.

151  def __init__(self, output_file_name):
152  """Constructor"""
153  super().__init__(foreach='CDCTrackVector',
154  output_file_name=output_file_name)
155 
156 
157  self.mc_track_lookup = None
158 
159  origin_track_fitter = TFCDC.CDCRiemannFitter()
160  origin_track_fitter.setOriginConstrained()
161 
162  self.track_fitter = origin_track_fitter
163 

Member Function Documentation

◆ initialize()

def initialize (   self)
Receive signal at the start of event processing

Definition at line 164 of file record.py.

◆ peel()

def peel (   self,
  track 
)
Aggregate the track and MC information for dE/dx analysis

Definition at line 197 of file record.py.

◆ pick()

def pick (   self,
  track 
)
Select tracks with at least 4 segments and associated primary MC particle with pt >= 0.25 GeV/c

Definition at line 180 of file record.py.

◆ prepare()

def prepare (   self)
Initialize the MC-hit lookup method

Definition at line 176 of file record.py.

Member Data Documentation

◆ mc_track_lookup

mc_track_lookup

by default, there is no method to find matching MC tracks

Method to find matching MC tracks.

Definition at line 157 of file record.py.

◆ save_bz_profiles

save_bz_profiles
static
Initial value:
= refiners.save_profiles(
x='r',
y='bz',
)

Save the magnetic-field profile histogram in a sub folder.

Definition at line 433 of file record.py.

◆ save_cid_histogram

save_cid_histogram
static
Initial value:
= refiners.save_histograms(
select=[
# 'mc_disp2D',
# 'mc_delossed_disp2D',
'first_disp2D',
'first_delossed_disp2D',
'bz',
# 'disp2D',
],
outlier_z_score=5.0,
groupby=["layer_cid"],
# stackby="pdg_code",
)

Save the eloss-displacement histograms in a sub folder.

Definition at line 442 of file record.py.

◆ save_cid_profiles

save_cid_profiles
static
Initial value:
= refiners.save_profiles(
x=["mc_pt"],
y=[
# 'mc_disp2D',
# 'mc_delossed_disp2D',
'first_disp2D',
'first_delossed_disp2D',
],
outlier_z_score=5.0,
groupby=["layer_cid"],
stackby="pdg_code",
)

Save the eloss-displacement profile histograms in a sub folder.

Definition at line 457 of file record.py.

◆ save_cid_scatters

save_cid_scatters
static
Initial value:
= refiners.save_scatters(
x=["mc_pt"],
y=[
# 'mc_disp2D',
# 'mc_delossed_disp2D',
'first_disp2D',
'first_delossed_disp2D',
],
outlier_z_score=5.0,
groupby=["layer_cid"],
stackby="pdg_code",
)

Save the eloss-displacement scatterplots in a sub folder.

Definition at line 471 of file record.py.

◆ save_energy_cid_histogram

save_energy_cid_histogram
static
Initial value:
= refiners.save_histograms(
select=[
'pdg_code',
'first_eloss_estimate',
'first_eloss_truth',
],
outlier_z_score=5.0,
groupby=["layer_cid"],
stackby="pdg_code",
folder_name='energy/{groupby_addition}',
)

Save the eloss histograms in a sub folder.

Definition at line 487 of file record.py.

◆ save_energy_cid_profiles

save_energy_cid_profiles
static
Initial value:
= refiners.save_profiles(
x=["mc_pt"],
y=[
'first_eloss_truth',
'first_eloss_estimate',
],
outlier_z_score=5.0,
groupby=["layer_cid"],
stackby="pdg_code",
folder_name='energy/{groupby_addition}',
)

Save the eloss profile histograms in a sub folder.

Definition at line 500 of file record.py.

◆ save_energy_cid_scatters

save_energy_cid_scatters
static
Initial value:
= refiners.save_scatters(
x=["mc_pt"],
y=[
'first_eloss_truth',
'first_eloss_estimate',
],
outlier_z_score=5.0,
groupby=["layer_cid"],
stackby="pdg_code",
folder_name='energy/{groupby_addition}',
)

Save the eloss profile scatterplots in a sub folder.

Definition at line 513 of file record.py.

◆ save_histograms

save_histograms
static
Initial value:
= refiners.save_histograms(
outlier_z_score=5.0,
allow_discrete=True,
)

Save histograms in a sub folder.

Definition at line 390 of file record.py.

◆ save_histograms_stackby_charge

save_histograms_stackby_charge
static
Initial value:
= refiners.save_histograms(
select=[
# "mc_disp2D",
"first_disp2D",
"charge",
],
outlier_z_score=5.0,
allow_discrete=True,
fit="gaus",
fit_z_score=1,
groupby="charge",
)

Save the histograms, stacked by charge, in a sub folder.

Definition at line 396 of file record.py.

◆ save_profiles

save_profiles
static
Initial value:
= refiners.save_profiles(
x=['mc_s2D'],
y=[
# 'mc_disp2D',
'first_disp2D',
'disp2D',
],
groupby=[None, "charge"],
)

Save the profile histograms to the output ROOT file.

Definition at line 422 of file record.py.

◆ save_scatter

save_scatter
static
Initial value:
= refiners.save_scatters(
x=['mc_s2D'],
y=[
# 'mc_disp2D',
'disp2D',
],
groupby=[None, "charge"],
filter=lambda x: x == 211,
filter_on="pdg_code",
)

Save the scatterplots in a sub folder.

Definition at line 410 of file record.py.


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