widget_ao_expert.py

 
 
import numpy as np
 
import shesha.ao as ao
import shesha.constants as scons
from shesha.constants import CONST
from shesha.ao.wfs import comp_new_pyr_ampl
 
from shesha.widgets.widget_base import uiLoader
ExpertWidgetTemplate, ExpertClassTemplate = uiLoader('widget_ao_expert')
 
 
class WidgetAOExpert(ExpertClassTemplate):
 
    def __init__(self) -> None:
        ExpertClassTemplate.__init__(self)
        self.simsim = None
 
        self.uiExpertuiExpert = ExpertWidgetTemplate()
        self.uiExpertuiExpert.setupUi(self)
 
        
 
        
 
        
        self.uiExpertuiExpert.wao_setAtmos.clicked.connect(self.setAtmosParamssetAtmosParams)
        self.uiExpertuiExpert.wao_setWfs.clicked.connect(self.setWfsParamssetWfsParams)
        self.uiExpertuiExpert.wao_setDM.clicked.connect(self.setDmParamssetDmParams)
        self.uiExpertuiExpert.wao_setControl.clicked.connect(self.setRtcParamssetRtcParams)
        self.uiExpertuiExpert.wao_setCentro.clicked.connect(self.setRtcParamssetRtcParams)
        self.uiExpertuiExpert.wao_setTelescope.clicked.connect(self.setTelescopeParamssetTelescopeParams)
        self.uiExpertuiExpert.wao_reset_dm.clicked.connect(self.reset_dmreset_dm)
        self.uiExpertuiExpert.wao_update_gain.clicked.connect(self.updateGainupdateGain)
        self.uiExpertuiExpert.wao_update_pyr_ampl.clicked.connect(self.updatePyrAmplupdatePyrAmpl)
        self.uiExpertuiExpert.wao_selectRtcMatrix.currentIndexChanged.connect(
                self.displayRtcMatrixdisplayRtcMatrix)
        self.uiExpertuiExpert.wao_commandBtt.clicked.connect(self.BttCommandBttCommand)
        self.uiExpertuiExpert.wao_commandKL.clicked.connect(self.KLCommandKLCommand)
 
        self.uiExpertuiExpert.wao_dmUnitPerVolt.valueChanged.connect(self.updateDMrangeGUIupdateDMrangeGUI)
        self.uiExpertuiExpert.wao_dmpush4iMat.valueChanged.connect(self.updateDMrangeGUIupdateDMrangeGUI)
        self.uiExpertuiExpert.wao_pyr_ampl.valueChanged.connect(self.updateAmpliCompGUIupdateAmpliCompGUI)
        self.uiExpertuiExpert.wao_dmActuPushArcSecNumWFS.currentIndexChanged.connect(
                self.updateDMrangeupdateDMrange)
 
    
 
    def setSim(self, sim) -> None:
        self.simsim = sim
 
    def updateGain(self) -> None:
        if (self.simsim.rtc):
            self.simsim.rtc.set_gain(0, float(self.uiExpertuiExpert.wao_controlGain.value()))
            print("Loop gain updated on GPU")
 
    def updateAmpliCompGUI(self) -> None:
        diffract = self.simsim.config.p_wfss[0].Lambda * \
            1e-6 / self.simsim.config.p_tel.diam * CONST.RAD2ARCSEC
        self.uiExpertuiExpert.wao_pyr_ampl_arcsec.setValue(
                self.uiExpertuiExpert.wao_pyr_ampl.value() * diffract)
 
    def updateAmpliComp(self) -> None:
        diffract = self.simsim.config.p_wfss[0].Lambda * \
            1e-6 / self.simsim.config.p_tel.diam * CONST.RAD2ARCSEC
        self.uiExpertuiExpert.wao_pyr_ampl_arcsec.setValue(
                self.simsim.config.p_wfss[0].pyr_ampl * diffract)
 
    def updatePyrAmpl(self) -> None:
        if (self.simsim.rtc):
            comp_new_pyr_ampl(0, self.uiExpertuiExpert.wao_pyr_ampl.value(), self.simsim.config.p_wfss,
                              self.simsim.config.p_tel)
            self.simsim.config.p_wfss[0].set_pyr_ampl(self.uiExpertuiExpert.wao_pyr_ampl.value())
            print("Pyramid modulation updated on GPU")
            self.updatePlotWfsupdatePlotWfs()
 
    def updateDMrangeGUI(self) -> None:
        push4imat = self.uiExpertuiExpert.wao_dmpush4iMat.value()
        unitpervolt = self.uiExpertuiExpert.wao_dmUnitPerVolt.value()
        self.updateDMrangeupdateDMrange(push4imat=push4imat, unitpervolt=unitpervolt)
 
    def updateDMrange(self, push4imat: float = None, unitpervolt: float = None) -> None:
        numdm = str(self.uiExpertuiExpert.wao_selectDM.currentText())
        numwfs = str(self.uiExpertuiExpert.wao_dmActuPushArcSecNumWFS.currentText())
        if ((numdm is not "") and (numwfs is not "") and (push4imat != 0) and
            (unitpervolt != 0)):
            arcsecDMrange = self.computeDMrangecomputeDMrange(
                    int(numdm), int(numwfs), push4imat=push4imat,
                    unitpervolt=unitpervolt)
            self.uiExpertuiExpert.wao_dmActPushArcsec.setValue(arcsecDMrange)
 
    def updateTelescopePanel(self) -> None:
        self.uiExpertuiExpert.wao_zenithAngle.setValue(self.simsim.config.p_geom.zenithangle)
        self.uiExpertuiExpert.wao_diamTel.setValue(self.simsim.config.p_tel.diam)
        self.uiExpertuiExpert.wao_cobs.setValue(self.simsim.config.p_tel.cobs)
 
    def updateDmPanel(self) -> None:
        ndm = self.uiExpertuiExpert.wao_selectDM.currentIndex()
        if (ndm < 0):
            ndm = 0
        self.uiExpertuiExpert.wao_dmActuPushArcSecNumWFS.clear()
        self.uiExpertuiExpert.wao_dmActuPushArcSecNumWFS.addItems([
                str(i) for i in range(len(self.simsim.config.p_dms))
        ])
        self.uiExpertuiExpert.wao_numberofDMs.setText(str(len(self.simsim.config.p_dms)))
        self.uiExpertuiExpert.wao_dmTypeSelector.setCurrentIndex(
                self.uiExpertuiExpert.wao_dmTypeSelector.findText(
                        str(self.simsim.config.p_dms[ndm].type)))
        self.uiExpertuiExpert.wao_dmAlt.setValue(self.simsim.config.p_dms[ndm].alt)
        if (self.simsim.config.p_dms[ndm].type == scons.DmType.KL):
            self.uiExpertuiExpert.wao_dmNactu.setValue(self.simsim.config.p_dms[ndm].nkl)
        else:
            self.uiExpertuiExpert.wao_dmNactu.setValue(self.simsim.config.p_dms[ndm].nact)
        self.uiExpertuiExpert.wao_dmUnitPerVolt.setValue(self.simsim.config.p_dms[ndm].unitpervolt)
        self.uiExpertuiExpert.wao_dmpush4iMat.setValue(self.simsim.config.p_dms[ndm].push4imat)
        self.uiExpertuiExpert.wao_dmCoupling.setValue(self.simsim.config.p_dms[ndm].coupling)
        self.uiExpertuiExpert.wao_dmThresh.setValue(self.simsim.config.p_dms[ndm].thresh)
        self.updateDMrangeupdateDMrange()
 
    def updateWfsPanel(self) -> None:
        nwfs = self.uiExpertuiExpert.wao_selectWfs.currentIndex()
        if (nwfs < 0):
            nwfs = 0
        self.uiExpertuiExpert.wao_numberofWfs.setText(str(len(self.simsim.config.p_dms)))
        self.uiExpertuiExpert.wao_wfsType.setText(str(self.simsim.config.p_wfss[nwfs].type))
        self.uiExpertuiExpert.wao_wfsNxsub.setValue(self.simsim.config.p_wfss[nwfs].nxsub)
        self.uiExpertuiExpert.wao_wfsNpix.setValue(self.simsim.config.p_wfss[nwfs].npix)
        self.uiExpertuiExpert.wao_wfsPixSize.setValue(self.simsim.config.p_wfss[nwfs].pixsize)
        self.uiExpertuiExpert.wao_wfsXpos.setValue(self.simsim.config.p_wfss[nwfs].xpos)
        self.uiExpertuiExpert.wao_wfsYpos.setValue(self.simsim.config.p_wfss[nwfs].ypos)
        self.uiExpertuiExpert.wao_wfsFracsub.setValue(self.simsim.config.p_wfss[nwfs].fracsub)
        self.uiExpertuiExpert.wao_wfsLambda.setValue(self.simsim.config.p_wfss[nwfs].Lambda)
        self.uiExpertuiExpert.wao_wfsMagnitude.setValue(self.simsim.config.p_wfss[nwfs].gsmag)
        self.uiExpertuiExpert.wao_wfsZp.setValue(np.log10(self.simsim.config.p_wfss[nwfs].zerop))
        self.uiExpertuiExpert.wao_wfsThrough.setValue(self.simsim.config.p_wfss[nwfs].optthroughput)
        self.uiExpertuiExpert.wao_wfsNoise.setValue(self.simsim.config.p_wfss[nwfs].noise)
        self.uiExpertuiExpert.wao_pyr_ampl.setValue(self.simsim.config.p_wfss[nwfs].pyr_ampl)
 
        # LGS panel
        if (self.simsim.config.p_wfss[nwfs].gsalt > 0):
            self.uiExpertuiExpert.wao_wfsIsLGS.setChecked(True)
            self.uiExpertuiExpert.wao_wfsGsAlt.setValue(self.simsim.config.p_wfss[nwfs].gsalt)
            self.uiExpertuiExpert.wao_wfsLLTx.setValue(self.simsim.config.p_wfss[nwfs].lltx)
            self.uiExpertuiExpert.wao_wfsLLTy.setValue(self.simsim.config.p_wfss[nwfs].llty)
            self.uiExpertuiExpert.wao_wfsLGSpower.setValue(
                    self.simsim.config.p_wfss[nwfs].laserpower)
            self.uiExpertuiExpert.wao_wfsReturnPerWatt.setValue(
                    self.simsim.config.p_wfss[nwfs].lgsreturnperwatt)
            self.uiExpertuiExpert.wao_wfsBeamSize.setValue(self.simsim.config.p_wfss[nwfs].beamsize)
            self.uiExpertuiExpert.wao_selectLGSProfile.setCurrentIndex(
                    self.uiExpertuiExpert.wao_selectLGSProfile.findText(
                            str(self.simsim.config.p_wfss[nwfs].proftype)))
 
        else:
            self.uiExpertuiExpert.wao_wfsIsLGS.setChecked(False)
 
        if (self.simsim.config.p_wfss[nwfs].type == "pyrhr" or
                    self.simsim.config.p_wfss[nwfs].type == "pyr"):
            self.uiExpertuiExpert.wao_wfs_plotSelector.setCurrentIndex(3)
        self.updatePlotWfsupdatePlotWfs()
 
    def updateAtmosPanel(self) -> None:
        nscreen = self.uiExpertuiExpert.wao_selectAtmosLayer.currentIndex()
        if (nscreen < 0):
            nscreen = 0
        self.uiExpertuiExpert.wao_r0.setValue(self.simsim.config.p_atmos.r0)
        self.uiExpertuiExpert.wao_atmosNlayers.setValue(self.simsim.config.p_atmos.nscreens)
        self.uiExpertuiExpert.wao_atmosAlt.setValue(self.simsim.config.p_atmos.alt[nscreen])
        self.uiExpertuiExpert.wao_atmosFrac.setValue(self.simsim.config.p_atmos.frac[nscreen])
        self.uiExpertuiExpert.wao_atmosL0.setValue(self.simsim.config.p_atmos.L0[nscreen])
        self.uiExpertuiExpert.wao_windSpeed.setValue(self.simsim.config.p_atmos.windspeed[nscreen])
        self.uiExpertuiExpert.wao_windDirection.setValue(
                self.simsim.config.p_atmos.winddir[nscreen])
        if (self.simsim.config.p_atmos.dim_screens is not None):
            self.uiExpertuiExpert.wao_atmosDimScreen.setText(
                    str(self.simsim.config.p_atmos.dim_screens[nscreen]))
        self.uiExpertuiExpert.wao_atmosWindow.canvas.axes.cla()
        width = (self.simsim.config.p_atmos.alt.max() / 20. + 0.1) / 1000.
        self.uiExpertuiExpert.wao_atmosWindow.canvas.axes.barh(
                self.simsim.config.p_atmos.alt / 1000. - width / 2.,
                self.simsim.config.p_atmos.frac, width, color="blue")
        self.uiExpertuiExpert.wao_atmosWindow.canvas.draw()
 
    def updateRtcPanel(self) -> None:
        # Centroider panel
        ncentro = self.uiExpertuiExpert.wao_selectCentro.currentIndex()
        if (ncentro < 0):
            ncentro = 0
        type = self.simsim.config.p_centroiders[ncentro].type
        self.uiExpertuiExpert.wao_centroTypeSelector.setCurrentIndex(
                self.uiExpertuiExpert.wao_centroTypeSelector.findText(str(type)))
 
        self.uiExpertuiExpert.wao_centroThresh.setValue(
                self.simsim.config.p_centroiders[ncentro].thresh)
        if type == scons.CentroiderType.BPCOG:
            self.uiExpertuiExpert.wao_centroNbrightest.setValue(
                    self.simsim.config.p_centroiders[ncentro].nmax)
        if type == scons.CentroiderType.TCOG:
            self.uiExpertuiExpert.wao_centroThresh.setValue(
                    self.simsim.config.p_centroiders[ncentro].thresh)
        if type in [scons.CentroiderType.CORR, scons.CentroiderType.WCOG]:
            self.uiExpertuiExpert.wao_centroFunctionSelector.setCurrentIndex(
                    self.uiExpertuiExpert.wao_centroFunctionSelector.findText(
                            str(self.simsim.config.p_centroiders[ncentro].type_fct)))
            self.uiExpertuiExpert.wao_centroWidth.setValue(
                    self.simsim.config.p_centroiders[ncentro].width)
 
        # Controller panel
        type_contro = self.simsim.config.p_controllers[0].type
        if (type_contro == scons.ControllerType.LS and
                    self.simsim.config.p_controllers[0].modopti == 0):
            self.uiExpertuiExpert.wao_controlTypeSelector.setCurrentIndex(0)
        elif (type_contro == scons.ControllerType.MV):
            self.uiExpertuiExpert.wao_controlTypeSelector.setCurrentIndex(1)
        elif (type_contro == scons.ControllerType.GEO):
            self.uiExpertuiExpert.wao_controlTypeSelector.setCurrentIndex(2)
        elif (type_contro == scons.ControllerType.LS and
              self.simsim.config.p_controllers[0].modopti):
            self.uiExpertuiExpert.wao_controlTypeSelector.setCurrentIndex(3)
        elif (type_contro == scons.ControllerType.CURED):
            self.uiExpertuiExpert.wao_controlTypeSelector.setCurrentIndex(4)
        elif (type_contro == scons.ControllerType.GENERIC):
            self.uiExpertuiExpert.wao_controlTypeSelector.setCurrentIndex(5)
        else:
            print("Controller type enumeration invalid.")
 
        self.uiExpertuiExpert.wao_controlCond.setValue(self.simsim.config.p_controllers[0].maxcond)
        self.uiExpertuiExpert.wao_controlDelay.setValue(self.simsim.config.p_controllers[0].delay)
        self.uiExpertuiExpert.wao_controlGain.setValue(self.simsim.config.p_controllers[0].gain)
        self.uiExpertuiExpert.wao_controlTTcond.setValue(
                self.simsim.config.p_controllers[0].maxcond)  # TODO : TTcond
 
    def updateTargetPanel(self) -> None:
        ntarget = self.uiExpertuiExpert.wao_selectTarget.currentIndex()
        if (ntarget < 0):
            ntarget = 0
        self.uiExpertuiExpert.wao_numberofTargets.setText(str(len(self.simsim.config.p_targets)))
        self.uiExpertuiExpert.wao_targetMag.setValue(self.simsim.config.p_target[ntarget].mag)
        self.uiExpertuiExpert.wao_targetXpos.setValue(self.simsim.config.p_target[ntarget].xpos)
        self.uiExpertuiExpert.wao_targetYpos.setValue(self.simsim.config.p_target[ntarget].ypos)
        self.uiExpertuiExpert.wao_targetLambda.setValue(self.simsim.config.p_target[ntarget].Lambda)
 
    def updatePanels(self) -> None:
        self.updateTelescopePanelupdateTelescopePanel()
        self.updateAtmosPanelupdateAtmosPanel()
        self.updateWfsPanelupdateWfsPanel()
        self.updateDmPanelupdateDmPanel()
        self.updateRtcPanelupdateRtcPanel()
        self.updateTargetPanelupdateTargetPanel()
 
    def setTelescopeParams(self) -> None:
        self.simsim.config.p_tel.set_diam(self.uiExpertuiExpert.wao_diamTel.value())
        self.simsim.config.p_tel.set_cobs(self.uiExpertuiExpert.wao_cobs.value())
        self.simsim.config.p_geom.set_zenithangle(self.uiExpertuiExpert.wao_zenithAngle.value())
        print("New telescope parameters set")
 
    def setAtmosParams(self) -> None:
        nscreen = self.uiExpertuiExpert.wao_selectAtmosLayer.currentIndex()
        if (nscreen < 0):
            nscreen = 0
        self.simsim.config.p_atmos.alt[nscreen] = self.uiExpertuiExpert.wao_atmosAlt.value()
        self.simsim.config.p_atmos.frac[nscreen] = self.uiExpertuiExpert.wao_atmosFrac.value()
        self.simsim.config.p_atmos.L0[nscreen] = self.uiExpertuiExpert.wao_atmosL0.value()
        self.simsim.config.p_atmos.windspeed[nscreen] = self.uiExpertuiExpert.wao_windSpeed.value()
        self.simsim.config.p_atmos.winddir[nscreen] = self.uiExpertuiExpert.wao_windDirection.value(
        )
        print("New atmos parameters set")
 
    def setRtcParams(self) -> None:
        # Centroider params
        ncentro = self.uiExpertuiExpert.wao_selectCentro.currentIndex()
        if (ncentro < 0):
            ncentro = 0
        self.simsim.config.p_centroiders[ncentro].set_type(
                str(self.uiExpertuiExpert.wao_centroTypeSelector.currentText()))
        self.simsim.config.p_centroiders[ncentro].set_thresh(
                self.uiExpertuiExpert.wao_centroThresh.value())
        self.simsim.config.p_centroiders[ncentro].set_nmax(
                self.uiExpertuiExpert.wao_centroNbrightest.value())
        self.simsim.config.p_centroiders[ncentro].set_thresh(
                self.uiExpertuiExpert.wao_centroThresh.value())
        self.simsim.config.p_centroiders[ncentro].set_type_fct(
                str(self.uiExpertuiExpert.wao_centroFunctionSelector.currentText()))
        self.simsim.config.p_centroiders[ncentro].set_width(
                self.uiExpertuiExpert.wao_centroWidth.value())
 
        # Controller panel
        type_contro = str(self.uiExpertuiExpert.wao_controlTypeSelector.currentText())
        if (type_contro == "LS"):
            self.simsim.config.p_controllers[0].set_type(scons.ControllerType.LS)
        elif (type_contro == "MV"):
            self.simsim.config.p_controllers[0].set_type(scons.ControllerType.MV)
        elif (type_contro == "PROJ"):
            self.simsim.config.p_controllers[0].set_type(scons.ControllerType.GEO)
        elif (type_contro == "OptiMods"):
            self.simsim.config.p_controllers[0].set_type(scons.ControllerType.LS)
            self.simsim.config.p_controllers[0].set_modopti(1)
 
        self.simsim.config.p_controllers[0].set_maxcond(
                self.uiExpertuiExpert.wao_controlCond.value())
        self.simsim.config.p_controllers[0].set_delay(
                self.uiExpertuiExpert.wao_controlDelay.value())
        self.simsim.config.p_controllers[0].set_gain(self.uiExpertuiExpert.wao_controlGain.value())
        # self.sim.config.p_controllers[0].set_TTcond(self.uiExpert.wao_controlTTcond.value())
        # # TODO : TTcond
        print("New RTC parameters set")
 
    def setWfsParams(self) -> None:
        nwfs = self.uiExpertuiExpert.wao_selectWfs.currentIndex()
        if (nwfs < 0):
            nwfs = 0
        self.simsim.config.p_wfss[nwfs].set_nxsub(self.uiExpertuiExpert.wao_wfsNxsub.value())
        self.simsim.config.p_wfss[nwfs].set_npix(self.uiExpertuiExpert.wao_wfsNpix.value())
        self.simsim.config.p_wfss[nwfs].set_pixsize(self.uiExpertuiExpert.wao_wfsPixSize.value())
        self.simsim.config.p_wfss[nwfs].set_xpos(self.uiExpertuiExpert.wao_wfsXpos.value())
        self.simsim.config.p_wfss[nwfs].set_ypos(self.uiExpertuiExpert.wao_wfsYpos.value())
        self.simsim.config.p_wfss[nwfs].set_fracsub(self.uiExpertuiExpert.wao_wfsFracsub.value())
        self.simsim.config.p_wfss[nwfs].set_Lambda(self.uiExpertuiExpert.wao_wfsLambda.value())
        self.simsim.config.p_wfss[nwfs].set_gsmag(self.uiExpertuiExpert.wao_wfsMagnitude.value())
        # TODO: find a way to correctly set zerop (limited by the maximum value
        # allowed by the double spin box)
        self.simsim.config.p_wfss[nwfs].set_zerop(10**(self.uiExpertuiExpert.wao_wfsZp.value()))
        self.simsim.config.p_wfss[nwfs].set_optthroughput(
                self.uiExpertuiExpert.wao_wfsThrough.value())
        self.simsim.config.p_wfss[nwfs].set_noise(self.uiExpertuiExpert.wao_wfsNoise.value())
 
        # LGS params
        if (self.uiExpertuiExpert.wao_wfsIsLGS.isChecked()):
            self.simsim.config.p_wfss[nwfs].set_gsalt(self.uiExpertuiExpert.wao_wfsGsAlt.value())
            self.simsim.config.p_wfss[nwfs].set_lltx(self.uiExpertuiExpert.wao_wfsLLTx.value())
            self.simsim.config.p_wfss[nwfs].set_llty(self.uiExpertuiExpert.wao_wfsLLTy.value())
            self.simsim.config.p_wfss[nwfs].set_laserpower(
                    self.uiExpertuiExpert.wao_wfsLGSpower.value())
            self.simsim.config.p_wfss[nwfs].set_lgsreturnperwatt(
                    self.uiExpertuiExpert.wao_wfsReturnPerWatt.value())
            self.simsim.config.p_wfss[nwfs].set_beamsize(
                    self.uiExpertuiExpert.wao_wfsBeamSize.value())
            self.simsim.config.p_wfss[nwfs].set_proftype(
                    str(self.uiExpertuiExpert.wao_selectLGSProfile.currentText()))
        print("New WFS parameters set")
 
    def setDmParams(self) -> None:
        ndm = self.uiExpertuiExpert.wao_selectDM.currentIndex()
        if (ndm < 0):
            ndm = 0
        self.simsim.config.p_dms[ndm].set_type(
                str(self.uiExpertuiExpert.wao_dmTypeSelector.currentText()))
        self.simsim.config.p_dms[ndm].set_alt(self.uiExpertuiExpert.wao_dmAlt.value())
        self.simsim.config.p_dms[ndm].set_nact(self.uiExpertuiExpert.wao_dmNactu.value())
        self.simsim.config.p_dms[ndm].set_unitpervolt(
                self.uiExpertuiExpert.wao_dmUnitPerVolt.value())
        self.simsim.config.p_dms[ndm].set_coupling(self.uiExpertuiExpert.wao_dmCoupling.value())
        self.simsim.config.p_dms[ndm].set_thresh(self.uiExpertuiExpert.wao_dmThresh.value())
        print("New DM parameters set")
 
    def reset_dm(self) -> None:
        if (self.simsim.dms):
            ndm = self.uiExpertuiExpert.wao_selectDM.currentIndex()
            if (ndm > -1):
                self.simsim.dms.resetdm(
                        str(self.uiExpertuiExpert.wao_dmTypeSelector.currentText()),
                        self.uiExpertuiExpert.wao_dmAlt.value())
                self.updateDisplay()
                print("DM #%d reset" % ndm)
            else:
                print("Invalid DM : please select a DM to reset")
        else:
            print("There is no DM to reset")
 
    def BttCommand(self) -> None:
        if (self.simsim.rtc):
            nfilt = int(self.uiExpertuiExpert.wao_filterBtt.value())
            ao.command_on_Btt(self.simsim.rtc, self.simsim.dms, self.simsim.config.p_dms,
                              self.simsim.config.p_geom, nfilt)
            print("Loop is commanded from Btt basis now")
 
    def KLCommand(self) -> None:
        if (self.simsim.rtc):
            nfilt = int(self.uiExpertuiExpert.wao_filterBtt.value())
            cmat = ao.command_on_KL(
                    self.simsim.rtc, self.simsim.dms, self.simsim.config.p_controllers[0],
                    self.simsim.config.p_dms, self.simsim.config.p_geom,
                    self.simsim.config.p_atmos, self.simsim.config.p_tel, nfilt)
            self.simsim.rtc.set_cmat(0, cmat.astype(np.float32))
            print("Loop is commanded from KL basis now")
 
    def updatePlotWfs(self) -> None:
        typeText = str(self.uiExpertuiExpert.wao_wfs_plotSelector.currentText())
        n = self.uiExpertuiExpert.wao_selectWfs.currentIndex()
        self.uiExpertuiExpert.wao_wfsWindow.canvas.axes.clear()
        ax = self.uiExpertuiExpert.wao_wfsWindow.canvas.axes
        if ((self.simsim.config.p_wfss[n].type == scons.WFSType.PYRHR or
             self.simsim.config.p_wfss[n].type == scons.WFSType.PYRLR) and
                    typeText == "Pyramid mod. pts" and self.simsim.is_init):
            scale_fact = 2 * np.pi / self.simsim.config.p_wfss[n]._Nfft * \
                self.simsim.config.p_wfss[n].Lambda * \
                    1e-6 / self.simsim.config.p_tel.diam / \
                    self.simsim.config.p_wfss[n]._qpixsize * CONST.RAD2ARCSEC
            cx = self.simsim.config.p_wfss[n]._pyr_cx / scale_fact
            cy = self.simsim.config.p_wfss[n]._pyr_cy / scale_fact
            ax.scatter(cx, cy)
 
    def displayRtcMatrix(self) -> None:
        if not self.simsim.is_init:
            # print(" widget not fully initialized")
            return
 
        data = None
        if (self.simsim.rtc):
            type_matrix = str(self.uiExpertuiExpert.wao_selectRtcMatrix.currentText())
            if (
                    type_matrix == "imat" and
                    self.simsim.config.p_controllers[0].type != scons.ControllerType.GENERIC
                    and
                    self.simsim.config.p_controllers[0].type != scons.ControllerType.GEO):
                data = self.simsim.rtc.get_imat(0)
            elif (type_matrix == "cmat"):
                data = self.simsim.rtc.get_cmat(0)
            elif (type_matrix == "Eigenvalues"):
                if (self.simsim.config.p_controllers[0].type == scons.ControllerType.LS or
                            self.simsim.config.p_controllers[0].type == scons.
                            ControllerType.MV):
                    data = self.simsim.rtc.getEigenvals(0)
            elif (type_matrix == "Cmm" and
                  self.simsim.config.p_controllers[0].type == scons.ControllerType.MV):
                tmp = self.simsim.rtc.get_cmm(0)
                ao.do_tomo_matrices(0, self.simsim.rtc, self.simsim.config.p_wfss,
                                    self.simsim.dms, self.simsim.atm, self.simsim.wfs,
                                    self.simsim.config.p_rtc, self.simsim.config.p_geom,
                                    self.simsim.config.p_dms, self.simsim.config.p_tel,
                                    self.simsim.config.p_atmos)
                data = self.simsim.rtc.get_cmm(0)
                self.simsim.rtc.set_cmm(0, tmp)
            elif (type_matrix == "Cmm inverse" and
                  self.simsim.config.p_controllers[0].type == "mv"):
                data = self.simsim.rtc.get_cmm(0)
            elif (type_matrix == "Cmm eigen" and
                  self.simsim.config.p_controllers[0].type == "mv"):
                data = self.simsim.rtc.getCmmEigenvals(0)
            elif (type_matrix == "Cphim" and
                  self.simsim.config.p_controllers[0].type == "mv"):
                data = self.simsim.rtc.get_cphim(0)
 
            if (data is not None):
                self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.clear()
                ax = self.uiExpertuiExpert.wao_rtcWindow.canvas.axes
                if (len(data.shape) == 2):
                    self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.matshow(
                            data, aspect="auto", origin="lower", cmap="gray")
                elif (len(data.shape) == 1):
                    # TODO : plot it properly, interactivity ?
                    self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.plot(
                            list(range(len(data))), data, color="black")
                    ax.set_yscale('log')
                    if (type_matrix == "Eigenvalues"):
                        #    major_ticks = np.arange(0, 101, 20)
                        #    minor_ticks = np.arange(0, 101, 5)
 
                        #    self.uiExpert.wao_rtcWindow.canvas.axes.set_xticks(major_ticks)
                        #    self.uiExpert.wao_rtcWindow.canvas.axes.set_xticks(minor_ticks, minor=True)
                        #    self.uiExpert.wao_rtcWindow.canvas.axes.set_yticks(major_ticks)
                        #    self.uiExpert.wao_rtcWindow.canvas.axes.set_yticks(minor_ticks, minor=True)
 
                        # and a corresponding grid
 
                        self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.grid(which='both')
 
                        # or if you want differnet settings for the grids:
                        self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.grid(
                                which='minor', alpha=0.2)
                        self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.grid(
                                which='major', alpha=0.5)
                        nfilt = self.simsim.rtc.get_nfiltered(0, self.simsim.config.p_rtc)
                        self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.plot([
                                nfilt - 0.5, nfilt - 0.5
                        ], [data.min(), data.max()], color="red", linestyle="dashed")
                        if (nfilt > 0):
                            self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.scatter(
                                    np.arange(0, nfilt, 1), data[0:nfilt], color="red"
                            )  # TODO : plot it properly, interactivity ?
                            self.uiExpertuiExpert.wao_rtcWindow.canvas.axes.scatter(
                                    np.arange(nfilt, len(data),
                                              1), data[nfilt:], color="blue"
                            )  # TODO : plot it properly, interactivity ?
                            tt = "%d modes Filtered" % nfilt
                            # ax.text(nfilt + 2, data[nfilt-1], tt)
                            ax.text(0.5, 0.2, tt, horizontalalignment='center',
                                    verticalalignment='center', transform=ax.transAxes)
 
                self.uiExpertuiExpert.wao_rtcWindow.canvas.draw()
 
    def computeDMrange(self, numdm: int, numwfs: int, push4imat: float = None,
                       unitpervolt: float = None) -> float:
        i = numdm
        if (push4imat is None or push4imat == 0):
            push4imat = self.simsim.config.p_dms[i].push4imat
        if (unitpervolt is None or unitpervolt == 0):
            unitpervolt = self.simsim.config.p_dms[i].unitpervolt
 
        actuPushInMicrons = push4imat * unitpervolt
        coupling = self.simsim.config.p_dms[i].coupling
        a = coupling * actuPushInMicrons
        b = 0
        c = actuPushInMicrons
        d = coupling * actuPushInMicrons
        if (self.simsim.config.p_dms[i].type is not scons.DmType.TT):
            dist = self.simsim.config.p_tel.diam
        else:
            dist = self.simsim.config.p_tel.diam / \
                self.simsim.config.p_wfss[numwfs].nxsub
        Delta = (1e-6 * (((c + d) / 2) - ((a + b) / 2)))
        actuPushInArcsecs = CONST.RAD2ARCSEC * Delta / dist
        return actuPushInArcsecs