ParseCurrentFile Namespace Reference

Functions
def	GetKeyNames (self, dir="")
def	ReadCurrentHistograms (aSystFile, newjetsdict=[])

Function Documentation

GetKeyNames()

def ParseCurrentFile.GetKeyNames	(		self,
			dir = ""
	)

Definition at line 10 of file ICHEP2016/ParseCurrentFile.py.

def GetKeyNames(self,dir=""):
  self.cd(dir)
  return [key.GetName() for key in gDirectory.GetListOfKeys()]
TFile.GetKeyNames = GetKeyNames
 

ReadCurrentHistograms()

def ParseCurrentFile.ReadCurrentHistograms	(		aSystFile,
			newjetsdict = []
	)

Definition at line 15 of file ICHEP2016/ParseCurrentFile.py.

def ReadCurrentHistograms(aSystFile,newjetsdict=[]):
 
    # Dictionary to translate from Bogdan's names to provider names
    useJetCollections = ["AntiKt4EMTopo","AntiKt4LCTopo"]
   
    # Just one file here with everything in it.
    systematicFile = TFile(aSystFile,"READ")
 
    if newjetsdict!=[] :
      useJetCollections = newjetsdict
 
    histos = {}
    for key in systematicFile.GetKeyNames() : 
 
      # Only keep things of the jet collections we are using.
      foundJetType = False
      print "looping through useJetCollections",useJetCollections
      for type in useJetCollections :
        if key.endswith(type) :
          jettype = type
          foundJetType = True
      if not foundJetType :
        print "No jet type matches"
        continue
      if jettype not in histos :
        print "Adding extra dict..."
        histos[jettype] = {}
 
      # Only keep things for which we do not have a more current version.
      systname = key.replace("_"+jettype,"")
 
      histo = systematicFile.Get(key)
      histo.SetDirectory(0)
      histos[jettype][systname] = histo
 
    # Done, return dictionary of histos
    return histos