Esempio n. 1
0
    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)
Esempio n. 2
0
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 &copy; 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()