import numpy as np
import sys
sys.path.insert(0, '/Users/joost/Dropbox/Werk/PhD/Analyses/Python/Analytical/Fourier Optics')
from fourierOptics import *

if __name__ == '__main__':
	# Initiate MICADO paraxial model
	mcd = ParaxialMICADO()

	# Initiate WFE object
	wfe = WavefrontError(mcd.n, mcd.m)

	# Generate PSD surfaces
	psd = wfe.generate_height_map(200e-6, 2, 1)

	# Add it to ADC-P1
	mcd.ELT_ADC.p1.add_wavefront_error(psd)

	# Generate the PSF and plot it
	E, x, y = mcd.propagate(l=1.35e-3,surface=-1, zoom=True, show=True)