コード例 #1
0
ファイル: mark.py プロジェクト: aluckyi/MarkGT
class Mark(MainUI):

    def __init__(self):
        super().__init__()

        self.exe = Execute()
        self.img_lbl.initXYBoxObjs(self.x_box, self.y_box)
        self.img_lbl.initReferLineNumObj(self.ref_num_lbl)
        self.img_lbl.initHorizonModifyObj(self.horizon_modify_btn)
        self.img_lbl.initSLObsNumObjs(self.small_obs_num_lbl, self.large_obs_num_lbl)

        self.start_flag = False
        self.mode = 'Horizon'
        self.horizon_method = 'Manual'
        self.refer_lines_flag = False
        self.obs_cls = 'Small Obstacle'

    def srcBtnRespond(self):
        home_dir = osp.expanduser('~')
        dir_name = QFileDialog.getExistingDirectory(self, 'Select Directory', home_dir)
        self.src_show_box.setText(dir_name)
        self.exe.setSrcDir(dir_name)

    def dstBtnRespond(self):
        home_dir = osp.expanduser('~')
        dir_name = QFileDialog.getExistingDirectory(self, 'Select Directory', home_dir)
        self.dst_show_box.setText(dir_name)
        self.exe.setDstDir(dir_name)

    def modeRespond(self):
        source = self.sender()
        # print(source.text())
        if source.text() == 'Horizon':
            self.mode = 'Horizon'
            self.img_lbl.setMode(self.mode)
            self.h_square.setStyleSheet("QWidget { background-color: rgb(0, 255, 0); }")
            self.o_square.setStyleSheet("QWidget { background-color: rgb(180, 180, 180); }")
        else:
            self.mode = 'Obstacle'
            self.img_lbl.setMode(self.mode)
            self.h_square.setStyleSheet("QWidget { background-color: rgb(180, 180, 180); }")
            self.o_square.setStyleSheet("QWidget { background-color: rgb(0, 255, 0); }")

            self.img_lbl.setHorizonModifyFlag(False)
            self.horizon_modify_btn.setChecked(False)

    def horizonMethodRespond(self):
        source = self.sender()
        if source.text() == 'Manual':
            self.horizon_method = 'Manual'
        else:
            self.horizon_method = 'Auto'

    def referLinesBtnRespond(self, value):
        self.img_lbl.setReferLines(value)

    def horizonModifyBtnRespond(self, value):
        if not self.start_flag:
            QMessageBox.warning(self, 'warning', 'Please press [Start] or [Restore]!!!')
            self.horizon_modify_btn.setChecked(False)
            return

        if value:
            if self.mode != 'Horizon':
                QMessageBox.warning(self, 'warning', 'The [Mode] must be "Horizon" !!!')
                self.horizon_modify_btn.setChecked(False)
                return

            ret = self.img_lbl.setHorizonModifyFlag(value)
            if ret < 0:
                QMessageBox.warning(self, 'warning', 'The horizon has not been marked!!!')
                self.horizon_modify_btn.setChecked(False)
                return
        else:
            self.img_lbl.setHorizonModifyFlag(value)

    def obsClsBtnRespond(self):
        source = self.sender()
        if source.text() == 'Small Obstacle':
            self.obs_cls = 'Small Obstacle'
            self.img_lbl.setObsCls(self.obs_cls)
            obs_cls_col = QColor(*cfg.SMALL_OBS_COLOR)
            self.obs_cls_square.setStyleSheet("QWidget { background-color: %s }" % obs_cls_col.name())
        else:
            self.obs_cls = 'Large Obstacle'
            self.img_lbl.setObsCls(self.obs_cls)
            obs_cls_col = QColor(*cfg.LARGE_OBS_COLOR)
            self.obs_cls_square.setStyleSheet("QWidget { background-color: %s }" % obs_cls_col.name())
        # print(self.obs_cls)

    def startBtnRespond(self):
        if self.start_flag:
            reply = QMessageBox.question(self, 'warning', "Are you sure to restart?",
                                         QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
            if reply == QMessageBox.No:
                return
            else:
                self.exe.reset()

        ret = self.exe.start()
        if ret == -1:
            QMessageBox.warning(self, 'warning', 'Please Select SRC Path!!!')
            return
        if ret == -2:
            QMessageBox.warning(self, 'warning', 'Please Select DST Path!!!')
            return
        if ret == -3:
            QMessageBox.warning(self, 'warning', 'No jpg or png images in SRC Path!!!')
            return

        self.display(isFinished=False)
        self.img_lbl.setEnableImageFlag(True)
        self.img_lbl.reset()

        self.start_flag = True

    def restoreBtnRespond(self):
        reply = QMessageBox.question(self, 'warning', "Are you sure to restore?",
                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
        if reply == QMessageBox.No:
            return

        ret = self.exe.restore()
        if ret < 0:
            QMessageBox.warning(self, 'warning', 'Some problems exist, cannot restore!!!')
            return

        self.display(isFinished=False)
        self.img_lbl.setEnableImageFlag(True)
        self.img_lbl.reset()

        self.start_flag = True

    def undoBtnRespond(self):
        if not self.start_flag:
            QMessageBox.warning(self, 'warning', 'Please press [Start] or [Restore]!!!')
            return
        ret = self.img_lbl.undo()
        if ret == -1:
            QMessageBox.warning(self, 'warning', 'All marks are removed!!!')

    def nextBtnRespond(self):
        if not self.start_flag:
            QMessageBox.warning(self, 'warning', 'Please press [Start] or [Restore]!!!')
            return

        horizon_data, obs_data = self.img_lbl.getData()
        ret = self.exe.saveData(horizon_data, obs_data)
        if ret == -1:
            QMessageBox.warning(self, 'warning', 'Marking Horizon has not completed!!!')
            return
        if ret == -2:
            self.display(isFinished=True)
            reply = QMessageBox.question(self, 'warning',
                                         'All marking has completed!!!, Are you sure to finish?',
                                         QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
            if reply == QMessageBox.No:
                self.img_lbl.reset()
                return
            self.reset()
            return

        self.img_lbl.reset()
        self.display(isFinished=False)

    def reset(self):
        self.start_flag = False

        self.src_show_box.clear()
        self.dst_show_box.clear()

        self.img_lbl.clear()
        self.img_lbl.setEnableImageFlag(False)
        self.img_lbl.reset()

        self.cur_img_box.clear()
        self.no_box.clear()
        self.total_box.clear()
        self.pbar.reset()
        self.x_box.clear()
        self.y_box.clear()

        self.ref_num_lbl.clear()
        self.small_obs_num_lbl.clear()
        self.large_obs_num_lbl.clear()

        self.exe.setSrcDir('')
        self.exe.setDstDir('')
        self.exe.reset()
        self.exe.saveConfig()

    def display(self, isFinished):
        cur_image_name = self.exe.getCurrentImageName()
        cur_image_no = self.exe.getCurrentImageNo()
        total_image_num = self.exe.getTotalImageNum()

        src_dir = self.exe.getSrcDir()
        dst_dir = self.exe.getDstDir()
        self.src_show_box.setText(src_dir)
        self.dst_show_box.setText(dst_dir)
        self.cur_img_box.setText(cur_image_name)
        self.no_box.setText(str(cur_image_no + 1))
        self.total_box.setText(str(total_image_num))

        if isFinished:
            pb_val = int((float(cur_image_no + 1) / total_image_num) * 100)
        else:
            pb_val = int((float(cur_image_no) / total_image_num) * 100)
        self.pbar.setValue(pb_val)

        self.ref_num_lbl.setText(str(0))
        self.small_obs_num_lbl.setText(str(0))
        self.large_obs_num_lbl.setText(str(0))

        self.img_lbl.loadFromDisk(osp.join(cfg.SRC_DIR, cur_image_name))
コード例 #2
0
class Gui_Main(QtGui.QMainWindow):
    """
    The main window of the app
    """
    def __init__(self, main):
        """
        Parent constructor call,
        call setup for gui, 
        create thread object for 
        the script and connect signals
        to slots
        """
        super(Gui_Main, self).__init__()
        self.setupUi(main)
        self.execute = Execute()
        self.connect(self.execute, SIGNAL("setOutput(QString)"),
                     self.setOutput)
        self.connect(self.execute, SIGNAL("finished(bool)"), self.finished)
        self.step = 0

    def setupUi(self, MainWindow):
        """
        Create gui widget etc 
        and connect signals to slots
        """
        MainWindow.setObjectName(_fromUtf8("Despeckle"))
        MainWindow.resize(716, 873)

        self.centralwidget = QtGui.QWidget(MainWindow)
        self.centralwidget.setObjectName(_fromUtf8("centralwidget"))

        self.widget = QtGui.QWidget(self.centralwidget)
        self.widget.setGeometry(QtCore.QRect(10, 750, 291, 41))
        self.widget.setObjectName(_fromUtf8("widget"))

        self.progressBar = QtGui.QProgressBar(self.widget)
        self.progressBar.setGeometry(QtCore.QRect(107, 10, 221, 23))
        self.progressBar.setProperty("value", 0)
        self.progressBar.setObjectName(_fromUtf8("progressBar"))

        self.widget_2 = QtGui.QWidget(self.centralwidget)
        self.widget_2.setGeometry(QtCore.QRect(410, 750, 291, 41))
        self.widget_2.setObjectName(_fromUtf8("widget_2"))

        self.widget_3 = QtGui.QWidget(self.centralwidget)
        self.widget_3.setGeometry(QtCore.QRect(10, 40, 691, 331))
        self.widget_3.setObjectName(_fromUtf8("widget_3"))

        MainWindow.setCentralWidget(self.centralwidget)

        self.textEdit = QtGui.QTextEdit(self.widget_3)
        self.textEdit.setGeometry(QtCore.QRect(110, 50, 501, 31))
        self.textEdit.setObjectName(_fromUtf8("textEdit"))

        self.textEdit_2 = QtGui.QTextEdit(self.widget_3)
        self.textEdit_2.setGeometry(QtCore.QRect(110, 100, 501, 31))
        self.textEdit_2.setObjectName(_fromUtf8("textEdit_2"))

        self.textEdit_3 = QtGui.QTextEdit(self.widget_3)
        self.textEdit_3.setGeometry(QtCore.QRect(110, 150, 501, 31))
        self.textEdit_3.setObjectName(_fromUtf8("textEdit_3"))

        self.textEdit_4 = QtGui.QTextEdit(self.widget_3)
        self.textEdit_4.setGeometry(QtCore.QRect(110, 200, 251, 31))
        self.textEdit_4.setObjectName(_fromUtf8("textEdit_4"))

        self.textEdit_5 = QtGui.QTextEdit(self.widget_3)
        self.textEdit_5.setGeometry(QtCore.QRect(110, 250, 251, 31))
        self.textEdit_5.setObjectName(_fromUtf8("textEdit_5"))

        self.plainTextEdit = QtGui.QPlainTextEdit(self.centralwidget)
        self.plainTextEdit.setGeometry(QtCore.QRect(119, 420, 501, 321))
        self.plainTextEdit.setObjectName(_fromUtf8("plainTextEdit"))
        self.plainTextEdit.setReadOnly(True)

        self.label = QtGui.QLabel(self.widget_2)
        self.label.setGeometry(QtCore.QRect(20, 10, 201, 21))
        self.label.setObjectName(_fromUtf8("label"))

        self.label_2 = QtGui.QLabel(self.widget_3)
        self.label_2.setGeometry(QtCore.QRect(30, 50, 41, 31))
        self.label_2.setObjectName(_fromUtf8("label_2"))

        self.label_3 = QtGui.QLabel(self.widget_3)
        self.label_3.setGeometry(QtCore.QRect(30, 100, 21, 31))
        self.label_3.setObjectName(_fromUtf8("label_3"))

        self.label_4 = QtGui.QLabel(self.widget_3)
        self.label_4.setGeometry(QtCore.QRect(30, 150, 51, 31))
        self.label_4.setObjectName(_fromUtf8("label_4"))

        self.label_5 = QtGui.QLabel(self.widget_3)
        self.label_5.setGeometry(QtCore.QRect(30, 200, 81, 31))
        self.label_5.setObjectName(_fromUtf8("label_5"))

        self.label_6 = QtGui.QLabel(self.widget_3)
        self.label_6.setGeometry(QtCore.QRect(30, 250, 51, 31))
        self.label_6.setObjectName(_fromUtf8("label_6"))

        self.label_7 = QtGui.QLabel(self.centralwidget)
        self.label_7.setGeometry(QtCore.QRect(310, 380, 101, 31))
        self.label_7.setObjectName(_fromUtf8("label_7"))

        self.label_8 = QtGui.QLabel(self.widget_3)
        self.label_8.setGeometry(QtCore.QRect(450, 200, 151, 20))
        self.label_8.setObjectName(_fromUtf8("label_8"))

        self.label_9 = QtGui.QLabel(self.widget_3)
        self.label_9.setGeometry(QtCore.QRect(450, 250, 151, 20))
        self.label_9.setObjectName(_fromUtf8("label_9"))

        icon = QtGui.QIcon()
        icon.addPixmap(
            QtGui.QPixmap(
                _fromUtf8("/opt/Despeckle_pyqt/icons/view_tree.png")),
            QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.pushButton = QtGui.QPushButton(self.widget_3)
        self.pushButton.setGeometry(QtCore.QRect(630, 50, 41, 31))
        self.pushButton.setText(_fromUtf8(""))
        self.pushButton.setIcon(icon)
        self.pushButton.setIconSize(QtCore.QSize(42, 42))
        self.pushButton.setObjectName(_fromUtf8("pushButton"))
        self.pushButton.clicked.connect(self.inputButtonClicked)

        self.pushButton_2 = QtGui.QPushButton(self.widget_3)
        self.pushButton_2.setGeometry(QtCore.QRect(630, 100, 41, 31))
        self.pushButton_2.setText(_fromUtf8(""))
        self.pushButton_2.setIcon(icon)
        self.pushButton_2.setIconSize(QtCore.QSize(42, 42))
        self.pushButton_2.setObjectName(_fromUtf8("pushButton_2"))
        self.pushButton_2.clicked.connect(self.outputButtonClicked)

        self.pushButton_3 = QtGui.QPushButton(self.widget_3)
        self.pushButton_3.setGeometry(QtCore.QRect(630, 150, 41, 31))
        self.pushButton_3.setText(_fromUtf8(""))
        self.pushButton_3.setIcon(icon)
        self.pushButton_3.setIconSize(QtCore.QSize(42, 42))
        self.pushButton_3.setObjectName(_fromUtf8("pushButton_3"))
        self.pushButton_3.clicked.connect(self.inputKernelButtonClicked)

        self.pushButton_4 = QtGui.QPushButton(self.widget_3)
        self.pushButton_4.setGeometry(QtCore.QRect(630, 200, 41, 31))
        self.pushButton_4.setObjectName(_fromUtf8("pushButton_4"))
        self.pushButton_4.setIcon(icon)
        self.pushButton_4.setIconSize(QtCore.QSize(42, 42))
        self.pushButton_4.clicked.connect(self.statsButton)

        self.pushButton_5 = QtGui.QPushButton(self.widget_3)
        self.pushButton_5.setGeometry(QtCore.QRect(630, 250, 41, 31))
        self.pushButton_5.setObjectName(_fromUtf8("pushButton_5"))
        self.pushButton_5.setIcon(icon)
        self.pushButton_5.setIconSize(QtCore.QSize(42, 42))
        self.pushButton_5.clicked.connect(self.statsButton2)

        self.menubar = QtGui.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 716, 20))
        self.menubar.setObjectName(_fromUtf8("menubar"))

        self.menu_File = QtGui.QMenu(self.menubar)
        self.menu_File.setObjectName(_fromUtf8("menu_File"))
        self.menu_Help = QtGui.QMenu(self.menubar)
        self.menu_Help.setObjectName(_fromUtf8("menu_Help"))
        self.menu_Help.triggered.connect(self.aboutClicked)

        MainWindow.setMenuBar(self.menubar)

        self.toolBar = QtGui.QToolBar(MainWindow)
        self.toolBar.setObjectName(_fromUtf8("toolBar"))
        MainWindow.addToolBar(QtCore.Qt.TopToolBarArea, self.toolBar)

        self.action_Run = QtGui.QAction(MainWindow)
        icon1 = QtGui.QIcon()
        icon1.addPixmap(
            QtGui.QPixmap(_fromUtf8("/opt/Despeckle_pyqt/icons/guiRun.png")),
            QtGui.QIcon.Normal, QtGui.QIcon.On)
        self.action_Run.setIcon(icon1)
        self.action_Run.setObjectName(_fromUtf8("action_Run"))
        self.action_Run.triggered.connect(self.action_Run_Clicked)

        self.actionStop = QtGui.QAction(MainWindow)
        icon2 = QtGui.QIcon()
        icon2.addPixmap(
            QtGui.QPixmap(_fromUtf8("/opt/Despeckle_pyqt/icons/guiStop.png")),
            QtGui.QIcon.Normal, QtGui.QIcon.On)
        self.actionStop.setIcon(icon2)
        self.actionStop.setObjectName(_fromUtf8("actionStop"))
        self.actionStop.triggered.connect(self.action_Stop_Clicked)

        self.action_Exit = QtGui.QAction(MainWindow)

        icon3 = QtGui.QIcon()
        icon3.addPixmap(
            QtGui.QPixmap(
                _fromUtf8("/opt/Despeckle_pyqt/icons/fileclose.png")),
            QtGui.QIcon.Normal, QtGui.QIcon.On)

        self.action_Exit.setIcon(icon3)
        self.action_Exit.setObjectName(_fromUtf8("action_Exit"))
        self.action_Exit.triggered.connect(QtGui.qApp.quit)

        self.toolBar.addAction(self.action_Run)
        self.toolBar.addAction(self.actionStop)
        self.toolBar.addAction(self.action_Exit)
        self.toolBar.addSeparator()

        self.action_About_Despeckle = QtGui.QAction(MainWindow)
        self.action_About_Despeckle.setObjectName(
            _fromUtf8("action_About_Despeckle"))

        self.menu_File.addAction(self.action_Run)
        self.menu_File.addAction(self.actionStop)
        self.menu_File.addAction(self.action_Exit)
        self.menu_File.addSeparator()
        self.menu_Help.addAction(self.action_About_Despeckle)

        self.menubar.addAction(self.menu_File.menuAction())
        self.menubar.addAction(self.menu_Help.menuAction())

        self.retranslateUi(MainWindow)

        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):

        MainWindow.setWindowTitle(_translate("Despeckle", "Despeckle", None))

        self.label.setText(_translate("MainWindow", "ERROR", None))
        self.label_2.setText(_translate("MainWindow", "FROM", None))
        self.label_3.setText(_translate("MainWindow", "TO", None))
        self.label_4.setText(_translate("MainWindow", "KERNEL", None))
        self.label_5.setText(_translate("MainWindow", "ITERATIONS", None))
        self.label_6.setText(_translate("MainWindow", "L value", None))
        self.label_7.setText(_translate("MainWindow", "Console Output", None))
        self.label_8.setText(
            _translate("MainWindow", "Input image statistics", None))
        self.label_9.setText(
            _translate("MainWindow", "Output image statistics", None))

        self.menu_File.setTitle(_translate("MainWindow", "&File", None))
        self.menu_Help.setTitle(_translate("MainWindow", "&Help", None))

        self.toolBar.setWindowTitle(_translate("MainWindow", "toolBar", None))

        self.action_Run.setText(_translate("MainWindow", "&Run", None))
        self.actionStop.setText(_translate("MainWindow", "Stop", None))
        self.action_Exit.setText(_translate("MainWindow", "&Exit", None))
        self.action_About_Despeckle.setText(
            _translate("MainWindow", "&About Despeckle", None))

    def inputButtonClicked(self):
        """
        When button pushed, select input .cub file
        and set the path to textedit field and prevent 
        user from changing path manualy
        """
        filename = QtGui.QFileDialog.getOpenFileName(None, 'Select file',
                                                     "*.cub", "*.cub")
        self.textEdit.setText(filename)
        self.textEdit.setEnabled(0)

    def outputButtonClicked(self):
        """
        When button pushed, select output .cub file
        and set the path to textedit field and prevent 
        user from changing path manualy
        """
        filename = QtGui.QFileDialog.getSaveFileName(None, "Select file",
                                                     ".cub", ".cub")

        self.textEdit_2.setText(filename)
        self.textEdit_2.setEnabled(0)

    def inputKernelButtonClicked(self):
        """
        When button pushed, select kernel.txt file
        and set the path to textedit field and prevent 
        user from changing path manualy
        """
        filename = QtGui.QFileDialog.getOpenFileName(None, 'Select file',
                                                     "*.txt", "*.txt")
        self.textEdit_3.setText(filename)
        self.textEdit_3.setEnabled(0)

    def action_Run_Clicked(self):
        """
        Main function of the program. 
        Gets the paths, sets default values
        when new values aren't given,sets
        the command to the thread object (execute)
        and starts the thread.
        Also sets the output of the bottom label
        and the progress bar
        """

        input_cub_path = self.textEdit.toPlainText()
        output_cub_path = self.textEdit_2.toPlainText()
        input_kernel_path = self.textEdit_3.toPlainText()
        l_value = self.textEdit_5.toPlainText()
        iterations = self.textEdit_4.toPlainText()

        if (not l_value):
            l_value = '-l 0.08'
        else:
            l_value = '-l ' + l_value

        if (not iterations):
            iterations = '-i 100'
            self.maxIterations = 100
        else:
            self.maxIterations = int(iterations)
            iterations = '-i ' + iterations

        # Set max number to progress bar --> number of total iterations, 100 by default
        self.progressBar.setMaximum(self.maxIterations)

        if (not input_cub_path or not output_cub_path
                or not input_kernel_path):
            print("Error")
        else:
            self.cmd = "sh /opt/Despeckle_pyqt/Despeckle/a_despeckle.sh " + input_cub_path + ' ' + input_kernel_path + ' ' + ' ' + output_cub_path + ' ' + l_value + ' ' + iterations

            self.execute.setCmd(self.cmd)

            self.execute.start()

            print("is finished", self.execute.isFinished())

    def finished(self, completed):
        if (self.execute.isRunning()):
            return
        if (completed):
            print("Thread finished")
            self.label.setText("Finished")
        else:
            print("stopped")
            self.label.setText("Cancelled")

    def setOutput(self, mystr):
        """
        Redirects stdout given from signal
        to plainTextEdit and send given 
        string to bar and label
        """
        self.plainTextEdit.appendPlainText(mystr)
        self.setOutputBar(mystr)
        self.setOutputLabel(mystr)

    def setOutputBar(self, mystr):
        """
        In bash script is echoed
        the word "Iteration" following
        a number showing us the current
        iteration, so when our line starts
        with the word "Iteration" we take 
        the string after that word (the number),
        we cast it to integer and we set
        the value of the progress bar 
        """
        if (mystr.startswith("Iteration")):
            currentIteration = int(mystr[9:])
            self.progressBar.setValue(currentIteration)

    def setOutputLabel(self, mystr):
        """
        In bash script is echoed
        the word "Iteration" following
        a number showing us the current
        iteration, so when our line starts
        with the word "Iteration" we take 
        the whole line (string) and we
        set the value of the label  
        """
        if (mystr.startswith("Iteration")):
            self.label.setText(mystr)

    def action_Stop_Clicked(self):
        """
        A not-an-elegant
        way to stop the thread 
        """
        self.execute.stop()

    def statsButton(self):
        """
        When button pushed, a stats
        frame pops up with the stats
        of the input image.
        """
        path = self.textEdit.toPlainText()
        if (not path):
            pass
        else:
            self.stats1 = StatsPopup(path)
            self.stats1.start()

    def statsButton2(self):
        """
        When button pushed, a stats
        frame pops up with the stats
        of the output image.
        """
        path = self.textEdit_2.toPlainText()
        if (not path):
            pass
        else:
            self.stats2 = StatsPopup(path)
            self.stats2.start()

    def aboutClicked(self):
        """
        When button pushed, pops up
        a frame with a text read from
        a file
        """
        text = []
        file = open("/opt/Despeckle_pyqt/about.txt", 'r')
        for line in file:
            text.append(line)

        self.about = MyPopup("about")
        self.about.setText("".join(text))