Ejemplo n.º 1
0
	def initUI(self):

		# If not in offline mode, do not initialize setup
		if not self.offline_mode:
			self.setup = Setup()
		
		# Create widgets, pass self so they can talk to each other
		print('Setting up widgets')

		# set up widgets
		if not self.offline_mode:
			self.axisControl    = AxisControl(self)
			self.savedPositions = SavedPositions(self)

			self.cameraWidget   = CameraWidget(self)
			
			self.mologramTab    = MologramWidget(self)
			self.saveImage      = SaveImage(self)
		else:
			self.axisControl    = QtGui.QLabel('')
			self.cameraWidget   = QtGui.QLabel('')
			self.savedPositions = QtGui.QLabel('')
			self.mologramTab    = QtGui.QLabel('')
			self.saveImage      = QtGui.QLabel('')

		# set up tabs
		if not self.offline_mode:
			self.imageAquisitionTab = ImageAquisitionTab(self)
			self.manualControlTab   = ManualControlTab(self)
			self.calibrationTab     = CalibrationTab(self)
			self.imageProcessingTab = ImageProcessingTab(self)
		else:
			self.imageAquisitionTab = QtGui.QLabel('')
			self.manualControlTab   = QtGui.QLabel('')
			self.calibrationTab     = QtGui.QLabel('')
			self.imageProcessingTab = QtGui.QLabel('')

		self.plotTab   			= PlotTab(self)
		self.postProcessingTab  = PostProcessingTab(self)
		print('Done setting up widgets')

		self.tabs_top = QtGui.QTabWidget()
		self.tabs_top.addTab(self.mologramTab,'Mologram Position')
		self.tabs_top.addTab(self.plotTab, 'Time Stack Plots')

		self.tabs_bot = QtGui.QTabWidget()
		self.tabs_bot.addTab(self.manualControlTab,'Manual Motor Control')
		self.tabs_bot.addTab(self.imageAquisitionTab, 'Image Acquisition')
		self.tabs_bot.addTab(self.calibrationTab, 'Calibration')
		self.tabs_bot.addTab(self.postProcessingTab, 'Post Processing')
		self.tabs_bot.addTab(self.imageProcessingTab, 'Image Processing')
	
		# declare layouts
		main_hbox  = QtGui.QHBoxLayout()
		left_vbox  = QtGui.QVBoxLayout()
		right_vbox = QtGui.QVBoxLayout()

		top_left_hbox = QtGui.QHBoxLayout()
		btm_left_vbox = QtGui.QVBoxLayout()

		v_splitter = QtGui.QSplitter(QtCore.Qt.Vertical)
		v_splitter.addWidget(self.tabs_top)
		v_splitter.addWidget(self.tabs_bot)

		h_splitter = QtGui.QSplitter(QtCore.Qt.Horizontal)
		h_splitter.addWidget(v_splitter)
		h_splitter.addWidget(self.cameraWidget)

		main_hbox.addWidget(h_splitter)

		self.setLayout(main_hbox)
