Inheritance diagram for shesha.widgets.widget_bench.widgetBenchWindow:

Collaboration diagram for shesha.widgets.widget_bench.widgetBenchWindow:

Public Member Functions
None	__init__ (self, Any config_file=None, bool brahma=False, str devices=None)

def	updateForever (self, state)
	METHODS #. More...

None	add_dispDock (self, str name, parent, str type="pg_image")

None	load_config (self)
	Callback when 'LOAD' button is hit. More...

None	aoLoopClicked (self, bool pressed)

None	aoLoopOpen (self, bool pressed)

None	init_config (self)

None	init_configThread (self)

None	init_configFinished (self)

None	updateDisplay (self)

None	loop_once (self)

def	run (self)

Public Member Functions inherited from shesha.widgets.widget_base.WidgetBase
None	__init__ (self, parent=None, hide_histograms=False)

None	gui_timer_config (self, state)

None	closeEvent (self, Any event)

None	quitGUI (self, Any event=None)

def	saveArea (self, widget, filename=None)
	METHODS #. More...

def	showDock (self, name)

def	restoreMyState (self, state)

def	loadArea (self, widget=None, filename=None)

None	addConfigFromFile (self)

def	update_displayDock (self)

None	load_config (self, args, *kwargs)
	Callback when 'LOAD' button is hit. More...

None	loadDefaultConfig (self)

def	addSHGrid (self, pg_image, valid_sub, sspsize, pitch)

None	printInPlace (self, str text)

Public Attributes
	brahma

	devices

	uiBench

	supervisor
	ATTRIBUTES #. More...

	config

	stop

	nbiter

	refreshTime

	loopThread

	assistant

	defaultParPath

	defaultAreaPath

	dispStatsInTerminal

	nwfs

Public Attributes inherited from shesha.widgets.widget_base.WidgetBase
	uiBase

	gui_timer
	ATTRIBUTES #. More...

	loopLock

	hide_histograms

	area
	PYQTGRAPH DockArea INIT #. More...

	gridSH

	defaultParPath

	defaultAreaPath

	disp_checkboxes

	docks

	viewboxes

	imgs

	hists

	PupilLines

	wao_phasesgroup_cb

	wao_graphgroup_cb

	wao_imagesgroup_cb

	thread

Detailed Description

Definition at line 86 of file widget_bench.py.

Constructor & Destructor Documentation

◆ init()

None shesha.widgets.widget_bench.widgetBenchWindow.__init__	(		self,
		Any	config_file = `None`,
		bool	brahma = `False`,
		str	devices = `None`
	)

Definition at line 88 of file widget_bench.py.

     def __init__(self, config_file: Any = None, brahma: bool = False,
                  devices: str = None) -> None:
         WidgetBase.__init__(self)
         BenchClassTemplate.__init__(self)
  
         self.brahma = brahma
         self.devices = devices
  
         self.uiBench = BenchWindowTemplate()
         self.uiBench.setupUi(self)
  
         
  
         self.supervisor = None
         self.config = None
         self.stop = False  # type: bool  # Request quit
  
         self.uiBench.wao_nbiters.setValue(1000)  # Default GUI nIter box value
         self.nbiter = self.uiBench.wao_nbiters.value()
         self.refreshTime = 0  # type: float  # System time at last display refresh
         self.loopThread = None  # type: QThread
         self.assistant = None  # type: Any
  
         
         self.defaultParPath = os.environ["SHESHA_ROOT"] + "/data/par/bench-config"
         self.defaultAreaPath = os.environ["SHESHA_ROOT"] + "/data/layouts"
         self.loadDefaultConfig()
  
         self.uiBench.wao_run.setCheckable(True)
         self.uiBench.wao_run.clicked[bool].connect(self.aoLoopClicked)
         self.uiBench.wao_open_loop.setCheckable(True)
         self.uiBench.wao_open_loop.clicked[bool].connect(self.aoLoopOpen)
         self.uiBench.wao_next.clicked.connect(self.loop_once)
  
         self.uiBench.wao_forever.stateChanged.connect(self.updateForever)
  
         self.dispStatsInTerminal = False
  
         self.uiBench.wao_run.setDisabled(True)
         self.uiBench.wao_next.setDisabled(True)
         self.uiBench.wao_unzoom.setDisabled(True)
  
         self.addDockWidget(Qt.DockWidgetArea(1), self.uiBase.wao_ConfigDock)
         self.addDockWidget(Qt.DockWidgetArea(1), self.uiBase.wao_DisplayDock)
         self.uiBase.wao_ConfigDock.setFloating(False)
         self.uiBase.wao_DisplayDock.setFloating(False)
  
         self.adjustSize()
  
         if config_file is not None:
             self.uiBase.wao_selectConfig.clear()
             self.uiBase.wao_selectConfig.addItem(config_file)
             self.load_config()
             self.init_config()
  

