Functions
	GetKeyNames (self, dir="")
	CombineRootFiles (fullFileName, reducedFileName, combinedFileName)
	readSanitizedLine (file)
	sanitizeComponentName (aName)
	getDescFromName (aName)
	getCategoryFromName (aName)
	checkIfWasCategoryReduction (listOfNames)
	replaceComponentNumber (line, newNumber)
	CombineConfigFiles (fullFileName, combinedFileName, combinedRootFileName, componentNames)

Variables
	sharePath = sys.argv[1]
	fullConfigFile = sys.argv[2]
	fullRootFile = sys.argv[3]
	reducedRootFile = sys.argv[4]
	outConfigFile = sys.argv[5]
	outRootFile = sys.argv[6]
	GetKeyNames
	compNames = CombineRootFiles(sharePath+"/"+fullRootFile,reducedRootFile,sharePath+"/"+outRootFile)

Function Documentation

◆ checkIfWasCategoryReduction()

CombineRootAndConfigFilesAfterReduction.checkIfWasCategoryReduction ( listOfNames )

Definition at line 126 of file CombineRootAndConfigFilesAfterReduction.py.

def checkIfWasCategoryReduction(listOfNames):
  for aName in listOfNames:
    if "Stat" in aName:
      return True
    elif "Detector" in aName:
      return True
    elif "Model" in aName:
      return True
    elif "Mixed" in aName:
      return True
  return False
 

◆ CombineConfigFiles()

CombineRootAndConfigFilesAfterReduction.CombineConfigFiles	(	fullFileName,
		combinedFileName,
		combinedRootFileName,
		componentNames )

Definition at line 144 of file CombineRootAndConfigFilesAfterReduction.py.

def CombineConfigFiles(fullFileName,combinedFileName,combinedRootFileName,componentNames):
  print "Creating reduced config file..."
 
  # Open the input file for reading, output file for writing
  inputFile = open(fullFileName,"r")
  outputFile = open(combinedFileName,"w")
 
  # Determine the type of reduction (for the header)
  reductionString = ""
  if "GlobalReduction" in combinedFileName:
    reductionString = "Reduction was applied globally"
  elif "CategoryReduction" in combinedFileName:
    reductionString = "Reduction was applied within each category"
  else:
    reductionString = "Unknown type of reduction"
 
  # Create the file header
  now = datetime.datetime.now()
  outputFile.write("\n#######################################\n#\n")
  outputFile.write("#  JESProvider Input Settings\n")
  outputFile.write("#    Reduced set of nuisance parameters from final 2011 iteration of the in-situ JES\n")
  outputFile.write("#    %s\n"%(reductionString))
  #outputFile.write("#      %s\n"%(fullFileName))
  outputFile.write("#    %s %d, %d\n#\n"%(calendar.month_name[now.month],now.day,now.year))
  #outputFile.write("#    \n") # AUTHORS HERE
  outputFile.write("#\n#######################################\n\n")
  
  # Specify the root file that goes with the header
  outputFile.write("JESUncertaintyRootFile:       %s\n"%(combinedRootFileName))
 
  # Settings header
  outputFile.write("\n#######################################\n#\n")
  outputFile.write("#   Settings for JES Components\n")
  outputFile.write("#\n#######################################\n\n")
 
  # JES Components
  # Parse the base names
  compNamesSanitized = []
  for aName in componentNames:
    sanitizedName = sanitizeComponentName(aName)
    if sanitizedName not in compNamesSanitized:
      compNamesSanitized.append(sanitizedName)
 
  i=1
  for aName in compNamesSanitized:
    outputFile.write("JESComponent.%d.Desc:            %s\n"%(i,getDescFromName(aName)))
    outputFile.write("JESComponent.%d.Name:            %s\n"%(i,aName))
    outputFile.write("JESComponent.%d.Type:            %s\n"%(i,getCategoryFromName(aName)))
    outputFile.write("\n")
    i = i + 1
 
  # Special components
  # Currently, read from other config file
  outputFile.write("\n#######################################\n\n")
 
  # Look for "Special" type components in the original file
  # EOF returns an empty string
  inputFile = open(fullFileName,"r")
  nextLine = readSanitizedLine(inputFile)
  currLine = ""
  previousLine = ""
  while (nextLine != ""):
    if "Special" not in nextLine and "EtaIntercal" not in currLine:
      # Not interesting (except for eta intercal), just iterate
      previousLine = currLine
      currLine = nextLine
      nextLine = readSanitizedLine(inputFile)
    else:
      # Found a special component, so save it
      outputFile.write(replaceComponentNumber(previousLine,i))
      outputFile.write(replaceComponentNumber(currLine,i))
      outputFile.write(replaceComponentNumber(nextLine,i))
      
      # Iterate
      previousLine = currLine
      currLine = nextLine
      nextLine = readSanitizedLine(inputFile)
      
      # Check if sub-components or correlated sub-components
      if ("SubComponent" in nextLine):
        outputFile.write(replaceComponentNumber(nextLine,i))
        previousLine = currLine
        currLine = nextLine
        nextLine = readSanitizedLine(inputFile)
             
      outputFile.write("\n")
      i = i + 1
 
 
  outputFile.write("\n#\n#######################################\n")
 
  # Done, close files
  inputFile.close()
  outputFile.close()
  print "Created config file: %s\n"%(combinedFileName)
 
