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

def style_histo(h,col):
    h.SetMaximum(250)
    h.SetMarkerStyle(7)
    h.SetMarkerColor(col)
    h.SetLineColor(col)
    h.SetTitle(';PC time of flight (ns); entries (arb. units)')
    return h.Clone()

def make_legend(h1,t1,h2,t2):
    l = TLegend(0.8,0.8,0.9,0.9)
    l.SetBorderSize(1)
    l.AddEntry(h1,t1,'l')
    l.AddEntry(h2,t2,'l')
    return l

c = TCanvas()
c.Divide(1,3)

f1 = TFile('sum84-9-1_sum584-1_Tam.root')

## __________
c.cd(1)

US_data = f1.Get('Data_wintype7plus14_Norm')
US_sim  = f1.Get('MC_wintype7plus14_Norm')

US_data = style_histo(US_data,1)
US_sim  = style_histo(US_sim,2)
    
l1 = make_legend(US_data,'Upstream, data',US_sim,'Upstream, simulation')

US_data.Draw()
US_sim.Draw('same')
l1.Draw()

## __________
c.cd(2)

DS_data = f1.Get('Data_wintype8plus15_Norm')
DS_sim  = f1.Get('MC_wintype8plus15_Norm')

DS_data = style_histo(DS_data,1)
DS_sim = style_histo(DS_sim,2)

l2 = make_legend(DS_data,'Downstream, data',US_sim,'Downstream, simulation')

DS_data.Draw()
DS_sim.Draw('same')
l2.Draw()

## __________
f2 = TFile('sum583-1_sum542-1-1_Tam.root')
c.cd(3)
DS_with_DSBP    = f2.Get('Data_wintype8plus15_Norm')
DS_without_DSBP = f2.Get('MC_wintype8plus15_Norm')

DS_with_DSBP = style_histo(DS_with_DSBP,1)
DS_without_DSBP = style_histo(DS_without_DSBP,2)

l3 = make_legend(DS_with_DSBP,'Sim: beam package IN',DS_without_DSBP,'Sim: beam package OUT')

DS_with_DSBP.Draw()
DS_without_DSBP.Draw('same')
l3.Draw()

au = raw_input('>')