USBPrinterOutputDevice.py 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432
  1. # Copyright (c) 2020 Ultimaker B.V.
  2. # Cura is released under the terms of the LGPLv3 or higher.
  3. import os
  4. from UM.i18n import i18nCatalog
  5. from UM.Logger import Logger
  6. from UM.Mesh.MeshWriter import MeshWriter #To get the g-code output.
  7. from UM.Message import Message #Show an error when already printing.
  8. from UM.PluginRegistry import PluginRegistry #To get the g-code output.
  9. from UM.Qt.Duration import DurationFormat
  10. from cura.CuraApplication import CuraApplication
  11. from cura.PrinterOutput.PrinterOutputDevice import PrinterOutputDevice, ConnectionState, ConnectionType
  12. from cura.PrinterOutput.Models.PrinterOutputModel import PrinterOutputModel
  13. from cura.PrinterOutput.Models.PrintJobOutputModel import PrintJobOutputModel
  14. from cura.PrinterOutput.GenericOutputController import GenericOutputController
  15. from .AutoDetectBaudJob import AutoDetectBaudJob
  16. from .AvrFirmwareUpdater import AvrFirmwareUpdater
  17. from io import StringIO #To write the g-code output.
  18. from queue import Queue
  19. from serial import Serial, SerialException, SerialTimeoutException
  20. from threading import Thread, Event
  21. from time import time
  22. from typing import Union, Optional, List, cast, TYPE_CHECKING
  23. import re
  24. import functools # Used for reduce
  25. if TYPE_CHECKING:
  26. from UM.FileHandler.FileHandler import FileHandler
  27. from UM.Scene.SceneNode import SceneNode
  28. catalog = i18nCatalog("cura")
  29. class USBPrinterOutputDevice(PrinterOutputDevice):
  30. def __init__(self, serial_port: str, baud_rate: Optional[int] = None) -> None:
  31. super().__init__(serial_port, connection_type = ConnectionType.UsbConnection)
  32. self.setName(catalog.i18nc("@item:inmenu", "USB printing"))
  33. self.setShortDescription(catalog.i18nc("@action:button Preceded by 'Ready to'.", "Print via USB"))
  34. self.setDescription(catalog.i18nc("@info:tooltip", "Print via USB"))
  35. self.setIconName("print")
  36. self._serial = None # type: Optional[Serial]
  37. self._serial_port = serial_port
  38. self._address = serial_port
  39. self._timeout = 3
  40. # List of gcode lines to be printed
  41. self._gcode = [] # type: List[str]
  42. self._gcode_position = 0
  43. self._use_auto_detect = True
  44. self._baud_rate = baud_rate
  45. self._all_baud_rates = [115200, 250000, 500000, 230400, 76800, 57600, 38400, 19200, 9600]
  46. # Instead of using a timer, we really need the update to be as a thread, as reading from serial can block.
  47. self._update_thread = Thread(target = self._update, daemon = True, name = "USBPrinterUpdate")
  48. self._last_temperature_request = None # type: Optional[int]
  49. self._firmware_idle_count = 0
  50. self._is_printing = False # A print is being sent.
  51. ## Set when print is started in order to check running time.
  52. self._print_start_time = None # type: Optional[float]
  53. self._print_estimated_time = None # type: Optional[int]
  54. self._accepts_commands = True
  55. self._paused = False
  56. self._printer_busy = False # When printer is preheating and waiting (M190/M109), or when waiting for action on the printer
  57. self.setConnectionText(catalog.i18nc("@info:status", "Connected via USB"))
  58. # Queue for commands that need to be sent.
  59. self._command_queue = Queue() # type: Queue
  60. # Event to indicate that an "ok" was received from the printer after sending a command.
  61. self._command_received = Event()
  62. self._command_received.set()
  63. self._firmware_name_requested = False
  64. self._firmware_updater = AvrFirmwareUpdater(self)
  65. plugin_path = PluginRegistry.getInstance().getPluginPath("USBPrinting")
  66. if plugin_path:
  67. self._monitor_view_qml_path = os.path.join(plugin_path, "MonitorItem.qml")
  68. else:
  69. Logger.log("e", "Cannot create Monitor QML view: cannot find plugin path for plugin [USBPrinting]")
  70. self._monitor_view_qml_path = ""
  71. CuraApplication.getInstance().getOnExitCallbackManager().addCallback(self._checkActivePrintingUponAppExit)
  72. # This is a callback function that checks if there is any printing in progress via USB when the application tries
  73. # to exit. If so, it will show a confirmation before
  74. def _checkActivePrintingUponAppExit(self) -> None:
  75. application = CuraApplication.getInstance()
  76. if not self._is_printing:
  77. # This USB printer is not printing, so we have nothing to do. Call the next callback if exists.
  78. application.triggerNextExitCheck()
  79. return
  80. application.setConfirmExitDialogCallback(self._onConfirmExitDialogResult)
  81. application.showConfirmExitDialog.emit(catalog.i18nc("@label", "A USB print is in progress, closing Cura will stop this print. Are you sure?"))
  82. def _onConfirmExitDialogResult(self, result: bool) -> None:
  83. if result:
  84. application = CuraApplication.getInstance()
  85. application.triggerNextExitCheck()
  86. def resetDeviceSettings(self) -> None:
  87. """Reset USB device settings"""
  88. self._firmware_name = None
  89. def requestWrite(self, nodes: List["SceneNode"], file_name: Optional[str] = None, limit_mimetypes: bool = False,
  90. file_handler: Optional["FileHandler"] = None, filter_by_machine: bool = False, **kwargs) -> None:
  91. """Request the current scene to be sent to a USB-connected printer.
  92. :param nodes: A collection of scene nodes to send. This is ignored.
  93. :param file_name: A suggestion for a file name to write.
  94. :param filter_by_machine: Whether to filter MIME types by machine. This
  95. is ignored.
  96. :param kwargs: Keyword arguments.
  97. """
  98. if self._is_printing:
  99. message = Message(text = catalog.i18nc("@message", "A print is still in progress. Cura cannot start another print via USB until the previous print has completed."), title = catalog.i18nc("@message", "Print in Progress"))
  100. message.show()
  101. return # Already printing
  102. self.writeStarted.emit(self)
  103. # cancel any ongoing preheat timer before starting a print
  104. controller = cast(GenericOutputController, self._printers[0].getController())
  105. controller.stopPreheatTimers()
  106. CuraApplication.getInstance().getController().setActiveStage("MonitorStage")
  107. #Find the g-code to print.
  108. gcode_textio = StringIO()
  109. gcode_writer = cast(MeshWriter, PluginRegistry.getInstance().getPluginObject("GCodeWriter"))
  110. success = gcode_writer.write(gcode_textio, None)
  111. if not success:
  112. return
  113. self._printGCode(gcode_textio.getvalue())
  114. def _printGCode(self, gcode: str):
  115. """Start a print based on a g-code.
  116. :param gcode: The g-code to print.
  117. """
  118. self._gcode.clear()
  119. self._paused = False
  120. self._gcode.extend(gcode.split("\n"))
  121. # Reset line number. If this is not done, first line is sometimes ignored
  122. self._gcode.insert(0, "M110")
  123. self._gcode_position = 0
  124. self._print_start_time = time()
  125. self._print_estimated_time = int(CuraApplication.getInstance().getPrintInformation().currentPrintTime.getDisplayString(DurationFormat.Format.Seconds))
  126. for i in range(0, 4): # Push first 4 entries before accepting other inputs
  127. self._sendNextGcodeLine()
  128. self._is_printing = True
  129. self.writeFinished.emit(self)
  130. def _autoDetectFinished(self, job: AutoDetectBaudJob):
  131. result = job.getResult()
  132. if result is not None:
  133. self.setBaudRate(result)
  134. self.connect() # Try to connect (actually create serial, etc)
  135. def setBaudRate(self, baud_rate: int):
  136. if baud_rate not in self._all_baud_rates:
  137. Logger.log("w", "Not updating baudrate to {baud_rate} as it's an unknown baudrate".format(baud_rate=baud_rate))
  138. return
  139. self._baud_rate = baud_rate
  140. def connect(self):
  141. self._firmware_name = None # after each connection ensure that the firmware name is removed
  142. if self._baud_rate is None:
  143. if self._use_auto_detect:
  144. auto_detect_job = AutoDetectBaudJob(self._serial_port)
  145. auto_detect_job.start()
  146. auto_detect_job.finished.connect(self._autoDetectFinished)
  147. return
  148. if self._serial is None:
  149. try:
  150. self._serial = Serial(str(self._serial_port), self._baud_rate, timeout=self._timeout, writeTimeout=self._timeout)
  151. except SerialException:
  152. Logger.warning("An exception occurred while trying to create serial connection.")
  153. return
  154. except OSError as e:
  155. Logger.warning("The serial device is suddenly unavailable while trying to create a serial connection: {err}".format(err = str(e)))
  156. return
  157. CuraApplication.getInstance().globalContainerStackChanged.connect(self._onGlobalContainerStackChanged)
  158. self._onGlobalContainerStackChanged()
  159. self.setConnectionState(ConnectionState.Connected)
  160. self._update_thread.start()
  161. def _onGlobalContainerStackChanged(self):
  162. container_stack = CuraApplication.getInstance().getGlobalContainerStack()
  163. if container_stack is None:
  164. return
  165. num_extruders = container_stack.getProperty("machine_extruder_count", "value")
  166. # Ensure that a printer is created.
  167. controller = GenericOutputController(self)
  168. controller.setCanUpdateFirmware(True)
  169. self._printers = [PrinterOutputModel(output_controller = controller, number_of_extruders = num_extruders)]
  170. self._printers[0].updateName(container_stack.getName())
  171. def close(self):
  172. super().close()
  173. if self._serial is not None:
  174. self._serial.close()
  175. # Re-create the thread so it can be started again later.
  176. self._update_thread = Thread(target=self._update, daemon=True, name = "USBPrinterUpdate")
  177. self._serial = None
  178. def sendCommand(self, command: Union[str, bytes]):
  179. """Send a command to printer."""
  180. if not self._command_received.is_set():
  181. self._command_queue.put(command)
  182. else:
  183. self._sendCommand(command)
  184. def _sendCommand(self, command: Union[str, bytes]):
  185. if self._serial is None or self._connection_state != ConnectionState.Connected:
  186. return
  187. new_command = cast(bytes, command) if type(command) is bytes else cast(str, command).encode() # type: bytes
  188. if not new_command.endswith(b"\n"):
  189. new_command += b"\n"
  190. try:
  191. self._command_received.clear()
  192. self._serial.write(new_command)
  193. except SerialTimeoutException:
  194. Logger.log("w", "Timeout when sending command to printer via USB.")
  195. self._command_received.set()
  196. except SerialException:
  197. Logger.logException("w", "An unexpected exception occurred while writing to the serial.")
  198. self.setConnectionState(ConnectionState.Error)
  199. def _update(self):
  200. while self._connection_state == ConnectionState.Connected and self._serial is not None:
  201. try:
  202. line = self._serial.readline()
  203. except:
  204. continue
  205. if not self._firmware_name_requested:
  206. self._firmware_name_requested = True
  207. self.sendCommand("M115")
  208. if b"FIRMWARE_NAME:" in line:
  209. self._setFirmwareName(line)
  210. if self._last_temperature_request is None or time() > self._last_temperature_request + self._timeout:
  211. # Timeout, or no request has been sent at all.
  212. if not self._printer_busy: # Don't flood the printer with temperature requests while it is busy
  213. self.sendCommand("M105")
  214. self._last_temperature_request = time()
  215. if re.search(b"[B|T\d*]: ?\d+\.?\d*", line): # Temperature message. 'T:' for extruder and 'B:' for bed
  216. extruder_temperature_matches = re.findall(b"T(\d*): ?(\d+\.?\d*)\s*\/?(\d+\.?\d*)?", line)
  217. # Update all temperature values
  218. matched_extruder_nrs = []
  219. for match in extruder_temperature_matches:
  220. extruder_nr = 0
  221. if match[0] != b"":
  222. extruder_nr = int(match[0])
  223. if extruder_nr in matched_extruder_nrs:
  224. continue
  225. matched_extruder_nrs.append(extruder_nr)
  226. if extruder_nr >= len(self._printers[0].extruders):
  227. Logger.log("w", "Printer reports more temperatures than the number of configured extruders")
  228. continue
  229. extruder = self._printers[0].extruders[extruder_nr]
  230. if match[1]:
  231. extruder.updateHotendTemperature(float(match[1]))
  232. if match[2]:
  233. extruder.updateTargetHotendTemperature(float(match[2]))
  234. bed_temperature_matches = re.findall(b"B: ?(\d+\.?\d*)\s*\/?(\d+\.?\d*)?", line)
  235. if bed_temperature_matches:
  236. match = bed_temperature_matches[0]
  237. if match[0]:
  238. self._printers[0].updateBedTemperature(float(match[0]))
  239. if match[1]:
  240. self._printers[0].updateTargetBedTemperature(float(match[1]))
  241. if line == b"":
  242. # An empty line means that the firmware is idle
  243. # Multiple empty lines probably means that the firmware and Cura are waiting
  244. # for eachother due to a missed "ok", so we keep track of empty lines
  245. self._firmware_idle_count += 1
  246. else:
  247. self._firmware_idle_count = 0
  248. if line.startswith(b"ok") or self._firmware_idle_count > 1:
  249. self._printer_busy = False
  250. self._command_received.set()
  251. if not self._command_queue.empty():
  252. self._sendCommand(self._command_queue.get())
  253. elif self._is_printing:
  254. if self._paused:
  255. pass # Nothing to do!
  256. else:
  257. self._sendNextGcodeLine()
  258. if line.startswith(b"echo:busy:"):
  259. self._printer_busy = True
  260. if self._is_printing:
  261. if line.startswith(b'!!'):
  262. Logger.log('e', "Printer signals fatal error. Cancelling print. {}".format(line))
  263. self.cancelPrint()
  264. elif line.lower().startswith(b"resend") or line.startswith(b"rs"):
  265. # A resend can be requested either by Resend, resend or rs.
  266. try:
  267. self._gcode_position = int(line.replace(b"N:", b" ").replace(b"N", b" ").replace(b":", b" ").split()[-1])
  268. except:
  269. if line.startswith(b"rs"):
  270. # In some cases of the RS command it needs to be handled differently.
  271. self._gcode_position = int(line.split()[1])
  272. def _setFirmwareName(self, name):
  273. new_name = re.findall(r"FIRMWARE_NAME:(.*);", str(name))
  274. if new_name:
  275. self._firmware_name = new_name[0]
  276. Logger.log("i", "USB output device Firmware name: %s", self._firmware_name)
  277. else:
  278. self._firmware_name = "Unknown"
  279. Logger.log("i", "Unknown USB output device Firmware name: %s", str(name))
  280. def getFirmwareName(self):
  281. return self._firmware_name
  282. def pausePrint(self):
  283. self._paused = True
  284. def resumePrint(self):
  285. self._paused = False
  286. self._sendNextGcodeLine() #Send one line of g-code next so that we'll trigger an "ok" response loop even if we're not polling temperatures.
  287. def cancelPrint(self):
  288. self._gcode_position = 0
  289. self._gcode.clear()
  290. self._printers[0].updateActivePrintJob(None)
  291. self._is_printing = False
  292. self._paused = False
  293. # Turn off temperatures, fan and steppers
  294. self._sendCommand("M140 S0")
  295. self._sendCommand("M104 S0")
  296. self._sendCommand("M107")
  297. # Home XY to prevent nozzle resting on aborted print
  298. # Don't home bed because it may crash the printhead into the print on printers that home on the bottom
  299. self.printers[0].homeHead()
  300. self._sendCommand("M84")
  301. def _sendNextGcodeLine(self):
  302. """
  303. Send the next line of g-code, at the current `_gcode_position`, via a
  304. serial port to the printer.
  305. If the print is done, this sets `_is_printing` to `False` as well.
  306. """
  307. try:
  308. line = self._gcode[self._gcode_position]
  309. except IndexError: # End of print, or print got cancelled.
  310. self._printers[0].updateActivePrintJob(None)
  311. self._is_printing = False
  312. return
  313. if ";" in line:
  314. line = line[:line.find(";")]
  315. line = line.strip()
  316. # Don't send empty lines. But we do have to send something, so send M105 instead.
  317. # Don't send the M0 or M1 to the machine, as M0 and M1 are handled as an LCD menu pause.
  318. if line == "" or line == "M0" or line == "M1":
  319. line = "M105"
  320. checksum = functools.reduce(lambda x, y: x ^ y, map(ord, "N%d%s" % (self._gcode_position, line)))
  321. self._sendCommand("N%d%s*%d" % (self._gcode_position, line, checksum))
  322. print_job = self._printers[0].activePrintJob
  323. try:
  324. progress = self._gcode_position / len(self._gcode)
  325. except ZeroDivisionError:
  326. # There is nothing to send!
  327. if print_job is not None:
  328. print_job.updateState("error")
  329. return
  330. elapsed_time = int(time() - self._print_start_time)
  331. if print_job is None:
  332. controller = GenericOutputController(self)
  333. controller.setCanUpdateFirmware(True)
  334. print_job = PrintJobOutputModel(output_controller = controller, name = CuraApplication.getInstance().getPrintInformation().jobName)
  335. print_job.updateState("printing")
  336. self._printers[0].updateActivePrintJob(print_job)
  337. print_job.updateTimeElapsed(elapsed_time)
  338. estimated_time = self._print_estimated_time
  339. if progress > .1:
  340. estimated_time = self._print_estimated_time * (1 - progress) + elapsed_time
  341. print_job.updateTimeTotal(estimated_time)
  342. self._gcode_position += 1