Inheritance diagram for SegmentQuadTreePlotter:

Public Member Functions
def	calculateIntersectionInQuadTreePicture (self, first, second)

def	calculatePositionInQuadTreePicture (self, position)

def	forAllAxialSegments (self, f)

def	convertToQuadTreePicture (self, phi, mag, charge)

def	event (self)

Public Attributes
	range_x_min
	lower x bound for polar angle

	range_x_max
	upper x bound for polar angle

	range_y_min
	lower y bound

	range_y_max
	upper y bound

Static Public Attributes
bool	draw_segment_intersection = True and False
	by default, do not draw a segment intersection

bool	draw_segment = True and False
	by default, do not draw a segment

bool	draw_segment_averaged = True and False
	by default, do not draw an averaged segment

bool	draw_segment_fitted = True and False
	by default, do not draw a fitted segment

bool	draw_mc_information = True and False
	by default, do not draw the MC information

bool	draw_mc_hits = True and False
	by default, do not draw the MC hits

bool	theta_shifted = False
	by default, polar angles and cuts are in the range (0,pi) rather than (-pi/2,+pi/2)

np	maximum_theta = np.pi
	an alias for the maximum value of the polar angle

Detailed Description

Implementation of a quad tree plotter for SegmentQuadTrees

Definition at line 125 of file quadTreePlotter.py.

Member Function Documentation

◆ calculateIntersectionInQuadTreePicture()

def calculateIntersectionInQuadTreePicture	(	self,
		first,
		second
	)

Calculate the point where the two given hits intersect

params
------
first: hit
second: hit

Definition at line 149 of file quadTreePlotter.py.

    def calculateIntersectionInQuadTreePicture(self, first, second):
        """
        Calculate the point where the two given hits intersect
 
        params
        ------
        first: hit
        second: hit
        """
        positionFront = first.getRecoPos2D().conformalTransformed()
        positionBack = second.getRecoPos2D().conformalTransformed()
 
        theta_cut = np.arctan2((positionBack - positionFront).x(), (positionFront - positionBack).y())
 
        if self.theta_shifted:
            while theta_cut < - self.maximum_theta / 2:
                theta_cut += self.maximum_theta
        else:
            while theta_cut < 0:
                theta_cut += self.maximum_theta
 
        r_cut = positionFront.x() * np.cos(theta_cut) + positionFront.y() * np.sin(theta_cut)
 
        return theta_cut, r_cut
 

◆ calculatePositionInQuadTreePicture()

def calculatePositionInQuadTreePicture	(	self,
		position
	)

Transform a given normal coordinate position to a legendre position (conformal transformed)

params
------
position: TrackFindingCDC.Vector2D

Definition at line 174 of file quadTreePlotter.py.

    def calculatePositionInQuadTreePicture(self, position):
        """
        Transform a given normal coordinate position to a legendre position (conformal transformed)
 
        params
        ------
        position: TrackFindingCDC.Vector2D
        """
        position = position.conformalTransformed()
 
        theta = np.linspace(self.range_x_min, self.range_x_max, 100)
        r = position.x() * np.cos(theta) + position.y() * np.sin(theta)
 
        return theta, r
 

◆ convertToQuadTreePicture()

def convertToQuadTreePicture	(	self,
		phi,
		mag,
		charge
	)

Convert given track parameters into a point in the legendre space

params
------
phi: phi of the track
mag: magnitude of pt
charge: charge of the track

Definition at line 205 of file quadTreePlotter.py.

    def convertToQuadTreePicture(self, phi, mag, charge):
        """
        Convert given track parameters into a point in the legendre space
 
        params
        ------
        phi: phi of the track
        mag: magnitude of pt
        charge: charge of the track
        """
        theta = phi + np.pi / 2
        r = 1 / mag * 1.5 * 0.00299792458 * charge
        if self.theta_shifted:
            if theta > self.maximum_theta / 2 or theta < -self.maximum_theta / 2:
                theta = theta % self.maximum_theta - self.maximum_theta / 2
            else:
                r *= -1
        else:
            if theta > self.maximum_theta or theta < 0:
                theta = theta % self.maximum_theta
            else:
                r *= -1
        return theta, r
 

