Ejemplo n.º 1
0
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)()
Ejemplo n.º 2
0
    "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)