# Main method here

◆ CombineRootFiles()

CombineRootAndConfigFilesAfterReduction.CombineRootFiles	(	fullFileName,
		reducedFileName,
		combinedFileName )

Definition at line 17 of file CombineRootAndConfigFilesAfterReduction.py.

def CombineRootFiles(fullFileName,reducedFileName,combinedFileName):
  print "Creating reduced root file..."
  print "Full file:     " + fullFileName
  print "Reduced file:  " + reducedFileName
  print "Output file:   " + combinedFileName
  AllNuisanceParametersFile = TFile.Open(fullFileName,"READ")
  ReducedNuisanceParametersFile = TFile.Open(reducedFileName,"READ")
  OutputFile = TFile.Open(combinedFileName,"RECREATE")
  
  for histName in AllNuisanceParametersFile.GetKeyNames():
    hist = AllNuisanceParametersFile.Get(histName)
    OutputFile.cd()
    hist.Write(histName)
 
  reducedComponentNames = []
  for histName in ReducedNuisanceParametersFile.GetKeyNames():
    if "EffectiveNP_" in histName:
      hist = ReducedNuisanceParametersFile.Get(histName)
      OutputFile.cd()
      hist.Write(histName)
      reducedComponentNames.append(histName)
  
  ''' This method is no longer relevant due to Reduction changes, retained in case this changes
 
  for histName in ReducedNuisanceParametersFile.GetKeyNames():
    hist = ReducedNuisanceParametersFile.Get(histName)
    OutputFile.cd()
    hist.Write(histName)
    reducedComponentNames.append(histName)
 
  for histName in AllNuisanceParametersFile.GetKeyNames():
    alreadyCopied = False
    for aComponent in reducedComponentNames:
      if histName in aComponent:
        alreadyCopied = True
        break
    if not alreadyCopied:
      hist = AllNuisanceParametersFile.Get(histName)
      OutputFile.cd()
      hist.Write(histName)
  '''
  
  AllNuisanceParametersFile.Close()
  ReducedNuisanceParametersFile.Close()
  OutputFile.Close()
 
  return reducedComponentNames
 
# Needed due to end of line differences between OS

◆ getCategoryFromName()

CombineRootAndConfigFilesAfterReduction.getCategoryFromName ( aName )

Definition at line 105 of file CombineRootAndConfigFilesAfterReduction.py.

