PXDHitErrorsTTree Class Reference

Inheritance diagram for PXDHitErrorsTTree:

Collaboration diagram for PXDHitErrorsTTree:

Public Member Functions
def	__init__ (self)
def	beginRun (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

A simple module to check the reconstruction of PXDTrueHits.
This module writes its output to a ROOT tree.

Definition at line 51 of file PXDHitErrorsTTree.py.

Constructor & Destructor Documentation

__init__()

def __init__

(

self

)

Initialize the module

Definition at line 58 of file PXDHitErrorsTTree.py.

    def __init__(self):
        """Initialize the module"""
 
        super().__init__()
        
        self.file = ROOT.TFile('PXDHitErrorOutput.root', 'recreate')
        
        self.tree = ROOT.TTree('tree', 'Event data of PXD simulation')
        
        self.data = EventData()
        # Declare tree branches
        for key in EventData.__dict__:
            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

beginRun()

def beginRun

(

self

)

 Does nothing 

Definition at line 82 of file PXDHitErrorsTTree.py.

    def beginRun(self):
        """ Does nothing """
 

event()

def event

(

self

)

Find clusters with a truehit and print some stats.

Definition at line 85 of file PXDHitErrorsTTree.py.

    def event(self):
        """Find clusters with a truehit and print some stats."""
 
        clusters = Belle2.PyStoreArray('PXDClusters')
 
        # Start with clusters and use the relation to get the corresponding
        # digits and truehits.
        for cluster in clusters:
            cluster_truehits = cluster.getRelationsTo('PXDTrueHits')
 
            # Here we ask only for clusters with exactly one TrueHit.
            if len(cluster_truehits) != 1:
                continue
 
            for truehit in cluster_truehits:
                # Now let's store some data
                # Sesnor identification
                vxd_id = truehit.getSensorID()
                self.data.vxd_id = vxd_id.getID()
                self.data.layer = vxd_id.getLayerNumber()
                self.data.ladder = vxd_id.getLadderNumber()
                self.data.sensor = vxd_id.getSensorNumber()
                self.data.truehit_index = truehit.getArrayIndex()
                self.data.cluster_index = cluster.getArrayIndex()
 
                # Get sensor geometry information
                sensor_info = Belle2.VXD.GeoCache.get(vxd_id)
                thickness = sensor_info.getThickness()
 
                # TrueHit information
                self.data.truehit_u = truehit.getU()
                self.data.truehit_v = truehit.getV()
                self.data.truehit_time = truehit.getGlobalTime()
                self.data.truehit_charge = truehit.getEnergyDep()
                self.data.theta_u = math.atan2(
                    truehit.getExitU() - truehit.getEntryU(), thickness)
                self.data.theta_v = math.atan2(
                    truehit.getExitV() - truehit.getEntryV(), thickness)
                # Cluster information
                self.data.cluster_u = cluster.getU()
                self.data.cluster_v = cluster.getV()
                self.data.cluster_uError = cluster.getUSigma()
                self.data.cluster_vError = cluster.getVSigma()
                self.data.cluster_rho = cluster.getRho()
                self.data.cluster_charge = cluster.getCharge()
                self.data.cluster_seed = cluster.getSeedCharge()
                self.data.cluster_size = cluster.getSize()
                self.data.cluster_uSize = cluster.getUSize()
                self.data.cluster_vSize = cluster.getVSize()
                self.data.cluster_uPull = (
                    cluster.getU() - truehit.getU()) / cluster.getUSigma()
                self.data.cluster_vPull = (
                    cluster.getV() - truehit.getV()) / cluster.getVSigma()
                self.file.cd()
                self.tree.Fill()
 

terminate()

def terminate

(

self

)

 Close the output file.

Definition at line 141 of file PXDHitErrorsTTree.py.

    def terminate(self):
        """ Close the output file."""
 
        self.file.cd()
        self.file.Write()
        self.file.Close()

Member Data Documentation

data

data

Instance of EventData class.

Definition at line 67 of file PXDHitErrorsTTree.py.

file

file

Output ROOT file.

Definition at line 63 of file PXDHitErrorsTTree.py.

tree

tree

TTree for output data.

Definition at line 65 of file PXDHitErrorsTTree.py.

---

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

• pxd/examples/PXDHitErrorsTTree.py