WriteData Class Reference

Inheritance diagram for WriteData:

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

Public Attributes
	data
	A list containing arrays with each event's data.

Detailed Description

Data collector module. Gathers TOP Cherenkov photon data for each event.
On terminate(), writes gathered data to given output file.

Attributes:
    data: A list containing AwkwardArrays with each event's data.

Definition at line 51 of file genPixelLikelihoods.py.

Member Function Documentation

event()

def event

(

self

)

Collects event data about TOP Cherenkov photons.

Definition at line 65 of file genPixelLikelihoods.py.

    def event(self):
        """Collects event data about TOP Cherenkov photons."""
        mcps = Belle2.PyStoreArray("MCParticles")
        mcp = sorted(mcps, key=lambda x: x.getEnergy(), reverse=True)[0]
        mcp_pdg = abs(mcp.getPDG())
        track = mcp.getRelated('Tracks')
        if not track:
            print('no related track')
            return
        logl = track.getRelated('TOPLikelihoods')
        if not logl:
            print('no related loglihoods')
            return
        pixlogl = track.getRelated('TOPPixelLikelihoods')
        if not pixlogl:
            print('no related pixel loglihoods')
            return
        digits = Belle2.PyStoreArray("TOPDigits")
        if not digits:
            print('no digits')
            return
 
        moduleID = 0
        photon_detections = []
        for d in digits:
            if moduleID == 0:
                moduleID = d.getModuleID()
            simhit = d.getRelated("TOPSimHits")
            if not simhit:
                continue
            photon = simhit.getRelated("TOPSimPhotons")
            if not photon:
                continue
 
            chanID = d.getPixelID() - 1  # pixelID is 1-based
            time = photon.getDetectionTime()
            point = photon.getEmissionPoint()
            emitTime = photon.getEmissionTime()
            photon_detections.append([
                chanID, time,
                point.X(), point.Y(), point.Z(), emitTime
            ])
 
        self.data.append(
            awk.fromiter([
                np.array(pixlogl.getPixelLogL_e()),
                np.array(pixlogl.getPixelLogL_mu()),
                np.array(pixlogl.getPixelLogL_pi()),
                np.array(pixlogl.getPixelLogL_K()),
                np.array(pixlogl.getPixelLogL_p()),
                np.array(pixlogl.getPixelSigPhot_e()),
                np.array(pixlogl.getPixelSigPhot_mu()),
                np.array(pixlogl.getPixelSigPhot_pi()),
                np.array(pixlogl.getPixelSigPhot_K()),
                np.array(pixlogl.getPixelSigPhot_p()),
                logl.getLogL_e(),
                logl.getLogL_mu(),
                logl.getLogL_pi(),
                logl.getLogL_K(),
                logl.getLogL_p(),
                photon_detections,
                mcp_pdg
            ])
        )
 

initialize()

def initialize

(

self

)

Inits WriteData object.

Definition at line 60 of file genPixelLikelihoods.py.

    def initialize(self):
        """Inits WriteData object."""
        
        self.data = []
 

terminate()

def terminate

(

self

)

Writes data to given output file.

Definition at line 130 of file genPixelLikelihoods.py.

    def terminate(self):
        """Writes data to given output file."""
        awk.save(opts.output, self.data, mode="w")
 
 
# Suppress messages and warnings during processing:

Member Data Documentation

data

data

A list containing arrays with each event's data.

Definition at line 63 of file genPixelLikelihoods.py.

---

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

• top/examples/genPixelLikelihoods.py