#   -1:     scale to background Integral
#   float:  scale with float value
#   False:  dont scale

# additional variables to plot
additional_variables = []

ignoredVariables = [
    "Weight_XS",
    "Weight_GEN_nom",
]

# initialize plotter
plotter = variablePlotter(output_dir=plot_dir,
                          variable_set=None,
                          add_vars=additional_variables,
                          max_entries=opts.max_events,
                          plotOptions=plotOptions)

# add samples
plotter.addSample(sampleName="ttZ(qq)",
                  sampleFile=data_dir + "/ttZqq.h5",
                  signalSample=False,
                  plotColor=ROOT.kYellow,
                  apply_cut=True)

#plotter.addSample(
#    sampleName      = "ttZ(ll)",
#    sampleFile      = data_dir+"/ttZll.h5",
#    signalSample    = True,
#    plotColor       = ROOT.kBlack,
    #"KSscore":      True,
}

# additional variables to plot
additional_variables = variable_set.plot_variables
additional_variables.append("TTXmatcher_chi2")

# ignored variables
ignored_variables = [
    "Weight_XS", "Weight_GEN_nom", "Weight_CSV", "class_label"
]

# initialize plotter
plotter = variablePlotter(output_dir=plot_dir,
                          variable_set=variable_set,
                          add_vars=additional_variables,
                          ignored_vars=ignored_variables,
                          plotOptions=plotOptions)

# add samples
plotter.addSample(
    sampleName="ttH",
    sampleFile=data_dir + "/ttHbb_dnn.h5",
    plotColor=ROOT.kBlue + 1,
    XSscaling=2.,  # due to even/odd splitting
    signalSample=True)
'''
plotter.addSample(
    sampleName      = "ttZ",
    sampleFile      = data_dir+"/ttZbb_dnn.h5",
    plotColor       = ROOT.kGreen+1,
    "ratio": False,
    "logscale": False,
    "scaleSignal": -1,
    "lumiScale": 1
}
#   scaleSignal:
#   -1:     scale to background Integral
#   float:  scale with float value
#   False:  dont scale

# additional variables to plot
additional_variables = []

# initialize plotter
plotter = variablePlotter(output_dir=plot_dir,
                          variable_set=None,
                          add_vars=additional_variables,
                          plotOptions=plotOptions)

# add samples
plotter.addSample(sampleName="ttH",
                  sampleFile=data_dir + "/ttH.h5",
                  signalSample=False,
                  plotColor=ROOT.kYellow,
                  apply_cut=False)

plotter.addSample(sampleName="ttZ",
                  sampleFile=data_dir + "/ttZ.h5",
                  signalSample=True,
                  plotColor=ROOT.kBlack,
                  apply_cut=False)