PixelMapPlotter.HistogramManager Class Reference

GLOBAL VARS. More...

Public Member Functions
def	__init__ (self)
def	drawLine (self, lineObj, x1, x2, y1, y2, width=2, style=1, color=1)
def	drawRectangle (self, lineObj, x1, x2, y1, y2, width=2, style=1, color=1)
def	fillHistograms (self, barrelObjs, forwardObjs)
def	prettifyCanvas (self, hist)
def	saveHistograms (self)

Public Attributes
	barrelHists
	forwardHists

Detailed Description

GLOBAL VARS.

GLOBAL VARS

Definition at line 36 of file PixelMapPlotter.py.

Constructor & Destructor Documentation

def PixelMapPlotter.HistogramManager.__init__

(

self

)

Definition at line 37 of file PixelMapPlotter.py.

  def __init__(self):
    self.barrelHists = []
    self.forwardHists = []
    
    # barrel histograms
    i = 1
    for maxLadder in onlineMaxLadder:
      nBinsX = (maxOnlineModule * 2 + 1) * rocsInRow
      nBinsY = (maxLadder * 2 + 1) * rocsInCol

      name = "PXBarrel_Layer" + str(i);
      title = "PXBarrel_Layer" + str(i);

      histObj = ROOT.TH2F(name, title, nBinsX, -(maxOnlineModule + 0.5), maxOnlineModule + 0.5, nBinsY, -(maxLadder + 0.5), maxLadder + 0.5)
      histObj.GetXaxis().SetTitle("OnlineSignedModule");
      histObj.GetYaxis().SetTitle("OnlineSignedLadder");
      histObj.SetOption("COLZ")
      histObj.SetStats(0)
      histObj.SetTitle("OnlineSignedModule")

      self.barrelHists.append(deepcopy(histObj))
      i = i + 1

    # forward histograms
    i = 1
    for maxBlade in onlineMaxBlade:
      nBinsX = (maxOnlineDisk * 2 + 1) * rocsInRow
      nBinsY = (maxBlade * 2 + 1) * 2 * rocsInCol

      name = "PXForward_Ring" + str(i);
      title = "PXForward_Ring" + str(i);

      histObj = ROOT.TH2F(name, title, nBinsX, -(maxOnlineDisk + 0.5), maxOnlineDisk + 0.5, nBinsY, -(maxBlade + 0.5), maxBlade + 0.5)
      histObj.GetXaxis().SetTitle("OnlineSignedDisk");
      histObj.GetYaxis().SetTitle("OnlineSignedBladePanel");
      histObj.SetOption("COLZ")
      histObj.SetStats(0)
      

      self.forwardHists.append(deepcopy(histObj))
      i = i + 1

Member Function Documentation

def PixelMapPlotter.HistogramManager.drawLine	(		self,
			lineObj,
			x1,
			x2,
			y1,
			y2,
			width = 2,
			style = 1,
			color = 1
	)

Definition at line 109 of file PixelMapPlotter.py.

Referenced by PixelMapPlotter.HistogramManager.drawRectangle(), and PixelMapPlotter.HistogramManager.prettifyCanvas().

  def drawLine(self, lineObj, x1, x2, y1, y2, width=2, style=1, color=1):  
    lineObj.SetLineWidth(width)
    lineObj.SetLineStyle(style)
    lineObj.SetLineColor(color)

    lineObj.DrawLine(x1, y1, x2, y2)

def PixelMapPlotter.HistogramManager.drawRectangle	(		self,
			lineObj,
			x1,
			x2,
			y1,
			y2,
			width = 2,
			style = 1,
			color = 1
	)

Definition at line 116 of file PixelMapPlotter.py.

References PixelMapPlotter.HistogramManager.drawLine().

Referenced by PixelMapPlotter.HistogramManager.prettifyCanvas().

  def drawRectangle(self, lineObj, x1, x2, y1, y2, width=2, style=1, color=1):
    self.drawLine(lineObj, x1, x2, y2, y2, width, style, color)
    self.drawLine(lineObj, x2, x2, y2, y1, width, style, color)
    self.drawLine(lineObj, x2, x1, y1, y1, width, style, color)
    self.drawLine(lineObj, x1, x1, y1, y2, width, style, color)

def PixelMapPlotter.HistogramManager.fillHistograms	(		self,
			barrelObjs,
			forwardObjs
	)

Definition at line 79 of file PixelMapPlotter.py.

