SVDValidationTTreeRecoDigit Class Reference

Inheritance diagram for SVDValidationTTreeRecoDigit:

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

Public Attributes
	file
	output file
	tree
	ttree
	data
	instance of event data class

Detailed Description

class to create reco digitis ttree

Definition at line 45 of file SVDValidationTTreeRecoDigit.py.

Constructor & Destructor Documentation

__init__()

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

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

def 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

instance of event data class

Definition at line 57 of file SVDValidationTTreeRecoDigit.py.

file

file

output file

Definition at line 53 of file SVDValidationTTreeRecoDigit.py.

tree

tree

ttree

Definition at line 55 of file SVDValidationTTreeRecoDigit.py.

---

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

• svd/validation/SVDValidationTTreeRecoDigit.py