SaveTreeRefiner Class Reference

Inheritance diagram for SaveTreeRefiner:

Public Member Functions
def	__init__ (self, name=None, title=None)
def	refine (self, harvesting_module, crops, tdirectory=None, groupby_part_name=None, groupby_value=None, **kwds)
def	add_branch (self, output_ttree, part_name, parts)

Public Attributes
	name
	cached name for this TTree
	title
	cached title for this TTree

Static Public Attributes
str	default_name = "{module.id}_tree"
	default name for this TTree
str	default_title = "Tree of {module.id}"
	default title for this TTree

Detailed Description

Refiner for ROOT TTrees

Definition at line 817 of file refiners.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			name = None,
			title = None
	)

Constructor for this refiner

Reimplemented from Refiner.

Definition at line 825 of file refiners.py.

                 title=None):
        """Constructor for this refiner"""
        super().__init__()
 
        
        self.name = name
        
        self.title = title
 

Member Function Documentation

add_branch()

def add_branch	(		self,
			output_ttree,
			part_name,
			parts
	)

Add a TBranch to the TTree

Definition at line 867 of file refiners.py.

    def add_branch(self, output_ttree, part_name, parts):
        """Add a TBranch to the TTree"""
        input_value = np.zeros(1, dtype=float)
 
        branch_type_spec = f'{part_name}/D'
        tbranch = output_ttree.Branch(part_name, input_value, branch_type_spec)
 
        if output_ttree.GetNbranches() == 1:
            # On filling of the first branch we need to use the fill method of the TTree
            # For all other branches we can use the one of the branch
            # #justrootthings
            for value in parts:
                input_value[0] = value
                output_ttree.Fill()
 
        else:
            for value in parts:
                input_value[0] = value
                tbranch.Fill()
 
        output_ttree.GetEntry(0)
        output_ttree.ResetBranchAddress(tbranch)
        also_subbranches = True  # No subbranches here but we drop the buffers just in case.
        output_ttree.DropBranchFromCache(tbranch, also_subbranches)
 
 

refine()

def refine	(		self,
			harvesting_module,
			crops,
			tdirectory = None,
			groupby_part_name = None,
			groupby_value = None,
		**	kwds
	)

Process the TTree

Reimplemented from Refiner.

Definition at line 836 of file refiners.py.

               **kwds):
        """Process the TTree"""
 
        replacement_dict = dict(
            refiner=self,
            module=harvesting_module,
            groupby_key='_' + groupby_part_name + groupby_value if groupby_part_name else "",
            groupby=groupby_part_name,  # deprecated
            groupby_value=groupby_value,  # deprecated
        )
 
        with root_cd(tdirectory):
            name = self.name or self.default_name
            title = self.title or self.default_title
 
            name = formatter.format(name, **replacement_dict)
            title = formatter.format(title, **replacement_dict)
 
            output_ttree = ROOT.TTree(root_save_name(name), title)
            for part_name, parts in iter_items_sorted_for_key(crops):
                self.add_branch(output_ttree, part_name, parts)
 
            output_ttree.FlushBaskets()
            output_ttree.Write()
 

Member Data Documentation

default_name

str default_name = "{module.id}_tree"

static

default name for this TTree

Definition at line 821 of file refiners.py.

default_title

str default_title = "Tree of {module.id}"

static

default title for this TTree

Definition at line 823 of file refiners.py.

name

name

cached name for this TTree

Definition at line 832 of file refiners.py.

title

title

cached title for this TTree

Definition at line 834 of file refiners.py.

---

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

• tracking/scripts/tracking/harvest/refiners.py