Member Function Documentation

◆ add_dispDock()

None shesha.widgets.widget_bench.widgetBenchWindow.add_dispDock	(		self,
		str	name,
			parent,
		str	type = `"pg_image"`
	)

Reimplemented from shesha.widgets.widget_base.WidgetBase.

Definition at line 158 of file widget_bench.py.

     def add_dispDock(self, name: str, parent, type: str = "pg_image") -> None:
         d = WidgetBase.add_dispDock(self, name, parent, type)
         if type == "SR":
             d.addWidget(self.uiBench.wao_Strehl)
  

◆ aoLoopClicked()

None shesha.widgets.widget_bench.widgetBenchWindow.aoLoopClicked	(		self,
		bool	pressed
	)

Definition at line 207 of file widget_bench.py.

     def aoLoopClicked(self, pressed: bool) -> None:
         if pressed:
             self.stop = False
             self.refreshTime = time.time()
             self.nbiter = self.uiBench.wao_nbiters.value()
             if self.dispStatsInTerminal:
                 if self.uiBench.wao_forever.isChecked():
                     print("LOOP STARTED")
                 else:
                     print("LOOP STARTED FOR %d iterations" % self.nbiter)
             self.run()
         else:
             self.stop = True
  

Here is the call graph for this function:

◆ aoLoopOpen()

None shesha.widgets.widget_bench.widgetBenchWindow.aoLoopOpen	(		self,
		bool	pressed
	)

Definition at line 221 of file widget_bench.py.

     def aoLoopOpen(self, pressed: bool) -> None:
         if (pressed):
             self.supervisor.open_loop()
             self.uiAO.wao_open_loop.setText("Close Loop")
         else:
             self.supervisor.close_loop()
             self.uiAO.wao_open_loop.setText("Open Loop")
  

◆ init_config()

None shesha.widgets.widget_bench.widgetBenchWindow.init_config ( self )

Reimplemented from shesha.widgets.widget_base.WidgetBase.

Definition at line 229 of file widget_bench.py.

     def init_config(self) -> None:
         WidgetBase.init_config(self)
  

Here is the caller graph for this function:

◆ init_configFinished()

None shesha.widgets.widget_bench.widgetBenchWindow.init_configFinished ( self )

Reimplemented from shesha.widgets.widget_base.WidgetBase.

Definition at line 241 of file widget_bench.py.

     def init_configFinished(self) -> None:
         # Thread carmaWrap context reload:
         try:
             self.supervisor.force_context()
         except:
             print('Warning: could not call supervisor.force_context().')
  
         for i in range(self.nwfs):
             if self.config.p_wfss[i].type == WFSType.SH:
                 key = "wfs_%d" % i
                 self.addSHGrid(self.docks[key].widgets[0],
                                self.config.p_wfss[i].get_validsub(),
                                self.config.p_wfss[i].npix, self.config.p_wfss[i].npix)
  
         self.updateDisplay()
  
         self.uiBench.wao_run.setDisabled(False)
         self.uiBench.wao_next.setDisabled(False)
         self.uiBench.wao_open_loop.setDisabled(False)
         self.uiBench.wao_unzoom.setDisabled(False)
  
         WidgetBase.init_configFinished(self)
  

Here is the call graph for this function:

◆ init_configThread()

None shesha.widgets.widget_bench.widgetBenchWindow.init_configThread ( self )

Reimplemented from shesha.widgets.widget_base.WidgetBase.

