123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778 |
- # Copyright (c) 2018 Ultimaker B.V.
- # Cura is released under the terms of the LGPLv3 or higher.
- from PyQt5.QtCore import QObject, QUrl, pyqtSignal, pyqtProperty
- from enum import IntEnum
- from threading import Thread
- from typing import Union
- MYPY = False
- if MYPY:
- from cura.PrinterOutput.PrinterOutputDevice import PrinterOutputDevice
- class FirmwareUpdater(QObject):
- firmwareProgressChanged = pyqtSignal()
- firmwareUpdateStateChanged = pyqtSignal()
- def __init__(self, output_device: "PrinterOutputDevice") -> None:
- super().__init__()
- self._output_device = output_device
- self._update_firmware_thread = Thread(target=self._updateFirmware, daemon=True)
- self._firmware_file = ""
- self._firmware_progress = 0
- self._firmware_update_state = FirmwareUpdateState.idle
- def updateFirmware(self, firmware_file: Union[str, QUrl]) -> None:
- # the file path could be url-encoded.
- if firmware_file.startswith("file://"):
- self._firmware_file = QUrl(firmware_file).toLocalFile()
- else:
- self._firmware_file = firmware_file
- self._setFirmwareUpdateState(FirmwareUpdateState.updating)
- self._update_firmware_thread.start()
- def _updateFirmware(self) -> None:
- raise NotImplementedError("_updateFirmware needs to be implemented")
- ## Cleanup after a succesful update
- def _cleanupAfterUpdate(self) -> None:
- # Clean up for next attempt.
- self._update_firmware_thread = Thread(target=self._updateFirmware, daemon=True)
- self._firmware_file = ""
- self._onFirmwareProgress(100)
- self._setFirmwareUpdateState(FirmwareUpdateState.completed)
- @pyqtProperty(int, notify = firmwareProgressChanged)
- def firmwareProgress(self) -> int:
- return self._firmware_progress
- @pyqtProperty(int, notify=firmwareUpdateStateChanged)
- def firmwareUpdateState(self) -> "FirmwareUpdateState":
- return self._firmware_update_state
- def _setFirmwareUpdateState(self, state: "FirmwareUpdateState") -> None:
- if self._firmware_update_state != state:
- self._firmware_update_state = state
- self.firmwareUpdateStateChanged.emit()
- # Callback function for firmware update progress.
- def _onFirmwareProgress(self, progress: int, max_progress: int = 100) -> None:
- self._firmware_progress = int(progress * 100 / max_progress) # Convert to scale of 0-100
- self.firmwareProgressChanged.emit()
- class FirmwareUpdateState(IntEnum):
- idle = 0
- updating = 1
- completed = 2
- unknown_error = 3
- communication_error = 4
- io_error = 5
- firmware_not_found_error = 6
|