StereoQuadTreePlotter Class Reference

Inheritance diagram for StereoQuadTreePlotter:

Public Member Functions
def	create_trajectory_from_track (self, track)
def	create_reco_hit3D (self, cdcHit, trajectory3D, rlInfo)
def	get_plottable_line (self, recoHit3D)
def	plot_hit_line (self, recoHit3D, color)
def	event (self)

Public Attributes
	draw_quad_tree_content
	by default, draw the QuadTree
	range_x_min
	default lower x bound
	range_x_max
	default upper x bound
	range_y_min
	default lower y bound
	range_y_max
	default upper y bound

Static Public Attributes
bool	draw_mc_hits = False
	by default, do not draw the MC hits
bool	draw_mc_tracks = False
	by default, do not draw the MC tracks
bool	draw_track_hits = False
	by default, do not draw the track hits
bool	draw_last_track = True
	by default, draw the last track
bool	delete_bad_hits = False
	by default, do not delete the bad track hits

Detailed Description

Implementation of a quad tree plotter for StereoHitAssignment

Definition at line 354 of file quadTreePlotter.py.

Member Function Documentation

create_reco_hit3D()

def create_reco_hit3D	(		self,
			cdcHit,
			trajectory3D,
			rlInfo
	)

Use a cdc hit and a trajectory to reconstruct a CDCRecoHit3D

params
------
cdcHit: CDCHit
trajectory3D: TrackFindingCDC.CDCTrajectory3D
rlInfo: RightLeftInfo ( = short)

Definition at line 388 of file quadTreePlotter.py.

    def create_reco_hit3D(self, cdcHit, trajectory3D, rlInfo):
        """
        Use a cdc hit and a trajectory to reconstruct a CDCRecoHit3D
 
        params
        ------
        cdcHit: CDCHit
        trajectory3D: TrackFindingCDC.CDCTrajectory3D
        rlInfo: RightLeftInfo ( = short)
        """
        storedWireHits = Belle2.PyStoreObj('CDCWireHitVector')
        wireHits = storedWireHits.obj().get()
 
        CDCRecoHit3D = Belle2.TrackFindingCDC.CDCRecoHit3D
        wireHit = wireHits.at(bisect.bisect_left(wireHits, cdcHit))
        rightLeftWireHit = Belle2.TrackFindingCDC.CDCRLWireHit(wireHit, rlInfo)
        if rightLeftWireHit.getStereoType() != 0:
            recoHit3D = CDCRecoHit3D.reconstruct(rightLeftWireHit, trajectory3D.getTrajectory2D())
            return recoHit3D
        else:
            return None
 

create_trajectory_from_track()

def create_trajectory_from_track	(		self,
			track
	)

Convert a genfit::TrackCand into a TrackFindingCDC.CDCTrajectory3D

params
------
track: genfit::TrackCand

Definition at line 371 of file quadTreePlotter.py.

    def create_trajectory_from_track(self, track):
        """
        Convert a genfit::TrackCand into a TrackFindingCDC.CDCTrajectory3D
 
        params
        ------
        track: genfit::TrackCand
        """
        Vector3D = Belle2.TrackFindingCDC.Vector3D
        Trajectory3D = Belle2.TrackFindingCDC.CDCTrajectory3D
 
        position = Vector3D(track.getPosSeed())
        momentum = Vector3D(track.getMomSeed())
        charge = track.getChargeSeed()
 
        return Trajectory3D(position, momentum, charge)
 

event()

def event

(

self

)

Draw the hit content according to the attributes

Attributes
----------
draw_mc_hits
draw_mc_tracks
draw_track_hits
draw_last_track
delete_bad_hits

Reimplemented from QuadTreePlotter.

