arjan.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
Copyright © 2013, W. van Ham, Radboud University Nijmegen
This file is part of Sleelab.

Sleelab is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Sleelab is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Sleelab.  If not, see <http://www.gnu.org/licenses/>.
'''

from __future__ import print_function
import logging, signal, argparse, csv, numpy as np
import OpenGL
OpenGL.ERROR_ON_COPY = True   # make sure we do not accidentally send other structures than numpy arrays
# PyQt (package python-qt4-gl on Ubuntu)
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyQt4.QtOpenGL import * 

# project files
from field import *
import utils
import qtriggerjoystick


class Main(QMainWindow):
	def __init__(self, args):
		super(Main, self).__init__()

		self.initUI()
		self.args = args
		QTimer.singleShot(0, self.processCommandLine)
		
	def quit(self, signum=None, frame=None):
		logging.info("quitting")
		self.field.quit()
		qApp.quit()

	def initUI(self):
		self.j = qtriggerjoystick.QTriggerJoystick()

		#contents
		self.field = Field(self)
		# alternative content
		self.text = QLabel("Paused")
		self.text.setStyleSheet('text-align: right; color:#a00000; background-color:#000000; font-size: 100pt; font-family: Sans-Serif;')
		self.text.setMinimumSize(1400, 525)
		
		# together in stack
		self.stack = QStackedWidget(self)
		self.stack.addWidget(self.field)
		self.stack.addWidget(self.text)
		self.setCentralWidget(self.stack)
		
		# dialogs
		self.errorMessageDialog = QErrorMessage(self)
		
		## menubar
		self.startIcon = QIcon('icon/start.png')
		self.stopIcon = QIcon('icon/pause.png')
		self.startAction = QAction(self.startIcon, '&Start/Stop', self)
		self.startAction.setShortcut(' ')
		self.startAction.setStatusTip('Start/Stop')
		self.startAction.triggered.connect(self.startStop)
		
		loadAction = QAction(QIcon('icon/load.png'), '&Load', self)
		loadAction.setShortcut('Ctrl+L')
		loadAction.setStatusTip('Load experiment file')
		loadAction.triggered.connect(self.load)

		exitAction = QAction(QIcon('icon/quit.png'), '&Exit', self)
		exitAction.setShortcut('Ctrl+Q')
		exitAction.setStatusTip('Quit application')
		#exitAction.triggered.connect(qApp.quit)
		exitAction.triggered.connect(self.quit)
		
		self.fullIcon = QIcon('icon/full.png')
		self.fullAction = QAction(self.fullIcon, '&Full Screen', self)
		self.fullAction.setShortcut('ctrl+F')
		self.fullAction.setStatusTip('Toggle Full Screen')
		self.fullAction.triggered.connect(self.toggleFullScreen)

		self.stereoIcon = QIcon('icon/stereo.png')
		self.stereoAction = QAction(self.stereoIcon, '&Stereoscopic', self)
		self.stereoAction.setShortcut('ctrl+S')
		self.stereoAction.setStatusTip('Toggle Stereoscopic')
		self.stereoAction.triggered.connect(self.field.toggleStereo)

		self.leftAction = QAction('Left intensity', self)
		self.leftAction.setShortcut("<")
		self.leftAction.setStatusTip('Increase left eye intensity')
		self.leftAction.triggered.connect(lambda: self.field.stereoIntensity(relative=-1))
		self.leftAction.setEnabled(False)

		self.rightAction = QAction('Right intensity', self)
		self.rightAction.setShortcut(">")
		self.rightAction.setStatusTip('Increase right eye intensity')
		self.rightAction.triggered.connect(lambda: self.field.stereoIntensity(relative=1))
		self.rightAction.setEnabled(False)

		nextIcon = QIcon('icon/next.png')
		self.nextAction = QAction(nextIcon, 'Next trial', self)
		self.nextAction.setShortcut(Qt.Key_Right)
		self.nextAction.setStatusTip('next trial')
		self.nextAction.triggered.connect(self.field.conditions.next)
		self.nextAction.setEnabled(False)

		downIcon = QIcon('icon/down.png')
		self.downAction = QAction(downIcon, 'Down trial', self)
		self.downAction.setShortcut(Qt.Key_Down)
		self.downAction.setShortcut(Qt.Key_Left)
		self.downAction.setStatusTip('decrease value')
		self.downAction.triggered.connect(lambda: self.field.addData(False))
		self.downAction.setEnabled(False)

		upIcon = QIcon('icon/up.png')
		self.upAction = QAction(upIcon, 'Up trial', self)
		self.upAction.setShortcut(Qt.Key_Up)
		self.upAction.setShortcut(Qt.Key_Right)
		self.upAction.setStatusTip('increase value')
		self.upAction.triggered.connect(lambda: self.field.addData(True))
		self.upAction.setEnabled(False)
		
		# populate the menu bar
		menubar = self.menuBar()
		fileMenu = menubar.addMenu('&File')
		fileMenu.addAction(loadAction)
		fileMenu.addAction(self.startAction)
		fileMenu.addAction(exitAction)

		viewMenu = menubar.addMenu('&View')
		viewMenu.addAction(self.fullAction)
		viewMenu.addAction(self.stereoAction)
		viewMenu.addAction(self.leftAction)
		viewMenu.addAction(self.rightAction)
		
		experimentMenu = menubar.addMenu('&Experiment')
		experimentMenu.addAction(self.nextAction)
		experimentMenu.addAction(self.upAction)
		experimentMenu.addAction(self.downAction)
		
		# make it also work when the menubar is hidden
		self.addAction(self.startAction) 
		self.addAction(self.fullAction)
		self.addAction(self.stereoAction)
		self.addAction(self.leftAction)
		self.addAction(self.rightAction)
		self.addAction(self.upAction)
		self.addAction(self.downAction)
		self.addAction(self.nextAction)
		self.addAction(loadAction)
		self.addAction(exitAction)

		self.statusBar().showMessage('Ready')
		self.setWindowTitle('Arjan, flying triangles')
		self.show()

	def startStop(self, event=None):
		if self.field.state=="sleep":
			logging.info("request state: end sleep")
			self.toggleText(False)
			self.field.changeState()
			self.startAction.setIcon(self.stopIcon)
		else:
			logging.info("request state: sleep")
			self.field.requestSleep = True
			self.startAction.setIcon(self.startIcon)

	def load(self, event=None, fileName=None):
		"""Load new list of trials."""
		self.nextAction.setEnabled(False)
		if fileName==None:
			fileName = QFileDialog.getOpenFileName(self, "Open file", filter="Experiment file (.csv)(*.csv)")
		if fileName==None:
			QMessageBox.question(self, 'Error', "No filename given?", QtGui.QMessageBox.Ok)
			return
		#try:
		self.field.conditions.load(fileName)
		self.field.initializeObjects()
		#except Exception as e:
		#	self.errorMessageDialog.showMessage("Could not read file: {}\n{}".format(fileName, e))
		#	raise e
			
		logging.info("load file {} with {} conditions".format(fileName, len(self.field.conditions.conditions)))

	def toggleFullScreen(self, event=None):
		if(self.isFullScreen()):
			self.showNormal()
			self.menuBar().setVisible(True)
			self.statusBar().setVisible(True)
			self.setCursor(QCursor(Qt.ArrowCursor))
		else:
			self.showFullScreen()
			self.menuBar().setVisible(False)
			self.statusBar().setVisible(False)
			self.setCursor(QCursor(Qt.BlankCursor))
				
	def toggleText(self, text=True):
		if text:
			self.stack.setCurrentIndex(1)
			#self.centralWidget().setParent(None) # if you do not do this the field widget will be deleted in the next line
			#self.setCentralWidget(self.text)
		else:
			self.stack.setCurrentIndex(0)
			#self.centralWidget().setParent(None)
			#self.setCentralWidget(self.field)
			

	def processCommandLine(self):
		# process command line arguments (to be called with running event loop)
		args = self.args
		if args.fullscreen:
			self.toggleFullScreen()
		if args.experiment:
			self.load(fileName=args.experiment)
		if args.lifetime:
			self.field.lifetime=int(args.lifetime)
		if args.running:
			self.startStop()
		self.field.connectSledServer(args.sledServer)     # defaults to simulator
		self.field.connectPositionServer(args.positionServer) # defaults to sledserver
		if args.stereo:
			self.field.toggleStereo(True)
		if args.stereoSim:
			self.field.toggleStereo(True, sim=True)
		if args.stereoIntensity:
			self.field.stereoIntensity(int(args.stereoIntensity))
		if args.subject:
			self.field.subject = args.subject
		else:
			self.field.subject = ""
		

def main(): 
	# parse command line arguments
	parser = argparse.ArgumentParser()
	parser.add_argument("-geometry", help="X11 option")
	parser.add_argument("-display", help="X11 option")
	parser.add_argument("-e", "--experiment", help="experiment input file (.csv)")
	parser.add_argument("-f", "--fullscreen", help="start in full screen mode", action="store_true")
	parser.add_argument("-l", "--lifetime", help="lifetime in frame of moving stars (0=infinite)")
	parser.add_argument("-s", "--sledServer", help="sled server, default to sled server simulator")
	parser.add_argument("-p", "--positionServer", help="position server, defaults to the sled server, but must be a first principles server when explicitly given, use 'mouse' for mouse")
	parser.add_argument("--stereo", help="Side by side stereoscopic view", action="store_true")
	parser.add_argument("--stereoSim", help="Simulating stereoscopic view", action="store_true")
	parser.add_argument("--stereoIntensity", help="Stereoscopic intensity balance -9 — 9")
	parser.add_argument("--subject", help="Subject ID")
	parser.add_argument("-r", "--running", help="start in running mode", action="store_true")
	args = parser.parse_args()
	
	
	# make application and main window
	a = QApplication(sys.argv)
	a.setApplicationName("Arjan")
	w = Main(args); 
	a.lastWindowClosed.connect(w.quit) # make upper right cross work
	signal.signal(signal.SIGINT, w.quit) # make ctrl-c work
	# main loop
	sys.exit(a.exec_())  # enter main loop (the underscore prevents using the keyword)

if __name__ == '__main__':
	utils.openLog();
	main()