TreeWriterModule Class Reference

Inheritance diagram for TreeWriterModule:

Collaboration diagram for TreeWriterModule:

Public Member Functions
def	__init__ (self)
def	event (self)
def	terminate (self)

Public Attributes
	file
	Output ROOT file.
	tree
	TTree for output data.
	data
	Instance of EventData class.

Detailed Description

This module writes its output to a ROOT tree.
Adapted from pxd/validation/PXDValidationTTreeSimHit.py

Definition at line 46 of file pid_ttree_writer.py.

Constructor & Destructor Documentation

__init__()

def __init__

(

self

)

Initialize the module

Definition at line 53 of file pid_ttree_writer.py.

    def __init__(self):
        """Initialize the module"""
 
        super(TreeWriterModule, self).__init__()
 
        
        self.file = ROOT.TFile('PID_TTree.root', 'recreate')
        
        self.tree = ROOT.TTree('tree', '')
        
        self.data = TreeStruct()
        """ Declare tree branches """
        for key in TreeStruct.__dict__.keys():
            if '__' not in key:
                formstring = '/F'
                if isinstance(self.data.__getattribute__(key), int):
                    formstring = '/I'
                self.tree.Branch(key, AddressOf(self.data, key), key + formstring)
 

Member Function Documentation

event()

def event

(

self

)

Store TOP and dE/dx info in tree

Definition at line 72 of file pid_ttree_writer.py.

terminate()

def terminate

(

self

)

Close the output file.

Definition at line 126 of file pid_ttree_writer.py.

---

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

• reconstruction/examples/pid_ttree_writer.py