drawFromPickle.PadWithHits Class Reference

Inheritance diagram for drawFromPickle.PadWithHits:

Collaboration diagram for drawFromPickle.PadWithHits:

Public Member Functions
def	__init__ (self, **kargs)
def	setFont (self, font)
def	isPivot (self)
def	rotateVecToSec5 (self, vec)
def	determineAvgChargeXyz (self)
def	determineXminmax (self)
def	determineYminmax (self)
def	determineZminmax (self)
def	determinePhiminmax (self)
def	determineHminmax (self)
def	draw (self, view='yx')
def	drawEdges (self, color, width, view='yx')
def	printPadCoord (self, view, line)
def	drawCharge (self, color, view='yx')

Public Attributes
	cornerXyz
	avgChargeXyz
	rotatedCornerXyz
	rotatedAvgChargeXyz
	maxX
	maxY
	maxZ
	maxH
	maxPhi
	lineColor
	lineWidth
	font

Private Attributes
	_lines
	_labels
	_markers

Detailed Description

Definition at line 55 of file drawFromPickle.py.

Constructor & Destructor Documentation

__init__()

def drawFromPickle.PadWithHits.__init__	(		self,
		**	kargs
	)

Reimplemented from drawFromPickle.BaseClassFromDict.

Definition at line 56 of file drawFromPickle.py.

    def __init__(self, **kargs) :
        super(PadWithHits, self).__init__(**kargs)
        self.cornerXyz    = [tuple2vec(t) for t in self.cornerXyz] # convert (x,y,z) tuple to vector
        self.avgChargeXyz = self.determineAvgChargeXyz()
        self.rotatedCornerXyz    = [self.rotateVecToSec5(v) for v in self.cornerXyz]
        self.rotatedAvgChargeXyz = self.rotateVecToSec5(self.avgChargeXyz)
        self.minX,   self.maxX   = self.determineXminmax()
        self.minY,   self.maxY   = self.determineYminmax()
        self.minZ,   self.maxZ   = self.determineZminmax()
        self.minH,   self.maxH   = self.determineHminmax()
        self.minPhi, self.maxPhi = self.determinePhiminmax()
        def color(layer) : return [r.kBlack, r.kRed, r.kBlue, r.kViolet, r.kGreen][layer]
        def width(isPivot) : return 2 if isPivot else 1
        self.lineColor = color(self.layer)
        self.lineWidth = width(self.isPivot())

Member Function Documentation

determineAvgChargeXyz()

def drawFromPickle.PadWithHits.determineAvgChargeXyz

(

self

)

Definition at line 77 of file drawFromPickle.py.

    def determineAvgChargeXyz(self) :
        z = average([p.z() for p in self.cornerXyz]) # although the corners should be at the same z...
        x, y = xyFromEtaPhiZ(self.avgEta, self.avgPhi, z)
        return xyzv(x, y, z)

determineHminmax()

def drawFromPickle.PadWithHits.determineHminmax

(

self

)

Definition at line 93 of file drawFromPickle.py.

    def determineHminmax(self): return self.loHei, self.hiHei

determinePhiminmax()

def drawFromPickle.PadWithHits.determinePhiminmax

(

self

)

Definition at line 90 of file drawFromPickle.py.

    def determinePhiminmax(self):
        phis = [p.phi() for p in self.cornerXyz]
        return min(phis), max(phis)

determineXminmax()

def drawFromPickle.PadWithHits.determineXminmax

(

self

)

Definition at line 81 of file drawFromPickle.py.

    def determineXminmax(self):
        xs = [p.x() for p in self.cornerXyz]
        return min(xs), max(xs)

determineYminmax()

def drawFromPickle.PadWithHits.determineYminmax

(

self

)

Definition at line 84 of file drawFromPickle.py.

    def determineYminmax(self):
        ys = [p.y() for p in self.cornerXyz]
        return min(ys), max(ys)

determineZminmax()

def drawFromPickle.PadWithHits.determineZminmax

(

self

)

Definition at line 87 of file drawFromPickle.py.

    def determineZminmax(self):
        zs = [p.z() for p in self.cornerXyz]
        return min(zs), max(zs)

draw()

def drawFromPickle.PadWithHits.draw	(		self,
			view = 'yx'
	)

Definition at line 94 of file drawFromPickle.py.

    def draw(self, view='yx') :
        self.drawEdges(self.lineColor, self.lineWidth, view)
        self.drawCharge(self.lineColor, view)

drawCharge()

def drawFromPickle.PadWithHits.drawCharge	(		self,
			color,
			view = 'yx'
	)

Definition at line 135 of file drawFromPickle.py.

    def drawCharge(self, color, view='yx') :
        v = self.rotatedAvgChargeXyz if view=='hz' else self.avgChargeXyz
        x, y = 0., 0.
        if   view=='yx'   : x, y = v.x(),   v.y()
        elif view=='hz'   : x, y = v.z(),   v.y()
        elif view=='zphi' : x, y = v.phi(), v.z()
        marker = r.TMarker(x, y, r.kMultiply)
        marker.SetMarkerColor(color)
        marker.Draw()
        if hasattr(self, '_markers') : self._markers.append(marker)
        else : self._markers = [marker]
        def drawCloud() : print 'drawCloud not implemented yet'
        if len(self.hitIndices)>1 : drawCloud()
    
#_________________________________________________

drawEdges()

def drawFromPickle.PadWithHits.drawEdges	(		self,
			color,
			width,
			view = 'yx'
	)