Definition at line 232 of file widget_bench.py.

     def init_configThread(self) -> None:
         # self.uiBench.wao_deviceNumber.setDisabled(True)
         # self.config.p_loop.devices = self.uiBench.wao_deviceNumber.value()  # using GUI value
         # gpudevice = "ALL"  # using all GPU avalaible
         # gpudevice = np.array([2, 3], dtype=np.int32)
         # gpudevice = np.arange(4, dtype=np.int32) # using 4 GPUs: 0-3
         # gpudevice = 0  # using 1 GPU : 0
         self.supervisor.init_config()
  

Here is the call graph for this function:

◆ load_config()

None shesha.widgets.widget_bench.widgetBenchWindow.load_config ( self )

Callback when 'LOAD' button is hit.

Definition at line 166 of file widget_bench.py.

         '''
         WidgetBase.load_config(self)
         config_file = str(self.uiBase.wao_selectConfig.currentText())
         sys.path.insert(0, self.defaultParPath)
  
         self.supervisor = Supervisor(config_file, self.brahma)
  
         self.config = self.supervisor.get_config()
  
         # if self.devices:
         #     self.config.p_loop.set_devices([
         #             int(device) for device in self.devices.split(",")
         #     ])
  
         try:
             sys.path.remove(self.defaultParPath)
         except:
             pass
  
         try:
             self.nwfs = len(self.config.p_wfss)
         except:
             self.nwfs = 1  # Default. config may very well not define p_wfss
         for wfs in range(self.nwfs):
             name = 'slpComp_%d' % wfs
             self.add_dispDock(name, self.wao_graphgroup_cb, "MPL")
             name = 'wfs_%d' % wfs
             self.add_dispDock(name, self.wao_imagesgroup_cb)
  
         self.uiBench.wao_run.setDisabled(True)
         self.uiBench.wao_next.setDisabled(True)
         self.uiBench.wao_unzoom.setDisabled(True)
  
         self.uiBase.wao_init.setDisabled(False)
  
         if (hasattr(self.config, "layout")):
             area_filename = self.defaultAreaPath + "/" + self.config.layout + ".area"
             self.loadArea(filename=area_filename)
  
         self.adjustSize()
  

◆ loop_once()

None shesha.widgets.widget_bench.widgetBenchWindow.loop_once ( self )

Definition at line 311 of file widget_bench.py.

     def loop_once(self) -> None:
         if not self.loopLock.acquire(False):
             print("Display locked")
             return
         else:
             try:
                 start = time.time()
                 self.supervisor.single_next()
                 loopTime = time.time() - start
                 refreshDisplayTime = 1. / self.uiBase.wao_frameRate.value()
  
                 if (time.time() - self.refreshTime > refreshDisplayTime):
                     currentFreq = 1 / loopTime
                     self.uiBench.wao_currentFreq.setValue(currentFreq)
                     self.refreshTime = start
             except:
                 pass
             finally:
                 self.loopLock.release()
  

Here is the caller graph for this function:

◆ run()

def shesha.widgets.widget_bench.widgetBenchWindow.run ( self )

Reimplemented from shesha.widgets.widget_base.WidgetBase.

Definition at line 331 of file widget_bench.py.

     def run(self):
         WidgetBase.run(self)
         if not self.uiBench.wao_forever.isChecked():
             self.nbiter -= 1
  
         if self.nbiter <= 0:
             self.stop = True
             self.uiBench.wao_run.setChecked(False)
  
  

◆ updateDisplay()

None shesha.widgets.widget_bench.widgetBenchWindow.updateDisplay ( self )

Reimplemented from shesha.widgets.widget_base.WidgetBase.

Definition at line 264 of file widget_bench.py.

     def updateDisplay(self) -> None:
         if (self.supervisor is None) or (not self.supervisor.is_init()) or (
                 not self.uiBase.wao_Display.isChecked()):
             # print("Widget not fully initialized")
             return
         if not self.loopLock.acquire(False):
             return
         else:
             try:
                 for key, dock in self.docks.items():
                     if key == "Strehl":
                         continue
                     elif dock.isVisible():
                         index = int(key.split("_")[-1])
                         data = None
                         if "wfs" in key:
                             data = self.supervisor.wfs.get_wfs_image(index)
                             if (data is not None):
                                 autoscale = True  # self.uiBench.actionAuto_Scale.isChecked()
                                 # if (autoscale):
                                 #     # inits levels
                                 #     self.hist.setLevels(data.min(), data.max())
                                 self.imgs[key].setImage(data, autoLevels=autoscale)
                                 # self.p1.autoRange()
  
                         elif "slp" in key:  # Slope display
                             if (self.config.p_wfss[index].type == WFSType.PYRHR or
                                         self.config.p_wfss[index].type == WFSType.PYRLR):
                                 raise RuntimeError("PYRHR not usable")
                             self.imgs[key].canvas.axes.clear()
                             x, y = self.supervisor.config.p_wfss[index].get_validsub()
  
                             nssp = x.size
                             centroids = self.supervisor.rtc.get_slopes()
                             vx = centroids[:nssp]
                             vy = centroids[nssp:]
  
                             offset = (self.supervisor.config.p_wfss[index].npix - 1) / 2
                             self.imgs[key].canvas.axes.quiver(
                                     x + offset, y + offset, vy, vx, angles='xy',
                                     scale_units='xy',
                                     scale=1)  # wao.supervisor.config.p_wfss[0].pixsize)
                             self.imgs[key].canvas.draw()
  
             finally:
                 self.loopLock.release()
  

Here is the caller graph for this function:

◆ updateForever()

def shesha.widgets.widget_bench.widgetBenchWindow.updateForever	(	self,
		state
	)

METHODS #.

Definition at line 155 of file widget_bench.py.

     def updateForever(self, state):
         self.uiBench.wao_nbiters.setDisabled(state)
  

Member Data Documentation

◆ assistant

shesha.widgets.widget_bench.widgetBenchWindow.assistant

Definition at line 110 of file widget_bench.py.

◆ brahma

shesha.widgets.widget_bench.widgetBenchWindow.brahma

Definition at line 92 of file widget_bench.py.

◆ config

shesha.widgets.widget_bench.widgetBenchWindow.config

Definition at line 103 of file widget_bench.py.

◆ defaultAreaPath

shesha.widgets.widget_bench.widgetBenchWindow.defaultAreaPath

Definition at line 117 of file widget_bench.py.

◆ defaultParPath

shesha.widgets.widget_bench.widgetBenchWindow.defaultParPath

            CONNECTED BUTTONS                         #

Default path for config files

Definition at line 116 of file widget_bench.py.

◆ devices

shesha.widgets.widget_bench.widgetBenchWindow.devices

Definition at line 93 of file widget_bench.py.

◆ dispStatsInTerminal

shesha.widgets.widget_bench.widgetBenchWindow.dispStatsInTerminal

Definition at line 128 of file widget_bench.py.

◆ loopThread

shesha.widgets.widget_bench.widgetBenchWindow.loopThread

Definition at line 109 of file widget_bench.py.

◆ nbiter

shesha.widgets.widget_bench.widgetBenchWindow.nbiter

Definition at line 107 of file widget_bench.py.

◆ nwfs

shesha.widgets.widget_bench.widgetBenchWindow.nwfs

Definition at line 186 of file widget_bench.py.

◆ refreshTime

shesha.widgets.widget_bench.widgetBenchWindow.refreshTime

Definition at line 108 of file widget_bench.py.

◆ stop

shesha.widgets.widget_bench.widgetBenchWindow.stop

Definition at line 104 of file widget_bench.py.

◆ supervisor

shesha.widgets.widget_bench.widgetBenchWindow.supervisor

ATTRIBUTES #.

Definition at line 102 of file widget_bench.py.

◆ uiBench

shesha.widgets.widget_bench.widgetBenchWindow.uiBench

Definition at line 95 of file widget_bench.py.

The documentation for this class was generated from the following file:

widget_bench.py

Public Member Functions

Public Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

Member Function Documentation

◆ add_dispDock()

◆ aoLoopClicked()

◆ aoLoopOpen()

◆ init_config()

◆ init_configFinished()

◆ init_configThread()

◆ load_config()

◆ loop_once()

◆ run()

◆ updateDisplay()

◆ updateForever()

Member Data Documentation

◆ assistant

◆ brahma

◆ config

◆ defaultAreaPath

◆ defaultParPath

◆ devices

◆ dispStatsInTerminal

◆ loopThread

◆ nbiter

◆ nwfs

◆ refreshTime

◆ stop

◆ supervisor

◆ uiBench

◆ init()