ParseFlavourInput Namespace Reference

Functions
def	GetKeyNames (self, dir="")
def	ReadFlavourHistograms (dirName)
def	ReadFlavourHistograms (dirName, freezepT=False)

Variables
dictionary	SystematicNameDictionary
dictionary	jetDefDict
list	SystematicNames = ['flavorCompLight','flavorCompGlu','FlavorResponse']

Function Documentation

GetKeyNames()

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

Definition at line 11 of file Final2012/ParseFlavourInput.py.

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

ReadFlavourHistograms() [1/2]

def ParseFlavourInput.ReadFlavourHistograms

(

dirName

)

Definition at line 30 of file Final2012/ParseFlavourInput.py.

def ReadFlavourHistograms(dirName):
    if not dirName.endswith("/"):
        dirName = dirName + "/"
    
    # Run over each file (one per jet definition for LQ/Gluon and one per jet definition for HvP)
    histos = {}
    files = sorted(glob.glob(dirName+"*.root"))
    for aFileName in files:
        # Determine the jet definition
        jetDef = ""
        for aFileDef,aJetDef in jetDefDict.iteritems():
            if aFileDef in aFileName:
                jetDef = aJetDef
                break
        if jetDef == "":
            print "Failed to determine jet definition for file:",aFileName
            return None
        if jetDef not in histos:
            histos[jetDef] = {}
 
        # Read in the histograms from the file
        # Note that we want all the histograms, but with this scheme not everything in the dict is in each file
        # So instead loop over all histos in the file
        inFile = TFile(aFileName,"READ")
        for histName in inFile.GetKeyNames():
            systematicName = SystematicNameDictionary[histName]+"_"+jetDef
            histo = inFile.Get(histName)
            if histo is None:
                print "Failed to get histogram:",systematicName
                return None
 
            histoSym = ProviderHistoHelpers.SymmetrizeHistoInEta(histo,systematicName)
            histoSym.SetDirectory(0)
            histos[jetDef][SystematicNameDictionary[histName]] = histoSym
        # Done reading, close the file
        inFile.Close()
 
    return histos
 
 
 

ReadFlavourHistograms() [2/2]

def ParseFlavourInput.ReadFlavourHistograms	(		dirName,
			freezepT = False
	)

Definition at line 23 of file ICHEP2016/ParseFlavourInput.py.

def ReadFlavourHistograms(dirName,freezepT=False):
    if not dirName.endswith("/") and not dirName.endswith(".root"):
        dirName = dirName + "/"
    if dirName.endswith("/") :
      files = sorted(glob.glob(dirName+"*.root"))
    else :
      files = [dirName]
 
    # Run over each file (one per jet definition for LQ/Gluon and one per jet definition for HvP)
    histos = {}
    for aFileName in files:
        # Determine the jet definition
        jetDef = ""
        for jetDef in jetDefDict.keys() :
          print "Beginning jetDef",jetDef
          if jetDefDict[jetDef] not in histos:
              histos[jetDefDict[jetDef]] = {}
 
          # Read in the histograms from the file
          # Note that we want all the histograms, but with this scheme not everything in the dict is in each file
          # So instead loop over all histos in the file
          inFile = TFile(aFileName,"READ")
          for histName in inFile.GetKeyNames():
              systName = histName.split("_")[0]  
              if not systName in SystematicNames :
                print "Error!!!"
                continue
              if not jetDef in histName :
                continue
              systematicName = systName + "_" + jetDefDict[jetDef]
              histo = inFile.Get(histName)
              if histo is None:
                  print "Failed to get histogram:",systematicName
                  return None
 
              if freezepT :
                # Loop over eta bins.
                for ybin in range(histo.GetNbinsY()+2) :
                  # Loop out in pT.
                  freezeval = 0
                  for xbin in range(histo.GetNbinsX()+2) :
                    inbin = histo.GetBinContent(xbin,ybin)
                    #print "Bin at pT ",histo.GetXaxis().GetBinCenter(xbin)," and [ylow,y,yhigh] = ", binYLow, histo.GetYaxis().GetBinCenter(ybin), binYHigh
                    # Store bin contents as we go out: last one is one we want as frozen value.
                    if inbin > 0 :
                      freezeval = inbin
                    else :
                      histo.SetBinContent(xbin,ybin,freezeval)
 
              if systName.find("FlavorResponse")!=-1 :
                histo.Scale(-1.0)
 
              histoSym = ProviderHistoHelpers.SymmetrizeHistoInEta(histo,systematicName)
              histoSym.SetDirectory(0)
              histos[jetDefDict[jetDef]][systName] = histoSym
 
        # Done reading, close the file
        inFile.Close()
 
    return histos

Variable Documentation

jetDefDict

dictionary ParseFlavourInput.jetDefDict

Initial value:

=     {
                    '4EM' : 'AntiKt4Topo_EMJES',
                    '6EM' : 'AntiKt6Topo_EMJES',
                    '4LC' : 'AntiKt4Topo_LCJES',
                    '6LC' : 'AntiKt6Topo_LCJES'
                }

Definition at line 23 of file Final2012/ParseFlavourInput.py.

SystematicNameDictionary

dictionary ParseFlavourInput.SystematicNameDictionary

Initial value:

=  {
                            'LQ'                    : 'flavorCompLight',
                            'Gluon'                 : 'flavorCompGlu',
                            'HerwigVsPythia_Gluon'  : 'FlavorResponse'
                           }

Definition at line 17 of file Final2012/ParseFlavourInput.py.

SystematicNames

list ParseFlavourInput.SystematicNames = ['flavorCompLight','flavorCompGlu','FlavorResponse']

Definition at line 17 of file ICHEP2016/ParseFlavourInput.py.