#!/usr/bin/python
import os,sys
from accessories import *

from ROOT import *
from clean_canvas import *
from nice_colours import *
gStyle = makegstyle(gStyle)
gStyle = nice_colours(gStyle)

#PARK: why does Alex use +1 in uss.py

#     ct_min,ct_max,pmax, pt_min, plong_min, pt_max
#fid = [0.54, 0.96,  52.0, 10.0,   14.0,      38.0] ##intermediate fiducial
fid = [0.50, 0.92,  51.5, 10.0,   13.7,      39.7] ##robs fiducial

fn = '/home/rbayes/scripts/upstream_stops/upstreamAnal/test/positron_interations/inefficency/g689a1s1_raw_s89a1s1_rawUSineff.root'
#fn = '/home/rbayes/scripts/upstream_stops/upstreamAnal/test/positron_interations/inefficency/g689a1s1_raw_s89a1s1_rawDSineff.root'
hn_data = 'effhist_s89a1s1_raw'
hn_sim  = 'effhist_g689a1s1_raw'

f = TFile(fn)
dat = f.Get(hn_data)
sim = f.Get(hn_sim)
dif = add_histos(dat,sim,-1.)

dat.SetTitle('DATA')
sim.SetTitle('SIM')
dif.SetTitle('DIFFERENCE')

dat_fid = fid_only(dat,fid,999.,999.)
sim_fid = fid_only(sim,fid,999.,999.)
dif_fid = fid_only(dif,fid,999.,999.)

dat_fid_p = fid_only(dat,fid,24.999,28.001)
sim_fid_p = fid_only(sim,fid,24.999,28.001)
dif_fid_p = fid_only(dif,fid,24.999,28.001)
#dat_fid_p = fid_only(dat,fid)
#sim_fid_p = fid_only(sim,fid)
#dif_fid_p = fid_only(dif,fid)

c1 = TCanvas('c1','c1')
c1.Divide(3,3)
for ci in [c1_1,c1_2,c1_3,c1_4,c1_5,c1_6,c1_7,c1_8,c1_9]:
    ci.SetLogz()
    ci.SetRightMargin(.15)

hs = [dat,sim,dif,dat_fid,sim_fid,dif_fid,dat_fid_p,sim_fid_p,dif_fid_p]
for ih,h in enumerate(hs):
    c1.cd(ih+1)
    h.Draw('colz')

c2 = TCanvas('c2','c2')
c2.Divide(2,2)
c2.cd(1)
pp1 = project_vs_p(dat_fid_p)
pp2 = project_vs_p(sim_fid_p)
pp3 = project_vs_p(dif_fid_p)
##
pp1.SetMarkerColor(2)
pp2.SetMarkerColor(4)
pp3.SetMarkerColor(8)
pp1.Draw('p')
pp2.Draw('samep')
c2.cd(3)
pp3.Draw('p')
#pp3.Fit('pol1')

c2.cd(2)
pp1.Draw('p')
pp2.Draw('samep')
c2.cd(4)
pp3.Draw('p')

c3 = TCanvas('c3','c3')
c3.Divide(2,2)
c3.cd(1)
pct1 = project_vs_ct(dat_fid_p)
pct2 = project_vs_ct(sim_fid_p)
pct3 = project_vs_ct(dif_fid_p)
##
pct1.SetMarkerColor(2)
pct2.SetMarkerColor(4)
pct3.SetMarkerColor(8)
pct1.Draw('p')
pct2.Draw('samep')
c3.cd(3)
pct3.Draw('p')
pct3.Fit('pol1')

c3.cd(2)
pct1.Draw('p')
pct2.Draw('samep')
c3.cd(4)
pct3.Draw('p')

c4 = TCanvas('c4','c4')
c4.Divide(2,1)
c4.cd(1)
print 'data:'
ineff_dat = simple_histo(dat_fid_p)
ineff_dat.Draw()
c4.cd(2)
print 'sim:'
ineff_sim = simple_histo(sim_fid_p)
ineff_sim.Draw()

au = raw_input('>')

#PARK: can't confirm that pp1-pp2=pp3
#      p cut clearly wrong
#      need error bars