Inheritance diagram for PhokharaEvtgenAnalysisModule:

Public Member Functions
	__init__ (self)

	event (self)

	terminate (self)

Public Attributes
	output_file = ROOT.TFile('PhokharaEvtgenAnalysis.root', 'recreate')
	Output file.

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

	ecms = numpy.zeros(1, dtype=numpy.float32)
	Beam energy.

	gamma_e = numpy.zeros(1, dtype=numpy.float32)
	Virtual photon energy.

	gamma_px = numpy.zeros(1, dtype=numpy.float32)
	Virtual photon momentum (x component).

	gamma_py = numpy.zeros(1, dtype=numpy.float32)
	Virtual photon momentum (y component).

	gamma_pz = numpy.zeros(1, dtype=numpy.float32)
	Virtual photon momentum (z component).

	jpsi_e = numpy.zeros(1, dtype=numpy.float32)
	J/psi energy.

	jpsi_px = numpy.zeros(1, dtype=numpy.float32)
	J/psi momentum (x component).

	jpsi_py = numpy.zeros(1, dtype=numpy.float32)
	J/psi momentum (y component).

	jpsi_pz = numpy.zeros(1, dtype=numpy.float32)
	J/psi momentum (z component).

	lepton_e = numpy.zeros(1, dtype=numpy.float32)
	Lepton energy.

	lepton_px = numpy.zeros(1, dtype=numpy.float32)
	Lepton momentum (x component).

	lepton_py = numpy.zeros(1, dtype=numpy.float32)
	Lepton momentum (y component).

	lepton_pz = numpy.zeros(1, dtype=numpy.float32)
	Lepton momentum (z component).

Detailed Description

 Analysis module for PhokharaEvtgen.

Definition at line 26 of file PhokharaEvtgenAnalyze.py.

Constructor & Destructor Documentation

◆ init()

__init__ ( self )

Initialization.

Definition at line 29 of file PhokharaEvtgenAnalyze.py.

    def __init__(self):
        """Initialization."""
        super().__init__()
        
        self.output_file = ROOT.TFile('PhokharaEvtgenAnalysis.root', 'recreate')
        
        self.tree = ROOT.TTree('tree', '')
        
        self.ecms = numpy.zeros(1, dtype=numpy.float32)
        
        self.gamma_e = numpy.zeros(1, dtype=numpy.float32)
        
        self.gamma_px = numpy.zeros(1, dtype=numpy.float32)
        
        self.gamma_py = numpy.zeros(1, dtype=numpy.float32)
        
        self.gamma_pz = numpy.zeros(1, dtype=numpy.float32)
        
        self.jpsi_e = numpy.zeros(1, dtype=numpy.float32)
        
        self.jpsi_px = numpy.zeros(1, dtype=numpy.float32)
        
        self.jpsi_py = numpy.zeros(1, dtype=numpy.float32)
        
        self.jpsi_pz = numpy.zeros(1, dtype=numpy.float32)
        
        self.lepton_e = numpy.zeros(1, dtype=numpy.float32)
        
        self.lepton_px = numpy.zeros(1, dtype=numpy.float32)
        
        self.lepton_py = numpy.zeros(1, dtype=numpy.float32)
        
        self.lepton_pz = numpy.zeros(1, dtype=numpy.float32)
        self.tree.Branch('ecms', self.ecms, 'ecms/F')
        self.tree.Branch('gamma_e', self.gamma_e, 'gamma_e/F')
        self.tree.Branch('gamma_px', self.gamma_px, 'gamma_px/F')
        self.tree.Branch('gamma_py', self.gamma_py, 'gamma_py/F')
        self.tree.Branch('gamma_pz', self.gamma_pz, 'gamma_pz/F')
        self.tree.Branch('jpsi_e', self.jpsi_e, 'jpsi_e/F')
        self.tree.Branch('jpsi_px', self.jpsi_px, 'jpsi_px/F')
        self.tree.Branch('jpsi_py', self.jpsi_py, 'jpsi_py/F')
        self.tree.Branch('jpsi_pz', self.jpsi_pz, 'jpsi_pz/F')
        self.tree.Branch('lepton_e', self.lepton_e, 'lepton_e/F')
        self.tree.Branch('lepton_px', self.lepton_px, 'lepton_px/F')
        self.tree.Branch('lepton_py', self.lepton_py, 'lepton_py/F')
        self.tree.Branch('lepton_pz', self.lepton_pz, 'lepton_pz/F')
 

Member Function Documentation

◆ event()

event ( self )

 Event function.

