ROOT.gROOT.ProcessLine('.L VectorTLorentzVector_h.so') group = sys.argv[1] #group = 'ZH50' #group = 'ZH' #group = 'ZH_inclusive' #group = 'ZH_MG_012j' #group = 'ZH_MG_0j' if 'MG' in group: clone = True else: clone = False collection = parse.samples('../data/samples_highstat.cfg') #collection = parse.samples('../data/samples.cfg') outfile = ROOT.TFile('../data/tree_%s.root'%group,'RECREATE') outfile.cd() mytree = ROOT.TTree('mytree','mytree') # Higgs branches H = ROOT.std.vector(ROOT.TLorentzVector)() mytree.Branch( "H", "vector<TLorentzVector>", H) genH = ROOT.std.vector(ROOT.TLorentzVector)() mytree.Branch( "genH", "vector<TLorentzVector>", genH) h_dau =ROOT.std.vector(ROOT.TLorentzVector)() mytree.Branch( "h_dau", "vector<TLorentzVector>", h_dau) genh_dau =ROOT.std.vector(ROOT.TLorentzVector)()
"aMC@NLO + Herwig", "Madgraph MLM(0-2j) + Pythia", "Madgraph + Pythia", "Powheg + Herwig", "Sherpa (AMEGIC+OpenLoops)", ] colors = [98, 98, 9, 9, 8, 15] styles = [1, 2, 1, 2, 1, 1] norm = [0.0478840946816, 0.0497891290506, 0.0140102168962, 0.022549493317, 19531.0, 99414.0] # Powheg PU: 19531.0 # Powheg: 33292.0 outfile = ROOT.TFile("test.root") vars = parse.samples(opts.conf) def fix_pdf(filename): os.system("pdf2ps %s %sps" % (filename, filename.rstrip("pdf"))) os.system("ps2pdf %sps %s" % (filename.rstrip("pdf"), filename)) os.system("rm %sps" % filename.rstrip("pdf")) os.system('pdfcrop %s %s --margins "20 20 20 20"' % (filename, filename)) def make_legend(): l = ROOT.TLegend(0.59, 0.67, 0.92, 0.88) l.SetLineWidth(2) l.SetBorderSize(0) l.SetFillColor(0) l.SetFillStyle(4000)