#!/usr/bin/python
import os,sys
from ROOT import *
from clean_canvas import *
gStyle = makegstyle(gStyle)



C_2 = {'nom':-2E-8,  '1p96':11.4E-8, '2p04':-19.7E-8}
C_3 = {'nom':-8E-10, '1p96':2.0E-10, '2p04':-2.2E-10}
C_r = {'nom':-8E-6,  '1p96':-1.1E-6, '2p04':-2.8E-6}

def dbz(z,r,map):
    return C_2[map]*(z**2) + C_2[map]*(z**3) + C_r[map]*r

gr_nom = TGraph()
gr_1p96 = TGraph()
gr_2p04 = TGraph()

for i,z in enumerate(range(-60,60)):
    gr_nom.SetPoint(i,z,dbz(z,0,'nom'))
    gr_1p96.SetPoint(i,z,dbz(z,0,'1p96'))
    gr_2p04.SetPoint(i,z,dbz(z,0,'2p04'))


c = TCanvas()
gr_nom.Draw('AC')
gr_1p96.Draw('C')
gr_2p04.Draw('C')

l = TLegend(0.6,0.6,0.9,0.9)
l.SetBorderSize(0)
l.AddEntry(gr_nom,'nominal map, r=0 cm','l')
l.AddEntry(gr_1p96,'B=1.96 T, r=0 cm','l')
l.AddEntry(gr_2p04,'B=2.04 T, r=0 cm','l')
l.Draw()

au = raw_input('>')