#!/usr/bin/python
import os,sys

# __________ check usage __________
if len(sys.argv)!=2:
    print 'usage: make_ts_ts_plot.py   treesum_output.root'
    sys.exit(-1)
fn = sys.argv[1]
if not os.path.isfile(fn):
    print 'ERROR: cannot find',fn
    sys.exit(-1)
from ROOT import *
from clean_canvas import *
gStyle = makegstyle(gStyle)

# __________ read in inputs __________
f = TFile(sys.argv[1])
lls = file('inputs.txt','r').readlines()
c1 = TCanvas('c1','c1')
c1.Divide(2,6)
i=0
for l in lls:
    if l.startswith('#'):
        continue
    else:
        i+=1
##        spec = None
        ls = l.split()
        h_name = ls[0]+'/'+ls[1]
        h_before = f.Get(h_name+'_before').Clone()
        h_after  = f.Get(h_name+'_after').Clone()
        if ls[2] == 'normal':
            pass
        elif ls[2] == 'projx':
            h_before = h_before.ProjectionX().Clone()
            h_after  = h_after.ProjectionX().Clone()
        elif ls[2] == 'projy':
            h_before = h_before.ProjectionY().Clone()
            h_after  = h_after.ProjectionY().Clone()
##        elif ls[3] == 'spec':
##            spec = 'colour'
        else:
            print 'ERROR: did not understand',ls[2]
            sys.exit(-1)
        h_before.SetTitle(ls[3].replace('_',' ')+';'+ls[4].replace('_',' ')+'; events')
        h_before.GetXaxis().SetRangeUser(float(ls[5]),float(ls[6]))
        h_before.SetMaximum(1.2*h_before.GetMaximum())

        h_before.GetXaxis().SetLabelSize(0.15)
        h_before.GetYaxis().SetLabelSize(0.15)

        h_after.SetFillColor(2)
        h_after.SetFillStyle(3004)
        h_after.SetLineWidth(0)
        c1.cd(i)
        h_before.Draw()
        h_after.Draw('same')

#for i,(h_before,h_after) in enumerate(hs):
#    print 'i=',i
#    c1.cd(i+1)
#    h_before.Clone().Draw()
#    h_after.Clone().Draw('same')
#c1.Divide(2,6)
au = raw_input('>')