References PixelMapPlotter.HistogramManager.barrelHists, objects.autophobj.float, PixelMapPlotter.HistogramManager.forwardHists, PixelMapPlotter.TranslateReasonStringBPix(), and PixelMapPlotter.TranslateReasonStringFPix().

  def fillHistograms(self, barrelObjs, forwardObjs):
    for b in barrelObjs:
      coord = b.GetXYCoords()
      histRef = self.barrelHists[b.layer - 1]
      
      binNum = histRef.FindBin(coord[0], coord[1])
      if colorCoded:
        binContent = TranslateReasonStringBPix(b.reason)
      elif pixelAlive:
        binContent = float(b.reason)
      else:
        binContent = b.roc + 1

      histRef.SetBinContent(binNum, binContent)
      # self.barrelHists[b.layer - 1].Fill(coord[0], coord[1], b.roc + 1) 

    for f in forwardObjs:
      coord = f.GetXYCoords()
      histRef = self.forwardHists[f.ring - 1]
      
      binNum = histRef.FindBin(coord[0], coord[1])

      if colorCoded:
        binContent = TranslateReasonStringFPix(f.reason)
      else:
        binContent = f.roc + 1
      histRef.SetBinContent(binNum, binContent)      

      # self.forwardHists[f.ring - 1].Fill(coord[0], coord[1], f.roc + 1)
  

def PixelMapPlotter.HistogramManager.prettifyCanvas	(		self,
			hist
	)

Definition at line 122 of file PixelMapPlotter.py.

References PixelMapPlotter.HistogramManager.drawLine(), PixelMapPlotter.HistogramManager.drawRectangle(), objects.autophobj.float, createfilelist.int, and reco.print().

Referenced by PixelMapPlotter.HistogramManager.saveHistograms().

  def prettifyCanvas(self, hist):
    nBinsX = hist.GetXaxis().GetNbins()
    xMin = hist.GetXaxis().GetXmin()
    xMax = hist.GetXaxis().GetXmax()
    nBinsY = hist.GetYaxis().GetNbins()
    yMin = hist.GetYaxis().GetXmin()
    yMax = hist.GetYaxis().GetXmax()

    xLen = int(xMax)
    yLen = int(yMax)

    name = hist.GetName()[0:3]
    isBarrel = True if name != "PXF" else False
    print(name, isBarrel)

    xBaseStep = 1
    xRange = (nBinsX - 1) // (rocsInRow * 2) + 1
    yBaseStep = (yMax - yMin) / nBinsY
    yRange = (nBinsY - 1) // (2) + 1
    if not isBarrel:
      yBaseStep = yBaseStep * 2
      yRange = yRange // 2

    # horizontal 
    x1 = xMin
    x2 = xMin + xLen
    y1 = yMin
    y2 = yMin

    lineObj = ROOT.TLine()
    lineObj.SetBit(ROOT.kCanDelete)

    for i in range(yRange):
      w = 1 if i % 2 else 2
      self.drawLine(lineObj, x1, x2, y1, y2, w)
      self.drawLine(lineObj, x1, x2, -y1, -y2, w)
      self.drawLine(lineObj, -x1, -x2, -y1, -y2,w )
      self.drawLine(lineObj, -x1, -x2, y1, y2, w)

      y1 = y1 + yBaseStep
      y2 = y2 + yBaseStep

    # vertical
    x1 = xMin
    x2 = xMin
    y1 = yMin
    y2 = yMin + yLen

    for i in range(xRange):
      self.drawLine(lineObj, x1, x2, y1, y2, style = 9)
      self.drawLine(lineObj, x1, x2, -y1, -y2, style = 9)
      self.drawLine(lineObj, -x1, -x2, -y1, -y2, style = 9)
      self.drawLine(lineObj, -x1, -x2, y1, y2, style = 9)

      x1 = x1 + xBaseStep
      x2 = x2 + xBaseStep

    # mark zero ROC
    zeroModuleHeight = yBaseStep if isBarrel else yBaseStep * 0.5 # because there are two panels height of roc is smaller

    yRange = int(yMax) if isBarrel else int(yMax) * 2
    
    x1_base = 0                         + xMin
    x2_base = xBaseStep / float(rocsInRow) + xMin
    y1_base = zeroModuleHeight          + yMin
    y2_base = 2 * zeroModuleHeight      + yMin

    for i in range(yRange):
      y1 = y1_base + i * (zeroModuleHeight * 2) - (zeroModuleHeight if i % 2 else 0)
      y2 = y2_base + i * (zeroModuleHeight * 2) - (zeroModuleHeight if i % 2 else 0)
      
      #negative ladders/blades
      for j in range(int(xMax)):
        x1 = x1_base + j * xBaseStep
        x2 = x2_base + j * xBaseStep
        if yMax == 6.5 and x1 <0:
          y1 = y1_base + i * (zeroModuleHeight * 2) + (zeroModuleHeight if i % 2 else 0)
          y2 = y2_base + i * (zeroModuleHeight * 2) + (zeroModuleHeight if i % 2 else 0)
          self.drawRectangle(lineObj,(xBaseStep+x1-(x2-x1)),(xBaseStep+x2-(x2-x1)), y1+(y1-y2), y2+(y1-y2), color=8)
          x1, x2 = -x1, -x2
          yPosChange = -zeroModuleHeight if i % 2 else zeroModuleHeight
          self.drawRectangle(lineObj, x1, x2, y1 - yPosChange-2*(zeroModuleHeight if i % 2 else 0), y2 - yPosChange-2*(zeroModuleHeight if i % 2 else 0), color=8)
        else:
          self.drawRectangle(lineObj, x1, x2, y1, y2, color=8)

          x1, x2 = -x1, -x2
          yPosChange = -zeroModuleHeight if i % 2 else zeroModuleHeight
          self.drawRectangle(lineObj, x1, x2, y1 - yPosChange, y2 - yPosChange, color=8)


      # positive ladders/blades
      y1 = y1 - yMin + yBaseStep
      y2 = y2 - yMin + yBaseStep

      for j in range(int(xMax)):
        x1 = x1_base + j * xBaseStep
        x2 = x2_base + j * xBaseStep

        if yMax== 6.5 and x1 <0:
          self.drawRectangle(lineObj, xBaseStep+x1-(x2-x1), xBaseStep+x2-(x2-x1), y1+(y1-y2), y2+(y1-y2), color=8)
          x1, x2 = -x1, -x2
          self.drawRectangle(lineObj, x1, x2, y1 - yPosChange- 2*(zeroModuleHeight if i % 2 else 0), y2 - yPosChange-2*(zeroModuleHeight if i % 2 else 0), color=8)
        else:
          self.drawRectangle(lineObj, x1, x2, y1, y2, color=8)
          x1, x2 = -x1, -x2
          self.drawRectangle(lineObj, x1, x2, y1 - yPosChange, y2 - yPosChange, color=8)

