makeSigFits.py

import os
from time import sleep
from subprocess import call
import shlex
from optparse import OptionParser
from getMasses import getMasses
parser = OptionParser()
parser.add_option("-c", "--category"  ,       dest="category",
                  help = "either 'btag' or 'antibtag'"              )
parser.add_option("-s", "--step"  ,       dest="step",
                  help = "either 'fullsim' or 'interpolated'"       )
parser.add_option("-i", "--massIndex"  , type=int,  dest="massIndex",
                  help = "either 'fullsim' or 'interpolated'"       )
(options, args) = parser.parse_args()

if not options.category in ["antibtag", "btag"]:
  print "invalid category given, must be either 'btag' or 'antibtag'."
  exit(1)

cmsenv = os.environ

dirName = "signalFitsHack650_%s_%s" % (options.category, options.step)
if not os.path.exists(dirName):
  os.makedirs(dirName)
masses = []
inputDir = ""
if options.step == "interpolated":
  masses=getMasses("all")
  inputDir = "GenSignal"
elif options.step == "fullsim":
  masses=getMasses("fullsim")
  inputDir = "../vgHists_600"
else:
  print "please use the -s option and pick step 'interpolated' or 'fullsim'"
  exit(1)

if options.massIndex > len(masses) or options.massIndex < 0:
  print "invalid mass index: it must be equal to or greater than 0 and less than %i" % len(masses)
  exit(1)

mass=masses[options.massIndex]
incantation = shlex.split("root -x -b -l -q 'Display_SignalFits.cc(\"%s\",\"%s\",\"\",\"histos_sig_m\",%i,1,%s%s)'" % 
                           (options.category, inputDir, mass, '"'+dirName+'"', ',"interpolated"' if options.step == "interpolated" else "")
                         )
print incantation
call(incantation, env=cmsenv)
sleep(.2)

mass += 10