SVDValidationTTreeRecoDigit Class Reference

Inheritance diagram for SVDValidationTTreeRecoDigit:

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

Public Attributes
	file = ROOT.TFile('../SVDValidationTTreeRecoDigit.root', 'recreate')
	output file
	tree = ROOT.TTree('tree', 'Event data of SVD validation events')
	ttree
	data = EventDataRecoDigit()
	instance of event data class

Detailed Description

class to create reco digits ttree

Definition at line 45 of file SVDValidationTTreeRecoDigit.py.

Constructor & Destructor Documentation

__init__()

__init__

(

self

)

Initialize the module

Definition at line 48 of file SVDValidationTTreeRecoDigit.py.

    def __init__(self):
        """Initialize the module"""
 
        super().__init__()
        
        self.file = ROOT.TFile('../SVDValidationTTreeRecoDigit.root', 'recreate')
        
        self.tree = ROOT.TTree('tree', 'Event data of SVD validation events')
        
        self.data = EventDataRecoDigit()
 
        # Declare tree branches
        for key in EventDataRecoDigit.__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

event()

event

(

self

)

Take digits from SVDRecoDigits with at least one truehit and save needed information

Definition at line 67 of file SVDValidationTTreeRecoDigit.py.

    def event(self):
        """Take digits from SVDRecoDigits with at least one truehit and save needed information"""
        digits = Belle2.PyStoreArray('SVDRecoDigits')
        shaperDigits = Belle2.PyStoreArray('SVDShaperDigits')
        for digit in digits:
            # get the true hit from the related SVDShaperDigit
            # it works because there is a 1-to-1 correspondence between
            # ShaperDigits and RecoDigits
            digit_truehits = shaperDigits[digit.getArrayIndex()].getRelationsTo('SVDTrueHits')
 
            if len(digit_truehits) == 0:
                continue
 
            # find the trueHit with highest energy deposit (the "best" match)
            energy = 0
            bestTrueHitIndex = 0
 
            for i, trueHit in enumerate(digit_truehits):
                if trueHit.getEnergyDep() > energy:
                    energy = trueHit.getEnergyDep()
                    bestTrueHitIndex = i
            bestTrueHit = digit_truehits[bestTrueHitIndex]
 
            # Sensor identification
            sensorID = digit.getSensorID()
            self.data.sensor_id = int(sensorID)
            sensorNum = sensorID.getSensorNumber()
            self.data.sensor = sensorNum
            layerNum = sensorID.getLayerNumber()
            self.data.layer = layerNum
            if (layerNum == 3):
                sensorType = 1  # Barrel
            else:
                if (sensorNum == 1):
                    sensorType = 0
                else:
                    sensorType = 1
            self.data.sensor_type = sensorType
            ladderNum = sensorID.getLadderNumber()
            self.data.ladder = ladderNum
            if digit.isUStrip():
                self.data.strip_dir = 0
            else:
                self.data.strip_dir = 1
            self.data.recodigit_charge = digit.getCharge()
            self.data.recodigit_time = digit.getTime()
            self.data.truehit_time = bestTrueHit.getGlobalTime()
            # Fill tree
            self.file.cd()
            self.tree.Fill()
 

terminate()

terminate

(

self

)

Close the output file. 

Definition at line 118 of file SVDValidationTTreeRecoDigit.py.

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

Member Data Documentation

data

data = EventDataRecoDigit()

instance of event data class

Definition at line 57 of file SVDValidationTTreeRecoDigit.py.

file

file = ROOT.TFile('../SVDValidationTTreeRecoDigit.root', 'recreate')

output file

Definition at line 53 of file SVDValidationTTreeRecoDigit.py.

tree

tree = ROOT.TTree('tree', 'Event data of SVD validation events')

ttree

Definition at line 55 of file SVDValidationTTreeRecoDigit.py.

---

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

• svd/validation/SVDValidationTTreeRecoDigit.py