mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-03 09:14:24 +08:00
119 lines
3.5 KiB
Python
119 lines
3.5 KiB
Python
# Copyright (c) 2017 Ultimaker B.V.
|
|
# Uranium is released under the terms of the AGPLv3 or higher.
|
|
|
|
from threading import Thread, Event
|
|
|
|
from PyQt5.QtCore import Qt
|
|
from PyQt5.QtGui import QPixmap, QColor, QFont, QPen, QPainter
|
|
from PyQt5.QtWidgets import QSplashScreen
|
|
|
|
from UM.Resources import Resources
|
|
from UM.Application import Application
|
|
|
|
|
|
class CuraSplashScreen(QSplashScreen):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._scale = 0.7
|
|
|
|
splash_image = QPixmap(Resources.getPath(Resources.Images, "cura.png"))
|
|
self.setPixmap(splash_image)
|
|
|
|
self._current_message = ""
|
|
|
|
self._loading_image_rotation_angle = 0
|
|
|
|
self._to_stop = False
|
|
self._loading_tick_thread = LoadingTickThread(self)
|
|
|
|
def show(self):
|
|
super().show()
|
|
self._loading_tick_thread.start()
|
|
|
|
def updateLoadingImage(self):
|
|
if self._to_stop:
|
|
return
|
|
|
|
self._loading_image_rotation_angle -= 10
|
|
self.repaint()
|
|
|
|
def drawContents(self, painter):
|
|
if self._to_stop:
|
|
return
|
|
|
|
painter.save()
|
|
painter.setPen(QColor(255, 255, 255, 255))
|
|
painter.setRenderHint(QPainter.Antialiasing)
|
|
painter.setRenderHint(QPainter.Antialiasing, True)
|
|
|
|
version = Application.getInstance().getVersion().split("-")
|
|
buildtype = Application.getInstance().getBuildType()
|
|
if buildtype:
|
|
version[0] += " (%s)" % buildtype
|
|
|
|
# draw version text
|
|
font = QFont() # Using system-default font here
|
|
font.setPointSize(34)
|
|
painter.setFont(font)
|
|
painter.drawText(300, 110, 330 * self._scale, 230 * self._scale, Qt.AlignLeft | Qt.AlignTop, version[0])
|
|
if len(version) > 1:
|
|
font.setPointSize(12)
|
|
painter.setFont(font)
|
|
painter.setPen(QColor(200, 200, 200, 255))
|
|
painter.drawText(343, 160, 330 * self._scale, 255 * self._scale, Qt.AlignLeft | Qt.AlignTop, version[1])
|
|
painter.setPen(QColor(255, 255, 255, 255))
|
|
|
|
# draw the loading image
|
|
pen = QPen()
|
|
pen.setWidth(6 * self._scale)
|
|
pen.setColor(QColor(255, 255, 255, 255))
|
|
painter.setPen(pen)
|
|
painter.drawArc(60, 350, 32 * self._scale, 32 * self._scale, self._loading_image_rotation_angle * 16, 300 * 16)
|
|
|
|
# draw message text
|
|
if self._current_message:
|
|
font = QFont() # Using system-default font here
|
|
font.setPointSize(16)
|
|
painter.setFont(font)
|
|
painter.drawText(100, 328, 260, 64,
|
|
Qt.AlignLeft | Qt.AlignVCenter | Qt.TextWordWrap,
|
|
self._current_message)
|
|
|
|
painter.restore()
|
|
super().drawContents(painter)
|
|
|
|
def showMessage(self, message, *args, **kwargs):
|
|
if self._to_stop:
|
|
return
|
|
|
|
self._current_message = message
|
|
self.messageChanged.emit(message)
|
|
|
|
def close(self):
|
|
# set stop flags
|
|
self._to_stop = True
|
|
self._loading_tick_thread.setToStop()
|
|
super().close()
|
|
|
|
|
|
class LoadingTickThread(Thread):
|
|
|
|
def __init__(self, splash):
|
|
super().__init__(daemon = True)
|
|
self._splash = splash
|
|
self._to_stop = False
|
|
self._time_interval = 0.05
|
|
self._event = Event()
|
|
|
|
def setToStop(self):
|
|
self._to_stop = True
|
|
self._event.set()
|
|
|
|
def run(self):
|
|
while not self._to_stop:
|
|
self._event.wait(self._time_interval)
|
|
if self._event.is_set():
|
|
break
|
|
|
|
self._splash.updateLoadingImage()
|