#!/usr/bin/env python
"""

Plotter transmisjons og reflesksjonskoeffisienter.
Vi bruker m = 0.511 MeV

@author Are Raklev
"""

from numpy import *
from matplotlib.pyplot import *

mcsq  = 0.511E6  # MeV
hbarc = 197.3    # nm eV
alpha = 1.0      # nm eV
a = 0.3          # nm
V0 = 15.0        # eV

# Definerer transmisjonskoeffisienter
def T_delta( E ):
    beta = mcsq*alpha**2/2./hbarc**2/E
    return 1./(1+beta**2)

def T_ebronn( E ):
    return 1./(1.+V0**2/4./E/(E+V0)*sin(2*a/hbarc*sqrt(2*mcsq*(E+V0)))**2)


# Lager energi-verdier
E = linspace( 0, 50, 1e4 )
               
               
# Plotter for deltafunksjonen
figure()
plot( E, T_delta( E ), label='$T(E)$')
plot( E, 1-T_delta( E ), label='$R(E)$')

# Tekst langs x-aksen
xlabel('$E$ [eV]')

# Tekst langs y-aksen
ylabel('Sannsynlighet')
ylim([0,1.1])

#
legend(loc='best')

# Lager nok en figur for aa plotte Hermite polynomer
figure()
plot( E, T_ebronn( E ), label='$T(E)$')
plot( E, 1-T_ebronn( E ), label='$R(E)$')

# Tekst langs x-aksen
xlabel('$E$ [eV]')
xlim([0,40])

# Tekst langs y-aksen
ylabel('Sannsynlighet')
ylim([0,1.1])

#
legend(loc='best')

# Viser plottet
show()