diff --git a/progressWindow.py b/progressWindow.py deleted file mode 100644 index 304690e..0000000 --- a/progressWindow.py +++ /dev/null @@ -1,149 +0,0 @@ -# window for showing progress while decensoring -# spliting windows by main to check Censor Types, Variations, ect will be better for -# understanding flow of code - -import sys -import PySide2 -from PySide2.QtWidgets import (QApplication, QMainWindow, QDesktopWidget, QPushButton, - QToolTip, QLabel, QProgressBar, QAction, qApp) -# from PyQt5.QtCore import QThread -from signals import Signals -import threading -import time - -class ProgressWindow(QMainWindow): - # debug for setting UI - def __init__(self, MainWindow, decensor, debug = False): - super().__init__() - self.width = 700 - self.height = 500 - self.resize(self.width, self.height) - self.initUI() - - # signal class that could share update progress ui from decensor class (Decensor) - self.setSignals() - - self.center() - self.setWindowTitle("DeepCreamPy v2.2.0 Decensoring...") - self.show() - - if not debug: - print("not debug") - # decensor class initialized with options selected from MainWindow - self.decensor = decensor - self.decensor.signals = self.signals - - # to go back to MainWindow after finshed decensoring - self.mainWindow = MainWindow - - self.runDecensor() - - - def initUI(self): - ''' - Must Todo UI: - 1. add goto decensored file button - 2. Two progress bars - 2-1. total images to decenesor (images in ./decensor_input) - 2-2. current decensoring image's censored area (example marmaid image in DCPv2, 2 / 17) - 3. go back to main button - - Could Do UI: - 1. showing live image decensoring (decensored one by one) - ''' - # progress bar showing images left to be decensored - def setProgressBar(): - bar_X = 50 - bar_Y = 300 - bar_width = 600 - bar_height = 30 - - # images waiting to be decensored - self.total_images_ProgressBar = QProgressBar(self) - # setGeometry(left top x cordinate, left top y cordinate, width, height) - self.total_images_ProgressBar.setGeometry(bar_X, bar_Y, bar_width,bar_height ) - self.total_images_ProgressBar.setMaximum(100) - self.total_images_ProgressBar.setValue(0) - - # showing progress of decensored area - self.signal_image_decensor_ProgressBar = QProgressBar(self) - self.signal_image_decensor_ProgressBar.setGeometry(bar_X, bar_Y+80, bar_width,bar_height ) - self.signal_image_decensor_ProgressBar.setMaximum(100) - self.signal_image_decensor_ProgressBar.setValue(0) - - progress_Label_1 = QLabel(self) - progress_Label_1.move(50, 270) - progress_Label_1.setText("Decensor progress of all images") - progress_Label_1.resize(progress_Label_1.sizeHint()) - - progress_Label_2 = QLabel(self) - progress_Label_2.move(50, 300 + 50) - progress_Label_2.setText("Decensor progress of current image") - progress_Label_2.resize(progress_Label_2.sizeHint()) - - self.progress_status_LABEL = QLabel(self) - self.progress_status_LABEL.move(100, 100) - self.progress_status_LABEL.setText("Decensoring...") - self.progress_status_LABEL.resize(self.progress_status_LABEL.sizeHint()) - - setProgressBar() - - def center(self): - qr = self.frameGeometry() - cp = QDesktopWidget().availableGeometry().center() - qr.moveCenter(cp) - self.move(qr.topLeft()) - - def setSignals(self): - self.signals = Signals() - # set signal variable name same as method name preventing confusion - self.signals.total_ProgressBar_update_MAX_VALUE.connect(self.total_ProgressBar_update_MAX_VALUE) - self.signals.total_ProgressBar_update_VALUE.connect(self.total_ProgressBar_update_VALUE) - self.signals.signal_ProgressBar_update_MAX_VALUE.connect(self.signal_ProgressBar_update_MAX_VALUE) - self.signals.signal_ProgressBar_update_VALUE.connect(self.signal_ProgressBar_update_VALUE) - self.signals.update_progress_LABEL.connect(self.update_progress_LABEL) - - # total_images_to_decensor_ProgressBar - def total_ProgressBar_update_MAX_VALUE(self, msg, max): - # print msg for debugging - print(msg) - self.total_images_ProgressBar.setMaximum(max) - - def total_ProgressBar_update_VALUE(self, msg, val): - # print msg for debugging - print(msg) - self.total_images_ProgressBar.setValue(val) - - def signal_ProgressBar_update_MAX_VALUE(self, msg, max): - # print msg for debugging - print(msg) - self.signal_image_decensor_ProgressBar.setMaximum(max) - - def signal_ProgressBar_update_VALUE(self, msg, val): - # print msg for debugging - print(msg) - self.signal_image_decensor_ProgressBar.setValue(val) - - def update_progress_LABEL(self, msg, status): - print(msg) - self.progress_status_LABEL.setText(status) - self.progress_status_LABEL.resize(self.progress_status_LABEL.sizeHint()) - - def runDecensor(self): - # start decensor in other thread, preventing UI Freezing - # print("start run") - self.decensor.start() - -if __name__ == "__main__": - # only use for debuging window - - import os - # you could remove this if statement if there's no error without this - if os.name == 'nt': - import PySide2 - pyqt = os.path.dirname(PySide2.__file__) - QApplication.addLibraryPath(os.path.join(pyqt, "plugins")) - app = QApplication(sys.argv) - ex = ProgressWindow(1, 1, debug = True) - ex.show() - sys.exit( app.exec_() )