Ejemplo n.º 2
0
class MainWidget(QtGui.QWidget):
	# Constructor
	def __init__(self,app):

		# Call parent class constructor
		super(MainWidget,self).__init__()

		# Initialize member variables		
		self.enable_image_processing = False
		self.app = app

		# Determine whether in offline mode
		self.offline_mode = self.ask_offline_mode()

		# Call child constructor
		self.initUI()
		self.set_offline_mode()

		# Update image and positions
		self.updateAll()

	# Initialization of GUI
	def initUI(self):

		# If not in offline mode, do not initialize setup
		if not self.offline_mode:
			self.setup = Setup()
		
		# Create widgets, pass self so they can talk to each other
		print('Setting up widgets')

		# set up widgets
		if not self.offline_mode:
			self.axisControl    = AxisControl(self)
			self.savedPositions = SavedPositions(self)

			self.cameraWidget   = CameraWidget(self)
			
			self.mologramTab    = MologramWidget(self)
			self.saveImage      = SaveImage(self)
		else:
			self.axisControl    = QtGui.QLabel('')
			self.cameraWidget   = QtGui.QLabel('')
			self.savedPositions = QtGui.QLabel('')
			self.mologramTab    = QtGui.QLabel('')
			self.saveImage      = QtGui.QLabel('')

		# set up tabs
		if not self.offline_mode:
			self.imageAquisitionTab = ImageAquisitionTab(self)
			self.manualControlTab   = ManualControlTab(self)
			self.calibrationTab     = CalibrationTab(self)
			self.imageProcessingTab = ImageProcessingTab(self)
		else:
			self.imageAquisitionTab = QtGui.QLabel('')
			self.manualControlTab   = QtGui.QLabel('')
			self.calibrationTab     = QtGui.QLabel('')
			self.imageProcessingTab = QtGui.QLabel('')

		self.plotTab   			= PlotTab(self)
		self.postProcessingTab  = PostProcessingTab(self)
		print('Done setting up widgets')

		self.tabs_top = QtGui.QTabWidget()
		self.tabs_top.addTab(self.mologramTab,'Mologram Position')
		self.tabs_top.addTab(self.plotTab, 'Time Stack Plots')

		self.tabs_bot = QtGui.QTabWidget()
		self.tabs_bot.addTab(self.manualControlTab,'Manual Motor Control')
		self.tabs_bot.addTab(self.imageAquisitionTab, 'Image Acquisition')
		self.tabs_bot.addTab(self.calibrationTab, 'Calibration')
		self.tabs_bot.addTab(self.postProcessingTab, 'Post Processing')
		self.tabs_bot.addTab(self.imageProcessingTab, 'Image Processing')
	
		# declare layouts
		main_hbox  = QtGui.QHBoxLayout()
		left_vbox  = QtGui.QVBoxLayout()
		right_vbox = QtGui.QVBoxLayout()

		top_left_hbox = QtGui.QHBoxLayout()
		btm_left_vbox = QtGui.QVBoxLayout()

		v_splitter = QtGui.QSplitter(QtCore.Qt.Vertical)
		v_splitter.addWidget(self.tabs_top)
		v_splitter.addWidget(self.tabs_bot)

		h_splitter = QtGui.QSplitter(QtCore.Qt.Horizontal)
		h_splitter.addWidget(v_splitter)
		h_splitter.addWidget(self.cameraWidget)

		main_hbox.addWidget(h_splitter)

		self.setLayout(main_hbox)

	def set_experiment_folder_path(self,experiment_folder_path):

		self.experiment_folder_path = experiment_folder_path

	def get_experiment_folder_path(self):

		return self.experiment_folder_path

	def ask_offline_mode(self):
		msg = QtGui.QMessageBox()
		msg.setWindowTitle("Offline mode")
		msg.setIcon(QtGui.QMessageBox.Question)
		msg.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
		msg.setText("Do you wish to start in offline mode?")
		retval = msg.exec_()

		return retval == QtGui.QMessageBox.Yes

	def set_offline_mode(self):

		# Disable tabs that are not useful in offline mode		
		self.tabs_top.setTabEnabled(0,not self.offline_mode)	# Mologram Position		
		self.tabs_bot.setTabEnabled(0,not self.offline_mode)	# Manual motor control		
		self.tabs_bot.setTabEnabled(1,not self.offline_mode)	# Image Acquisition		
		self.tabs_bot.setTabEnabled(2,not self.offline_mode)	# Calibration			
	#	self.tabs_bot.setTabEnabled(3,not self.offline_mode)	# Post Processing			
		self.tabs_bot.setTabEnabled(4,			# Image Processing
			not self.offline_mode and self.enable_image_processing)

	# linked to the main calibration button
	def calib(self):

		if not self.offline_mode:
			self.setup.calib()

			msg = QtGui.QMessageBox()
			msg.setWindowTitle("Calibration")
			msg.setIcon(QtGui.QMessageBox.Information)
			msg.setStandardButtons(QtGui.QMessageBox.Ok)

			msg.setText("Calibration is finshed")
			msg.exec_()

	def updatePos(self):
		if not self.offline_mode:
			self.axisControl.updatePos()
			self.savedPositions.updatePos()
			self.mologramTab.updateDisplay()

	def updateImg(self,Stack=False):

		time1 = time.time()

		if not self.offline_mode:

			self.cameraWidget.updateImage(Stack)
			
		time2 = time.time()
		self.plotTab.updateImage()
		time3 = time.time()

		print('Total update time: ' + str(time3-time1))
		if not self.offline_mode:
			print('  >> camera widget update time: ' + str(time2-time1))
		print('  >> plot tab update time:      ' + str(time3-time2))

		if self.enable_image_processing == True:
			self.imageProcessingTab.updateCanvas()


	def updateAll(self):
		print("default update all launched")
		self.updatePos()
		self.updateImg()

	def updateAllStack(self):
		# special update function during stack execution, because the camera should not
		# be used to get additional images otherwise the software crashes.
		self.updatePos()
		self.updateImg(Stack=True)

	def toggle_enable_image_processing(self):
		self.enable_image_processing = not(self.enable_image_processing)
		self.tabs_bot.setTabEnabled(4,self.enable_image_processing)
		self.updateImg()

	def redraw(self):
		self.app.processEvents()

	""" Tell the GUI that the current motor is in movemnt or not """
	def set_in_movement(self,state):

		# When in movement, disable buttons that would enable a simultaneous movement
		widget_list = [self.axisControl, self.mologramTab]

		for widget in widget_list:
			widget.set_in_movement(state)

	def stop_moving(self):
		self.set_in_movement(False)
		self.updateAll()