◆ event()

def event ( self )

Draw everything according to the given options

Attributes
----------
draw_segment_intersection
draw_segment
draw_segment_averaged
draw_segment_fitted
draw_mc_information
draw_mc_hits

Reimplemented from QuadTreePlotter.

Definition at line 229 of file quadTreePlotter.py.

    def event(self):
        """
        Draw everything according to the given options
 
        Attributes
        ----------
        draw_segment_intersection
        draw_segment
        draw_segment_averaged
        draw_segment_fitted
        draw_mc_information
        draw_mc_hits
        """
        if self.theta_shifted:
            
            self.range_x_min = -self.maximum_theta / 2
            
            self.range_x_max = self.maximum_theta / 2
        else:
            self.range_x_min = 0
            self.range_x_max = self.maximum_theta
 
        
        self.range_y_min = -0.08
        
        self.range_y_max = 0.08
 
        self.init_plotting()
        # self.plotQuadTreeContent()
 
        if self.draw_segment_intersection:
            map = attributemaps.SegmentMCTrackIdColorMap()
 
            def f(segment):
                theta, r = self.calculateIntersectionInQuadTreePicture(segment.front(), segment.back())
                plt.plot(theta, r, color=list(map(0, segment)), marker="o")
 
            self.forAllAxialSegments(f)
 
        if self.draw_segment:
            map = attributemaps.SegmentMCTrackIdColorMap()
 
            def f(segment):
                theta, r = self.calculatePositionInQuadTreePicture(segment.front().getRecoPos2D())
                plt.plot(theta, r, color=list(map(0, segment)), marker="", ls="-")
 
            self.forAllAxialSegments(f)
 
        if self.draw_segment_averaged:
            map = attributemaps.SegmentMCTrackIdColorMap()
 
            def f(segment):
                middle_index = int(np.round(segment.size() / 2.0))
                middle_point = list(segment.items())[middle_index]
                theta_front, r_front = self.calculateIntersectionInQuadTreePicture(segment.front(), middle_point)
                theta_back, r_back = self.calculateIntersectionInQuadTreePicture(middle_point, segment.back())
 
                plt.plot([theta_front, theta_back], [r_front, r_back], color=list(map(0, segment)), marker="o", ls="-")
 
            self.forAllAxialSegments(f)
 
        if self.draw_segment_fitted:
            map = attributemaps.SegmentMCTrackIdColorMap()
            fitter = Belle2.TrackFindingCDC.CDCRiemannFitter.getOriginCircleFitter()
 
            def f(segment):
                trajectory = fitter.fit(segment)
                momentum = trajectory.getUnitMom2D(Belle2.TrackFindingCDC.Vector2D(0, 0)).scale(trajectory.getAbsMom2D())
                theta, r = self.convertToQuadTreePicture(momentum.phi(), momentum.norm(), trajectory.getChargeSign())
                plt.plot(theta, r, color=list(map(0, segment)), marker="o")
 
            self.forAllAxialSegments(f)
 
        if self.draw_hits:
            cdcHits = Belle2.PyStoreArray("CDCHits")
            storedWireHits = Belle2.PyStoreObj('CDCWireHitVector')
            wireHits = storedWireHits.obj().get()
 
            array = Belle2.PyStoreArray("MCTrackCands")
            cdc_hits = [cdcHits[i] for track in array for i in track.getHitIDs()]
 
            for cdcHit in cdcHits:
                if cdcHit in cdc_hits:
                    continue
                wireHit = wireHits.at(bisect.bisect_left(wireHits, cdcHit))
                theta, r = self.calculatePositionInQuadTreePicture(wireHit.getRefPos2D())
 
                plt.plot(theta, r, marker="", color="black", ls="-", alpha=0.8)
 
        if self.draw_mc_hits:
            storedWireHits = Belle2.PyStoreObj('CDCWireHitVector')
            wireHits = storedWireHits.obj().get()
 
            map = attributemaps.listColors
            array = Belle2.PyStoreArray("MCTrackCands")
            cdcHits = Belle2.PyStoreArray("CDCHits")
 
            for track in array:
                mcTrackID = track.getMcTrackId()
 
                for cdcHitID in track.getHitIDs(Belle2.Const.CDC):
                    cdcHit = cdcHits[cdcHitID]
                    wireHit = wireHits.at(bisect.bisect_left(wireHits, cdcHit))
 
                    theta, r = self.calculatePositionInQuadTreePicture(wireHit.getRefPos2D())
 
                    plt.plot(theta, r, marker="", color=map[mcTrackID % len(map)], ls="-", alpha=0.2)
 
        if self.draw_mc_information:
            map = attributemaps.listColors
            array = Belle2.PyStoreArray("MCTrackCands")
 
            for track in array:
                momentum = track.getMomSeed()
 
                # HARDCODED!!! Temporary solution
                theta, r = self.convertToQuadTreePicture(momentum.Phi(), momentum.Mag(), track.getChargeSeed())
                mcTrackID = track.getMcTrackId()
 
                plt.plot(theta, r, marker="o", color="black", ms=10)
                plt.plot(theta, r, marker="o", color=map[mcTrackID % len(map)], ms=5)
 
        self.save_and_show_file()
 
 

