Gabriele Cosmo
89 lines
2.0 KiB
Python
Executable File
89 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import ROOT
|
|
from array import array
|
|
|
|
def plot_1_file (file):
|
|
ROOT.gROOT.Reset()
|
|
ROOT.gROOT.SetBatch(True)
|
|
|
|
input_file_1=ROOT.TFile(file+'.root','READ')
|
|
h11 = input_file_1.Get('histo/h1.1')
|
|
h12 = input_file_1.Get('histo/h1.2')
|
|
h13 = input_file_1.Get('histo/h1.3')
|
|
h14 = input_file_1.Get('histo/h1.4')
|
|
h21 = input_file_1.Get('histo/h2.1')
|
|
h22 = input_file_1.Get('histo/h2.2')
|
|
h23 = input_file_1.Get('histo/h2.3')
|
|
h24 = input_file_1.Get('histo/h2.4')
|
|
|
|
c1 = ROOT.TCanvas('c1', file+'.h1', 200, 10, 1200, 700)
|
|
c1.Divide(4,2)
|
|
histos1 = [h11, h12, h13, h14]
|
|
pad = 1
|
|
for h1 in histos1:
|
|
if h1:
|
|
c1.cd(pad)
|
|
h1.Draw()
|
|
pad = pad + 1
|
|
|
|
histos2 = [h21, h22, h23, h24]
|
|
pad = 5
|
|
for h2 in histos2:
|
|
if h2:
|
|
c1.cd(pad)
|
|
h2.Draw()
|
|
pad = pad + 1
|
|
c1.Print(file+'.png')
|
|
|
|
input_file_1.Close()
|
|
|
|
def plot_2_files (file):
|
|
ROOT.gROOT.Reset()
|
|
ROOT.gROOT.SetBatch(True)
|
|
|
|
input_file_1=ROOT.TFile(file+"a.root",'READ')
|
|
input_file_2=ROOT.TFile(file+"b.root",'READ')
|
|
|
|
input_file_1.cd()
|
|
h_1_1 = input_file_1.Get('histo/h1.1')
|
|
|
|
c1 = ROOT.TCanvas('c1', file, 200, 10, 700, 500)
|
|
c1.SetGridx()
|
|
c1.SetGridy()
|
|
c1.SetLogx()
|
|
c1.SetLogy()
|
|
|
|
#histogram for energy spectra
|
|
n = 41
|
|
bin = array( 'f' )
|
|
|
|
for i in range( n ):
|
|
bin.append(pow(10,(-2+0.1*i)))
|
|
|
|
h_1 = ROOT.TH1F('unbiased','Source Spectrum',40,bin)
|
|
h_2 = ROOT.TH1F('biased','Source Spectrum',40,bin)
|
|
|
|
input_file_1.cd()
|
|
#get the tuple t1
|
|
t1 = input_file_1.Get('ntuple/101')
|
|
for i in range(t1.GetEntries()):
|
|
t1.GetEntry(i)
|
|
h_1.Fill(t1.Ekin,t1.weight)
|
|
|
|
input_file_2.cd()
|
|
# get the tuple t1
|
|
t2 = input_file_2.Get('ntuple/101')
|
|
for i in range(t2.GetEntries()):
|
|
t2.GetEntry(i)
|
|
h_2.Fill(t2.Ekin,t2.weight)
|
|
|
|
h_2.Draw();
|
|
h_1.Draw("same") ;
|
|
c1.Update()
|
|
c1.Print(file+".png")
|
|
|
|
input_file_1.Close()
|
|
input_file_2.Close()
|
|
|