DecayHashMap Class Reference

Public Member Functions
def	__init__ (self, rootfile, removeRadiativeGammaFlag=False)
def	get_string (self, decayHash, decayHashExtended)
def	get_original_decay (self, decayHash, decayHashExtended)
def	get_reconstructed_decay (self, decayHash, decayHashExtended)
def	print_hash (self, decayHash, decayHashExtended)

Protected Attributes
	_string
	Dict Int -> DecayStrings.
	_forest
	Dict Int -> Reconstructed DecayTree.

Detailed Description

DecayHashMap using the C++ implementation of DecayTree and DecayNode

Definition at line 49 of file decayHash.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			rootfile,
			removeRadiativeGammaFlag = False
	)

Constructor

Definition at line 54 of file decayHash.py.

    def __init__(self, rootfile, removeRadiativeGammaFlag=False):
        """Constructor"""
        # Always avoid the top-level 'import ROOT'.
        import ROOT  # noqa
        with uproot.open(rootfile) as rf:
            trees = rf.keys()
            assert len(trees) == 1
            ntuple = rf[trees[0]].arrays(library='np')
        # self._removeGammaFlag = removeRadiativeGammaFlag
        
        self._string = {}
        
        self._forest = {}
        for decayHash, decayHashExtended, decayString in zip(
                ntuple['decayHash'], ntuple['decayHashExtended'], ntuple['decayString']):
            decayInt = ROOT.Belle2.DecayForest.decayHashFloatToInt(decayHash, decayHashExtended)
            if decayInt in self._string:
                continue
            self._string[decayInt] = decayString
            self._forest[decayInt] = ROOT.Belle2.DecayForest(decayString, True, removeRadiativeGammaFlag)
 

Member Function Documentation

get_original_decay()

def get_original_decay	(		self,
			decayHash,
			decayHashExtended
	)

Return original (MC) DecayTree given the decayHash and decayHashExtended
@param decayHash output of extraInfo(decayHash)
@param decayHashExtended output of extraInfo(decayHashExtended)

Definition at line 85 of file decayHash.py.

    def get_original_decay(self, decayHash, decayHashExtended):
        """
        Return original (MC) DecayTree given the decayHash and decayHashExtended
        @param decayHash output of extraInfo(decayHash)
        @param decayHashExtended output of extraInfo(decayHashExtended)
        """
        # Always avoid the top-level 'import ROOT'.
        import ROOT  # noqa
        return self._forest[ROOT.Belle2.DecayForest.decayHashFloatToInt(decayHash, decayHashExtended)].getOriginalTree()
 

get_reconstructed_decay()

def get_reconstructed_decay	(		self,
			decayHash,
			decayHashExtended
	)

Return reconstructed DecayTree given the decayHash and decayHashExtended
@param decayHash output of extraInfo(decayHash)
@param decayHashExtended output of extraInfo(decayHashExtended)

Definition at line 95 of file decayHash.py.

    def get_reconstructed_decay(self, decayHash, decayHashExtended):
        """
        Return reconstructed DecayTree given the decayHash and decayHashExtended
        @param decayHash output of extraInfo(decayHash)
        @param decayHashExtended output of extraInfo(decayHashExtended)
        """
        # Always avoid the top-level 'import ROOT'.
        import ROOT  # noqa
        return self._forest[ROOT.Belle2.DecayForest.decayHashFloatToInt(decayHash, decayHashExtended)].getReconstructedTree()
 

get_string()

def get_string	(		self,
			decayHash,
			decayHashExtended
	)

Return DecayString given the decayHash and decayHashExtended
@param decayHash output of extraInfo(decayHash)
@param decayHashExtended output of extraInfo(decayHashExtended)

Definition at line 75 of file decayHash.py.

    def get_string(self, decayHash, decayHashExtended):
        """
        Return DecayString given the decayHash and decayHashExtended
        @param decayHash output of extraInfo(decayHash)
        @param decayHashExtended output of extraInfo(decayHashExtended)
        """
        # Always avoid the top-level 'import ROOT'.
        import ROOT  # noqa
        return self._string[ROOT.Belle2.DecayForest.decayHashFloatToInt(decayHash, decayHashExtended)]
 

print_hash()

def print_hash	(		self,
			decayHash,
			decayHashExtended
	)

Print the DecayString in a fancy way given the decayHash and decayHashExtended
@param decayHash output of extraInfo(decayHash)
@param decayHashExtended output of extraInfo(decayHashExtended)

Definition at line 105 of file decayHash.py.

    def print_hash(self, decayHash, decayHashExtended):
        """
        Print the DecayString in a fancy way given the decayHash and decayHashExtended
        @param decayHash output of extraInfo(decayHash)
        @param decayHashExtended output of extraInfo(decayHashExtended)
        """
        entry = self.get_string(decayHash, decayHashExtended)
        entries = entry.split('|')
        all_particles = re.findall(r"(-?[0-9]+)", entries[0])
 
        if len(all_particles) != len(entries) - 1:
            print(entry)
            raise RuntimeError("Bad format of decay string: " +
                               str(len(all_particles)) + " " + str(len(entries)) + " " + str(entries))
 
        table = []
        table.append(["Decay ", prettify_pdg_codes(entries[0])])
        for particle, mc_decay_string in zip(all_particles, entries[1:]):
            table.append([prettify_pdg_codes(particle), prettify_pdg_codes(mc_decay_string)])
 
        basf2.pretty_print_table(table, column_widths=[6, '*'])
 
 

Member Data Documentation

_forest

_forest

protected

Dict Int -> Reconstructed DecayTree.

Definition at line 66 of file decayHash.py.

_string

_string

protected

Dict Int -> DecayStrings.

Definition at line 64 of file decayHash.py.

---

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

• analysis/scripts/decayHash.py