PXDHitErrorsTTree Class Reference

Inheritance diagram for PXDHitErrorsTTree:

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

Public Attributes
	file = ROOT.TFile('PXDHitErrorOutput.root', 'recreate')
	Output ROOT file.
	tree = ROOT.TTree('tree', 'Event data of PXD simulation')
	TTree for output data.
	data = EventData()
	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__()

__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()

beginRun

(

self

)

 Does nothing 

Definition at line 82 of file PXDHitErrorsTTree.py.

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

event()

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
                # Sensor 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.getInstance().getSensorInfo(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()

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 = EventData()

Instance of EventData class.

Definition at line 67 of file PXDHitErrorsTTree.py.

file

file = ROOT.TFile('PXDHitErrorOutput.root', 'recreate')

Output ROOT file.

Definition at line 63 of file PXDHitErrorsTTree.py.

tree

tree = ROOT.TTree('tree', 'Event data of PXD simulation')

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