PostReconstruction Class Reference

Public Member Functions
	__init__ (self, typing.Sequence[config.Particle] particles, config.FeiConfiguration config)
typing.Sequence[str]	get_missing_channels (self)
bool	available (self)
pybasf2.Path	reconstruct (self)

Public Attributes
	particles = particles
	list of config.Particle objects
	config = config
	config.FeiConfiguration object

Detailed Description

Steers the reconstruction phase after the mva method was applied
It Includes:
    - The application of the mva method itself.
    - Copying all channel lists in a common one for each particle defined in particles
    - Tag unique signal candidates, to avoid double counting of channels with overlap

Definition at line 486 of file core.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
		typing.Sequence[config.Particle]	particles,
		config.FeiConfiguration	config )

Create a new PostReconstruction object
@param particles list of config.Particle objects
@param config config.FeiConfiguration object

Definition at line 495 of file core.py.

    def __init__(self, particles: typing.Sequence[config.Particle], config: config.FeiConfiguration):
        """
        Create a new PostReconstruction object
        @param particles list of config.Particle objects
        @param config config.FeiConfiguration object
        """
        
        self.particles = particles
        
        self.config = config
 

Member Function Documentation

available()

bool available

(

self

)

Check if the relevant information is already available

Definition at line 519 of file core.py.

    def available(self) -> bool:
        """
        Check if the relevant information is already available
        """
        return len(self.get_missing_channels()) == 0
 

get_missing_channels()

typing.Sequence[str] get_missing_channels

(

self

)

Returns all channels for which the weightfile is missing

Definition at line 506 of file core.py.

    def get_missing_channels(self) -> typing.Sequence[str]:
        """
        Returns all channels for which the weightfile is missing
        """
        missing = []
        for particle in self.particles:
            for channel in particle.channels:
                # weightfile = self.config.prefix + '_' + channel.label
                weightfile = f'{channel.label}.xml'
                if not basf2_mva.available(weightfile):
                    missing += [channel.label]
        return missing
 

reconstruct()

pybasf2.Path reconstruct

(

self

)

Returns pybasf2.Path which reconstructs the particles and does the vertex fitting if necessary

