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)