SteerMultipleCompare Namespace Reference

Functions
def	CleanArguments (argv, option)
def	CreateBaseDirectory (options)
def	CreateDirectory (dir, addToExisting=False)
def	CreateSubDirectory (basedir, path)
def	main (argv=None)
def	plotDefault (argv, outputDir, name, type, plots, addArgv=[])
def	plotDefaults (argv, options, outputDir)
def	plotOneByOne (argv, outputDir, histoList, histoSubNames, paths)
def	StripPath (name)

Variables
	__author__
	__doc__

Function Documentation

CleanArguments()

def SteerMultipleCompare.CleanArguments	(		argv,
			option
	)

Definition at line 54 of file SteerMultipleCompare.py.

Referenced by main().

def CleanArguments(argv, option):
  #remove existing output arguments
  while argv.count(option) > 0:
    index = argv.index(option)
    if index < len(argv)-1:
      argv.pop(index+1)#drop the corresponding value
    argv.pop(index)#drop the option itself


#execute Multicompare for each plot as a comparison one by one
#argv was modified to contain only one plot each

CreateBaseDirectory()

def SteerMultipleCompare.CreateBaseDirectory

(

options

)

Definition at line 39 of file SteerMultipleCompare.py.

References CreateDirectory().

Referenced by main().

def CreateBaseDirectory(options):
  if options.out == 'MultipleCompare.png' or options.out.find('.')!=-1:
    #default case, so no directory was given
    #or a filename was given
    outputDirName = 'MultipleCompareOutput'
  else:
    outputDirName = options.out
  outputDir = os.path.join(os.getcwd(), outputDirName)
  CreateDirectory(outputDir)
  return outputDir

CreateDirectory()

def SteerMultipleCompare.CreateDirectory	(		dir,
			addToExisting = False
	)

Definition at line 26 of file SteerMultipleCompare.py.

References print().

Referenced by CreateBaseDirectory(), and CreateSubDirectory().

def CreateDirectory(dir,addToExisting=False):
  if os.path.exists(dir) and not addToExisting:
    print("Output directory %s already exists!  OK to overwrite?" % dir)
    while True:
      input = raw_input("Please enter [y/n] ")
      if (input == 'y'):
        break
      elif (input == 'n'):
        print(" ...exiting.")
        sys.exit()
  if not os.path.exists(dir):
    os.makedirs(dir)

CreateSubDirectory()

def SteerMultipleCompare.CreateSubDirectory	(		basedir,
			path
	)

Definition at line 50 of file SteerMultipleCompare.py.

References CreateDirectory().

Referenced by plotOneByOne().

def CreateSubDirectory(basedir, path):
  outputDir = os.path.join(basedir, path)
  CreateDirectory(outputDir,True)

main()

def SteerMultipleCompare.main

(

argv = None

)

Definition at line 121 of file SteerMultipleCompare.py.

References CleanArguments(), CreateBaseDirectory(), MultipleCompare.LoadCommandlineOptions(), MultipleCompare.MapDirStructure(), MultipleCompare.Match(), plotDefaults(), plotOneByOne(), print(), and StripPath().

def main(argv=None):
  if argv is None:
    argv = sys.argv
  
  options, toPlot = MultipleCompare.LoadCommandlineOptions(argv)
      
  gROOT.SetBatch()

  testFile = TFile(options.test)
  refFile = None
  if options.ref != '':
    refFile = TFile(options.ref)

  plotList = []
  MultipleCompare.MapDirStructure( testFile,'',plotList)

  if len(plotList)<1:
    print('\tError: Please specify at least one histogram. The following ones are available in the root file.')
    print(plotList)
    sys.exit()

  histoList = []
  histoSubNames = []
  paths = []
  massiveMode = False
  for plot in toPlot:
    #clean the arguments. toPlot contains the list of positional arguments leftover after parsing options
    argv.remove(plot)
    for path in plotList:
      if MultipleCompare.Match(plot.lower(),path.lower()):
        histoList.append(path)
        strippedPath, strippedPlot = StripPath(path)
        paths.append(strippedPath)
        histoSubNames.append(strippedPlot)
        #print histoSubNames[-1]
      elif plot.find('massiveMode') != -1:
        massiveMode = True

  CleanArguments(argv,'--output')
  CleanArguments(argv,'-o')
          
  outputDir = CreateBaseDirectory(options)

  if massiveMode:
    print("Massive mode: scan all subdirs and make plots comparing each histogram one by one.")
    plotOneByOne(argv, outputDir, histoList, histoSubNames, paths)          
  else:
    print("Default mode: Make default plot combinations.")
    plotDefaults(argv, options, outputDir)


#only execute main() if manually run

plotDefault()

def SteerMultipleCompare.plotDefault	(		argv,
			outputDir,
			name,
			type,
			plots,
			addArgv = []
	)

Definition at line 75 of file SteerMultipleCompare.py.

References MultipleCompare.main().

Referenced by plotDefaults().

