validator Namespace Reference

Classes
class	Validator

Functions
	GoToDirectory (directory)
	SameHist (refHist, testHist)
	SimilarHist (refHist, testHist)
	main (argv)

Variables
list	LABELS = []

Detailed Description

This script compares two given root files and saves plots where they differ
as pdfs.

author: Felix Socher  <Felix.Socher@cern.ch>

Function Documentation

GoToDirectory()

validator.GoToDirectory

(

directory

)

Definition at line 19 of file validator.py.

def GoToDirectory(directory):
  cwd = os.getcwd()
  print os.getcwd()
  newdir = cwd + "/" + directory 
  if not os.path.isdir( newdir ):
    print "Creating folder " + newdir
    os.makedirs( newdir )
  os.chdir(newdir)
 
 

main()

validator.main

(

argv

)

Main function to be executed when starting the code.

Definition at line 276 of file validator.py.

def main( argv ):
  """
  Main function to be executed when starting the code.
  """
 
  parser = argparse.ArgumentParser( description = 'Distribution Plotter' )
  parser.add_argument( '-s', '--structDirs', default = True, action = "store_true", help = ' if true, it creates directories following the same structure as in the root file to store the pdf plots')
  parser.add_argument( '-r', '--reference', help = 'The reference' )
  parser.add_argument( '-t', '--test', help = 'The test' )
  parser.add_argument( '-t2', '--test2', default ='', help = 'The additional test' )
  parser.add_argument( '-d', '--directory', default = "/", help = 'Print cutflow fror systematic variations' )
  parser.add_argument( '-e', '--exclude', default = "_bin_", help = 'histograms whose names contain the provided strings are not examined')
  parser.add_argument( '-n', '--normalize', default = False, action = "store_true", help = 'normalize histograms with larger stats for better comparison')
  parser.add_argument( '-l', '--labels', default ='', help = 'Add text to legend for ref/test/test2, split with commas' )
 
  args = parser.parse_args()
 
 
  #ROOT.gROOT.Macro("rootlogon.C")
  ROOT.gROOT.SetBatch()
  #ROOT.gROOT.LoadMacro("./AtlasUtils.C") 
  ROOT.gROOT.LoadMacro("./AtlasStyle.C")
  ROOT.TH1.SetDefaultSumw2(ROOT.kTRUE)
  
  SetAtlasStyle()
 
  LABELS = []
  if args.labels is '':
    LABELS = [ 'Ref','Test' ]
    if args.test2 is not '':
      LABELS.append( 'Test2' )
  else:
    LABELS = args.labels.split(",")
    #print LABELS
    
  # if you want to pass labels to legend directly from the file name...:
  #LABELS.append( args.test.split('.')[2]+' '+args.test.split('.')[3] )
  #
    
  validator = Validator( args.structDirs, args.exclude, SameHist, args.normalize )
  if args.test2 == '':
    allFiles = [ os.path.abspath(args.reference), os.path.abspath(args.test) ]
  else:
    allFiles = [ os.path.abspath(args.reference), os.path.abspath(args.test), os.path.abspath(args.test2) ]
 #======================================================================
 

SameHist()

validator.SameHist	(		refHist,
			testHist )

Definition at line 29 of file validator.py.

def SameHist(refHist, testHist):
  chi2 = 0
  for x in range(refHist.GetNbinsX()):
    for y in range(refHist.GetNbinsY()):
      for z in range(refHist.GetNbinsZ()):
        if refHist.GetBinContent(x,y,z) != testHist.GetBinContent(x,y,z): return False
  return True
 

SimilarHist()

validator.SimilarHist	(		refHist,
			testHist )

Definition at line 37 of file validator.py.

def SimilarHist(refHist, testHist):
  for x in range(refHist.GetNbinsX()):
    for y in range(refHist.GetNbinsY()):
      for z in range(refHist.GetNbinsZ()):
        if abs(refHist.GetBinContent(x,y,z) - testHist.GetBinContent(x,y,z)) > 0.01: return False
  return True
 
 
 
 

Variable Documentation

LABELS

list validator.LABELS = []

Definition at line 17 of file validator.py.