def getCategoryFromName(aName):
  # Special for eta intercalibration components
  if "etaintercalibration_totalstat" in aName.lower():
    return "Statistical"
  elif "etaintercalibration_modelling" in aName.lower():
    return "Modelling"
  elif "stat" in aName.lower():
    return "Statistical"
  elif "detector" in aName.lower():
    return "Detector"
  elif "model" in aName.lower():
    return "Modelling"
  elif "mixed" in aName.lower():
    return "Mixed"
  elif "special" in aName.lower():
    return "Special"
  elif "EffectiveNP_" in aName:
    return "Special" # Global reduction parameters are of type "Special"
  else:
    return "UNKNOWN"
 

◆ getDescFromName()

CombineRootAndConfigFilesAfterReduction.getDescFromName ( aName )

Definition at line 81 of file CombineRootAndConfigFilesAfterReduction.py.

def getDescFromName(aName):
  if "Stat" in aName:
    return "Effective JES Statistical and Method Component %c"%(aName[-1])
  elif "Detector" in aName:
    return "Effective JES Detector Component %c"%(aName[-1])
  elif "Model" in aName:
    return "Effective JES Modelling Component %c"%(aName[-1])
  elif "Mixed" in aName:
    return "Effective JES Mixed Modelling and Detector Component %c"%(aName[-1])
  elif "Special" in aName:
    print "WARNING: Including a special component directly - %s"%(aName)
    return "Effective JES Special Component %c"%(aName[-1])
  elif "UNKNOWN" in aName:
    print "WARNING: Including an unknown component - %s"%(aName)
    return "Effecitve JES Unknown Category Component %c"%(aName[-1])
  elif "EffectiveNP_" in aName:
    if "restTerm" in aName:
      return "Effective JES Uncertainty Component %c"%(aName[-9])
    else:
      return "Effective JES Uncertainty Component %c"%(aName[-1])
  else:
    print "ERROR: Including an unrecognized component"
    exit(1)
 

◆ GetKeyNames()

CombineRootAndConfigFilesAfterReduction.GetKeyNames	(		self,
			dir = "" )

Definition at line 11 of file CombineRootAndConfigFilesAfterReduction.py.

def GetKeyNames(self,dir=""):
  self.cd(dir)
  return [key.GetName() for key in gDirectory.GetListOfKeys()]
 
#Copy everything from the reduced file to the output file
#Copy everything excluding any overlap with the reduced file from the full file to the output file

◆ readSanitizedLine()

CombineRootAndConfigFilesAfterReduction.readSanitizedLine ( file )

Definition at line 66 of file CombineRootAndConfigFilesAfterReduction.py.

def readSanitizedLine(file):
  line = file.readline()
  sanitizedLine = re.sub('\r','',line)
#  line = re.sub('\n','',sanitizedLine)
#  sanitizedLine = line+"\n\r"
  return sanitizedLine 
 

◆ replaceComponentNumber()

CombineRootAndConfigFilesAfterReduction.replaceComponentNumber	(		line,
			newNumber )

Definition at line 138 of file CombineRootAndConfigFilesAfterReduction.py.

def replaceComponentNumber(line,newNumber):
  # Line is in the following form:
  #    JESComponent.#.etc
  # Replace the first part
  return re.sub("JESComponent.[0-9]*.","JESComponent.%d."%(newNumber),line)
 

◆ sanitizeComponentName()

CombineRootAndConfigFilesAfterReduction.sanitizeComponentName ( aName )

Definition at line 73 of file CombineRootAndConfigFilesAfterReduction.py.

def sanitizeComponentName(aName):
  sanitizedName = re.sub(';1','',aName)
  sanitizedName = re.sub('_AntiKt4Topo_EMJES','',sanitizedName)
  sanitizedName = re.sub('_AntiKt6Topo_EMJES','',sanitizedName)
  sanitizedName = re.sub('_AntiKt4Topo_LCJES','',sanitizedName)
  sanitizedName = re.sub('_AntiKt6Topo_LCJES','',sanitizedName)
  return sanitizedName
 