def plotDefault(argv, outputDir, name, type, plots, addArgv=[]):
  tmpArgv = argv[:]
  tmpArgv.append('-o')  
  tmpArgv.append(outputDir+'/'+name+type)
  tmpArgv.extend(addArgv)
  tmpArgv.extend(plots)
  MultipleCompare.main(tmpArgv)

#make some default plots grouping several histograms

plotDefaults()

def SteerMultipleCompare.plotDefaults	(		argv,
			options,
			outputDir
	)

Definition at line 84 of file SteerMultipleCompare.py.

References plotDefault().

Referenced by main().

def plotDefaults(argv, options, outputDir):
  name = 'Validation_'
  if options.testLabel != None:
    name += options.testLabel+'_'
  else:
    name += options.test+'_vs_'
  if options.refLabel != None:
    name += options.refLabel+'_'
  else:
    name += options.ref+'_'
  outputType = '.eps'
  additionalArgv = []
  if outputDir.find('QCD')!=-1:
    additionalArgv.append('-f') #fakerate
  plotDefault(argv, outputDir, name, 'LeptonRejectionEffphi'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*Rejection/*Effphi'], additionalArgv)
  plotDefault(argv, outputDir, name, 'LeptonRejectionEffeta'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*Rejection/*Effeta'], additionalArgv)
  plotDefault(argv, outputDir, name, 'LeptonRejectionEffpt'+outputType,  ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*Rejection/*Effpt'], additionalArgv)

  if outputDir.find('QCD')!=-1:
    additionalArgv.append('--logScale')
  plotDefault(argv, outputDir, name, 'Effphi'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*Effphi', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*Effphi'], additionalArgv)
  plotDefault(argv, outputDir, name, 'Effeta'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*Effeta', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*Effeta'], additionalArgv)
  plotDefault(argv, outputDir, name, 'Effpt'+outputType,  ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*Effpt', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*Effpt'], additionalArgv)

  plotDefault(argv, outputDir, name, 'pTRatio_allHadronic'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_pTRatio_allHadronic', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_pTRatio_allHadronic'])
  plotDefault(argv, outputDir, name, 'pTRatio_oneProng1Pi0'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_pTRatio_oneProng1Pi0', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_pTRatio_oneProng1Pi0'])
  plotDefault(argv, outputDir, name, 'pTRatio_threeProng0Pi0'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_pTRatio_threeProng0Pi0', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_pTRatio_threeProng0Pi0'])

  plotDefault(argv, outputDir, name, 'Size_isolationPFChargedHadrCands'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_Size_isolationPFChargedHadrCands', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_Size_isolationPFChargedHadrCands'])
  plotDefault(argv, outputDir, name, 'Size_isolationPFNeutrHadrCands'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_Size_isolationPFNeutrHadrCands', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_Size_isolationPFNeutrHadrCands'])
  plotDefault(argv, outputDir, name, 'Size_isolationPFGammaCands'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_Size_isolationPFGammaCands', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_Size_isolationPFGammaCands'])

  plotDefault(argv, outputDir, name, 'SumPt_isolationPFChargedHadrCands'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_SumPt_isolationPFChargedHadrCands', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_SumPt_isolationPFChargedHadrCands'])
  plotDefault(argv, outputDir, name, 'SumPt_isolationPFNeutrHadrCands'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_SumPt_isolationPFNeutrHadrCands', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_SumPt_isolationPFNeutrHadrCands'])
  plotDefault(argv, outputDir, name, 'SumPt_isolationPFGammaCands'+outputType, ['DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationByDecayModeFinding/*_SumPt_isolationPFGammaCands', 'DQMData/RecoTauV/hpsPFTauProducer_hpsPFTauDiscriminationBy*CombinedIsolationDBSumPtCorr/*_SumPt_isolationPFGammaCands'])

plotOneByOne()

def SteerMultipleCompare.plotOneByOne	(		argv,
			outputDir,
			histoList,
			histoSubNames,
			paths
	)

Definition at line 65 of file SteerMultipleCompare.py.

References CreateSubDirectory(), MultipleCompare.main(), and ComparisonHelper.zip().

Referenced by main().

def plotOneByOne(argv, outputDir, histoList, histoSubNames, paths):
  for hist, name, path in zip(histoList, histoSubNames, paths):
    CreateSubDirectory(outputDir, path)
    #now give modified arguments to MultipleCompare
    tmpArgv = argv[:]
    tmpArgv.append('-o')
    tmpArgv.append(outputDir+'/'+path+'/'+name+'.png')
    tmpArgv.append(hist)
    MultipleCompare.main(tmpArgv)

StripPath()

def SteerMultipleCompare.StripPath

(

name

)

Definition at line 17 of file SteerMultipleCompare.py.

Referenced by main().

def StripPath(name):
  path = ''
  plot = ''
  matches = re.match(r'(.*)\/(.*)$', name)
  if matches:
    path = matches.group(1)
    plot = matches.group(2)
  return [path, plot]

Variable Documentation

__author__

SteerMultipleCompare.__author__

private

Definition at line 11 of file SteerMultipleCompare.py.

__doc__

SteerMultipleCompare.__doc__

private

Definition at line 12 of file SteerMultipleCompare.py.