sourceCompass.py

 
import numpy as np
from typing import List
 
class SourceCompass(object):
    """ Source handler for compass simulation
 
    Attributes:
        sources : (List) : List of SutraSource instances
    """
    def __init__(self, sources : List):
        """ Initialize a SourceCompass component for target and wfs source related supervision
 
        Parameters:
            sources : (List) : List of SutraSource instances
        """
        self.sources = sources
 
    def raytrace(self, index, *, tel=None, atm=None, dms=None, ncpa : bool=True, reset : bool = True) -> None:
        """ Performs the raytracing operation through provided object phase screens 
        to obtain the phase screen of the SutraSource
 
        Parameters :
            index : (int) : Index of the source  to raytrace in self.sources list 
 
            tel : (TelescopeCompass, optional) : TelescopeCompass instance.
                                                 If provided, raytrace through the telescope aberration phase in the pupil
 
            atm : (AtmosCompass, optional) : AtmosCompass instance.
                                            If provided, raytrace through the layers phase screens 
 
            dms : (dmsCompass, optional) : DmCompass instance.
                                            If provided, raytrace through the DM shapes
 
            ncpa : (bool, optional) : If True (default), raytrace through NCPA phase screen of the source (default is array of 0, i.e. no impact) 
 
            rst: (bool): reset the phase screen before raytracing. Default is True
        """
        if (reset):
            self.sources[index].d_phase.reset()
 
        if atm is not None:
            self.sources[index].raytrace(atm.atmos) # Must be done first because of automatic reset of the phase screen when call
        if tel is not None:
            self.sources[index].raytrace(tel.tel)
        if dms is not None:
            self.sources[index].raytrace(dms.dms)
        if ncpa:
            self.sources[index].raytrace()