Definition at line 76 of file PhokharaEvtgenAnalyze.py.

    def event(self):
        """ Event function. """
        mc_initial_particles = Belle2.PyStoreObj('MCInitialParticles')
        self.ecms[0] = mc_initial_particles.getMass()
        mc_particles = Belle2.PyStoreArray('MCParticles')
        for mc_particle in mc_particles:
            # Select virtual photons.
            if (mc_particle.getPDG() != 10022):
                continue
            p = mc_particle.getMomentum()
            self.gamma_e[0] = mc_particle.getEnergy()
            self.gamma_px[0] = p.X()
            self.gamma_py[0] = p.Y()
            self.gamma_pz[0] = p.Z()
            jpsi = mc_particle.getDaughters()[0]
            p = jpsi.getMomentum()
            self.jpsi_e[0] = jpsi.getEnergy()
            self.jpsi_px[0] = p.X()
            self.jpsi_py[0] = p.Y()
            self.jpsi_pz[0] = p.Z()
            lepton = jpsi.getDaughters()[0]
            p = lepton.getMomentum()
            self.lepton_e[0] = lepton.getEnergy()
            self.lepton_px[0] = p.X()
            self.lepton_py[0] = p.Y()
            self.lepton_pz[0] = p.Z()
            self.tree.Fill()
 

◆ terminate()

terminate ( self )

 Termination function.

Definition at line 104 of file PhokharaEvtgenAnalyze.py.

    def terminate(self):
        """ Termination function. """
        self.output_file.cd()
        self.tree.Write()
        self.output_file.Close()
 
 
# Input.

Member Data Documentation

◆ ecms

ecms = numpy.zeros(1, dtype=numpy.float32)

Beam energy.

Definition at line 37 of file PhokharaEvtgenAnalyze.py.

◆ gamma_e

gamma_e = numpy.zeros(1, dtype=numpy.float32)

Virtual photon energy.

Definition at line 39 of file PhokharaEvtgenAnalyze.py.

◆ gamma_px

gamma_px = numpy.zeros(1, dtype=numpy.float32)

Virtual photon momentum (x component).

Definition at line 41 of file PhokharaEvtgenAnalyze.py.

◆ gamma_py

gamma_py = numpy.zeros(1, dtype=numpy.float32)

Virtual photon momentum (y component).

Definition at line 43 of file PhokharaEvtgenAnalyze.py.

◆ gamma_pz

gamma_pz = numpy.zeros(1, dtype=numpy.float32)

Virtual photon momentum (z component).

Definition at line 45 of file PhokharaEvtgenAnalyze.py.

◆ jpsi_e

jpsi_e = numpy.zeros(1, dtype=numpy.float32)

J/psi energy.

Definition at line 47 of file PhokharaEvtgenAnalyze.py.

◆ jpsi_px

jpsi_px = numpy.zeros(1, dtype=numpy.float32)

J/psi momentum (x component).

Definition at line 49 of file PhokharaEvtgenAnalyze.py.

◆ jpsi_py

jpsi_py = numpy.zeros(1, dtype=numpy.float32)

J/psi momentum (y component).

Definition at line 51 of file PhokharaEvtgenAnalyze.py.

◆ jpsi_pz

jpsi_pz = numpy.zeros(1, dtype=numpy.float32)

J/psi momentum (z component).

Definition at line 53 of file PhokharaEvtgenAnalyze.py.

◆ lepton_e

lepton_e = numpy.zeros(1, dtype=numpy.float32)

Lepton energy.

Definition at line 55 of file PhokharaEvtgenAnalyze.py.

◆ lepton_px

lepton_px = numpy.zeros(1, dtype=numpy.float32)

Lepton momentum (x component).

Definition at line 57 of file PhokharaEvtgenAnalyze.py.

◆ lepton_py

lepton_py = numpy.zeros(1, dtype=numpy.float32)

Lepton momentum (y component).

Definition at line 59 of file PhokharaEvtgenAnalyze.py.

◆ lepton_pz

lepton_pz = numpy.zeros(1, dtype=numpy.float32)

Lepton momentum (z component).

Definition at line 61 of file PhokharaEvtgenAnalyze.py.

◆ output_file

output_file = ROOT.TFile('PhokharaEvtgenAnalysis.root', 'recreate')

Output file.

Definition at line 33 of file PhokharaEvtgenAnalyze.py.

◆ tree

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

Output tree.

Definition at line 35 of file PhokharaEvtgenAnalyze.py.

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

generators/validation/PhokharaEvtgenAnalyze.py

Public Member Functions

Public Attributes