#!/usr/bin/python
# -*- coding: iso-8859-15 -*-

import os,sys
# import getopt
##maybe it's supposed to be sys.argv[:] ... maybe
tmpargv = sys.argv
sys.argv = [ '-n' ]
from ROOT import *
sys.argv = tmpargv

from robs_draw_fiducial import *
ls = fiducial_lines()

# import psyco
# psyco.full()

# from time import sleep
# from optparse import OptionParser

# from ROOT_Tools import RootHandler
# from CommandLineMod import CommandLine

fid = [0.54, 0.96,  52.0, 10.0,   14.0,      38.0] ##intermediate fiducial

def in_fid(p,ct,cvals):
    if not cvals[0]<abs(ct)<cvals[1]: return False
    if not p<cvals[2]: return False
    pz=p*ct
    if not abs(pz)>cvals[4]: return False
    pt=sqrt(p**2-pz**2)
    if not cvals[3]<pt<cvals[5]: return False
    return True


if __name__ == "__main__":

	gStyle.SetTitleBorderSize	( 0			)   # duh
	# gStyle.SetTitleBorderSize	( 1			)
	gStyle.SetPadBorderMode		( 0			)
	gStyle.SetOptStat			( 111111	)	# Good standard stats display.
	gStyle.SetOptFit			( 1111		)	# Display fit info if available.
	gStyle.SetMarkerStyle		( 24		)	# Open circle.
	gStyle.SetOptTitle			( 1			)	# Display histogram titles.
	gStyle.SetCanvasColor		( 10		)	# Background color (white)
	gStyle.SetPadColor			( 10		)	# Background color (white)
	gStyle.SetFillColor			( 10		)	# Background color (white)
	gStyle.SetHistFillColor		( 10		)	# Background color (white)
	gStyle.SetFrameFillColor	( 10		)	# Background color (white)
	gStyle.SetStatColor			( 10		)	# Background color (white)

	gStyle.SetTitleFillColor	( 10		)	# Background color (white)
	gStyle.SetTitleFontSize		( 0.08		)	# Title Font Size
	gStyle.SetTitleX			( 0.10		)	# Title box X
	gStyle.SetTitleY			( 0.50		)	# Title box Y
	gStyle.SetTitleW			( 0.50		)	# Title box width
	gStyle.SetTitleH			( 0.09		)	# Title box height

	gStyle.SetStatX				( 0.9		)	# Statistic box X
	gStyle.SetStatY				( 0.9		)	# Statistic box Y
	gStyle.SetStatW				( 0.25		)	# Statistic box width
	# gStyle.SetStatH				( 0.2		)	# Statistic box height
	gStyle.SetStatBorderSize	( 1			)
	gStyle.SetLegendBorderSize	( 1			)
	gStyle.SetPalette			( 1			)	# Nicer colour scale for 2D histograms.
	gStyle.SetPaperSize			( 20., 24.	)
	gStyle.SetHistLineWidth		( 1			)
	
	gStyle.SetLabelFont(132)
	gStyle.SetLabelFont(132,"X")
	gStyle.SetLabelFont(132,"Y")
	gStyle.SetLabelFont(132,"Z")
	gStyle.SetLabelFont(132,"T")
	gStyle.SetTitleFont(132)
	gStyle.SetTitleFont(132,"X")
	gStyle.SetTitleFont(132,"Y")
	gStyle.SetTitleFont(132,"Z")
	gStyle.SetTitleFont(132,"T")
	gStyle.SetLabelSize(0.04)
	gStyle.SetLabelSize(0.04,"X")
	gStyle.SetLabelSize(0.04,"Y")
	gStyle.SetLabelSize(0.04,"Z")
	gStyle.SetLabelSize(0.04,"T")
	gStyle.SetTitleSize(0.04   )
	gStyle.SetTitleSize(0.04,"X")
	gStyle.SetTitleSize(0.04,"Y")
	gStyle.SetTitleSize(0.04,"Z")
	gStyle.SetTitleSize(0.04,"T")
	gStyle.SetTextFont(132)
	gStyle.SetStatFont(132)

	gStyle.SetPadLeftMargin		(0.10)
	gStyle.SetPadBottomMargin	(0.13)
	gStyle.SetPadRightMargin	(0.14)
	gStyle.SetPadTopMargin		(0.05)

	gROOT.SetBatch()

	# No Stats !
	# gStyle.SetOptStat(1110)
	# gStyle.SetOptFit(111)
	gStyle.SetOptStat			( 0	)	# Good standard stats display.
	# gStyle.SetOptFit			( 0	)	# Display fit info if available.
	gStyle.SetTitleW			( 0.70		)	# Title box width
	gStyle.SetStatX				( 1.00		)	# Statistic box width
	gStyle.SetStatY				( 0.99		)	# Statistic box width
	gStyle.SetStatW				( 0.23		)	# Statistic box width
	gStyle.SetStatH				( 0.25		)	# Statistic box width

	gStyle.SetPaperSize(14,14)
	gStyle.SetCanvasDefH(800)  # Defaults
	gStyle.SetCanvasDefW(1100)  # Defaults

	gROOT.ForceStyle()

	# We can have many files in input
	# FileList	= ['sum74-4-1_sum474-1_2_21-1_michel.root']
	FileList	= ['sum84-9-1_sum584-1_2_21-1_michel.root']


	HistName	= 'full_residuals2'
	OutputFile	= '../MCfitter_Residuals.eps'
	OutputFile2	= '../MCfitter_ResidualsFitted.eps'
	F=[]

	Canv = TCanvas()
	# Leg = TLegend(0.7, 0.7,0.9,0.9)

	# MinX = -3.0
	# MaxX = 3.0
	# MinY = -3.0
	# MaxY = 3.0
	# MinY = 3620
	# MaxY = 3830

	# Title	= 'TDC hit time, wire 12;[ns]'
	Title	= ';Momentum [MeV];cos(#theta);Residuals: (data-MC)/#sigma'
	# LegName = ['Bla','Blabla']

	for i,f in enumerate(FileList):
		F.append( TFile(f) )
		H	= F[i].Get(HistName)

		# H.GetXaxis().SetLabelFont(132)
		# H.GetYaxis().SetLabelFont(132)
		H.Draw('colz')
		H.SetTitle(Title)
		# H.SetAxisRange(MinX,MaxX,'X')
		# H.SetAxisRange(MinY,MaxY,'Y')
		# H.SetLineWidth(1)
		H.SetLineColor(0)
		H.SetTitleOffset( 1.0, 'Y')
		# H.SetTitleOffset( 1.2, 'Z')
		H.SetMinimum(-4)
		H.SetMaximum(4)
		
		for l in ls: l.Draw('same')
		# H.Fit('gaus', "Q","", 3612.0, 3812.0)

	# Leg.Draw()

	# For Postscript
	# Canv.Print(OutputFile,"Landscape")
	# For eps
	Canv.Print(OutputFile)



	Canv.Clear()

	gStyle.SetPaperSize(12,12)
	gStyle.SetLabelSize(0.05)
	gStyle.SetLabelSize(0.05,"X")
	gStyle.SetLabelSize(0.05,"Y")
	gStyle.SetLabelSize(0.05,"Z")
	gStyle.SetLabelSize(0.05,"T")
	gStyle.SetTitleSize(0.05   )
	gStyle.SetTitleSize(0.05,"X")
	gStyle.SetTitleSize(0.05,"Y")
	gStyle.SetTitleSize(0.05,"Z")
	gStyle.SetTitleSize(0.05,"T")

	def list_residuals(h,ctmin,ctmax):
	#DS = 0<ct<1, US = -1<ct<0
		rs = []
		nx,ny = h.GetNbinsX(),h.GetNbinsY()
		for ibx in range(1,nx+1):
			p  = h.GetXaxis().GetBinCenter(ibx)
			for iby in range(1,ny+1):
				ct = h.GetYaxis().GetBinCenter(iby)
				if in_fid(p,ct,fid) and ctmin<ct<ctmax:
					rs.append(h.GetBinContent(ibx,iby))
		return rs
	
	#h_residuals_fid_only = fid_only(h_all_residuals,fid,999.,999.)
	#h_residuals_fid_only.Draw('colz')
	
	US = list_residuals(H,0.,1.)
	DS = list_residuals(H,-1.,0.)
	US_and_DS = US+DS
	
	#h_DS        = TH1D('h_DS','h_DS',50,-5.,5.)
	#h_US        = TH1D('h_US','h_US',50,-5.,5.)
	h_US_and_DS	= TH1D('h_US_and_DS','h_US_and_DS',50,-5.,5.)
	
	for r in US_and_DS:
		h_US_and_DS.Fill(r,1.)
	#for r in US:
	#	h_US.Fill(r,1.)
	#	h_US_and_DS.Fill(r,1.)
	#for r in DS:
	#	h_DS.Fill(r,1.)
	#	h_US_and_DS.Fill(r,1.)
	
	h_US_and_DS.Fit('gaus')
	h_US_and_DS.SetTitle(';Residuals in fiducial: (data-simulation)/#sigma;Counts')
	h_US_and_DS.Draw()



	Canv.Print(OutputFile2)