Ntuple Class Reference

Inheritance diagram for Ntuple:

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

Public Attributes
	file = ROOT.TFile('cosmicT0FinderNtuple.root', 'recreate')
	root file
	tree = ROOT.TTree('tree', '')
	root tree
	data = TreeStruct()
	data structure to be written to tree
	reso = TH1F('reso', 'T0 residuals', 200, -1.0, 1.0)
	histogram of t0 residuals
	pull = TH1F('pull', 'T0 pulls', 200, -10.0, 10.0)
	histogram of t0 pulls

Detailed Description

 makes ntuple for MC studies of TOPCosmicT0Finder performance 

Definition at line 66 of file testCosmicT0Finder.py.

Member Function Documentation

event()

event

(

self

)

 event processing: fill ntuple and histograms 

Definition at line 92 of file testCosmicT0Finder.py.

    def event(self):
        ''' event processing: fill ntuple and histograms '''
 
        mcParticles = Belle2.PyStoreArray('MCParticles')
        timeZeros = Belle2.PyStoreArray('TOPTimeZeros')
        geo = Belle2.TOP.TOPGeometryPar.Instance().getGeometry()
        for timeZero in timeZeros:
            extHit = timeZero.getRelated('ExtHits')
            if not extHit:
                b2.B2WARNING('no related extHits')
                continue
            barHit = None
            for barhit in mcParticles[0].getRelationsWith('TOPBarHits'):
                if barhit.getModuleID() == extHit.getCopyID():
                    barHit = barhit
            if not barHit:
                b2.B2WARNING('no corresponding TOPBarHit')
                continue
 
            moduleID = timeZero.getModuleID()
            if extHit.getCopyID() != moduleID:
                b2.B2ERROR('moduleID in extHit differs: ' + str(extHit.getCopyID()) +
                           ' ' + str(moduleID))
            if barHit.getModuleID() != moduleID:
                b2.B2ERROR('moduleID in barHit differs: ' + str(barHit.getModuleID()) +
                           ' ' + str(moduleID))
 
            self.data.slot = moduleID
            self.data.nfot = timeZero.getNumPhotons()
            self.data.pdg = barHit.getPDG()
            pos = barHit.getLocalPosition()
            self.data.x = pos.X()
            self.data.y = pos.Y()
            self.data.z = pos.Z()
            self.data.t = barHit.getTime()
            module = geo.getModule(moduleID)
            mom = module.momentumToLocal(barHit.getMomentum())
            self.data.p = mom.R()
            self.data.theta = math.degrees(mom.Theta())
            self.data.phi = math.degrees(mom.Phi())
            dr = module.momentumToLocal(extHit.getPosition() - XYZVector(barHit.getPosition()))
            self.data.dx = dr.X()
            self.data.dy = dr.Y()
            self.data.dz = dr.Z()
            self.data.dt = extHit.getTOF() - self.data.t
            momExt = module.momentumToLocal(extHit.getMomentum())
            self.data.dp = momExt.R() - self.data.p
            self.data.dtheta = math.degrees(momExt.Theta()) - self.data.theta
            self.data.dphi = math.degrees(momExt.Phi()) - self.data.phi
            self.data.t0 = timeZero.getTime()
            self.data.t0err = timeZero.getError()
            self.data.timeShift = timeShift
            self.data.dt0 = self.data.dt + self.data.t0 - self.data.timeShift
            self.file.cd()
            self.tree.Fill()
            self.reso.Fill(self.data.dt0)
            self.pull.Fill(self.data.dt0/self.data.t0err)
            if timeZero.getChi2().GetEntries() > 0:
                timeZero.getChi2().Write()
                timeZero.getPDF().Write()
                timeZero.getHits().Write()
 

initialize()

initialize

(

self

)

 initialize 

Definition at line 69 of file testCosmicT0Finder.py.

    def initialize(self):
        ''' initialize '''
 
        
        self.file = ROOT.TFile('cosmicT0FinderNtuple.root', 'recreate')
        
        self.tree = ROOT.TTree('tree', '')
        
        self.data = TreeStruct()
 
        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)
 
        
        self.reso = TH1F('reso', 'T0 residuals', 200, -1.0, 1.0)
        self.reso.SetXTitle('#Delta t0 [ns]')
        
        self.pull = TH1F('pull', 'T0 pulls', 200, -10.0, 10.0)
 

terminate()

terminate

(

self

)

 terminate: write and close root file 

Definition at line 154 of file testCosmicT0Finder.py.

    def terminate(self):
        ''' terminate: write and close root file '''
 
        self.file.cd()
        self.file.Write()
        self.file.Close()
 
 
# Suppress messages and warnings during processing:

Member Data Documentation

data

data = TreeStruct()

data structure to be written to tree

Definition at line 77 of file testCosmicT0Finder.py.

file

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

root file

Definition at line 73 of file testCosmicT0Finder.py.

pull

pull = TH1F('pull', 'T0 pulls', 200, -10.0, 10.0)

histogram of t0 pulls

Definition at line 90 of file testCosmicT0Finder.py.

reso

reso = TH1F('reso', 'T0 residuals', 200, -1.0, 1.0)

histogram of t0 residuals

Definition at line 87 of file testCosmicT0Finder.py.

tree

tree = ROOT.TTree('tree', '')

root tree

Definition at line 75 of file testCosmicT0Finder.py.

---

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

• top/examples/testCosmicT0Finder.py