1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- # Copyright (c) 2018 Ultimaker B.V.
- # Cura is released under the terms of the LGPLv3 or higher.
- from UM.Logger import Logger
- from cura.CuraApplication import CuraApplication
- from cura.PrinterOutput.FirmwareUpdater import FirmwareUpdater, FirmwareUpdateState
- from .avr_isp import stk500v2, intelHex
- from serial import SerialException
- from time import sleep
- MYPY = False
- if MYPY:
- from cura.PrinterOutputDevice import PrinterOutputDevice
- class AvrFirmwareUpdater(FirmwareUpdater):
- def __init__(self, output_device: "PrinterOutputDevice") -> None:
- super().__init__(output_device)
- def _updateFirmware(self) -> None:
- try:
- hex_file = intelHex.readHex(self._firmware_file)
- assert len(hex_file) > 0
- except (FileNotFoundError, AssertionError):
- Logger.log("e", "Unable to read provided hex file. Could not update firmware.")
- self._setFirmwareUpdateState(FirmwareUpdateState.firmware_not_found_error)
- return
- programmer = stk500v2.Stk500v2()
- programmer.progress_callback = self._onFirmwareProgress
- # Ensure that other connections are closed.
- if self._output_device.isConnected():
- self._output_device.close()
- try:
- programmer.connect(self._output_device._serial_port)
- except:
- programmer.close()
- Logger.logException("e", "Failed to update firmware")
- self._setFirmwareUpdateState(FirmwareUpdateState.communication_error)
- return
- # Give programmer some time to connect. Might need more in some cases, but this worked in all tested cases.
- sleep(1)
- if not programmer.isConnected():
- Logger.log("e", "Unable to connect with serial. Could not update firmware")
- self._setFirmwareUpdateState(FirmwareUpdateState.communication_error)
- try:
- programmer.programChip(hex_file)
- except SerialException as e:
- Logger.log("e", "A serial port exception occured during firmware update: %s" % e)
- self._setFirmwareUpdateState(FirmwareUpdateState.io_error)
- return
- except Exception as e:
- Logger.log("e", "An unknown exception occured during firmware update: %s" % e)
- self._setFirmwareUpdateState(FirmwareUpdateState.unknown_error)
- return
- programmer.close()
- # Try to re-connect with the machine again, which must be done on the Qt thread, so we use call later.
- CuraApplication.getInstance().callLater(self._output_device.connect)
- self._cleanupAfterUpdate()
|