import numpy as np
import sys
sys.path.insert(0, '/Users/joost/Dropbox/Werk/PhD/Documents/Published/FourierOptics GitHub release/')
from fourierOptics import *

'''
	Description:
	This script generates a point spread function of the MICADO optical model.
	During the setup a wavefront error of 200 nm RMS is added to the first 
	prisms of the MICADO atmospheric dispersion corrector.
'''


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)