# -*- coding: utf-8 -*-
"""
Created on Thu Mar  9 15:27:48 2017

@author: lbnc
"""

import numpy as np
import matplotlib.pyplot as plt

# brytningsindekser
n1 = 1.0
n2 = 2.0

mur1 = 1.
mur2 = 1.

# innfallsvinkel
alpha = np.arange(0,90)/180.*np.pi

rt = np.sqrt( n2**2 - n1**2*np.sin( alpha )**2 )
mur = mur1/mur2
ca = np.cos( alpha )

# E felt har ingen komponenete normal til innfallsplanet
Tp = 2.*n1*ca/( n1*ca + mur*rt )
Rp = ( n1*ca - mur*rt )/( n1*ca + mur*rt )

# E felt har komponente normal til innfallsplanet
Ts = 2.*n1*n2*ca/( mur*n2**2*ca + n1*rt )
Rs = ( mur*n2**2*ca - n1*rt )/( mur*n2**2*ca + n1*rt )

plt.plot( alpha/np.pi*180., R*2 )
plt.xlabel("vinkel")
plt.ylabel("T/R")