def __init__(self, parent=None):
"""
Constructor
"""
super(LtMainWindow, self).__init__(parent)
self.setProjectName()
self.setObjectName("ltMainWindow")
# ==== Creating core functional units ====
self.cvProcessor = CvProcessor(self)
# ==== Creating GUI ====
# status = self.statusBar()
# status.setSizeGripEnabled(False)
# ---- Creating main video widget ----
self.cvGraphics = CvGraphics(self)
self.cvGraphics.setAlignment(QtCore.Qt.AlignCenter)
self.cvGraphics.setContextMenuPolicy(QtCore.Qt.ActionsContextMenu)
self.cvGraphics.setObjectName("cvGraphics")
self.setCentralWidget(self.cvGraphics)
self.cvProcessor.signalNextFrame.connect(self.cvGraphics.putImage)
self.cvProcessor.signalClearFrame.connect(self.cvGraphics.setNullImage)
self.cvProcessor.project.signalProjectOpened.connect(self.setProjectName)
self.cvProcessor.project.signalProjectClosed.connect(self.setProjectName)
# ---- Creating dock panel for video player ----
videoDockPanel = QtGui.QDockWidget("Video Control", self)
videoDockPanel.setObjectName("videoDockWidget")
videoDockPanel.setAllowedAreas(QtCore.Qt.BottomDockWidgetArea)
videoDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, videoDockPanel)
self.videoWidget = VideoWidget(self.cvProcessor.cvPlayer)
videoDockPanel.setWidget(self.videoWidget)
# ---- Creating dock panel for project ----
projectDockPanel = QtGui.QDockWidget("Project", self) # Created and set caption
projectDockPanel.setObjectName("projectDockWidget")
projectDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea)
projectDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, projectDockPanel)
self.projectWidget = ProjectWidget(self.cvProcessor.project)
projectDockPanel.setWidget(self.projectWidget)
# ---- Creating dock panel for chambers ----
chambersDockPanel = QtGui.QDockWidget("Chambers", self) # Created and set caption
chambersDockPanel.setObjectName("chambersDockWidget")
chambersDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea)
chambersDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, chambersDockPanel)
self.chambersWidget = ChambersWidget()
# self.chambersWidget.analysisMethod.stateChanged.connect(self.cvProcessor.setAnalysisMethod)
chambersDockPanel.setWidget(self.chambersWidget)
# ---- chambersWidget ----
self.chambersWidget.signalScaleSelect.connect(self.cvGraphics.selectScale)
self.chambersWidget.signalChamberSelect.connect(self.cvGraphics.selectChamber)
self.cvGraphics.signalChamberSetted.connect(self.chambersWidget.chamberSetted)
self.cvGraphics.signalScaleSetted.connect(self.chambersWidget.scaleSetted)
self.chambersWidget.signalChamberSelected.connect(self.cvGraphics.selectChamberGui)
self.cvGraphics.signalChamberSelected.connect(self.chambersWidget.selectChamber)
self.cvProcessor.project.signalChamberAdded.connect(self.chambersWidget.addChamber)
self.cvProcessor.project.signalChamberAdded.connect(self.cvGraphics.addChamberGui)
self.cvProcessor.project.signalChamberDeleted.connect(self.chambersWidget.removeChamber)
self.cvProcessor.project.signalChamberDeleted.connect(self.cvGraphics.delChamberGui)
self.cvProcessor.project.signalChambersMangerChanged.connect(self.chambersWidget.setChambersManager)
self.cvProcessor.project.signalChambersMangerChanged.connect(self.cvGraphics.delScaleGui)
# ---- Creating dock panel for image processing ----
cvProcessorDockPanel = QtGui.QDockWidget("Image processor", self) # Created and set caption
cvProcessorDockPanel.setObjectName("chambersDockWidget")
cvProcessorDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea | QtCore.Qt.RightDockWidgetArea)
cvProcessorDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, cvProcessorDockPanel)
self.cvProcessorWidget = CvProcessorWidget(self.cvProcessor)
cvProcessorDockPanel.setWidget(self.cvProcessorWidget)
# ---- Creating dock panel for image processing ----
cvPreProcessorDockPanel = QtGui.QDockWidget("Preprocessor", self) # Created and set caption
cvPreProcessorDockPanel.setObjectName("PreDockWidget")
cvPreProcessorDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea | QtCore.Qt.RightDockWidgetArea)
cvPreProcessorDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, cvPreProcessorDockPanel)
self.cvPreProcessorWidget = PreprocessorWidget(self.cvProcessor.preprocessor)
cvPreProcessorDockPanel.setWidget(self.cvPreProcessorWidget)
# ---- Creating dock panel for trajectory Widget
cvTrajectoryDockPanel = QtGui.QDockWidget("Trajectory", self) # Created and set caption
cvTrajectoryDockPanel.setObjectName("TrajectoryDockWidget")
cvTrajectoryDockPanel.setAllowedAreas(QtCore.Qt.RightDockWidgetArea)
cvTrajectoryDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, cvTrajectoryDockPanel)
self.trajectoryWidget = TrajectoryWidget(self.cvProcessor.project.trajectoryAnalysis)
cvTrajectoryDockPanel.setWidget(self.trajectoryWidget)
self.presetsWidget = PresetsWidget(self)
# ==== Creating menu ====
projectMenu = self.menuBar().addMenu("&Project")
setPresetAction = createAction(self, "Set preset...", "", "", "")
projectMenu.addAction(setPresetAction)
setPresetAction.triggered.connect(self.presetDialogDisplay)
self.presetsWidget.signalSetPreset.connect(self.cvProcessorWidget.setPreset)
self.presetsWidget.signalSetPreset.connect(self.trajectoryWidget.setPreset)
self.quitAction = createAction(self, "Exit...", QtGui.QKeySequence.Quit, "application-exit", "Exit program")
addActions(projectMenu, self.projectWidget.actions)
projectMenu.addSeparator()
projectMenu.addAction(self.quitAction)
"""
projectToolbar = self.addToolBar("Project")
projectToolbar.setObjectName("projectToolBar")
addActions(projectToolbar, (self.quitAction,))
"""
videoMenu = self.menuBar().addMenu("&Video")
addActions(videoMenu, self.videoWidget.actions)
chamberMenu = self.menuBar().addMenu("&Chamber")
# addActions(chamberMenu, self.chambersWidget.actions)
trajectoryMenu = self.menuBar().addMenu("&Trajectory")
# addActions(trajectoryMenu, self.trajectoryWidget.actions)
helpMenu = self.menuBar().addMenu("&Help")
helpAboutAction = createAction(self, "&About LocoTrack", "", "help-about", "")
helpAboutAction.triggered.connect(self.helpAbout)
helpHelpAction = createAction(self, "&Help", QtGui.QKeySequence.HelpContents, "help-contents")
helpHelpAction.triggered.connect(self.helpHelp)
addActions(helpMenu, (helpAboutAction, helpHelpAction))
# ---- cvProcessorWidget ----
self.cvProcessorWidget.showProcessedChechBox.stateChanged.connect(self.cvProcessor.setShowProcessed)
self.cvProcessorWidget.showContourChechBox.stateChanged.connect(self.cvProcessor.setShowContour)
self.cvProcessorWidget.thresholdSlider.valueChanged.connect(self.cvProcessor.setTreshold)
self.cvProcessorWidget.ellipseCropCheckBox.stateChanged.connect(self.cvProcessor.setEllipseCrop)
# ---- Main menu ----
# Project menu
self.quitAction.triggered.connect(self.close)
# Restore window settings
settings = QtCore.QSettings()
self.restoreGeometry(settings.value("ltMainWindow/Geometry").toByteArray())
self.restoreState(settings.value("ltMainWindow/State").toByteArray())
preset, isInt = settings.value("ltMainWindow/Preset").toInt()
self.trajectoryWidget.analyseDialog.lastDirectory = settings.value(
"ltTrajectoryWidget/lastDirectory"
).toString()
self.projectWidget.lastDirectory = settings.value("ltProjectWidget/lastDirectory").toString()
print "load preset", preset
self.presetsWidget.setPreset(preset)
class LtMainWindow(QtGui.QMainWindow):
"""
Locotrack main window
"""
def __init__(self, parent=None):
"""
Constructor
"""
super(LtMainWindow, self).__init__(parent)
self.setProjectName()
self.setObjectName("ltMainWindow")
# ==== Creating core functional units ====
self.cvProcessor = CvProcessor(self)
# ==== Creating GUI ====
# status = self.statusBar()
# status.setSizeGripEnabled(False)
# ---- Creating main video widget ----
self.cvGraphics = CvGraphics(self)
self.cvGraphics.setAlignment(QtCore.Qt.AlignCenter)
self.cvGraphics.setContextMenuPolicy(QtCore.Qt.ActionsContextMenu)
self.cvGraphics.setObjectName("cvGraphics")
self.setCentralWidget(self.cvGraphics)
self.cvProcessor.signalNextFrame.connect(self.cvGraphics.putImage)
self.cvProcessor.signalClearFrame.connect(self.cvGraphics.setNullImage)
self.cvProcessor.project.signalProjectOpened.connect(self.setProjectName)
self.cvProcessor.project.signalProjectClosed.connect(self.setProjectName)
# ---- Creating dock panel for video player ----
videoDockPanel = QtGui.QDockWidget("Video Control", self)
videoDockPanel.setObjectName("videoDockWidget")
videoDockPanel.setAllowedAreas(QtCore.Qt.BottomDockWidgetArea)
videoDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, videoDockPanel)
self.videoWidget = VideoWidget(self.cvProcessor.cvPlayer)
videoDockPanel.setWidget(self.videoWidget)
# ---- Creating dock panel for project ----
projectDockPanel = QtGui.QDockWidget("Project", self) # Created and set caption
projectDockPanel.setObjectName("projectDockWidget")
projectDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea)
projectDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, projectDockPanel)
self.projectWidget = ProjectWidget(self.cvProcessor.project)
projectDockPanel.setWidget(self.projectWidget)
# ---- Creating dock panel for chambers ----
chambersDockPanel = QtGui.QDockWidget("Chambers", self) # Created and set caption
chambersDockPanel.setObjectName("chambersDockWidget")
chambersDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea)
chambersDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, chambersDockPanel)
self.chambersWidget = ChambersWidget()
# self.chambersWidget.analysisMethod.stateChanged.connect(self.cvProcessor.setAnalysisMethod)
chambersDockPanel.setWidget(self.chambersWidget)
# ---- chambersWidget ----
self.chambersWidget.signalScaleSelect.connect(self.cvGraphics.selectScale)
self.chambersWidget.signalChamberSelect.connect(self.cvGraphics.selectChamber)
self.cvGraphics.signalChamberSetted.connect(self.chambersWidget.chamberSetted)
self.cvGraphics.signalScaleSetted.connect(self.chambersWidget.scaleSetted)
self.chambersWidget.signalChamberSelected.connect(self.cvGraphics.selectChamberGui)
self.cvGraphics.signalChamberSelected.connect(self.chambersWidget.selectChamber)
self.cvProcessor.project.signalChamberAdded.connect(self.chambersWidget.addChamber)
self.cvProcessor.project.signalChamberAdded.connect(self.cvGraphics.addChamberGui)
self.cvProcessor.project.signalChamberDeleted.connect(self.chambersWidget.removeChamber)
self.cvProcessor.project.signalChamberDeleted.connect(self.cvGraphics.delChamberGui)
self.cvProcessor.project.signalChambersMangerChanged.connect(self.chambersWidget.setChambersManager)
self.cvProcessor.project.signalChambersMangerChanged.connect(self.cvGraphics.delScaleGui)
# ---- Creating dock panel for image processing ----
cvProcessorDockPanel = QtGui.QDockWidget("Image processor", self) # Created and set caption
cvProcessorDockPanel.setObjectName("chambersDockWidget")
cvProcessorDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea | QtCore.Qt.RightDockWidgetArea)
cvProcessorDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, cvProcessorDockPanel)
self.cvProcessorWidget = CvProcessorWidget(self.cvProcessor)
cvProcessorDockPanel.setWidget(self.cvProcessorWidget)
# ---- Creating dock panel for image processing ----
cvPreProcessorDockPanel = QtGui.QDockWidget("Preprocessor", self) # Created and set caption
cvPreProcessorDockPanel.setObjectName("PreDockWidget")
cvPreProcessorDockPanel.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea | QtCore.Qt.RightDockWidgetArea)
cvPreProcessorDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, cvPreProcessorDockPanel)
self.cvPreProcessorWidget = PreprocessorWidget(self.cvProcessor.preprocessor)
cvPreProcessorDockPanel.setWidget(self.cvPreProcessorWidget)
# ---- Creating dock panel for trajectory Widget
cvTrajectoryDockPanel = QtGui.QDockWidget("Trajectory", self) # Created and set caption
cvTrajectoryDockPanel.setObjectName("TrajectoryDockWidget")
cvTrajectoryDockPanel.setAllowedAreas(QtCore.Qt.RightDockWidgetArea)
cvTrajectoryDockPanel.setFeatures(QtGui.QDockWidget.DockWidgetMovable | QtGui.QDockWidget.DockWidgetFloatable)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, cvTrajectoryDockPanel)
self.trajectoryWidget = TrajectoryWidget(self.cvProcessor.project.trajectoryAnalysis)
cvTrajectoryDockPanel.setWidget(self.trajectoryWidget)
self.presetsWidget = PresetsWidget(self)
# ==== Creating menu ====
projectMenu = self.menuBar().addMenu("&Project")
setPresetAction = createAction(self, "Set preset...", "", "", "")
projectMenu.addAction(setPresetAction)
setPresetAction.triggered.connect(self.presetDialogDisplay)
self.presetsWidget.signalSetPreset.connect(self.cvProcessorWidget.setPreset)
self.presetsWidget.signalSetPreset.connect(self.trajectoryWidget.setPreset)
self.quitAction = createAction(self, "Exit...", QtGui.QKeySequence.Quit, "application-exit", "Exit program")
addActions(projectMenu, self.projectWidget.actions)
projectMenu.addSeparator()
projectMenu.addAction(self.quitAction)
"""
projectToolbar = self.addToolBar("Project")
projectToolbar.setObjectName("projectToolBar")
addActions(projectToolbar, (self.quitAction,))
"""
videoMenu = self.menuBar().addMenu("&Video")
addActions(videoMenu, self.videoWidget.actions)
chamberMenu = self.menuBar().addMenu("&Chamber")
# addActions(chamberMenu, self.chambersWidget.actions)
trajectoryMenu = self.menuBar().addMenu("&Trajectory")
# addActions(trajectoryMenu, self.trajectoryWidget.actions)
helpMenu = self.menuBar().addMenu("&Help")
helpAboutAction = createAction(self, "&About LocoTrack", "", "help-about", "")
helpAboutAction.triggered.connect(self.helpAbout)
helpHelpAction = createAction(self, "&Help", QtGui.QKeySequence.HelpContents, "help-contents")
helpHelpAction.triggered.connect(self.helpHelp)
addActions(helpMenu, (helpAboutAction, helpHelpAction))
# ---- cvProcessorWidget ----
self.cvProcessorWidget.showProcessedChechBox.stateChanged.connect(self.cvProcessor.setShowProcessed)
self.cvProcessorWidget.showContourChechBox.stateChanged.connect(self.cvProcessor.setShowContour)
self.cvProcessorWidget.thresholdSlider.valueChanged.connect(self.cvProcessor.setTreshold)
self.cvProcessorWidget.ellipseCropCheckBox.stateChanged.connect(self.cvProcessor.setEllipseCrop)
# ---- Main menu ----
# Project menu
self.quitAction.triggered.connect(self.close)
# Restore window settings
settings = QtCore.QSettings()
self.restoreGeometry(settings.value("ltMainWindow/Geometry").toByteArray())
self.restoreState(settings.value("ltMainWindow/State").toByteArray())
preset, isInt = settings.value("ltMainWindow/Preset").toInt()
self.trajectoryWidget.analyseDialog.lastDirectory = settings.value(
"ltTrajectoryWidget/lastDirectory"
).toString()
self.projectWidget.lastDirectory = settings.value("ltProjectWidget/lastDirectory").toString()
print "load preset", preset
self.presetsWidget.setPreset(preset)
# ==== Slots to handle actions ====
def setProjectName(self, name=QtCore.QString("untitled")):
title = QtCore.QString(applicationName + " " + applicationVersion) + " - " + name
self.setWindowTitle(title)
def presetDialogDisplay(self):
self.presetsWidget.exec_()
def okToContinue(self):
if False:
reply = QtGui.QMessageBox.question(
self,
" Unsaved Changes",
"Save unsaved changes?",
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No | QtGui.QMessageBox.Cancel,
)
if reply == QtGui.QMessageBox.Cancel:
return False
elif reply == QtGui.QMessageBox.Yes:
self.saveProject()
return True
def showInfo(self, messageKind, messageText):
"""
Show info message and log it to logFile
"""
if messageKind == 0:
self.statusBar().showMessage(messageText, 5000)
elif messageKind == -1:
QtGui.QMessageBox("Error: " + messageText)
# Close Event handler
def closeEvent(self, event):
# Asking user to confirm
# if self.okToContinue():
# Save settings and exit
settings = QtCore.QSettings()
settings.setValue("ltMainWindow/Geometry", QtCore.QVariant(self.saveGeometry()))
settings.setValue("ltMainWindow/State", QtCore.QVariant(self.saveState()))
settings.setValue("ltMainWindow/Preset", QtCore.QVariant(self.presetsWidget.currentPreset()))
settings.setValue(
"ltTrajectoryWidget/lastDirectory", QtCore.QVariant(self.trajectoryWidget.analyseDialog.lastDirectory)
)
settings.setValue("ltProjectWidget/lastDirectory", QtCore.QVariant(self.projectWidget.lastDirectory))
def helpAbout(self):
QtGui.QMessageBox.about(
self,
"About LocoTrack",
"""<b>LocoTrack</b> v {0}
<p>Copyright © 2012-2013.
All rights reserved.
<p>This application can be used to .
<p>Python {1} - Qt {2} - PyQt {3} on {4}""".format(
applicationVersion,
platform.python_version(),
QtCore.QT_VERSION_STR,
QtCore.PYQT_VERSION_STR,
platform.system(),
),
)
def helpHelp(self):
form = HelpWidget("index.html", self)
form.show()