Definition at line 97 of file drawFromPickle.py.

    def drawEdges(self, color, width, view='yx') :
        lineColor = color
        lineWidth = width
        corners = self.rotatedCornerXyz if view=='hz' else self.cornerXyz
        segments, lines = [], []
         # see Pad::fillCornerCoords to remember which corner is where
        if   view=='yx'   : segments = [(corners[i], corners[j]) for i,j in [(0, 1), (1, 3), (2, 3), (0, 2)]]
        elif view=='hz'   : segments = [(corners[i], corners[j]) for i,j in [(0, 1)]]
        elif view=='zphi' : segments = [(corners[i], corners[j]) for i,j in [(0, 2)]]
        l = r.TLine
        if   view=='yx'   : lines = [l(p1.x(),   p1.y(), p2.x(),   p2.y()) for p1,p2 in segments]
        elif view=='hz'   : lines = [l(p1.z(),   p1.y(), p2.z(),   p2.y()) for p1,p2 in segments]
        elif view=='zphi' : lines = [l(p1.phi(), p1.z(), p2.phi(), p2.z()) for p1,p2 in segments]
        for l in lines :
            l.SetLineColor(lineColor)
            l.SetLineWidth(lineWidth)
            l.Draw()
        if hasattr(self, '_lines') : self._lines.append(lines)
        else : self._lines = lines
        self.printPadCoord(view, lines[0])

isPivot()

def drawFromPickle.PadWithHits.isPivot

(

self

)

Definition at line 72 of file drawFromPickle.py.

    def isPivot(self) : return self.multiplet in [0,2] # ASM 28/04/2016 : Check that this is correct

printPadCoord()

def drawFromPickle.PadWithHits.printPadCoord	(		self,
			view,
			line
	)

Definition at line 117 of file drawFromPickle.py.

    def printPadCoord(self, view, line) :
        if view not in ['hz','zphi'] : return
        coord = {'hz':'ieta', 'zphi':'iphi'}[view]
        leftOrRight = self.layer%2
        angle = 90. if view=='hz' else 0.
        xs = sorted([line.GetX1(), line.GetX2()], reverse=leftOrRight)
        ys = sorted([line.GetY1(), line.GetY2()], reverse=leftOrRight)
        def outside(xx) : return xx[1] + 0.75*(xx[1]-xx[0])
        x = average(xs) if view=='hz'   else outside(xs)
        y = average(ys) if view=='zphi' else outside(ys)
        text = r.TLatex(x, y, "%d" % getattr(self, coord))
        text.SetTextAlign(22)
        text.SetTextAngle(angle)
        text.SetTextColor(line.GetLineColor())
        if hasattr(self, 'font') : text.SetTextFont(self.font)
        text.Draw()
        if hasattr(self, '_labels') : self._labels.append(text)
        else : self._labels = [text]

rotateVecToSec5()

def drawFromPickle.PadWithHits.rotateVecToSec5	(		self,
			vec
	)

Definition at line 73 of file drawFromPickle.py.

    def rotateVecToSec5(self, vec) :
        "Rotate a vector to sector5 coordinates (note: delta phi depends only on the sector)"
        deltaPhi = midSectorPhi(self.sector) - midSectorPhi(5)
        return repv(vec.rho(), vec.eta(), vec.phi()).SetPhi(vec.phi() - deltaPhi)

setFont()

def drawFromPickle.PadWithHits.setFont	(		self,
			font
	)

Definition at line 71 of file drawFromPickle.py.

    def setFont(self, font) : self.font = font

Member Data Documentation

_labels

drawFromPickle.PadWithHits._labels

private

Definition at line 134 of file drawFromPickle.py.

_lines

drawFromPickle.PadWithHits._lines

private

Definition at line 115 of file drawFromPickle.py.

_markers

drawFromPickle.PadWithHits._markers

private

Definition at line 145 of file drawFromPickle.py.

avgChargeXyz

drawFromPickle.PadWithHits.avgChargeXyz

Definition at line 59 of file drawFromPickle.py.

cornerXyz

drawFromPickle.PadWithHits.cornerXyz

Definition at line 58 of file drawFromPickle.py.

font

drawFromPickle.PadWithHits.font

Definition at line 71 of file drawFromPickle.py.

lineColor

drawFromPickle.PadWithHits.lineColor

Definition at line 69 of file drawFromPickle.py.

lineWidth

drawFromPickle.PadWithHits.lineWidth

Definition at line 70 of file drawFromPickle.py.

maxH

drawFromPickle.PadWithHits.maxH

Definition at line 65 of file drawFromPickle.py.

maxPhi

drawFromPickle.PadWithHits.maxPhi

Definition at line 66 of file drawFromPickle.py.

maxX

drawFromPickle.PadWithHits.maxX

Definition at line 62 of file drawFromPickle.py.

maxY

drawFromPickle.PadWithHits.maxY

Definition at line 63 of file drawFromPickle.py.

maxZ

drawFromPickle.PadWithHits.maxZ

Definition at line 64 of file drawFromPickle.py.

rotatedAvgChargeXyz

drawFromPickle.PadWithHits.rotatedAvgChargeXyz

Definition at line 61 of file drawFromPickle.py.

rotatedCornerXyz

drawFromPickle.PadWithHits.rotatedCornerXyz

Definition at line 60 of file drawFromPickle.py.

---

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

• drawFromPickle.py