validator.Validator Class Reference

Inheritance diagram for validator.Validator:

Collaboration diagram for validator.Validator:

Public Member Functions
def	__init__ (self, structDirs, exclude, compareFunc, normalization)
def	CompareFiles (self, fileNames, dirName)
def	CompareDirectories (self, dirs, dirName)
def	CompareHistograms (self, hists, path)
def	MakeComparisonPlot (self, hists, path)

Public Attributes
	structDirs
	CompareFunc
	DoNormalization

Detailed Description

docstring for Validator

Definition at line 47 of file validator.py.

Constructor & Destructor Documentation

__init__()

def validator.Validator.__init__	(		self,
			structDirs,
			exclude,
			compareFunc,
			normalization
	)

Definition at line 49 of file validator.py.

  def __init__(self, structDirs, exclude, compareFunc, normalization):
      super(Validator, self).__init__()
      self.structDirs = structDirs
      #self.excludedStrings = exclude.split("|")
      #print self.excludedStrings
      self.CompareFunc = compareFunc
      self.DoNormalization = normalization
  

Member Function Documentation

CompareDirectories()

def validator.Validator.CompareDirectories	(		self,
			dirs,
			dirName
	)

Definition at line 61 of file validator.py.

  def CompareDirectories(self, dirs, dirName):
    if (self.structDirs): GoToDirectory(dirName)
    refDir = dirs[0]
    newRefDir = refDir.GetDirectory(dirName)
    for testDir in dirs:
      if testDir == refDir: continue
      if not testDir:
        print 'ERROR --------- test dir not found: ', dirName, testDir
        return
 
    newTestDirs = [ testDir.GetDirectory(dirName) for testDir in dirs ]
    
    theDir = newRefDir
    # if not newRefDir:
    #   print 'ERROR --------- reference dir not found: ', dirName
    #   theDir = newTestDir
 
    for key in theDir.GetListOfKeys():
          
      obj = key.ReadObj()
 
      #2D histograms cannot be compared; skip
      if obj.IsA().InheritsFrom(ROOT.TH2.Class()):
        continue
      
      if obj.IsA().InheritsFrom(ROOT.TH1.Class()):
        print key.GetName()
        #testHist = newTestDir.Get(key.GetName())
        testHists = [ newTestDir.Get(key.GetName()) for newTestDir in newTestDirs ]
        self.CompareHistograms( testHists, newRefDir.GetPath())
      elif obj.IsA().InheritsFrom(ROOT.TDirectory.Class()):
        self.CompareDirectories( newTestDirs, obj.GetName())
    if (self.structDirs): os.chdir("..")
  

CompareFiles()

def validator.Validator.CompareFiles	(		self,
			fileNames,
			dirName
	)

Definition at line 57 of file validator.py.

  def CompareFiles(self, fileNames, dirName):
    openedFiles = [ TFile.Open(f) for f in fileNames ]
    self.CompareDirectories(openedFiles, dirName)
  

CompareHistograms()

def validator.Validator.CompareHistograms	(		self,
			hists,
			path
	)

Definition at line 95 of file validator.py.

  def CompareHistograms(self, hists, path):
    # print self.excludedStrings, refHist.GetName()
    # if any(exString in refHist.GetName() for exString in self.excludedStrings): 
    #   print "Skipped " + path + "/" + refHist.GetName()
    #   return
    #@@@if self.CompareFunc(refHist, testHist): print path + refHist.GetName() + " looks ok"
    #else:
    #self.MakeComparisonPlot(refHist, testHist, path)
    self.MakeComparisonPlot(hists, path)
  

MakeComparisonPlot()

def validator.Validator.MakeComparisonPlot	(		self,
			hists,
			path
	)

