import sys

sys.path.append('/home/mberger/fopiroot/fopiroot_dev/macro/tpc/FOPI/mberger')
import ROOT,glob,math,argparse
from anaFile import anaFile
from functions import *

parser=argparse.ArgumentParser(description='plot snr for one or more files')
parser.add_argument('anafiles',help='the ana files',type=str,nargs='+')
parser.add_argument('--titles',help='the titles',type=str,default=[],nargs='+')
parser.add_argument('--hlp',help='show help',action='store_true')
args=parser.parse_args()

ROOT.gROOT.ProcessLine(".x rootlogon.C") 
ROOT.gROOT.ProcessLine('gStyle->SetPalette(1)')
ROOT.gROOT.ProcessLine('gROOT->SetStyle("Plain")')

if args.hlp:
    parser.print_help()
    exit(0)


rfiles=[]

sampleSNR=[]
digiSNR=[]
clustSNR=[]
legendSNR=[]
digiSNRslice=[]
clustSNRslice=[]
zSlices=[]

colors=getColors2()

sampleSNR_max=[]
sampleSNR_mpv=[]
sampleSNR_mpvErr=[]
digiSNR_max=[]
clustSNR_max=[]

fcounter=-1
for ifile in range(len(args.anafiles)):
    fcounter+=1
    rfiles.append(anaFile(args.anafiles[ifile]))
    sampleSNR.append(rfiles[-1].getHist("ssnr"))
    digiSNR.append(rfiles[-1].getHist("hsnr"))
    clustSNR.append(rfiles[-1].getHist("csnr"))
    legendSNR.append(clustSNR[-1].Clone())
    digiSNRslice.append(rfiles[-1].getSlicedHist("nsnr"))
    clustSNRslice.append(rfiles[-1].getSlicedHist("csnr"))
    zSlices.append(rfiles[-1].getDriftSlices())
    sampleSNR_max.append(-1)
    digiSNR_max.append(-1)
    clustSNR_max.append(-1)
    sampleSNR_mpv.append(-1)
    sampleSNR_mpvErr.append(-1)

counter=0
while len(args.titles)<len(args.anafiles):
    args.titles.append(int(round( float(rfiles[counter].getInfo("gain") )/100. ,0)*100 ))
    counter+=1
print args.titles

casnr=ROOT.TCanvas("casnr","snr",1500,500)
caclsnrslice=ROOT.TCanvas("cavlsnslice","cluster snr slice")
cadisnrslice=ROOT.TCanvas("cadisnslice","digi snr slice")
casnr.Divide(3,1)

caclsnrslice.Divide(3,2)
cadisnrslice.Divide(3,2)


cSNRg=ROOT.TCanvas("cSNRg","SNR graph",1500,500)
cSNRg.Divide(3,1)
cSNRg_all=ROOT.TCanvas("cSNRg_all","All SNR graphs",700,500)

gsampleSNR=ROOT.TGraphErrors()
gdigiSNR=ROOT.TGraph()
gclustSNR=ROOT.TGraph()

leg=ROOT.TLegend(0.75,0.5,0.89,0.89)
leg.SetFillColor(4000)
leg.SetLineWidth(0)
leg.SetBorderSize(0)

