class ProgressBar(QDialog):
def __init__(self,
parent=None,
max_items=100,
window_title='Progress Bar',
label='',
on_top=False):
if on_top:
QDialog.__init__(self,
parent=parent,
flags=Qt.WindowStaysOnTopHint)
else:
QDialog.__init__(self, parent=parent)
self.application = Application
self.setWindowTitle(window_title)
self.l = QVBoxLayout(self)
self.setLayout(self.l)
self.label = QLabel(label)
self.label.setAlignment(Qt.AlignHCenter)
self.l.addWidget(self.label)
self.progressBar = QProgressBar(self)
self.progressBar.setRange(0, max_items)
self.progressBar.setValue(0)
self.l.addWidget(self.progressBar)
def increment(self):
self.progressBar.setValue(self.progressBar.value() + 1)
self.refresh()
def refresh(self):
self.application.processEvents()
def set_label(self, value):
self.label.setText(value)
self.refresh()
def left_align_label(self):
self.label.setAlignment(Qt.AlignLeft)
def set_maximum(self, value):
self.progressBar.setMaximum(value)
self.refresh()
def set_value(self, value):
self.progressBar.setValue(value)
self.refresh()
def set_progress_format(self, progress_format=None):
pass
class ProgressBar(QDialog):
def __init__(self, parent=None, max_items=100, window_title='Progress Bar',
label='Label goes here', on_top=False):
if on_top:
QDialog.__init__(self, parent=parent, flags=Qt.WindowStaysOnTopHint)
else:
QDialog.__init__(self, parent=parent)
self.application = Application
self.setWindowTitle(window_title)
self.l = QVBoxLayout(self)
self.setLayout(self.l)
self.label = QLabel(label)
self.label.setAlignment(Qt.AlignHCenter)
self.l.addWidget(self.label)
self.progressBar = QProgressBar(self)
self.progressBar.setRange(0, max_items)
self.progressBar.setValue(0)
self.l.addWidget(self.progressBar)
def increment(self):
self.progressBar.setValue(self.progressBar.value() + 1)
self.refresh()
def refresh(self):
self.application.processEvents()
def set_label(self, value):
self.label.setText(value)
self.refresh()
def set_maximum(self, value):
self.progressBar.setMaximum(value)
self.refresh()
def set_value(self, value):
self.progressBar.setValue(value)
self.refresh()
class SplashScreen(QSplashScreen):
""" Display a Splashscreen """
signal_done = QtCore.pyqtSignal()
def __init__(self):
QSplashScreen.__init__(self, QtGui.QPixmap(),
QtCore.Qt.WindowStaysOnTopHint)
self.rootDir = Path(__file__).parent
# default values
self.image = None
self.version = "Version: 3.x"
# pixmap.height - ?
self.progress_y = 68
# pixmap.width - ?
self.vx = 24
# pixmap.height - ?
self.vy = 32
# message font size
self.msize = 12
self.cv = OpenCVLib()
self.progressBar = QProgressBar(self)
self.progressBar.setMinimum(0)
self.progressBar.setValue(0)
self.progressBar.setMaximum(100)
self.progressBar.setTextVisible(False)
self.setPositionProgressBar()
self.message = QLabel(self)
self.message.setFont(QFont("Arial", self.msize))
self.message.setStyleSheet("QLabel { color : #000000; }")
self.message.setAlignment(Qt.AlignCenter)
# Shadow Effekt
effect = QGraphicsDropShadowEffect(self)
effect.setBlurRadius(5)
effect.setColor(QColor("#ffffff"))
effect.setOffset(1, 1)
self.message.setGraphicsEffect(effect)
self.setPositionMessage()
# self.message.hide()
# CSS Styling
self.setStyleSheet("""
QProgressBar{
border: 1px solid #eeeeee;
text-align: center;
padding: 0;
margin-top: 10px;
}
QProgressBar::chunk{
background-color: #3daee9;
width: 0.34em;
margin: 0 1px;
}
""")
def show(self, *args, **kwargs):
if self.image == None:
raise ValueError('Specify an Image via SplashScreen::setImage()')
return QSplashScreen.show(self, *args, **kwargs)
def setMessage(self, msg):
self.message.setText("%s ..." % (msg))
self.message.show()
def setPositionMessage(self):
""" place Message on screen """
# where is progress?
p = self.progressBar.geometry()
self.message.setGeometry(0,
p.y() - self.msize,
self.pixmap().width(), 2 * self.msize)
self.message.updateGeometry()
def setPositionProgressBar(self):
""" place ProgressBar on screen """
margin = 10
# x, y, w, h
self.progressBar.setGeometry(margin,
self.pixmap().height() - self.progress_y,
self.pixmap().width() - 2 * margin, 20)
self.progressBar.updateGeometry()
def scale(self, pix):
gold = 0.618
h = pix.size().height()
w = pix.size().width()
# max width 68% of screen
screen = QGuiApplication.screens()[0]
new_w = screen.geometry().width() * gold
new_h = h * new_w / w
# python resize
# return pix.scaled(new_w, new_h, Qt.KeepAspectRatioByExpanding | Qt.SmoothTransformation)
# resize with opencv
Qimg = pix.toImage()
img = self.cv.QImage2MAT(Qimg)
resized = self.cv.resizeTo(img, new_w, new_h)
return self.cv.MAT2QPixmap(resized)
def setVersion(self, version):
""" adds a Version Number and updates the image """
self.version = "Version: %s" % version
def paintEvent(self, *args, **kwargs):
""" override, important to work """
return QSplashScreen.paintEvent(self, *args, **kwargs)
def setImage(self, img):
""" sets the image and adds a Version Number """
self.image = self.rootDir.joinpath(img).as_posix()
splash_pix = QtGui.QPixmap(self.image, format='jpg')
if splash_pix.isNull():
raise ValueError('Error loading Image [self.image]')
# Add version
painter = QtGui.QPainter(splash_pix)
painter.setRenderHint(QtGui.QPainter.Antialiasing)
painter.setFont(QFont("Arial", 18))
painter.setPen(QColor("#666666"))
painter.drawText(0, 0,
splash_pix.size().width() - self.vx,
splash_pix.size().height() - self.vy,
QtCore.Qt.AlignBottom | QtCore.Qt.AlignRight,
self.version)
painter.end()
self.setPixmap(self.scale(splash_pix))
# ----------------------------------------------
# other stuff
self.setPositionProgressBar()
self.setPositionMessage()
def step(self):
""" a preloading step is done """
self.progressBar.setValue(self.progressBar.value() + 1)
self.progressBar.repaint()
if (self.progressBar.value() >= (self.progressBar.maximum() - 1)):
self.signal_done.emit()
def setProgressMax(self, maxval):
self.progressBar.setMaximum(maxval)
class ProgressDialog(QDialog):
_default = '''
QProgressBar {
min-width: 350px;
}
'''
def __init__(self, parent, title, total_progress):
super().__init__(parent)
self.setWindowTitle(title)
self._total_progress = total_progress
self._progress_bar = None
self._setup_ui()
self._prepare()
def _setup_ui(self):
self.setStyleSheet(self._default)
self.setWindowFlags(
Qt.Window | Qt.WindowTitleHint | Qt.CustomizeWindowHint
)
self.setStyleSheet(self._default)
layout = make_layout(
horizon=False,
margin=24,
spacing=24
)
self._progress_bar = QProgressBar()
self._progress_bar.setRange(0, self._total_progress)
self._progress_bar.setTextVisible(True)
self._progress_bar.setFormat(r'%p% (%v/%m)')
self._progress_bar.setValue(0)
layout.addWidget(
self._progress_bar
)
self.setLayout(layout)
move_center(self)
def _prepare(self):
return
def increase(self, step=1):
self._progress_bar.setValue(self._progress_bar.value() + step)
if self._progress_bar.value() == self._progress_bar.maximum():
self.close()
def _on_show(self):
return
def _on_close(self):
return
def showEvent(self, event):
self._on_show()
event.accept()
def closeEvent(self, event):
self._on_close()
event.accept()
