def loadVideoData(self, videoDataFileName):
try:
self.videoData = VideoData()
self.videoData.load(videoDataFileName)
self.drawingMenu.enableVideoData(self.videoData.length())
self.setVideoData(0)
self.loadRobotData(self.videoData.robotNames)
self.setRobotData(0)
self.bindActions()
except BaseException as e:
self.displayErrorMsg('Error while reading videoData: ' + str(e))
class MainWin(QtGui.QMainWindow):
def __init__(self):
""" Creates main app win and tries to load videodata from last used file """
QtGui.QMainWindow.__init__(self)
"""UI part"""
self.setWindowTitle('MGR VIEWER 2.0')
self.setGeometry(10,10,840,840)
self.createContainer()
self.createTopMenu()
"""data init part"""
self.config = Config()
self.config.load('mgrViewerConfig')
if self.config.videoDataFileNameOK():
self.loadVideoData(self.config.videoDataFileName)
def createContainer(self):
"""
Creates main container for window content. Populates it with widgets.
Container is placed in scroll to make it possible to resize window with no clipping.
"""
scroll = QtGui.QScrollArea(self)
self.setCentralWidget(scroll)
container = QtGui.QWidget()
self.drawingArea = DrawingArea(self)
self.drawingMenu = DrawingMenu(self)
hbox = QtGui.QHBoxLayout()
hbox.addWidget(self.drawingArea)
hbox.addWidget(self.drawingMenu)
container.setLayout(hbox)
scroll.setWidget(container)
def createTopMenu(self):
"""Creates top menu with file option"""
self.statusBar().showMessage('Select file with videoData')
exit = QtGui.QAction('Exit', self)
self.connect(exit, QtCore.SIGNAL('triggered()'), QtCore.SLOT('close()'))
videoData = QtGui.QAction('Load VideoData', self)
self.connect(videoData, QtCore.SIGNAL('triggered()'), self.videoDataDialog)
menubar = self.menuBar()
file = menubar.addMenu('&File')
file.addAction(videoData)
file.addAction(exit)
def initVideoData(self):
if self.config.video_data_OK():
self.loadVideoData()
def videoDataDialog(self):
tmp = QtGui.QFileDialog.getOpenFileName(self, 'Open file','.')
if tmp:
self.config.videoDataFileName = tmp
self.loadVideoData(self.config.videoDataFileName)
def loadVideoData(self, videoDataFileName):
try:
self.videoData = VideoData()
self.videoData.load(videoDataFileName)
self.drawingMenu.enableVideoData(self.videoData.length())
self.setVideoData(0)
self.loadRobotData(self.videoData.robotNames)
self.setRobotData(0)
self.bindActions()
except BaseException as e:
self.displayErrorMsg('Error while reading videoData: ' + str(e))
def loadRobotData(self, robotNames):
try:
self.robotData = dict()
dir = os.path.dirname(str(self.config.videoDataFileName))
for robot in robotNames:
filename = dir +'/'+ robot+'.txt'
if os.path.isfile(filename) == True:
tmp = RobotData()
tmp.load(filename)
if tmp.length() > 0:
self.robotData[robot] = tmp
self.drawingMenu.enableRobotData(self.robotData.keys())
except BaseException as e:
self.displayErrorMsg('Error while reading robotData: ' + str(e))
def closeEvent(self, event):
self.config.disp()
self.config.save()
def bindActions(self):
#self.connect(self.drawingMenu.slider, QtCore.SIGNAL('valueChanged(int)'), self.setVideoData)
self.drawingMenu.slider.valueChanged.connect(self.setVideoData)
if len(self.robotData) > 0:
self.drawingMenu.slider.valueChanged.connect(self.setRobotData)
self.drawingMenu.robotCombo.currentIndexChanged.connect(self.setRobotData)
def setVideoData(self, val):
self.drawingArea.setVideoData(self.videoData.steps[val])
def setRobotData(self, val):
robot = str(self.drawingMenu.robotCombo.currentText())
if self.robotData[robot].steps[val] != None:
self.drawingMenu.setRobotData(self.robotData[robot].steps[val])
self.drawingArea.setRRT(self.robotData[robot].steps[val].rrt, self.robotData[robot].steps[val].rrtRes)
def switchRobotData(self, val):
self.setRobotData(self.drawingMenu.slider.value())
def displayErrorMsg(self, msg):
QtGui.QMessageBox.critical(self, 'Error', msg , QtGui.QMessageBox.Ok)