rtcCompass.py

 
 
from shesha.init.rtc_init import rtc_init
from shesha.supervisor.components.sourceCompass import SourceCompass
import shesha.constants as scons
import numpy as np
from typing import Union
 
from shesha.supervisor.components.rtc.rtcAbstract import RtcAbstract, carmaWrap_context
 
 
class RtcCompass(RtcAbstract):
    """ RTC handler for compass simulation
    """
 
    def __init__(self, context: carmaWrap_context, config, tel, wfs, dms, atm, *,
                 brahma: bool = False, fp16: bool = False, cacao: bool = False):
        """ Initialize a RtcCompass component for rtc related supervision
 
        Args:
            context : (carmaContext) : CarmaContext instance
 
            config : (config module) : Parameters configuration structure module
 
            tel: (Telescope) : Telescope object
 
            wfs: (Sensors) : Sensors object
 
            dms: (Dms) : Dms object
 
            atm: (Atmos) : Atmos object
 
        Kwargs:
            brahma : (bool, optional) : If True, enables BRAHMA features in RTC (Default is False)
                                       Requires BRAHMA to be installed
 
            fp16 : (bool, optional) : If True, enables FP16 features in RTC (Default is False)
                                       Requires CUDA_SM>60 to be installed
 
            cacao : (bool) : If True, enables CACAO features in RTC (Default is False)
                                       Requires OCTOPUS to be installed
        """
        RtcAbstract.__init__(self, context, config, brahma=brahma, fp16=fp16,
                             cacao=cacao)
        self.rtc_initrtc_initrtc_init(tel, wfs, dms, atm)
 
    def rtc_init(self, tel, wfs, dms, atm):
        """ Initialize a RtcCompass component for rtc related supervision
 
        Args:
            tel: (Telescope) : Telescope object
 
            wfs: (Sensors) : Sensors object
 
            dms: (Dms) : Dms object
 
            atm: (Atmos) : Atmos object
        """
        self._rtc_rtc_rtc = rtc_init(self._context_context, tel._tel, wfs._wfs, dms._dms, atm._atmos,
                             self._config_config.p_wfss, self._config_config.p_tel,
                             self._config_config.p_geom, self._config_config.p_atmos,
                             self._config_config.p_loop.ittime, self._config_config.p_centroiders,
                             self._config_config.p_controllers, self._config_config.p_dms,
                             cacao=self.cacaocacao)