CDCPlotModule Class Reference

Inheritance diagram for CDCPlotModule:

Collaboration diagram for CDCPlotModule:

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

Static Public Attributes
int	num_events = 0
	event counter

Detailed Description

An example python module.

It gathers the x/y position off all CDCSimHits and draws them using
matplotlib. The result is saved as a PNG.

Definition at line 61 of file cdcplotmodule.py.

Member Function Documentation

event()

def event

(

self

)

reimplementation of Module::event().

loops over the CDCSimHits in the current event.

Definition at line 71 of file cdcplotmodule.py.

    def event(self):
        """reimplementation of Module::event().
 
        loops over the CDCSimHits in the current event.
        """
        simhits = Belle2.PyStoreArray('CDCSimHits')
 
        # list of lists of simhit positions, one list per mcpart
        trackhits_x = []
        trackhits_y = []
 
        mcparts = []
        for hit in simhits:
            mcpart = hit.getRelatedFrom("MCParticles")
            if mcpart not in mcparts:
                mcparts.append(mcpart)
                trackhits_x.append([])
                trackhits_y.append([])
            # add simhit to the list corresponding to this particle
            idx = mcparts.index(mcpart)
            hitpos = hit.getPosWire()  # TVector3
            trackhits_x[idx].append(hitpos.X())
            trackhits_y[idx].append(hitpos.Y())
 
        npart = len(mcparts)
        if npart > 0:
            # plot the (x,y) list on a matplotlib figure
            col = [colormap.jet(1.0 * c / (npart - 1)) for c in range(npart)]
            fig = plot(trackhits_x, trackhits_y, col)
 
            filename = 'cdchits_%i.png' % (self.num_events)
            if os.path.lexists(filename):
                B2WARNING(filename + ' exists, overwriting ...')
            else:
                B2INFO('creating ' + filename + ' ...')
            fig.savefig(filename)
 
        self.num_events += 1
 

terminate()

def terminate

(

self

)

reimplementation of Module::terminate().

Definition at line 110 of file cdcplotmodule.py.

---

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

• framework/examples/cdcplotmodule.py