#!/usr/bin/python
import os,sys
from ROOT import *
from clean_canvas import *
from nice_colours import *
gStyle = makegstyle(gStyle)
gStyle = nice_colours(gStyle)

data_fn = '/home/e614/analysis_2006_2007/treesum/data/set84anal5/ecal_edge_set84a5s1_jb.root'
sim_fn  = '/home/e614/analysis_2006_2007/treesum/mc/gen501anal1_2/ecal_edge_gen501a1_2s2.root'

data_f = TFile(data_fn)
sim_f  = TFile(sim_fn)

c1 = TCanvas()

##c1.Divide(2,2)
##bins = ['051','130','231','310']
##for i,b in enumerate(bins):
##    c1.cd(i+1)
##    h_data = data_f.Get('best_iteration/slices/bin'+b)
##    h_sim  = sim_f.Get('best_iteration/slices/bin'+b)
##    h_data.SetLineColor(2)
##    h_sim.SetLineColor(4)
##    h_data.SetMaximum(400.)
##    h_data.GetXaxis().SetRangeUser(51.9,53.1)
##    h_data.Draw()
##    h_sim.Draw('same')

h_data = data_f.Get('best_iteration/slices/bin130')
h_sim  = sim_f.Get('best_iteration/slices/bin130')
h_data.SetLineColor(2)
h_sim.SetLineColor(4)
h_sim.SetLineStyle(7)

print 'data integral=',h_data.Integral()
print 'sim  integral=',h_sim.Integral()
h_sim.Scale(h_data.Integral()/h_sim.Integral()) #new: rescale

h_data.SetMaximum(400.)
h_data.GetXaxis().SetRangeUser(51.9,53.1)

h_data.SetTitle('115#circ < #theta < 115.5#circ; '
                +'reconstructed momentum (MeV/c); ')

h_data.Draw()
h_sim.Draw('same')

l = TLegend(.8,.8,.9,.9)
l.SetBorderSize(1)
l.AddEntry(h_data,'data','l')
l.AddEntry(h_sim,'simulation','l')
l.Draw()

au = raw_input('>')