Definition at line 106 of file validator.py.

  def MakeComparisonPlot(self, hists, path):
    def SetBounds(hists, ymin=0, ymax=0):
      if (ymin==0 and ymax==0):
        ymin = min( [ h.GetMinimum() for h in hists ] )
        ymax = min( [ h.GetMaximum() for h in hists ] )
        if ymin>0 and ymax>0:
          ymin = 0
        else:          
          ymin = ymin - 0.15*abs(ymin)
        ymax = ymax+ 0.15*ymax
        # if ymin<0 and ymax>0:
        #     if ymax>0 and abs(minimum)>ymax:
        #   ymax = -minimum
      hists[0].SetMinimum(ymin)
      hists[0].SetMaximum(ymax)
      return hists[0]
    
          
    # try:
    #   refHist.Scale(1.0/refHist.Integral())
    #   testHist.Scale(1.0/testHist.Integral())
    # except:
    #   return
    canvas = ROOT.TCanvas( "", "", 900, 900 )
    padMain = ROOT.TPad( 'padMain', 'padMain', 0, 0.3, 1, 1 )
    padMain.SetBottomMargin( 0.02 )
    padMain.Draw()
    #padMain.SetLogy(1) #@@@
    padRatio = ROOT.TPad( 'padRatio', 'padRatio', 0, 0, 1, 0.3 )
    padRatio.SetTopMargin( 0.01 )
    padRatio.SetBottomMargin( 0.25 )
    padRatio.Draw()
    ROOT.TLine()
    padMain.cd()
    
    leg = ROOT.TLegend(0.82,0.78,0.96,0.94)  
    #leg = ROOT.TLegend(0.52,0.76,0.96,0.94)  
    #leg.SetFillColor(ROOT.kWhite)
    leg.SetFillStyle(0)
    leg.SetBorderSize(0)
    leg.SetTextFont(43)
    leg.SetTextSizePixels(32)    
    
    refHist = hists[0]
    refHist.SetLineColor(17)
    #refHist.SetFillColor(30)
 
    
    for i in range(0,len(hists)):      
      leg.AddEntry(hists[i], LABELS[i], 'lp')
              
    if self.DoNormalization and not "_Eff_" in refHist.GetName() and not "_eff" in refHist.GetName() and not "_Eff" in refHist.GetName():
      for h in hists:
        n = h.Integral()
        if n>0:
          h.Scale(1./n)
                
    refHist = SetBounds(hists)
    
    ref_textsize = 32./(padMain.GetWh()*padMain.GetAbsHNDC())
    refHist.GetYaxis().SetLabelSize( ref_textsize )
    refHist.GetXaxis().SetLabelSize( 0 )
    refHist.GetXaxis().SetTitleSize( 0 )
    refHist.GetYaxis().SetTitleSize( 1.3*ref_textsize )
    refHist.GetYaxis().SetTitleOffset(1)
    refHist.GetYaxis().SetTitleColor( kAzure )
 
    # testHist.GetYaxis().SetTextFont(43)
    # testHist.GetYaxis().SetTextSizePixels(20)
    refHist.SetMarkerSize(0)
    refHist.SetLineColor(ROOT.kRed)
    #refHist.GetYaxis().SetRangeUser(0.00001,refHist.GetMaximum()*5) ##@@@
 
    #testHist.Rebin(2)
    #refHist.Rebin(2);
    histmax = refHist
    histmax.DrawCopy('e')
    # for h in hists:
    #   if h.GetMaximum() > refHist.GetMaximum():
    #     histmax = h
    #   if histmax == refHist:
    #     histmax.DrawCopy("e")
    #   else:
    #     histmax.DrawCopy("ehist")
    nh=0
    for h in hists:
      if h is not histmax:
        if h==refHist:
          h.DrawCopy('histsame')
        else:
          nh=nh+1
          if nh>1:
            h.SetMarkerStyle(0)
            h.SetMarkerColor(0)
            h.SetLineColor(8) # green
            h.SetLineStyle(ROOT.kDashed)
          h.DrawCopy('ehistsame')
    # if refHist.GetMaximum()>histmax.GetMaximum():
    #   refHist.DrawCopy("e")
    #   testHist.DrawCopy("ehistsame")
    #   refHist.DrawCopy("ehistsame")
    # else:
    #   testHist.DrawCopy("ehist")
    #   refHist.DrawCopy("ehistsame")
      
    leg.Draw()
    padRatio.cd()
 
    for h in hists:
      if h is refHist: continue
      ratioHist = h.Clone()
      ratioHist.Divide(refHist)
      ratioHist = SetBounds( [ratioHist, ratioHist], 0.84,1.16) # 0.941,1.059
      for i in range(ratioHist.GetNbinsX()):
        nref = refHist.GetBinContent(i)
        ntest = h.GetBinContent(i)
        if nref == 0 or ntest == 0:
          ratioHist.SetBinError(i, 0)
        else:
          #error = nref/ntest*math.sqrt((refHist.GetBinError(i)/nref)**2 + (testHist.GetBinError(i)/ntest)**2)
          error = nref/ntest* max(refHist.GetBinError(i)/nref, h.GetBinError(i)/ntest) 
          ratioHist.SetBinError(i, error)
          
      ratioHist_textsize = 32./(padRatio.GetWh()*padRatio.GetAbsHNDC())
      ratioHist.GetYaxis().SetLabelSize( ratioHist_textsize )
      ratioHist.GetXaxis().SetLabelSize( ratioHist_textsize )
      ratioHist.GetXaxis().SetTitleSize( 1.2*ratioHist_textsize )
      ratioHist.GetXaxis().SetTitleOffset(0.75)
      ratioHist.GetXaxis().SetTitleColor(kAzure)
      ratioHist.GetYaxis().SetTitleSize( ratioHist_textsize )
      ratioHist.GetYaxis().SetTitleOffset(0.6)
 
    
      ratioHist.SetLineColor(ROOT.kBlack)
      ratioHist.SetMarkerStyle(24)
      ratioHist.SetYTitle("test / ref")
      if h == hists[1]:
        ratioHist.DrawCopy("p")
        lineRatio = ROOT.TLine( ratioHist.GetXaxis().GetXmin(), 1, 
                              ratioHist.GetXaxis().GetXmax(), 1 ) 
        lineRatio.SetLineColor( ROOT.kRed )
        lineRatio.SetLineWidth( 2 )
        lineRatio.Draw("same")
      else:
        ratioHist.SetMarkerStyle(25)
        ratioHist.SetMarkerColor(8)
        ratioHist.SetLineColor(8)
        ratioHist.DrawCopy("psame")
      
    npath = ""
    if not self.structDirs:
      npath = path[path.find(":/")+2:] + "/"
      npath = re.sub(r"[:,./]", "_", npath+"/")
 
    canvas.cd()
    t = ROOT.TLatex();
    t.SetNDC()
    t.SetTextColor(1)
    t.SetTextSize(0.03);
    t.DrawLatex(0.,0.97,refHist.GetName()) #@@@
 
    canvas.SaveAs(npath + refHist.GetName() + ".pdf")
    canvas.Close()
 
#======================================================================

Member Data Documentation

CompareFunc

validator.Validator.CompareFunc

Definition at line 54 of file validator.py.

DoNormalization

validator.Validator.DoNormalization

Definition at line 55 of file validator.py.

structDirs

validator.Validator.structDirs

Definition at line 51 of file validator.py.

---

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

• validator.py