parser = UserInput.getDefaultParser()

parser.add_argument("-s", "--selection", type=str, required=True,

help="Specificy selection level to run over")

parser.add_argument("-b", "--branches", type=str, default="all",

help="List (separate by commas) of names of branches "

"in root and config file to plot")

parser.add_argument("--systematics", type=str, required=True,

help="Specificy systematics to plot")

if rebin:

if len(rebin) == 1:

hist.Rebin(int(rebin[0]))

else:

bins = array.array('d', rebin)

hist = hist.Rebin(len(bins)-1, "", bins)

args = getComLineArgs()

ROOT.gROOT.SetBatch(True)

branches = [x.strip() for x in args.branches.split(",")]

(plot_path, html_path) = helper.getPlotPaths(args.selection, args.folder_name, True)

meta_info = '-'*80 + '\n' + \

'Script called at %s\n' % datetime.datetime.now() + \

'The command was: %s\n' % ' '.join(sys.argv) + \

'-'*80 + '\n'

rtfile = ROOT.TFile(args.hist_file)

for branch in args.branches.split(","):

with open("temp.txt", "w") as mc_file:

mc_file.write(meta_info)

mc_file.write("Selection: %s" % args.selection)

mc_file.write("\nPlotting branch: %s\n" % branch)

systematics = args.systematics.split(",")

systs = "_".join(systematics)

plot_name = "_".join([branch, systs]) if args.append_to_name == "" \

else "_".join([branch, systs, args.append_to_name])

hist_stack = ROOT.THStack("stack_"+branch, "stack_"+branch)

file_names = args.files_to_plot.split(",")

for i, file_name in enumerate(file_names):

systematic = systematics[i]

central_hist = 0

for chan in args.channels.split(","):

hist_name = file_name.replace("_standalone", "") + "/"+ "_".join([branch, chan])

uphist_name = hist_name.replace(chan, "_".join([systematic+"Up", chan]))

downhist_name = hist_name.replace(chan, "_".join([systematic+"Down", chan]))

if not central_hist:

central_hist = rtfile.Get(hist_name)

up_hist = rtfile.Get(uphist_name)

down_hist = rtfile.Get(downhist_name)

central_hist = rebinHist(central_hist, args.rebin)

up_hist = rebinHist(up_hist, args.rebin)

down_hist = rebinHist(down_hist, args.rebin)

central_hist.Draw()

else:

if i == 0 or file_name != file_names[i-1]:

central_hist_chan = rtfile.Get(hist_name)

central_hist_chan = rebinHist(central_hist_chan, args.rebin)

central_hist.Add(central_hist_chan)

up_hist_chan = rtfile.Get(uphist_name)

down_hist_chan = rtfile.Get(downhist_name)

up_hist_chan = rebinHist(up_hist_chan, args.rebin)

down_hist_chan = rebinHist(down_hist_chan, args.rebin)

up_hist.Add(up_hist_chan)

down_hist.Add(down_hist_chan)

path = "/cms/uhussain" if "hep.wisc.edu" in os.environ['HOSTNAME'] else \

"/afs/cern.ch/user/u/uhussain/work"

config_factory = ConfigHistFactory(

"%s/ZZ4lAnalysisDatasetManager" % path,

args.selection.split("_")[0],

)

scale_fac = 1

scale = True

# Normalize to data in CR

if file_name in ["wz", "wz_standalone"]:

scale_fac = 0.8224

elif file_name in ["wz-powheg", "wz-powheg_standalone"]:

scale_fac =(191-(28.78+3.84+18.35+22.27))/139.43

if scale:

central_hist.Scale(scale_fac)

up_hist.Scale(scale_fac)

down_hist.Scale(scale_fac)

with open("temp.txt", "a") as mc_file:

mc_file.write("\nYield for %s is %0.2f" % (file_name, central_hist.Integral()))

branch_name = branch.replace("_Fakes", "")

config_factory.setHistAttributes(central_hist, branch_name, file_name)

config_factory.setHistAttributes(up_hist, branch_name, file_name)

config_factory.setHistAttributes(down_hist, branch_name, file_name)

central_hist.SetMinimum(0.001)

central_hist.SetFillColor(0)

up_hist.SetFillColor(0)

down_hist.SetFillColor(0)

central_hist.SetLineStyle(1)

central_hist.SetLineWidth(2)

up_hist.SetLineWidth(2)

down_hist.SetLineWidth(2)

up_hist.SetLineStyle(5)

down_hist.SetLineStyle(5)

#down_hist.SetLineColor(ROOT.TColor.GetColor("#980000"))

if "wzQCDModeling" in up_hist.GetName():

print "Oui oui"

up_hist.SetLineColor(ROOT.TColor.GetColor("#980000"))

down_hist.SetLineColor(ROOT.TColor.GetColor("#980000"))

#elif "wz-mgmlm_scale" in up_hist.GetName():

# up_hist.SetLineColor(ROOT.TColor.GetColor("#016300"))

hist_stack.Add(central_hist)

hist_stack.Add(up_hist)

hist_stack.Add(down_hist)

canvas_dimensions = [800, 800] if "unrolled" not in branch else [1200, 800]

canvas = ROOT.TCanvas("canvas", "canvas", *canvas_dimensions)

hist_stack.Draw("nostack hist")

hist_stack.SetMinimum(central_hist.GetMinimum()*args.scaleymin)

hist_stack.SetMaximum(central_hist.GetMaximum()*args.scaleymax)

hist_stack.GetHistogram().GetYaxis().SetTitle("Events / bin")

hist_stack.GetHistogram().GetYaxis().SetTitleOffset(1.05)

if not args.no_decorations:

ROOT.dotrootImport('uhussain/CMSPlotDecorations')

scale_label = "Normalized to Unity" if args.luminosity < 0 else \

"%0.1f fb^{-1}" % args.luminosity

lumi_text = ""

if args.thesis:

lumi_text = "Thesis"

elif args.preliminary:

lumi_text = "Preliminary"

if args.simulation:

lumi_text += "Simulation"

ROOT.CMSlumi(canvas, 0, 11, "%s (13 TeV)" % scale_label,lumi_text)

#"Preliminary Simulation" if args.simulation else "Preliminary")

text_box = ROOT.TPaveText(0.65, 0.75-0.1*len(systematics), 0.9, 0.85, "NDCnb")

text_box.SetFillColor(0)

text_box.SetTextFont(42)

text_box.AddText("Systematic variation")

for s in systematics:

text_box.AddText(s)

text_box.AddText("; ".join(file_names))

for line, h in zip(text_box.GetListOfLines()[1:], hist_stack.GetHists()[1::3]):

line.SetTextColor(h.GetLineColor())

text_box.Draw()

if args.logy:

canvas.SetLogy()

if not args.no_ratio:

canvas = plotter.splitCanvas(canvas, canvas_dimensions,

"syst./cent.",

[float(i) for i in args.ratio_range]

)

ratioPad = canvas.FindObject("ratioPad")

hist = ratioPad.GetListOfPrimitives().FindObject("canvas_central_ratioHist")

hist.GetXaxis().SetLabelSize(0.175)

hist.GetXaxis().SetLabelOffset(0.03)

hist.GetXaxis().SetTitleOffset(1.15)

canvas.Update()

helper.savePlot(canvas, plot_path, html_path, plot_name, True, args)