from project_histos import Cuts, filter_alnum, Sample
from make_plots import plot_hists, canvas_margin, legend
import argparse
from collections import OrderedDict as dict
if __name__=="__main__":
parser = argparse.ArgumentParser()
parser.add_argument('-i', '--infile', type=str, required=True)
parser.add_argument('--normalize_lumi', action="store_true", default=True)
args = parser.parse_args()
#fname = "/Users/joosep/Documents/stpol/data/step3_trees_Mar28/T_t.root"
sample = Sample.fromFile(args.infile)
# ev_ids = dict()
# for ev in sample.tree:
# evid = int(ev.event_id)
# if evid not in ev_ids.keys():
# ev_ids[evid] = 1
# else:
# ev_ids[evid] += 1
cuts = dict()
cuts["MT_mu"] = Cuts.mt_mu
cuts["2J"] = Cuts.n_jets(2)
cuts["1T"] = Cuts.n_tags(1)
#cuts["eta_lj"] = Cuts.eta_lj
cuts["rms_lj"] = Cuts.rms_lj
cuts["eta_jet"] = Cuts.eta_jet
cuts["top_mass_sig"] = Cuts.top_mass_sig
import ROOT
from project_histos import Sample, Cuts, Cut
import argparse
from common.odict import OrderedDict as dict
#from make_plots import plot_hists_stacked, Styling, merge_hists_g, g_merge_cmd, legend
import copy
from common.utils import merge_cmds, merge_hists
if __name__=="__main__":
parser = argparse.ArgumentParser()
parser.add_argument("-t", "--type", type=str, default="iso")
parser.add_argument("-d", "--dir", type=str, default="/Users/joosep/Documents/stpol/data/step3_trees_Apr04")
lumi_total = 18062
args = parser.parse_args()
samples_mc = Sample.fromDirectory(args.dir + "/" + args.type + "/mc")
samples_data = Sample.fromDirectory(args.dir + "/" + args.type + "/data")
# def compare_plot(var, plot_range, weight, cut, **kwargs):
# histsD = dict()
#
# for samp in samples_mc:
# histsD[samp.name] = samp.drawHistogram(var, str(cut), weight=weight, plot_range=plot_range)
#
# for name, hist in histsD.items():
# hist.normalize_lumi(lumi_total)
#
# for samp in samples_data:
# histsD[samp.name] = samp.drawHistogram(var, str(cut), plot_range=plot_range)
#
#
from project_histos import Cuts, filter_alnum, Sample
from make_plots import plot_hists, canvas_margin, legend
import argparse
if __name__=="__main__":
parser = argparse.ArgumentParser()
args = parser.parse_args()
#fname = "/Users/joosep/Documents/stpol/data/step3_trees_Mar28/T_t.root"
fnames = [
"/Users/joosep/Documents/stpol/data/T_t.root",
"/Users/joosep/Documents/stpol/data/step3_trees_Apr04/iso/mc/T_t.root",
"/Users/joosep/Documents/stpol/data/step3_trees_Apr05_noPuClean/iso/mc/T_t.root",
]
samples = [Sample.fromFile(f) for f in fnames]
sample_names = ["Apr16 (fixed norm)", "Apr04", "Apr05 (no PU cl.)"]
samps = zip(samples, sample_names)
cut = Cuts.mt_mu*Cuts.n_jets(2)*Cuts.n_tags(1)*Cuts.eta_lj*Cuts.top_mass_sig
hists = []
for sample, name in samps:
hist = sample.drawHistogram("cos_theta", cut.cut_str, plot_range=[40, -1, 1])
#normalize to 20/fb
hist.normalize_lumi(20000)
hist.hist.SetName(name)
hist.hist.SetTitle("%s: %.3E" % (name, hist.hist.Integral()))
hist.update()
from project_histos import Cuts, filter_alnum, Sample
from make_plots import plot_hists, canvas_margin, legend
import argparse
from collections import OrderedDict as dict
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("-i", "--infile", type=str, required=True)
parser.add_argument("--normalize_lumi", action="store_true", default=True)
args = parser.parse_args()
# fname = "/Users/joosep/Documents/stpol/data/step3_trees_Mar28/T_t.root"
sample = Sample.fromFile(args.infile)
# ev_ids = dict()
# for ev in sample.tree:
# evid = int(ev.event_id)
# if evid not in ev_ids.keys():
# ev_ids[evid] = 1
# else:
# ev_ids[evid] += 1
cuts = dict()
cuts["MT_mu"] = Cuts.mt_mu
cuts["2J"] = Cuts.n_jets(2)
cuts["1T"] = Cuts.n_tags(1)
# cuts["eta_lj"] = Cuts.eta_lj
cuts["rms_lj"] = Cuts.rms_lj
cuts["eta_jet"] = Cuts.eta_jet
cuts["top_mass_sig"] = Cuts.top_mass_sig
import ROOT
from project_histos import Sample, Cuts, Cut
import argparse
from collections import OrderedDict as dict
from make_plots import plot_hists_stacked, Styling, merge_hists_g, g_merge_cmd, legend, plot_hists
import copy
if __name__=="__main__":
parser = argparse.ArgumentParser()
parser.add_argument("-d", "--dir", type=str, default="/Users/joosep/Documents/stpol/data/step3_trees_Apr04")
args = parser.parse_args()
samples_mc_iso = Sample.fromDirectory(args.dir + "/iso/mc")
samples_data_iso = Sample.fromDirectory(args.dir + "/iso/data")
samples_mc_aiso = Sample.fromDirectory(args.dir + "/anti-iso/mc")
samples_data_aiso = Sample.fromDirectory(args.dir + "/anti-iso/data")
samplesD_iso = dict()
for sample in samples_mc_iso + samples_data_iso:
samplesD_iso[sample.name] = sample
samplesD_aiso = dict()
for sample in samples_mc_aiso + samples_data_aiso:
samplesD_aiso[sample.name] = sample
def qcd_comp_plot(var, cut, plot_range, **kwargs):
lumi = 6471
hist_qcd_iso = samplesD_iso["QCDMu"].drawHistogram(var, str(cut), plot_range=plot_range)
hist_qcd_iso.pretty_name = "simulation: %d ev." % hist_qcd_iso.hist.Integral()
print "MC qcd events: %d" % hist_qcd_iso.hist.Integral()
hist_qcd_iso.normalize_lumi(lumi)
hist_data_aiso = samplesD_aiso["SingleMuD"].drawHistogram(var, str(cut), plot_range=plot_range)
if __name__ == "__main__":
from common.tdrstyle import tdrstyle
tdrstyle()
parser = argparse.ArgumentParser()
parser.add_argument("-t", "--type", type=str, default="iso")
parser.add_argument(
"-d",
"--dir",
type=str,
default=
"/Users/joosep/Documents/stpol/data/05_02/out_step3_05_01_22_36/")
lumi_total = 19300
args = parser.parse_args()
samples_mc = Sample.fromDirectory(args.dir + "/" + args.type + "/mc")
samples_data = Sample.fromDirectory(args.dir + "/" + args.type + "/data")
def compare_plot(var, plot_range, weight, cut, **kwargs):
histsD = dict()
for samp in samples_mc:
histsD[samp.name] = samp.drawHistogram(var,
str(cut),
weight=weight,
plot_range=plot_range)
Styling.mc_style(histsD[samp.name].hist, samp.name)
for name, hist in histsD.items():
hist.normalize_lumi(lumi_total)
