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))
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))