#      hist.GetZaxis().SetRangeUser(-0.5,15.5)

def PixelMapPlotter.HistogramManager.saveHistograms

(

self

)

Definition at line 231 of file PixelMapPlotter.py.

References PixelMapPlotter.HistogramManager.barrelHists, PixelMapPlotter.HistogramManager.forwardHists, and PixelMapPlotter.HistogramManager.prettifyCanvas().

  def saveHistograms(self):
    for hists in [self.barrelHists, self.forwardHists]:
      for hist in hists:
        # if hist.GetEntries():
        c1 = ROOT.TCanvas(hist.GetName(), hist.GetName(), hRes, vRes)
        if colorCoded:
          hist.GetZaxis().SetRangeUser(0,5)
          ROOT.gStyle.SetPalette(55)
        elif pixelAlive:
          hist.GetZaxis().SetRangeUser(0,4160)
          ROOT.gStyle.SetPalette(70)
        hist.Draw()

        txt = TLatex()
        txt.SetNDC()
        txt.SetTextFont(1)
        txt.SetTextColor(1)
        txt.SetTextAlign(22)
        txt.SetTextAngle(0)
        txt.SetTextSize(0.05)
        txt.DrawLatex(0.5, 0.95, hist.GetName())

        xMin = hist.GetXaxis().GetXmin()

        yMin = hist.GetYaxis().GetXmin()

        box1 = TBox(xMin*1.1,yMin*1.25,xMin*1,yMin*1.15);
        box1.SetFillColor(kRed+3)
        box1.Draw()
        txt.SetTextSize(0.035)
        txt.DrawLatex(0.25, 0.077, "Dead At Beginning")


        box2 = TBox(xMin*0.45,yMin*1.25,xMin*0.35,yMin*1.15);
        box2.SetFillColor(kAzure+2)
        box2.Draw()
        txt.SetTextSize(0.035)
        txt.DrawLatex(0.47, 0.077, "Dead At End")


        self.prettifyCanvas(hist)
        colorString = ""
        if colorCoded:
          colorString = "_coded"
        elif pixelAlive:
          colorString = "_pixelalive"
        if useFileSuffix:
          c1.Print(hist.GetName() + colorString + "_" + inputFileName[:-4] + ".png")
        else:
          c1.Print(hist.GetName() + colorString + ".png")
            

Member Data Documentation

PixelMapPlotter.HistogramManager.barrelHists

Definition at line 38 of file PixelMapPlotter.py.

Referenced by PixelMapPlotter.HistogramManager.fillHistograms(), and PixelMapPlotter.HistogramManager.saveHistograms().

PixelMapPlotter.HistogramManager.forwardHists

Definition at line 39 of file PixelMapPlotter.py.

Referenced by PixelMapPlotter.HistogramManager.fillHistograms(), and PixelMapPlotter.HistogramManager.saveHistograms().