Definition at line 525 of file core.py.

    def reconstruct(self) -> pybasf2.Path:
        """
        Returns pybasf2.Path which reconstructs the particles and does the vertex fitting if necessary
        """
        import ROOT  # noqa
        path = basf2.create_path()
 
        for particle in self.particles:
            for channel in particle.channels:
                expert = basf2.register_module('MVAExpert')
                expert.set_name(f'MVAExpert_{channel.name}')
                if self.config.training:
                    expert.param('identifier', f'{channel.label}.xml')
                else:
                    expert.param('identifier', f'{self.config.prefix}_{channel.label}')
                expert.param('extraInfoName', 'SignalProbability')
                expert.param('listNames', [channel.name])
                # suppress warning that signal probability won't be overwritten if it already exists
                expert.set_log_level(basf2.logging.log_level.ERROR)
                path.add_module(expert)
 
                if self.config.monitor:
                    if self.config.monitor == 'simple':
                        hist_variables = [channel.mvaConfig.target, 'extraInfo(decayModeID)']
                        hist_variables_2d = [(channel.mvaConfig.target, 'extraInfo(decayModeID)')]
                    else:
                        hist_variables = ['mcErrors',
                                          'mcParticleStatus',
                                          'extraInfo(SignalProbability)',
                                          channel.mvaConfig.target,
                                          'extraInfo(decayModeID)'] + list(channel.mvaConfig.spectators.keys())
                        hist_variables_2d = [('extraInfo(SignalProbability)', channel.mvaConfig.target),
                                             ('extraInfo(SignalProbability)', 'mcErrors'),
                                             ('extraInfo(SignalProbability)', 'mcParticleStatus'),
                                             ('extraInfo(decayModeID)', channel.mvaConfig.target),
                                             ('extraInfo(decayModeID)', 'mcErrors'),
                                             ('extraInfo(decayModeID)', 'mcParticleStatus')]
                        for specVar in channel.mvaConfig.spectators:
                            hist_variables_2d.append(('extraInfo(SignalProbability)', specVar))
                            hist_variables_2d.append(('extraInfo(decayModeID)', specVar))
                            hist_variables_2d.append((channel.mvaConfig.target, specVar))
                    filename = os.path.join(self.config.monitoring_path, 'Monitor_PostReconstruction_AfterMVA.root')
                    ma.variablesToHistogram(
                        channel.name,
                        variables=config.variables2binnings(hist_variables),
                        variables_2d=config.variables2binnings_2d(hist_variables_2d),
                        filename=filename,
                        ignoreCommandLineOverride=True,
                        directory=f'{channel.label}',
                        path=path)
 
            cutstring = ''
            if particle.postCutConfig.value > 0.0:
                cutstring = f'{particle.postCutConfig.value} < extraInfo(SignalProbability)'
 
            ma.mergeListsWithBestDuplicate(particle.identifier, [c.name for c in particle.channels],
                                           variable='particleSource', writeOut=True, path=path)
 
            if self.config.monitor:
                if self.config.monitor == 'simple':
                    hist_variables = [particle.mvaConfig.target, 'extraInfo(decayModeID)']
                    hist_variables_2d = [(particle.mvaConfig.target, 'extraInfo(decayModeID)')]
                else:
                    hist_variables = ['mcErrors',
                                      'mcParticleStatus',
                                      'extraInfo(SignalProbability)',
                                      particle.mvaConfig.target,
                                      'extraInfo(decayModeID)'] + list(particle.mvaConfig.spectators.keys())
                    hist_variables_2d = [('extraInfo(decayModeID)', particle.mvaConfig.target),
                                         ('extraInfo(decayModeID)', 'mcErrors'),
                                         ('extraInfo(decayModeID)', 'mcParticleStatus')]
                    for specVar in particle.mvaConfig.spectators:
                        hist_variables_2d.append(('extraInfo(SignalProbability)', specVar))
                        hist_variables_2d.append(('extraInfo(decayModeID)', specVar))
                        hist_variables_2d.append((particle.mvaConfig.target, specVar))
                filename = os.path.join(self.config.monitoring_path, 'Monitor_PostReconstruction_BeforePostCut.root')
                ma.variablesToHistogram(
                    particle.identifier,
                    variables=config.variables2binnings(hist_variables),
                    variables_2d=config.variables2binnings_2d(hist_variables_2d),
                    filename=filename,
                    ignoreCommandLineOverride=True,
                    directory=config.removeJPsiSlash(f'{particle.identifier}'),
                    path=path)
 
            ma.applyCuts(particle.identifier, cutstring, path=path)
 
            if self.config.monitor:
                filename = os.path.join(self.config.monitoring_path, 'Monitor_PostReconstruction_BeforeRanking.root')
                ma.variablesToHistogram(
                    particle.identifier,
                    variables=config.variables2binnings(hist_variables),
                    variables_2d=config.variables2binnings_2d(hist_variables_2d),
                    filename=filename,
                    ignoreCommandLineOverride=True,
                    directory=config.removeJPsiSlash(f'{particle.identifier}'),
                    path=path)
 
            ma.rankByHighest(particle.identifier, 'extraInfo(SignalProbability)',
                             particle.postCutConfig.bestCandidateCut, 'postCut_rank', path=path)
 
            uniqueSignal = basf2.register_module('TagUniqueSignal')
            uniqueSignal.param('particleList', particle.identifier)
            uniqueSignal.param('target', particle.mvaConfig.target)
            uniqueSignal.param('extraInfoName', 'uniqueSignal')
            uniqueSignal.set_name(f'TagUniqueSignal_{particle.identifier}')
            # suppress warning that unique signal extra info won't be overwritten if it already exists
            uniqueSignal.set_log_level(basf2.logging.log_level.ERROR)
            path.add_module(uniqueSignal)
 
            if self.config.monitor:
                if self.config.monitor != 'simple':
                    hist_variables += ['extraInfo(postCut_rank)']
                    hist_variables_2d += [('extraInfo(decayModeID)', 'extraInfo(postCut_rank)'),
                                          (particle.mvaConfig.target, 'extraInfo(postCut_rank)'),
                                          ('mcErrors', 'extraInfo(postCut_rank)'),
                                          ('mcParticleStatus', 'extraInfo(postCut_rank)')]
                    for specVar in particle.mvaConfig.spectators:
                        hist_variables_2d.append(('extraInfo(postCut_rank)', specVar))
                filename = os.path.join(self.config.monitoring_path, 'Monitor_PostReconstruction_AfterRanking.root')
                ma.variablesToHistogram(
                    particle.identifier,
                    variables=config.variables2binnings(hist_variables),
                    variables_2d=config.variables2binnings_2d(hist_variables_2d),
                    filename=filename,
                    ignoreCommandLineOverride=True,
                    directory=config.removeJPsiSlash(f'{particle.identifier}'),
                    path=path)
 
                filename = os.path.join(self.config.monitoring_path, 'Monitor_Final.root')
                if self.config.monitor == 'simple':
                    hist_variables = ['extraInfo(uniqueSignal)', 'extraInfo(decayModeID)']
                    hist_variables_2d = [('extraInfo(uniqueSignal)', 'extraInfo(decayModeID)')]
                    ma.variablesToHistogram(
                        particle.identifier,
                        variables=config.variables2binnings(hist_variables),
                        variables_2d=config.variables2binnings_2d(hist_variables_2d),
                        filename=filename,
                        ignoreCommandLineOverride=True,
                        directory=config.removeJPsiSlash(f'{particle.identifier}'),
                        path=path)
                else:
                    variables = ['extraInfo(SignalProbability)', 'mcErrors', 'mcParticleStatus', particle.mvaConfig.target,
                                 'extraInfo(uniqueSignal)', 'extraInfo(decayModeID)'] + list(particle.mvaConfig.spectators.keys())
 
                    ma.variablesToNtuple(
                        particle.identifier,
                        variables,
                        treename=ROOT.Belle2.MakeROOTCompatible.makeROOTCompatible(
                            config.removeJPsiSlash(f'{particle.identifier} variables')),
                        filename=filename,
                        ignoreCommandLineOverride=True,
                        path=path)
        return path
 
 

Member Data Documentation

config

config = config

config.FeiConfiguration object

Definition at line 504 of file core.py.

particles

particles = particles

list of config.Particle objects

Definition at line 502 of file core.py.

---

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

• analysis/scripts/fei/core.py