Definition at line 430 of file quadTreePlotter.py.

    def event(self):
        """
        Draw the hit content according to the attributes
 
        Attributes
        ----------
        draw_mc_hits
        draw_mc_tracks
        draw_track_hits
        draw_last_track
        delete_bad_hits
        """
        
        self.draw_quad_tree_content = True
 
        
        self.range_x_min = -2 - np.sqrt(3)
        
        self.range_x_max = 2 + np.sqrt(3)
 
        
        self.range_y_min = -100
        
        self.range_y_max = -self.range_y_min
 
        self.init_plotting()
        self.plot_quad_tree_content()
 
        map = attributemaps.listColors
        cdcHits = Belle2.PyStoreArray("CDCHits")
 
        items = Belle2.PyStoreObj("CDCTrackVector")
        wrapped_vector = items.obj()
        track_vector = wrapped_vector.get()
 
        mcHitLookUp = Belle2.TrackFindingCDC.CDCMCHitLookUp().getInstance()
        mcHitLookUp.fill()
 
        storedWireHits = Belle2.PyStoreObj('CDCWireHitVector')
        wireHits = storedWireHits.obj().get()
 
        if self.draw_mc_hits:
            mc_track_cands = Belle2.PyStoreArray("MCTrackCands")
 
            for track in mc_track_cands:
                mcTrackID = track.getMcTrackId()
                trajectory = self.create_trajectory_from_track(track)
 
                for cdcHitID in track.getHitIDs(Belle2.Const.CDC):
                    cdcHit = cdcHits[cdcHitID]
 
                    leftRecoHit3D = self.create_reco_hit3D(cdcHit, trajectory, -1)
                    rightRecoHit3D = self.create_reco_hit3D(cdcHit, trajectory, 1)
 
                    self.plot_hit_line(leftRecoHit3D, color=map[mcTrackID % len(map)])
                    self.plot_hit_line(rightRecoHit3D, color=map[mcTrackID % len(map)])
 
        if self.draw_mc_tracks:
            mc_track_cands = Belle2.PyStoreArray("MCTrackCands")
 
            for track in mc_track_cands:
                mcTrackID = track.getMcTrackId()
                trajectory = self.create_trajectory_from_track(track)
                z0 = trajectory.getTrajectorySZ().getZ0()
 
                for cdcHitID in track.getHitIDs(Belle2.Const.CDC):
                    cdcHit = cdcHits[cdcHitID]
                    recoHit3D = self.create_reco_hit3D(cdcHit, trajectory, mcHitLookUp.getRLInfo(cdcHit))
 
                    if recoHit3D:
                        arr_l = (recoHit3D.getRecoPos3D().z() - z0) / recoHit3D.getArcLength2D()
                        plt.plot(arr_l, z0, marker="o", color=map[mcTrackID % len(map)], ls="", alpha=0.2)
 
        if self.draw_track_hits:
            for id, track in enumerate(track_vector):
                for recoHit3D in list(track.items()):
                    self.plot_hit_line(recoHit3D, color=map[id % len(map)])
 
        if self.draw_last_track and len(track_vector) != 0:
 
            last_track = track_vector[-1]
            trajectory = last_track.getStartTrajectory3D().getTrajectory2D()
 
            for wireHit in wireHits:
                for rlInfo in (-1, 1):
                    recoHit3D = Belle2.TrackFindingCDC.CDCRecoHit3D.reconstruct(wireHit, rlInfo, trajectory)
 
                    if (self.delete_bad_hits and
                        (wireHit.getRLInfo() != mcHitLookUp.getRLInfo(wireHit.getWireHit().getHit()) or
                         not recoHit3D.isInCellZBounds())):
                        continue
 
                    if recoHit3D in list(last_track.items()):
                        color = map[len(track_vector) % len(map)]
                    else:
                        if wireHit.getRLInfo() == 1:
                            color = "black"
                        else:
                            color = "gray"
                    self.plot_hit_line(recoHit3D, color)
 
        plt.xlabel(r"$\tan \ \lambda$")
        plt.ylabel(r"$z_0$")
        self.save_and_show_file()
 
 

get_plottable_line()

def get_plottable_line	(		self,
			recoHit3D
	)

Minim the task of the StereoQuadTree by showing the line of quadtree nodes
a hit belongs to

Definition at line 410 of file quadTreePlotter.py.

    def get_plottable_line(self, recoHit3D):
        """
        Minim the task of the StereoQuadTree by showing the line of quadtree nodes
        a hit belongs to
        """
        z0 = [self.range_y_min, self.range_y_max]
        arr_l = np.array((np.array(recoHit3D.getRecoPos3D().z()) - z0) / recoHit3D.getArcLength2D())
        return arr_l, z0
 

plot_hit_line()

def plot_hit_line	(		self,
			recoHit3D,
			color
	)

Draw one recoHit3D

Definition at line 419 of file quadTreePlotter.py.

    def plot_hit_line(self, recoHit3D, color):
        """
        Draw one recoHit3D
        """
        if recoHit3D:
            if recoHit3D.getStereoType() == 0:
                return
 
            arr_l, z0 = self.get_plottable_line(recoHit3D)
            plt.plot(arr_l, z0, marker="", ls="-", alpha=0.4, color=color)
 

Member Data Documentation

delete_bad_hits

bool delete_bad_hits = False

static

by default, do not delete the bad track hits

Definition at line 369 of file quadTreePlotter.py.

draw_last_track

bool draw_last_track = True

static

by default, draw the last track

Definition at line 367 of file quadTreePlotter.py.

draw_mc_hits

bool draw_mc_hits = False

static

by default, do not draw the MC hits

Definition at line 361 of file quadTreePlotter.py.

draw_mc_tracks

bool draw_mc_tracks = False

static

by default, do not draw the MC tracks

Definition at line 363 of file quadTreePlotter.py.

draw_quad_tree_content

draw_quad_tree_content

by default, draw the QuadTree

Definition at line 443 of file quadTreePlotter.py.

draw_track_hits

bool draw_track_hits = False

static

by default, do not draw the track hits

Definition at line 365 of file quadTreePlotter.py.

range_x_max

range_x_max

default upper x bound

Definition at line 448 of file quadTreePlotter.py.

range_x_min

range_x_min

default lower x bound

Definition at line 446 of file quadTreePlotter.py.

range_y_max

range_y_max

default upper y bound

Definition at line 453 of file quadTreePlotter.py.

range_y_min

range_y_min

default lower y bound

Definition at line 451 of file quadTreePlotter.py.

---

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

• tracking/trackFindingCDC/scripts/trackfindingcdc/quadtree/quadTreePlotter.py