Ntuple Class Reference

Inheritance diagram for Ntuple:

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

Public Attributes
	file
	root file
	tree
	root tree
	data
	data structure to be written to tree
	reso
	histogram of t0 residuals
	pull
	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()

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

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

def 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

data structure to be written to tree

Definition at line 77 of file testCosmicT0Finder.py.

file

file

root file

Definition at line 73 of file testCosmicT0Finder.py.

pull

pull

histogram of t0 pulls

Definition at line 90 of file testCosmicT0Finder.py.

reso

reso

histogram of t0 residuals

Definition at line 87 of file testCosmicT0Finder.py.

tree

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