import pylab as pl
from intensity import *
from propagate import *

def  testfraunhoferRect(x) :
       i = intensity()
       i.rect_flattop(5e-4,5e-4)
       i.plot()

       pl.figure(2)
       p = propagate()
       ip = p.fraunhofer(x,i)
       ip.plot2()


def  testfraunhoferCirc() :
       i = intensity(1e-6,-1e-4,1e-4,1e-6,-1e-4,1e-4,523e-9)
       i.circular_flattop(5e-5)
       i.plot()

       pl.figure(2)
       p = propagate()
       ip = p.fraunhofer(0.1,i)
       ip.plot2()


def  testfresnelRect() :
       i = intensity()
       i.rect_flattop(5e-4,5e-4);
       i.plot()

       pl.figure(2)
       p = propagate()
       ip = p.singlefresnel(0.01,i)
       ip.plot()

def  testfresnelCirc() :
       i = intensity()
       i.circular_flattop(5e-4)
       i.plot()

       pl.figure(2)
       p = propagate()
       ip = p.singlefresnel(0.1,i)
       ip.plot()

def testfraunhoferGau():
       i = intensity(1e-6,-1e-4,1e-4,1e-6,-1e-4,1e-4,523e-9)
       i.gaussian(5e-5)
       i.plot()

       pl.figure(2)
       p = propagate()
       ip = p.fraunhofer(20,i)
       ip.plot2()
 
def testfresnelGau():
       i = intensity()
       i.gaussian(5e-4);
       i.plot()

       pl.figure(2)
       p = propagate()
       ip = p.singlefresnel(0.02,i)     
       ip.plot()