Inheritance diagram for ParticleStatisticsModule:

Public Member Functions
	__init__ (self)

	initialize (self)

	event (self)

	terminate (self)

Public Attributes
	tfile = ROOT.TFile("ParticleStatistics.root", "recreate")
	save statistics in here

Detailed Description

Collect statistics on particles - also parallel processable

Definition at line 58 of file register_pystorearray.py.

Constructor & Destructor Documentation

◆ init()

__init__ ( self )

set module flags

Definition at line 61 of file register_pystorearray.py.

    def __init__(self):
        """set module flags"""
        super().__init__()
        self.set_property_flags(
            basf2.ModulePropFlags.PARALLELPROCESSINGCERTIFIED |
            basf2.ModulePropFlags.TERMINATEINALLPROCESSES)
 

Member Function Documentation

◆ event()

event ( self )

actually collect info

Definition at line 98 of file register_pystorearray.py.

    def event(self):
        """actually collect info"""
        ntuple = Belle2.PyStoreObj("ParticleMomenta", Belle2.DataStore.c_Persistent)
        hist = Belle2.PyStoreObj("AbsMomentum", Belle2.DataStore.c_Persistent)
 
        mcParticles = Belle2.PyStoreArray(Belle2.MCParticle.Class())
 
        for mcParticle in mcParticles:
            momentum = mcParticle.getMomentum()
            ntuple.get().Fill(momentum.X(),
                              momentum.Y(),
                              momentum.Z())
 
            hist.get().Fill(momentum.Mag())
 

◆ initialize()

initialize ( self )

init

Definition at line 68 of file register_pystorearray.py.

    def initialize(self):
        """init"""
        
        self.tfile = ROOT.TFile("ParticleStatistics.root", "recreate")
 
        ntuple = Belle2.PyStoreObj(Belle2.RootMergeable("TNtuple").Class(), "ParticleMomenta", Belle2.DataStore.c_Persistent)
        ntuple.registerInDataStore()
        print("IsValid", ntuple.isValid())
        ntuple.create()
        print("IsValid", ntuple.isValid())
        ntuple.obj().assign(ROOT.TNtuple("Particles",  # ROOT name
                                         "Momentum components of the particles",  # ROOT title
                                         "px:py:pz"  # Var list
                                         ))
 
        hist = Belle2.PyStoreObj(Belle2.RootMergeable("TH1D").Class(), "AbsMomentum", Belle2.DataStore.c_Persistent)
        hist.registerInDataStore()
        print("IsValid", hist.isValid())
        hist.create()
        print("IsValid", hist.isValid())
        hist.obj().assign(ROOT.TH1D("AbsMomentum",  # ROOT name
                                    "Absolute momentum of particles",  # ROOT title
                                    20,  # n bins
                                    0,  # upper bound
                                    2  # lower bound
                                    ))
 
        print("IsValid", ntuple.isValid())
        print("IsValid", hist.isValid())
 

◆ terminate()

terminate ( self )

terminate

Definition at line 113 of file register_pystorearray.py.

    def terminate(self):
        """terminate"""
        if (not Belle2.ProcHandler.parallelProcessingUsed() or
                Belle2.ProcHandler.isOutputProcess()):
            print("Writing objects")
            ntuple = Belle2.PyStoreObj("ParticleMomenta", Belle2.DataStore.c_Persistent)
            ntuple.write(self.tfile)
 
            hist = Belle2.PyStoreObj("AbsMomentum", Belle2.DataStore.c_Persistent)
            hist.write(self.tfile)
 
            self.tfile.Close()
 
 

Member Data Documentation

◆ tfile

tfile = ROOT.TFile("ParticleStatistics.root", "recreate")

save statistics in here

Definition at line 71 of file register_pystorearray.py.

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

framework/examples/register_pystorearray.py

Public Member Functions

Public Attributes