◆ forAllAxialSegments()

def forAllAxialSegments	(	self,
		f
	)

Loop over all segments and execute a function

params
------
f: function

Definition at line 189 of file quadTreePlotter.py.

    def forAllAxialSegments(self, f):
        """
        Loop over all segments and execute a function
 
        params
        ------
        f: function
        """
        items = Belle2.PyStoreObj("CDCSegment2DVector")
        wrapped_vector = items.obj()
        vector = wrapped_vector.get()
 
        for quad_tree_item in vector:
            if quad_tree_item.getStereoType() == 0:
                f(quad_tree_item)
 

Member Data Documentation

◆ draw_mc_hits

bool draw_mc_hits = True and False

static

by default, do not draw the MC hits

Definition at line 142 of file quadTreePlotter.py.

◆ draw_mc_information

bool draw_mc_information = True and False

static

by default, do not draw the MC information

Definition at line 140 of file quadTreePlotter.py.

◆ draw_segment

bool draw_segment = True and False

static

by default, do not draw a segment

Definition at line 134 of file quadTreePlotter.py.

◆ draw_segment_averaged

bool draw_segment_averaged = True and False

static

by default, do not draw an averaged segment

Definition at line 136 of file quadTreePlotter.py.

◆ draw_segment_fitted

bool draw_segment_fitted = True and False

static

by default, do not draw a fitted segment

Definition at line 138 of file quadTreePlotter.py.

◆ draw_segment_intersection

bool draw_segment_intersection = True and False

static

by default, do not draw a segment intersection

Definition at line 132 of file quadTreePlotter.py.

◆ maximum_theta

np maximum_theta = np.pi

static

an alias for the maximum value of the polar angle

Definition at line 147 of file quadTreePlotter.py.

◆ range_x_max

range_x_max

upper x bound for polar angle

Definition at line 246 of file quadTreePlotter.py.

◆ range_x_min

range_x_min

lower x bound for polar angle

Definition at line 244 of file quadTreePlotter.py.

◆ range_y_max

range_y_max

upper y bound

Definition at line 254 of file quadTreePlotter.py.

◆ range_y_min

range_y_min

lower y bound

Definition at line 252 of file quadTreePlotter.py.

◆ theta_shifted

bool theta_shifted = False

static

by default, polar angles and cuts are in the range (0,pi) rather than (-pi/2,+pi/2)

Definition at line 145 of file quadTreePlotter.py.

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

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

Public Member Functions

Public Attributes

Static Public Attributes

Detailed Description

Member Function Documentation

◆ calculateIntersectionInQuadTreePicture()

◆ calculatePositionInQuadTreePicture()

◆ convertToQuadTreePicture()

◆ event()

◆ forAllAxialSegments()

Member Data Documentation

◆ draw_mc_hits

◆ draw_mc_information

◆ draw_segment

◆ draw_segment_averaged

◆ draw_segment_fitted

◆ draw_segment_intersection

◆ maximum_theta

◆ range_x_max

◆ range_x_min

◆ range_y_max

◆ range_y_min

◆ theta_shifted