normalizator Namespace Reference

Functions
def	browseDir (dir)
def	main (argv)

Function Documentation

browseDir()

def normalizator.browseDir

(

dir

)

Definition at line 15 of file normalizator.py.

def browseDir( dir ):
    #print dir.GetPath().split(':', 1)[-1]
    path = dir.GetPath().split(':/', 1)[-1]
    pathdir = dir #.GetDirectory(path)
    #print pathdir.GetPath()
    keylist = dir.GetListOfKeys()
    
    for key in keylist:
        obj = key.ReadObj()
 
        #2D histograms cannot be compared, tprofiles and efficiencies should not be normalized; skip
        if obj.IsA().InheritsFrom(ROOT.TH2.Class()) or obj.IsA().InheritsFrom(ROOT.TProfile.Class()) or "fficiency" in obj.GetName() or 'RecoFraction' in obj.GetName() or 'purity' in obj.GetName() or "PtResol" in obj.GetName() or "PtScale" in obj.GetName() or "Prof" in obj.GetName() or "Fit" in obj.GetName():
            continue
 
        if obj.IsA().InheritsFrom(ROOT.TH1.Class()) :
            #normalize
            #print '----- ', obj.GetName()
            ii = 1.*obj.Integral()
            hname = obj.GetName()
            if ii>0:
                 obj.Scale(1./ii)
                 #print '   ',hname
                 pathdir.WriteTObject( obj , hname , "WriteDelete" )
            continue
 
        if obj.IsA().InheritsFrom(ROOT.TDirectory.Class()):
            #print obj.GetName()
            browseDir(obj)
        
  
 
#======================================================================

main()

def normalizator.main

(

argv

)

Main function to be executed when starting the code.

Definition at line 47 of file normalizator.py.

def main( argv ):
  """
  Main function to be executed when starting the code.
  """
 
  if len( argv ) < 2:
      print( 'No filename given' )
      print( 'Usage: python '+argv[0]+' physval_filename' )
      exit(1)
 
  filename = argv[1]
  if not os.path.exists( filename ):
      print ( 'File not found: ' + filename )
      exit(1)
 
  try:
      infile = ROOT.TFile.Open( filename, 'update' )
  except:
      print( 'Cannot open file: {0}'.format( filename ) )
      exit(1)
 
  ROOT.gROOT.SetBatch()
 
  infile.cd("Muons")
  topDir = infile.GetDirectory("Muons")
 
  browseDir(topDir)
  infile.Close()
  
#======================================================================