opt=""
for i in range(len(args.anafiles)):
    opt="same"
    if i==0:
        opt=""
    casnr.cd(1)
    sampleSNR[i].SetLineColor(colors[i])
    sampleSNR[i].SetLineWidth(2)
    sampleSNR[i].SetTitle("Sample SNR")
    sampleSNR[i].SetStats(0)
    sampleSNR[i].Sumw2()
    sampleSNR[i].Scale(1./sampleSNR[i].Integral())
    sampleSNR[i].Draw(opt)
    sampleSNR[i].GetYaxis().SetTitle("Normalized Counts")
    sampleSNR_max[i]=sampleSNR[i].GetBinCenter(sampleSNR[i].GetMaximumBin())
    landau=ROOT.TF1("landau","landau",0,200)
    sampleSNR[i].Fit("landau")
    sampleSNR_mpv[i]=landau.GetParameter(1)
    sampleSNR_mpvErr[i]=landau.GetParError(1)
    gsampleSNR.SetPoint(i,float(rfiles[i].getInfo("gain")),sampleSNR_mpv[i])
    gsampleSNR.SetPointError(i,0,sampleSNR_mpvErr[i])

    casnr.cd(2)
    digiSNR[i].SetLineColor(colors[i])
    digiSNR[i].SetTitle("Digi SNR ")
    digiSNR[i].SetStats(0)
    digiSNR[i].GetXaxis().SetTitle("Signal to Noise Ratio")
    digiSNR[i].Sumw2()
    #digiSNR[i].Rebin(2)
    digiSNR[i].Scale(1./digiSNR[i].Integral())
    digiSNR[i].Draw(opt)
    digiSNR_max[i]=digiSNR[i].GetBinCenter(digiSNR[i].GetMaximumBin())
    gdigiSNR.SetPoint(i,float(rfiles[i].getInfo("gain")),digiSNR_max[i])

    casnr.cd(3)
    clustSNR[i].SetLineColor(colors[i])
    clustSNR[i].SetTitle("Cluster SNR ")
    clustSNR[i].SetStats(0)
    clustSNR[i].GetXaxis().SetTitle("Signal to Noise Ratio")
    #clustSNR[i].Rebin(2)
    clustSNR[i].Sumw2()
    clustSNR[i].Scale(1./clustSNR[i].Integral())
    clustSNR[i].Draw(opt)
    clustSNR_max[i]=clustSNR[i].GetBinCenter(clustSNR[i].GetMaximumBin())
    gclustSNR.SetPoint(i,float(rfiles[i].getInfo("gain")),clustSNR_max[i])
    
    legendSNR[i].SetLineColor(colors[i])
    legendSNR[i].SetLineWidth(8)
    
    leg.AddEntry(legendSNR[i],str(args.titles[i]),"lp")
    casnr.cd(1)
    leg.Draw()
    casnr.cd(2)
    leg.Draw()
                   
    for j in range(len(zSlices[i])):
        caclsnrslice.cd(j+1)
        clustSNRslice[i][j].SetStats(0)
        clustSNRslice[i][j].SetLineColor(colors[i])
        clustSNRslice[i][j].SetTitle("Cluster SNR "+str(zSlices[i][j]))
        clustSNRslice[i][j].GetXaxis().SetTitle("Signal to Noise Ratio")
        clustSNRslice[i][j].Draw(opt)

        cadisnrslice.cd(j+1)
        digiSNRslice[i][j].SetStats(0)
        digiSNRslice[i][j].SetLineColor(colors[i])
        digiSNRslice[i][j].SetTitle("Digi SNR "+str(zSlices[i][j]))
        digiSNRslice[i][j].GetXaxis().SetTitle("Signal to Noise Ratio")
        digiSNRslice[i][j].Rebin(4)
        digiSNRslice[i][j].Draw(opt)


cSNRg.cd(1)
gsampleSNR.SetMarkerStyle(20)
gsampleSNR.SetMarkerColor(ROOT.kViolet)
gsampleSNR.Draw("AP")
cSNRg.cd(2)
gdigiSNR.SetMarkerStyle(20)
gdigiSNR.SetMarkerColor(ROOT.kAzure)
gdigiSNR.Draw("AP")
cSNRg.cd(3)
gclustSNR.SetMarkerStyle(20)
gclustSNR.SetMarkerColor(ROOT.kSpring)
gclustSNR.Draw("AP")

#cSNRg_all.cd()
#gsampleSNR.Draw("AP")
#gclustSNR.Draw("P")
#gdigiSNR.Draw("P")
fontsize=0.06
leftMargin=0.14
bottomMargin=0.14

cextra1=ROOT.TCanvas()
cextra1.SetBottomMargin(bottomMargin)
cextra1.SetLeftMargin(leftMargin)
opt=""
for i in range(len(args.anafiles)):
    sampleSNR[i].GetYaxis().SetLabelFont(132)
    sampleSNR[i].GetYaxis().SetLabelSize(fontsize)
    sampleSNR[i].GetYaxis().SetTitleFont(132)
    sampleSNR[i].GetYaxis().SetTitleSize(fontsize)
    sampleSNR[i].GetXaxis().SetLabelFont(132)
    sampleSNR[i].GetXaxis().SetLabelSize(fontsize)
    sampleSNR[i].GetXaxis().SetTitleFont(132)
    sampleSNR[i].GetXaxis().SetTitleSize(fontsize)

    sampleSNR[i].SetTitle("")
    sampleSNR[i].Draw(opt)
    opt="same"
leg.SetTextFont(132)
leg.SetTextSize(fontsize)
leg.Draw()

cextra2=ROOT.TCanvas()
cextra2.SetBottomMargin(bottomMargin)
cextra2.SetLeftMargin(leftMargin)
gsampleSNR.Draw("AP")
gsampleSNR.GetXaxis().SetTitle("Gain")
gsampleSNR.GetYaxis().SetTitle("Signal to Noise Ratio")
gsampleSNR.GetXaxis().SetTitleFont(132)
gsampleSNR.GetXaxis().SetTitleSize(fontsize)
gsampleSNR.GetYaxis().SetTitleFont(132)
gsampleSNR.GetYaxis().SetTitleSize(fontsize)
gsampleSNR.GetXaxis().SetLabelFont(132)
gsampleSNR.GetXaxis().SetLabelSize(fontsize)
gsampleSNR.GetYaxis().SetLabelFont(132)
gsampleSNR.GetYaxis().SetLabelSize(fontsize)
gsampleSNR.GetXaxis().SetNdivisions(505)

thisIsTheEnd()