![]() |
COMPASS
5.4.4
End-to-end AO simulation tool using GPU acceleration
|
Public Member Functions | |
None | __init__ (self, carmaWrap.context context, Source d_source, int im_dimx, int im_dimy, int fpm_dimx, int fpm_dimy, numpy.ndarray[numpy.float32] wavelength, int nWavelength, int babinet, int device) |
Instantiates a StellarCoronagraph object. More... | |
int | compute_image_normalization (self) |
Compute image for normalization. More... | |
int | set_apodizer (self, numpy.ndarray[numpy.float32] mask) |
Set apodizer for coronagraphic image computation. More... | |
int | set_focal_plane_mask (self, numpy.ndarray[numpy.float32] mask) |
Set focal plane mask for coronagraphic image computation. More... | |
int | set_lyot_stop (self, numpy.ndarray[numpy.float32] mask) |
Set lyot_stop for coronagraphic image computation. More... | |
int | set_mft (self, numpy.ndarray[numpy.complex64] A, numpy.ndarray[numpy.complex64] B, numpy.ndarray[numpy.float32] norm, str mft_type) |
Set MFT matrices for coronagraphic image computation. More... | |
![]() | |
int | compute_electric_field (self, int arg0) |
Computes the electric field from the specified wavelength. More... | |
int | compute_image (self, bool accumulate=True) |
Computes the coronagraphic image from the source phase screen. More... | |
int | compute_psf (self, bool accumulate=True) |
Computes the psf from the source phase screen. More... | |
int | reset (self) |
Reset long exposure image and counter. More... | |
int | set_amplitude (self, numpy.ndarray[numpy.float32] arg0) |
Set the electric field amplitude. More... | |
Properties | |
AA = property | |
A MFT matrices. More... | |
BB = property | |
B MFT matrices. More... | |
babinet = property | |
Babinet trick flag. More... | |
d_apodizer = property | |
Apodizer. More... | |
d_lyot_stop = property | |
Lyot stop. More... | |
focal_plane_mask = property | |
Focal plane mask. More... | |
fpmDimx = property | |
Focal plane dimension. More... | |
fpmDimy = property | |
Focal plane dimension. More... | |
norm = property | |
MFT normalization. More... | |
![]() | |
amplitude = property | |
Electric field amplitude in the pupil. More... | |
cntImg = property | |
Accumulation counter of coronagraphic images. More... | |
cntPsf = property | |
Accumulation counter of psf. More... | |
d_complex_image = property | |
Complex coronagraphic image. More... | |
d_electric_field = property | |
Electric field in the pupil. More... | |
d_image_le = property | |
Long exposure coronagraphic image. More... | |
d_image_se = property | |
Short exposure coronagraphic image. More... | |
d_psf_le = property | |
Long exposure coronagraphic psf. More... | |
d_psf_se = property | |
Short exposure coronagraphic psf. More... | |
d_pupil = property | |
Telescope pupil. More... | |
d_source = property | |
SutraSource used as OPD input. More... | |
device = property | |
GPU device index. More... | |
imageDimx = property | |
Coronagraphic image dimension along X-axis. More... | |
imageDimy = property | |
Coronagraphic image dimension along Y-axis. More... | |
pupDimx = property | |
Pupil dimension along X-axis. More... | |
pupDimy = property | |
Pupil dimension along Y-axis. More... | |
type = property | |
Coronagraph type. More... | |
wavelength = property | |
Vector of wavelength used to compute coronagraphic image. More... | |
Definition at line 8489 of file sutraWrap.py.
None sutraWrap.StellarCoronagraph.__init__ | ( | self, | |
carmaWrap.context | context, | ||
Source | d_source, | ||
int | im_dimx, | ||
int | im_dimy, | ||
int | fpm_dimx, | ||
int | fpm_dimy, | ||
numpy.ndarray[numpy.float32] | wavelength, | ||
int | nWavelength, | ||
int | babinet, | ||
int | device | ||
) |
Instantiates a StellarCoronagraph object.
context | (CarmaContext): context |
d_source | (SutraSource): Coronagraph source input |
im_dimx | (int): Coronagraphic image dimension along x axis |
im_dimy | (int): Coronagraphic image dimension along y axis |
fpm_dimx | (int): Focal plane dimension along x axis |
fpm_dimy | (int): Focal plane dimension along y axis |
wavelength | (np.ndarray[ndim=1, dtype=np.float32]): vector of wavelengths |
nWavelength | (int): number of wavelength |
babinet | (bool): Flag to enable Babinet trick |
device | (int): GPU device index |
Definition at line 8514 of file sutraWrap.py.
int sutraWrap.StellarCoronagraph.compute_image_normalization | ( | self | ) |
Compute image for normalization.
Definition at line 8519 of file sutraWrap.py.
int sutraWrap.StellarCoronagraph.set_apodizer | ( | self, | |
numpy.ndarray[numpy.float32] | mask | ||
) |
Set apodizer for coronagraphic image computation.
mask | (np.ndarray[ndim=2, dtype=np.float32]): apodizer |
Definition at line 8526 of file sutraWrap.py.
int sutraWrap.StellarCoronagraph.set_focal_plane_mask | ( | self, | |
numpy.ndarray[numpy.float32] | mask | ||
) |
Set focal plane mask for coronagraphic image computation.
mask | (np.ndarray[ndim=3, dtype=np.float32]): Focal plane mask for each wavelength |
Definition at line 8533 of file sutraWrap.py.
int sutraWrap.StellarCoronagraph.set_lyot_stop | ( | self, | |
numpy.ndarray[numpy.float32] | mask | ||
) |
Set lyot_stop for coronagraphic image computation.
mask | (np.ndarray[ndim=2, dtype=np.float32]): lyot_stop |
Definition at line 8540 of file sutraWrap.py.
int sutraWrap.StellarCoronagraph.set_mft | ( | self, | |
numpy.ndarray[numpy.complex64] | A, | ||
numpy.ndarray[numpy.complex64] | B, | ||
numpy.ndarray[numpy.float32] | norm, | ||
str | mft_type | ||
) |
Set MFT matrices for coronagraphic image computation.
A | (np.ndarray[dtype=np.complex32, ndims=3]): A MFT matrix for each wavelength |
B | (np.ndarray[dtype=np.complex32, ndims=3]): B MFT matrix for each wavelength |
norm | (np.ndarray[dtype=np.complex32, ndims=3]): MFT normalization for each wavelength |
mft_type | (str): MFT matrices to set, i.e. "img" or "psf" |
Definition at line 8553 of file sutraWrap.py.
|
static |
A MFT matrices.
@n type of : typing.Dict[str, typing.Tuple[typing.List[carmaWrap.obj_float_complex], typing.List[int]]]
Definition at line 8560 of file sutraWrap.py.
|
static |
|
static |
B MFT matrices.
@n type of : typing.Dict[str, typing.Tuple[typing.List[carmaWrap.obj_float_complex], typing.List[int]]]
Definition at line 8569 of file sutraWrap.py.
|
static |
|
static |
|
static |
Focal plane mask.
@n type of : typing.List[carmaWrap.obj_float]
Definition at line 8605 of file sutraWrap.py.
|
static |
|
static |
|
static |
MFT normalization.
@n type of : typing.Dict[str, typing.List[float]]
Definition at line 8632 of file sutraWrap.py.