Inheritance diagram for WriteData:

Public Member Functions
	initialize (self)

	event (self)

	terminate (self)

Public Attributes
list	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()

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

initialize ( self )

Inits WriteData object.

Definition at line 60 of file genPixelLikelihoods.py.

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

◆ terminate()

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

Public Member Functions