import sys,copy
sys.path.append('/home/mberger/fopiroot/fopiroot_dev/macro/tpc/FOPI/mberger')
import ROOT,glob,math,argparse
from functions import Divideh,set_palette,setRange,setGraph,getColors,thisIsTheEnd,getGraphYMaxMin


parser=argparse.ArgumentParser(description="compare digiresiduals with theoretical values")
parser.add_argument('infiles',help='inupt files. root files from calib_gemSpread.py',type=str,nargs='+')
parser.add_argument('--titles',help='the titles to use',default='',type=str,nargs='+')

args=parser.parse_args()
ROOT.gROOT.ProcessLine(".x rootlogon.C") 
ROOT.gROOT.ProcessLine('gROOT->SetStyle("Plain")')
set_palette()

canvas=ROOT.TCanvas('canvas','Comparison',1000,1000)
canvas.Divide(2,2)

vargraphs=[]
diffgraphs=[]
fracgraphs=[]
ifile=-1
thetas=[]
for f in args.infiles:
    ifile+=1
    if len(args.titles)<len(args.infiles):
        print f
        if f.find('theta')!=-1:
            title=f[f.find('theta')+5:f.find('.root')]
            thetas.append(float(title))
            print title,thetas
        else:
            title=str(ifile)
            theta=ifile
    else:
        title=args.titles[ifile]
        theta=ifile

    rfile=ROOT.TFile(f,'read')

    vargraphs.append(copy.deepcopy(rfile.Get('graph_variance')))
    diffgraphs.append(copy.deepcopy(rfile.Get('graph_diffusion')))
    fracgraphs.append(copy.deepcopy(rfile.Get('graph_varDifFrac')))

    setGraph(vargraphs[-1],'Standard Deviation',20,1,getColors()[ifile],'Z-Pos (cm)','Standard Deviation (cm)',1,1.2)
    setGraph(diffgraphs[-1],'Theoretical Diffusion',24,1,getColors()[ifile],'Z-Pos (cm)','Diffusion (cm)',1,1.2)
    setGraph(fracgraphs[-1],'Fraction',20,1,getColors()[ifile],'Z-Pos (cm)','Fraction',1,1.2)


dvmax,dvmin=getGraphYMaxMin(vargraphs+diffgraphs)
vargraphs[0].GetYaxis().SetRangeUser(dvmin,dvmax)
canvas.cd(1)
opt='AP'
fitfuncs=[]
par0graph=ROOT.TGraphErrors()
par1graph=ROOT.TGraphErrors()
for g in vargraphs:
    g.Draw(opt)
    opt='P'

for g in diffgraphs:
    g.Draw(opt)

opt='AP'
canvas.cd(2)
ig=-1
for g in fracgraphs:
    ig+=1
    fitfuncs.append(ROOT.TF1('fitfunc','[0]+[1]*sqrt(x)',0,75))
    g.Fit(fitfuncs[-1])
    par0graph.SetPoint(ig,thetas[ig],fitfuncs[-1].GetParameter(0))
    par0graph.SetPointError(ig,1,fitfuncs[-1].GetParError(0))
    par1graph.SetPoint(ig,thetas[ig],fitfuncs[-1].GetParameter(1))
    par1graph.SetPointError(ig,1,fitfuncs[-1].GetParError(1))

    g.Draw(opt)
    opt='P'

setGraph(par0graph,'Fit Parameter 0',20,1,1,'Theta','Value',1,1.2)
setGraph(par1graph,'Fit Parameter 1',20,1,1,'Theta','Value',1,1.2)
canvas.cd(3)
par0graph.Draw('AP')
canvas.cd(4)
par1graph.Draw('AP')

thisIsTheEnd()