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

Plott av sannsynligheten for aa finne et K^0-meson eller antipartikkelen.
Vi antar at vi begynner med et K^0-meson ved t=0.

@author Are Raklev
"""

from numpy import *
from matplotlib.pyplot import *

# Noen parametre for K^0-mesoner
DeltaM  = 3.483E-6 # eV
#Gamma   = 3.683E-6 # eV
#Gamma12 = 3.670E-6 # eV
Gamma   = 0. # eV
Gamma12 = 0. # eV
hbar    = 6.582E-7 # eV ns

# Definer sannsynlighetene som funksjon av tiden t
def PK0( t ):
    return 0.25*(exp(-(Gamma+Gamma12)/hbar*t)+exp(-(Gamma-Gamma12)/hbar*t)+2.*exp(-Gamma/hbar*t)*cos(DeltaM*t/hbar))

def PK0bar( t ):
    return 0.25*(exp(-(Gamma+Gamma12)/hbar*t)+exp(-(Gamma-Gamma12)/hbar*t)-2.*exp(-Gamma/hbar*t)*cos(DeltaM*t/hbar))

# Lager t-verdier
t = linspace( 0, 2, 1e5 )

# Lager foerst en figur
figure()
plot( t, PK0( t ), label='$K^0$')
plot( t, PK0bar( t ), label=r'${\bar K}^0$')

# Setter tittel
title('')

# Tekst langs x-aksen
xlabel('$t$ [ns]')

# Tekst langs y-aksen
ylabel('$P(t)$')

#
legend(loc='best')

# Lagre som .eps fil
savefig('K0.eps')

# Viser plottet
show()