#!/usr/bin/python

import sys, glob
from ROOT import *

def makegstyle(cStyle):
    cStyle.SetPadBorderMode(0)
    cStyle.SetOptStat(0)           ## Hide stats
    cStyle.SetOptFit(0)          ## Display histogram titles.
    cStyle.SetOptTitle(0)
    cStyle.SetCanvasColor(10)      ## Background color (white)
    cStyle.SetPadColor(10)         ## Background color (white)
    cStyle.SetFillColor(10)        ## Background color (white)
    cStyle.SetTitleFillColor(10)   ## Background color (white)
    cStyle.SetStatColor(10)        ## Background color (white)
    cStyle.SetPalette(1)           ## Nicer colour scale for 2D histograms.
    cStyle.SetTitleBorderSize(0)   ##duh
    cStyle.SetTitleFont(132)
    cStyle.SetTitleFont(132,'X')
    cStyle.SetTitleFont(132,'Y')
    cStyle.SetTitleFont(132,'Z')
    cStyle.SetTitleFont(132,'T')
    cStyle.SetLabelFont(132)
    cStyle.SetLabelFont(132,'X')
    cStyle.SetLabelFont(132,'Y')
    cStyle.SetLabelFont(132,'Z')
    cStyle.SetLabelFont(132,'T')
    cStyle.SetTextFont(132)
    cStyle.SetStatFont(132)
    return cStyle


def plot_edge(file,pname,gStyle):
    f = TFile(file)
    g = f.Get('SliceFit_MPcorr/edge')
    gStyle = makegstyle(gStyle)
    c = TCanvas('c','edge',800,300)
    c.SetBottomMargin(0.15)
    c.SetLeftMargin(0.15)

    g.GetHistogram().SetLabelSize(0.07)
    g.GetHistogram().SetLabelSize(0.07,'X')
    g.GetHistogram().SetLabelSize(0.07,'Y')
    g.GetHistogram().SetTitleSize(0.07)
    g.GetHistogram().SetTitleSize(0.07,'X')
    g.GetHistogram().SetTitleSize(0.07,'Y')
    g.GetHistogram().SetTitleOffset(1)
    g.GetHistogram().SetTitleOffset(1,'X')
    g.GetHistogram().SetTitleOffset(1,'Y')
    
    g.Draw('ap')

    c.Print(pname)

def sequence_inputfiles(dir):
    files = glob.glob(dir + '/sum*_ecmc.root')
    files.sort()

    for f in files:
        print f
        plot_edge(f,f[len(dir) + 1:-10] + '_ec.eps',gStyle)

if __name__ == "__main__":
    sequence_inputfiles(sys.argv[1])