#!/usr/bin/python

from numpy import *
from scipy import *
from pylab import *
from matplotlib import rc, rcParams

# Bruk LaTeX
rc('text',usetex=True)
# Endre alle fonter til 'Computer Modern'
rc('font',**{'family':'serif','serif':['Computer Modern']})
# Eksperimentelle data for phi og S/g i enheter hhv radianer og gram
phi = [1.57080, 3.14159, 4.71239, 6.28319, 7.85398, 9.42478, 10.99557, 12.56637]
S = [240.0, 300.0, 440.0, 600.0, 800.0, 1000.0, 1100.0, 1400.0]
# S0 = 185.0 gram = loddets masse
S0 = 185.0
y = log(divide(S,S0))
mu = y/phi
musum = sum(mu)
imax = 8
# Middelverdi av mu
mumiddel = musum/imax
kvadratavvik = sum((mu-mumiddel)**2)
# Standardavvik og standardfeil i mu:
deltamu = sqrt(kvadratavvik/(imax-1))
deltamumiddel = deltamu/sqrt(imax)
# Skriver ut middelverdi, standardavvik og standardfeil
print('mumiddel = ',mumiddel)
print('deltamu = ',deltamu)
print('deltamumiddel = ',deltamumiddel)
# Plotter resultatene
scatter(phi,y,label='$\ln(S/S0)$')
hold(True)
phi = asarray(phi)
midtlinje = mumiddel*phi
lavlinje = (mumiddel-deltamu)*phi
hoylinje = (mumiddel+deltamu)*phi
plot(phi,midtlinje,label='$\overline{\mu}\,\phi$',color='b')
plot(phi,hoylinje,label='($\overline{\mu}+\Delta\mu)\,\phi$',color='g')
plot(phi,lavlinje,label='($\overline{\mu}-\Delta\mu)\,\phi$',color='r')
title('{\O}ving 3, Oppgave 1')
xlabel('$\phi$ (rad)')
legend(loc='upper left')
axis([0,14,0,2.5])
hold(False)
# Lagrer figur i format eps (encapsulated postscript) og pdf
savefig('pythonfigur.eps')
savefig('pythonfigur.pdf')
show()