drawinteraction.py

Go to the documentation of this file.

import ROOT

def DrawTogether(rootfile,directory,urange):
    hists=[]
    histogram=["h_aveint_vs_enu_tot","h_aveint_vs_enu_thin_pCpion","h_aveint_vs_enu_thin_pCkaon","h_aveint_vs_enu_thin_nCpion","h_aveint_vs_enu_thin_pCnucleon","h_aveint_vs_enu_thin_mesoninc","h_aveint_vs_enu_thin_nucleona","h_aveint_vs_enu_others"]
    legends=["Total HP","pC-->#pi X","pC-->KX","nC-->#pi X","pC-->nucleonX","meson inc.","nucloen-A","others"]
    color=[1,4,4,42,42,28,28,39]
    style=[1,1,2,1,2,1,2,1]
    histo_universe=[]
    for j in range(0,len(histogram)):
        in_dir = directory+"/"+histogram[j]
        histo_universe.append(rootfile.Get(str(in_dir)))
    c1 = ROOT.TCanvas("c1","c1",750,750)
    c1.cd()
    leg = ROOT.TLegend(0.3,0.3,0.5,0.9)
    histo_universe[0].SetLineWidth(3)
    histo_universe[0].SetLineColor(color[0])
    histo_universe[0].SetLineStyle(style[0])
    leg.AddEntry(histo_universe[0],str(legends[0]),"lp")
    #histo_universe[0].Scale(1/100)
    if urange:
        (histo_universe[0].GetXaxis()).SetRangeUser(0,20)
        histo_universe[0].SetMinimum(0)
        histo_universe[0].SetMaximum(220)
    histo_universe[0].Draw("hist")
    for i in range(1,len(histo_universe)):
        histo_universe[i].SetLineWidth(3)
        histo_universe[i].SetLineColor(color[i])
        histo_universe[i].SetLineStyle(style[i])
        leg.AddEntry(histo_universe[i],str(legends[i]),"lp")
        #histo_universe[i].Scale(1/100)
        histo_universe[i].Draw("Same")
    leg.Draw("")
    raw_input()
        
r_file = ROOT.TFile("output_interactionmap_opt.root")
DrawTogether(r_file,"Projectile",True)